Add regression test for wfss_contam step for NIRISS (#9830)

Co-authored-by: Melanie Clarke <[email protected]>

Author: emolter

jwst/regtest/test_niriss_wfss_contam.py

@@ -0,0 +1,45 @@
+"""Regression test for WFSS contam correction of NIRISS data"""
+
+import pytest
+
+from jwst.regtest import regtestdata as rt
+
+# Mark all tests in this module
+pytestmark = [pytest.mark.bigdata]
+
+
+@pytest.fixture(scope="module")
+def run_wfss_contam(rtdata_module, resource_tracker):
+    """Run the wfss_contam step"""
+    rtdata = rtdata_module
+
+    # Get input data files
+    rtdata.get_data("niriss/wfss/jw02738-o001_t001_niriss_clear-f200w_i2d.fits")
+    rtdata.get_data("niriss/wfss/jw02738-o001_t001_niriss_clear-f200w_segm.fits")
+    rtdata.get_data("niriss/wfss/jw02738-o001_t001_niriss_clear-f200w_cat.ecsv")
+    rtdata.get_data("niriss/wfss/jw02738001001_04201_00004_nis_srctype.fits")
+
+    # Run the step
+    step_params = {
+        "step": "jwst.wfss_contam.WfssContamStep",
+        "args": [
+            "--save_simulated_image=True",
+            "--save_contam_images=True",
+            "--skip=False",
+        ],
+    }
+    with resource_tracker.track():
+        rtdata = rt.run_step_from_dict(rtdata, **step_params)
+    return rtdata
+
+
+def test_log_tracked_resources_tsimg(log_tracked_resources, run_wfss_contam):
+    log_tracked_resources()
+
+
+@pytest.mark.parametrize("suffix", ["simul", "simul_slits", "contam", "wfsscontamstep"])
+def test_niriss_wfss_contam(run_wfss_contam, fitsdiff_default_kwargs, suffix):
+    """Regression test for wfss_contam applied to NIRISS WFSS data"""
+    rt.is_like_truth(
+        run_wfss_contam, fitsdiff_default_kwargs, suffix, "truth/test_niriss_wfss_contam"
+    )