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

Fix bug in get_container not passing specified ifc_class to the parent #6155

Browse Source
This commit is contained in:
Andrej730
2025-02-14 17:49:01 +05:00
parent 1014487d82
commit 20e1fc2618
2 changed files with 21 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/ifcopenshell-python/ifcopenshell/util/element.py
+1 -1
View File
@@ -935,7 +935,7 @@ def get_container(
return container
container = get_aggregate(container)
elif parent := get_parent(element):
return get_container(parent, should_get_direct)
return get_container(parent, should_get_direct, ifc_class)
def get_referenced_structures(element: ifcopenshell.entity_instance) -> list[ifcopenshell.entity_instance]:
src/ifcopenshell-python/test/util/test_element.py
+20
View File
@@ -793,6 +793,26 @@ class TestGetContainerIFC4(test.bootstrap.IFC4):
ifcopenshell.api.aggregate.assign_object(self.file, products=[subelement], relating_object=element)
assert subject.get_container(subelement, should_get_direct=True) is None
def test_getting_the_specific_spatial_container_of_an_element(self):
element = ifcopenshell.api.root.create_entity(self.file, ifc_class="IfcWall")
building = ifcopenshell.api.root.create_entity(self.file, ifc_class="IfcBuilding")
storey = ifcopenshell.api.root.create_entity(self.file, ifc_class="IfcBuildingStorey")
ifcopenshell.api.aggregate.assign_object(self.file, products=[storey], relating_object=building)
ifcopenshell.api.spatial.assign_container(self.file, products=[element], relating_structure=storey)
assert subject.get_container(element, ifc_class="IfcBuilding") == building
assert subject.get_container(element, ifc_class="IfcSite") == None
def test_getting_the_specific_spatial_container_of_an_element_indirectly(self):
element = ifcopenshell.api.root.create_entity(self.file, ifc_class="IfcElementAssembly")
subelement = ifcopenshell.api.root.create_entity(self.file, ifc_class="IfcWall")
ifcopenshell.api.aggregate.assign_object(self.file, products=[subelement], relating_object=element)
building = ifcopenshell.api.root.create_entity(self.file, ifc_class="IfcBuilding")
storey = ifcopenshell.api.root.create_entity(self.file, ifc_class="IfcBuildingStorey")
ifcopenshell.api.aggregate.assign_object(self.file, products=[storey], relating_object=building)
ifcopenshell.api.spatial.assign_container(self.file, products=[element], relating_structure=storey)
assert subject.get_container(subelement, ifc_class="IfcBuilding") == building
assert subject.get_container(subelement, ifc_class="IfcSite") == None
class TestGetReferencedStructures(test.bootstrap.IFC4):
def test_getting_references_of_an_element(self):