diff --git a/city_model_structure/city.py b/city_model_structure/city.py index d50f53ac..4af8df24 100644 --- a/city_model_structure/city.py +++ b/city_model_structure/city.py @@ -442,3 +442,9 @@ class City: Get a copy of the current city """ return copy.deepcopy(self) + + def merge(self, city) -> City: + _merge_city = self.copy + for city_object in city.city_objects: + _merge_city.add_city_object(city_object) + return _merge_city diff --git a/unittests/test_city_merge.py b/unittests/test_city_merge.py new file mode 100644 index 00000000..e1ecaac9 --- /dev/null +++ b/unittests/test_city_merge.py @@ -0,0 +1,36 @@ +""" +TestCityMerge test and validate the merge of several cities into one +SPDX - License - Identifier: LGPL - 3.0 - or -later +Copyright © 2022 Concordia CERC group +Project Coder Guille Gutierrez Guillermo.GutierrezMorote@concordia.ca +""" +from pathlib import Path +from unittest import TestCase + +from imports.geometry_factory import GeometryFactory + + +class TestCityMerge(TestCase): + """ + Functional TestCityMerge + """ + def setUp(self) -> None: + """ + Test setup + :return: None + """ + self._example_path = (Path(__file__).parent / 'tests_data').resolve() + + def _get_citygml(self, file): + file_path = (self._example_path / file).resolve() + city = GeometryFactory('citygml', file_path).city + self.assertIsNotNone(city, 'city is none') + return city + + def test_merge(self): + city_1 = self._get_citygml('one_building_in_kelowna.gml') + city_2 = self._get_citygml('pluto_building.gml') + city = city_1.merge(city_2) + self.assertEqual(len(city_1.city_objects), 1, 'Wrong amount of city_objects found in city_1') + self.assertEqual(len(city_2.city_objects), 1, 'Wrong amount of city_objects found in city_2') + self.assertEqual(len(city.city_objects), 2, 'Wrong amount of city_objects found in city') diff --git a/unittests/test_enrichement.py b/unittests/test_enrichement.py index 23a54027..e6ec7c88 100644 --- a/unittests/test_enrichement.py +++ b/unittests/test_enrichement.py @@ -47,8 +47,8 @@ class TestGeometryFactory(TestCase): for internal_zone in building.internal_zones: self.assertIsNotNone(internal_zone.usage_zones, 'usage zones are not defined') self.assertIsNotNone(internal_zone.thermal_zones, 'thermal zones are not defined') - #self.assertIsNotNone(building.basement_heated, 'building basement_heated is none') - #self.assertIsNotNone(building.attic_heated, 'building attic_heated is none') + self.assertIsNotNone(building.basement_heated, 'building basement_heated is none') + self.assertIsNotNone(building.attic_heated, 'building attic_heated is none') self.assertIsNotNone(building.average_storey_height, 'building average_storey_height is none') self.assertIsNotNone(building.storeys_above_ground, 'building storeys_above_ground is none') self.assertTrue(building.is_conditioned, 'building is_conditioned is not conditioned')