diff --git a/src/stcal/jump/jump.py b/src/stcal/jump/jump.py index 280ddb42..d197ce87 100644 --- a/src/stcal/jump/jump.py +++ b/src/stcal/jump/jump.py @@ -579,6 +579,7 @@ def flag_large_events( sat_flag, jump_flag, expansion=expand_factor, + num_grps_masked=0, max_extended_radius=max_extended_radius, ) return gdq, total_snowballs @@ -681,8 +682,6 @@ def extend_ellipses( jump_ellipse[saty, satx] = 0 out_gdq_cube[intg, flg_grp, :, :] = np.bitwise_or(gdq_cube[intg, flg_grp, :, :], jump_ellipse) diff_cube = out_gdq_cube - gdq_cube - print("") - print("extend ellipse corner", out_gdq_cube[0, 2, 0, 0]) return out_gdq_cube, num_ellipses def find_last_grp(grp, ngrps, num_grps_masked): diff --git a/tests/test_jump.py b/tests/test_jump.py index 7de81085..995a5367 100644 --- a/tests/test_jump.py +++ b/tests/test_jump.py @@ -172,11 +172,9 @@ def test_flag_large_events_groupedsnowball(): min_sat_radius_extend=0.5, sat_expand=1.1, ) - print("") - print("final corner", cube[0, 2, 0, 0]) # assert cube[0, 1, 2, 2] == 0 # assert cube[0, 1, 3, 5] == 0 - assert cube[0, 2, 0, 0] == DQFLAGS["JUMP_DET"] # Jump was extended + 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