id,ddlZddlmZmZmZmZmZmZm Z ddl Z ej dej dzdzZ GddZ dS)N)assert_ assert_equalassert_array_equalassert_almost_equalassert_array_almost_equal assert_raisesassert_allclose AllIntegerAllFloatOceZdZdZdZdZejde de ddfdZ dZ d Z d Zd Zd Zd ZdZdZdZdZdZdZdZdZdS)TestPolynomialctjgd}tt|dtt |dtjgd}tt|dtt |dtjgd}tt |dtt tjgd d dS) N?zpoly1d([1., 2., 3.])z 2 1 x + 2 x + 3@rzpoly1d([3., 2., 1.]) 2 3 x + 2 x + 1)y [f?@yg"GBtk~y@?z= 3 2 (1.9 + 2j) x - 3j x - 5.123 x + (2 + 1j))z 2 -3 x - 2 x - 1)nppoly1drreprstr)selfpqrs T/opt/cloudlinux/venv/lib/python3.11/site-packages/numpy/lib/tests/test_polynomial.pytest_poly1d_str_and_reprz'TestPolynomial.test_poly1d_str_and_reprs  Ijjj ! !T!WW4555SVV% & & & Ijjj ! !T!WW4555SVV% & & & I>>> ? ?SVV@ A A A S<<<0011& ' ' ' ' 'c&tjgd}tjgd}t|ddt|ddt|ddt|dddS) NrrrrgC@rgU@rrrrr r!s r#test_poly1d_resolutionz%TestPolynomial.test_poly1d_resolution%s Ijjj ! ! Ijjj ! !QQqTT3QQqTT4   QQqTT3QQqTT4     r%c tjgd}tjgd}t||z tjdgtjddgft|tjgdt|dtjgdtjgd}tjgd }t||ztjgd t||ztjgd t||z tjgd t|d ztjgdt||tjgdt||tjgdt|tjddgt|dtjdgttjtjgdtjddgtjddgtjdgfdS)N)rr)@rr?g?g@)gUUUUUU?rr-rrr)r @g,@r0r)r-r-r-)gr/@r,) rr0gB@gZ@gj@gs@g@t@gk@g@T@)g"@(@g0@r0g@)rr2g@@gD@gA@r1r)rrrrgr/)rrrintegderivpolydivr)s r#test_poly1d_mathzTestPolynomial.test_poly1d_math-se Ijjj ! ! Ijjj ! !QqS29dV,,bid .D.DEFFFQWWYY *;*;*; < <===QWWQZZ+<+<+>> Ijjj ! ! Ijjj ! !QUBI&;&;&;<<===QUBIlll33444QUBImmm44555Q!VRY'W'W'WXXYYYQQqTT29%;%;%;<<===QQqTT29%=%=%=>>???QWWYY 2r( 3 3444QWWQZZB4111RZ *** 5 5ry!Q7H7HIIiS **BIrdOO< > > > > >r% type_codereturnNctj|}tjgd|}tj|}t tj||t tj|j|t t |ddddddd}|D]C\}}||}t |||tjkr.t |j|DdS)Nrrr)dtyperrrr)rrrrr) rr;arrayrrasarraylenitemsobject_) rr7r;arr comparison_dctindexrefscalars r#test_poly1d_misczTestPolynomial.test_poly1d_miscBs## Xiiiu - - - IbMM RZ]]B'''RZ]](%000SVVQ A!a88(..00 2 2JE3uXF  % % % """V\51111  2 2r%ctjgdd}tt|dtjgdd}tt|ddS)Nry)variablez 2 1 y + 2 y + 3lambdaz! 2 1 lambda + 2 lambda + 3rrrr)rr!s r#test_poly1d_variable_argz'TestPolynomial.test_poly1d_variable_argWs| Ijjj3 / / /SVV% & & & Ijjj8 4 4 4SVV/ 0 0 0 0 0r%c ttjdtjd tjdggdgdgdgdg}ttj|gdt tjtjgdt tjtjgd t tjtjgd t tjtjgd t tjtjgd t tjtjd dgt tjtjddgt tjtjd dgtjdtj dd tj dzz}t tjtjtj |tj |fdS)Nrr)rrrr:)r,r'rN)r)riii)yZd;O?yCl@yClyZd;O)?y?@?y? @y? )rQrRrSy?@y?)rQrTrRrS)rQrT@yrQrTrry*d) rrpolysqrtr isrealobj iscomplexobjrandomseedrandn concatenate conjugate)rAas r# test_polyzTestPolynomial.test_polyasP!"'1rwqzzk271::*F"G"G"0.. 2 2 2YY 999 -!"'!**.?.?.?@@@  RW%I%I%IJJKKLLL RW&=&=&=>>?? @ @ @ RW%G%G%GHHIIJJJ RW%:%:%:;;<<=== RW%7%7%78899::: RWb#Y//00111 RWaW--..///[(9 : :;;<<< r IOOC 2biooc&:&:#: : RWR^Q Q4H%I%IJJKKLLLLLr%cRttjgdddgdS)N)rrrr)rrrootsrs r# test_rootszTestPolynomial.test_rootsys*28III..A77777r%ctjgd}d|d<tt|dtjddg}d|d<d|d<tt|ddS)N)r,rrrrrrrrz 0rKrr s r#test_str_leading_zerosz%TestPolynomial.test_str_leading_zeros|s Illl # #!SVV% & & & Iq!f  !!SVVV$$$$$r%c8 tjgd}tjddd}tj||}gd}tjddddzd z }t t tjdgdgdd tj|||zdd \}}gd }t||dgdgdgdg} t| |dtj|||zd|d \} } tgd| dgdgdgdg} t| | dtj|||zd|d\} }tgd| dgdgdgdg} t| |d|ddtj f}|ddtj f}t|tj||dtj ||fd}tj ||fd}t|tj||dtj||tj|ddtj fzdd \}}t||dddfdt||dddfdt| |dddddfdt| |dddddfdtj dtj d}tjtj|jd|dd \}}t!|d d!"t!tj|d d!"tjtj|jd|tj|jddd#\}}t!|d d!"ttj|d tj|jdd$}tjtj|jd||dd #\}}t!|d d!"t!tj|d d!"tjtj|jd||dd#\}}t!|d d!"ttj|d%dS)&N)rr1rrrrO)rrrrrrrrPrrg@T)degcov)rm)gz6>W@gH?gCl?r,)decimal)g ? 1w-!?)rogH}x@B>٬)rprqg~jt?)wrm)g@g*:HgY8m?)g@&SH}8?)rsg:@hs)rtrugz6>W[?unscaled)gгY?ŏ1wſ{?)rwgv/?ŏ1w-!)rxrygǺ?)axis{)r,i')sizeg?g{Gz?)atol)rrrlrmr1r.)rr<linspacepolyvalaranger ValueErrorpolyfitrnewaxisr_r\r]normalzerosshaper stdrYmeanonesfull)rcxrHerrweightsmrmestval0m2cov2valm3cov3yyccrrrs r# test_polyfitzTestPolynomial.test_polyfitst H\\\ " " K1a  Jq!  &&&)Aq"%%q(, j"*cA3A4 1 1 1 1AquaT2223%%%CA....+++++++++- D#q1111:a3W$???D555r1EEEE*********, Cq1111:a3W*EEED555r1EEEE*********, Cq1111 aaam  aaam Arz!Q22333 ^QF + + + ^QF + + +B 1b! 4 4555ArBHSMM!!!RZ-$@@!NNN3C111a4!4444C111a4!4444D#aaaAg,::::D#aaaAg,:::: s I  *  - -Jrx 33QA4HHH c Cd3333 ++St<<<<Jrx 33Q"'!'!*:M:M#$*666 c Cd3333BGCHHJJ//555 GAGAJ ' 'Jrx 33Q!MMM c Cd3333 ++St<<<<Jrx 33Q! SSS c Cd3333BGCHHJJ//66666r%cDddlm}tj|d|d|dg}||dz}t |d|dk|}t |d|d k|}t |d |d kt |d |d kt tj|jj tj tj |d|d g}ttj |d|d gd|d|d gdS)Nr)Decimalz4.0z3.0z2.0z1.333333333333333rz3.9999999999999990z8.0rz1.333333333333333333333333333rz1.5r) rnrrrrr4r3 issubdtypecoeffsr;r@rXr)rrr p2s r# test_objectszTestPolynomial.test_objectss###### Iwwu~~wwu~~wwu~~F G G ,-- -1!5666777 WWYY1'((( WWYY1!@AAABBB1'((( biorz::;;; GWWQZZ, - -RWggajj''!**566ggajj1 3 3 3 3 3r%c2tjgd}|}t|jgdk|}t|jddgkdS)N)y@rUrQ)rQrQrQry@rU)rrr3rrallr4rr rs r# test_complexzTestPolynomial.test_complexs Illl # # WWYYooo-2244555 WWYYr2h&++--.....r%ctjgd}|dgd}t|jgdkdS)N)rrrr) rOrN)k)g?gUUUUUU?gUUUUUU?g@rOrN)rrr3rrrrs r#test_integ_coeffsz TestPolynomial.test_integ_coeffssl Iiii  WWQ)))W $ $ YDDD D I I K K M M M M Mr%cv tjtjddS#t$rYdSwxYw)N)rr)rrXrrrfs r#test_zero_dimszTestPolynomial.test_zero_dimssI  GBHV$$ % % % % %    DD s &* 88ctjdd}ttj|tjtj|dS)z. Regression test for gh-5096. rN)rrrrXdiag)rvs r#test_poly_int_overflowz%TestPolynomial.test_poly_int_overflowsE Ia  BGAJJ (;(;<<<<>>* [[*552#2$222652(000MMM0888 % % %H7H7H7T 3 3 3///MMM  === 3 3 3$$$(((" I I I I Ir%r)numpyr numpy.testingrrrrrrr r typecodesrrrr%r#rs  \, '",z*B BS H aIaIaIaIaIaIaIaIaIaIr%