Skip to content

Commit 2a945d9

Browse files
committed
parametrise test further
1 parent 791b27f commit 2a945d9

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

tests/test_httomolibgpu.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -573,9 +573,10 @@ def test_recon_FBP_memoryhook(
573573

574574

575575
@pytest.mark.cupy
576-
@pytest.mark.parametrize("slices", [2, 3, 4, 5, 10, 15])
577-
def test_recon_LPRec_memoryhook(slices, ensure_clean_memory):
578-
angles_number = 1801
576+
@pytest.mark.parametrize("projections", [1801, 2560, 3601])
577+
@pytest.mark.parametrize("slices", [3, 4, 5, 10])
578+
def test_recon_LPRec_memoryhook(slices, projections, ensure_clean_memory):
579+
angles_number = projections
579580
detX_size = 2560
580581
data = cp.random.random_sample((angles_number, slices, detX_size), dtype=np.float32)
581582
kwargs = {}

0 commit comments

Comments
 (0)