K i0ddlZddlZddlmcmZddlmZmZm Z m Z m Z m Z m Z ejdejdzdzZGddZy)N)assert_assert_allcloseassert_almost_equalassert_array_almost_equalassert_array_equal assert_equal assert_raises AllIntegerAllFloatOceZdZdZdZdZejjde de ddfdZ dZ d Z d Zd Zd Zd ZdZdZdZdZdZdZdZdZy)TestPolynomialctjgd}tt|dtt |dtjgd}tt|dtt |dtjgd}tt |dtt tjgd d y) 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 e/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/numpy/lib/tests/test_polynomial.pytest_poly1d_str_and_reprz'TestPolynomial.test_poly1d_str_and_reprs IIj !T!W45SV% & IIj !T!W45SV% & II> ?SV@ A S<01& 'ctjgd}tjgd}t|ddt|ddt|ddt|ddy) NrrrrgC@rgU@rrrrr r!s r#test_poly1d_resolutionz%TestPolynomial.test_poly1d_resolution+sV IIj ! IIj !QqT3QqT4 QqT3QqT4 r%c Ptjgd}tjgd}t||z tjdgtjddgft|jtjgdt|jdtjgdtjgd}tjgd }t||ztjgd t||ztjgd t||z tjgd t|d ztjgdt||tjgdt||tjgdt|j tjddgt|j dtjdgttj tjgdtjddgtjddgtjdgfy)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_math3s IIj ! IIj !QURYYv. 3+0FGHQWWY *= >?QWWQZ+>!?@ IIj ! IIj !QUBII&;<=QUBIIl34QUBIIm45Q!VRYY'WXYQqT299%;<=QqT299%=>?QWWY 2r( 34QWWQZB41RZZ * 5ryy!Q7HIiiS *BIIrdO< >r% type_codereturnNctj|}tjgd|}tj|}t tj ||t tj |j|t t |ddddddd}|jD]R\}}||}t |||tjk(rt|tr;Jt |j|Ty)Nrrr)dtyperrrr)rrrrr) rr;arrayrrasarraylenitemsobject_ isinstanceint) rr7r;arr comparison_dctindexrefscalars r#test_poly1d_misczTestPolynomial.test_poly1d_miscHs# XXiu - IIbM RZZ]B'RZZ]((%0SVQ A!a8(..0 2JE3uXF  % "!&#...V\\51  2r%ctjgdd}tt|dtjgdd}tt|dy)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_arg]sJ IIj3 /SV% & IIj8 4SV/ 0r%c ttjdtjd tjdggdgdgdgdg}ttj|gdt tj tjgdt tj tjgd t tj tjgd t tj tjgd t tj tjgd t tj tjd dgt tj tjddgt tj tjd dgtjjdtjjdd tjjdzz}t tj tjtj|tj|fy)Nrr)rrrr:)r,r'rP)r)riii)yZd;O?yCl@yClyZd;O)?y?@?y? @y? )rSrTrUy?@y?)rSrVrTrU)rSrV@yrSrVrry*d) rrpolysqrtr isrealobj iscomplexobjrandomseedrandn concatenate conjugate)rAas r# test_polyzTestPolynomial.test_polygs!"''1rwwqzk2771:*F"G"0 2 9 -!"''!*.?@  RWW%IJKL RWW&CDE F RWW%OPQR RWW%>?@A RWW%789: RWWb#Y/01 RWWaW-./[(9 :;< r IIOOC 2 (<#< < RWWR^^Q Q4H%IJKLr%c ttjgdddgtjddddD]}tjdd|g}tj tjt j|ddd}t||d ttj|z tjddddD]}tjdd |g}tj tjt j|ddd}t||d ttj|z y) N)rrrr i)numbaserrg)\(?) rrrootslogspacer<sortrZ polyfromrootsrrBlog10)ritgtress r# test_rootszTestPolynomial.test_rootss288I.A7RB7 AA((B1:&C''"((4#5#5c#:4R4#@ABC S"s288A;/?*? @ A RB7 AA((Ba=)C''"((4#5#5c#:4R4#@ABC S"s288A;/?*? @ Ar%ctjgd}d|d<tt|dtjddg}d|d<d|d<tt|dy)N)r,rrrrrrrrz 0rMrr s r#test_str_leading_zerosz%TestPolynomial.test_str_leading_zeross` IIl #!SV% & IIq!f !!SVV$r%c 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|ddtjf}|ddtjf}t|tj||dtj||fd}tj||fd}t|tj||dtj||tj|ddtjfzdd \}}t||dddfdt||dddfdt| |dddddfdt| |dddddfdtjjdtjjd}tjtj|jd|dd \}}t!|j#d d!"t!tj$|j'd d!"tjtj|jd|tj(|jddd#\}}t!|j#d d!"ttj$|j'd tj*|jdd$}tjtj|jd||dd #\}}t!|j#d d!"t!tj$|j'd d!"tjtj|jd||dd#\}}t!|j#d d!"ttj$|j'd%y)&N)rr1rrrrQ)rrrrrrrrRrrg@T)degcov)rz)gz6>W@gH?gCl?r,)decimal)g ? 1w-!?)r|gH}x@B>٬)r}r~g~jt?)wrz)g@g*:HgY8m?)g@&SH}8?)rg:@hs)rrgz6>W[?unscaled)gгY?ŏ1wſ{?)rgv/?ŏ1w-!)rrgǺ?)axis{)r,i')sizeg?g{Gz?)atol)rryrzr1r.)rr<linspacepolyvalaranger ValueErrorpolyfitrnewaxisrar^r_normalzerosshaperstdr[meanonesfull)rcxrJerrweightsmrzestval0m2cov2valm3cov3yyccrrs r# test_polyfitzTestPolynomial.test_polyfits HH\ " KK1a  JJq! &))Aq"%q(3. j"**cA3A4 1Aq3wt43%CA.+++- D#q1::aS!wDAD5r1E***, Cq1::aS!wJGD5r1E***, Cq1 am  am Arzz!Q23 ^^QF + ^^QF +B 1b! 45ArBHHSM!RZZ-$@@!N3C1a4!4C1a4!4D#aAg,:D#aAg,: s II  *  -JJrxx 3QA4H c Cd3 +St<JJrxx 3Q"''!''!*:M#$*6 c Cd3BGGCHHJ/5 GGAGGAJ )JJrxx 3Q!M c Cd3 +St<JJrxx 3Q! S c Cd3BGGCHHJ/6r%cddlm}tj|d|d|dg}||dz}t |d|dk(|j }t |d|d k(|j }t |d |d k(t |d |d k(t tj|jjtjtj|d|d g}ttj|d|d gd|d|d gy)Nr)Decimalz4.0z3.0z2.0z1.333333333333333rz3.9999999999999990z8.0rz1.333333333333333333333333333rz1.5r) r{rrrrr4r3 issubdtypecoeffsr;r@rZr)rrr p2s r# test_objectszTestPolynomial.test_objectss# IIwu~wu~wu~F G ,- -1!5667 WWY1'( WWY1!@AAB1'( biioorzz:; GGWQZ, -RWWgaj'!*56gaj1 3r%ctjgd}|j}t|jgdk(j |j }t|jddgk(j y)N)y@rWrS)rSrSrSry@rW)rrr3rrallr4rr rs r# test_complexzTestPolynomial.test_complexs^ IIl # WWYo-2245 WWYr2h&++-.r%ctjgd}|jdgd}t|jgdk(j y)N)rrrr) rQrP)k)g?gUUUUUU?gUUUUUU?g@rQrP)rrr3rrrrs r#test_integ_coeffsz TestPolynomial.test_integ_coeffssA IIi  WWQ)W $ YYT T Y Y [ ]r%ct tjtjdy#t$rYywxYw)N)rr)rrZrr)rs r#test_zero_dimszTestPolynomial.test_zero_dimss.  GGBHHV$ %   s (+ 77ctjdd}ttj|tjtj|y)z. Regression test for gh-5096. rN)rrrrZdiag)rvs r#test_poly_int_overflowz%TestPolynomial.test_poly_int_overflows7 IIa BGGAJ (;tjgd}|xjdz c_t|jgd|jdxxdz cc<t|jgdt t t |dtjdy) z# Coefficients should be modifiable r:r)rrr,rrg)rrrkrN)rrrrr AttributeErrorsetattrr<rvs r#test_poly_coeffs_mutablez'TestPolynomial.test_poly_coeffs_mutable4se IIi  A QXXy)  r QXXz* ngq(BHHQKHr%)__name__ __module__ __qualname__r$r*r6pytestmark parametrize TYPE_CODESrrHrNrertrwrrrrrrrrrrr%r#rrs',!>* [[[*52#2$262(0M0 A %H7T 3/]  = 3$(" Ir%r)rnumpyrnumpy.polynomial.polynomial polynomialrZ numpy.testingrrrrrrr typecodesrrrr%r#rsQ **\\, '",,z*B BS H lIlIr%