-2 d Z ddlmZ ddlZddlmc mZ ddl m
Z
ddl m Z m
Z
mZmZ ej dg dz Z ej ddg dz Z ej g d d z Z ej g d
d z Z ej g d d
z Z ej g d dz Z ej g d dz ZeeeeeeegZd 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 laguerre module.
)reduceNpolyval)assert_almost_equal
assert_raises assert_equalassert_ ) r
r ) i r r ) iH ir
r )x iiX i8 r r ) i i ii ir
r c . t j | d S )Ngư>)tol)laglagtrimxs x/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/numpy/polynomial/tests/test_laguerre.pytrimr s ;qd # # ## c & e Zd Zd Zd Zd Zd ZdS )
TestConstantsc > t t j ddg d S )Nr r
)r r lagdomainselfs r test_lagdomainz TestConstants.test_lagdomain s S]QF+++++r c < t t j dg d S )Nr )r r lagzeror" s r test_lagzerozTestConstants.test_lagzero! s S[1#&&&&&r c < t t j dg d S Nr
)r r lagoner" s r test_lagonezTestConstants.test_lagone$ s SZ!%%%%%r c > t t j ddg d S )Nr
r )r r lagxr" s r test_lagxzTestConstants.test_lagx' s SX2w'''''r N)__name__
__module__ __qualname__r$ r' r+ r. r r r r sP , , ,' ' '& & &( ( ( ( (r r c V e Zd Z ej ddd Zd Zd Zd Zd Z d Z
d Z d
S ) TestArithmetic 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 lagaddr r r# ijmsgtgtress r test_lagaddzTestArithmetic.test_lagadd. q @ @A1XX
@
@(a((Q((hs1ayy1}--A! A! j!Q!qc!eqck:: 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 r9 )r? r@ rA rB r lagsubr r rD s r test_lagsubzTestArithmetic.test_lagsub8 rK r c J t t j dg dg t t j dg ddg t dd D ]H}dg|z dgz }dg|dz
z | d|z dz |dz gz }t t j | | Id S )Nr r
r r: r )r r lagmulxr? r )r# rE serrH s r test_lagmulxzTestArithmetic.test_lagmulxB s S[!%%s+++S[!%%2w///q! 7 7A#a%1#+C#q1u+!QqS1WAh 77C C 0 0# 6 6 6 6 7 7r 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 )Nr: r r
r; r< r= )r? r lagvalr lagmulr lenr )
r# rE pol1val1rF rG pol2val2pol3val3s
r test_lagmulzTestArithmetic.test_lagmulJ s q
B
BA3q5A3;D:dfd++D1XX
B
B(a((Q((s1us{z$&$//z$--z$&$//D 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 )Nr: r; r< r r
r= )r? r rC lagdivrU r r )
r# rE rF rG cicjrH quoremrI s
r test_lagdivzTestArithmetic.test_lagdivX s q G GA1XX
G
G(a((Q((SUaS[SUaS[jR(( :c2..SjC!4!4c::#DIItCyy#FFFFF
G G Gr c r t d D ]}t d D ]}d| d| }t j |dz }t t j |g|z t j dg }t j || }t t | t | | d S )Nr: r; r< r
r= )
r? r@ aranger r rU arraylagpowr r )r# rE rF rG crH rI s r test_lagpowzTestArithmetic.test_lagpowc s q @ @A1XX
@
@(a((Q((Ia!e$$ SZ!Q!
>>jA&& T#YYS 3?????
@ @ @r N) r/ r0 r1 r@ linspacer rJ rN rR r] rd rj r2 r r r4 r4 + s B3 A@ @ @@ @ @7 7 7 B B B G G G@ @ @ @ @r r4 c e Zd Z ej g d Z ej dee Z ej deee Zej d dz dz
Z
e e
g d Z d Z
d Zd
Zd Zd Zd
S )TestEvaluation)g "@g ,g @i,j->ij
i,j,k->ijkr6 r: r r
) ?g @g @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 r2 r ).0ri r s r
z.TestEvaluation.test_lagval..} s! * * *qWQ]] * * *r r; r= r6 r )r
r r ) r r rT sizer@ rk Llistr? r rA shape)r# yrE rG rH rI dimsr s @r test_lagvalzTestEvaluation.test_lagvalw sQ SZQC((-q111
KA * * * *E * * *q 7 7A !++CA$C*QA ,,C S# 6 6 6 6 6 q ? ?A3q5DA As++14 8 8 8 A1v..4d ; ; ; Ayyy117 > > > > ? ?r 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 Nr r r6 ) r rz r
ValueErrorr lagval2dc2dr r@ onesr ry
r# x1x2x3y1y2y3rH rI zs
r
test_lagval2dz TestEvaluation.test_lagval2d s V
BV
B j#,BrrFDHEEE el2r48,,C%%%
GFOOl1a** 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 rz r r r lagval3dc3dr r@ r r ry r s
r
test_lagval3dz TestEvaluation.test_lagval3d s V
BV
B j#,B2A2III eBhl2r2tx00C%%%
GFOOl1aDH-- V#$$$$$r c B | j \ }}}| j \ }}}t j d|| }t j ||| j }t || t j d } t j | | | j }t |j
dk d S )Nrn r )r r6 r r6 ) r rz r@ einsumr laggrid2dr r r r ry r s
r test_laggrid2dz TestEvaluation.test_laggrid2d s V
BV
B i 2r**mBDH--C%%%
GFOOmAq$(++ X%&&&&&r c H | j \ }}}| j \ }}}t j d||| }t j |||| j }t || t j d } t j | | | | j }t |j
dk d S )Nro r )r r6 r r6 r r6 ) r rz r@ r r laggrid3dr r r r ry r s
r test_laggrid3dz TestEvaluation.test_laggrid3d s V
BV
B i b"b11mBB11C%%%
GFOOmAq!TX.. X%&&&&&r N)r/ r0 r1 r@ rg c1dr r r randomr r rz r| r r r r r2 r r rm rm m s
"(>>>
"
"C
")IsC
(
(C
")L#sC
0
0C "Q&A<<< A? ? ?*% % %"% % %" ' ' ' ' ' ' ' 'r rm 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)sclaxisr r: )mk)r r r )r r r r ) r TypeErrorr lagintr r? r poly2laglag2polyr rT list)
r# rE r rI r polrH lagpolr rF s
r test_lagintzTestIntegral.test_lagint s iaS"555j#*qc2666j#*qc1q!f===j#*qc<<<<j#*qcs;;;;iaSr:::: q! . .AQU qc!A*aSA+++C aW - - - - q 6 6Aa%C#a%1#+C#A+3'C\#&&FZ!s333F,v&&C S 499 5 5 5 5 q ; ;Aa%C#a%1#+C\#&&FZ!s<<>>CCjT%((^^"EEE#DIItCyy9999
: q : :A1a[[
:
:c!eqck!!!f q = =A *SA!!<<.$ 6661A 666r r r c 6 g | ]}t j | S r2 r r s r ru z1TestIntegral.test_lagint_axis..( 4441A 444r r
c : g | ]}t j |d S )r6 )r r r s r ru z1TestIntegral.test_lagint_axis.., s' 999A+++999r r6 )r r )r@ r vstackTr r r r# r rH rI s r test_lagint_axisz TestIntegral.test_lagint_axis s i v&&i66666779j1%%%C%%%i4444455j1%%%C%%%i99S999::j***C%%%%%r N)r/ r0 r1 r r r2 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 r: r
r r )r r )
r r r lagderr r? r r r r )r# rE rH rI rF s r test_lagderzTestDerivative.test_lagder3 s iaS"555j#*qc2666 q / /A#a%1#+C*SA&&&C cDII . . . . q : :A1a[[
:
:c!eqckjC1!5!5!5;;;#DIItCyy9999
: q : :A1a[[
:
:c!eqckjC1!!.P r r r r c 6 g | ]}t j | S r2 r r s r ru z3TestDerivative.test_lagder_axis..T r r r
)r@ r r r r r r r s r test_lagder_axisz TestDerivative.test_lagder_axisL s i v&&i66666779j1%%%C%%%i4444455j1%%%C%%%%%r N)r/ r0 r1 r r r2 r r r r 1 s2 : : :2
&
&
&
&
&r r c ` e Zd Zej d dz dz
Zd Zd Zd ZdS )
TestVanderrp r 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 ) Nr6 r r r r
.r r: r )r6 r r )
r@ rf r lagvanderr ry r? r rT rg )r# r vrE coefs r test_lagvanderzTestVander.test_lagvander] s) IaLL M!Q 6!"""q @ @A3q5A3;D #q& 3:a+>+> ? ? ? ?
Hq!fq!fq!f- . . M!Q 9$%%%q @ @A3q5A3;D #q& 3:a+>+> ? ? ? ? @ @r 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
r )r
r: r ) r r@ r r lagvander2dr dotflatr r ry r# r r r ri vanrH rI s r test_lagvander2dzTestVander.test_lagvander2dn s V
B I V $ $ob"q!f--l2r1%%fS!&!!C%%% ordRD1a&11 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)r r6 r )r
r r6 )r
r: r ) r r@ r r lagvander3dr r r r r ry r s r test_lagvander3dzTestVander.test_lagvander3d{ s V
B I Y ' 'ob"b)))44l2r2q))fS!&!!C%%% ordRD2$ :: Z'(((((r N) r/ r0 r1 r@ r r r r r r2 r r r r Y sc
"Q&A@ @ @" ( ( ( ) ) ) ) )r r c e Zd Zd ZdS ) TestFittingc
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 |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 d S )Nc | | dz
z | dz
z S )Nr
r r2 r s r fz"TestFitting.test_lagfit..f s a!e9a!e$ $r r
r r r )w)r r r r6 r )r r
r r6 r: )r r
r r6 r )r
y ?r y )r r r lagfitr r@ rk r rV r rT rg r
zeros_likecopy) r# r r rz coef3coef4coef2dr ywwcoef3wcoef2ds r test_lagfitzTestFitting.test_lagfit s % % % j#*qcA3;;;iqcUQC;;;iR!a888iaSQC5'1===iaVaS!<<<iaS1a&!<<<iaS1#qaSEBBBBiaS1#qQFCCCCj#*qcA3>>>j#*qcA3
CCCiaS1#r:::
K1
AaDD
1a##SZZ###CJq% 0 0!444
1a ..SZZ###CJq% 0 0!444
1a##SZZ###CJq% 0 0!444
1a 11SZZ###CJq% 0 0!444ArxA / / 1155FBHeU^$<$<$>???ArxA / / 1<<<@@FBHeU^$<$<$>??? M!
VVXX!$Q$!$Q$Ar1***FE***Ar<<<1555FE****Q"b 2 2 4a1===GRXuen%=%=%?@@@*Q"b 2 2 4lllaHHHGRXuen%=%=%?@@@
CJq!Q / /!R999CJq!aV 4 4q"g>>>>>r N)r/ r0 r1 r r2 r r r r s# 9? 9? 9? 9? 9?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 lagcompanionr" s r test_raiseszTestCompanion.test_raises s4 j#"2B777j#"2QC88888r c t dd D ]8}dg|z dgz }t t j | j ||fk 9d S )Nr
r: r )r? r r r ry )r# rE r s r test_dimensionsz TestCompanion.test_dimensions s\ q! < r& s/ ' ' ' ' ' ' ' ' ' / / / / / / RXqc]]1_
RXq"gq
RXjjj!
RXooo q
RX###$$R'
RX,,,--c1
RX44455c9 RRR$$ $ $ ( ( ( ( ( ( ( ( ?@ ?@ ?@ ?@ ?@ ?@ ?@ ?@D[' [' [' [' [' [' [' ['|c& c& c& c& c& c& c& c&L%& %& %& %& %& %& %& %&P-) -) -) -) -) -) -) -)`;? ;? ;? ;? ;? ;? ;? ;?| 7 7 7 7 7 7 7 7 * * * * * * * *&0& 0& 0& 0& 0& 0& 0& 0& 0& 0&r