def test_map_to_yaml_2(self):
map = DuckietownMap()
map.set_tile_layer(
tiles_to_objects(data_from_file('../maps/test.yaml').get('tiles')))
map.set_item_layer(
map_objects_to_objects(
data_from_file('../maps/test.yaml').get('objects')))
map_to_yaml(map, '../maps/test_result.yaml')
new_map = DuckietownMap()
new_map.set_tile_layer(
tiles_to_objects(
data_from_file('../maps/test_result.yaml').get('tiles')))
new_map.set_item_layer(
map_objects_to_objects(
data_from_file('../maps/test_result.yaml').get('objects')))
for tile, new_tile in zip(
np.array(map.get_tile_layer().data).flat,
np.array(new_map.get_tile_layer().data).flat):
self.assertEqual(tile.kind, new_tile.kind)
self.assertEqual(tile.rotation, new_tile.rotation)
for obj, new_obj in zip(map.get_item_layer().data,
new_map.get_item_layer().data):
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.set_tile_layer(
tiles_to_objects(
data_from_file('../maps/regress_4way_drivable.yaml').get(
'tiles')))
map.set_item_layer(
map_objects_to_objects(
data_from_file('../maps/regress_4way_drivable.yaml').get(
'objects')))
map_to_yaml(map, '../maps/test_result.yaml')
new_map = DuckietownMap()
new_map.set_tile_layer(
tiles_to_objects(
data_from_file('../maps/test_result.yaml').get('tiles')))
new_map.set_item_layer(
map_objects_to_objects(
data_from_file('../maps/test_result.yaml').get('objects')))
for tile, new_tile in zip(
np.array(map.get_tile_layer().data).flat,
np.array(new_map.get_tile_layer().data).flat):
self.assertEqual(tile.kind, new_tile.kind)
self.assertEqual(tile.rotation, new_tile.rotation)
self.assertEqual(map.get_item_layer().data,
new_map.get_item_layer().data)