QՖ^. d dl Zd dlmZmZmZmZmZmZm Z d dl
Z
ej d ej d z dz Z G d d Z
dS ) N)assert_ assert_equalassert_array_equalassert_almost_equalassert_array_almost_equal
assert_raisesassert_allclose
AllIntegerAllFloatOc e Zd Zd Zd Zd Zej de de
ddfd Z d Z d Z
d
Zd Zd Zd
Zd Zd Zd Zd Zd Zd Zd Zd ZdS )TestPolynomialc t j g d }t t | d t t | d t j g d }t t | d t t | d t j g d }t t | d t t t j g d d
d S ) N ? zpoly1d([1., 2., 3.])z 2
1 x + 2 x + 3 @r zpoly1d([3., 2., 1.]) 2
3 x + 2 x + 1)y [f? @y g"GBtk~y @ ?z= 3 2
(1.9 + 2j) x - 3j x - 5.123 x + (2 + 1j))z 2
-3 x - 2 x - 1)nppoly1dr reprstr)selfpqrs s/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/numpy/lib/tests/test_polynomial.pytest_poly1d_str_and_reprz'TestPolynomial.test_poly1d_str_and_repr s Ijjj ! !T!WW 4555SVV% & & &
Ijjj ! !T!WW 4555SVV% & & &
I>>> ? ?SVV@ A A A S<<<0011& ' ' ' ' ' c & t j g d }t j g d }t |d d t |d d t |d d t |d d d S ) Nr r r r g C@r g U@r r r r r r! s r# test_poly1d_resolutionz%TestPolynomial.test_poly1d_resolution% s Ijjj ! ! Ijjj ! !QQqTT3 QQqTT4 QQqTT3 QQqTT4 r% c t j g d }t j g d }t ||z t j dg t j ddg f t | t j g d t | d t j g d t j g d }t j g d }t ||z t j g d
t ||z t j g d t ||z
t j g d t |d
z t j g d t || t j g d t || t j g d t | t j ddg t | d t j dg t t j t j g d t j ddg t j ddg t j dg f d S )N)r r ) @r r ?g ?g @)gUUUUUU?r r- r r r )r @g ,@r0 r )r- r- r- )g r/ @r, ) r r0 g B@g Z@g j@g s@g @t@g k@g @T@)g "@ (@g 0@r0 g @)r r2 g @@g D@g A@r1 r )r r r r g r/ )r r r integderivpolydivr) s r# test_poly1d_mathz TestPolynomial.test_poly1d_math- se Ijjj ! ! Ijjj ! !QqS29dV,,bid .D.DEFFFQWWYY *;*;*; < <===QWWQZZ+<+<+>> Ijjj ! ! Ijjj ! !QUBI&;&;&; < <===QUBIlll 3 3444QUBImmm 4 4555Q!VRY'W'W'W X XYYYQQqTT29%;%;%;<<===QQqTT29%=%=%=>>???QWWYY 2r( 3 3444QWWQZZB4 111RZ *** 5 5ry!Q7H7HIIiS **BIrdOO< > > > > >r% type_codereturnNc @ t j | }t j g d| }t j | }t t j | | t t j | j | t t
| d dddddd}| D ]Z\ }}|| }t || |t j k rt |t sJ Et |j | [d S )Nr r r )dtyper r r r )r r r r r ) r r; arrayr r asarraylenitemsobject_
isinstanceint) r r7 r; arr comparison_dctindexrefscalars r# test_poly1d_miscz TestPolynomial.test_poly1d_miscB s ##
Xiiiu
-
-
- IbMM RZ]]B'''RZ]](%000SVVQ A!a88(..00 2 2JE3uXF % % %
""!&#...... V\51111
2 2r% c t j g dd }t t | d t j g dd }t t | d d S )Nr y)variablez 2
1 y + 2 y + 3lambdaz! 2
1 lambda + 2 lambda + 3r r r r )r r! s r# test_poly1d_variable_argz'TestPolynomial.test_poly1d_variable_argW s| Ijjj3 / / /SVV% & & &
Ijjj8 4 4 4SVV/ 0 0 0 0 0r% c t t j dt j d t j d g g d g dg dg dg}t t j | g d t t j t j g d t t j t j g d t t j t j g d
t t j t j g d t t j t j g d t t j t j d
dg t t j t j ddg t t j t j d
dg t j d t j d d
t j d z z }t t j t j t j
|t j | f d S )Nr r )r r r r: )r, r' rP ) r )r iii)y Zd;O?y Cl@y Cly Zd;O) ?y ? @ ? y ? @y ? )rS rT rU y ? @y ? )rS rV rT rU )rS rV @y rS rV r r y * d ) r r polysqrtr isrealobj iscomplexobjrandomseedrandn concatenate conjugate)r Aas r# test_polyzTestPolynomial.test_polya sP !"'1rwqzzk271::*F"G"G"0.. 2 2 2 YY 999 -!"'!**.?.?.?@@@ RW%I%I%I J JKKLLL RW &= &= &= > > ? ? @ @ @ RW%G%G%G H HIIJJJ RW%:%:%: ; ;<<=== RW%7%7%7 8 899::: RWb#Y / /00111 RWaW - -..///[(9 : :;;<<<
r IOOC 2biooc&:&:#: : RWR^Q Q4H%I%I J JKKLLLLLr% c R t t j g d ddg d S )N)r r r r )r r rootsr s r#
test_rootszTestPolynomial.test_rootsy s* 28III..A77777r% c t j g d }d|d< t t | d t j ddg }d|d< d|d< t t | d d S )N)r, r r r r r r r r z
0rM r r s r# test_str_leading_zerosz%TestPolynomial.test_str_leading_zeros| s Illl # #!SVV% & & &
Iq!f !!SVVV$$$$$r% c 8 t j g d }t j ddd }t j || }g d}t j ddd dz d z }t t t j dgdgdd
t j |||z dd
\ }}g d
}t ||d g dg dg dg} t | |d t j |||z d|d
\ }
} t g d|
d g dg dg dg} t | | d t j |||z d|d \ }
}t g d|
d g dg dg dg} t | |d |d d t j f }|d d t j f }t |t j ||d t j
||fd }t j
||fd }t |t j ||d t j ||t j | d d t j f z dd
\ }}t ||d d df d t ||d d df d t | |d d d d df d t | |d d d d df d t j d t j
d }t j t j |j d |dd
\ }}t! | d d!" t! t j | d d!" t j t j |j d |t j |j d dd# \ }}t! | d d!" t t j | d t j |j d d$ }t j t j |j d ||dd
# \ }}t! | d d!" t! t j | d d!" t j t j |j d ||dd# \ }}t! | d d!" t t j | d% d S )&N)r r1 r r r rQ )r r r r r r r rR r r g @T)degcov)ro )gz6>W@g H?gCl?r, )decimal)g ? 1w-! ?)rq gH} x@B>٬ )rr rs g~jt?)wro )g @g*:HgY8m?)g@&SH} 8?)ru g:@hs)rv rw gz6>W[?unscaled)gгY?ŏ1wſ{?)ry gv/?ŏ1w-!)rz r{ gǺ?)axis{ )r, i' )sizeg ?g{Gz?)atol)rt rn ro r1 r. )r r< linspacepolyvalaranger
ValueErrorpolyfitr newaxisra r^ r_ normalzerosshaper stdr[ meanonesfull)r cxrJ errweightsmro estval0m2cov2valm3cov3yyccr rt s r# test_polyfitzTestPolynomial.test_polyfit st H\\\ " " K1a Jq! &&&)Aq"%%q(,
j"*cA3A4 1 1 1 1 AquaT2223%%%CA....+++++++++- D#q1111:a3W$???D 5 5 5r1EEEE*********, Cq1111:a3W*EEED 5 5 5r1EEEE*********, Cq1111
aaam
aaam Arz!Q 2 2333
^QF
+
+
+
^QF
+
+
+B
1b! 4 4555ArBHSMM!!!RZ-$@ @!NNN3C111a4!4444C111a4!4444D#aaaAg,::::D#aaaAg,:::: s I * - -Jrx
3 3QA4HHH c
Cd3333
++St<<<<Jrx
3 3Q"'!'!*:M:M#$*6 6 6 c
Cd3333BGCHHJJ / /555 GAGAJ ' 'Jrx
3 3Q!MMM c
Cd3333
++St<<<< Jrx
3 3Q!
SSS c
Cd3333BGCHHJJ / /66666r% c D ddl m} t j |d |d |d g }| |d z }t |d |d k | }t |d |d k | }t |d
|d k t |d |d
k t t j |j j t j
t j |d |d g }t t j |d |d g d |d |d g d S )Nr )Decimalz4.0z3.0z2.0z1.333333333333333r z3.9999999999999990z8.0r z 1.333333333333333333333333333r z1.5r )
rp r r r r r4 r3
issubdtypecoeffsr; r@ rZ r )r r r p2s r# test_objectszTestPolynomial.test_objects s ###### Iwwu~~wwu~~wwu~~F G G
,--
-1!5666777
WWYY1 '(((
WWYY1!@AAABBB1 '(((
biorz::;;; GWWQZZ, - -RWggajj''!** 566ggajj1 3 3 3 3 3r% c 2 t j g d }| }t |j g dk | }t |j ddgk d S )N)y @rW rS )rS rS rS r y @rW )r r r3 r r allr4 r r r s r# test_complexzTestPolynomial.test_complex s Illl # #
WWYYooo-2244555
WWYYr2h&++--.....r% c t j g d }| dg d }t |j g dk d S )N)r r r r ) rQ rP )k)g?gUUUUUU?gUUUUUU?g @rQ rP )r r r3 r r r r s r# test_integ_coeffsz TestPolynomial.test_integ_coeffs sl Iiii
WWQ)))W
$
$
YDDD
D I I K K M M M M Mr% c v t j t j d d S # t $ r Y d S w xY w)N)r r )r rZ r r rh s r# test_zero_dimsz TestPolynomial.test_zero_dims sI GBHV$$ % % % % % DD s &*
88c t j dd }t t j | t j t j | dS )z.
Regression test for gh-5096.
r N)r r r rZ diag)r vs r# test_poly_int_overflowz%TestPolynomial.test_poly_int_overflow sE
Ia BGAJJ
(;(;<<<< > >* [ [*552# 2$ 2 2 2 652(0 0 0M M M08 8 8
%
%
%H7 H7 H7T
3
3
3 / / /M M M = = = 3 3 3 $ $ $( ( (" I I I I Ir% r )numpyr
numpy.testingr r r r r r r r typecodesr r r r% r# r s
\,
'",z*B
BS
H
aI aI aI aI aI aI aI aI aI aIr%