Rename template name

Author: BrianMichell

src/mdio/builder/template_registry.py

@@ -26,7 +26,7 @@
 from mdio.builder.templates.seismic_3d_poststack import Seismic3DPostStackTemplate
 from mdio.builder.templates.seismic_3d_prestack_cdp import Seismic3DPreStackCDPTemplate
 from mdio.builder.templates.seismic_3d_prestack_coca import Seismic3DPreStackCocaTemplate
-from mdio.builder.templates.seismic_3d_prestack_field_records import SeismicPreStackTemplate
+from mdio.builder.templates.seismic_3d_prestack_field_records import Seismic3DPreStackFieldRecordsTemplate
 from mdio.builder.templates.seismic_3d_prestack_shot import Seismic3DPreStackShotTemplate
 
 if TYPE_CHECKING:
@@ -134,7 +134,7 @@ def _register_default_templates(self) -> None:
         self.register(Seismic3DPreStackCocaTemplate("depth"))
 
         # Field (shot) data
-        self.register(SeismicPreStackTemplate("time"))
+        self.register(Seismic3DPreStackFieldRecordsTemplate("time"))
         self.register(Seismic2DPreStackShotTemplate("time"))
         self.register(Seismic3DPreStackShotTemplate("time"))
 

src/mdio/builder/templates/seismic_3d_prestack_field_records.py

@@ -8,7 +8,7 @@
 from mdio.builder.templates.types import SeismicDataDomain
 
 
-class SeismicPreStackTemplate(AbstractDatasetTemplate):
+class Seismic3DPreStackFieldRecordsTemplate(AbstractDatasetTemplate):
     """Seismic pre-stack time Dataset template.
 
     This should be used for both 2D and 3D datasets. Common-shot or common-channel datasets
@@ -34,7 +34,7 @@ def __init__(self, data_domain: SeismicDataDomain):
 
     @property
     def _name(self) -> str:
-        return f"PreStackGathers3D{self._data_domain.capitalize()}"
+        return f"PreStackFieldRecords3D{self._data_domain.capitalize()}"
 
     def _load_dataset_attributes(self) -> dict[str, Any]:
         return {

tests/integration/test_import_streamer_grid_overrides.py

@@ -178,7 +178,7 @@ def test_import_6d_segy(  # noqa: PLR0913
 
         segy_to_mdio(
             segy_spec=segy_spec,
-            mdio_template=TemplateRegistry().get("PreStackGathers3DTime"),  # Placeholder for the template
+            mdio_template=TemplateRegistry().get("PreStackFieldRecords3DTime"),
             input_path=segy_path,
             output_path=zarr_tmp,
             overwrite=True,

tests/unit/v1/templates/test_seismic_3d_prestack_field_records.py

@@ -13,7 +13,7 @@
 from mdio.builder.schemas.v1.units import LengthUnitModel
 from mdio.builder.schemas.v1.units import TimeUnitEnum
 from mdio.builder.schemas.v1.units import TimeUnitModel
-from mdio.builder.templates.seismic_3d_prestack_field_records import SeismicPreStackTemplate
+from mdio.builder.templates.seismic_3d_prestack_field_records import Seismic3DPreStackFieldRecordsTemplate
 
 UNITS_METER = LengthUnitModel(length=LengthUnitEnum.METER)
 UNITS_SECOND = TimeUnitModel(time=TimeUnitEnum.SECOND)
@@ -143,15 +143,15 @@ def _validate_coordinates_headers_trace_mask(dataset: Dataset, headers: Structur
     assert group_coord_y.metadata.units_v1.length == LengthUnitEnum.METER
 
 
-class TestSeismic3DPreStackShotTemplate:
+class TestSeismic3DPreStackFieldRecordsTemplate:
     """Unit tests for SeismicPreStackTemplate."""
 
     def test_configuration(self) -> None:
         """Unit tests for SeismicPreStackTemplate in time domain."""
-        t = SeismicPreStackTemplate(data_domain="time")
+        t = Seismic3DPreStackFieldRecordsTemplate(data_domain="time")
 
         # Template attributes for prestack shot
-        assert t.name == "PreStackGathers3DTime"
+        assert t.name == "PreStackFieldRecords3DTime"
         assert t.default_variable_name == "amplitude"
         assert t.trace_domain == "time"
         assert t.spatial_dimension_names == ("shot_line", "gun", "shot_point", "cable", "channel")
@@ -177,11 +177,11 @@ def test_configuration(self) -> None:
         assert attrs == {"surveyDimensionality": "3D", "ensembleType": "shot_point", "processingStage": "pre-stack"}
         assert t.default_variable_name == "amplitude"
 
-        assert t.name == "PreStackGathers3DTime"
+        assert t.name == "PreStackFieldRecords3DTime"
 
     def test_build_dataset(self, structured_headers: StructuredType) -> None:
         """Unit tests for SeismicPreStackTemplate build in time domain."""
-        t = SeismicPreStackTemplate(data_domain="time")
+        t = Seismic3DPreStackFieldRecordsTemplate(data_domain="time")
         t.add_units({"source_coord_x": UNITS_METER, "source_coord_y": UNITS_METER})  # spatial domain units
         t.add_units({"group_coord_x": UNITS_METER, "group_coord_y": UNITS_METER})  # spatial domain units
         t.add_units({"time": UNITS_SECOND})  # data domain units
@@ -215,6 +215,6 @@ def test_build_dataset(self, structured_headers: StructuredType) -> None:
 @pytest.mark.parametrize("data_domain", ["Time", "TiME"])
 def test_domain_case_handling(data_domain: str) -> None:
     """Test that domain parameter handles different cases correctly."""
-    template = SeismicPreStackTemplate(data_domain=data_domain)
+    template = Seismic3DPreStackFieldRecordsTemplate(data_domain=data_domain)
     assert template._data_domain == data_domain.lower()
     assert template.name.endswith(data_domain.capitalize())

tests/unit/v1/templates/test_template_registry.py

@@ -31,7 +31,7 @@
     "PreStackCdpAngleGathers3DDepth",
     "PreStackCocaGathers3DTime",
     "PreStackCocaGathers3DDepth",
-    "PreStackGathers3DTime",
+    "PreStackFieldRecords3DTime",
     "PreStackShotGathers2DTime",
     "PreStackShotGathers3DTime",
 ]