mirror of
https://github.com/IfcOpenShell/IfcOpenShell.git
synced 2026-08-13 02:47:48 +00:00
Bruno Postle
114 lines
3.6 KiB
Python
114 lines
3.6 KiB
Python
# This file was generated with the assistance of an AI coding tool.
|
|
import pytest
|
|
|
|
from ifcmcp.server import ifc_info, ifc_relations, ifc_select, ifc_summary, ifc_tree
|
|
|
|
|
|
class TestNoModel:
|
|
"""All query tools should fail when no model is loaded."""
|
|
|
|
def test_summary_no_model(self):
|
|
with pytest.raises(ValueError, match="No model loaded"):
|
|
ifc_summary()
|
|
|
|
def test_tree_no_model(self):
|
|
with pytest.raises(ValueError, match="No model loaded"):
|
|
ifc_tree()
|
|
|
|
def test_info_no_model(self):
|
|
with pytest.raises(ValueError, match="No model loaded"):
|
|
ifc_info(1)
|
|
|
|
def test_select_no_model(self):
|
|
with pytest.raises(ValueError, match="No model loaded"):
|
|
ifc_select("IfcWall")
|
|
|
|
def test_relations_no_model(self):
|
|
with pytest.raises(ValueError, match="No model loaded"):
|
|
ifc_relations(1)
|
|
|
|
|
|
class TestSummary:
|
|
def test_schema(self, loaded_model):
|
|
result = ifc_summary()
|
|
assert result["schema"] == "IFC4"
|
|
|
|
def test_total_entities(self, loaded_model):
|
|
result = ifc_summary()
|
|
assert result["total_entities"] > 0
|
|
|
|
def test_project_name(self, loaded_model):
|
|
result = ifc_summary()
|
|
assert result["project"]["name"] == "TestProject"
|
|
|
|
def test_type_counts(self, loaded_model):
|
|
result = ifc_summary()
|
|
assert result["types"]["IfcWall"] == 1
|
|
assert result["types"]["IfcSlab"] == 1
|
|
|
|
|
|
class TestTree:
|
|
def test_root_is_project(self, loaded_model):
|
|
result = ifc_tree()
|
|
assert result["type"] == "IfcProject"
|
|
assert result["name"] == "TestProject"
|
|
|
|
def test_hierarchy_depth(self, loaded_model):
|
|
result = ifc_tree()
|
|
site = result["children"][0]
|
|
assert site["type"] == "IfcSite"
|
|
building = site["children"][0]
|
|
assert building["type"] == "IfcBuilding"
|
|
storey = building["children"][0]
|
|
assert storey["type"] == "IfcBuildingStorey"
|
|
|
|
|
|
class TestInfo:
|
|
def test_wall_info(self, loaded_model):
|
|
wall = loaded_model.by_type("IfcWall")[0]
|
|
result = ifc_info(wall.id())
|
|
assert result["id"] == wall.id()
|
|
assert result["type"] == "IfcWall"
|
|
|
|
def test_invalid_id(self, loaded_model):
|
|
with pytest.raises(Exception):
|
|
ifc_info(999999)
|
|
|
|
|
|
class TestSelect:
|
|
def test_select_walls(self, loaded_model):
|
|
result = ifc_select("IfcWall")
|
|
assert len(result) == 1
|
|
assert result[0]["type"] == "IfcWall"
|
|
assert result[0]["name"] == "Wall001"
|
|
|
|
def test_select_slabs(self, loaded_model):
|
|
result = ifc_select("IfcSlab")
|
|
assert len(result) == 1
|
|
assert result[0]["name"] == "Slab001"
|
|
|
|
def test_select_no_match(self, loaded_model):
|
|
result = ifc_select("IfcWindow")
|
|
assert result == []
|
|
|
|
|
|
class TestRelations:
|
|
def test_wall_relations(self, loaded_model):
|
|
wall = loaded_model.by_type("IfcWall")[0]
|
|
result = ifc_relations(wall.id())
|
|
assert result["id"] == wall.id()
|
|
assert result["type"] == "IfcWall"
|
|
assert "hierarchy" in result
|
|
|
|
def test_traverse_up(self, loaded_model):
|
|
wall = loaded_model.by_type("IfcWall")[0]
|
|
result = ifc_relations(wall.id(), traverse="up")
|
|
assert isinstance(result, list)
|
|
assert result[0]["type"] == "IfcWall"
|
|
assert result[-1]["type"] == "IfcProject"
|
|
|
|
def test_traverse_empty_string_means_no_traverse(self, loaded_model):
|
|
wall = loaded_model.by_type("IfcWall")[0]
|
|
result = ifc_relations(wall.id(), traverse="")
|
|
assert isinstance(result, dict)
|