$d[ȥ
d Z ddlmZ ddlZddlmc mZ ddlZddl m
Z
ddl m Z m
Z
mZmZmZmZmZ d ZdgZddgZg dZg d Zg d
Zg d Zg d Zg d
Zg dZ g dZ eeeeeeeee e g
Z G d d Z G d d Z G d d Z! G d d Z" G d d Z# G d d Z$ G d d Z% G d d Z&dS ) z Tests for polynomial module.
)reduceN)deepcopy)assert_almost_equal
assert_raises assert_equalassert_ assert_warnsassert_array_equalassert_raises_regexc . t j | d S )Ngư>)tol)polypolytrimxs z/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/numpy/polynomial/tests/test_polynomial.pytrimr s = % % %% )r )r r )r r ir )r r ir )r r r ir )r ir 8 r ir @ ) r r ir r i r )
r r ir i r ir c 2 e Zd Zd Zd Zd Zd Zd Zd ZdS )
TestConstantsc > t t j ddg d S )Nr r )r r
polydomainselfs r test_polydomainz TestConstants.test_polydomain" s T_r1g.....r c < t t j dg d S )Nr )r r polyzeror) s r
test_polyzerozTestConstants.test_polyzero% s T]QC(((((r c < t t j dg d S Nr )r r polyoner) s r test_polyonezTestConstants.test_polyone( s T\A3'''''r c > t t j ddg d S )Nr r )r r polyxr) s r
test_polyxzTestConstants.test_polyx+ s TZ!Q(((((r c p t j g d }t | }t || d S Nr r )r
Polynomialr r r* r ys r test_copyzTestConstants.test_copy. s6 OIII & & QKKQr c t j g d }t j t j | }t || d S r7 )r r: pickleloadsdumpsr r; s r test_picklezTestConstants.test_pickle3 sB OIII & & L a ) )Qr N) __name__
__module__ __qualname__r+ r. r2 r5 r= rB r r r& r& sn / / /) ) )( ( () ) )
r r& c 2 e Zd Zd Zd Zd Zd Zd Zd ZdS )TestArithmeticc t d D ]}t d D ]}d| d| }t j t || dz }||xx dz
cc< ||xx dz
cc< t j dg|z dgz dg|z dgz }t
t | t | | d S Nr At i=, j=r r err_msg)rangenpzerosmaxr polyaddr r r* ijmsgtgtress r test_polyaddzTestArithmetic.test_polyadd: q @ @A1XX
@
@(a((Q((hs1ayy1}--A! A! lA3q5A3;A << T#YYS 3?????
@ @ @r c t d D ]}t d D ]}d| d| }t j t || dz }||xx dz
cc< ||xx dz cc< t j dg|z dgz dg|z dgz }t
t | t | | d S rJ )rO rP rQ rR r polysubr r rT s r test_polysubzTestArithmetic.test_polysubD r[ r c 0 t t j dg dg t t j dg ddg t dd D ];}dg|z dgz }dg|dz z dgz }t t j | | > S#s33333
4 4 4r c r t d D ]}t d D ]}d| d| }t j |dz }t t j |g|z t j dg }t j || }t t | t | | d S )Nr rK rL r rM )
rO rP aranger r rd arraypolypowr r )r* rU rV rW crX rY s r test_polypowzTestArithmetic.test_polypowt s q @ @A1XX
@
@(a((Q((Ia!e$$ T\A3q5"(A3--@@l1a(( T#YYS 3?????
@ @ @r N) rC rD rE rZ r^ rb re rm rs rF r r rH rH 8 s| @ @ @@ @ @2 2 2@ @ @4 4 4*@ @ @ @ @r rH c e Zd Z ej g d Z ej dee Z ej deee Zej d dz dz
Z
e j e
g d Z
d Zd Zd Zd
Zd Zd Zd
S )TestEvaluation) ?g @g @i,j->ij
i,j,k->ijkr9 r r r c
t t j g dg j d t j dd
fdt
d D }t
d D ]7}|| }t j
dg|z dgz }t || 8
dz dz
z }t j
g d }t || t
d D ]}dg|z }t j |
t t j
dg j | t t j
ddg j | t t j
g d j | g d
}t j
g d | }t j g d
| }t |j
| G d dt j }t j g d | } t t! t j g d| | d S )Nr r r c g | ]}|z S rF rF .0rU r s r
z/TestEvaluation.test_polyval.. $ $ $aQT $ $ $r r r )r r r r r9 r r r )FTFr8 )mask) r r9 c e Zd ZdS )&TestEvaluation.test_polyval..CN)rC rD rE rF r r Cr s Dr r r r9 r )r r polyvalsizerP linspacerO r rQ shapemarp r
r ndarrayviewtype) r* r< rU rX rY dimsr mxr cxr s @r test_polyvalzTestEvaluation.test_polyval sA T\"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"*YYY3344a88888r c t t t j dgdgd t t j g dg j d t
t j g dg j dk t t j g dgdz g j d t
t j g dgdz g j dk t t j dd d t
t j dt j d j d k t j
d
d fd t d D }t dd D ]3}|| }t j dg|z }t || 4 dz
z dz z }t j g d }t || t d
D ]}dg|z }t j
| t t j dg j | t t j ddg j | t t j g d j | g d}t j | }t j
d
d t t j | t j | d}t j dd t j dd| }t j |d }t j |j dd }t |j D ]*} t j | |d d | f || < +t || t j d z g t j |d }t j |j dd j z }t |j d D ]L} t j d D ]/}
t j |
|d d | f || |
d d f< 0Mt || d S )Nr F)tensorr )r r )r r )r9 r9 )r9 r c g | ]}|z S rF rF r| s r r~ z8TestEvaluation.test_polyvalfromroots.. r r )r r r r9 r r ) r ir ry r )r T)r
ValueErrorr polyvalfromrootsr r r r rP onesr rO r rQ polyrootsr ro randomrandintemptyvstack) r* r< rU rX rY r ptestrrshapeiijjr s @r test_polyvalfromrootsz$TestEvaluation.test_polyvalfromroots sY j$"7cA3u . . . . T*2s338!<<<%b1#..4<=== T*2ay99>BBB%bA37)44:fDEEE T*1a00!444%a 99?4GHHH
KA $ $ $ $588 $ $ $q! * *AA$C'A3q511C S ) ) ) )QiQ #Azzz22C%%% q J JA3q5DA .q1#66 > > @% % %$% % %$ ' ' ' ' ' ' ' 'r ru c e Zd Zd Zd ZdS ) TestIntegralc t t t j dgd t t t j dgd t t t j dgdddg t t t j dgdg t t t j dgdg t t t j dgd t t 5 t j ddgd d d d n # 1 swxY w Y t d d
D ]9}dg|d z
z dgz }t j dg|| }t |ddg :t d
D ]d}|dz }dg|z dgz }|gdg|z z d|z gz }t j |d|g }t t | t | et d
D ]M}|dz }dg|z dgz }t j |d|gd }t t j
d| | Nt d
D ]e}|dz }dg|z dgz }|gdg|z z d |z gz }t j |d|gd
}t t | t | ft d
D ]}t d d
D ]~}dg|z dgz }|d d }t | D ]}t j |d }t j || }t t | t | t d
D ]}t d d
D ]}dg|z dgz }|d d }t | D ]}t j |d|g }t j ||t t | }t t | t | t d
D ]}t d d
D ]}dg|z dgz }|d d }t | D ]}t j |d|gd } t j ||t t | d }t t | t | t d
D ]}t d d
D ]}dg|z dgz }|d d }t | D ]}t j |d|gd
} t j ||t t | d
}t t | t | d S )Nr ?r r )lbnd)sclaxisrv r r )mk)r r r )r r r r ) r TypeErrorr polyintr r DeprecationWarningrO r r r list)r* rU r rY r polrX rV s r test_polyintzTestIntegral.test_polyint/ s i sB777j$,R888j$,QA???j$,1#>>>>j$,!====i s<<<<
,
-
- % % 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 5 q : :Aa%C#a%1#+C,saA3R888C R 5 5q 9 9 9 9 q 6 6Aa%C#a%1#+C#A+3'C,saA3A666C S 499 5 5 5 5 q : :A1a[[
:
:c!eqck!!!f q 1 1A ,sa000CCl3!,,,#DIItCyy9999
: q : :A1a[[
:
:c!eqck!!!f q 8 8A ,saA3777CCl3!tE!HH~~>>>#DIItCyy9999
: q : :A1a[[
:
:c!eqck!!!f q A AA ,saA3R@@@CCl3!tE!HH~~BGGG#DIItCyy9999
: q : :A1a[[
:
:c!eqck!!!f q ? ?A ,saA3A>>>CCl3!tE!HH~~1EEE#DIItCyy9999
: : :s )D
DDc t j d }t j d |j D j }t j |d }t
|| t j d |D }t j |d }t
|| t j d |D }t j |dd }t
|| d S )
Nr9 r c 6 g | ]}t j | S rF r r r} rr s r r~ z2TestIntegral.test_polyint_axis.. 888Q a 888r r r c 6 g | ]}t j | S rF r r s r r~ z2TestIntegral.test_polyint_axis.. 666Q a 666r r c : g | ]}t j |d S )r9 )r r r s r r~ z2TestIntegral.test_polyint_axis.. s' ;;;! a1---;;;r r9 )r r )rP r r Tr r r r* r rX rY s r test_polyint_axisz TestIntegral.test_polyint_axis s i v&&i88#%88899;l3Q'''C%%%i66#66677l3Q'''C%%%i;;s;;;<<l3!!,,,C%%%%%r N)rC rD rE r r rF r r r r - s6 N: N: N:`& & & & &r r c e Zd Zd Zd ZdS )TestDerivativec V t t t j dgd t t t j dgd t d D ]L}dg|z dgz }t j |d }t
t | t | Mt d D ]s}t dd D ]`}dg|z dgz }t j t j || | }t t | t | att d D ]u}t dd D ]b}dg|z dgz }t j t j ||d |d }t t | t | cvd S ) Nr r r r r r r )r r )
r r r polyderr rO r r r r )r* rU rX rY rV s r test_polyderzTestDerivative.test_polyder s i sB777j$,R888 q / /A#a%1#+C,sa(((C cDII . . . . q : :A1a[[
:
:c!eqckl4<q#9#9#9Q???#DIItCyy9999
: q : :A1a[[
:
:c!eqckl4<qa#@#@#@A2NNN#DIItCyy9999
: : :r c h t j d }t j d |j D j }t j |d }t
|| t j d |D }t j |d }t
|| d S )Nr c 6 g | ]}t j | S rF r r r s r r~ z4TestDerivative.test_polyder_axis.. r r r r c 6 g | ]}t j | S rF r r s r r~ z4TestDerivative.test_polyder_axis.. r r r )rP r r r r r r r s r test_polyder_axisz TestDerivative.test_polyder_axis s i v&&i88#%88899;l3Q'''C%%%i66#66677l3Q'''C%%%%%r N)rC rD rE r r rF r r r r s2 : : :2
&
&
&
&
&r r c f e Zd Zej d dz dz
Zd Zd Zd Zd Z dS )
TestVanderry r r c 8 t j d }t j |d }t |j dk t
d D ]7}dg|z dgz }t |d|f t j || 8t j ddgddgdd gg }t j |d }t |j d
k t
d D ]7}dg|z dgz }t |d|f t j || 8d S ) Nr9 r r r r .r r )r9 r r )
rP ro r
polyvanderr r rO r r rp )r* r vrU coefs r test_polyvanderzTestVander.test_polyvander s) 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 Br c n | j \ }}}t j d }t j ||ddg }t j ||| }t j ||j }t || t j |g|gddg }t |j
dk d S )Nr r r )r r r ) r rP r r polyvander2dr dotflatr r r r* r r r rr vanrX rY s r test_polyvander2dz TestVander.test_polyvander2d s V
B I V $ $ BA//nRQ''fS!&!!C%%% rdQF33 Y&'''''r c v | j \ }}}t j d }t j |||g d }t j |||| }t j ||j }t || t j |g|g|gg d }t |j
dk d S )Nr r8 )r r ) r rP r r polyvander3dr r r r r r r s r test_polyvander3dz TestVander.test_polyvander3d s V
B I Y ' ' BIII66nRR++fS!&!!C%%% rdRD)))<< Z'(((((r c p t j d }t t t j |d d S )Nr9 r )rP ro r r r r )r* r s r test_polyvandernegdegz TestVander.test_polyvandernegdeg s+ IaLLj$/1b99999r N)
rC rD rE rP r r r r r r rF r r r r sr
"Q&AB B B" ( ( ( ) ) ): : : : :r r c e Zd Zd Zd Zd ZdS )
TestCompanionc t t t j g t t t j dg d S r0 )r r r
polycompanionr) s r test_raiseszTestCompanion.test_raises s4 j$"4b999j$"4qc:::::r c t dd D ]8}dg|z dgz }t t j | j ||fk 9d S )Nr r r )rO r r r r )r* rU r s r test_dimensionsz TestCompanion.test_dimensions s\ q! > >A3q5A3;D D&t,,2q!f< = = = = > >r c ` t t j ddg d dk d S )Nr r )r r )r r r r) s r test_linear_rootz TestCompanion.test_linear_root s/ "Aq6**40C788888r N)rC rD rE r r r rF r r r r sA ; ; ;> > >
9 9 9 9 9r r c 2 e Zd Zd Zd Zd Zd Zd Zd ZdS )TestMiscc t j g }t t | dg t dd D ]}t j t j t
j dd|z dz dd d }t | }t j | d|dz
z z }t t | t | d S )Nr r r r )
r
polyfromrootsr r rO rP cosr piTlist)r* rY rU rootsrX s r test_polyfromrootszTestMisc.test_polyfromroots s $$DIIs+++q! 6 6AF2;vq!A#'::14a4@AAE(C$U++A!H4C S 499 5 5 5 5 6 6r c t t j dg g t t j ddg dg t dd D ]h}t j dd| }t j t j | }t t | t | id S )Nr r r r r )r r r rO rP r r r )r* rU rX rY s r test_polyrootszTestMisc.test_polyroots s DNA3 / /444DNAq6 2 2SE:::q! 6 6A+b!Q''C.!3C!8!899C S 499 5 5 5 5 6 6r c d }d }t t t j dgdgd t t t j dggdgd t t t j g dgd t t t j dgdgggd t t t j ddgdgd t t t j dgddgd t t t j dgdgddgg t t t j dgdgdddg t t t j dgdgdg t t t j dgdgg d t t t j dgdgg t j dd } || }t j ||d }t t | d
t t j
|| | t j ||g d }t t | d
t t j
|| | t j ||d
}t t | d t t j
|| | t j ||g d
}t t | d t t j
|| | t j |t j ||g j d }t |t j ||g j t j |t j ||g j g d }t |t j ||g j t j
| }| } d|dd d< d| dd d< t j || d | }
t |
| t j || g d | }
t |
| t j |t j | | g j d | } t | t j ||g j t j |t j | | g j g d | } t | t j ||g j g d}t t j ||d ddg t t j ||ddg ddg t j dd } || }t j ||d
} t t j
|| | t j ||g d }
t t j
||
| t | |
d S )Nc | | dz
z | dz
z S )Nr r rF r s r fz TestMisc.test_polyfit..f s a!e9a!e$ $r c | dz | dz z dz S )Nr r r rF r s r f2z!TestMisc.test_polyfit..f2 s a4!Q$;? "r r r r r )w)r r r r9 r )r r r r9 r )r r r r9 r )r y ?r y )r r r )r r r polyfitr rP r r lenr r rp r
zeros_likecopy)r* r r r r< coef3coef4coef2dr ywwcoef3wcoef2dcoef1coef2s r test_polyfitzTestMisc.test_polyfit sQ % % % # # # j$,aS"===i uqc1===i rA3:::i sqcUGQ???i 1vsA>>>i sQFA>>>i sQCseDDDDi sQCq!fEEEEj$,aS2%@@@j$,aS***EEEi sQC<<<
K1
AaDD Q1%%SZZ###DLE 2 2A666 Q<<<00SZZ###DLE 2 2A666 Q1%%SZZ###DLE 2 2A666 Q???33SZZ###DLE 2 2A666 a1a&!1!1!3Q77FBHeU^$<$<$>??? a1a&!1!1!3\\\BBFBHeU^$<$<$>??? M!
VVXX!$Q$14a4 aQ!,,,FE*** a\\\Q777FE***,q"(B8"4"4"6Q???GRXuen%=%=%?@@@,q"(B8"4"4"6 JJJGRXuen%=%=%?@@@
DLAq 1 1Aq6:::DLA1v 6 6A??? KA BqEE Q1%%DLE 2 2A666 Q999--DLE 2 2A666E5)))))r c B g d}t t t j |d t t j | |d d t t j |d |d d t t j |d dg d S )N)r r r r r r r r r )r r r r r )r* r s r
test_polytrimzTestMisc.test_polytrimT s }} j$-r::: T]4(($ss)444T]4++T#2#Y777T]4++aS11111r c P t t j dd ddg d S )Nr9 r r r polyliner) s r
test_polylinezTestMisc.test_polyline_ s( T]1a((1a&11111r c N t t j dd dg d S )Nr9 r r) r) s r test_polyline_zerozTestMisc.test_polyline_zerob s& T]1a((1#.....r N) rC rD rE r r r% r' r+ r- rF r r r r sr 6 6 66 6 6D* D* D*L 2 2 22 2 2/ / / / /r r )'__doc__ functoolsr numpyrP numpy.polynomial.polynomial
polynomialr r? r r
numpy.testingr r r r r r
r r T0T1T2T3T4T5T6T7T8T9r r& rH ru r r r r r rF r r r> s * * * * * * * * *
; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ;
& & & SVZZ]] ###)))--- RRRRR0 0C@ 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 9 h/ h/ h/ h/ h/ h/ h/ h/ h/ h/r