diff options
| author | Charles Harris <charlesr.harris@gmail.com> | 2013-09-02 13:21:48 -0600 |
|---|---|---|
| committer | Charles Harris <charlesr.harris@gmail.com> | 2013-09-03 05:22:10 -0600 |
| commit | d2c44c5a012d0a5db7d10a2586c8e9c58045d04b (patch) | |
| tree | b93306d17c5460a68d7e7fb29e6be729ad2a28dd /numpy/lib/tests/test_polynomial.py | |
| parent | 57245e4570fa84f2e40a6fa1cc22716d279f8c00 (diff) | |
| download | numpy-d2c44c5a012d0a5db7d10a2586c8e9c58045d04b.tar.gz | |
STY: Make numpy/lib/test/*.py PEP8 compliant.
Run autopep8 over the test files in numpy/lib/test and make fixes
to the result.
Also remove Python5 workaround.
Diffstat (limited to 'numpy/lib/tests/test_polynomial.py')
| -rw-r--r-- | numpy/lib/tests/test_polynomial.py | 14 |
1 files changed, 8 insertions, 6 deletions
diff --git a/numpy/lib/tests/test_polynomial.py b/numpy/lib/tests/test_polynomial.py index 617419eee..4adc6aed8 100644 --- a/numpy/lib/tests/test_polynomial.py +++ b/numpy/lib/tests/test_polynomial.py @@ -81,6 +81,7 @@ poly1d([ 2.]) from numpy.testing import * import numpy as np + class TestDocs(TestCase): def test_doctests(self): return rundocs() @@ -100,7 +101,7 @@ class TestDocs(TestCase): p[1] = 0 assert_equal(str(p), " \n0") - def test_polyfit(self) : + def test_polyfit(self): c = np.array([3., 2., 1.]) x = np.linspace(0, 2, 7) y = np.polyval(c, x) @@ -118,9 +119,9 @@ class TestDocs(TestCase): m2, cov2 = np.polyfit(x, y+err, 2, w=weights, cov=True) assert_almost_equal([4.8927, -1.0177, 1.7768], m2, decimal=4) - val = [[ 8.7929, -10.0103, 0.9756], + val = [[8.7929, -10.0103, 0.9756], [-10.0103, 13.6134, -1.8178], - [ 0.9756, -1.8178, 0.6674]] + [0.9756, -1.8178, 0.6674]] assert_almost_equal(val, cov2, decimal=4) # check 2D (n,1) case @@ -135,8 +136,8 @@ class TestDocs(TestCase): m, cov = np.polyfit(x, yy + np.array(err)[:, np.newaxis], 2, cov=True) assert_almost_equal(est, m[:, 0], decimal=4) assert_almost_equal(est, m[:, 1], decimal=4) - assert_almost_equal(val0, cov[:,:, 0], decimal=4) - assert_almost_equal(val0, cov[:,:, 1], decimal=4) + assert_almost_equal(val0, cov[:, :, 0], decimal=4) + assert_almost_equal(val0, cov[:, :, 1], decimal=4) def test_objects(self): from decimal import Decimal @@ -160,7 +161,8 @@ class TestDocs(TestCase): def test_integ_coeffs(self): p = np.poly1d([3, 2, 1]) p2 = p.integ(3, k=[9, 7, 6]) - assert_((p2.coeffs == [1/4./5., 1/3./4., 1/2./3., 9/1./2., 7, 6]).all()) + assert_( + (p2.coeffs == [1/4./5., 1/3./4., 1/2./3., 9/1./2., 7, 6]).all()) def test_zero_dims(self): try: |