Loosen saturation test.

spacetelescope · Dec 20, 2024 · 97c3349 · 97c3349
1 parent b2aa8e0
commit 97c3349
Showing 1 changed file with 4 additions and 1 deletion.
diff --git 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):