Synnefo » snf-ganeti » ganeti-local

from ganeti import compat

from ganeti.utils import text

def SequenceToDict(seq, key=compat.fst):

  """Converts a sequence to a dictionary with duplicate detection.

  @type seq: sequen

  @param seq: Input sequence

  @type key: callable

  @param key: Function for retrieving dictionary key from sequence element

  @rtype: dict

  """

  keys = map(key, seq)

  duplicates = FindDuplicates(keys)

  if duplicates:

    raise ValueError("Duplicate keys found: %s" % text.CommaJoin(duplicates))

  assert len(keys) == len(seq)

  return dict(zip(keys, seq))

from ganeti import compat

class TestSequenceToDict(unittest.TestCase):

  def testEmpty(self):

    self.assertEqual(algo.SequenceToDict([]), {})

    self.assertEqual(algo.SequenceToDict({}), {})

  def testSimple(self):

    data = [(i, str(i), "test%s" % i) for i in range(391)]

    self.assertEqual(algo.SequenceToDict(data),

      dict((i, (i, str(i), "test%s" % i))

           for i in range(391)))

  def testCustomKey(self):

    data = [(i, hex(i), "test%s" % i) for i in range(100)]

    self.assertEqual(algo.SequenceToDict(data, key=compat.snd),

      dict((hex(i), (i, hex(i), "test%s" % i))

           for i in range(100)))

    self.assertEqual(algo.SequenceToDict(data,

                                         key=lambda (a, b, val): hash(val)),

      dict((hash("test%s" % i), (i, hex(i), "test%s" % i))

           for i in range(100)))

  def testDuplicate(self):

    self.assertRaises(ValueError, algo.SequenceToDict,

                      [(0, 0), (0, 0)])

    self.assertRaises(ValueError, algo.SequenceToDict,

                      [(i, ) for i in range(200)] + [(10, )])