Merge branch 'poly'

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) :