yz3TestIndexing.test_index_no_floats.. s !C& c d S )N)r r r r s r r z3TestIndexing.test_index_no_floats.. !F) r c d S )N)r r r r s r r z3TestIndexing.test_index_no_floats.. r r c dd d f S r r r s r r z3TestIndexing.test_index_no_floats.. s !CE( r c d d df S r r r s r r z3TestIndexing.test_index_no_floats.. s !AAAsF) r c " d d dd d f S r r r s r r z3TestIndexing.test_index_no_floats.. s !AAAs111H+ r c " dd d d d f S r r r s r r z3TestIndexing.test_index_no_floats.. s !C!!!G* r c d S )N)r r r r r s r r z3TestIndexing.test_index_no_floats.. !I, r c d S )N)r r r r r s r r z3TestIndexing.test_index_no_floats.. r" r c d S )N)r r r r r s r r z3TestIndexing.test_index_no_floats.. r" r c d S Nffffffr r s r r z3TestIndexing.test_index_no_floats.. s !D' r c d S )N)r r' r r s r r z3TestIndexing.test_index_no_floats.. !G* r c d S )N)r' r r r s r r z3TestIndexing.test_index_no_floats..! r) r c dd d f S r& r r s r r z3TestIndexing.test_index_no_floats.." s !DF) r c d d df S r& r r s r r z3TestIndexing.test_index_no_floats..# s !AAAtG* r c " d d dd d f S r& r r s r r z3TestIndexing.test_index_no_floats..$ s !AAAtAAAI, r c " dd d d d f S r& r r s r r z3TestIndexing.test_index_no_floats..% s !D111H+ r c d S )N)r r r' r r s r r z3TestIndexing.test_index_no_floats..& !J- r c d S )N)r' r r r r s r r z3TestIndexing.test_index_no_floats..' r0 r c d S )N)r r' r r r s r r z3TestIndexing.test_index_no_floats..( r0 r c dd df S r r r s r r z3TestIndexing.test_index_no_floats..) s !CDD#I, r c " dd dd d f S r r r s r r z3TestIndexing.test_index_no_floats..* s !CDD#aaaK. r )nparrayr
IndexErrorselfr s @r test_index_no_floatsz!TestIndexing.test_index_no_floats sR HseW j....111j"3"3"3"3444j"3"3"3"3444j"2"2"2"2333j"3"3"3"3444j"5"5"5"5666j"4"4"4"4555j"6"6"6"6777j"6"6"6"6777j"6"6"6"6777j////222j"4"4"4"4555j"4"4"4"4555j"3"3"3"3444j"4"4"4"4555j"6"6"6"6777j"5"5"5"5666j"7"7"7"7888j"7"7"7"7888j"7"7"7"7888j"6"6"6"6777j"8"8"8"899999r c T t j dgg t t fd t t fd t t fd t t fd t t fd t t fd t t fd t t fd t t fd
t t fd t t fd t t fd
t t fd t t fd t t fd t t fd t t fd t t fd t t fd t t fd t t fd t t fd d S )Nr c dd S r r r s r r z5TestIndexing.test_slicing_no_floats..0 s 344 r c dd ddf S )Nr r r r s r r z5TestIndexing.test_slicing_no_floats..1 s 122s1u9 r c " dd dd df S Nr r> r r r s r r z5TestIndexing.test_slicing_no_floats..2 s 3662A2: r c " dddd d f S )Nr r> r r s r r z5TestIndexing.test_slicing_no_floats..3 s 3q71119 r c d d dd f S r r r s r r z5TestIndexing.test_slicing_no_floats..4 s 111cdd7 r c d d S r r r s r r z5TestIndexing.test_slicing_no_floats..6 s 4C4 r c d dddf S )Nr rB @r r s r r z5TestIndexing.test_slicing_no_floats..7 s 2A2qu9 r c " d ddd df S r@ r r s r r z5TestIndexing.test_slicing_no_floats..8 s 6C62A2: r c d dd d f S r r r s r r z5TestIndexing.test_slicing_no_floats..9 s 4C46 r c " d d dddf S )Nr @r> r r s r r z5TestIndexing.test_slicing_no_floats..: s 111aAg: r c d d d S )N ?r r s r r z5TestIndexing.test_slicing_no_floats..< 55S5 r c " dd d ddf S )Nr r> rF r r s r r z5TestIndexing.test_slicing_no_floats..= s 122v#v: r c " dd dd df S )NrB rJ r r r s r r z5TestIndexing.test_slicing_no_floats..> s 16c62A2: r c " d d dd d f S )N @r r s r r z5TestIndexing.test_slicing_no_floats..? s 55S57 r c " d d dddf S )Nr rF r r s r r z5TestIndexing.test_slicing_no_floats..@ s 111a#g: r c ddd S )NrL r> rF r r s r r z5TestIndexing.test_slicing_no_floats..B s 3q9 r c dd d S )NrL rF r r s r r z5TestIndexing.test_slicing_no_floats..C s 388 r c " dd d ddf S )Nr rF r r s r r z5TestIndexing.test_slicing_no_floats..D s 122xCx< r c " dddd df S )NrL rB rJ r r r s r r z5TestIndexing.test_slicing_no_floats..E s 3q9bqb=)9 r c " dddd d f S )NrL rQ r r s r r z5TestIndexing.test_slicing_no_floats..F s 3s3;qqq=)9 r c " d d dddf S )Ng?rJ rF r r s r r z5TestIndexing.test_slicing_no_floats..G s 111c#ck>): r c d d d S r r r s r r z5TestIndexing.test_slicing_no_floats..I rM r r5 r6 r TypeErrorr8 s @r test_slicing_no_floatsz#TestIndexing.test_slicing_no_floats, sP HqcUOO i 111i!5!5!5!5666i!6!6!6!6777i!5!5!5!5666i!3!3!3!3444i 111i!5!5!5!5666i!6!6!6!6777i!2!2!2!2333i!6!6!6!6777i!1!1!1!1222i!6!6!6!6777i!6!6!6!6777i!3!3!3!3444i!6!6!6!6777i!5!5!5!5666i!4!4!4!4555i!8!8!8!8999i!9!9!9!9:::i!9!9!9!9:::i!:!:!:!:;;;i!1!1!1!122222r c f t j dggg t t fd d S )NrB c S Nr r s r r z;TestIndexing.test_index_no_array_to_index..O s 1Qq5 r r[ r8 s @r test_index_no_array_to_indexz)TestIndexing.test_index_no_array_to_indexK s8 HseW i!1!1!1!122222r c t j g d }t |d |t j t |d j |j dz d S )NrB r> rB )r5 r6 r newaxisndimr8 s r test_none_indexz TestIndexing.test_none_indexQ sQ HYYY QtWa
m,,,QtW\16A:.....r c t j g d }t |d | t |d j |u t j d }t t |d t j d S )Nrc r r )r5 r6 r r base
isinstanceint_r8 s r test_empty_tuple_indexz#TestIndexing.test_empty_tuple_indexW sr HYYY QrUA "
a HQKK
1R5"'**+++++r c t j dd }t |d j |j t |d | t t |d t j d S )Nr V4dtype.)r5 zerosr rp typendarray)r9 ss r test_void_scalar_empty_tuplez)TestIndexing.test_void_scalar_empty_tuple_ sd HRt $ $ $QrU[!'***QrUA T!C&\\2:.....r c t j d }| t j }t j d }t || || t j d ||< t |t j d t j d dd }t || || t j d d d d f ||< t |t j d d d d f dd t j d dd }t |||f |||f d S )Nr
r> rB axis )r5 arangeastypeuintpr reshaperepeat)r9 indexu_indexarrs r test_same_kind_index_castingz)TestIndexing.test_same_kind_index_castinge sC ! ,,rx((imm3u:s7|444y||G 3 "
...imm##Aq))3u:s7|444y||AAAdF+G 3 ! QQQtV 4 ; ;AA ; F FGGGimm##Aq))3w /0#eUl2CDDDDDr c t j g d }t |g g t |g j |j t j g t j }t |g g t |g j |j t j g }t t |j | d S )Nrc ro )r5 r6 r rp intpr r7 __getitem__r9 r bs r test_empty_fancy_indexz#TestIndexing.test_empty_fancy_indexy s
HYYY QrUB QrU[!'*** HRrw ' ' 'QrUB QrU[!'*** HRLLj!-33333r c 8 t j g dg dg dg }t |d |u t |d | t |d j |u t |d |d t |d |dd d f t |d |d d df t |d t j d t j d
}d |t
f< t |d d S ) Nrc rS r .)r .r .r )r .rB r> rB )r5 r6 r r ri Ellipsisr s r test_ellipsis_indexz TestIndexing.test_ellipsis_index s Hiii ii ii!
"
" #a QsVQ # q !!! QvY!%%%QvY!AAA#'''QvY!!!Q$((( Qy\28A;;///
HQKK8+Qr c t j g dg dg dg }t |d g d t |d g d t t |j d t t |j d d S )Nrc r r r i @l )r5 r6 r r r7 r r8 s r test_single_int_indexz"TestIndexing.test_single_int_index s Hiii ii ii!
"
" QqT999%%%QrUIII&&& j!-999j!-99999r c t j g dg dg dg }t |t j d |d t |t j d |d dd d S )Nrc r r TFr r5 r6 r r8 s r test_single_bool_indexz#TestIndexing.test_single_bool_index s~ Hiii ii ii!
"
" Qrx~~&$000Qrx'41 66666r c t j d }t j dg }t t |j | t j dgdz }t t |j | t j dt }t t |j | t t |j t d |f d S )N)r rS rd TFr rS rS ro ) r5 onesr6 r r7 r rq boolslice)r9 r r s r test_boolean_shape_mismatchz(TestIndexing.test_boolean_shape_mismatch s gi $ j#/5999%1%%j#/5999t,,,j#/5999j#/E$KK3GHHHHHr c t j g dg }t j dgt }t || | d||< t |g dg d S )N)r r r Tro rL )rL rL rL )r5 r6 r r r s r test_boolean_indexing_onedimz)TestIndexing.test_boolean_indexing_onedim si
Hooo& ' ' HtWD ) ) )QqT1 !Q '''''r c t j d }d }t t ||g t t ||g d t t ||d d g d d S )NrS c || | dk < d S )Nr r )r vs r fz>TestIndexing.test_boolean_assignment_value_mismatch..f s Aa"fIIIr rc rB )r5 r{ r
ValueError)r9 r r s r &test_boolean_assignment_value_mismatchz3TestIndexing.test_boolean_assignment_value_mismatch st
IaLL j!Q+++j!Q 222j!QrrUIII66666r c t j d }t j dt }d|d d< t j dt ||< t j d }d|d d< t || d S )N ro Td rB )r5 rq r r objectr )r9 r indxexpecteds r !test_boolean_assignment_needs_apiz.TestIndexing.test_boolean_assignment_needs_api sz
htnnxD)))TcT
GCv...D 8D>>#3)))))r c t t j g dg dg dg }t j g dg dg dg }t || g d t ||d g dg t ||d ||d d||< t |g d
g d g d g d S )
Nrc r r TFT)FTF)rB rd r r r rB r r> )r r> r )rS r r )r r r r r s r test_boolean_indexing_twodimz)TestIndexing.test_boolean_indexing_twodim s
Hiii ii ii!
"
"
H+++++++++-
.
. QqT???+++QqtWyyyk***QqtWa!g&&& !Q""$ % % % % %r c t j g d }g d}t || ddg t |d |f ddgg d S )Nrc r rB rd r r s r test_boolean_indexing_listz'TestIndexing.test_boolean_indexing_list s^
HYYY QqTAq6"""QtQwZ1a&*****r c t j d }t j dt j d d d }t j d d d d }|||< t |d d t j d }t j d dd d d d }|||d d f< t |d ddg d S ) Nr ro r r r r> rw r> rB )r5 r rq r r{ r r~ )r9 r r cs r ,test_reverse_strides_and_subspace_bufferinitz9TestIndexing.test_reverse_strides_and_subspace_bufferinit s
GAJJ HQbg & & &ttt , IaLL2 !QqT1
GFOO IbMM ! !!Q ' '" -!QQQ$QqTAq6"""""r c * t j d d d d f }t j d d d d }t || || d t j d dd }d S )Nrw r C r> )r5 r{ r copyr~ )r9 r is r 'test_reversed_strides_result_allocationz4TestIndexing.test_reversed_strides_result_allocation sy
IbMM!!!T' " IbMM$$B$ 1Q4166#;; 000 IbMM ! !"a ( (r c j t j dd }t j dd }t j d ddd j |ddg< t j d ddd j |ddg< t
|| d S )N)rd rS r> r r> rS r rB )r5 fullr{ r~ Tr r r s r %test_uncontiguous_subspace_assignmentz2TestIndexing.test_uncontiguous_subspace_assignment s
GIr " " GIr " "Ii((00Aq99;1a& Ii((00Aq99;@@BB1a& Qr c t j d }t t |j t j dg fdz d S )N) rB rB rB rB rB rB rB rB rB rB rB rB rB rB rB rB rB rB rB rB rB rB rB rB rB rB rB rB rB rB rB rB r )r5 r r r7 r r6 r8 s r (test_too_many_fancy_indices_special_casez5TestIndexing.test_too_many_fancy_indices_special_case) s= GI j!-"(A3--1AB1FGGGGGr c . t j d }t |t j d |t j d t |t j d |t j d d S )NrB TF)r5 r6 r bool_r8 s r test_scalar_array_boolz#TestIndexing.test_scalar_array_bool. si HQKKQrx~~&"(4..(9:::Qrx'28E??);<<<< )r5 rq r r7 r __setitem__r8 s r test_broaderrors_indexingz&TestIndexing.test_broaderrors_indexingC s[ HV j!-1a&)))1DEEEj!-1a&)))1DaHHHHHr c t j d }t j dt j }d|d< t t
|j | t t
|j |d t j dt j }d|d< t t
|j | t t
|j |d d S )Nr r ro rw r r )r5 rq r r r r7 r r )r9 r inds r test_trivial_fancy_out_of_boundsz-TestIndexing.test_trivial_fancy_out_of_boundsH s HQKKgb(((Bj!-555j!-a888gb(((Aj!-555j!-a88888r c D t j d }t j dt j ddd d d d d df }t || | d||< t j d }d|d< d|d< t || d S )Nr ro r> rB rd r r )r5 r{ r r~ r )r9 r idxress r test_trivial_fancy_not_possiblez,TestIndexing.test_trivial_fancy_not_possibleS s
IaLLi)))11!Q::111aaa7C1S63''' #illAA1c"""""r c G d dt j }t j d }| | }| d ||g dd d f< t
|dk ||d d g df< t
|dk | d ||d< t
|dk d S ) Nc e Zd Zd ZdS )7TestIndexing.test_nonbaseclass_values..SubClassc 0 | d d S )Nc )fillr9 olds r __array_finalize__zJTestIndexing.test_nonbaseclass_values..SubClass.__array_finalize__d s "
r N__name__
__module__ __qualname__r r r r SubClassr c s#
r r r )rr rB r rB r> rd rS r .)r5 rs rq r viewr r all)r9 r r rt s r test_nonbaseclass_valuesz%TestIndexing.test_nonbaseclass_valuesb s rz
HV
FFHHMMxM ( ( q !///111
a !"!!!___
a q #a r c z t j d }t j dt j dd }t | |g dd d f< t
|| t | |d d g df< t
|| t | |d< t
|| d S )Nr rz ro r r .)r5 rq r{ float64r~
memoryviewr )r9 r rt s r test_array_like_valuesz#TestIndexing.test_array_like_valuesw s HV Ib
+ + + 3 3Aq 9 9 *1
///111
1a *1
!!!___
1a A #1a r c t j g dddg }t j g dt }t || j j t j ddg }t || j j t |d j j t |d j j d S ) N))NGC1001r )NGC1002rL )NGC1003rL )targetS20)V_magz>f4ro )FTTr rB .)r5 recr6 r r flags writeable)r9 dr s r test_subclass_writeablez$TestIndexing.test_subclass_writeable s FLLLLL 13C D
F
Fh,,,D999# &'''h1v # &'''# &'''!
$%%%%%r c 4 t j d }t j d dd j }t || j j | dd }t ||df j j d S )Nrw r r> r rB r )r5 r{ r~ r r r f_contiguousr s r test_memory_orderz TestIndexing.test_memory_order s~
IbMM IbMM ! !!A & & (!
'(((
IIb! !Q$
*+++++r c G d d } | } G d d }t j d }t t |d t j t j d }t t || t j t j d }t t ||t j d f t j t t || | f t j t j d }t j dt }||d< t t |d t j t j |d g }t t || t j t j |d gg }t t ||t j d f t j t t || | f t j d S )
Nc e Zd Zd ZdS )2TestIndexing.test_scalar_return_type..Zeroc dS Nr r r9 s r __index__z.Zero.__index__ qr N)r r r r r r r Zeror s#
r r c e Zd Zd ZdS )7TestIndexing.test_scalar_return_type..ArrayLikec * t j d S r r5 r6 r s r __array__zATestIndexing.test_scalar_return_type..ArrayLike.__array__ s x{{"r N)r r r r
r r r ArrayLiker s#
#
#
#
#
#r r r rB )rB rB r ro )r5 rq r rj float_r6 r rs )r9 r zr r r s r test_scalar_return_typez$TestIndexing.test_scalar_return_type s
DFF # # # # # # # #
HRLL
1R5"),,--- HQKK
1Q4++,,, HV
1Q ^,bi88999
1Q ^,bi88999
HQKK HQf % % %"
1R5"*--... HaY
1Q4,,--- Hq$i[ ! !
1Q ^,bj99:::
1Q ^,bj99:::::r c t j dg }t r0t j t j t j }d|t j dgt j < d|t j dgt j < t t |j
t j dgt j d t t |j
t j dgt j d t r@t t j t j t j | d S d S )Nr rB ro ) r5 r6 r sys getrefcountrp r uint8r r7 r r )r9 r refcounts r test_small_regressionsz#TestIndexing.test_small_regressions s HaSMM :rx'8'899H*+"(A3bg
&
&
&'+,"(A3bh
'
'
'(j!-hs"'222A 7 7 7j!-hs"(333Q 8 8 8 G "'):):;;X F F F F F G Gr c V t j dt j t d z dd }| t j d }t j dt j t d z dd }| t j d }t j dd |d
< t j |j }t || | |||< t j |j dz t j dt j
d j dz
}| t j
d |j }t j |j |d
< t || t j
| ||| t j
< ||d z d k |d d d ||d z d k < d S )
N@ ro r rB iS8r bbbbbbbbS.r r> )r5 rq int8ordr rp r6 r{ sizer r itemsizer| int16)r9 r r xr s r test_unalignedzTestIndexing.test_unaligned s
Xb
(
(
(3s88
3QrT :
FF28D>> " "
Xb
(
(
(3s88
3QrT :
FF28D>> " "'---# Iaf QqT1 !
HQVaZ % %bg . .q271::3F3J1K/K L
FF27OOGQVG $16""#Qqxx))*A... !!((28
!a%1*
##A#!a%1*
r c t j d } G d dt }dgdgf} || }t || j dk t ||f j dk d S )Nr c e Zd ZdS )7TestIndexing.test_tuple_subclass..TupleSubclassNr r r r r r
TupleSubclassr# Dr r% rB rB )r5 r tupler shape)r9 r r% r s r test_tuple_subclassz TestIndexing.test_tuple_subclass s gfoo E qc
e$$E
D()))EF !T)*****r c , G d d }t j d }t | | | | f t j dddg }t | | | | f d S )Nc e Zd Zd Zd Zd ZdS )DTestIndexing.test_broken_sequence_not_nd_index..SequenceLikec dS r r r s r r zNTestIndexing.test_broken_sequence_not_nd_index..SequenceLike.__index__ r r c dS )NrB r r s r __len__zLTestIndexing.test_broken_sequence_not_nd_index..SequenceLike.__len__ r r c t d )Nz Not possible)r7 )r9 items r r zPTestIndexing.test_broken_sequence_not_nd_index..SequenceLike.__getitem__ s 000r N)r r r r r0 r r r r SequenceLiker- sA
1
1
1
1
1r r3 rw r' )f1i8)f2r5 ro )r5 r{ r rq )r9 r3 r s r !test_broken_sequence_not_nd_indexz.TestIndexing.test_broken_sequence_not_nd_index s 1 1 1 1 1 1 1 1 imm3||~~.LLNNO0DEEE htL,#?@@@3||~~.LLNNO0DEEEEEr c t j d }t j d }t j d d d d d d f }t j |d }t || || t j dt j }t |||f || |f d S )Nrw )rw r> )rw 7 rS rS rS ro )r5 r r{ broadcast_tor r rq r )r9 r x2r zinds r !test_indexing_array_weird_stridesz.TestIndexing.test_indexing_array_weird_strides s
GBKK
WW
immAAAtT4 /0oc>22 1S61SXXZZ=111x)))2c4i="SXXZZ-=*>?????r c t j d }|d d dd d df }t d g df}d||< t |d d S )Nr r r rB r> rd rw g $@)r5 rq r r )r9 arror slicess r $test_indexing_array_negative_stridesz1TestIndexing.test_indexing_array_negative_strides s` x 44R42: ++|||,F 3$$$$$r c t j dd }t j d |d< t |d t j dd |d dk sJ d S )N)rB r r ro asdfgr r rB s)r5 rq str_r r6 )r9 r s r test_character_assignmentz&TestIndexing.test_character_assignment si hvS)))!!A3q628G3#?#?#?@@@4yD r r TFr numr (
original_ndimrB c $ t j d|z }t j t 5 ||f|z d d d n # 1 swxY w Y t j t 5 d||f|z < d d d d S # 1 swxY w Y d S )Nr' rL )r5 r pytestraisesr7 )r9 r rI rK r s r test_too_many_advanced_indicesz+TestIndexing.test_too_many_advanced_indices( s gd]*++
]:
&
& 3
]:
&
& % %"$C3 % % % % % % % % % % % % % % % % % %s#
A
AA.
BB B z no threading)reasonc ddl m} t j dg }t j d|fgdz }t j d|fgdz }t j d | d d df t j fd |d
fd t d D }|D ]}| j |u sJ d S )
Nr )ThreadPoolExecutor) f8rS r> )p r )r c x dt | dt j }| | d S )Nr rU )r rp )integerslenr5 r )r r rngs r funcz.funcD s1 <<3s88$bg<FFD IIIIr r ) max_workersc < g | ]} S r )submit).0_r rZ tpes r
zBTestIndexing.test_structured_advanced_indexing..I s' <<i##%% ! Q///<<<<<<%))<<< A
HHJJJJyBr c v t j d d }t |ddg t j |d |d g t |ddgddgf t j ddg t t |j t d g d S )Nrz r r rB r ) r5 r{ r~ r r6 r r7 r r r8 s r test_nontuple_ndindexz"TestIndexing.test_nontuple_ndindexO s IbMM ! !& ) )Q1vY!A$! 6 6777Q1v1v~&!Q(8(8999j!-%++ ?????r N)3r r r r: r] ra rg rl ru 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* r7 r= rB rH rM mark parametrizer5 r6 rO skipifr rj rl r r r r r s3 : : :4 3 3 3>3 3 3 / / / , , ,/ / / E E E( 4 4 4 0 : : : 7 7 7 I I I ( ( (
7
7
7 * * * % % %&+ + +# # #") ) ) H H H
= = = I I I
9 9 9
#
#
# * ! ! ! & & & , , , ; ; ;DG G G ,
+
+
+F F F0
@
@
@ % % %! ! ! [ W
5("(A3-- (* * [ URH-- [ _q"g66% % 76 .-* *% [77 87 0@ @ @ @ @r r c e Zd Zd ZdS )TestFieldIndexingc t j ddg }t t |d t j t t |dg t j d S )Nr )r rT r )r5 rq r rj rs r8 s r r z)TestFieldIndexing.test_scalar_return_typeW s[
HR* & &
1S62:..///
1cU8RZ0011111r N)r r r r r r r rq rq V s# 2 2 2 2 2r rq c e Zd Zd Zd Zd Zd Zej ddddg e
d fd
dgdd
fdddgddgfg d Z d Z d
Z
d S )TestBroadcastedAssignmentsc |||< |S r` r )r9 r r vals r assignz!TestBroadcastedAssignments.assign` s #r c t j d }t j d |d< t j d |g dd d f< t j d |d d ddgf< t j d |dgdgdggddgf< d S )N)rd r> )rB rd r> .r r rB r> )r5 rq r r8 s r test_prepending_onesz/TestBroadcastedAssignments.test_prepending_onesd s HV ### '),,)))QQQ,wy))!!!aV) %'WY%7%7A3aS/Aq6
!"""r c | j }t j }t j d }t t
|||d t j d t t
|||g df t j d t t
|||dgdggf t j d d S )Nr .r> rB rc rB r> )r> r> rB )rw r5 s_rq r r r r9 rw r| r s r test_prepend_not_onez/TestBroadcastedAssignments.test_prepend_not_onen s
U HQKK j&!RWrwvGGGj&!R
^RWV__MMMj&!R!qc
_bgg>N>NOOOOOr c R | j }t j }t j d }t t
|||d t j d t t
|||d t j d t t
|||d d dgf t j d t t
|||d d dgf t j d t t
|||dgd d f t j d d S )N)r rB .r )r r r r{ )rw r5 r| rq r r r} s r test_simple_broadcasting_errorsz:TestBroadcastedAssignments.test_simple_broadcasting_errorsx s
U HV j&!RWbhv6F6FGGGj&!RWbhv6F6FGGGj&!RA3Z&9I9IJJJj&!RA3Z&9I9IJJJj&!RQQQZ&9I9IJJJJJr r .rB r> Nr c p t j d }t j d }t || j dd }t j t 5 }|||< d d d n # 1 swxY w Y t |j | sJ d S )Nr r )rd rS r r r rS )
r5 rq strr) replacerM rN r valueendswith)r9 r valuesr shape_stres r *test_broadcast_error_reports_correct_shapezETestBroadcastedAssignments.test_broadcast_error_reports_correct_shape s
*%%h''E
())11#r::
]:
&
& ! CJ 17||$$Y///////s 0BB Bc t j d }g d|dgdgdggg df< t |d dd df g dk d S )Nr r> rd rS r rB r> r rd )r5 rq r r r8 s r test_index_is_largerz/TestBroadcastedAssignments.test_index_is_larger sm HV (1 A3aS/999
$%2A2rr6iii',,../////r c
t j d }t j d d d d f }t j d d d d }|||< t |d d d |k d S )Nr r r )r5 rq r{ r r )r9 r r r s r test_broadcast_subspacez2TestBroadcastedAssignments.test_broadcast_subspace sz HZ IcNN111T6 " IcNN44R4 !44R4A ""$$%%%%%r )r r r rw ry r~ r rM rm rn r r r r r r r rt rt _ s 8 8 8P P P K K K [ W
1a&%%++ &VS!
1a&1a& !'# $ $
0
0 $ $
00 0 0& & & & &r rt c e Zd Zd Zd Zd ZdS )TestSubclassesc G d dt j }t j d }| | }|d d }t t | |u t |j |u t ||d d |g d }t t | |u t |j |u t t |j t j u t ||g d t |j |g d ||dk }t t | |u t |j |u t t |j t j u t |||dk t |j ||dk d S )Nc e Zd ZdS )+TestSubclasses.test_basic..SubClassNr$ r r r r r r& r r r rd r r )r5 rs r{ r r rr ri r )r9 r r rt s_slices_fancys_bools r
test_basiczTestSubclasses.test_basic s
rz
IaLL
FF8 BQB%W
)*** !"""7AbqbE***III,W
)*** A%&&&W\""bj01117AiiiL1117<999 6661q5V ())) 1$%%%V[!!RZ/00061QU8,,,6;!a%11111r c G d dt j }t j d }| | }d|j _ |g d }t
|j j d S )Nc e Zd ZdS )8TestSubclasses.test_fancy_on_read_only..SubClassNr$ r r r r r r& r r r Fr )r5 rs r{ r r r r )r9 r r rt r s r test_fancy_on_read_onlyz&TestSubclasses.test_fancy_on_read_only s~ rz
IaLL
FF8 !III,
'(((((r c G d dt j }t j d | }|d d }t |j | t |j | |g d }t |j | t |j | ||dk }t |j | t |j | d S )Nc e Zd Zd ZdS )=TestSubclasses.test_finalize_gets_full_info..SubClassc F t j | | _ || _ d S r` )r5 r6 finalize_statusr r s r r zPTestSubclasses.test_finalize_gets_full_info..SubClass.__array_finalize__ s ')x~~$ r Nr r r r r r s#
r r rw rd r? r )r5 rs r{ r r r r )r9 r rt new_ss r test_finalize_gets_full_infoz+TestSubclasses.test_finalize_gets_full_info s rz
IbMM x ( ("1"50%88859a((())) 50%88859a(((!a%50%88859a(((((r N)r r r r r r r r r r r sA 2 2 2:
)
)
)) ) ) ) )r r c e Zd Zd ZdS )TestFancyIndexingCastc d}t j | t }d|d< t j | }t j dg ||< t |d d t
t j |j dgdgft j dg t |d d t
t j |j |t j dg t |d d d S ) N)r ? TrE rB r y @ ?r> y ?) r5 rq r| r r6 r r
ComplexWarningr )r9 r)
bool_index
zero_arrays r test_boolean_index_cast_assignz4TestFancyIndexingCast.test_boolean_index_cast_assign s Xe__++D11
4Xe__
"$1#
: Z%q))) R& +qcA3Z6(9K9K M M MZ%q))) R& +Z2$ I I IZ%q)))))r N)r r r r r r r r r s# * * * * *r r c e Zd Zd Zd ZdS ) TestFancyIndexingEquivalencec t j dt }| }g d|d d< g d|g d< t || t j dt d d d f }g dg|dgd df< t ||d |j }dgdgdgg|d ddgf< t ||d d df t j d t }| }dgd
gdgd ggg|d dd f< dgd
gdgd ggg|dgd f< t || | d
}dgd
gdgd ggg|dgd f< t || d S )Nr ro )rB rB r> rd rd r r rB r )rd rS r r> rS .F)r5 r{ r r r r r )r9 r r r cmp_arrs r test_object_assignz/TestFancyIndexingEquivalence.test_object_assign s
Iav & & &
FFHH
"1"$}}))) 1a
Iav & & &tQQQw /#mm_1#rr'
1ad###
CcE7QC("1"qc'
1a1g&&& giv...((** S1#sQC 01C #sQC!-.QCH
3(((hhsmm #sQC!-.QCH
3(((((r c t j d }| }t j g d |d d< t j g d |g d< t || t j d d d d f }t j g dg |dgd df< t ||d d S )Nr )2-3-1rd )r r r )r r> rB r )r5 r{ r r6 r r s r test_cast_equivalencez2TestFancyIndexingEquivalence.test_cast_equivalence s IaLL
FFHH***++"1"x 1 1 122))) 1a
IaLLqqq !X 0 0 0 1221#rr'
1ad#####r N)r r r r r r r r r r s3 ) ) )@ $ $ $ $ $r r c B e Zd ZdZd Zd Zd Zd Zd Zd Z d Z
d Z d
S ) TestMultiIndexingAutomateda:
These tests use code to mimic the C-Code indexing for selection.
NOTE:
* This still lacks tests for complex item setting.
* If you change behavior of indexing, you might want to modify
these tests to try more combinations.
* Behavior was written to match numpy version 1.8. (though a
first version matched 1.7.)
* Only tuple indices are supported by the mimicking code.
(and tested as of writing this)
* Error types should match most of the time as long as there
is only one error. For multiple errors, what gets raised
will usually not be the same one. They are *not* tested.
Update 2016-11-30: It is probably not worth maintaining this test
indefinitely and it can be dropped if maintenance becomes a burden.
c & t j t j g d dddd | _ t j d | _ dt dt j g d t j d
d gd d
gg t j d d gd d ggg t d dd
t ddd t ddd t d d d t j dt j
t j g d t j d
gdgdgg t j ddgddggt j d t j d
dgt j
t j dgdz t t j ddg g| _ t d ddgt j d
g dg| _ t d d dg| _ d S )N)rd rB r r rd rB r r )rd r r r skipr )TFFTFr> r rS r )r rB rB ro )r rB r r r rL )r5 r{ prodr~ r emptyr r r6 r r rp newbyteorderr rq intcomplex_indicessimple_indices fill_indicesr s r setup_methodz'TestMultiIndexingAutomated.setup_method> s 27<<<001199!Q1EE,'' &
H))) * * HtUmeT]3 4 4 Hu~u~67 8 8 "aOO !Q !R $b ! ! HYbg . . . HZZZ HqcA3_ % % Hq"g1v&bhv.>.>.K.K.M.M N N N HaWBG , , , HaSV3 ' ' ' Hb"X ' * (rA3$8H8H % ' #4 . . 2r c t | }g }d}d}d}d}d}d} t | D ])\ }
} | t | t j r9| j t k r)d}| j dk rt || j z
}|| j z
}\| t u r | |
} jt t | t r|dz
}t | t j s= t j | t j } n# t $ r t w xY w| ||
< n/| j j dk r | j j dk rt d | j dk rd}|dz
}|dz
}+|j |z
dk rt |dk rd|vr| |fS | $t dd g|j |z
z || | dz < t | D ]\ } } t | t rDt j | |j | } | d
| g _| l| d t j dgt j g | |j d| d z |j | d z }t | t j r:| j t k r)| j |j | | | j z k rt t j t j | |j | | | j z d
}
nH# t. $ r; d}t j dg| z t j }
Y nw xY w| j dk rh| |j d| t j |j | | | j z fz |j | | j z d z }|
} n t j | dd |d
}nD# t. $ r t d w xY wt j | dd |d }nt |& |' | } | |j d| |j z |j | dz d z }n# t $ r t w xY w| |j z
} |& | d | }| dz
} ||fS )a Mimic multi dimensional indexing.
Parameters
----------
arr : ndarray
Array to be indexed.
indices : tuple of index objects
Returns
-------
out : ndarray
An array equivalent to the indexing operation (but always a copy).
`arr[indices]` should be identical.
no_copy : bool
Whether the indexing operation requires a copy. If this is `True`,
`np.may_share_memory(arr, arr[indices])` should be `True` (with
some exceptions for scalars and possibly 0-d arrays).
Notes
-----
While the function may mostly match the errors of normal indexing this
is generally not the case.
Tr FNrB ro r r z;arrays used as indices must be of integer (or boolean) typert nr' raise)moder r zinvalid index into 0-sizedwraprx )(list enumeraterj r5 rs rp r rf r7 r r r6 r r kindr r{ indicesr) appendr~ ravel_multi_indexnonzero Exceptionsumr rX rf insertextendremove transposer| r broadcastr zipanytakeravel)r9 r r
in_indicesno_copy num_fancy fancy_dim error_unless_broadcast_to_emptyrf ellipsis_posr r ax flat_indx new_indicesaxes
fancy_axesniai
orig_shape
orig_slicer _indx_sizemis r _get_multi_indexz+TestMultiIndexingAutomated._get_multi_index[ s& 0 ']]
+0' ,, $ $ GAt|$
++
d0B0B 9>>$$ !TY& x '#$L $&&
dBJ//
> % 8D888DD! % % %$$% $
1
C''DJOs,B,B "= > > >yA~~ AID NII 8d?Q
199Z//88::w& & #8=dD8I8I7J8;48IJ|LN2 3 "*-- ; , ; ,HB$&&
y$,,sy}"="=>T{+++ RXqc%A%A%A BCCCkk39SbS>D#839RSS>#IKK$
++
)
d0B0B:2bl?!;;;$$H " 4RZ5E5E$'IbDI o$>W!N !N !NII H H H6:3 "!TXXZZ rw G G GIIIH
9>> ++sy"~%'WSYr"TY,-G%H%H$J(K$'Iblmm$<(= ? ?C %DD %$ 9>>sy},, "
~0E0E ((yA~~
39R=((DSYr]?,B,B$$G q BKNc)),&& ""4(((( Q T{++++ q===!!!!*Kch((DJ q3% ( ( (BB
$
$a7c>> N))$qrr(333#B!GB #BSabb]](: ; ; ( ("))"--- Bc$ii!m#!G #-*t"35C 6 6 DAw#~~t99>> Y
'2DH
+=(= >
kk39SbS>')wz':':'A'A#'F'F&H$I&)iSabb]]0B0C0C&D$E G G
lDH- 8q==6 ) (((+DHj(A(A 1 1 u :?? $ 6%5.11 1RVEUFN5K5K 1&0 01tABBx==C
OO33wz**a//K!#!5d122h
;B"D "D "DBB ( K K K #--I"J"J JK
1$qrr(J7= ? ? ? %$hhrxxzzh33% ++sy"~*,((3*-)BqDEE*:(; = =CC " % % %$$% bg
((47(,,C !GBBG|s7 C55D#>L""AM'&M' ]66^$=`""`4c | || \ }}n# t $ r}t rt j | }t t
| |j | t t
| |j |d t r"t |t j | Y d}~dS d}~ww xY w|
|||| dS )a Check a multi index item getting and simple setting.
Parameters
----------
arr : ndarray
Array to be indexed, must be a reshaped arange.
index : tuple of indexing objects
Index being tested.
r N r r r r r r rr r r r _compare_index_resultr9 r r mimic_getr r
prev_refcounts r _check_multi_indexz-TestMultiIndexingAutomated._check_multi_indexJ s !%!6!6sE!B!B Iww
5 # 4 4
$q''3?E : : : $q''3?E1 = = =
B ]COC,@,@AAA FFFFF
""3y'BBBBBs
B<B B77B<c | ||f \ }}n# t $ r}t rt j | }t t
| |j | t t
| |j |d t r"t |t j | Y d}~dS d}~ww xY w|
|||| dS )aW Check a single index item getting and simple setting.
Parameters
----------
arr : ndarray
Array to be indexed, must be an arange.
index : indexing object
Index being tested. Must be a single index and not a tuple
of indexing objects (see also `_check_multi_index`).
r Nr r s r _check_single_indexz.TestMultiIndexingAutomated._check_single_indexb s !%!6!6sUH!E!E Iww
5 # 4 4
$q''3?E : : : $q''3?E1 = = =
B ]COC,@,@AAA FFFFF
""3y'BBBBBs
B=B B88B=c z | }|| }t || |j dk r|j dk rtt t j || |k t rG|r#t t j
| d n"t t j
| d | }|dz ||< |j dk rdS |r"|j dk r|dz
}t || dS |j | xx dz
cc< t || dS )z4Compare mimicked result to indexing result.
r rd r> r N)
r r r rf r r5 may_share_memoryr r r r flatr )r9 r r r r indexed_arrr s r r z0TestMultiIndexingAutomated._compare_index_resultz sO hhjj%j ; 222 q [%5%:%: B' S99WD E E E
: : !5!5q9999 !5!5q999
HHJJ t#% 6Q;; F {'1,, 4 K sA & & & F ""$$%%%-%%%3"""""r c @ t j d }t |t j d d d|t j d < t |d | | j t j | j t f | | j t j | j t d f | | j t j | j t d f d S )Nr TrB ro r )N.)r5 r6 r r r
zeros_liker r8 s r test_booleanz'TestMultiIndexingAutomated.test_boolean s HQKKQrx~~&*** "(4..Q
FR]466668 : : : FR]46666v>@ B B B FR]46666yAC E E E E Er c p t j 5 t j ddt t j ddt j d dD ]}| j | j | j | j g}| j ||< t | D ]S}t fd|D }| | j | | | j
| T d d d n # 1 swxY w Y | | j d | | j
d t t | j j d t t | j j dd t t | j j ddd gddf t t | j j ddd gddfd d S )
NerrorrS c 8 t | t o| dk S )Nr )rj r )r s r isskipz8TestMultiIndexingAutomated.test_multidim..isskip s !#s++=v
=r )r r> rd c 3 2 K | ]} |
|V d S r` r )r^ r r s r z;TestMultiIndexingAutomated.test_multidim.. s0 !D!D&&))!D!!D!D!D!D!D!Dr )r r r r )r r r r r r rB )warningscatch_warningsfilterwarningsDeprecationWarningr5 VisibleDeprecationWarningr r r r r( r r r r r7 r r )r9
simple_postocheckr r s @r
test_multidimz(TestMultiIndexingAutomated.test_multidim s
$
&
& ; ;
#GR1C D D D #GR1M N N N
>
>
> (
;
;
,d.B ,d.?A&*&9
# $g . ; ;E !!D!D!D!DU!D!D!D D DE++DFE:::++DFE::::;
; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ;&
555 555j$&"4oFFFj$&"4oqIIIj$&"4q!aS!Q6GHHHj$&"4q!aS!Q6GKKKKKs CC''C+.C+c n t j d }| j D ]}| || d S )Nrw )r5 r{ r r )r9 r r s r test_1dz"TestMultiIndexingAutomated.test_1d sD IbMM) / /E $ $Q . . . . / /r N) r r r __doc__r r r r r r r r r r r r r ( s *3 3 3:m m m ^C C C0C C C0 # # #@E E E" L L L</ / / / /r r c 0 e Zd ZdZd Zd Zd Zd Zd ZdS )TestFloatNonIntegerArgumentz
These test that ``TypeError`` is raised when you try to use
non-integers as arguments to for indexing and slicing e.g. ``a[0.0:5]``
and ``a[0.5]``, or other functions like ``array.reshape(1., -1)``.
c t j dggg }|t j dg |ddg |d d ddgf |d d dd d f |d d d d d d f d S )Nr r r r8 s r test_valid_indexingz/TestFloatNonIntegerArgument.test_valid_indexing s HseW "(A3-- 1a& !!!aV) !!!Qqqq& !!!AAAaaa%r c t j dggg }|d d |dd |d d |dd |d d d |dd d |d dd |ddd d S )Nr r r> rB r r8 s r test_valid_slicingz.TestFloatNonIntegerArgument.test_valid_slicing s HseW """ !"" "1" !A# ##A# !$Q$ $1Q$ !Aa%r c t j dgg }t t t j |d t t t j |t j d df t t t j |dgd t t t j |dgt j d d S )Nr )rL rL r rL r r )r5 r6 r r\ r~ r r r8 s r test_non_integer_argument_errorsz.mult s q5Lr rB rd )r r\ r5 r rk )r9 r s r (test_non_integer_sequence_multiplicationzDTestFloatNonIntegerArgument.test_non_integer_sequence_multiplication sS isBIaLL999 aS"'!** r c 6 t j d }t t t j |d t t t j |d t t t j |d t t t j |d d S )N)rd rd rd ?)r! rB )rB g@)g?g333333?)r5 rq r r\ min)r9 r s r test_reduce_axis_float_indexz8TestFloatNonIntegerArgument.test_reduce_axis_float_index so HW iC000iH555iH555iI66666r N) r r r r r r r r r# r r r r r sl B B B 7 7 7 7 7r r c e Zd Zd Zd Zd ZdS )TestBooleanIndexingc t j dggg }t t t j |d t t t j |t j d df t t t j t j d t t t j t j
t t t j |dgdf d S )NrB )Tr Tr r F)args) r5 r6 r r\ r~ r operatorr r
r True_r r8 s r test_bool_as_int_argument_errorsz4TestBooleanIndexing.test_bool_as_int_argument_errors s HseW iQ
;;;iQ$ 0DEEE i $ @@@' BBBiA3??????r c t j d d j dk sJ dddgdddgdggf j dk sJ t t fd d S )
Nr )FT.)r r> rd rS Tr rB r> r c dddgdf S )NFr rB .r r s r r zETestBooleanIndexing.test_boolean_indexing_weirdness.. s !EAq63,>*? r )r5 r r) r r7 r8 s @r test_boolean_indexing_weirdnessz3TestBooleanIndexing.test_boolean_indexing_weirdness s| GI !"(L88881vtTA3!56<FFFFj"?"?"?"?@@@@@r c t j d t j dgdz g t t dfd t j dgdz dgz g t t dfd t j dgd z g t t dfd
t j d t j dgdggg t t d fd
d S )N)rd rd Fr ztboolean index did not match indexed array along dimension 0; dimension is 3 but corresponding boolean dimension is 1c S r` r )r idx1s r r zETestBooleanIndexing.test_boolean_indexing_fast_path..# AdG r r Tc S r` r )r idx2s r r zETestBooleanIndexing.test_boolean_indexing_fast_path..* r1 r rw c S r` r )r idx3s r r zETestBooleanIndexing.test_boolean_indexing_fast_path..1 r1 r )rB rB r> ztboolean index did not match indexed array along dimension 1; dimension is 1 but corresponding boolean dimension is 2c S r` r )r r s r r zETestBooleanIndexing.test_boolean_indexing_fast_path..9 s AcF r )r5 r r6 r r7 )r9 r r r0 r3 r5 s @@@@@r test_boolean_indexing_fast_pathz3TestBooleanIndexing.test_boolean_indexing_fast_path s3
GFOO x% $$J
F OOOO x%dV+,--J
F OOOO x% %%J
F OOOO
GI h$%)*++J
F NNNN r N)r r r r* r- r7 r r r r% r% sG @ @ @A A A r r% c e Zd ZdZd ZdS )TestArrayToIndexDeprecationz7Creating an index from array not 0-D is an error.
c & t j dggg }t t t j t j dg t t t j ||df t t t j |dg| d S )NrB r r )r5 r6 r r\ r( r r~ r r8 s r test_array_to_index_errorz5TestArrayToIndexDeprecation.test_array_to_index_error@ so HseW i 1# ???iQB888i!aS!44444r N)r r r r r; r r r r9 r9 < s- 5 5 5 5 5r r9 c e Zd ZdZd ZdS )TestNonIntegerArrayLikezTests that array_likes only valid if can safely cast to integer.
For instance, lists give IndexError when they cannot be safely cast to
an integer.
c t j d }t t |j ddg t t |j ddgf | g d S )Nrw r! g ?1r )r5 r{ r r7 r r8 s r r z"TestNonIntegerArrayLike.test_basicP s[ IbMMj!-#s<<<j!-3* ???
br Nr r r r r r r r r= r= I s- r r= c e Zd ZdZd ZdS )TestMultipleEllipsisErrorz/An index can only have a single ellipsis.
c t j d t t fd t t j t
fdz f t t j t
fdz f d S )Nrw c d S )N)..r r s r r z6TestMultipleEllipsisError.test_basic..` s !H+ r r> rd )r5 r{ r r7 r r r8 s @r r z$TestMultipleEllipsisError.test_basic^ sj IbMMj"5"5"5"5666j!-8+/1CDDDj!-8+/1CDDDDDr Nr@ r r r rB rB Z s2 E E E E Er rB c e Zd Zd Zd ZdS )TestCApiAccessc t j t d }t t |t j d d t t |t j d d t t |t j d d t t |t j d d t t |t j d d t j d }t |d ||d | dd }t |d
||d
d S ) Nr r rw r rw rw rS r r> )
functoolspartialr r r7 r5 r r{ r r~ )r9 subscriptr s r test_getitemzTestCApiAccess.test_getitemf s %na88 j)RWR[[!<<<j)RWR[["===j)RWR[[#>>>j)RWX->->CCCj)RWX->->DDD IbMM1Q41a 111
IIaOO1R5))Ar"2"233333r c t j t d }t t |t j d d t t |t j d dd t t |t j d dd t t |t j d dd t t |t j d dd t t |t j d dd t j d } ||dd t |d dk |
d d
} ||dd t |d ddg d S ) NrB rw r r r rH rI rS r r> r ) rK rL r r r r5 r r7 r{ r r~ r )r9 rw r s r test_setitemzTestCApiAccess.test_setitemv sF ">155 j&"'"++q999j&"'"++q!<<<j&"'"++r1===j&"'"++sA>>>j&"'(*;*;RCCCj&"'(*;*;S!DDD IbMMq!R!
IIaOOq!R1R52r(+++++r N)r r r rN rP r r r rF rF e s2 4 4 4 , , , , ,r rF )!r r rK r( rM numpyr5 numpy.core._multiarray_testsr itertoolsr
numpy.testingr r r r r r
r r r rq rt r r r r r r% r9 r= rB rF r r r rU s
7 7 7 7 7 7 B @ B @ B @ B @ B @ B @ B @ B @J2 2 2 2 2 2 2 2@& @& @& @& @& @& @& @&F=) =) =) =) =) =) =) =)@* * * * * * * *,-$ -$ -$ -$ -$ -$ -$ -$`d/ d/ d/ d/ d/ d/ d/ d/L
47 47 47 47 47 47 47 47n4 4 4 4 4 4 4 4 n
5
5
5
5
5
5
5
5 "E E E E E E E E$, $, $, $, $, $, $, $, $, $,r