f4Vu
d Z ddlmZ ddlZddlmc mZ ddl m
Z
ddl m Z m
Z
mZmZ d ZdgZddgZg dZg d Zg d
Zg d Zg d Zg d
Zg dZg dZeeeeeeeeeeg
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% G d$ d% Z& G d& d' Z'dS )(z Tests for chebyshev module.
)reduceNpolyval)assert_almost_equal
assert_raises assert_equalassert_c . t j | d S )Ngư>)tol)chebchebtrimxs y/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/numpy/polynomial/tests/test_chebyshev.pytrimr s = % % %% )r )r r )r r ir )r r ir )r r r ir )r ir 8 r ir @ ) r r ir r i r )
r r ir i r ir c e Zd Zd Zd ZdS ) TestPrivatec t d D ]{}t j dgdg|z z t j }t j dg|z dgz dg|z z t j }t j | }t
|| |d S )Nr r r ?)rangenparraydoubler _cseries_to_zseriesr selfiinptgtress r test__cseries_to_zseriesz$TestPrivate.test__cseries_to_zseries! s q # #A(A3!Q; 22C(B46QC<2$q&0")< t t j ddg d S )Nr r )r r
chebdomainr- s r test_chebdomainz TestConstants.test_chebdomain2 s T_r1g.....r c < t t j dg d S )Nr )r r chebzeror> s r
test_chebzerozTestConstants.test_chebzero5 s T]QC(((((r c < t t j dg d S Nr )r r cheboner> s r test_chebonezTestConstants.test_chebone8 s T\A3'''''r c > t t j ddg d S )Nr r )r r chebxr> s r
test_chebxzTestConstants.test_chebx; s TZ!Q(((((r N)r6 r7 r8 r? rB rF rI r9 r r r; r; 0 sP / / /) ) )( ( () ) ) ) )r r; c 2 e Zd Zd Zd Zd Zd Zd Zd ZdS )TestArithmeticc 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)r' r( zerosmaxr chebaddr r r- r. jmsgr0 r1 s r test_chebaddzTestArithmetic.test_chebaddA q @ @A1XX
@
@(a((Q((hs1ayy1}--A! A! lA3q5A3;A << 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 rM )r' r( rR rS r chebsubr r rU s r test_chebsubzTestArithmetic.test_chebsubK rY r c 2 t t j dg dg t t j dg ddg t dd D ]<}dg|z dgz }dg|dz
z g dz }t t j | | =d S )Nr r r )r& r r& )r r chebmulxr' )r- r. serr0 s r
test_chebmulxz TestArithmetic.test_chebmulxU s T]A3''!---T]A3''!Q000q! 2 2A#a%1#+C#q1u+ +C s++S 1 1 1 1 2 2r c t d D ]}t d D ]}d| d| }t j ||z dz }|||z xx dz
cc< |t ||z
xx dz
cc< t j dg|z dgz dg|z dgz }t
t | t | | d S )Nr rN rO r r& r rP )r' r( rR absr chebmulr r rU s r test_chebmulzTestArithmetic.test_chebmul] s q @ @A1XX
@
@(a((Q((hq1uqy))AE
b
CAJJ2%lA3q5A3;A << T#YYS 3?????
@ @ @r 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 rN rO r r rP )r' r rT chebdivrc r r )
r- r. rV rW cicjr0 quoremr1 s
r test_chebdivzTestArithmetic.test_chebdivg s q @ @A1XX
@
@(a((Q((SUaS[SUaS[l2r** <R00Sl4<R#8#8#>> 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 rN rO r rP )
r' r( aranger r rc r) chebpowr r )r- r. rV rW cr0 r1 s r test_chebpowzTestArithmetic.test_chebpowr s q @ @A1XX
@
@(a((Q((Ia!e$$ T\A3q5"(A3--@@l1a(( T#YYS 3?????
@ @ @r N) r6 r7 r8 rX r\ r` rd rk rp r9 r r rK rK ? s @ @ @@ @ @2 2 2@ @ @ @ @ @@ @ @ @ @r rK 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 @ @ ?i,j->ij
i,j,k->ijk r r r ) ?rs 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 r9 r ).0ro r s r
z/TestEvaluation.test_chebval.. s! * * *qWQ]] * * *r
rN rP rx r )r r r ) r r chebvalsizer( linspaceTlistr' r rR shape)r- yr. rW r0 r1 dimsr s @r test_chebvalzTestEvaluation.test_chebval sU T\"qc**/333
KA * * * *E * * *r 7 7A !++CA$C,q1#a%1#+..C S# 6 6 6 6 6 q A AA3q5DA a!--3T : : : a!Q006 = = = a3394 @ @ @ @ A Ar 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 rx ) r r r
ValueErrorr chebval2dc2dr r( onesr r
r- x1x2x3y1y2y3r0 r1 zs
r test_chebval2dz TestEvaluation.test_chebval2d s V
BV
B j$."b!fdhGGG enRTX..C%%%
GFOOnQ48,, 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 chebval3dc3dr r( r r r r s
r test_chebval3dz TestEvaluation.test_chebval3d s V
BV
B j$."b"RaR&$(KKK eBhnRR22C%%%
GFOOnQ1dh// V#$$$$$r c B | j \ }}}| j \ }}}t j d|| }t j ||| j }t || t j d } t j | | | j }t |j
dk d S )Nru r )r rx r rx ) r r r( einsumr
chebgrid2dr r r r r r s
r test_chebgrid2dz TestEvaluation.test_chebgrid2d s V
BV
B i 2r**ob"dh//C%%%
GFOOoaDH-- X%&&&&&r c H | j \ }}}| j \ }}}t j d||| }t j |||| j }t || t j d } t j | | | | j }t |j
dk d S )Nrv r )r rx r rx r rx ) r r r( r r
chebgrid3dr r r r r r s
r test_chebgrid3dz TestEvaluation.test_chebgrid3d s V
BV
B i b"b11ob"b$(33C%%%
GFOOoaAtx00 X%&&&&&r N)r6 r7 r8 r( r) c1dr r r randomr r r r r r r r r9 r r rr rr | s
"(>>>
"
"C
")IsC
(
(C
")L#sC
0
0C "Q&A<<< AA A A*% % %"% % %" ' ' ' ' ' ' ' 'r rr 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 r )lbnd)sclaxisr r )mk)r r r )r r r r ) r TypeErrorr chebintr r' r poly2cheb cheb2polyr r list)
r- r. r r1 r polr0 chebpolr rV s
r test_chebintzTestIntegral.test_chebint s i sB777j$,R888j$,QA???j$,1#>>>>j$,!====i s<<<< q! - -AQU qc!A,sa1---C aV , , , , q 6 6Aa%C#a%1#+C#A+3'CnS))Gl7aA3777G.))C S 499 5 5 5 5 q > >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 5 q : :A1a[[
:
:c!eqck!!!f q 1 1A ,sa000CCl3!,,,#DIItCyy9999
: q : :A1a[[
:
:c!eqck!!!f q 8 8A ,saA3777CCl3!tE!HH~~>>>#DIItCyy9999
: q : :A1a[[
:
:c!eqck!!!f q A AA ,saA3R@@@CCl3!tE!HH~~BGGG#DIItCyy9999
: q : :A1a[[
:
:c!eqck!!!f q ? ?A ,saA3A>>>CCl3!tE!HH~~1EEE#DIItCyy9999
: : :r c t j d }t j d |j D j }t j |d }t
|| t j d |D }t j |d }t
|| t j d |D }t j |dd }t
|| d S )
Nrx r c 6 g | ]}t j | S r9 r r r| ro s r r} z2TestIntegral.test_chebint_axis..3 888Q a 888r r r c 6 g | ]}t j | S r9 r r s r r} z2TestIntegral.test_chebint_axis..7 666Q a 666r r c : g | ]}t j |d S )rx )r r r s r r} z2TestIntegral.test_chebint_axis..; s' ;;;! a1---;;;r rx )r r )r( r vstackTr r r r- r r0 r1 s r test_chebint_axisz TestIntegral.test_chebint_axis/ s i v&&i88#%88899;l3Q'''C%%%i66#66677l3Q'''C%%%i;;s;;;<<l3!!,,,C%%%%%r N)r6 r7 r8 r r r9 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 chebderr r' r r r r )r- r. r0 r1 rV s r test_chebderzTestDerivative.test_chebderB s i sB777j$,R888 q / /A#a%1#+C,sa(((C cDII . . . . q : :A1a[[
:
:c!eqckl4<q#9#9#9Q???#DIItCyy9999
: q : :A1a[[
:
:c!eqckl4<qa#@#@#@A2NNN#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 r9 r r r s r r} z4TestDerivative.test_chebder_axis.._ r r r r c 6 g | ]}t j | S r9 r r s r r} z4TestDerivative.test_chebder_axis..c r r r )r( r r r r r r r s r test_chebder_axisz TestDerivative.test_chebder_axis[ s i v&&i88#%88899;l3Q'''C%%%i66#66677l3Q'''C%%%%%r N)r6 r7 r8 r r r9 r r r r @ s2 : : :2
&
&
&
&
&r r c ` e Zd Zej d dz dz
Zd Zd Zd ZdS )
TestVanderrw 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 ) Nrx r r r r .r r )rx r r )
r( rm r
chebvanderr r r' r r r) )r- r vr. coefs r test_chebvanderzTestVander.test_chebvanderl 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 Br 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 chebvander2dr dotflatr r r r- r r r ro vanr0 r1 s r test_chebvander2dz TestVander.test_chebvander2d} s V
B I V $ $ BA//nRQ''fS!&!!C%%% rdQF33 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 rx r )r r rx )r r ) r r( r r chebvander3dr r r r r r r s r test_chebvander3dz TestVander.test_chebvander3d s V
B I Y ' ' BIII66nRR++fS!&!!C%%% rdRD)))<< Z'(((((r N) r6 r7 r8 r( r r r r r r9 r r r r h sc
"Q&AB B B" ( ( ( ) ) ) ) )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 r9 r s r fz#TestFitting.test_chebfit..f s a!e9a!e$ $r c | dz | dz z dz S )Nr r r r9 r s r f2z$TestFitting.test_chebfit..f2 s a4!Q$;? "r r r r r )w)r r r rx r )r r r rx r )r r r rx r )r rx r r r )r y ?r y )r r r )r r r chebfitr r( r r lenr r r) r
zeros_likecopy)r- r r r r coef3coef4coef2dr ywwcoef3wcoef2dcoef1coef2s r test_chebfitzTestFitting.test_chebfit s % % % # # # j$,aS"===i uqc1===i rA3:::i sqcUGQ???i 1vsA>>>i sQFA>>>i sQCseDDDDi sQCq!fEEEEj$,aS2%@@@j$,aS***EEEi sQC<<<
K1
AaDD Q1%%SZZ###DLE 2 2A666 Q<<<00SZZ###DLE 2 2A666 Q1%%SZZ###DLE 2 2A666 Q???33SZZ###DLE 2 2A666 Q???33SZZ###DLE 2 2A666 a1a&!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%=%=%?@@@
DLAq 1 1Aq6:::DLA1v 6 6A??? KA BqEE Q1%%DLE 2 2A666 Q999--DLE 2 2A666E5)))))r N)r6 r7 r8 r r9 r r r r s( H* H* H* H* H*r r c & e Zd Zd Zd Zd Zd ZdS )TestInterpolatec ||dz
z |dz
z S r r9 )r- r s r r zTestInterpolate.f s AE{a!e$$r c t t t j | j d t t
t j | j d d S )Nr g $@)r r r chebinterpolater r r> s r test_raiseszTestInterpolate.test_raises s: j$"6CCCi!5tvsCCCCCr c t dd D ]6}t t j | j | j |dz fk 7d S )Nr r )r' r r r r r )r- degs r test_dimensionsz TestInterpolate.test_dimensions sU A;; K KC D(55;azI J J J J K Kr c d }t j ddd }t dd D ]^}t d|dz D ]H}t j |||f }t t j || ||| d I_d S )Nc | |z S )Nr9 )r ps r powxz0TestInterpolate.test_approximation..powx s a4Kr r r r~ r )decimal)r( r r' r r r r )r- r r r r ro s r test_approximationz"TestInterpolate.test_approximation s
KAr " "B<< P PC1cAg&&
P
P(sQD99#DLA$6$6Q
BOOOOO
P P Pr N)r6 r7 r8 r r r r r9 r r r r s[ % % %D D DK K K P P P P Pr r c e Zd Zd Zd Zd ZdS )
TestCompanionc t t t j g t t t j dg d S rD )r r r
chebcompanionr> s r r zTestCompanion.test_raises s4 j$"4b999j$"4qc:::::r c t dd D ]8}dg|z dgz }t t j | j ||fk 9d S )Nr r r )r' r r r r )r- r. r s r r z TestCompanion.test_dimensions s\ q! > >A3q5A3;D D&t,,2q!f< = = = = > >r c ` t t j ddg d dk d S )Nr r )r r )r r r r> s r test_linear_rootz TestCompanion.test_linear_root s/ "Aq6**40C788888r N)r6 r7 r8 r r r
r9 r r r r sA ; ; ;> > >
9 9 9 9 9r 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
}t | | d S )Nd c r ) r chebgaussr r( r r sqrtdiagonalr eyepisum)r- r r r vvvdr0 s r test_100zTestGauss.test_100 s ~c""1
OAr " "
VAC!GQ
rwr{{}}%%
%
4[2
"Bs ,,, eAEEGGS)))))r N)r6 r7 r8 r r9 r r r r
s# * * * * *r r c D e Zd Zd Zd Zd Zd Zd Zd Zd Z d 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 }dg|z dgz }t j | d|dz
z z }t t | t | d S )Nr r r r ) r
chebfromrootsr r r' r( cosr r )r- r1 r. rootsr0 s r test_chebfromrootszTestMisc.test_chebfromroots" s $$DIIs+++q! 6 6AF2;vq!A#'::14a4@AAE#a%1#+C$U++A!H4C S 499 5 5 5 5 6 6r 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 chebrootsr' r( r r r )r- r. r0 r1 s r test_chebrootszTestMisc.test_chebroots+ s DNA3 / /444DNAq6 2 2SE:::q! 6 6A+b!Q''C.!3C!8!899C 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_chebtrimzTestMisc.test_chebtrim3 s }} j$-r::: T]4(($ss)444T]4++T#2#Y777T]4++aS11111r c P t t j dd ddg d S )Nrx r )r r chebliner> s r
test_cheblinezTestMisc.test_chebline> s( T]1a((1a&11111r c t d D ]7}t t j dg|z dgz t | 8d S Nr~ r r )r' r r r r r- r. s r test_cheb2polyzTestMisc.test_cheb2polyA sR r G GA s1us{ ; ;U1X F F F F G Gr c t d D ]7}t t j t | dg|z dgz 8d S r, )r' r r r r r- s r test_poly2chebzTestMisc.test_poly2chebE sR r G GA uQx 8 81#a%1#+ F F F F G Gr c t j ddd dd }dt j d|z t j d|z
z z }t j | }t || d S )Nr r ry )r( r r r
chebweightr )r- r r0 r1 s r test_weightzTestMisc.test_weightI sj KAr " "1R4 ("'!a%..271q5>>12oa C%%%%%r c t t t j d t t t j d dg}t t j d | ddg}t t j d | g d}t t j d | g d }t t j d
| d S ) Nrt r r g;fg;f?r )gLXzr gLXz?rx )g( 1kgŜ}ؿgŜ}?g( 1k?r )r r r chebpts1r r- r0 s r
test_chebpts1zTestMisc.test_chebpts1O s j$-555j$-333 cDM! , ,c222#%89DM! , ,c222<<<DM! , ,c222HHHDM! , ,c22222r c t t t j d t t t j d ddg}t t j d | g d}t t j d | g d}t t j d | g d }t t j d
| d S ) Nrt r r r )r r r rx )r r r& r r )g gKfr gKf?ry r )r r r chebpts2r r7 s r
test_chebpts2zTestMisc.test_chebpts2^ s j$-555j$-333 1gDM! , ,c222jjDM! , ,c222 DM! , ,c222===DM! , ,c22222r N) r6 r7 r8 r" r% r' r* r. r0 r4 r8 r; r9 r r r r s 6 6 66 6 6 2 2 22 2 2G G GG G G& & &
3
3
3
3
3
3
3
3r r )(__doc__ functoolsr numpyr( numpy.polynomial.chebyshev
polynomial chebyshevr numpy.polynomial.polynomialr
numpy.testingr r r r r T0T1T2T3T4T5T6T7T8T9r r$ r; rK rr r r r r r r r r r9 r r rN sI ) ) ) ) ) ) ) ) ) / / / / / /
& & & SVZZ]] ###)))--- RRRRR0# # # # # # # #" ) ) ) ) ) ) ) ) :@ :@ :@ :@ :@ :@ :@ :@z[' [' [' [' [' [' [' ['|c& c& c& c& c& c& c& c&L%& %& %& %& %& %& %& %&P-) -) -) -) -) -) -) -)`J* J* J* J* J* J* J* J*ZP P P P P P P P2 9 9 9 9 9 9 9 9 * * * * * * * *&K3 K3 K3 K3 K3 K3 K3 K3 K3 K3r