mirror of
https://github.com/IfcOpenShell/IfcOpenShell.git
synced 2026-08-10 01:41:57 +00:00
Bruno Postle
38 lines
1.2 KiB
Python
38 lines
1.2 KiB
Python
# This file was generated with the assistance of an AI coding tool.
|
|
from ifcquery.tree import tree
|
|
|
|
|
|
class TestTree:
|
|
def test_root_is_project(self, model):
|
|
result = tree(model)
|
|
assert result["type"] == "IfcProject"
|
|
assert result["name"] == "TestProject"
|
|
|
|
def test_spatial_hierarchy(self, model):
|
|
result = tree(model)
|
|
# Project > Site > Building > Storey
|
|
site = result["children"][0]
|
|
assert site["type"] == "IfcSite"
|
|
assert site["name"] == "TestSite"
|
|
|
|
building = site["children"][0]
|
|
assert building["type"] == "IfcBuilding"
|
|
assert building["name"] == "TestBuilding"
|
|
|
|
storey = building["children"][0]
|
|
assert storey["type"] == "IfcBuildingStorey"
|
|
assert storey["name"] == "Ground Floor"
|
|
|
|
def test_contained_elements(self, model):
|
|
result = tree(model)
|
|
storey = result["children"][0]["children"][0]["children"][0]
|
|
elements = storey["elements"]
|
|
element_types = {e["type"] for e in elements}
|
|
assert "IfcWall" in element_types
|
|
assert "IfcSlab" in element_types
|
|
|
|
def test_element_ids_present(self, model):
|
|
result = tree(model)
|
|
assert "id" in result
|
|
assert isinstance(result["id"], int)
|