1313from mdio .builder .schemas .v1 .units import LengthUnitModel
1414from mdio .builder .schemas .v1 .units import TimeUnitEnum
1515from mdio .builder .schemas .v1 .units import TimeUnitModel
16- from mdio .builder .templates .seismic_3d_prestack_field_records import Seismic3DPreStackFieldRecordsTemplate
16+ from mdio .builder .templates .seismic_3d_prestack_streamer_field_records import Seismic3DPreStackStreamerFieldRecordsTemplate
1717
1818UNITS_METER = LengthUnitModel (length = LengthUnitEnum .METER )
1919UNITS_SECOND = TimeUnitModel (time = TimeUnitEnum .SECOND )
@@ -79,15 +79,15 @@ def _validate_coordinates_headers_trace_mask(dataset: Dataset, headers: Structur
7979 assert coord .metadata .units_v1 .length == LengthUnitEnum .METER
8080
8181
82- class TestSeismic3DPreStackFieldRecordsTemplate :
83- """Unit tests for Seismic3DPreStackFieldRecordsTemplate ."""
82+ class TestSeismic3DPreStackStreamerFieldRecordsTemplate :
83+ """Unit tests for Seismic3DPreStackStreamerFieldRecordsTemplate ."""
8484
8585 def test_configuration (self ) -> None :
86- """Unit tests for Seismic3DPreStackFieldRecordsTemplate ."""
87- t = Seismic3DPreStackFieldRecordsTemplate (data_domain = "time" )
86+ """Unit tests for Seismic3DPreStackStreamerFieldRecordsTemplate ."""
87+ t = Seismic3DPreStackStreamerFieldRecordsTemplate (data_domain = "time" )
8888
8989 # Template attributes
90- assert t .name == "PreStackFieldRecords3DTime "
90+ assert t .name == "PreStackStreamerFieldRecords3DTime "
9191 assert t ._dim_names == ("shot_line" , "gun" , "shot_point" , "cable" , "channel" , "time" )
9292 assert t ._physical_coord_names == ("source_coord_x" , "source_coord_y" , "group_coord_x" , "group_coord_y" )
9393 # TODO(Anyone): Disable chunking in time domain when support is merged.
@@ -104,17 +104,17 @@ def test_configuration(self) -> None:
104104 assert t .default_variable_name == "amplitude"
105105
106106 def test_build_dataset (self , structured_headers : StructuredType ) -> None :
107- """Unit tests for Seismic3DPreStackFieldRecordsTemplate build."""
108- t = Seismic3DPreStackFieldRecordsTemplate (data_domain = "time" )
107+ """Unit tests for Seismic3DPreStackStreamerFieldRecordsTemplate build."""
108+ t = Seismic3DPreStackStreamerFieldRecordsTemplate (data_domain = "time" )
109109 t .add_units ({"source_coord_x" : UNITS_METER , "source_coord_y" : UNITS_METER }) # spatial domain units
110110 t .add_units ({"group_coord_x" : UNITS_METER , "group_coord_y" : UNITS_METER }) # spatial domain units
111111 t .add_units ({"time" : UNITS_SECOND }) # data domain units
112112
113113 dataset = t .build_dataset (
114- "North Sea 3D Field Records" , sizes = (1 , 3 , 256 , 512 , 24 , 2048 ), header_dtype = structured_headers
114+ "North Sea 3D Streamer Field Records" , sizes = (1 , 3 , 256 , 512 , 24 , 2048 ), header_dtype = structured_headers
115115 )
116116
117- assert dataset .metadata .name == "North Sea 3D Field Records"
117+ assert dataset .metadata .name == "North Sea 3D Streamer Field Records"
118118 assert dataset .metadata .attributes ["surveyDimensionality" ] == "3D"
119119 assert dataset .metadata .attributes ["ensembleType" ] == "shot_point"
120120 assert dataset .metadata .attributes ["processingStage" ] == "pre-stack"
@@ -139,6 +139,6 @@ def test_build_dataset(self, structured_headers: StructuredType) -> None:
139139@pytest .mark .parametrize ("data_domain" , ["Time" , "TiME" ])
140140def test_domain_case_handling (data_domain : str ) -> None :
141141 """Test that domain parameter handles different cases correctly."""
142- template = Seismic3DPreStackFieldRecordsTemplate (data_domain = data_domain )
142+ template = Seismic3DPreStackStreamerFieldRecordsTemplate (data_domain = data_domain )
143143 assert template ._data_domain == data_domain .lower ()
144144 assert template .name .endswith (data_domain .capitalize ())
0 commit comments