Glassvision/IfcOpenShell
1
0
Fork 0
mirror of https://github.com/IfcOpenShell/IfcOpenShell.git synced 2026-08-09 17:31:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

ExtractElemenets - add some tests

Browse Source
This commit is contained in:
Andrej730
2024-03-27 17:52:18 +05:00
parent f68e8f0e00
commit 74adde3030
1 changed files with 49 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/ifcpatch/test/test_ExtractElements.py
+49 -1
View File
@@ -20,11 +20,59 @@ import os
import pytest
import ifcpatch
import ifcopenshell
import ifcopenshell.api
import ifcopenshell.util.element
class TestExtractElements:
def test_basic(self):
ifc_file = ifcopenshell.file()
project = ifcopenshell.api.run("root.create_entity", ifc_file, ifc_class="IfcProject")
wall = ifcopenshell.api.run("root.create_entity", ifc_file, ifc_class="IfcWall")
output = ifcpatch.execute({"file": ifc_file, "recipe": "ExtractElements", "arguments": ["IfcWall"]})
assert output.by_type("IfcProject")[0].GlobalId == project.GlobalId
assert output.by_type("IfcWall")[0].GlobalId == wall.GlobalId
def test_keep_spatial_structure(self):
ifc_file = ifcopenshell.file()
project = ifcopenshell.api.run("root.create_entity", ifc_file, ifc_class="IfcProject")
site = ifcopenshell.api.run("root.create_entity", ifc_file, ifc_class="IfcSite")
building = ifcopenshell.api.run("root.create_entity", ifc_file, ifc_class="IfcBuilding")
storey = ifcopenshell.api.run("root.create_entity", ifc_file, ifc_class="IfcBuildingStorey")
wall = ifcopenshell.api.run("root.create_entity", ifc_file, ifc_class="IfcWall")
ifcopenshell.api.run("aggregate.assign_object", ifc_file, product=building, relating_object=site)
ifcopenshell.api.run("aggregate.assign_object", ifc_file, product=storey, relating_object=building)
ifcopenshell.api.run("spatial.assign_container", ifc_file, product=wall, relating_structure=storey)
output = ifcpatch.execute({"file": ifc_file, "recipe": "ExtractElements", "arguments": ["IfcWall"]})
wall_new = output.by_type("IfcWall")[0]
assert (storey_new := ifcopenshell.util.element.get_container(wall_new)).GlobalId == storey.GlobalId
assert (building_new := ifcopenshell.util.element.get_aggregate(storey_new)).GlobalId == building.GlobalId
assert (site_new := ifcopenshell.util.element.get_aggregate(building_new)).GlobalId == site.GlobalId
def test_keep_aggregate_in_spatial_structure(self):
ifc_file = ifcopenshell.file()
project = ifcopenshell.api.run("root.create_entity", ifc_file, ifc_class="IfcProject")
element = ifcopenshell.api.run("root.create_entity", ifc_file, ifc_class="IfcElementAssembly")
container = ifcopenshell.api.run("root.create_entity", ifc_file, ifc_class="IfcBuildingStorey")
subelement = ifcopenshell.api.run("root.create_entity", ifc_file, ifc_class="IfcWall")
ifcopenshell.api.run("spatial.assign_container", ifc_file, product=element, relating_structure=container)
ifcopenshell.api.run("aggregate.assign_object", ifc_file, product=subelement, relating_object=element)
output = ifcpatch.execute({"file": ifc_file, "recipe": "ExtractElements", "arguments": ["IfcWall"]})
wall_new = output.by_type("IfcWall")[0]
assembly = output.by_type("IfcElementAssembly")[0]
assert ifcopenshell.util.element.get_aggregate(wall_new).GlobalId == element.GlobalId
assert ifcopenshell.util.element.get_container(assembly).GlobalId == container.GlobalId
def test_getting_the_psets_of_a_product_as_a_dictionary(self):
ifc = ifcopenshell.open(os.path.join(os.getcwd(), "test", "files", "basic.ifc"))
output = ifcpatch.execute({"input": ifc, "recipe": "ExtractElements", "arguments": [".IfcWall"]})
output = ifcpatch.execute({"file": ifc, "recipe": "ExtractElements", "arguments": ["IfcWall"]})
assert output.by_type("IfcWall")
assert not output.by_type("IfcSlab")