update tests

src/stcal/jump/jump.py

@@ -540,7 +540,7 @@ def flag_large_events(
 
     Returns
     -------
-    Nothing, gdq array is modified.
+    total Snowballs
 
     """
     log.info("Flagging Snowballs")

tests/test_jump.py

@@ -150,16 +150,16 @@ def test_flag_large_events_withsnowball():
 def test_flag_large_events_groupedsnowball():
     cube = np.zeros(shape=(1, 5, 7, 7), dtype=np.uint8)
     # cross of saturation surrounding by jump -> snowball
-    cube[0, 1, :, :] = DQFLAGS["JUMP_DET"]
+#    cube[0, 1, :, :] = DQFLAGS["JUMP_DET"]
+#    cube[0, 2, :, :] = DQFLAGS["JUMP_DET"]
+    cube[0, 2, 1:6, 1:6] = DQFLAGS["JUMP_DET"]
+    cube[0, 1, 1:6, 1:6] = DQFLAGS["JUMP_DET"]
+
     cube[0, 2, 3, 3] = DQFLAGS["SATURATED"]
     cube[0, 2, 2, 3] = DQFLAGS["SATURATED"]
     cube[0, 2, 3, 4] = DQFLAGS["SATURATED"]
     cube[0, 2, 4, 3] = DQFLAGS["SATURATED"]
     cube[0, 2, 3, 2] = DQFLAGS["SATURATED"]
-    cube[0, 2, 1, 1:6] = DQFLAGS["JUMP_DET"]
-    cube[0, 2, 5, 1:6] = DQFLAGS["JUMP_DET"]
-    cube[0, 2, 1:6, 1] = DQFLAGS["JUMP_DET"]
-    cube[0, 2, 1:6, 5] = DQFLAGS["JUMP_DET"]
     flag_large_events(
         cube,
         DQFLAGS["JUMP_DET"],
@@ -172,9 +172,6 @@ def test_flag_large_events_groupedsnowball():
         min_sat_radius_extend=0.5,
         sat_expand=1.1,
     )
-    #    assert cube[0, 1, 2, 2] == 0
-    #    assert cube[0, 1, 3, 5] == 0
-    assert cube[0, 2, 0, 0] == 0
     assert cube[0, 2, 1, 0] == DQFLAGS["JUMP_DET"]  # Jump was extended
     assert cube[0, 2, 2, 2] == DQFLAGS["SATURATED"]  # Saturation was extended