pre-commit

Author: BrianMichell

src/mdio/builder/templates/base.py

@@ -81,6 +81,9 @@ def build_dataset(
 
         Returns:
             Dataset: The constructed dataset
+
+        Raises:
+            ValueError: If coordinate already exists from subclass override.
         """
         self._dim_sizes = sizes
 

tests/integration/test_import_streamer_grid_overrides.py

@@ -27,7 +27,8 @@
 os.environ["MDIO__IMPORT__SAVE_SEGY_FILE_HEADER"] = "true"
 
 
-# @pytest.mark.parametrize("grid_override", [{"NonBinned": True, "chunksize": 4, "non_binned_dims": ["channel"]}, {"HasDuplicates": True}])
+# @pytest.mark.parametrize("grid_override", [{"NonBinned": True, "chunksize": 4, "non_binned_dims": ["channel"]},
+#                                          {"HasDuplicates": True}])
 @pytest.mark.parametrize("grid_override", [{"NonBinned": True, "chunksize": 4, "non_binned_dims": ["channel"]}])
 @pytest.mark.parametrize("chan_header_type", [StreamerShotGeometryType.C])
 class TestImport4DNonReg:
@@ -44,14 +45,13 @@ def test_import_4d_segy(  # noqa: PLR0913
         segy_spec: SegySpec = get_segy_mock_4d_spec()
         segy_path = segy_mock_4d_shots[chan_header_type]
 
-        path = "tmp.mdio"
         print(f"Running test with grid override: {grid_override}")
 
         segy_to_mdio(
             segy_spec=segy_spec,
             mdio_template=TemplateRegistry().get("PreStackShotGathers3DTime"),
             input_path=segy_path,
-            output_path=path,
+            output_path=zarr_tmp,
             overwrite=True,
             grid_overrides=grid_override,
         )
@@ -62,7 +62,7 @@ def test_import_4d_segy(  # noqa: PLR0913
         cables = [0, 101, 201, 301]
         receivers_per_cable = [1, 5, 7, 5]
 
-        ds = open_mdio(path)
+        ds = open_mdio(zarr_tmp)
 
         assert ds["segy_file_header"].attrs["binaryHeader"]["samples_per_trace"] == num_samples
         assert ds.attrs["attributes"]["gridOverrides"] == grid_override