Skip to content

Commit 1b3ff3a

Browse files
committed
fixing tests with logpolar
1 parent 8d8784d commit 1b3ff3a

File tree

1 file changed

+12
-5
lines changed

1 file changed

+12
-5
lines changed

zenodo-tests/test_recon/test_algorithm.py

Lines changed: 12 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
12
import cupy as cp
23
import numpy as np
34
import pytest
@@ -70,7 +71,7 @@ def test_reconstruct_FBP3d_tomobar_i12_dataset1(i12_dataset1):
7071
assert recon_data.shape == (2560, 50, 2560)
7172

7273

73-
def test_reconstruct_LPRec_i12_dataset1(i12_dataset1):
74+
def test_reconstruct_LPRec3d_tomobar_i12_dataset1(i12_dataset1):
7475
force_clean_gpu_memory()
7576
projdata = i12_dataset1[0]
7677
angles = i12_dataset1[1]
@@ -92,12 +93,14 @@ def test_reconstruct_LPRec_i12_dataset1(i12_dataset1):
9293
)
9394
assert recon_data.flags.c_contiguous
9495
recon_data = recon_data.get()
95-
assert_allclose(np.sum(recon_data), 8973.761, rtol=1e-07, atol=1e-6)
96+
assert isclose(np.sum(recon_data), 8973.755, abs_tol=10**-3)
97+
assert isclose(np.max(recon_data), 0.006396, abs_tol=10**-6)
98+
assert isclose(np.min(recon_data), -0.006159, abs_tol=10**-6)
9699
assert recon_data.dtype == np.float32
97100
assert recon_data.shape == (2560, 3, 2560)
98101

99102

100-
def test_reconstruct_LP_REC_i13_dataset1(i13_dataset1):
103+
def test_reconstruct_LPRec_tomobar_i13_dataset1(i13_dataset1):
101104
force_clean_gpu_memory()
102105
projdata = i13_dataset1[0]
103106
angles = i13_dataset1[1]
@@ -134,7 +137,9 @@ def test_reconstruct_LP_REC_i13_dataset1(i13_dataset1):
134137

135138
assert recon_data.flags.c_contiguous
136139
recon_data = recon_data.get()
137-
assert isclose(np.sum(recon_data), 1241.859, abs_tol=10**-3)
140+
assert isclose(np.sum(recon_data), 1241.8612, abs_tol=10**-4)
141+
assert isclose(np.max(recon_data), 0.008238, abs_tol=10**-6)
142+
assert isclose(np.min(recon_data), -0.006565, abs_tol=10**-6)
138143
assert recon_data.dtype == np.float32
139144
assert recon_data.shape == (4646, 1, 4646)
140145

@@ -200,7 +205,9 @@ def test_reconstruct_LPRec3d_tomobar_i13_dataset2(i13_dataset2):
200205
assert recon_data.flags.c_contiguous
201206
recon_data = recon_data.get()
202207

203-
assert isclose(np.sum(recon_data), 4095.577, abs_tol=10**-3)
208+
assert isclose(np.sum(recon_data), 4095.6272, abs_tol=10**-4)
209+
assert isclose(np.max(recon_data), 0.010567, abs_tol=10**-6)
210+
assert isclose(np.min(recon_data), -0.008396, abs_tol=10**-6)
204211
assert recon_data.dtype == np.float32
205212
assert recon_data.shape == (2560, 10, 2560)
206213

0 commit comments

Comments
 (0)