bg1P dZddlmZddlZddlmcmZddlZddl m Z ddl m Z m Z mZmZmZmZmZdZdgZddgZgdZgd Zgd Zgd Zgd Zgd ZgdZgdZeeeeeeeeeeg ZGddZGddZ GddZ!GddZ"GddZ#GddZ$GddZ%GddZ&dS) zTests for polynomial module. )reduceN)deepcopy)assert_almost_equal assert_raises assert_equalassert_ assert_warnsassert_array_equalassert_raises_regexc.tj|dS)Ngư>)tol)polypolytrimxs y/builddir/build/BUILD/cloudlinux-venv-1.0.7/venv/lib64/python3.11/site-packages/numpy/polynomial/tests/test_polynomial.pytrimrs = % % %%)r)rr)rrir)rrir)rrrir )rir8rir@) rrirrir) r riririrc2eZdZdZdZdZdZdZdZdS) TestConstantsc>ttjddgdS)Nrr)rr polydomainselfs rtest_polydomainzTestConstants.test_polydomain"sT_r1g.....rc<ttjdgdS)Nr)rrpolyzeror)s r test_polyzerozTestConstants.test_polyzero%sT]QC(((((rc<ttjdgdSNr)rrpolyoner)s r test_polyonezTestConstants.test_polyone(sT\A3'''''rc>ttjddgdS)Nrr)rrpolyxr)s r test_polyxzTestConstants.test_polyx+sTZ!Q(((((rcptjgd}t|}t||dSNrr)r Polynomialrrr*rys r test_copyzTestConstants.test_copy.s6 OIII & & QKKQrctjgd}tjtj|}t ||dSr7)rr:pickleloadsdumpsrr;s r test_picklezTestConstants.test_pickle3sB OIII & & La ) )QrN) __name__ __module__ __qualname__r+r.r2r5r=rBrrr&r& sn///)))((())) rr&c2eZdZdZdZdZdZdZdZdS)TestArithmeticctdD]}tdD]}d|d|}tjt||dz}||xxdz cc<||xxdz cc<t jdg|zdgzdg|zdgz}t t|t||dSNrAt i=, j=rrerr_msg)rangenpzerosmaxrpolyaddrrr*ijmsgtgtress r test_polyaddzTestArithmetic.test_polyadd:q @ @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||dSrJ)rOrPrQrRrpolysubrrrTs r test_polysubzTestArithmetic.test_polysubDr[rc0ttjdgdgttjdgddgtddD];}dg|zdgz}dg|dzzdgz}ttj||>S#s33333 4 4 4rc rtdD]}tdD]}d|d|}tj|dz}ttj|g|ztjdg}t j||}tt|t||dS)NrrKrLrrM) rOrParangerrrdarraypolypowrr)r*rUrVrWcrXrYs r test_polypowzTestArithmetic.test_polypowtsq @ @A1XX @ @(a((Q((Ia!e$$T\A3q5"(A3--@@l1a((T#YYS 3?????  @ @ @rN) rCrDrErZr^rbrermrsrFrrrHrH8s|@@@@@@222@@@444*@@@@@rrHceZdZejgdZejdeeZejdeeeZej ddzdz Z e j e gdZ dZdZd Zd Zd Zd Zd S)TestEvaluation)?g@g@i,j->ij i,j,k->ijkr9rrrc ttjgdgjdt jdd fdt dD}t dD]7}||}tj dg|zdgz}t||8 dzdz z}tj gd}t||t dD]}dg|z}t j| ttj dgj |ttj ddgj |ttj gd j |gd }tj gd | }t jgd |}t|j |Gddtj}t j gd |} tt!t jgd| |dS)Nrrrcg|]}|zSrFrF.0rUrs r z/TestEvaluation.test_polyval.. $ $ $aQT $ $ $rrr)rrrrr9rrr)FTFr8)mask)rr9ceZdZdS)&TestEvaluation.test_polyval..CN)rCrDrErFrrCrs Drrrr9r)rrpolyvalsizerPlinspacerOrrQshapemarpr rndarrayviewtype) r*r<rUrXrYdimsrmxrcxrs @r test_polyvalzTestEvaluation.test_polyvalsAT\"qc**/333 KA   $ $ $ $588 $ $ $q * *AA$C,q1#a%1#+..C S ) ) ) )All1mmm,,C%%%q A AA3q5DA a!--3T : : : a!Q006 = = = a3394 @ @ @ @$## U[[[ . .jB''38T***         Xiii % %a ( (T"*YYY3344a88888rc tttjdgdgdt tjgdgjdt tjgdgjdkt tjgdgdzgjdt tjgdgdzgjdkt tjdddt tjdtj djd ktj d d fd tdD}tddD]3}||}tj dg|z}t||4 dz z dzz}tj gd }t||td D]}dg|z}tj | t tj dgj|t tj ddgj|t tj gdj|gd}tj|}tj d d ttj |tj |d}tjdd tjdd|}tj |d}tj|jdd}t|jD]*} tj | |dd| f|| <+t ||tj d zg tj |d}tj|jdd jz}t|jdD]L} t jdD]/} tj | |dd| f|| | ddf<0Mt ||dS)NrF)tensorr)rr)rr)r9r9)r9rcg|]}|zSrFrFr|s rr~z8TestEvaluation.test_polyvalfromroots..rr)rrrr9rr)rirryr)rT)r ValueErrorrpolyvalfromrootsrrrrrPonesrrOrrQ polyrootsrrorandomrandintemptyvstack) r*r<rUrXrYrptestrrshapeiijjrs @rtest_polyvalfromrootsz$TestEvaluation.test_polyvalfromrootssY j$"7cA3u . . . . T*2s338!<<<%b1#..4<=== T*2ay99>BBB%bA37)44:fDEEE T*1a00!444%a99?4GHHH KA   $ $ $ $588 $ $ $q! * *AA$C'A3q511C S ) ) ) )QiQ#Azzz22C%%%q J JA3q5DA .q1#66>>@%%%$%%%$ ' ' ' ' ' ' ' 'rruceZdZdZdZdS) TestIntegralc  tttjdgdtttjdgdtttjdgdddgtttjdgdgtttjdgdgtttjdgdt t 5tjddgddddn #1swxYwYtd d D]9}dg|d z zdgz}tjdg|| }t|ddg:td D]d}|dz}dg|zdgz}|gdg|zzd|z gz}tj|d|g }tt|t|etd D]M}|dz}dg|zdgz}tj|d|gd }ttj d||Ntd D]e}|dz}dg|zdgz}|gdg|zzd |z gz}tj|d|gd }tt|t|ftd D]}td d D]~}dg|zdgz}|dd}t|D]}tj|d}tj||}tt|t|td D]}td d D]}dg|zdgz}|dd}t|D]}tj|d|g }tj||tt| }tt|t|td D]}td d D]}dg|zdgz}|dd}t|D]}tj|d|gd }tj||tt|d }tt|t|td D]}td d D]}dg|zdgz}|dd}t|D]}tj|d|gd }tj||tt|d }tt|t|dS)Nr?rr)lbnd)sclaxisrvrr)mk)rrr)rrrr) r TypeErrorrpolyintrr DeprecationWarningrOrrrlist)r*rUrrYrpolrXrVs r test_polyintzTestIntegral.test_polyint/sisB777j$,R888j$,QA???j$,1#>>>>j$,!====is<<<< , - - % % L!Q $ $ $ % % % % % % % % % % % % % % %q! - -AQU qc!A,sa1---C aV , , , ,q 6 6Aa%C#a%1#+C#A+3'C,saA3///C S 499 5 5 5 5q : :Aa%C#a%1#+C,saA3R888C  R 5 5q 9 9 9 9q 6 6Aa%C#a%1#+C#A+3'C,saA3A666C 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  : : :s)D  DDctjd}tjd|jDj}t j|d}t ||tjd|D}t j|d}t ||tjd|D}t j|dd }t ||dS) Nr9rc6g|]}tj|SrFrrr}rrs rr~z2TestIntegral.test_polyint_axis.. 888Qa888rrrc6g|]}tj|SrFrrs rr~z2TestIntegral.test_polyint_axis.. 666Qa666rrc:g|]}tj|dS)r9)rrrs rr~z2TestIntegral.test_polyint_axis..s';;;!a1---;;;rr9)rr)rPrrTrrrr*rrXrYs rtest_polyint_axiszTestIntegral.test_polyint_axissiv&&i88#%88899;l3Q'''C%%%i66#66677l3Q'''C%%%i;;s;;;<<l3!!,,,C%%%%%rN)rCrDrErrrFrrrr-s6N:N:N:`&&&&&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) Nrrrrrrr)rr) rrrpolyderrrOrrrr)r*rUrXrYrVs r test_polyderzTestDerivative.test_polydersisB777j$,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|SrFrrrs rr~z4TestDerivative.test_polyder_axis..rrrrc6g|]}tj|SrFrrs rr~z4TestDerivative.test_polyder_axis..rrr)rPrrrrrrrs rtest_polyder_axisz TestDerivative.test_polyder_axissiv&&i88#%88899;l3Q'''C%%%i66#66677l3Q'''C%%%%%rN)rCrDrErrrFrrrrs2:::2 & & & & &rrcfeZdZejddzdz ZdZdZdZdZ dS) TestVanderryrrc8tjd}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) Nr9rrrr.rr)r9rr) rPror polyvanderrrrOrrrp)r*rvrUcoefs rtest_polyvanderzTestVander.test_polyvanders) 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) rrPrr polyvander2drdotflatrrrr*rrrrrvanrXrYs rtest_polyvander2dzTestVander.test_polyvander2dsV 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)Nrr8)rr) rrPrr polyvander3drrrrrrrs rtest_polyvander3dzTestVander.test_polyvander3dsV B I  Y ' 'BIII66nRR++fS!&!!C%%%rdRD)))<< Z'(((((rcptjd}tttj|ddS)Nr9r)rProrrrr)r*rs rtest_polyvandernegdegz TestVander.test_polyvandernegdegs+ IaLLj$/1b99999rN) rCrDrErPrrrrrrrFrrrrsr   "Q&ABBB" ( ( ( ) ) ):::::rrc eZdZdZdZdZdS) TestCompanionctttjgtttjdgdSr0)rrr polycompanionr)s r test_raiseszTestCompanion.test_raisess4j$"4b999j$"4qc:::::rctddD]8}dg|zdgz}ttj|j||fk9dS)Nrrr)rOrrrr)r*rUrs rtest_dimensionszTestCompanion.test_dimensionss\q! > >A3q5A3;D D&t,,2q!f< = = = = > >rc`ttjddgddkdS)Nrr)rr)rrrr)s rtest_linear_rootzTestCompanion.test_linear_roots/"Aq6**40C788888rN)rCrDrErrrrFrrrrsA;;;>>> 99999rrc2eZdZdZdZdZdZdZdZdS)TestMiscc tjg}tt|dgt ddD]}t jt jt j dd|zdzddd}t|}tj|d|dz zz}tt|t|dS)Nrrrr) r polyfromrootsrrrOrPcosrpiTlist)r*rYrUrootsrXs rtest_polyfromrootszTestMisc.test_polyfromrootss $$DIIs+++q! 6 6AF2;vq!A#'::14a4@AAE(C$U++A!H4C S 499 5 5 5 5  6 6rcttjdggttjddgdgtddD]h}t jdd|}tjtj|}tt|t|idS)Nrrrrr)rrrrOrPrr r)r*rUrXrYs rtest_polyrootszTestMisc.test_polyrootssDNA3//444DNAq622SE:::q! 6 6A+b!Q''C.!3C!8!899C S 499 5 5 5 5 6 6rc  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|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 zS)NrrrFrs rfz TestMisc.test_polyfit..fsa!e9a!e$ $rc|dz|dzzdzS)NrrrrFrs rf2z!TestMisc.test_polyfit..f2sa4!Q$;? "rrrrr)w)rrrr9r)rrrr9r)rrrr9r)ry?ry)rrr)rrrpolyfitrrPrrlenrrrpr zeros_likecopy)r*rrrr<coef3coef4coef2drywwcoef3wcoef2dcoef1coef2s r test_polyfitzTestMisc.test_polyfitsQ % % % # # # 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###DLE22A666a1a&!1!1!3Q77FBHeU^$<$<$>???a1a&!1!1!3\\\BBFBHeU^$<$<$>??? M!   VVXX!$Q$14a4aQ!,,,FE***a\\\Q777FE***,q"(B8"4"4"6Q???GRXuen%=%=%?@@@,q"(B8"4"4"6 JJJGRXuen%=%=%?@@@   DLAq11Aq6:::DLA1v66A??? KA   BqEE Q1%%DLE22A666 Q999--DLE22A666E5)))))rcBgd}tttj|dt tj||ddt tj|d|ddt tj|ddgdS)N)rrrrrrrrr)rrrrr)r*rs r test_polytrimzTestMisc.test_polytrimTs}} j$-r::: T]4(($ss)444T]4++T#2#Y777T]4++aS11111rcPttjddddgdS)Nr9rrrpolyliner)s r test_polylinezTestMisc.test_polyline_s(T]1a((1a&11111rcNttjdddgdS)Nr9rr)r)s rtest_polyline_zerozTestMisc.test_polyline_zerobs&T]1a((1#.....rN) rCrDrErrr%r'r+r-rFrrr r sr666666D*D*D*L 2 2 2222/////rr )'__doc__ functoolsrnumpyrPnumpy.polynomial.polynomial polynomialrr?rr numpy.testingrrrrr r r rT0T1T2T3T4T5T6T7T8T9rr&rHrurrrrr rFrrr>s********* ;;;;;;;;;;;;;;;;;; &&&SVZZ]]###)))--- RRRRR00C@C@C@C@C@C@C@C@Ll'l'l'l'l'l'l'l'^`&`&`&`&`&`&`&`&F%&%&%&%&%&%&%&%&P1:1:1:1:1:1:1:1:h 9 9 9 9 9 9 9 9h/h/h/h/h/h/h/h/h/h/r