From 73d886a3842f0a9d1b6d2cc0d4f6005c8ab430c1 Mon Sep 17 00:00:00 2001
From: D Davis <49163225+ddavis-stsci@users.noreply.github.com>
Date: Tue, 21 Nov 2023 11:14:03 -0500
Subject: [PATCH] rcal-457 Add test for spectral preview files (#1021)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
---
 romancal/regtest/test_wfi_pipeline.py | 44 +++++++++++++++++++++++++++
 1 file changed, 44 insertions(+)

diff --git a/romancal/regtest/test_wfi_pipeline.py b/romancal/regtest/test_wfi_pipeline.py
index d72e7ec10..94d79866a 100644
--- a/romancal/regtest/test_wfi_pipeline.py
+++ b/romancal/regtest/test_wfi_pipeline.py
@@ -1,5 +1,7 @@
 """ Roman tests for flat field correction """
 import copy
+import os
+import subprocess
 
 import numpy as np
 import pytest
@@ -515,3 +517,45 @@ def test_processing_pipeline_all_saturated(rtdata, ignore_asdf_paths):
     assert model.meta.cal_step.assign_wcs == "SKIPPED"
     assert model.meta.cal_step.flat_field == "SKIPPED"
     assert model.meta.cal_step.photom == "SKIPPED"
+
+
+@pytest.mark.bigdata
+@pytest.mark.soctests
+@metrics_logger("DMS281")
+def test_level2_image_preview(rtdata, ignore_asdf_paths):
+    """Tests for flat field image preview requirements DMS 281"""
+    input_data = "r0000501001001001001_01101_0001_WFI01_cal.asdf"
+    rtdata.get_data(f"WFI/grism/{input_data}")
+    rtdata.input = input_data
+
+    # Test Pipeline
+    output = "r0000501001001001001_01101_0001_WFI01_cal_thumb.png"
+    rtdata.output = output
+    args = [
+        "roman_static_preview",
+        "thumbnail",
+        rtdata.input,
+    ]
+    subprocess.run(args)
+    assert os.path.exists(rtdata.output) is True
+
+
+@pytest.mark.bigdata
+@pytest.mark.soctests
+@metrics_logger("DMS278")
+def test_level2_grism_preview(rtdata, ignore_asdf_paths):
+    """Tests for flat field grism preview requirements DMS 278"""
+    input_data = "r0000501001001001001_01101_0001_WFI01_cal.asdf"
+    rtdata.get_data(f"WFI/grism/{input_data}")
+    rtdata.input = input_data
+
+    # Test Pipeline
+    output = "r0000501001001001001_01101_0001_WFI01_cal_thumb.png"
+    rtdata.output = output
+    args = [
+        "roman_static_preview",
+        "thumbnail",
+        rtdata.input,
+    ]
+    subprocess.run(args)
+    assert os.path.exists(rtdata.output) is True