m_osman/city_retrofit
1
0
Fork 0
forked from s_ranjbar/city_retrofit
Code Pull Requests Activity

Partially correct unit-test

Browse Source
This commit is contained in:
Guille Gutierrez 2023-07-25 15:36:49 -04:00
parent 0d63a6c4e9
commit 066a25bfee
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tests/test_city_merge.py
View File

@ -33,7 +33,7 @@ class TestCityMerge(TestCase):
self._executable = 'sra' self._executable = 'sra'
def test_merge(self): def test_merge(self):
file_path = Path('./tests_data/test.geojson').resolve() file_path = Path(self._example_path / 'test.geojson').resolve()
full_city = GeometryFactory('geojson', file_path, height_field='citygml_me').city full_city = GeometryFactory('geojson', file_path, height_field='citygml_me').city
self.assertEqual(17, len(full_city.buildings), 'Wrong number of buildings') self.assertEqual(17, len(full_city.buildings), 'Wrong number of buildings')
odd_city = City(full_city.lower_corner, full_city.upper_corner, full_city.srs_name) odd_city = City(full_city.lower_corner, full_city.upper_corner, full_city.srs_name)
@ -54,17 +54,17 @@ class TestCityMerge(TestCase):
def test_merge_with_radiation(self): def test_merge_with_radiation(self):
sra = distutils.spawn.find_executable('sra') sra = distutils.spawn.find_executable('sra')
file_path = Path('./tests_data/test.geojson').resolve() file_path = Path(self._example_path / 'test.geojson').resolve()
output_path = Path('./tests_outputs/')
full_city = GeometryFactory('geojson', file_path, height_field='citygml_me').city full_city = GeometryFactory('geojson', file_path, height_field='citygml_me').city
even_city = City(full_city.lower_corner, full_city.upper_corner, full_city.srs_name) even_city = City(full_city.lower_corner, full_city.upper_corner, full_city.srs_name)
for building in full_city.buildings: for building in full_city.buildings:
if int(building.name) % 2 == 0: if int(building.name) % 2 == 0:
even_city.add_city_object(copy.deepcopy(building)) even_city.add_city_object(copy.deepcopy(building))
ExportsFactory('sra', full_city, output_path).export() ExportsFactory('sra', full_city, self._output_path).export()
sra_file = str((output_path / f'{full_city.name}_sra.xml').resolve()) sra_file = str((self._output_path / f'{full_city.name}_sra.xml').resolve())
subprocess.run([sra, sra_file], stdout=subprocess.DEVNULL) subprocess.run([sra, sra_file], stdout=subprocess.DEVNULL)
ResultFactory('sra', full_city, output_path).enrich() ResultFactory('sra', full_city, self._output_path).enrich()
self.assertEqual(17, len(full_city.buildings), 'Wrong number of buildings') self.assertEqual(17, len(full_city.buildings), 'Wrong number of buildings')
merged_city = full_city.merge(even_city) merged_city = full_city.merge(even_city)