;.fj W
d Z ddlmZ ddlZddlmc mZ ddl m
Z
ddl m Z m
Z
mZmZ ej dg Z ej ddg Z ej g d dz Z ej g d dz Z ej g d
d z Z ej g d d z Z ej g d
dz Z ej g d dz Z ej g d dz Z ej g d dz 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 legendre module.
)reduceNpolyval)assert_almost_equal
assert_raises assert_equalassert_ )r )r r )r r ir # )r r ir ? )r i r ir )r ir ; r iKr i ) r r ir i r ir i# )
r r r ir ibF r itr i{/ c . t j | d S )Ngư>)tol)leglegtrimxs x/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/numpy/polynomial/tests/test_legendre.pytrimr! s ;qd # # ## c & e Zd Zd Zd Zd Zd ZdS )
TestConstantsc > t t j ddg d S )Nr r
)r r legdomainselfs r test_legdomainz TestConstants.test_legdomain! s S]RG,,,,,r" c < t t j dg d S )Nr )r r legzeror' s r test_legzerozTestConstants.test_legzero$ s S[1#&&&&&r" c < t t j dg d S Nr
)r r legoner' s r test_legonezTestConstants.test_legone' s SZ!%%%%%r" c > t t j ddg d S )Nr r
)r r legxr' s r test_legxzTestConstants.test_legx* s SX1v&&&&&r" N)__name__
__module__ __qualname__r) r, r0 r3 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 ) 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 Nr At i=, j=r
r err_msg)rangenpzerosmaxr legaddr r! r( ijmsgtgtress r test_legaddzTestArithmetic.test_legadd1 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 r< )rA rB rC rD r legsubr r! rF s r test_legsubzTestArithmetic.test_legsub; rM r" c V t t j dg dg t t j dg ddg t dd D ]N}d|z dz }dg|z dgz }dg|dz
z ||z d|dz |z gz }t t j | | Od S )Nr r
r r
)r r legmulxrA )r( rG tmpserrJ s r test_legmulxzTestArithmetic.test_legmulxE s S[!%%s+++S[!%%1v...q! 0 0AA#'C#a%1#+C#q1u+3AE3; 77C S))3 / / / / 0 0r" 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? )rA r legvalr legmulr lenr )
r( rG pol1val1rH rI pol2val2pol3val3s
r test_legmulzTestArithmetic.test_legmulN 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? )rA r rE legdivrX r r! )
r( rG rH rI cicjrJ quoremrK s
r test_legdivzTestArithmetic.test_legdiv\ s q @ @A1XX
@
@(a((Q((SUaS[SUaS[jR(( :c2..SjC!4!4c:: T#YYS 3?????
@ @ @r" 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? )
rA rB aranger r rX arraylegpowr r! )r( rG rH rI crJ rK s r test_legpowzTestArithmetic.test_legpowg s q @ @A1XX
@
@(a((Q((Ia!e$$ SZ!Q!
>>jA&& T#YYS 3?????
@ @ @r" N) r4 r5 r6 rB linspacer rL rP rU r` rg rm r7 r" r r9 r9 . s B3 A@ @ @@ @ @0 0 0 B B B @ @ @@ @ @ @ @r" r9 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) @rq rq i,j->ij
i,j,k->ijkr r r
r
) ?rq 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 r7 r ).0rl r s r
z.TestEvaluation.test_legval.. s! * * *qWQ]] * * *r"
r= r? r r
)r
r r ) r r rW sizerB rn LlistrA r rC shape)r( yrG rI rJ rK dimsr s @r test_legvalzTestEvaluation.test_legval{ sQ SZQC((-q111
KA * * * *E * * *r 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
r ) r r~ r
ValueErrorr legval2dc2dr rB onesr r}
r( x1x2x3y1y2y3rJ rK zs
r
test_legval2dz TestEvaluation.test_legval2d 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 r~ r r r legval3dc3dr rB r r r} r s
r
test_legval3dz TestEvaluation.test_legval3d 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 )Nrr r )r
r r
r ) r r~ rB einsumr leggrid2dr r r r r} r s
r test_leggrid2dz TestEvaluation.test_leggrid2d 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 )Nrs r )r
r r
r r
r ) r r~ rB r r leggrid3dr r r r r} r s
r test_leggrid3dz TestEvaluation.test_leggrid3d s V
BV
B i b"b11mBB11C%%%
GFOOmAq!TX.. X%&&&&&r" N)r4 r5 r6 rB rj c1dr r r randomr r r~ r r r r r r7 r" r rp rp q s
"(<<<
C
")IsC
(
(C
")L#sC
0
0C "Q&A<<< A? ? ?*% % %"% % %" ' ' ' ' ' ' ' 'r" rp c e Zd 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 legintr rA r poly2legleg2polyr! rW list)
r( rG r rK r polrJ legpolr rH s
r test_legintzTestIntegral.test_legint s iaS"555j#*qc2666j#*qc1q!f===j#*qc<<<<j#*qcs;;;;iaSr:::: q! - -AQU qc!A*aSA+++C aV , , , , 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 r7 r r s r ry z1TestIntegral.test_legint_axis.., 4441A 444r" r
c : g | ]}t j |d S )r )r r r s r ry z1TestIntegral.test_legint_axis..0 s' 999A+++999r" r )r r )rB r vstackTr r r r( r rJ rK s r test_legint_axisz TestIntegral.test_legint_axis$ s i v&&i66666779j1%%%C%%%i4444455j1%%%C%%%i99S999::j***C%%%%%r" c L t t j dd d d S )Nr
r
r r )r r r r' s r test_legint_zerointordz#TestIntegral.test_legint_zerointord4 s$ SZ 1--y99999r" N)r4 r5 r6 r r r r7 r" r r r sE Q: Q: Q:f& & & : : : : :r" r c e Zd 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 legderr rA r r! r r )r( rG rJ rK rH s r test_legderzTestDerivative.test_legder: s iaS"555j#*qc2666 q / /A#a%1#+C*SA&&&C cDII . . . . q : :A1a[[
:
:c!eqckjC1!5!5!5;;;#DIItCyy9999
: q : :A1a[[
:
:c!eqckjC1!!.W r r" r r c 6 g | ]}t j | S r7 r r s r ry z3TestDerivative.test_legder_axis..[ r r" r
)rB r r r r r r r s r test_legder_axisz TestDerivative.test_legder_axisS s i v&&i66666779j1%%%C%%%i4444455j1%%%C%%%%%r" c R d}t t j |d dg d S )N)r
r
r r r r )r r r )r( rl s r test_legder_orderhigherthancoeffz/TestDerivative.test_legder_orderhigherthancoeff_ s+ SZ1%%s+++++r" N)r4 r5 r6 r r r r7 r" r r r 8 sA : : :2
&
&
&, , , , ,r" r c f e Zd Zej d dz dz
Zd Zd Zd Zd Z dS )
TestVanderrt 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 ) Nr r r r r
.r
r )r r
r )
rB ri r legvanderr r} rA r rW rj )r( r vrG coefs r test_legvanderzTestVander.test_legvanderg 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 rB r r legvander2dr dotflatr r r} r( r r r rl vanrJ rK s r test_legvander2dzTestVander.test_legvander2dx 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
r r r )r
r ) r rB r r legvander3dr r r r r r} r s r test_legvander3dzTestVander.test_legvander3d s V
B I Y ' 'ob"b)))44l2r2q))fS!&!!C%%% ordRD2$ :: Z'(((((r" c H t t t j dd d S )Nr r )r r r r r' s r test_legvander_negdegz TestVander.test_legvander_negdeg s j#-B?????r" N)
r4 r5 r6 rB r r r r r r r7 r" r r r c sw
"Q&A@ @ @" ( ( ( ) ) )@ @ @ @ @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
r
r7 r s r fz"TestFitting.test_legfit..f s a!e9a!e$ $r" c | dz | dz z dz S )Nr r
r
r7 r s r f2z#TestFitting.test_legfit..f2 s a4!Q$;? "r" r
r r r
)w)r
r r r r )r r
r
r r )r r
r
r r )r
r r r
r )r
y ?r y )r r
r )r r r legfitr rB rn r rY r rW rj r
zeros_likecopy)r( r r r r~ coef3coef4coef2dr ywwcoef3wcoef2dcoef1coef2s r test_legfitzTestFitting.test_legfit 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!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 / /!Q888CJq!aV 4 4q!f=== KA BqEE
1a##CJq% 0 0!444
1a++CJq% 0 0!444E5)))))r" N)r4 r5 r6 r r7 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 legcompanionr' 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 )rA r r r r} )r( rG r s r test_dimensionsz TestCompanion.test_dimensions s\ q! < e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d S )
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
r r r
r )
r legfromrootsr r! rA rB cosrn pirW r rY r )r( rK rG rootsr rJ s r test_legfromrootszTestMisc.test_legfromroots s r""DIIs+++q! * *AF2;vq!A#'::14a4@AAE"5))C*UC((CC CHHA% & & & S 1 1" 5q 9 9 9 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
r r r )r r legrootsrA rB rn r r! )r( rG rJ rK s r
test_legrootszTestMisc.test_legroots s CL! - -r222CL!Q 0 03%888q! 6 6A+b!Q''C,s / 4 455C 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)r
r r
r r r
r r
r )r r r r r )r( r s r test_legtrimzTestMisc.test_legtrim s }} j#+tR888 S[&&SbS 222S[q))49555S[q))A3/////r" c P t t j dd ddg d S )Nr r r r legliner' s r test_leglinezTestMisc.test_legline& s( S[A&&A/////r" c N t t j dd dg d S )Nr r r r' s r test_legline_zerosclz TestMisc.test_legline_zeroscl) s& S[A&&,,,,,r" c t d D ]7}t t j dg|z dgz t | 8d S Nrz r r
)rA r r r r| r( rG s r
test_leg2polyzTestMisc.test_leg2poly, sR r E EA aSUaS[ 9 958 D D D D E Er" c t d D ]7}t t j t | dg|z dgz 8d S r! )rA r r r r| r" s r
test_poly2legzTestMisc.test_poly2leg0 sR r E EA U1X 6 6A D D D D E Er" c ~ t j ddd }d}t j | }t || d S )Nr r
ru )rB rn r legweightr )r( r rJ rK s r test_weightzTestMisc.test_weight4 s? KAr " "mA C%%%%%r" N) r4 r5 r6 r r r r r r# r% r) r7 r" r r r s
*
*
*6 6 6 0 0 00 0 0- - -E E EE E E& & & & &r" r )'__doc__ functoolsr numpyrB numpy.polynomial.legendre
polynomiallegendrer numpy.polynomial.polynomialr
numpy.testingr r r r rj L0L1L2L3L4L5L6L7L8L9r| r! r$ r9 rp r r r r r r r r7 r" r r< s ' ' ' ' ' ' ' ' ' / / / / / / RXqc]]
RXq!f
RXjjj!
RXmmm Q
RX !!!#
RX$$$%%a'
RX+++,,R/
RX///003
RX999::3>
RX???@@D RRRRR0$ $ $ ' ' ' ' ' ' ' ' @@ @@ @@ @@ @@ @@ @@ @@F[' [' [' [' [' [' [' ['|f: f: f: f: f: f: f: f:R), ), ), ), ), ), ), ),V0@ 0@ 0@ 0@ 0@ 0@ 0@ 0@fJ* J* J* J* J* J* J* J*Z 7 7 7 7 7 7 7 7 * * * * * * * *&3& 3& 3& 3& 3& 3& 3& 3& 3& 3&r"