bgI dZddlmZddlZddlmcmZddl m Z ddl m Z m Z mZmZejdgZejddgZejgdZejgdZejgd Zejgd Zejgd Zejgd Zejgd ZejgdZeeeeeeeeeeg ZdZGddZGddZGddZGddZ GddZ!GddZ"GddZ#GddZ$Gd d!Z%Gd"d#Z&dS)$zTests for hermite_e module. )reduceNpolyval)assert_almost_equal assert_raises assert_equalassert_)rr )rrr )rirr )rrirr )r-rrrr )riririrr ) rri\rrirr ) riririzrirr c.tj|dS)Ngư>)tol)herme hermetrimxs x/builddir/build/BUILD/cloudlinux-venv-1.0.7/venv/lib64/python3.11/site-packages/numpy/polynomial/tests/test_hermite_e.pytrimrs ?1$ ' ' ''c&eZdZdZdZdZdZdS) TestConstantsc>ttjddgdS)Nr r )rr hermedomainselfs rtest_hermedomainzTestConstants.test_hermedomain!sU&Q00000rc<ttjdgdS)Nr)rr hermezeror s rtest_hermezerozTestConstants.test_hermezero$sU_qc*****rc<ttjdgdSNr )rrhermeoner s r test_hermeonezTestConstants.test_hermeone'sU^aS)))))rc>ttjddgdS)Nrr )rrhermexr s r test_hermexzTestConstants.test_hermex*sU\Aq6*****rN)__name__ __module__ __qualname__r"r%r)r,rrrrsP111+++***+++++rrcVeZdZejdddZdZdZdZdZ dZ d Z d S) TestArithmeticr r dctdD]}tdD]}d|d|}tjt||dz}||xxdz cc<||xxdz cc<t jdg|zdgzdg|zdgz}t t|t||dSNAt i=, j=r rerr_msg)rangenpzerosmaxrhermeaddrrr!ijmsgtgtress r test_hermeaddzTestArithmetic.test_hermeadd1q @ @A1XX @ @(a((Q((hs1ayy1}--A! A! naSUaS[1#a%1#+>>T#YYS 3?????  @ @ @rctdD]}tdD]}d|d|}tjt||dz}||xxdz cc<||xxdzcc<t jdg|zdgzdg|zdgz}t t|t||dSr5)r;r<r=r>rhermesubrrr@s r test_hermesubzTestArithmetic.test_hermesub;rGrc4ttjdgdgttjdgddgtddD]=}dg|zdgz}dg|dz z|ddgz}ttj||>dS)Nrr r6)rr hermemulxr;)r!rAserrDs rtest_hermemulxzTestArithmetic.test_hermemulxEsU_aS))A3///U_aS))Aq6222q! 4 4A#a%1#+C#q1u+Aq )C --s 3 3 3 3 4 4rctdD]}dg|zdgz}tj|j|}tdD]}d|d|}dg|zdgz}tj|j|}tj||}tj|j|} t t |||zdzk|t| ||z|dS)Nr6rr r7r8r9)r;rhermevalrhermemulr lenr) r!rApol1val1rBrCpol2val2pol3val3s r test_hermemulzTestArithmetic.test_hermemulMsq B BA3q5A3;D>$&$//D1XX B B(a((Q((s1us{~dfd33~dD11~dfd33D QUQY.444#D$t)SAAAAA B B BrctdD]}tdD]}d|d|}dg|zdgz}dg|zdgz}tj||}tj||\}}tjtj|||} t t | t ||dS)Nr6r7r8rr r9)r;rr?hermedivrQrr) r!rArBrCcicjrDquoremrEs r test_hermedivzTestArithmetic.test_hermediv[sq @ @A1XX @ @(a((Q((SUaS[SUaS[nR,, >#r22SnU^C%<%ij i,j,k->ijkr r6r )g?rjrkcttjgdgjdt jddfdt D}tdD]>}d|}||}tjdg|zdgz}t|||?tdD]}d g|z}t j |ttjdgj |ttjddgj |ttjgd j |dS) Nr rr c0g|]}t|Sr0r).0rers r z0TestEvaluation.test_hermeval..s! + + +qWQ]] + + +r r7r9r ro)r rr) rrrPsizer<rgHelistr;rr=shape)r!yrArCrDrEdimsrs @r test_hermevalzTestEvaluation.test_hermevalzsUU^B,,11555 KA   + + + +F + + +r 7 7A!++CA$C.QCEQCK00C S# 6 6 6 6 6q C CA3q5DA A3//5t < < < Aq6228$ ? ? ? 99955;T B B B B  C Crc~|j\}}}|j\}}}tttj||dd|j||z}t j|||j}t||tj d} t j| | |j}t|j dkdSNroror ) rrxr ValueErrorr hermeval2dc2drr<onesr rw r!x1x2x3y1y2y3rDrEzs rtest_hermeval2dzTestEvaluation.test_hermeval2dsV BV B j%"2B2A2IIIer2tx00C%%% GFOOq!TX.. V#$$$$$rc |j\}}}|j\}}}tttj|||dd|j||z|z}t j||||j}t||tj d} t j| | | |j}t|j dkdSr|) rrxrr~r hermeval3dc3drr<rr rwrs rtest_hermeval3dzTestEvaluation.test_hermeval3dsV BV B j%"2BBrrFDHMMMeBhr2r4844C%%% GFOOq!Q11 V#$$$$$rcB|j\}}}|j\}}}tjd||}t j|||j}t||tjd} t j| | |j}t|j dkdS)Nrlr})ror ror ) rrxr<einsumr hermegrid2drrrr rwrs rtest_hermegrid2dzTestEvaluation.test_hermegrid2dsV BV Bi 2r**B11C%%% GFOO1dh// X%&&&&&rcH|j\}}}|j\}}}tjd|||}t j||||j}t||tjd} t j| | | |j}t|j dkdS)Nrmr})ror ror ror ) rrxr<rr hermegrid3drrrr rwrs rtest_hermegrid3dzTestEvaluation.test_hermegrid3dsV BV Bi b"b11BDH55C%%% GFOO1a22 X%&&&&&rN)r-r.r/r<rcc1drrrrandomrrrxrzrrrrr0rrririps "(<<< C ")IsC ( (C ")L#sC 0 0C   "Q&A<<<  ACCC*%%%"%%%" ' ' ' ' ' ' ' 'rriceZdZdZdZdS) 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)Nr?r r )lbnd)sclaxisror6)mk)rrr)rrrr) r TypeErrorrhermeintr~r;r poly2herme herme2polyrrPlist) r!rArrErpolrDhermepolrrBs r test_hermeintzTestIntegral.test_hermeintsi!b999j%.1#r:::j%.1#q1a&AAAj%.1#QC@@@@j%.1#A3????i!2>>>>q! - -AQU qc!A.!Q///C aV , , , ,q 6 6Aa%C#a%1#+C#A+3'C',,H~h!s;;;H"8,,C S 499 5 5 5 5q A AAa%C#a%1#+C',,H~h!sDDDH r8 < .'":::q**:::rrrc6g|]}tj|Sr0rrs rrsz3TestIntegral.test_hermeint_axis..+"888q**888rr c:g|]}tj|dS)r )rrrs rrsz3TestIntegral.test_hermeint_axis../s'===AQ///===rr )rr)r<rvstackTrrrr!rrDrEs rtest_hermeint_axiszTestIntegral.test_hermeint_axis#siv&&i::CE:::;;=nSq)))C%%%i88C88899nSq)))C%%%i=====>>nSAA...C%%%%%rN)r-r.r/rrr0rrrrs6Q: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) Nrrr r6r rro)rr) rrrhermederr~r;rrrr)r!rArDrErBs r test_hermederzTestDerivative.test_hermeder6si!b999j%.1#r:::q / /A#a%1#+C.***C cDII . . . .q : :A1a[[ : :c!eqcknU^C1%=%=%=CCC#DIItCyy9999 : q : :A1a[[ : :c!eqcknN3!333qbBBB#DIItCyy9999  : : :rchtjd}tjd|jDj}t j|d}t ||tjd|D}t j|d}t ||dS)Nrc6g|]}tj|Sr0rrrs rrsz5TestDerivative.test_hermeder_axis..Trrrrc6g|]}tj|Sr0rrs rrsz5TestDerivative.test_hermeder_axis..Xrrr )r<rrrrrrrs rtest_hermeder_axisz!TestDerivative.test_hermeder_axisPsiv&&i::CE:::;;=nSq)))C%%%i88C88899nSq)))C%%%%%rN)r-r.r/rrr0rrrr4s2:::4 & & & & &rrc`eZdZejddzdz ZdZdZdZdS) TestVanderrnror 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) Nr rrrr .ror6)r ror) r<rbr hermevanderr rwr;rrPrc)r!rvrAcoefs rtest_hermevanderzTestVander.test_hermevanderas+ IaLL  a # #6!"""q D DA3q5A3;D #q& 5>!T+B+B C C C C Hq!fq!fq!f- . .  a # #9$%%%q D DA3q5A3;D #q& 5>!T+B+B C C C C D Drcn|j\}}}tjd}tj||ddg}tj|||}tj||j}t||tj|g|gddg}t|j dkdS)Nr}r ro)r r6r) rr<rr hermevander2drdotflatrr rwr!rrrrevanrDrEs rtest_hermevander2dzTestVander.test_hermevander2drsV B I  V $ $!"b1a&11r2q))fS!&!!C%%%!2$q!f55 Y&'''''rcv|j\}}}tjd}tj|||gd}tj||||}tj||j}t||tj|g|g|ggd}t|j dkdS)N)ror r)r ror )r r6) rr<rr hermevander3drrrrr rwrs rtest_hermevander3dzTestVander.test_hermevander3dsV B I  Y ' '!"b"iii88r2r1--fS!&!!C%%%!2$rdIII>> Z'(((((rN) r-r.r/r<rrrrrr0rrrr]sc   "Q&ADDD" ( ( ( ) ) ) ) )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 ror0rs rfz$TestFitting.test_hermefit..fsa!e9a!e$ $rc|dz|dzzdzS)Nrror r0rs rf2z%TestFitting.test_hermefit..f2sa4!Q$;? "rr r rro)w)ror rr r)rr ror r6)rr ror r)ror rr r)r y?r y)rror)rr~rhermefitrr<rgrrRrrPrcr zeros_likecopy)r!rrrrxcoef3coef4coef2drywwcoef3wcoef2dcoef1coef2s r test_hermefitzTestFitting.test_hermefits % % % # # # j%.1#sB???i1#Q???iaS!<<<i!ugqAAAi!Q!a@@@i!q!fa@@@i!qc1!FFFFi!qc1AGGGGj%.1#sREBBBj%.1#sJJJGGGi!qc2>>> K1   AaDDq!Q''SZZ###EN1e44a888q!\\\22SZZ###EN1e44a888q!Q''SZZ###EN1e44a888q!___55SZZ###EN1e44a888q!___55SZZ###EN1e44a88828QF#3#3#5q99FBHeU^$<$<$>???28QF#3#3#5|||DDFBHeU^$<$<$>??? M!   VVXX!$Q$!$Q$2qA...FE***2|||q999FE***.BHb"X$6$6$8!qAAAGRXuen%=%=%?@@@.BHb"X$6$6$8,,,!LLLGRXuen%=%=%?@@@   EN1a33aV<<<EN1a!Q881a&AAA KA   BqEEq!Q''EN1e44a888q!YYY//EN1e44a888E5)))))rN)r-r.r/rr0rrrrs(H*H*H*H*H*rrc eZdZdZdZdZdS) TestCompanionctttjgtttjdgdSr')rr~rhermecompanionr s r test_raiseszTestCompanion.test_raisess4j%"6;;;j%"6<<<< ? ? ? ? @ @rc`ttjddgddkdS)Nr ro)rr)r rrr s rtest_linear_rootzTestCompanion.test_linear_roots/$aV,,T2c9:::::rN)r-r.r/rrrr0rrrrsD===@@@ ;;;;;rrceZdZdZdS) TestGaussctjd\}}tj|d}tj|j|z|}dtj|z }|dddf|z|z}t|tj dtjdtj z}t| |dS)Nr3cr ro) r hermegaussrr<rrsqrtdiagonalreyepisum)r!rrrvvvdrDs rtest_100zTestGauss.test_100s$$1  a $ $ VAC!GQ   rwr{{}}%% % 4[2  "Bs ,,,gagAEEGGS)))))rN)r-r.r/rr0rrrrs#*****rrc8eZdZdZdZdZdZdZdZdZ dS) TestMiscc  tjg}tt|dgt ddD]}t jt jt j dd|zdzddd}tj|}tj ||}d}tt||dzkttj |ddt||dS)Nr r6rror ) rhermefromrootsrrr;r<cosrgrrPr rRr)r!rErArootsrrDs rtest_hermefromrootszTestMisc.test_hermefromrootss"2&&DIIs+++q! * *AF2;vq!A#'::14a4@AAE&u--C.,,CC CHHA% & & &  0 5 5b 91 = = = S ) ) ) ) * *rcttjdggttjddgdgtddD]h}t jdd|}tjtj|}tt|t|idS)Nr r ror6)rr hermerootsr;r<rgrr)r!rArDrEs rtest_hermerootszTestMisc.test_hermeroots sE,aS112666E,aV44rd;;;q! 6 6A+b!Q''C"5#7#<#<==C S 499 5 5 5 5 6 6rcBgd}tttj|dt tj||ddt tj|d|ddt tj|ddgdS)N)ror r rr r r ror)rr~rrr)r!rs rtest_hermetrimzTestMisc.test_hermetrims}} j%/4<<< U_T**D"I666U_T1--tCRCy999U_T1--s33333rcPttjddddgdS)Nr r)rr hermeliner s rtest_hermelinezTestMisc.test_hermelines(U_Q**QF33333rctdD]7}ttjdg|zdgzt|8dSNrtrr )r;rrrrvr!rAs rtest_herme2polyzTestMisc.test_herme2poly sSr J JA  0!Q! = =vay I I I I J JrctdD]7}ttjt|dg|zdgz8dSr)r;rrrrvrs rtest_poly2hermezTestMisc.test_poly2herme$sSr J JA  0 ; ;aSUaS[ I I I I J Jrctjddd}tjd|dzz}tj|}t ||dS)Nr6 rro)r<rgexpr hermeweightr)r!rrDrEs r test_weightzTestMisc.test_weight(sR KAr " "fSAX""C%%%%%rN) r-r.r/rr r rrrrr0rrrrs * * *666 4 4 4444JJJJJJ&&&&&rr)'__doc__ functoolsrnumpyr<numpy.polynomial.hermite_e polynomial hermite_ernumpy.polynomial.polynomialr numpy.testingrrrr rcHe0He1He2He3He4He5He6He7He8He9rvrrr2rirrrrrrrr0rrr.sc*********//////bhsmmbh1vbhzzzbh}}}bh  bh$$$%%bh)))**bh...//bh33344bh77788 sCc3S#s ;((( + + + + + + + +?@?@?@?@?@?@?@?@D['['['['['['['['|c&c&c&c&c&c&c&c&L&&&&&&&&&&&&&&&&R-)-)-)-)-)-)-)-)`J*J*J*J*J*J*J*J*Z ; ; ; ; ; ; ; ;********&0&0&0&0&0&0&0&0&0&0&r