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

tests for system.get_connected_to / get_connected_from

Browse Source
This commit is contained in:
Andrej730
2023-08-01 13:37:59 +05:00
parent 3c8dc70198
commit 2cc9ea0539
2 changed files with 28 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/ifcopenshell-python/ifcopenshell/util/system.py
-2
View File
@@ -84,7 +84,6 @@ def get_connected_port(port):
return rel.RelatingPort
# TODO: cover in tests
def get_connected_to(element):
results = []
for port in ifcopenshell.util.system.get_ports(element):
@@ -102,7 +101,6 @@ def get_connected_to(element):
return results
# TODO: cover in tests
def get_connected_from(element):
results = []
for port in ifcopenshell.util.system.get_ports(element):
src/ifcopenshell-python/test/util/test_system.py
+28
View File
@@ -79,3 +79,31 @@ class TestGetConnectedPort(test.bootstrap.IFC4):
ifcopenshell.api.run("system.connect_port", self.file, port1=port1, port2=port2)
assert subject.get_connected_port(port1) == port2
assert subject.get_connected_port(port2) == port1
class TestGetConnectedToFrom(test.bootstrap.IFC4):
def test_run(self):
port1 = ifcopenshell.api.run("system.add_port", self.file)
element1 = ifcopenshell.api.run("root.create_entity", self.file, ifc_class="IfcFlowSegment")
ifcopenshell.api.run("system.assign_port", self.file, element=element1, port=port1)
port2 = ifcopenshell.api.run("system.add_port", self.file)
element2 = ifcopenshell.api.run("root.create_entity", self.file, ifc_class="IfcFlowSegment")
ifcopenshell.api.run("system.assign_port", self.file, element=element2, port=port2)
ifcopenshell.api.run("system.connect_port", self.file, port1=port1, port2=port2, direction="SOURCE")
assert subject.get_connected_to(element1) == [element2]
assert subject.get_connected_from(element1) == []
assert subject.get_connected_to(element2) == []
assert subject.get_connected_from(element2) == [element1]
ifcopenshell.api.run("system.connect_port", self.file, port1=port1, port2=port2, direction="SINK")
assert subject.get_connected_to(element1) == []
assert subject.get_connected_from(element1) == [element2]
assert subject.get_connected_to(element2) == [element1]
assert subject.get_connected_from(element2) == []
class TestGetConnectedToFromIFC2X3(test.bootstrap.IFC2X3):
def test_run(self):
TestGetConnectedToFrom.test_run(self)