diff options
Diffstat (limited to 'numpy/polynomial/tests/test_chebyshev.py')
| -rw-r--r-- | numpy/polynomial/tests/test_chebyshev.py | 33 |
1 files changed, 33 insertions, 0 deletions
diff --git a/numpy/polynomial/tests/test_chebyshev.py b/numpy/polynomial/tests/test_chebyshev.py index cc23ba701..b80ff61de 100644 --- a/numpy/polynomial/tests/test_chebyshev.py +++ b/numpy/polynomial/tests/test_chebyshev.py @@ -345,6 +345,39 @@ class TestMisc(TestCase) : for i in range(10) : assert_equal(ch.poly2cheb(Tlist[i]), [0]*i + [1]) + def test_chebpts1(self): + #test exceptions + yield assert_raises(ValueError, ch.chebpts1, 1.5) + yield assert_raises(ValueError, ch.chebpts1, 0) + + #test points + tgt = [0] + yield assert_almost_equal(ch.chebpts1(1), tgt) + tgt = [-0.70710678118654746, 0.70710678118654746] + yield assert_almost_equal(ch.chebpts1(2), tgt) + tgt = [-0.86602540378443871, 0, 0.86602540378443871] + yield assert_almost_equal(ch.chebpts1(3), tgt) + tgt = [-0.9238795325, -0.3826834323, 0.3826834323, 0.9238795325] + yield assert_almost_equal(ch.chebpts1(4), tgt) + + + def test_chebpts2(self): + #test exceptions + yield assert_raises(ValueError, ch.chebpts2, 1.5) + yield assert_raises(ValueError, ch.chebpts2, 1) + + #test points + tgt = [-1, 1] + yield assert_almost_equal(ch.chebpts2(2), tgt) + tgt = [-1, 0, 1] + yield assert_almost_equal(ch.chebpts2(3), tgt) + tgt = [-1 -0.5, .5, 1] + yield assert_almost_equal(ch.chebpts2(4), tgt) + tgt = [-1.0, -0.707106781187, 0, 0.707106781187, 1.0] + yield assert_almost_equal(ch.chebpts2(5), tgt) + + + class TestChebyshevClass(TestCase) : |