TST: Add test for weight modification.

numpy · Dec 15, 2011 · 008e5e9 · 008e5e9
1 parent eb042bf
commit 008e5e9
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/numpy/ma/tests/test_extras.py b/numpy/ma/tests/test_extras.py
@@ -632,11 +632,13 @@ def test_polyfit(self):
             assert_almost_equal(a, a_)
         #
         w =  np.random.rand(10) + 1
+        wo = w.copy()
         xs = x[1:-1]
         ys = y[1:-1]
         ws = w[1:-1]
         (C, R, K, S, D) = polyfit(x, y, 3, full=True, w=w)
         (c, r, k, s, d) = np.polyfit(xs, ys, 3, full=True, w=ws)
+        assert_equal(w, wo)
         for (a, a_) in zip((C, R, K, S, D), (c, r, k, s, d)):
             assert_almost_equal(a, a_)