ó 2ÄÈ[c @`s$dZddlmZmZmZddlZddljjZ ddl m Z m Z m Z mZd„ZdgZddgZdddgZdd dd gZddd dd gZdd ddddgZdddddddgZddddddddgZdddddddddg Zddddddddddg Zeeeeeeeeeeg Zd efd!„ƒYZd"efd#„ƒYZd$efd%„ƒYZd&efd'„ƒYZd(efd)„ƒYZ d*efd+„ƒYZ!d,efd-„ƒYZ"d.efd/„ƒYZ#dS(0sTests for polynomial module. i(tdivisiontabsolute_importtprint_functionN(tassert_almost_equalt assert_raisest assert_equaltassert_cC`stj|ddƒS(Nttolgíµ ÷Æ°>(tpolytpolytrim(tx((sE/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_polynomial.pyttrim siiÿÿÿÿiiýÿÿÿiiøÿÿÿiiiìÿÿÿiiiÐÿÿÿi iùÿÿÿi8iÿÿÿi@iàÿÿÿi iÿÿÿi€i iˆÿÿÿi°iÀýÿÿit TestConstantscB`s,eZd„Zd„Zd„Zd„ZRS(cC`sttjddgƒdS(Niÿÿÿÿi(RRt polydomain(tself((sE/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_polynomial.pyttest_polydomain scC`sttjdgƒdS(Ni(RRtpolyzero(R((sE/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_polynomial.pyt test_polyzero#scC`sttjdgƒdS(Ni(RRtpolyone(R((sE/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_polynomial.pyt test_polyone&scC`sttjddgƒdS(Nii(RRtpolyx(R((sE/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_polynomial.pyt test_polyx)s(t__name__t __module__RRRR(((sE/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_polynomial.pyR s   tTestArithmeticcB`s5eZd„Zd„Zd„Zd„Zd„ZRS(cC`sËxÄtdƒD]¶}x­tdƒD]Ÿ}d||f}tjt||ƒdƒ}||cd7<||cd7ijs i,j,k->ijkiiiicC`s|ttjgdgƒjdƒtjddƒ}gtdƒD]}||^q>}xKtdƒD]=}||}tj|dg|dgƒ}t||ƒqaW||dd}tj|ddddgƒ}t||ƒx–tdƒD]ˆ}dg|}tj|ƒ}ttj|dgƒj |ƒttj|ddgƒj |ƒttj|dddgƒj |ƒqìWdS(Niiiÿÿÿÿiii( RRtpolyvaltsizeRtlinspaceRRRtshape(RR RtyR"R#tdims((sE/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_polynomial.pyt test_polyvalts #    "c C`s*tttjdgdgdtƒttjgdgƒjdƒttjgdgƒjdkƒttjgdgdgƒjdƒttjgdgdgƒjdkƒttjddƒdƒttjdt j dƒƒjdkƒt j ddƒ}gt dƒD]}||^q}xGt ddƒD]6}||}tj|dg|ƒ}t ||ƒq8W||d|d}tj|dddgƒ}t ||ƒx–t dƒD]ˆ}dg|}t j|ƒ}ttj|dgƒj|ƒttj|ddgƒj|ƒttj|dddgƒj|ƒq½Wddd d dg}tj|ƒ}t j ddƒ}t tj||ƒtj||ƒƒd} t jd dƒ}t jjd dd | ƒ}tj||dtƒ}t j|jdƒ}xAt |jƒD]0} tj|| |dd…| fƒ|| ijii(ii(ii(iiii( R R8RteinsumRt polygrid2dRNRR?RR7( RRORPRQRRRSRTR"R#RU((sE/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_polynomial.pyttest_polygrid2dís c C`s¥|j\}}}|j\}}}tjd|||ƒ}tj||||jƒ}t||ƒtjdƒ} tj| | | |jƒ}t |j dkƒdS(Ns i,j,k->ijkii(ii(ii(iiiiii( R R8RRZRt polygrid3dRXRR?RR7( RRORPRQRRRSRTR"R#RU((sE/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_polynomial.pyttest_polygrid3dûs (ii(RRRtarraytc1dRZRNRXRBR RR4R8R:RLRVRYR\R^(((sE/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_polynomial.pyR3js  @   t TestIntegralcB`seZd„Zd„ZRS(c C`s¤tttjdgdƒtttjdgdƒtttjdgdddgƒtttjdgddgƒtttjdgddgƒtttjdgddƒx`tdd ƒD]O}dg|ddg}tjdgd |d |ƒ}t|ddgƒq¾Wxˆtd ƒD]z}|d}dg|dg}|gdg|d|g}tj|d dd |gƒ}tt|ƒt|ƒƒqWxotd ƒD]a}|d}dg|dg}tj|d dd |gddƒ}ttjd|ƒ|ƒq©WxŽtd ƒD]€}|d}dg|dg}|gdg|d|g}tj|d dd |gddƒ}tt|ƒt|ƒƒqWx£td ƒD]•}xŒtdd ƒD]{}dg|dg}|}x)t|ƒD]}tj|d dƒ}qðWtj|d |ƒ}tt|ƒt|ƒƒqÂWq¬Wx¾td ƒD]°}x§tdd ƒD]–}dg|dg}|}x2t|ƒD]$}tj|d dd |gƒ}q–Wtj|d |d tt|ƒƒƒ}tt|ƒt|ƒƒqhWqRWxÊtd ƒD]¼}x³tdd ƒD]¢}dg|dg}|}x8t|ƒD]*}tj|d dd |gddƒ}qWWtj|d |d tt|ƒƒddƒ}tt|ƒt|ƒƒq)WqWxÊtd ƒD]¼}x³tdd ƒD]¢}dg|dg}|}x8t|ƒD]*}tj|d dd |gddƒ}q$Wtj|d |d tt|ƒƒddƒ}tt|ƒt|ƒƒqöWqàWdS( Nigà?iÿÿÿÿitlbndtscltaxisiitmtk( RR<RtpolyintRRR R4tlist(RRRfR#RctpolR"R ((sE/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_polynomial.pyt test_polyint sv"  $ $!"'!(-!(-cC`stjjdƒ}tjg|jD]}tj|ƒ^q"ƒj}tj|ddƒ}t||ƒtjg|D]}tj|ƒ^qrƒ}tj|ddƒ}t||ƒtjg|D]}tj|ddƒ^q¿ƒ}tj|ddddƒ}t||ƒdS(NiiRdiiRf(ii(RRBREtTRRgR(RRNtcR"R#((sE/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_polynomial.pyttest_polyint_axisZs1 + 1(RRRjRm(((sE/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_polynomial.pyRa s NtTestDerivativecB`seZd„Zd„ZRS(c C`sŸtttjdgdƒtttjdgdƒxVtdƒD]H}dg|dg}tj|ddƒ}tt|ƒt|ƒƒq?WxtdƒD]q}xhtddƒD]W}dg|dg}tjtj|d|ƒd|ƒ}tt|ƒt|ƒƒq®Wq˜Wx‹tdƒD]}}xttddƒD]c}dg|dg}tjtj|d|ddƒd|ddƒ}tt|ƒt|ƒƒq0WqWdS( Nigà?iÿÿÿÿiiReiRc( RR<RtpolyderRRR RgR(RRR"R#R ((sE/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_polynomial.pyt test_polyderms $!0cC`s¶tjjdƒ}tjg|jD]}tj|ƒ^q"ƒj}tj|ddƒ}t||ƒtjg|D]}tj|ƒ^qrƒ}tj|ddƒ}t||ƒdS(NiiRdii(ii(RRBRERkRRoR(RRNRlR"R#((sE/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_polynomial.pyttest_polyder_axis†s1 +(RRRpRq(((sE/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_polynomial.pyRnks t TestVandercB`s=eZejjdƒddZd„Zd„Zd„ZRS(iiiicC`s#tjdƒ}tj|dƒ}t|jd kƒxKtdƒD]=}dg|dg}t|d|ftj||ƒƒqAWtj ddgddgddggƒ}tj|dƒ}t|jd kƒxKtdƒD]=}dg|dg}t|d|ftj||ƒƒqÞWdS( Niiii.iii(ii(iii( RRARt polyvanderRR7RRR4R_(RR tvRtcoef((sE/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_polynomial.pyttest_polyvander—s'*cC`s®|j\}}}tjjdƒ}tj||ddgƒ}tj|||ƒ}tj||jƒ}t||ƒtj|g|gddgƒ}t |j dkƒdS(Niiiii(ii(iii( R RRBRt polyvander2dRMtdottflatRRR7(RRORPRQRltvanR"R#((sE/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_polynomial.pyttest_polyvander2d¨s !cC`sÀ|j\}}}tjjdƒ}tj|||dddgƒ}tj||||ƒ}tj||jƒ}t||ƒtj|g|g|gdddgƒ}t |j dkƒdS( Niiiiii(iii(iii( R RRBRt polyvander3dRWRxRyRRR7(RRORPRQRlRzR"R#((sE/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_polynomial.pyttest_polyvander3dµs! *(ii(RRRRBR RvR{R}(((sE/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_polynomial.pyRr“s  t TestCompanioncB`s#eZd„Zd„Zd„ZRS(cC`s-tttjgƒtttjdgƒdS(Ni(RR<Rt polycompanion(R((sE/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_polynomial.pyt test_raisesÅscC`sTxMtddƒD]<}dg|dg}ttj|ƒj||fkƒqWdS(Niii(RRRRR7(RRRu((sE/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_polynomial.pyttest_dimensionsÉscC`s'ttjddgƒddkƒdS(Niiigà¿(ii(RRR(R((sE/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_polynomial.pyttest_linear_rootÎs(RRR€RR‚(((sE/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_polynomial.pyR~Ãs  tTestMisccB`s5eZd„Zd„Zd„Zd„Zd„ZRS(cC`s¸tjgƒ}tt|ƒdgƒxŒtddƒD]{}tjtjtj dd|dƒddd…ƒ}t |}tj|ƒd|d}tt|ƒt|ƒƒq5WdS(Niiii( Rt polyfromrootsRR RRtcosR6tpitTlist(RR#RtrootsR"((sE/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_polynomial.pyttest_polyfromrootsÔs7 cC`sœttjdgƒgƒttjddgƒdgƒx]tddƒD]L}tjdd|ƒ}tjtj|ƒƒ}tt|ƒt|ƒƒqHWdS(Niigà¿iiÿÿÿÿ(RRR@RRR6R„R (RRR"R#((sE/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_polynomial.pyttest_polyrootsÝs c C`sd„}d„}tttjdgdgdƒtttjdggdgdƒtttjgdgdƒtttjdgdgggdƒtttjddgdgdƒtttjdgddgdƒtttjdgdgdddggƒtttjdgdgddddgƒtttjdgdgdgƒtttjdgdgdddgƒtttjdgdggƒtjddƒ}||ƒ}tj||d ƒ}tt|ƒd ƒt tj ||ƒ|ƒtj||dddd gƒ}tt|ƒd ƒt tj ||ƒ|ƒtj||d ƒ}tt|ƒd ƒt tj ||ƒ|ƒtj||dddd d gƒ}tt|ƒd ƒt tj ||ƒ|ƒtj|tj ||gƒj d ƒ}t |tj ||gƒj ƒtj|tj ||gƒj dddd gƒ}t |tj ||gƒj ƒtj |ƒ}|jƒ} d|ddd…s0 "   !$$= a(0