1+
12import cupy as cp
23import numpy as np
34import 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