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')