Fix failing uncertainty test

astropy · Sep 17, 2019 · 31a4b53 · 31a4b53
1 parent e804847
commit 31a4b53
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/specutils/tests/test_fitting.py b/specutils/tests/test_fitting.py
@@ -216,16 +216,18 @@ def calculate_rms(x, init_mod, implicit_weights):
             spec = Spectrum1D(spectral_axis=x, flux=y * u.Jy,
                               uncertainty=StdDevUncertainty(unc * u.Jy))
 
-            weights = 'unc' if implicit_weights else 1 / unc
+            weights = 'unc' if implicit_weights else unc ** -2
 
             spec_fit = fit_lines(spec, init_mod, weights=weights)
 
             rms.append(np.std(spec_fit(x).value - y))
 
         return np.median(rms)
 
-    assert np.allclose(calculate_rms(x, init_mod, implicit_weights=True), 5.101611033799086)
-    assert np.allclose(calculate_rms(x, init_mod, implicit_weights=False), 5.113697654869089)
+    assert np.allclose(calculate_rms(x, init_mod, implicit_weights=True), 
+                       5.113708262419985)
+    assert np.allclose(calculate_rms(x, init_mod, implicit_weights=False), 
+                       5.147348340711497)
 
 
 def test_single_peak_fit_window():