# This file was generated with the assistance of an AI coding tool.

import ifcopenshell

import ifcopenshell.api.aggregate

import ifcopenshell.api.material

import ifcopenshell.api.owner.settings

import ifcopenshell.api.project

import ifcopenshell.api.pset

import ifcopenshell.api.root

import ifcopenshell.api.spatial

import ifcopenshell.api.unit

import pytest

@pytest.fixture

def model():

"""Create an IFC4 model with a spatial hierarchy and a wall."""

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

@pytest.fixture

def model_file(model, tmp_path):

"""Write the model fixture to a temp file and return the path."""

path = tmp_path / "test.ifc"

model.write(str(path))

return str(path)

@pytest.fixture

def library():

"""Create an IFC4 library with a single IfcWallType asset."""

lib = 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]

ifcopenshell.api.root.create_entity(lib, ifc_class="IfcProject", name="TestLibrary")

ifcopenshell.api.unit.assign_unit(lib)

ifcopenshell.api.root.create_entity(lib, ifc_class="IfcWallType", name="WAL01")

return lib

@pytest.fixture

def library_file(library, tmp_path):

"""Write the library fixture to a temp file and return the path."""

path = tmp_path / "library.ifc"

library.write(str(path))

return str(path)