Add method to merge cities

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

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

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