Skip to content

Commit a4466d1

Browse files
committed
fix: update test setup.
1 parent 9d763aa commit a4466d1

File tree

2 files changed

+19
-17
lines changed

2 files changed

+19
-17
lines changed

tests/operators/test_orc.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -84,7 +84,7 @@ def test_orc_forward(
8484
ksp = orc_nufft.op(image_data)
8585
ksp_ideal = ext_mat @ image_data.ravel()
8686

87-
assert_correlate(ksp.squeeze(), ksp_ideal)
87+
assert_correlate(ksp.squeeze(), ksp_ideal, slope_err=0.005)
8888

8989

9090
def test_orc_adjoint(orc_info, kspace_data):
@@ -93,4 +93,4 @@ def test_orc_adjoint(orc_info, kspace_data):
9393
img = orc_nufft.adj_op(kspace_data)
9494
img_ideal = ext_mat.conj().T @ kspace_data.ravel()
9595
img_ideal = img_ideal.reshape(orc_nufft.shape)
96-
assert_correlate(img.squeeze(), img_ideal)
96+
assert_correlate(img.squeeze(), img_ideal, slope_err=0.005)

tests/test_offres_exp_approx.py

Lines changed: 17 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -26,10 +26,10 @@ def case_real2D(self, N=64, b0_range=(-300, 300)):
2626
b0_map, mask = make_b0map(2 * [N])
2727
return b0_map, None, mask
2828

29-
# def case_real3D(self, N=32, b0range=(-300, 300)):
30-
# """Create a real (B0 only) 3D field map."""
31-
# b0_map, mask = make_b0map(3 * [N], b0range)
32-
# return b0_map, None, mask
29+
def case_real3D(self, N=32, b0range=(-300, 300)):
30+
"""Create a real (B0 only) 3D field map."""
31+
b0_map, mask = make_b0map(3 * [N], b0range)
32+
return b0_map, None, mask
3333

3434
def case_complex2D(self, N=64, b0range=(-300, 300), t2svalue=15.0):
3535
"""Create a complex (R2* + 1j * B0) 2D field map."""
@@ -44,21 +44,23 @@ def case_complex2D(self, N=64, b0range=(-300, 300), t2svalue=15.0):
4444

4545
return b0_map, r2s_map, mask
4646

47-
# def case_complex3D(self, N=32, b0range=(-300, 300), t2svalue=15.0):
48-
# """Create a complex (R2* + 1j * B0) 3D field map."""
49-
# # Generate real and imaginary parts
50-
# t2s_map, _ = make_t2smap(3 * [N], t2svalue)
51-
# b0_map, mask = make_b0map(3 * [N], b0range)
47+
def case_complex3D(self, N=32, b0range=(-300, 300), t2svalue=15.0):
48+
"""Create a complex (R2* + 1j * B0) 3D field map."""
49+
# Generate real and imaginary parts
50+
t2s_map, _ = make_t2smap(3 * [N], t2svalue)
51+
b0_map, mask = make_b0map(3 * [N], b0range)
5252

53-
# # Convert T2* map to R2* map
54-
# t2s_map = t2s_map * 1e-3 # ms -> s
55-
# r2s_map = 1.0 / (t2s_map + 1e-9) # Hz
56-
# r2s_map = mask * r2s_map
57-
# return b0_map, r2s_map, mask
53+
# Convert T2* map to R2* map
54+
t2s_map = t2s_map * 1e-3 # ms -> s
55+
r2s_map = 1.0 / (t2s_map + 1e-9) # Hz
56+
r2s_map = mask * r2s_map
57+
return b0_map, r2s_map, mask
5858

5959

6060
@_param_array_interface_np_cp
61-
@parametrize_with_cases("b0_map, r2s_map, mask", cases=CasesB0maps)
61+
@parametrize_with_cases(
62+
"b0_map, r2s_map, mask", cases=[CasesB0maps.case_real2D, CasesB0maps.case_complex2D]
63+
)
6264
@parametrize("method", ["svd-full", "mti", "mfi"])
6365
@parametrize("L, lazy", [(40, True), (-1, True), (40, False)])
6466
def test_b0map_coeff(b0_map, r2s_map, mask, method, L, lazy, array_interface):

0 commit comments

Comments
 (0)