s_ranjbar/city_retrofit
1
0
Fork 1
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Clean mapping test

Browse Source
This commit is contained in:
Guille Gutierrez 2023-02-24 07:19:01 -05:00
parent e070ee5779
commit def1bd221f
2 changed files with 5 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
hub/helpers/geometry_helper.py
View File

@ -62,7 +62,7 @@ class GeometryHelper:
return MapPoint(((city.upper_corner[0] - coordinate[0]) * 0.5), ((city.upper_corner[1] - coordinate[1]) * 0.5))
@staticmethod
def city_mapping(city, building_names=None):
def city_mapping(city, building_names=None, plot=False):
if building_names is None:
building_names = [b.name for b in city.buildings]
x = int((city.upper_corner[0] - city.lower_corner[0]) * 0.5) + 1
@ -101,7 +101,8 @@ class GeometryHelper:
neighbour.neighbours = [building]
elif building not in neighbour.neighbours:
neighbour.neighbours.append(building)
img.show()
if plot:
img.show()
@staticmethod
def segment_list_to_trimesh(lines) -> Trimesh:

14
hub/unittests/test_geometry_factory.py
View File

@ -152,20 +152,10 @@ class TestGeometryFactory(TestCase):
"""
Test neighbours map creation
"""
start = datetime.datetime.now()
file = 'concordia.geojson'
city = self._get_city(file, 'geojson',
height_field='citygml_me',
year_of_construction_field='ANNEE_CONS',
function_field='LIBELLE_UT')
city_end = datetime.datetime.now()
print(f'city load {city_end-start}')
GeometryHelper.city_mapping(city)
end = datetime.datetime.now()
print(f'city map {end-city_end}')
for building in city.buildings:
if building.neighbours is not None:
print(f'{building.name} [{[b.name for b in building.neighbours]}]')
else:
print(f'{building.name} has no neighbours')
GeometryHelper.city_mapping(city, plot=False)
self.assertTrue(False)