U
|
P ±dR ã
|
@ s d Z ddlmZ ddlZddlm mZ ddlZddl m
|
Z
|
ddl m Z m Z mZmZmZmZmZ dd ZdgZddgZd dd
|
gZdd dd gZddd ddgZddddddgZd ddddddgZddddddddgZddddddddd g Z dd dd dd dd dd!g
|
Z eeeeeeeee e g
|
Z G d"d# d#Z G d$d% d%Z G d&d' d'Z!G d(d) d)Z"G d*d+ d+Z#G d,d- d-Z$G d.d/ d/Z%G d0d1 d1Z&dS )2z Tests for polynomial module.
|
|
é )ÚreduceN)Údeepcopy)Úassert_almost_equalÚ assert_raisesÚ assert_equalÚassert_Ú assert_warnsÚassert_array_equalÚassert_raises_regexc C s t j| ddS )Ngíµ ÷Æ°>)Ztol)ÚpolyÚpolytrim©Úx© r ú]d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/polynomial/tests/test_polynomial.pyÚtrim s r é éÿÿÿÿé éýÿÿÿé iøÿÿÿé é iìÿÿÿé é iÐÿÿÿé iùÿÿÿé8 iÿÿÿé@ iàÿÿÿé i ÿÿÿé é iÿÿÿi° iÀýÿÿé c @ s< e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd S )Ú TestConstantsc C s t tjddg d S )Nr r )r r Z
|
polydomain©Úselfr r r Útest_polydomain" s z TestConstants.test_polydomainc C s t tjdg d S )Nr )r r Zpolyzeror# r r r Ú test_polyzero% s zTestConstants.test_polyzeroc C s t tjdg d S ©Nr )r r Zpolyoner# r r r Ú test_polyone( s zTestConstants.test_polyonec C s t tjddg d S )Nr r )r r Zpolyxr# r r r Ú
|
test_polyx+ s zTestConstants.test_polyxc C s& t dddg¡}t|}t|| d S ©Nr r é )r Ú
|
Polynomialr r ©r$ r Úyr r r Ú test_copy. s zTestConstants.test_copyc C s. t dddg¡}t t |¡¡}t|| d S r* )r r, ÚpickleÚloadsÚdumpsr r- r r r Ú test_pickle3 s zTestConstants.test_pickleN) Ú__name__Ú
|
__module__Ú __qualname__r% r& r( r) r/ r3 r r r r r" s r" c @ s< e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd S )ÚTestArithmeticc C s t dD ]}t dD ]}d| d| }t t||d ¡}|| d7 < || d7 < t dg| dg dg| dg ¡}tt|t||d qqd S ©Nr úAt i=ú, j=r r ©Úerr_msg)ÚrangeÚnpÚzerosÚmaxr Úpolyaddr r ©r$ ÚiÚjÚmsgÚtgtÚresr r r Ú test_polyadd: s $zTestArithmetic.test_polyaddc C s t dD ]}t dD ]}d| d| }t t||d ¡}|| d7 < || d8 < t dg| dg dg| dg ¡}tt|t||d qqd S r8 )r= r> r? r@ r Zpolysubr r rB r r r Ú test_polysubD s $zTestArithmetic.test_polysubc C sr t t dg¡dg t t dg¡ddg tddD ]8}dg| dg }dg|d dg }t t |¡| q4d S )Nr r r )r r Zpolymulxr= )r$ rC ZserrF r r r Ú test_polymulxN s z TestArithmetic.test_polymulxc C s t dD ]}t dD ]t}d| d| }t || d ¡}||| d7 < t dg| dg dg| dg ¡}tt|t||d qqd S r8 )r= r> r? r Úpolymulr r rB r r r Ú test_polymulV s $zTestArithmetic.test_polymulc
|
C sì t ttjdgdg t dgdg¡\}}t||fd t ddgdg¡\}}t||fd tdD ]}tdD ]v}d| d| }dg| ddg }dg| ddg }t ||¡}t ||¡\}}t t ||¡|¡} t| ||d qnqbd S )
|
Nr r r )r r ))r r r r r9 r: r; )r ÚZeroDivisionErrorr Zpolydivr r= rA rK )
|
r$ ZquoÚremrC rD rE ÚciÚcjrF rG r r r Ú test_polydiv_ s zTestArithmetic.test_polydivc C s| t dD ]n}t dD ]`}d| d| }t |d ¡}ttj|g| t dg¡}t ||¡}tt |t ||d qqd S )Nr r9 r: r r; )
|
r= r> Úaranger r rK ÚarrayZpolypowr r )r$ rC rD rE ÚcrF rG r r r Ú test_polypowt s zTestArithmetic.test_polypowN) r4 r5 r6 rH rI rJ rL rQ rU r r r r r7 8 s
|
|
r7 c @ s e Zd Ze dddg¡Ze dee¡Ze deee¡Zej d¡d d Z
|
e e
|
dddg¡Z d d
|
Zd d Zd d Zdd Zdd Zdd ZdS )ÚTestEvaluationç ð?g @g @úi,j->ijú
|
i,j,k->ijk©r+ r r r c
|
s t t g dg¡jd t dd¡ fddtdD }tdD ].}|| }t dg| dg ¡}t|| q@ d d }t ddddg¡}t|| tdD ]`}dg| }t |¡ t t dg¡j | t t ddg¡j | t t dddg¡j | q¦d d
|
d g}tj
|
j dddg|d }t d ddg|¡}t |j | G d d dtj}t dddg¡ |¡} t tt dddg| ¡| d S )Nr r r c s g | ] } | qS r r ©Ú.0rC r r r Ú
|
<listcomp> s z/TestEvaluation.test_polyval.<locals>.<listcomp>r r r+ FT)Úmaské c @ s e Zd ZdS )z&TestEvaluation.test_polyval.<locals>.CN)r4 r5 r6 r r r r ÚC¦ s r` r )r r ÚpolyvalÚsizer> Úlinspacer= r r? ÚshapeÚmarS r r^ ZndarrayÚviewÚtype)
|
r$ r. rC rF rG Údimsr^ Zmxr` Úcxr r r Ú test_polyval s.
|
|
|
|
zTestEvaluation.test_polyvalc s t ttjdgdgdd tt g dg¡jd tt g dg¡jdk tt g dgd g¡jd tt g dgd g¡jdk tt dd¡d tt dt d¡¡jd k t
|
d
|
d¡ fd d t dD }t ddD ](}|| }t dg| ¡}t || qÚ d d }t d
|
ddg¡}t || t d D ]b}dg| }t |¡ tt dg¡j| tt ddg¡j| tt dddg¡j| q<dddddg}t |¡}t
|
d
|
d¡ t t |¡t |¡ d}t dd¡ tjjdd|d}tj |dd}t |jdd
¡}t |jD ]&} t | |d d
| f ¡|| < q.t|| t d g¡ tj |dd}t |jdd
j ¡}t |jd D ]F} t jd D ]0}
|
t |
|
|d d
| f ¡|| |
|
d d
f< q¼qªt|| d S )Nr F)Ztensorr )r r )r r )r+ r+ )r+ r c s g | ] } | qS r r r[ r r r r] À s z8TestEvaluation.test_polyvalfromroots.<locals>.<listcomp>r+ r é iðÿÿÿéþÿÿÿrZ r éûÿÿÿ)rb T)r Ú
|
ValueErrorr Zpolyvalfromrootsr rb r rd r> Úonesrc r= r r? Ú polyrootsra rR ÚrandomÚrandintÚemptyÚvstack) r$ r. rC rF rG rh ZptestÚrZrshapeÚiiZjjr r r Útest_polyvalfromroots¬ s^ ÿ
|
|
|
|
|
ÿ $
|
2z$TestEvaluation.test_polyvalfromrootsc
|
C s | j \}}}| j\}}}ttdtj||d d
| j || }t ||| j¡}t|| t d¡} t | | | j¡}t
|
|j dk d S ©NZ incompatibler ©r r+ ) r r. r
|
rn r Ú polyval2dÚc2dr r> ro r rd ©
|
r$ Úx1Úx2Úx3Úy1Úy2Zy3rF rG Úzr r r Útest_polyval2dì s
|
ÿ
|
|
z TestEvaluation.test_polyval2dc
|
C s | j \}}}| j\}}}ttdtj|||d d
| j || | }t |||| j¡}t|| t d¡} t | | | | j¡}t
|
|j dk d S rx ) r r. r
|
rn r Ú polyval3dÚc3dr r> ro r rd r| r r r Útest_polyval3dþ s
|
ÿ
|
|
z TestEvaluation.test_polyval3dc
|
C sl | j \}}}| j\}}}t d||¡}t ||| j¡}t|| t d¡} t | | | j¡}t |j
|
dk d S )NrX ry )r r+ r r+ ) r r. r> Úeinsumr Z
|
polygrid2dr{ r ro r rd r| r r r Útest_polygrid2d s
|
|
z TestEvaluation.test_polygrid2dc
|
C sr | j \}}}| j\}}}t d|||¡}t |||| j¡}t|| t d¡} t | | | | j¡}t |j
|
dk d S )NrY ry )r r+ r r+ r r+ ) r r. r> r r Z
|
polygrid3dr
r ro r rd r| r r r Útest_polygrid3d s
|
|
z TestEvaluation.test_polygrid3dN)r4 r5 r6 r> rS Zc1dr r{ r
rq r r ra r. rj rw r r r r r r r r rV ~ s $@rV c @ s e Zd Zdd Zdd ZdS )Ú TestIntegralc C s$ t ttjdgd t ttjdgd t ttjdgdddg t ttjdgdgd t ttjdgdgd t ttjdgdd tt t ddgd¡ W 5 Q R X td d
|
D ]8}dg|d dg }tjdg||d }t|ddg qªtd
|
D ]Z}|d }dg| dg }|gdg| d| g }tj|d|gd }tt |t | qìtd
|
D ]D}|d }dg| dg }tj|d|gdd }tt
|
d|¡| qPtd
|
D ]^}|d }dg| dg }|gdg| d | g }tj|d|gd d }tt |t | qtd
|
D ]r}td d
|
D ]`}dg| dg }|d d
}t|D ]}tj|dd}q<tj||d}tt |t | qqtd
|
D ]}td d
|
D ]n}dg| dg }|d d
}t|D ]}tj|d|gd }q¸tj||t t|d }tt |t | qqtd
|
D ]}td d
|
D ]r}dg| dg }|d d
}t|D ]}tj|d|gdd }qBtj||t t|dd }tt |t | qq td
|
D ]}td d
|
D ]r}dg| dg }|d d
}t|D ]}tj|d|gd d }qÐtj||t t|d d }tt |t | q¨qd S )Nr ç à?r r )Úlbnd)Úscl©ÚaxisrW r r )ÚmÚk)r r r )r r r ©r ) r Ú TypeErrorr Úpolyintrn r ÚDeprecationWarningr= r r ra Úlist)r$ rC r rG r ZpolrF rD r r r Ú test_polyint/ sz
|
zTestIntegral.test_polyintc C s t j d¡}t dd |jD ¡j}tj|dd}t|| t dd |D ¡}tj|dd}t|| t dd |D ¡}tj|d dd
|
}t|| d S ) N©r+ r c S s g | ]}t |¡qS r ©r r ©r\ rT r r r r] s z2TestIntegral.test_polyint_axis.<locals>.<listcomp>r r c S s g | ]}t |¡qS r r r r r r r] s r c S s g | ]}t j|d dqS )r+ )r r r r r r r] s r+ )r r )r> rq rt ÚTr r r ©r$ r{ rF rG r r r Útest_polyint_axis s
|
|
z TestIntegral.test_polyint_axisN)r4 r5 r6 r r r r r r r - s Pr c @ s e Zd Zdd Zdd ZdS )ÚTestDerivativec C s t ttjdgd t ttjdgd tdD ]4}dg| dg }tj|dd}tt|t| q,tdD ]N}tddD ]>}dg| dg }tjtj||d|d}t t|t| qxqjtdD ]R}tddD ]B}dg| dg }tjtj||dd|dd}t t|t| qÐqÂd S ) Nr r r r r r r )r r )
|
r r r Úpolyderrn r= r r r r )r$ rC rF rG rD r r r Ú test_polyder s zTestDerivative.test_polyderc C sl t j d¡}t dd |jD ¡j}tj|dd}t|| t dd |D ¡}tj|dd}t|| d S )Nr c S s g | ]}t |¡qS r ©r r r r r r r] ¯ s z4TestDerivative.test_polyder_axis.<locals>.<listcomp>r r c S s g | ]}t |¡qS r r¡ r r r r r] ³ s r )r> rq rt r r r r r r r r Útest_polyder_axis« s
|
z TestDerivative.test_polyder_axisN)r4 r5 r6 r r¢ r r r r r s r c @ s@ e Zd Zej d¡d d Zdd Zdd Zdd Zd
|
d Z d S ) Ú
|
TestVanderrZ r r c C sÎ t d¡}t |d¡}t|jdk tdD ].}dg| dg }t|d|f t ||¡ q,t ddgddgdd gg¡}t |d¡}t|jd
|
k tdD ].}dg| dg }t|d|f t ||¡ qd S ) Nr+ r r r r .r r é )r+ r r )
|
r> rR r Ú
|
polyvanderr rd r= r ra rS )r$ r ÚvrC Úcoefr r r Útest_polyvander¼ s
|
zTestVander.test_polyvanderc C sx | j \}}}tj d¡}t ||ddg¡}t |||¡}t ||j¡}t|| t |g|gddg¡}t |j
|
dk d S )Nry r r )r r r¤ ) r r> rq r Z polyvander2drz ÚdotÚflatr r rd ©r$ r} r~ r rT ZvanrF rG r r r Útest_polyvander2dÍ s
|
z TestVander.test_polyvander2dc C s | j \}}}tj d¡}t |||dddg¡}t ||||¡}t ||j¡}t|| t |g|g|gdddg¡}t |j
|
dk d S )N)r r+ r r r r+ )r r é ) r r> rq r Z polyvander3dr r© rª r r rd r« r r r Útest_polyvander3dÚ s
|
z TestVander.test_polyvander3dc C s t d¡}tttj|d d S )Nr+ r )r> rR r rn r r¥ )r$ r r r r Útest_polyvandernegdegç s
|
z TestVander.test_polyvandernegdegN)
|
r4 r5 r6 r> rq r r¨ r¬ r® r¯ r r r r r£ ¸ s
|
r£ c @ s$ e Zd Zdd Zdd Zdd ZdS )Ú TestCompanionc C s" t ttjg t ttjdg d S r' )r rn r Ú polycompanionr# r r r Ú test_raisesî s zTestCompanion.test_raisesc C s< t ddD ],}dg| dg }tt |¡j||fk q
|
d S )Nr r r )r= r r r± rd )r$ rC r§ r r r Útest_dimensionsò s z TestCompanion.test_dimensionsc C s t t ddg¡d dk d S )Nr r )r r ç à¿)r r r± r# r r r Útest_linear_root÷ s z TestCompanion.test_linear_rootN)r4 r5 r6 r² r³ rµ r r r r r° ì s r° c @ s< e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd S )ÚTestMiscc C s t g ¡}tt|dg tddD ]^}t t tj dd| d ¡dd d
¡}t | }t |¡d|d }tt|t| q$d S )Nr r r r )
|
r Ú polyfromrootsr r r= r> Úcosrc ÚpiÚTlist)r$ rG rC ÚrootsrF r r r Útest_polyfromrootsý s
|
*zTestMisc.test_polyfromrootsc C sl t t dg¡g t t ddg¡dg tddD ]4}t dd|¡}t t |¡¡}t t|t| q2d S )Nr r r´ r r )r r rp r= r> rc r· r )r$ rC rF rG r r r Útest_polyroots s zTestMisc.test_polyrootsc C sî dd }dd }t ttjdgdgd t ttjdggdgd t ttjg dgd t ttjdgdgggd t ttjddgdgd t ttjdgddgd t ttjdgdgddggd t ttjdgdgdddgd t ttjdgdgdg t ttjdgdgddd
|
g t ttjdgdgg t dd¡}||}t ||d ¡}tt|d t t
|
||¡| t ||dddd g¡}tt|d t t
|
||¡| t ||d ¡}tt|d t t
|
||¡| t ||dddd d g¡}tt|d t t
|
||¡| t |t ||g¡j d ¡}t |t ||g¡j t |t ||g¡j dddd g¡}t |t ||g¡j t |¡}| ¡ } d|dd d
< d| dd d
< tj|| d |d }
|
t |
|
| tj|| dddd g|d }
|
t |
|
| tj|t | | g¡j d |d } t | t ||g¡j tj|t | | g¡j dddd g|d } t | t ||g¡j ddddg}t t ||d¡ddg t t ||ddg¡ddg t dd¡}||}t ||d ¡} t t
|
|| ¡| t ||ddd g¡} t t
|
|| ¡| t | | d S )Nc S s | | d | d S )Nr r r r r r r Úf s z TestMisc.test_polyfit.<locals>.fc S s | d | d d S )Nr r r r r r r r Úf2 s z!TestMisc.test_polyfit.<locals>.f2r r r r )Úwr¤ r+ r r y ð?y ð¿)r rn r Zpolyfitr r> rc r Úlenr ra rS r Z
|
zeros_likeÚcopy)r$ r¾ r¿ r r. Zcoef3Zcoef4Zcoef2drÀ ZywZwcoef3Zwcoef2dZcoef1Zcoef2r r r Ú test_polyfit sj "
|
|
|
& zTestMisc.test_polyfitc C sf ddddg}t ttj|d tt |¡|d d
tt |d¡|d d
tt |d¡dg d S )Nr r r r r )r rn r r r )r$ r§ r r r Ú test_polytrimT s
|
zTestMisc.test_polytrimc C s t t dd¡ddg d S )Nr+ r ©r r Zpolyliner# r r r Ú test_polyline_ s zTestMisc.test_polylinec C s t t dd¡dg d S )Nr+ r rÅ r# r r r Útest_polyline_zerob s zTestMisc.test_polyline_zeroN) r4 r5 r6 r¼ r½ rà rÄ rÆ rÇ r r r r r¶ û s F r¶ )'Ú__doc__Ú functoolsr Únumpyr> Znumpy.polynomial.polynomialZ
|
polynomialr r0 r r Z numpy.testingr r r r r r r
|
r ZT0ZT1ZT2ZT3ZT4ZT5ZT6ZT7ZT8ZT9rº r" r7 rV r r r£ r° r¶ r r r r Ú<module> s6 $
|
F 0c(4
|
