diff --git a/romancal/regtest/test_wfi_pipeline.py b/romancal/regtest/test_wfi_pipeline.py index 330b91baa..d734dfb9e 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 @@ -509,3 +511,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