split test_nirspec_ifu_spec3_regtest

spacetelescope · Sep 20, 2024 · b7bc19c · b7bc19c
1 parent 217d660
commit b7bc19c
Show file tree

Hide file tree

Showing 2 changed files with 54 additions and 53 deletions.
diff --git a/jwst/regtest/test_nirspec_ifu_spec3.py b/jwst/regtest/test_nirspec_ifu_spec3.py
@@ -3,11 +3,6 @@
 
 from jwst.regtest import regtestdata as rt
 
-# Define artifactory source and truth
-INPUT_PATH = 'nirspec/ifu'
-TRUTH_PATH = 'truth/test_nirspec_ifu'
-TRUTH_PATH_PIXEL_REPLACE = 'truth/test_nirspec_ifu_pixel_replace'
-
 
 @pytest.fixture(scope='module')
 def run_spec3_multi(rtdata_module):
@@ -31,27 +26,6 @@ def run_spec3_multi(rtdata_module):
     return rtdata
 
 
-@pytest.fixture(scope='module')
-def run_spec3_multi_pixel_replace(rtdata_module):
-    """Run Spec3Pipeline with pixel replacement"""
-    rtdata = rtdata_module
-
-    step_params = {
-        'input_path': 'nirspec/ifu/jw01249-o005_20230622t074431_rp_spec3_00001_asn.json',
-        'step': 'calwebb_spec3',
-        'args': {
-            '--steps.pixel_replace.skip=false',
-            '--steps.pixel_replace.save_results=true',
-            '--steps.pixel_replace.algorithm=mingrad',
-            '--steps.cube_build.save_results=true',
-            '--steps.extract_1d.save_results=true',
-        }
-    }
-
-    rtdata = rt.run_step_from_dict(rtdata, **step_params)
-    return rtdata
-
-
 @pytest.mark.slow
 @pytest.mark.bigdata
 @pytest.mark.parametrize(
@@ -73,32 +47,6 @@ def test_spec3_multi(run_spec3_multi, fitsdiff_default_kwargs, output):
     """Regression test matching output files"""
     rt.is_like_truth(
         run_spec3_multi, fitsdiff_default_kwargs, output,
-        truth_path=TRUTH_PATH,
-        is_suffix=False
-    )
-
-
-@pytest.mark.slow
-@pytest.mark.bigdata
-@pytest.mark.parametrize(
-    'output',
-    [
-        'jw01249005001_03101_00001_nrs1_o005_pixel_replace.fits',
-        'jw01249005001_03101_00001_nrs2_o005_pixel_replace.fits',
-        'jw01249005001_03101_00002_nrs1_o005_pixel_replace.fits',
-        'jw01249005001_03101_00002_nrs2_o005_pixel_replace.fits',
-        'jw01249005001_03101_00003_nrs1_o005_pixel_replace.fits',
-        'jw01249005001_03101_00003_nrs2_o005_pixel_replace.fits',
-        'jw01249005001_03101_00004_nrs1_o005_pixel_replace.fits',
-        'jw01249005001_03101_00004_nrs2_o005_pixel_replace.fits',
-        'jw01249-o005_t001_nirspec_rp_test_g395h-f290lp_s3d.fits',
-        'jw01249-o005_t001_nirspec_rp_test_g395h-f290lp_x1d.fits',
-    ]
-)
-def test_spec3_multi_pixel_replace(run_spec3_multi_pixel_replace, fitsdiff_default_kwargs, output):
-    """Regression test matching output files"""
-    rt.is_like_truth(
-        run_spec3_multi_pixel_replace, fitsdiff_default_kwargs, output,
-        truth_path=TRUTH_PATH_PIXEL_REPLACE,
+        truth_path='truth/test_nirspec_ifu',
         is_suffix=False
     )
diff --git a/jwst/regtest/test_nirspec_ifu_spec3_pixelreplace.py b/jwst/regtest/test_nirspec_ifu_spec3_pixelreplace.py
@@ -0,0 +1,53 @@
+"""Regression tests for NIRSpec IFU"""
+import pytest
+
+from jwst.regtest import regtestdata as rt
+
+TRUTH_PATH_PIXEL_REPLACE = 'truth/test_nirspec_ifu_pixel_replace'
+
+
+@pytest.fixture(scope='module')
+def run_spec3_multi_pixel_replace(rtdata_module):
+    """Run Spec3Pipeline with pixel replacement"""
+    rtdata = rtdata_module
+
+    step_params = {
+        'input_path': 'nirspec/ifu/jw01249-o005_20230622t074431_rp_spec3_00001_asn.json',
+        'step': 'calwebb_spec3',
+        'args': {
+            '--steps.pixel_replace.skip=false',
+            '--steps.pixel_replace.save_results=true',
+            '--steps.pixel_replace.algorithm=mingrad',
+            '--steps.cube_build.save_results=true',
+            '--steps.extract_1d.save_results=true',
+        }
+    }
+
+    rtdata = rt.run_step_from_dict(rtdata, **step_params)
+    return rtdata
+
+
+@pytest.mark.slow
+@pytest.mark.bigdata
+@pytest.mark.parametrize(
+    'output',
+    [
+        'jw01249005001_03101_00001_nrs1_o005_pixel_replace.fits',
+        'jw01249005001_03101_00001_nrs2_o005_pixel_replace.fits',
+        'jw01249005001_03101_00002_nrs1_o005_pixel_replace.fits',
+        'jw01249005001_03101_00002_nrs2_o005_pixel_replace.fits',
+        'jw01249005001_03101_00003_nrs1_o005_pixel_replace.fits',
+        'jw01249005001_03101_00003_nrs2_o005_pixel_replace.fits',
+        'jw01249005001_03101_00004_nrs1_o005_pixel_replace.fits',
+        'jw01249005001_03101_00004_nrs2_o005_pixel_replace.fits',
+        'jw01249-o005_t001_nirspec_rp_test_g395h-f290lp_s3d.fits',
+        'jw01249-o005_t001_nirspec_rp_test_g395h-f290lp_x1d.fits',
+    ]
+)
+def test_spec3_multi_pixel_replace(run_spec3_multi_pixel_replace, fitsdiff_default_kwargs, output):
+    """Regression test matching output files"""
+    rt.is_like_truth(
+        run_spec3_multi_pixel_replace, fitsdiff_default_kwargs, output,
+        truth_path='truth/test_nirspec_ifu_pixel_replace',
+        is_suffix=False
+    )