From 4d8ece780b4ecace8292ea035222b882f53a4928 Mon Sep 17 00:00:00 2001 From: Hannu Parviainen Date: Tue, 10 Dec 2024 16:11:21 +0000 Subject: [PATCH] Modified test_tracing.py. --- specreduce/tests/test_tracing.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/specreduce/tests/test_tracing.py b/specreduce/tests/test_tracing.py index ec372b8..3f54052 100644 --- a/specreduce/tests/test_tracing.py +++ b/specreduce/tests/test_tracing.py @@ -373,7 +373,7 @@ def test_fit_trace_fully_masked_cols(self, mask_treatment): @pytest.mark.parametrize("peak_method,expected", [("max", [5., 3., 5., 5., 7., 5., np.nan, 5., 5., 5., 2., 5.]), - ("gaussian", [5., 2.10936004, 5., 5., 7.80744334, + ("gaussian", [5., 1.696159, 5., 5., 7.80744334, 5., np.nan, 5., 5., 5., 1.28216332, 5.]), ("centroid", [4.27108332, 2.24060342, 4.27108332, 4.27108332, 6.66827608, 4.27108332, @@ -409,7 +409,7 @@ def test_mask_treatment_filter(self, peak_method, expected): # check that final fit to all bins, accouting for fully-masked bins, # matches the trace - fitter = fitting.LevMarLSQFitter() + fitter = fitting.LMLSQFitter() mask = np.isfinite(y_bins) all_bin_fit = fitter(trace.trace_model, x_bins[mask], y_bins[mask]) all_bin_fit = all_bin_fit((np.arange(12)))