8eͩ$r d dl Z d dlZd dlZd dlZd dlmZ d dlZd dlZd dlZd dl m
Z
d dl m Z
d dlmZ d dlZd dlmZ d dlmZmZmZmZmZmZmZmZm Z m Z m Z m Z m Z d dl!m"c m#Z$ d dl%m&Z& d dl'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;mZ>m?Z?m@Z@mAZAmBZBmCZCmDZDmEZEmFZFmGZG d d lHmIZI d
ZJd ZK G d d
ZL G d d ZM G d d ZN G d d ZO G d d ZP G d d ZQ G d d ZR G d d ZS G d d ZT G d d ZU G d d! ZV G d" d# ZW G d$ d% ZX G d& d' ZY G d( d) ZZ G d* d+ Z[ G d, d- Z\ G d. d/ Z] G d0 d1 Z^ G d2 d3 Z_dsd5Z`dtd7Za G d8 d9 Zb G d: d; Zc G d< d= Zd G d> d? Zeejf g d@dAejh dB z ejh dC z ejf g dDg dE G dF dG Zi G dH dI Zj G dJ dK Zk G dL dM Zl G dN dO Zm G dP dQ Zn G dR dS Zo G dT dU Zp G dV dW Zq G dX dY Zr G dZ d[ Zs G d\ d] Zt G d^ d_ Zu G d` da Zv G db dc Zwg ddZx G de df Zy G dg dh Zz G di dj Z{ G dk dl Z| G dm dn Z} G do dp Z~ G dq dr ZdS )u NFraction)arrays)partial)ma)
assert_ assert_equalassert_array_equalassert_almost_equalassert_array_almost_equal
assert_raisesassert_allcloseIS_PYPY assert_warnsassert_raises_regexsuppress_warnings HAS_REFCOUNTIS_WASM)rand) add_newdoc_ufuncangleaveragebartlettblackmancorrcoefcovdeletediffdigitizeextractflipudgradienthamminghanningi0insertinterpkaisermeshgridmsort piecewiseplacerot90selectsetxor1dsinctrapz
trim_zerosunwrapunique vectorize)normalize_axis_tuplec n t j | }t j || }|S N)nparangeaddouter)ndatas v/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/numpy/lib/tests/test_function_base.pyget_matr@ s)
9Q<>>
@
@
@
@ @ @rA N)__name__
__module__ __qualname__rn rv r rA r? rL rL 0 sM L L L@C C C @ @ @ @ @rA rL c D e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
S ) TestFlipc t t j t j t j d d t t j t j t j d d t t j t j t j d d t t j t j t j d d d S )NrN rR axisrN rN rP r^ r rZ )r
r9 AxisErrorfliprf rh s r? rv zTestFlip.test_axesv s blBGRWQZZa@@@@blBGRWV__1EEEEblBGRWV__2FFFFblBGRWV__6JJJJJJrA c t d }|d d d d df }t t j |d | g dg dg}g dg dg}t t j |d | d S )NrN ra rR rQ rY r] r\ r@ r r9 r rh ri bs r?
test_basic_lrzTestFlip.test_basic_lr| s AJJ
aaa2gJRWQ]]A&&&
YY
YY
YY
YY
RWQ]]A&&&&&rA c t d }|d d dd d f }t t j |d | g dg dg}g dg dg}t t j |d | d S )NrN ra r rQ rY r r s r?
test_basic_udzTestFlip.test_basic_ud s AJJ
dddAAAgJRWQ]]A&&&
YY
YY
YY
YY
RWQ]]A&&&&&rA c t j ddgddggddgddggg }t j ddgddggddgddggg }t t j |d | d S Nr rR rP rZ rN r[ ry rz r9 arrayr r r s r? test_3d_swap_axis0zTestFlip.test_3d_swap_axis0 H1v1v 1v1v
!
!
H1v1v 1v1v
!
!
RWQ]]A&&&&&rA c t j ddgddggddgddggg }t j ddgddggddgddggg }t t j |d | d S r r r s r? test_3d_swap_axis1zTestFlip.test_3d_swap_axis1 r rA c t j ddgddggddgddggg }t j ddgddggddgddggg }t t j |d | d S r r r s r? test_3d_swap_axis2zTestFlip.test_3d_swap_axis2 r rA c
B t j d dddd }t |j D ]_}t t j || t j | d| |d `d S )Nx rP rZ rN r[ r ) r9 r: r{ rg ndimr r r! swapaxes)rh ri is r? test_4dzTestFlip.test_4d s Im $ $ , ,Q1a 8 8qv E EA A 1::a#3#344==aCC
E
E
E
E E ErA c t j g dg dg }t j g dg dg }t t j | | d S )NrR rP rZ rN r[ ry )ry r[ rN rZ rP rR r r s r? test_default_axiszTestFlip.test_default_axis sf Hiii ii!
"
" Hiii ii!
"
"RWQZZ#####rA c t j ddgddggddgddggg }t t j |d
| t j ddgddggddgddggg }t t j |d
| t j ddgddggddgddggg }t t j |d
| d S )
Nr rR rP rZ rN r[ ry rz r r rV rs r rh ri r cs r? test_multiple_axeszTestFlip.test_multiple_axes s H1v1v 1v1v
!
!
RWQR(((!,,, H1v1v 1v1v
!
!
RWQV,,,a000 H1v1v 1v1v
!
!
RWQV,,,a00000rA N) r r r rv r r r r r r r r r rA r? r r t s K K K ' ' '' ' ' ' ' ' ' ' ' ' ' 'E E E $ $ $1 1 1 1 1rA r c e Zd Zd Zd ZdS )TestAnyc g d}g d}g d}t t j | t t j | t t j | d S )Nr r rR r r r r r )rR r rR r )r r9 anyrh y1y2y3s r? rn zTestAny.test_basic sg
\\
\\
\\r
r
BF2JJ rA c g dg dg dg}t t j | t t j |d g d t t j |d g d d S )Nr r r r rR r rR rR r r r rR r rR rR )r r9 r r
rh r s r? test_ndzTestAny.test_nd s| iiIII
.r
26"1---yyy99926"1---yyy99999rA Nr r r rn r r rA r? r r s2 : : : : :rA r c e Zd Zd Zd ZdS )TestAllc P g d}g d}g d}t t j | t t j | t t j | t t j t j | d S )N)r rR rR r r rR rR rR rR )r r9 allr r s r? rn zTestAll.test_basic s
\\
\\
\\BF2JJ r
BF2JJ }%%&&&&&rA c g dg dg dg}t t j | t t j |d g d t t j |d g d d S )Nr r rR r rR rR rR r r rR )r r9 r r
r s r? r zTestAll.test_nd s~ iiIII
.BF2JJ 26"1---yyy99926"1---yyy99999rA Nr r rA r? r r s2 ' ' ': : : : :rA r c e Zd Zd Zd Zd ZdS )TestCopyc t j ddgddgg }t j | }t || d|d< t |d d t |d d d S )NrR rP rZ rN
)r r )r9 r copyr
r )rh ri a_copys r? rn zTestCopy.test_basic sr Hq!fq!f% & &1f%%%t QtWa VD\2&&&&&rA c Z t j ddgddgg }t |j j t |j j t j ddgddggd }t |j j t |j j t j | }t |j j t |j j t j | }t |j j t |j j d S )NrR rP rZ rN Forder)r9 r r flags c_contiguous f_contiguousr )rh ri a_fortr a_fort_copys r?
test_orderzTestCopy.test_order s
Hq!fq!f% & &$%%%AG(()))Aq6Aq6*#666FL--... )*** )***FL--...gfoo K%22333 !./////rA c l t j d }t t j t j |d t t j t j |d t t j t j | d S )Nr[ F)subokT)r rf r
isMaskedArrayr9 r )rh mxs r?
test_subokzTestCopy.test_subok s
WQZZB$RWRu%=%=%=>>>??? 4!8!8!899:::B$RWR[[11122222rA N)r r r rn r r r rA r? r r sA ' ' '0 0 0"3 3 3 3 3rA r c e Zd Zd Zej dg dddgg ddgdgfg d g d
gd g d gd
dgg dgg dgfg d Zd Zd Z d Z
d Z d Z d Z
dS ) TestAveragec t j g d }t t |d dk t j g d }t t |d dk g d}t t |d dk t j d }d|d < d
|d < t | d t |d t | d t |d t d
d
}t | d t |d t | d t |d d S )Nr r r @ ?r @) r r r r rb rP rc rR r[ )r9 r r r rf r meanr )rh r r r y4y5s r? rn zTestAverage.test_basic# sK
Xiii
###r)***
Xlll
#
####r)***
\\###r)***
WV__44BGGAJJA777BGGAJJA777
!QZZBGGAJJA777BGGAJJA77777rA z.subclassNr r r r rA r? subclassr DrA r rR rP rZ rN r )r9 ndarrayr viewr typer )rh r ri r s r? test_subclasseszTestAverage.test_subclasses s rz HqeQqE] # # ( ( 2 2 HqeQqE] # # ( ( 2 2T"*Q--(((333T"*Q22233X>>>>>rA c g d}|D ]y\ }}}t j ddgddgg| }t j ddgddgg| }t t j || j t j | zd S )N))i4r f8)r f4r )r r r )r r r )r r r rR rP rZ rN r r )r9 r r r r )rh typsatwtrtri r s r? test_upcastingzTestAverage.test_upcasting s 9 9 9 G GJBB1Q%1 b111A1Q%1 b111A Aq1117" F F F F G GrA c D t j d t d D }t j d t d D }|| z }t | d t
|| d S )Nc 6 g | ]}t j | S r decimalDecimal).0r s r?
z1TestAverage.test_object_dtype.. " <<. r. rA r r )r9 r rg r r r r )rh ri r s r? test_object_dtypez TestAverage.test_object_dtype s H<<%))<<< = = H<<%))<<< = = QUUWW AFF1IIwq!'<'<'<=====rA c t j t dd t dd g }t t j | t dd d S )NrR r[ rZ rP )r9 r r r r ru s r? test_average_class_without_dtypez,TestAverage.test_average_class_without_dtype sL Hhq!nnhq!nn5 6 6RZ]]HQNN33333rA )r r r rn pytestmark parametrizer r r r r& r2 r4 r rA r? r r ! s 8 8 8$ [ F
))TC5)))dVcU ; 99jjj
!1&7 a&???#iii[
2 3 0 0
0"*, *, *,X4 4 44? ? ?G G G> > > 4 4 4 4 4rA r c e Zd Z ej g d ej g d ej g d gZ ej g d ej g d ej g d gZddZd Zd
Z d Z
d Z d
Z d Z
dS )
TestSelectr r rz rx )FFFFTFFFTr c g }t t | D ]$|fdt || D p|gz
}%|S )Nc 6 g | ]\ }}|
| S r r )r, VCms r? r- z&TestSelect._select.. s* BBB1QqTBqtBBBrA )rg lenzip)rh condvaluesdefaultoutputrB s @r? _selectzTestSelect._select s^ s4yy!! P PA BBBBFD(9(9BBBOwi OFF
rA c | j }| j }t t ||d | ||d t t
| d t t
| d d S )N rG rZ )choices
conditionsr
r. rI r rC rh rM rN s r? rn zTestSelect.test_basic s ,_
6*grBBB <<
GR<HH J J J S\\1%%%S__a(((((rA c ^ t j d t j g d g}dt j d dd g}t t || t j d t t dgdgdg j d
d S ) NTr< rR rN rZ rN rZ r rL r0 ) r9 r r: r{ r
r. rf r rE rh rN rM s r? test_broadcastingz TestSelect.test_broadcasting s htnnbh/C/C/C&D&DE
bimm++Aq1126*g66HHHVTFQC!555;TBBBBBrA c t t | j | j d j t
j d | j D }t t | j | j t
j t j dddt
j ddg }t j
| }t t |g|g dddt
j ddg d S ) N ?c L g | ]!}| t j "S r )r r9 int8)r, choices r? r- z0TestSelect.test_return_dtype.. s& EEEf6==))EEErA rR rP rZ r[ rz r ) r r. rN rM r r9 rF rX r nanisnan)rh rM drB s r? test_return_dtypez TestSelect.test_return_dtype s VDOT\2>>D[ " " " FE EEEVDOW55;RWEEE HaArvq!, - - HQKKVQC!%%1aA'>?????rA c x t t t g g d t t t g g d S )N @)r
re r. r s r? test_deprecated_emptyz TestSelect.test_deprecated_empty s2 j&"b"555j&"b11111rA c z | j }| j d d }|d t j |d< t t t || |d t j |d< t t t || t t t || d S Nr ) rM rN r r9 int_r
TypeErrorr. uint8rO s r? test_non_bool_deprecationz$TestSelect.test_non_bool_deprecation s ,_QQQ'
"1
,,RW55
1
iW==="1
,,RX66
1
iW===iW=====rA c t j dg gdz }t j dg gdz }t || d S )NFd rR )r9 r r. rS s r? test_many_argumentsz TestSelect.test_many_arguments sG hw''(3.
8QC==/C'z7#####rA Nr )r r r r9 r rM rN rI rn rT r] r` rf ri r rA r? r9 r9 s rx ""rx ""rx ""$G "(00011"(///00"(///002J ) ) )C C C
@
@
@2 2 2> > >$ $ $ $ $rA r9 c e Zd Zd Zd Zd Zd Zd Zd Zd Z e
j dd d
g d Z
d S )
TestInsertc
$ g d}t t |dd g d t t |dd g d t t |g dg d g d t t |dg d g d t t |g d d
g d t t |t d d d d
g d
t t |g dg d g d t j ddgt j }t t |d|d g d t t |g g | t
j d 5 }t
j ddt t t |t j dgdz d
g d t |d j t u d d d d S # 1 swxY w Y d S )Nr r rR )rR rR rP rZ rZ )rR rP rZ rR r )rR rR rP rZ rP rZ )rR ra rZ r; )rR r; rP r; rZ r; ra )r; rR r; rP r; rZ ra rR rZ r: )rR rx rP rz rZ r; r! r r r Trecordalways rN )rR r; r; r; r; rP rZ ) r r& slicer9 r r warningscatch_warningsfilterwarnings
FutureWarningr category)rh ri r r s r? rn zTestInsert.test_basic sV IIVAq!__lll333VAq!__lll333VAyyy)))446H6H6HIIIVAq))),,.@.@.@AAAVAzzz1--/A/A/ABBBVAuRr 2 2A668J8J8JKKKVAzzz999557I7I7IJJJ HaV2: . . .VAq!A$'' 666VAr2&&***
$D
1
1
1 4Q #Hb- @ @ @ q"(D6A:..224I4I4I
K
K
K AaDM]2 3 3 3 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4s A4HH H c g dg}g dg dg}t t |ddg g d t t |dg dd | t t |ddd | t t |ddd g dg t j ddgddgd d gg }t j dd
d d d }t j |d d ddf t j dd
d d d j |d d ddf fd }t t |dgdgdgd ggd | t t |dgg d d | t t |dg d d | t t |ddgdgd ggd | t j d
dd }t t |d d d df d|d d df d | t t |d dd d f d|dd d f d | t j d d
}t t |d|d d d d d f d t |d|d d d d d f d t t |d|d d dd d f d t |d|d d dd d f d t t j
t |d|d d dd d f d t t j
t |d|d d dd d f d t j d d
}t t |d|d d d d d f d t |d|d d d d d f d t t |d|d d dd d f d t |d|d d dd d f d d S )Nr rO r rR r r rP )rR rR rP rR rZ rN r r ra rX ) r r& r9 r r: repeatr{ concatenateTr
r )rh ri rr r s r?
test_multidimzTestInsert.test_multidim s
YYK
YY
YY
VAq1#&& 555VAq)))!444a888VAq!!,,,a000VAq!!,,,|||n=== Hq!fq!fq!f- . . IaOO " "1 % % - -a 3 3 N
qqq!A#vY !Q..q1199!Q??A
qqq!A#vY
!
!
! VAsaS1#sO!<<>> IaLL A & &VAaaa!eHa111a4q9991===VAbqbdGQ!AAA#Q777;;;
IbMM ! !) , ,VAq!AAAaaaF)"555Aq!AAAaaaF)!444 6 6 6VAq!AAAqF)"555Aq!AAAqF)!444 6 6 6 blFAq!AAAq!!!G*1EEEEblFAq!AAAq!!!G*2FFFF
IbMM ! !) , ,VAq!AAAqqq!G*2666Aq!AAAqqq!G*1555 7 7 7VAq!AAAq!!!G*2666Aq!AAAq!!!G*1555 7 7 7 7 7rA c J t j d }t j t j 5 t |g dd d d d n # 1 swxY w Y t j t 5 t |g dd d d d d S # 1 swxY w Y d S )NrR rP r r nonsense)r9 r r5 raisesr r& rd ru s r? test_0dzTestInsert.test_0d, s HQKK
]2<
(
( % % 1b!! $ $ $ $ % % % % % % % % % % % % % % %
]9
%
% . . 1b!* - - - - . . . . . . . . . . . . . . . . . .s# AAA7BB B c h G d dt j }t j d | }t t t j |ddg | t t t j |g g | t t t j |ddgddg | t t t j |t dd ddg | t t t j |t ddd g | t j d | }t t t j |ddg | d S ) Nc e Zd ZdS )*TestInsert.test_subclass..SubClassNr r rA r? SubClassr 4 r rA r r r rR rP rX ra ) r9 r r: r r
isinstancer& rt r rh r ri s r?
test_subclasszTestInsert.test_subclass3 sr rz IbMM x ( (
29QA3//::;;;
29QB//::;;;
29QAA77BBCCC
29Qa aV<>>j1n Ia!Q % %1aV9bhs!'&B&B&BCCCCCrA c t j t 5 t j g dt j ddg ddg d d d n # 1 swxY w Y t j t 5 t j g dt j g t g d d d d S # 1 swxY w Y d S )NrQ r r r r! )r5 r
IndexErrorr9 r& r floatr s r? test_index_floatsz TestInsert.test_index_floatsO sQ
]:
&
& A A Iiii3*!5!5Bx @ @ @ A A A A A A A A A A A A A A A
]:
&
& @ @ Iiii"E!:!:!:B ? ? ? @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @s# /AA A92B88B<?B<idxrN r| c t j t d 5 t j g d|gddg d d d d S # 1 swxY w Y d S )Nz
out of boundsmatchrQ rZ rN )r5 r r r9 r& )rh r s r? test_index_out_of_boundsz#TestInsert.test_index_out_of_boundsU s
]:_
=
=
= 0 0 Iiii#A / / / 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0s AA A N)r r r rn r r r r r r r5 r6 r7 r r rA r? rl rl s 4 4 4((7 (7 (7T. . . < < <- - -
D D D@ @ @ [ UQG,,0 0 -,0 0 0rA rl c e Zd Zd ZdS )TestAmaxc g d}t t j | d g dg dg dg}t t j |d g d t t j |d g d
d S ) NrZ rN r[ r r^ r $@rZ r "@rN r r rx r r r r )r r r rR )r r r )r r9 amaxr s r? rn zTestAmax.test_basic] s & & &RWQZZ&&&
]]
^^
]]
RWQQ''')9)9)9:::RWQQ''')9)9)9:::::rA Nr r r rn r rA r? r r [ s# ; ; ; ; ;rA r c e Zd Zd ZdS )TestAminc g d}t t j | d g dg dg dg}t t j |d g d t t j |d g d
d S ) Nr g r r r r r )r r r rR )r r r )r r9 aminr s r? rn zTestAmin.test_basici s & & &RWQZZ&&&
]]
^^
]]
RWQQ''' 999RWQQ''' 99999rA Nr r rA r? r r g s# : : : : :rA r c e Zd Zd ZdS )TestPtpc t j g d }t | d d t j g dg dg dg }t | d g d t | d g d
t | dd g dg t | d
d dgg d S )Nr r r .@r r r )r @r ra )r r r Tr rb r )r9 r r ptpr s r? rn zTestPtp.test_basicu s H/// 0 0QUUU]]D))) Hmmm$nn#mm%
&
& QUUU]]OOO444QUUU^^___555QUUDU11OOO3DEEEQUUU55w?????rA Nr r rA r? r r s s(
@
@
@
@
@rA r c e Zd Zd ZdS )
TestCumsumc
g d}g dg dg dg}t j t j t j t j t j t j t j t j t j t j
f
D ]}t j || }t j || }t j g d| }t t j
|d | t j g dg dg d g| }t t j
|d | t j g d
g d g d g| }t t j
|d
| d S )NrR rP r r ry r[ rN rR rP rZ rN r[ ry rz r; r rZ rN r[ )rR rZ r_ r{ # ' r r )ry rx r r_ ) r )rR rZ ry r )r[ r r )r r_ rR )r9 rX re int16uint16int32uint32r r complex64
complex128r r
cumsum)rh baba2ctyperi a2tgts r? rn zTestCumsum.test_basic sc
$
$
$||\\\===9grx29bhiRZ m% ; ;E U##A#u%%B(555u==C ry 3 3 3S 9 9 9( ~~~/?/?/?@%I IC ry! 4 4 4c : : :( 1A1A1ABEK KC ry! 4 4 4c : : : : ; ;rA Nr r rA r? r r s# ; ; ; ; ;rA r c e Zd Zd ZdS )TestProdc g d}g dg dg dg}t j t j t j t j t j t j t j t j fD ] }t j || }t j || }|dv rBt t t j | t t t j |d st | d d t | d t j g d
| t | d t j g d | d S )
Nr r r r 1r rR r r g rp $ T ra )r{ b X )r9 r r r r r r r r r r
ArithmeticErrorprodr r
rh r r r ri r s r? rn zTestProd.test_basic sT
$
$
$||\\\===9h 28RYj"*blBMK E EEU##A#u%%B
"" orw::: orwA>>>> QVVV^^U333"2777??#%8,=,=,=u#E#EG G G"2777#3#3#%8OOOU#C#CE E E E E ErA Nr r rA r? r r s( E E E E ErA r c e Zd Zd ZdS ) TestCumprodc
t g d}g dg dg dg}t j t j t j t j t j t j t j t j fD ]L}t j || }t j || }|dv rbt t t j | t t t j |d t t t j | t t j |d t j g d | t t j |d
t j g dg d g d g| t t j |d t j g d
g dg dg| Nd S )Nr r r r r rR ra r )rR rP r i( i r r )r[ rQ r r )rR rP ry r{ )r[ r r )r r r r )r9 r r r r r r r r r r
r cumprodr
r s r? rn zTestCumprod.test_basic s
$
$
$||\\\===9h 28RYj"*blBMK J JEU##A#u%%B
"" orz1=== orz2qAAA orz1===="2:ab#9#9#9#%8 -@ -@ -@AF$H $HI I I #2:bq#9#9#9#%8\\\-<__->->->-@AF$H $HI I I #2:br#:#:#:#%8]]]-?-?-?-?-?-?-ABG$I $IJ J J J! J JrA Nr r rA r? r r s( J J J J JrA r c > e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d S )
TestDiffc g d}t j g d }t j g d }t j ddg }t t | | t t |d | t t |d | g d }t j g d
}t t | | g d }t j g d }t j d
d
g }t t | | t t |d | d S )N)rR rN ry rz rQ )rZ rP rR r[ )ra ra rN r r[ rP r= rZ )皙?皙@r gɿ皙)r g?g 皙?)TTFFr< T)r9 r r
r r )rh r outout2out3s r? rn zTestDiff.test_basic s1 h|||$$x $$xA 477C(((4Q<<<...4Q<<<... ' ' 'h,,,--DGGS))) & & &h+++,,xt %%477C(((4Q<<<.....rA c , t j d }d|d d dd dd d f< t j d }d|d d dd dd d f< t t | t j d t t |d t j d t t |d t j d t t |d | t t |d
| t t j t |d t t j t |d t j d
t j }t t t | d S )N)r r r rR rP )r r ra )r r r r )r; r r rX rZ r| gq ?)
r9 rC rf r
r r
r r r re )rh r exps r? test_axiszTestDiff.test_axis sY H\ " "!!!QTT111*
gl## AAAqt!tQQQJ477BH\$:$:;;;4+++RXl-C-CDDD4???BH[,A,ABBB4???C0004+++S111blD!!4444blD!"5555 H]BJ / /j$*****rA c < dt ddd z }|d d d d dd f |d d d d d df z
}|d d d d dd f |d d d d d df z
}|dd d d d d f |d dd d d d f z
}|dd d d d d f |d dd d d d f z
}t t | | t t |d | t t |d | t t |dd
| d S ) Nr r r rR ra rP r r r )r= r )r r
r )rh r out1r r out4s r? r zTestDiff.test_nd s_ b"b!! !AAAqrr{Qqqq!!!SbSy\)AAAqqq!""H~QQQ3B3Y/QQQ{QssAAAqqqy\)ABB111H~SbS!!!QQQY/477D)))4Q<<<...4???D1114QQ///66666rA c
x t t d t t t d fdt dd D }ddgdgg g g}t t d u t
t || d D ]\ }\ }}t t | t j
u t || t |j
t j t t | t! dt |z
d S ) NrZ ra r c 2 g | ]}t | S )r )r )r, r= r s r? r- z#TestDiff.test_n.. s# 4441$qA,,,444rA rR r[ r start)listrg r
re r r enumeraterD r r9 r r
r r rc rC max)rh rH expectedr= r r s @r? test_nzTestDiff.test_n s" qNNj$R00004444a 444FQCR(Q! !""""+C&,A,A"K"K"K 7 7 A# DII+ , , , sH - - - BG , , , S3q#a&&1*#5#5 6 6 6 6 7 7rA c t j ddt j }t j ddgd t j dgd g}| t j g d gdz t |d D ]@\ }}t
|| }t || t |j |j Ad S )
Nz
1066-10-13z
1066-10-16r! rR timedelta64[D]r rZ r r )
r9 r:
datetime64r extendr r r
r r )rh r r r= r r s r?
test_timeszTestDiff.test_times s IlL
F F F HaV#3 4 4 4 HaS 0 1 1 1
",<===>BCCC 222 / /FAsqA,,,C sC ( ( ( CI . . . . / /rA c > t j ddgddgddgddgd d
ggd d gd d gd d gd d gd d gg
}t | }t |j dgdgdgdgdgg t |j d gd gd gd gd gg t
t | t | u t |d }t |j g g g g g g t |j g g g g g g t
t | t | u d S )NrR rP rZ rN r[ ry rz rx r; r FTmaskr )r r r r
r> r r r )rh r r r s r? r zTestDiff.test_subclass sV Hq!fq!fq!fq!fq"g> !5>D%= !4=4,H
I
I
I 1gg38qcA3aS1#%>???38ugv'+ftfug&? @ @ @S T!WW$%%%A|||49r2r2r&:;;;49r2r2r&:;;;T
d1gg%&&&&&rA c 0 t j d dz }t t |d t j d t t |dg t j d t t j t j |d | t t |ddg t j d t j d dd }t j |dd }ddgddgg}t || t j |ddgdgg }t || t j |dd }ddgddgg}t || t j |dddgg }t || t t t j |t j d
t t j
t |dd d S )
Nr[ rR r )prependra ry rN rP )r r
rZ rZ rZ )r
r ) r9 r: r
r rf r r{ r
re rC r rh r resultr s r? test_prependzTestDiff.test_prepend s IaLL1 41---rwqzz:::4A3///<<<29RWQ%:%:%:;;Q???4B7333RWQZZ@@@ IaLL A & &A...FQF#68,,,aS1#J77768,,,A...FQF#68,,,aVH55568,,,j"'1bhuooFFFFblD!QQ??????rA c r t j d }t |d }g d}t || t |dg }t || t |ddg }|dgz }t || t j d dd }t j |dd }dd gdd
gg}t || t j |ddgdgg }t || t j |dd }ddgd d
gg}t || t j |dddgg }t || t t t j |t j d t t j t |dd
d S )Nr[ r )append)rR rR rR rR r| rP rN rR )r r ra r^ rX r rZ )r r ) r9 r: r r
r{ r
re rC r r s r? test_appendzTestDiff.test_append7 s IaLLa"""###68,,,a$$$68,,,aA'''qc>68,,, IaLL A & &1---GaW%68,,,QC!:66668,,,1---FRH%68,,,QF844468,,,j"'1RXe__EEEEblD!AA>>>>>>rA N) r r r rn r r r r r r r r rA r? r r s / / /&+ + + 7 7 7
7
7
7
/
/
/
'
'
' @ @ @0? ? ? ? ?rA r c e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z e j
d
ej d g d gg d Zd
ZdS )
TestDeletec t j d | _ t j d d ddd | _ d S )Nr[ rP rR )r9 r: ri r} r{ nd_ar s r? setup_methodzTestDelete.setup_methodV sD 1 IaLL''**221a;; rA c ` t | j | }t | j |d }d|z }t t || j |f | j | t |dd d df | j d|df }t || j dd d df | d S )NrR r zDelete failed for obj: %r)err_msgr )r ri r r
r/ )rh indicesa_delnd_a_delmsgxors r? _check_inverse_of_slicingz$TestDelete._check_inverse_of_slicingZ s tvw''$)W1555)G38E46'(+<==tv#& ( ( ( (x!!!Q'1gq=)ABB3 !AAAq& 13??????rA c g d}g d}|D ]2}|D ]-}|D ](}t ||| }| | ).3d S )N)rX r rR rP rN r[ )r^ ra rR rZ )rt r! )rh limsstepsr stopstepss r? test_sliceszTestDelete.test_slicesc s &&& 6 6E
6
6 ! 6 6D eT400A22155556
6 6 6rA c | t j ddgddgg t j t
5 t
| j dg d d d n # 1 swxY w Y t j t
5 t
| j dg d d d n # 1 swxY w Y | g d | g d t j t 5 t
| j d d d d n # 1 swxY w Y t j t 5 t
| j d d d d n # 1 swxY w Y t j t 5 t
| j d gd
z d d d d S # 1 swxY w Y d S ) Nr rR rP rh i)r ra rP rP )TFFTFTFrN ) r! r9 r r5 r r r ri re r s r?
test_fancyzTestDelete.test_fancyl s &&rx!Q!Q0@'A'ABBB
]:
&
& " " 46C5 ! ! ! " " " " " " " " " " " " " " "
]:
&
& # # 46D6 " " " # # # # # # # # # # # # # # #
&&}}}555 &&'H'H'HIII ]:
&
& ! ! 464 ! ! ! ! ! ! ! ! ! ! ! ! ! ! !
]:
&
& " " 465 ! ! ! " " " " " " " " " " " " " " " ]:
&
& & & 46E719 % % % & & & & & & & & & & & & & & & & & &sZ A**A.1A.B11B58B5D%%D),D) E++E/2E/F66F:=F:c Z | d | d d S )Nr r| )r! r s r? test_singlezTestDelete.test_single s0 &&q))) &&r*****rA c F t j d }t j t j 5 t |g d d d d n # 1 swxY w Y t j t 5 t |g d d d d d S # 1 swxY w Y d S )NrR r r r )r9 r r5 r r r rd ru s r? r zTestDelete.test_0d s HQKK
]2<
(
( " " 1bq ! ! ! ! " " " " " " " " " " " " " " "
]9
%
% + + 1bz * * * * + + + + + + + + + + + + + + + + + +s# AAA6BB Bc T G d dt j }| j | }t t t
|d | t t t
|g | t t t
|ddg | t t t
|t dd | t t t
|t dd | d S )Nc e Zd ZdS )*TestDelete.test_subclass..SubClassNr r rA r? r r1 r rA r r rR rP rX )r9 r ri r r r r rt r s r? r zTestDelete.test_subclass s rz FKK ! !
6!Q<<22333
6!R==(33444
6!aV,,h77888
6!U1a[[118<<===
6!U1b\\22H==>>>>>rA c * t j d ddd }t |t dd d }t |j j |j j t |j j |j j d S ) Nr rP r[ r r < rR r ) r9 r: r{ r rt r r r r )rh r` rB s r? test_array_order_preservez$TestDelete.test_array_order_preserve s| IbMM ! !!Qc ! 2 2 1eBooA . . . QW)17+?@@@QW)17+?@@@@@rA c t j t 5 t j g dt j ddg d d d n # 1 swxY w Y t j t 5 t j g dt j g t d d d d S # 1 swxY w Y d S )NrQ r r r! )r5 r r r9 r r r r s r? r z TestDelete.test_index_floats s$
]:
&
& 7 7 Iiii3*!5!5 6 6 6 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7
]:
&
& < < Iiii"E!:!:!: ; ; ; < < < < < < < < < < < < < < < < < > 271::rx8 < < < = = = > > > > > > > > > > > > > > > > > >s$
=EE E:8F??GGN)r r r r r! r) r+ r- r r r4 r r5 r6 r7 r9 r r: rA r rA r? r r T s < < <@ @ @6 6 6& & &*+ + ++ + +? ? ?A A A< < < [ Y1# (<==- - >=-> > > > >rA r c 6 e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
Z d Z
ej d ej ej ej ej g d
Zej d ej ej ej ej g d Zej dej ej ej ej g d Z ej dej ej ej ej g d Z d Z dS ) TestGradientc ddgddgg}t j | }t j ddgddgg t j ddgddgg g}t t | | t t | | d S )NrR rZ rN r r r r )r9 r r
r" rh vr dxs r? rn zTestGradient.test_basic s VaV HQKKhR2r(+,,hR2r(+,,.8A;;+++8A;;+++++rA c
t j t j d }g d}t j d dd }t t j d d t t j d t j d t t j d | t |d t |t j d t ||| t ||d t ||d t t d t
|t j |gdz d
d d S ) Nr[ )r r r r &@ r ?rP rR r z.*scalars or 1dra )
r9 r rf r: r{ r" r r re stack)rh rG dx_unevenf_2ds r? test_argszTestGradient.test_args s9
Yrwqzz
"
"))) y}}$$Q** 1 r"""1 rx||,,,1 r"""srx}}%%%y),,, r1 r"""" J(9 dBHbT!V" 5 5 5q : : : : :rA c H t j d dd }t j t j d }t t t ||t j d t t t |dt j d t t t |t j d t j d t t t || t t t ||d t t t |||| t t t |ddd t t t |||d t t t |ddd d S )NrJ r[ rP rR rb r ) r9 r: r{ r rf r
re r" rd )rh rN r s r? test_badargszTestGradient.test_badargs s3 y}}$$Q** Ibgajj ! ! j(D!RWQZZ@@@j(D!RWQZZ@@@j(D"'!**bgajjIIIi4333i4????i4Aq999i4Aq999i4AA>>>>i4AA>>>>>>rA c t j g dd }t j g dd }t t | | t |j t j d k d S )N)z
1910-08-16z
1910-08-11z
1910-08-10z
1910-08-12z
1910-10-12z
1910-12-12z
1912-12-12z
datetime64[D]r! )r r^ r = i i r r9 r r
r" r r rh r rG s r? test_datetime64z TestGradient.test_datetime64 s H
7
7
7!
#
#
# X ) ) )"$ $ $ 8A;;+++BH%5 6 6677777rA c t j ddgddggddgddgg }t | d }t t | t | t
|j |j u t j d }t j j |d< t j |d t |j g d
d S ) NrR rZ rN Fr
r r[ rP
edge_order)FFTFF) r9 r r r" r r r _maskr: maskedr
r )rh r r x2s r? test_maskedzTestGradient.test_masked s EKK!Q!Q( $enuen =
?
?qkk!nT#YYQ((( sy()))
U\\!__ 1
B1%%%%27$F$F$FGGGGGrA c t j ddd }|d |d z
}d|dz z d|dz z z d|z z }d|dz z d|z z dz }t j t j ||d |z dz
}t t j |d
k d k t j d t j t j d }d|dz z d|dz z z d|z z }d|dz z d|z z dz }t j t j ||d |z dz
}t t j |d
k d k d S ) Nr rR r rP rZ rN ry rx rY g Q?T) r9 linspaceabsr" r r randomseedsort)rh r rG r
analytical num_errors r? test_second_order_accuratez'TestGradient.test_second_order_accurate sw
K1b ! !
qTAaD[
QJQ!V #a!e +aZ!a%'!+
FBK2! < < ">">FF '"A"A"A5II &'
$6M8J#K . . JAr*EEEDAvv!&:? ? ?DAvv!%*> > >D tT * * * tT * * * Q 3 3 3 Q 1 1 1ArjAAADAvA*EEED DJ$*, - - - gi 0 0 0ArjAAADAvA*EEED DJ$*, - - - tW - - - - &'$81o:N#O
/
/ JAx!&:? ? ?DAx!%*> > >D tT * * * Q 3 3 3 Q 1 1 1AxaJGGGD gi 0 0 0AxaJGGGD g . . . . 68%AFFF8V%AFFF47DG,,,47DG,,,DG]_555DG_55568%AFFF8V%AFFF47DG,,,47DG,,,DG]_555DG_55555rA c ddgddgg}t j | }t j ddgddgg t j ddgddgg g}t t |d |d t t |d |d t t |d
|d t t |d |d |d g t t |d |d |d g t t |d t | t t |d dd |d dz |d dz g t t t |dd d t t j t |d t t j t |d
d S )NrR rZ rN r r r r r r ra rc rP r^ )r9 r r
r" r r
rd r rE s r? test_specific_axesz TestGradient.test_specific_axesk s VaV HQKKhR2r(+,,hR2r(+,,.8AA...16668AA...16668AB///A7778AF333beRU^DDD HQT 2 2 2RUBqENCCCHQT 2 2 2HQKK@@@ 8Aq!&999 qE#Ir!uSy1 3 3 3 i1a;;;;blHaa8888blHab999999rA c t j g dd }t j g dd }t t | | t |j t j d k d S )N)r r^ r rQ rT iA , r r! )rP rz rz rJ w irU rV s r? test_timedelta64z TestGradient.test_timedelta64 s H * * *"
$
$
$ X ( ( ("$ $ $ 8A;;+++BH%5 6 6677777rA c t j t j t j fD ]S}t j g d| }t t
| j t j | j Td S )Nr r! ) r9 float16r r r r r" r r )rh dtr s r? test_inexact_dtypesz TestGradient.test_inexact_dtypes sg :rz2:6 > >B"---A !*BGAJJ,< = = = = > >rA c h t t j d d t t j d d t t t t j d d t t t t j d d t t t t j d d t t t t j d d t t t t j d d d S )NrP rR rY rZ r )r" r9 r: r
re r s r? test_valueszTestGradient.test_values s 1 !,,,,1 !,,,,j(BIaLLQGGGGj(BIaLLQGGGGj(BIaLLQGGGGj(BIaLLQGGGGj(BIaLLQGGGGGGrA f_dtypec t j g d| }t | }t |dgt | z d S )N)r[ rN rZ rP rR r! ra r9 r r" r
rC )rh r ro gs r? test_f_decreasing_unsigned_intz+TestGradient.test_f_decreasing_unsigned_int sI
H___G 4 4 4 QKK1rd3q66k*****rA c t j | j }t j ddg }t j d|g| }t || }t ||dz dz gdz d S )NrR rZ ra r! rP )r9 iinfor r r" r
)rh r maxintr ro dfdxs r? test_f_signed_int_big_jumpz'TestGradient.test_f_signed_int_big_jump su '""& HaV Hb&\ 1 1 11~~46A:!"3!4Q!677777rA x_dtypec t j g d| }t j g d }t || }t |dgt | z d S )Nr r! r rP rN rX r )rh r r ro r s r? test_x_decreasing_unsignedz'TestGradient.test_x_decreasing_unsigned s^
HYYYg . . . HYYY 1~~4"c!ff-----rA c t j | j }t j | j }t j d|g| }t j |dz dg }t || }t
|ddg d S )Nra r! rP r ri )r9 r minr r r" r
)rh r minintr r ro r s r? test_x_signed_int_big_jumpz'TestGradient.test_x_signed_int_big_jump s '""&'""& Hb&\ 1 1 1 Hfk1% & &1~~4#s,,,,,rA c t j ddgddgf }t j rt | t u sJ d S t | t
u sJ d S )NrR rP rZ )r9 r" _using_numpy2_behaviorr tupler rh r> s r? test_return_typez TestGradient.test_return_type si kAq6Aq6*++
$ & & %99%%%%%%99$$$$$$rA N) r r r rn rO rQ rW r^ rg rz r| r r r r5 r6 r7 r9 re r r uint64r rX r r int64r r r r r rA r? rC rC s , , ,: : :2? ? ?
8
8
8
H
H
H 2 2 2(>6 >6 >6@: : :4 8 8 8> > >
H
H
H [ Y29)+BI)? @ @+ +@ @+
[ Y"()+28)= > >8 8> >8 [ Y29)+BI)? @ @. .@ @. [ Y"()+28)= > >- -> >-% % % % %rA rC c e Zd Zd Zd ZdS ) TestAnglec P dt j d dz dt j d z dz z dddddd g}t | }t j d
t j d d t j dz t j t j dz t j d
t j t j d
z
g}t |d
}t j | dz t j z }t
||d t
||d d S )N ? @rP r rV rR ra ? y @r r r T)degr r )r9 sqrtr arctanpir r )rh r r yozzos r? rn zTestAngle.test_basic s
RWQZZ#%RWQZZ!(;;
BVW
. !HH Ii IcNNAruqy"%"%#
Yy
!
! !2529Y+?+?#?A
!
Xb\\C
"%
'!!R,,,!!R,,,,,rA c t j ddt j d dz dz g }t j j |d< t j t j d dt j d g }t j j |d< t
| }t t | t | t |j |j t || d S )Nr rR rP ? ?r r r )
r9 r r r r\ r r r r r )rh r r r s r? r zTestAngle.test_subclass s EKKBGAJJqLF$;< = =u|!5;; ) 4 4a3 HIIel qT&\\4>>222V[(-000VX&&&&&rA N)r r r rn r r rA r? r r s2 - - -' ' ' ' 'rA r c e Zd Z ej g d Ze e Ze e Z
e e Z d Z
d Zd Zd Zd Zd Zej d ej g d ej g d
ej g d g d Zd
Zd ZdS )
TestTrimZeros)r r rR r rP rZ rN r c $ d} fd|D S )N)ri r r r\ c 3 8 K | ]}t | V d S r8 )getattr)r, namerh s r? z'TestTrimZeros.values.. s- ;;d##;;;;;;rA r )rh
attr_namess ` r? rF zTestTrimZeros.values s" )
;;;;
;;;;rA c t j dd }| D ]'}t | }t ||| (d S )NrP ra r9 s_rF r2 r
rh slcarrr> s r? rn zTestTrimZeros.test_basic sT eAbDk;;== . .C S//C sCH - - - - . .rA c t j d d }| D ])}t |d }t ||| *d S )Nra r trimr r s r? test_leading_skipz TestTrimZeros.test_leading_skip s[ eCRCj;;== . .C Ss+++C sCH - - - - . .rA c t j dd }| D ])}t |d }t ||| *d S )NrP r r r r s r? test_trailing_skipz TestTrimZeros.test_trailing_skip s[ eABBi;;== . .C Ss+++C sCH - - - - . .rA c | D ]i}t j ||j }t |d }t | dk sJ t |d }t | dk sJ jd S )Nr! Br r ro )rF r9
zeros_liker r2 rC )rh _arrr rw rx s r?
test_all_zerozTestTrimZeros.test_all_zero s KKMM " "D-DJ777C c,,,Dt99>>>> c,,,Dt99>>>>> " "rA c l t j d }t | }t || d S rb )r9 rC r2 r
rh r r> s r? test_size_zeroz TestTrimZeros.test_size_zero
s0 hqkkoo3$$$$$rA r )r l r )r l r )r l r c x t j dd }t | }t ||| d S )NrR rP )r9 r r2 r
)rh r r r> s r?
test_overflowzTestTrimZeros.test_overflow s8 eAaCjoo3C)))))rA c p t j g d }t | }t || d S )N)NrR N)r9 r r2 r
r s r? test_no_trimzTestTrimZeros.test_no_trim s6 h''oo3$$$$$rA c t | j }t |t sJ d S r8 )r2 ri tolistr r r s r? test_list_to_listz TestTrimZeros.test_list_to_list" s5 ))#t$$$$$$$rA N)r r r r9 r ri r r r complexr r= r\ rF rn r r r r r5 r6 r7 r r r r rA r? r r sF )))**A A A A< < <. . . . . . . . . " " "% % %
[
--- --- --- " * *
*
% % %
% % % % %rA r c e Zd Zd Zd Zd ZdS )
TestExtinsc ~ t j g d }t |dk | }t |g d d S )N)rR rZ rP rR rP rZ rZ rR )rZ rP rP rZ rZ )r9 r r r
r s r? rn zTestExtins.test_basic) sE H*** + + AE1 1ooo.....rA c J t t t g dddgddg t j g d t g dg d t g d t t j d
g t t j dd t g d d d
g t g d t t dfd t j ddg t ddgd t ddg d S )Nr TFr rR )rR rN rZ rP r[ rx rz )r rR r rR r rR r rP rN ry )rR rP rZ rN r[ ry rz rz rx )rR r rR r rR r rR r; )rx rP r; rN rx ry r; z!Cannot insert from an empty arrayc * t g dg S )N)r r r r r rR r )r, ri s r? z'TestExtins.test_place..= s E!-B-B-BB$G$G rA 12349)
r
rd r, r9 r r
rC r: r re ru s @r?
test_placezTestExtins.test_place. sD i D%=1a&III H*** + +
a&&& 2221 3 3 3444
a!b!!!1bi1oo...
a&&&A///1 3 3 3444J(K G G G G I I I
HdD\ " "
a!Q 1tSk*****rA c t d }|dk }| }t || }t ||d t ||| t || d S )Nr ri r )r r r r, r
)rh ri r acr s r? test_bothzTestExtins.test_bothD sl HH3w
VVXX D!
aq
aq1b!!!!!rA N)r r r rn r r r rA r? r r ' sA / / /
+ + +," " " " "rA r r c 0 |t j | z S r8 mathfloorr r s r? _foo1r P s TZ]]?rA r c 6 |t j | z |z S r8 r )r r r s r? _foo2r T s TZ]]?Q rA c e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
Z d Z
d Zd
Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Z d Z d Z d Z d Z d Z!d Z"d! Z#d" Z$d# Z%d$ Z&d% Z'd& Z(d' Z)d( Z*d) Z+d* Z,d+ Z-d,S )-
TestVectorizec n d }t | } |g dg d }t |g d d S )Nc " | |k r| |z
S | |z S r8 r ri r s r? addsubtractz.TestVectorize.test_simple..addsubtract[ 1uu1u 1u rA r rZ ry r; rR rZ r[ rz rR ry rR rP r5 r
rh r ro r s r? test_simplezTestVectorize.test_simpleZ sS
k " "
AlllLLL ) )1lll+++++rA c j d }t | } |g dd }t |g d d S )Nc " | |k r| |z
S | |z S r8 r r s r? r z.TestVectorize.test_scalar..addsubtractf r rA r r[ )r[ rx rR rN r r s r? test_scalarzTestVectorize.test_scalare sO
k " "
AlllA 1lll+++++rA c t j ddd }t d } || }t || d S )Nr^ rP i' c | S r8 r r s r? r z*TestVectorize.test_large..r rA )r9 r` r5 r
)rh r ro r s r?
test_largezTestVectorize.test_largep sG KAu % % kk " "
AaDD1a rA c * t t j }t j ddt j z t j dt j z dt j z g } || }t j | }t
|| d S )Nr ri rK rP )r5 r cosr9 r r r )rh ro argsr1r2s r?
test_ufunczTestVectorize.test_ufuncv so dh xC"%Kbe QYGHH
QtWW
VD\\!"b)))))rA c dd}t | }t j g d } || }t j g d }t || ||d }t j g d }t || d S )NrR c | |z S r8 r r s r? r z(TestVectorize.test_keywords..foo q5LrA r r rP rY r0 r5 r9 r r
rh r ro r r r s r?
test_keywordszTestVectorize.test_keywords} s
cNNx ""
QtWW
Xiii
2r"""
QtQZZ
Xiii
2r"""""rA c t t t g } |t j d d } |t j d }t || d S Notypesr r r5 r r r9 r: r
rh ro r r s r? test_keywords_with_otypes_order1z.TestVectorize.test_keywords_with_otypes_order1 s`
eUG , , , Qry~~s
#
#
Qry~~
2r"""""rA c t t t g } |t j d } |t j d d }t || d S r
r r s r? test_keywords_with_otypes_order2z.TestVectorize.test_keywords_with_otypes_order2 s`
eUG , , , Qry~~
Qry~~s
#
#2r"""""rA c 0 t t t g } |t j d } |t j d d } |t j d }t || t || d S )Nr r r r r rh ro r r r3s r? test_keywords_with_otypes_order3z.TestVectorize.test_keywords_with_otypes_order3 s
eUG , , , Qry~~
Qry~~
%
%
%
Qry~~
2r"""2r"""""rA c > t t t g } |dd } |dd } |d }t |t dd t |t dd t |t d d S )Nr $@rh )r ra r r5 r r r r s r? +test_keywords_with_otypes_several_kwd_args1z9TestVectorize.test_keywords_with_otypes_several_kwd_args1 s
eUG , , , Qts^^^
Qtr]]]
QtWWRts+++,,,Rtr***+++Rt%%%%%rA c t t t g } |ddd } |ddd }t |t ddd t |t ddd d S ) Nr rh r ra )r r r rR rP rZ r r s r? +test_keywords_with_otypes_several_kwd_args2z9TestVectorize.test_keywords_with_otypes_several_kwd_args2 s
eUG , , , Q
#
#
#
Qq!QZZR333444Rq!Q (((((rA c p dd l } t |j d S # t $ r t w xY wrb )rb r5 randrange ExceptionAssertionError)rh rb s r? test_keywords_no_func_codez(TestVectorize.test_keywords_no_func_code sP
# f& ' ' ' ' ' # # # "" " #s 5c \ d
d}t | }t j g d } || }t j g d }t || |d| }t || ||d }t j g d }t || d S ) NrR c | |z S r8 r r s r? r z5TestVectorize.test_keywords2_ticket_2100..foo r rA r r r )r ri rP r rY r0 r r
s r? test_keywords2_ticket_2100z(TestVectorize.test_keywords2_ticket_2100 s
cNNx ""
QYYY
Xiii
2r"""
Qd^^^2r"""
Qtq\\\
Xiii
2r"""""rA c d }t j |ddg }ddg}t | |ddgg d t | |ddgg d
t | |ddgg d d S ) Nc t | }| d }|r || z | d z }| |S rb )r pop)r p_pr> s r? mypolyvalz;TestVectorize.test_keywords3_ticket_2100..mypolyval sK aB&&))C
*Agq )
*JrA r, rR )excludedrZ ry r r )r r, )r, )r9 r5 r
)rh r. vpolyvalanss r? test_keywords3_ticket_2100z(TestVectorize.test_keywords3_ticket_2100 s < S!H===!f3Aq6YYY ? ? ?@@@3!Q999 = = =>>>3!Q ; ;<<<< _ks r? ro z3TestVectorize.test_keywords4_ticket_2100..f s* C
r"v
JrA rR rP rZ rN r rx r rh ro s r? test_keywords4_ticket_2100z(TestVectorize.test_keywords4_ticket_2100 sN
111v!Q0001a&99999rA c f t d }t |ddgddg ddg d S )Nc * t j | S r8 )r9 r )rF s r? ro z3TestVectorize.test_keywords5_ticket_2100..f s 71:: rA rR rP rZ rN rx r r8 s r? test_keywords5_ticket_2100z(TestVectorize.test_keywords5_ticket_2100 sK
11aVaV,,q!f55555rA c Z d }t | }t | d d S )Nc dS NrR r r rA r? r z5TestVectorize.test_coverage1_ticket_2100..foo 1rA rR r rh r ro s r? test_coverage1_ticket_2100z(TestVectorize.test_coverage1_ticket_2100 s:
cNN1133"""""rA c d }t | }t |j |j d}t || }t |j | d S )Nc | S )zOriginal documentationr r s r? r z3TestVectorize.test_assigning_docstring..foo HrA zProvided documentation)doc)r5 r __doc__)rh r ro rF s r? test_assigning_docstringz&TestVectorize.test_assigning_docstring sa
cNNQY ,,,& cs # # #QY$$$$$rA c G d d }t t | j t j d t j d dz t t |j | t j d t j d dz d S )Nc e Zd ZdZd ZdS )9TestVectorize.test_UnboundMethod_ticket_1156..FoorP c || j z S r8 r' ru s r? r z=TestVectorize.test_UnboundMethod_ticket_1156..Foo.bar s DF{"rA N)r r r r r r rA r? FoorK s( A
#
#
#
#
#rA rM r; rP )r
r5 r r9 r: )rh rM s r? test_UnboundMethod_ticket_1156z,TestVectorize.test_UnboundMethod_ticket_1156 s # # # # # # # # /9SSUUY// ! == 9Q<<1, . . .-9SW--cceeRYq\\BB 9Q<<1, . . . . .rA c r t d } |t j d } |t j dd }t d } |t j dd } |t j d }t || t || d S )Nc | S r8 r r s r? r z@TestVectorize.test_execution_order_ticket_1487..# rA rZ r c | S r8 r r s r? r z@TestVectorize.test_execution_order_ticket_1487..& rQ rA )r5 r9 r: r )rh f1res1ares1bf2res2bres2as r? test_execution_order_ticket_1487z.TestVectorize.test_execution_order_ticket_1487! s
{{
#
#29Q<< 29S!$$%%
{{
#
#29S!$$%%29Q<< UE"""UE"""""rA c f t j d }d}t | || d S )Nc | S r8 r r s r? r z7TestVectorize.test_string_ticket_1892... 1 rA d0123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789)r9 r5 r )rh ro r( s r? test_string_ticket_1892z%TestVectorize.test_string_ticket_1892, s6 L % % Q! rA c dgt fd }d|_ t j d }t || ||z t d t
| d S )Nr c . dxx dz
cc< | dz S )Nr rR rP r )r _callss r? ro z#TestVectorize.test_cache..f6 s! 1IIINIII6MrA Tr[ )r5 cacher9 r: r
r rC )rh ro r ra s @r?
test_cachezTestVectorize.test_cache2 s
IaLL11Q44Q'''VAYA'''''rA c t j d }d|_ t j d }t || | d S )Nc | S r8 r r s r? r z+TestVectorize.test_otypes..A r\ rA r r[ )r9 r5 r r: r
rh ro r s r? test_otypeszTestVectorize.test_otypes@ sE L % % IaLL11Q44#####rA c $ t t j d dgdgf t t j d dgdgf t t j d ddgdgf t t j d dgdgf t t j d dgddgf t t j d
g d d gf t t j d
dgdgf t t j d dgdgf t t j d ddgdgf t t j d dgdgf t t j d dgddgf t t j d g d d gf t t 5 t j d d d d n # 1 swxY w Y t t 5 t j d d d d n # 1 swxY w Y t t 5 t j d d d d d S # 1 swxY w Y d S )Nz(x)->()r r z (x,y)->()r z (x),(y)->()r z(x)->(y)z (x)->(y),()z(),(a,b,c),(d)->(d,e))r )ri 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 nfb_parse_gufunc_signaturer
re r s r? test_parse_gufunc_signaturez)TestVectorize.test_parse_gufunc_signatureF s S0;;vh=MNNNS0==!lRD) + + +S0 ?? v&- / / /S0<< h) + + +S0 ?? h - / / /S01HII333j\B D D D S0<<x">NOOOS01BCC!lRD) + + +S0 AA v&- / / /S01ABB h) + + +S01ABB h - / / /S0AC C333j\B D D D :
&
& 6 6 ' 5 5 5 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6
:
&
& 5 5 ' 4 4 4 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5
:
&
& 6 6 ' 5 5 5 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6s6 #HH H#II I#JJ J c r d }t |d } |g dg d }t |g d d S )Nc " | |k r| |z
S | |z S r8 r r s r? r z8TestVectorize.test_signature_simple..addsubtracti r rA z (),()->() signaturer r r r r s r? test_signature_simplez#TestVectorize.test_signature_simpleh sX
k[ 9 9 9
AlllLLL ) )1lll+++++rA c t d }t |d } |ddgddgg }t |ddg d S )Nc * | S r8 r r s r? r z4TestVectorize.test_signature_mean_last..meant s 6688OrA z(n)->()ro rR rZ rP rN r )rh r ro r s r? test_signature_mean_lastz&TestVectorize.test_signature_mean_lasts s[
di 0 0 0
A1v1v 1q!f%%%%%rA c | d }t |d } |ddgddgg }t |ddgddgg d S ) Nc 0 | | z
S r8 rt r s r? centerz3TestVectorize.test_signature_center..center| s qvvxx< rA (n)->(n)ro rR rZ rP rN ra r )rh rx ro r s r? test_signature_centerz#TestVectorize.test_signature_center{ sd
f
3 3 3
A1v1v 1AwQ 011111rA c t d d } |g d }t t |t ot | dk t |d g d t |d g d d S )Nc
| | fS r8 r r s r? r z:TestVectorize.test_signature_two_outputs..
A rA ()->(),()ro r rP r rR )r5 r r r rC r
rh ro r s r? test_signature_two_outputsz(TestVectorize.test_signature_two_outputs s &&+ > > >
AiiiLL
1e$$4Q15551Q4+++1Q4+++++rA c t t j d } |ddgg d }t |g dg dg |ddgggg d }t |g dg dggg |ddgddggg d }t |g dg dgg dg dgg |ddgg dg dg }t |g dg dgg dg dgg d S ) Nz(a),(b)->(a,b)ro rR rP r r r r )r5 r9 r< r
r s r? test_signature_outerz"TestVectorize.test_signature_outer sM bh*: ; ; ;
Aq!fiii 1yyy))) 4555
AAxj))) $ $1III 6 7 8999
A1v1v * *1 999 5 ) 999 5 7 8 8 8
Aq!fyyy))), - -1 999 5 ) 999 5 7 8 8 8 8 8rA c t d d } |g d }t |ddg |g dg dg }t |ddgddgg d S ) Nc | d d S Nra r r s r? r z. #2# rA z(n)->(m)ro r rR rP r rZ r r s r? test_signature_computed_sizez*TestVectorize.test_signature_computed_size s| &&* = = =
AiiiLL1q!f%%%
Ayyy)))$ % %11v1v ./////rA c d
d}t |ddh }t |g d g d t |g dd g d d S ) NrR c | |z S r8 r r s r? r z2TestVectorize.test_signature_excluded..foo r rA ()->()r )rp r/ r r r r' r0 r rA s r? test_signature_excludedz%TestVectorize.test_signature_excluded s{
cX > > >11YYY<<33311YYY!,,,iii88888rA c t d ddg } |g d }t |j t j d t |g d d S )Nc | S r8 r r s r? r z5TestVectorize.test_signature_otypes.. r rA ry r rp r r )r5 r r r9 r
r s r? test_signature_otypesz#TestVectorize.test_signature_otypes sa kkZ L L L
AiiiLLQWbhy 1 12221iii(((((rA c F t t j d }t t d 5 |ddg d d d n # 1 swxY w Y t t
d 5 |dd d d d n # 1 swxY w Y t t
d 5 |ddgg d d d d n # 1 swxY w Y t t j d }t t d 5 |dd d d d d S # 1 swxY w Y d S )
Nz (n),(n)->(n)ro zwrong number of positionalrR rP z does not have enough dimensions$inconsistent size for core dimensionr r )r5 operatorr; r rd re r8 s r? test_signature_invalid_inputsz+TestVectorize.test_signature_invalid_inputs s* hln = = =
,H
I
I
Aq!fIII
=? ?
AaGGG ! BD D ! !
Aq!fiii ! ! ! ! ! ! ! ! ! ! ! ! ! ! !
hlh 7 7 7
,H
I
I
AaGGG sG A AA+
BB B$CCC<
DD Dc t d d }t t d 5 |g d d d d n # 1 swxY w Y t d d }t t d 5 |d d d d n # 1 swxY w Y t d
d }t t d 5 |d d g d d d d S # 1 swxY w Y d S )
Nc | d d S r r r s r? r z>TestVectorize.test_signature_invalid_outputs.. r rA ry ro r r c | S r8 r r s r? r z>TestVectorize.test_signature_invalid_outputs.. r rA r~ zwrong number of outputsrR c
| | fS r8 r r s r? r z>TestVectorize.test_signature_invalid_outputs.. r} rA r rP )r5 r re r8 s r? test_signature_invalid_outputsz,TestVectorize.test_signature_invalid_outputs s &&* = = =
BD D
AiiiLLL
kk[ 9 9 9
-F
G
G
AaDDD
&&( ; ; ;
-F
G
G
Aq!fIII s5 AA A4 B BB>CC C c t j d }t j ddgt }t t
d 5 || d d d n # 1 swxY w Y d|_ t || | t j d d }t t
d 5 || d d d n # 1 swxY w Y t j d
dd }t || | t j d d
d }t || | t j d d
}t ||j |j t j d dd }t t
d 5 || d d d d S # 1 swxY w Y d S )Nc | S r8 r r s r? r z5TestVectorize.test_size_zero_output.. r\ rA r r[ r! r r c | S r8 r r s r? r z5TestVectorize.test_size_zero_output.. r\ rA r ro c | S r8 r r s r? r z5TestVectorize.test_size_zero_output.. r\ rA r c | S r8 r r s r? r z5TestVectorize.test_size_zero_output.. r\ rA ry c | S r8 r r s r? r z5TestVectorize.test_size_zero_output.. r\ rA c | gS r8 r r s r? r z5TestVectorize.test_size_zero_output.. s A3 rA z()->(n)znew output dimensions) r9 r5 rC intr re r r
r rf s r? test_size_zero_outputz#TestVectorize.test_size_zero_output st L % % HaV3 ' ' '
X
6
6
AaDDD 11Q44### L 9 9 9
X
6
6
AaDDD
L E E E11Q44### L
3 G G G11Q44### L
; ; ;11QS6613''' L )C H H H
-D
E
E
AaDDD s6 A A$'A$7 CCC# F<<G G c \ G d dt j }t j g dg dg dg | }t j g dg dg dg | }t j t j d
} ||| }t
t | | t
|g d g d g d
g t j d } ||| }t
t | | t
|||z d S )Nc e Zd ZdS )/TestVectorize.test_subclasses..subclassNr r rA r? r r r rA r )r r r ro )r r r r )r r r )r r r z(m,m),(m)->(m)ro )r r r )r r r )r r r c | |z S r8 r r s r? r z/TestVectorize.test_subclasses.. s
1 rA )r9 r r r r5 matmulr r )rh r rB rF matvecr mults r? r z TestVectorize.test_subclasses sM rz
Hlll"ll"ll$
%
%%)T(^^
HlllLLL,,,? @ @ E Eh O O bi3CDDD F1aLLT!WWh'''Q |||\\\BCCC | , ,-- DAJJT!WWh'''QA rA c J t j d }|j dk sJ d S )Nc | |z S r8 r r s r? rV z#TestVectorize.test_name..f2 s q5LrA rV )r9 r5 r )rh rV s r? test_namezTestVectorize.test_name s8
{d""""""rA c j t d } |g dg d }t |g d d S )Nc " | |k r| |z
S | |z S r8 r r s r? r z1TestVectorize.test_decorator..addsubtract s 1uu1u 1u rA r r r r )rh r r s r? test_decoratorz TestVectorize.test_decorator sP
K lll 3 31lll+++++rA c n t d }t j j dk r
|j dk sJ d S d S )Nc | S ) Docstringr r s r? ro z'TestVectorize.test_docstring..f s HrA rP r )r5 sysr optimizerG r8 s r? test_docstringz TestVectorize.test_docstring sN
9 ! !9 ++++ " !++rA c v d }t |d }t j | } |d dk sJ d S )Nc | |z S r8 r r s r? r z'TestVectorize.test_partial..foo r rA rZ rR rN )r r9 r5 )rh r r vbars r? test_partialzTestVectorize.test_partial sH c1oo|C tAww!||||||rA c t ddg d } |g d }t |j t j d t |g d |j dk sJ d S )Nry r r c | S r8 r r s r? ro z8TestVectorize.test_signature_otypes_decorator..f# rE rA r ro )r5 r r r9 r
r r s r? test_signature_otypes_decoratorz-TestVectorize.test_signature_otypes_decorator" s Z < < <
= <
AiiiLLQWbhy 1 12221iii(((zS rA c t t 5 t j d }d d d d S # 1 swxY w Y d S )NrZ )pyfuncr
rd r9 r5 rh As r? test_bad_inputz TestVectorize.test_bad_input, s
9
%
% ) ) a(((A ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) )s 8<<c t t 5 t j d d }d d d d S # 1 swxY w Y d S )Nstringc dS )Nr r r rA r? r z+TestVectorize.test_no_keywords..foo2 s urA r )rh r s r? test_no_keywordsz TestVectorize.test_no_keywords0 s
9
%
%
\(
#
#
$
#
s A AAc t d dg } |dg }t |j t j d d S )Nc | S r8 r r s r? r z?TestVectorize.test_positional_regression_9477..: rQ rA r rP )r5 r r r9 r s r? test_positional_regression_9477z-TestVectorize.test_positional_regression_94776 sH
{{i[ 1 1
AqcFFQWbhy 1 122222rA N).r r r r r r r r r r r r r r$ r( r2 r9 r< rB rH rN rY r^ rc rg rl rq ru rz r r r r r r r r r r r r r r r r r r rA r? r r X s , , , , , ,! ! ! * * * # # # # # # # # #
#
#
# & & & ) ) )# # ## # #"
=
=
= : : :6 6 6# # #
%
%
% . . . # # # ( ( ( $ $ $ 6 6 6D , , ,& & &2 2 2, , ,8 8 8 0 0 09 9 9) ) )
6 (# # # , , ,, , , ! ! !) ) ) 3 3 3 3 3rA r c e Zd Z G d d Zej e d ej ddej fdg d Z
d S )
TestLeaksc . e Zd ZdZd Zed ZdS ) TestLeaks.Ar c dS rb r rh r s r? boundzTestLeaks.A.boundC r@ rA c dS rb r )r s r? unboundzTestLeaks.A.unboundF s 1rA N)r r r itersr staticmethodr r rA r? r r @ sC
rA r Python lacks refcountsreasonz
name, incrr )r r c dd l }t | j | }| t j | }t
| j j D ]f}| }t j t || dd |_
|
t j d }gd }t t j | ||z t
d D ]}|
t t j | | | d S # | w xY w)Nr rR r r[ )gcr r disabler getrefcountrg r r9
frompyfuncro r: r collectenable) rh r incrr A_funcrefcountr ri r s r? test_frompyfunc_leaksz TestLeaks.test_frompyfunc_leaksJ s% &&
v..H46<((
)
)FFHHmGAt$4$4a;;cc")B--((A 00(T/ B B B1XX
00( ; ; ; IIKKKKKBIIKKKKs DE E N) r r r r r5 r6 skipifr r7 r r r rA r? r r ? s [L(1IJJ [ \
ag , KJ
rA r c e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
Z d Z
ej d
d ZdS ) TestDigitizec t j dd }t j dd }t t || t j d d S )Nr# r[ r r r9 r: r
r rh r binss r? test_forwardzTestDigitize.test_forwardj sL Ib! yQ 8At,,bimm<<<<.ANr r rA r? r r r rA r r[ rR rZ FT)r9 r r: r r r r )rh r ri r s r? r z TestDigitize.test_return_type s
IaLL a IaOO # #Jx1e 4 4a888999Jx1d 3 3Q77788888rA c ` d}t t j ||dz
|dz g d d S Nl rR r r9 r r s r? test_large_integers_increasingz+TestDigitize.test_large_integers_increasing s5 R[QUAEN33Q77777rA z8gh-11022: np.core.multiarray._monoticity loses precisionr c ` d}t t j ||dz |dz
g d d S r r r s r? test_large_integers_decreasingz+TestDigitize.test_large_integers_decreasing s7
R[QUAEN33Q77777rA N)r r r r r r r r r r r r r r r5 r6 xfailr r rA r? r r h s = = =
= = =
/ / /
B B BC C C
C C C
7 7 7
5 5 5 4 4 49 9 98 8 8
[I K K8 8K K8 8 8rA r c e Zd Zd Zd ZdS )
TestUnwrapc t t dddt j z z g ddg t t j t
t t d dz t j k d S )NrR rP r rh )r
r3 r9 r r r r r r s r? r zTestUnwrap.test_simple sl 61a!be)m"4551v>>>tF488c> 2 233be;<<=====rA c t t ddgd ddg t t j t t t
d dz d dk t j g d }t j |d }t t |d | t j g d }t j |d
}t |d
}t |g d t |d
d
}t |g d |j |j k sJ d S )NrR i periodrP r )r K r~ )r r r r r~ i )r r r r r~ r )r discont)
r
r3 r r9 r r r r modr )rh
simple_seqwrap_seq
uneven_seq wrap_uneven
no_discont
sm_disconts r? test_periodzTestUnwrap.test_period s? 61g,s;;;aVDDDtF488d?3 ? ? ?@@3FGGHHHX 4 4 455
6*c**6(3777DDDX 9 9 9::
fZ-- K444
:'B'B'BCCCKSAAA
:'B'B'BCCC ;#4444444rA N)r r r r r r rA r? r r s2 > > > 5 5 5 5 5rA r r O
AllIntegerFloatM)r rR r c r e Zd ZdededdfdZdededdfdZdededdfdZdededdfdZdededdfd Z dS )
TestFilterwindowsr r! returnNc t j || d }t | }|dk r
t j }n$t j |j t j }|j |k sJ t
|t | |dk r$t |t j g d S |dk r$t |t j d d S t t j |d dd d S ) Nr! r r rR r r @rN ) r9 r r$ r r r r r! r
rf r r rh r r! scalarr ref_dtypes r? test_hanningz TestFilterwindows.test_hanning s !5)))"- FOO C<<
IIv|RZ@@Iw)#### Qq """ A:: q"(2,, / / / / /
q[[ q"'!** - - - - - qq 1 1 15! < < < < # ># >$ > > > >*>3 >3 >4 > > > >*>3 >3 >4 > > > >*; ; ; ; ; ; ; ; ;rA r# c e Zd Zd Zd Zd ZdS ) TestTrapzc t j ddd }t t j d|dz z t j dt j z z d }t
|dd d S ) Nr r rk rP )rG rR rz )r9 r: r1 r r r r )rh r r s r? r zTestTrapz.test_simpleJ sf Ic2r " " "&qAv &&RU););; D D DAq!$$$$$rA c 2 t j ddd }t j ddd }t j ddd }t j | |d |d z
z }|dxx dz cc< |dxx dz cc< t j | |d |d z
z }|dxx dz cc< |dxx dz cc< t j | |d |d z
z }|dxx dz cc< |dxx dz cc< |d d d d f |d d d d f z |d d d d f z }||d d d d f z d }||d d d d f z d } ||d d d d f z d }
t ||d d d d f d } t | | t ||d d d d f d } t | | t ||d d d d f d } t | |
t ||d } t | | t ||d } t | | t ||d } t | |
d S )
Nr rR rZ rP rx r_ ra r )r r )r9 r` ones_liker r1 r ) rh r r r wxwywzqqxqyqzr s r? test_ndimzTestTrapz.test_ndimP s K1a K1a K1b ! !
\!__!qt
,
1
2!
\!__!qt
,
1
2!
\!__!qt
,
1
2!
aaatm qaaa .4aaa< @"QQQd]##
(
(a
(
0
0"T111d]##
(
(a
(
0
0"T4]##
(
(a
(
0
0
!qD$ 'a 0 0 0Ar""" !qaaa Q / / /Ar""" !qtAAA Q / / /Ar"""
!qq ! ! !Ar""" !qq ! ! !Ar""" !qq ! ! !Ar"""""rA c t j d }||z }|dk }t j || }d}t t || | t j || }t t || | t j || }t t || | d S )Nr[ rP r
g *@)r9 r: r r r r1 )rh r r r ymr xms r? r^ zTestTrapz.test_maskedu s
IaLL
EAv
U[[[
&
& E"aLL!,,,
U[[[
&
&E"bMM1---
U[[[
&
&E!RLL!,,,,,rA N)r r r r rB r^ r rA r? r6 r6 H sB % % % ## ## ##J- - - - -rA r6 c e Zd Zd Zd ZdS )TestSincc t t d dk t t j ddd }t |t | d d S )Nr rR ra rh rz )r r0 r9 r` r r! rh r s r? r zTestSinc.test_simple sS Q1 RC(( ) )!!VAYY22222rA c ddg}t t j | }t t | }t t | }t || t || d S )Nr ri )r0 r9 r r r r
)rh r r r r s r? test_array_likezTestSinc.test_array_like sh
H
"(1++
$q'']]
%((^^2r"""2r"""""rA N)r r r r rK r rA r? rG rG s2 3 3 3 # # # # #rA rG c e Zd Zd ZdS )
TestUniquec , t j g d }t t j t | g dk t t t j g d t j dg k g d}t t j t | g dk t j g d }t t j t | g dk d S ) N) rN rZ rP rR rR rP rZ rN r )r rR rP rZ rN )rR rR rR rR rR rR )widgethamr r r rP )r r rP rO ) @ @r ? $@r rQ )r rR rQ r )r9 r r r r4 r s r? r zTestUnique.test_simple s H000 1 1vayyOOO344555rx0011RXqc]]BCCC 9 9 9vayy$C$C$CCDDEEE H::: ; ;vayy$A$A$AABBCCCCCrA Nr r r r r rA r? rM rM s( D D D D DrA rM c e Zd Zd Zd ZdS )TestCheckFinitec g d}ddt j g}ddt j g}t j | t t t j j | t t t j j | d S )Nr rR rP )r9 infrZ libasarray_chkfiniter
re r s r? r zTestCheckFinite.test_simple so II
26N
26N
###j"&":A>>>j"&":A>>>>>rA c g d}t j |dt j }t |j t j k d S )Nr r )r r )r9 rX rY r r r ru s r? test_dtype_orderz TestCheckFinite.test_dtype_order sF II F $ $Qc $ D D2:%&&&&&rA N)r r r r r[ r rA r? rU rU s2 ? ? ?' ' ' ' 'rA rU c e Zd Z ej g dg dg dg Z ej g dg dg dg Z ej g dg dg d g Z ej g d
g d g d g d
g dg dg Zd Z d Z
d Z d Z d Z
d Zd Zd Zej dej ej ej ej g d ZdS ) TestCorrCoef)g^?g.{<?g $,?)g4?gp)?gJG?)gY?g?gވ0v?)gة ?gB FU?g G)t?)g3TB?gL\?gA?)g>
?gqFE`?g5 ?)r ޛ?r@)r^ r p]4?)r_ r` r )r r^ r_ ?'L?(8?4V}?)r^ r r` w{,H5?jS,
?)r_ r` r )h?5#A?)ra rd rg r toB C,)rb re rh rj r @02v?)rc rf ri rk rl r c ` t t j g dg d ddgddgg d S )Nr )rR r rR r )r r9 r r s r? test_non_arrayzTestCorrCoef.test_non_array sD BK 999 = = #Yr
3 5 5 5 5 5rA c t | j }t || j t t j t j | dk t | j | j }t || j t t j t j | dk d S r5 )
r r r rw r r9 r ra r rx )rh tgt1tgt2s r? r zTestCorrCoef.test_simple s D$),,,rvd||s*++,,,''D$),,,rvd||s*++,,,,,rA c T t 5 }t j d t t t
| j d | t t t | j d | j t t | j | j
d | j t t | j d | j t t | j | j
d | j d d d d S # 1 swxY w Y d S )Nrr ra ddofrZ ) r ru simplefilterr DeprecationWarningr r filterr rw r rx rh sups r? test_ddofzTestCorrCoef.test_ddof sL
MC !( + + + +XtvB G G G G JJ) * * * b 9 9 949 E E E b A A A49 M M M a 8 8 8$) D D D a @ @ @$) L L L M M M M M M M M M M M M M M M M M Ms DD D!$D!c t 5 }t j d t t t
| j | j dd t t t
| j d | t t t | j d | j
d d d d S # 1 swxY w Y d S )Nrr rR r )bias) r ru rv r rw r r r rx r rw ry s r? test_biaszTestCorrCoef.test_bias s
EC !( + + + +Xtvtvq! L L L +XtvA F F F F JJ) * * * a 8 8 8$) D D D
E E E E E E E E E E E E E E E E E Es B"B>>CCc t j g dg dg }t | }t j ddgddgg }t || t t j t j | dk d S Nr rV y @r_ r r rV )r9 r r r r r ra )rh r r> r s r? test_complexzTestCorrCoef.test_complex s Hiii . / /qkkhT
S"I.//S!!!rvc{{c)**+++++rA c t j g dg }t j g dg }t t j || t j ddgddgg d S r )r9 r r r rh r r s r? test_xyzTestCorrCoef.test_xy se Hiii[ ! ! Hlll^ $ $ Aq))28b$Z#r4K+L+LMMMMMrA c t j d 5 t j dt t t t
j g t j t t t
j g dd t
j g dd t t t
j g dd t
j t j t j gt j t j gg d d d d S # 1 swxY w Y d S NTrp rr r rP )
ru rv rv RuntimeWarningr
r r9 r rZ r{ r s r?
test_emptyzTestCorrCoef.test_empty sm
$D
1
1
1 O O !(N ; ; ; x 5 5rv > > > x (<(> test_typec v | j | }t || }||j k sJ d S Nr! )r r r r )rh r cast_Ar> s r? test_corrcoef_dtypez TestCorrCoef.test_corrcoef_dtype s@ y))vY///CI%%%%%%rA N)r r r r9 r r r rw rx ro r r{ r~ r r r r r5 r6 r7 halfsingledouble
longdoubler r rA r? r] r] s - - - - - - , , , . / /A , , , - - - , , , . / /A 28 % % % $ $ $ & & & ( ) )D 28 I I I I I I K K K M M M J J J J J J L M MD5 5 5- - -
M
M
ME E E, , ,N N N
O O O* * * [ [27BIry"-*XYY& & ZY& & &rA r] c e Zd Z ej ddgddgddgg j Z ej ddgddgg Z ej g dd Z ej g d Z ej d gdgdgdgdgd
gg j Z
ej d d gd d gg Z ej d
ej
Z ej g d Z ej ddgddgg Z ej d
Z ej g d Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Z e j d ej ej! ej" ej# g d Z$d S ) TestCovr rP rR r rn )r r r )ndmin)rR rN rR r r 皙?gٿrZ r! )r r r gUUUUUU?gUUUUUU)g&S:?go_?g#~j?gZӼ?gQ|?c T t t | j | j d S r8 )r r x1rw r s r? rn zTestCov.test_basic s" DG di00000rA c
t j g dg dg }t j ddgddgg }t t | | t t |t j d | d S )Nr r r r rV rZ aweights)r9 r r r rf )rh r r> s r? r zTestCov.test_complex s{ Hiii . / /hT
S"I.//A$$$A
333S99999rA c t j g dg }t j g dg }t t || t j ddgddgg d S r )r9 r r r r s r? r zTestCov.test_xy% sa Hiii[ ! ! Hlll^ $ $Aq 28b$Z#r,C#D#DEEEEErA c t j d 5 t j dt t t t
j g t j t t t
j g dd t
j g dd t t t
j g dd t
j t j t j gt j t j gg d d d d S # 1 swxY w Y d S r )
ru rv rv r r
r r9 r rZ r{ r s r? r zTestCov.test_empty* sm
$D
1
1
1 O O !(N ; ; ; s28B<< 0 0"& 9 9 9 s28B<<#7#71#=#= > > !x|| 3 3Aq 9 9
;
;
; s28B<<#7#71#=#= > > !x"&"&)9BFBF;K(L M M
O
O
O O O O O O O O O O O O O O O O O O Or c b t j d 5 t j dt t t | j d t j t j t j gt j t j gg d d d d S # 1 swxY w Y d S )NTrp rr r[ rt )
ru rv rv r r
r r r9 r rW r s r? test_wrong_ddofzTestCov.test_wrong_ddof3 s
$D
1
1
1 > > !(N ; ; ; s47 3 3 3 !x"&26'):+-6'26):)< = =
>
>
> > > > > > > > > > > > > > > > > > >s BB$$B(+B(c t t | j t | j d t j g d }t t | j | t | j |d d S )NF)rowvar)g+?g4@?g&S?go_?gJ4?)r r x3r9 r rh r s r? test_1D_rowvarzTestCov.test_1D_rowvar: sp DG c$'%&@&@&@AAA H=== > >DGQTWa)F)F)FGGGGGrA c t t | j d t j | j d d S )NrR rt )r r r r9 varr s r? test_1D_variancezTestCov.test_1D_variance? s8 DG!,,,bfTW1.E.E.EFFFFFrA c t t | j | j t | j t t | j | j | j t t | j | j | j | j dz }t t t | j | t j dt j
}t t t | j | t j dt j
}t t t | j | dt j dt j
z }t t t | j | d S )N)fweightsri rP rZ r! rP ra rZ )r r r] frequencies
x2_repeatsr rx unit_frequenciesrw r
rd r9 rf rc RuntimeErrorre )rh nonintro s r?
test_fweightszTestCov.test_fweightsB sI DGd.>???DO,, . . .DGd.>??? # # #DGd.CDDD # # #!C'idg???? GF"' * * *lC1==== GARW % % %lC1==== "'*** *j#tw;;;;;;rA c t t | j | j | j t t | j d| j z t | j | j t t | j | j | j t j d }t t t | j | t j d }t t t | j | dt j d z }t t t | j | d S )Nr r r rP rn rZ ) r r r r ry unit_weightsrw r9 rf r
r re rI s r?
test_aweightszTestCov.test_aweightsR s DGdl;;;TYGGGDGcDL.@AAADGdl;;; = = =DGd.?@@@$)LLL GFOOlC1==== GAJJlC1==== 271:: j#tw;;;;;;rA c t t | j | j | j t | j t t | j | j | j | j t t | j | j | j | j t t | j | j | j
| j t t | j | j d| j
z t | j | j
t t | j | j | j | j d S )N)r r r r ) r r r] r r r r rx r rw r ry r s r? test_unit_fweights_and_aweightsz'TestCov.test_unit_fweights_and_aweights^ sz DGd.>%)%68 8 8DO,, . . . DGd.>%)%68 8 8 # # # DGd.C%)%68 8 8 # # # DGd.C%)\3 3 3 # # # DGd.C%(4<%79 9 9DGdl;;; = = = DGd.C%)%68 8 8 # # # # #rA r c v | j | }t || }||j k sJ d S r )r r r r )rh r cast_x1r> s r? test_cov_dtypezTestCov.test_cov_dtyper s@ '..++'+++CI%%%%%%rA N)%r r r r9 r r r rw r] r r rx rf rc r r ry r r rn r r r r r r r r r r5 r6 r7 r r r r r r rA r? r r s* Aq6Aq6Aq6* + + -B 28b#Yb * + +D /// + + +B"(999%%KC53%#useDEEGJ 28c4[4+. / /D rwq000bh''G 28gx(8W*=> ? ?D271::L ::: ; ;B1 1 1: : : F F F
O O O> > >H H H
G G G< < <
<
<
<# # #( [ [27BIry"-*XYY& & ZY& & &rA r c e Zd Zd Zd Zd ZdS )Test_I0c t t d t j d t j g d }t j g d }t t | | t t | | t j ddgddgd d
gd d gd
dgg }t t | t j ddgddgddgddgddgg t j dg }t |j d t
t j dg t j dg d S ) Nri gǒ#?)gC7?gmM?g3a,?g$WV?ۏ?r )gֿN^?gE ?g閡1?g%m? ?gDn@g>@v?guͥ?gVX˳?gua&?gB- sJ?g-^ ?g_V?g
X.?gVaX?gj ?gp?g~-@?g D/b?gݦ?g,Y1ؔ?g[!?gQL?gWsU+?g@?gfxw?r r0 r )r r% r9 r r rE r
)rh r r r i0_0s r? r zTest_I0.test_simple{ si sGG H' ( ( * * *
HIII J J8 [ [ [\\BqEE8,,,BrFFH--- Hx,!:.!:.!:.!:. 0
1
1
qEE Hz:.!:.!:.!:.!:. 0
1
1 2 2 2 ubT{{TZ&&&25";;"77777rA c & t j d G fdd }t t j | | sJ t j }t j | }t || d S )NrN c $ e Zd Z j Zd ZdS )*Test_I0.test_non_array..array_likec | S r8 r )rh r s r? __array_wrap__z9Test_I0.test_non_array..array_like.__array_wrap__ s rA N)r r r __array_interface__r r s r?
array_liker s- "#"7
rA r )r9 r: r ra r% r
)rh r r r> ri s @r? ro zTest_I0.test_non_array s IaLL "& ..
;;;;;eAhheJJLL!!3$$$$$rA c t j ddg }t j t d 5 t | }d d d d S # 1 swxY w Y d S )Nr y ? @z#i0 not supported for complex valuesr )r9 r r5 r rd r% )rh ri r> s r? r zTest_I0.test_complex s Ha[ ! !
]9,Q
R
R
R Q%%C s AAAN)r r r r ro r r rA r? r r y sA 8 8 88% % % rA r c e Zd Zd Zd ZdS )
TestKaiserc t t j t dd t t dd t j g t t dd t j ddg t t dd t j g d t t dd t j g d d S )
NrR r r rP oF?r[ )r vT=?r r r gg?)?YJ?r r r )r r9 isfiniter( r r r s r? r zTestKaiser.test_simple s F1cNN++,,,F1cNN HRLL * * *F1cNN Hj*%= > > @ @ @F1cNN H &> &> &> ? ? @ @ @ F1g . . H &> &> &> ? ? @ @ @ @ @rA c & t dd d S )NrZ rN )r( r s r?
test_int_betazTestKaiser.test_int_beta s q! rA N)r r r r r r rA r? r r s5 @ @ @ rA r c e Zd Zd ZdS ) TestMsortc
t j g dg dg dg }t j t d 5 t t
| t j g dg dg dg d d d d S # 1 swxY w Y d S ) N)\?bC Q?הs?)w? kl?*dta?)ӯ?6H?ܘC?zmsort is deprecatedr )r r r )r r r )r r r )r9 r r5 warnsrw r r* r s r? r zTestMsort.test_simple s H:::::::::<
=
= \,4I
J
J
J B B a>>>>>>>>>@ A A
B
B
B B B B B B B B B B B B B B B B B B Bs 9A??BBNrS r rA r? r r s( B B B B BrA r c P e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
Z d Z
d S )
TestMeshgridc
t g dg d \ }}t |t j g dg dg dg dg t |t j g dg dg dg dg d S )Nr rN r[ ry rz r )r[ r[ r[ )ry ry ry )rz rz rz r) r
r9 r rh XYs r? r zTestMeshgrid.test_simple s )))\\\22A1bh (1 (1 (1 (3 4 4 5 5 5 1bh (1 (1 (1 (3 4 4 5 5 5 5 5rA c t t g d \ }t |t j g d d S )Nr r )rh r s r? test_single_inputz TestMeshgrid.test_single_input s9 |||$$1bh||| 4 455555rA c l g }t g t | t g t |ddi d S )Nr F)r
r) r s r?
test_no_inputzTestMeshgrid.test_no_input s> 2x///2x :E : :;;;;;rA c g d}g d}t ||d \ }}t |t j g dg dg dg t |t j g dg dg dg dd g}t t || d
j d k t t ||d d
j d k t t ||| d
j d
k t t |||d d
j dk t
t t ||d d S )Nr r ijindexingr )rP rP rP rP )rZ rZ rZ rZ rx r; r rR rZ rN )rN rZ rP )rZ rN rP notvalid)r) r
r9 r r rE r
re )rh r r r r r s r?
test_indexingzTestMeshgrid.test_indexing sn II LL!Q...A1bh (4 (4 (6 7 7 8 8 8 1bh (4 (4 (6 7 7 8 8 8
FAq!'61222A---a06&@AAAAq!!!$*i7888Aq400039YFGGGj(Aq:FFFFFFrA c t g dg dd \ }}t |t j g dg t |t j dgdgdgdgg d S ) Nr r TsparserN r[ ry rz r r s r? test_sparsezTestMeshgrid.test_sparse ss )))\\\$???A1bh { 3 34441bhaS1#s'; < <=====rA c J t t t g dg dd d S )Nr r r )r )r
rd r) r s r? test_invalid_argumentsz#TestMeshgrid.test_invalid_arguments s8 i ii t = = = = = =rA c ~ t j ddt j }t j ddt j }t j || \ }}t |j |j k t |j |j k t j ||d \ }}t |j |j k t |j |j k t j ||d \ }}t |j |j k t |j |j k d S )Nr r r! r T)r r )r9 r: r r r) r r )rh r r r r s r? r z TestMeshgrid.test_return_type
s
Ia2: . . . Ib"BJ / / /{1Q 117"###17"### {1QT***117"###17"### {1Qt,,,117"###17"#####rA c t j ddg }t j ddg }t j ||dd \ }}d|dd d f< t |dd d f d t |d d d f | d S )
Nr r gffffff
@g@FT)r r r rR )r9 r r) r )rh r r r r s r? test_writebackzTestMeshgrid.test_writeback
s Hc3Z Hc3Z {1aD9991!QQQ$Qq!!!tWa Qq!!!tWa rA c 4 t j d t dd D \ }}}}}d}t |j | t |j | t |j | t |j | t |j | d S )Nc 3 " K | ]
}d g|z V dS )r Nr r, r s r? r z-TestMeshgrid.test_nd_shape..'
s( %C%C!qcAg%C%C%C%C%C%CrA rR ry )rP rR rZ rN r[ )r9 r) rg r rE )rh ri r r r\ ri expected_shapes r?
test_nd_shapezTestMeshgrid.test_nd_shape&
s %C%CuQ{{%C%C%CD
1aA(QWn---QWn---QWn---QWn---QWn-----rA c t j dgddgg d \ }}}t |g dgg dgg t |g dgg dgg t |g dgg dgg d S )Nr rR rP rY r r rO r9 r) r r s r? test_nd_valueszTestMeshgrid.test_nd_values/
s +qcAq6999551aQ)))yyyk2333Q)))yyyk2333Q)))yyyk233333rA c t j dgddgg dd \ }}}t |g dg dgg t |g dg d gg t |g dg dgg d S )
Nr rR rP rY r r r r rO r r s r? test_nd_indexingz TestMeshgrid.test_nd_indexing5
s +qcAq6999tDDD1aQ)))YYY/0111Q)))YYY/0111Q)))YYY/011111rA N)r r r r r r r r r r r r r r r rA r? r r s 5 5 56 6 6< < <
G G G(> > >
= = =$ $ $0! ! !. . .4 4 4 2 2 2 2 2rA r c D e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
S )
TestPiecewisec t ddgddgdg }t |ddg t ddgddggdg }t |ddg t ddgt j ddg dg }t |ddg t ddgt j ddg dg }t |ddg t ddgt j ddg gdg }t |ddg t ddgddggd g }t |ddg t t
dt ddgddggg t t
dt ddgddggg d d S ) Nr TFrR c dS r r r s r? r z+TestPiecewise.test_simple..S
s rA ra z 1 or 2 functions are expectedr )r+ r
r9 r r re r s r? r zTestPiecewise.test_simple>
s q!ftUmaS 1 11q!f%%%
q!fe}os 3 31q!f%%%
q!fbhe} 5 5s ; ;1q!f%%%
q!fbh1v . . 4 41q!f%%%
q!frxA / / 01# 6 61q!f%%% q!ft}o } = =1q"g&&&J(G 1v
4 4 4J(G 1v
; ; ; ; ;rA c ` t ddgddgddggddg }t |ddg d S )NrR rP TFrZ rN r+ r
r s r? test_two_conditionsz!TestPiecewise.test_two_conditions[
sB q!fe}udm .subclassNr r rA r? r r
r rA r r r rN )rn r r )rn rn r r r )r9 r r: r r+ r r )rh r r r s r? r z TestPiecewise.test_subclasses
s rz IbMM x ( ( a!B$1 }}} 5 5T!WWh'''Q.../////rA N) r r r r r
r r r r r r r r rA r? r r <
s ; ; ;:& & & & & &
!
!
!
9
9
9 8 8 80 0 0 0 0rA r c e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
Z e
j e d d
Ze
j dddggdg d ZdS ) TestBincountc t j t j d }t |t j d d S )NrN )r9 bincountr: r
rf r s r? r zTestBincount.test_simple
s7 K ! % %1bgajj)))))rA c t j t j g d }t |t j g d d S )N)rR r[ rP rN rR )r rP rR r rR rR )r9 r r r
r s r? test_simple2zTestBincount.test_simple2
sF K 11 2 21bh'9'9'9 : :;;;;;rA c t j d }t j g d }t j || }t || d S )NrN )皙?333333?ri r )r9 r: r r r
rh r r r s r? test_simple_weightz TestBincount.test_simple_weight
sL IaLL H))) * * K1 1a rA c t j g d }t j g d }t j || }t |t j g d d S )NrR rP rN r[ rP r# r$ ri r r# )r r# ri r ri r r9 r r r
r% s r? test_simple_weight2z TestBincount.test_simple_weight2
sc H___ % % H... / / K1 1bh'A'A'A B BCCCCCrA c t j g d }t j |d }t |t j g d g }t j |d }t |t j g d S )N)r rR r rR rR rZ minlength)rP rZ r r r* r s r? test_with_minlengthz TestBincount.test_with_minlength
s~ H___ % % KQ ' ' '1bhyyy 1 1222 KQ ' ' '1bhrll+++++rA c t j g d }t j |d }t |t j g d t j |d }t |t j g d d S )N)r rR rR rP rP rZ rZ rP r- )rR rP rP rP r r* r s r? )test_with_minlength_smaller_than_maxvaluez6TestBincount.test_with_minlength_smaller_than_maxvalue
s H*** + + KQ ' ' '1bh||| 4 4555 KQ ' ' '1bh||| 4 455555rA c t j g d }t j g d }t j ||d }t |t j g d d S )Nr( r) rx )r r# ri r ri r r r r* r% s r? test_with_minlength_and_weightsz,TestBincount.test_with_minlength_and_weights
se H___ % % H... / / K1a 1bh'G'G'G H HIIIIIrA c t j g t }t j | }t || d S r )r9 r r r r
r s r? r zTestBincount.test_empty
s9 HRs # # # KNN1a rA c t j g t }t j |d }t |t j dt d S )Nr! r[ r- )r9 r r r r
rC r s r? test_empty_with_minlengthz&TestBincount.test_empty_with_minlength
sO HRs # # # KQ ' ' '1bhq 4 4 455555rA c . t j g t t t dfd t t
dfd t j d t t dfd t t
dfd d S ) Nr! z"'str' object cannot be interpretedc 0 t j d S Nfoobarr- r9 r r s r? r z.
BKX$F$F$F rA zmust not be negativec 0 t j d S Nra r- r; r s r? r z.
BKR$@$@$@ rA r[ c 0 t j d S r9 r; r s r? r z.
r< rA c 0 t j d S r> r; r s r? r z.
r? rA )r9 r r r rd re r: r s @r? test_with_incorrect_minlengthz*TestBincount.test_with_incorrect_minlength
s HRs # # #I @ F F F F H H H J 2 @ @ @ @ B B B
IaLLI @ F F F F H H H J 2 @ @ @ @ B B B B BrA r r c \ t j t j t j }t j t j t j }t
d D ]}t j g d t t j t j t j | t t j t j t j | t
d D ]}t j g dg d t t j t j t j | t t j t j t j | d S )Nr r r ) r r r9 r intpr rg r r )rh
intp_refcountdouble_refcountjs r? test_dtype_reference_leaksz'TestBincount.test_dtype_reference_leaks
s6 (9(9::
/"(29*=*=>>r # #A K " " " "S_RXbg%6%677 GGGS_RXbi%8%899?KKKr . .A K 999 - - - -S_RXbg%6%677 GGGS_RXbi%8%899?KKKKKrA valsrP c 0 t j | }t t 5 t j | d d d n # 1 swxY w Y t t 5 t j | d d d d S # 1 swxY w Y d S r8 )r9 asarrayr
re r )rh rI vals_arrs r? test_error_not_1dz TestBincount.test_error_not_1d
s :d##
:
&
& " " K ! ! ! " " " " " " " " " " " " " " "
:
&
& K s# A
AA)B BBN)r r r r r! r& r+ r/ r1 r3 r r6 rB r5 r6 r r rH r7 rM r rA r? r r
s$ * * *< < <! ! ! D D D , , ,6 6 6J J J ! ! !
6 6 6
B B B" [L(1IJJ
L
L KJ
L [ V1vh]33 43 rA r c e Zd Zd Zd Zd Zd Zd Z ej d d d d gg d
d Z
d
Z d Z d Z
d Zd Zd Zd Zd Zd ZdS )
TestInterpc H t t t dg g t t t ddgddg t t t dddgddgd t t t dg g d t t t ddgddgd d S )Nr rR rP r
h )r
re r' r s r? test_exceptionszTestInterp.test_exceptions s j&!R444j&!aS1a&999j&!aVaVAFFFFj&!RC@@@@j&!aS1a&EEEEEErA c t j ddd }t j ddd }t j ddd }t t j ||| | d S )Nr rR r[ rp )r9 r` r r' rh r r x0s r? rn zTestInterp.test_basic s_ K1a K1a
[Ar
"
"BIb!Q / /44444rA c t dd D ]+}t j |t j }t j |t j }t j dd|dz
|gt j }|d d d }t
||| }t
||| }t j g dt }|d d d } t || t || t
|||d }t
|||d }t j g dt }|d d d } t || t || t
|||d
}t
|||d
}t j g d t }|d d d } t || t || t
|||dd }t
|||dd }t j g d
t }|d d d } t || t || -d S )NrR r r! ra r r left)r rR rR rR rP right)rR rR rR rP )rX rZ )r rR rR rP ) rg r9 r: r rf r r' r r )
rh sizexpypincptsdecptsincresdecresinctgtdectgts
r? test_right_left_behaviorz#TestInterp.test_right_left_behavior sM !RLL ) )D4ry111BRY///BXr1dQh 5RYGGGFDDbD\FFB++FFB++FXlll%888FDDbD\F ( ( ( ( ( (FB333FFB333FXlll%888FDDbD\F ( ( ( ( ( (FB!444FFB!444FXlll%888FDDbD\F ( ( ( ( ( (FB!<<L s ")A,, rA c " t | d S rb rJ r s r? r zTestInterp.M -1%% rA c " t d| S rb rn r s r? r zTestInterp.N ro rA c H t | t j | d S )NrX )rJ r9 multiplyr s r? r zTestInterp.O s -2;q"#5#566 rA )rG z complex-realz complex-imagz complex-both)paramsidsc |j S )z( scale function used by the below tests )param)rh requests r? scz
TestInterp.scK s } rA c
v t t j dt j dg |ddg |t j t t j ddt j g |ddg |t j t t j dddg |t j dg |t j t t j dddg |dt j g |t j dS )z test that nans are propagated ri rR r r N)r r9 r' rZ rh rx s r? test_non_finite_any_nanz"TestInterp.test_non_finite_any_nanZ s RYsRV!$4bbqb9I6J6JKKRRPRPVZZXXXRYs!RV$4bbqb9I6J6JKKRRPRPVZZXXXRYs!!$4bb"&b9I6J6JKKRRPRPVZZXXXRYs!!$4bbq"&9I6J6JKKRRPRPVZZXXXXXrA c
t t j dt j t j
g |ddg |t j t t j dddg |t j t j
g |t j t t j dddg |t j
t j g |t j t t j dt j t j
g |ddg |d dS )z2 Test that interp between opposite infs gives nan ri r r rR Nr r9 r' rW rZ rz s r? test_non_finite_infz TestInterp.test_non_finite_infa s/ RYsbfWrvg$6!";M8N8NOOQSQSTVTZQ[Q[\\\RYs1A$6RVGbfW;M8N8NOOQSQSTVTZQ[Q[\\\RYs1A$6RVGbfW;M8N8NOOQSQSTVTZQ[Q[\\\ RYsbfWrvg$6";M8N8NOOQSQSTVQWQWXXXXXrA c
t t j dt j dg |t j dg |t j t t j dt j dg |t j
dg |t j t t j dt j dg |dt j g |t j t t j dt j dg |dt j
g |t j t t j ddt j
g |t j dg |t j t t j ddt j
g |t j
dg |t j t t j ddt j
g |dt j g |t j t t j ddt j
g |dt j
g |t j dS )z@ Test that interp where both axes have a bound at inf gives nan ri rR r r Nr} rz s r? test_non_finite_half_inf_xfz&TestInterp.test_non_finite_half_inf_xfj sX RYsbfWA$6RVG";M8N8NOOQSQSTVTZQ[Q[\\\RYsbfWA$6RVG";M8N8NOOQSQSTVTZQ[Q[\\\RYsbfWA$6!bfW;M8N8NOOQSQSTVTZQ[Q[\\\RYsbfWA$6!bfW;M8N8NOOQSQSTVTZQ[Q[\\\RYs1rvg$6RVG";M8N8NOOQSQSTVTZQ[Q[\\\RYs1rvg$6RVG";M8N8NOOQSQSTVTZQ[Q[\\\RYs1rvg$6!bfW;M8N8NOOQSQSTVTZQ[Q[\\\RYs1rvg$6!bfW;M8N8NOOQSQSTVTZQ[Q[\\\\\rA c
Z t t j dt j t j g |ddg |d t t j dt j dg |ddg |d t t j ddt j
g |ddg |d t t j dt j
t j
g |ddg |d dS )z1 Test interp where the x axis has a bound at inf ri r r rR Nr r9 r' rW rz s r? test_non_finite_half_inf_xz%TestInterp.test_non_finite_half_inf_xu s RYsbfWrvg$6Ar7 DDbbffMMMRYsbfWa$6Ar7 DDbbffMMMRYs1rvg$6Ar7 DDbbeeLLLRYsbfWrvg$6Ar7 DDbbeeLLLLLrA c
t t j dddg |dt j g |t j t t j dddg |dt j
g |t j
t t j dddg |t j dg |t j t t j dddg |t j
dg |t j
t t j dddg |t j t j g |t j t t j dddg |t j
t j
g |t j
dS )z1 Test interp where the f axis has a bound at inf ri r rR r Nr rz s r? test_non_finite_half_inf_fz%TestInterp.test_non_finite_half_inf_f| s RYsQFBBa"&/A,B,BCCRR[[QQQRYsQFBBa"&/A,B,BCCRR[[QQQRYsQFBBb/A,B,BCCRR[[QQQRYsQFBBb/A,B,BCCRR[[QQQRYsQFBB"&/A,B,BCCRR[[QQQRYsQFBB"&/A,B,BCCRR[[QQQQQrA c t j ddd }t j ddd dt j ddd z dz z }d}|d|z dz z }t t j ||| | d}d}t t j |||| | d }d}t t j ||||
| g d }g d }dd
t j dg}dd
t j dz t j dg}t t j ||| | g d}g d}g d}g d}t t j |||d | d S )Nr rR r[ rV r$ ra @ @rW r rY rh r @ ?rN y ?iLiViG r8 r r im iBi^ i) @ ?y $@ @ @ @y @ @)y @ ?r y !@ ?y @ ?r y
@
@y @ @y @ @rQ r
)r9 r` r r' rW ) rh r r rU y0rX rZ r\ ri s r? test_complex_interpz TestInterp.test_complex_interp s K1a K1a A Aq!(<(<$. . . .Xa^^ . . .rA rx r rh rz rP rp rN )rg r9 r r r r r r )rh r r, s r?
test_fractionz TestPercentile.test_fraction sD . .U1XX . . . M!Xa[[ ) )Q $$$T!WWh''' M!Xc]] + +QA'''T!WWh''' M!Xb\\ * *QA'''T!WWh''' M!hrll^ , ,Q(1a..!122333T!WWbj)))))rA c t j d }t j |dd d d t j |dd d dd t j d }t j |dd |dd d S )Nr[ Flinearr0 )r9 rf r )rh r\ os r? test_apizTestPercentile.test_api sn GAJJ
aD$...
aD$x888 GDMM
aD!UH55555rA c \ t j g dd }t t t j |d t j g dd }t t t j |d t j g dd }t t t j |d d S N)y ? @y @ ?y?ffffff@Gr! ri Dr )r9 r r
rd r rh arr_cs r? r zTestPercentile.test_complex s 777sCCCi s;;;777sCCCi s;;;777sCCCi s;;;;;rA c t j g dg dg dg dg dg }t t j |dd g d d S )Nr )rN rN rZ rp r r )r9 r r
r r s r? test_2DzTestPercentile.test_2D sg Hiii ii ii ii ii !
"
"
2=BQ777CCCCCrA r r c & t j dt j dddg| }t j |dd }t j |t j t j |j |j d S )Nr A@ D@ I@r! r r )r9 rK NANr testingr r )rh r r r> s r? test_linear_nan_1Dz!TestPercentile.test_linear_nan_1D s j$dD9GGGm
RV,,,
3955555rA c * g | ]}|t j fS r )r9 r )r, int_types r? r- zTestPercentile. s/ " , rA r r input_dtypeexpected_dtyper r ) ) inverted_cdfr )averaged_inverted_cdf ;@)closest_observationr )interpolated_inverted_cdfr )hazenr )weibull )r r )median_unbiasedr )normal_unbiasedg ;@c " t j | }t j dk r t j |t j }t j g d| }t j |d| }t j || | d |dv r|dk rCt j
t j | j t j d S t j
t j | j t j | d S t j
t j | j t j | d S ) Nlegacy)r g 4@r r r r! r r r )r r r ) r9 r _get_promotion_state
promote_typesr rK r r r r r )rh r r r r r r s r? test_linear_interpolationz(TestPercentile.test_linear_interpolation
sc " .11
" $ $ 0 0 -nbj I INj777{KKK sD888
&& N''112 7 7 7 < < <c!!
''
6(:(:(@"*MMMMM
''
6(:(:(@(*(=(=? ? ? ? ?
J # #BJv$6$6$<$&H^$<$<
>
>
>
>
>rA c t t j t j d| dd d t t j t j d| dd d d S ) Nr r! rp lowerr rN higherr[ r r9 r r: rh r s r? test_lower_higherz TestPercentile.test_lower_higher1 s R]29Ru#=#=#=r*13 3 345 7 7 7R]29Ru#=#=#=r*24 4 456 8 8 8 8 8rA c t t j t j d| dd d t t j t j d| dz d d d
t t j t j d | dd d t t j t j d | d d d
d S )
Nr r! 3 midpointr r& r; rR rp r[ r @r r s r?
test_midpointz TestPercentile.test_midpoint8 s R]29Ru#=#=#=r*46 6 67: < < <R]29Qe#<#<# IaLL3 R]1lll33^^^DDDDDrA c | t j d dd }t t j |d g d g dg dg dg}t t j |dd
| g d g d g d
g}t t j |dd
t j | j t j d dddd }t t j |d j d t t j |d j d t t j |dd
j d t t j |dd
j d t t j |dd
j d t t j |dd
j d t t j |dd
j d t t j |dd j d t t j |dd j d t t j |dd d j d t t j |ddd j d t t j |ddd j d t t j |ddd j d t t j |ddd j d d S ) NrQ rZ rN )rJ rp rh )g @r rI )rP rZ rN r[ r )rx r; r r r r ) ?rK rZ ) @r rz )r r r rR rQ r[ ry rJ rp r )rJ rp r rZ )rP rN r[ ry )rP rZ r[ ry rP )rP rZ rN ry )rZ rZ r[ ry r r r r )r9 r: r{ r r r r rE )rh r r0r s r? r zTestPercentile.test_axisW sa IbMM ! !!Q ' 'R]1m446G6G6GHHHllLLL...
9R]1m!<<R]1m!<<>> BM!R 0 011222
X)))
*
*
]1bq
9
9
9R RXrx(((
Xnnn
%
%
]1bq
9
9
9B###RXrx(((hr))) M!RS 9 9 9QS! hq((( M!RaS A A AQS" hq((( M!RaS A A AQS" rA c
t t t j ddgdd t t t j dgd t t t j dgd t t t j dgt t d dgz t t t j dgt t d d gz d S )
NrR rP 8 r: r e ra rp r )r
re r9 r r rg r s r? test_exceptionz TestPercentile.test_exception s j"-!Q % ' ' ' 'j"-!c:::j"-!b999j"-!d599oo6MNNNj"-!d599oo6NOOOOOrA c P t t j g dd d d S )Nr r rR )r r9 r r s r? test_percentile_listz#TestPercentile.test_percentile_list s( R]999a00!44444rA c t j g d }t j d }d}t j ||| t t j || | t j g dg dg }t j d }t j ||d| t t j ||d | t j d }t j ||d
| t t j ||d
| t j d d d
}t j g dg dg }t j d }t t j |dd| | t || t j g dg dg }t j d }t t j |dd
| | t || t j g dg dg }t j d|j }t j |ddd| }t || t || t j g dg dg }t j d|j }t j |ddd
| }t || t || d S ) Nr r r r r r r r rZ rP rR rQ rZ rN )r r r r r )rP rN r )r r r r r r rR rP rZ r r! r r )r rN rx )rR r[ r; ) r9 r rC r r r: r{ r r )rh r r r, r r r r s r? test_percentile_outz"TestPercentile.test_percentile_out s HYYY HTNN
a""""R]1a((!,,, Hiii ii!
"
"
HV
a****R]1aa000!444 HV
a****R]1aa000!444 IbMM ! !!Q ' '
X)))+;+;+;<
=
=hv R]1hQC@@@"EEES"
X+++->->->?
@
@hv R]1hQC@@@"EEES" X~~~|||4
5
5hvQW--- M!XgA3 G G GQS"
X{{{KKK0
1
1hvQW--- M!XgA3 G G GQS" rA c t j d dddd }t t j |dd j d t t j |dd j d t t j |dd j d
t t j |dd j d t t j |dd
j d t t j |dd j d
t t j |dd j d t t j |dd j d t t j |ddd j d
t t j |ddd j d
t t j t j |ddgd j d t t j t j |ddgd j d t t j t j |ddgd j d t t j t j |ddgd j d d S )Nr r rR rP rp r r )rR rP rR )r rP rR )r rR rR rZ )r rR rP ra rX r^ r| r r r )rP rR rP rR )rP r rP rR )rP r rR rR )rP r rR rP )r9 r: r{ r
r rE r rh r\ s r? test_percentile_empty_dimz(TestPercentile.test_percentile_empty_dim s If % %b!Q 2 22=BQ777=yIII2=BQ777=zJJJ2=BQ777=zJJJ2=BQ777=zJJJ2=BR888>
KKK2=BR888>
KKK2=BR888>
KKK2=BR888> JJJ2=BQ0: < < < > >R]1bv666t<]1Q!!!AAAX;#6#6#8#8"== ? ? ?R]1bw777=]1QqqqAX;#6#6#8#8"== ? ? ?R]1bv666t<]1Qqqq!AAAX;#6#6#8#8"== ? ? ? ? ?rA c @ t j d }t t j t j |dd t t j t j |dd t t j t j |dd t t j t j |dd t t
t j |dd t t
t j |d d t t
t j |d
d d S ) Nr r rJ )r r> r r rN r rN rW )ra ra )rZ ra )r9 rf r
r r re r s r? test_extended_axis_invalidz)TestPercentile.test_extended_axis_invalid4
s GM " "blBM12DDDDblBM17bIIIIblBM11CCCCblBM16RHHHHj"-2FFFFj"-RHHHHj"-BGGGGGGrA c t j d }t t j |dd d j d t t j |ddd j d t t j |ddd j d t t j |dd
d j d t t j |dd d
j d t t j |ddd j d t t j |ddgdd j d t t j |ddgdd j d d S )Nr rz Tr r rb rR rR rz r r rR r[ rz rR r0 rZ rR rz r r )r r rR rR rz rR rR )rP rR rR rz rR )rP rR r[ rz rR )r9 rf r r rE r s r?
test_keepdimsz TestPercentile.test_keepdims?
s GM " "R]1adTBBBH! # # #R]1aftDDDJ" $ $ $R]1aftDDDJ! # # #R]1adTBBBH" $ $ $R]1a EEEK! # # #R]1ai$GGGM! # # # R]1q!f9,02 2 227 J J JR]1q!f6,02 2 227 J J J J JrA r> rz rR r NrR r0 rb )r^ ra argnames argvaluesc t j d | dj z }nCt |j t fdt j D }t j | |z }t j | }t j ||d| }||u sJ t |j | d S )Nr r0 c 3 > K | ]}|v rd n j | V dS rR NrE r, r axis_normr\ s r? r z3TestPercentile.test_keepdims_out..d
N L L89Q)^^ L L L L L LrA Tr r r )
r9 rf r r6 r rg rE r r r )rh r> r shape_outr r r3 r\ s @@r? test_keepdims_outz TestPercentile.test_keepdims_outS
s
GM " " < qv
II,T16::I L L L L L=B16]] L L L L LIHQKK)+ hy!! q!$3GGG}}}}V\9-----rA c p t j d }t j d }t t j |dd| | t t j |ddd| | t j d }t t j |dd| | t t j |ddd| | t j d }t t j |d
| | t t j |d
d| | d S ) NrN r r r r r r rR r rP )r9 rC rf r r rh r r\ s r? test_outzTestPercentile.test_outm
s HTNN GFOOR]1a222A666R]1a9!DDDaHHH HTNNR]1a222A666R]1a9!DDDaHHH HRLLR]1aQ///333R]1a qAAA1EEEEErA c
t j d 5 t j ddt t j d }t j d }t j |d< t t j |dd| | t t j |ddd
| | t j d }t t j |d
d
| | t t j |d
d
d
| | t j d }t t j |d
| | t t j |d
d
| | d d d d S # 1 swxY w Y d S )NTrp rr rs r9 r r r r r r r rR r )
ru rv rw r r9 rC rf rZ r r r: s r? test_out_nanzTestPercentile.test_out_nanz
s
$D
1
1
1 A A #Hb. A A AAAfAdG q!QA666 : : :
aAiQ???
D
D
DA q!QA666 : : :
aAiQ???
D
D
D A q!333Q 7 7 7
a9!<< s r?
test_nan_qzTestPercentile.test_nan_q
s
]:-E
F
F
F 2 2 M..."& 1 1 1 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2
]:-E
F
F
F 4 4 M...26( 3 3 3 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 KT2 & &v!
]:-E
F
F
F - - M...! , , , - - - - - - - - - - - - - - - - - -s5 "A
AA0#B B#&B#*DDDzm8[D]zM8[s]posr r
r c t j dd| }d||< t j |d }|j |k sJ t j | sJ t j |ddg }|j |k sJ t j | sJ t j dd| dd }d||d
f< t j |dd }t t j | g d d S
Nr r{ r! NaTr r3 H ra rZ rR r r< )r9 r: r r isnatr r{ r
rh r rC ri r> s r? test_nat_basicz TestPercentile.test_nat_basic
s
Ia5 ) ) )#mAr""yE!!!!x}} }mABx((yE!!!!x}} """"" IaU + + + 3 3B : :#q& mAr***28C==*>*>*>?????rA N)-r r r rn r r r r r5 r6 r7 r9 typecodesr r r r r r H_F_TYPE_CODESr
TYPE_CODESr r r r r r r r r r r r r r r r# r) r7 r; r= r@ rB rK r rA r? r r s D D D* * *&6 6 6< < <D D D [ Wbl7&;<<6 6 =<6 &(l<&@ J
3 J
3 J
3 M2=9 $BHSMM2:6 88N [ m-= >OO [ h
3
!
!> >
!
! PO>2 l+bl7.CCcIJ [ Wj118 8 218 [ Wj11: : 21: [ Wj119 9 219 3 3 3E E E"I "I "IH. . . `P P P5 5 5' ' ' R* * *80 0 0E E E) ) ) ? ? ?@ H H HJ J J( [ S1q!f+.. [
.
. /.
. F F FA A A$;9 ;9 ;9z - - - [ Ww&899 [ UKKK00@ @ 10 :9@ @ @rA r )
r r r r r r r r r r r r r c e Zd Zd Zd Zd Zd Zd Zd Zd Z e
j de
j d d
Ze
j d e d Z ej ee
j ej d
d ej dddd d Zd Ze
j d e e
j dg d d Ze
j d e e
j dg d d Z d S ) TestQuantilec ||k |z
S r8 r )rh r r alphas r? r@ zTestQuantile.V
s Q% rA c | g d}t j |d }t j |dd d S )N)r r# r g?g
ףp=
?rR )maxulp)r9 quantiler assert_array_max_ulp)rh r ri s r? test_max_ulpzTestQuantile.test_max_ulp
sA OO K4
''4':::::rA c t j d dz }t t j |d d t t j |d d t t j |d d d S )Nrx ri r r rR r r )r9 r: r rU r s r? rn zTestQuantile.test_basic
sn IaLL3 R[A&&+++R[A&&,,,R[C(($/////rA c t t j dg }t j dd }t ||d t t | |j t j g d }t j || }t
|| t |j |j d S )NTFr )FTT)r9 r rU r r r r
)rh ri tf_quant quant_ress r? test_correct_quantile_valuez(TestQuantile.test_correct_quantile_value s HdV ;tU++Xqt$$$T(^^QW--- H((( ) )K1%% 9a(((Y_ag.....rA c d t d D }t j |d }t |d t t | t
t j |d }t |t dd t t | t
t j |t dd }t |t dd t t | t
t j |t dd g }t |t j t dd g t t | t j t j |t dd gg }t |t j t dd gg t t | t j t j d }t t j |t dd t dd d S )Nc . g | ]}t |d S r r r s r? r- z.TestQuantile.test_fraction.. r rA rx r rR rz rP rN ) rg r9 rU r r r r r r: )rh r r> s r? r zTestQuantile.test_fraction s . .U1XX . . . K1 QT!WWh''' K1 QA'''T!WWh''' K8Aq>> * *QA'''T!WWh''' KHQNN+ , ,Q(1a..!122333T!WWbj))) KXa^^ ,- . .Q8Aq>>"2!344555T!WWbj)))
IaLLR[HQNN33Xa^^DDDDDrA c \ t j g dd }t t t j |d t j g dd }t t t j |d t j g dd }t t t j |d d S r )r9 r r
rd rU r s r? r zTestQuantile.test_complex) s 777sCCCieS999777sCCCieS999777sCCCieS99999rA c t j g d }| }t j t j d |d t || | }| }t j t j d |d t || d S )N)r r rm ri r r r r )r9 r r rU r: r
r )rh p0r, s r? r z TestQuantile.test_no_p_overwrite2 s
X///
0
0 GGII
BIdOOQz::::1b!!!
YY[[
HHJJ
BIdOOQz::::1b!!!!!rA r r c | t j t j ddg| dgd }|j |k sJ d S )NrR rP r! ri r r )r9 rU r r )rh r r> s r? test_quantile_preserve_int_typez,TestQuantile.test_quantile_preserve_int_type> sM k"(Aq6777#!*, , ,yE!!!!!!rA r c H t j ddd }t j t j g d dz || }t t j | | t j g d|| }t t j | | d S )Nr rR r )r rR rR rP rP rZ rZ rN r[ r[ rR rR r; r; r; rx rx rz r r )r r r r )r9 r` rU r r rd )rh r ra rU s r? test_quantile_monotonicz$TestQuantile.test_quantile_monotonicD s
[As
#
#;rx )2 )2 )2 3 358 9: r s r? 'test_quantile_add_and_multiply_constantz4TestQuantile.test_quantile_add_and_multiply_constant su i##D))
JJqMM K5 0 0 0 AE5@@@!a%HHH AE5@@@!a%HHH
[!QYv
6
6
6 6 ^ # #E SU^^++8AI&&#a%i..1*<<< 2;q%#I#I#IJJJJJ 2;q%#H#H#HIIIII
,
,
,5yCE NN** 2;q%#I#I#IJJJJJ!e)$$AI(::: r{1eack(CCCE E E E E 2;q%#H#H#HIIIII
2
2
2 Ar{1eack& I I I J J J J J
y
5yCE NN** 2;q%!A#+f#M#M#MNNNNN 2;q%#G#G#GHHHHH
w
Ar{1eH E E E F F F F F
x
Ar{1eG D D D E E E E E
Ar{1eF C C C D D D D DrA N) r r r r@ rW rn r\ r r r r5 r6 r7 r9 rL rc quantile_methodsre
hypothesisgivenr r stintegersfloatsrn rp rw rz r rA r? rP rP
s! ; ; ;0 0 0 / / /E E E6: : :
"
"
" [ Wbl<&@AA" " BA"
[ X'788 2 2 98 2 ZRZ (R[1 E E E ) %4:e!M !M !MN N NO O O
2 2 O O
2
2 2 2 [ X'788 [ Wooo66: : 76 98:8 [ X'788 [ Wooo666E 6E 76 986E 6E 6ErA rP c & e Zd Z ej ej dddd ej dddd ej dddd ej dddd d Z ej ej dddd ej dddd ej dddd d
Z ej ej dddd ej dddd ej dddd d Z d Z
d
S )TestLerpFr rR )rk rj rf rg rh ri )t0t1ri r c t j ||| }t j ||| }||k s||k r
||k sJ d S ||k ||k k r
||k sJ d S ||k sJ d S r8 rj _lerp)rh r r ri r l0l1s r? +test_linear_interpolation_formula_monotonicz4TestLerp.test_linear_interpolation_formula_monotonic s Yq!R
Yq!R
88qAvv8888882g1q5
!
!888888888888rA )tri r c ||k r'|t j ||| cxk r|k sn J d S |t j ||| cxk r|k sn J d S r8 r )rh r ri r s r? )test_linear_interpolation_formula_boundedz2TestLerp.test_linear_interpolation_formula_bounded s
66 !Q**////a//////// !Q**////a////////rA c t j ||dd|z
z
}t j ||d|z
}t || d S r? )rj r r )rh r ri r rX rZ s r? +test_linear_interpolation_formula_symmetricz4TestLerp.test_linear_interpolation_formula_symmetric sL yAqAE{++ !QA&&e$$$$$rA c t j d }t j d }t j d }t j ||| dk sJ d S )NrP r[ r# g@)r9 r rj r )rh ri r r s r? +test_linear_interpolation_formula_0d_inputsz4TestLerp.test_linear_interpolation_formula_0d_inputs sN HQKK HQKK HSMMyAq!!S((((((rA N) r r r r| r} r~ r r r r r r rA r? r r s ZU5-.!= = ="U5-.!= = =#")eE.4G G G#")eE.4G G G
H H H H H Z E%,-< < <! E%,2eE E E! E%,2eE E E F F F 0 0
F F 0 Z E%,-< < <! E%,2eE E E! E%,2eE E E F F F % %
F F % ) ) ) ) )rA r c e Zd Zd Zd Zd Zd Zd Zej
dg dde j d ge j d
f d Z
d Zd
Zd Zej ed d Zd Zd Zd Zd Zej
dg d d Zej
ddg ej
d g d d Zd S )
TestMedianc t j d }t j d }t j d dd }t t j | d t
t j | d t
t j | d t
t j |d g d t t j |d dd
g t
t j |d d t j g d }t |d |d z d z t j | t j g d
}t |d t j | t j g d }t |d t j | t t j | j d t j |d< t t j | j d d S )NrR rP ry rZ ri rj r r rK rj r rN )6-¦? 0?}r?g4 ?r )r r r )r r r ra )
r9 r r: r{ r medianr r r rZ )rh a0r r ri s r? rn zTestMedian.test_basic s
Xa[[
Yq\\
Yq\\
!
!!Q
'
'RYr]]A&&& "
s+++ "
s+++ "1---/@/@/@AAARYr***QF333 "4000#666 H@@@ A AQqTAaD[B . ! === H555 6 6QqT29Q<<((( H555 6 6QrUBIaLL)))RYq\\&***v!RYq\\&*****rA c t j ddgddgddgddgg }|t j dd d
fD ]i}| }t j |d t
|j D ]}t j || t || jt t j |d ddg t t j |j
d ddg t t j | d
t t j |d d
t t j |j
d
d S )NrP rZ r rR ry rz rN r[ rh r r r r ) r9 r rb randintr r rg r r
r r )rh a3ri origaxs r? test_axis_keywordz TestMedian.test_axis_keyword sc
X1v1v1v1v bi''3Y'??@ ( (A6688D Iad # # # #AFmm
&
& !"%%%%% q$ ' ' ' ' "1---Aw777 "$Q///!a999 "
s+++ "4000#666 "$-----rA c t j ddgddgddgddgg }t j d }t j d }t j d dd }t t j | d
d t t j | d
d t t j | d
d t t j | d d
g d t t j | d d
ddg t t j | d d
d t t j | d d
ddg t t j |j d d
ddg t j dt j d }t j
| t t j |d t j | d d t t j |d t j | dd t t j |d t j | dd t t j |d t j | dd d S )NrP rZ r rR ry rz rN r[ Tr ri rj )r r r r3 r! rY r )r r ) r9 r r: r{ r r r r r rb r r )rh r r r r a4s r? test_overwrite_keywordz!TestMedian.test_overwrite_keyword sF
X1v1v1v1v Xa[[
Yq\\
Yq\\
!
!!Q
'
' "''))TBBBAFFF "''))TBBBCHHH "''))TBBBCHHH "''))TJJJ))) + + + IbggiiA > > >A H H H IbggiiD A A A3 H H H IbggiiA > > >Q I I I "$))++t!LLLQ ! ! ! Yy
3
3
3
;
;I
F
F
"((**%%% "4000 "''))$MMM O O O "1--- "''))!TJJJ L L L "1--- "''))!TJJJ L L L "1--- "''))!TJJJ L L L L LrA c g d}t t j | d |g}t t j | d t t j |d | d S )Nr rP r r )r r9 r r )rh r r] s r? rK zTestMedian.test_array_like9 sg IIBIaLL!,,,SBIbMM1--- "1---q11111rA c G d dt j } |g d }t t j | d d S )Nc e Zd ZddZddZdS ),TestMedian.test_subclass..MySubClassNc b t j | | }||_ |S r8 )r9 rK r info)cls input_arrayr objs r? __new__z4TestMedian.test_subclass..MySubClass.__new__D s+ j--22377
rA c dS )Nr )rh r r r s r? r z1TestMedian.test_subclass..MySubClass.meanI s rrA r8 )NNN)r r r r r r rA r?
MySubClassr B s<
rA r r r )r9 r r r )rh r ri s r? r zTestMedian.test_subclass@ sa
Jyyy ! !RYq\\2&&&&&rA r r r r r c G d dt j }t j t j | | }t || sJ dS )z6Check that we return subclasses, even if a NaN scalar.c e Zd ZdS )-TestMedian.test_subclass2..MySubclassNr r rA r?
MySubclassr S r rA r N)r9 r r r r r )rh r r rB s r? test_subclass2zTestMedian.test_subclass2O sq
Ibhsmm((44 5 5!Z(((((((rA c t j d }t j d }t t j |d| | t j d }t t j |d| | t j d }t t j || | d S )Nr9 r r r r rR r )r9 rC rf r r r: s r? r; zTestMedian.test_outY s HTNN GFOORYq!+++Q/// HTNNRYq!+++Q/// HRLLRYqa(((!,,,,,rA c 2 t j d 5 t j ddt t j d }t j d }t j |d< t t j |d| | t j d
}t t j |d | | t j d }t t j || | d d d d S # 1 swxY w Y d S )
NTrp rr rs r9 r r r r r rR r )
ru rv rw r r9 rC rf rZ r r r: s r? r= zTestMedian.test_out_nanb s/
$D
1
1
1 1 1 #Hb. A A AAAfAdG 1aQ/// 3 3 3A 1aQ/// 3 3 3 A 1!,,,a 0 0 0 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1s C)D DDc h t j dt }t j |d< t t j | t j t t j |d t j t j dt ddd }t j |d< t j |d < t t j | t j t t j | j d t j t j dt ddd d }t j |d
< t j |d < t t j |d | t j t j dt ddd d }t j |d
< t j |d < t t j |d | t j t j dt ddd d }t j |d < t j |d< t t j |d | d S )Nr{ r! rP r r rZ rN r r? r rs rR r rV )r9 r: r rZ r r r{ r r s r? r@ z TestMedian.test_nan_behaviorn s Ib & & &v!RYq\\26***RYqq)))26222 Ib & & & . .q!Q 7 7V'
V'
RYq\\26***RYq\\&***
Ibi%00088AqAA1 E E&$&$RYq!__a(((
Ibi%00088AqAA1 E E&$&$RYq!__a(((
Ibi%00088AqAA6 J Jv!v!RYq&))1-----rA z fp errors don't work correctlyr c 2 t j g t }t j d 5 }t j ddt t t j | t j t |d j t u t t | d d d d n # 1 swxY w Y t j g t d }t j d 5 }t j ddt t t j | t j t |d j t u d d d n # 1 swxY w Y t j g t d }t t j |d
| t t j |d
| t j t j t d }t j d 5 }t j ddt t t j |d
| t |d j t u d d d d S # 1 swxY w Y d S ) Nr! Trp rr rs r rP rZ )r r r rR )
r9 r r ru rv rw r r r rZ r ry rC )rh ri r r s r? r zTestMedian.test_empty s
HRu % % %
$D
1
1
1 $Q #Hb. A A A 1 rv . . . AaDM^3 4 4 4 Q # # # $ $ $ $ $ $ $ $ $ $ $ $ $ $ $
HRuA . . .
$D
1
1
1 5Q #Hb. A A A 1 rv . . . AaDM^3 4 4 4 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5
HRuA . . .RYqq)))1---RYqq)))1---
HRV5 2 2 2
$D
1
1
1 5Q #Hb. A A A 11---q 1 1 1 AaDM^3 4 4 4 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5s8 BCC
C?A)E44E8;E8 A!J JJc | t j d }t t t j | t t t j |d< t t t j | t t d S )Nr rP ) r9 r: r r r r r= r rZ )rh r s r? test_objectzTestMedian.test_object s| IbMMRYqxx / /00115999v!RYqxx / /0011599999rA c N t j d }t j |gdz }t t j |d t j | t j |dd }t t j |d t j | | dd }t t j |d
t j | t t j |d t j |d t t j |d t j |d t t j |d
t j |d t j d
d }t j | t t j |d d t j |d d d d d d df
t t j |d d t j |d d d d d d d f
t t j |d d t j |d d d d dd d f
t t j |d d t j |dd d d d d d f
t t j |d d t j |dd d d d d f
t t j |d d t j |dd d d d d f
t t j |d d t j |dd d dd d f
d S )Nr r r rb r ra r rt rR rr rQ rj )ra r r r r rP r r r r r rU )r9 rb r r r r r r r r: r{ r r r r s r? r z TestMedian.test_extended_axis s I ( + + IqcBh RYqv... ! === K2q ! !RYqx000")A,,???
JJq! ! ! # #RYqw///1 >>>RYqy11129QT3J3J3JKKKRYqu---ry/C/C/CDDDRYqv... !"0E0E0EFFF In % % - -m < <
!''))$$$RYqy111!4Yq111QQQ{ 2 2 4 455 7 7 7RYqy111!4Yq111a{ 2 2 4 455 7 7 7RYqz22215Yq111a{ 2 2 4 455 7 7 7RYqy111!4Yq111QQQqqqz 1 1 3 344 6 6 6RYqv...t4YqAaaa{ 2 2 4 455 7 7 7RYqw///5Yq111QQQ{ 2 2 4 455 7 7 7RYqv...t4Yq111a{ 2 2 4 455 7 7 7 7 7rA c t j d }t t j t j |d t t j t j |d t t j t j |d t t j t j |d t t
t j |d d S )Nr r r r! rN r" rW )r9 rf r
r r re r s r? r# z%TestMedian.test_extended_axis_invalid s GM " "blBIqr::::blBIqw????blBIqq9999blBIqv>>>>j")QV<<<<< K | ]}|v rd n j | V dS r0 r1 r2 s r? r z/TestMedian.test_keepdims_out.. r4 rA Tr5 )
r9 rf r r6 r rg r r r rE )rh r r6 r r r3 r\ s @@r? r7 z TestMedian.test_keepdims_out s
GM " " < qv
II,T16::I L L L L L=B16]] L L L L LIhy!!14$C@@@}}}}V\9-----rA r zm8[s]rC rD c t j dd| }d||< t j | }|j |k sJ t j | sJ t j |ddg }|j |k sJ t j | sJ t j dd| dd }d||d
f< t j |d }t t j | g d d S rF ) r9 r: r r rI r r r{ r
rJ s r? test_nat_behaviorz TestMedian.test_nat_behavior s
Ia5 ) ) )#illyE!!!!x}} }mABx((yE!!!!x}} """"" IaU + + + 3 3B : :#q& i"""28C==*>*>*>?????rA N)r r r rn r r rK r r5 r6 r7 r9 rZ r r; r= r@ r r r r r r# r) r7 r r rA r? r r s + + +,. . .$ L L LB2 2 2
'
'
' [ U *llR, > > > >rA r c e Zd Zej ej j dk d ej e
d d Z ej ej j dk d d Z dS )TestAdd_newdocrP Python running -OOr PyPy does not modify tp_docc j d}t t j j j j d t
| | t t
t j j j j dk t t
t j
j j j dk d S )Nz"Current flat index into the array.r~ )
r r9 coreflatiterindexrG rC r ufuncidentityrX index_tricksmgrid)rh r s r? test_add_doczTestAdd_newdoc.test_add_doc s 3RW%+3ISXXI>DDDBGM*233c9:::BF'-566<=====rA c t j j j j }t j ddd |t j j j j k sJ d S )Nz
numpy.corer )r z
bad docstring)r9 r r r rG
add_newdoc)rh prev_docs r? test_errors_are_ignoredz&TestAdd_newdoc.test_errors_are_ignored' sK 7#)1
lJ0JKKK27+19999999rA N)
r r r r5 r6 r r r r r r r r r rA r? r r s [ *a/8LMM [w'DEE> > FE NM> [ *a/8LMM: : NM: : :rA r c e Zd Zej ej j dk d ej e d d Z
ej ej j dk d d Z dS )TestAddDocstringrP r r r c t j t j j t j j j d }t j ||j d S )Nc dS docstringNr r rA r? funcz6TestAddDocstring.test_add_same_docstring..func9 FrA )r9
add_docstringr flatrG rh r s r? test_add_same_docstringz(TestAddDocstring.test_add_same_docstring3 sN "*/*ABBB t|,,,,,rA c 0 t t 5 t j t j j d d d d n # 1 swxY w Y d }t t 5 t j |d d d d d S # 1 swxY w Y d S )Nzdifferent docstringc dS r r r rA r? r z=TestAddDocstring.test_different_docstring_fails..funcE r rA )r
r r9 r r r r s r? test_different_docstring_failsz/TestAddDocstring.test_different_docstring_fails? s' <
(
( E E RZ_.C D D D E E E E E E E E E E E E E E E <
(
( : : T#8 9 9 9 : : : : : : : : : : : : : : : : : :s# %AA
A
(B BBN) r r r r5 r6 r r r r r r r r rA r? r r 0 s [ *a/8LMM [(EFF- - GF NM- [ *a/8LMM
:
: NM
:
:
:rA r c ` e Zd Zej dg d d Zd ZdS )TestSortComplexztype_in, type_out))lr )hr )Hr )r r )r r )r r c t j g d| }t j | }t j | | }t || t |j |j d S )N)r[ rZ ry rP rR r! )r9 r sort_complexrd r r r )rh type_intype_outri r r s r? test_sort_realz TestSortComplex.test_sort_realO sp
H___G 4 4 4 ##71::$$X..VX&&&V\8>22222rA c t j g dd }t j g dd }t j | }t || t |j |j d S )N)r ? r r r r! )r r r r )r9 r r r r )rh ri r r s r? test_sort_complexz!TestSortComplex.test_sort_complex_ sq H555S A A A8 < < 22222rA N)r r r r5 r6 r7 r r r rA r? r r M sc [ 0 3
3
3
3 3 33 3 3 3 3rA r )r )r r )r ru r r* fractionsr r r5 r| hypothesis.extra.numpyr hypothesis.strategies
strategiesr~ functoolsr numpyr9 r
numpy.testingr r r
r r r
r r r r r r r numpy.lib.function_baserX
function_baserj numpy.randomr 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/ r0 r1 r2 r3 r4 r5 numpy.core.numericr6 r@ rJ rL r r r r r r9 rl r r r r r r r r rC r r r r r r r r r r6 r7 rL r# r6 rG rM rU r] r r r r r r r rO r r{ rP r r r r r r r rA r? r s
) ) ) ) ) ) " " " " " "
& % % % % % % % % 4 3 3 3 3 3 A@ A@ A@ A@ A@ A@ A@ A@Hd1 d1 d1 d1 d1 d1 d1 d1N: : : : : : : :": : : : : : : :$ 3 3 3 3 3 3 3 3FK4 K4 K4 K4 K4 K4 K4 K4Z;$ ;$ ;$ ;$ ;$ ;$ ;$ ;$|l0 l0 l0 l0 l0 l0 l0 l0^ ; ; ; ; ; ; ; ; : : : : : : : : @ @ @ @ @ @ @ @ ; ; ; ; ; ; ; ;.E E E E E E E E(J J J J J J J J6I? I? I? I? I? I? I? I?Xr> r> r> r> r> r> r> r>j@% @% @% @% @% @% @% @%F' ' ' ' ' ' ' '4>% >% >% >% >% >% >% >%B$" $" $" $" $" $" $" $"R d3 d3 d3 d3 d3 d3 d3 d3N& & & & & & & &RR8 R8 R8 R8 R8 R8 R8 R8j5 5 5 5 5 5 5 56 S2< -
- W0E
E jjj))i; i; i; i; i; i; i; *) i;X;- ;- ;- ;- ;- ;- ;- ;-|# # # # # # # #" D D D D D D D D' ' ' ' ' ' ' '"X& X& X& X& X& X& X& X&vg& g& g& g& g& g& g& g&T1 1 1 1 1 1 1 1h & B B B B B B B B k2 k2 k2 k2 k2 k2 k2 k2\a0 a0 a0 a0 a0 a0 a0 a0H] ] ] ] ] ] ] ] @A A A A A A A ADc@ c@ c@ c@ c@ c@ c@ c@L JE JE JE JE JE JE JE JEZ/) /) /) /) /) /) /) /)da@ a@ a@ a@ a@ a@ a@ a@H > > > > > > > >: : : : : : : :(: : : : : : : ::3 3 3 3 3 3 3 3 3 3rA