Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
def test_build_index(self, lod):
co = models.CityObject(id='one')
geom = models.Geometry(type='Solid', lod=lod)
co.geometry.append(geom)
geometry, vtx_lookup, vtx_idx = co.build_index()
assert 'semantics' not in geometry[0]
def test_get_surface_boundaries(self, data_geometry):
"""Test how to get the boundaries (geometry) of semantic surfaces"""
geometry, vertices = data_geometry
geom = models.Geometry(type=geometry[0]['type'],
lod=geometry[0]['lod'],
boundaries=geometry[0]['boundaries'],
semantics_obj=geometry[0]['semantics'],
vertices=vertices)
roofsurfaces = geom.get_surfaces('roofsurface')
rsrf_bndry = [geom.get_surface_boundaries(rsrf)
for i,rsrf in roofsurfaces.items()]
roof_geom = [
[
[[(1.0, 1.0, 0.0), (1.0, 1.0, 0.0), (1.0, 1.0, 0.0), (1.0, 1.0, 0.0)]],
[[(3.0, 1.0, 0.0), (3.0, 1.0, 0.0), (3.0, 1.0, 0.0), (3.0, 1.0, 0.0)]]
]
]
assert rsrf_bndry == roof_geom
doorsurfaces = geom.get_surfaces('door')
def cm_rdam_subset(rotterdam_subset):
rotterdam_subset.cityobjects = dict()
for co_id, co in rotterdam_subset.j['CityObjects'].items():
# do some verification here
children = co['children'] if 'children' in co else None
parents = co['parents'] if 'parents' in co else None
attributes = co['attributes'] if 'attributes' in co else None
# cast to objects
geometry = []
for geom in co['geometry']:
semantics = geom['semantics'] if 'semantics' in geom else None
geometry.append(
models.Geometry(
type=geom['type'],
lod=geom['lod'],
boundaries=geom['boundaries'],
semantics_obj=semantics,
vertices=rotterdam_subset.j['vertices']
)
)
rotterdam_subset.cityobjects[co_id] = models.CityObject(
id=id,
type=co['type'],
attributes=attributes,
children=children,
parents=parents,
geometry=geometry
)
yield rotterdam_subset
def cm_zur_subset(zurich_subset):
zurich_subset.cityobjects = dict()
for co_id, co in zurich_subset.j['CityObjects'].items():
# do some verification here
children = co['children'] if 'children' in co else None
parents = co['parents'] if 'parents' in co else None
attributes = co['attributes'] if 'attributes' in co else None
# cast to objects
geometry = []
for geom in co['geometry']:
semantics = geom['semantics'] if 'semantics' in geom else None
geometry.append(
models.Geometry(
type=geom['type'],
lod=geom['lod'],
boundaries=geom['boundaries'],
semantics_obj=semantics,
vertices=zurich_subset.j['vertices']
)
)
zurich_subset.cityobjects[co_id] = models.CityObject(
id=co_id,
type=co['type'],
attributes=attributes,
children=children,
parents=parents,
geometry=geometry
)
return zurich_subset
def test_get_surface_boundaries(self, boundaries, surface, surfaces):
geom = models.Geometry()
geom.boundaries = boundaries
res = geom.get_surface_boundaries(surface)
assert res == surfaces
def cm_rdam_subset(rotterdam_subset):
rotterdam_subset.cityobjects = dict()
for co_id, co in rotterdam_subset.j['CityObjects'].items():
# do some verification here
children = co['children'] if 'children' in co else None
parents = co['parents'] if 'parents' in co else None
attributes = co['attributes'] if 'attributes' in co else None
# cast to objects
geometry = []
for geom in co['geometry']:
semantics = geom['semantics'] if 'semantics' in geom else None
geometry.append(
models.Geometry(
type=geom['type'],
lod=geom['lod'],
boundaries=geom['boundaries'],
semantics_obj=semantics,
vertices=rotterdam_subset.j['vertices']
)
)
rotterdam_subset.cityobjects[co_id] = models.CityObject(
id=id,
type=co['type'],
attributes=attributes,
children=children,
parents=parents,
geometry=geometry
)
yield rotterdam_subset