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)