Removing contexts now also reassign to parent or remove references to the context.

This commit is contained in:
Dion Moult
2021-10-01 09:29:38 +10:00
parent a7a1616f9d
commit 42d321d553
3 changed files with 85 additions and 2 deletions
@@ -0,0 +1,32 @@
import test.bootstrap
import ifcopenshell.api
class TestRemoveContext(test.bootstrap.IFC4):
def test_removing_a_context(self):
context = self.file.createIfcGeometricRepresentationContext()
ifcopenshell.api.run("context.remove_context", self.file, context=context)
assert len(self.file.by_type("IfcGeometricRepresentationContext")) == 0
def test_removing_a_context_with_subcontexts(self):
context = self.file.createIfcGeometricRepresentationContext()
subcontext = self.file.createIfcGeometricRepresentationSubcontext()
subcontext.ParentContext = context
ifcopenshell.api.run("context.remove_context", self.file, context=context)
assert len(self.file.by_type("IfcGeometricRepresentationContext")) == 0
def test_removing_a_subcontext_and_reassigning_references_to_its_parent(self):
context = self.file.createIfcGeometricRepresentationContext()
subcontext = self.file.createIfcGeometricRepresentationSubcontext()
subcontext.ParentContext = context
representation = self.file.createIfcRepresentation(ContextOfItems=subcontext)
ifcopenshell.api.run("context.remove_context", self.file, context=subcontext)
assert len(self.file.by_type("IfcGeometricRepresentationSubcontext")) == 0
assert representation in self.file.get_inverse(context)
def test_removing_a_context_with_references(self):
subcontext = self.file.createIfcGeometricRepresentationSubContext()
representation = self.file.createIfcRepresentation(ContextOfItems=subcontext)
map_conversion = self.file.createIfcMapConversion(SourceCRS=subcontext)
ifcopenshell.api.run("context.remove_context", self.file, context=subcontext)
assert len([e for e in self.file]) == 0