h k
d Z ddlmZ ddlZddlmc mZ ddl m
Z
ddl m Z m
Z
mZmZ ej dg Z ej ddg Z ej g d Z ej g d Z ej g d Z ej g d
Z ej g d Z ej g d Z ej g d
Z ej g d Zeeeeeeeeeeg
Zd Z G d d Z G d d Z G d d Z G d d Z G d d Z! G d d Z" G d d Z# G d d Z$ G d d! Z% G d" d# Z&dS )$z Tests for hermite_e module.
)reduceNpolyval)assert_almost_equal
assert_raises assert_equalassert_ )r r
)r r r
) r ir r
)r r ir r
)r - r r r r
)r ir i r ir r
) r r i\r r ir r
)
r i r ir iz r ir r
c . t j | d S )Ngư>)tol)herme hermetrimxs y/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/numpy/polynomial/tests/test_hermite_e.pytrimr s ?1$ ' ' '' c & e Zd Zd Zd Zd Zd ZdS )
TestConstantsc > t t j ddg d S )Nr r
)r r hermedomainselfs r test_hermedomainz TestConstants.test_hermedomain! s U&Q00000r c < t t j dg d S )Nr )r r hermezeror s r test_hermezeroz TestConstants.test_hermezero$ s U_qc*****r c < t t j dg d S Nr
)r r hermeoner s r
test_hermeonezTestConstants.test_hermeone' s U^aS)))))r c > t t j ddg d S )Nr r
)r r hermexr s r test_hermexzTestConstants.test_hermex* s U\Aq6*****r N)__name__
__module__ __qualname__r" r% r) r, r r r r sP 1 1 1+ + +* * *+ + + + +r r c V e Zd Z ej ddd Zd Zd Zd Zd Z d Z
d Z d
S ) TestArithmeticr r
d c t d D ]}t d D ]}d| d| }t j t || dz }||xx dz
cc< ||xx dz
cc< t j dg|z dgz dg|z dgz }t
t | t | | d S N At i=, j=r
r err_msg)rangenpzerosmaxr hermeaddr r r! ijmsgtgtress r
test_hermeaddz TestArithmetic.test_hermeadd1 q @ @A1XX
@
@(a((Q((hs1ayy1}--A! A! naSUaS[1#a%1#+>> T#YYS 3?????
@ @ @r c t d D ]}t d D ]}d| d| }t j t || dz }||xx dz
cc< ||xx dz cc< t j dg|z dgz dg|z dgz }t
t | t | | d S r5 )r; r< r= r> r hermesubr r r@ s r
test_hermesubz TestArithmetic.test_hermesub; rG r c 4 t t j dg dg t t j dg ddg t dd D ]=}dg|z dgz }dg|dz
z |ddgz }t t j | | >d S )Nr r
r6 )r r hermemulxr; )r! rA serrD s r test_hermemulxz TestArithmetic.test_hermemulxE s U_aS))A3///U_aS))Aq6222q! 4 4A#a%1#+C#q1u+Aq )C --s 3 3 3 3 4 4r c t d D ]}dg|z dgz }t j | j | }t d D ]}d| d| }dg|z dgz }t j | j | }t j || }t j | j | } t t
| ||z dz k | t | ||z | d S )Nr6 r r
r7 r8 r9 )r; r hermevalr hermemulr lenr )
r! rA pol1val1rB rC pol2val2pol3val3s
r
test_hermemulz TestArithmetic.test_hermemulM s q
B
BA3q5A3;D>$&$//D1XX
B
B(a((Q((s1us{ ~dfd33 ~dD11 ~dfd33D QUQY.444#D$t)SAAAAA
B
B
Br c t d D ]}t d D ]}d| d| }dg|z dgz }dg|z dgz }t j || }t j || \ }}t j t j || | } t t
| t
| | d S )Nr6 r7 r8 r r
r9 )r; r r? hermedivrQ r r )
r! rA rB rC cicjrD quoremrE s
r
test_hermedivz TestArithmetic.test_hermediv[ s q @ @A1XX
@
@(a((Q((SUaS[SUaS[nR,, >#r22SnU^C%<%ij
i,j,k->ijkr
r6 r
)g ?rj rk c t t j g dg j d t j dd fdt D }t d D ]>}d| }|| }t j dg|z dgz }t ||| ?t d D ]}d g|z }t j | t t j dg j
| t t j ddg j
| t t j g d
j
| d S ) Nr
r r c 0 g | ]}t | S r0 r ).0re r s r
z0TestEvaluation.test_hermeval.. s! + + +qWQ]] + + +r
r7 r9 r
ro )r
r r ) r r rP sizer< rg Helistr; r r= shape)r! yrA rC rD rE dimsr s @r
test_hermevalz TestEvaluation.test_hermevalz sU U^B,,11555
KA + + + +F + + +r 7 7A !++CA$C.QCEQCK00C S# 6 6 6 6 6 q C CA3q5DA A3//5t < < < Aq6228$ ? ? ? 99955;T B B B B C Cr c ~ | j \ }}}| j \ }}}t t t j ||d d | j ||z }t j ||| j }t || t j d } t j | | | j }t |j dk d S Nro ro r
) r rx r
ValueErrorr
hermeval2dc2dr r< onesr rw
r! x1x2x3y1y2y3rD rE zs
r test_hermeval2dz TestEvaluation.test_hermeval2d s V
BV
B j%"2B2A2III e r2tx00C%%%
GFOO q!TX.. V#$$$$$r c | j \ }}}| j \ }}}t t t j |||d d | j ||z |z }t j |||| j }t || t j d } t j | | | | j }t |j dk d S r| ) r rx r r~ r
hermeval3dc3dr r< r r rw r s
r test_hermeval3dz TestEvaluation.test_hermeval3d s V
BV
B j%"2BBrrFDHMMM eBh r2r4844C%%%
GFOO q!Q11 V#$$$$$r c B | j \ }}}| j \ }}}t j d|| }t j ||| j }t || t j d } t j | | | j }t |j
dk d S )Nrl r} )ro r
ro r
) r rx r< einsumr hermegrid2dr r r r rw r s
r test_hermegrid2dz TestEvaluation.test_hermegrid2d s V
BV
B i 2r** B11C%%%
GFOO 1dh// X%&&&&&r c H | j \ }}}| j \ }}}t j d||| }t j |||| j }t || t j d } t j | | | | j }t |j
dk d S )Nrm r} )ro r
ro r
ro r
) r rx r< r r hermegrid3dr r r r rw r s
r test_hermegrid3dz TestEvaluation.test_hermegrid3d s V
BV
B i b"b11 BDH55C%%%
GFOO 1a22 X%&&&&&r N)r- r. r/ r< rc c1dr r r randomr r rx rz r r r r r0 r r ri ri p s
"(<<<
C
")IsC
(
(C
")L#sC
0
0C "Q&A<<< AC C C*% % %"% % %" ' ' ' ' ' ' ' 'r ri c e Zd Zd Zd ZdS ) TestIntegralc T t t t j dgd t t t j dgd t t t j dgdddg t t t j dgdg t t t j dgdg t t t j dgd t dd D ]9}dg|dz
z dgz }t j dg||
}t
|ddg :t d D ]}|dz }dg|z dgz }|gdg|z z d|z gz }t j | }t j |d|g
}t j | }t
t | t | t d D ]a}|dz }dg|z dgz }t j | }t j |d|gd }t
t j
d| | bt d D ]}|dz }dg|z dgz }|gdg|z z d|z gz }t j | }t j |d|gd }t j | }t
t | t | t d D ]}t dd D ]~} dg|z dgz }|d d }t | D ]}t j |d
}t j ||
}t
t | t | t d D ]}t dd D ]} dg|z dgz }|d d }t | D ]}t j |d|g
}t j || t t |
}t
t | t | t d D ]}t dd D ]} dg|z dgz }|d d }t | D ]}t j |d|gd } t j || t t | d }t
t | t | t d D ]}t dd D ]} dg|z dgz }|d d }t | D ]}t j |d|gd } t j || t t | d }t
t | t | d S )Nr ?r r
)lbnd)sclaxisro r6 )mk)r r r )r r r r ) r TypeErrorr hermeintr~ r; r
poly2herme
herme2polyr rP list)
r! rA r rE r polrD hermepolr rB s
r
test_hermeintzTestIntegral.test_hermeint s i !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 5 q A AAa%C#a%1#+C ',,H ~h!sDDDH r8 < .' " :::q **:::r r r c 6 g | ]}t j | S r0 r r s r rs z3TestIntegral.test_hermeint_axis..+ " 888q **888r r
c : g | ]}t j |d S )r
)r r r s r rs z3TestIntegral.test_hermeint_axis../ s' ===A Q///===r r
)r r )r< r vstackTr r r r! r rD rE s r test_hermeint_axisz TestIntegral.test_hermeint_axis# s i v&&i::CE:::;;=nSq)))C%%%i88C88899nSq)))C%%%i=====>>nSAA...C%%%%%r N)r- r. r/ r r r0 r r r r s6 Q: Q: Q:f& & & & &r r c e Zd Zd Zd ZdS )TestDerivativec V t t t j dgd t t t j dgd t d D ]L}dg|z dgz }t j |d }t
t | t | Mt d D ]s}t dd D ]`}dg|z dgz }t j t j || | }t t | t | att d D ]u}t dd D ]b}dg|z dgz }t j t j ||d |d }t t | t | cvd S ) Nr r r r6 r
r ro )r r )
r r r hermederr~ r; r r r r )r! rA rD rE rB s r
test_hermederz TestDerivative.test_hermeder6 s i !b999j%.1#r::: q / /A#a%1#+C.***C cDII . . . . q : :A1a[[
:
:c!eqcknU^C1%=%=%=CCC#DIItCyy9999
: q : :A1a[[
:
:c!eqcknN3!333qbB B B#DIItCyy9999
: : :r c h t j d }t j d |j D j }t j |d }t
|| t j d |D }t j |d }t
|| d S )Nr c 6 g | ]}t j | S r0 r r r s r rs z5TestDerivative.test_hermeder_axis..T r r r r c 6 g | ]}t j | S r0 r r s r rs z5TestDerivative.test_hermeder_axis..X r r r
)r< r r r r r r r s r test_hermeder_axisz!TestDerivative.test_hermeder_axisP s i v&&i::CE:::;;=nSq)))C%%%i88C88899nSq)))C%%%%%r N)r- r. r/ r r r0 r r r r 4 s2 : : :4
&
&
&
&
&r r c ` e Zd Zej d dz dz
Zd Zd Zd ZdS )
TestVanderrn ro r
c 8 t j d }t j |d }t |j dk t
d D ]7}dg|z dgz }t |d|f t j || 8t j ddgddgdd gg }t j |d }t |j d
k t
d D ]7}dg|z dgz }t |d|f t j || 8d S ) Nr
r r r r
.ro r6 )r
ro r )
r< rb r hermevanderr rw r; r rP rc )r! r vrA coefs r test_hermevanderzTestVander.test_hermevandera s+ 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 Dr c n | j \ }}}t j d }t j ||ddg }t j ||| }t j ||j }t || t j |g|gddg }t |j
dk d S )Nr} r
ro )r
r6 r ) r r< r r
hermevander2dr dotflatr r rw r! r r r re vanrD rE s r test_hermevander2dz TestVander.test_hermevander2dr s V
B I V $ $!"b1a&11 r2q))fS!&!!C%%% !2$q!f55 Y&'''''r c v | j \ }}}t j d }t j |||g d }t j |||| }t j ||j }t || t j |g|g|gg d }t |j
dk d S )N)ro r
r )r
ro r
)r
r6 ) r r< r r
hermevander3dr r r r r rw r s r test_hermevander3dz TestVander.test_hermevander3d s V
B I Y ' '!"b"iii88 r2r1--fS!&!!C%%% !2$rdIII>> Z'(((((r N) r- r. r/ r< r r r r r r0 r r r r ] sc
"Q&AD D D" ( ( ( ) ) ) ) )r r c e Zd Zd ZdS ) TestFittingc d }d }t t t j dgdgd t t t j dggdgd t t t j g dgd t t t j dgdgggd t t t j ddgdgd t t t j dgddgd t t t j dgdgddgg t t t j dgdgdddg t t t j dgdgdg t t t j dgdgg d t t t j dgdgg t j dd } || }t j ||d }t t | d
t t j
|| | t j ||g d }t t | d
t t j
|| | t j ||d
}t t | d t t j
|| | t j ||g d
}t t | d t t j
|| | t j ||g d }t t | d t t j
|| | t j |t j ||g j d }t |t j ||g j t j |t j ||g j g d }t |t j ||g j t j
| }| } d|dd d< d|dd d< t j || d | }
t |
| t j || g d | }
t |
| t j |t j | | g j d | } t | t j ||g j t j |t j | | g j g d | } t | t j ||g j g d}t t j ||d ddg t t j ||ddg ddg t j dd } || }t j ||d
} t t j
|| | t j ||g d }
t t j
||
| t | |
d S )Nc | | dz
z | dz
z S )Nr
ro r0 r s r fz$TestFitting.test_hermefit..f s a!e9a!e$ $r c | dz | dz z dz S )Nr ro r
r0 r s r f2z%TestFitting.test_hermefit..f2 s a4!Q$;? "r r
r r ro )w)ro r r r
r )r r
ro r
r6 )r r
ro r
r )ro r
r r
r )r
y ?r y )r ro r )r r~ r hermefitr r< rg r rR r rP rc r
zeros_likecopy)r! r r r rx coef3coef4coef2dr ywwcoef3wcoef2dcoef1coef2s r
test_hermefitzTestFitting.test_hermefit s % % % # # # j%.1#sB???i 1#Q???i aS!<<<i !ugqAAAi !Q!a@@@i !q!fa@@@i !qc1!FFFFi !qc1AGGGGj%.1#sREBBBj%.1#sJJJGGGi !qc2>>>
K1
AaDDq!Q''SZZ###EN1e 4 4a888q!\\\22SZZ###EN1e 4 4a888q!Q''SZZ###EN1e 4 4a888q!___55SZZ###EN1e 4 4a888q!___55SZZ###EN1e 4 4a888 28QF#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%=%=%?@@@
EN1a 3 3aV<<<EN1a!Q 8 81a&AAA KA BqEEq!Q''EN1e 4 4a888q!YYY//EN1e 4 4a888E5)))))r N)r- r. r/ r r0 r r r r s( H* H* H* H* H*r r c e Zd Zd Zd Zd ZdS )
TestCompanionc t t t j g t t t j dg d S r' )r r~ r hermecompanionr s r test_raiseszTestCompanion.test_raises s4 j%"6;;;j%"6<<<< ? ? ? ? @ @r c ` t t j ddg d dk d S )Nr
ro )r r )r r r r s r test_linear_rootz TestCompanion.test_linear_root s/ $aV,,T2c9:::::r N)r- r. r/ r r r r0 r r r r sD = = =@ @ @
; ; ; ; ;r r c e Zd Zd ZdS ) TestGaussc t j d \ }}t j |d }t j |j |z | }dt j | z }|d d d f |z |z }t |t j d t j dt j
z }t | | d S )Nr3 c r
ro ) r
hermegaussr r< r r sqrtdiagonalr eyepisum)r! r r r vvvdrD s r test_100zTestGauss.test_100 s $$1
a $ $
VAC!GQ
rwr{{}}%%
%
4[2
"Bs ,,, gag AEEGGS)))))r N)r- r. r/ r r0 r r r r s# * * * * *r r c 8 e Zd Zd Zd Zd Zd Zd Zd Zd Z dS ) TestMiscc t j g }t t | dg t dd D ]}t j t j t
j dd|z dz dd d }t j | }t j || }d}t t | |dz k t t j | d d t || d S )Nr
r6 r ro r )
r hermefromrootsr r r; r< cosrg r rP r rR r )r! rE rA rootsr rD s r test_hermefromrootsz TestMisc.test_hermefromroots s "2&&DIIs+++q! * *AF2;vq!A#'::14a4@AAE&u--C.,,CC CHHA% & & & 0 5 5b 91 = = = S ) ) ) ) * *r c t t j dg g t t j ddg dg t dd D ]h}t j dd| }t j t j | }t t | t | id S )Nr
r ro r6 )r r
hermerootsr; r< rg r r )r! rA rD rE s r test_hermerootszTestMisc.test_hermeroots
s E ,aS 1 12666E ,aV 4 4rd;;;q! 6 6A+b!Q''C"5#7#<#<==C S 499 5 5 5 5 6 6r c B g d}t t t j |d t t j | |d d t t j |d |d d t t j |d dg d S )N)ro r r
r r r
r ro r )r r~ r r r )r! r s r test_hermetrimzTestMisc.test_hermetrim s }} j%/4<<< U_T**D"I666U_T1--tCRCy999U_T1--s33333r c P t t j dd ddg d S )Nr
r )r r hermeliner s r test_hermelinezTestMisc.test_hermeline s( U_Q**QF33333r c t d D ]7}t t j dg|z dgz t | 8d S Nrt r r
)r; r r r rv r! rA s r test_herme2polyzTestMisc.test_herme2poly sS r J JA 0!Q! = =vay I I I I J Jr c t d D ]7}t t j t | dg|z dgz 8d S r )r; r r r rv r s r test_poly2hermezTestMisc.test_poly2herme$ sS r J JA 0 ; ;aSUaS[ I I I I J Jr c t j ddd }t j d|dz z }t j | }t || d S )Nr6 r ro )r< rg expr hermeweightr )r! r rD rE s r test_weightzTestMisc.test_weight( sR KAr " "fSAX ""C%%%%%r N)
r- r. r/ r r
r r r r r r0 r r r r s
*
*
*6 6 6 4 4 44 4 4J J JJ J J& & & & &r r )'__doc__ functoolsr numpyr< numpy.polynomial.hermite_e
polynomial hermite_er numpy.polynomial.polynomialr
numpy.testingr r r r rc He0He1He2He3He4He5He6He7He8He9rv r r r2 ri r r r r r r r r0 r r r. 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