ó
ÐEe]c        
   @` sœ  d  Z  d d l m Z m Z m Z d d l 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 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' d0 e  f d1 „  ƒ  YZ( d2 e  f d3 „  ƒ  YZ) d4 e  f d5 „  ƒ  YZ* d6 e  f d7 „  ƒ  YZ+ d8 e  f d9 „  ƒ  YZ, d S(:   s   Tests for chebyshev module.

i    (   t   divisiont   absolute_importt   print_function(   t   reduceN(   t   polyval(   t   assert_almost_equalt   assert_raisest   assert_equalt   assert_c         C` s   t  j |  d d ƒS(   Nt   tolgíµ ÷Æ°>(   t   chebt   chebtrim(   t   x(    (    sD   lib/python2.7/site-packages/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   lib/python2.7/site-packages/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   lib/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt   test__zseries_to_cseries*   s
    .#(   t   __name__t
   __module__R   R   (    (    (    sD   lib/python2.7/site-packages/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   lib/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt   test_chebdomain4   s    c         C` s   t  t j d g ƒ d  S(   Ni    (   R   R
   t   chebzero(   R   (    (    sD   lib/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt   test_chebzero7   s    c         C` s   t  t j d g ƒ d  S(   Ni   (   R   R
   t   chebone(   R   (    (    sD   lib/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt   test_chebone:   s    c         C` s   t  t j d d g ƒ d  S(   Ni    i   (   R   R
   t   chebx(   R   (    (    sD   lib/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt
   test_chebx=   s    (   R   R   R    R"   R$   R&   (    (    (    sD   lib/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyR   2   s   			t   TestArithmeticc           B` s>   e  Z d  „  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   lib/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt   test_chebaddC   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   lib/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt   test_chebsubM   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   lib/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt   test_chebmulxW   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   lib/python2.7/site-packages/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   chebdivR5   R   R   (
   R   R   R,   R-   t   cit   cjR   t   quot   remR   (    (    sD   lib/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt   test_chebdivi   s    c         C` s®   x§ t  d ƒ D]™ } x t  d ƒ D]‚ } d | | f } t j | d ƒ } t t j | g | t j d g ƒ ƒ } t j | | ƒ } t t	 | ƒ t	 | ƒ d | ƒq  Wq Wd  S(   Ni   s   At i=%d, j=%di   R(   (
   R   R   t   arangeR   R
   R5   R   t   chebpowR   R   (   R   R   R,   R-   t   cR   R   (    (    sD   lib/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt   test_chebpowt   s    ((   R   R   R.   R0   R3   R6   R<   R@   (    (    (    sD   lib/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyR'   A   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   R?   t   yR   R-   R   R   t   dims(    (    sD   lib/python2.7/site-packages/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   RG   R   t
   ValueErrorR
   t	   chebval2dt   c2dR   R   t   onesR   RF   (
   R   t   x1t   x2t   x3t   y1t   y2t   y3R   R   t   z(    (    sD   lib/python2.7/site-packages/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   RG   R   RJ   R
   t	   chebval3dt   c3dR   R   RM   R   RF   (
   R   RN   RO   RP   RQ   RR   RS   R   R   RT   (    (    sD   lib/python2.7/site-packages/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   RG   R   t   einsumR
   t
   chebgrid2dRL   R   RM   R   RF   (
   R   RN   RO   RP   RQ   RR   RS   R   R   RT   (    (    sD   lib/python2.7/site-packages/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   RG   R   RY   R
   t
   chebgrid3dRW   R   RM   R   RF   (
   R   RN   RO   RP   RQ   RR   RS   R   R   RT   (    (    sD   lib/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt   test_chebgrid3dÍ   s    (   i   i   (   R   R   R   R   t   c1dRY   RL   RW   t   randomR   R   RG   RI   RU   RX   R[   R]   (    (    (    sD   lib/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyRA   ~   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 ƒ t  t t j d g d d g ƒt  t t j d g d d g ƒt  t t j d g d d ƒx` t d d	 ƒ D]O } d g | d d g } t j d g d
 | d | ƒ} t | d d g ƒ q¾ 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 | ƒ | ƒ qÇ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 | ƒ ƒ qH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 | ƒ ƒ 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 | ƒ ƒ qt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 ƒ} qoWt j | d
 |	 d t
 t |	 ƒ ƒ d d ƒ} t t | ƒ t | ƒ ƒ qAWq+Wd  S(   Ni    g      à?iÿÿÿÿi   t   lbndt   sclt   axisi   i   t   mt   k(   R   RJ   R
   t   chebintR   R   t	   poly2chebt	   cheb2polyR   RB   t   list(
   R   R   Re   R   Rb   t   polR   t   chebpolRf   R,   (    (    sD   lib/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt   test_chebintÞ   s€    "

$
$!"'!(-!(-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   Rc   i    i   Re   (   i   i   (   R   R_   t   vstackt   TR
   Rf   R   (   R   RL   R?   R   R   (    (    sD   lib/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt   test_chebint_axis1  s    1+1(   R   R   Rl   Ro   (    (    (    sD   lib/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyR`   Ü   s   	St   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   Rd   i   Rb   (	   R   RJ   R
   t   chebderR   R   R   Rf   R   (   R   R   R   R   R,   (    (    sD   lib/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt   test_chebderD  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   Rc   i    i   (   i   i   (   R   R_   Rm   Rn   R
   Rq   R   (   R   RL   R?   R   R   (    (    sD   lib/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt   test_chebder_axis]  s    1+(   R   R   Rr   Rs   (    (    (    sD   lib/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyRp   B  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   R=   R
   t
   chebvanderR   RF   R   R   RB   R   (   R   R   t   vR   t   coef(    (    sD   lib/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt   test_chebvandern  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   chebvander2dRK   t   dott   flatR   R   RF   (   R   RN   RO   RP   R?   t   vanR   R   (    (    sD   lib/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt   test_chebvander2d  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   chebvander3dRV   Rz   R{   R   R   RF   (   R   RN   RO   RP   R?   R|   R   R   (    (    sD   lib/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt   test_chebvander3dŒ  s    !*(   i   i   (   R   R   R   R_   R   Rx   R}   R   (    (    (    sD   lib/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyRt   j  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   lib/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt   f  s    c         S` s   |  d |  d d S(   Ni   i   i   (    (   R   (    (    sD   lib/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt   f2   s    i   iÿÿÿÿi    i   t   wi   i   i   i   y              ð?y              ð¿(   R   RJ   R
   t   chebfitt	   TypeErrorR   RD   R   t   lenR   RB   R   Rn   t
   zeros_liket   copy(   R   R   R‚   R   RG   t   coef3t   coef4t   coef2dRƒ   t   ywt   wcoef3t   wcoef2dt   coef1t   coef2(    (    sD   lib/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt   test_chebfitœ  sp    		"%""++"(!$$'3'-9"((   R   R   R‘   (    (    (    sD   lib/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyR€   š  s   t   TestInterpolatec           B` s,   e  Z d  „  Z d „  Z d „  Z d „  Z RS(   c         C` s   | | d | d S(   Ni   i   (    (   R   R   (    (    sD   lib/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyR   é  s    c         C` s6   t  t t j |  j d ƒ t  t t j |  j d ƒ d  S(   Niÿÿÿÿg      $@(   R   RJ   R
   t   chebinterpolateR   R…   (   R   (    (    sD   lib/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt   test_raisesì  s    c         C` sG   x@ t  d d ƒ D]/ } t t j |  j | ƒ j | d f k ƒ q Wd  S(   Ni   i   (   R   R   R
   R“   R   RF   (   R   t   deg(    (    sD   lib/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt   test_dimensionsð  s    c         C` sš   d „  } t  j d d d ƒ } xu t d d ƒ D]d } x[ t d | d ƒ D]F } t j | | | f ƒ } t t j | | ƒ | | | ƒ d d ƒqH Wq. Wd  S(   Nc         S` s   |  | S(   N(    (   R   t   p(    (    sD   lib/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt   powxö  s    iÿÿÿÿi   i
   i    t   decimali   (   R   RD   R   R
   R“   R   RB   (   R   R˜   R   R•   R—   R?   (    (    sD   lib/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt   test_approximationô  s    	(   R   R   R   R”   R–   Rš   (    (    (    sD   lib/python2.7/site-packages/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   RJ   R
   t   chebcompanion(   R   (    (    sD   lib/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyR”     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œ   RF   (   R   R   Rw   (    (    sD   lib/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyR–     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   lib/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt   test_linear_root  s    (   R   R   R”   R–   R   (    (    (    sD   lib/python2.7/site-packages/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	   chebgaussRu   R   Rz   Rn   t   sqrtt   diagonalt   NoneR   t   eyet   pit   sum(   R   R   Rƒ   Rv   t   vvt   vdR   (    (    sD   lib/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt   test_100  s    	(   R   R   R¨   (    (    (    sD   lib/python2.7/site-packages/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   cosRD   R¤   (   R   R   R   t   rootsR   (    (    sD   lib/python2.7/site-packages/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   RD   Rª   R   (   R   R   R   R   (    (    sD   lib/python2.7/site-packages/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   RJ   R
   R   R   (   R   Rw   (    (    sD   lib/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt   test_chebtrim5  s
    c         C` s#   t  t j d d ƒ d d g ƒ d  S(   Ni   i   (   R   R
   t   chebline(   R   (    (    sD   lib/python2.7/site-packages/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
   Rh   RE   (   R   R   (    (    sD   lib/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt   test_cheb2polyC  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
   Rg   RE   (   R   R   (    (    sD   lib/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt   test_poly2chebG  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   RD   R    R
   t
   chebweightR   (   R   R   R   R   (    (    sD   lib/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt   test_weightK  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   RJ   R
   t   chebpts1R   (   R   R   (    (    sD   lib/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt   test_chebpts1Q  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   RJ   R
   t   chebpts2R   (   R   R   (    (    sD   lib/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt   test_chebpts2`  s    (   R   R   R­   R¯   R°   R²   R³   R´   R¶   R¸   Rº   (    (    (    sD   lib/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyR©   "  s   									(-   t   __doc__t
   __future__R    R   R   t	   functoolsR   t   numpyR   t   numpy.polynomial.chebyshevt
   polynomialt	   chebyshevR
   t   numpy.polynomial.polynomialR   t   numpy.testingR   R   R   R   R   t   T0t   T1t   T2t   T3t   T4t   T5t   T6t   T7t   T8t   T9RE   t   objectR   R   R'   RA   R`   Rp   Rt   R€   R’   R›   Rž   R©   (    (    (    sD   lib/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt   <module>   s<   "		!$$=^f(0M