Skip to content

Commit 4d5df93

Browse files
committed
Fix test_remove_stripe_fw_calc_mem
1 parent 1653ef2 commit 4d5df93

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/test_prep/test_stripe.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -86,14 +86,14 @@ def ensure_clean_memory():
8686

8787
@pytest.mark.parametrize("wname", ["haar", "db4", "sym5", "sym16", "bior4.4"])
8888
@pytest.mark.parametrize("slices", [55, 80])
89-
@pytest.mark.parametrize("level", [1, 3, 7, 11])
89+
@pytest.mark.parametrize("level", [None, 1, 3, 7, 11])
9090
@pytest.mark.parametrize("dim_x", [128, 140])
9191
def test_remove_stripe_fw_calc_mem(slices, level, dim_x, wname, ensure_clean_memory):
9292
dim_y = 159
9393
data = cp.random.random_sample((slices, dim_x, dim_y), dtype=np.float32)
9494
hook = MaxMemoryHook()
9595
with hook:
96-
remove_stripe_fw(cp.copy(data), level=level)
96+
remove_stripe_fw(cp.copy(data), wname=wname, level=level)
9797
actual_mem_peak = hook.max_mem
9898

9999
hook = MaxMemoryHook()

0 commit comments

Comments
 (0)