Skip to content

Commit f4790fc

Browse files
committed
zenodo test on data padding
1 parent 4d99b5c commit f4790fc

File tree

1 file changed

+31
-0
lines changed

1 file changed

+31
-0
lines changed

zenodo-tests/test_recon/test_algorithm.py

Lines changed: 31 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -130,6 +130,37 @@ def test_reconstruct_LPRec3d_tomobar_i12_dataset1(i12_dataset1: tuple):
130130
assert recon_data.shape == (2560, 3, 2560)
131131

132132

133+
def test_reconstruct_LPRec3d_tomobar_i12_dataset1_pad(i12_dataset1: tuple):
134+
force_clean_gpu_memory()
135+
projdata = i12_dataset1[0]
136+
angles = i12_dataset1[1]
137+
flats = i12_dataset1[2]
138+
darks = i12_dataset1[3]
139+
del i12_dataset1
140+
141+
data_normalised = normalize(projdata, flats, darks, minus_log=True)
142+
data_normalised_cut = data_normalised[:, 5:8, :]
143+
del flats, darks, projdata, data_normalised
144+
force_clean_gpu_memory()
145+
146+
recon_data = LPRec3d_tomobar(
147+
data_normalised_cut,
148+
np.deg2rad(angles),
149+
center=1253.75,
150+
detector_pad=100,
151+
filter_type="shepp",
152+
filter_freq_cutoff=1.0,
153+
recon_mask_radius=2.0,
154+
)
155+
assert recon_data.flags.c_contiguous
156+
recon_data = recon_data.get()
157+
assert isclose(np.sum(recon_data), 9629.825, abs_tol=10**-3)
158+
assert pytest.approx(np.max(recon_data), rel=1e-3) == 0.006357904
159+
assert pytest.approx(np.min(recon_data), rel=1e-3) == -0.0061424887
160+
assert recon_data.dtype == np.float32
161+
assert recon_data.shape == (2560, 3, 2560)
162+
163+
133164
def test_reconstruct_LPRec_tomobar_i13_dataset1(i13_dataset1: tuple):
134165
force_clean_gpu_memory()
135166
projdata = i13_dataset1[0]

0 commit comments

Comments
 (0)