bg*P dZddlmZddlZddlmcmZddl m Z ddl m Z m Z mZmZdZdgZddgZgdZgd Zgd Zgd Zgd Zgd ZgdZgdZeeeeeeeeeeg ZGddZGddZGddZGddZGddZ GddZ!GddZ"GddZ#Gd d!Z$Gd"d#Z%Gd$d%Z&Gd&d'Z'dS)(zTests for chebyshev module. )reduceNpolyval)assert_almost_equal assert_raises assert_equalassert_c.tj|dS)Ngư>)tol)chebchebtrimxs x/builddir/build/BUILD/cloudlinux-venv-1.0.7/venv/lib64/python3.11/site-packages/numpy/polynomial/tests/test_chebyshev.pytrimrs = % % %%)r)rr)rrir)rrir)rrrir )rir8rir@) rrirrir) r riririrceZdZdZdZdS) TestPrivatectdD]{}tjdgdg|zztj}tjdg|zdgzdg|zztj}t j|}t |||dS)Nrrr?)rangenparraydoubler _cseries_to_zseriesrselfiinptgtress rtest__cseries_to_zseriesz$TestPrivate.test__cseries_to_zseries!sq # #A(A3!Q; 22C(B46QC<2$q&0")<ttjddgdS)Nrr)rr chebdomainr-s rtest_chebdomainzTestConstants.test_chebdomain2sT_r1g.....rc<ttjdgdS)Nr)rr chebzeror>s r test_chebzerozTestConstants.test_chebzero5sT]QC(((((rc<ttjdgdSNr)rr cheboner>s r test_chebonezTestConstants.test_chebone8sT\A3'''''rc>ttjddgdS)Nrr)rr chebxr>s r test_chebxzTestConstants.test_chebx;sTZ!Q(((((rN)r6r7r8r?rBrFrIr9rrr;r;0sP///)))((()))))rr;c2eZdZdZdZdZdZdZdZdS)TestArithmeticctdD]}tdD]}d|d|}tjt||dz}||xxdz cc<||xxdz cc<t jdg|zdgzdg|zdgz}t t|t||dSNrAt i=, j=rrerr_msg)r'r(zerosmaxr chebaddrrr-r.jmsgr0r1s r test_chebaddzTestArithmetic.test_chebaddAq @ @A1XX @ @(a((Q((hs1ayy1}--A! A! lA3q5A3;A <<T#YYS 3?????  @ @ @rctdD]}tdD]}d|d|}tjt||dz}||xxdz cc<||xxdzcc<t jdg|zdgzdg|zdgz}t t|t||dSrM)r'r(rRrSr chebsubrrrUs r test_chebsubzTestArithmetic.test_chebsubKrYrc2ttjdgdgttjdgddgtddD]<}dg|zdgz}dg|dz zgdz}ttj||=dS)Nrrr)r&rr&)rr chebmulxr')r-r.serr0s r test_chebmulxzTestArithmetic.test_chebmulxUsT]A3''!---T]A3''!Q000q! 2 2A#a%1#+C#q1u+ +C s++S 1 1 1 1 2 2rctdD]}tdD]}d|d|}tj||zdz}|||zxxdz cc<|t||z xxdz cc<t jdg|zdgzdg|zdgz}t t|t||dS)NrrNrOrr&rrP)r'r(rRabsr chebmulrrrUs r test_chebmulzTestArithmetic.test_chebmul]sq @ @A1XX @ @(a((Q((hq1uqy))AE b CAJJ2%lA3q5A3;A <<T#YYS 3?????  @ @ @rctdD]}tdD]}d|d|}dg|zdgz}dg|zdgz}tj||}tj||\}}tjtj|||} t t | t ||dS)NrrNrOrrrP)r'r rTchebdivrcrr) r-r.rVrWcicjr0quoremr1s r test_chebdivzTestArithmetic.test_chebdivgsq @ @A1XX @ @(a((Q((SUaS[SUaS[l2r**<R00Sl4<R#8#8#>>T#YYS 3????? @ @ @rc rtdD]}tdD]}d|d|}tj|dz}ttj|g|ztjdg}t j||}tt|t||dS)NrrNrOrrP) r'r(arangerr rcr)chebpowrr)r-r.rVrWcr0r1s r test_chebpowzTestArithmetic.test_chebpowrsq @ @A1XX @ @(a((Q((Ia!e$$T\A3q5"(A3--@@l1a((T#YYS 3?????  @ @ @rN) r6r7r8rXr\r`rdrkrpr9rrrKrK?s@@@@@@222@@@ @ @ @@@@@@rrKceZdZejgdZejdeeZejdeeeZej ddzdz Z e e gdZ dZ d Zd Zd Zd Zd S)TestEvaluation)g@@?i,j->ij i,j,k->ijkrrr)?rsg@cttjgdgjdt jddfdt D}tdD]>}d|}||}tjdg|zdgz}t|||?tdD]}d g|z}t j |ttjdgj |ttjddgj |ttjgd j |dS) Nrrrc0g|]}t|Sr9r).0rors r z/TestEvaluation.test_chebval..s! * * *qWQ]] * * *r rNrPrxr)rrr) rr chebvalsizer(linspaceTlistr'rrRshape)r-yr.rWr0r1dimsrs @r test_chebvalzTestEvaluation.test_chebvalsUT\"qc**/333 KA   * * * *E * * *r 7 7A!++CA$C,q1#a%1#+..C S# 6 6 6 6 6q A AA3q5DA a!--3T : : : a!Q006 = = = a3394 @ @ @ @  A Arc~|j\}}}|j\}}}tttj||dd|j||z}t j|||j}t||tj d} t j| | |j}t|j dkdSNrrrx) rrr ValueErrorr chebval2dc2drr(onesr r r-x1x2x3y1y2y3r0r1zs rtest_chebval2dzTestEvaluation.test_chebval2dsV BV B j$."b!fdhGGGenRTX..C%%% GFOOnQ48,, V#$$$$$rc |j\}}}|j\}}}tttj|||dd|j||z|z}t j||||j}t||tj d} t j| | | |j}t|j dkdSr) rrrrr chebval3dc3drr(rr rrs rtest_chebval3dzTestEvaluation.test_chebval3dsV BV B j$."b"RaR&$(KKKeBhnRR22C%%% GFOOnQ1dh// V#$$$$$rcB|j\}}}|j\}}}tjd||}t j|||j}t||tjd} t j| | |j}t|j dkdS)Nrur)rrxrrx) rrr(einsumr chebgrid2drrrr rrs rtest_chebgrid2dzTestEvaluation.test_chebgrid2dsV BV Bi 2r**ob"dh//C%%% GFOOoaDH-- X%&&&&&rcH|j\}}}|j\}}}tjd|||}t j||||j}t||tjd} t j| | | |j}t|j dkdS)Nrvr)rrxrrxrrx) rrr(rr chebgrid3drrrr rrs rtest_chebgrid3dzTestEvaluation.test_chebgrid3dsV BV Bi b"b11ob"b$(33C%%% GFOOoaAtx00 X%&&&&&rN)r6r7r8r(r)c1drrrrandomrrrrrrrrr9rrrrrr|s "(>>> " "C ")IsC ( (C ")L#sC 0 0C   "Q&A<<<  AAAA*%%%"%%%" ' ' ' ' ' ' ' 'rrrceZdZdZdZdS) 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)Nrr&rr)lbnd)sclaxisrr)mk)rrr)rrrr) r TypeErrorr chebintrr'r poly2cheb cheb2polyrrlist) r-r.rr1rpolr0chebpolrrVs r test_chebintzTestIntegral.test_chebintsisB777j$,R888j$,QA???j$,1#>>>>j$,!====is<<<<q! - -AQU qc!A,sa1---C aV , , , ,q 6 6Aa%C#a%1#+C#A+3'CnS))Gl7aA3777G.))C S 499 5 5 5 5q > >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 5q : :A1a[[ : :c!eqck!!!fq11A,sa000CCl3!,,,#DIItCyy9999  :q : :A1a[[ : :c!eqck!!!fq88A,saA3777CCl3!tE!HH~~>>>#DIItCyy9999  :q : :A1a[[ : :c!eqck!!!fqAAA,saA3R@@@CCl3!tE!HH~~BGGG#DIItCyy9999  :q : :A1a[[ : :c!eqck!!!fq??A,saA3A>>>CCl3!tE!HH~~1EEE#DIItCyy9999  : : :rctjd}tjd|jDj}t j|d}t ||tjd|D}t j|d}t ||tjd|D}t j|dd }t ||dS) Nrxrc6g|]}tj|Sr9r rr|ros rr}z2TestIntegral.test_chebint_axis..3 888Qa888rrrc6g|]}tj|Sr9rrs rr}z2TestIntegral.test_chebint_axis..7 666Qa666rrc:g|]}tj|dS)rx)rrrs rr}z2TestIntegral.test_chebint_axis..;s';;;!a1---;;;rrx)rr)r(rvstackTr rrr-rr0r1s rtest_chebint_axiszTestIntegral.test_chebint_axis/siv&&i88#%88899;l3Q'''C%%%i66#66677l3Q'''C%%%i;;s;;;<<l3!!,,,C%%%%%rN)r6r7r8rrr9rrrrs6Q: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) Nrr&rrrrr)rr) rrr chebderrr'rrrr)r-r.r0r1rVs r test_chebderzTestDerivative.test_chebderBsisB777j$,R888q / /A#a%1#+C,sa(((C cDII . . . .q : :A1a[[ : :c!eqckl4<q#9#9#9Q???#DIItCyy9999 : q : :A1a[[ : :c!eqckl4<qa#@#@#@A2NNN#DIItCyy9999 : : :rchtjd}tjd|jDj}t j|d}t ||tjd|D}t j|d}t ||dS)Nrc6g|]}tj|Sr9r rrs rr}z4TestDerivative.test_chebder_axis.._rrrrc6g|]}tj|Sr9rrs rr}z4TestDerivative.test_chebder_axis..crrr)r(rrrr rrrs rtest_chebder_axisz TestDerivative.test_chebder_axis[siv&&i88#%88899;l3Q'''C%%%i66#66677l3Q'''C%%%%%rN)r6r7r8rrr9rrrr@s2:::2 & & & & &rrc`eZdZejddzdz ZdZdZdZdS) TestVanderrwrrc8tjd}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) Nrxrrrr.rr)rxrr) r(rmr chebvanderr rr'rrr))r-rvr.coefs rtest_chebvanderzTestVander.test_chebvanderls) 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 Brcn|j\}}}tjd}tj||ddg}tj|||}tj||j}t||tj|g|gddg}t|j dkdS)Nrrr)rrr) rr(rr chebvander2drdotflatrr rr-rrrrovanr0r1s rtest_chebvander2dzTestVander.test_chebvander2d}sV B I  V $ $BA//nRQ''fS!&!!C%%%rdQF33 Y&'''''rcv|j\}}}tjd}tj|||gd}tj||||}tj||j}t||tj|g|g|ggd}t|j dkdS)N)rrxr)rrrx)rr) rr(rr chebvander3drrrrr rrs rtest_chebvander3dzTestVander.test_chebvander3dsV B I  Y ' 'BIII66nRR++fS!&!!C%%%rdRD)))<< Z'(((((rN) r6r7r8r(rrrrrr9rrrrhsc   "Q&ABBB" ( ( ( ) ) ) ) )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 zSNrrr9rs rfz#TestFitting.test_chebfit..fsa!e9a!e$ $rc|dz|dzzdzS)Nrrrr9rs rf2z$TestFitting.test_chebfit..f2sa4!Q$;? "rrrrr)w)rrrrxr)rrrrxr)rrrrxr)rrxrrr)ry?ry)rrr)rrr chebfitrr(rrlenrrr)r zeros_likecopy)r-rrrrcoef3coef4coef2drywwcoef3wcoef2dcoef1coef2s r test_chebfitzTestFitting.test_chebfits % % % # # # j$,aS"===iuqc1===irA3:::isqcUGQ???i1vsA>>>isQFA>>>isQCseDDDDisQCq!fEEEEj$,aS2%@@@j$,aS***EEEisQC<<< K1   AaDD Q1%%SZZ###DLE22A666 Q<<<00SZZ###DLE22A666 Q1%%SZZ###DLE22A666 Q???33SZZ###DLE22A666 Q???33SZZ###DLE22A666a1a&!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%=%=%?@@@   DLAq11Aq6:::DLA1v66A??? KA   BqEE Q1%%DLE22A666 Q999--DLE22A666E5)))))rN)r6r7r8rr9rrrrs(H*H*H*H*H*rrc&eZdZdZdZdZdZdS)TestInterpolatec||dz z|dz zSrr9)r-rs rrzTestInterpolate.fsAE{a!e$$rctttj|jdtt tj|jddS)Nrg$@)rrr chebinterpolaterrr>s r test_raiseszTestInterpolate.test_raisess:j$"6CCCi!5tvsCCCCCrctddD]6}ttj|j|j|dzfk7dS)Nrr)r'r r rrr)r-degs rtest_dimensionszTestInterpolate.test_dimensionssUA;; K KC D(55;azI J J J J K Krc d}tjddd}tddD]^}td|dzD]H}tj|||f}t tj|||||dI_dS)Nc ||zS)Nr9)rps rpowxz0TestInterpolate.test_approximation..powxs a4Krrrr~r )decimal)r(rr'r rrr)r-rrrrros rtest_approximationz"TestInterpolate.test_approximations    KAr " "B<< P PC1cAg&& P P(sQD99#DLA$6$6Q BOOOOO P P PrN)r6r7r8rrrrr9rrrrs[%%%DDDKKK P P P P Prrc eZdZdZdZdZdS) TestCompanionctttjgtttjdgdSrD)rrr chebcompanionr>s rrzTestCompanion.test_raisess4j$"4b999j$"4qc:::::rctddD]8}dg|zdgz}ttj|j||fk9dS)Nrrr)r'r r r r)r-r.rs rrzTestCompanion.test_dimensionss\q! > >A3q5A3;D D&t,,2q!f< = = = = > >rc`ttjddgddkdS)Nrr)rr)r r r r>s rtest_linear_rootzTestCompanion.test_linear_root s/"Aq6**40C788888rN)r6r7r8rrr r9rrrrsA;;;>>> 99999rrceZdZdZdS) TestGaussctjd\}}tj|d}tj|j|z|}dtj|z }|dddf|z|z}t|tj dtj }t| |dS)Ndcr) r chebgaussrr(rrsqrtdiagonalreyepisum)r-rrrvvvdr0s rtest_100zTestGauss.test_100s~c""1 OAr " " VAC!GQ   rwr{{}}%% % 4[2  "Bs ,,,eAEEGGS)))))rN)r6r7r8rr9rrrr s#*****rrcDeZdZdZdZdZdZdZdZdZ dZ d Z d S) TestMiscc tjg}tt|dgt ddD]}t jt jt j dd|zdzddd}dg|zdgz}tj|d|dz zz}tt|t|dS)Nrrrr) r chebfromrootsrrr'r(cosrr)r-r1r.rootsr0s rtest_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 6rcttjdggttjddgdgtddD]h}t jdd|}tjtj|}tt|t|idS)Nrrr rr)rr chebrootsr'r(rrr)r-r.r0r1s rtest_chebrootszTestMisc.test_chebroots+sDNA3//444DNAq622SE:::q! 6 6A+b!Q''C.!3C!8!899C S 499 5 5 5 5 6 6rcBgd}tttj|dt tj||ddt tj|d|ddt tj|ddgdS)N)rrrrrrrrr)rrr r r)r-rs r test_chebtrimzTestMisc.test_chebtrim3s}} j$-r::: T]4(($ss)444T]4++T#2#Y777T]4++aS11111rcPttjddddgdS)Nrxr)rr chebliner>s r test_cheblinezTestMisc.test_chebline>s(T]1a((1a&11111rctdD]7}ttjdg|zdgzt|8dSNr~rr)r'rr rrr-r.s rtest_cheb2polyzTestMisc.test_cheb2polyAsRr G GA s1us{ ; ;U1X F F F F G GrctdD]7}ttjt|dg|zdgz8dSr,)r'rr rrr-s rtest_poly2chebzTestMisc.test_poly2chebEsRr G GA uQx 8 81#a%1#+ F F F F G Grctjddddd}dtjd|ztjd|z zz }tj|}t ||dS)Nrr ry)r(rrr chebweightr)r-rr0r1s r test_weightzTestMisc.test_weightIsj KAr " "1R4 ("'!a%..271q5>>12oa  C%%%%%rctttjdtttjddg}t tjd|ddg}t tjd|gd}t tjd|gd }t tjd |dS) Nrtrrg;fg;f?r)gLXzrgLXz?rx)g(1kgŜ}ؿgŜ}?g(1k?r)rrr chebpts1rr-r0s r test_chebpts1zTestMisc.test_chebpts1Osj$-555j$-333cDM!,,c222#%89DM!,,c222<<<DM!,,c222HHHDM!,,c22222rctttjdtttjdddg}t tjd|gd}t tjd|gd}t tjd|gd }t tjd |dS) Nrtrrr)rrrrx)rr r&rr)ggKfrgKf?ryr)rrr chebpts2rr7s r test_chebpts2zTestMisc.test_chebpts2^sj$-555j$-3331gDM!,,c222jjDM!,,c222DM!,,c222===DM!,,c22222rN) r6r7r8r"r%r'r*r.r0r4r8r;r9rrrr s666666 2 2 2222GGGGGG&&& 3 3 3 3 3 3 3 3rr)(__doc__ functoolsrnumpyr(numpy.polynomial.chebyshev polynomial chebyshevr numpy.polynomial.polynomialr numpy.testingrrrr rT0T1T2T3T4T5T6T7T8T9rr$r;rKrrrrrrrrrrr9rrrNsI)))))))))////// &&&SVZZ]]###)))--- RRRRR0########" ) ) ) ) ) ) ) ):@:@:@:@:@:@:@:@z['['['['['['['['|c&c&c&c&c&c&c&c&L%&%&%&%&%&%&%&%&P-)-)-)-)-)-)-)-)`J*J*J*J*J*J*J*J*ZPPPPPPPP2 9 9 9 9 9 9 9 9********&K3K3K3K3K3K3K3K3K3K3r