TST: Add tests for basis and cast static class methods.

A new test file, test_classes, has been added so that conversions
between all the class types can be tested. Several tests common to
all the classes were also moved to this file. Ideally all the common
tests will be moved, but that isn't done yet.

1 files changed, 5 insertions, 63 deletions

diff --git a/numpy/polynomial/tests/test_polynomial.py b/numpy/polynomial/tests/test_polynomial.py
index 09028686f..5715d8b2d 100644
--- a/numpy/polynomial/tests/test_polynomial.py
+++ b/numpy/polynomial/tests/test_polynomial.py
@@ -5,7 +5,9 @@ from __future__ import division
 
 import numpy as np
 import numpy.polynomial.polynomial as poly
-from numpy.testing import *
+from numpy.testing import (
+        TestCase, assert_almost_equal, assert_raises,
+        assert_equal, assert_, run_module_suite)
 
 def trim(x) :
     return poly.polytrim(x, tol=1e-6)
@@ -533,9 +535,6 @@ class TestPolynomialClass(TestCase) :
         xx = 2*x - 1
         assert_almost_equal(self.p2(x), self.p1(xx))
 
-    def test_degree(self) :
-        assert_equal(self.p1.degree(), 2)
-
     def test_cutdeg(self) :
         assert_raises(ValueError, self.p1.cutdeg, .5)
         assert_raises(ValueError, self.p1.cutdeg, -1)
@@ -544,11 +543,6 @@ class TestPolynomialClass(TestCase) :
         assert_equal(len(self.p1.cutdeg(1)), 2)
         assert_equal(len(self.p1.cutdeg(0)), 1)
 
-    def test_convert(self) :
-        x = np.linspace(-1,1)
-        p = self.p1.convert(domain=[0,1])
-        assert_almost_equal(p(x), self.p1(x))
-
     def test_mapparms(self) :
         parms = self.p2.mapparms()
         assert_almost_equal(parms, [-1, 2])
@@ -568,10 +562,6 @@ class TestPolynomialClass(TestCase) :
         assert_equal(len(self.p1.truncate(2)), 2)
         assert_equal(len(self.p1.truncate(1)), 1)
 
-    def test_copy(self) :
-        p = self.p1.copy()
-        assert_(self.p1 == p)
-
     def test_integ(self) :
         p = self.p2.integ()
         assert_almost_equal(p.coef, poly.polyint([1,2,3], 1, 0, scl=.5))
@@ -582,55 +572,6 @@ class TestPolynomialClass(TestCase) :
         p = self.p2.integ(2, [1, 2])
         assert_almost_equal(p.coef, poly.polyint([1,2,3], 2, [1, 2], scl=.5))
 
-    def test_deriv(self) :
-        p = self.p2.integ(2, [1, 2])
-        assert_almost_equal(p.deriv(1).coef, self.p2.integ(1, [1]).coef)
-        assert_almost_equal(p.deriv(2).coef, self.p2.coef)
-
-    def test_roots(self) :
-        p = poly.Polynomial([0, -1, 0, 1], [0, 1])
-        res = p.roots()
-        tgt = [0, .5, 1]
-        assert_almost_equal(res, tgt)
-
-    def test_linspace(self):
-        xdes = np.linspace(0, 1, 20)
-        ydes = self.p2(xdes)
-        xres, yres = self.p2.linspace(20)
-        assert_almost_equal(xres, xdes)
-        assert_almost_equal(yres, ydes)
-
-    def test_fromroots(self) :
-        roots = [0, .5, 1]
-        p = poly.Polynomial.fromroots(roots, domain=[0, 1])
-        res = p.coef
-        tgt = [0, -1, 0, 1]
-        assert_almost_equal(res, tgt)
-
-    def test_fit(self) :
-        def f(x) :
-            return x*(x - 1)*(x - 2)
-        x = np.linspace(0,3)
-        y = f(x)
-
-        # test default value of domain
-        p = poly.Polynomial.fit(x, y, 3)
-        assert_almost_equal(p.domain, [0,3])
-
-        # test that fit works in given domains
-        p = poly.Polynomial.fit(x, y, 3, None)
-        assert_almost_equal(p(x), y)
-        assert_almost_equal(p.domain, [0,3])
-        p = poly.Polynomial.fit(x, y, 3, [])
-        assert_almost_equal(p(x), y)
-        assert_almost_equal(p.domain, [-1, 1])
-        # test that fit accepts weights.
-        w = np.zeros_like(x)
-        yw = y.copy()
-        w[1::2] = 1
-        yw[0::2] = 0
-        p = poly.Polynomial.fit(x, yw, 3, w=w)
-        assert_almost_equal(p(x), y)
 
     def test_identity(self) :
         x = np.linspace(0,3)
@@ -638,7 +579,8 @@ class TestPolynomialClass(TestCase) :
         assert_almost_equal(p(x), x)
         p = poly.Polynomial.identity([1,3])
         assert_almost_equal(p(x), x)
-#
+
+
 
 if __name__ == "__main__":
     run_module_suite()