@@ -37,12 +37,12 @@ def test_generated(ivim_algorithm, ivim_data, SNR, rtol, atol, fit_count, rician
37
37
# else:
38
38
# signal = data["data"]
39
39
start_time = datetime .datetime .now ()
40
- [ f_fit , Dp_fit , D_fit ] = fit .osipi_fit (signal , bvals ) #, prior_in=prior
40
+ fit_result = fit .osipi_fit (signal , bvals ) #, prior_in=prior
41
41
time_delta += datetime .datetime .now () - start_time
42
42
if save_file is not None :
43
- save_file .writerow ([ivim_algorithm , name , SNR , idx , f , Dp , D , f_fit , Dp_fit , D_fit , * signal ])
43
+ save_file .writerow ([ivim_algorithm , name , SNR , idx , f , Dp , D , fit_result [ "f" ], fit_result [ "D*" ], fit_result [ "D" ] , * signal ])
44
44
# save_results(save_file, ivim_algorithm, name, SNR, idx, [f, Dp, D], [f_fit, Dp_fit, D_fit])
45
- npt .assert_allclose ([f , Dp , D ], [f_fit , Dp_fit , D_fit ], rtol , atol )
45
+ npt .assert_allclose ([f , Dp , D ], [fit_result [ "f" ], fit_result [ "D*" ], fit_result [ "D" ] ], rtol , atol )
46
46
if save_duration_file is not None :
47
47
save_duration_file .writerow ([ivim_algorithm , name , SNR , time_delta / datetime .timedelta (microseconds = 1 ), fit_count ])
48
48
# save_duration(save_duration_file, ivim_algorithm, name, SNR, time_delta, fit_count)
0 commit comments