Current File : //opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyc
�
� 9Zc @` s� d Z d d l m Z m Z m Z d d l Z d d l m Z d d l Z
d d l m Z m
Z
m Z m Z m Z m Z d d l m Z m Z m Z m Z m Z d d l m Z e e
e e e e f Z d � Z e
j j Z d d
� Z d � Z d � Z d
� Z d � Z! d � Z" d � Z# d � Z$ d � Z% d � Z& d � Z' d � Z( d � Z) d � Z* d � Z+ d � Z, d � Z- d � Z. d � Z/ d � Z0 d � Z1 d � Z2 d � Z3 d! � Z4 d" � Z5 d# � Z6 d$ � Z7 d% � Z8 e9 d& k r�e � n d S(' s� Test inter-conversion of different polynomial classes.
This tests the convert and cast methods of all the polynomial classes.
i ( t divisiont absolute_importt print_functionN( t Number( t
Polynomialt Legendret Chebyshevt Laguerret Hermitet HermiteE( t assert_almost_equalt
assert_raisest assert_equalt assert_t run_module_suite( t longc c` sf x; t D]3 } x* t D]" } t | | f Vt | | f Vq Wq Wx!t D]} t | f Vt | f Vt | f Vt | f Vt | f Vt | f Vt | f Vt
| f Vt | f Vt | f Vt
| f Vt | f Vt | f Vt | f Vt | f Vt | f Vt | f Vt | f Vt | f Vt | f Vt | f Vt | f Vt | f Vt | f Vt | f VqE Wd S( N( t classest check_conversiont
check_castt
check_callt check_identityt check_basist check_fromrootst check_fitt check_equalt check_not_equalt check_addt check_subt check_mult check_floordivt
check_truedivt check_modt check_divmodt check_powt check_integt check_derivt check_rootst check_linspacet check_mapparmst check_degreet
check_copyt check_cutdegt check_truncatet
check_trim( t Poly1t Poly2t Poly( ( sV /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyt test_class_methods s<
t c C` s� yU t t j | j | j k � � t t j | j | j k � � t | j | j � Wn/ t k
r� d | | f f } t | � � n Xd S( Ns Result: %s
Target: %s( R
t npt allt domaint windowR
t coeft AssertionError( t p1t p2t msg( ( sV /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyt assert_poly_almost_equal@ s
c
C` s� t j d d d � } t d
� } | j t d � d } | j t d � d } | | d | d | �} | j t d
� d } | j t d � d } | j d | d | d | � } t | j | � t | j | � t | | � | | � � d S( Ni i i
i i g �?R3 R4 t kind( i ( i ( i ( i ( i ( R1 t linspacet randomR3 R4 t convertR
(
R, R- t xR5 t d1t w1R7 t d2t w2R8 ( ( sV /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyR O s c
C` s� t j d d d � } t d � } | j t d
� d } | j t d � d } | | d | d | �} | j t d � d } | j t d
� d } | j | d | d | �} t | j | � t | j | � t | | � | | � � d S( Ni i i
i i g �?R3 R4 ( i ( i ( i ( i ( i ( R1 R< R= R3 R4 t castR
(
R, R- R? R5 R@ RA R7 RB RC R8 ( ( sV /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyR ` s c C` s� | j t d � d } | j t d � d } t j | d | d d � } | j d | d | � } t | j | � t | j | � t | | � | � d S(
Ni g �?i i i R3 R4 ( i ( i ( R3 R= R4 R1 R< t identityR R
( R. t dt wR? t p( ( sV /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyR v s c C` s� | j t d � d } | j t d � d } | j d d | d | �} t | j | � t | j | � t | j d g d d g � d S(
Ni g �?i R3 R4 i i ( i ( i ( R3 R= R4 t basisR R5 ( R. RF RG RH ( ( sV /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyR � s c C` s� | j t d � d } | j t d
� d } t d � } | j | d | d | �} t | j � t | � � t | j | � t | j | � t | | � d � t j } t j } t j | d | d | �} t | j
d d � d S( Ni g �?i R3 R4 i i����i ( i ( i ( i ( R3 R= R4 t fromrootsR t degreet lenR
R RD R5 ( R. RF RG t rR7 t pdomt pwinR8 ( ( sV /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyR � s c C` s� d � } t j d d � } | | � } | j | | d � } t | j d d g � t | | � | � t | j � d � | j t d
� d } | j t d � d } | j | | d d | d | �} t | | � | � t | j | � t | j | � | j | | d d d d g d | d | �} t | | � | � t | j | � t | j | � | j | | d g � } t | j | j � t | j | j � | j | | d d d d g g � } t | j | j � t | j | j � t j | � } | t | j
� d } d | d d d � <| j | d d d � | d d d � d � } | j | | d d | �} | j | | d d d d g d | �}
t | | � | | � � t | | � |
| � � d S( Nc S` s | | d | d S( Ni i ( ( R? ( ( sV /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyt f� s i i i g �?R3 R4 i RG ( i ( i ( R1 R< t fitR
R3 R RK R= R4 t
zeros_liket shape( R. RP R? t yRH RF RG t zR7 R8 t p3( ( sV /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyR � s> !-$/'c C` s� | d d d g d d d g d d d g �} | d d d g d d d g d d d g �} | d d d g d d d g d d d g �} | d d d g d d d g d d d g �} t | | k � t | | k � t | | k � t | | k � d S( Ni i i R3 i R4 ( R
( R. R7 R8 RV t p4( ( sV /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyR � s ----c C` s� | d d d g d d d g d d d g �} | d d d g d d d g d d d g �} | d d d g d d d g d d d g �} | d d d g d d d g d d d g �} t | | k � t | | k � t | | k � t | | k � d S( Ni i i R3 i R4 ( R
( R. R7 R8 RV RW ( ( sV /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyR � s ----c C` s� t t d � d � } t t d � d � } | | � } | | � } | | } t | | | � t | | | � t | | | � t | t | � | � t t | � | | � t | t j | � | � t t j | � | | � t t t j | | d g d | j
d �� t t t j | | d g d | j d �� | t k rit t t j | t
d g � � n t t t j | t d g � � d S(
Ni g �?i i R3 i R4 ( i ( i ( t listR= R: t tupleR1 t arrayR t TypeErrort opt addR3 R4 R R ( R. t c1t c2R7 R8 RV ( ( sV /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyR � s"
,, "c C` s� t t d � d � } t t d � d � } | | � } | | � } | | } t | | | � t | | | � t | | | � t | t | � | � t t | � | | � t | t j | � | � t t j | � | | � t t t j | | d g d | j
d �� t t t j | | d g d | j d �� | t k rmt t t j | t
d g � � n t t t j | t d g � � d S(
Ni g �?i i R3 i R4 ( i ( i ( RX R= R: RY R1 RZ R R[ R\ t subR3 R4 R R ( R. R^ R_ R7 R8 RV ( ( sV /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyR � s"
,, "c C` s� t t d � d � } t t d
� d � } | | � } | | � } | | } t | | | � t | | | � t | | | � t | t | � | � t t | � | | � t | t j | � | � t t j | � | | � t | d | | d g � � t d | | | d g � � t t t j | | d g d | j
d �� t t t j | | d g d | j d �� | t k r�t t t j | t
d g � � n t t t j | t d g � � d S( Ni g �?i i i R3 i R4 ( i ( i ( RX R= R: RY R1 RZ R R[ R\ t mulR3 R4 R R ( R. R^ R_ R7 R8 RV ( ( sV /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyR
s&
,, "c C` s� t t d � d � } t t d
� d � } t t d � d � } | | � } | | � } | | � } | | | } t | j � } t | | | � t | | | � t | | | � t | t | � | � t t | � | | � t | t j | � | � t t j | � | | � t d | | d g � � t | d d | � t t t j
| | d g d | j d �� t t t j
| | d g d | j d �� | t
k r�t t t j
| t d g � � n t t t j
| t
d g � � d S( Ni g �?i i i R3 i R4 ( i ( i ( i ( RX R= R5 R: RY R1 RZ R R[ R\ t floordivR3 R4 R R ( R. R^ R_ t c3R7 R8 RV RW t c4( ( sV /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyR ! s0 )) "c C` s� | d d d g � } | d } xq t j D]f } t | t � s) t | t � rT q) n | d � } t t j | | � | � t t t j | | � q) WxR t
t t f D]A } | d � } t t j | | � | � t t t j | | � q� WxO t
g D]D } | d d � } t t j | | � | � t t t j | | � q� Wxa t � t � t � t � t j d g � g D]2 } t t t j | | � t t t j | | � qhWx* t D]" } t t t j | | d � � q�Wd S( Ni i i i i ( R1 t
ScalarTypet
issubclassR t boolR: R\ t truedivR R[ t intR t floatt complexRY RX t dictRZ R ( R. R7 R8 t stypet st ptype( ( sV /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyR = s*
4
c C` s� t t d � d � } t t d
� d � } t t d � d � } | | � } | | � } | | � } | | | } t | j � } t | | | � t | | | � t | | | � t | t | � | � t t | � | | � t | t j | � | � t t j | � | | � t d | | d g � � t | d | d g � � t t t j
| | d g d | j d �� t t t j
| | d g d | j d �� | t
k r�t t t j
| t d g � � n t t t j
| t
d g � � d S( Ni g �?i i i R3 i R4 ( i ( i ( i ( RX R= R5 R: RY R1 RZ R R[ R\ t modR3 R4 R R ( R. R^ R_ Rc R7 R8 RV RW Rd ( ( sV /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyR X s, ,, "c C` s t t d � d � } t t d
� d � } t t d � d � } | | � } | | � } | | � } | | | } t | j � } t | | � \ } }
t | | � t |
| � t | | � \ } }
t | | � t |
| � t | | � \ } }
t | | � t |
| � t | t | � � \ } }
t | | � t |
| � t t | � | � \ } }
t | | � t |
| � t | t j | � � \ } }
t | | � t |
| � t t j | � | � \ } }
t | | � t |
| � t | d � \ } }
t | d | � t |
| d g � � t d | � \ } }
t | | d g � � t |
| d g � � t t t | | d g d | j
d �� t t t | | d g d | j d �� | t k r�t t t | t
d g � � n t t t | t d g � � d S( Ni g �?i i i R3 i R4 ( i ( i ( i ( RX R= R5 t divmodR: RY R1 RZ R R[ R3 R4 R R ( R. R^ R_ Rc R7 R8 RV RW Rd t quot rem( ( sV /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyR s sP
)) c C` s� | j t d � d } | j t d � d } t j t d � � } t j | j | d | d | �j � � } t | | � t j | j | � j � � } t | | � d S( Ni g �?i R3 R4 ( i ( i ( i ( R3 R= R4 R1 t sortRJ t rootsR
( R. RF RG t tgtt res( ( sV /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyR$ � s *
c C` s& | j d � } t | j � d � d S( Ni ( RI R RK ( R. RH ( ( sV /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyR' � s c C` s� | j d � } | j � } t | | k � t | | k � t | j | j k � t | j | j k � t | j | j k � d S( Ni ( RI t copyR
R5 R3 R4 ( R. R7 R8 ( ( sV /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyR( � s c C` sk t } | j | d d d
g � � } | j | j � � } | j | j d � � } t | | d d d d g � � t | | d d d d d g � � | j | d d d g � � } | j | j d d � � } | j | j d d d d g �� } t | | d d d d g � � t | | d d d d d g � � | j | d d d g � � } | j | j d d � � } | j | j d d d �� } t | | d d d d g � � t | | d d d d d g � � d | j } | j | d d d g � d
| �} | j | j � � } | j | j d � � } t | | d d d d g � � t | | d d d d d g � � d S( Ni i i i i t kt lbndi����i R3 i i i i i i i i i i i i ( R RD t integR: R3 ( R. t Pt p0R7 R8 RF ( ( sV /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyR" � s, " $ " "
$ c C` s> | j t d � d } | j t d � d } | d d d g d | d | �} | j d d d d g �} | j d d d g �} t | j d � j | j � t | j d � j | j � | d d d g � } | j d d d d g �} | j d d d g �} t | j d � j | j � t | j d � j | j � d S(
Ni g �?i i R3 R4 Ry ( i ( i ( R3 R= R4 R{ R
t derivR5 ( R. RF RG R7 R8 RV ( ( sV /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyR# � s ! c C` s | j t d � d } | j t d
� d } | d d d g d | d | �} t j | d | d d � } | | � } | j d � \ } } t | | � t | | � t j d d d � } | | � } | j d d d d g �\ } } t | | � t | | � d S( Ni g �?i i R3 R4 i i ( i ( i ( R3 R= R4 R1 R< R
( R. RF RG RH t xtgtt ytgtt xrest yres( ( sV /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyR% � s !
!
c C` s" | j t d
� d } | j t d � d } | d g d | d | �} | d d d g d | d | �} x/ t d � D]! } t | | | � | | } qw W| d g � } | d d d g � } x/ t d � D]! } t | | | � | | } q� Wt t t j | d � t t t j | d � d S( Ni g �?i R3 R4 i i g �?i����( i ( i ( R3 R= R4 t rangeR: R t
ValueErrorR\ t pow( R. RF RG Rv t tstt i( ( sV /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyR! � s !c C` s} t } | j } t j | d | d d � } | j | d d d g � � } d | d d | } | | � } t | | � d S( Ni i i i i ( R R3 R1 R< RD R
( R. R| RF R? RH Rv Rw ( ( sV /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyR s c C` s� | d d d g � } t t | j d � t t | j d � t t | j d � � d � t t | j d � � d � t t | j d � � d � t t | j d � � d � d S( Ni i i g �?i����i ( R R� t cutdegR RL ( R. RH ( ( sV /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyR) s c C` s� | d d d g � } t t | j d � t t | j d � t t | j d � � d � t t | j d � � d � t t | j d � � d � t t | j d � � d � d S( Ni i i g �?i i ( R R� t truncateR RL ( R. RH ( ( sV /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyR* $ s c C` sv d d d d g } | | � } t | j � j | d � t | j d � j | d � t | j d � j | d � d S( Ni g���ư>g�-���q=i i g�����|�=i g�h㈵��>( R t trimR5 ( R. t cRH ( ( sV /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyR+ . s
c C` s� | j } | j } | d g d | d | �} t d d g | j � � d | d } | d g d | d | �} t d d g | j � � d S( Ni R3 R4 i i ( R3 R4 R
t mapparms( R. RF RG RH ( ( sV /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyR& 6 s t __main__(: t __doc__t
__future__R R R t operatorR\ t numbersR t numpyR1 t numpy.polynomialR R R R R R t
numpy.testingR
R R R
R t numpy.compatR R R/ R= R: R R R R R R R R R R R R R R R R$ R' R( R" R# R% R! R R) R* R+ R& t __name__( ( ( sV /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyt <module> sR .( $
, -
Mini Shell