Skip to content

Commit 3a26534

Browse files
committed
Update memory hook test naming
1 parent efde97e commit 3a26534

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/test_httomolibgpu.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -576,7 +576,7 @@ def test_recon_FBP3d_tomobar_memoryhook(
576576
@pytest.mark.cupy
577577
@pytest.mark.parametrize("projections", [1801, 2560, 3601])
578578
@pytest.mark.parametrize("slices", [3, 4, 5, 10, 15, 20])
579-
def test_recon_LPRec_memoryhook(slices, projections, ensure_clean_memory):
579+
def test_recon_LPRec3d_tomobar_memoryhook(slices, projections, ensure_clean_memory):
580580
angles_number = projections
581581
detX_size = 2560
582582
data = cp.random.random_sample((angles_number, slices, detX_size), dtype=np.float32)
@@ -590,7 +590,7 @@ def test_recon_LPRec_memoryhook(slices, projections, ensure_clean_memory):
590590

591591
hook = MaxMemoryHook()
592592
with hook:
593-
recon_data = LPRec(cp.copy(data), **kwargs)
593+
recon_data = LPRec3d_tomobar(cp.copy(data), **kwargs)
594594

595595
# make sure estimator function is within range (80% min, 100% max)
596596
max_mem = (
@@ -601,7 +601,7 @@ def test_recon_LPRec_memoryhook(slices, projections, ensure_clean_memory):
601601
input_data_type = np.float32()
602602

603603
# now we estimate how much of the total memory required for this data
604-
(estimated_memory_bytes, subtract_bytes) = _calc_memory_bytes_LPRec(
604+
(estimated_memory_bytes, subtract_bytes) = _calc_memory_bytes_LPRec3d_tomobar(
605605
non_slice_dims_shape, dtype=input_data_type, **kwargs
606606
)
607607

0 commit comments

Comments
 (0)