Glassvision/IfcOpenShell
1
0
Fork 0
mirror of https://github.com/IfcOpenShell/IfcOpenShell.git synced 2026-08-10 09:48:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

resource.add_resource_quantity to ensure quantity type is supported

Browse Source
This commit is contained in:
Andrej730
2024-06-03 12:26:50 +05:00
parent 5ccdf187d5
commit c399eb259a
4 changed files with 54 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/ifcopenshell-python/test/api/resource/test_add_resource_quantity.py
+30 -12
View File
@@ -16,27 +16,45 @@
# You should have received a copy of the GNU Lesser General Public License
# along with IfcOpenShell. If not, see <http://www.gnu.org/licenses/>.
import pytest
import test.bootstrap
import ifcopenshell.api
import ifcopenshell.api.resource
import ifcopenshell.util.resource
class TestAddResourceQuantity(test.bootstrap.IFC4):
def test_run(self):
schema = ifcopenshell.schema_by_name(self.file.schema)
quantity_types = [t.name() for t in schema.declaration_by_name("IfcPhysicalSimpleQuantity").subtypes()]
self.file.create_entity("IfcProject") # add_resource
resource = ifcopenshell.api.run("resource.add_resource", self.file, ifc_class="IfcCrewResource")
resource_types = [t.name() for t in schema.declaration_by_name("IfcConstructionResource").subtypes()]
for quantity_type in quantity_types:
quantity = ifcopenshell.api.run(
"resource.add_resource_quantity", self.file, resource=resource, ifc_class=quantity_type
)
assert quantity.is_a(quantity_type)
assert quantity.Name == "Unnamed"
assert quantity[3] == 0.0
# previous quantity is reassigned and removed
assert resource.BaseQuantity == quantity
assert len(self.file.by_type("IfcPhysicalSimpleQuantity")) == 1
self.file.create_entity("IfcProject") # add_resource
for resource_type in resource_types:
resource = ifcopenshell.api.resource.add_resource(self.file, ifc_class=resource_type)
available_quantities = ifcopenshell.util.resource.RESOURCES_TO_QUANTITIES[resource_type]
for quantity_type in quantity_types:
if quantity_type not in available_quantities:
with pytest.raises(ValueError):
quantity = ifcopenshell.api.resource.add_resource_quantity(
self.file, resource=resource, ifc_class=quantity_type
)
continue
else:
quantity = ifcopenshell.api.resource.add_resource_quantity(
self.file, resource=resource, ifc_class=quantity_type
)
assert quantity.is_a(quantity_type)
assert quantity.Name == "Unnamed"
assert quantity[3] == 0.0
# previous quantity is reassigned and removed
assert resource.BaseQuantity == quantity
assert len(self.file.by_type("IfcPhysicalSimpleQuantity")) == 1
ifcopenshell.api.resource.remove_resource(self.file, resource)
class TestAddResourceQuantityIFC2X3(test.bootstrap.IFC2X3, TestAddResourceQuantity):