diff options
Diffstat (limited to 'numpy/polynomial/tests')
| -rw-r--r-- | numpy/polynomial/tests/test_chebyshev.py | 9 | ||||
| -rw-r--r-- | numpy/polynomial/tests/test_polynomial.py | 9 |
2 files changed, 10 insertions, 8 deletions
diff --git a/numpy/polynomial/tests/test_chebyshev.py b/numpy/polynomial/tests/test_chebyshev.py index 6f42d06b9..b2c4ee105 100644 --- a/numpy/polynomial/tests/test_chebyshev.py +++ b/numpy/polynomial/tests/test_chebyshev.py @@ -416,11 +416,12 @@ class TestChebyshevClass(TestCase) : assert_equal(p.trim(1e-5).coef, coef[:1]) def test_truncate(self) : - assert_raises(ValueError, self.p1.truncate, 0) - assert_equal(len(self.p1.truncate(4)), 3) + assert_raises(ValueError, self.p1.truncate, .5) + assert_raises(ValueError, self.p1.truncate, -1) assert_equal(len(self.p1.truncate(3)), 3) - assert_equal(len(self.p1.truncate(2)), 2) - assert_equal(len(self.p1.truncate(1)), 1) + assert_equal(len(self.p1.truncate(2)), 3) + assert_equal(len(self.p1.truncate(1)), 2) + assert_equal(len(self.p1.truncate(0)), 1) def test_copy(self) : p = self.p1.copy() diff --git a/numpy/polynomial/tests/test_polynomial.py b/numpy/polynomial/tests/test_polynomial.py index 34a3d10f3..7ba5b6c8f 100644 --- a/numpy/polynomial/tests/test_polynomial.py +++ b/numpy/polynomial/tests/test_polynomial.py @@ -390,11 +390,12 @@ class TestPolynomialClass(TestCase) : assert_equal(p.trim(1e-5).coef, coef[:1]) def test_truncate(self) : - assert_raises(ValueError, self.p1.truncate, 0) - assert_equal(len(self.p1.truncate(4)), 3) + assert_raises(ValueError, self.p1.truncate, .5) + assert_raises(ValueError, self.p1.truncate, -1) assert_equal(len(self.p1.truncate(3)), 3) - assert_equal(len(self.p1.truncate(2)), 2) - assert_equal(len(self.p1.truncate(1)), 1) + assert_equal(len(self.p1.truncate(2)), 3) + assert_equal(len(self.p1.truncate(1)), 2) + assert_equal(len(self.p1.truncate(0)), 1) def test_copy(self) : p = self.p1.copy() |