ó 2ÄÈ[c @`sºdZddlmZmZmZddlZddljj Z ddl m Z ddl mZmZmZmZejdgƒZejddgƒZejdddgƒZejddddgƒZejd dd ddgƒZejdd dd ddgƒZejd dddd ddgƒZejddddddddgƒZejdddddddddg ƒZejddddddddddg ƒZeeeeeeeeeeg Zd„Zdefd„ƒYZ defd„ƒYZ!defd„ƒYZ"d efd!„ƒYZ#d"efd#„ƒYZ$d$efd%„ƒYZ%d&efd'„ƒYZ&d(efd)„ƒYZ'd*efd+„ƒYZ(d,efd-„ƒYZ)dS(.sTests for hermite_e module. i(tdivisiontabsolute_importtprint_functionN(tpolyval(tassert_almost_equalt assert_raisest assert_equaltassert_iiÿÿÿÿiýÿÿÿiiúÿÿÿiiöÿÿÿiñÿÿÿi-i—ÿÿÿiiiëÿÿÿi\þÿÿiÒiäÿÿÿi±iûÿÿiziÜÿÿÿcC`stj|ddƒS(Nttolgíµ ÷Æ°>(thermet hermetrim(tx((sD/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_hermite_e.pyttrimst TestConstantscB`s,eZd„Zd„Zd„Zd„ZRS(cC`sttjddgƒdS(Niÿÿÿÿi(RR t hermedomain(tself((sD/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_hermite_e.pyttest_hermedomain!scC`sttjdgƒdS(Ni(RR t hermezero(R((sD/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_hermite_e.pyttest_hermezero$scC`sttjdgƒdS(Ni(RR thermeone(R((sD/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_hermite_e.pyt test_hermeone'scC`sttjddgƒdS(Nii(RR thermex(R((sD/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_hermite_e.pyt test_hermex*s(t__name__t __module__RRRR(((sD/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_hermite_e.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ð?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ÿÿÿÿi sAt i=%dRii( RR R+tsizeRR;tHelistRRRRtshape( RR tctyR R"R#R$tdims((sD/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_hermite_e.pyt test_hermevalqs"    "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 t hermeval2dtc2dRRtonesRR?( Rtx1tx2tx3ty1ty2ty3R#R$tz((sD/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_hermite_e.pyttest_hermeval2d†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 t hermeval3dtc3dRRRGRR?( RRHRIRJRKRLRMR#R$RN((sD/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_hermite_e.pyttest_hermeval3d—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 hermegrid2dRFRRGRR?( RRHRIRJRKRLRMR#R$RN((sD/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_hermite_e.pyttest_hermegrid2d¨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 hermegrid3dRQRRGRR?( RRHRIRJRKRLRMR#R$RN((sD/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_hermite_e.pyttest_hermegrid3d¶s (ii(RRRtarraytc1dRSRFRQtrandomR RRARCRORRRURW(((sD/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_hermite_e.pyR<gs    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 thermeintRRt poly2hermet herme2polyR R+tlist( RR R`R$R]tpolR#thermepolRaR!((sD/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_hermite_e.pyt test_hermeintÇ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$((sD/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_hermite_e.pyttest_hermeint_axiss1 + 1(RRRgRj(((sD/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_hermite_e.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 thermederRRR RaR(RR R#R$R!((sD/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_hermite_e.pyt test_hermeder-s"$!*cC`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$((sD/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_hermite_e.pyttest_hermeder_axisGs1 +(RRRmRn(((sD/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_hermite_e.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 hermevanderRR?RRR+RX(RR tvR tcoef((sD/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_hermite_e.pyttest_hermevanderXs'*cC`s®|j\}}}tjjdƒ}tj||ddgƒ}tj|||ƒ}tj||jƒ}t||ƒtj|g|gddgƒ}t |j dkƒdS(Niiiii(ii(iii( R RRZR t hermevander2dREtdottflatRRR?(RRHRIRJR@tvanR#R$((sD/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_hermite_e.pyttest_hermevander2dis !cC`sÀ|j\}}}tjjdƒ}tj|||dddgƒ}tj||||ƒ}tj||jƒ}t||ƒtj|g|g|gdddgƒ}t |j dkƒdS( Niiiiii(iii(iii( R RRZR t hermevander3dRPRvRwRRR?(RRHRIRJR@RxR#R$((sD/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_hermite_e.pyttest_hermevander3dvs! *(ii(RRRRZR RtRyR{(((sD/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_hermite_e.pyRoTs  t TestFittingcB`seZd„ZRS(c C`síd„}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||dd d ddgƒ}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…(RR ((sD/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_hermite_e.pyttest_herme2polyscC`sCx<tdƒD].}ttjt|ƒdg|dgƒq WdS(Ni ii(RRR RbR>(RR ((sD/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_hermite_e.pyttest_poly2hermescC`sLtjdddƒ}tjd|dƒ}tj|ƒ}t||ƒdS(Niûÿÿÿii gà¿i(RR;texpR t hermeweightR(RR R#R$((sD/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_hermite_e.pyt test_weights( RRR¡R£R¤R¦R§R¨R«(((sD/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_hermite_e.pyRós    (*t__doc__t __future__RRRtnumpyRtnumpy.polynomial.hermite_et polynomialt hermite_eR tnumpy.polynomial.polynomialRt numpy.testingRRRRRXtHe0tHe1tHe2tHe3tHe4tHe5tHe6tHe7tHe8tHe9R>R tobjectR RR<R[RkRoR|RR’R(((sD/tmp/pip-build-fiC0ax/numpy/numpy/polynomial/tests/test_hermite_e.pyts6 "!$'*-$ 9^f)0M