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)