def test_uuids(self):
expected_map = {
0: 'a',
75: 'b',
124769: 'c',
}
dms = DataMemorySet()
dms._element_map = expected_map
self.assertEqual(dms.uuids(), {0, 75, 124769})
def test_count(self):
expected_map = {
0: 'a',
75: 'b',
124769: 'c',
}
dms = DataMemorySet()
dms._element_map = expected_map
self.assertEqual(dms.count(), 3)
def test_iter(self):
expected_map = {
0: 'a',
75: 'b',
124769: 'c',
}
expected_map_values = {'a', 'b', 'c'}
dms = DataMemorySet()
dms._element_map = expected_map
self.assertEqual(set(dms), expected_map_values)
self.assertEqual(set(iter(dms)), expected_map_values)
def test_get_data_valid_uuid(self):
expected_map = {
0: 'a',
75: 'b',
124769: 'c',
}
dms = DataMemorySet()
dms._element_map = expected_map
self.assertEqual(dms.get_data(0), 'a')
self.assertEqual(dms.get_data(75), 'b')
self.assertEqual(dms.get_data(124769), 'c')
def test_has_uuid(self):
expected_map = {
0: 'a',
75: 'b',
124769: 'c',
}
dms = DataMemorySet()
dms._element_map = expected_map
self.assertTrue(dms.has_uuid(0))
self.assertTrue(dms.has_uuid(75))
self.assertTrue(dms.has_uuid(124769))
def test_cacheing_with_map(self):
expected_cache = DataMemoryElement()
expected_map = {
0: 'a',
75: 'b',
124769: 'c',
}
dms = DataMemorySet(expected_cache)
dms._element_map = expected_map
dms.cache()
self.assertFalse(expected_cache.is_empty())
self.assertEqual(pickle.loads(expected_cache.get_bytes()), expected_map)
