diff --git a/simpeg_drivers-assets/uijson/plate_sweep.ui.json b/simpeg_drivers-assets/uijson/plate_sweep.ui.json index 8a1853b7..9d72ef1f 100644 --- a/simpeg_drivers-assets/uijson/plate_sweep.ui.json +++ b/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": "", diff --git a/simpeg_drivers/plate_simulation/sweep/uijson.py b/simpeg_drivers/plate_simulation/sweep/uijson.py index a9a3e866..73fa9297 100644 --- a/simpeg_drivers/plate_simulation/sweep/uijson.py +++ b/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 diff --git a/tests/plate_simulation/runtest/sweep_test.py b/tests/plate_simulation/runtest/sweep_test.py index 9fa38827..01ceb818 100644 --- a/tests/plate_simulation/runtest/sweep_test.py +++ b/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)