Skip to content

Commit bce6ec7

Browse files
author
IvanARashid
committed
Changed to dictionary outputs from osipi_fit
1 parent 14e77f9 commit bce6ec7

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/IVIMmodels/unit_tests/test_ivim_synthetic.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -37,12 +37,12 @@ def test_generated(ivim_algorithm, ivim_data, SNR, rtol, atol, fit_count, rician
3737
# else:
3838
# signal = data["data"]
3939
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
4141
time_delta += datetime.datetime.now() - start_time
4242
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])
4444
# 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)
4646
if save_duration_file is not None:
4747
save_duration_file.writerow([ivim_algorithm, name, SNR, time_delta/datetime.timedelta(microseconds=1), fit_count])
4848
# save_duration(save_duration_file, ivim_algorithm, name, SNR, time_delta, fit_count)

0 commit comments

Comments
 (0)