ó É9Zc@`södZddlmZmZmZddlZddlmZddl Z ddl m Z m Z mZmZmZmZddlmZmZmZmZmZddlmZe e eeeefZd„Ze jjZd d „Zd „Zd „Zd „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„Z0d„Z1d„Z2d „Z3d!„Z4d"„Z5d#„Z6d$„Z7d%„Z8e9d&kròeƒndS('s€Test inter-conversion of different polynomial classes. This tests the convert and cast methods of all the polynomial classes. i(tdivisiontabsolute_importtprint_functionN(tNumber(t PolynomialtLegendret ChebyshevtLaguerretHermitetHermiteE(tassert_almost_equalt assert_raisest assert_equaltassert_trun_module_suite(tlongcc`sfx;tD]3}x*tD]"}t||fVt||fVqWqWx!tD]}t|fVt|fVt|fVt|fVt|fVt|fVt |fVt |fVt |fVt |fVt |fVt|fVt|fVt|fVt|fVt|fVt|fVt|fVt|fVt|fVt|fVt|fVt|fVt|fVt|fVqEWdS(N(tclassestcheck_conversiont check_castt check_calltcheck_identityt check_basistcheck_fromrootst check_fitt check_equaltcheck_not_equalt check_addt check_subt check_multcheck_floordivt check_truedivt check_modt check_divmodt check_powt check_integt check_derivt check_rootstcheck_linspacetcheck_mapparmst check_degreet check_copyt check_cutdegtcheck_truncatet check_trim(tPoly1tPoly2tPoly((sV/opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyttest_class_methodss<                           tcC`s‹yUttj|j|jkƒƒttj|j|jkƒƒt|j|jƒWn/tk r†d||ff}t|ƒ‚nXdS(NsResult: %s Target: %s(R tnptalltdomaintwindowR tcoeftAssertionError(tp1tp2tmsg((sV/opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pytassert_poly_almost_equal@s c C`sðtjdddƒ}td ƒ}|jtd ƒd}|jtd ƒd}||d|d|ƒ}|jtd ƒd}|jtdƒd}|jd |d|d|ƒ} t| j|ƒt| j|ƒt| |ƒ||ƒƒdS(Niii iigÐ?R3R4tkind(i(i(i(i(i(R1tlinspacetrandomR3R4tconvertR ( R,R-txR5td1tw1R7td2tw2R8((sV/opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyROs c C`sítjdddƒ}td ƒ}|jtd ƒd}|jtd ƒd}||d|d|ƒ}|jtd ƒd}|jtd ƒd}|j|d|d|ƒ} t| j|ƒt| j|ƒt| |ƒ||ƒƒdS(Niii iigÐ?R3R4(i(i(i(i(i(R1R<R=R3R4tcastR ( R,R-R?R5R@RAR7RBRCR8((sV/opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyR`s cC`sš|jtdƒd}|jtd ƒd}tj|d|ddƒ}|jd|d|ƒ}t|j|ƒt|j|ƒt||ƒ|ƒdS( NigÐ?iii R3R4(i(i(R3R=R4R1R<tidentityR R (R.tdtwR?tp((sV/opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyRvscC`s‹|jtdƒd}|jtd ƒd}|jdd|d|ƒ}t|j|ƒt|j|ƒt|jdgddgƒdS( NigÐ?iR3R4ii(i(i(R3R=R4tbasisR R5(R.RFRGRH((sV/opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyR€s cC`sæ|jtd ƒd}|jtd ƒd}td ƒ}|j|d|d|ƒ}t|jƒt|ƒƒt|j|ƒt|j|ƒt||ƒdƒtj}tj}tj |d|d|ƒ}t|j ddƒdS( NigÐ?iR3R4iiÿÿÿÿi(i(i(i( R3R=R4t fromrootsR tdegreetlenR RRDR5(R.RFRGtrR7tpdomtpwinR8((sV/opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyR‰s   c C`sÂd„}tjddƒ}||ƒ}|j||dƒ}t|jddgƒt||ƒ|ƒt|jƒdƒ|jtd ƒd}|jtd ƒd}|j||dd|d|ƒ}t||ƒ|ƒt|j|ƒt|j|ƒ|j||ddddgd|d|ƒ}t||ƒ|ƒt|j|ƒt|j|ƒ|j||dgƒ}t|j|jƒt|j|jƒ|j||ddddggƒ}t|j|jƒt|j|jƒtj |ƒ}|t|j ƒd}d|ddd…<|j|ddd…|ddd…dƒ}|j||dd |ƒ} |j||ddddgd |ƒ} t||ƒ| |ƒƒt| |ƒ| |ƒƒdS( NcS`s||d|dS(Nii((R?((sV/opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pytfžsiiigÐ?R3R4iRG(i(i( R1R<tfitR R3R RKR=R4t zeros_liketshape( R.RPR?tyRHRFRGtzR7R8tp3((sV/opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyRœs>  !-$/'cC`sû|dddgdddgdddgƒ}|dddgdddgdddgƒ}|dddgdddgdddgƒ}|dddgdddgdddgƒ}t||kƒt||k ƒt||k ƒt||k ƒdS(NiiiR3iR4(R (R.R7R8RVtp4((sV/opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyRÈs----cC`sù|dddgdddgdddgƒ}|dddgdddgdddgƒ}|dddgdddgdddgƒ}|dddgdddgdddgƒ}t||k ƒt||kƒt||kƒt||kƒdS(NiiiR3iR4(R (R.R7R8RVRW((sV/opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyRÓs----c C`sŒttdƒdƒ}ttd ƒdƒ}||ƒ}||ƒ}||}t|||ƒt|||ƒt|||ƒt|t|ƒ|ƒtt|ƒ||ƒt|tj|ƒ|ƒttj|ƒ||ƒtttj ||dgd|j dƒƒtttj ||dgd|j dƒƒ|t kritttj |t dgƒƒntttj |t dgƒƒdS( Nigà?iiR3iR4(i(i(tlistR=R:ttupleR1tarrayR t TypeErrortoptaddR3R4RR(R.tc1tc2R7R8RV((sV/opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyRÞs"   ,, "c C`sttdƒdƒ}ttd ƒdƒ}||ƒ}||ƒ}||}t||| ƒt|||ƒt||| ƒt|t|ƒ|ƒtt|ƒ|| ƒt|tj|ƒ|ƒttj|ƒ|| ƒtttj ||dgd|j dƒƒtttj ||dgd|j dƒƒ|t krmtttj |t dgƒƒntttj |t dgƒƒdS( Nigà?iiR3iR4(i(i(RXR=R:RYR1RZR R[R\tsubR3R4RR(R.R^R_R7R8RV((sV/opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyRôs"   ,, "c C`sÈttd ƒdƒ}ttd ƒdƒ}||ƒ}||ƒ}||}t|||ƒt|||ƒt|||ƒt|t|ƒ|ƒtt|ƒ||ƒt|tj|ƒ|ƒttj|ƒ||ƒt|d||dgƒƒtd|||dgƒƒtttj ||dgd|j dƒƒtttj ||dgd|j dƒƒ|t kr¥tttj |t dgƒƒntttj |t dgƒƒdS( Nigà?iiiR3iR4(i(i(RXR=R:RYR1RZR R[R\tmulR3R4RR(R.R^R_R7R8RV((sV/opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyR s&   ,, "c C`sðttd ƒdƒ}ttd ƒdƒ}ttd ƒdƒ}||ƒ}||ƒ}||ƒ}|||}t|jƒ}t|||ƒt|||ƒt|||ƒt|t|ƒ|ƒtt|ƒ||ƒt|tj|ƒ|ƒttj|ƒ||ƒtd||dgƒƒt|dd|ƒttt j ||dgd|j dƒƒttt j ||dgd|j dƒƒ|t krÍttt j |tdgƒƒnttt j |t dgƒƒdS( Nigà?iiiR3iR4(i(i(i(RXR=R5R:RYR1RZR R[R\tfloordivR3R4RR( R.R^R_tc3R7R8RVRWtc4((sV/opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyR!s0   )) "cC`sÏ|dddgƒ}|d}xqtjD]f}t|tƒ s)t|tƒrTq)n|dƒ}ttj||ƒ|ƒtt tj||ƒq)WxRt t t fD]A}|dƒ}ttj||ƒ|ƒtt tj||ƒq£WxOt gD]D}|ddƒ}ttj||ƒ|ƒtt tj||ƒqòWxatƒtƒtƒtƒtjdgƒgD]2}tt tj||ƒtt tj||ƒqhWx*tD]"}tt tj||dƒƒq¥WdS(Niiiii(R1t ScalarTypet issubclassRtboolR:R\ttruedivR R[tintRtfloattcomplexRYRXtdictRZR(R.R7R8tstypetstptype((sV/opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyR=s*   4 c C`sõttd ƒdƒ}ttd ƒdƒ}ttd ƒdƒ}||ƒ}||ƒ}||ƒ}|||}t|jƒ}t|||ƒt|||ƒt|||ƒt|t|ƒ|ƒtt|ƒ||ƒt|tj|ƒ|ƒttj|ƒ||ƒtd||dgƒƒt|d|dgƒƒttt j ||dgd|j dƒƒttt j ||dgd|j dƒƒ|t krÒttt j |tdgƒƒnttt j |t dgƒƒdS( Nigà?iiiR3iR4(i(i(i(RXR=R5R:RYR1RZR R[R\tmodR3R4RR( R.R^R_RcR7R8RVRWRd((sV/opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyRXs,   ,, "c C`sttd ƒdƒ}ttd ƒdƒ}ttd ƒdƒ}||ƒ}||ƒ}||ƒ}|||}t|jƒ}t||ƒ\} } t| |ƒt| |ƒt||ƒ\} } t| |ƒt| |ƒt||ƒ\} } t| |ƒt| |ƒt|t|ƒƒ\} } t| |ƒt| |ƒtt|ƒ|ƒ\} } t| |ƒt| |ƒt|tj|ƒƒ\} } t| |ƒt| |ƒttj|ƒ|ƒ\} } t| |ƒt| |ƒt|dƒ\} } t| d|ƒt| |dgƒƒtd|ƒ\} } t| |dgƒƒt| |dgƒƒtt t||dgd|j dƒƒtt t||dgd|j dƒƒ|t krätt t|t dgƒƒntt t|t dgƒƒdS( Nigà?iiiR3iR4(i(i(i(RXR=R5tdivmodR:RYR1RZR R[R3R4RR( R.R^R_RcR7R8RVRWRdtquotrem((sV/opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyR ssP                 )) cC`s©|jtdƒd}|jtdƒd}tjtdƒƒ}tj|j|d|d|ƒjƒƒ}t||ƒtj|j|ƒjƒƒ}t||ƒdS( NigÐ?iR3R4(i(i(i(R3R=R4R1tsortRJtrootsR (R.RFRGttgttres((sV/opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyR$ s* cC`s&|jdƒ}t|jƒdƒdS(Ni(RIR RK(R.RH((sV/opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyR'«scC`s|jdƒ}|jƒ}t||kƒt||k ƒt|j|jk ƒt|j|jk ƒt|j|jk ƒdS(Ni(RItcopyR R5R3R4(R.R7R8((sV/opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyR(°s cC`skt}|j|d d d gƒƒ}|j|jƒƒ}|j|jdƒƒ}t||ddddgƒƒt||dddddgƒƒ|j|dddgƒƒ}|j|jddƒƒ}|j|jddddgƒƒ}t||ddddgƒƒt||dddddgƒƒ|j|dddgƒƒ}|j|jddƒƒ}|j|jdddƒƒ}t||ddddgƒƒt||d ddddgƒƒd|j}|j|dddgƒd |ƒ}|j|jƒƒ}|j|jdƒƒ}t||ddddgƒƒt||dddddgƒƒdS(Niiiiitktlbndi÷ÿÿÿiR3iii iii iii iii (RRDtintegR:R3(R.tPtp0R7R8RF((sV/opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyR"ºs,"$"" $cC`s>|jtdƒd}|jtd ƒd}|dddgd|d|ƒ}|jddddgƒ}|jdddgƒ}t|jdƒj|jƒt|jdƒj|jƒ|dddgƒ}|jddddgƒ}|jdddgƒ}t|jdƒj|jƒt|jdƒj|jƒdS( NigÐ?iiR3R4Ry(i(i(R3R=R4R{R tderivR5(R.RFRGR7R8RV((sV/opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyR#×s!cC`s|jtd ƒd}|jtd ƒd}|dddgd|d|ƒ}tj|d|ddƒ}||ƒ}|jdƒ\}}t||ƒt||ƒtjdddƒ}||ƒ}|jddddgƒ\}}t||ƒt||ƒdS( NigÐ?iiR3R4ii(i(i(R3R=R4R1R<R (R.RFRGRHtxtgttytgttxrestyres((sV/opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyR%és!    ! cC`s"|jtd ƒd}|jtd ƒd}|dgd|d|ƒ}|dddgd|d|ƒ}x/tdƒD]!}t|||ƒ||}qwW|dgƒ}|dddgƒ}x/tdƒD]!}t|||ƒ||}qÍWtttj|dƒtttj|d ƒdS( NigÐ?iR3R4iigø?iÿÿÿÿ(i(i( R3R=R4trangeR:R t ValueErrorR\tpow(R.RFRGRvttstti((sV/opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyR!ûs!cC`s}t}|j}tj|d|ddƒ}|j|dddgƒƒ}d|dd|}||ƒ}t||ƒdS(Niii ii(RR3R1R<RDR (R.R|RFR?RHRvRw((sV/opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyRs  cC`s¯|dddgƒ}tt|jdƒtt|jdƒtt|jdƒƒdƒtt|jdƒƒdƒtt|jdƒƒdƒtt|jdƒƒdƒdS(Niiigà?iÿÿÿÿi(R R„tcutdegR RL(R.RH((sV/opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyR)scC`s¯|dddgƒ}tt|jdƒtt|jdƒtt|jdƒƒdƒtt|jdƒƒdƒtt|jdƒƒdƒtt|jdƒƒdƒdS(Niiigà?ii(R R„ttruncateR RL(R.RH((sV/opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyR*$scC`svddddg}||ƒ}t|jƒj|d ƒt|jdƒj|d ƒt|jdƒj|d ƒdS( Nigíµ ÷ư>gê-™—q=iig»½×Ùß|Û=igñh㈵øä>(R ttrimR5(R.tcRH((sV/opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyR+.s  cC`sŒ|j}|j}|dgd|d|ƒ}tddg|jƒƒd|d}|dgd|d|ƒ}tddg|jƒƒdS(NiR3R4ii(R3R4R tmapparms(R.RFRGRH((sV/opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pyR&6s  t__main__(:t__doc__t __future__RRRtoperatorR\tnumbersRtnumpyR1tnumpy.polynomialRRRRRR t numpy.testingR R R R Rt numpy.compatRRR/R=R:RRRRRRRRRRRRRRR R$R'R(R"R#R%R!RR)R*R+R&t__name__(((sV/opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_classes.pytsR  .(   $      ,       -