def test_remove_by_first_or_second(self):
'''
Test for removing triplets form the TwoKeyDict
'''
triplet_first = (1, 'I', 'one')
triplet_second = (2, 'II', 'two')
tw_dict = TwoKeyDict(triplet_first, triplet_second)
tw_dict._remove_by_primary(triplet_first[0])
tw_dict._remove_by_secondary(triplet_second[1])
missing_first = 0
missing_second = 0
try:
tw_dict._get_by_primary(triplet_first[0])
except KeyError:
missing_first += 1
try:
tw_dict._get_by_secondary(triplet_second[0])
except KeyError:
missing_first += 1
try:
tw_dict._get_by_secondary(triplet_first[1])
except KeyError:
missing_second += 1
try:
tw_dict._get_by_secondary(triplet_second[1])
except KeyError:
missing_second += 1
self.assertEqual(missing_first, 2)
self.assertEqual(missing_second, 2)
#check for memory leaks
dict_len = 0
for key in tw_dict._primary_to_value.iterkeys():
dict_len += 1
self.assertEqual(dict_len, 0)
def test_remove_by_first_or_second(self):
""" Test for removing triplets form the TwoKeyDict """
triplet_first = (1, 'I', 'one')
triplet_second = (2, 'II', 'two')
tw_dict = TwoKeyDict(triplet_first, triplet_second)
tw_dict._remove_by_primary(triplet_first[0])
tw_dict._remove_by_secondary(triplet_second[1])
missing_first = 0
missing_second = 0
try:
tw_dict._get_by_primary(triplet_first[0])
except KeyError:
missing_first += 1
try:
tw_dict._get_by_secondary(triplet_second[0])
except KeyError:
missing_first += 1
try:
tw_dict._get_by_secondary(triplet_first[1])
except KeyError:
missing_second += 1
try:
tw_dict._get_by_secondary(triplet_second[1])
except KeyError:
missing_second += 1
self.assertEqual(missing_first, 2)
self.assertEqual(missing_second, 2)
# check for memory leaks
dict_len = 0
for key in tw_dict._primary_to_value.iterkeys():
dict_len += 1
self.assertEqual(dict_len, 0)
def test_add_and_gets(self):
""" Test for the __init__, _get_by_first, _get_by_second function """
triplets = [(uuid.uuid4(), uuid.uuid4(), uuid.uuid4())
for a in range(10)]
tw_dict = TwoKeyDict(*triplets)
for triplet in triplets:
self.assertEqual(tw_dict._get_by_primary(triplet[0]), triplet[2])
self.assertEqual(tw_dict._get_by_secondary(triplet[1]), triplet[2])
def test_add_and_gets(self):
""" Test for the __init__, _get_by_first, _get_by_second function """
triplets = [(uuid.uuid4(), uuid.uuid4(), uuid.uuid4())
for a in xrange(10)]
tw_dict = TwoKeyDict(*triplets)
for triplet in triplets:
self.assertEqual(tw_dict._get_by_primary(triplet[0]), triplet[2])
self.assertEqual(tw_dict._get_by_secondary(triplet[1]), triplet[2])
