.eKϯ d Z ddlZddlmZ ddlZddlZddlm Z m
Z
m Z m Z m
Z
mZ ddlmZmZmZmZ ddlmZ e e
e e e
efZ ed eD Z ej ee d Zej j Zd+d Z eZ eZ d Z d
Z d Z!d Z"d Z#d Z$d Z%d Z&d Z'd Z(d Z)d Z*d Z+d Z,d Z-d Z.d Z/d Z0d Z1d Z2d Z3d! Z4d" Z5d# Z6d$ Z7d% Z8d& Z9d' Z:d( Z; G d) d* Zr s$ 11# 111111 )paramsidsc | j S r )param)requests r Polyr s
=r c B t t j | j |j k t t j | j |j k t | j |j d S # t $ r d| d| }t | w xY w)NzResult: z
Target: )r
npalldomainwindowr
coefAssertionError)p1p2msgs r assert_poly_almost_equalr( & s "ryBI-..///ryBI-..///BGRW----- " " "+++r++ S!!!"s A8A< <"B c t j ddd }t d }| j t d dz z }| j t d dz z } | ||| }|j t d dz z }|j t d dz z }| ||| } t
| j | t
| j | t
| | || d S ) Nr
?r! r" )kindr! r" )r linspacerandomr! r" convertr
Poly1Poly2xr# d1w1r% d2w2r& s
r test_conversionr> 8 s
Aq" A $<.f s !a%y!a% r r r- r. r0 r1 )r r* r/ r- r* r/ )rF ) r r3 r\ r
r! r rQ r4 r"
zeros_likeshape) r rf r9 r_ rG rE rF zr% r& p3s r test_fitrk s ! ! !
AqA !A
AqA1a&)))!a Q
fTll3&&A fTll3&&A Aq1--A!a !$$$!$$$ A|||Aa88A!a !$$$!$$$
Aq" A4;'''4;''' A|||R((A4;'''4;'''
aA F17OOC A
AcccF
!CCaC&!CCaC&! $ $B
!QQ B
!Q * *B1rr!uu%%%1rr!uu%%%%%r c D | g dddgddg } | g dddgddg } | g dddgddg } | g dddgddg }t ||k t ||k t ||k t ||k d S Nr* r/ r- r r* r/ r- r1 )r* r* r* r
r r% r& rj p4s r
test_equalrr s
iiiA1v 6 6 6B
iiiA1v 6 6 6B
iiiA1v 6 6 6B
iiiA1v 6 6 6B B"H bL bL bLr c @ | g dddgddg } | g dddgddg } | g dddgddg } | g dddgddg }t ||k t ||k t ||k t ||k d S rm ro rp s r test_not_equalrt s
iiiA1v 6 6 6B
iiiA1v 6 6 6B
iiiA1v 6 6 6B
iiiA1v 6 6 6B bL B"H B"H B"Hr c
t t d dz }t t d dz } | | } | | }||z }t ||z | t ||z | t ||z | t |t | z | t t | |z | t |t j | z | t t j | |z | t
t t j | | dg| j
dz t
t t j | | dg| j dz | t u r1t
t t j |t dg d S t
t t j |t dg d S N ?r, r r* r! r" )listr4 r( tupler arrayr TypeErroropaddr! r" r r r c1c2r% r& rj s r test_addr s
fTllR B
fTllR B
bB
bB bB R"Wb))) R"Wb))) R"Wb))) R%))^R000 U2YY^R000 R"(2,, .333 RXb\\B .333)RVRqc$+/)J)J)JKKK)RVRqc$+/)J)J)JKKK ziYs^^<<<<<iZ__=====r c
t t d dz }t t d dz } | | } | | }||z
}t ||z
| t ||z
| t ||z
| t |t | z
| t t | |z
| t |t j | z
| t t j | |z
| t
t t j | | dg| j
dz t
t t j | | dg| j dz | t u r1t
t t j |t dg d S t
t t j |t dg d S rv )r| r4 r( r} r r~ r r r subr! r" r r r s r test_subr s
fTllR B
fTllR B
bB
bB bB R"Wrc*** R"Wb))) R"Wrc*** R%))^R000 U2YY^bS111 R"(2,, .333 RXb\\B .444)RVRqc$+/)J)J)JKKK)RVRqc$+/)J)J)JKKK ziYs^^<<<<<iZ__=====r c
d t t d dz }t t d dz } | | } | | }||z }t ||z | t ||z | t ||z | t |t | z | t t | |z | t |t j | z | t t j | |z | t |dz | | dg z t d|z | | dg z t
t t j | | dg| j
dz t
t t j | | dg| j dz | t u r1t
t t j |t dg d S t
t t j |t dg d S ) Nrw ry r, r/ r r* rz r{ )r| r4 r( r} r r~ r r r mulr! r" r r r s r test_mulr s
fTllR B
fTllR B
bB
bB bB R"Wb))) R"Wb))) R"Wb))) R%))^R000 U2YY^R000 R"(2,, .333 RXb\\B .333 R!VR$$s))^444 QVR$$s))^444)RVRqc$+/)J)J)JKKK)RVRqc$+/)J)J)JKKK ziYs^^<<<<<iZ__=====r c
t t d dz }t t d dz }t t d dz } | | } | | } | | }||z |z }t |j }t ||z | t ||z | t ||z | t |t | z | t t | |z | t |t j | z | t t j | |z | t d|z | dg t |dz d|z t t t j
| | dg| j dz t t t j
| | dg| j dz | t u r1t t t j
|t dg d S t t t j
|t dg d S
Nrw ry r, r. r/ r r* rz r{ )r| r4 r# r( r} r r~ r r r floordivr! r" r r r r r c3r% r& rj rq c4s r
test_floordivr s,
fTllR B
fTllR B
fTllR B
bB
bB
bB b2 B
bg B R2Xr*** R2Xr*** R2Xr*** R599_b111 U2YY"_b111 R28B<< /444 RXb\\R /444 Q"WddA3ii000 R1Wc"f---2;DD!T[1_$E$E$EG G G2;DD!T[1_$E$E$EG G G zib)QC..AAAAAib*aS//BBBBBr c 8 | g d }|dz }t j D ]|}t |t rt |t r- |d }t t
j || | t t t j || }t t fD ]Q} |d }t t
j || | t t t j || Rt fD ]R} |dd }t t
j || | t t t j || St t t t t j dg fD ]D}t t t j || t t t j || Et" D ],}t t t j | |d -d S )Nrn rJ r r* )r
ScalarType
issubclassr boolr( r truedivr r intfloatcomplexr} r| dictr~ classes)r r% r& stypesptypes r test_truedivr 1 s
ggg B aB 4 4%(( Jud,C,C E!HH B!2!2B777iQ3333u 4 4 E!HH B!2!2B777iQ3333 4 4 E!QKK B!2!2B777iQ3333ggtvvtvvtvvrx}}
= 4 4iR333iQ3333 ; ;iRq::::; ;r c
t t d dz }t t d dz }t t d dz } | | } | | } | | }||z |z }t |j }t ||z | t ||z | t ||z | t |t | z | t t | |z | t |t j | z | t t j | |z | t d|z | dg t |dz | dg t t t j
| | dg| j dz t t t j
| | dg| j dz | t u r1t t t j
|t dg d S t t t j
|t dg d S r )r| r4 r# r( r} r r~ r r r modr! r" r r r s r test_modr L s
fTllR B
fTllR B
fTllR B
bB
bB
bB b2 B
bg B R"Wb))) R"Wb))) R"Wb))) R%))^R000 U2YY^R000 R"(2,, .333 RXb\\B .333 QVTT1#YY/// R!VTT1#YY///)RVRqc$+/)J)J)JKKK)RVRqc$+/)J)J)JKKK ziYs^^<<<<<iZ__=====r c
t t d dz }t t d dz }t t d dz } | | } | | } | | }||z |z }t |j }t || \ } }
t | | t |
| t || \ } }
t | | t |
| t || \ } }
t | | t |
| t |t | \ } }
t | | t |
| t t | | \ } }
t | | t |
| t |t
j | \ } }
t | | t |
| t t
j | | \ } }
t | | t |
| t |d \ } }
t | d|z t |
| dg t d| \ } }
t | | dg t |
| dg t t t | | dg| j
dz t t t | | dg| j dz | t u r,t t t |t dg d S t t t |t dg d S r )r| r4 r# divmodr( r} r r~ r r r! r" r r ) r r r r r% r& rj rq r quorems r test_divmodr g s/
fTllR B
fTllR B
fTllR B
bB
bB
bB b2 B
bg Bb"~~HC S"%%% S"%%%b"~~HC S"%%% S"%%%b"~~HC S"%%% S"%%%b%))$$HC S"%%% S"%%%eBii$$HC S"%%% S"%%%b"(2,,''HC S"%%% S"%%%bhrllB''HC S"%%% S"%%%b!}}HC S#b&))) S$$s)),,,a}}HC S$$s)),,, S$$s)),,,)VRqc$+/)J)J)JKKK)VRqc$+/)J)J)JKKK ziYs^^<<<<<iZ__=====r c | j dz dz }| j }t j |d |d d }t j | ||| }t || t j | | }t || d S )Ng ?r0 r r* rJ r1 )r! r" r r3 sortrP rootsr
)r rE rF tgtress r
test_rootsr s dS A A
+adAaD!
$
$C
'$..Qq.99??AA
B
BCS!!!
'$..%%++--
.
.CS!!!!!r c t | d }t | d d S NrJ )rK r rQ r rG s r test_degreer s/
1
AQ r c B | d }| }t ||k t ||u t |j |j u t |j |j u t |j |j u d S r )rK copyr
r# r! r" )r r% r& s r test_copyr s
A B B B"H BbL BG27 "### BIRY &''' BIRY &'''''r c h t }| |g d }| | }| | d }t | |g d t | |g d | |g d }| | d }| | dddg }t | |g d t | |g d | |g d }| | d }| | dd }t | |g d
t | |g d d| j z }| |g d | }| | }| | d }t | |g d t | |g d d S )
N)r/ r/ )r r/ r- rx )r r r* r* r* r* k)r* r/ r- rx )r* r* r* r* r* )lbnd)r/ r- rx )r r r* r* r* rz )r r@ integr( r! )r Pp0r% r& rE s r
test_integr so A
11___%% & &B
B
B R<<< 111 R???!3!3444
11___%% & &B
1
B
q!f%% & &B R<<< 111 R???!3!3444
11___%% & &B
a ! !B
## $ $B R===!1!1222 R#3#3#3!4!4555 $+
A
11___%%a 0 0B
B
B R<<< 111 R???!3!344444r c | j t d dz z }| j t d dz z } | g d|| }| dddg }| ddg }t | d j |j t | d j |j | g d }| dddg }| ddg }t | d j |j t | d j |j d S )Nr. r0 rn r1 r/ r* r )r! r4 r" r r
derivr# )r rE rF r% r& rj s r
test_derivr s?
fTll3&&A fTll3&&A
iii! , , ,B !1v B !s B ("'222 ("'222
iii B !1v B !s B ("'222 ("'22222r c | j t d dz z }| j t d dz z } | g d|| }t j |d |d d } || }| d \ }}t || t || t j ddd } || }| dddg \ }}t || t || d S )
Nr. r0 rn r1 r r* r/ rz )r! r4 r" r r3 r
)r rE rF rG xtgtytgtxresyress r
test_linspacer s fTll3&&A fTll3&&A YYYq+++A
;qtQqT2 & &D 1T77DB JD$d###d###
;q!R D 1T77DB1v..JD$d###d#####r c
| j t d dz z }| j t d dz z } | dg|| } | g d|| }t d D ]}t ||z | ||z } | dg } | g d }t d D ]}t ||z | ||z }t t t j |d t t t j |d d S ) Nr. r0 r* r1 rn rJ g ?rO ) r! r4 r" ranger( r
ValueErrorr pow)r rE rF r tstis r test_powr s fTll3&&A fTll3&&A
$s1Q
'
'
'C
$yyy1
-
-
-C
1XX a---Ci
$s))C
$yyy//C
1XX a---Ci*bfc3///*bfc2.....r c t }| j }t j |d |d d }| |g d }d|dd|z z z z } || }t || d S )Nr r* rC rn r/ r- )r r! r r3 r@ r
)r r rE r9 rG r r s r test_callr s A A
AaD!A$##A
!!III,, A
aQqSk/C
!A$$CS!!!!!r c | g d }t t |j d t t |j d t t | d d t t | d d t t | d d t t | d d d S )Nrn ry rO r- r/ r* r )r r cutdegr rR r s r test_cutdegr s YYYA*ah+++*ah+++QXXa[[!!1%%%QXXa[[!!1%%%QXXa[[!!1%%%QXXa[[!!1%%%%%r c | g d }t t |j d t t |j d t t | d d t t | d d t t | d d t t | d d d S )Nrn ry r rx r- r/ r* )r r truncater rR r s r
test_truncater s YYYA*aj"---*aj!,,,QZZ]]##Q'''QZZ]]##Q'''QZZ]]##Q'''QZZ]]##Q'''''r c B g d} | | }t | j |d d t | d j |d d t | d j |d d d S )N)r* gư>g-q=r r- g|=r/ gh㈵>r* )r trimr# )r crG s r test_trimr " s A QA "1"&&& #QrrU+++ "AbqbE*****r c | j }| j } | dg|| }t ddg| d|z dz } | dg|| }t ddg| d S )Nr* r1 r r/ )r! r" r
mapparmsrL s r
test_mapparmsr * s A A aS1%%%AA
--- !aA aS1%%%AA
-----r c | g d }t j d }t t t j || t t t j || d S )Nrn r- )r onesr r r )r rG r9 s r test_ufunc_overrider 6 sR YYYA
A)RVQ***)RVQ*****r c & e Zd Zd Zd Zd Zd ZdS )TestInterpolatec ||dz
z |dz
z S rd re )selfr9 s r rf zTestInterpolate.fD s AE{a!e$$r c t t t j | j d t t
t j | j d d S )NrO g $@)r r r interpolaterf r )r s r test_raiseszTestInterpolate.test_raisesG s: j)"7DDDi!6DDDDDr c t dd D ]?}t t j | j | |k @d S )Nr* rJ )r r
r r rf rQ )r degs r test_dimensionsz TestInterpolate.test_dimensionsK sX A;; H HC I)$B==??3F G G G G H Hr c
d }t j ddd }t dd D ]X}t d|dz D ]B}t j ||ddg|f }t || ||| d CYd S ) Nc | |z S r re )r9 rG s r powxz0TestInterpolate.test_approximation..powxQ s a4Kr r r/ r+ r* )r! r^ rC )decimal)r r3 r r r r
)r r r9 r trG s r test_approximationz"TestInterpolate.test_approximationO s
K1b ! !B<< B BC1cAg&&
B
B )$QF!NNN#AAaDD$$q!**bAAAAA
B B Br N)r
__module__ __qualname__rf r r r re r r r r B s[ % % %E E EH H H B B B B Br r )r )=__doc__operatorr numbersr rZ numpyr numpy.polynomialr r r r r r
numpy.testingr
r r r
numpy.polynomial.polyutilsr r r} classidsfixturer r4 r( r7 r8 r> rA rH rM rV ra rk rr rt r r r r r r r r r r r r r r r r r r r r r re r r r sy
B B B B B B B B B B B B B B B B 3 2 2 2 2 2 )X X 511111 1 1 wH--- .-
" " " "
& & &"& & &,! ! !& & &( ( (&L L L)& )& )&X > > >,> > >,> > >.C C C8; ; ;6> > >6*> *> *>Z" " "
( ( (5 5 5:3 3 3$$ $ $$/ / /& " " "& & &( ( (+ + + . . .+ + +B B B B B B B B B Br