y G d< d= Z? G d> d? Z@ G d@ dA ZA G dB dC ZB G dD dE ZC G dF dG ZDdH ZE G dI dJ ZF G dK dL ZG G dM dN ZH G dO dP ZI G dQ dR ZJdS )S N)Decimal)umath)randrandintrandn) assert_ assert_equal
assert_raisesassert_raises_regexassert_array_equalassert_almost_equalassert_array_almost_equal assert_warnsassert_array_max_ulp HAS_REFCOUNTIS_WASM)rational)given
strategies)numpyc 2 e Zd Zd Zd Zd Zd Zd Zd ZdS )
TestResizec t j ddgddgg }t j g dg dg }t t j |d | t j ddgddgddgddgg }t t j |d | t j g dg d g d
g d g }t t j |d | d S )
N r r r r r r r r r r r )r r r )r r r r r r r r nparrayr resizeselfAAr1Ar2Ar3s q/builddir/build/BUILD/cloudlinux-venv-1.0.12/venv/lib64/python3.11/site-packages/numpy/core/tests/test_numeric.py test_copieszTestResize.test_copies s Hq!fq!f% & &h lll344RYq&))3///hAAAA788RYq&))3///h 999iiiCDDRYq&))3///// c t j g d }t j g dg dg }t t j |d | t j ddgddgddgddgg }t t j |d | t j g dg dg dg dg }t t j |d | d S )
Nr! )r r r r )r r r r r r r r r r# r$ r( s r. test_repeatszTestResize.test_repeats# s HYYY h lll344RYq&))3///hAAAA788RYq&))3///h 999iiiCDDRYq&))3/////r0 c t j ddgddgg }t j |d }t |t j g t |j |j t j |d }t |j d t j |d }t |j d d S )Nr r r r r r r r r )r% r& r' r r dtypeshaper) r* Ars r. test_zeroresizezTestResize.test_zeroresize. s Hq!fq!f% & &
Yq$
2rx||,,,QWbh'''
Yq&
!
!RXv&&&
Yq&
!
!RXv&&&&&r0 c t j ddt j fg }t j |d }t |t j d|j t
|j |j d S )Nr ar7 r r )r% zerosfloat32r' r r7 r r9 s r. test_reshape_from_zeroz!TestResize.test_reshape_from_zero: si HQRZ 0 1 2 2 2
Yq&
!
!2rx 9 9:::QWbh'''''r0 c t j ddt j }d}t j t
d 5 t j || d d d d S # 1 swxY w Y d S )Nr
r> )negativematch) new_shape)r% arangerA pytestraises
ValueErrorr' )r) r* rJ s r. test_negative_resizez TestResize.test_negative_resizeA s Ia2: . . .
]:[
9
9
9 . . Ia9 - - - - . . . . . . . . . . . . . . . . . .s A##A'*A'c G d dt j }t j dg | }t t j |d |u sJ t t j |d |u sJ t j g | }t t j |d |u sJ d S )Nc e Zd ZdZdS ))TestResize.test_subclass..MyArray ?N)__name__
__module__ __qualname____array_priority__ r0 r. MyArrayrR H s !# r0 rY r r )r% ndarrayr& viewtyper' )r) rY my_arrs r.
test_subclasszTestResize.test_subclassG s $ $ $ $ $bj $ $ $ 1# ##G,,BIfa(())W4444BIfa(())W4444" ""7++BIfa(())W444444r0 N) rT rU rV r/ r2 r; rB rO r_ rX r0 r. r r sn 0 0 0 0 0 0
'
'
'( ( (. . . 5 5 5 5 5r0 r c e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
Z d Z
d Zd
Zej dej ej ej ej ej ej ej ej ej ej ej g d Z ej d ej ddej ! d dd e"j# e"j$ d z
fd e"j# e"j$ d fg d Z%d Z&ej ! e'd d Z(d Z)d Z*d Z+d Z,d Z-d! Z.d" Z/d# Z0d$ Z1d% Z2d&S )'TestNonarrayArgsc v g dg dg dg}g d}g d}t j || }t || d S )Nr r r r r rZ rZ )rZ r rZ )r r r )r% chooser )r) choicestgtr= outs r. test_choosez TestNonarrayArgs.test_chooseU sY 99 99 99 ii IIi7##S# r0 c b g d}t j |dd }g d}t || d S )N)rF rZ r r rD ir rg )r rZ r r rg r r r% clipr )r) arrrk rj s r. test_clipzTestNonarrayArgs.test_clip_ sA '''gc1a ###S# r0 c r g dg dg}g dg}t j ddg|d }t || d S )Nr r r r r rZ rf rg r r axis)r% compressr r) rq rj rk s r.
test_compressz TestNonarrayArgs.test_compresse sS k1a&#A...S# r0 c g dg dg}t j ddg }t j |d }t || d S )Nr r rg r r r r r r r r r rx )r% r&
count_nonzeror r{ s r. test_count_nonzeroz#TestNonarrayArgs.test_count_nonzerol sW !h1v s+++S# r0 c
g dg dg}t t 5 t j g d }t t j t j | |k d d d d S # 1 swxY w Y d S )Nr! r rZ rf )r r rf x i )r DeprecationWarningr% r& r all
cumproduct)r) r* expecteds r. test_cumproductz TestNonarrayArgs.test_cumproducts s
YY "
,
-
- : :x 7 7 788H BF2=++x788 9 9 9 : : : : : : : : : : : : : : : : : :s AA88A<?A<c l g dg dg dg}t j | }g d}t || d S )Nr r r r )r rZ rf rg )rv rw rD )r rZ rD )r% diagonalr )r) r= rk rj s r.
test_diagonalz TestNonarrayArgs.test_diagonaly sL
\\
\\
^^
k!nnjjS# r0 c g dg dg}t t j | dk t t j t j |d t j g d k t t j t j |d t j ddg k t j d
5 }t j d d t t t j t j g t |d j
t u d d d d S # 1 swxY w Y d S )
Nr! r @r )g @r g @r @ @Trecordalways ) r r% meanr r& warningscatch_warningsfilterwarningsRuntimeWarningisnancategoryr) r* ws r. test_meanzTestNonarrayArgs.test_mean s\
YY "
c!"""rwq!}} (A(AABBCCCrwq!}}"b(:(::;;<<<
$D
1
1
1 5Q #Hb. A A A BHRWR[[)) * * * AaDM^3 4 4 4 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5s A0EE E c V g d}t t j |d d d S )N)r r rZ rD g @r rx g .@)r r% ptpr) r= s r. test_ptpzTestNonarrayArgs.test_ptp s2 & & &RVAA&&&-----r0 c l g dg dg dg}g d}t t j |d | d S )Nr )rZ rf rg rw )rD r r rZ )r ib iX rF rx )r r% prodr) rq rj s r. test_prodzTestNonarrayArgs.test_prod sN |||| }} ooRWSr***C00000r0 c n g dg dg dg dg}g d}t t j | | d S )Nr! r rg rv rw rD r ) r r r r rZ rf rg rv rw rD r r )r r% ravel)r) r= rj s r.
test_ravelzTestNonarrayArgs.test_ravel sI
YY 999lll ;555RXa[[#&&&&&r0 c ` g d}g d}t j |d }t || d S )Nr! )r r r r r r r )r% repeatr )r) r= rj rk s r. test_repeatz TestNonarrayArgs.test_repeat s; II i1ooS# r0 c x g dg dg dg dg}g dg dg}t t j |d | d S )Nr! r r r )r r r r rZ rf )rg rv rw rD r r )r rf )r r% reshaper s r. test_reshapez TestNonarrayArgs.test_reshape sY yy)))YYY =!!!#8#8#89RZV,,c22222r0 c < g d}g d}t t j |d | t j d }t t | t j t | d d S )N)g(\?g(\"R@gffffff@g
@)g?g R@g@g @r )decimalsrS )r r% aroundfloat64r
isinstanceround)r) rq rj ss r.
test_roundzTestNonarrayArgs.test_round s '''###RYsQ///555 JrNN
17799bj11222QWWYY#####r0 r7 c |d }t t t | t t t t |d t t t t |d t t t | d t t |d d t t |d d d S )Nr ndigits)r r r intr )r) r7 r s r. test_dunder_roundz"TestNonarrayArgs.test_dunder_round s
E!HH
588S))***
5D>>3//000
5D111377888U1XXq!!!U1d^^Q'''U1d+++Q/////r0 z val, ndigitsirF zOut of range of int32reason)marksr c t t || t t j | | d S N)r r r% int32)r) valr s r. test_dunder_round_edgecasesz,TestNonarrayArgs.test_dunder_round_edgecases s6 U3((%
w*G*GHHHHHr0 c t j dddz z }t t t |d t j t t |d dddz z t t t |d t j t t |d dddz z t j d }t t t |d t j t t |d d t t t |d t j t t |d d d S )
Nffffff@rD I r r i i )r% r r r r r int64)r) fis r. test_dunder_round_accuracyz+TestNonarrayArgs.test_dunder_round_accuracy s8 JsRV| $ $
5C=="*55666 U1c]]C"b&L999
5C000"*==>>> U1c 2 2 2C"b&LAAA HSMM
5B<<22333 U1b\\3///
5B///::;;; U1b 1 1 1377777r0 zgh-15896)rM r c dddz z }t t t j | d t |d d S )Nr rD r r )r r r% r )r) r s r. test_round_py_consistencyz*TestNonarrayArgs.test_round_py_consistency s? "b&LU2:a==#..a
>>>>>r0 c X g d}t j |d }t |d d S )N)ir rF r rf rD r r )r% searchsortedr )r) rq rk s r. test_searchsortedz"TestNonarrayArgs.test_searchsorted s4 $$$oc1%%S! r0 c g dg dg}t t j | dk t t j |d dk t t j |d dk d S )Nr! r rf r r r r )r r% sizer) r* s r. test_sizezTestNonarrayArgs.test_size sn
YY "
a 1
"###1
"#####r0 c g dg dg dgg}t t j | j d t t j t j d j d t t j t j d d j d t t j t j d d
j d t t j t j d d j d t t j t j d g j d t t j t j d gd j d t t j t j d gd j d t t j t j d gd
j d d S )
Nr r r r r r r r r r r )r r r )r r rx r r rF r r r )r r% squeezer8 r@ r s r. test_squeezez TestNonarrayArgs.test_squeeze s iiIII
. /RZ]](&111RZ 3 344:DAAARZ 3 3!<<<BFKKKRZ 3 3"===CVLLLRZ 3 3!<<<BFKKKRZ&!1!1 23394@@@RZ&!1!1 2;;;A6JJJRZ&!1!1 2;;;A6JJJRZ&!1!1 2<<<BFKKKKKr0 c p g dg dg}t t j | d t t j |d t j g d t t j |d t j ddg t j d 5 }t j d
d t t t j t j g t |d j
t u d d d d S # 1 swxY w Y d S ) Nr! r g bp@S?r ) ?r r r gɪo ?Tr r r ) r
r% stdr& r r r r r r r r s r. test_stdzTestNonarrayArgs.test_std sA
YY "BF1II'8999BF1aLL"(???*C*CDDDBF1aLL"(J
3K*L*LMMM
$D
1
1
1 5Q #Hb. A A A BHRVBZZ(( ) ) ) AaDM^3 4 4 4 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5s .A0D++D/2D/c ddgddggddgddggg}ddgddggddgddggg}t j |dd }t || d S ) Nr r r rf r rZ r rg )r% swapaxesr )r) rj r= rk s r.
test_swapaxesz TestNonarrayArgs.test_swapaxes ss AA 1a&1a&!12!fq!f
AA / 0k!Q""S# r0 c z g dg dg dg}dgdgdgg}t j |dd }t || d S )
Nr! r r rf r r Try keepdims)r% sumr )r) mrj rk s r. test_sumzTestNonarrayArgs.test_sum s^
YY
YY
YY
sRD2$ fQQ...S# r0 c h g d}g d}g d}t j || }t || d S )N)r r rZ )r r r r r r r rZ )r% taker )r) rj indicesr= rk s r. test_takezTestNonarrayArgs.test_take
sD ii)) OOga!!S# r0 c ` ddgddgddgg}t t j | d d S )Nr r r r rZ rf )r r% trace)r) cs r.
test_tracezTestNonarrayArgs.test_trace s8 VaVaV $RXa[[!$$$$$r0 c r ddgddgddgg}g dg dg}t t j |d | d S )
Nr r r r rZ rf )r r rZ )r r rf r r )r r% transposer s r. test_transposez TestNonarrayArgs.test_transpose sN 1v1v1v&yy)))$R\#v..44444r0 c g dg dg}t t j | d t t j |d t j g d t t j |d t j ddg t j d 5 }t j d
d t t t j t j g t |d j
t u d d d n # 1 swxY w Y t j d dg }d |d< t t j | d
d S )Nr! r gUUUUUU@r ) @r r r gv WUU?Tr r r ?g ?) r
r% varr& r r r r r r r )r) r* r Bs r. test_varzTestNonarrayArgs.test_var sq
YY "BF1II'9:::BF1aLL"(3E3E3E*F*FGGGBF1aLL"(J
3K*L*LMMM
$D
1
1
1 5Q #Hb. A A A BHRVBZZ(( ) ) ) AaDM^3 4 4 4 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5
HdAY !BF1IIt,,,,,s .A0D**D.1D.N)3rT rU rV rl rr r| r r r r r r r r r r rL mark parametrizer% int8int16r r uint8uint16uint32uint64float16rA r r paramxfailmathceillog10r r AssertionErrorr r r r r r r r r r r rX r0 r. ra ra S s : : : 5 5 5. . .1 1 1' ' '
3 3 3
$ $ $ [ W
28RX
")RY
BJ
'
0 0
0 [ ^ Y+##+B#CC
Aidi
9 5 56667 YTYztz) 4 45556 . I I I 8 8 8 [nZ@@? ? A@?
$ $ $
L
L
L 5 5 5 % % %5 5 5
-
-
-
-
-r0 ra c e Zd Zd ZdS ) TestIsscalarc | t t j d t t j t j d t t j d t t j d t t j dg t t j d ddlm} t t j |dd dd lm} t t j | d S )
Ng@i90 Fr r )FractionrZ )Number)r r% isscalarr
fractionsr numbersr )r) r r s r.
test_isscalarzTestIsscalar.test_isscalar/ s C !!! BHUOO,,--- E""### G$$%%%BK&&&'''BK%%%&&& '&&&&& HHQOO,,---"""""" FFHH%%&&&&&r0 N)rT rU rV r" rX r0 r. r r . s# ' ' ' ' 'r0 r c & e Zd Zd Zd Zd Zd ZdS )TestBoolScalarc t j }t j }d}t |o||u t |o||u d S )Nxyzr% False_True_r )r) r tr s r. test_logicalzTestBoolScalar.test_logical? sG I H qQ qQ r0 c t j }t j }t ||z |u t ||z |u t ||z |u t ||z |u d S r r' r) r r* s r. test_bitwise_orz TestBoolScalar.test_bitwise_orF n I HQ1 Q1 Q1 Q1 r0 c t j }t j }t ||z |u t ||z |u t ||z |u t ||z |u d S r r' r- s r. test_bitwise_andz TestBoolScalar.test_bitwise_andN r/ r0 c t j }t j }t ||z |u t ||z |u t ||z |u t ||z |u d S r r' r- s r. test_bitwise_xorz TestBoolScalar.test_bitwise_xorV r/ r0 N)rT rU rV r+ r. r1 r3 rX r0 r. r$ r$ > sP r0 r$ c & e Zd Zd Zd Zd Zd ZdS )
TestBoolArrayc t j dgdz t dd | _ t j dgdz t dd | _ t j dgdz t dd | _ | j | _ | j | _ d| j d< d| j d < d| j d< d| j d < d S )
NT) r> r F* r r r ) r% r& boolr* r ocopynmimr) s r. setup_methodzTestBoolArray.setup_method` s 4&2+T22213375'B,d333ACC85'B,d333ACC8&++--&++--
r0 c ` t | j t | j t | j t | j t | j t | j t | j t | j t d D ]}t j dgdz t dd }d||< t t j | t j dgdz t dd }d||< t t j | t || t t ddd g d
z D ]}t j dgd z t }d||< t t j | d |z
t j dgd z t }d||< t t j | d |z
d S )N F r> rg Trw ip i )iT i_ rE iˆ z%rmsg)
r r* r anyr r< r= ranger% r& r9 r list)r) r des r. test_all_anyzTestBoolArray.test_all_anyl sA
DFJJLL !!!DFJJLL !!!
DGKKMM!"""DGKKMM!"""w & &A%3 d333ACC8AAaD BF1II $# T2221337AAaD q M " " " q1" % % % %eAtS))**-?-?-?? 1 1A%6)666AAaD BF1II4!8 , , , ,$& 555AAaD q Mtax 0 0 0 0 0
1 1r0 c > t | j | j t t j | j | j t t j | j | j t t j | j | j t t j | j | j t t j | j | j t t j | j | j t j | j | j t | j | j t j | j | j t | j | j d S )Nrk ) r r* r r% absr< r= logical_notr: r> s r. test_logical_not_absz"TestBoolArray.test_logical_not_abs s, DF7DF+++2646'??DF3332646'??DF33326$&>>46222BF46NN?DF333BF46NN?DF3332647(++TW555
tv46****4646***
tv46""""4646*****r0 c t | j | j z | j t | j | j z | j t | j | j z | j t | j | j z | j t j | j | j | j t | j | j t | j | j z | j t | j | j z | j t | j | j z | j t | j | j z | j t j | j | j | j t | j | j t | j | j z | j t | j | j z | j t | j | j z | j t | j | j z | j t j | j | j | j t | j | j t | j | j z | j t | j | j z d t | j dz | j t | j dz | j t | j | j z | j t | j | j z | j t | j dz | j t | j dz | j t | j | j z | j t | j | j z | j t | j dz | j t | j dz | j d S )NrL FT)
r r* r r%
logical_orr: logical_and logical_xorr< r= r> s r. test_logical_and_or_xorz%TestBoolArray.test_logical_and_or_xor s 46DF?DF33346DF?DF33346DF?DF33346DF?DF333
dfdf$&11114646***46DF?DF33346DF?DF33346DF?DF33346DF?DF333
tvtv4622224646***46DF?DF33346DF?DF33346DF?DF33346DF?DF333
tvtv4622224646***47TV+TW55547TV+U33347T>4733347U?DF33347TV+TV44447TV+TW55547T>4622247U?DG44447TV+TW55547TV+TW55547T>4733347U?DG44444r0 N)rT rU rV r? rJ rO rT rX r0 r. r5 r5 _ sP
1 1 16 + + + 5 5 5 5 5r0 r5 c e Zd Zd Zd Zd ZdS ) TestBoolCmpc t j dt j | _ t j | j j t
| _ t j dt j | _ t j | j j t
| _ d}t d D ]Wfdt d D | j ||dz < fdt d D | j ||dz < |dz
}Xd}t d D ]Wfd
t d D | j ||d z < fd t d D | j ||d z < |d z
}X| j | _ | j | _
t j | j | j < t j | j
| j < | j | _ | j | _ t j | j d d d
| j d d d
< t j | j d d d
| j d d d
< t j | j dd d
| j dd d
< t j | j dd d
| j dd d
< t j | j dd d
| j dd d
< t j | j dd d
| j dd d
< | j | _ d| j dd d
< | j | _ d| j dd d
< | j | _ | j | _ | j | j xx dz cc< | j | j xx dz cc< t j | j dd d | j dd d < t j | j dd d | j dd d < t j | j d
d d | j d
d d < t j | j d
d d | j d
d d < d| j d d d | j d d d < d| j d d d | j d d d < d S )NrB r> r c g | ]
}d |z z S r rX .0xr s r.
z,TestBoolCmp.setup_method.. ! 8 8 8!QAX 8 8 8r0 rv c ( g | ]}d |z z dk S r6 rX r\ s r. r_ z,TestBoolCmp.setup_method.. % @ @ @!q1a4xAo @ @ @r0 c g | ]
}d |z z S r[ rX r\ s r. r_ z,TestBoolCmp.setup_method.. r` r0 r c ( g | ]}d |z z dk S r6 rX r\ s r. r_ z,TestBoolCmp.setup_method.. rb r0 r r r F rf g )r% onesrA r r r9 efr rH edrF r; nfndnaninffinfdinfefnonanednonansignfsignd)r) r r s @r. r? zTestBoolCmp.setup_method s5 BJ///'$&+T222BJ///'$&+T222
r A 8 8 8 8uQxx 8 8 8DF1QqS5M @ @ @ @uQxx @ @ @DGAacEN
FAA
r A 8 8 8 8uQxx 8 8 8DF1QqS5M @ @ @ @uQxx @ @ @DGAacEN
FAA&++--&++-- 6 6FKKMM FKKMM ')v ##A#twsss|$')v ##A#twsss|$*,& !$Q$1
&*,& !$Q$1
&)+ !$Q$1
&)+ !$Q$1
&w||~~ " QTTw||~~ " QTTV[[]]
V[[]]
47s"
47s"+-6'
14a4A '+-6'
14a4A '+-6'
14a4A '+-6'
14a4A '*-
14a4A '*-
14a4A '''r0 c N t d D ]}t | j |d dk | j |d t | j |d dz
dk | j |d t | j |d dk | j |d t | j |d dk | j |d t | j |d dz dk | j |d | j |d dk }t || j |d | j |d t j | j |d k }d| j |d k }t || t || t | t j | t j t | t j | t j t | t j | t j t t j | j
|d | j |d t t j | j
|d | j |d t t j | j |d | j |d t t j
| j |d | j |d t t j | j |d | j |d d S )Nr r r )rF r r rh r%
zeros_liker\ r astyper rj isfiniterm isinfrp signbitrr r) r rr2r3s r.
test_floatzTestBoolCmp.test_float q H HA tvabbzA~twqrr{ ; ; ; tvabbzA~ 2DGABBK @ @ @ tvabbzQ = = = qrr
{Q < < < qrr
{Q! 3TWQRR[ A A Aqrr
aA q$'!""+ . . .r}TVABBZ 8 88BdfQRRj B q" % % % q" % % % qvvbg0A0A B B B rwwrw / /271C1C D D D rwwrw / /271C1C D D D
rx 4 4dgabbk B B B r{47122; 7 7$'!""+ F F F r{49QRR= 9 9DGABBK< H H H rx !""
6 6 QRR8H I I I rz$*QRR. 9 947122; G G G G- H Hr0 c N t d D ]}t | j |d dk | j |d t | j |d dz
dk | j |d t | j |d dk | j |d t | j |d dk | j |d t | j |d dz dk | j |d | j |d dk }t || j |d | j |d t j | j |d k }d| j |d k }t || t || t | t j | t j t | t j | t j t | t j | t j t t j | j
|d | j |d t t j | j
|d | j |d t t j | j |d | j |d t t j
| j |d | j |d t t j | j |d | j |d d S )Nr r r )rF r rH ri r% ru r\ r rv r rk rw rn rx rq ry rs rz s r. test_doublezTestBoolCmp.test_double r r0 N)rT rU rV r? r~ r rX r0 r. rV rV sJ ,. ,. ,.\H H H4H H H H Hr0 rV c e Zd Zd Zd Zej ed ej e j
dk d d Z ej ed d Z d Z
d
S )
TestSeterrc n t j }t |t dddd d S )Nwarnignore)divideinvalidoverunder)r% geterrr dict)r) errs r. test_defaultzTestSeterr.test_default sH ikkS"( % (* * * r0 c F t j 5 t j }t j d }t ||k t j }t |d dk t j d t t j d dk t |d dk t j di | t t j |k d d d d S # 1 swxY w Y d S )Nprintr r raise)r r rX )r% errstateseterrr r )r) r oldnews r. test_setzTestSeterr.test_set$ s:
[]]
(
()++C)7+++C C3J )++C CMW, - - - I7 # # # # BIKK'72 3 3 3 CMW, - - - I BIKK3& ' ' '
(
(
(
(
(
(
(
(
(
(
(
(
(
(
(
(
(
(s C5DD D no wasm fp exception supportr armv5telz See gh-413.c t j d 5 t t 5 t j dg t j dg z d d d n # 1 swxY w Y t j d t j dg t j dg z d d d d S # 1 swxY w Y d S )Nr r rS r )r% r r
FloatingPointErrorr& r r> s r. test_divide_errzTestSeterr.test_divide_err1 s2 [
(
(
( , , 122
0
0"2$ //
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
IX & & & & HbTNNRXrd^^ + + , , , , , , , , , , , , , , , , , ,s5 B;,A# B;#A' 'B;*A' +AB;;B?B?c * t j }d _ t j d 5 }t j d t j d 5 t j g d t j dg t j d g z t t | d
d d d n # 1 swxY w Y d d d n # 1 swxY w Y fd }t j d 5 t j d
d|g t j dg t j d g z d d d n # 1 swxY w Y t j d
t j | t j d 5 t j dd d
d|g d d d n # 1 swxY w Y t j d t j | `d S # t j | `w xY w)Nr Tr r r r ) N r NrS r r c xj dz
c_ | }t t | dk t d|d v d S )Nr r r r )calledr len)args
extobj_errr) s r. log_errz'TestSeterr.test_errobj..log_errG sN q !
J1,---JqM122222r0 r r r )extobjr ) r% geterrobjr r r simplefilterr seterrobjr& r r r )r) olderrobjr r s ` r. test_errobjzTestSeterr.test_errobj; s LNN (555
,%h///[/// , ,L!1!1!1222HbTNNRXrd^^33 Q+++, , , , , , , , , , , , , , ,
,
,
,
,
,
,
,
,
,
,
,
,
,
,
,
3
3
3
3
3 H---
0
0 eQ 0111"2$ //
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
a ( ( ( L # # #H---
>
> "b%G)<====
>
>
>
>
>
>
>
>
>
>
>
>
>
>
> a ( ( ( L # # #
L # # # OOOOs G: *C A C; C C C C C G: C""G: %C"& G: AE G: EG: EAG: G6 G: GG: G
G: :Hc t j } t j t j t j dz d g t j t j dg t d D ]-}t j t j t j d g .t j t j dg t j | d S # t j | w xY w)Nr rf i' ) r% r r r UFUNC_BUFSIZE_DEFAULT ERR_DEFAULTr r& rF )r) r r s r. test_errobj_noerrmaskz TestSeterr.test_errobj_noerrmaskZ s LNN
$ L%5 *Q.6
7
7
7
HRXqc]] # # # 5\\
$
$ e 95;L ! # $ $ $ $ HRXqc]] # # # L # # # # #BL # # # #s B9C$ $C:N)rT rU rV r r rL r
skipifr platformmachiner r r rX r0 r. r r s ( ( ( [(FGG [((**j8 OO, , PO HG, [(FGG HG <$ $ $ $ $r0 r c e Zd Zd Zd Zej ed ej de
j d d Z ej ed d Z
d S )
TestFloatExceptionsc t | } ||| t dd|d|d d S # t $ rF}t t | | dk d|d|d Y d }~d S d }~ww xY w)NFzType z did not raise fpe error 'z'.r z raised wrong fpe error ')r] r r strfind)r) fpeerrflopr^ yftypeexcs r. assert_raises_fpez%TestFloatExceptions.assert_raises_fpen s Q K DAJJJ EE?DuufffM
O
O
O
O
O! K K K CHHMM&))Q..>CeeSSSI
K
K
K
K
K
K
K
K
K Ks #6
B ;BBc | |||| | |||d | | ||||d | |||d |d d S NrX )r )r) r r sc1sc2s r. assert_op_raises_fpez(TestFloatExceptions.assert_op_raises_fpex s
vtS#666 vtSWc::: vtS#b'::: vtSWc"g>>>>>r0 r r typecodeAllFloatc
t j d 5 t j | }t j | j dk r3t j | }|j j }|j }|j }d}d}njt |d j } t j | } ||j j } ||j } ||j }d}d}d}
d } t j | s2|
|d
|| |
|d || |
|
d | |d
|
|
d | |d |
|
d |||z |
|
d | ||z |
|
t j |d
|d
|j z |
|d |d |d |
| d |t j |t j |
| d |d |d |
| d |t j |t j |
| d |t j |t j |
| d |d |t j d d d d S # 1 swxY w Y d S )Nr r r underflowdivide by zeror r overflowr c | |z S r rX r= bs r. z>TestFloatExceptions.test_floating_exceptions..
1 r0 c | |z S r rX r s r. r z>TestFloatExceptions.test_floating_exceptions.. r r0 c | |z S r rX r s r. r z>TestFloatExceptions.test_floating_exceptions..
! r0 r c | |z S r rX r s r. r z>TestFloatExceptions.test_floating_exceptions.. r r0 ?c | |z S r rX r s r. r z>TestFloatExceptions.test_floating_exceptions.. r r0 c | |z
S r rX r s r. r z>TestFloatExceptions.test_floating_exceptions.. r r0 c | |z S r rX r s r. r z>TestFloatExceptions.test_floating_exceptions.. r r0 r c | |z S r rX r s r. r z>TestFloatExceptions.test_floating_exceptions..
ac r0 c | |z S r rX r s r. r z>TestFloatExceptions.test_floating_exceptions.. r r0 c | |z
S r rX r s r. r z>TestFloatExceptions.test_floating_exceptions.. r r0 c | |z S r rX r s r. r z>TestFloatExceptions.test_floating_exceptions.. r r0 c | |z S r rX r s r. r z>TestFloatExceptions.test_floating_exceptions.. r r0 )r% r
obj2sctyper7 kindfinfo_machartinymaxepsr] realr r powernexpro ) r) r r fift_tinyft_maxft_epsr divbyzerortyper r s r. test_floating_exceptionsz,TestFloatExceptions.test_floating_exceptions s [W
%
%
% 6 N 6 NM(++Ex#s**Xe__ */ ' , UU1XX]++Xe__ %
00 rv rv !H G 8G$$
H&&y$4$4gvG G G&&y$4$4gwH H H " "8#3#3VUU1XX
G
G
G " "8#3#3VUU3ZZ
I
I
I " "8#3#3VVF]
L
L
L " "8#3#3fWfVm
M
M
M " "8#%8UU1XXuuQZ7H7H
J
J
J " "9#3#3UU1XXuuQxx
I
I
I " "))55==%%--
" "7#3#3UU1XXuuQxx
I
I
I " "))55==%%--
" "))55==%%..
" "7#3#3UU1XXuuRV}}
N
N
Nk6 N 6 N 6 N 6 N 6 N 6 N 6 N 6 N 6 N 6 N 6 N 6 N 6 N 6 N 6 N 6 N 6 N 6 Ns LL>>MMc \ t j d 5 }t j d t j d 5 t j dd t
t | d t dt |d j
v t j d
t j d
z t
t | d t d t |d
j
v t j t j t j t j z
t
t | d t dt |d
j
v t j d t j d z t
t | d t dt |d
j
v d d d n # 1 swxY w Y d d d d S # 1 swxY w Y d S )NTr r r r r r r r gu <7~r r rF r z
invalid valuegY nr r )
r r r r% r r r r r r messager& ro )r) r s r.
test_warningsz!TestFloatExceptions.test_warnings s>
$D
1
1
1 ;Q !( + + +(((
;
; !R SVVQ'''(C! ,=,==>>>"(5//11 SVVQ'''
c!B%-&8&88999 28BF#3#333 SVVQ'''3qu}+=+==>>> 28F#3#333 SVVQ''' s1R5='9'99:::
;
;
;
;
;
;
;
;
;
;
;
;
;
;
; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ;s6 *H! F=H = H! H
H!H
H!!H%(H%N)rT rU rV r r rL r
r r r r% typecodesr r rX r0 r. r r m s K K K
?
?
? [(FGG [ Z j)ABB8N 8N CB HG8Nt [(FGG; ; HG; ; ;r0 r c e Zd Zd Zd Zd Zd Zd Zej
dddg ej
dd d
g d Z ej
d d
g e j
d e j
d g e j
dg e j
dg gg d Zej
d d
g e j
d e j
d g e j
dg e j
d e j
d fg g e j
d e j
d g e j
d e j
d gg d Zej
d ee j d dd d d d d egz d! Zej j ej d" ej
d d
g ej ee j d dd d d d egz d#$ d% Zd& Zd' Zd( Z d) Z d*S )+ TestTypesc ` t j d }t j d t j d t j d t j d f\ }}}}t j d t j d t j d t j d f\ }}} }
t j
d t j d t j d }
} } t j
d t j d t j d }}}t! ||| t j t j t! ||| t j t j t! ||| t j t j t! ||| t j t j t! || | t j t j t! ||
| t j t j t! || |
t j t j t! ||| t j t j t! ||| t j t j t! |||
t j t j t! ||| t j t j t! ||| t j t j t! ||| t j t j t! ||| t j t j t! ||| t j t j t! ||
| t j t j t! ||| t j t j t! ||| t j t j t! || | t j t j
t! || | t j t j t! || | t j t j
t! ||| t j t j t! ||| t j t j t! |t j |g | t j t j t! |t j |g | t j t j t! |t j |g | t j t j t! |t j |g | t j t j t! |t j |g | t j t j t! ||
t j |g t j t j t! ||t j | g t j t j t! |t j d t j |
g t j t j t! || t j | g t j t j
t! ||
t j | g t j t j
t! |t j | g |
t j t j t! ||
t j |g t j t j
t! ||t j | g t j t j t! |t j
d t j | g t j t j t! |t j |g | t j t j t! |t j |g | t j t j t! |t j |g |
t j t j t! |t j |g | t j t j t! |t j |g | t j t j t! |t j |g | t j t j t! |t j | g | t j t j
d S )Nr rF y @)r% bool_r r
r r r r r r rA r
longdouble complex64
complex128 clongdoubler r7 r& )r) promote_funcr i8i16i32i64u8u16u32u64f32f64fldc64c128clds r. check_promotion_casesz TestTypes.check_promotion_cases s HQKK GAJJ RXa[["(1++ MCc HQKK1 ry||RYq\\ QCc
1
rz!}}bmA6F6F#S a "-*:*:BN1,>???\\#r**BHRX,>,>???\\#s++RXbh-?-?@@@\\"c**BHRY,?,?@@@\\#s++RXbj-A-ABBB\\#s++RXbm-D-DEEE\\#s++RXbm-D-DEEE\\$,,bhr}.E.EFFF\\#t,,bhr~.F.FGGG\\#s++RXbn-E-EFFF \\!S))28BH+=+=>>>\\!R(("(28*<*<===\\"b))28BH+=+=>>>\\"c**BHRX,>,>???\\#s++RXbh-?-?@@@\\#s++RXbj-A-ABBB\\#s++RXbj-A-ABBB\\#s++RXbj-A-ABBB\\#s++RXbj-A-ABBB\\#s++RXbj-A-ABBB\\#s++RXbl-C-CDDD\\$,,bhr}.E.EFFF\\#s++RXbn-E-EFFF \\"(A3--44bhrw6G6GHHH\\"(A3--44bhrx6H6HIII\\"(A3--55rx7I7IJJJ\\"(A3--55rx 7J7JKKK\\"(B4..#668I8IJJJ\\#rx77"(9K9KLLL\\#rx77")9L9LMMM\\"(2,,#@@Xbj)) + + +\\#rx77"*9M9MNNN\\#rx77"*9M9MNNN\\"(C5//377"*9M9MNNN\\#rx77Xbl++ - - -\\#rx77Xbm,, . . .\\",r"2"2BHcUOODDXbm,, . . .
\\"(A3--55rx
7K7KLLL\\"(A3--55rx7I7IJJJ\\"(A3--55rx 7J7JKKK\\"(B4..#668L8LMMM\\"(C5//377"*9M9MNNN \\"(C5//377")9L9LMMM \\"(C5//488Xbl++ - - - - -r0 c d }| | t j ddg t j ddgt j fD ]r}d|z }t |j t j d d |j z t j d |z }t |j t j t j d |j z t j d |z }t |j t j d d |j z t j d |z }t |j t j d
d |j z t j d |z }t |j t j d d |j z d |z }t |j t j d
d |j z t j
d |z }t |j t j t j
d |j z t j d |z }t |j t j d
d |j z t j d |z }t |j t j d d |j z td S )Nc 6 t j | | j S r )r% addr7 r s r. res_typez)TestTypes.test_coercion..res_type! s 6!Q<<% %r0 TFr r r> gX9v?f8z
array type %sf4f2y X9v?c16c8)
r r% r& r r r7 r r rA r r r r )r) r
r= r s r.
test_coercionzTestTypes.test_coercion s` & & &
""8,,, (D%=))28RHBG+L+L+LM M MA A "(4../AG2K L L L
e$$q(A "(2="9"9(172
4
4
4
5!!A%A "(4../AG2K L L L
5!!A%A "(4../AG2K L L L
5!!A%A "(4../AG2K L L L
A "(5//?QW3L M M Mv&&*A "(2>":":(172
4
4
4
f%%)A "(5//?QW3L M M M V$$q(A "(4../AG2K L L L L+ M Mr0 c | t j t t j d t j d k d S r )r r% result_typer r7 r> s r. test_result_typezTestTypes.test_result_typeV sC ""2>222t$$677777r0 c t t j dd t j d t t j dd t j d t t j dd t j d t t j dd t j d t t j dd t j d t t j dd t j d t t j dd t j d t t j d
d t j d t t j dd t j d t t j d d
t j d t t j d d t j d t t j d d
t j d t t j dd t j d t t j dd t j d t t j dd t j d t t j dd t j d d S )Ni8z>U16U21zS5z>U8zU5zM8zm8)r r%
promote_typesr7 r> s r. test_promote_types_endianz#TestTypes.test_promote_types_endianZ sX R%eU33RXd^^DDDR%eU33RXd^^DDDR%eV44bhuooFFFR%eV44bhuooFFFR%fe44bhuooFFFR%fe44bhuooFFFR%eU33RXd^^DDDR%eU33RXd^^DDDR%eU33RXd^^DDDR%eU33RXd^^DDDR%eU33RXd^^DDDR%eU33RXd^^DDDR%eU33RXd^^DDDR%eU33RXd^^DDDR%eU33RXd^^DDDR%eU33RXd^^DDDDDr0 c J g d}d}t j t }|D ]F}t j | }t j || sJ t j || |u sJ G|D ]s}t j | }t j || rJ t j t 5 t j || d d d n # 1 swxY w Y tt j d }t j || sJ t j || |u sJ d S )N)r r
r r r9
BHILQPFDGmMfVdouble)r% r7 r can_castr rL rM TypeError)r) valid_types
invalid_types rational_dt numpy_dtype double_dts r. #test_can_cast_and_promote_usertypesz-TestTypes.test_can_cast_and_promote_usertypesp sy
BAA ;
hx(( & M MK (;//K;{K88 8 88#K==LLLLL( ; ;K (;//K{; << < << y))
;
; k:::
;
;
;
;
;
;
;
;
;
;
;
;
;
;
; HX&& {; 22222 ;779DDDDDDs 1CC C swapr string_dtypeUSc |dk rd }n t j }|}t |d| t j |dz t |d| t j |dz t |d| t j |dz t |d | t j |dz t |d
| t j |d z t |d | t j |d
z t |d| t j |dz t |d| t j |dz t |d| t j |dz t |d| t j |dz t |d|dz t j |dz t |d|dz t j |dz t |d|dz t j |dz t |d|dz t j |dz t |d|dz t j |dz t |d|dz t j |dz t |d |dz t j |dz t |d |dz t j |dz t |d
|dz t j |d z t |d
|dz t j |dz t |d |dz t j |d
z t |d |dz t j |dz t |d|dz t j d d S )Nr+ c , t j || S r r% r r s r. r z6TestTypes.test_promote_types_strings.. s )9!Q)?)? r0 r9 5r 4u13u2u410r 20i1i26i411r 21130O)r% r r r7 )r) r+ r, r r. s r. test_promote_types_stringsz$TestTypes.test_promote_types_strings s 6>> ? ?MM ,M ]]61--rx#???]]3**BHQsUOO<<<]]4++RXae__===]]4++RXae__===]]4++RXaf-=-=>>>]]4++RXaf-=-=>>>]]4++RXae__===]]4++RXae__===]]4++RXaf-=-=>>>]]4++RXaf-=-=>>>]]61S51128AcE??CCC]]61T622BHQtV4D4DEEE]]3#..3@@@]]3$//!D&1A1ABBB]]43//!C% AAA]]4400"(1T62B2BCCC]]43//!C% AAA]]4400"(1T62B2BCCC]]43//!D&1A1ABBB]]4400"(1T62B2BCCC]]43//!D&1A1ABBB]]4400"(1T62B2BCCC]]3$//# ?????r0 dtype1dtype2V6V10)name1r )name2r c t j t 5 t j || d d d d S # 1 swxY w Y d S r )rL rM r$ r% r r) rD rE s r. test_invalid_void_promotionz%TestTypes.test_invalid_void_promotion s ]9
%
% - - VV , , , - - - - - - - - - - - - - - - - - -s =AArH r zi8,i8zi8,>i8i4,i4c < t j || |k sJ d S r r1 rK s r. test_valid_void_promotionz#TestTypes.test_valid_void_promotion s( //6999999r0 r7 Allzi,i10iS3S100U3U100c B ddi}t j || }t j || }|j |j k sJ | }|j rd S t j || }|j dk r
|j J n
|j |k sJ |j sJ d S )Nr metadatar- )r% r7 r rX newbyteorderisnativechar)r) r7 rX ress r. %test_promote_identical_types_metadataz/TestTypes.test_promote_identical_types_metadata s q6222 ue,,|u~---- ""$$ > F ue,, : <''''<8++++||r0 z)ignore:Promotion of numbers:FutureWarningr )r c H ddi}ddi}t j || }t j || } t j || }n# t $ r Y dS w xY w|j dvs|j |j dk r
|j J nP||k r||u sJ nC||k r4t j ||j |k r
|j du n|j |k n |j J | }|j |k sJ t j || }||k sJ |j |j k sJ dS )a Metadata handling in promotion does not appear formalized
right now in NumPy. This test should thus be considered to
document behaviour, rather than test the correct definition of it.
This test is very ugly, it was useful for rewriting part of the
promotion, but probably should eventually be replaced/deleted
(i.e. when metadata handling in promotion is better defined).
r r rW NUSVrX )
r% r7 r r$ r[ namesr8 rX r rY )r) rD rE metadata1 metadata2r\ res_bss r. test_promote_types_metadataz%TestTypes.test_promote_types_metadata sw F F &9555&9555 "6622CC FF 85 CI$9SY"__ <''''
F]]&=====
F]] 44>> $$$ )))<''' $$&&)++++!&&11}}}}#,......s A
AAc 4 t t j t j t j t t j t j t t t j t t t t j dd t t j dd t t j dd t t j ddd t t j ddd t t j ddd t t j d
dd t t j d
dd t t j dd d t t j dd d
t t j ddd
t t j ddd t t j dd t t j dd t t j dd t t j dd t t j dd t t j dd t t j dd t t j dd t t j dd t t j dd t t j dd t t j dd t t j d d t t j d d t t j d d t t j d d t t j dd t t j dd t t j dd! t t j dd t t j dd t t j dd t t j dd" t t j dd# t t j dd# t t j dd$ t t j dd% t t j dd" t t j dd& t t j dd' t t j dd( t t j dd) t t j d d" t t j d d# t t j d d* t t j d d% t t j dd+ t t j dd& t t j dd, t t j dd( t t t j dd t t t j d d t t j t j t j - d S ).Nr r r r= S11nor r equivzi4 same_kindz>u4unsafer9 S5S4r rR r4 S2r6 r7 S10S9r S20S19r: r; S6S21U4rT U2U5U10U9U20U19U6U11r )from_to)
r r% r# r r r complexfloatr
r$ r> s r.
test_can_castzTestTypes.test_can_cast sS BHbh//000 BJ00111BK///000 D$''(((BKd+++,,, D%(())) D$--...BKud333444 E5'22333BKug666777 E5&11222BKuf555666 E5+66777BKuk:::;;; E5(33444 FD))***BK---... C&&'''BKT***+++ D$''(((BKd+++,,, D$''(((BKd+++,,, D%(()))BKd+++,,, D%(()))BKe,,,--- D$''(((BKd+++,,, D$''(((BKd+++,,, D%(()))BKe,,,--- D%(()))BKe,,,--- FD))***BK---... C&&'''BKT***+++ D$''(((BKd+++,,, D$''(((BKd+++,,, D%(()))BKd+++,,, D%(()))BKe,,,--- D$''(((BKd+++,,, D$''(((BKd+++,,, D%(()))BKe,,,--- D%(()))BKe,,,---idD999idD999 "(rx88899999r0 c $ t t j dd t t j dd t t j ddd t t j ddd t t j ddg t t j ddgd t t j ddgd t t j dd g t t j dd gd t t j dd
g t t j dd
gd d S ) Nr= rM zi4,i2rk castingr; f1r= rj )r rM )r z(2,3)i4r r% r# r> s r. "test_can_cast_simple_to_structuredz,TestTypes.test_can_cast_simple_to_structured^ sj BKg...///BKg.../// D'8<<<=== D'8<<<===BK|n555666BK|nkJJJJKKK D<.(CCCDDDBK&7888999 D?"3XFFFGGGBK'8&9:::;;; D#4"5xHHHIIIIIr0 c t t j dgd t t j dgdd t t j dgdd t t j ddd t t j ddgfgd t t j ddgfgdd t t j d gd t t j d gdd t t j d
gdd d S ) Nr r= rk r r; rM r )r^ r= )f0z(3,)i4)r rM r[ r r> s r. "test_can_cast_structured_to_simplez,TestTypes.test_can_cast_structured_to_simplen s\ BK 555666 \ND(CCCDDD \ND(CCCDDD BKx@@@@AAABK$
!6 7>>>??? d[M 2 3T8LLLMMM BK!1 2D999::: - .hGGGHHHBK!8 94(02 2 2 2 3 3 3 3 3r0 c t j d t j d z D ]}t j | }t t j |j | t t j |j | t t j |j dz
| t t j |j dz | t j d D ]d}t j | }t t j |j | t t j |j | ed S )Nr uintr r )r% sctypesiinfor r# minr r )r) dtiir s r. test_can_cast_valuesz TestTypes.test_can_cast_values s *U#bj&88 5 5B" B BK++ , , , BK++ , , , BFQJ333 4 4 4 BFQJ333 4 4 4 4*W% - -B" B BK++ , , , BK++ , , , , - -r0 N) rT rU rV r r r r r* rL r
r rC r% r7 rL rY rO rG r r r] slowr itertoolsproductrd r r r r rX r0 r. r r sl H- H- H-T M M Ml8 8 8E E E,E E E0 [ Vb&\22 [ ^c3Z88!@ !@ 98 32!@F [ h 1bhtnnhbhuoo
.bh())828_4E+F+F
G
- -
- [ h 1bhuooxrx
/bh())bh("(4.."="="?"?@ABBDbhw (!3!3
4bhw '!2!2
3
: : : [ W D e$ % %
E4vx @
AB B B B 2 [ [ KLL [ h 1 I R\%())fdFH=>
,/ ,/ ML ,/\K: K: K:ZJ J J 3 3 3& - - - - -r0 r c e Zd ZdS )
NIterErrorNrT rU rV rX r0 r. r r s Dr0 r c
e Zd Zd Zd Zd Zd Zd Zej
de e g ej
ddgd d
g d Z
ej
dg d d
Zej
ddgdg dfdg dfdg dfdg df ej d ddgfg ej
dddg d Zd Zd Zd Zd S ) TestFromiterc 4 d t d D S )Nc 3 K | ] }|d z V
dS )r NrX )r] r^ s r. z'TestFromiter.makegen.. s& ((1((((((r0 r )rF r> s r. makegenzTestFromiter.makegen s ((eBii((((r0 c P t j | t j }t j | t j }t j | t
}t
|j t j t j k t
|j t j t j k t
|j t j t
k d S r )r% fromiterr r r r r r7 )r) ai32ai64afs r.
test_typeszTestFromiter.test_types s {4<<>>2844{4<<>>2844
[
/
/
bhrx 0 00111
bhrx 0 00111BHUOO+,,,,,r0 c J t j t | }t j | t
}t j | t
d }t
t | t | k t
t | dk t t t j | t
t | dz d S )N rD )
r% r& rG r r r r r r
rN r) r r= a20s r. test_lengthszTestFromiter.test_lengths s 8D 0 011 K , ,k$,,..#r22A#h--'(((CB j"+ ,,..#s8}}r/A C C C C Cr0 c t j t | }t j | t
}t j | t
d }t
t j ||k d t
t j ||d d k d d S )Nr r rx )r% r& rG r r r r r r s r. test_valueszTestFromiter.test_values s 8D 0 011 K , ,k$,,..#r22qH}1---...shssm+!44455555r0 c # f K t | D ] }||k rt d|z |V d S )Nzerror at index %s)rF r )r) neindexrI s r. load_datazTestFromiter.load_data sO q AF{{ !4v!=>>>GGGG r0 r7 count error_index)rD rZ )rD rw c | || }t j t 5 t j ||| d d d d S # 1 swxY w Y d S )N)r7 r )r rL rM r r% r )r) r r r7 iterables r. test_2592zTestFromiter.test_2592 s
>>%55
]:
&
& < < KU ; ; ; ; < < < < < < < < < < < < < < < < < )rL rM rN r% r )r) r7 s r. test_empty_not_structuredz&TestFromiter.test_empty_not_structured s ]:-B
C
C
C ) ) K% ( ( ( ( ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) )s A AAdatarH ) r r r r rZ rf rg rv rw rB zi,O)r r )rZ r r r )rw rv rf rg 2i)rB r )r! rd ))r r r )rZ r r length_hintr r c t j | }dz G fdd }t j | | }t j | }t || d S )Nd c $ e Zd ZfdZ fdZdS )?TestFromiter.test_growth_and_complicated_dtypes..MyIterc S r rX )r) r s r. __length_hint__zOTestFromiter.test_growth_and_complicated_dtypes..MyIter.__length_hint__ s ""r0 c " t S r iter)r) r s r. __iter__zHTestFromiter.test_growth_and_complicated_dtypes..MyIter.__iter__ s Dzz!r0 NrT rU rV r r )r r s r. MyIterr sG
#
#
#
#
#
"
"
"
"
"
"
"r0 r r> )r% r7 r r& r )r) r7 r r r r\ r s `` r. "test_growth_and_complicated_dtypesz/TestFromiter.test_growth_and_complicated_dtypes s cz " " " " " " " " " " " k&&((%0008D...3)))))r0 c G d d }t j | d }|j dk sJ |j dk sJ d S )Nc e Zd Zd Zd ZdS ).TestFromiter.test_empty_result..MyIterc dS NrD rX r> s r. r z>TestFromiter.test_empty_result..MyIter.__length_hint__ s rr0 c t g S r r r> s r. r z7TestFromiter.test_empty_result..MyIter.__iter__ s
Bxxr0 Nr rX r0 r. r r s2
r0 r rH r> r4 )r% r r8 r7 )r) r r\ s r. test_empty_resultz TestFromiter.test_empty_result sr k&&((#...yD yC r0 c d}t j t | 5 t j g ddt
d d d d S # 1 swxY w Y d S )Nz>iterator too short: Expected 10 but iterator had only 3 items.rH r! rD )r r7 )rL rM rN r% r r )r) rD s r. test_too_few_itemsz TestFromiter.test_too_few_items s N
]:S
1
1
1 8 8 K 3 7 7 7 7 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8s A
AAc t j t 5 t j g dt
d d d n # 1 swxY w Y d t
d D }t j t 5 t j |t j t
df d d d d S # 1 swxY w Y d S )N)r Nr r> c 3 K | ]}d V dS )r" NrX )r] r s r. r z7TestFromiter.test_failed_itemsetting.. s" 00!I000000r0 rZ r ) rL rM r$ r% r r rF rN r7 )r) r s r. test_failed_itemsettingz$TestFromiter.test_failed_itemsetting s2
]9
%
% 1 1 K C 0 0 0 0 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 10uQxx000
]:
&
& < < K#q(:(: ; ; ; ; < < < < < < < < < < < < < < < < < >CCN)rT rU rV r r r r r rL r
r r objectr r r% r7 r r r r rX r0 r. r r s ) ) )- - -C C C6 6 6 [ WsFm44 [ g} 57IJJ< < KJ 54< [ W&=&=&=>>) ) ?>) [ gv ....
/...
/===
> <<<
=bh}%%%'=>@
AB B [ ]QF33* * 43B B*&
8 8 8
< < < < r )r r% r r& nonzeror> s r. test_nonzero_trivialz TestNonzero.test_nonzero_trivial si R%bhrll33Q777R%bhr&=&=&=>>BBBRZ --u555R%bhsmm44a888R%bhs#&>&>&>??CCCRZ!
..666R%bhsmm44a888R%bhs#&>&>&>??CCCRZ!
..!77777r0 c 6 t t j t j d d t t j t j dd d t t
5 t t j t j d g f d d d n # 1 swxY w Y t t j t j d d t t j t j dd d t t
5 t t j t j d dgf d d d d S # 1 swxY w Y d S r )r r% r r& r r r r> s r. test_nonzero_zerodz TestNonzero.test_nonzero_zerod s R%bhqkk22A666R%bhq&<&<&<==qAAA
,
-
- 9 9 BHQKK002% 8 8 8 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 R%bhqkk22A666R%bhq&<&<&<==qAAA
,
-
- : : BHQKK00A3& 9 9 9 : : : : : : : : : : : : : : : : : :s$ ?6CCC
7FFFc $ t j g d }t t j | d t t j | d t t j | g df t j g dg d }t t j |d d t t j |d d t t j |d
d t t j |d d t t j |d g d f t t j |d g d
f d S )N)r r r rF r r rv r )r r r rf ))r r r r )r r r rg )r r r r )rF r r r )r rg r r ))r= r= )r r; )r r: )rH r r> r= r r r rH )r r r )r r r r )r% r& r r r r) r^ s r. test_nonzero_onedimz TestNonzero.test_nonzero_onedim sc H+++ , ,R%a((!,,,R%a((!,,,RZ]]\\\O444
H^^^OOO
Q
Q
QR%af--q111R%af--q111R%af--q111R%af--q111RZ#''))) 666RZ#'',,, 99999r0 c t j g dg dg }t t j | d d t t j | d d t t j | d d t t j | d d t t j | g dg d f t j d }t t j | d d t t j | d d t t j | d d t t j | d d t t j | g d
g d
f t j g d g d g d
gddg }t t j |d d t t j |d d t t j |d g dg df t t j |d g dg df t |d j j j
t t j |d j d t t j |d j d t t j |d j g dg df t t j |d j g dg df d S )N)r r r )r r r r: r r; r= r r r r )r r r rc )r r r r )r r )r r r r )r )r rZ r )r= r )r r4 r> r= r r rZ )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 ) r% r& r r rv r eyer Tflagsalignedr s r. test_nonzero_twodimz TestNonzero.test_nonzero_twodim* s2 Hiii+ , ,R%ahhtnn55q999R%ahhtnn55q999R%ahhtnn55q999R%ahhtnn55q999RZ]]YYY $:;;; F1IIR%ahhtnn55q999R%ahhtnn55q999R%ahhtnn55q999R%ahhtnn55q999RZ]]YYY $:;;; H///++++++-5@+4N
P
P
P R%af--q111R%af--q111RZ#'',,, )EFFFRZ#''///???)KLLLAcFHN**+++R%afh//333R%afh//333RZ#))LLL,,,+GHHHRZ#))OOO___+MNNNNNr0 c t d D ]}t j dt }d||d d< t t j | d t j |d|z d t j dt }d|d|z d|z < d|d|dz z < t t j | d t j t j d|z d|z d|dz z gf d S ) Nr r> Tr i rD r )rF r% r@ r9 r r rK concatenate)r) r r s r. test_sparsezTestNonzero.test_sparseG s r R RAD)))AAaeeH A q)29Qa+D+D E E ED)))A #Ab1fR!Vm Ab1Q3hK A q) 2626)B)BR!A#XJ(OPP
R
R
R
R R Rr0 c 8 G d dt j }|t j fD ]}t dd D ]}t t dd|z }t j t j | | | }t j | | fD ]H}|D ]C}t t | t j u t |j j DId S )Nc e Zd ZdS )'TestNonzero.test_return_type..CNr rX r0 r. Cr U Dr0 r r r r )
r% r[ rF tuplerK r r r\ r r r] r writeable)r) r r\ rk r8 r^ nzxnzx_is r. test_return_typez TestNonzero.test_return_typeT s%
O 7 7DAqkk
7
7 eAqtnn--Ibgenn--55e<<AA$GG JqMM199;;7 7 7C!$ 7 7 U rz 9::: 5666677
7 7 7r0 c h t j g dg dg }t j g d }t t j |d | t j ddg }t t j |d | t t
t j |d t t t j |d
t t j t j |d t t t j |t j dgdgg d S ) Nr~ r )r r r r r r rx r r r r foo)r% r& r r r
rN r$ AxisError)r) r r s r. test_count_nonzero_axisz#TestNonzero.test_count_nonzero_axisa s Hooo'7'7'78 9 98OOO,,R%aa000(;;;8QF##R%aa000(;;;j""2AFCCCCi!115AAAAblB$4aa@@@@i!1bhaSz 2 2 4 4 4 4 4 4r0 c h d}d }t j d D ]}|t j | j fz }|dk r|dk rCt j d| }t j d| }|d |d
< |d |d < nLt j d gd
z }| d }d|d
< d|d < | | }t j g dt j } |t j
|d || t j g dt j } |t j
|d || t j d }t t j
|d || t t j
|d || t t j
| || |dk rIt j t j d gdz d }t j g dt j } |t j
|d || t j d d gt j } |t j
|d || t j d }t t j
|d || t t j
|d || t t j
| || d S )NzMismatch for dtype: %sc b t | j |j | t | || d S )Nerr_msg)r r7 )r= r r s r. assert_equal_w_dtzITestNonzero.test_count_nonzero_axis_all_dtypes..assert_equal_w_dtw s6 !'7 ; ; ; ; Aw / / / / / /r0 rP VMr r> r r r r z
1970-01-01rw z
1970-01-12)r r r rx r )r r r r r rf r r r r )
r% r r7 namer@ rg r& r rv intpr r void)r) rD r r r r r r s r. "test_count_nonzero_axis_all_dtypesz.TestNonzero.test_count_nonzero_axis_all_dtypesq s ' 0 0 0 ,u% 6 8 6 8BRXb\\ . 00GSyy99r222A,,,A dAdG dAdGG , !!344A &))A *AdG *AdG A 8IIIRW===!!""211"="="="*G= = = = 8IIIRW===!!""211"="="="*G= = = = 8A;; R -af = = = %w8 8 8 8 R -ad ; ; ; %w8 8 8 8 R -a 0 0 %w8 8 8 8 Syy Hbgajj\A -..66v>> 8IIIRW===!!""211"="="="*G= = = = 8QF"':::!!""211"="="="*G= = = = 8A;; R -af = = = %w8 8 8 8 R -ad ; ; ; %w8 8 8 8 R -a 0 0 %w8 8 8 8k6 8 6 8r0 c
ddl m}m} d}d}d}t j d }| dd| }| t }t t | D ]]} ||| D ]N}
||
D ]@} t t j ||
t j ||
|| fz AO^d S ) Nr ) combinations permutationsr )rZ rZ rZ rZ zMismatch for axis: %si ir r rx r )
r r r r% random RandomStater rv r rF r r r ) r) r r ry r rD rngr r lengthcomboperms r. "test_count_nonzero_axis_consistentz.TestNonzero.test_count_nonzero_axis_consistent s
98888888%i##D)) KKcK - -
HHV CII&& / /F% dF33
/
/ (L / / / /D (666(666 #tg
/ / / / //
/ / /r0 c t j g dg dg }t t j |d | t
d S )N)r r r )r r r rX rx )r% r& r r rv r9 r s r. test_countnonzero_axis_emptyz(TestNonzero.test_countnonzero_axis_empty sN Hiii+ , ,R%ab111188D>>BBBBBr0 c 2 t j g dg dg dg }t t j |dd g dg t t j |dd dgd gd
gg t t j |d d gg d S )
N)r r r r )r r rZ r )rg rw r r r Tr )r r r r r r r r rf )r% r& r r r s r. test_countnonzero_keepdimsz&TestNonzero.test_countnonzero_keepdims s Hlll"ll"ll$
%
% R%aa$???"ll^ % % %R%aa$???cA3_ & & &R%a$777cU r0 c t j g dg dg }g dg dg}t | | d S )N)r r r )r r rf r )r r r )r% r& r r )r) r rj s r. test_array_methodz TestNonzero.test_array_method sP
Hiii+ , ,yy)))$QYY[[#&&&&&r0 c 8 t j t j ddg dgt }t t t j | G d d }t t t j t j | g d S )Nr r r r> c e Zd Zd ZdS );TestNonzero.test_nonzero_invalid_object..BoolErrorsc t d )Nz Not allowed)rN r> s r. __bool__zDTestNonzero.test_nonzero_invalid_object..BoolErrors.__bool__ s ///r0 N)rT rU rV r& rX r0 r.
BoolErrorsr$ s#
0
0
0
0
0r0 r' )r% r& r r
rN r )r) r= r' s r. test_nonzero_invalid_objectz'TestNonzero.test_nonzero_invalid_object s Hbh1v&&*& 9 9 9j"*a000 0 0 0 0 0 0 0 0 j"*bh
~.F.FGGGGGr0 c 2 G d d } G d d }t j d | g }t t t j | t j dg | gg }t t t j | t j d | g }t t t j | t j dg | gg }t t t j | d S )Nc e Zd ZdZd ZdS )ATestNonzero.test_nonzero_sideeffect_safety..FalseThenTrueFc 8 | j d| _ S # d| _ w xY w)NT_valr> s r. r& zJTestNonzero.test_nonzero_sideeffect_safety..FalseThenTrue.__bool__ s( % 9 $DIIDI$$$$ NrT rU rV r. r& rX r0 r.
FalseThenTruer+ s( D
%
%
%
%
%r0 r1 c e Zd ZdZd ZdS )ATestNonzero.test_nonzero_sideeffect_safety..TrueThenFalseTc 8 | j d| _ S # d| _ w xY w)NFr- r> s r. r& zJTestNonzero.test_nonzero_sideeffect_safety..TrueThenFalse.__bool__ s( & 9 %DIIDI%%%%r/ Nr0 rX r0 r.
TrueThenFalser3 s( D
&
&
&
&
&r0 r5 TF)r% r& r
RuntimeErrorr )r) r1 r5 r= s r. test_nonzero_sideeffect_safetyz*TestNonzero.test_nonzero_sideeffect_safety s % % % % % % % % & & & & & & & &
HdMMOO, - -lBJ222 Htf}} /0 1 1lBJ222
He]]__- . .lBJ222 Hug
01 2 2lBJ22222r0 c t j dd }|j j sJ |d j j rJ t j | |j j sJ t j | |j j sJ d S )NrZ zi1,i8,i8r> r )r% r@ r r r r r) rq s r. 'test_nonzero_sideffects_structured_voidz3TestNonzero.test_nonzero_sideffects_structured_void s hq
+++y t9?****
3y
y r0 c G d d } t j |d gdz }t t t j | t j |d gdz }t t t j | t j |d ggdz }t t t j | d S )Nc e Zd Zd Zd ZdS ).ThrowsAfterc || _ d S r )
iters_left)r) iterss r. __init__zETestNonzero.test_nonzero_exception_safe..ThrowsAfter.__init__ s
"'r0 c Z | j dk rt d | xj dz c_ dS )Nr zcalled `iters` timesr T)r? rN r> s r. r& zETestNonzero.test_nonzero_exception_safe..ThrowsAfter.__bool__ s3 ?a''$%;<<<1$tr0 N)rT rU rV rA r& rX r0 r. ThrowsAfterr= s2
(
(
(
r0 rC rZ rD r )r% r& r
rN r )r) rC r= s r. test_nonzero_exception_safez'TestNonzero.test_nonzero_exception_safe s
Hkk!nn%b( ) )j"*a000
Hkk"oo&r) * *j"*a000
H{{2'(+ , ,j"*a00000r0 zwasm doesn't have threadsr c ddl m} t j dg }t j d|fg }t j d|fgdz }t j d | d d df d |d
fd t d D }|D ]}| j |u sJ d S )
Nr )ThreadPoolExecutor)r r r r )i r r c . | d S r )r )rq s r. funcz6TestNonzero.test_structured_threadsafety..funcE s KKMMMMMr0 rv ) max_workersc < g | ]} S rX )submit)r] _rq rH tpes r. r_ z.I s' << ! Q///<<<<<<%))<<< A
HHJJJJyBr0 N)rT rU rV r r r r r r r r r r r r! r( r7 r: rD rL r
r r rR rX r0 r. r r sI 8 8 8 : : :: : :"O O O: R R R 7 7 74 4 4 @8 @8 @8D/ / /.C C C ' ' ' H H H 3 3 3@
!
!
! 1 1 1B [(CDD ED r0 r c e Zd Zd Zd ZdS ) TestIndexc t ddd }t dd }t ddd }t ddd }|||f |||f< t t j |d |dk |d |dk |d |dk g |d d |dk f k d S ) Nr rZ rv r r r r r )r r r r% r& r )r) r= r g1g2s r. test_booleanzTestIndex.test_booleanQ s AqMM AJJ
Q
#
#
#
Q
#
#
#r2vYJ"b& 1Q4A;!QU QqT!a%[ABBa1q5kQVVXXYYYYYr0 c t j g d }t j g d }|| }t |g t |j t j d d S )Nr r> r9 )r% r& r r7 )r) r= r r s r. test_boolean_edgecasez TestIndex.test_boolean_edgecaseY se HRw ' ' ' HRv & & &
aDQQWbhw / /00000r0 N)rT rU rV rX rZ rX r0 r. rT rT P s5 Z Z Z1 1 1 1 1r0 rT c 2 e Zd Zd Zd Zd Zd Zd Zd ZdS )TestBinaryReprc J t t j d d d S )Nr 0r r% binary_reprr> s r. test_zerozTestBinaryRepr.test_zerob s" R^A&&,,,,,r0 c t t j d d t t j d d t t j d d d S )NrD 1010i0 11000011101010iԣ 101000111101010011010000r_ r> s r.
test_positivez TestBinaryRepr.test_positivee sh R^B''000R^E**% ' ' 'R^H--/ 1 1 1 1 1r0 c t t j d d t t j d d t t j d d t t j d d d S ) NrF z-1rE z-1010iz-11000011101010i0+\z-101000111101010011010000r_ r> s r.
test_negativez TestBinaryRepr.test_negativel s R^B''...R^C(('222R^F++& ( ( (R^I..0 2 2 2 2 2r0 c t t j dd d t t j dd d t t j dd d d S )
Nr rZ width00000rD rg 0001010r 1111011r_ r> s r. test_sufficient_widthz$TestBinaryRepr.test_sufficient_widtht se R^AQ///999R^Ba000)<<<R^Ba000)<<<</Cc))C 59993 ? ? ? ? @ @r0 c r t t j t j d d d d S )Nl @ rj @1100000000000000000000000000000000000000000000000000000000000000)r r% r` r r> s r. test_large_neg_int64z#TestBinaryRepr.test_large_neg_int64 s< R^BHV$4$4B???" $ $ $ $ $r0 N) rT rU rV ra rf rh ro rt rx rX r0 r. r\ r\ a sq - - -1 1 12 2 2= = =
@
@
@$ $ $ $ $r0 r\ c & e Zd Zd Zd Zd Zd ZdS ) TestBaseReprc L t t j dd d d S )N r 100000r r% base_reprr> s r.
test_base3zTestBaseRepr.test_base3 s$ R\$**H55555r0 c t t j dd d t t j ddd d t t j dd d t t j dd d d S )
Nr rD 12r 000012rA l 5j $ 10QR0ROFCEWr~ r> s r. rf zTestBaseRepr.test_positive s{ R\"b))4000R\"b!,,h777R\"a(($///R\"2B77 GGGGGr0 c t t j dd d t t j ddd d t t j dd d d S )NrD z-12r z-000012z-30r~ r> s r. rh zTestBaseRepr.test_negative s^ R\#r**E222R\#r1--y999R\#q))511111r0 c t t 5 t j dd d d d n # 1 swxY w Y t t 5 t j dd d d d d S # 1 swxY w Y d S )Nr % )r
rN r% r r> s r. test_base_rangez TestBaseRepr.test_base_range s
:
&
& LA
:
&
& LB s 7;;A99A= A=N)rT rU rV r rf rh r rX r0 r. rz rz sS 6 6 6H H H 2 2 2
r0 rz c r e Zd Zd Zd Zd Zd Zej dg d d Z
dS ) TestArrayComparisonsc H t j t j ddg t j ddg }t | t t | t
u t j t j ddg t j g d }t | t t | t
u t j t j ddg t j ddg }t | t t | t
u t j t j ddg t j ddg }t | t t | t
u t j t j dgd t j dgd }t | t t | t
u t j t j d gd
t j d gd
}t | t t | t
u d S ) Nr r r! r r r= S1r> )r= r zS1,u4)r% array_equalr& r r] r9 r) r\ s r. test_array_equalz%TestArrayComparisons.test_array_equal s nRXq!f - -rxA/?/?@@ S T!"""nRXq!f - -rx /B/BCCCS T!"""nRXq!f - -rxA/?/?@@CS T!"""nRXq!f - -rxA/?/?@@CS T!"""nRXse4 8 8 8"(C5PT:U:U:UVV S T!"""nRXxj @ @ @ Xxj @ @ @B B S T!"""""r0 c d t j ddt j g }t j dt j dg }t j ddt j g }t t j || t t j ||d t t j ||d t t j ||d t j t j }t t j || t t j ||d t j g dt }t t j || t t j ||d t j ddgt j dgg }t t j || t t j ||d t j dg gdz \ }}t j t j c|_ |_ t t j ||d t t j ||d d S )
Nr r T equal_nanr! r> r y ? ?F) r% r& rl ro r r r r imag)r) a1a2a3r= r s r. test_array_equal_equal_nanz/TestArrayComparisons.test_array_equal_equal_nan s8
Xq!RVn
%
%
Xq"&!n
%
%
Xq!RVn
%
% BN2r***+++r2666777BN2rT::::;;;BN2rT::::;;; HRV BN1a((()))q!t444555 HYYYc * * *q!$$%%%q!t444555 Hq!frvqk* + +BN1a((()))q!t444555&""#A%1BN1a59999:::q!t44455555r0 c t j g dt }t |d k g d t |d k g d t j d }t |d k g d t |d k g d d S )N)Nr Nr> )TFT)FTFr FFF)TTT)r% r& r r rg r s r. test_none_compares_elementwisez3TestArrayComparisons.test_none_compares_elementwise s H___F 3 3 3Q$Y 3 3 3444Q$Y 4 4 4555 GAJJQ$Y 5 5 5666Q$Y 2 2 233333r0 c t j t j ddg t j ddg }t | t t | t
u t j t j ddg t j g d }t | t t | t
u t j t j ddg t j ddg }t | t t | t
u t j t j ddg t j ddg }t | t t | t
u t j t j ddg t j dg }t | t t | t
u t j t j ddg t j dgdgg }t | t t | t
u t j t j ddg t j dg }t | t t | t
u t j t j ddg t j dgdgg }t | t t | t
u t j t j ddg t j g dg dg dg }t | t t | t
u d S )Nr r r! r r r r )r% array_equivr& r r] r9 r s r. test_array_equivz%TestArrayComparisons.test_array_equiv s nRXq!f - -rxA/?/?@@ S T!"""nRXq!f - -rx /B/BCCCS T!"""nRXq!f - -rxA/?/?@@CS T!"""nRXq!f - -rxA/?/?@@CS T!"""nRXq!f - -rx}}== S T!"""nRXq!f - -rx!qc
/C/CDD S T!"""nRXq!f - -rx}}==CS T!"""nRXq!f - -rx!qc
/C/CDDCS T!"""nRXq!f - -rxIIIyyy8Y/Z/Z[[CS T!"""""r0 r7 )r V3rG c . t j d| }t || ||k rJ |dk rd S t j g d| }||k rJ ||k sJ d S )Nr r> r ) 1 2 3)r% r@ r rE r& r )r) r7 r@ nonzeross r. test_compare_unstructured_voidsz4TestArrayComparisons.test_compare_unstructured_voids s %(((5%(((UN''))))) D== F8 . . .e<<<X%**,,,,,!&&(((((((r0 N) rT rU rV r r r r rL r
r r rX r0 r. r r s # # #* 6 6 6<4 4 4 # # #< [ W&9&9&9::
)
) ;:
)
)
)r0 r c t | | | j j dk s!t j j j dk r.t j dk r t | j |j k nt | j j
|j j
k t | j j |j j k t | j j |j j k t | j j
|j j
k t | j j |j j k t | j j |j j k d S )Nrv r win32)r r7 alignmentr% r
itemsizesysr r r owndatar c_contiguous f_contiguouswritebackifcopyrZ )r^ r s r. assert_array_strict_equalr
s q!
a GIIO $ ) ) LG # #17"####17?2333!QW%66777$(<<===$(<<==='17+BBCCC AG 0 011111r0 c . e Zd Zd ZdOdZdOdZd Zd Zd Zd Z d Z
d
Z d Z e
j d d
d Zd Zd Zd Zd Zd Zd Zd Zd Ze
j dddg d 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. Z.d/ Z/d0 Z0d1 Z1d2 Z2d3 Z3d4 Z4d5 Z5d6 Z6d7 Z7d8 Z8d9 Z9e
j d:d;d< e:j; d= f e:j< d= e:j; d= fg d> Z=e
j d? e:j; d=e:j> @ dAdB e:j? d=dBe@@ f e:j; d=dC@ d @ e:jB e:jC e:j; d=e:jD @ fg dF ZEe
j F dGH e
j dE e:jB d s r. r? zTestClip.setup_method s r0 Nc $ |j ||fd|i|S )Nrk )rp )r) r= r r
rk kwargss r. fastclipzTestClip.fastclip# s# qva...v...r0 c t j || dt j || z z }| |||f| S )Nr rL )r% lessgreaterrh )r) r= r r
rk selectors r. rp z
TestClip.clip& sC 71a==1RZ1%5%5#551ayc222r0 c " t || S r )r r) r r s r. _generate_datazTestClip._generate_data, s Q{{r0 c J t || dt || z z S )Nr )r r r s r. _generate_data_complexz TestClip._generate_data_complex/ s" Q{{S41:: ---r0 c \ t || t j S r )r rv r% rA r s r. _generate_flt_datazTestClip._generate_flt_data2 s a ##BJ///r0 c t j | }t j dk r.| |j d }n-| |j d }|S )Nlittle><)r% asarrayr byteorderrv r7 rY r s r. _neg_byteorderzTestClip._neg_byteorder5 se JqMM =H $ $--c2233AA--c2233Ar0 c t || }| | }t |j j |S r )r r r r7 rZ )r) r r r s r. _generate_non_native_dataz"TestClip._generate_non_native_data= s= Q{{""4((DJ''((( r0 c b dt || z t j S r )r rv r% r r s r. _generate_int_datazTestClip._generate_int_dataC $ T!QZZ''111r0 c b dt || z t j S r )r rv r% r r s r. _generate_int32_dataz TestClip._generate_int32_dataF r r0 r7 z?bhilqpBHILQPefdgFDGOc t j d| }t j d| }t j |dd }|dk r.| | k sJ d S t || d S )NrD r> r r rB )r% rg r@ rp tolistr )r) r7 rq r actuals r. test_ones_pathologicalz TestClip.test_ones_pathologicalK s
gb&&&8Be,,,a## C<<==??hoo&7&7777777 * * * * *r0 c | | j | j }d}d}| ||| }| ||| }t || d S )Ng?333333?r r r r rp r r) r= r r
acacts r. test_simple_doublezTestClip.test_simple_doubleX se 1 1
]]1a
#
#ii1a !"c*****r0 c | | j | j }| t }d}d}| ||| }| ||| }t || d S Nr r )r r r rv r r rp r r s r. test_simple_intzTestClip.test_simple_inta st # #DGTW 5 5
HHSMM
]]1a
#
#ii1a !"c*****r0 c | | j | j }t j |j }|dz }| ||| }| ||| }t || d S )Nr r r r r% r@ r8 r rp r r s r. test_array_doublezTestClip.test_array_doublek su 1 1 HQW
G
]]1a
#
#ii1a !"c*****r0 c | | j | j }d}d}| ||| }| ||| }t || | | j | j }d}| d }t |j j
| ||| }| ||| }t || d S N r ) r r r r rp r r r r r7 rZ r s r. test_simple_nonnativez TestClip.test_simple_nonnativet s
* *47DG < <
]]1a
#
#ii1a 2s###
1 1 $ $AG$$%%%
]]1a
#
#ii1a 2s#####r0 c d| | j | j z }d}d}| ||| }| ||| }t || d| | j | j z }d}d}| ||| }| ||| }t || d S )Nr r rS y ?y ? @)r r r r rp r r r s r. test_simple_complexz TestClip.test_simple_complex s
++DGTW== =
]]1a
#
#ii1a !"c***
##DGTW55 5
]]1a
#
#ii1a !"c*****r0 c ( t j dt }| }| }| ||d }| |d | }t
|| t
|| d S NrD r> )r% rg r r r r r )r) r= r r
amaMs r. test_clip_complexzTestClip.test_clip_complex s
GBg & & &
EEGG
EEGG
]]1a
&
&
]]1dA
&
&!"a(((!"a(((((r0 c Z | | j dz | j dz }|d d dd d df }t |j d t |j d | |dd }| |dd }t || d S )Nr r F_CONTIGUOUS C_CONTIGUOUSgg333333?)r r r r r r rp r )r) r= r r s r. test_clip_non_contigz TestClip.test_clip_non_contig s ! TWq[ 9 9
ccc33Q3hKAGN++,,,AGN++,,,
]]1dC
(
(ii4%%!"c*****r0 c 2 | | j | j }d}d}t j |j }t j |j }| |||| | |||| t || d S r r r s r. test_simple_outzTestClip.test_simple_out 1 1
Xag
hqw
aAr""" !Q3 !"c*****r0 r rk c , | | j | j }t j d }t j d }t j |j t j }| }|Nt j
t 5 | ||||| d d d d S # 1 swxY w Y d S | ||||| |
|||| t || d S )Nr r r> r )r r r r% r r@ r8 r r; rL rM r$ r rp r )r) r r= r r
r r s r. test_simple_int32_inoutz TestClip.test_simple_int32_inout s=
% %dgtw 7 7 JqMM JqMM
XagRX
.
.
.ggii ? y))
<
<
aAr7
;;;
<
<
<
<
<
<
<
<
<
<
<
<
<
<
<
<
<
<
MM!Q2wM 7 7 7 IIaAs # # # %b# . . . . .s CC C c | | j | j }t j d }t j d }t j |j t j }| }| |||| |
|||| t || d S NrF r r> r r r r% r r@ r8 r r; r rp r r s r. test_simple_int64_outz TestClip.test_simple_int64_out % %dgtw 7 7 HRLL HQKK
XagRX
.
.
.ggii
aAr""" !Q3 !"c*****r0 c | | j | j }t j |j t j }t j d }t j |j t j }| }| ||||d |
|||| t || d S Nr r> rk rk r r r r r% r@ r8 r r r; r rp r r s r. test_simple_int64_inoutz TestClip.test_simple_int64_inout % %dgtw 7 7 HQWbj ) ) JqMM
XagRX
.
.
.ggii
aA2x
888 !Q3 !"c*****r0 c D | | j | j }d}d}t j |j t j }| }| ||||d | |||| t || d S Nrf r r> rk r r r r r% r@ r8 r r; r rp r r s r. test_simple_int32_outz TestClip.test_simple_int32_out 1 1
XagRX
.
.
.ggii
aA2x
888 !Q3 !"c*****r0 c $ | | j | j }| }t j |j }d}| |||| | |||| t || d S NrS
r r r r; r% r@ r8 r rp r r) r= r r r
s r. test_simple_inplace_01z TestClip.test_simple_inplace_01 1 1
VVXX HQW
aAq!!! !Q2 !!R(((((r0 c | | j | j }| }d}d}| |||| | |||| t
|| d S r r r r r; r rp r r
s r. test_simple_inplace_02z TestClip.test_simple_inplace_02 sx 1 1
VVXX
aAq!!! "aB !!R(((((r0 c | | j dz | j dz }|d d dd d df }t |j d t |j d | }d}d}| |||| | |||| t || d S )Nr r r r r r ) r r r r r r; r rp r r
s r. test_noncontig_inplacez TestClip.test_noncontig_inplace s ! TWq[ 9 9
ccc33Q3hKAGN++,,,AGN++,,,
VVXX
aAq!!! "aB 1b!!!!!r0 c | | j | j }d}d}| ||| }| ||| }t || d S r r r s r. test_type_cast_01zTestClip.test_type_cast_01
se 1 1
]]1a
#
#ii1a !"c*****r0 c | | j | j }| t j }d}d}| ||| }| ||| }t || d S r ) r r r rv r% r r rp r r s r. test_type_cast_02zTestClip.test_type_cast_02 sx # #DGTW 5 5
HHRX
]]1a
#
#ii1a !"c*****r0 c Z | | j | j }d}d}| |t j | t j | }| |t j | t j | }t || d S r )r r r r r% r rp r r s r. test_type_cast_03zTestClip.test_type_cast_03 s % %dgtw 7 7
]]1bjmmRZ]]
;
;ii2:a=="*Q--88!"c*****r0 c | | j | j }t j d }t j d }| ||| }| ||| }t || d S r )r r r r% rA r rp r )r) r= r r
r r s r. test_type_cast_04zTestClip.test_type_cast_04) su % %dgtw 7 7 JrNN JqMMmmAq!$$
YYq!Q
!"c*****r0 c 2 | | j | j }d}d}| ||t j |j z | }| ||t j |j z | }t || d S Nr rS ) r r r r r% r@ r8 rp r r s r. test_type_cast_05zTestClip.test_type_cast_052 s # #DGTW 5 5
]]1a"(17"3"3 3Q
7
7ii1rx 0 00!44!"c*****r0 c | | j | j }d}| | }d}| ||| }| ||| }t
|| d S )Nr rS )r r r r rp r r )r) r= r m_sr
r r s r. test_type_cast_06zTestClip.test_type_cast_06; sx 1 1 !!!$$ ii3""
]]1c1
%
%!"c*****r0 c Z | | j | j }dt j |j z }d}| | }t |j j |
|| }| ||| }t || d S r )
r r r r% rg r8 r r r7 rZ rp r r )r) r= r r
a_sr r s r. test_type_cast_07zTestClip.test_type_cast_07E s 1 1 2717## # !!!$$CI&&'''hhq!nn
]]31
%
%!"c*****r0 c & | | j | j }d}d}| | }t |j j | ||| }| || }t || d S r )
r r r r r r7 rZ r rp r )r) r= r r
r! r r s r. test_type_cast_08zTestClip.test_type_cast_08P s 1 1 !!!$$CI&&'''
]]31
%
%hhq!nn!"c*****r0 c \ | | j | j }dt j |j z }d}| | }t |j j |
||| }| ||| }t || d S r )
r r r r% rg r8 r r r7 rZ r rp r )r) r= r r
r r r s r. test_type_cast_09zTestClip.test_type_cast_09[ s 1 1 2717## # !!!$$CI&&'''
]]1c1
%
%ii3""!"c*****r0 c d | | j | j }t j |j t j }t j d }t j d }| |||| }| |||| }t || d S )Nr> r r rL )
r r r r% r@ r8 rA rp r r r) r= r r r
r r s r. test_type_cast_10zTestClip.test_type_cast_10f s # #DGTW 5 5 HQWBJ / / / Jt JqMMii1aQi''
]]1a]
*
*!"c*****r0 c | | | j | j }| }| |j d }| }d}d}| |||| | |||| t || d S )Nr r rS rL )
r r r r; rv r7 rY r rp r )r) r= r btr r
s r. test_type_cast_11zTestClip.test_type_cast_11p s * *47DG < <
FFHH
HHQW))#.. / /
VVXX
aA1
%%% !Qr """!!R(((((r0 c d | | j | j }t j |j t j }t j d }t j d }| |||| }| |||| }t || d S )Nr> r r rL ) r r r r% r@ r8 rA r rp r r r( s r. test_type_cast_12zTestClip.test_type_cast_12| s # #DGTW 5 5 HQWBJ / / / HQKK HQKKii1aQi''
]]1a]
*
*!"c*****r0 c 2 | | j | j }d}d}t j |j }t j |j }| |||| | |||| t || d S r r r s r. test_clip_with_out_simplez"TestClip.test_clip_with_out_simple r r0 c | | j | j }t j d }t j d }t j |j t j }| }| ||||d |
|||| t || d S )Nr r r> rk r ) r r r r% r r@ r8 r r; r rp r r s r. test_clip_with_out_simple2z#TestClip.test_clip_with_out_simple2 s % %dgtw 7 7 JqMM JqMM
XagRX
.
.
.ggii
aA2x
888 !Q3 !"c*****r0 c | | j | j }t j d }t j d }t j |j t j }| }| |||| |
|||| t || d S r r r s r. test_clip_with_out_simple_int32z(TestClip.test_clip_with_out_simple_int32 r r0 c | | j | j }t j |j t j }t j d }t j |j t j }| }| ||||d |
|||| t || d S r r r s r. test_clip_with_out_array_int32z'TestClip.test_clip_with_out_array_int32 r r0 c D | | j | j }d}d}t j |j t j }| }| ||||d | |||| t || d S r r r s r. !test_clip_with_out_array_outint32z*TestClip.test_clip_with_out_array_outint32 r r0 c t j d dd }t j | j }| dd| | |dd }t
|| d S )Nrc r rD rL )r% rK r
empty_liker rp r )r) r= rk r s r. test_clip_with_out_transposedz&TestClip.test_clip_with_out_transposed st IbMM ! !!Q ' 'mA q"# 99Q2&&3)))))r0 c < t j d dd }| }|d d dd|dd | |d d dd }t |dd | d S )Nrc r rF rD r rL )r% rK r r; rp r )r) r= r r s r. !test_clip_with_out_memory_overlapz*TestClip.test_clip_with_out_memory_overlap s IbMM ! !!Q ' '
VVXX #2# Arqu %%%99RWa,,1QRR5(+++++r0 c $ | | j | j }| }t j |j }d}| |||| | |||| t || d S r r r
s r. test_clip_inplace_arrayz TestClip.test_clip_inplace_array r r0 c | | j | j }| }d}d}| |||| | |||| t
|| d S r r r
s r. test_clip_inplace_simplez!TestClip.test_clip_inplace_simple sx 1 1
VVXX
aAq!!! !Q2 !!R(((((r0 c | | j | j }| }d}d}t j |||| }| |||| t
|| t ||u d S )Nr r rL )r r r r; r% rp r r )r) r= r r r
r s r. test_clip_func_takes_outz!TestClip.test_clip_func_takes_out s 1 1
VVXX
WQ1!
$
$
$ !Q2 !"b)))ar0 c x t j d }t | t j t j t | t j t j t | t j t j t j t | dt j t j t | t j d t j d S )Ng @)r )r )r r r rD )r% rK r rp rl )r) rH s r.
test_clip_nanzTestClip.test_clip_nan s IbMMQVVV''000QVVV''000QVVBFV33RV<<<QVVV//888QVVBV//88888r0 c t j dt }t j |dd }t j g d }| | k sJ d S )NrD r> r rZ )
r r r r r rZ rZ rZ rZ rZ )r% rK r rp r& r )r) r= r r s r. test_object_clipzTestClip.test_object_clip sh Ib ' ' 'Aq!!8 : : :;;}}(//"3"3333333r0 c t j dt }t t d 5 t j |d d d d d d S # 1 swxY w Y d S )NrD r> z
max or min)r% rK r r rN rp r s r. test_clip_all_nonezTestClip.test_clip_all_none s Ib ' ' '
\
:
: # # GAtT " " " # # # # # # # # # # # # # # # # # #s AA Ac t j dt }t t d 5 | |ddd d d d d S # 1 swxY w Y d S )NrD r> zcasting must be one ofr rv garbager )r% rK r r rN r r s r. test_clip_invalid_castingz"TestClip.test_clip_invalid_casting s Ib ' ' '
!9; ; 6 6 MM!Q9M 5 5 5 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6s AA A z
amin, amaxr r rD c t j dt j }t j t j || | }t j ||| }t
|| d S r )r% rK r minimummaximumrp r )r) aminamaxr= r r s r. test_clip_value_min_max_flipz%TestClip.test_clip_value_min_max_flip s`
Ib ) ) ):bjD 1 1488D$''VX&&&&&r0 zarr, amin, amax, expr> r l r c R t j ||| }t || d S r ro )r) rq rP rQ rs r s r. test_clip_problem_casesz TestClip.test_clip_problem_cases s- dD))VS!!!!!r0 zarr, amin, amaxc t j t j || | }t j ||| }t || d S r r% rN rO rp r r) rq rP rQ r r s r. test_clip_scalar_nan_propagationz)TestClip.test_clip_scalar_nan_propagation& sJ :bjd 3 3T::dD))VX&&&&&r0 z propagation doesn't match specr NaTzignore::DeprecationWarningc t j t j || | }t j ||| }t || d S r rV rW s r. test_NaT_propagationz TestClip.test_NaT_propagation3 sJ :bjd 3 3T::dD))VX&&&&&r0 )r7 r8 )r rq c t j t j z }| t j d|j \ }}| d }| |t j ||d ddi z }| |t j ||d ddi z }t j ||| } t j
||| }
t j |t j |||
|
} | j
|
k sJ t | | d
S ) a A property-based test using Hypothesis.
This aims for maximum generality: it could in principle generate *any*
valid inputs to np.clip, and in practice generates much more varied
inputs than human testers come up with.
Because many of the inputs have tricky dependencies - compatible dtypes
and mutually-broadcastable shapes - we use `st.data()` strategy draw
values *inside* the test function, from strategies we construct based
on previous values. An alternative would be to define a custom strategy
with `@st.composite`, but until we have duplicated code inline is fine.
That accounts for most of the function; the actual test is just three
lines to calculate and compare actual vs expected results!
r )
num_shapes
base_shapec . t j | d S )NF) allow_nan)hynp
from_dtyper^ s r. r z-TestClip.test_clip_property..c s doa5999 r0 r r` F)r7 r8 elementsr r> N)ra integer_dtypesfloating_dtypesdraw mutually_broadcastable_shapesr8 flatmaparraysr% rp r rN rO r7 r ) r) r rq numeric_dtypes in_shapes result_shaper rP rQ rP r* r s r. test_clip_propertyzTestClip.test_clip_propertyA si . ,..1E1G1GG #')) .
#
#
<
" " 9 9
;
;yyT[~A,+u)= ? ? ? ? @ @yyT[~A,+u)= ? ? ? ? @ @ dD)) N3d + +:dBJsD$B$B$B!LLL|q 68,,,,,r0 r )SrT rU rV r? r rp r r r r r r r rL 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& r) r, r. r0 r2 r4 r6 r8 r; r= r? rA rC rE rG rI rL r% r@ rg rR r fullr rT r& rl r rX r timedelta64r r[ r str ra rj re rf array_shapesrn rX r0 r. r r s2 / / / /3 3 3 3 . . .0 0 0 2 2 22 2 2
[ W&=>>
+
+ ?>
++ + ++ + ++ + +$ $ $&+ + +$
)
)
)+ + + + + + [ Yx(899/ / :9/ + + + + + + + + +) ) )) ) ) " " "+ + ++ + ++ + ++ + ++ + ++ + + + + + + + + + + ++ + +
)
)
)+ + + + + + + + + + + + + + + + + +* * *, , ,) ) )) ) ) 9 9 94 4 4 # # #
6 6 6 [ \
HBHRLL hbhrll#
,
' ' ' [ 3
"BH % % %
XV , , , .
"D ! ! !A %
"D ! ! ! #
6
" "
" [ . "BH % % % "& "BH % % % '1 ' '
' [ >?? [ . 1#($ ' ' ' "BH % % % '1
[ <==' ' >= @?' U
RWYY DK%$%''*>$*>*@*@@#$#%%
(- (- (- (- (-r0 r c R e Zd ZdZdZd Zd Zd Zd Zd Z d Z
d Z d
Z d Z
d Zd
S ) TestAllcloseh㈵>:0yE>c : t j d | _ d S )Nr r r% r olderrr> s r. r? zTestAllclose.setup_methodw s i111 r0 c 2 t j di | j d S r ry r> s r. teardown_methodz TestAllclose.teardown_methodz s
DK r0 c X t t j || |d|d d S N and z
not closer r% allcloser) r^ r s r. tst_allclosezTestAllclose.tst_allclose} s1 Aq!!AAAqqq#ABBBBBr0 c Z t t j || |d|d d S )Nr z shouldn't be closer r s r. tst_not_allclosez TestAllclose.tst_not_allclose s4 BK1%%%AAA'NOOOOOr0 c t j ddg }t j d d }| j }| j }ddgddgf|gdgfdgd|z |z gf||||z z f||||z z |dz z f||||z z ft j t j ft j t j gfg}|D ]\ }}| || d S )Nr } rZ rZ rZ r r r )r% r& rK r atolrtolro r )r) rq aranr r r r^ r s r. test_ip_allclosez TestAllclose.test_ip_allclose s hT{##y~~%%i00yyQ!Q !
qvd{m$cCHn%cCHntAv-.td4i'( "&"$ $ $FQ a # # # # $ $r0 c t j d d }| j }| j }t j dgdt j gft j dgddgft j t j gdt j gft j t j gddgft j dgt j dgft j dgt j dgf|dz gdgfdgd|z |dz z gf||||z z |dz z ft j t j dg t j dt j g fg
}|D ]\ }}| || d S )Nr r r r r ) r% rK r r r ro rl r& r )r) r r r r r^ r s r. test_ip_not_allclosez!TestAllclose.test_ip_not_allclose sP y~~%%i00yy&!q"&k*&!q!f%&"&!Arv;/&"&!Aq6*6'1 {+&!rvqk*q&A3qvd1f}o&td4i'$q&01261+&&!RV(=(=> @ ( (FQ ! !!Q ' ' ' ' ( (r0 c h t j t j dg }t j dt j g }t j || t |t j t j dg t |t j dt j g d S Nr r )r% r& ro r r r s r. test_no_parameter_modificationz+TestAllclose.test_no_parameter_modification s Hbfa[ ! ! Ha[ ! !
Aq1bh{ 3 34441bh26{ 3 344444r0 c t j t j j }t j |gt j }t t j || d S )Nr> )r% r int_r r& r r )r) min_intr= s r. test_min_intzTestAllclose.test_min_int sN (27##' HgYbg . . . Aq!!"""""r0 c t j dt j g }t t j ||d d S )NrS Tr )r% r& rl r r r s r.
test_equalnanzTestAllclose.test_equalnan s< Hc26] # # AqD11122222r0 c G d dt j } |dg }t t t j || t
u d S )Nc e Zd Zd ZdS )6TestAllclose.test_return_class_is_ndarray..Fooc J t j |i | | S r )r% r& r\ )clsr r s r. __new__z>TestAllclose.test_return_class_is_ndarray..Foo.__new__ s% x00055c:::r0 N)rT rU rV r rX r0 r. Foor s#
;
;
;
;
;r0 r r )r% r[ r r] r r9 )r) r r= s r. test_return_class_is_ndarrayz)TestAllclose.test_return_class_is_ndarray sl ; ; ; ; ;"* ; ; ;
CHHR[A&&''4/00000r0 N)rT rU rV r r r? r| r r r r r r r r rX r0 r. rt rt s s D D2 2 2! ! !C C CP P P$ $ $(( ( (*5 5 5# # # 3 3 31 1 1 1 1r0 rt c j e Zd ZdZdZd Zd Zd Zd Zd Z d Z
d Z d
Z d Z
d Zd
Zd Zd Zd ZdS ) TestIscloseru rv c
| j }| j }t j ddg }t j d d }ddgddgf|gdgfdgd|z |z gf||||z z f||||z z |z f||||z z ft j t j ft j t j gft j t j gt j t j gfg | _ t j dgdt j gft j t j gddgft j t j gdt j gft j t j gddgft j dgt j t j gf|dz gdgfdgd|z |dz z gf|||dz |z z |dz z ft j t j dg t j dt j g fg | _ t j dgt j |dz gf|dd dd|z z z |z gdt j d gft j d
g d ft j t j t j t j gfdg|t j t j t j gfd|t j t j t j gfg| _
d d
gg dg dg dg dg dg| _ d S )Nr r r r r r r g?g .Ar )r r g @TF)TFFTTFr )TFFF) r r r% r& rK r ro all_close_testsrl none_close_testssome_close_testssome_close_results)r) r r rq r s r. _setupzTestIsclose._setup s yyhT{##y~~%%i00 Q!Q !
q4x$'(cCHn%cCHnt+,td4i'( "&"&26'"RVbfW$56
&!q"&k*&26'"QF+&"&!Aw<0&"&!Aq6*&!rvw /0q&A3q4x$q&()*td3htm+d3h67261+&&!RV(=(=>
! &!rvtAv ./3AdF
+d23a5EF1 {{{+"&"&"&12tRVbfWbf56T26BF7BF34
! u
$$$###%%%++++++
# r0 c | | j }| j }t || D ]+\ \ }}}t t j || | ,d S r )r r r zipr r% isclose)r) testsresultsr^ r rP s r. test_ip_isclosezTestIsclose.test_ip_isclose si
%) !% 1 1 9 9NFQF rz!Q / / 8 8 8 8 9 9r0 c | t t j t j || |d|d d S r~ )r r% r r r s r. tst_all_isclosezTestIsclose.tst_all_isclose
s; rz!Q''((111aaa*HIIIIIr0 c d}t t j t j || |||fz d S )Nz %s and %s shouldn't be close)r r% rE r )r) r^ r rD s r. tst_none_isclosez TestIsclose.tst_none_isclose
s> ,BF2:a++,,,cQFl;;;;;r0 c ~ d}d}t j | rVt j | rBt t j || t j || k |||fz d S t t j || t j || |||fz d S )Nz4isclose.all() and allclose aren't same for %s and %sz.isclose and allclose aren't same for %s and %srC )r% r r r r r r )r) r^ r rD msg2s r. tst_isclose_allclosez TestIsclose.tst_isclose_allclose
s D?
;q>> Xbk!nn X BJq!$$ Aq(9(99tq!f} M M M M M M rz!Q / / 3 3 5 5r{1a7H7H#QRTUPV, W W W W Wr0 c t | | j D ]\ }}| || d S r )r r r r s r. test_ip_all_isclosez TestIsclose.test_ip_all_isclose
sH
* ' 'FQ A & & & & ' 'r0 c t | | j D ]\ }}| || d S r )r r r r s r. test_ip_none_isclosez TestIsclose.test_ip_none_isclose
sH
+ ( (FQ ! !!Q ' ' ' ' ( (r0 c | | j | j z | j z }|D ]\ }}| || d S r )r r r r r )r) r r^ r s r. test_ip_isclose_allclosez$TestIsclose.test_ip_isclose_allclose
s`
%(==&' , ,FQ % %a + + + + , ,r0 c t t j t j t j d dg t j dt j g }t t j ||d ddg d S )NTr rS )r r% r rl r& r9 s r. test_equal_nanzTestIsclose.test_equal_nan
si 2:bfbfEEEvNNNhRV}%%2:c3$???$ NNNNNr0 c V t j g dt j d }t t | t t j d| u t t | t t j |d u t j g dt j t j t j g }t t | t t j t j | u t t | t t j |t j u t j g dt j t j t j g }t j t j |d }t t | t | u t g d|j
t j |t j d }t t | t | u t g d|j
t j g dt j t j t j g }t j ||d }t t | t | u t g d|j
d S )Nr r r Tr ) r% ma masked_whererK r r] r rl ro r maskr s r. test_masked_arraysz TestIsclose.test_masked_arrays%
sW
E 2 2 2BIaLL A AQ4
1a 0 0111222Q4
1a 0 0111222 E 2 2 2RVRVRV4L M MQ4
261 5 5666777Q4
1bf 5 5666777 E 2 2 2RVRVRV4L M M JrvqD 1 1 1Q477"###...777 Jq"&D 1 1 1Q477"###...777 E 2 2 2RVRVRV4L M M Jq!t , , ,Q477"###...77777r0 c n t t j t j dd d S Nr )r r% r r r> s r. test_scalar_returnz TestIsclose.test_scalar_return@
s, BJq! , ,--.....r0 c h t j t j dg }t j dt j g }t j || t |t j t j dg t |t j dt j g d S r )r% r& ro r r r s r. r z*TestIsclose.test_no_parameter_modificationC
s Hbfa[ ! ! Ha[ ! !
1a1bh{ 3 34441bh26{ 3 344444r0 c t t j t j t j t j u t t j dt j t j u t t t j dt j t j u d S Nr )r r% r ro r( r] r r> s r. test_non_finite_scalarz"TestIsclose.test_non_finite_scalarJ
s
26BF7++ry8999
1bf%%2333RZ26**++rx788888r0 c t j g dgd }t j ||dd sJ t j ||t j dd d sJ t j ||dd sJ t j ||t j dd d sJ d S ) N)r r r rY zm8[ns]r> r T)r r r ns)r% r& r r rp r r s r. test_timedeltazTestIsclose.test_timedeltaQ
s
H&&&'x 8 8 8z!QQ$777;;=====z!QR^At%<%<MMMQQSSSSS{1aa4888888{1abnQ&=&=NNNNNNNNr0 N)rT rU rV r r r r r r r r r r r r r r r r rX r0 r. r r s D D+ + +Z9 9 9J J J< < <X X X' ' '
( ( (
, , ,O O O
8 8 86/ / /5 5 59 9 9O O O O Or0 r c 2 e Zd Zd Zd Zd Zd Zd Zd ZdS )
TestStdVarc J t j g d | _ d| _ d S )N)r rF r rF r )r% r& r* real_varr> s r. r? zTestStdVar.setup_method\
s" ...))
r0 c t t j | j | j t t j | j dz | j d S )Nr )r
r% r r* r r r> s r.
test_basiczTestStdVar.test_basic`
sE BF46NNDM:::BF46NNA -t}=====r0 c t t j d d t t j d d d S r r r% r r r> s r. test_scalarszTestStdVar.test_scalarsd
s8 RVAYY"""RVAYY"""""r0 c x t t j | j d | j t | j z t | j dz
z t t j | j d dz | j t | j z t | j dz
z d S )Nr ddofr r
r% r r* r r r r> s r.
test_ddof1zTestStdVar.test_ddof1h
s BF46 2 2 2 MCKK 73tv;;? K M M MBF46 2 2 2A 5 M#df++ 5TVq I K K K K Kr0 c x t t j | j d | j t | j z t | j dz
z t t j | j d dz | j t | j z t | j dz
z d S )Nr r r r> s r.
test_ddof2zTestStdVar.test_ddof2n
s BF46 2 2 2 MCKK 73tv;;? K M M MBF46 2 2 2A 5 MCKK 73tv;;? K M M M M Mr0 c t j d }t j d }t j || }t ||u t || t j || }t ||u t || t j || }t ||u t || d S )NrD r rL )r% rK r& r r r r r )r) rH rk r{ s r. test_out_scalarzTestStdVar.test_out_scalart
s IbMMhrll F1# S1c""" F1# S1c""" GA3 S1c"""""r0 N) rT rU rV r? r r r r r rX r0 r. r r [
st > > ># # #K K K M M M # # # # #r0 r c e Zd Zd Zd ZdS )TestStdVarComplexc t j g d }d}t t j | | t t j | dz | d S )N)r r rF y r r )r% r& r
r r )r) r* r s r. r z TestStdVarComplex.test_basic
sW H''' ( (BF1IIx000BF1IIqL(33333r0 c t t j d d t t j d d d S )Nr r r r> s r. r z TestStdVarComplex.test_scalars
s8 RVBZZ###RVBZZ#####r0 N)rT rU rV r r rX r0 r. r r
s2 4 4 4 $ $ $ $ $r0 r c e Zd Zd Zd dZd Zd Zd Zd Ze j
e d d
Z
dS ) TestCreationFuncsc d t j t j D }d |D }t ||z
d |D z d | _ ddd| _ d | _ d S )
Nc 6 h | ]}t j | S rX )r% r7 r] tps r. z1TestCreationFuncs.setup_method..
s OOO2"(2,,OOOr0 c F h | ] }|j d | S )r^ )r endswithr s r. r z1TestCreationFuncs.setup_method..
s+ FFF 1E1EF"FFFr0 c h | ]O}t d d D ]<}t j |j dt | =PS )r rD r^ )rF r% r7 r replace)r] r r s r. r z1TestCreationFuncs.setup_method..
sm N N N"$a N N;< !hrv~~c3q66'B'B C C N N N Nr0 c | j S r )r r> s r. r z0TestCreationFuncs.setup_method..
s uy r0 )keyr r )r FrD ) r chainr% r valuessorteddtypesordersndims)r) r variable_sizeds r. r? z TestCreationFuncs.setup_method
s OO "*:K:K:M:M)NOOOFFvFFF Vn 4 N N(6 N N N N "9!8: : : + @@
r0 Nc
dt | j | j | j f}i }|d|i}t j | D ]\ }}}}||gz } |r|j d r* || f||d|}
t |
j | t t |
j | j | |O|j d rt
| } n|} t |
|
| d S )Nrc
fill_valuez|Vorderr7 z|S)rF r r r r r r
startswithr r7 r getattrr r] ) r) rH r par
fill_kwargr r r r7 r8 rq r s r. check_functionz TestCreationFuncs.check_function
s< TZ {{
!&
3J)2):C)@ 3 3 %D%TFNE
ei 2 24 8 8
$u %E % %#% %C
E * * * GCIt{5'9:: ; ; ;%9''-- % j//CC$C S%**S//222% 3 3r0 c D | t j d S r )r r% r@ r> s r.
test_zerosz TestCreationFuncs.test_zeros
BH%%%%%r0 c D | t j d S r )r r% rg r> s r. test_oneszTestCreationFuncs.test_ones
s BG$$$$$r0 c D | t j d S r )r r% emptyr> s r.
test_emptyz TestCreationFuncs.test_empty
r r0 c | t j d | t j d d S )Nr r )r r% ro r> s r. test_fullzTestCreationFuncs.test_full
s8 BGQ''' BGQ'''''r0 zPython lacks refcountsr c d}t j | }t j |gdz t t j | |k t j |gdz t t j | |k t j |gdz t t j | |k t j |gdz d t t j | |k d S )Nr rD r )r getrefcountr% r@ r rg r ro )r) dimbegs r. test_for_reference_leakz)TestCreationFuncs.test_for_reference_leak
s oc""
#r$$+,,,
b$$+,,,
#r$$+,,,
b! $$+,,,,,r0 r )rT rU rV r? r r r r r rL r
r r r rX r0 r. r r
s 3 3 3 3:& & &% % %& & &( ( ( [L(1IJJ - - KJ - - -r0 r c e Zd ZdZd Zd ZddZd Zd Zd Z d Z
e j
d
ej ej ej ej g e j
d eeg d Zd
S )
TestLikeFuncsz4Test ones_like, zeros_like, empty_like and full_likec f t j d d ft j d dft j dd d ft j d dft j d dd d ft j d dd d ft j d d
d d ft j d d
d d ft j d ddd d ft j d ddd dft j d dd d ft j d dd dft j d ddd dd d ft j d ddd dd dfg| _ g d| _ d S )Ng @r r rf r r> r r r: r r )r r r r r r" )r r r r r r )rX rZ )rZ rf re )r% r& rK r r r shapesr> s r. r? zTestLikeFuncs.setup_method
s " t$!d#1D)))401 u%1 %%a++T21 %%a++T21 %%fC%88$?1 %%fC%88$?2 &&q!Q//62 &&q!Q//62 &&y&<>E2 &&q!Q//88A>>D+ . 322 r0 c |||rSt j | |j }t t j ||k d S t t j ||k d S d S r )r% r& rv r7 r r )r) dzvaluer zs r. compare_array_valuez!TestLikeFuncs.compare_array_value
sy
- HUOO**2844rQw(((((rU{++,,,,, r0 Fc
|rd|i}ni }| j D ]\ }} ||fd|i|}t |j |j t t j |j |j j z t j |j |j j z t |j j |j j t |j j
|j j
|t |j |j n't |j t j | | ||| ||fd|d|}t |j |j t |j j |t |j |j n't |j t j | | ||| ||fd|d|}t |j |j t |j j
|t |j |j n't |j t j | | ||| ||fd|d|}t |j |j |j j
rt |j j
nt |j j |t |j |j n't |j t j | | ||| | j
D ]}dD ]} ||f||| d|}
t |
j | |t |
j |j n't |
j t j | | dk s| dk r&|j j rt |
j j n1| dk s| dk r%|j j
rt |
j j
| |
|| |j t | k r`t t j ||f||d d|j t j t j ||d
j `t t j ||f||d d|j t j |j G d d t j } t j d
dgddgg | } || fi |}
t t) |
| u || fddi|}
t t) |
| u d S )Nr r7 r r r r* CFA)r7 r8 r K)r7 r c e Zd ZdS )4TestLikeFuncs.check_like_function..MyNDArrayNr rX r0 r. MyNDArrayr N r r0 r! r r r r subokF)r r r8 r% r& stridesr7 r r r r r r r ndimr argsortr r[ r\ r] )r)
like_functionr r r rH r7 r r r: szr! r= r s r. check_like_functionz!TestLikeFuncs.check_like_function
s &.JJJ I 8 I 8HAu q<<<<>!),,RX->>
@
@
@ -rx/D E E E -rx/D E E E} RXqw//// RXrx777 $ $R
; ; ; qG5GGJGGB 17 + + + BH) * * *} RXqw//// RXrx777 $ $R
; ; ; qG5GGJGGB 17 + + + BH) * * *} RXqw//// RXrx777 $ $R
; ; ; qG5GGJGGB 17 + + +w#
/-....-...} RXqw//// RXrx777 $ $R
; ; ; [
8
8 D DA& q 5Qa 5 5)35 5B 1--- }$RXqw7777$RXrx???CxxAHH1EH 56666ca3hh173Gh 5666,,R
CCCCFc!ff$$ MM! -H5@A-H -H-@ -@ -@@G"I "IJ J J J !MM! -H5@A-H -H s r. test_ones_likez TestLikeFuncs.test_ones_likeY s q11111r0 c F | t j d d S r )r( r% ru r> s r. test_zeros_likez TestLikeFuncs.test_zeros_like\ s 22222r0 c F | t j d d S r )r( r% r: r> s r. test_empty_likez TestLikeFuncs.test_empty_like_ s 55555r0 c | t j dd | t j dd | t j dd | t j dd t j d 5 | t j t j d d d d d S # 1 swxY w Y d S )Nr Tr r gw/^@r rx )r( r% full_liker ro r> s r. test_filled_likez TestLikeFuncs.test_filled_likeb s q$777 q$777 tT::: w===
[
*
*
* A A $ $R\264 @ @ @ A A A A A A A A A A A A A A A A A As ,CCClikefuncr7 c t j d dd }|d d d d df }|t j k rddini } ||fd|i|}|t k r|j dk sJ d S |j dk sJ d S ) Nrc r rv r r r7 )rc r )r r )r% rK r r1 r r# )r) r3 r7 r= r r rP s r. test_dtype_str_bytesz"TestLikeFuncs.test_dtype_str_bytesk s
IbMM ! !!Q ' '
aaa1fI'/2<'?'?,##R!3353F33 C<<>W,,,,,, >V++++++r0 N)F)rT rU rV __doc__r? r r( r+ r- r/ r2 rL r
r r% r: r1 ru r* r bytesr5 rX r0 r. r r
s >>3 3 34 - - -Z* Z* Z* Z*x2 2 23 3 36 6 6A A A [ Z"- *,- *G H H [ WsEl33
,
, 43H H
,
,
,r0 r c 8 e Zd Zd Zd Zd Zd Zd Zd Zd Z dS )
TestCorrelatec t j g d| | _ t j dd d d d | _ t j g d| | _ t j g d| | _ t j g d| | _ t j g d | | _ t j g d
| | _ t j g d | | _
t j g d | | _ d S )
Nr r> r r r )rF r r ) , 4 :r= )g r@ r< g &r= r@ ) . 6rB 0 $ rf )r@ r= r? r> r= r< r; )rf rE rD rC rB rB rA ) r; r= g >g Hg Pg Ug Yg Kg 3) r% r& r^ rK xsr z1z1_4z1rz2z2rzs)r) r s r. r zTestCorrelate._setup| s ///444)Ar""33Q3',,,b111(BBB"MMMH = = =RHHH 8 D D DBOOO(BBB"MMM8 D D DBOOO( . . .579 9 9r0 c | t t j | j | j d }t
|| j t j | j | j d d d }t
|| j t j | j | j d }t
|| j t j | j d d d | j d }t
|| j
t j | j | j d d d d }t
|| j t j | j | j d }t
|| j
d S )Nro rF )r r r% correlater^ r r rG rH rJ rI rK rF rL r) r s r. r~ zTestCorrelate.test_float s, E L 0 0!!TW--- Lf 5 5!!TY/// L 0 0!!TW--- L" tvv 6 6!!TX... L" v 6 6!!TX... L$&& 1 1!!TW-----r0 c | t t j | j | j d }t
|| j t j | j | j d }t
|| j d S )Nro ) r r r% rN r^ r r rG rJ rO s r. test_objectzTestCorrelate.test_object sl G L 0 0!!TW--- L 0 0!!TW-----r0 c t j d }t j d }t j || t |t j d t |t j d d S Nr r )r% rg rN r r) rH ks r. test_no_overwritez TestCorrelate.test_no_overwrite sd GCLL GAJJ
Q1bgcll+++1bgajj)))))r0 c < t j g dt }t j g dt }t j g dt }|d d d }t j ||d }t || d S )N)r r r y @ ?r> )rF y y @ ?)y @ rf y @ ?y &@ @y @y rF ro mode)r% r& r conjugaterN r )r) r^ r r_zr s r. test_complexzTestCorrelate.test_complex s H___G 4 4 4 H___G 4 4 4h;;;7KKK$$B$i!!## LAF + + +!!S)))))r0 c t j t 5 t j t j g t j d d d d d n # 1 swxY w Y t j t 5 t j t j d t j g d d d d d S # 1 swxY w Y d S )Nr ro rX )rL rM rN r% rN r& rg r> s r. test_zero_sizez TestCorrelate.test_zero_size sN
]:
&
& C C L" rwt}}6 B B B B C C C C C C C C C C C C C C C
]:
&
& C C L 6 B B B B C C C C C C C C C C C C C C C C C Cs# >JJJJJr0 c t j d }t j d }t j || t |t j d t |t j d d S rS )r% rg rj r rT s r. rV z TestConvolve.test_no_overwrite sd GCLL GAJJ
Aq1bgcll+++1bgajj)))))r0 c t j d }t j d }t j ||d }t t 5 t j ||d }d d d n # 1 swxY w Y t || t
t 5 t j ||d d d d n # 1 swxY w Y t t j ||d | t
t 5 t j ||d d d d d S # 1 swxY w Y d S )Nr r ro rX r rF r ) r% rg rj r r r r
rN r$ )r) rH rU rb full_modes r. rd zTestConvolve.test_mode s GCLL GAJJ{1af555
,
-
- 4 4 Aqs333I 4 4 4 4 4 4 4 4 4 4 4 4 4 4 49l333
:
&
& ' ' K12 & & & & ' ' ' ' ' ' ' ' ' ' ' ' ' ' '2;q!!444i@@@
9
%
% ) ) K14 ( ( ( ( ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) )re N)rT rU rV rQ rV rd rX r0 r. rg rg sD K K K
* * *
)
)
)
)
)r0 rg c f e Zd Zej dg d d Zd Zd ZdS ) TestArgwhererk rc c t j d|z t }d|d< t t j | j d|f d|d< d|j d< t t j | j d|f d|d< d|j d< t t j | j |j dz
|f d|d< t t j | j |j |f d S )Nr[ F.r Tr )r% r r9 r argwherer8 flatr )r) rk r^ s r. test_ndzTestArgwhere.test_nd s
HT"Wd # # #R[^^)Ar7333 #q R[^^)Ar7333 #q R[^^)AFQJ+;<<< #R[^^)AFB<88888r0 c t j d d }t t j |dk ddgddgddgddgg d S )Nrf r r r r )r% rK r r rq r s r. test_2DzTestArgwhere.test_2D sj IaLL ( (2;q1u-- F F F F $ % % % % %r0 c ^ t t j g d dgdgdgdgg d S )N)r r r r r r r r r )r r% rq r> s r. test_listzTestArgwhere.test_list s7 R[ 11QC!qcA33GHHHHHr0 N) rT rU rV rL r
r rs ru rw rX r0 r. ro ro sg [ T999--9 9 .-9,% % %I I I I Ir0 ro c e Zd Zd ZdS )TestStringFunctionc t j dg }t j d d t t | d t j d d t t | d t j d d t t | d t j d d t t | d d S )
Nr c dS NFOOrX rc s r. r z=TestStringFunction.test_set_string_function.. r0 T)reprr} z
array([1])c dS r| rX rc s r. r z=TestStringFunction.test_set_string_function.. r~ r0 Fz[1])r% r& set_string_functionr r r r s r. test_set_string_functionz+TestStringFunction.test_set_string_function s HaSMM
T::::T!WWe$$$
t$////T!WWl+++
U;;;;SVVU###
t%0000SVVU#####r0 N)rT rU rV r rX r0 r. ry ry s#
$
$
$
$
$r0 ry c e Zd Zd Zd Zd ZdS )TestRollc t j d }t j |d }t |t j g d d S )NrD r )
rv rw r r r r r rZ rf rg )r% rK rollr r& )r) r^ xrs r. test_roll1dzTestRoll.test_roll1d sE IbMM
WQ]]R"@"@"@AABBBBBr0 c 4 t j t j d d }t j |d }t |t j g dg dg t j |dd }t |t j g dg d g t j |dd }t |t j g d
g d g t j |dd }t |t j g d g d
g t j |d
d }t |t j g dg d g t j |dd }t |t j g dg d g t j |d d }t |t j g d
g d g t j |dd }t |t j g dg dg t j |dd }t |t j g d g d
g t j |dd }t |t j g dg dg t j |dd }t |t j g d g dg t j |dd }t |t j g dg dg t j |dd }t |t j g d
g d g t j |dd }t |t j g d
g d g d S )NrD r rZ r )rw r r r r )r rZ rf rg rv r rx ru rt )r r r r r )rw rZ rf rg rv r r )rF r )r rF )r r r r r )rf rg rv rw rZ r )rF rF r )r r r r r )rv rw rZ rf rg rf rn )r% r rK r r r& )r) x2x2rs r. test_roll2dzTestRoll.test_roll2d sv
Z "
v
.
.gb!nnS"(OOO___#EFFGGGgb!!$$$S"(OOO___#EFFGGGgb!!$$$S"(OOO___#EFFGGG gb!&)))S"(OOO___#EFFGGGgb&v...S"(OOO___#EFFGGGgb'///S"(OOO___#EFFGGGgb&v...S"(OOO___#EFFGGGgb'///S"(OOO___#EFFGGGgb&v...S"(OOO___#EFFGGGgb(000S"(OOO___#EFFGGG gb!&)))S"(OOO___#EFFGGGgb!&)))S"(OOO___#EFFGGG gb!!$$$S"(OOO___#EFFGGGgb"1%%%S"(OOO___#EFFGGGGGr0 c t j g }t t j |d t j g d S r )r% r& r r r s r. test_roll_emptyzTestRoll.test_roll_emptyI s7 HRLLRWQ]]BHRLL11111r0 N)rT rU rV r r r rX r0 r. r r sH C C C
-H -H -H^2 2 2 2 2r0 r c e Zd Zi dddddddddd d
dd dd dd
ddddddddddddddddddddd Zd Zd Zd S ) TestRollaxisr r r r5 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 r r r r6 )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 r c t j d dddd }t t j t j |dd t t j t j |dd t t j t j |dd t t j t j |dd d S ) Nr r r r r r r rZ )r% rK r r
r rollaxisr s r. test_exceptionsz TestRollaxis.test_exceptions_ s Ig & &q!Q 2 2blBKB:::blBKAr:::blBKAq999blBKAq99999r0 c
t j d dddd }t j |j }t
|j d | j D ]\ }}t j ||| }|t j
|j dz
\ }}}} t
t j ||||| f |k t
|j | j ||f k t ||f t
|j d |dz }
t j ||
| }|t j
|j dz
\ }}}} t
t j ||||| f |k t
|j | j d|
z
|f k t
|j d |dk r|dz n|} t j ||| }|t j
|j dz
\ }}}} t
t j ||||| f |k t
|j | j |d| z
f k t
|j d |dz }
|dk r|dz n|} t j ||
| }|t j
|j dz
\ }}}} t
t j ||||| f |k t
|j | j d|
z
d| z
f k t
|j d d S )Nr r r r r OWNDATA)ry start)
r% rK r r; r r8 r r tgtshaper r& r r ) r) r= aindr jr\ i0r: r; i3ipjps r. test_resultszTestRollaxis.test_resultsf sL Ig & &q!Q 2 2 7 7 9 9z!'"" "###m . .FQ+aaq111C !"(39"5"5"9 :NBB BF3r2r2~.!344 5 5 5 CI 1v!66QqE
C C C ),, - - - QB+arc333C !"(39"5"5"9 :NBB BF3r2r2~.!344 5 5 5 CI B{!;; < < < ),, - - - a%%QQB+aas333C !"(39"5"5"9 :NBB BF3r2r2~.!344 5 5 5 CI 1r6{!;; < < < ),, - - - QBa%%QQB+arc"555C !"(39"5"5"9 :NBB BF3r2r2~.!344 5 5 5 CI BB/?!@@ A A A ),, - - - -? . .r0 N)rT rU rV r r r rX r0 r. r r N s & &fl & &&,l & & & '-l & & '-l &
&
& '-l
& & '-l & & & '-l & %l$ & & &H: : :#. #. #. #. #.r0 r c 2 e Zd Zd Zd Zd Zd Zd Zd ZdS ) TestMoveaxisc t j ddd }dD ]0\ }}t j ||d j }t || 1d S )NrZ rf rg ))r )rf rg rZ )r )rZ rg rf )r re )rF re rF r% r r moveaxisr8 r )r) r^ sourcer r s r. test_move_to_endz TestMoveaxis.test_move_to_end sd IOOAq! $ $!2 & & FH [FB//5F FH % % % % & &r0 c t j dddd }dD ]1\ }}}t j ||| j }t || 2d S )Nr r r r ))r r r )r r r )r rF r r r) r^ r destinationr r s r. test_move_new_positionz#TestMoveaxis.test_move_new_position sh IOOAq!Q ' '. & & )FK
[FK88>F FH % % % %
& &r0 c t j d }dddddgddgfddgddgft d t d ffD ]0\ }}t j ||| j }t |d 1d S ) Nr r )r rF )rF r r rF r r )r% r@ rF r r8 r )r) r^ r r r s r. test_preserve_orderz TestMoveaxis.test_preserve_order s H\ " "R1b'"Q!Q q588$
$ * * FK [FK88>F FL ) ) ) ) * *r0 c t j d }ddgddgdfddgddgdfg dg ddfddgddgd fddgddgd ffD ]1\ }}}t j ||| j }t || 2d S )
Nr r r r r )r r r r rc )r r r )r r r r )r% r@ r r8 r r s r. test_move_multiplesz TestMoveaxis.test_move_multiples s H\ " "Q!Q .Q!Q .III|4Q!Q .Q!Q . . & & )FK [FK88>F FH % % % % & &r0 c j t j ddd }t t j dt j |dd t t j dt j |dd t t j dt j |dd t t d t j |ddgddg t t d
t j |ddgddg t t d t j |dddg t t d t j |ddgdg d S ) Nr r r zsource.*out of boundsr rn zdestination.*out of boundsrZ zrepeated axis in `source`z repeated axis in `destination`zmust have the same number)r% r r r r r rN r s r. test_errorszTestMoveaxis.test_errors s1 IOOAq! $ $BL*A KAq 2 2 2BL*A KB 3 3 3BL*F KAq 2 2 2J(C KQFQF < < <J(H KQFQF < < <J(C KA1v 7 7 7J(C KQFQC 9 9 9 9 9r0 c t j d }t j |dd }t |j |j t t
|t j j g d}t j |dd }t |t | t t
|t j d S )Nr! r )
r% r r@ r r r8 r MaskedArrayrG r[ )r) r^ rP s r. test_array_likesz TestMoveaxis.test_array_likes s EKK " "Q1%% &&&
625#455666 IIQ1%%4<<
62:../////r0 N) rT rU rV r r r r r r rX r0 r. r r sn & & && & & * * *
&
&
&9 9 9" 0 0 0 0 0r0 r c 2 e Zd Zd Zd Zd Zd Zd Zd ZdS ) TestCrossc ddg}ddg}d}t j || }t || t j || }t || d S )Nr r r r r )r% crossr r) ura r cps r. test_2x2zTestCross.test_2x2 s_
F
F
Xa^^R
Xa^^R! r0 c ddg}g d}t j g d }t j || }t || t j || }t || d S )Nr r rd rD r r r% r& r r r s r. test_2x3zTestCross.test_2x3 sm
F II H\\\ " "
Xa^^R
Xa^^R! r0 c g d}g d}t j g d }t j || }t || t j || }t || d S )Nr! r r rf r r r s r. test_3x3zTestCross.test_3x3 sm II II H[[[ ! !
Xa^^R
Xa^^R! r0 c t j ddgd }t j ddgd }d}t t j || | t t j || | t t j || d t j ddgd j }t j g dd }t j g d d }t t j ||d
| t t j ||j | t t j || d t j g d d j }t j ddgd j }t j g d d }t t j ||dd
| t t j |j |j | t t j |j |j d t j g d d }t j g dd j }t j g dd }t t j ||d | t t j |j | | t t j || d d S )Nr r )r r r r r r rd r )axisar! )r rw r r axisb)rZ r r r )r )r% tiler r r r) r ra r s r. test_broadcastingzTestCross.test_broadcasting sk GQFG $ $ GQFG $ $ RXa^^Q'''RXa^^aR(((RXa^^Q''' GQFG $ $ & GIIIw ' ' GLLL' * *RXa!,,,a000RXa%%r***RXa^^Q''' GIIIw ' ' ) GQFG $ $ & GLLL' * *RXa!1555q999RXac13''!,,,RXac13''+++ GIIIv & & GIIIv & & ( GKKK ( (RXa!,,,a000RXac1%%r***RXa^^Q'''''r0 c t j d }t j d }t t j || j d t j d }t j d }t t j ||dd j d t t j t j ||dd
t t j t j ||d d t j d }t j d
}t t j ||dd
j d t t j t j ||dd
t t j t j ||dd t j d }t dd
D ],}t t j ||| j d -d S )N)r r r )rZ r )r rZ r )rD r rZ r r r r )rD rZ r r r )rD r rZ rg )rZ rg r )r axisc)rD rZ r rg r rn )r r r r )r r )r% rg r r r8 r
r rF )r) r ra r s r. test_broadcasting_shapesz"TestCross.test_broadcasting_shapes
s GI GFOORXa^^)9555 GJ GFOORXa!1555;ZHHHblBHa!1EEEEblBHa!1EEEE GM " " GI RXa!1555;]KKKblBHa"AFFFFblBHa!2FFFF GI 2q\\ D DE !Qe444:F C C C C D Dr0 c ` t j g dgt j }t j g dt j }t j g dgt j }t t j || | t t j || | d S )N) rv rw ) D )i i+ i^r> )r% r& r r r r r s r. test_uint8_int32_mixed_dtypesz'TestCross.test_uint8_int32_mixed_dtypes#
s Hkkk]BH - - H^^^RX . . H***+28 < < <RXa^^Q'''RXa^^aR(((((r0 N) rT rU rV r r r r r r rX r0 r. r r sq ( ( (<D D D&) ) ) ) )r0 r c j t j d } t j d }t j ddd }t j d }t j d }t j | || }t || t t j ||| | d S ) Nr r[ r r rZ )rZ rZ )r8 r )r% rg linspacer[ outerr )arr1arr2arr3out1out2res1s r. test_outer_out_paramr ,
s
74==D
74==D
;r1a D
:E " " "D
:F # # #D
8D$ % %Dt $d++T22222r0 c e Zd Zd Zd Zd Zd Zej de
j e
j e
j
e
j g ej dg d d Zd S )
TestIndicesc
t j d \ }}t |t j g dg dg dg dg t |t j g dg dg dg dg d S )Nr# r r r r rc r% r r r& r s r. test_simplezTestIndices.test_simple9
s F##A1bh (1 (1 (1 (3 4 4 5 5 5 1bh (1 (1 (1 (3 4 4 5 5 5 5 5r0 c t j d \ }t |t j g d t j dd \ }t |t j g d d S )N)r r Tsparser r s r. test_single_inputz TestIndices.test_single_inputD
sk j 1bh||| 4 4555jd+++1bh||| 4 455555r0 c " t g t j d t g t j dd t g gt j d t g gt j dd d S )NrX Tr r4 )r r% r r> s r. test_scalar_inputz TestIndices.test_scalar_inputK
s| 2rz"~~...2rz"T : : :;;;B4D!1!1222B4D!>!>!>?????r0 c t j dd \ }}t |t j dgdgdgdgg t |t j g dg d S ) Nr# Tr r r r r rc r r s r. r zTestIndices.test_sparseQ
sk E$///A1bhaS1#s'; < <===1bh { 3 344444r0 r7 dims)rX r4 r# c t j || }t |j |k t j ||d D ]}t |j |k d S )Nr> T)r7 r )r% r r r7 )r) r7 r indsrq s r. r z TestIndices.test_return_typeV
so z$e,,,
e#$$$:d%=== ( (C CI& ' ' ' ' ( (r0 N)rT rU rV r r r r rL r
r r% r r rA r r rX r0 r. r r 7
s 5 5 56 6 6@ @ @ 5 5 5
[ Wrx2:rz&RSS [ V%7%7%788( ( 98 TS( ( (r0 r c F e Zd Zg dZd Zd Zd Zd Zd Zd Z d Z
d Z d
S ) TestRequire) r r
CONTIGUOUSr r FORTRANr* ALIGNEDW WRITEABLErB r c t j ddd|fg }| d |d }t |j d t |j d t |j d t |j d t |j d
|S ) Nr )junkr: r= F)writer r rB r r* )r% r@ setflagsr r )r) r7 rq r= s r. generate_all_falsez TestRequire.generate_all_falseg
s hve =>> 5 !!! HAGCL !!!AGCL !!!AGCL !!!AGCL !!!AGCL !!!r0 c J ||j }t j |||g }t |j | t |j |k t j |d |g }|d dk rt ||u d S t |j | d S )Nr rB )r7 r% requirer r )r) flagr7 rq r r s r. set_and_check_flagz TestRequire.set_and_check_flagr
s =IE JsED6 * *
5 !!!
Jq$ ' ' 7c>> AFOOOOO AGDM " " " " "r0 c ddg}g d}t j ||| j D ]2\ }}}| | }| ||| 3d S )Nr r= )Nr r )r r
flag_namesr r )r) idfdidtypefdtyper r= s r. test_require_eachz TestRequire.test_require_each
su D\
$-$5b"do$N$N 6 6 FFD''//A # #D&1 5 5 5 5 6 6r0 c t | d }t t t j |d d d S )Nr Q)r r
KeyErrorr% r r s r. test_unknown_requirementz$TestRequire.test_unknown_requirement
s3 # #D ) )h
AtS99999r0 c * t j g ddg d }t |j d t |j d t |j d t |j dk t |g d d S )Nr r= )r r* rB rB r r* )r% r r r r7 r r s r. test_non_array_inputz TestRequire.test_non_array_input
s J|||T??? ; ; 4 Q %%%%%r0 c x | d }t t t j |d ddg d S )Nr r r )r r
rN r% r r s r. test_C_and_F_simulz TestRequire.test_C_and_F_simul
s7 # #D ) )j"*aSzBBBBBr0 c G d dt j } |d }t j |d dg }t t | t j u d S )Nc e Zd ZdS )4TestRequire.test_ensure_array..ArraySubclassNr rX r0 r.
ArraySubclassr
r r0 r r E)r% r[ r r r] )r) r r= r s r. test_ensure_arrayz TestRequire.test_ensure_array
su BJ
M& ! ! Jq$ & &Q2:%&&&&&r0 c G d dt j }| j D ]$} |d }| |d | %d S )Nc e Zd ZdS )8TestRequire.test_preserve_subtype..ArraySubclassNr rX r0 r. r r
r r0 r r )r% r[ r r )r) r r r= s r. test_preserve_subtypez!TestRequire.test_preserve_subtype
su BJ O 3 3D
f%%A # #D$ 2 2 2 2 3 3r0 N) rT rU rV r r r r r
r r r r rX r0 r. r r `
s " " "J
#
#
# 6 6 6: : :& & &C C C' ' '3 3 3 3 3r0 r c , e Zd Zd Zd Zd Zd Zd ZdS )
TestBroadcastc t j d t j d t j d t j d g}t j | t j t j |d d t j |dd t j t j |d d t j |dd t j t j |d d t j |dd t j |d t j |dd |d g}|D ]}t |j d t |j d
t |j d
t |j d t ||j D ]\ }}t ||j u d S ) Nr )rZ rf r )rg )rZ r rg r r r rF re r r ) r% r broadcastr r8 r$ rk numiterr r@ r base)r) arrsmitsmitr= ias r. test_broadcast_in_argsz$TestBroadcast.test_broadcast_in_args
s "(9"5"5rx~~##% d# R\48 4blDH6MNN R\48 4blDH6MNN R\48 4blDH6MNN T!WblD2J&?bJJ L
& &C I . . . 1 % % % # # # a ( ( ( T39--
&
&2RW %%%%
& & &r0 c B t j d g}t j | }t |j d t |j d t |j d t |j d t |d |j d j
u d S )Nre r r r ) r% r r r r8 r$ rk r r r@ r! )r) r" r$ s r. test_broadcast_single_argz'TestBroadcast.test_broadcast_single_arg
s ##$lD!SY ***SXq!!!SVQ S[!$$$Q39Q<,,-----r0 c t j d }t d D ]O}|g|z }|dk r t t t j g|R ,t j | }t
|j | Pd S )Nr # rY )r% r rF r
rN r r r )r) rq r r" r$ s r. test_number_of_argumentsz&TestBroadcast.test_number_of_arguments
s htnnr - -A519D2vv j",>>>>>>lD) S[!,,,,
- -r0 c t j d g}t j | }t j |i i }t |j |j t |j |j t |j |j t |j |j t |j d j
|j d j
u t t t j dfi ddi d S )Nre r r r^ )
r% r r r r8 r$ rk r r r@ r! r
rN )r) r" r$ mit2s r. test_broadcast_error_kwargsz)TestBroadcast.test_broadcast_error_kwargs
s ##$|T"|T(R((SY
+++SXty)))SVTW%%%S[$,/// ! !TZ]%77888j",>>c1X>>>>>r0 c t j t d 5 t j g dgdgdggddg d d d d S # 1 swxY w Y d S )Nz5arg 0 with shape \(1, 3\) and arg 2 with shape \(2,\)rH r! r rZ rf rg )rL rM rN r% r r> s r. !test_shape_mismatch_error_messagez/TestBroadcast.test_shape_mismatch_error_message
s
]: .H I I I : : L)))sQCj1a& 9 9 9 : : : : : : : : : : : : : : : : : :s A A
A
N)rT rU rV r& r( r+ r. r0 rX r0 r. r r
s_ & & &". . .- - - ? ? ?: : : : :r0 r c : e Zd Z G d dej Zd ZdS ) TestKeepdimsc e Zd ZddZdS )TestKeepdims.sub_arrayNc J t j | |||d S )NTr )r% r[ r )r) ry r7 rk s r. r zTestKeepdims.sub_array.sum
s :>>$eS4>HH Hr0 )NNN)rT rU rV r rX r0 r. sub_arrayr4
s. I I I I I Ir0 r6 c | j }t j d | }t t
t j |d d S )N Tr )r6 r% rK r\ r
r$ r )r) sub_classr^ s r.
test_raisezTestKeepdims.test_raise
sC N IbMM y ) )iT::::::r0 N)rT rU rV r% r[ r6 r: rX r0 r. r2 r2
sW I I I I IBJ I I I; ; ; ; ;r0 r2 c e Zd Zd Zd ZdS )
TestTensordotc t j d }t j d }t j ||d }t |t j || t |t j d|| d S )Nr r r zij,jk)r% r[ tensordotr doteinsum)r) r= r tds r. test_zero_dimensionz!TestTensordot.test_zero_dimension
sq Ju Ju
\!Q
'
'2rva||,,,2ry!Q 7 788888r0 c ~ t j d }t j ||g g f }t || d S r )r% r& r> r )r) arr_0drets r. test_zero_dimensionalz#TestTensordot.test_zero_dimensional
s< !l66B8443'''''r0 N)rT rU rV rB rF rX r0 r. r< r<
s2 9 9 9( ( ( ( (r0 r< )Kr r r r rL r decimalr r r%
numpy.corer numpy.randomr r r
numpy.testingr r r
r r r
r r r r r numpy.core._rational_testsr
hypothesisr r rq hypothesis.extrara r ra r r$ r5 rV r r r Exceptionr r r rT r\ rz r r r rt r r r r r r9 rg ro ry r r r r r r r r r2 r< rX r0 r. rO s
- - - - - - - - - -
0 / / / / / . . . . . . . . * * * * * *95 95 95 95 95 95 95 95xX- X- X- X- X- X- X- X-v
'
'
'
'
'
'
'
' BT5 T5 T5 T5 T5 T5 T5 T5naH aH aH aH aH aH aH aHHP$ P$ P$ P$ P$ P$ P$ P$fe; e; e; e; e; e; e; e;Px- x- x- x- x- x- x- x-x
i< i< i< i< i< i< i< i