mirror of
https://github.com/IfcOpenShell/IfcOpenShell.git
synced 2026-08-14 11:24:19 +00:00
88 lines
3.4 KiB
Python
88 lines
3.4 KiB
Python
|
# This file was generated with the assistance of an AI coding tool.
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import ifcopenshell
|
||
|
|
import ifcopenshell.api.aggregate
|
||
|
|
import ifcopenshell.api.owner.settings
|
||
|
|
import ifcopenshell.api.project
|
||
|
|
import ifcopenshell.api.root
|
||
|
|
import ifcopenshell.api.spatial
|
||
|
|
import ifcopenshell.api.unit
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from ifcedit.quantify import AVAILABLE_RULES, list_rules, run_quantify
|
||
|
|
|
||
|
|
|
||
|
|
class TestListRules:
|
||
|
|
def test_returns_list(self):
|
||
|
|
result = list_rules()
|
||
|
|
assert isinstance(result, list)
|
||
|
|
|
||
|
|
def test_each_entry_has_name(self):
|
||
|
|
result = list_rules()
|
||
|
|
for entry in result:
|
||
|
|
assert "name" in entry
|
||
|
|
|
||
|
|
def test_ifc4_rule_present(self):
|
||
|
|
result = list_rules()
|
||
|
|
names = [r["name"] for r in result]
|
||
|
|
assert "IFC4QtoBaseQuantities" in names
|
||
|
|
|
||
|
|
def test_ifc4x3_rule_present(self):
|
||
|
|
result = list_rules()
|
||
|
|
names = [r["name"] for r in result]
|
||
|
|
assert "IFC4X3QtoBaseQuantities" in names
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
def quantify_model():
|
||
|
|
"""Create an IFC4 model with a wall element."""
|
||
|
|
f = ifcopenshell.api.project.create_file()
|
||
|
|
ifcopenshell.api.owner.settings.get_user = lambda ifc: (ifc.by_type("IfcPersonAndOrganization") or [None])[0]
|
||
|
|
ifcopenshell.api.owner.settings.get_application = lambda ifc: (ifc.by_type("IfcApplication") or [None])[0]
|
||
|
|
|
||
|
|
project = ifcopenshell.api.root.create_entity(f, ifc_class="IfcProject", name="TestProject")
|
||
|
|
ifcopenshell.api.unit.assign_unit(f)
|
||
|
|
|
||
|
|
site = ifcopenshell.api.root.create_entity(f, ifc_class="IfcSite", name="TestSite")
|
||
|
|
building = ifcopenshell.api.root.create_entity(f, ifc_class="IfcBuilding", name="TestBuilding")
|
||
|
|
storey = ifcopenshell.api.root.create_entity(f, ifc_class="IfcBuildingStorey", name="Ground Floor")
|
||
|
|
|
||
|
|
ifcopenshell.api.aggregate.assign_object(f, products=[site], relating_object=project)
|
||
|
|
ifcopenshell.api.aggregate.assign_object(f, products=[building], relating_object=site)
|
||
|
|
ifcopenshell.api.aggregate.assign_object(f, products=[storey], relating_object=building)
|
||
|
|
|
||
|
|
wall = ifcopenshell.api.root.create_entity(f, ifc_class="IfcWall", name="Wall001")
|
||
|
|
ifcopenshell.api.spatial.assign_container(f, products=[wall], relating_structure=storey)
|
||
|
|
|
||
|
|
return f
|
||
|
|
|
||
|
|
|
||
|
|
class TestRunQuantify:
|
||
|
|
def test_returns_ok_true(self, quantify_model):
|
||
|
|
result = run_quantify(quantify_model, "IFC4QtoBaseQuantities")
|
||
|
|
assert result["ok"] is True
|
||
|
|
|
||
|
|
def test_returns_rule_name(self, quantify_model):
|
||
|
|
result = run_quantify(quantify_model, "IFC4QtoBaseQuantities")
|
||
|
|
assert result["rule"] == "IFC4QtoBaseQuantities"
|
||
|
|
|
||
|
|
def test_returns_elements_quantified(self, quantify_model):
|
||
|
|
result = run_quantify(quantify_model, "IFC4QtoBaseQuantities")
|
||
|
|
assert "elements_quantified" in result
|
||
|
|
assert isinstance(result["elements_quantified"], int)
|
||
|
|
|
||
|
|
def test_unknown_rule_returns_error(self, quantify_model):
|
||
|
|
result = run_quantify(quantify_model, "NonExistentRule")
|
||
|
|
assert result["ok"] is False
|
||
|
|
assert "error" in result
|
||
|
|
|
||
|
|
def test_selector_restricts_elements(self, quantify_model):
|
||
|
|
result = run_quantify(quantify_model, "IFC4QtoBaseQuantities", selector="IfcWall")
|
||
|
|
assert result["ok"] is True
|
||
|
|
assert result["rule"] == "IFC4QtoBaseQuantities"
|
||
|
|
|
||
|
|
def test_empty_selector_runs_on_all(self, quantify_model):
|
||
|
|
result = run_quantify(quantify_model, "IFC4QtoBaseQuantities", selector=None)
|
||
|
|
assert result["ok"] is True
|