Merge pull request #314 from MiraGeoscience/GEOPY-2587

GEOPY-2587: Add dip_direction to list of supported variables by plate_sweep

Author: benk-mira

simpeg_drivers-assets/uijson/plate_sweep.ui.json

@@ -193,6 +193,26 @@
         "label": "number of samples",
         "value": 1
     },
+    "dip_direction_start": {
+        "main": true,
+        "group": "Dip Dir",
+        "groupOptional": true,
+        "enabled": false,
+        "label": "starting",
+        "value": 60.0
+    },
+    "dip_direction_stop": {
+        "main": true,
+        "group": "Dip Dir",
+        "label": "ending",
+        "value": 60.0
+    },
+    "dip_direction_count": {
+        "main": true,
+        "group": "Dip Dir",
+        "label": "number of samples",
+        "value": 1
+    },
     "out_group": {
         "label": "SimPEGGroup",
         "value": "",

simpeg_drivers/plate_simulation/sweep/uijson.py

@@ -41,5 +41,8 @@ class PlateSweepUIJson(BaseUIJson):
     dip_start: FloatForm
     dip_stop: FloatForm
     dip_count: IntegerForm
+    dip_direction_start: FloatForm
+    dip_direction_stop: FloatForm
+    dip_direction_count: IntegerForm
     out_group: GroupForm | None
     workdir: StringForm | None

tests/plate_simulation/runtest/sweep_test.py

@@ -8,7 +8,6 @@
 #                                                                                   '
 # '''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''
 
-import os
 
 from geoh5py import Workspace
 from geoh5py.groups import SimPEGGroup
@@ -72,8 +71,7 @@ def setup_plate_sweep(workspace) -> SimPEGGroup:
 
 def test_sweep(tmp_path):
     workdir = tmp_path / "my_workdir"
-    os.mkdir(workdir)
-    assert workdir.exists()
+
     with Workspace.create(tmp_path / "test.geoh5") as ws:
         plate_simulation = setup_plate_sweep(ws)
 
@@ -95,6 +93,8 @@ def test_sweep(tmp_path):
         ifile.write_ui_json(name="plate_sweep.ui.json", path=tmp_path)
     PlateSweepDriver.start(tmp_path / "plate_sweep.ui.json")
 
+    assert workdir.exists()
+
     with Workspace(tmp_path / "test.geoh5"):
         ifile = InputFile.read_ui_json(tmp_path / "plate_sweep.ui.json")
         ifile.set_data_value("background_count", 3)