Update tests/test_jump.py

spacetelescope · Jan 31, 2024 · a3cbe27 · a3cbe27
1 parent e4017c3
commit a3cbe27
Showing 1 changed file with 9 additions and 9 deletions.
diff --git a/tests/test_jump.py b/tests/test_jump.py
@@ -356,12 +356,12 @@ def test_calc_num_slices():
 
 
 def test_find_last_grp():
-    assert (find_last_grp(grp=5, ngrps=7, num_grps_masked_after=0) == 6)
-    assert (find_last_grp(grp=5, ngrps=7, num_grps_masked_after=2) == 7)
-    assert (find_last_grp(grp=5, ngrps=7, num_grps_masked_after=3) == 7)
-    assert (find_last_grp(grp=5, ngrps=6, num_grps_masked_after=1) == 6)
-    assert (find_last_grp(grp=5, ngrps=6, num_grps_masked_after=0) == 6)
-    assert (find_last_grp(grp=5, ngrps=6, num_grps_masked_after=2) == 6)
-    assert (find_last_grp(grp=5, ngrps=8, num_grps_masked_after=0) == 6)
-    assert (find_last_grp(grp=5, ngrps=8, num_grps_masked_after=1) == 7)
-    assert (find_last_grp(grp=5, ngrps=8, num_grps_masked_after=2) == 8)
+    assert (find_last_grp(grp=5, ngrps=7, num_grps_masked=0) == 6)
+    assert (find_last_grp(grp=5, ngrps=7, num_grps_masked=2) == 7)
+    assert (find_last_grp(grp=5, ngrps=7, num_grps_masked=3) == 7)
+    assert (find_last_grp(grp=5, ngrps=6, num_grps_masked=1) == 6)
+    assert (find_last_grp(grp=5, ngrps=6, num_grps_masked=0) == 6)
+    assert (find_last_grp(grp=5, ngrps=6, num_grps_masked=2) == 6)
+    assert (find_last_grp(grp=5, ngrps=8, num_grps_masked=0) == 6)
+    assert (find_last_grp(grp=5, ngrps=8, num_grps_masked=1) == 7)
+    assert (find_last_grp(grp=5, ngrps=8, num_grps_masked=2) == 8)