U
|
P ±d²- ã @ s^ d dl Zd dlmZmZmZmZmZmZm Z d dl
|
Z
|
ej d ej d d Z G dd dZ dS )é N)Úassert_Ú assert_equalÚassert_array_equalÚassert_almost_equalÚassert_array_almost_equalÚ assert_raisesÚassert_allcloseZ
|
AllIntegerZAllFloatÚOc @ s² e Zd Zdd Zdd Zdd Zej de ¡e
|
dd d
|
d Z d d Z dd Z dd Zdd Zdd Zdd Zdd Zdd Zd d Zd d Zd d! Zd"d# Zd$d% Zd&d' ZdS )(ÚTestPolynomialc C s t dddg¡}tt|d tt|d t dddg¡}tt|d tt|d t d
|
d d d g¡}tt|d ttt dddg¡d d S )Nç ð?é é zpoly1d([1., 2., 3.])z 2
|
1 x + 2 x + 3ç @é zpoly1d([3., 2., 1.])ú 2
|
3 x + 2 x + 1yõ¡ ê[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)ÚnpÚpoly1dr ÚreprÚstr)ÚselfÚpÚqÚr© r úVd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/lib/tests/test_polynomial.pyÚtest_poly1d_str_and_repr s" ÿÿÿÿz'TestPolynomial.test_poly1d_str_and_reprc C s\ t dddg¡}t dddg¡}t|dd t|dd t|dd t|dd d S )
|
Nr r r r r r é g C@g U@©r r r ©r r r r r r Útest_poly1d_resolution% s z%TestPolynomial.test_poly1d_resolutionc C sÄ t dddg¡}t dddg¡}t|| t dg¡t ddg¡f t| ¡ t d ddd
|
g¡ t| d¡t d ddd
|
g¡ t ddd g¡}t d ddg¡}t|| t d d dd d g¡ t|| t dddg¡ t|| t dd
|
dg¡ t|d t dd dddddddg ¡ t||t dddd dg¡ t||t d dd d d g¡ t| ¡ t ddg¡ t| d¡t dg¡ tt t dd d g¡t ddg¡¡t dd!g¡t d
|
g¡f d S )"Nr r é g @r ç Ð?g ø?g @gUUUUUUÕ?g r r g @g ,@g Àç @g B@g Z@g Àj@g s@g @t@g k@g @T@g "@g (@g 0@g @g @@g D@g A@r r g ð¿)r r r ÚintegÚderivÚpolydivr! r r r Útest_poly1d_math- s$ $ & "ÿz TestPolynomial.test_poly1d_mathÚ type_codeN)r* Úreturnc C s¶ t |¡}t jdddg|d}t |¡}tt |¡| tt |¡j| tt|d dddddd}| ¡ D ]@\}}|| }t|| |t jkr¤t |t
|
s°t qpt|j| qpd S )Nr r r )Údtyper )r r r r r ) r r, Úarrayr r ZasarrayÚlenÚitemsÚobject_Ú
|
isinstanceÚintÚAssertionError) r r* r, Úarr Zcomparison_dctÚindexÚrefZscalarr r r Útest_poly1d_miscB s
|
|
|
|
z TestPolynomial.test_poly1d_miscc C sH t jdddgdd}tt|d t jdddgdd}tt|d d S ) Nr r r Úy)Úvariablez 2
|
1 y + 2 y + 3Úlambdaz! 2
|
1 lambda + 2 lambda + 3©r r r r )r r r r r Útest_poly1d_variable_argW s ÿÿz'TestPolynomial.test_poly1d_variable_argc C s t t dt d¡ t d¡g¡ddddg dddgdddgd d
|
d gg}t t |¡dd d dg tt t ddddg¡¡ tt t ddddddg¡¡ tt t ddddddg¡¡ tt t ddddg¡¡ tt t ddd d g¡¡ tt t ddg¡¡ tt t dd g¡¡ tt t dd g¡¡ tj d ¡ tj d!¡dtj d!¡ }tt t t
|
|t |¡f¡¡¡ d S )"Nr r r r r é r# r é é r iúÿÿÿi¸ÿÿÿiåÿÿÿy Zd;ßOñ?y Clç@y ClçÀy Zd;ßOñ¿ù ð?y ð¿ù ð? @y ð? Ày ð? @y ð? Àù ð¿y ð? @y ð? Àù @y Àr y ò× ð¿é* éd ) r r ÚpolyÚsqrtr Z isrealobjZ iscomplexobjÚrandomÚseedZrandnÚ concatenateÚ conjugate)r ÚAÚar r r Ú test_polya s(
|
ÿ ÿ
|
zTestPolynomial.test_polyc C s t t dddg¡ddg d S )Nr r )r r Úroots©r r r r Ú
|
test_rootsy s zTestPolynomial.test_rootsc C sX t ddddg¡}d|d< tt|d t ddg¡}d|d< d|d< tt|d d S )Nr# r r r r r z
|
0r; ©r r r r r Útest_str_leading_zeros| s ÿz%TestPolynomial.test_str_leading_zerosc C sø t dddg¡}t ddd¡}t ||¡}dddddddg}t d dd¡d d
|
}ttt jdgdgdd d t j||| dd d \}}dddg}t||dd dddgdddgdddgg} t| |dd t j||| d|d d\}
|
} tddd g|
|
dd d d d gd d d!gd d!d"gg} t| | dd t j||| d|d#d\} }tddd g| dd d$d%d&gd%d'd(gd&d(d)gg} t| |dd |d d
t j f }|d d
t j f }t|t ||d¡ t j
|
||fdd*}t j
|
||fdd*}t|t ||d¡ t j||t |¡d d
t j f dd d \}}t||d d
df dd t||d d
df dd t| |d d
d d
df dd t| |d d
d d
df dd t j d+¡ t j j d,d-}t jt |jd ¡|dd d \}}t| ¡ d.d/d0 tt | ¡ ¡d.d/d0 t jt |jd ¡|t |jd ¡dd#d1\}}t| ¡ d.d/d0 tt | ¡ ¡d. t |jd d¡}t jt |jd ¡||dd d1\}}t| ¡ d.d/d0 tt | ¡ ¡d.d/d0 t jt |jd ¡||dd#d1\}}t| ¡ d.d/d0 tt | ¡ ¡d2 d S )3Nr r% r r r r> r r r? g @T)ÚdegÚcov)rU gz6«>WÛ@gæ §èHÒ?gClçù?r# )Údecimalgßà ©÷?gßà ©Àg1w-! ê?gH¿} x@gB>èÙ¬ú Àgú~j¼tò?)ÚwrU gäò @g*:ËHð¿gY8ÖÅmü?g °áé@g¾Á&SÀgH¿} 8ß?g»:@gh³êsµí¿gz6«>W[Õ?ZunscaledgгYõ¹ÚÂ?gÅ1wÅ¿gÅþ²{ò°?gÉv¾/Í?gÅ1w-!¿gǺ¸ð?)Zaxisé{ )r# i' )Úsizeg à?g{®Gáz?)Zatol)rW rT rU r$ )r r- ZlinspaceZpolyvalÚaranger Ú
|
ValueErrorZpolyfitr ZnewaxisrJ rH rI ÚnormalÚzerosÚshaper ZstdrG ÚmeanZonesÚfull)r ÚcÚxr8 ÚerrÚweightsÚmrU ZestZval0Úm2Zcov2ÚvalZm3Zcov3ÚyyÚccr_ rW r r r Ú test_polyfit s| ÿ
|
þ þ þ. "" ÿ
|
$$zTestPolynomial.test_polyfitc C sî ddl m} t |d|d|dg¡}||d }t|d |dk | ¡ }t|d |d k | ¡ }t|d
|
|d k t|d |d k tt |jj tj
|
¡ t |d|d g¡}t t |d|d g¡d|d|d g d S )Nr )ÚDecimalz4.0z3.0z2.0z1.333333333333333r z3.9999999999999990z8.0r z 1.333333333333333333333333333r z1.5r ) rV rk r r r r' r& Z
|
issubdtypeÚcoeffsr, r0 rF r )r rk r Úp2r r r Ú test_objectsÒ s ÿzTestPolynomial.test_objectsc C sT t dddg¡}| ¡ }t|jddddgk ¡ | ¡ }t|jddgk ¡ d S )Ny @rC r@ r y @)r r r& r rl Úallr' ©r r rm r r r Ú test_complexá s
|
zTestPolynomial.test_complexc C sF t dddg¡}|jddddgd}t|jdd d
|
d ddgk ¡ d S ) Nr r r é r> r= )Úkg©?gUUUUUUµ?gUUUUUUÅ?g @)r r r& r rl ro rp r r r Útest_integ_coeffsè s
|
ÿz TestPolynomial.test_integ_coeffsc C s. zt t d¡¡ W n tk
|
r( Y nX d S )N)r r )r rF r] r[ rP r r r Útest_zero_dimsî s z TestPolynomial.test_zero_dimsc C s, t dd¡}tt |¡t t |¡¡ dS )z.
|
Regression test for gh-5096.
|
r é N)r rZ r rF Zdiag)r Úvr r r Útest_poly_int_overflowô s z%TestPolynomial.test_poly_int_overflowc C sz t dddg¡}t | t j¡¡}t|jjt j t | t j¡¡}t|jjt j t | t j ¡¡}t|jjt j dS )z/
|
Regression test for gh-16354.
|
r N)
|
r r- r ZastypeÚint64r rl r, Úfloat32Z complex64)r Úzr r r r Útest_zero_poly_dtypeû s z#TestPolynomial.test_zero_poly_dtypec C sj t dddg¡}t dddg¡}t|d kd t|d kd t||kd t||kd t||kd d S )Nr r r r# FTr rp r r r Ú test_poly_eq s zTestPolynomial.test_poly_eqc C sä t ddddg¡}t ddddg¡}t ||¡\}}t|jjt j t|jjt j t|| | | dddg}t dddg¡}t ||¡\}}t|t js tt|t js°tt ||¡\} }
|
t| t jsÐtt|
|
t jsàtd S )Nr r= r rB rA y À ð¿r ) r r r( r rl r, Z
|
complex128r1 r3 ) r ÚbrM r r ra ÚdÚsÚtÚurw r r r Ú test_polydiv s
|
zTestPolynomial.test_polydivc C sn t dddg¡}| jd7 _t|jdddg |jd d7 < t|jdddg ttt|dt d¡ dS ) z# Coefficients should be modifiable r r r r# é
|
é rl N)r r rl r r ÚAttributeErrorÚsetattrr- rR r r r Útest_poly_coeffs_mutable# s z'TestPolynomial.test_poly_coeffs_mutable)Ú__name__Ú
|
__module__Ú __qualname__r r" r) ÚpytestÚmarkZ parametrizeÚ
|
TYPE_CODESr r7 r< rN rQ rS rj rn rq rt ru rx r| r} r r r r r r r
|
s&
|
J r
|
)Únumpyr Z numpy.testingr r r r r r r r Z typecodesr r
|
r r r r Ú<module> s $
|
