From 6bbd88a8c1296b313996ed6d657a16c305bffafc Mon Sep 17 00:00:00 2001 From: Fabian Joswig Date: Wed, 25 May 2022 18:03:46 +0100 Subject: [PATCH] tests: fit_corr_independent now runs for various methods. --- tests/fits_test.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/tests/fits_test.py b/tests/fits_test.py index 374550e4..88b8438a 100644 --- a/tests/fits_test.py +++ b/tests/fits_test.py @@ -171,12 +171,13 @@ def test_fit_corr_independent(): y = a[0] * anp.exp(-a[1] * x) return y - out = pe.least_squares(x, oy, func) - out_corr = pe.least_squares(x, oy, func, correlated_fit=True) + for method in ["Levenberg-Marquardt", "migrad", "Nelder-Mead"]: + out = pe.least_squares(x, oy, func, method=method) + out_corr = pe.least_squares(x, oy, func, correlated_fit=True, method=method) - assert np.isclose(out.chisquare, out_corr.chisquare) - assert (out[0] - out_corr[0]).is_zero(atol=1e-5) - assert (out[1] - out_corr[1]).is_zero(atol=1e-5) + assert np.isclose(out.chisquare, out_corr.chisquare) + assert (out[0] - out_corr[0]).is_zero(atol=1e-5) + assert (out[1] - out_corr[1]).is_zero(atol=1e-5) def test_total_least_squares():