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

Unassigning the last unit now cleans up empty unit assignments

Browse Source
This commit is contained in:
Dion Moult
2021-10-14 12:42:33 +11:00
parent 179a970974
commit 464b0c2ebc
2 changed files with 26 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/ifcopenshell-python/ifcopenshell/api/unit/unassign_unit.py
+2 -1
View File
@@ -14,4 +14,5 @@ class Usecase:
units = units - set(self.settings["units"])
if units:
unit_assignment.Units = list(units)
return unit_assignment
return unit_assignment
self.file.remove(unit_assignment)
src/ifcopenshell-python/test/api/unit/test_unassign_unit.py
+24
View File
@@ -0,0 +1,24 @@
import test.bootstrap
import ifcopenshell.api
class TestUnassignUnit(test.bootstrap.IFC4):
def test_run(self):
project = self.file.createIfcProject()
unit1 = ifcopenshell.api.run("unit.add_monetary_unit", self.file, currency="FOO")
unit2 = ifcopenshell.api.run("unit.add_monetary_unit", self.file, currency="BAR")
assignment = ifcopenshell.api.run("unit.assign_unit", self.file, units=[unit1, unit2])
ifcopenshell.api.run("unit.unassign_unit", self.file, units=[unit1])
assert unit1 not in assignment.Units
assert unit2 in assignment.Units
def test_unassigning_the_last_unit(self):
project = self.file.createIfcProject()
unit = ifcopenshell.api.run("unit.add_monetary_unit", self.file, currency="FOO")
ifcopenshell.api.run("unit.assign_unit", self.file, units=[unit])
ifcopenshell.api.run("unit.unassign_unit", self.file, units=[unit])
assert project.UnitsInContext is None
def test_doing_nothing_if_the_unit_is_not_assigned(self):
unit = ifcopenshell.api.run("unit.add_monetary_unit", self.file, currency="FOO")
assert ifcopenshell.api.run("unit.unassign_unit", self.file, units=[unit]) is None