Fix test for new property groups

Author: domfournier

simpeg_drivers/utils/utils.py

@@ -22,6 +22,7 @@
 from geoapps_utils.utils.conversions import string_to_numeric
 from geoapps_utils.utils.numerical import running_mean, traveling_salesman
 from geoh5py import Workspace
+from geoh5py.data import NumericData
 from geoh5py.groups import Group, SimPEGGroup
 from geoh5py.objects import DrapeModel, Octree
 from geoh5py.objects.surveys.direct_current import PotentialElectrode
@@ -38,12 +39,6 @@
 from simpeg.survey import BaseSurvey
 from simpeg.utils import mkvc
 
-
-if TYPE_CHECKING:
-    from simpeg_drivers.components.data import InversionData
-    from simpeg_drivers.driver import InversionDriver
-
-
 from simpeg_drivers import DRIVER_MAP
 from simpeg_drivers.utils.surveys import (
     compute_alongline_distance,
@@ -52,6 +47,11 @@
 )
 
 
+if TYPE_CHECKING:
+    from simpeg_drivers.components.data import InversionData
+    from simpeg_drivers.driver import InversionDriver
+
+
 def calculate_2D_trend(
     points: np.ndarray, values: np.ndarray, order: int = 0, method: str = "all"
 ):
@@ -258,12 +258,19 @@ def drape_to_octree(
                     f"Found more than one data set with name {names[ind]} in"
                     f"model {model.name}."
                 )
+
+            if not isinstance(datum[0], NumericData):
+                continue
+
             if method == "nearest":
                 octree_model.append(datum[0].values)
             else:
                 lookup_inds = mesh.get_containing_cells(model.centroids)
                 octree_model[lookup_inds] = datum[0].values
 
+        if len(octree_model) == 0:
+            continue
+
         if method == "nearest":
             octree_model = np.hstack(octree_model)[lookup_inds]
 

tests/run_tests/driver_mvi_test.py

@@ -154,7 +154,7 @@ def test_magnetic_vector_run(
 
         out_group = run_ws.get_entity("Magnetic Vector Inversion")[0]
         mesh = out_group.get_entity("mesh")[0]
-        assert len(mesh.property_groups) == 3
+        assert len(mesh.property_groups) == 5
         assert len(mesh.property_groups[0].properties) == 2
         assert mesh.property_groups[1].property_group_type == GroupTypeEnum.DIPDIR