bgH dZddlmZddlZddlmcmZddl m Z ddl m Z m Z mZmZejdgZejddgZejgdZejgd Zejgd Zejgd Zejgd Zejgd ZejgdZejgdZeeeeeeeeeeg ZdZGddZGddZGddZGddZ GddZ!GddZ"GddZ#Gdd Z$Gd!d"Z%Gd#d$Z&dS)%zTests for hermite module. )reduceNpolyval)assert_almost_equal assert_raises assert_equalassert_)r)rir) rir)rxri`r )iriri r@)ripri rir) iriri4rir) ri vriririric.tj|dS)Ngư>)tol)hermhermtrimxs v/builddir/build/BUILD/cloudlinux-venv-1.0.7/venv/lib64/python3.11/site-packages/numpy/polynomial/tests/test_hermite.pytrimrs = % % %%c&eZdZdZdZdZdZdS) TestConstantsc>ttjddgdS)Nr )rr hermdomainselfs rtest_hermdomainzTestConstants.test_hermdomain!sT_r1g.....rc<ttjdgdS)Nr)rrhermzeror$s r test_hermzerozTestConstants.test_hermzero$sT]QC(((((rc<ttjdgdSNr )rrhermoner$s r test_hermonezTestConstants.test_hermone'sT\A3'''''rc>ttjddgdS)Nr?)rrhermxr$s r test_hermxzTestConstants.test_hermx*sTZ!R)))))rN)__name__ __module__ __qualname__r&r)r-r1rrr r sP///)))(((*****rr cVeZdZejdddZdZdZdZdZ dZ d Z d S) TestArithmeticdctdD]}tdD]}d|d|}tjt||dz}||xxdz cc<||xxdz cc<t jdg|zdgzdg|zdgz}t t|t||dSNAt i=, j=r rerr_msg)rangenpzerosmaxrhermaddrrr%ijmsgtgtress r test_hermaddzTestArithmetic.test_hermadd1q @ @A1XX @ @(a((Q((hs1ayy1}--A! A! lA3q5A3;A <<T#YYS 3?????  @ @ @rctdD]}tdD]}d|d|}tjt||dz}||xxdz cc<||xxdzcc<t jdg|zdgzdg|zdgz}t t|t||dSr<)rBrCrDrErhermsubrrrGs r test_hermsubzTestArithmetic.test_hermsub;rNrc4ttjdgdgttjdgddgtddD]=}dg|zdgz}dg|dz z|ddgz}ttj||>dS)Nrr r/r=)rrhermmulxrB)r%rHserrKs r test_hermmulxzTestArithmetic.test_hermmulxEsT]A3''!---T]A3''!R111q! 2 2A#a%1#+C#q1u+Ar *C s++S 1 1 1 1 2 2rctdD]}dg|zdgz}tj|j|}tdD]}d|d|}dg|zdgz}tj|j|}tj||}tj|j|} t t |||zdzk|t| ||z|dS)Nr=rr r>r?r@)rBrhermvalrhermmulr lenr) r%rHpol1val1rIrJpol2val2pol3val3s r test_hermmulzTestArithmetic.test_hermmulMsq B BA3q5A3;D<--D1XX B B(a((Q((s1us{|DFD11|D$//|DFD11D QUQY.444#D$t)SAAAAA B B BrctdD]}tdD]}d|d|}dg|zdgz}dg|zdgz}tj||}tj||\}}tjtj|||} t t | t ||dS)Nr=r>r?rr r@)rBrrFhermdivrXrr) r%rHrIrJcicjrKquoremrLs r test_hermdivzTestArithmetic.test_hermdiv[sq @ @A1XX @ @(a((Q((SUaS[SUaS[l2r**<R00Sl4<R#8#8#>>T#YYS 3????? @ @ @rc rtdD]}tdD]}d|d|}tj|dz}ttj|g|ztjdg}t j||}tt|t||dS)Nr=r>r?r r@) rBrCarangerrrXarrayhermpowrr)r%rHrIrJcrKrLs r test_hermpowzTestArithmetic.test_hermpowfsq @ @A1XX @ @(a((Q((Ia!e$$T\A3q5"(A3--@@l1a((T#YYS 3?????  @ @ @rN) r2r3r4rClinspacerrMrQrUr`rgrmr5rrr7r7.s B3A@@@@@@222 B B B @ @ @@@@@@rr7ceZdZejgdZejdeeZejdeeeZej ddzdz Z e e gdZ dZ d Zd Zd Zd Zd S)TestEvaluation)g@?g?i,j->ij i,j,k->ijkr9r=r r )rqg@g@cttjgdgjdt jddfdt D}tdD]>}d|}||}tjdg|zdgz}t|||?tdD]}d g|z}t j |ttjdgj |ttjddgj |ttjgd j |dS) Nr rr"c0g|]}t|Sr5r).0rlrs r z/TestEvaluation.test_hermval..s! * * *qWQ]] * * *r r>r@r9r )r rr) rrrWsizerCrnHlistrBrrDshape)r%yrHrJrKrLdimsrs @r test_hermvalzTestEvaluation.test_hermvalzsUT\"qc**/333 KA   * * * *E * * *r 7 7A!++CA$C,q1#a%1#+..C S# 6 6 6 6 6q A AA3q5DA a!--3T : : : a!Q006 = = = a3394 @ @ @ @  A Arc~|j\}}}|j\}}}tttj||dd|j||z}t j|||j}t||tj d} t j| | |j}t|j dkdSNr r r9) rr}r ValueErrorr hermval2dc2drrConesr r| r%x1x2x3y1y2y3rKrLzs rtest_hermval2dzTestEvaluation.test_hermval2dsV BV B j$."b!fdhGGGenRTX..C%%% GFOOnQ48,, V#$$$$$rc |j\}}}|j\}}}tttj|||dd|j||z|z}t j||||j}t||tj d} t j| | | |j}t|j dkdSr) rr}rrr hermval3dc3drrCrr r|rs rtest_hermval3dzTestEvaluation.test_hermval3dsV BV B j$."b"RaR&$(KKKeBhnRR22C%%% GFOOnQ1dh// V#$$$$$rcB|j\}}}|j\}}}tjd||}t j|||j}t||tjd} t j| | |j}t|j dkdS)Nrrr)r r9r r9) rr}rCeinsumr hermgrid2drrrr r|rs rtest_hermgrid2dzTestEvaluation.test_hermgrid2dsV BV Bi 2r**ob"dh//C%%% GFOOoaDH-- X%&&&&&rcH|j\}}}|j\}}}tjd|||}t j||||j}t||tjd} t j| | | |j}t|j dkdS)Nrsr)r r9r r9r r9) rr}rCrr hermgrid3drrrr r|rs rtest_hermgrid3dzTestEvaluation.test_hermgrid3dsV BV Bi b"b11ob"b$(33C%%% GFOOoaAtx00 X%&&&&&rN)r2r3r4rCrjc1drrrrandomrrr}rrrrrr5rrrprpps "(>>> " "C ")IsC ( (C ")L#sC 0 0C   "Q&A<<<  AAAA*%%%"%%%" ' ' ' ' ' ' ' 'rrpceZdZdZdZdS) TestIntegralc T tttjdgdtttjdgdtttjdgdddgtttjdgdgtttjdgdgtttjdgdt dd D]9}dg|dz zdgz}tjdg|| }t |ddg:t d D]}|dz}dg|zdgz}|gdg|zzd|z gz}tj|}tj|d|g }tj|}t t|t|t d D]a}|dz}dg|zdgz}tj|}tj|d|gd }t tj d||bt d D]}|dz}dg|zdgz}|gdg|zzd|z gz}tj|}tj|d|gd }tj|}t t|t|t d D]}t dd D]~} dg|zdgz}|dd}t | D]}tj|d }tj|| }t t|t|t d D]}t dd D]} dg|zdgz}|dd}t | D]}tj|d|g }tj|| tt |  }t t|t|t d D]}t dd D]} dg|zdgz}|dd}t | D]}tj|d|gd }tj|| tt | d }t t|t|t d D]}t dd D]} dg|zdgz}|dd}t | D]}tj|d|gd }tj|| tt | d }t t|t|dS)Nrr/r"r )lbnd)sclaxisr r=)mk)rrr)rrrr) r TypeErrorrhermintrrBr poly2herm herm2polyrrWlist) r%rHrrLrpolrKhermpolrrIs r test_hermintzTestIntegral.test_hermintsisB777j$,R888j$,QA???j$,1#>>>>j$,!====is<<<<q! . .AQU qc!A,sa1---C aW - - - -q 6 6Aa%C#a%1#+C#A+3'CnS))Gl7aA3777G.))C S 499 5 5 5 5q > >Aa%C#a%1#+CnS))Gl7aA3R@@@G  R 9 91 = = = =q 6 6Aa%C#a%1#+C#A+3'CnS))Gl7aA3A>>>G.))C S 499 5 5 5 5q : :A1a[[ : :c!eqck!!!fq11A,sa000CCl3!,,,#DIItCyy9999  :q : :A1a[[ : :c!eqck!!!fq88A,saA3777CCl3!tE!HH~~>>>#DIItCyy9999  :q : :A1a[[ : :c!eqck!!!fqAAA,saA3R@@@CCl3!tE!HH~~BGGG#DIItCyy9999  :q : :A1a[[ : :c!eqck!!!fq??A,saA3A>>>CCl3!tE!HH~~1EEE#DIItCyy9999  : : :rctjd}tjd|jDj}t j|d}t ||tjd|D}t j|d}t ||tjd|D}t j|dd }t ||dS) Nr9r c6g|]}tj|Sr5rrrwrls rrxz2TestIntegral.test_hermint_axis..' 888Qa888rrrc6g|]}tj|Sr5rrs rrxz2TestIntegral.test_hermint_axis..+ 666Qa666rr c:g|]}tj|dS)r9)rrrs rrxz2TestIntegral.test_hermint_axis../s';;;!a1---;;;rr9)rr)rCrvstackTrrrr%rrKrLs rtest_hermint_axiszTestIntegral.test_hermint_axis#siv&&i88#%88899;l3Q'''C%%%i66#66677l3Q'''C%%%i;;s;;;<<l3!!,,,C%%%%%rN)r2r3r4rrr5rrrrs6Q:Q:Q:f&&&&&rrceZdZdZdZdS)TestDerivativec Vtttjdgdtttjdgdt dD]L}dg|zdgz}tj|d}t t|t|Mt dD]s}t ddD]`}dg|zdgz}tjtj|||}tt|t|att dD]u}t ddD]b}dg|zdgz}tjtj||d|d}tt|t|cvdS) Nrr/r"r=r rr )rr) rrrhermderrrBrrrr)r%rHrKrLrIs r test_hermderzTestDerivative.test_hermder6sisB777j$,R888q / /A#a%1#+C,sa(((C cDII . . . .q : :A1a[[ : :c!eqckl4<q#9#9#9Q???#DIItCyy9999 : q : :A1a[[ : :c!eqckl4<qa#@#@#@A2NNN#DIItCyy9999 : : :rchtjd}tjd|jDj}t j|d}t ||tjd|D}t j|d}t ||dS)Nrc6g|]}tj|Sr5rrrs rrxz4TestDerivative.test_hermder_axis..Srrrrc6g|]}tj|Sr5rrs rrxz4TestDerivative.test_hermder_axis..Wrrr )rCrrrrrrrs rtest_hermder_axisz TestDerivative.test_hermder_axisOsiv&&i88#%88899;l3Q'''C%%%i66#66677l3Q'''C%%%%%rN)r2r3r4rrr5rrrr4s2:::2 & & & & &rrc`eZdZejddzdz ZdZdZdZdS) TestVanderrtr r c8tjd}tj|d}t |jdkt dD]7}dg|zdgz}t|d|ftj||8tj ddgddgdd gg}tj|d}t |jd kt dD]7}dg|zdgz}t|d|ftj||8dS) Nr9rr rr .r r=)r9r r ) rCrir hermvanderr r|rBrrWrj)r%rvrHcoefs rtest_hermvanderzTestVander.test_hermvander`s) IaLL OAq ! !6!"""q B BA3q5A3;D #q& 4<4+@+@ A A A A Hq!fq!fq!f- . . OAq ! !9$%%%q B BA3q5A3;D #q& 4<4+@+@ A A A A B Brcn|j\}}}tjd}tj||ddg}tj|||}tj||j}t||tj|g|gddg}t|j dkdS)Nrr r )r r=r) rrCrr hermvander2drdotflatrr r|r%rrrrlvanrKrLs rtest_hermvander2dzTestVander.test_hermvander2dqsV B I  V $ $BA//nRQ''fS!&!!C%%%rdQF33 Y&'''''rcv|j\}}}tjd}tj|||gd}tj||||}tj||j}t||tj|g|g|ggd}t|j dkdS)N)r r9r )r r r9)r r=) rrCrr hermvander3drrrrr r|rs rtest_hermvander3dzTestVander.test_hermvander3d~sV B I  Y ' 'BIII66nRR++fS!&!!C%%%rdRD)))<< Z'(((((rN) r2r3r4rCrrrrrr5rrrr\sc   "Q&ABBB" ( ( ( ) ) ) ) )rrceZdZdZdS) TestFittingc  d}d}tttjdgdgdtttjdggdgdtttjgdgdtttjdgdgggdtttjddgdgdtttjdgddgdtttjdgdgddggtttjdgdgdddgtttjdgdgdgtttjdgdggdtttjdgdggt jdd}||}tj||d }tt|d ttj |||tj||gd }tt|d ttj |||tj||d }tt|d ttj |||tj||gd }tt|d ttj |||tj||gd}tt|d ttj |||tj|t j ||gj d }t|t j ||gj tj|t j ||gj gd }t|t j ||gj t j |}|} d|ddd<d|ddd<tj|| d |} t| |tj|| gd |} t| |tj|t j | | gj d |} t| t j ||gj tj|t j | | gj gd |} t| t j ||gj gd}ttj||dddgttj||ddgddgt jdd}||}tj||d } ttj || |tj||gd} ttj || |t| | dS)Nc||dz z|dz zS)Nr r r5rs rfz#TestFitting.test_hermfit..fsa!e9a!e$ $rc|dz|dzzdzS)Nr r r r5rs rf2z$TestFitting.test_hermfit..f2sa4!Q$;? "rr r"rr )w)r r"rr9r )rr r r9r=)rr r r9r )r r9r r r)r y?r"yr/)rr r )rrrhermfitrrCrnrrYrrWrjr zeros_likecopy)r%rrrr}coef3coef4coef2drywwcoef3wcoef2dcoef1coef2s r test_hermfitzTestFitting.test_hermfits % % % # # # j$,aS"===iuqc1===irA3:::isqcUGQ???i1vsA>>>isQFA>>>isQCseDDDDisQCq!fEEEEj$,aS2%@@@j$,aS***EEEisQC<<< K1   AaDD Q1%%SZZ###DLE22A666 Q<<<00SZZ###DLE22A666 Q1%%SZZ###DLE22A666 Q???33SZZ###DLE22A666 Q???33SZZ###DLE22A666a1a&!1!1!3Q77FBHeU^$<$<$>???a1a&!1!1!3\\\BBFBHeU^$<$<$>??? M!   VVXX!$Q$!$Q$aQ!,,,FE***a\\\Q777FE***,q"(B8"4"4"6Q???GRXuen%=%=%?@@@,q"(B8"4"4"6 JJJGRXuen%=%=%?@@@   DLAq11Ar7;;;DLA1v66B@@@ KA   BqEE Q1%%DLE22A666 Q999--DLE22A666E5)))))rN)r2r3r4rr5rrrrs(H*H*H*H*H*rrc eZdZdZdZdZdS) TestCompanionctttjgtttjdgdSr+)rrr hermcompanionr$s r test_raiseszTestCompanion.test_raisess4j$"4b999j$"4qc:::::rctddD]8}dg|zdgz}ttj|j||fk9dS)Nr r=r)rBr rrr|)r%rHrs rtest_dimensionszTestCompanion.test_dimensionss\q! > >A3q5A3;D D&t,,2q!f< = = = = > >rc`ttjddgddkdS)Nr r )rrgп)r rrr$s rtest_linear_rootzTestCompanion.test_linear_roots/"Aq6**40D899999rN)r2r3r4rrrr5rrrrsA;;;>>> :::::rrceZdZdZdS) TestGaussctjd\}}tj|d}tj|j|z|}dtj|z }|dddf|z|z}t|tj dtjtj }t| |dS)Nr:cr ) r hermgaussrrCrrsqrtdiagonalreyepisum)r%rrrvvvdrKs rtest_100zTestGauss.test_100s~c""1 OAr " " VAC!GQ   rwr{{}}%% % 4[2  "Bs ,,,gbennAEEGGS)))))rN)r2r3r4rr5rrrrs#*****rrc8eZdZdZdZdZdZdZdZdZ dS) TestMiscc  tjg}tt|dgt ddD]}t jt jt j dd|zdzddd}tj|}tj ||}d}tt||dzkttj |ddt||dS)Nr r=rr r") r hermfromrootsrrrBrCcosrnrrWr rYr)r%rLrHrootsrrKs rtest_hermfromrootszTestMisc.test_hermfromrootss $$DIIs+++q! * *AF2;vq!A#'::14a4@AAE$U++C,uc**CC CHHA% & & & s 3 3B 7 ; ; ; S ) ) ) ) * *rcttjdggttjddgdgtddD]h}t jdd|}tjtj|}tt|t|idS)Nr gr r=r")rr hermrootsrBrCrnrr)r%rHrKrLs rtest_hermrootszTestMisc.test_hermroots sDNA3//444DNAq622SE:::q! 6 6A+b!Q''C.!3C!8!899C S 499 5 5 5 5 6 6rcBgd}tttj|dt tj||ddt tj|d|ddt tj|ddgdS)N)r r"r rr"r r8r r)rrrrr)r%rs r test_hermtrimzTestMisc.test_hermtrims}} j$-r::: T]4(($ss)444T]4++T#2#Y777T]4++aS11111rcPttjddddgdS)Nr9r r )rrhermliner$s r test_hermlinezTestMisc.test_hermlines(T]1a((1a&11111rctdD]7}ttjdg|zdgzt|8dSNryrr )rBrrrr{r%rHs rtest_herm2polyzTestMisc.test_herm2polysRr G GA s1us{ ; ;U1X F F F F G GrctdD]7}ttjt|dg|zdgz8dSr)rBrrrr{rs rtest_poly2hermzTestMisc.test_poly2herm#sRr G GA uQx 8 81#a%1#+ F F F F G Grctjddd}tj|dz }tj|}t ||dS)Nr= r )rCrnexpr hermweightr)r%rrKrLs r test_weightzTestMisc.test_weight'sM KAr " "fadUmmoa  C%%%%%rN) r2r3r4r r rrrrrr5rrrrs * * *666 2 2 2222GGGGGG&&&&&rr)'__doc__ functoolsrnumpyrCnumpy.polynomial.hermite polynomialhermiternumpy.polynomial.polynomialr numpy.testingrrrr rjH0H1H2H3H4H5H6H7H8H9r{rr r7rprrrrrrrr5rrr0sc'''''''''//////RXqc]] RXq!f RXjjj RXnnn RX!!!"" RX&&&'' RX,,,-- RX33344 RX;;;<< RX???@@ RRRRR0&&& * * * * * * * *?@?@?@?@?@?@?@?@D['['['['['['['['|c&c&c&c&c&c&c&c&L%&%&%&%&%&%&%&%&P-)-)-)-)-)-)-)-)`J*J*J*J*J*J*J*J*Z : : : : : : : :********&0&0&0&0&0&0&0&0&0&0&r