@@ -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 )])
6466def test_b0map_coeff (b0_map , r2s_map , mask , method , L , lazy , array_interface ):
0 commit comments