U
|
P ±dëV ã @ s@ d dl Z d dlZd dlZd dlZd dlmZ d dlZd dlZd dlZd dl m
|
Z
|
d dl m Z d dlZd dlmZ d dlmZmZmZmZmZmZmZmZmZmZm Z m Z m Z d dl m m!Z" d dl#m$Z$ d dl%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;m<Z<m=Z=m>Z>m?Z?m@Z@mAZAmBZBmCZCmDZDmEZE d dlFmGZG d d
|
ZHd d ZIG d d dZJG dd dZKG dd dZLG dd dZMG dd dZNG dd dZOG dd dZPG dd d ZQG d d d ZRG d d d ZSG d!d" d"ZTG d#d$ d$ZUG d%d& d&ZVG d'd( d(ZWG d)d* d*ZXG d+d, d,ZYG d-d. d.ZZG d/d0 d0Z[G d1d2 d2Z\G d3d4 d4Z]dvd6d7Z^dwd9d:Z_G d;d< d<Z`G d=d> d>ZaG d?d@ d@ZbG dAdB dBZcejd edCdDejfdE ejfdF ¡ejd edGd dHdIg¡G dJdK dKZgG dLdM dMZhG dNdO dOZiG dPdQ dQZjG dRdS dSZkG dTdU dUZlG dVdW dWZmG dXdY dYZnG dZd[ d[ZoG d\d] d]ZpG d^d_ d_ZqG d`da daZrG dbdc dcZsG ddde deZtG dfdg dgZuG dhdi diZvG djdk dkZwG dldm dmZxG dndo doZyG dpdq dqZzG drds dsZ{G dtdu duZ|dS )xé N©ÚFraction)Úarrays)Úma) Úassert_Ú assert_equalÚassert_array_equalÚassert_almost_equalÚassert_array_almost_equalÚ assert_raisesÚassert_allcloseÚIS_PYPYÚ assert_warnsÚassert_raises_regexÚsuppress_warningsÚ HAS_REFCOUNTÚIS_WASM)Úrand) Úadd_newdoc_ufuncÚangleÚaverageÚbartlettÚblackmanÚcorrcoefÚcovÚdeleteÚdiffÚdigitizeÚextractÚflipudÚgradientÚhammingÚhanningÚi0ÚinsertÚinterpÚkaiserÚmeshgridÚmsortÚ piecewiseÚplaceÚrot90ÚselectÚsetxor1dÚsincÚtrapzÚ
|
trim_zerosÚunwrapÚuniqueÚ vectorize)Únormalize_axis_tuplec C s t | ¡}t j ||¡}|S ©N)ÚnpÚarangeÚaddÚouter)ÚnÚdata© r< úYd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/lib/tests/test_function_base.pyÚget_mat s
|
r> c C s( t t | |¡jt j¡}| |_||_|S )za
|
Like real + 1j * imag, but behaves as expected when imag contains non-finite
|
values
|
)r6 ÚzerosÚ broadcastÚshapeÚcomplex_ÚrealÚimag)rC rD Úretr< r< r= Ú _make_complex% s rF c @ s$ e Zd Zdd Zdd Zdd ZdS )Ú TestRot90c C s¤ t ttt d¡ t ttt d¡dd t ttt d¡dd t ttt d¡dd t ttt d¡dd d d
|
d gd dd gg}d d gd
|
dgd d gg}d dd gd d
|
d gg}d d gdd
|
gd d gg}d d
|
d gd dd gg}tdddD ]}tt||d| qÞtdddD ]}tt||d| qtdddD ]}tt||d| q(td ddD ]}tt||d| qNttt|dddd| tt|d
|
ddt|ddd d S )Né ©é rJ rJ ©r é rJ ©Úaxes©rJ rJ ©r rJ ©rL rL )éþÿÿÿrL r rL rJ é é éýÿÿÿé )ÚkrR éÿÿÿÿ©r rL ©rL r ©rW rN )r Ú
|
ValueErrorr+ r6 ÚonesÚranger )ÚselfÚaZb1Zb2Zb3Zb4rW r< r< r= Ú
|
test_basic1 s@ ÿþÿþÿzTestRot90.test_basicc C sR t d¡}tt|jd tt|ddt|dd tt|ddt|dd d S )N)é2 é( rS )rc rb rS rP rM ©r rX ©rL rJ ©rR rX )r6 r] r r+ rA ©r_ r` r< r< r= Ú test_axesQ s
|
zTestRot90.test_axesc C sü t d¡ d¡}ddgddggddgd d
|
ggg}ddgddggd
|
dgd dggg}d dgddggd
|
dgddggg}d d
|
gddggddgddggg}tt|d d | tt|d d | tt|dd | tdd
|
D ]&}tt||ddt||d dd qÐd S )Né rI rJ rS é é r rL rH rT rY rM rZ re )rJ r r[ )r6 r7 Úreshaper r+ r^ )r_ r` Z
|
a_rot90_01Z
|
a_rot90_12Z
|
a_rot90_20Z
|
a_rot90_10rW r< r< r= Útest_rotation_axesW sH ÿÿþÿÿþÿÿþÿÿþÿz TestRot90.test_rotation_axesN)Ú__name__Ú
|
__module__Ú __qualname__ra rh rm r< r< r< r= rG 0 s rG c @ sT e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dS )ÚTestFlipc C sl t tjtjt d¡dd t tjtjt d¡dd t tjtjt d¡dd t tjtjt d¡dd d S )NrH rL ©Úaxis©rH rH rJ rU ©r rS )r r6 Ú AxisErrorÚflipr] ©r_ r< r< r= rh v s zTestFlip.test_axesc C sn t d}|d d
d d d
f }tt |d¡| dddgdddgg}dddgdddgg}tt |d¡| d S )NrH rX rL r rJ rS rT ©r> r r6 rw ©r_ r` Úbr< r< r= Ú test_basic_lr| s ÿÿzTestFlip.test_basic_lrc C sn t d}|d d d
d d
f }tt |d¡| dddgdddgg}dddgdddgg}tt |d¡| d S )NrH rX r rL rJ rS rT ry rz r< r< r= Ú test_basic_ud s ÿÿzTestFlip.test_basic_udc C sb t ddgddggddgddggg¡}t ddgddggddgddggg¡}tt |d¡| d S © Nr rL rJ rS rH rT rj rk ©r6 Úarrayr rw rz r< r< r= Útest_3d_swap_axis0 s
|
ÿÿþ
|
ÿÿþzTestFlip.test_3d_swap_axis0c C sb t ddgddggddgddggg¡}t ddgddggddgddggg¡}tt |d¡| d S r~ r rz r< r< r= Útest_3d_swap_axis1 s
|
ÿÿþ
|
ÿÿþzTestFlip.test_3d_swap_axis1c C sb t ddgddggddgddggg¡}t ddgddggddgddggg¡}tt |d¡| d S r~ r rz r< r< r= Útest_3d_swap_axis2ª s
|
ÿÿþ
|
ÿÿþzTestFlip.test_3d_swap_axis2c C sR t d¡ dddd¡}t|jD ],}tt ||¡t | d|¡¡ |d¡ q d S )Néx rJ rS rH rT r ) r6 r7 rl r^ Úndimr rw r Úswapaxes)r_ r` Úir< r< r= Útest_4d· s
|
ÿzTestFlip.test_4dc C sH t dddgdddgg¡}t dddgdddgg¡}tt |¡| d S )NrL rJ rS rH rT rj r rz r< r< r= Útest_default_axis½ s ÿ ÿzTestFlip.test_default_axisc C s² t ddgddggddgddggg¡}tt j|d d
|
| t ddgddggddgddggg¡}tt j|d d
|
| t ddgddggddgddggg¡}tt j|d d
|
| d S ) Nr rL rJ rS rH rT rj rk r< rr rP re r ©r_ r` r{ Úcr< r< r= Útest_multiple_axesÄ s0
|
ÿÿþ
|
ÿÿþ
|
ÿÿþzTestFlip.test_multiple_axesN) rn ro rp rh r| r} r r r r r r r< r< r< r= rq t s
|
|
rq c @ s e Zd Zdd Zdd ZdS )ÚTestAnyc C sT ddddg}ddddg}ddddg}t t |¡ t t |¡ t t |¡ d S ©Nr rL )r r6 Úany©r_ Úy1Úy2Úy3r< r< r= ra Ý s zTestAny.test_basicc C sb dddgdddgdddgg}t t |¡ ttj|dddddg ttj|dddddg d S ©Nr rL rr )r r6 r r Zsometrue©r_ r r< r< r= Útest_ndå s zTestAny.test_ndN©rn ro rp ra r r< r< r< r= r Û s r c @ s e Zd Zdd Zdd ZdS )ÚTestAllc C sl ddddg}ddddg}ddddg}t t |¡ t t |¡ t t |¡ t t t |¡ ¡ d S r )r r6 Úallr r r< r< r= ra î s zTestAll.test_basicc C sd dddgdddgdddgg}t t |¡ ttj|dddddg ttj|dddddg d S r )r r6 r r Zalltruer r< r< r= r ÷ s zTestAll.test_ndNr r< r< r< r= r ì s r c @ s$ e Zd Zdd Zdd Zdd ZdS )ÚTestCopyc C sR t ddgddgg¡}t |¡}t|| d|d< t|d d t|d d d S )NrL rJ rS rH é
|
)r r )r6 r Úcopyr r )r_ r` Úa_copyr< r< r= ra s
|
|
zTestCopy.test_basicc C s° t ddgddgg¡}t|jj t|jj t jddgddggdd}t|jj t|jj t |¡}t|jj t|jj t |¡}t|jj t|jj d S )NrL rJ rS rH ÚF©Úorder)r6 r r ÚflagsÚ c_contiguousÚ f_contiguousr )r_ r` Za_fortr Z a_fort_copyr< r< r= Ú
|
test_order s
|
|
zTestCopy.test_orderc C sV t d¡}tt tj|dd¡ tt tj|dd¡ tt t |¡¡ d S )NrT F)ZsubokT)r r] r Z isMaskedArrayr6 r )r_ Zmxr< r< r= Ú
|
test_subok s
|
zTestCopy.test_subokN)rn ro rp ra r¤ r¥ r< r< r< r= r þ s r c @ s´ e Zd Zdd Zej ddddgddgdd dgd
|
gd gfddd gdd dggdddd ggddgdddggd d d ggfg¡dd Zdd Zdd Z dd Z
|
d d Z d d Z d d! Z dS )"Ú TestAveragec C sæ t dddg¡}tt|dddk t dddg¡}tt|dddk d d d g}tt|ddd k t d
|
¡}d|d < d|d < t| d¡t|d t| d¡t|d td d }t| d¡t|d t| d¡t|d d S )NrL rJ rS r rr ç @ç ð?ç @ç rt rY rZ rT )r6 r r r r] r Úmeanr )r_ r r r Zy4Zy5r< r< r= ra # s
|
|
|
zTestAverage.test_basicz<x, axis, expected_avg, weights, expected_wavg, expected_wsumrL rJ rS Nr§ rH ç ü?ç @rT rj é r r¨ ç @ç @ç #@c
|
C s¶ t j||dd}|jt |¡ks$tt|| t j|||dd}|jt |¡ksTtt|| t j|||ddd\}} |jt |¡kstt|| | jt |¡ks¨tt| | d S )NT©rs Úkeepdims)rs Úweightsr³ )rs r´ Úreturnedr³ )r6 r rA ÚAssertionErrorr )
|
r_ Úxrs Z expected_avgr´ Z expected_wavgZ expected_wsumÚavgZwavgZwsumr< r< r= Útest_basic_keepdims5 s
|
|
ÿ
|
|
z TestAverage.test_basic_keepdimsc C sâ t d¡}t d¡}t||d}t d¡d ¡ d t d¡ ¡ }t|| t dddgddd gg¡}ddg}t||d
|
d }t d d dg¡}t|| d
|
d
|
dg}t||dd }t d dg¡}t|| d
|
d
|
dgd
|
d
|
dgg}t d dg¡}tt||dd | tt||dd td t j
|
¡} td t j ¡}
|
t t j| |
|
dj t | |
|
¡k t dddg¡ dd¡} t ddd g¡ dd¡}t j| |ddd}t dd d g¡}t|| t j| |ddd}t dgd gd gg¡}t|| d S )Nr ©r´ rJ r¨ rL rS rH rT rj r )r´ rs r© r¯ r° ç @F)r´ rs r³ r§ T)r6 r7 r Úsumr r r r r ÚastypeÚfloat32Úfloat64r ÚdtypeÚ result_typerl ) r_ ÚyÚwÚactualZdesiredr Úw0Úw1Úw2r Zw3r· r< r< r= Ú test_weightsL s:
|
|
$
|
|
|
|
|
zTestAverage.test_weightsc C s& t dddgdddgg¡}t|dd\}}t|d t|d
|
dd\}}t|t d d d g¡ t|ddd\}}t|t d d g¡ ddg}t||d
|
dd \}}t|t d d d g¡ dddg}t||ddd \}}t|t d d g¡ d
|
d
|
dgdddgg}t||ddd \}}t|t dd g¡ d S )NrL rJ rS rH rT rj T)rµ r» r r§ r© )r´ rs rµ r¨ )r6 r r r r )r_ r r¸ ZsclrÅ rÆ rÇ r< r< r= Ú test_returnedx s
|
|
zTestAverage.test_returnedc C sz G dd dt j}t ddgddgg¡ |¡}t ddgddgg¡ |¡}ttt |¡| ttt j||d| d S )Nc @ s e Zd ZdS )z-TestAverage.test_subclasses.<locals>.subclassN©rn ro rp r< r< r< r= Úsubclass s rË rL rJ rS rH rº )r6 Úndarrayr Úviewr Útyper )r_ rË r` rà r< r< r= Útest_subclasses s
|
zTestAverage.test_subclassesc C sr dddddg}|D ]Z\}}}t jddgdd gg|d
|
}t jddgdd gg|d
|
}tt j||d jt |¡ qd S ) N)Úi4rÐ Úf8)rÐ Úf4rÑ )rÒ rÐ rÑ )rÒ rÒ rÒ )rÒ rÑ rÑ rL rJ rS rH ©rÀ rº )r6 r r r rÀ )r_ ZtypsÚatÚwtÚrtr` rà r< r< r= Útest_upcasting s ÿzTestAverage.test_upcastingc C sX t dd tdD ¡}t dd tdD ¡}|| ¡ }t| d¡t||d d S )Nc S s g | ]}t |¡qS r< ©ÚdecimalÚDecimal)Ú.0r· r< r< r= Ú
|
<listcomp>¤ s z1TestAverage.test_object_dtype.<locals>.<listcomp>r c S s g | ]}t d ¡qS ©rL rØ )rÛ Ú_r< r< r= rÜ ¥ s r rº )r6 r r^ r¼ r r« r )r_ r` rà r< r< r= Útest_object_dtype£ s z TestAverage.test_object_dtypec C s4 t tddtddg¡}tt |¡tdd d S )NrL rT rS rJ )r6 r r r r rg r< r< r= Ú test_average_class_without_dtype© s z,TestAverage.test_average_class_without_dtype)rn ro rp ra ÚpytestÚmarkÚ parametrizer¹ rÈ rÉ rÏ r× rß rà r< r< r< r= r¦ ! s$
|
|
ÿÿþ
|
, r¦ c @ s¢ e Zd Ze dddg¡e dddg¡e ddd g¡gZe d
|
d
|
d
|
g¡e d
|
d d
|
g¡e d
|
d
|
d g¡gZd d dZdd Zdd Z dd Z
|
dd Z dd Z dd Z dS ) Ú
|
TestSelectrL rJ rS rH rT rj rk ri é FTr c s< g }t t|D ]& | fddt||D p2|g7 }q|S )Nc s g | ]\}}| r| qS r< r< )rÛ ÚVÚC©Úmr< r= rÜ ¹ s z&TestSelect._select.<locals>.<listcomp>)r^ ÚlenÚzip)r_ ZcondÚvaluesÚdefaultÚoutputr< rè r= Ú_select¶ s $zTestSelect._selectc C sL | j }| j}tt||dd| j||dd tt|d tt|d d S )Né ©rí rS )ÚchoicesÚ
|
conditionsr r, rï r rê ©r_ rò ró r< r< r= ra ¼ s ÿzTestSelect.test_basicc C sf t d¡t dddg¡g}dt d¡ dd¡g}tt||t d¡ ttdgdgdgd jd
|
d S ) NTFrL é rH rS ©rH rS r rñ rÝ ) r6 r r7 rl r r, r] r rA ©r_ ró rò r< r< r= Útest_broadcastingÅ s z TestSelect.test_broadcastingc C s t t| j| jdjtj dd | jD }t t| j|jtj t dddtj ddg¡}t
|
|¡}t t|g|gd d d tj d d g d S )
|
Nù ð?c S s g | ]}| tj¡qS r< )r½ r6 Úint8)rÛ Úchoicer< r< r= rÜ Ñ s z0TestSelect.test_return_dtype.<locals>.<listcomp>rL rJ rS rT rk r ) r r, ró rò rÀ r6 rB rú r ÚnanÚisnan)r_ rò Údré r< r< r= Útest_return_dtypeÌ s ÿ
|
z TestSelect.test_return_dtypec C s" t ttg g d t ttg g d S )Nù @)r r\ r, rx r< r< r= Útest_deprecated_emptyØ s z TestSelect.test_deprecated_emptyc C sj | j }| jd d
}|d tj¡|d< ttt|| |d tj¡|d< ttt|| ttt|| d S ©Nr ) rò ró r½ r6 Úint_r Ú TypeErrorr, Úuint8rô r< r< r= Útest_non_bool_deprecationÜ s z$TestSelect.test_non_bool_deprecationc C s2 t dg¡gd }t dg¡gd }t|| d S )NFéd rL )r6 r r, r÷ r< r< r= Útest_many_argumentså s z TestSelect.test_many_argumentsN)r )rn ro rp r6 r rò ró rï ra rø rÿ r r r r< r< r< r= rä ® s þþ
|
rä c @ s^ e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z e
|
j dddg¡dd Z dS )Ú
|
TestInsertc
|
C sª dddg}t t|ddddddg t t|ddddddg t t|dddgdddgddddddg t t|ddddgddddddg t t|dddgdddddddg t t|tdd ddddddddg t t|dddgdddgddddddg tjddgtjd }t t|d|d d
|
d
|
d g t t|g g | tjd d R}t ddt ¡ t t|t d gd ¡ddddddddg t
|
|d j t k W 5 Q R X d S )NrL rJ rS r rX rå rk ri rÓ rª r¨ T©ÚrecordÚalwaysÚ rH ) r r$ Úslicer6 r r¿ ÚwarningsÚcatch_warningsÚfilterwarningsÚ FutureWarningr Úcategory)r_ r` r{ rà r< r< r= ra î s$
|
*$$&* ÿzTestInsert.test_basicc
|
C s dddgg}dddgdddgg}t t|ddgddddg t t|ddddgdd| t t|dddd| t t|ddddddddgg t ddgddgddgg¡}t dd¡ d¡ dd¡}tj|d d
dd
f t dd¡ d¡ dd¡j|d d
dd
f fdd}t t|dgdgdgdggdd| t t|dgdddgdd| t t|ddddgdd| t t|ddgdgdggdd| t d¡ dd¡}t t|d d
d d
f d|d d
df dd| t t|d d
d d
f d|dd d
f dd| t d¡ d¡}t t|d|d d
d d
df d dt|d|d d
d d
df dd t t|d|d d
dd d
f d
|
dt|d|d d
dd d
f dd t tj
|
t|d|d d
dd d
f dd t tj
|
t|d|d d
dd d
f d d t d¡ d¡}t t|d|d d
d d
df d dt|d|d d
d d
df dd t t|d|d d
dd d
f d
|
dt|d|d d
dd d
f dd d S ) NrL rJ r rr rS rH é ©rJ rS rH rX rR éüÿÿÿ) r r$ r6 r r7 Úrepeatrl Z concatenateÚTr rv )r_ r` Úrr{ r r< r< r= Ú test_multidim sN ÿ ,ÿþ$ "22" ÿ" ÿ((" ÿ" ÿzTestInsert.test_multidimc C s\ t d¡}t t j¡ t|g ddd W 5 Q R X t t¡ t|g ddd W 5 Q R X d S )NrL rJ r rr Únonsense)r6 r rá Úraisesrv r$ r rg r< r< r= Útest_0d, s
|
|
zTestInsert.test_0dc C sä G dd dt j}t d¡ |¡}ttt |ddg¡| ttt |g g ¡| ttt |ddgddg¡| ttt |tddddg¡| ttt |tdddg ¡| t d¡ |¡}ttt |ddg¡| d S ) Nc @ s e Zd ZdS )z*TestInsert.test_subclass.<locals>.SubClassNrÊ r< r< r< r= ÚSubClass4 s r r r rL rJ rR rX ) r6 rÌ r7 rÍ r Ú
|
isinstancer$ r r ©r_ r r` r< r< r= Ú test_subclass3 s " zTestInsert.test_subclassc C sD t dddg¡}t dddg|dddg¡ t|t dddg¡ d S )NrL r rJ rS rH rT )r6 r r$ r ©r_ r· r< r< r= Útest_index_array_copied@ s z"TestInsert.test_index_array_copiedc C s t jdddgddgd}d}t |d|¡}t|d t j||jd dgd }t |dd g|¡}t|dd
|
g t j||jd d S ) N)rL r` )rJ r{ )rS r )Úfoor )ÚbarÚa1rÓ )rH rþ r rJ rS )r6 r r$ r rÀ )r_ r` Úvalr{ r< r< r= Útest_structured_arrayE s ÿ
|
z TestInsert.test_structured_arrayc C sp t t¡( t dddgt ddg¡ddg¡ W 5 Q R X t t¡$ t dddgtjg tdg ¡ W 5 Q R X d S ) Nr rL rJ r¨ r§ r é rÓ )rá r Ú
|
IndexErrorr6 r$ r Úfloatrx r< r< r= Útest_index_floatsO s , z TestInsert.test_index_floatsÚidxrH r c C s8 t jtdd t dddg|gddg¡ W 5 Q R X d S )Nz out of bounds©Úmatchr rL rJ rS rH )rá r r* r6 r$ )r_ r- r< r< r= Útest_index_out_of_boundsU s z#TestInsert.test_index_out_of_boundsN)rn ro rp ra r r r! r# r( r, rá râ rã r0 r< r< r< r= r ì s *
|
r c @ s e Zd Zdd ZdS )ÚTestAmaxc C sv dddddddg}t t |¡d ddd gddd
|
gd d d gg}t tj|ddddd g t tj|ddd ddg d S )NrS rH rT r rU éûÿÿÿr» ç $@ç "@r° ri r© r§ r rr r rL )r r6 Zamaxrz r< r< r= ra ] s þzTestAmax.test_basicN©rn ro rp ra r< r< r< r= r1 [ s r1 c @ s e Zd Zdd ZdS )ÚTestAminc C sv dddddddg}t t |¡d ddd gdd
|
d gd d dgg}t tj|ddd d dg t tj|ddd ddg d S )NrS rH rT r rU r2 r» g Àr4 r3 r° ri r© r§ r rr rL r¯ )r r6 Zaminrz r< r< r= ra i s þzTestAmin.test_basicNr5 r< r< r< r= r6 g s r6 c @ s e Zd Zdd ZdS )ÚTestPtpc C s´ t dddddddg¡}t|jdd d
|
t ddd gdd d gdddgg¡}t|jdd d ddg t|jdd dddg t|jdddd ddgg t|jddddgg d S )NrS rH rT r rU r2 r» r rr ç .@r4 r3 r° ri r© r§ ç @rX Tr² rY r )r6 r r Zptprz r< r< r= ra u s þ zTestPtp.test_basicNr5 r< r< r< r= r7 s s r7 c @ s e Zd Zdd ZdS )Ú
|
TestCumsumc
|
C s, dddddddg}ddddgddd d
|
gddddgg}t jt jt jt jt jt jt jt jt j t j
|
f
|
D ]Æ}t ||¡}t ||¡}t ddd d d ddg|¡}t t j |dd| t ddddgdddd gddddgg|¡}t t j |dd| t ddddgddddgdd ddgg|¡}t t j |dd| q`d S )NrL rJ r r® rj rT rH rS rk rå rV r é é# é' r rr ri é é é é é é )r6 rú r Úint16Úuint16Úint32Úuint32r¾ r¿ Ú complex64Ú
|
complex128r r Úcumsum)r_ ÚbaÚba2Úctyper` Úa2Útgtr< r< r= ra s. " þ ÿ ÿzTestCumsum.test_basicNr5 r< r< r< r= r: s r: c @ s e Zd Zdd ZdS )ÚTestProdc C sö dddddddg}ddddgddd d
|
gddddgg}t jt jt jt jt jt jt jt jfD ]}t ||¡}t ||¡}|d krt
|
t t j | t
|
t t j |d qXt |j d d d t|j d d t ddddg|¡ t|j dd t dddg|¡ qXd S )NrL rJ r r® rj rT rH rS rk rå ©Ú1r{ r rr é g rb é$ éT é´ rX r éb éX )r6 rD rE rF rG r¾ r¿ rH rI r r ÚArithmeticErrorÚprodr r ©r_ rK rL rM r` rN r< r< r= ra s( " ÿ ÿ ÿzTestProd.test_basicNr5 r< r< r< r= rP s rP c @ s e Zd Zdd ZdS )Ú TestCumprodc C sN dddddddg}ddddgddd d
|
gddddgg}t jt jt jt jt jt jt jt jfD ]ð}t ||¡}t ||¡}|d krªt
|
t t j | t
|
t t j |d t
|
t t j | qXt t j |d d t dddddddg|¡ t t j |dd t ddddgddddgddddgg|¡ t t j |d d t ddddgddd d gddd d gg|¡ qXd S ) NrL rJ r r® rj rT rH rS rk rå rQ rX rr r) éÜ i( iÈ rS r rõ é rT rb rU rV r r; éÒ rW r rX )r6 rD rE rF rG r¾ r¿ rH rI r r rY Zcumprodr r[ r< r< r= ra ¯ sN " ÿ ÿÿÿ
|
|
þþÿ
|
|
þþÿzTestCumprod.test_basicNr5 r< r< r< r= r\ s r\ c @ sL e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dS )ÚTestDiffc C sì dddddg}t ddddg¡}t d d dg¡}t d
|
dg¡}tt|| tt|dd | tt|dd | d d dddg}t d dddg¡}tt|| ddddg}t dddg¡}t ddg¡}tt|| tt|dd | d S )NrL rH rj rk rõ rS rJ rT rX r ©r: çñ?ç@r© gɿ繿gé?g Àç¹?TF)r6 r r r r )r_ r· ÚoutÚout2Úout3r< r< r= ra Ê s zTestDiff.test_basicc C sö t d¡}d|d d
dd d
d d
f< t d¡}d|d d
dd d
d d
f< tt|t d¡ tt|ddt d¡ tt|ddt d ¡ tt|dd| tt|d
|
d| tt jt|d d tt jt|d d t d t j¡}tt t| d S )N)r r) r; rL rJ )r é r; rX )r r) é rr r )rå r) r; rR rS r gæÇq Çñ?)
|
r6 r? r] r r r rv r r¿ r\ )r_ r· Úexpr< r< r= Ú test_axisÝ s
|
|
zTestDiff.test_axisc C s* dt ddd }|d d
d d
dd
f |d d
d d
d d
f }|d d
d d
dd
f |d d
d d
d d
f }|dd
d d
d d
f |d d
d d
d d
f }|dd
d d
d d
f |d d
d d
d d
f }tt|| tt|dd| tt|dd | tt|ddd
|
| d S ) Nr) r r; rL rX rJ ra r rr )r: rs )r r r )r_ r· Zout1rg rh Zout4r< r< r= r í s 4444zTestDiff.test_ndc sÄ t td ttt dd fddtddD }ddgdgg g g}tt dd k tt||dd D ]R\}\}}tt|t j
|
k t || t |j t j t t|tdt | qld S )
|
NrS rX ra c s g | ]}t |d qS )ra )r )rÛ r: ©r· r< r= rÜ û s z#TestDiff.test_n.<locals>.<listcomp>rL rT r ©Ústart)Úlistr^ r r\ r r Ú enumeraterë rÎ r6 rÌ r r rÀ r rê Úmax)r_ rî Úexpectedr: rf r< rm r= Útest_nø s
|
zTestDiff.test_nc C s t jddt jd}t jddgddt jdgddg}| t jg ddgd ¡ t|ddD ],\}}t||d }t|| t|j |j qZd S )
|
Nz
|
1066-10-13z
|
1066-10-16rÓ rL útimedelta64[D]r rS rn ra )
|
r6 r7 Z
|
datetime64r Úextendrq r r r rÀ )r_ r· rs r: rk rf r< r< r= Ú
|
test_times s þ
|
zTestDiff.test_timesc C sö t jddgddgddgddgd d
|
ggd d gd d gd d gd d gd d ggd }t|}t|jdgdgdgdgdgg t|jd gd gd gd gd gg tt|t|k t|dd}t|jg g g g g g t|jg g g g g g tt|t|k d S )NrL rJ rS rH rT rj rk ri rå r FT©Úmaskra )r r r r r; ry r rÎ )r_ r· rf rh r< r< r= r! s& $ ÿÿ ÿ zTestDiff.test_subclassc C sN t d¡d }tt|ddt d¡ tt|dgdt d¡ tt t j|dd¡| tt|ddgdt d¡ t d¡ dd¡}t j|ddd }ddgddgg}t|| t j|ddgdggd }t|| t j|ddd }ddgddgg}t|| t j|dddggd }t|| ttt j|t d
|
¡d tt j
|
t|dd d d S ) NrT rL r )ÚprependrX rj rH rJ )rs rz ©rS rS rS )rz rs ) r6 r7 r r r] rJ rl r r\ r? rv ©r_ r· Úresultrs r< r< r= Ú test_prepend s$
|
|
|
|
zTestDiff.test_prependc C sB t d¡}t|dd}dddddg}t|| t|dgd}t|| t|ddgd}|dg }t|| t d¡ dd¡}t j|ddd}dd gdd
|
gg}t|| t j|ddgdggd}t|| t j|ddd}ddgd d
|
gg}t|| t j|dddggd}t|| ttt j|t d ¡d tt jt|dd d d S )NrT r )ÚappendrL r rJ rH )rs r rX rU rR r{ rS )r rs ) r6 r7 r r rl r r\ r? rv r| r< r< r= Ú test_append7 s,
|
|
|
|
|
|
|
|
|
zTestDiff.test_appendN) rn ro rp ra rl r rt rw r! r~ r r< r< r< r= r` È s r` c @ s e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z e j de dg¡dgg¡dd Zdd ZdS )Ú
|
TestDeletec C s, t d¡| _t d¡ d¡ ddd¡| _d S )NrT rJ rL )r6 r7 r` r rl Únd_arx r< r< r= Ú setup_methodV s zTestDelete.setup_methodc C s t | j|}t | j|dd}d| }tt|| j|f | j|d t|dd d
df | jd|df }t|| jdd d
df |d d S )NrL rr zDelete failed for obj: %r)Úerr_msgr )r r` r r r- )r_ ÚindicesÚa_delÚnd_a_delÚmsgÚxorr< r< r= Ú_check_inverse_of_slicingZ s ÿ$z$TestDelete._check_inverse_of_slicingc C sV dddddddg}dd dd
|
g}|D ].}|D ]$}|D ]}t |||}| |¡ q2q*q"d S ) NéúÿÿÿrR r rL rJ rH rT rU rX rS )r r )r_ ZlimsZstepsro ÚstopÚstepÚsr< r< r= Ú test_slicesc s zTestDelete.test_slicesc C sú | t ddgddgg¡¡ t t¡ t| jdg W 5 Q R X t t¡ t| jdg W 5 Q R X | ddddg¡ | dddddg¡ t t¡ t| jd W 5 Q R X t t¡ t| jd W 5 Q R X t t¡ t| jdgd W 5 Q R X d S )
|
Nr rL rJ r iÿÿÿrX TFrH ) r r6 r rá r r* r r` r\ rx r< r< r= Ú
|
test_fancyl s zTestDelete.test_fancyc C s | d¡ | d¡ d S )Nr r )r rx r< r< r= Ú test_single s
|
zTestDelete.test_singlec C sX t d¡}t t j¡ t|g dd W 5 Q R X t t¡ t|g dd W 5 Q R X d S )NrL r rr r )r6 r rá r rv r r rg r< r< r= r
s
|
|
zTestDelete.test_0dc C s G dd dt j}| j |¡}ttt|d| ttt|g | ttt|ddg| ttt|tdd| ttt|tdd| d S )Nc @ s e Zd ZdS )z*TestDelete.test_subclass.<locals>.SubClassNrÊ r< r< r< r= r s r r rL rJ rR )r6 rÌ r` rÍ r r r r r r< r< r= r! s zTestDelete.test_subclassc C sR t d¡jdddd}t|tdd dd}t|jj|jj t|jj|jj d S ) Nr rJ rT r r é< rL rr ) r6 r7 rl r r r r¡ r¢ r£ )r_ rW ré r< r< r= Útest_array_order_preserve s z$TestDelete.test_array_order_preservec C sh t t¡" t dddgt ddg¡¡ W 5 Q R X t t¡" t dddgtjg td¡ W 5 Q R X d S )Nr rL rJ r¨ r§ rÓ )rá r r* r6 r r r+ rx r< r< r= r, s & z TestDelete.test_index_floatsÚindexerrL c C sX t | jd}t | j|}t|| t | jddd}t | jt dg¡dd}t|| d S )NrL rr )r r` r r r6 r )r_ r Z a_del_intr Z nd_a_del_intr r< r< r= Útest_single_item_array¦ s
|
z!TestDelete.test_single_item_arrayc C sú t t d¡t dg¡}t|t d¡ t d¡}tjdgtd}tjdgtd}t ||dd}t|| t ||dd}t||d d
d d
f t t¡" t t d ¡tjdgt d W 5 Q R X t t¡" t t d ¡tjdgd
|
d W 5 Q R X d S ) NrL F)rS rL rÓ TrX rr r rJ zm8[ns])
|
r r6 r] r r Úboolrá r r* Úobject)r_ Úresr· Z
|
false_maskZ true_maskr< r< r= Ú test_single_item_array_non_int° s
|
|
& z)TestDelete.test_single_item_array_non_intN)rn ro rp r r r r r r r! r r, rá râ rã r6 r r r r< r< r< r= r T s
|
|
|
r c @ sü e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z ej dejejejejg¡dd Zej dejejejejg¡dd Zej d ejejejejg¡d d Z ej d ejejejejg¡d d Z d!S )"Ú TestGradientc C sf ddgddgg}t |¡}t ddgddgg¡t ddgddgg¡g}tt|| tt|| d S )NrL rS rH r§ r© rª r¨ )r6 r r r ©r_ Úvr· Údxr< r< r= ra Ë s
|
ÿzTestGradient.test_basicc C sÌ t t d¡¡}dddddg}t d¡ dd¡}tt d¡d tt d¡t d¡ tt d¡| t|d t|t d ¡ t||| t||d
|
t||d d ttd t|t j |gd
|
dd d d S )NrT r¨ r§ r° r4 ç &@é r© ç ø?rJ rL rr z.*scalars or 1drX )
|
r6 rJ r] r7 rl r r r r\ Ústack)r_ r Z dx_unevenÚf_2dr< r< r= Ú test_argsÓ s"
|
ÿzTestGradient.test_argsc C sÚ t d¡ dd¡}t t d¡¡}ttt||t d¡ ttt|dt d¡ ttt|t d¡t d¡ ttt|| ttt||dd ttt|||| ttt|ddd ttt|||dd ttt|dddd d S )Nr rT rJ rL rY rr ) r6 r7 rl rJ r] r r\ r r )r_ r¢ r· r< r< r= Ú test_badargsì s zTestGradient.test_badargsc C s^ t jdddddddgdd }t jd
|
d d d dddgdd }tt|| t|jt d¡k d S )Nz
|
1910-08-16z
|
1910-08-11z
|
1910-08-10z
|
1910-08-12z
|
1910-10-12z
|
1910-12-12z
|
1912-12-12z datetime64[D]rÓ r2 rU r é é= i iÛ ru ©r6 r r r r rÀ ©r_ r· r r< r< r= Útest_datetime64ü s ÿýþz TestGradient.test_datetime64c C s t jjddgddggddgddggd}t|d }tt|t| t|j|jk t j d¡}t jj |d< t j|dd t
|
|j ddd
|
ddg d S ) NrL rS rH Frx r rT rJ ©Ú
|
edge_orderT) r6 r r r r rÎ r Z_maskr7 Úmaskedr ry )r_ r· rf Úx2r< r< r= Ú test_masked s ÿ zTestGradient.test_maskedc C s t ddd¡}|d |d }d|d d|d d| }d|d d| d }t t j||dd | d ¡}tt |d
|
k ¡d k t j d¡ t t j d¡¡}d|d d|d d| }d|d d| d }t t j||dd | d ¡}tt |d
|
k ¡d k d S ) Nr rL r rJ rS rH rj ri rª g¸
ëQ¸?T) r6 ÚlinspaceÚabsr r r ÚrandomÚseedÚsort)r_ r· r r Z
|
analyticalZ num_errorr< r< r= Útest_second_order_accurate s z'TestGradient.test_second_order_accuratec C s t ddddddg¡}t |d¡| dd¡ }t d d
|
d ddd g¡}t d ¡}t ddd d d
|
d gd¡}t ddd d d
|
dgd¡}t dddd
|
dd gd¡}t dddd
|
ddgd¡}d|fd|ffD ]ä\}} t|d d|d}
|
t|||d|d} t|||d |d} t|
|
| t| | t|
|
d | j t|
|
d | t|d d|d}
|
t||d|d} t |
|
j
|
| j
|
k t| | j t|d d|d}
|
t||d|d} t |
|
j
|
| j
|
k t| | q¾d|fd|ffD ]\}} t|||d|d}
|
t|||d |d} t|
|
| t|
|
d | j t|
|
d | t||d|d}
|
t|
|
| j t||d|d}
|
t|
|
| q´t|||ddd}
|
t|||ddd} t|
|
d | d t|
|
d | d t|
|
d |j t|
|
d | t|||ddd}
|
t|||ddd} t|
|
d | d t|
|
d | d t|
|
d |j t|
|
d | d S )Nr r§ r© r¯ r° )rj rL rX rL rª ç à?r¨ r9 r» r ç @ç à¿g333333û?ç Ð?g пrJ rY )rs r« rZ ) r6 r Ztilerl r7 r r r r r rA ) r_ ÚfZx_unevenZx_evenZ fdx_even_ord1Z fdx_even_ord2Zfdx_uneven_ord1Zfdx_uneven_ord2r« Zexp_resÚres1Úres2Úres3r< r< r= Ú test_spacing+ st
|
ÿ ÿ
|
|
ÿ ÿ
|
zTestGradient.test_spacingc C s@ ddgddgg}t |¡}t ddgddgg¡t ddgddgg¡g}tt|dd |d tt|dd |d tt|d
|
d |d tt|d d |d |d g tt|d d |d |d g tt|d d t| tt|d dd d |d d |d d g ttt|dd dd tt jt|dd tt jt|d d d S )NrL rS rH r§ r© rª r¨ r rr rX rZ rJ rU )r6 r r r r r r rv r r< r< r= Útest_specific_axesk s"
|
ÿ ÿz TestGradient.test_specific_axesc C s^ t jdddddddgdd }t jd
|
d d d d ddgdd }tt|| t|jt d¡k d S )Nr2 rU r rõ r¦ iA é, ru rÓ rJ rk r é éw iëÿÿÿr§ r¨ r< r< r= Útest_timedelta64
s þþz TestGradient.test_timedelta64c C sF t jt jt jfD ]0}t jdddg|d}tt|jt |¡j qd S )NrL rJ rS rÓ ) r6 Úfloat16r¾ r¿ r r r rÀ r )r_ Údtr· r< r< r= Útest_inexact_dtypes s z TestGradient.test_inexact_dtypesc C s t t d¡dd t t d¡dd ttt t d¡dd ttt t d¡dd ttt t d¡dd ttt t d¡dd ttt t d¡dd d S )NrJ rL rª rS r )r r6 r7 r r\ rx r< r< r= Ú test_values s zTestGradient.test_valuesÚf_dtypec C s8 t jdddddg|d}t|}t|dgt| d S )NrT rH rS rJ rL rÓ rX ©r6 r r r rê )r_ rÇ r¹ Úgr< r< r= Ú test_f_decreasing_unsigned_int¢ s z+TestGradient.test_f_decreasing_unsigned_intc C sR t |¡j}t ddg¡}t jd|g|d}t||}t||d d gd d S )NrL rS rX rÓ rJ )r6 Úiinforr r r r )r_ rÇ Úmaxintr· r¹ Údfdxr< r< r= Útest_f_signed_int_big_jump© s
|
|
z'TestGradient.test_f_signed_int_big_jumpÚx_dtypec C sF t jdddg|d}t dddg¡}t||}t|dgt| d S )NrS rJ rL rÓ r rH rR rÈ )r_ rÏ r· r¹ rÍ r< r< r= Útest_x_decreasing_unsigned² s
|
z'TestGradient.test_x_decreasing_unsignedc C sX t |¡j}t |¡j}t jd|g|d}t |d dg¡}t||}t|ddg d S )NrX rÓ rJ r rµ )r6 rË Úminrr r r r )r_ rÏ ZminintrÌ r· r¹ rÍ r< r< r= Útest_x_signed_int_big_jumpº s
|
z'TestGradient.test_x_signed_int_big_jumpN) rn ro rp ra r£ r¤ r© r® r´ r½ r¾ r rÅ rÆ rá râ rã r6 r rE rG Zuint64rÊ rú rD rF Úint64rÎ rÐ rÒ r< r< r< r= r É s> @ ÿ
|
ÿ
|
ÿ
|
ÿr c @ s e Zd Zdd Zdd ZdS )Ú TestAnglec C s¸ dt d¡d dt d¡ d dddddd g}t|}t d
|
¡t d ¡d t jd t jt j d t d
|
¡ t jt d
|
¡ g}t|d d}t |¡d t j }t||d t||d d S )Nù ð? @rJ r§ rù rL rX ù ð¿ù ð? Ày ð¿ @r© r¨ r T)ÚdegrV r® )r6 Úsqrtr ÚarctanÚpir r
|
)r_ r· r ZyoÚzZzor< r< r= ra Ç s, ÿ
|
|
ý zTestAngle.test_basicc C s t j ddt d¡d d g¡}t jj|d< t j t d¡dt d¡g¡}t jj|d< t|}tt|t| t|j |j t|| d S )NrÕ rL rJ ù ð? ð?r© r r¨ )
|
r6 r r rÙ r¬ rÚ r r rÎ ry )r_ r· rs rÄ r< r< r= r! Ô s zTestAngle.test_subclassN)rn ro rp ra r! r< r< r< r= rÔ Å s rÔ c
|
@ sÄ e Zd Ze ddddddddg¡Ze e¡Ze e ¡Z
|
e e ¡Z dd Z dd Zd
|
d Zd d Zdd Zdd Zej de dddg¡e dddg¡e dddg¡g¡dd Zdd Zdd Zd S ) Ú TestTrimZerosr rL rJ rS rH c s d} fdd|D S )N)r` r{ r rþ c 3 s | ]}t |V qd S r5 )Úgetattr)rÛ Únamerx r< r= Ú <genexpr>è s z'TestTrimZeros.values.<locals>.<genexpr>r< )r_ Z
|
attr_namesr< rx r= rì æ s zTestTrimZeros.valuesc C s6 t jdd
}| ¡ D ]}t|}t||| qd S )NrJ rX ©r6 Ús_rì r0 r ©r_ ÚslcÚarrr r< r< r= ra ê s zTestTrimZeros.test_basicc C s: t jd d
}| ¡ D ] }t|dd}t||| qd S )NrX r{ ©Ztrimrâ rä r< r< r= Útest_leading_skipð s z TestTrimZeros.test_leading_skipc C s: t jdd
}| ¡ D ] }t|dd}t||| qd S )NrJ r rç râ rä r< r< r= Útest_trailing_skipö s z TestTrimZeros.test_trailing_skipc C sZ | ¡ D ]L}tj||jd}t|dd}t|dks8tt|dd}t|dkstqd S )NrÓ ÚBrç r r¹ )rì r6 Z
|
zeros_likerÀ r0 rê r¶ )r_ Z_arrræ rº r» r< r< r= Ú test_all_zeroü s zTestTrimZeros.test_all_zeroc C s t d¡}t|}t|| d S r )r6 r? r0 r ©r_ ræ r r< r< r= Útest_size_zero s
|
z TestTrimZeros.test_size_zeroræ l l l c C s( t jdd
}t|}t||| d S )NrL rJ )r6 rã r0 r )r_ ræ rå r r< r< r= Ú test_overflow s zTestTrimZeros.test_overflowc C s& t d dd g¡}t|}t|| d S ©NrL )r6 r r0 r rì r< r< r= Ú test_no_trim s zTestTrimZeros.test_no_trimc C s t | j ¡ }t|ts td S r5 )r0 r` Útolistr rp r¶ )r_ r r< r< r= Útest_list_to_list s z TestTrimZeros.test_list_to_listN)rn ro rp r6 r r` r½ r+ r{ Úcomplexr r rþ rì ra rè ré rë rí rá râ rã rî rð rò r< r< r< r= rÞ ß s(
|
|
|
|
þþ
|
rÞ c @ s$ e Zd Zdd Zdd Zdd ZdS )Ú
|
TestExtinsc C s> t dddddddg¡}t|dk|}t|dddddg d S )NrL rS rJ )r6 r r r rz r< r< r= ra " s zTestExtins.test_basicc s t ttdddgddgddg t dddddd d
|
g¡ t dddddddgddd g t dddddd d
|
g t t d
|
¡g t t dd ¡ t dddddddgd d g t d dd dd d d g tt d fdd t ddg¡ t ddgd t ddg d S )NrL rJ rS TFr rH rT ri rk rj rå z!Cannot insert from an empty arrayc s t dddddddgg S r )r* r< ©r` r< r= Ú<lambda>6 ó z'TestExtins.test_place.<locals>.<lambda>Z12Z34Ú9)
|
r r r* r6 r r r? r7 r r\ rx r< rõ r= Ú
|
test_place' s
|
ÿzTestExtins.test_placec C sH t d}|dk}| ¡ }t||}t||d t||| t|| d S )Nr rµ r )r r r r* r )r_ r` ry Úacr r< r< r= Ú test_both= s
|
zTestExtins.test_bothN)rn ro rp ra rù rû r< r< r< r= rô s rô r¨ c C s |t | ¡ S r5 ©ÚmathÚfloor©r· r r< r< r= Ú_foo1I s r rª c C s |t | ¡ | S r5 rü )r· r rÜ r< r< r= Ú_foo2M s r c @ s$ e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z dd Zdd Zdd Zd d Zd d Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zd+d, Zd-d. Zd/d0 Zd1d2 Zd3d4 Z d5d6 Z d7d8 Z d9d: Z d;d< Z d=d> Z!d?d@ Z"dAdB Z#dCdD Z$dEdF Z%dGS )HÚ TestVectorizec C s@ dd }t |}|ddddgdddd g}t|dddd
|
g d S ) Nc S s | |kr| | S | | S d S r5 r< ©r` r{ r< r< r= Ú addsubtractT s z.TestVectorize.test_simple.<locals>.addsubtractr rS rj rå rL rT rk rJ ©r3 r ©r_ r r¹ r r< r< r= Ú test_simpleS s zTestVectorize.test_simplec C s8 dd }t |}|ddddgd}t|ddd d
|
g d S ) Nc S s | |kr| | S | | S d S r5 r< r r< r< r= r _ s z.TestVectorize.test_scalar.<locals>.addsubtractr rS rj rå rT ri rL rH r r r< r< r= Ú test_scalar^ s zTestVectorize.test_scalarc C s0 t ddd¡}tdd }||}t|| d S )NrU rJ i' c S s | S r5 r< rm r< r< r= rö k r÷ z*TestVectorize.test_large.<locals>.<lambda>)r6 r¯ r3 r )r_ r· r¹ r r< r< r= Ú
|
test_largei s zTestVectorize.test_largec C sR t tj}t ddtj tjdtj dtj g¡}||}t |¡}t|| d S )Nr rµ r rJ )r3 rý Úcosr6 r rÛ r
|
)r_ r¹ ÚargsÚr1Úr2r< r< r= Ú
|
test_ufunco s
|
|
(
|
zTestVectorize.test_ufuncc C sl ddd}t |}t dddg¡}||}t dddg¡}t|| ||d}t dddg¡}t|| d S ) NrL c S s | | S r5 r< r r< r< r= r$ x s z(TestVectorize.test_keywords.<locals>.foorJ rS rH rT )rL ©r3 r6 r r ©r_ r$ r¹ r r r r< r< r= Ú test_keywordsv s
|
|
|
zTestVectorize.test_keywordsc C s: t ttgd}|t d¡d}|t d¡}t|| d S ©N©Úotypesr© r¨ ©r3 r r+ r6 r7 r ©r_ r¹ r r r< r< r= Ú test_keywords_with_otypes_order1 s z.TestVectorize.test_keywords_with_otypes_order1c C s: t ttgd}|t d¡}|t d¡d}t|| d S r r r r< r< r= Ú test_keywords_with_otypes_order2 s z.TestVectorize.test_keywords_with_otypes_order2c C sT t ttgd}|t d¡}|t d¡dd}|t d¡}t|| t|| d S )Nr r© r¨ ©r r ©r_ r¹ r r Zr3r< r< r= Ú test_keywords_with_otypes_order3 s
|
z.TestVectorize.test_keywords_with_otypes_order3c C sd t ttgd}|ddd}|ddd}|d}t|tddd t|tddd t|td d S )Nr çÍÌÌÌÌÌ$@r )rÜ rX r ©r3 r r+ r r r< r< r= Ú+test_keywords_with_otypes_several_kwd_args1¥ s z9TestVectorize.test_keywords_with_otypes_several_kwd_args1c C sR t ttgd}|dddd}|ddd}t|tdddd t|tddd d S ) Nr r r rX )rÜ r· r rL rJ rS r r r< r< r= Ú+test_keywords_with_otypes_several_kwd_args2² s
|
z9TestVectorize.test_keywords_with_otypes_several_kwd_args2c C s6 dd l }zt|j W n tk
|
r0 t Y nX d S r )r± r3 Ú randrangeÚ Exceptionr¶ )r_ r± r< r< r= Útest_keywords_no_func_code½ s
|
z(TestVectorize.test_keywords_no_func_codec C s d dd}t |}t dddg¡}||d}t dddg¡}t|| |d|d}t|| ||dd }t ddd
|
g¡}t|| d S ) NrL c S s | | S r5 r< r r< r< r= r$ Ê s z5TestVectorize.test_keywords2_ticket_2100.<locals>.foorJ rS rõ rH )r{ r` ©r{ rT )rL r r r< r< r= Útest_keywords2_ticket_2100Ç s
|
|
|
|
z(TestVectorize.test_keywords2_ticket_2100c C sx dd }t j|ddgd}ddg}t||ddgdd dgd
|
t||ddgdd dgd t||ddgdd dg d S ) Nc S s. t |}| d¡}|r*|| | d¡ }q|S r )rp Úpop)r· ÚpZ_pr r< r< r= Ú mypolyvalÚ s
|
|
z;TestVectorize.test_keywords3_ticket_2100.<locals>.mypolyvalr& rL )ÚexcludedrS rj r rJ )r· r& )r& )r6 r3 r )r_ r' ZvpolyvalZansr< r< r= Útest_keywords3_ticket_2100Ø s z(TestVectorize.test_keywords3_ticket_2100c C s. t dd }t|ddgddgdddg d S ) Nc [ s d}| D ]}|| | 9 }q|S ©Nr¨ r< )Úkwr Z_kr< r< r= r¹ é s z3TestVectorize.test_keywords4_ticket_2100.<locals>.frL rJ rS rH r ri r ©r_ r¹ r< r< r= Útest_keywords4_ticket_2100ç s
|
z(TestVectorize.test_keywords4_ticket_2100c C s, t dd }t|ddgddgddg d S )Nc W s
|
t | ¡S r5 )r6 rZ )r r< r< r= r¹ ô s z3TestVectorize.test_keywords5_ticket_2100.<locals>.frL rJ rS rH ri r r, r< r< r= Útest_keywords5_ticket_2100ò s
|
z(TestVectorize.test_keywords5_ticket_2100c C s dd }t |}t| d d S )Nc S s dS rï r< r< r< r< r= r$ û s z5TestVectorize.test_coverage1_ticket_2100.<locals>.foorL r ©r_ r$ r¹ r< r< r= Útest_coverage1_ticket_2100ú s z(TestVectorize.test_coverage1_ticket_2100c C s> dd }t |}t|j|j d}t ||d}t|j| d S )Nc S s | S )zOriginal documentationr< rm r< r< r= r$ s z3TestVectorize.test_assigning_docstring.<locals>.foozProvided documentation)Údoc)r3 r Ú__doc__)r_ r$ r¹ r1 r< r< r= Útest_assigning_docstring s z&TestVectorize.test_assigning_docstringc C s` G dd d}t t| jt d¡t d¡d t t|j| t d¡t d¡d d S )Nc @ s e Zd ZdZdd ZdS )z9TestVectorize.test_UnboundMethod_ticket_1156.<locals>.FoorJ c S s
|
|| j S r5 r# rg r< r< r= r% s z=TestVectorize.test_UnboundMethod_ticket_1156.<locals>.Foo.barN)rn ro rp r{ r% r< r< r< r= ÚFoo s r4 rå rJ )r r3 r% r6 r7 )r_ r4 r< r< r= Ú test_UnboundMethod_ticket_1156 s ÿ ÿz,TestVectorize.test_UnboundMethod_ticket_1156c C sl t dd }|t d¡}|t dd¡}t dd }|t dd¡}|t d¡}t|| t|| d S )Nc S s | S r5 r< rm r< r< r= rö r÷ z@TestVectorize.test_execution_order_ticket_1487.<locals>.<lambda>rS re c S s | S r5 r< rm r< r< r= rö r÷ )r3 r6 r7 r )r_ Úf1Zres1aZres1bÚf2Zres2bZres2ar< r< r= Ú test_execution_order_ticket_1487 s
|
z.TestVectorize.test_execution_order_ticket_1487c C s$ t dd ¡}d}t||| d S )Nc S s | S r5 r< rm r< r< r= rö ' r÷ z7TestVectorize.test_string_ticket_1892.<locals>.<lambda>Zd0123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789)r6 r3 r )r_ r¹ r r< r< r= Útest_string_ticket_1892% s z%TestVectorize.test_string_ticket_1892c sN dg t fdd}d|_t d¡}t|||| t d t| d S )Nr c s d d7 < | d S )Nr rL rJ r< rm ©Z_callsr< r= r¹ / s z#TestVectorize.test_cache.<locals>.fTrT )r3 Úcacher6 r7 r r rê ©r_ r¹ r· r< r: r= Ú
|
test_cache+ s
|
zTestVectorize.test_cachec C s0 t dd ¡}d|_t d¡}t||| d S )Nc S s | S r5 r< rm r< r< r= rö : r÷ z+TestVectorize.test_otypes.<locals>.<lambda>r rT )r6 r3 r r7 r r< r< r< r= Ú test_otypes9 s
|
zTestVectorize.test_otypesc C s t t d¡dgdgf t t d¡dgdgf t t d¡ddgdgf t t d¡dgdgf t t d ¡dgddgf t t d
|
¡dd d gd gf t t d¡dgdgf t t d¡dgdgf t t d¡ddgdgf t t d¡dgdgf t t d¡dgddgf t t d¡dd d gd gf tt t d¡ W 5 Q R X tt t d¡ W 5 Q R X tt t d¡ W 5 Q R X d S )Nz(x)->()rm r< z (x,y)->()rÿ z (x),(y)->()r z(x)->(y)z (x)->(y),()z(),(a,b,c),(d)->(d,e))r` r{ r )rþ )rþ Úez(x )->()z( x , y )->( )z (x),( y) ->()z( x)-> (y ) z (x)->( y),( )z*( ), ( a, b,c ) ,( d) -> (d , e)z
|
(x)(y)->()z (x),(y)->z
|
((x))->(x))r ÚnfbZ_parse_gufunc_signaturer r\ rx r< r< r= Útest_parse_gufunc_signature? sP
|
|
ÿ
|
ÿ
|
|
ÿ
|
ÿ
|
ÿ
|
|
ÿ
|
ÿ
|
|
ÿ
|
ÿÿþ
|
|
|
z)TestVectorize.test_parse_gufunc_signaturec C sD dd }t |dd}|ddddgd dd
|
d g}t|d dd d g d S ) Nc S s | |kr| | S | | S d S r5 r< r r< r< r= r b s z8TestVectorize.test_signature_simple.<locals>.addsubtractz (),()->()©Ú signaturer rS rj rå rL rT rk rJ r r r< r< r= Útest_signature_simplea s z#TestVectorize.test_signature_simplec C s: dd }t |dd}|ddgddgg}t|ddg d S ) Nc S s | ¡ S r5 ©r« rõ r< r< r= r« m s z4TestVectorize.test_signature_mean_last.<locals>.meanz(n)->()rB rL rS rJ rH r )r_ r« r¹ r r< r< r= Útest_signature_mean_lastl s z&TestVectorize.test_signature_mean_lastc C sB dd }t |dd}|ddgddgg}t|d dgd dgg d S )
|
Nc S s | | ¡ S r5 rE rõ r< r< r= Úcenteru s z3TestVectorize.test_signature_center.<locals>.centerú(n)->(n)rB rL rS rJ rH rX r )r_ rG r¹ r r< r< r= Útest_signature_centert s z#TestVectorize.test_signature_centerc C sd t dd dd}|dddg}tt|to4t|dk t|d dddg t|d dddg d S ) Nc S s | | fS r5 r< rm r< r< r= rö } r÷ z:TestVectorize.test_signature_two_outputs.<locals>.<lambda>ú ()->(),()rB rL rJ rS r )r3 r r Útuplerê r ©r_ r¹ r r< r< r= Útest_signature_two_outputs| s
|
z(TestVectorize.test_signature_two_outputsc C s t tjdd}|ddgdddg}t|dddgdddgg |ddgggdddg}t|dddgdddgggg |ddgddggdddg}t|dddgdddggdddgdddggg |ddgdddgdddgg}t|dddgdddggdddgdddggg d S ) Nz(a),(b)->(a,b)rB rL rJ rS rH rj r )r3 r6 r9 r rL r< r< r= Útest_signature_outer s ÿ ÿz"TestVectorize.test_signature_outerc C s^ t dd dd}|dddg}t|ddg |dddgdddgg}t|ddgddgg d S ) Nc S s | d d
S ©NrX r< rm r< r< r= rö r÷ z<TestVectorize.test_signature_computed_size.<locals>.<lambda>z(n)->(m)rB rL rJ rS rH r rL r< r< r= Ú test_signature_computed_size s
|
z*TestVectorize.test_signature_computed_sizec C sV d dd}t |ddhd}t|dddgddd g t|dddgd
|
d dddg d S ) NrL c S s | | S r5 r< r r< r< r= r$ s z2TestVectorize.test_signature_excluded.<locals>.fooú()->()r{ )rC r( rJ rS rH r r# )rL r r/ r< r< r= Útest_signature_excluded s
|
z%TestVectorize.test_signature_excludedc C sH t dd ddgd}|dddg}t|jt d¡ t|dddg d S ) Nc S s | S r5 r< rm r< r< r= rö ¥ r÷ z5TestVectorize.test_signature_otypes.<locals>.<lambda>rH r¿ ©rC r rL rJ rS )r3 r rÀ r6 r rL r< r< r= Útest_signature_otypes¤ s z#TestVectorize.test_signature_otypesc C s¬ t tjdd}ttd |ddg W 5 Q R X ttd |dd W 5 Q R X ttd |ddgdddg W 5 Q R X t tjd d}ttd |dd W 5 Q R X d S )
|
Nz (n),(n)->(n)rB zwrong number of positionalrL rJ z does not have enough dimensionsú$inconsistent size for core dimensionrS rQ )r3 Úoperatorr8 r r r\ r, r< r< r= Ú test_signature_invalid_inputsª s ÿ ÿ z+TestVectorize.test_signature_invalid_inputsc C s t dd dd}ttd |dddg W 5 Q R X t d d d
|
d}ttd |d W 5 Q R X t d d d d}ttd |ddg W 5 Q R X d S )Nc S s | d d
S rO r< rm r< r< r= rö » r÷ z>TestVectorize.test_signature_invalid_outputs.<locals>.<lambda>rH rB rU rL rJ rS c S s | S r5 r< rm r< r< r= rö À r÷ rJ zwrong number of outputsc S s | | fS r5 r< rm r< r< r= rö Ä r÷ rQ )r3 r r\ r, r< r< r= Ú test_signature_invalid_outputs¹ s ÿ z,TestVectorize.test_signature_invalid_outputsc C s t dd ¡}t jddgtd}ttd || W 5 Q R X d|_t||| t jdd d d
|
}ttd || W 5 Q R X t jd d d dd }t||| t jd d ddd }t||| t jdd dd
|
}t||j|j t jdd ddd }ttd || W 5 Q R X d S )Nc S s | S r5 r< rm r< r< r= rö Ê r÷ z5TestVectorize.test_size_zero_output.<locals>.<lambda>r rT rÓ r r c S s | S r5 r< rm r< r< r= rö Ò r÷ rQ rB c S s | S r5 r< rm r< r< r= rö Ö r÷ rS c S s | S r5 r< rm r< r< r= rö Ù r÷ rH c S s | S r5 r< rm r< r< r= rö Ü r÷ c S s | gS r5 r< rm r< r< r= rö ß r÷ z()->(n)znew output dimensions) r6 r3 r? Úintr r\ r r r r< r< r< r= Útest_size_zero_outputÈ s$ z#TestVectorize.test_size_zero_outputc C sä G dd dt j}t dddgdddgdddgg¡ |¡}t dddgddd gd
|
d d gg¡ |¡}t jt jd d}|||}tt|| t|dddgdd dgd
|
d d gg t dd ¡}|||}tt|| t||| d S )Nc @ s e Zd ZdS )z/TestVectorize.test_subclasses.<locals>.subclassNrÊ r< r< r< r= rË ä s rË r¨ rª r§ r© r¯ r° r» r9 r r4 z(m,m),(m)->(m)rB c S s | | S r5 r< rÿ r< r< r= rö ò r÷ z/TestVectorize.test_subclasses.<locals>.<lambda>)r6 rÌ r rÍ r3 Úmatmulr rÎ )r_ rË ré r Zmatvecr Zmultr< r< r= rÏ ã s þþ(
|
"
|
z TestVectorize.test_subclassesN)&rn ro rp r r r r r r r r r r r" r$ r) r- r. r0 r3 r5 r8 r9 r= r> rA rD rF rI rM rN rP rR rT rW rX rZ rÏ r< r< r< r= r Q sF
|
|
" r c @ sL e Zd ZG dd dZejje ddej ddej fdg¡dd Z
|
d
|
S ) Ú TestLeaksc @ s$ e Zd ZdZdd Zedd ZdS )z TestLeaks.Ar) c G s dS r r< ©r_ r r< r< r= Úboundü s zTestLeaks.A.boundc G s dS r r< )r r< r< r= Úunboundÿ s zTestLeaks.A.unboundN)rn ro rp Úitersr^ Ú staticmethodr_ r< r< r< r= ÚAù s rb úPython lacks refcounts©Úreasonz
|
name, incrr^ )r_ r c C s¸ dd l }t| j|}| ¡ zt |¡}t| jjD ]2}| ¡ }t
|
t||dd¡|_ | t d¡¡}q4d }t t |¡|| tdD ] }| ¡ qt t |¡| W 5 | ¡ X d S )Nr rL r rT )Úgcrß rb ÚdisableÚenableÚsysÚ getrefcountr^ r` r6 Z
|
frompyfuncr¹ r7 r Zcollect) r_ rà Úincrrf ZA_funcÚrefcountr r` rf r< r< r= Útest_frompyfunc_leaks s
|
|
z TestLeaks.test_frompyfunc_leaksN) rn ro rp rb rá râ Úskipifr rã r` rm r< r< r< r= r\ ø s
|
þr\ c @ sz e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z ejjdddd ZdS ) Ú TestDigitizec C s2 t dd¡}t dd¡}tt||t d¡ d S )Nr rT r2 r® ©r6 r7 r r ©r_ r· Úbinsr< r< r= Ú test_forward# s zTestDigitize.test_forwardc C s6 t ddd¡}t ddd¡}tt||t d¡ d S )NrT r rX r2 r® rp rq r< r< r= Ú test_reverse( s zTestDigitize.test_reversec C s: t d}t | ¡ | ¡ d¡}tt t||dk¡ d S )Nr r ©r r6 r¯ rÑ rr r r r )r_ r· Úbinr< r< r= Ú test_random- s zTestDigitize.test_randomc C sf dddddddg}dddg}dddd dd dg}t t||| dddddd dg}t t||d
|
| d S ) NrL rT rH r ri r® r rJ rS T)r r )r_ r· rr Zdefault_answerZ right_answerr< r< r= Útest_right_basic2 s
|
z TestDigitize.test_right_basicc C s4 t dd¡}t dd¡}tt||dt d¡ d S )Nr rT rH Tr® rp rq r< r< r= Útest_right_open: s z TestDigitize.test_right_openc C s8 t ddd¡}t ddd¡}tt||dt d¡ d S )NrT r rX rH Tr® rp rq r< r< r= Útest_right_open_reverse? s z$TestDigitize.test_right_open_reversec C s< t d}t | ¡ | ¡ d¡}tt t||ddk¡ d S )Nr Tru rq r< r< r= Útest_right_open_randomD s z#TestDigitize.test_right_open_randomc C s ddddg}dddg}t t||dddddg t t||dddddg dddg}t t||dddddg t t||dddddg ddddg}t t||dddddg t t||dddddg ddddg}ttt|| ddddg}ttt|| d S ) NrX r rL rJ FrS TrH )r r r r\ rq r< r< r= Útest_monotonicI s
|
|
zTestDigitize.test_monotonicc C s> dddg}dddg}t tt|| || }}t tt|| d S )NrL rJ y @ ð?rS )r r r rq r< r< r= Útest_casting_errorY s
|
|
|
|
z TestDigitize.test_casting_errorc C sh G dd dt j}t d¡ |¡}t dd¡ |¡}ttt||d| ttt||d| d S )Nc @ s e Zd ZdS )z(TestDigitize.test_return_type.<locals>.ANrÊ r< r< r< r= rb b s rb rT rL rS FT)r6 rÌ r7 rÍ r r r )r_ rb r` r{ r< r< r= Útest_return_type` s
|
z TestDigitize.test_return_typec C s& d}t t ||d |d g¡d d S ©Nl rL ©r r6 r r" r< r< r= Ú test_large_integers_increasingi s z+TestDigitize.test_large_integers_increasingz8gh-11022: np.core.multiarray._monoticity loses precisionrd c C s& d}t t ||d |d g¡d d S r r r" r< r< r= Ú test_large_integers_decreasingn s z+TestDigitize.test_large_integers_decreasingN)rn ro rp rs rt rw rx ry rz r{ r| r} r~ r rá râ Úxfailr r< r< r< r= ro ! s ÿro c @ s e Zd Zdd Zdd ZdS )Ú
|
TestUnwrapc C sH t tdddtj gddg tt tttdd tjk ¡ d S )NrL rJ r r )r r1 r6 rÛ r r r r rx r< r< r= r x s zTestUnwrap.test_simplec C sî t tddgddddg tt tttdd dddk ¡ t dd d
|
d d g¡}t |d¡}t t|dd| t dd d
|
d d d g¡}t |d¡}t|dd}t |dd d
|
d d dg t|ddd}t |dd d
|
d d d g |j |j ksêt
|
d S )NrL i éÿ ©ÚperiodrJ r éè r éK é éá r¿ i® éú rV é )r Zdiscont) r r1 r r6 r r r r ÚmodrÀ r¶ )r_ Z
|
simple_seqZwrap_seqZ
|
uneven_seqZ wrap_unevenZ
|
no_discontZ
|
sm_discontr< r< r= Ú test_period~ s & zTestUnwrap.test_periodN)rn ro rp r r r< r< r< r= r v s r rÀ ÚOÚ
|
AllIntegerZFloatÚMrL r c @ sf e Zd ZeeddddZeeddddZeeddddZeeddd d
|
Zeeddd d Z dS ) ÚTestFilterwindowsN)rÀ r Úreturnc C s¤ t j||dd }t|}|dkr*t j}nt |jt j¡}|j|ksHtt|t| |dk rpt |t g ¡ n0|dkrt |t
|
d¡ nt t j |dddd d S ) NrÓ r< r rL r rr ç @rH ) r6 r r" r¿ rÁ rÀ r¶ r r r r] r r¼ ©r_ rÀ r Zscalarrà Z ref_dtyper< r< r= Ú test_hanning s z TestFilterwindows.test_hanningc C s¤ t j||dd }t|}|dkr*t j}nt |jt j¡}|j|ksHtt|t| |dk rpt |t g ¡ n0|dkrt |t
|
d¡ nt t j |dddd d S ) NrÓ r< r rL r rr gÃõ(\Â@rH ) r6 r r! r¿ rÁ rÀ r¶ r r r r] r r¼ r r< r< r= Ú test_hamming¬ s z TestFilterwindows.test_hammingc C s¤ t j||dd }t|}|dkr*t j}nt |jt j¡}|j|ksHtt|t| |dk rpt |t g ¡ n0|dkrt |t
|
d¡ nt t j |dddd d S ) NrÓ r< r rL r rr g^)ËÇ@rH ) r6 r r r¿ rÁ rÀ r¶ r r r r] r r¼ r r< r< r= Ú test_bartlettÁ s z TestFilterwindows.test_bartlettc C s¤ t j||dd }t|}|dkr*t j}nt |jt j¡}|j|ksHtt|t| |dk rpt |t g ¡ n0|dkrt |t
|
d¡ nt t j |dddd d S ) NrÓ r< r rL r rr g=
|
×£p=@rH ) r6 r r r¿ rÁ rÀ r¶ r r r r] r r¼ r r< r< r= Ú test_blackmanÖ s z TestFilterwindows.test_blackmanc C s¦ t j||dd }t|d}|dkr,t j}nt |jt j¡}|j|ksJtt|t| |dk rrt |t g ¡ n0|dkrt |t
|
d¡ nt t j |dddd d S ) NrÓ r< r r rL rr r rð ) r6 r r& r¿ rÁ rÀ r¶ r r r r] r r¼ r r< r< r= Ú test_kaiserë s
|
z TestFilterwindows.test_kaiser)
|
rn ro rp ÚstrrY r r r r r r< r< r< r= r s
|
r c @ s$ e Zd Zdd Zdd Zdd ZdS )Ú TestTrapzc C sH t ddd¡}tt d|d ¡t dt j ¡ dd}t|dd d S ) Néöÿÿÿr re r· rJ )r rL rk )r6 r7 r/ rk rÙ rÛ r )r_ r· r r< r< r= r s *zTestTrapz.test_simplec C s& t ddd¡}t ddd¡}t ddd¡}t |¡|d |d }|d d < |d d < t |¡|d |d }|d d < |d d < t |¡|d |d }|d d < |d d < |d d
d d f |d d d
d f |d d d d
f }||d d
d d f jdd}||d d d
d f jdd} ||d d d d
f jdd}
|
t||d d
d d f dd } t| | t||d d d
d f dd } t| | t||d d d d
f dd } t| |
|
t||dd } t| | t||dd } t| | t||dd } t| |
|
d S )
|
Nr rL rS rJ ri rV rX rr )r· rs )r6 r¯ Z ones_liker¼ r/ r ) r_ r· r rÜ ZwxZwyZwzÚqZqxZqyZqzr r< r< r= Ú test_ndim s8 6
|
|
|
|
|
zTestTrapz.test_ndimc C s t d¡}|| }|dk}t jj||d}d}tt||| t jj||d}tt||| t jj||d}tt||| d S )NrT rJ rx g *@)r6 r7 r r r r/ )r_ r· r ry Zymr Zxmr< r< r= r® . s
|
zTestTrapz.test_maskedN)rn ro rp r r r® r< r< r< r= r s %r c @ s e Zd Zdd Zdd ZdS )ÚTestSincc C s6 t tddk tt ddd¡}t|t|d d S )Nr rL rX r rk )r r. r6 r¯ r
|
r ©r_ rà r< r< r= r A s zTestSinc.test_simplec C sF ddg}t t |¡}t t|}t t|}t|| t|| d S )Nr rµ )r. r6 r rp rK r )r_ r· r r r r< r< r= Útest_array_likeG s
|
zTestSinc.test_array_likeN)rn ro rp r r£ r< r< r< r= r¡ ? s r¡ c @ s e Zd Zdd ZdS )Ú
|
TestUniquec C sÈ t dddddddddg ¡}tt t|dddddgk¡ ttt dddddg¡t dg¡k dddd ddg}tt t|d dddgk¡ t d
|
d d d d
|
g¡}tt t|d d d
|
d gk¡ d S )NrH rS rJ rL r ZwidgetZhamr$ r% y @ @rÝ y ð? $@r )r6 r r r r2 r" r< r< r= r R s ( zTestUnique.test_simpleN©rn ro rp r r< r< r< r= r¤ P s r¤ c @ s e Zd Zdd Zdd ZdS )ÚTestCheckFinitec C sR dddg}ddt jg}ddt jg}t j |¡ ttt jj| ttt jj| d S )NrL rJ rS )r6 Úinfrü ÚlibÚasarray_chkfiniter r\ r r< r< r= r ^ s
|
zTestCheckFinite.test_simplec C s2 dddg}t jj|dt jd}t|jt jk d S )NrL rJ rS r )r rÀ )r6 r¨ r© r¿ r rÀ rg r< r< r= Útest_dtype_orderf s
|
z TestCheckFinite.test_dtype_orderN)rn ro rp r rª r< r< r< r= r¦ \ s r¦ c @ s6 e Zd Ze dddgdddgddd gg¡Ze d
|
d d gd ddgdddgg¡Ze dddgdddgdddgg¡Ze ddddddgdddddd gdddd d d gddd dd d!gddd d dd"gdd d d!d"dgg¡Zd#d$ Z d%d& Z
|
d'd( Z d)d* Z d+d, Z d-d. Zd/d0 Zd1d2 Zej d3ejejejejg¡d4d5 Zd6S )7Ú TestCorrCoefgÿÁ°^³Ã?g.{§<Ç?g ¥$,Â?g¯ßá4æ?gpÖ)ßî?gÕJ¶ôGÜÑ?gY»Àí?g½¬«Ô?gáÞ0v©È?gØ©¢ º?gBÈ FUá?g ¢G)tàß?g3äTBâê?gLÁ\åè?gïÃAçÙ?g>
|
Îí?gqµFëE`å?g5Ø ¢?r¨ gÞ£¶î?gßÉár@©¿gp¾è]4Ó?g?Ä'LÓ?gÝë(÷Ð8å?gîê4V}à?gÕØw¨{¨¿g,H¨×5ì?gjS, úè?gªÀâÓóî¿gµ)h¤ïßæ?gå5#AÀê?gätÄçoà¿gB CÃæ,å¿g@§ÿ02vï?c C s. t t dddgdddg¡ddgddgg d S )Nr rL r¨ ç ð¿)r r6 r rx r< r< r= Útest_non_array s ÿzTestCorrCoef.test_non_arrayc C sd t | j}t|| j tt t |¡dk¡ t | j| j}t|| j tt t |¡dk¡ d S r* )
|
r rb r rº r r6 r r° rê r» )r_ Ztgt1Ztgt2r< r< r= r s
|
zTestCorrCoef.test_simplec C s t }t d¡ ttt| jdd | t¡ tt| jdd| j tt| j| j
|
dd| j tt| jdd| j tt| j| j
|
dd| j W 5 Q R X d S )Nr rX ©ZddofrS ) r r Ú simplefilterr ÚDeprecationWarningr rb Úfilterr rº rê r» ©r_ Úsupr< r< r= Ú test_ddof s
|
|
zTestCorrCoef.test_ddofc C sh t X}t d¡ ttt| j| jdd ttt| jdd | t¡ t t| jdd| j
|
W 5 Q R X d S )Nr rL r )Zbias) r r r¯ r r° r rb rê r± r rº r² r< r< r= Ú test_bias s
|
|
zTestCorrCoef.test_biasc C s^ t dddgdddgg¡}t|}t ddgddgg¡}t|| tt t |¡dk¡ d S © NrL rJ rS rù ù @r r¨ rÖ )r6 r r r r r r° )r_ r· r rO r< r< r= Ú test_complex¥ s
|
|
zTestCorrCoef.test_complexc C sL t dddgg¡}t dddgg¡}tt ||¡t ddgddgg¡ d S r¶ )r6 r r r ©r_ r· r r< r< r= Útest_xy¬ s zTestCorrCoef.test_xyc C s t jdd t dt¡ ttt g ¡tj ttt g ¡ dd¡t g ¡ dd¡ ttt g ¡ dd¡t tjtjgtjtjgg¡ W 5 Q R X d S ©NTr
|
r r rJ )
|
r r r¯ ÚRuntimeWarningr r r6 r rü rl rx r< r< r= Ú
|
test_empty± s ÿ ÿzTestCorrCoef.test_emptyc C sh ddgddgg}t jdd t|}W 5 Q R X t|t ddgddgg¡ tt t |¡dk¡ d S )Ng0ä.ÿ++g}Ã%I²TÚraise)r r¨ r¬ )r6 Zerrstater r
|
r r r r° )r_ r· r r< r< r= Ú test_extremeº s
|
zTestCorrCoef.test_extremeÚ test_typec C s* | j |¡}t||d}||jks&td S ©NrÓ )rb r½ r rÀ r¶ )r_ rÀ Zcast_Ar r< r< r= Útest_corrcoef_dtypeÁ s z TestCorrCoef.test_corrcoef_dtypeN)rn ro rp r6 r rb rê rº r» r r r´ rµ r¸ rº r½ r¿ rá râ rã ÚhalfÚsingleÚdoubleÚ
|
longdoubler r< r< r< r= r« m sJ þÿþÿþÿûÿ
|
r« c @ sh e Zd Ze ddgddgddgg¡jZe ddgddgg¡ZejdddgddZe dd dg¡Z e dgdgdgdgdgdgg¡jZ
|
e d
|
d gd d
|
gg¡Z ej d ej d Ze dddg¡Ze ddgddgg¡Ze d ¡Ze dddddg¡Zdd Zdd Zdd Zd d Zd d Zd d! Zd"d# Zd$d% Zd&d' Zd(d) Z e j d*ej ej!ej"ej#g¡d+d, Z$d-S ).ÚTestCovr rJ rL r¨ r¬ rª r§ )ÚndminrH çÙ?gÙ¿rS rÓ r¯ gUUUUUUå?gUUUUUUå¿g&S£:Ù?gÒo_Îã?g#Ûù~j¼è?g
ZÓ¼ï?gÎQÚ|é?c C s t t| j| j d S r5 )r r Úx1rº rx r< r< r= ra Õ s zTestCov.test_basicc C sZ t dddgdddgg¡}t ddgddgg¡}tt|| tt|t d¡d | d S )
|
NrL rJ rS rù r· r r¨ rÖ ©Úaweights)r6 r r r r] )r_ r· r r< r< r= r¸ Ø s zTestCov.test_complexc C sJ t dddgg¡}t dddgg¡}tt||t ddgddgg¡ d S r¶ )r6 r r r r¹ r< r< r= rº Þ s zTestCov.test_xyc C s t jdd t dt¡ ttt g ¡tj ttt g ¡ dd¡t g ¡ dd¡ ttt g ¡ dd¡t tjtjgtjtjgg¡ W 5 Q R X d S r» )
|
r r r¯ r¼ r r r6 r rü rl rx r< r< r= r½ ã s ÿ ÿzTestCov.test_emptyc C sZ t jddD t dt¡ tt| jddt tj tj gtj tj gg¡ W 5 Q R X d S )NTr
|
r rT r® )
|
r r r¯ r¼ r r rÊ r6 r r§ rx r< r< r= Útest_wrong_ddofì s ÿÿzTestCov.test_wrong_ddofc C sP t t| jt| jdd t dddddg¡}t t| j|t| j|dd d S )NF)Zrowvarg+ÙÎ÷³?g4·@âÓ?g¾Á&SË?gÒo_Ρ?gJê4ë?)r r Úx3r6 r ©r_ r r< r< r= Útest_1D_rowvaró s zTestCov.test_1D_rowvarc C s$ t t| jddtj| jdd d S )NrL r® )r r rÎ r6 Úvarrx r< r< r= Útest_1D_varianceø s zTestCov.test_1D_variancec C sÖ t t| j| jdt| j t t| j| jd| j t t| j| jd| j | jd }t t
|
t| j|d t j dt j d}t tt| j|d t j dt j d}t tt| j|d dt j dt j d }t tt| j|d d S )N)Úfweightsrµ ©rJ rS rÓ rJ rX rS )r r r Ú frequenciesÚ
|
x2_repeatsrÊ r» Úunit_frequenciesrº r r r6 r] r Ú RuntimeErrorr\ )r_ Znonintr¹ r< r< r= Ú test_fweightsû s" ÿÿÿ
|
zTestCov.test_fweightsc C s² t t| j| jd| j t t| jd| j dt| j| jd t t| j| jd| j t d¡}t t
|
t| j|d t d¡}t t
|
t| j|d dt d¡ }t t t| j|d d S )NrË r© rÔ rJ r¬ rS ) r r rÊ r´ r¼ Ú unit_weightsrº r6 r] r rØ r\ r¢ r< r< r= Ú test_aweights s ÿ
|
|
zTestCov.test_aweightsc C s¾ t t| j| j| jdt| j t t| j| j| jd| j t t| j| j| jd| j t t| j| j| j
|
d| j t t| j| jd| j
|
dt| j| j
|
d t t| j| j| jd| j d S )N)rÓ rÌ r© rË ) r r r rÕ rÚ rÖ rÊ r» r× rº r´ r¼ rx r< r< r= Ú test_unit_fweights_and_aweights s< ÿþ ÿþ ÿþ ÿþ ÿþ ÿþz'TestCov.test_unit_fweights_and_aweightsrÀ c C s* | j |¡}t||d}||jks&td S rÁ )rÊ r½ r rÀ r¶ )r_ rÀ Zcast_x1r r< r< r= Útest_cov_dtype+ s zTestCov.test_cov_dtypeN)%rn ro rp r6 r r rÊ rº r rÕ rÖ r» r] r r× r´ r¼ rÚ rÎ ra r¸ rº r½ rÍ rÐ rÒ rÙ rÛ rÜ rá râ rã rà rÄ rÅ rÆ rÝ r< r< r< r= rÇ È s. $
|
rÇ c @ s$ e Zd Zdd Zdd Zdd ZdS )ÚTest_I0c
|
C sê t tdt d¡ t dddddg¡}t dd d
|
d d g¡}t t|| t t| | t d dgddgddgddgddgg¡}t t|t ddgddgdd gd d gd d gg¡ t d!g¡}t|jd" tt d!g¡t d#g¡ d S )$Nrµ gÿÇ#ñ?gCê¬7æß?gùmãªMæ?g¡3aâ,Ì?gª$WV?Û?r3 gÖ¿N^ñ?gE á¿ ò?gé¡Ä1ð?gÔ%©m? ð?gDíénÿ¥@g>@÷åÌvê?gÏu¯Í¥üï?gVºX¤Ë³ì?g÷ua&Ù?gB- ¬sJØ?g-^ þ§ª?g§_¿³V×?g ¨X±Ï.ç?gVØëaXÓÞ?gõj ¨à?gpÁÛÚò?g~-ØÛ@ô?g¦ ¼D/bó?gݦ¬ð?g,êY1Øð?g[!¬Æð?gQÐLð?gWè¤sU+ò?g÷ ±@ñð?gfx»wñ?rª rÝ r¨ )r r# r6 r r rA r )r_ rb rs rê Zi0_0r< r< r= r 4 s4 þ
|
ü
|
üþ zTest_I0.test_simplec sV t d¡ G fddd}tt | ¡|s2tt ¡}t | ¡}t|| d S )NrH c s e Zd Z jZdd ZdS )z*Test_I0.test_non_array.<locals>.array_likec S s | S r5 r< )r_ ræ r< r< r= Ú__array_wrap__V s z9Test_I0.test_non_array.<locals>.array_like.__array_wrap__N)rn ro rp Z__array_interface__rß r< rõ r< r= Ú
|
array_likeS s rà )r6 r7 r r° r¶ r# r )r_ rà rk r r< rõ r= r P s
|
|
zTest_I0.test_non_arrayc C s4 t ddg¡}tjtdd t|}W 5 Q R X d S )Nr y ð? @z#i0 not supported for complex valuesr. )r6 r rá r r r# )r_ r` r r< r< r= r¸ ` s zTest_I0.test_complexN)rn ro rp r r r¸ r< r< r< r= rÞ 2 s rÞ c @ s e Zd Zdd Zdd ZdS )Ú
|
TestKaiserc C s t t tdd¡ ttddt g ¡ ttddt ddg¡ ttddt dddddg¡ ttddt d d
|
dd
|
d g¡ d S ) NrL r¨ r rJ g öoFé?rT gÀvT¶=î?ggÓù?gø½¦â?g÷ñY
Jì?)r r6 Úisfiniter& r r rx r< r< r= r h s&
|
ÿ
|
ÿ
|
|
ÿÿ
|
|
ÿÿzTestKaiser.test_simplec C s t dd d S )NrS rH )r& rx r< r< r= Ú test_int_betau s zTestKaiser.test_int_betaN)rn ro rp r rã r< r< r< r= rá f s rá c @ s e Zd Zdd ZdS )Ú TestMsortc
|
C sl t dddgdddgddd gg¡}tjtd
|
d 2 tt|t ddd gdddgdddgg¡ W 5 Q R X d S ) Ngî\é
Ü?gb¥ÝC Qé?g×sá?gwø×?g¥ ¹klã×?gø*dtaÀî?gÐèÓ¯Áä?g6Hòïà?gÜC÷Ù?zmsort is deprecatedr. )r6 r rá Zwarnsr° r r( )r_ rb r< r< r= r { s þ þþzTestMsort.test_simpleNr¥ r< r< r< r= rä y s rä c @ sd e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z dS )Ú TestMeshgridc
|
C s t dddgddddg\}}t|t dddgdddgdddgdddgg¡ t|t dddgdddgdddgdddgg¡ d S )NrL rJ rS rH rT rj rk ©r' r r6 r ©r_ ÚXÚYr< r< r= r s ýýzTestMeshgrid.test_simplec C s. t ddddg\}t|t ddddg¡ d S )NrL rJ rS rH ræ )r_ rè r< r< r= Útest_single_input s z TestMeshgrid.test_single_inputc C s* g }t g t| t g t|ddi d S )Nr F)r r' r] r< r< r= Ú test_no_input s zTestMeshgrid.test_no_inputc
|
C s dddg}ddddg}t ||dd \}}t|t ddddgddddgddddgg¡ t|t ddddgddddgddddgg¡ d
|
d g}tt ||d jd k tt ||dd d jdk tt |||d jdk tt |||dd d jdk ttt ||dd d S )NrL rJ rS rH rT rj rk Úij©Zindexingri rå r rö ©rS rH )rH rS rJ )rS rH rJ Znotvalid)r' r r6 r r rA r r\ )r_ r· r rè ré rÜ r< r< r= Ú test_indexing s"
|
|
|
þ
|
|
þ zTestMeshgrid.test_indexingc C s\ t dddgddddgdd \}}t|t dddgg¡ t|t dgdgdgdgg¡ d S )
|
NrL rJ rS rH rT rj rk T©Úsparseræ rç r< r< r= Ú test_sparse± s zTestMeshgrid.test_sparsec C s$ t ttdddgddddgdd d S )
|
NrL rJ rS rH rT rj rk rì )r
)r r r' rx r< r< r= Útest_invalid_arguments¶ s
|
|
ÿz#TestMeshgrid.test_invalid_argumentsc C sÀ t jddt jd}t jddt jd}t ||¡\}}t|j|jk t|j|jk t j||dd\}}t|j|jk t|j|jk t j||dd\}}t|j|jk t|j|jk d S )Nr r rÓ r) T)r rð )r6 r7 r¾ r¿ r' r rÀ )r_ r· r rè ré r< r< r= r~ ½ s z TestMeshgrid.test_return_typec C sr t ddg¡}t ddg¡}t j||ddd\}}d|dd d
f< t|dd d
f d t|d d d
f | d S )
|
Nrb rc gffffff
|
@g@FT)rñ r r rL )r6 r r' r )r_ rè ré r· r r< r< r= Útest_writebackÕ s zTestMeshgrid.test_writebackc C sh t jdd tddD \}}}}}d}t|j| t|j| t|j| t|j| t|j| d S )Nc s s | ]}d g| V qdS )r Nr< ©rÛ r r< r< r= rá à s z-TestMeshgrid.test_nd_shape.<locals>.<genexpr>rL rj )rJ rL rS rH rT )r6 r' r^ r rA )r_ r` r{ r rþ r? Zexpected_shaper< r< r= Ú test_nd_shapeß s $ zTestMeshgrid.test_nd_shapec C s~ t dgddgdddg¡\}}}t|dddggdddggg t|dddggdddggg t|dddggdddggg d S )Nr rL rJ rS rH rT ©r6 r' r r r< r< r= Útest_nd_valuesè s zTestMeshgrid.test_nd_valuesc C s| t jdgddgdddgdd\}}}t|dddgdddggg t|dddgdddggg t|dddgdddggg d S ) Nr rL rJ rS rH rT rì rí r÷ r r< r< r= Útest_nd_indexingî s $ z TestMeshgrid.test_nd_indexingN)rn ro rp r rê rë rï rò ró r~ rô rö rø rù r< r< r< r= rå s
|
rå c @ sT e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dS )Ú TestPiecewisec C s6 t ddgddgdg}t|ddg t ddgddggdg}t|ddg t ddgt ddg¡dg}t|ddg t ddgt ddg¡dg}t|ddg t ddgt ddg¡gdg}t|ddg t ddgddggdd g}t|ddg ttdt ddgddggg ttdt ddgddggdd d
|
g d S ) Nr TFrL c S s dS rO r< rm r< r< r= rö
|
r÷ z+TestPiecewise.test_simple.<locals>.<lambda>rX z 1 or 2 functions are expectedrJ rS )r) r r6 r r r\ r" r< r< r= r ÷ s0 ÿ ÿzTestPiecewise.test_simplec C s2 t ddgddgddggddg}t|ddg d S )NrL rJ TFrS rH ©r) r r" r< r< r= Útest_two_conditions
|
s z!TestPiecewise.test_two_conditionsc C s& t ddddgdddg}t|d d S )NrS TFrH rJ r )r) r r" r< r< r= Ú$test_scalar_domains_three_conditions
|
s z2TestPiecewise.test_scalar_domains_three_conditionsc C sN t ddgddgdg}t|ddg t ddgddgddg}t|ddg d S )NrL rJ TFr rS rû r" r< r< r= Ú test_default
|
s zTestPiecewise.test_defaultc C s t d¡}t||dkddg}t|jdk t|dk d}t|ddgddg}t|jdk t|dk t|dddgdddg}t|d d S ) NrS rH r rT TFrL rJ )r6 r r) r r
r r¹ r< r< r= r %
|
s
|
zTestPiecewise.test_0dc
|
C s® d}t ||dk|dkgddg}t|d d}t ||dk|dk|dk |dkgdddg}t|d ttdt ||dk|dkgdg ttdt ||dk|dkgddddg d S )NrS rH r rT rL rJ z 2 or 3 functions are expected)r) r r r r\ r¹ r< r< r= Útest_0d_comparison4
|
s$
|
,
|
ÿ
|
ÿz TestPiecewise.test_0d_comparisonc C s8 t d¡}t |dk¡}t||gddg}t|d d S )NrS rL rJ )r6 r r) r )r_ r· r r r< r< r= Útest_0d_0d_conditionC
|
s
|
z"TestPiecewise.test_0d_0d_conditionc C s\ t dddgdddgg¡}t||dk |dkgd d
|
d g}t|t d d d gd d dgg¡ d S )Ng Àg ø¿r· rµ r r¶ r rJ rX rL rS r¬ r© r¨ )r6 r r) r r¹ r< r< r= Ú test_multidimensional_extrafuncI
|
s ÿ ÿz-TestPiecewise.test_multidimensional_extrafuncc C sf G dd dt j}t d¡ |¡}t||dk |dkgdddg}tt|| t|dddddg d S ) Nc @ s e Zd ZdS )z/TestPiecewise.test_subclasses.<locals>.subclassNrÊ r< r< r< r= rË Q
|
s rË r° r§ rH r¬ r¨ rª )r6 rÌ r7 rÍ r) r rÎ )r_ rË r· r r< r< r= rÏ P
|
s
|
z TestPiecewise.test_subclassesN) rn ro rp r rü rý rþ r rÿ r r rÏ r< r< r< r= rú õ s rú c @ s e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z e jje dddd Ze j dddggdg¡dd Zd S ) Ú TestBincountc C s$ t t d¡¡}t|t d¡ d S )NrH )r6 Úbincountr7 r r] rÏ r< r< r= r [
|
s zTestBincount.test_simplec
|
C s: t t dddddg¡¡}t|t ddddddg¡ d S )NrL rT rJ rH r )r6 r r r rÏ r< r< r= Ú test_simple2_
|
s zTestBincount.test_simple2c C s6 t d¡}t ddddg¡}t ||¡}t|| d S )NrH çÉ?ç333333Ó?rµ re )r6 r7 r r r ©r_ r· rà r r< r< r= Útest_simple_weightc
|
s
|
z TestBincount.test_simple_weightc
|
C sT t dddddg¡}t dddddg¡}t ||¡}t|t d ddd ddg¡ d S )
|
NrL rJ rH rT r r rµ re r ©r6 r r r r r< r< r= Útest_simple_weight2i
|
s z TestBincount.test_simple_weight2c C s^ t dddddg¡}t j|dd}t|t dddg¡ g }t j|dd}t|t g ¡ d S )Nr rL rS ©Z minlengthrJ r r¹ r< r< r= Útest_with_minlengtho
|
s z TestBincount.test_with_minlengthc C sh t dddddddg¡}t j|dd}t|t ddddg¡ t j|dd}t|t ddddg¡ d S )Nr rL rJ rS r r r¹ r< r< r= Ú)test_with_minlength_smaller_than_maxvaluew
|
s
|
z6TestBincount.test_with_minlength_smaller_than_maxvaluec C sZ t dddddg¡}t dddddg¡}t ||d ¡}t|t d
|
ddd
|
ddd
|
d
|
g¡ d S ) NrL rJ rH rT r r rµ re ri r r r r< r< r= Ú test_with_minlength_and_weights~
|
s z,TestBincount.test_with_minlength_and_weightsc C s& t jg td}t |¡}t|| d S rÁ )r6 r rY r r r¹ r< r< r= r½
|
s
|
zTestBincount.test_emptyc C s4 t jg td}t j|dd}t|t jdtd d S )NrÓ rT r )r6 r rY r r r? r¹ r< r< r= Útest_empty_with_minlength
|
s z&TestBincount.test_empty_with_minlengthc sl t jg td ttd fdd ttd fdd t d¡ ttd fdd ttd fd d d S )
|
NrÓ z"'str' object cannot be interpretedc s t j ddS ©NÚfoobarr ©r6 r r< rm r< r= rö
|
r÷ z<TestBincount.test_with_incorrect_minlength.<locals>.<lambda>zmust not be negativec s t j ddS ©NrX r r r< rm r< r= rö
|
r÷ rT c s t j ddS r r r< rm r< r= rö
|
r÷ c s t j ddS r r r< rm r< r= rö
|
r÷ )r6 r rY r r r\ r7 rx r< rm r= Ú test_with_incorrect_minlength
|
s$
|
þ
|
þ
|
|
þ
|
þz*TestBincount.test_with_incorrect_minlengthrc rd c C sÌ t t tj¡¡}t t tj¡¡}tdD ]}t dddg¡ q,tt t tj¡¡| tt t tj¡¡| tdD ] }t dddgdddg¡ qztt t tj¡¡| tt t tj¡¡| d S )Nr rL rJ rS rH rT rj ) ri rj r6 rÀ ZintprÅ r^ r r )r_ Z intp_refcountZdouble_refcountÚjr< r< r= Útest_dtype_reference_leaks
|
s z'TestBincount.test_dtype_reference_leaksÚvalsrJ c C sJ t |¡}tt t |¡ W 5 Q R X tt t |¡ W 5 Q R X d S r5 )r6 Úasarrayr r\ r )r_ r Zvals_arrr< r< r= Útest_error_not_1d¯
|
s
|
|
|
|
z TestBincount.test_error_not_1dN)rn ro rp r r r r
|
r r r r½ r r rá râ rn r r rã r r< r< r< r= r Y
|
s
|
r c @ s² e Zd Zdd Zdd Zdd Zdd Zd d
|
Zej d d d d dd dd gddddgddd Z
|
dd Z dd Z dd Z d d Zd d Zd!d" Zd#d$ Zd%d& Zd'd( Zd)S )*Ú
|
TestInterpc C st t ttdg g t ttddgddg t ttdddgddgdd t ttdg g dd t ttddgddgdd d S )Nr rL rJ r éh )r r\ r% rx r< r< r= Útest_exceptions»
|
s
|
zTestInterp.test_exceptionsc C sB t ddd¡}t ddd¡}t ddd¡}tt |||¡| d S )Nr rL rT rb )r6 r¯ r r% ©r_ r· r Úx0r< r< r= ra Â
|
s zTestInterp.test_basicc
|
C s¼ t ddD ]ª}tj|tjd}tj|tjd}tjdd|d |gtjd}|d d d
}t|||}t|||}tjddddgtd}|d d d
} t|| t|| t|||dd}t|||dd}tjddddgtd}|d d d
} t|| t|| t|||dd}t|||dd}tjddddgtd}|d d d
} t|| t|| t|||ddd }t|||ddd }tjddddgtd}|d d d
} t|| t|| q
|
d S )
|
NrL r rÓ rX r ©ÚleftrJ ©Úright)r r" ) r^ r6 r7 rÅ r] r r% r+ r )
|
r_ ÚsizeÚxpZypZincptsZdecptsZincresZdecresZinctgtZdectgtr< r< r= Útest_right_left_behaviorÈ
|
s:
|
|
|
|
|
|
|
z#TestInterp.test_right_left_behaviorc C s¦ t ddd¡}t ddd¡}d}tt |||¡| d}tt |||¡| t d¡}tt |||¡| t d¡}tt |||¡| t j}tt |||¡| d S )Nr rL rT r )r6 r¯ r r% r¾ r¿ rü r r< r< r= Ú test_scalar_interpolation_pointî
|
s
|
|
z*TestInterp.test_scalar_interpolation_pointc C s~ dddddg}ddddg}ddt jdg}tt |||¡ddt jt jdg ddt jdg}tt |||¡ddt jt jdg d S )NrL rJ r¶ rS rH )r6 r§ r r% rü )r_ r· r$ Úfpr< r< r= Ú test_non_finite_behavior_exact_xü
|
s "z+TestInterp.test_non_finite_behavior_exact_xc C s
|
t | ¡S r5 )r6 Zfloat_rm r< r< r= rö r÷ zTestInterp.<lambda>c C s
|
t | dS r ©rF rm r< r< r= rö r÷ c C s
|
t d| S r r) rm r< r< r= rö r÷ c C s t | t | d¡S )NrR )rF r6 Úmultiplyrm r< r< r= rö r÷ rC z complex-realz complex-imagz complex-both)ÚparamsÚidsc C s |j S )z( scale function used by the below tests )Úparam)r_ Úrequestr< r< r= Úsc s z TestInterp.scc C s¤ t t dtjdg|ddg¡|tj t t ddtjg|ddg¡|tj t t dddg|tjdg¡|tj t t dddg|dtjg¡|tj dS )z test that nans are propagated rµ rL r r N)r r6 r% rü ©r_ r/ r< r< r= Útest_non_finite_any_nan s (((z"TestInterp.test_non_finite_any_nanc C sº t t dtj tj
|
g|ddg¡|tj t t dddg|tj tj
|
g¡|tj t t dddg|tj
|
tj g¡|tj t t dtj tj
|
g|ddg¡|d dS )z2 Test that interp between opposite infs gives nan rµ r r rL N©r r6 r% r§ rü r0 r< r< r= Útest_non_finite_inf s ...z TestInterp.test_non_finite_infc C st t t dtj dg|tj dg¡|tj t t dtj dg|tj
|
dg¡|tj t t dtj dg|dtj g¡|tj t t dtj dg|dtj
|
g¡|tj t t ddtj
|
g|tj dg¡|tj t t ddtj
|
g|tj
|
dg¡|tj t t ddtj
|
g|dtj g¡|tj t t ddtj
|
g|dtj
|
g¡|tj dS )z@ Test that interp where both axes have a bound at inf gives nan rµ rL r r Nr2 r0 r< r< r= Útest_non_finite_half_inf_xf# s .......z&TestInterp.test_non_finite_half_inf_xfc C s¬ t t dtj tj g|ddg¡|d t t dtj dg|ddg¡|d t t ddtj
|
g|ddg¡|d t t dtj
|
tj
|
g|ddg¡|d dS )z1 Test interp where the x axis has a bound at inf rµ r r rL N©r r6 r% r§ r0 r< r< r= Útest_non_finite_half_inf_x. s ,((z%TestInterp.test_non_finite_half_inf_xc C s t t dddg|dtj g¡|tj t t dddg|dtj
|
g¡|tj
|
t t dddg|tj dg¡|tj t t dddg|tj
|
dg¡|tj
|
t t dddg|tj tj g¡|tj t t dddg|tj
|
tj
|
g¡|tj
|
dS )z1 Test interp where the f axis has a bound at inf rµ r rL r Nr5 r0 r< r< r= Útest_non_finite_half_inf_f5 s ,,,,0z%TestInterp.test_non_finite_half_inf_fc C sH t ddd¡}t ddd¡dt ddd¡ d }d}|d| d }tt |||¡| d}d}tt j||||d| d }d}tt j||||d
|
| dd d d dg}dd d dg}ddt jdg}ddt jd t jdg}tt |||¡| ddddddddg}ddddg}d d d d g}d d d!d"d d#d$d%g}tt j|||d&d'| d S )(Nr rL rT rù r rX ù @ @r r§ r! rJ r¶ rS rH ù @ ð?y à?éLÿÿÿéVÿÿÿéGÿÿÿé¹ r r2 ém é¾ éBÿÿÿé^ é¢þÿÿy @ ð?y $@ @y @ @y @ @y @ ø?y !@ ü?y @ ô?y
|
@
|
@y @ @y @ @r r )r6 r¯ r r% r§ ) r_ r· r r Zy0r r" r$ r' r< r< r= Útest_complex_interp> s0 $ ÿz TestInterp.test_complex_interpc C s¾ t ddd¡}t ddd¡}t d¡}tt |||¡| t dddg¡}t dddg¡}t t d¡||¡}t|d tt|t j t jt d¡||dd }t|d
|
tt|t j d S ) Nr rL rT r rJ rH rX r r rµ ) r6 r¯ r r r% r r r r¿ )r_ r· r r r$ r' rÄ r< r< r= Ú)test_zero_dimensional_interpolation_pointZ s
|
|
|
z4TestInterp.test_zero_dimensional_interpolation_pointc C s2 t ddd¡}t |¡}tt t j||¡d d S )Nr r g-C ëâ6?rª )r6 r7 Úsinr r% rÛ )r_ r$ r' r< r< r= Útest_if_len_x_is_smallk s
|
z!TestInterp.test_if_len_x_is_smallc C s ddddddddg}d d
|
d d g}d dddg}ddddddddg}t tj|||dd| tj|dd d d ¡}tj|d d d d ¡}t tj|||dd| d S ) Nr: r; r< r= r r2 r r> r? r@ rA rB rT r rS rH g @r° ç !@g @r© g
|
@ç @g @r r r r rJ rX rç )r r6 r% r rl )r_ r· r$ r' r r< r< r= r p s zTestInterp.test_periodN)rn ro rp r ra r% r& r( rá Zfixturer/ r1 r3 r4 r6 r7 rC rD rF r r< r< r< r= r ¹
|
s6 &üüû
|
r c @ s e Zd Zdd Zdd Zdd Zdd Zej d e
|
j d
|
¡d d Z d d e
|
j d D e
|
j e
|
j fe
|
je
|
jfe
|
je
|
jfe
|
je
|
jfe
|
je
|
jfe
|
je
|
jfe
|
je
|
jfe
|
d¡e
|
jfg Zej ddge¡ej ddgdddddddd d g ¡d d Ze
|
j d e
|
j d
|
d Zej d e¡d d! Zej d e¡d"d# Zej d e¡d$d% Zd&d' Zd(d) Z d*d+ Z d,d- Z d.d/ Z d0d1 Z d2d3 Z!d4d5 Z"d6d7 Z#d8d9 Z$d:d; Z%d<d= Z&d>d? Z'd@dA Z(ej dBdCdDdCgg¡ejj dEdFdDdGdHdIgdJdKdL Z)dMdN Z*dOdP Z+dQdR Z,dSdT Z-dFS )UÚTestPercentilec C s~ t d¡d }tt |d¡d tt |d¡d tt |d¡d t j|d < tt |d¡t j tt j|dd
|
d t j d S ) Nri rµ r rª r rH rb r¬ rL Únearest©Úmethod)r6 r7 r Ú
|
percentilerü r" r< r< r= ra } s
|
zTestPercentile.test_basicc C sØ dd t dD }t |td¡}t|td tt|t t |td¡}t|tdd tt|t t |td¡}t|tdd tt|t t |tdg¡}t|t tdd g¡ tt|tj d S )
|
Nc S s g | ]}t |d qS ©rJ r rõ r< r< r= rÜ s z0TestPercentile.test_fraction.<locals>.<listcomp>ri r r rk rJ rb rH )r^ r6 rM r r rÎ r rÌ )r_ r· r& r< r< r= Ú test_fraction s z TestPercentile.test_fractionc C sR t d¡}t |dd d d¡ t |dd d dd¡ t d¡}t |dd |dd¡ d S )NrT FÚlinearrÝ )r6 r] rM )r_ rþ Úor< r< r= Útest_api s
|
|
|
zTestPercentile.test_apic C sR t dddgdddgdddgdddgdddgg¡}tt j|ddddddg d S )NrL rH rS rb r rr )r6 r r rM r" r< r< r= Útest_2D s üzTestPercentile.test_2DrÀ ZAllFloatc C sP t jdt jdddg|d}t j|ddd}t j |t j¡ t j |j|j¡ d S )Nr8 ç A@ç D@ç I@rÓ rP rK )r6 r ZNANrM Útestingr rÀ )r_ rÀ ræ r r< r< r= Útest_linear_nan_1D¨ s ýz!TestPercentile.test_linear_nan_1Dc C s g | ]}|t jfqS r< )r6 r¿ )rÛ Zint_typer< r< r= rÜ ³ s ÿzTestPercentile.<listcomp>r r Ú input_dtypeÚexpected_dtyperL rs )Ú inverted_cdfr) )Úaveraged_inverted_cdfç ;@)Úclosest_observationr) )Úinterpolated_inverted_cdfr) )Úhazenr] )Úweibullé )rP rj )Úmedian_unbiasedrA )Únormal_unbiasedg ;@c C sÊ t |¡}t ¡ dkr$t |t j¡}t jdddddg|d}t j|d|d}t j || |¡d ¡ |d
|
krª|d krt j
|
t |¡jt j¡ qÆt j
|
t |¡jt |¡¡ n t j
|
t |¡jt |¡¡ d S ) NÚlegacyr8 g 4@rT rU rV rÓ rK r? )r[ r^ r ) r6 rÀ Z_get_promotion_stateZ promote_typesr¿ r rM rW r rÎ r )r_ rL rs rY rZ ræ rÄ r< r< r= Útest_linear_interpolation¾ s&
|
ÿÿÿz(TestPercentile.test_linear_interpolationc C sD t tjtjd|ddddd t tjtjd|ddddd d S ) Nr rÓ rb ÚlowerrK rH ÚhigherrT ©r r6 rM r7 ©r_ rÀ r< r< r= Útest_lower_higherå s ÿÿÿÿz TestPercentile.test_lower_higherc C s t tjtjd|ddddd t tjtjd|dd d ddd
|
t tjtjd |ddddd t tjtjd |dd ddd
|
d S ) Nr rÓ é3 ÚmidpointrK r rå rL rb rT r® ç @ri rj r< r< r= Ú test_midpointì s( ÿÿÿÿÿÿÿÿz TestPercentile.test_midpointc C sD t tjtjd|ddddd t tjtjd|ddddd d S ) Nr rÓ rl rJ rK rT é1 rH ri rj r< r< r= Ú test_nearest÷ s ÿÿÿÿzTestPercentile.test_nearestc C sL t j d¡}t |d¡}t j || ¡ ¡ t |d¡}t j || ¡ ¡ d S )NrT r r )r6 r± r rM rW r rr rÑ )r_ ræ rÄ r< r< r= Ú'test_linear_interpolation_extrapolationþ s
|
z6TestPercentile.test_linear_interpolation_extrapolationc C s0 t d¡d }tt |dddg¡dddg d S )Nri rµ r r rb rH r¬ )r6 r7 r rM r" r< r< r= Ú test_sequence s z TestPercentile.test_sequencec C s t d¡ dd¡}tt |d¡dddg dddd gdd d
|
d gd d ddgg}tt j|ddd| dddgddd gdddgg}tt j|dddt |¡j t d¡ ddd d
|
¡}tt |d¡jd tt |d¡jd tt j|dddjd tt j|dddjd tt j|dddjd tt j|dddjd tt j|dddjd! tt j|dd"d#jd tt j|dd"d#jd tt j|ddd"d$jd tt j|ddd"d$jd tt j|ddd"d$jd tt j|ddd"d$jd tt j|ddd"d$jd! d S )%Nrõ rS rH )r rb r g @rn r rJ rT rj rk ri rå r r® r rr ç è?r ç @rG r± rL r ©r rb rN )r rb r ©rS )rJ rH rT rj )rJ rS rT rj )rJ rS rH rj )rJ rS rH rT )rS rS rT rj rh rK ©rs rL )r6 r7 rl r rM r r rA )r_ r· Úr0r r< r< r= rl sh " ÿ
|
ÿÿ
|
ÿÿ ÿÿ ÿÿ ÿÿ ÿÿ ÿÿzTestPercentile.test_axisc C s t d¡ dd¡}tt |d¡d tt t |d¡¡ t dddd g¡}tt j|dd
|
d | tt j|dd
|
d j|j t d dd g¡}t t j|ddd | tt j|ddd j|j t
|
d¡}tt j|d|dd t|d t
|
d¡}tt j|dd
|
|d| t|| t
|
d¡}tt j|dd|d| t|| t d¡ dd¡}tt j|dddd tt t |d¡¡ t dddd g¡}t j|ddd
|
d}t|| t|j|j t dddg¡}t j|dddd}t || t|j|j t j
|
d|j d}t j|dd|d}t|d t|d t j
|
d|j d}t j|ddd
|
|d}t|| t|| t j
|
d|j d}t j|ddd|d}t|| t|| d S )Nrõ rS rH rb rn r¯ r° r» r9 r rr r r± rL ©rf ©rs rf rg rK )rL rs r¨ r4 r< rÓ ©rL rf rT ©rL rs rf ) r6 r7 rl r rM r Úisscalarr rA r ÚemptyrÀ )r_ r· ry r rf Zc0Zc1r r< r< r= Ú test_scalar_q/ sR
|
|
|
|
|
|
|
|
|
|
|
|
|
z TestPercentile.test_scalar_qc C s t ttjddgddd t ttjdgd t ttjdgd t ttjdgttddg t ttjdgttdd g d S )
|
NrL rJ é8 r rK ée rX rb rd )r r\ r6 rM rp r^ rx r< r< r= Útest_exception_ s ÿ z TestPercentile.test_exceptionc C s t t dddgd¡d d S )NrL rJ rS r )r r6 rM rx r< r< r= Útest_percentile_listg s z#TestPercentile.test_percentile_listc C s t dddg¡}t d¡}d}t j|||d tt ||¡| t dddgddd gg¡}t d
|
¡}t j||d |d tt j||d d | t d¡}t j||d|d tt j||dd | t d¡ dd¡}t ddddgddddgg¡}t d¡}tt j|dd |d | t|| t dddgdd d gg¡}t d ¡}tt j|dd|d | t|| t d dddgddd d gg¡}t jd|jd }t j|dd!d |d"}t|| t|| t d dd#gddd$gg¡}t jd |jd }t j|dd!d|d"}t|| t|| d S )%NrL rJ rS rw ©rL rJ rS rz rH rT rj r{ r r{ rr ©rS rJ rõ r§ r© r¯ r° r» r9 )rJ rH rv rt ru rG r rn r± rÔ rk rÓ rg r} ri rå ) r6 r r? rM r r7 rl r rÀ )r_ r· r r& ry rf r r r< r< r= Útest_percentile_outj sB
|
ÿ
|
|
|
|
|
|
|
|
|
z"TestPercentile.test_percentile_outc C s t d¡ dddd¡}tt j|dddjd tt j|dddjd tt j|dddjd
|
tt j|dd djd tt j|dd djd tt j|dddjd
|
tt j|dddjd tt j|dddjd tt j|ddddjd
|
tt j|ddddjd
|
tt t j|ddgdd¡jd tt t j|ddgdd¡jd tt t j|ddgdd¡jd tt t j|ddgd d¡jd d S )NrC r® rL rJ rb r rr )rL rJ rL )r® rJ rL )r® rL rL rS )r® rL rJ rX rR rU r rm rx r )rJ rL rJ rL )rJ r® rJ rL )rJ r® rL rL )rJ r® rL rJ )r6 r7 rl r rM rA r ©r_ rþ r< r< r= Útest_percentile_empty_dim s> ÿþ ÿþ ÿ ÿ ÿ ÿz(TestPercentile.test_percentile_empty_dimc C sx t ddddg¡}t j|dgdd t|t ddddg¡ t ddddg¡}t |dg¡ t|t ddddg¡ d S )NrJ rS rH rL rb F©Úoverwrite_input©r6 r rM r rg r< r< r= Ú test_percentile_no_overwrite¯ s z+TestPercentile.test_percentile_no_overwritec C s t jdddd}t jt d¡|dd t|t jdddd t jdddd ¡ }t jt d¡|dd t|t jdddd ¡ d S )Nrª ç Y@rT )Únumrm rK )r6 r¯ rM r7 r rñ )r_ r& r< r< r= Útest_no_p_overwrite¸ s z"TestPercentile.test_no_p_overwritec C sf t ddddg¡}t j|dgdd}t|t dg¡ t jddddgdgdd}t|t dg¡ d S ) NrJ rS rH rL rb Tr r¶ r rz r< r< r= Útest_percentile_overwriteÀ s
|
z(TestPercentile.test_percentile_overwritec
|
C sÔ t jjdd}t |gd ¡}tt j|dddt |d¡ t |dd¡}tt j|dd dt |d¡ | dd
|
¡ ¡ }tt j|dd dt |d¡ | dd
|
¡ ¡ }tt j|d d gddt j|d d gd d tt j|d d gddt j|d d gdd t d¡
|
d¡}t j | ¡ ¡ tt j|d ddd t |d d
d d
d d
df ¡ d ¡ tt j|ddgddd d
d
|
f t |d d
d d
d
|
d d
f ¡ ddg¡ tt j|d ddd t |d d
d d
dd d
f ¡ d ¡ tt j|d ddd t |dd d
d d
d d
f ¡ d ¡ tt j|d ddd t |dd
|
d d
d d
f ¡ d ¡ tt j|d ddd t |dd d
d d
d
|
f ¡ d ¡ tt j|d ddd t |dd d
dd d
f ¡ d ¡ d S ) N©éG é ©r# r r; rY rr rX r rf rL rd r r rK ©r é ©rS rT rk r® éZ ©r rL rS ©rS rL r rJ ©rS rL rJ r ©rJ rL ©rL rR ©rL rS rO )r6 r± ÚnormalÚdstackr rM Úmoveaxisr r r7 rl ÚshuffleÚravelÚflatten©r_ rQ r· rþ r< r< r= Útest_extended_axisÈ sJ ÿÿ&ÿ *ÿ&ÿ&ÿ"ÿ"ÿ"ÿz!TestPercentile.test_extended_axisc C s¢ t d¡}tt jt j|ddd tt jt j|ddd tt jt j|ddd tt jt j|ddd ttt j|ddd ttt j|d dd ttt j|d
|
dd d S ) Nr r2 r )rs r ©r r2 rH ©r rH rQ )rX rX )rS rX )r6 r] r rv rM r\ r r< r< r= Útest_extended_axis_invalidè s
|
z)TestPercentile.test_extended_axis_invalidc C sæ t d¡}tt j|dd ddjd tt j|ddddjd tt j|ddddjd tt j|dd
|
ddjd tt j|dd dd jd tt j|ddddjd tt j|ddgdddjd tt j|ddgdddjd d S )Nr rk Tr² ©rL rL rL rL rY ©rL rL rk r® ru ©rL rT rk rL rÝ ©rS rL rk r® ©r rL rJ rS )r³ r ©rL rL rk rL rL )rJ rL rL rk rL )rJ rL rT rk rL )r6 r] r rM rA r r< r< r= Ú test_keepdimsó s:
|
ÿÿÿÿÿÿÿÿÿÿz TestPercentile.test_keepdimsr rk rL rs NrÝ rY ©rU rX ©ÚargnamesZ argvaluesc s t d¡|d kr dj }n*t|j t fddtjD }t |¡| }t |¡}t j||d|d}||kst t
|
|j| d S )Nr rÝ c 3 s$ | ] }| krd nj | V qdS ©rL N©rA rõ ©Z axis_normrþ r< r= rá s z3TestPercentile.test_keepdims_out.<locals>.<genexpr>T©rs r³ rf ) r6 r] r
r4 rK r^ rA r rM r¶ r )r_ r rs Ú shape_outrf r} r< r· r= Útest_keepdims_out s
|
ÿ
|
z TestPercentile.test_keepdims_outc C s¾ t d¡}t d¡}tt j|dd|d| tt j|ddd|d| t d¡}tt j|dd|d| tt j|ddd|d| t d ¡}tt j|d
|
|d| tt j|d
|
d|d| d S ) N©rH rî r rz rJ r| rw rL r< rJ )r6 r? r] r rM ©r_ rQ rþ r< r< r= Útest_out! s
|
|
|
|
zTestPercentile.test_outc C sî t jddØ t ddt¡ t d¡}t d¡}tj|d< ttj |dd|d | ttj |ddd
|
|d | t d ¡}ttj |d d |d | ttj |d d d
|
|d | t d¡}ttj |d |d | ttj |d d
|
|d | W 5 Q R X d S )NTr
|
r r r» rî r r rz rJ r| rw rL r< )
|
r r r r¼ r6 r? r] rü r rM r¼ r< r< r= Ú test_out_nan. s,
|
|
|
ÿ
|
ÿ
|
ÿzTestPercentile.test_out_nanc C s: t jdtd}t j|d< tt |d¡t j tt j|dddt j tt j|ddgddt t jgd ¡ t jdtd ddd ¡}t j|d
|
< t j|d < tt |d¡t j tt |d¡jd t t jdtd ddd ¡dd¡}t j|d < t j|d < tt |dd¡| t t jdtd ddd ¡ddgd¡}t j|d d
ddf< t j|d d
ddf< tt |ddgd¡| t t jdtd ddd ¡dd¡}t j|d< t j|d < tt |dd¡| t t jdtd ddd ¡ddgd¡}t j|d d
ddf< t j|d d
ddf< tt |ddgd¡| t t jdtd ddd ¡dd¡}t j|d< t j|d< tt |dd¡| t t jdtd ddd ¡ddgd¡}t j|d d
df< t j|d d
df< tt |ddgd¡| t jt jdtd ddd ¡ddgddd}t j|d d
df< t j|d d
df< tt j|ddgddd| d S )Nr rÓ rJ r r rr g333333ã?rS rH r
©rL rL rJ rÔ re rL r rP rJ rK ) r6 r7 r+ rü r rM r rl r
rz r< r< r= Útest_nan_behavior@ s
|
ÿ
|
|
"
|
|
ÿ"
|
|
ÿ ÿ
|
|
ÿ ÿ ÿÿz TestPercentile.test_nan_behaviorc C s¬ t jtdd t ddddgtj¡ W 5 Q R X t jtdd t ddddgtjg¡ W 5 Q R X t ddd ¡}tj|d
|
< t jtdd t ddddg|¡ W 5 Q R X d S ) NzPercentiles must be inr. rL rJ rS r¯ r¨ g ÀX@r> r )rá r r\ r6 rM rü r¯ )r_ r r< r< r= Ú
|
test_nan_q} s "
|
zTestPercentile.test_nan_q).rn ro rp ra rO rR rS rá râ rã r6 Ú typecodesrX rà r¾ r¿ rÆ rH rI Z clongdoublerÀ ZH_F_TYPE_CODESrf Z
|
TYPE_CODESrk ro rq rr rs rl r r r r r r r r r§ rª r± rº r½ r¾ rÀ rÁ r< r< r< r= rI { s
|
|
ÿ
|
|
|
|
|
|
|
ùþ øÿ
|
|
|
|
$0) ûþ
|
=rI c @ sÎ e Zd Zdd Zdd Zejjdddd Zd d
|
Z d d Z
|
ej d e j d ¡dd Zej dddddddddddd d d g ¡d d Zejee jejd!d"d#ejd$d$d%d&d'd(d)d*d+ Zd,d- Zd.S )/Ú TestQuantilec C s, dddg}t |d¡}t jj|ddd d S )Nrª r rÉ gÍÌÌÌÌÌÜ?g
|
×£p=
|
Ç?rL )Zmaxulp)r6 ÚquantilerW Zassert_array_max_ulp)r_ r· r` r< r< r= Ú test_max_ulp s
|
zTestQuantile.test_max_ulpc C sH t d¡d }tt |d¡d tt |d¡d tt |d¡d d S )Nri rµ r rª rL rH r¬ )r6 r7 r rÄ r" r< r< r= ra s zTestQuantile.test_basicz See gh-19154rd c C sn t dg¡}t dd¡}t||d tt||j t dddg¡}t ||¡}t|| t|j|j d S )NTFr )r6 r rÄ r rÎ rÀ r )r_ r` Ztf_quantZ quant_resr< r< r= Útest_correct_quantile_value s
|
z(TestQuantile.test_correct_quantile_valuec C s8 dd t dD }t |d¡}t|d tt|t t |d¡}t|tdd tt|t t |tdd¡}t|tdd tt|t t |tddg¡}t|t tddg¡ tt|tj t |tddgg¡}t|t tddgg¡ tt|tj t d¡}tt |tdd¡tdd d S ) Nc S s g | ]}t |d qS rN r rõ r< r< r= rÜ § s z.TestQuantile.test_fraction.<locals>.<listcomp>ri r rL rk rJ rH ) r^ r6 rÄ r rÎ r r rÌ r7 )r_ r· r r< r< r= rO ¥ s$
|
|
zTestQuantile.test_fractionc C sp t dddddg¡}| ¡ }t jt d¡|dd t|| | ¡ }| ¡ }t jt d¡|dd t|| d S ) Nr rt r¸ rµ r¨ r rm rK )r6 r r rÄ r7 r rñ )r_ Úp0r& r< r< r= r À s
|
z TestQuantile.test_no_p_overwriterÀ r c C s2 t jt jddg|ddgdd}|j|ks.td S )NrL rJ rÓ rµ rJ rK )r6 rÄ r rÀ r¶ )r_ rÀ r r< r< r= Ú test_quantile_preserve_int_typeÌ s ÿz,TestQuantile.test_quantile_preserve_int_typerL r[ r\ r^ r_ r` ra rP rc rd rJ rg rh rm c C s t ddd¡}t jt dddddddddddddddd d d
|
g¡d ||d }tt |¡| t jd dddg||d }tt |¡| d S )Nr rL r rJ rS rH rT rå ri rk re rK rª r¨ r§ r© )r6 r¯ rÄ r r r³ )r_ rL rÇ rÄ r< r< r= Útest_quantile_monotonicÒ s & ÿÿ ÿz$TestQuantile.test_quantile_monotonicrS r )Ú min_valueÚ max_valueFçu <ä7þçu <ä7~)Úallow_infinityÚ allow_nanrÊ rË )rÀ rA Úelements)ræ c C s. t ddd¡}t ||¡}tt |¡| d S )Nr rL g{®Gáz?)r6 r7 rÄ r r³ )r_ ræ rÇ rÄ r< r< r= Ú test_quantile_monotonic_hypoä s z)TestQuantile.test_quantile_monotonic_hypoc C sZ t dddgdddgg¡}t j|d d< t |d ¡}t |¡sBttt |d ¡t j d S )
|
Nr3 r9 r¯ r© r§ r¨ r rL rµ )r6 r rü rÄ r~ r¶ r )r_ r` rÄ r< r< r= Útest_quantile_scalar_nanî s
|
z%TestQuantile.test_quantile_scalar_nanN)rn ro rp rÅ ra rá râ r rÆ rO r rã r6 r rÈ rÉ Ú
|
hypothesisÚgivenr r¿ ÚstZintegersÚfloatsrÑ rÒ r< r< r< r= rà sD
|
|
|
ýÿ
|
ÿþÿ
|
rà c
|
@ sê e Zd Zejejdddddejdddddejdddddejdddddddd Zejejdddddejdddddejdddddd
|
d d Zejejdddddejdddddejdddddd
|
d d Z dd Z
|
dS )ÚTestLerpFr rL )rÏ rÎ rÊ rË rÌ rÍ )Út0Út1r` r{ c C sh t |||¡}t |||¡}||ks,||kr:||ksdtn*||k ||k krX||ksdtn ||ksdtd S r5 ©r@ Ú_lerpr¶ )r_ rØ rÙ r` r{ Zl0Úl1r< r< r= Ú+test_linear_interpolation_formula_monotonic÷ s z4TestLerp.test_linear_interpolation_formula_monotonic)Útr` r{ c C sV ||kr.|t |||¡ kr&|ksRn tn$|t |||¡ krL|ksRn td S r5 rÚ )r_ rÞ r` r{ r< r< r= Ú)test_linear_interpolation_formula_bounded s &z2TestLerp.test_linear_interpolation_formula_boundedc C s6 t ||dd| ¡}t ||d| ¡}t|| d S rï )r@ rÛ r )r_ rÞ r` r{ r r" r< r< r= Ú+test_linear_interpolation_formula_symmetric s z4TestLerp.test_linear_interpolation_formula_symmetricc C s8 t d¡}t d¡}t d¡}t |||¡dks4td S )NrJ rT r gÍÌÌÌÌÌ@)r6 r r@ rÛ r¶ )r_ r` r{ rÞ r< r< r= Ú+test_linear_interpolation_formula_0d_inputs! s
|
|
|
z4TestLerp.test_linear_interpolation_formula_0d_inputsN) rn ro rp rÓ rÔ rÕ rÖ rÝ rß rà rá r< r< r< r= r× ö s^ ÿ ÿ ÿ ÿú
|
|
ÿ ÿ ÿü
|
ÿ ÿ ÿü
|
r× c @ sÔ e Zd Zdd Zdd Zdd Zdd Zd d
|
Zej
|
d d d dgd e j dge j df¡dd Z dd Zdd Zdd Zej jedddd Zd d Zd d Zd d! Zd"d# Zej j
|
d$d%d&d'd(d)gd*d+d, Zd%S )-Ú
|
TestMedianc C sH t d¡}t d¡}t d¡ dd¡}tt |¡d tt |¡d tt |¡d tt j|ddd dd
|
g tt j|dddd g tt j|d dd t d d ddg¡}t|d |d d t |¡ t d d dg¡}t|d t |¡ t d dd g¡}t|d t |¡ tt |¡jd t j |d< tt |¡jd d S )NrL rJ rj rS rµ r¶ r rr r rH rH gð6-¦?gº ¤0¸§?gºÝË}rÂ?g©Á4 ¯?r§ rX )
|
r6 r r7 rl r Úmedianr r r
rü )r_ Úa0r& rN r` r< r< r= ra * s$
|
|
|
zTestMedian.test_basicc C sè t ddgddgddgddgg¡}|t jjdd d
|
d fD ]B}| ¡ }t j|d d t|jD ]}t j||d q\t|| q8t t j|dd ddg t t j|j
|
dd ddg t t |¡d t t j|d d d t t |j
|
¡d d S )NrJ rS r rL rj rk rH rT r r r rr rH ) r6 r r± Úrandintr rã r^ r
r r r )r_ Úa3r` ÚorigZaxr< r< r= Útest_axis_keyword@ s
|
ý z TestMedian.test_axis_keywordc C sà t ddgddgddgddgg¡}t d¡}t d¡}t d¡ dd¡}tt j| ¡ d d
|
d tt j| ¡ d d
|
d tt j| ¡ d d
|
d tt j| ¡ d dd dd dg tt j| ¡ d dd ddg tt j| ¡ d d d d tt j| ¡ d dd ddg tt j|j ¡ d dd ddg t jdt jd d¡}t j
|
| ¡ ¡ tt j|d dt j| ¡ d d d tt j|ddt j| ¡ dd d tt j|ddt j| ¡ dd d tt j|ddt j| ¡ dd d d S )NrJ rS r rL rj rk rH rT Tr rµ r¶ )r rs r rH r rÓ )rS rH rT rr )rs r ) r6 r r7 rl r rã r r r¾ r± r£ r¤ )r_ ræ rä r& rN Za4r< r< r= Útest_overwrite_keywordR sV
|
ý
|
|
ÿ ÿ ÿ ÿÿÿÿÿÿz!TestMedian.test_overwrite_keywordc C sH dddg}t t |¡d |g}t t |¡d ttj|dd| d S )NrL rJ rS r rr )r r6 rã r )r_ r· r r< r< r= r£ s s
|
|
zTestMedian.test_array_likec C s4 G dd dt j}|dddg}tt |¡d d S )Nc @ s e Zd ZdddZdddZdS )z,TestMedian.test_subclass.<locals>.MySubClassNc S s t |¡ | ¡}||_|S r5 )r6 r rÍ Úinfo)ÚclsZ input_arrayrê Úobjr< r< r= Ú__new__~ s z4TestMedian.test_subclass.<locals>.MySubClass.__new__c S s dS )Néùÿÿÿr< )r_ rs rÀ rf r< r< r= r« s z1TestMedian.test_subclass.<locals>.MySubClass.mean)N)NNN)rn ro rp rí r« r< r< r< r= Ú
|
MySubClass| s
|
rï rL rJ rS rî )r6 rÌ r rã )r_ rï r` r< r< r= r! z s
|
zTestMedian.test_subclassræ r¨ r§ r© rª c C s: G dd dt j}t t |¡ |¡¡}t||s6tdS )z6Check that we return subclasses, even if a NaN scalar.c @ s e Zd ZdS )z-TestMedian.test_subclass2.<locals>.MySubclassNrÊ r< r< r< r= Ú
|
MySubclass s rð N)r6 rÌ rã r rÍ r r¶ )r_ ræ rð ré r< r< r= Útest_subclass2 s zTestMedian.test_subclass2c C sl t d¡}t d¡}tt j|d|d| t d¡}tt j|d|d| t d¡}tt j||d| d S )Nr» rî r rz rw rL r< )r6 r? r] r rã r¼ r< r< r= r½ s
|
|
|
|
zTestMedian.test_outc C s t jdd t ddt¡ t d¡}t d¡}tj|d< ttj |d|d | t d
|
¡}ttj |d |d | t d ¡}ttj ||d | W 5 Q R X d S ) NTr
|
r r r» rî r r rz rw rL r< )
|
r r r r¼ r6 r? r] rü r rã r¼ r< r< r= r¾ s
|
|
|
|
|
zTestMedian.test_out_nanc C sf t jdtd}t j|d< tt |¡t j tt j|ddt j t jdtd ddd¡}t j|d< t j|d < tt |¡t j tt |¡jd t t jdtd ddd¡d¡}t j|d
|
< t j|d < tt |d¡| t t jdtd ddd¡d ¡}t j|d < t j|d < tt |d ¡| t t jdtd ddd¡d¡}t j|d < t j|d< tt |d¡| d S )Nr rÓ rJ r rr rS rH r
r¿ rÔ re rL r rP )r6 r7 r+ rü r rã rl r
rz r< r< r= rÀ ¨ s*
|
|
|
|
|
|
|
|
|
z TestMedian.test_nan_behaviorz fp errors don't work correctlyrd c C sZ t jg td}tjddF}t ddt¡ tt |¡t j t
|
|d j tk tt |d W 5 Q R X t jg tdd }tjdd8}t ddt¡ tt |¡t j t
|
|d j tk W 5 Q R X t jg tdd }tt j|dd
|
| tt j|d d
|
| t jt j tdd }tjdd:}t ddt¡ tt j|dd
|
| t
|
|d j tk W 5 Q R X d S ) NrÓ Tr
|
r r r rJ rS )rÀ rÈ rr rL ) r6 r r+ r r r r¼ r rã rü r r rê )r_ r` rà r{ r< r< r= r½ È s& zTestMedian.test_emptyc C sL t d¡}ttt | t¡¡t t j|d< ttt | t¡¡t d S )Nr9 rJ ) r6 r7 r rÎ rã r½ r r+ rü )r_ rQ r< r< r= Ú test_objectæ s
|
|
zTestMedian.test_objectc
|
C s t jjdd}t |gd ¡}tt j|ddt |¡ t |dd¡}tt j|ddt |¡ | dd ¡ ¡ }tt j|d
|
dt |¡ tt j|d dt j|d d tt j|d dt j|dd tt j|d dt j|dd t d¡
|
d¡}t j | ¡ ¡ tt j|d dd t |d d
d d
d d
df ¡ ¡ tt j|ddd t |d d
d d
d d d
f ¡ ¡ tt j|ddd t |d d
d d
dd d
f ¡ ¡ tt j|ddd t |dd d
d d
d d
f ¡ ¡ tt j|ddd t |dd d d
d d
f ¡ ¡ tt j|ddd t |dd d
d d
d f ¡ ¡ tt j|ddd t |dd d
dd d
f ¡ ¡ d S )Nr r r rY rr rX r rf rL rd rK r )rX r r r r rJ r r r r r rO )r6 r± r r¡ r rã r¢ r r r7 rl r£ r¤ r¥ r¦ r< r< r= r§ ì sB $ÿ$ÿ$ÿ$ÿ ÿ ÿ ÿz TestMedian.test_extended_axisc C sp t d¡}tt jt j|dd tt jt j|dd tt jt j|dd tt jt j|dd ttt j|dd d S )Nr r2 rr r¨ rH r© rQ )r6 r] r rv rã r\ r r< r< r= rª
|
s
|
z%TestMedian.test_extended_axis_invalidc C s t d¡}tt j|d ddjd tt j|dddjd tt j|dddjd tt j|d ddjd
|
tt j|d ddjd tt j|d ddjd d S )Nr Tr² r« rY r¬ ru r rÝ r® r¯ r r° )r6 r] r rã rA r r< r< r= r± s&
|
ÿÿÿÿÿÿzTestMedian.test_keepdimsrs NrL rÝ rY r² r³ c s t d¡|d kr dj }n*t|j t fddtjD }t |¡}t j|d|d}||ksptt |j
|
| d S )Nr rÝ c 3 s$ | ] }| krd nj | V qdS rµ r¶ rõ r· r< r= rá 1 s z/TestMedian.test_keepdims_out.<locals>.<genexpr>Tr¸ ) r6 r] r
r4 rK r^ r rã r¶ r rA )r_ rs r¹ rf r} r< r· r= rº ! s
|
ÿ
|
z TestMedian.test_keepdims_out)rn ro rp ra rè ré r£ r! rá râ rã r6 rü rñ r½ r¾ rÀ rn r r½ rò r§ rª r± rº r< r< r< r= râ ( s8 !ÿ
|
|
ûþ
|
râ c @ s e Zd Zdd Zdd ZdS )ÚTestAdd_newdoc_ufuncc C s" t ttdd t tttjd d S )NrJ Zblah)r r r r\ r6 r8 rx r< r< r= Útest_ufunc_arg; s z#TestAdd_newdoc_ufunc.test_ufunc_argc C s t tttjd d S )NrS )r r r r6 r8 rx r< r< r= Útest_string_arg? s z$TestAdd_newdoc_ufunc.test_string_argN)rn ro rp rô rõ r< r< r< r= ró 9 s ró c @ s\ e Zd Zejjejjdkddejj e
|
dddd Z ejjejjdkdddd Z d S )
|
ÚTestAdd_newdocrJ úPython running -OOrd úPyPy does not modify tp_docc C sV d}t tjjjjd t|
| tttjjj jdk tttj
|
j j jdk d S )Nz"Current flat index into the array.r¿ ) r r6 ÚcoreÚflatiterÚindexr2 rê r ZufuncÚidentityr¨ Z index_tricksZmgrid)r_ rO r< r< r= Ú test_add_docE s zTestAdd_newdoc.test_add_docc C s2 t jjjj}t ddd¡ |t jjjjks.td S )Nz
|
numpy.corerú )rû z bad docstring)r6 rù rú rû r2 Z
|
add_newdocr¶ )r_ Zprev_docr< r< r= Útest_errors_are_ignoredN s z&TestAdd_newdoc.test_errors_are_ignoredN) rn ro rp rá râ rn ri r¡ Úoptimizer r rý rþ r< r< r< r= rö C s
|
rö c @ s\ e Zd Zejjejjdkddejje dddd Z
|
ejjejjdkdddd Z d S )
|
ÚTestAddDocstringrJ r÷ rd rø c C s0 t t jjt jjj¡ dd }t ||j¡ d S )Nc S s dS ©Z docstringNr< r< r< r< r= Úfunc` s z6TestAddDocstring.test_add_same_docstring.<locals>.func)r6 Ú add_docstringrÌ Úflatr2 ©r_ r r< r< r= Útest_add_same_docstringZ s z(TestAddDocstring.test_add_same_docstringc C sP t t t tjjd¡ W 5 Q R X dd }t t t |d¡ W 5 Q R X d S )Nzdifferent docstringc S s dS r r< r< r< r< r= r l s z=TestAddDocstring.test_different_docstring_fails.<locals>.func)r rØ r6 r rÌ r r r< r< r= Ú test_different_docstring_failsf s
|
|
|
z/TestAddDocstring.test_different_docstring_failsN) rn ro rp rá râ rn ri r¡ rÿ r r r r< r< r< r= r W s
|
|
r c @ s6 e Zd Zej dddddddg¡dd Zd
|
d Zd S ) ÚTestSortComplexztype_in, type_out)ÚlÚD)Úhr )ÚHr )r{ r )rê r )rÉ ÚGc C sN t jdddddg|d}t |¡}t |¡ |¡}t|| t|j|j d S )NrT rS rj rJ rL rÓ )r6 r Ú sort_complexr³ r½ r rÀ )r_ Ztype_inZtype_outr` rÄ rs r< r< r= Útest_sort_realv s
|
|
|
|
z TestSortComplex.test_sort_realc C sR t jddddgdd}t jddddgdd}t |¡}t|| t|j|j d S )Nr8 y ð? Àr× r9 r
|
rÓ )r6 r r r rÀ )r_ r` rs rÄ r< r< r= Útest_sort_complex s
|
|
|
z!TestSortComplex.test_sort_complexN)rn ro rp rá râ rã r r r< r< r< r= r t s ú
|
r )r¨ )r¨ rª )}rV r ri rÙ Z fractionsr rý rá rÓ Zhypothesis.extra.numpyr Zhypothesis.strategiesZ
|
strategiesrÕ Únumpyr6 r Z numpy.testingr r r r r
|
r r r r r r r r Znumpy.lib.function_baser¨ Z function_baser@ Z numpy.randomr Z numpy.libr r r r r r r r r r r r r r! r" r# r$ r% r& r' r( r) r* r+ r, r- r. r/ r0 r1 r2 r3 Znumpy.core.numericr4 r> rF rG rq r r r r¦ rä r r1 r6 r7 r: rP r\ r` r r rÔ rÞ rô r r r r\ ro r râ rã r r r r¡ r¤ r¦ r« rÇ rÞ rá rä rå rú r r rI rà r× râ ró rö r r r< r< r< r= Ú<module> s¨ < Dg# >o u }A)
|
|
*)U ÿl> [j4nd` C m2
|
|
