Fix dq test to actually produce dq flags

spacetelescope · Oct 6, 2023 · 9d1e54a · 9d1e54a
1 parent afe4712
commit 9d1e54a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_jump_cas22.py b/tests/test_jump_cas22.py
@@ -16,6 +16,7 @@
 FLUX = 100
 JUMP_VALUE = 10_000
 CHI2_TOL = 0.03
+GOOD_PROB = 0.7
 
 
 @pytest.fixture(scope="module")
@@ -334,7 +335,7 @@ def test_fit_ramps(detector_data, use_jump, use_dq):
     """
     resultants, read_noise, read_pattern = detector_data
     dq = (
-        (RNG.uniform(size=resultants.shape) > 1).astype(np.int32) if use_dq else
+        (RNG.uniform(size=resultants.shape) > GOOD_PROB).astype(np.int32) if use_dq else
         np.zeros(resultants.shape, dtype=np.int32)
     )