From a9e80989fe4283367e730c00b5acae09996d6abd Mon Sep 17 00:00:00 2001 From: Fabian Joswig Date: Wed, 8 Dec 2021 22:37:38 +0000 Subject: [PATCH] test: test for Corr.fit added --- tests/correlators_test.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/tests/correlators_test.py b/tests/correlators_test.py index d2bb6a2c..f27bba54 100644 --- a/tests/correlators_test.py +++ b/tests/correlators_test.py @@ -79,6 +79,18 @@ def test_T_symmetry(): T_symmetric = my_corr.T_symmetry(my_corr) +def test_fit_correlator(): + my_corr = pe.correlators.Corr([pe.pseudo_Obs(1.01324, 0.05, 't'), pe.pseudo_Obs(2.042345, 0.0004, 't')]) + + def f(a, x): + y = a[0] + a[1] * x + return y + + fit_res = my_corr.fit(f) + assert fit_res[0] == my_corr[0] + assert fit_res[1] == my_corr[1] - my_corr[0] + + def test_utility(): corr_content = [] for t in range(8):