@@ -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