| Current File : //usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyc |
�
E�`Qc @ s� d 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 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 � Z, d � Z- d � Z. d � Z/ d � Z0 d! � Z1 d" � Z2 d# � Z3 e4 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 divisionN( t
Polynomialt Legendret Chebyshevt Laguerret Hermitet HermiteE( t TestCaset assert_almost_equalt
assert_raisest assert_equalt assert_t run_module_suitet dec( t KnownFailurec c s[ x; t D]3 } x* t D]" } t | | f Vt | | f Vq Wq Wxt 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 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_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( ( sI /usr/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( ( sI /usr/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 �?R1 R2 t kind( i ( i ( i ( i ( i ( R/ t linspacet randomR1 R2 t convertR (
R* R+ t xR3 t d1t w1R5 t d2t w2R6 ( ( sI /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyR M 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 �?R1 R2 ( i ( i ( i ( i ( i ( R/ R: R; R1 R2 t castR (
R* R+ R= R3 R> R? R5 R@ RA R6 ( ( sI /usr/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 R1 R2 ( i ( i ( R1 R; R2 R/ R: t identityR
R ( R, t dt wR= t p( ( sI /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyR t 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 R1 R2 i i ( i ( i ( R1 R; R2 t basisR
R3 ( R, RD RE RF ( ( sI /usr/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 | d | d | �} t | j
d d � d S( Ni g �?i R1 R2 i i����i ( i ( i ( i ( R1 R; R2 t fromrootsR
t degreet lenR R RB R3 ( R, RD RE t rR5 R6 ( ( sI /usr/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 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 | �} t | | � | | � � d S( Nc S s | | d | d S( Ni i ( ( R= ( ( sI /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyt f� s i i i g �?R1 R2 i RE ( i ( i ( R/ R: t fitR R1 R
RI R; R2 t
zeros_liket shape(
R, RL R= t yRF RD RE t zR5 R6 ( ( sI /usr/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 R1 i R2 ( R ( R, R5 R6 t p3t p4( ( sI /usr/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 R1 i R2 ( R ( R, R5 R6 RR RS ( ( sI /usr/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 | j | d g d | j d �� t t | j | d g d | j
d �� | t k r`t t | j t d g � � n t t | j t d g � � d S(
Ni g �?i i R1 i R2 ( i ( i (
t listR; R8 t tupleR/ t arrayR t TypeErrort __add__R1 R2 R R ( R, t c1t c2R5 R6 RR ( ( sI /usr/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 | j | d g d | j d �� t t | j | d g d | j
d �� | t k rdt t | j t d g � � n t t | j t d g � � d S(
Ni g �?i i R1 i R2 ( i ( i (
RT R; R8 RU R/ RV R RW t __sub__R1 R2 R R ( R, RY RZ R5 R6 RR ( ( sI /usr/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 | j | d g d | j d �� t t | j | d g d | j
d �� | t k r�t t | j t d g � � n t t | j t d g � � d S( Ni g �?i i i R1 i R2 ( i ( i (
RT R; R8 RU R/ RV R RW t __mul__R1 R2 R R ( R, RY RZ R5 R6 RR ( ( sI /usr/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 | j | d g d | j
d �� t t | j | d g d | j d �� | t k r�t t | j t
d g � � n t t | j t d g � � d S( Ni g �?i i i R1 i R2 ( i ( i ( i ( RT R; R3 R8 RU R/ RV R RW t __floordiv__R1 R2 R R ( R, RY RZ t c3R5 R6 RR RS t c4( ( sI /usr/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 g � � t t | j | d g d | j
d �� t t | j | d g d | j d �� | t k r�t t | j t
d g � � n t t | j t d g � � d S( Ni g �?i i i R1 i R2 ( i ( i ( i ( RT R; R3 R8 RU R/ RV R RW t __mod__R1 R2 R R ( R, RY RZ R^ R5 R6 RR RS R_ ( ( sI /usr/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 |
| � 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 R1 i R2 ( i ( i ( i ( RT R; R3 t divmodR8 RU R/ RV R RW R1 R2 R R ( R, RY RZ R^ R5 R6 RR RS R_ t quot rem( ( sI /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyR J sP
)) c C sr | j t d � d } | j t d � d } t j t d � � } t j | j | � j � � } t | | � d S( Ni g �?i ( i ( i ( i ( R1 R; R2 R/ t sortRH t rootsR ( R, RD RE t tgtt res( ( sI /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyR" w s
c C s& | j d � } t | j � d � d S( Ni ( RG R
RI ( R, RF ( ( sI /usr/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 ( RG t copyR R3 R1 R2 ( R, R5 R6 ( ( sI /usr/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 R1 i i i i i i i i i i i i ( R RB t integR8 R1 ( R, t Pt p0R5 R6 RD ( ( sI /usr/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 S(
Ni g �?i i R1 R2 Ri ( i ( i ( R1 R; R2 Rk R t derivR3 ( R, RD RE R5 R6 RR ( ( sI /usr/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 R1 R2 i i ( i ( i ( R1 R; R2 R/ R: R ( R, RD RE RF t xtgtt ytgtt xrest yres( ( sI /usr/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 Wt t | j d � t t | j d � d S( Ni g �?i R1 R2 i i g �?i����( i ( i ( R1 R; R2 t rangeR8 R t
ValueErrort __pow__( R, RD RE Rf t tstt i( ( sI /usr/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 R1 R/ R: RB R ( R, Rl RD R= RF Rf Rg ( ( sI /usr/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 Rt t cutdegR
RJ ( R, RF ( ( sI /usr/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 Rt t truncateR
RJ ( R, RF ( ( sI /usr/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 trimR3 ( R, t cRF ( ( sI /usr/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 R1 R2 i i ( R1 R2 R t mapparms( R, RD RE RF ( ( sI /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyR$ � s t __main__(5 t __doc__t
__future__R t numpyR/ t numpy.polynomialR R R R R R t
numpy.testingR R R R
R R R
t numpy.testing.noseclassesR R R- R; R8 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__( ( ( sI /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyt <module> sL .4 %
# -