Modified test_tracing.py.

cshanahan1 · Dec 10, 2024 · 4d8ece7 · 4d8ece7
1 parent 7b64a0a
commit 4d8ece7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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)))