From c621e33466e9dcab5fea311d40a8f0103bed8a59 Mon Sep 17 00:00:00 2001
From: guille <guillermo.gutierrezmorote@concordia.ca>
Date: Thu, 20 Oct 2022 11:34:22 -0400
Subject: [PATCH] Add method to merge cities

---
 city_model_structure/city.py  |  6 ++++++
 unittests/test_city_merge.py  | 36 +++++++++++++++++++++++++++++++++++
 unittests/test_enrichement.py |  4 ++--
 3 files changed, 44 insertions(+), 2 deletions(-)
 create mode 100644 unittests/test_city_merge.py

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