lint and fix 1 bug

Author: tasansal

tests/unit/v1/templates/test_seismic_2d_poststack.py

@@ -96,10 +96,7 @@ def test_configuration(self, data_domain: SeismicDataDomain) -> None:
 
         # Verify dataset attributes
         attrs = t._load_dataset_attributes()
-        assert attrs == {
-            "surveyType": "2D",
-            "gatherType": "stacked",
-        }
+        assert attrs == {"surveyType": "2D", "gatherType": "stacked"}
 
         assert t.default_variable_name == "amplitude"
 

tests/unit/v1/templates/test_seismic_2d_prestack_cdp.py

@@ -110,10 +110,7 @@ def test_configuration(self, data_domain: SeismicDataDomain, gather_domain: CdpG
 
         # Verify prestack CDP attributes
         attrs = t._load_dataset_attributes()
-        assert attrs == {
-            "surveyType": "2D",
-            "gatherType": "cdp",
-        }
+        assert attrs == {"surveyType": "2D", "gatherType": "cdp"}
         assert t.default_variable_name == "amplitude"
 
     def test_build_dataset(

tests/unit/v1/templates/test_seismic_3d_poststack.py

@@ -113,10 +113,7 @@ def test_configuration(self, data_domain: SeismicDataDomain) -> None:
 
         # Verify dataset attributes
         attrs = t._load_dataset_attributes()
-        assert attrs == {
-            "surveyType": "3D",
-            "gatherType": "stacked",
-        }
+        assert attrs == {"surveyType": "3D", "gatherType": "stacked"}
         assert t.default_variable_name == "amplitude"
 
     def test_build_dataset(self, data_domain: SeismicDataDomain, structured_headers: StructuredType) -> None:

tests/unit/v1/templates/test_seismic_3d_prestack_cdp.py

@@ -119,10 +119,7 @@ def test_configuration(self, data_domain: SeismicDataDomain, gather_domain: CdpG
 
         # Verify prestack CDP attributes
         attrs = t._load_dataset_attributes()
-        assert attrs == {
-            "surveyType": "3D",
-            "gatherType": "cdp",
-        }
+        assert attrs == {"surveyType": "3D", "gatherType": "cdp"}
         assert t.default_variable_name == "amplitude"
 
     def test_build_dataset(

tests/unit/v1/templates/test_seismic_3d_prestack_coca.py

@@ -131,10 +131,7 @@ def test_configuration(self, data_domain: SeismicDataDomain) -> None:
 
         # Verify dataset attributes
         attrs = t._load_dataset_attributes()
-        assert attrs == {
-            "surveyType": "3D",
-            "gatherType": "cdp_coca",
-        }
+        assert attrs == {"surveyType": "3D", "gatherType": "common_offset_common_azimuth"}
         assert t.default_variable_name == "amplitude"
 
     def test_build_dataset(self, data_domain: SeismicDataDomain, structured_headers: StructuredType) -> None:
@@ -150,7 +147,7 @@ def test_build_dataset(self, data_domain: SeismicDataDomain, structured_headers:
 
         assert dataset.metadata.name == "Permian Basin 3D CDP Coca Gathers"
         assert dataset.metadata.attributes["surveyType"] == "3D"
-        assert dataset.metadata.attributes["gatherType"] == "cdp_coca"
+        assert dataset.metadata.attributes["gatherType"] == "common_offset_common_azimuth"
 
         _validate_coordinates_headers_trace_mask(dataset, structured_headers, data_domain)
 

tests/unit/v1/templates/test_seismic_3d_prestack_shot.py

@@ -143,10 +143,7 @@ def test_configuration(self) -> None:
 
         # Verify prestack shot attributes
         attrs = t._load_dataset_attributes()
-        assert attrs == {
-            "surveyType": "3D",
-            "gatherType": "common_shot",
-        }
+        assert attrs == {"surveyType": "3D", "gatherType": "common_shot"}
 
         assert t.name == "PreStackShotGathers3DTime"