| Current File : //usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyo |
�
E�`Qc
@ sv d 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 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' e( d5 k rre � n d S(6 s Tests for chebyshev module.
i����( t divisionN( 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( ( sK /usr/lib64/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( ( sK /usr/lib64/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 ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt test__zseries_to_cseries' s
.#( t __name__t
__module__R R ( ( ( sK /usr/lib64/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 ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt test_chebdomain1 s c C s t t j d g � d S( Ni ( R R t chebzero( R ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt
test_chebzero4 s c C s t t j d g � d S( Ni ( R R t chebone( R ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt test_chebone7 s c C s t t j d d g � d S( Ni i ( R R t chebx( R ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt
test_chebx: s ( R R R R! R# R% ( ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyR / 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 ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt test_chebadd@ 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 ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt test_chebsubJ 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 ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt
test_chebmulxT 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 ( ( sK /usr/lib64/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 chebdivR4 R R (
R R R+ R, t cit cjR t quot remR ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt test_chebdivf s ( R R R- R/ R2 R5 R; ( ( ( sK /usr/lib64/python2.7/site-packages/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( ( sK /usr/lib64/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 RC R t
ValueErrorR t chebval2dt c2dR R t onesR RA (
R t x1t x2t x3t y1t y2t y3R R t z( ( sK /usr/lib64/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 RC R RF R t chebval3dt c3dR R RI R RA (
R RJ RK RL RM RN RO R R RP ( ( sK /usr/lib64/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 RC R t einsumR t
chebgrid2dRH R RI R RA (
R RJ RK RL RM RN RO R R RP ( ( sK /usr/lib64/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 RC R RU R t
chebgrid3dRS R RI R RA (
R RJ RK RL RM RN RO R R RP ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt test_chebgrid3d� s
( i i ( R R R R t c1dRU RH RS t randomR R RC RE RQ RT RW RY ( ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyR< r 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 | � | � qmWx� 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 | � � qYWqCWx� t d � D]� } x� t d d � D]� } d g | d g } | } x8 t | � D]* } t j | d d d | g d d �} qBWt j | d | d t | � d d �} t t | � t | � � qWq�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 | � d
d �} t t | � t | � � q�Wq�Wd S( Ni g �?i����i i i t mt kt lbndt scl(
R RF R t chebintR R t poly2chebt cheb2polyR R= (
R R R^ R R` t polR t chebpolRa R+ ( ( sK /usr/lib64/python2.7/site-packages/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 Ra R ( R RH RB R R ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt test_chebint_axis# s 1
+
1( R R Rf Rj ( ( ( sK /usr/lib64/python2.7/site-packages/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 | � � q0WqWd S( Ni g �?i����i i R] i R` ( R RF R t chebderR R R Ra R ( R R R R R+ ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt test_chebder6 s $!0c 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 Rg i i ( i i ( R R[ Rh Ri R Rl R ( R RH RB R R ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt test_chebder_axisO s 1
+( R R Rm Rn ( ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyRk 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 RA R R R= R ( R R t vR t coef( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt test_chebvandera 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 chebvander2dRG t dott flatR R RA ( R RJ RK RL RB t vanR R ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt test_chebvander2dr 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 chebvander3dRR Rv Rw R R RA ( R RJ RK RL RB Rx R R ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt test_chebvander3d� s !
*( i i ( R R R R[ R Rt Ry R{ ( ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyRo \ s t TestFittingc B s e Z d � Z RS( c C s� 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 j d d � } | | � } t j | | d � } t t | � d � t t j
| | � | � t j | | d � } t t | � d � t t j
| | � | � t j | t j | | g � j d � } t | t j | | g � j � t j
| � } | j � } d | d d d � <d | d d d � <t j | | d d | �} t | | � t j | t j | | g � j d d | �}
t |
t j | | g � j � d d
d d g } t t j | | d � d d g � d S( Nc S s | | d | d S( Ni i ( ( R ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt f� s i i����i i t wi i i y �?y �( R RF R t chebfitt TypeErrorR R? R t lenR R= R Ri t
zeros_liket copy( R R} R RC t coef3t coef4t coef2dR~ t ywt wcoef3t wcoef2d( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt test_chebfit� s: " %""++ '
- ( R R R� ( ( ( sK /usr/lib64/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 chebgaussRq R Rv Ri t sqrtt diagonalt NoneR t eyet pit sum( R R R~ Rr t vvt vdR ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt test_100� s ( R R R� ( ( ( sK /usr/lib64/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 cosR? R� ( R R R t rootsR ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt test_chebfromroots� s 7c 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 R? R� R ( R R R R ( ( sK /usr/lib64/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 RF R R
R ( R Rs ( ( sK /usr/lib64/python2.7/site-packages/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 ( ( sK /usr/lib64/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 Rc R@ ( R R ( ( sK /usr/lib64/python2.7/site-packages/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 Rb R@ ( R R ( ( sK /usr/lib64/python2.7/site-packages/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 R? R� R t
chebweightR ( R R R R ( ( sK /usr/lib64/python2.7/site-packages/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 RF R t chebpts1R ( R R ( ( sK /usr/lib64/python2.7/site-packages/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�Kf���g�Kf���?g �?i ( R RF R t chebpts2R ( R R ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt
test_chebpts2
s ( R R R� R� R� R� R� R� R� R� R� ( ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyR� � s t __main__() t __doc__t
__future__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 T9R@ R
R R& R< R\ Rk Ro R| R� R� R ( ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_chebyshev.pyt <module> s: . !$$4_c(2-N