properly remove nans

jason-neal · Nov 5, 2017 · 17fca4d · 17fca4d
1 parent 59f7474
commit 17fca4d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/spectrum_overload/test/test_Spectrum.py b/spectrum_overload/test/test_Spectrum.py
@@ -471,9 +471,9 @@ def test_normalization_method_match_degree(method, degree):
     x = np.arange(1000)
     y = np.arange(1000)
     s = Spectrum(xaxis=x, flux=y)
-    named_method = s.normalize(method=method)
+    named_method = named_method = s.normalize(method=method)
     named_method.remove_nans()  # hack for geting to run on < py34 
     poly_deg = s.normalize(method='poly', degree=degree)
-    poly_deg.remove_nans() # hack for getting to pass on < py 34
+    poly_deg = poly_deg.remove_nans() # hack for getting to pass on < py 34
     assert np.allclose(named_method.flux, poly_deg.flux)