def test_map_to_yaml_without_yaml(self):
map = DuckietownMap()
map.tiles = tiles_to_objects(
get_tiles('../maps/regress_4way_adam.yaml'))
map.items = map_objects_to_objects(
get_objects('../maps/regress_4way_adam.yaml'))
map_to_yaml(map, '../maps/test_result')
new_map = DuckietownMap()
new_map.tiles = tiles_to_objects(get_tiles('../maps/test_result.yaml'))
new_map.items = map_objects_to_objects(
get_objects('../maps/test_result.yaml'))
for tile, new_tile in zip(
np.array(map.tiles).flat,
np.array(new_map.tiles).flat):
self.assertEqual(tile.kind, new_tile.kind)
self.assertEqual(tile.rotation, new_tile.rotation)
for obj, new_obj in zip(map.items, new_map.items):
self.assertEqual(obj.kind, new_obj.kind)
self.assertEqual(obj.position, new_obj.position)
self.assertEqual(obj.rotation, new_obj.rotation)
self.assertEqual(obj.height, new_obj.height)
self.assertEqual(obj.optional, new_obj.optional)
self.assertEqual(obj.static, new_obj.static)
def test_map_to_yaml_without_objects(self):
map = DuckietownMap()
map.tiles = tiles_to_objects(
get_tiles('../maps/regress_4way_drivable.yaml'))
map.items = map_objects_to_objects(
get_objects('../maps/regress_4way_drivable.yaml'))
map_to_yaml(map, '../maps/test_result.yaml')
new_map = DuckietownMap()
new_map.tiles = tiles_to_objects(get_tiles('../maps/test_result.yaml'))
new_map.items = map_objects_to_objects(
get_objects('../maps/test_result.yaml'))
for tile, new_tile in zip(
np.array(map.tiles).flat,
np.array(new_map.tiles).flat):
self.assertEqual(tile.kind, new_tile.kind)
self.assertEqual(tile.rotation, new_tile.rotation)
self.assertEqual(map.items, new_map.items)
def save(self, map):
# deep copy
backup = DuckietownMap()
backup.tiles = [[]]
for i in range(len(map.tiles)):
backup.tiles.append([])
for j in range(len(map.tiles[i])):
backup.tiles[i].append(copy.copy(map.tiles[i][j]))
if map.items:
for i in range(len(map.items)):
backup.items.append(map.items[i])
if len(self.memento) == self.memento.maxlen - 1:
self.memento.popleft()
self.memento.append(backup)