ó
ØÆ÷Xc        
   @` s˜  d  Z  d d l m Z m Z m Z d d l Z d d l j j	 Z
 d d l m Z d d l m Z m Z m Z m Z m Z m Z d „  Z d g Z d d g Z d d d	 g Z d d
 d d g Z d d d d d g Z d d d d d d g Z d d d d d d d g Z d d d d d d d d g Z d d d d d d d d d g	 Z d d d d d d d d d d  g
 Z e e e e e e e e e e g
 Z d! e f d" „  ƒ  YZ  d# e f d$ „  ƒ  YZ! d% e f d& „  ƒ  YZ" d' e f d( „  ƒ  YZ# d) e f d* „  ƒ  YZ$ d+ e f d, „  ƒ  YZ% d- e f d. „  ƒ  YZ& d/ e f d0 „  ƒ  YZ' d1 e f d2 „  ƒ  YZ( d3 e f d4 „  ƒ  YZ) d5 e f d6 „  ƒ  YZ* e+ d7 k r”e ƒ  n  d S(8   s   Tests for chebyshev module.

i    (   t   divisiont   absolute_importt   print_functionN(   t   polyval(   t   TestCaset   assert_almost_equalt   assert_raisest   assert_equalt   assert_t   run_module_suitec         C` s   t  j |  d d ƒS(   Nt   tolgíµ ÷Æ°>(   t   chebt   chebtrim(   t   x(    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyt   trim   s    i   iÿÿÿÿi   iýÿÿÿi   iøÿÿÿi   i   iìÿÿÿi   i   iÐÿÿÿi    iùÿÿÿi8   iÿÿÿi@   iàÿÿÿi    i ÿÿÿi€   i	   iˆÿÿÿi°  iÀýÿÿi   t   TestPrivatec           B` s   e  Z d  „  Z d „  Z RS(   c         C` sˆ   x t  d ƒ D]s } t j d g d g | t j ƒ } t j d g | d g d g | t j ƒ } t j | ƒ } t | | ƒ q Wd  S(   Ni   i   i   g      à?(   t   ranget   npt   arrayt   doubleR   t   _cseries_to_zseriesR   (   t   selft   it   inpt   tgtt   res(    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyt   test__cseries_to_zseries!   s
    #.c         C` sˆ   x t  d ƒ D]s } t j d g | d g d g | t j ƒ } t j d g d g | t j ƒ } t j | ƒ } t | | ƒ q Wd  S(   Ni   g      à?i   i   (   R   R   R   R   R   t   _zseries_to_cseriesR   (   R   R   R   R   R   (    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyt   test__zseries_to_cseries(   s
    .#(   t   __name__t
   __module__R   R   (    (    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyR      s   	t   TestConstantsc           B` s,   e  Z d  „  Z d „  Z d „  Z d „  Z RS(   c         C` s   t  t j d d g ƒ d  S(   Niÿÿÿÿi   (   R   R   t
   chebdomain(   R   (    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyt   test_chebdomain2   s    c         C` s   t  t j d g ƒ d  S(   Ni    (   R   R   t   chebzero(   R   (    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyt   test_chebzero5   s    c         C` s   t  t j d g ƒ d  S(   Ni   (   R   R   t   chebone(   R   (    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyt   test_chebone8   s    c         C` s   t  t j d d g ƒ d  S(   Ni    i   (   R   R   t   chebx(   R   (    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyt
   test_chebx;   s    (   R   R   R!   R#   R%   R'   (    (    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyR   0   s   			t   TestArithmeticc           B` s5   e  Z d  „  Z d „  Z d „  Z d „  Z d „  Z RS(   c         C` sË   xÄ t  d ƒ D]¶ } x­ t  d ƒ D]Ÿ } d | | f } t j t | | ƒ d ƒ } | | c d 7<| | c d 7<t j d g | d g d g | d g ƒ } t t | ƒ t | ƒ d | ƒq  Wq Wd  S(   Ni   s   At i=%d, j=%di   i    t   err_msg(   R   R   t   zerost   maxR   t   chebaddR   R   (   R   R   t   jt   msgR   R   (    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyt   test_chebaddA   s    .c         C` sË   xÄ t  d ƒ D]¶ } x­ t  d ƒ D]Ÿ } d | | f } t j t | | ƒ d ƒ } | | c d 7<| | c d 8<t j d g | d g d g | d g ƒ } t t | ƒ t | ƒ d | ƒq  Wq Wd  S(   Ni   s   At i=%d, j=%di   i    R)   (   R   R   R*   R+   R   t   chebsubR   R   (   R   R   R-   R.   R   R   (    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyt   test_chebsubK   s    .c         C` s¡   t  t j d g ƒ d g ƒ t  t j d g ƒ d d g ƒ x_ t d d ƒ D]N } d g | d g } d g | d d d d g } t  t j | ƒ | ƒ qK Wd  S(   Ni    i   i   g      à?(   R   R   t   chebmulxR   (   R   R   t   serR   (    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyt   test_chebmulxU   s    c         C` sÔ   xÍ t  d ƒ D]¿ } x¶ t  d ƒ D]¨ } d | | f } t j | | d ƒ } | | | c d 7<| t | | ƒ c d 7<t j d g | d g d g | d g ƒ } t t | ƒ t | ƒ d | ƒq  Wq Wd  S(   Ni   s   At i=%d, j=%di   g      à?i    R)   (   R   R   R*   t   absR   t   chebmulR   R   (   R   R   R-   R.   R   R   (    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyt   test_chebmul]   s    .c   
      C` sÑ   xÊ t  d ƒ D]¼ } x³ t  d ƒ D]¥ } d | | f } d g | d g } d g | d g } t j | | ƒ } t j | | ƒ \ } } t j t j | | ƒ | ƒ }	 t t |	 ƒ t | ƒ d | ƒq  Wq Wd  S(   Ni   s   At i=%d, j=%di    i   R)   (   R   R   R,   t   chebdivR6   R   R   (
   R   R   R-   R.   t   cit   cjR   t   quot   remR   (    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyt   test_chebdivg   s    (   R   R   R/   R1   R4   R7   R=   (    (    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyR(   ?   s
   	
	
		
t   TestEvaluationc           B` s¬   e  Z e j d  d d g ƒ Z e j d e e ƒ Z e j d e e e ƒ Z e j j d ƒ d d Z	 e
 e	 d	 d d
 g ƒ Z d „  Z d „  Z d „  Z d „  Z d „  Z RS(   g      @g       @g      ø?s   i,j->ijs
   i,j,k->ijki   i   i   i   g      ð?g      @c   	      C` sN  t  t j g  d g ƒ j d ƒ t j d d ƒ } g  t D] } t | | ƒ ^ q8 } x[ t d ƒ D]M } d | } | | } t j | d g | d g ƒ } t	 | | d | ƒq` Wx– t d ƒ D]ˆ } d g | } t j
 | ƒ } t  t j | d g ƒ j | ƒ t  t j | d d g ƒ j | ƒ t  t j | d d d g ƒ j | ƒ q¾ Wd  S(	   Ni   i    iÿÿÿÿi
   s   At i=%dR)   i   i   (   R   R   t   chebvalt   sizeR   t   linspacet   TlistR   R   R   R*   t   shape(	   R   R   t   ct   yR   R.   R   R   t   dims(    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyt   test_chebval}   s    "

 "c   
      C` s±   |  j  \ } } } |  j \ } } } t t t j | | d  |  j ƒ | | } t j | | |  j ƒ } t | | ƒ t j	 d ƒ }	 t j |	 |	 |  j ƒ } t
 | j d k ƒ d  S(   Ni   i   (   i   i   (   i   i   (   R   RE   R   t
   ValueErrorR   t	   chebval2dt   c2dR   R   t   onesR   RC   (
   R   t   x1t   x2t   x3t   y1t   y2t   y3R   R   t   z(    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyt   test_chebval2d’   s     
c   
      C` s¾   |  j  \ } } } |  j \ } } } t t t j | | | d  |  j ƒ | | | } t j | | | |  j ƒ } t | | ƒ t j	 d ƒ }	 t j |	 |	 |	 |  j ƒ } t
 | j d k ƒ d  S(   Ni   i   (   i   i   (   i   i   (   R   RE   R   RH   R   t	   chebval3dt   c3dR   R   RK   R   RC   (
   R   RL   RM   RN   RO   RP   RQ   R   R   RR   (    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyt   test_chebval3d£   s    #c   
      C` sœ   |  j  \ } } } |  j \ } } } t j d | | ƒ } t j | | |  j ƒ } t | | ƒ t j d ƒ }	 t j |	 |	 |  j ƒ } t	 | j
 d k ƒ d  S(   Ns   i,j->iji   i   (   i   i   (   i   i   (   i   i   i   i   (   R   RE   R   t   einsumR   t
   chebgrid2dRJ   R   RK   R   RC   (
   R   RL   RM   RN   RO   RP   RQ   R   R   RR   (    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyt   test_chebgrid2d´   s    c   
      C` s¥   |  j  \ } } } |  j \ } } } t j d | | | ƒ } t j | | | |  j ƒ } t | | ƒ t j d ƒ }	 t j |	 |	 |	 |  j ƒ } t	 | j
 d k ƒ d  S(   Ns
   i,j,k->ijki   i   (   i   i   (   i   i   (   i   i   i   i   i   i   (   R   RE   R   RW   R   t
   chebgrid3dRU   R   RK   R   RC   (
   R   RL   RM   RN   RO   RP   RQ   R   R   RR   (    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyt   test_chebgrid3dÂ   s    (   i   i   (   R   R   R   R   t   c1dRW   RJ   RU   t   randomR   R   RE   RG   RS   RV   RY   R[   (    (    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyR>   s   s   				t   TestIntegralc           B` s   e  Z d  „  Z d „  Z RS(   c   
      C` s•  t  t t j d g d ƒ t  t t j d g d ƒ t  t t j d g d d d g ƒ x` t d d ƒ D]O } d g | d d g } t j d g d | d | ƒ} t | d d g ƒ qd Wx¦ t d ƒ D]˜ } | d } d g | d g } | g d g | d | g } t j | ƒ } t j | d d d | g ƒ} t j | ƒ } t t | ƒ t | ƒ ƒ qÄ Wx~ t d ƒ D]p } | d } d g | d g } t j | ƒ } t j | d d d | g d	 d ƒ} t t j	 d | ƒ | ƒ qmWx¬ t d ƒ D]ž } | d } d g | d g } | g d g | d | g } t j | ƒ } t j | d d d | g d
 d ƒ} t j | ƒ } t t | ƒ t | ƒ ƒ qîWx£ t d ƒ D]• } xŒ t d d ƒ D]{ }	 d g | d g } | } x) t |	 ƒ D] } t j | d d ƒ} qáWt j | d |	 ƒ} t t | ƒ t | ƒ ƒ q³WqWx¾ t d ƒ D]° } x§ t d d ƒ D]– }	 d g | d g } | } x2 t |	 ƒ D]$ } t j | d d d | g ƒ} q‡Wt j | d |	 d t
 t |	 ƒ ƒ ƒ} t t | ƒ t | ƒ ƒ qYWqCWxÊ t d ƒ D]¼ } x³ t d d ƒ D]¢ }	 d g | d g } | } x8 t |	 ƒ D]* } t j | d d d | g d	 d ƒ} qHWt j | d |	 d t
 t |	 ƒ ƒ d	 d ƒ} t t | ƒ t | ƒ ƒ qWqWxÊ t d ƒ D]¼ } x³ t d d ƒ D]¢ }	 d g | d g } | } x8 t |	 ƒ D]* } t j | d d d | g d
 d ƒ} qWt j | d |	 d t
 t |	 ƒ ƒ d
 d ƒ} t t | ƒ t | ƒ ƒ qçWqÑWd  S(   Ni    g      à?iÿÿÿÿi   i   i   t   mt   kt   lbndt   scl(   R   RH   R   t   chebintR   R   t	   poly2chebt	   cheb2polyR   R?   t   list(
   R   R   R`   R   Rb   t   polR   t   chebpolRc   R-   (    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyt   test_chebintÓ   sz    "

$
$!"'!(-!(-c         C` s  t  j j d ƒ } t  j g  | j D] } t j | ƒ ^ q" ƒ j } t j | d d ƒ} t | | ƒ t  j g  | D] } t j | ƒ ^ qr ƒ } t j | d d ƒ} t | | ƒ t  j g  | D] } t j | d d ƒ^ q¿ ƒ } t j | d d d d ƒ} t | | ƒ d  S(   Ni   i   t   axisi    i   R`   (   i   i   (   R   R]   t   vstackt   TR   Rc   R   (   R   RJ   RD   R   R   (    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyt   test_chebint_axis#  s    1+1(   R   R   Ri   Rm   (    (    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyR^   Ñ   s   	Pt   TestDerivativec           B` s   e  Z d  „  Z d „  Z RS(   c      	   C` sŸ  t  t t j d g d ƒ t  t t j d g d ƒ xV t d ƒ D]H } d g | d g } t j | d d ƒ} t t | ƒ t | ƒ ƒ q? Wx t d ƒ D]q } xh t d d ƒ D]W } d g | d g } t j t j | d | ƒd | ƒ} t t | ƒ t | ƒ ƒ q® Wq˜ Wx‹ t d ƒ D]} } xt t d d ƒ D]c } d g | d g } t j t j | d | d d ƒd | d d ƒ} t t | ƒ t | ƒ ƒ q0WqWd  S(	   Ni    g      à?iÿÿÿÿi   i   R_   i   Rb   (	   R   RH   R   t   chebderR   R   R   Rc   R   (   R   R   R   R   R-   (    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyt   test_chebder6  s     $!0c         C` s¶   t  j j d ƒ } t  j g  | j D] } t j | ƒ ^ q" ƒ j } t j | d d ƒ} t | | ƒ t  j g  | D] } t j | ƒ ^ qr ƒ } t j | d d ƒ} t | | ƒ d  S(   Ni   i   Rj   i    i   (   i   i   (   R   R]   Rk   Rl   R   Ro   R   (   R   RJ   RD   R   R   (    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyt   test_chebder_axisO  s    1+(   R   R   Rp   Rq   (    (    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyRn   4  s   	t
   TestVanderc           B` s=   e  Z e j j d ƒ d d Z d „  Z d „  Z d „  Z RS(   i   i   i   i   c         C` s#  t  j d ƒ } t j | d ƒ } t | j d	 k ƒ xK t d ƒ D]= } d g | d g } t | d | f t j | | ƒ ƒ qA Wt  j	 d d g d d g d d g g ƒ } t j | d ƒ } t | j d
 k ƒ xK t d ƒ D]= } d g | d g } t | d | f t j | | ƒ ƒ qÞ Wd  S(   Ni   i   i    i   .i   i   i   (   i   i   (   i   i   i   (
   R   t   arangeR   t
   chebvanderR   RC   R   R   R?   R   (   R   R   t   vR   t   coef(    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyt   test_chebvander`  s    '*c         C` s®   |  j  \ } } } t j j d ƒ } t j | | d d g ƒ } t j | | | ƒ } t j | | j ƒ } t | | ƒ t j | g | g d d g ƒ } t	 | j
 d k ƒ d  S(   Ni   i   i   i   i   (   i   i   (   i   i   i   (   R   R   R]   R   t   chebvander2dRI   t   dott   flatR   R   RC   (   R   RL   RM   RN   RD   t   vanR   R   (    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyt   test_chebvander2dq  s    !c         C` sÀ   |  j  \ } } } t j j d ƒ } t j | | | d d d g ƒ } t j | | | | ƒ } t j | | j ƒ } t | | ƒ t j | g | g | g d d d g ƒ } t	 | j
 d k ƒ d  S(	   Ni   i   i   i   i   i   (   i   i   i   (   i   i   i   (   R   R   R]   R   t   chebvander3dRT   Ry   Rz   R   R   RC   (   R   RL   RM   RN   RD   R{   R   R   (    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyt   test_chebvander3d~  s    !*(   i   i   (   R   R   R   R]   R   Rw   R|   R~   (    (    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyRr   \  s   		t   TestFittingc           B` s   e  Z d  „  Z RS(   c      	   C` sí  d „  } d „  } t  t t j d g d g d ƒ t  t t j d g g d g d ƒ t  t t j g  d g d ƒ t  t t j d g d g g g d ƒ t  t t j d d g d g d ƒ t  t t j d g d d g d ƒ t  t t j d g d g d d d g g ƒt  t t j d g d g d d d d g ƒt  t t j d g d g d g ƒ t  t t j d g d g d d d g ƒ t  t t j d g d g g  ƒ t j d d ƒ } | | ƒ } t j | | d	 ƒ } t t | ƒ d
 ƒ t	 t j
 | | ƒ | ƒ t j | | d d d d	 g ƒ } t t | ƒ d
 ƒ t	 t j
 | | ƒ | ƒ t j | | d
 ƒ } t t | ƒ d ƒ t	 t j
 | | ƒ | ƒ t j | | d d d d	 d
 g ƒ } t t | ƒ d ƒ t	 t j
 | | ƒ | ƒ t j | | d d	 d
 d d g ƒ } t t | ƒ d ƒ t	 t j
 | | ƒ | ƒ t j | t j | | g ƒ j d	 ƒ } t	 | t j | | g ƒ j ƒ t j | t j | | g ƒ j d d d d	 g ƒ } t	 | t j | | g ƒ j ƒ t j | ƒ } | j ƒ  }	 d | d d  d … <d | d d  d … <t j | |	 d	 d | ƒ}
 t	 |
 | ƒ t j | |	 d d d d	 g d | ƒ}
 t	 |
 | ƒ t j | t j |	 |	 g ƒ j d	 d | ƒ} t	 | t j | | g ƒ j ƒ t j | t j |	 |	 g ƒ j d d d d	 g d | ƒ} t	 | t j | | g ƒ j ƒ d d d d g } t	 t j | | d ƒ d d g ƒ t	 t j | | d d g ƒ d d g ƒ t j d d ƒ } | | ƒ } t j | | d
 ƒ } t	 t j
 | | ƒ | ƒ t j | | d d d
 g ƒ } t	 t j
 | | ƒ | ƒ t	 | | ƒ d  S(   Nc         S` s   |  |  d |  d S(   Ni   i   (    (   R   (    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyt   f  s    c         S` s   |  d |  d d S(   Ni   i   i   (    (   R   (    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyt   f2’  s    i   iÿÿÿÿi    i   t   wi   i   i   i   y              ð?y              ð¿(   R   RH   R   t   chebfitt	   TypeErrorR   RA   R   t   lenR   R?   R   Rl   t
   zeros_liket   copy(   R   R€   R   R   RE   t   coef3t   coef4t   coef2dR‚   t   ywt   wcoef3t   wcoef2dt   coef1t   coef2(    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyt   test_chebfitŽ  sp    		"%""++"(!$$'3'-9"((   R   R   R   (    (    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyR   Œ  s   t   TestCompanionc           B` s#   e  Z d  „  Z d „  Z d „  Z RS(   c         C` s-   t  t t j g  ƒ t  t t j d g ƒ d  S(   Ni   (   R   RH   R   t   chebcompanion(   R   (    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyt   test_raisesÛ  s    c         C` sT   xM t  d d ƒ D]< } d g | d g } t t j | ƒ j | | f k ƒ q Wd  S(   Ni   i   i    (   R   R   R   R’   RC   (   R   R   Rv   (    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyt   test_dimensionsß  s    c         C` s'   t  t j d d g ƒ d d k ƒ d  S(   Ni   i   i    g      à¿(   i    i    (   R   R   R’   (   R   (    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyt   test_linear_rootä  s    (   R   R   R“   R”   R•   (    (    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyR‘   Ù  s   		t	   TestGaussc           B` s   e  Z d  „  Z RS(   c         C` s­   t  j d ƒ \ } } t  j | d ƒ } t j | j | | ƒ } d t j | j ƒ  ƒ } | d  d  … d  f | | } t	 | t j
 d ƒ ƒ t j } t	 | j ƒ  | ƒ d  S(   Nid   ic   i   (   R   t	   chebgaussRt   R   Ry   Rl   t   sqrtt   diagonalt   NoneR   t   eyet   pit   sum(   R   R   R‚   Ru   t   vvt   vdR   (    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyt   test_100ê  s    	(   R   R   R    (    (    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyR–   è  s   t   TestMiscc           B` sY   e  Z d  „  Z d „  Z d „  Z d „  Z d „  Z d „  Z d „  Z d „  Z	 d „  Z
 RS(	   c         C` sÂ   t  j g  ƒ } t t | ƒ d g ƒ x– t d d ƒ D]… } t j t j t j d d | d ƒ d d  d … ƒ } d g | d g } t  j | ƒ d | d } t t | ƒ t | ƒ ƒ q5 Wd  S(   Ni   i   i    i   (	   R   t   chebfromrootsR   R   R   R   t   cosRA   Rœ   (   R   R   R   t   rootsR   (    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyt   test_chebfromrootsý  s    7c         C` sœ   t  t j d g ƒ g  ƒ t  t j d d g ƒ d g ƒ x] t d d ƒ D]L } t j d d | ƒ } t j t j | ƒ ƒ } t  t | ƒ t | ƒ ƒ qH Wd  S(   Ni   i   g      à¿i   iÿÿÿÿ(   R   R   t	   chebrootsR   R   RA   R¢   R   (   R   R   R   R   (    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyt   test_chebroots  s    c         C` s   d d d d g } t  t t j | d ƒ t t j | ƒ | d  ƒ t t j | d ƒ | d  ƒ t t j | d ƒ d g ƒ d  S(   Ni   iÿÿÿÿi   i    iýÿÿÿ(   R   RH   R   R   R   (   R   Rv   (    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyt   test_chebtrim  s
    c         C` s#   t  t j d d ƒ d d g ƒ d  S(   Ni   i   (   R   R   t   chebline(   R   (    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyt   test_chebline  s    c         C` sC   x< t  d ƒ D]. } t t j d g | d g ƒ t | ƒ q Wd  S(   Ni
   i    i   (   R   R   R   Re   RB   (   R   R   (    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyt   test_cheb2poly  s    c         C` sC   x< t  d ƒ D]. } t t j t | ƒ d g | d g ƒ q Wd  S(   Ni
   i    i   (   R   R   R   Rd   RB   (   R   R   (    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyt   test_poly2cheb   s    c         C` sd   t  j d d d ƒ d d !} d t  j d | ƒ t  j d | ƒ } t j | ƒ } t | | ƒ d  S(   Niÿÿÿÿi   i   g      ð?(   R   RA   R˜   R   t
   chebweightR   (   R   R   R   R   (    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyt   test_weight$  s    (c         C` s¸   t  t t j d ƒ t  t t j d ƒ d g } t t j d ƒ | ƒ d d g } t t j d ƒ | ƒ d d d g } t t j d	 ƒ | ƒ d
 d d d g } t t j d ƒ | ƒ d  S(   Ng      ø?i    i   gÌ;fž æ¿gÌ;fž æ?i   g«LXèz¶ë¿g«LXèz¶ë?i   g( 1Ïkí¿gÅœ¦â}Ø¿gÅœ¦â}Ø?g( 1Ïkí?i   (   R   RH   R   t   chebpts1R   (   R   R   (    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyt   test_chebpts1*  s    	c         C` sÄ   t  t t j d ƒ t  t t j d ƒ d d g } t t j d ƒ | ƒ d d d g } t t j d ƒ | ƒ d d d d g } t t j d	 ƒ | ƒ d
 d d d d g } t t j d ƒ | ƒ d  S(   Ng      ø?i   iÿÿÿÿi   i    i   g      à¿g      à?i   g      ð¿g¸Kfž æ¿g¸Kfž æ?g      ð?i   (   R   RH   R   t   chebpts2R   (   R   R   (    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyt   test_chebpts29  s    (   R   R   R¥   R§   R¨   Rª   R«   R¬   R®   R°   R²   (    (    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyR¡   û  s   									t   __main__(,   t   __doc__t
   __future__R    R   R   t   numpyR   t   numpy.polynomial.chebyshevt
   polynomialt	   chebyshevR   t   numpy.polynomial.polynomialR   t   numpy.testingR   R   R   R   R   R	   R   t   T0t   T1t   T2t   T3t   T4t   T5t   T6t   T7t   T8t   T9RB   R   R   R(   R>   R^   Rn   Rr   R   R‘   R–   R¡   R   (    (    (    sD   /tmp/pip-build-X4mzal/numpy/numpy/polynomial/tests/test_chebyshev.pyt   <module>   s<   .		!$$4^c(0MM