Fix bug where imported grids were not linked and didn't auto track edits

src/blenderbim/test/tool/test_collector.py

@@ -178,3 +178,39 @@ class TestAssign(NewFile):
         subject.assign(element_obj)
         assert element_obj.users_collection[0].name == "IfcOpeningElements"
         assert bpy.data.collections.get("IfcProject/My Project").children.get("IfcOpeningElements")
+
+    def test_in_decomposition_mode_grids_are_placed_in_their_own_collection(self):
+        bpy.ops.bim.create_project()
+        element_obj = bpy.data.objects.new("IfcGrid/Name", None)
+        element = tool.Ifc.get().createIfcGrid()
+        tool.Ifc.link(element, element_obj)
+        ifcopenshell.api.run(
+            "spatial.assign_container",
+            tool.Ifc.get(),
+            product=element,
+            relating_structure=tool.Ifc.get().by_type("IfcSite")[0],
+        )
+        bpy.context.scene.collection.objects.link(element_obj)
+        subject.assign(element_obj)
+        assert element_obj.users_collection[0].name == "IfcGrid/Name"
+        assert bpy.data.collections.get("IfcSite/My Site").children.get("IfcGrid/Name")
+
+    def test_in_decomposition_mode_grids_axes_are_placed_in_an_axis_collection_of_the_grid(self):
+        bpy.ops.bim.create_project()
+        element_obj = bpy.data.objects.new("IfcGrid/Name", None)
+        axis_obj = bpy.data.objects.new("IfcGrid/Name", None)
+        axis = tool.Ifc.get().createIfcGridAxis()
+        element = tool.Ifc.get().createIfcGrid(UAxes=[axis])
+        tool.Ifc.link(element, element_obj)
+        tool.Ifc.link(axis, axis_obj)
+        ifcopenshell.api.run(
+            "spatial.assign_container",
+            tool.Ifc.get(),
+            product=element,
+            relating_structure=tool.Ifc.get().by_type("IfcSite")[0],
+        )
+        bpy.context.scene.collection.objects.link(element_obj)
+        subject.assign(element_obj)
+        subject.assign(axis_obj)
+        assert axis_obj.users_collection[0].name == "UAxes"
+        assert bpy.data.collections.get("IfcGrid/Name").children.get("UAxes")

src/blenderbim/test/tool/test_spatial.py

@@ -80,6 +80,17 @@ class TestCanContain(NewFile):
         tool.Ifc.link(element, element_obj)
         assert subject.can_contain(structure_obj, element_obj) is False
 
+    def test_other_non_elements_that_have_a_contained_in_structure_attribute_can_be_contained(self):
+        ifc = ifcopenshell.file()
+        tool.Ifc.set(ifc)
+        structure = ifc.createIfcSite()
+        structure_obj = bpy.data.objects.new("Object", None)
+        tool.Ifc.link(structure, structure_obj)
+        element = ifc.createIfcGrid()
+        element_obj = bpy.data.objects.new("Object", None)
+        tool.Ifc.link(element, element_obj)
+        assert subject.can_contain(structure_obj, element_obj) is True
+
 
 class TestDisableEditing(NewFile):
     def test_run(self):