U
|
P ±dóI ã @ s¸ d Z ddlZddlmZ ddlZddlZddlm Z m
|
Z
|
m Z m Z m Z mZ ddlmZmZmZmZ ddlmZ e e
|
e e e efZedd eD Zejeed d
|
d ZejjZdKd dZ eZ eZ dd Z dd Z dd Z!dd Z"dd Z#dd Z$dd Z%d d Z&d d Z'd!d" Z(d#d$ Z)d%d& Z*d'd( Z+d)d* Z,d+d, Z-d-d. Z.d/d0 Z/d1d2 Z0d3d4 Z1d5d6 Z2d7d8 Z3d9d: Z4d;d< Z5d=d> Z6d?d@ Z7dAdB Z8dCdD Z9dEdF Z:dGdH Z;G dIdJ dJZ<dS )LzTest inter-conversion of different polynomial classes.
|
|
This tests the convert and cast methods of all the polynomial classes.
|
|
é N)ÚNumber)Ú
|
PolynomialÚLegendreÚ ChebyshevÚLaguerreÚHermiteÚHermiteE)Úassert_almost_equalÚ assert_raisesÚ assert_equalÚassert_)Ú RankWarningc c s | ] }|j V qd S ©N)Ú__name__)Ú.0Úcls© r úZd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/polynomial/tests/test_classes.pyÚ <genexpr> s r )ÚparamsÚidsc C s | j S r )Úparam)Úrequestr r r ÚPoly s r Ú c C sp z>t t | j|jk¡ t t | j|jk¡ t| j|j W n, tk
|
rj d| d| }t|Y nX d S )NzResult: z
|
Target: )r ÚnpÚallÚdomainÚwindowr ÚcoefÚAssertionError)Úp1Úp2Úmsgr r r Úassert_poly_almost_equal& s r$ c
|
C sª t ddd¡}td}| jtdd }| jtdd }| |||d}|jtdd }|jtdd }|j|||d} t| j| t| j| t| ||| d S ) Nr é é
|
©é ©é ç Ð?©r r )Úkindr r )r ÚlinspaceÚrandomr r Úconvertr ©
|
ÚPoly1ÚPoly2Úxr Úd1Zw1r! Zd2Zw2r" r r r Útest_conversion8 s r6 c
|
C sª t ddd¡}td}| jtdd }| jtdd }| |||d}|jtdd }|jtdd }|j|||d} t| j| t| j| t| ||| d S )Nr r% r& r' r) r+ r, )r r. r/ r r Úcastr r1 r r r Ú test_castI s r8 c C sr | j tdd }| jtdd }t |d |d d¡}| j||d}t|j | t|j| t||| d S )Nr) r+ r r% é r, )r r/ r r r. Úidentityr r )r ÚdÚwr4 Úpr r r Ú test_identity_ s r> c C sh | j tdd }| jtdd }| jd||d}t|j | t|j| t|jdgd dg d S )Nr) r+ é r, r r% )r r/ r Úbasisr r ©r r; r< r= r r r Ú
|
test_basisi s rB c C s¤ | j tdd }| jtdd }td}| j|||d}t| ¡ t| t|j | t|j| t||d tj }tj}tj |||d}t|j
|
d d d S )Nr) r+ )r? r, r éÿÿÿÿr% ) r r/ r Ú fromrootsr ÚdegreeÚlenr r r7 r )r r; r< Úrr! ZpdomZpwinr" r r r Útest_fromrootsr s rH c C sT dddg}dddg}t t¡}| ||d¡ W 5 Q R X |d jjd dksPtd S )Ng g ð?g @g @r* r z!The fit may be poorly conditioned)ÚpytestZwarnsr ÚfitÚmessageÚargsr )r r4 ÚyÚrecordr r r Útest_bad_conditioned_fit
s
|
|
|
rO c C sö dd }t dd¡}||}| ||d¡}t|jddg t||| t| ¡ d | jtdd }| jtdd }| j||d||d}t||| t|j| t|j| | j||ddd dg||d}t||| t|j| t|j| | ||dg ¡}t|j| j t|j| j | ||ddd dgg ¡}t|j| j t|j| j t |¡}|t|j
|
d }d|d d d
< | |d d d
|d d d
d¡}| j||d|d
|
} | j||ddd dg|d
|
}
|
t||| | t| ||
|
| d S ) Nc S s | | d | d S ©Nr% r* r )r4 r r r Úf s ztest_fit.<locals>.fr r( r) r+ r, r% r* )r< ) r r. rJ r r r rE r/ r Z
|
zeros_likeÚshape) r rQ r4 rM r= r; r< Úzr! r" Úp3r r r Útest_fit s>
|
"rU c C sª | dddgddgddgd}| dddgddgddgd}| dddgddgddgd}| dddgddgddgd}t ||k t ||k t ||k t ||k d S ©Nr% r* r( r r, ©r ©r r! r" rT Úp4r r r Ú
|
test_equal¼ s rZ c C s¦ | dddgddgddgd}| dddgddgddgd}| dddgddgddgd}| dddgddgddgd}t ||k t ||k t ||k t ||k d S rV rW rX r r r Útest_not_equalÇ s r[ c C s* t tdd }t tdd }| |}| |}|| }t|| | t|| | t|| | t|t| | tt|| | t|t |¡ | tt |¡| | tttj || dg| j
|
d d tttj || dg| j d d | t krtttj |t dg ntttj |t dg d S ©N©é ç à?r' r r% ©r ©r )Úlistr/ r$ Útupler Úarrayr
|
Ú TypeErrorÚopÚaddr r r r ©r Úc1Úc2r! r" rT r r r Útest_addÒ s"
|
rk c C s2 t tdd }t tdd }| |}| |}|| }t|| | t|| | t|| | t|t| | tt|| | t|t |¡ | tt |¡| | tttj || dg| j
|
d d tttj || dg| j d d | t krtttj |t dg ntttj |t dg d S r\ )rb r/ r$ rc r rd r
|
re rf Úsubr r r r rh r r r Útest_subè s"
|
rm c C sZ t tdd }t tdd }| |}| |}|| }t|| | t|| | t|| | t|t| | tt|| | t|t |¡ | tt |¡| | t|d || dg td| || dg tttj || dg| j
|
d d tttj || dg| j d d | t kr@tttj |t dg ntttj |t dg d S ) Nr] r_ r' r* r r% r` ra )rb r/ r$ rc r rd r
|
re rf Úmulr r r r rh r r r Útest_mulþ s&
|
ro c C sv t tdd }t tdd }t tdd }| |}| |}| |}|| | }t |j}t|| | t|| | t|| | t|t| | tt|| | t|t |¡ | tt |¡| | td| | dg t|d d| ttt j
|
|| dg| j d d ttt j
|
|| dg| j d d | t kr\ttt j
|
|tdg nttt j
|
|t dg d S ©
|
Nr] r_ r' r) r* r r% r` ra )rb r/ r r$ rc r rd r
|
re rf Úfloordivr r r r © r ri rj Úc3r! r" rT rY Úc4r r r Ú test_floordiv s@
|
ÿ ÿ
|
ru c C s: | dddg}|d }t jD ]D}t|tr t|tr6q |d}tt ||¡| tt tj|| q t
|
t fD ].}|d}tt ||¡| tt tj|| qjt fD ]0}|dd}tt ||¡| tt tj|| q t t t t t dg¡fD ]$}tt tj|| tt tj|| qðtD ]}tt tj||d qd S )Nr% r* r( r? r )r Z
|
ScalarTypeÚ
|
issubclassr Úboolr$ rf Útruedivr
|
re ÚintÚfloatÚcomplexrc rb Údictrd Úclasses)r r! r" ÚstypeÚsÚptyper r r Ú test_truediv1 s*
|
|
|
"r c C sx t tdd }t tdd }t tdd }| |}| |}| |}|| | }t |j}t|| | t|| | t|| | t|t| | tt|| | t|t |¡ | tt |¡| | td| | dg t|d | dg ttt j
|
|| dg| j d d ttt j
|
|| dg| j d d | t kr^ttt j
|
|tdg nttt j
|
|t dg d S rp )rb r/ r r$ rc r rd r
|
re rf Úmodr r r r rr r r r Útest_modL s,
|
|
r c C s. t tdd }t tdd }t tdd }| |}| |}| |}|| | }t |j}t||\} }
|
t| | t|
|
| t||\} }
|
t| | t|
|
| t||\} }
|
t| | t|
|
| t|t|\} }
|
t| | t|
|
| tt||\} }
|
t| | t|
|
| t|t |¡\} }
|
t| | t|
|
| tt |¡|\} }
|
t| | t|
|
| t|d\} }
|
t| d| t|
|
| dg td|\} }
|
t| | dg t|
|
| dg tt t|| dg| j
|
d d tt t|| dg| j d d | t krtt t|t dg ntt t|t dg d S rp )rb r/ r Údivmodr$ rc r rd r
|
re r r r r ) r ri rj rs r! r" rT rY rt ZquoÚremr r r Ú test_divmodg sP
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
r c C sp | j d d }| j}t |d |d d¡}t | j|||d ¡ ¡}t|| t | |¡ ¡ ¡}t|| d S )Ng ô?r+ r r% r? r, )r r r r. ÚsortrD Úrootsr )r r; r< ÚtgtÚresr r r Ú
|
test_roots s
|
r c C s | d¡}t| ¡ d d S ©Nr? )r@ r rE ©r r= r r r Ú test_degree s
|
r c C s^ | d¡}| ¡ }t||k t||k t|j|jk t|j|jk t|j|jk d S r )r@ Úcopyr r r r )r r! r" r r r Ú test_copy¤ s
|
r c C sª t }| |dddg¡}| | ¡ ¡}| | d¡¡}t||ddddg t||dddddg | |dddg¡}| |jdd¡}| |jdddgd¡}t||ddddg t||dddddg | |dddg¡}| |jdd ¡}| |jddd ¡}t||d
|
dddg t||dd
|
dddg d| j }| j|dddg|d }| | ¡ ¡}| | d¡¡}t||ddddg t||dddddg d S ) Nr* é é r r( r^ r% ©Úk)Zlbndi÷ÿÿÿr` )r r7 Úintegr$ r )r ÚPZp0r! r" r; r r r Ú
|
test_integ® s,
|
r c C sÞ | j tdd }| jtdd }| dddg||d}|jdddgd}|jddgd}t| d¡j|j t| d¡j|j | dddg}|jdddgd}|jddgd}t| d¡j|j t| d¡j|j d S )Nr) r+ r% r* r( r, r )r r/ r r r Zderivr )r r; r< r! r" rT r r r Ú
|
test_derivË s r c C s¼ | j tdd }| jtdd }| dddg||d}t |d |d d¡}||}| d¡\}}t|| t|| t ddd¡}||}|jdddgd \}}t|| t|| d S )
|
Nr) r+ r% r* r( r, r é r` )r r/ r r r. r )r r; r< r= ZxtgtZytgtZxresZyresr r r Ú test_linspaceÝ s
|
|
|
r c C sÌ | j tdd }| jtdd }| dg||d}| dddg||d}tdD ]}t|| | || }qP| dg}| dddg}tdD ]}t|| | || }qtttj|d tttj|d d S )
|
Nr) r+ r% r, r* r( r? g ø?rC ) r r/ r Úranger$ r
|
Ú
|
ValueErrorrf Úpow)r r; r< r ZtstÚir r r Útest_powï s
|
|
|
r c C s^ t }| j}t |d |d d¡}| |dddg¡}d|dd| }||}t|| d S )Nr r% r9 r* r( )r r r r. r7 r )r r r; r4 r= r r r r r Ú test_call s r c C s~ | dddg}t t|jd t t|jd tt| d¡d tt| d¡d tt| d¡d tt| d¡d d S )Nr% r* r( r_ rC r )r
|
r Zcutdegr rF r r r r Ú test_cutdeg s r¡ c C s~ | dddg}t t|jd t t|jd tt| d¡d tt| d¡d tt| d¡d tt| d¡d d S )Nr% r* r( r_ r r^ )r
|
r Útruncater rF r r r r Ú test_truncate s r£ c C sd ddddg}| |}t | ¡ j|d d
t | d¡j|d d
t | d¡j|d d
d S ) Nr% gíµ ÷Æ°>gê-q=r r( g»½×Ùß|Û=r* gñhãµøä>)r Ztrimr )r Úcr= r r r Ú test_trim" s
|
r¥ c C s` | j }| j}| dg||d}tddg| ¡ d| d }| dg||d}tddg| ¡ d S )Nr% r, r r* )r r r ZmapparmsrA r r r Ú test_mapparms* s r¦ c C s< | dddg}t d¡}ttt j|| ttt j|| d S )Nr% r* r( )r Zonesr
|
re rg )r r= r4 r r r Útest_ufunc_override6 s
|
r§ c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
|
ÚTestInterpolatec C s ||d |d S rP r )Úselfr4 r r r rQ D s zTestInterpolate.fc C s( t ttj| jd t ttj| jd d S )NrC g $@)r
|
r r Ú interpolaterQ re )r© r r r Ú test_raisesG s zTestInterpolate.test_raisesc C s. t ddD ] }tt | j|¡ ¡ |k q
|
d S )Nr% r? )r r r rª rQ rE )r© Údegr r r Útest_dimensionsK s z TestInterpolate.test_dimensionsc C sn dd }t ddd¡}tddD ]H}td|d D ]4}tj||ddg|fd}t|||||dd q2q d S )
|
Nc S s | | S r r )r4 r= r r r ÚpowxQ s z0TestInterpolate.test_approximation.<locals>.powxr r* r& r% )r rL r9 )Údecimal)r r. r r rª r )r© r® r4 r¬ Útr= r r r Útest_approximationO s z"TestInterpolate.test_approximationN)r Ú
|
__module__Ú __qualname__rQ r« r r± r r r r r¨ B s r¨ )r )=Ú__doc__Úoperatorrf Znumbersr rI Únumpyr Znumpy.polynomialr r r r r r Z numpy.testingr r
|
r r Znumpy.polynomial.polyutilsr r} rc ZclassidsZfixturer r/ r$ r2 r3 r6 r8 r> rB rH rO rU rZ r[ rk rm ro ru r r r r r r r r r r r r¡ r£ r¥ r¦ r§ r¨ r r r r Ú<module> sf þ
|
|
|
, -
|
|
|
|
