diff options
Diffstat (limited to 'numpy/lib/tests/test_regression.py')
-rw-r--r-- | numpy/lib/tests/test_regression.py | 11 |
1 files changed, 8 insertions, 3 deletions
diff --git a/numpy/lib/tests/test_regression.py b/numpy/lib/tests/test_regression.py index 189b2e481..e465dc4b6 100644 --- a/numpy/lib/tests/test_regression.py +++ b/numpy/lib/tests/test_regression.py @@ -1,10 +1,9 @@ from numpy.testing import * import numpy as np -rlevel = 1 -class TestRegression(TestCase): - def test_polyfit_build(self,level=rlevel): +class TestRegression(object): + def test_polyfit_build(self): """Ticket #628""" ref = [-1.06123820e-06, 5.70886914e-04, -1.13822012e-01, 9.95368241e+00, -3.14526520e+02] @@ -25,6 +24,12 @@ class TestRegression(TestCase): tested = np.polyfit(x, y, 4) assert_array_almost_equal(ref, tested) + def test_polyint_type(self) : + """Ticket #944""" + msg = "Wrong type, should be complex" + x = np.polyint(np.ones(3, dtype=np.complex)) + assert_(np.asarray(x).dtype == np.complex, msg) + if __name__ == "__main__": run_module_suite() |