ó 2ÄÈ[c@`sOdZddlmZmZmZddlZddljj Z ddl m Z ddl mZmZmZmZejdgƒdZejddgƒdZejdd dgƒdZejd d d dgƒd Zejd ddddgƒd ZejddddddgƒdZejdddddddgƒdZeeeeeeegZd„Zdefd„ƒYZdefd „ƒYZd!efd"„ƒYZd#efd$„ƒYZ d%efd&„ƒYZ!d'efd(„ƒYZ"d)efd*„ƒYZ#d+efd,„ƒYZ$d-efd.„ƒYZ%d/efd0„ƒYZ&dS(1sTests for laguerre module. i(tdivisiontabsolute_importtprint_functionN(tpolyval(tassert_almost_equalt assert_raisest assert_equaltassert_iiÿÿÿÿiiüÿÿÿiiîÿÿÿi ii ÿÿÿiHiðÿÿÿixi¨ýÿÿiXi8ÿÿÿiiÐi ïÿÿii öÿÿiÂiÜÿÿÿcC`stj|ddƒS(Nttolgíµ ÷Æ°>(tlagtlagtrim(tx((sC/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_laguerre.pyttrimst TestConstantscB`s,eZd„Zd„Zd„Zd„ZRS(cC`sttjddgƒdS(Nii(RR t lagdomain(tself((sC/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_laguerre.pyttest_lagdomainscC`sttjdgƒdS(Ni(RR tlagzero(R((sC/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_laguerre.pyt test_lagzero!scC`sttjdgƒdS(Ni(RR tlagone(R((sC/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_laguerre.pyt test_lagone$scC`sttjddgƒdS(Niiÿÿÿÿ(RR tlagx(R((sC/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_laguerre.pyt test_lagx's(t__name__t __module__RRRR(((sC/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_laguerre.pyR s   tTestArithmeticcB`sJeZejdddƒZd„Zd„Zd„Zd„Zd„Z RS(iýÿÿÿiidcC`sËxÄtdƒD]¶}x­tdƒD]Ÿ}d||f}tjt||ƒdƒ}||cd7<||cd7ijs i,j,k->ijkiiiigð?g@g@c C`sNttjgdgƒjdƒtjddƒ}gtD]}t||ƒ^q8}x[tdƒD]M}d|}||}tj|dg|dgƒ}t ||d|ƒq`Wx–tdƒD]ˆ}dg|}tj |ƒ}ttj|dgƒj |ƒttj|ddgƒj |ƒttj|dddgƒj |ƒq¾WdS( NiiiÿÿÿÿisAt i=%dRii( RR R+tsizeRR;tLlistRRRRtshape( RR tctyR R"R#R$tdims((sC/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_laguerre.pyt test_lagvalns"    "c C`s±|j\}}}|j\}}}tttj||d |jƒ||}tj|||jƒ}t||ƒtj dƒ} tj| | |jƒ}t |j dkƒdS(Nii(ii(ii( R RARt ValueErrorR tlagval2dtc2dRRtonesRR?( Rtx1tx2tx3ty1ty2ty3R#R$tz((sC/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_laguerre.pyt test_lagval2dƒs   c C`s¾|j\}}}|j\}}}tttj|||d |jƒ|||}tj||||jƒ}t||ƒtj dƒ} tj| | | |jƒ}t |j dkƒdS(Nii(ii(ii( R RARRDR tlagval3dtc3dRRRGRR?( RRHRIRJRKRLRMR#R$RN((sC/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_laguerre.pyt test_lagval3d”s# c C`sœ|j\}}}|j\}}}tjd||ƒ}tj|||jƒ}t||ƒtjdƒ} tj| | |jƒ}t |j dkƒdS(Nsi,j->ijii(ii(ii(iiii( R RARteinsumR t laggrid2dRFRRGRR?( RRHRIRJRKRLRMR#R$RN((sC/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_laguerre.pyttest_laggrid2d¥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 RARRSR t laggrid3dRQRRGRR?( RRHRIRJRKRLRMR#R$RN((sC/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_laguerre.pyttest_laggrid3d³s (ii(RRRtarraytc1dRSRFRQtrandomR RRARCRORRRURW(((sC/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_laguerre.pyR<ds    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]˜}|d}dg|dg}|gdg|d|g}tj|ƒ}tj|d dd |gƒ}tj|ƒ}tt|ƒt|ƒƒqWx~td ƒD]p}|d}dg|dg}tj|ƒ}tj|d dd |gddƒ}ttj d|ƒ|ƒqÇWx¬td ƒD]ž}|d}dg|dg}|gdg|d|g}tj|ƒ}tj|d dd |gddƒ}tj|ƒ}tt|ƒt|ƒƒqHWx£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 t t| ƒƒƒ}tt|ƒt|ƒƒq³WqWxÊtd ƒD]¼}x³tdd ƒD]¢} dg|dg}|}x8t| ƒD]*}tj|d dd |gddƒ}q¢Wtj|d | d t t| ƒƒddƒ}tt|ƒt|ƒƒqtWq^WxÊtd ƒD]¼}x³tdd ƒD]¢} dg|dg}|}x8t| ƒD]*}tj|d dd |gddƒ}qoWtj|d | d t t| ƒƒddƒ}tt|ƒt|ƒƒqAWq+WdS( Nigà?iÿÿÿÿitlbndtscltaxisiitmtk( RRDR tlagintRRtpoly2lagtlag2polyR R+tlist( RR R`R$R]tpolR#tlagpolRaR!((sC/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_laguerre.pyt test_lagintÄs€"  $ $!"'!(-!(-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(NiiR^iiR`(ii(RRZtvstacktTR RaR(RRFR@R#R$((sC/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_laguerre.pyttest_lagint_axiss1 + 1(RRRgRj(((sC/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_laguerre.pyR[Âs StTestDerivativecB`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ÿÿÿÿiiR_iR]( RRDR tlagderRRR RaR(RR R#R$R!((sC/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_laguerre.pyt test_lagder*s $!0cC`s¶tjjdƒ}tjg|jD]}tj|ƒ^q"ƒj}tj|ddƒ}t||ƒtjg|D]}tj|ƒ^qrƒ}tj|ddƒ}t||ƒdS(NiiR^ii(ii(RRZRhRiR RlR(RRFR@R#R$((sC/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_laguerre.pyttest_lagder_axisCs1 +(RRRmRn(((sC/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_laguerre.pyRk(s 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( RtarangeR t lagvanderRR?RRR+RX(RR tvR tcoef((sC/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_laguerre.pyttest_lagvanderTs'*cC`s®|j\}}}tjjdƒ}tj||ddgƒ}tj|||ƒ}tj||jƒ}t||ƒtj|g|gddgƒ}t |j dkƒdS(Niiiii(ii(iii( R RRZR t lagvander2dREtdottflatRRR?(RRHRIRJR@tvanR#R$((sC/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_laguerre.pyttest_lagvander2des !cC`sÀ|j\}}}tjjdƒ}tj|||dddgƒ}tj||||ƒ}tj||jƒ}t||ƒtj|g|g|gdddgƒ}t |j dkƒdS( Niiiiii(iii(iii( R RRZR t lagvander3dRPRvRwRRR?(RRHRIRJR@RxR#R$((sC/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_laguerre.pyttest_lagvander3drs! *(ii(RRRRZR RtRyR{(((sC/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_laguerre.pyRoPs  t TestFittingcB`seZd„ZRS(c C`sd„}tttjdgdgdƒtttjdggdgdƒtttjgdgdƒtttjdgdgggdƒtttjddgdgdƒtttjdgddgdƒtttjdgdgdddggƒtttjdgdgddddgƒtttjdgdgdgƒtttjdgdgdddgƒtttjdgdggƒtjddƒ}||ƒ}tj||dƒ}tt|ƒd ƒt tj ||ƒ|ƒtj||ddddgƒ}tt|ƒd ƒt tj ||ƒ|ƒtj||d ƒ}tt|ƒd ƒt tj ||ƒ|ƒtj||ddddd gƒ}tt|ƒd ƒt tj ||ƒ|ƒtj|tj ||gƒj dƒ}t |tj ||gƒj ƒtj|tj ||gƒj ddddgƒ}t |tj ||gƒj ƒtj |ƒ}|jƒ}d|ddd…(RR ((sC/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_laguerre.pyt test_lag2polyscC`sCx<tdƒD].}ttjt|ƒdg|dgƒq WdS(Niii(RRR RbR>(RR ((sC/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_laguerre.pyt test_poly2lagscC`sEtjdddƒ}tj| ƒ}tj|ƒ}t||ƒdS(Nii i (RR;texpR t lagweightR(RR R#R$((sC/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_laguerre.pyt test_weight s( RRRžR R¡R£R¤R¥R¨(((sC/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_laguerre.pyR™às    ('t__doc__t __future__RRRtnumpyRtnumpy.polynomial.laguerret polynomialtlaguerreR tnumpy.polynomial.polynomialRt numpy.testingRRRRRXtL0tL1tL2tL3tL4tL5tL6R>R tobjectR RR<R[RkRoR|RŠRR™(((sC/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_laguerre.pyts0 ""%( 9^f(0>