From 97c334922ea8f32e1a86123f3c0a68f50f95fd29 Mon Sep 17 00:00:00 2001 From: Eddie Schlafly Date: Fri, 20 Dec 2024 11:06:54 -0500 Subject: [PATCH] Loosen saturation test. --- romancal/saturation/tests/test_saturation.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/romancal/saturation/tests/test_saturation.py b/romancal/saturation/tests/test_saturation.py index 6d080af45..4d5e0a5ba 100644 --- a/romancal/saturation/tests/test_saturation.py +++ b/romancal/saturation/tests/test_saturation.py @@ -83,7 +83,10 @@ def test_read_pattern_saturation_flagging(setup_wfi_datamodels): output = flag_saturation(ramp, satmap) # Make sure that groups after the third get flagged - assert np.all(output.groupdq[2:, 5, 5] == group.SATURATED) + # To accommodate https://github.com/spacetelescope/stcal/pull/321 + # this test was loosed to be SATURATED | DO_NOT_USE + assert np.all( + (output.groupdq[2:, 5, 5] & (group.SATURATED | group.DO_NOT_USE)) != 0) def test_ad_floor_flagging(setup_wfi_datamodels):