yd- Z?ej @ d.ejA d/ ej @ d0ejA d/ ej B d1 d2 ZCd3 ZDd4 ZEd5 ZFd6 ZGd7 ZHd8 ZId9 ZJd: ZKd; ZLd< ZMd= ZNd> ZOd? ZPd@ ZQdA ZRdB ZSdC ZTdD ZUdE ZVdF ZWdG ZXdH ZYdI ZZdJ Z[dK Z\ej edL ej e dM dN Z]dO Z^dP Z_dQ Z`dR Zaej jb dS ZcdT Zd G dU dV ZedW ZfdX ZgdY ZhdZ Zid[ Zjd\ Zkej @ d] ejl d^d_` ejl dad_` dddbddf ejl dcd_` dddbdddddbddf ejl ddd_` ddde ejl dfd_` dddbdddgf ejl dcd_` dddbdddddbdddhf g di Zmej @ djdkgddhd gf ejl dldm` d dngf ejl dldm` df ejl dodm` dhd gf ejl dpdm` dhdhgf ejl dpdm` dfg dq Zndr Zods Zpdt Zqdu Zrdv Zsdw Ztdx Zudy Zvej @ dz ejw d{ex` dddddbddf ejw d{exd|} dddddbddf ejw d{ex` dddddbddf y d| g d~ Zzd Z{d Z|d Z}d Z~d Zd Zej e d d Zej e j ddb dk o
e j dk d ej @ ddgg d ej @ dg d d Zej e d ej @ ddgddg d Zd ZdS ) N)arrayarangenditerall)assert_ assert_equalassert_array_equal
assert_raisesIS_WASM HAS_REFCOUNTsuppress_warnings break_cyclesc g }| j s5| | j | | j 5|S N)finishedappend multi_indexiternextirets p/builddir/build/BUILD/cloudlinux-venv-1.0.12/venv/lib64/python3.11/site-packages/numpy/core/tests/test_nditer.pyiter_multi_indexr sE
Cj
1=!!!
j J c g }| j s5| | j | | j 5|S r )r r indexr r s r iter_indicesr sE
Cj
17
j Jr c g }| j s5| | j | | j 5|S r )r r iterindexr r s r iter_iterindicesr sE
Cj
1;
j Jr zPython lacks refcounts)reasonc V t d } t j d }t j | }t j | }t
| g ddggd|g 5 }t |j t t j | |k t t j | |k d d d n # 1 swxY w Y d }t t j | | t t j | | t dd } t j d }t j | }t j | }t
| g dgg|g }t j | }t j | }|
}t t j | |k t t j | |k d }t t j | | t t j | | d }t t j | | t t j | | ~d S ) N f4 readwrite updateifcopyunsafecasting op_dtypesdtyper* ) r npr, newbyteordersys getrefcountr r iterationneedsapir copy)adtrc_arc_dtitrc2_arc2_dtit2s r test_iter_refcountr< % s
q A $ $ $ & &B ?1 D OB E 2 ~./ $
- $&B(()))""T)***##e+,,,
- - - - - - - - - - - - - - -
B ##T*** $$e,,, q A $ B ?1 D OB E 2 $
B
OA E
_R
F
''))C COA &''' COB & ()))
B ##U+++ $$f---
C ##T*** $$e,,, s 6A C""C&)C&c dD ]} t t j | }t dt | z D ];}t d gt | z }t t | D ] }d|z |z rt d d d ||< t
| }| | | }t |g dgg }t d |D | t |j
g dgg }t d |D | t | dk rAt | dd g dgg }t d |D | =d S )
N) rA rB )rD rA rB rA )rD rA rD rD rA rD readonlyc g | ]}|S rH .0xs r
z(test_iter_best_order..b ! r c g | ]}|S rH rH rI s r rL z(test_iter_best_order..e rM r r c g | ]}|S rH rH rI s r rL z(test_iter_best_order..i ^^^Aa^^^r ) r r. prodrangelenslicetuplereshaper r Tswapaxesshaper4 dirs
dirs_indexbitaviewr s r test_iter_best_orderr` P s
J 0 0 275>> " "!SZZ-(( 0 0D ++ s5zz1J SZZ((
<
<VtO <&+D$&;&;JsO z**JIIe$$Z0EubJ<.11A Q + + +uwj\N33A Q + + +5zzA~~5>>!Q//zl^DD ^^^^^Q///# 00 0r c dD ]} t t j | }t dt | z D ]}t d gt | z }t t | D ] }d|z |z rt d d d ||< t
| }| | | }t |d }t d |D |
d t |j d }t d |D |j
d t | dk rgt | dd d }t d
|D | dd
d d S ) Nr> rD rE Corderc g | ]}|S rH rH rI s r rL z%test_iter_c_order..| rM r c g | ]}|S rH rH rI s r rL z%test_iter_c_order.. rM r r rO c g | ]}|S rH rH rI s r rL z%test_iter_c_order.. rQ r
r r. rR rS rT rU rV rW r r ravelrX rY rZ s r test_iter_c_orderrj k J K K 275>> " "!SZZ-(( K KD ++ s5zz1J SZZ((
<
<VtO <&+D$&;&;JsO z**JIIe$$Z0EuC(((A Q 3)?)? @ @ @uwc***A Q S )A)A B B B5zzA~~5>>!Q//s;;; ^^^^^$)NN1a$8$8$>$>S$>$I$IK K K# KK Kr c dD ]} t t j | }t dt | z D ]}t d gt | z }t t | D ] }d|z |z rt d d d ||< t
| }| | | }t |d }t d |D |
d t |j d }t d |D |j
d t | dk rgt | dd d }t d
|D | dd
d d S ) Nr> rD rE Frc c g | ]}|S rH rH rI s r rL z%test_iter_f_order.. rM r c g | ]}|S rH rH rI s r rL z%test_iter_f_order.. rM r r rO c g | ]}|S rH rH rI s r rL z%test_iter_f_order.. rQ r rh rZ s r test_iter_f_orderrq rk r c dD ]} t t j | }t dt | z D ]}t d gt | z }t t | D ] }d|z |z rt d d d ||< t
| }| | | }t |d }t d |D |
d t |j d }t d |D |j
d t | dk rgt | dd d }t d
|D | dd
d d S ) Nr> rD rE Arc c g | ]}|S rH rH rI s r rL z*test_iter_c_or_f_order.. rM r c g | ]}|S rH rH rI s r rL z*test_iter_c_or_f_order.. rM r r rO c g | ]}|S rH rH rI s r rL z*test_iter_c_or_f_order.. rQ r rh rZ s r test_iter_c_or_f_orderrw rk r c t j d dd } t j | dg }d|_ t d |D g d d S ) Nr# rD rA r flagsr rD c g | ]}|S rH rH )rJ r s r rL z/test_nditer_multi_index_set.. s ! r )rD rA rB r? )r. r rW r r r )r4 r8 s r test_nditer_multi_index_setr} sb
! Q""A 1]O , , ,B BN R ,,,/////r c d} t j t j g d dg }t j | }| f|_ t j | }t
|| d S )Nr )o iM i r ry )r. r r r0 r1 r r )r r start_count end_counts r $test_nditer_multi_index_set_refcountr sm
E
"(///00 HHHA/%((KHAM&&Ii(((((r c t d } t | dgdgg }t t | g d t | d d d dgdgg }t t | g d d S )NrB r rF )r rO rD rA rE )r r r r )r r r r r4 r s r #test_iter_best_order_multi_index_1dr s q Aq=/ZL>22A!!$$&>&>&>???q2w :, 88A!!$$&>&>&>?????r c t d } t | dd dgdgg }t t | g d t | dd d dgdgg }t t | g d t | dd d d d
dgdgg }t t | g d t | dd d d d d d
f dgdgg }t t | g d t | dd d d d
d d d
f dgdgg }t t | g d
t | dd d d d d
dgdgg }t t | g d t | dd d d d d d d
f dgdgg }t t | g d t | dd d d d d
d d d
f dgdgg }t t | g d d S )Nr# rD rA r rF )r r r rO r{ rO r rO rO rO rD rm rc )r r r r r{ r rE )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 r{ r r r r r r rW r r r3 r s r #test_iter_best_order_multi_index_2dr s q AqyyA :, @@A!!$$&V&V&VWWWqyyA####.. :, PPA!!$$&V&V&VWWWqyyAttt$}o ~FFA!!$$&V&V&VWWWqyyAqqq$$B$w'- J<.IIA!!$$&V&V&VWWWqyyAtttTTrTz*]Ozl^LLA!!$$&V&V&VWWWqyyA####..ttt4}o ~VVA!!$$&V&V&VWWWqyyA####..qqq$$B$w74A?j\N T TA!!$$&V&V&VWWWqyyA####..tttTTrTz:4A?j\N T TA!!$$&V&V&VWWWWWr c t d } t | ddd dgdgg }t t | g d t | ddd d dgdgg }t t | g d t | ddd d d d
dgdgg }t t | g d t | ddd d d d d d
f dgdgg }t t | g d t | ddd d d d d d d d
f dgdgg }t t | g d
t | ddd d d d d
dgdgg }t t | g d t | ddd d d d d d d
f dgdgg }t t | g d t | ddd d d d d d d d d
f dgdgg }t t | g d d S )N rD rA r rF ) r r r r r rO r rO r r rO rO r rD r r rD rO rO r r rO r rO rO rO r rO rO rO rO rD r rO rD rO rm rc ) r r r r r r r r r r r r rE ) 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 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 r r r r r r r r r r r r s r #test_iter_best_order_multi_index_3dr s r
AqyyAq!!M?j\NCCA!!$$ _ _ _` ` ` qyyAq!!&&S&11M?j\NSSA!!$$ _ _ _` ` ` qyyAq!!$$B$'- J<.IIA!!$$ _ _ _` ` ` qyyAq!!!!!TTrT'*]Ozl^LLA!!$$ _ _ _` ` ` qyyAq!!!!!AAAttt),}o ~NNA!!$$ _ _ _` ` ` qyyAq!!&&S&11$$B$75BOzl^ U UA!!$$ _ _ _` ` ` qyyAq!!&&S&11!!!TTrT':5BOzl^ U UA!!$$ _ _ _` ` ` qyyAq!!&&S&11!!!AAAttt)<5BOzl^ U UA!!$$ _ _ _` ` ` ` `r c t d } t | dgdgg }t t | g d t | d d d dgdgg }t t | g d d S )NrB c_indexrF r rO rD rA rE rA rD rO r r r r r r s r test_iter_best_order_c_index_1dr % q Aq9+ ~..A a ,,,///q2w zl^44A a ,,,/////r c t d } t | dd dgdgg }t t | g d t | dd d dgdgg }t t | g d t | dd d d d
dgdgg }t t | g d t | dd d d d d d
f dgdgg }t t | g d t | dd d d d
d d d
f dgdgg }t t | g d
t | dd d d d d
dgdgg }t t | g d t | dd d d d d d d
f dgdgg }t t | g d t | dd d d d d
d d d
f dgdgg }t t | g d d S )Nr# rD rA r rF r rO rD rA rB r? rm rc )r rA rO rB rD r? rE )rA rB r? r rO rD )rD rO r r? rB rA r? rB rA rD rO r )rA r rB rO r? rD )rD r? rO rB r rA )r? rD rB rO rA r r r rW r r r3 r s r test_iter_best_order_c_index_2dr 0 q AqyyA zl^<BBA a "4"4"4555qyyAqqq$$B$w')
|nEEA a "4"4"4555qyyAtttTTrTz*YK:, HHA a "4"4"4555qyyA####..ttt4%.K:, A AA a "4"4"4555qyyA####..qqq$$B$w7%.K:, A AA a "4"4"4555qyyA####..tttTTrTz:%.K:, A AA a "4"4"455555r c t d } t | ddd dgdgg }t t | g d t | ddd d dgdgg }t t | g d t | ddd d d d
dgdgg }t t | g d t | ddd d d d d d
f dgdgg }t t | g d t | ddd d d d d d d d
f dgdgg }t t | g d
t | ddd d d d d
dgdgg }t t | g d t | ddd d d d d d d
f dgdgg }t t | g d t | ddd d d d d d d d d
f dgdgg }t t | g d d S )Nr rD rA r rF r rO rD rA rB r? r#
rm rc r r# rD r rB r rO r rA r r? r rE r# r r r r r r rO rD rA rB r? rB r? rD rA r rO r r r r r# r rO r rA rD r? rB r r# r r r r r# r r rD r rB r rO r rA r r? rB r rD r r r# r? r rA r rO r rO r rA r r? r r r# rD r rB r r r s r test_iter_best_order_c_index_3dr M W r
AqyyAq!!I;* ??A a B B BD D D qyyAq!!&&S&11%.K:, A AA a B B BD D D qyyAq!!$$B$')
|nEEA a B B BD D DqyyAq!!!!!TTrT'*YK:, HHA a B B BD D DqyyAq!!!!!AAAttt),ykZL>JJA a B B BD D D qyyAq!!&&S&11$$B$7%.K:, A AA a B B BD D DqyyAq!!&&S&11!!!TTrT':%.K:, A AA a B B BD D DqyyAq!!&&S&11!!!AAAttt)<%.K:, A AA a B B BD D D D Dr c t d } t | dgdgg }t t | g d t | d d d dgdgg }t t | g d d S )NrB f_indexrF r rE r r r s r test_iter_best_order_f_index_1dr r r r c t d } t | dd dgdgg }t t | g d t | dd d dgdgg }t t | g d t | dd d d d
dgdgg }t t | g d t | dd d d d d d
f dgdgg }t t | g d t | dd d d d
d d d
f dgdgg }t t | g d
t | dd d d d d
dgdgg }t t | g d t | dd d d d d d d
f dgdgg }t t | g d t | dd d d d d
d d d
f dgdgg }t t | g d d S )Nr# rD rA r rF r rD rB rO rA r? rm rc r rE )rO rA r? r rD rB )rB rD r r? rA rO )r? rA rO rB rD r )rO r rA rD r? rB )rB r? rD rA r rO r r r s r test_iter_best_order_f_index_2dr } r r c t d } t | ddd dgdgg }t t | g d t | ddd d dgdgg }t t | g d t | ddd d d d
dgdgg }t t | g d t | ddd d d d d d
f dgdgg }t t | g d t | ddd d d d d d d d
f dgdgg }t t | g d
t | ddd d d d d
dgdgg }t t | g d t | ddd d d d d d d
f dgdgg }t t | g d t | ddd d d d d d d d d
f dgdgg }t t | g d d S )Nr rD rA r rF r rm rc r rE r r r r r r r r s r test_iter_best_order_f_index_3dr r r c dD ]} t j | }t | }t dt | z D ]}t d gt | z }t t | D ] }d|z |z rt d d d ||< t
| }| | | }t |dgdgg }t |j
d t |d j |f t |j dgdgg }t |j
d t |d j |f t | dk rYt |
dd dgdgg }t |j
d t |d j |f d S )Nr> rD rE
external_looprF rO r )r. rR r rS rT rU rV rW r r ndimr[ rX rY )r[ sizer4 r\ r] r^ r_ r s r test_iter_no_inner_full_coalescer s J 2 2wu~~ 4LL!SZZ-(( 2 2D ++ s5zz1J SZZ((
<
<VtO <&+D$&;&;JsO z**JIIe$$Z0Eu /:, @@A # # # 1dW - - -uw 1ZL>BBA # # # 1dW - - -5zzA~~5>>!Q//%4$5 ~G G QVQ''' QqTZ$111+ 2 2 2r c
t d ddd d d d d d df } t | dgdgg }t |j d t |d j d t d ddd d d d dd d f } t | dgdgg }t |j d t |d j d
t d ddd d dd d d d f } t | dgdgg }t |j d t |d j d t d d d dd d dd d dd d } t | dgdgg }t |j d t |d j d
d S )N rD rA rB rE r rF r r r rO )r r )r rW r r r r[ r s r !test_iter_no_inner_dim_coalescingr s
r
1a##AAAaaa"H-Aq?#zl^44A1T"""r
1a##AAAss111H-Aq?#zl^44A1T"""r
1a##CRC!!!G,Aq?#zl^44A1U### r
1aAq!Q1a;;Aq?#zl^44A1U#####r c t d ddd } t | dgdgg }t |j d t d ddd }t |dgdgg }t |j d t | d d dgdgg }t |j d t |j dgdgg }t |j d t |j d
gdgg }t |j d t |j d d d
gdgg }t |j d t d ddd }t |d }t |j d t |j d }t |j d t |d
}t |j d t |j d
}t |j d t |d }t |j d t |j d }t |j d d S )Nr rD rA rB r rF r rO r r rb rc rm rs )r rW r r r rY rX )r4 r a3ds r test_iter_dim_coalescingr s[ r
1a##Aq=/ZL>22A **
Q1
%
%CsYK:, 00As||Aq!!I;* ??AsuykZL>22AsuykZL>22Asu~~a##i[J<.AAA **
Q1
%
%Cs# AsuC As# AsuC As# AsuC Ar c
t t d t j d gdgdggdz } t | j d t | j d t t d dd t j d gdgdggdz } t | j d t | j d t t d dd t d gdgdggdz } t | j d t | j d t t d dd t d gdgdggdz } t | j d t | j d t t d dd t d dd gdgdggdz } t | j d t | j d t t j d t d d
dd gdgdggdz } t | j d t | j d t t d t d d
dd gdgdggdz } t | j d t | j d t t d t d d
dd gdgdggdz } t | j d t | j d t t d dd t d d
dd gdgdggdz } t | j d t | j d t t d dd t d d
dd gdgdggdz } t | j d t | j d t t d dd t d d
dd gdgdggdz } t | j d t | j d t t d ddd t d ddd t d
d
dd gdgdggdz } t | j d t | j d t t d ddd t d
d
dd gdgdggdz } t | j d t | j d t t d d
dd t d
d
dd gdgdggdz } t | j d t | j d d S )Nr# rD r rF r# rA rD rA rO r rB rB rD rA r r )r r r. int32r itersizer[ rW r s r test_iter_broadcastingr s: q 28A;;'- J<.:JKKAQ $ q !!!Q''!5&:, )9 ; ;AQ &!!!q !!!Q''3&:, )9 ; ;AQ &!!!q !!!Q''3&:, )9 ; ;AQ &!!!q !!!Q''):):1a)@)@A&:, )9 ; ;AQ &!!! VBZZ / /1a 8 89&:, )9 ; ;AR )$$$q 6"::--aA667&:, )9 ; ;AR )$$$q 6!99,,Q1556&:, )9 ; ;AR )$$$q !!!Q''););Aq!)D)DE&:, )9 ; ;AR )$$$q !!!Q''););Aq!)D)DE&:, )9 ; ;AR )$$$q !!!Q''):):1a)C)CD&:, )9 ; ;AR )$$$q !!!Q**F1II,=,=aA,F,F q ))!Q224&:, )9 ; ;A R )$$$q !!!Q**F1II,=,=aA,F,FG&:, )9 ; ;AR )$$$r
""1a++VBZZ-?-?1a-H-HI&:, )9 ; ;AR )$$$$$r c V t j dd dd } t | d gg dgddggg d d gd
}t |j d j d
t |j d j d t | j d gg dgddggg d d gd
}t |j d j d t |j d j d t | j d gg dgddggdg d d gd
}t |j d j d t |j d j d t t t | d gg dgddggg d d gd t j t j d d d gd }d S )Nr# i2r+ rD rA rF writeonlyallocate)r rO N)rE rE rB op_axes itershaperO rC )r r rD )rA rD rB )r r rD rm )rd r r )rD r# r )rE rO rB r r ) r. r rW r r operandsr[ stridesrX r
ValueErroronesr s r test_iter_itershaper _ s
!4 ((A..A4y"
|k:-F G%1\\4$8&1 3 3 3A A $i000A &
333T{B* Z/H I%1\\4$8&1 3 3 3A A $i000A &
333T{B* Z/H I"%%1\\4$8&1 3 3 3A A $i000A &
333 *fq$i (\K+D E%1\\4$8&02 2 2 2
271::tT*d;;;AAAr c
t t t t d t d gg dggdz t t t t d dd t d gg dggdz t t t t d dd t d dd gg dggdz t t t t d dd t d dd gg dggdz t t t t d ddd t d ddd gg dggdz t t t t d ddd
t d ddd gg dggdz t t d d
dd
t d d
d t d dd gg dgdgd d gg t d
# t $ rm} t
| }t | d dk d|z t | d dk d|z Y d } ~ nd } ~ ww xY w t t d dd t d gg dgdggdd
gdt j
ggd t d
# t $ r} t
| }t | d dk d|z t | d dk d|z t | d dk d|z Y d } ~ nd } ~ ww xY w t t j d t j d gg d d gdgg t d
# t $ rn} t
| }t | d dk d |z t | d dk d |z Y d } ~ d S d } ~ ww xY w)!NrD rA rF r# r rB $ r r rO r no_broadcastz$Should have raised a broadcast errorz(2,3)r z0Message "%s" doesn't contain operand shape (2,3)z(1,2,3)z4Message "%s" doesn't contain broadcast shape (1,2,3))rB rA r z (2,3)->(2,3)z7Message "%s" doesn't contain operand shape (2,3)->(2,3)z(2,)->(2,newaxis)zDMessage "%s" doesn't contain remapped operand shape(2,)->(2,newaxis)z(4,3)z6Message "%s" doesn't contain itershape parameter (4,3)rD rO rO r z(2,1,1)z2Message "%s" doesn't contain operand shape (2,1,1)z(2,1,2)z8Message "%s" doesn't contain the broadcast shape (2,1,2)) r
r r r rW AssertionErrorstrr findr. newaxiszeros)emsgs r test_iter_broadcasting_errorsr ~ s *fvayy&))&<* q(* * * *fAYY&&q!,,fQii8* q(* * * *fAYY&&q!,,fQii.?.?1.E.EF* q(* * * *fAYY&&q!,,fQii.?.?1.E.EF* q(* * * *fBZZ''1a00&**2D2DQ12M2MN* q(* * * *fAYY&&q!Q//1C1CAq!1L1LM* q(* * *
Oq !!!Q**q !!!Q''q !!!Q'') zl[.,IJ L L L
CDDD O O O!ff!!Q&CcI K K K ##q(G#M O O O O O O O O
OQq !!!Q''3 zl+ABJ0 ! ! ! !
CDDD
Q
Q
Q!ff ((A- F L N N N,--2$'*+ , , , !!Q&ICO Q Q Q Q Q Q Q Q
Q S##RXd^^4 n-
|< > > > CDDD S S S!ff##q( AC G I I I ##q(KcQ S S S S S S S S S
SsN .BJ< <
L3A#L..L37A'N
P?(B
P::P?A R
TA#TTc t d } t t t g g g t t t | gdz g dggdz t t t | gdgdgg t t t | gg ddgg t t t | gg dggd t t t | gg dggd t t t | gd z g dggd
z t t t | d d gdgg t t t | d
dgdgg t t t | d
d gdgg t t t | d
d gdgg t t t | g g g t t t | g ddgg t t t | g ddgg t t t | g ddgg t t t | g g dg t t t dg dgg t t t dg dgg t t t t j d g dgg t t t t j d g dgg d| j _ t t t | g dgg t t t | g dgg d| j _ t t d g dgg }t t d | t t d | d }d }d }d }t t d d
g }t t || t t || t t || t t || t t d d g }t t || t t || t t || t t t t j g d S ) Nr# d rF zbad flagGrc noonr) rA rD r r r r r r% rF r r% ?rO FTc | j S r r r s r z(test_iter_flags_errors.. q} r c | j S r r r s r r z(test_iter_flags_errors.. qw r c d| _ d S )Nr r r s r assign_multi_indexz2test_iter_flags_errors..assign_multi_index s
r c d| _ d S Nr r r s r assign_indexz,test_iter_flags_errors..assign_index s
r c d| _ d S r )r r s r assign_iterindexz0test_iter_flags_errors..assign_iterindex s
r c d| _ d S )Nr ) iterranger s r assign_iterrangez0test_iter_flags_errors..assign_iterrange s
r buffered)
r r
r r TypeErrorr. r rz writeabler )r4 r r r r r s r test_iter_flags_errorsr s@ q A *fb"b111*fqc#grZL>#3EFFF*fqcJ<:, HHH*fqc2Z0H/IJJJ*fqc2 ~SIIII*fqc2 ~vNNNN*fqc!eR:, 1ABBB*faI&* 8 8 8 *fa -0J<.B B B*fa ),
|n> > >*fa ),
|n> > > *fabT222*faz;.G-HIII*faz;.G-HIII*fa{K.H-IJJJ*fa;;;<> > > )VS" }o>>>)VS" }o>>>)VRXa[[" }oFFF)VRXa[[" }oFFF AG*fa{m_===*fa{m_=== AGvayy"
|n--A* 4 4a888* . .222 vayy?+,,A* 0!444*lA...* .222* .222vayy:,''A* 0!444*lA...* .222*fbhrll33333r c t j d t j d t j d }}} t | ||gg dg }|5 d|dd< t | g d t |g d t |g d d |d
< t |dd dd g d d d d S # 1 swxY w Y d S ) NrA @r% )rA rA r rD )rA rO rD r rO rD r rO )r. r r r )r4 bcr s r test_iter_slicer s illBIaLL")B--!qA1ay"{m,,A
& &!A#Q """Q """Q """!QqsVaW%%%
& & & & & & & & & & & & & & & & & &s A B>>CCc f t j dd ddd j } t j | g ddggd t j d
g }|5 d|j d d
< d|j d d
< d d d n # 1 swxY w Y t | d t j | g ddggd t j d
g }|5 |j d dd }d|d
< d|j d d
< d d d n # 1 swxY w Y t | d d }d }d S )Nr f8r+ rD rA rB r% r& same_kindr$ r( r . rE rO i.)r. r rW rX r r, r r )r4 r8 rK s r test_iter_assign_mappingr ! s
"D!!!))!Q224A 1bK 89 *rx~~6F
H
H
HB ! ! As As! ! ! ! ! ! ! ! ! ! ! ! ! ! ! B 1bK 89 *rx~~6F
H
H
HB $ $ KN2a4 # # As$ $ $ $ $ $ $ $ $ $ $ $ $ $ $ E A
BBBs$ "!BBB+DDDc t j dd } | }t | j j |j j k t |g ddggdt j d g }|5 t |j d j | j j t |j
d j j | j j t |j
d | d |j
d d d < d d d n # 1 swxY w Y t |d gdz ~t j dd } | }t | j j |j j k t |g g d
gd 5 }t |j d j | j j t |j
d j j | j j t |j
d | d |j
d d d < d |j
d d d < d d d n # 1 swxY w Y t |d gdz t j d
d dd } d| _ t j dd | d d < t | j j
t | g dgg }t |j
d j j
t |j
d | t | g g dg 5 }t |j
d j j
t |j
d | d|j
d d d < d d d n # 1 swxY w Y t | dgdz t d } t | d d g dgg }t |j
d j j t |j
d | d d t | d d d ddgddggd }t |d j j t |d | d d d d S )Nr# r$ r+ r% r& equivr( r rD )r% r& nbor i90 ) i1rO rF )r% r& alignedrA r r
r contigr
buffersize)r. r byteswapr/ r r, byteorderr r dtypesr r rz r!
contiguous)r4 aur s r test_iter_nbo_align_contigr* 4 s !4 A
" " $ $B AG !3 3444r2n56 '#%8D>>"2 4 4 4A
QXa[*AG,=>>>QZ](2AG4EFFFQZ]A&&&
1
aaa aSU
!4 A
" " $ $B AG !3 3444 B===> '
)
)
) ,-QXa[*AG,=>>>QZ](2AG4EFFFQZ]A&&&
1
aaa
1
aaa aSU &&&qrr*AAG
9Qd # # #AaaaD q"
|n%%A
1
#+ +,,,A """ 2@@@A B B a
1
#+,,,QZ]A&&&
1
aaa
QCE r
Aq!ubJ<.))A AJqM *+++A "1"&&&q1v
O4$h/0#% ' ' 'A AaDJ !"""1q1v s9 BD##D'*D'f4r( r rB rO rF r3 safer )` r rE r )rB 0 g )r i4r r' )rB rO rD rA ) r. r rW r r, r r rY r rX r r s r test_iter_array_castr6 r s !4 ((A..Aq" }o"(4..1ABBBA
: :QZ]A&&&QZ]("(4..999: : : : : : : : : : : : : : :
!5!!!))!Q//A 2n56x'
)
)
) ;,-QZ]A&&&QZ]("(5//::: ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; "D!!!))!Q22;;AqAAAq"
F+,x~~& ( ( (A A """A $bhtnn555 A & 444 $$B$qqq$$B$,Aq"
F+,x~~& ( ( (A A """A $bhtnn555A & 444 "D!!!))!Q224A 2>
* + x~~&
(
(
( % ,-QZ]A&&&QZ]("(4..999QZ]*K888!&
1
g '
e#$$$ % % % % % % % % % % % % % % % 7U###
!4 2&A 2>
* + x~~&
(
(
( % ,-QZ]("(4..999 QZ]*D111$99
1
aaa% % % % % % % % % % % % % % % III sL AB//B36B3AE((E,/E,
BO00O47O4A'S
SSc f t t t t dd g dggt j d g t t t t dd g ddggdt j d g t t t t dd g ddggd
t j d g t t t t dd g d d ggdt j d g t t t t dd g d d ggd
t j d g t t t t dd
g ddggdt j d g t t t t dd g dd ggdt j d g t t t t dd g dd ggdt j d g t t t t dd g ddggdt j d g t t t t dd g d d ggdt j d g d S )NrD r$ r+ rF r r- r3 nor( r r r& r, r- r% r. r r3 )r
r r r r. r, rH r r test_iter_array_cast_errorsr9 s )VVAT%:%:%:B 28D>>*:< < < < )VVAT%:%:%:B f%& 8D>>*, , , , )VVAT%:%:%:B f%& 8D>>*, , , , )VVAT%:%:%:B ~./ 8D>>*, , , , )VVAT%:%:%:B ~./ 8D>>*, , , ,
)VVAU%;%;%;R f%& 8E??+- - - - )VVAT%:%:%:B ~./ 8D>>*, , , , )VVAT%:%:%:B ~./ 8D>>*, , , ,
)VVAT%:%:%:B f%&# 8D>>*, , , , )VVAT%:%:%:B ~./# 8D>>*, , , , , ,r c ( t t j d g dggt j d g } t | j d t j d t | j j t j d t | j d t t j d g ddggdt j d g } t | j d t j d t | j j t j d t | j d t t j d g ddggd
t j d g } t | j d t j d t | j j t j d t | j d t t j d g ddggd t j d
g } t | j d t j d
t | j j t j d
t | j d t dg dggt j d g } t | d j t j d t | d d d S )N @rF r$ r- r r3 r. r r( r r r' r3 rA )r r. float32r, r r' valuefloat64r s r test_iter_scalar_castr? s rz#j\N !x~~ . 0 0 0A!bhtnn--- ///# rz# &)* " !x~~ . 0 0 0A !bhtnn--- ///# rz# &)* ' !x~~ . 0 0 0A !bhtnn--- ///# rz# &)* $ !x~~ . 0 0 0A !bhtnn--- ///! q"
|n$ 0@AAAA1RXd^^,,,1rr c t t t t j d g dggt j d g t t t dg dggt j d g t t t t j d g dggd t j d g
t t t t j d g dggd t j d g
d S )
NrD r% r r- r; r$ gDcLrF r. r( r r3 )r
r r r. r< r, r> rH r r test_iter_scalar_cast_errorsrA s )VRZ]]B BHTNN+;= = = =)VS" BHTNN+;= = = = )VRZ%5%5r 8D>>*, , , ,
)VRZ]]B # 8D>>*, , , , , ,r c ddd} t j g dd | d gd }t rt j | }t t t | t r"t t j | | t |dgdg }d |D }t t j |d | d gdz \ }}}t r"t t j | | t | d
d
j
dd gdgd
}t |j d |D }t t j |d | d
d
d
d gdz \ }}}t r"t t j | | t | d
d
j
dd gdgd
}|5 |D ]}d |d< d gdz \ }}}d d d n # 1 swxY w Y t r(t t j | |dz
k t |t j d gdz d d S )NrA dr4 r r5 Or+ refs_okrF c g | ]
}|d S rH rH rJ x_s r rL z1test_iter_object_arrays_basic..# rBrF r rD r
rb rc c g | ]
}|d S rH rH rI s r rL z1test_iter_object_arrays_basic.., rK r rm r% .rO rB )r. r r r0 r1 r
r r r rW rX r r2 ri )objr4 rcr valsrK s r test_iter_object_arrays_basicrP s S//C
)))T3-S999A "
_S
!
! )VQ''' /S_S))2...q9+
|,,A Q D$c***A...JD!Q /S_S))2...qyyA 9j"9# C 1 1 1A A Q D$c***AIIaOO,A,A,A,L,LMMMJD!Q /S_S))2...qyyA 9j"9$
S 2 2 2A
AAcFFVAX
a .$$1,---BHdVAXS11122222s 6HH Hc X t j dd } t | ddgdgdd }|5 |D ]}|d
xx d z
cc< d d d n # 1 swxY w Y t | t j d d z t j dd } t | ddgdgdd }|5 |D ]}|d
xx d z
cc< d d d n # 1 swxY w Y t | t j d d z t j d d
dg } | d } t j d | d d < t | ddgdgdd }|5 |D ]}|d
xx d z
cc< d d d n # 1 swxY w Y t | t j d d z t j d d
dg } | d } t j d dz | d d < t | ddgdgdd }|5 |d d }t
rt
j | }|D ]}|d
xx d z
cc< d d d n # 1 swxY w Y t
r(t t
j | |d z
k t | t j d dz d S )Nr# rE r+ rF r
r% r' r3 r( .rO r )pr r4 rE r4 )r4 r3 iHr rH iI) r. r r r r r r0 r1 r )r4 r rK obrN s r #test_iter_object_arrays_conversionsrU < s
!3 Aq9j)K= $ 6 6 6A
A
cFFFaKFFFF BIaLLN###
!4 Aq9j)K= $ 5 5 5A
A
cFFFaKFFFF BIaLLN### k: 6777A #A
9Q<E>??I IIc > t t dgd t dgd gdgddggd z d
} t | j d t j d t | j d t j d t t dgd
t dgd gdgddggd z d
} t | j d t j d t | j d t j d t t dgd t dd gdgddggd z d } t | j d t j d t | j d t j d t t dgd t dd
gdgddggd z d
} t | j d t j d t | j d t j d t t dgd t dd
gdgddggd z d
} t | j d t j d t | j d t j d t t dgd t dd
t dgd t dgd gdgddggdz d
} t | j d t j d t | j d t j d t | j d t j d t | j d t j d t | j d t t dgd
d t dgd gg ddgddgdggd
} t | j d t j d
t | j d t j d
t | j d t j d t t dgd
d t dgd gdgddgddgdggd
} t | j d t j d t | j d t j d t | j d t j d d S )NrA r$ r+ r r common_dtyperF r3 rD r. r rO r3 r u4i8 @c8r rB c16)rA rY r[ r r r )r r r r' r. r, r= r s r test_iter_common_dtyper^ i sL qc&&&qc(>(>(>?#$ &)*1, " $ $ $A !bhtnn---!bhtnn---qc&&&qc(>(>(>?#$ &)*1, " $ $ $A !bhtnn---!bhtnn---qc&&&at(<(<(<=#$ &)*1, ' ) ) )A !bhtnn---!bhtnn---qc&&&at(<(<(<=#$ &)*1, " $ $ $A !bhtnn---!bhtnn---qc&&&c(>(>(>?#$ &)*1, " $ $ $A !bhtnn---!bhtnn---qc&&&c(>(>(>t4(((%4*@*@*@B#$ &)*1, " $ $ $A
!bhuoo...!bhuoo...!bhuoo...!bhuoo.../*** qc&&&eRD.F.F.FG &)!:.!]$ # $ $ $A
!bhtnn---!bhtnn---!bhuoo...qc&&&eRD.F.F.FG#$ &)!:.!]$ # $ $ $A !bhuoo...!bhuoo...!bhuoo.....r c dD ]\} t d }t |gdg| gg }|5 t |j d |u d d d n # 1 swxY w Y ]t d }|dd }|d d }t ||gdgdgdgg 5 }t t j |j d d d n # 1 swxY w Y t d }|}|}t ||gdgdd gdd gg }|5 t |j d |u o|j d |u d d d n # 1 swxY w Y t ||gdgdgdgg 5 }t |j d |u o t j |j d | d d d n # 1 swxY w Y t d }|d d d
}|dd d
}t ||gdgdgd gg }t |j d |u o|j d |u t d t j
}|dd }| t j d d }t ||gdgdgd gg 5 }t t j |j d d d n # 1 swxY w Y dD ]} t j ddg }|}|j
}|}t |||gdgdgdg| gg 5 }|j \ }}}t t j || t t j || d d d n # 1 swxY w Y t j ddg }|}|j
}|}t |||gdgdgdgdgg }|j \ }}}t ||u t ||u t ||u t j ddg }|}t j ddg }|j
}t |||gdgdgd gdgg }|j \ }}}t ||u t ||u t ||u t j d }|d d }|dd }|d d }t |||gdgdgd gd gg }|j \ }}}t ||u t ||u t ||u d S )Nr r copy_if_overlapr rO rE rF r% overlap_assume_elementwiserD r rB r+ rA )r r% r r# ) r r r r r.
shares_memoryint8viewr r rX ) flagr4 r rK r r a2b2c2s r test_iter_copy_if_overlapri s 7 ( ( 2JJ A3*+tfX 6 6
( ( AJqMQ& ' ' ' ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( r
A !""A #2#A A*+zl[M-J K K 3qB$aj112223 3 3 3 3 3 3 3 3 3 3 3 3 3 3 r
A A A1v)*j:V-W.9;W-X-Z [ [A
; ;
1
"9qz!}'9:::; ; ; ; ; ; ; ; ; ; ; ; ; ; ; A*+zl[M-J K K Oq
1
"M2+;AJqM1+M+M'MNNNO O O O O O O O O O O O O O O r
A ##A#A !$Q$A1v)*j\K=,IJJA AJqMQ 51:a=A#5666 q A !""A rx! A A*+zl[M-J K K 3qB$aj112223 3 3 3 3 3 3 3 3 3 3 3 3 3 3 + 2 2 GRH
C
Q1I 1 2 L:,79 9 2<=JBB (R000 1 1 1 (R000 1 1 1 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 RA A A A1ay,- zlZL9 ; ;AJBB AG AG AG RA A
RA A1ay,- {mj\: < F((F,/F,?"J--J14J18AMM M c t d dd } t | | j gg dggdz ddgddgg }t t d |D t d ddd
} t | j | gg dggdz g d d g }t t d |D t dd
ddd } t dd }t | |gg dggdz d g dg }t
d |D | | ddd z t dd
}t | |gg dggdz d g dg }t
d |D | | ddd z t dd }t | |gg dggdz d t j t j dgg }t
d |D | | ddd z t d ddd
} t d ddd
}t | |gdgdggdz g dg dg }t
|j
d t d dd
} t d d
d }t | |gdgdggdz ddgddgg }t
|j
d d S ) Nr# rD rA rF r rO r c g | ] \ }}||k S rH rH rJ rK ys r rL z%test_iter_op_axes.. (((FQa(((r r rB )rD rO r c g | ] \ }}||k S rH rH rm s r rL z%test_iter_op_axes.. ro r r? r rE rE c g | ]
\ }}||z S rH rH rm s r rL z%test_iter_op_axes.. &&&&1a!A#&&&r )rE r rE c g | ]
\ }}||z S rH rH rm s r rL z%test_iter_op_axes.. rt r c g | ]
\ }}||z S rH rH rm s r rL z%test_iter_op_axes.. rt r ( r )r rO rE rE )rE rE r rO )rD rA r? rD r rE )rA r? ) r rW r rX r r r ri r. r r[ )r4 r r s r test_iter_op_axesry s q !Q A13xzl^A -AA7GHHHA C((a((( ) )***r
1a##AQxzl^A - 47HIIIA C((a((( ) )*** q"
aA&&Aq! A1vrZL>!+dKKK5HIIIA&&A&&&199Q1+=+=)=(D(D(F(FGGGq! A1vrZL>!+dKKK5HIIIA&&A&&&199Q1+=+=)=(D(D(F(FGGGq! A1vrZL>!+%)BJ
A+F$G I I IA&&A&&&199Q1+=+=)=(D(D(F(FGGG r
1a##Ar
1a##A1v
* q(8%3^^^^^$D F F FA,''' r
1a Ar
1a A1v
* q(8&'Wr1g$6 8 8 8A&!!!!!r c
t d dd } t t t | | gg dggdz dgdgdgg t t t | | gg dggdz ddgddgg t t t | | gg dggdz ddgddgg t t t | | gg dggdz ddgddgg t t t | | gg dggdz ddgddgg t t t | | gg dggdz ddgg d g t t t | | gg dggdz ddgddgg d S )
Nr# rD rA rF r rO rk rE r )r rW r
r r r4 s r test_iter_op_axes_errorsr| ) s q !Q A*fq!fbJ<.2B./S1#sO= = = = *fq!fbJ<.2B./VaV,<> > > >*fq!fbJ<.2B./VaW,=? ? ? ? *fq!fbJ<.2B./VaV,<> > > >*fq!fbJ<.2B./VaV,<> > > > *fq!fbJ<.2B./VYYY,?A A A A *fq!fbJ<.2B./VaV,<> > > > > >r c t d ddd } t | }| }t d |D d |D d|_ | }t d |D d |D t | d d
gd d }| }t d
|D d |D d|_ | }t d |D d |D d|_ | }t d |D d |D d|_ t | t | | }t d |D d |D t | d gd ddd 5 }| }d d d n # 1 swxY w Y t d |D | d t dd ddd } t | d gd dd d 5 }| }d d d n # 1 swxY w Y t d |D | d d S )!Nr rD rA rB c g | ]
}|d S rH rH rI s r rL z"test_iter_copy..J ###A!B%###r c g | ]
}|d S rH rH rI s r rL z"test_iter_copy..J %7%7%7ae%7%7%7r c g | ]
}|d S rH rH rI s r rL z"test_iter_copy..N r r c g | ]
}|d S rH rH rI s r rL z"test_iter_copy..N r r r
rangedrm rd r$ c g | ]
}|d S rH rH rI s r rL z"test_iter_copy..S r r c g | ]
}|d S rH rH rI s r rL z"test_iter_copy..S r r c g | ]
}|d S rH rH rI s r rL z"test_iter_copy..W r r c g | ]
}|d S rH rH rI s r rL z"test_iter_copy..W r r rA r c g | ]
}|d S rH rH rI s r rL z"test_iter_copy..[ r r c g | ]
}|d S rH rH rI s r rL z"test_iter_copy..[ r r )rD c g | ]
}|d S rH rH rI s r rL z"test_iter_copy..a r r c g | ]
}|d S rH rH rI s r rL z"test_iter_copy..a r r r' r r? )rd r) r* r$ c g | ]
}|d S rH rH rI s r rL z"test_iter_copy..g r r rc zf8c g | ]
}|d S rH rH rI s r rL z"test_iter_copy..m r r ) r rW r r3 r r r nextri )r4 r js r test_iter_copyr C so r
1a##A q A A#####%7%7Q%7%7%7888AK A#####%7%7Q%7%7%7888 q:x(BBBA A#####%7%7Q%7%7%7888AK A#####%7%7Q%7%7%7888AK A#####%7%7Q%7%7%7888AKGGGGGG A#####%7%7Q%7%7%7888
Ji,d,S17,>d,(3)f,O,i1rE z>i,>i,S17,>d,>U3,(3)d,i1,Or r+ r r' r )
r. r, r r r3 r r r namesbase) in_dtype out_dtyper r8 r res1res2r fields r test_iter_copy_casts_structuredr s x#rx~~. rx(C D DEG H HH3
. )E F FGI J JI
'$h
'
'
'C C6CCC '[(
<
<
!,bgav.F.F.FGGG3s8D>276+E+E+EFFF3x~#v----3s8D>!,bgav.F.F.FGGG3s8D>!,bgav.F.F.FGGG3s8D>!,bgas.C.C.CDDDD
E Er c . t d } t | d gg dgddggd t j d g }t |j d j | j t |j d j t j d d S )Nr# rF r r r$ r- rO )r r r. r, r r r[ r s r test_iter_allocate_output_simpler s q A4y"
|k:-F G#'$ "8 : : :AA $ag...A $bhtnn55555r c d t d } t | d gddgdgddgg }|5 d|j d d d < | |D ]$}|d dxx |d d z
cc< %t |j d | dz d d d d S # 1 swxY w Y d S )
Nr# r
delay_bufallocrF r r% rO .r )r r r resetr r4 r rK s r ,test_iter_allocate_output_buffered_readwriter s q A4y:'78$
K'@A C CA
) )
1
aaa # #A
aDIII1c "IIIIQZ]AaC((( ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) )s A,B%%B),B)c
t dd dd } t | d gg dgddggd t j d g
}t |j d j | j t |j d j | j t |j d j t j d t d d ddd
j } t | d gg dgddggd t j d g
}t |j d j | j t |j d j | j t |j d j t j d t d d ddd
dd } t | d gg dgddggdd t j d g }t |j d j | j t |j d j d t |j d j t j d d S )Nr# r3 r+ rD rA rF r r r$ r- rO r rB r rb )rd r* )r0 r1 rB ) r rW r r. r, r r r[ r rX rY r s r !test_iter_allocate_output_itorderr s> q %%a++A4y"
|k:-F G#'$ "8 : : :AA $ag...A & 222A $bhtnn555r &&q!Q//1A4y"
|k:-F G#'$ "8 : : :AA $ag...A & 222A $bhtnn555r &&q!Q//88A>>A4y"$ Z'@A !#'$ "8 : : :A A $ag...A & 444A $bhtnn55555r c t dd ddd } t d | gg ddgd ggt j d
d gg d d g }t |j d
j d t |j d
j d t |j d
j t j d
d S )Nr r3 r+ rD rA rB r r rF rX r r* r r r )rB r2 r1 ) r rW r r. r, r r r[ r r s r test_iter_allocate_output_opaxesr s r &&q!Q//Aay" Z 8:, G#%8D>>4"8!*D 1 3 3 3A A $i000A & 444A $bhtnn55555r c 8 t t dgd t dgd d gg dggdz dd ggz } t | j d t j d t t dgd
t dgd d gg dggdz dd ggz } t | j d t j d t t dgd t dd d gg dggdz dd ggz } t | j d t j d t t dgd t dd
d gg dggdz dd ggz } t | j d t j d t t dgd t d d
d gg dggdz dd ggz } t | j d t j d
d S )NrA r$ r+ r r rF rD r r r3 rX rY rZ )r r r r' r. r, r s r )test_iter_allocate_output_types_promotionr s9 qc&&&qc(>(>(>Er \N1${J&?%@@ B BA!bhtnn---qc&&&qc(>(>(>Er \N1${J&?%@@ B BA!bhtnn---qc&&&at(<(<((>(>Er \N1${J&?%@@ B BA!bhtnn-----r c t dgd } t | d gg dgddgg }t |j d |j d t | | d gg dgdgddgg }t |j d |j d k t |j d d
|j d d S ) NrA rX r+ rF r r r rO rD =)r r/ r r r' r r s r *test_iter_allocate_output_types_byte_orderr $ s qc ++--A4y" \K#<= ? ?A!ahqk***1d|R \J<+z1JK M MA AHQK18A; &'''!))#.. <<<<.>? z*+
|nQ.>> @ @AA $bh|&<&<===A #Q'''''r c G d dt j } t j ddgddgg | }t j d dd j }t ||d gg dgdgdd gg }t t | t |j
d t t | t |j
d u t |j
d j d
t ||d gg dgdgg d g }t t | t |j
d t t | t |j
d u t |j
d j d
d S ) Nc e Zd ZdZdS )4test_iter_allocate_output_subtype..MyNDArray N)__name__
__module__ __qualname____array_priority__rH r r MyNDArrayr < s r r rO rD rA rB rF r r rD rD )r r
no_subtype)
r. ndarrayr rd r rW rX r r r r r r[ )r r4 r r s r !test_iter_allocate_output_subtyper : s BJ 1a&1a&!""'' 22A
! Q""$A1d|R zl[*,EF H HAa$qz!}--... DGG4
1
.. .///A $f--- 1d|R zl7779 : :A a$qz!}--... DGG4
1
.. .///A $f-----r c , t d } t t t | d gg dgddgg t t t | d gg dgddgg t t t | d gdgddg t t t d d gg ddgddggd t j d g t d d
d d
d } t t t | d gg dgddggd t j d gd dt
j dgg t t t | d gg dgddggd t j d gd g dg t t t | d gg dgddggd t j d gd g dg t d d
d d
d } t t t | d gdgdgddggd t j d gd dt
j d gg d S )Nr# r r rF r
r% r$ r- r r3 r+ rD rA rB r rO r )r rA rO )r rD rO r reduce_ok) r r
r r r r. r, rW r r{ s r test_iter_allocate_output_errorsr P s q A)VaY% j(ABD D D *fq$i$
J'?@B B B *fq$i* -7,EG G G )VdD\2%z2%z24#'$ "8: : : :
r &&q!Q//A*fq$i$ Z'@A#'$ "8!%2:q'9 :< < < <
*fq$i$ Z'@A#'$ "8!%yyy 13 3 3 3
*fq$i$ Z'@A#'$ "8!%||| 46 6 6 6 r &&q!Q//A*fq$i+ $ Z'@A#'$ "8!%2:q'9 :< < < < < . s ^^^Aa^^^r c g | ]}|S rH rH rI s r rL z;test_iter_remove_multi_index_inner_loop.. s NNN1QNNNr rO c | j S r r[ r s r r z9test_iter_remove_multi_index_inner_loop.. r r r )r rW r r r r[ itviewsremove_multi_indexr
r r r rV enable_external_loopr= )r4 r beforeafters r 'test_iter_remove_multi_index_inner_loopr s r
1a##Aq=/""A)$$$1 #Y/// ^^^^F NNNNE * . .2221 #U+++ GGIIIR 1UWW%%% R 1U###&**%%%%%r c
d} t d ddd }g dgfD ]&}t ||| }t t | t t
d d|_ t t | t t
dd t ||d| }t t | t t
d d|_ t t | t t
dd t |d d d
|d| }t t | t t
d d |_ t t | t t
d d t |d d d
d d d
f |d | }t t | t t
d d
|_ t t | t t
d
d t |d d dd d d
f || }t t | t t
d d|_ t t | t t
dd | d|_ t t | t t
dd (d S )Nr? r rB rA rD r
r# rm r rE r rb
rO ) r rW r r r listrS r r )r$ r4 rz r s r test_iter_iterindexr s Jr
1a##Azl# > > 1e
3 3 3%a(($uRyy//::: %a(($uQ||*<*<=== 1e3: > > >%a(($uRyy//::: %a(($uQ||*<*<=== 1TTrT7E D D D%a(($uRyy//::: %a(($uQ||*<*<=== 1TTrT44R4Z=%sz J J J%a(($uRyy//::: %a(($uR}}*=*=>>> 1SSqS$$B$Y<: > > >%a(($uRyy//::: %a(($uR}}*=*=>>> %a(($uQ||*<*<====7> >r c d} t dd ddd }| d }t |d
gd gd| }t |j d
t d |D | dD ]L}||_ t |j | t d |D ||d |d Mt |d
dgd gdd| }t |j d
t d |D | dD ]L}||_ t |j | t d |D ||d |d Md }t |g dd gdd| }t |j d
t || | dD ]K}||_ t |j | t || ||d |d Ld S )Nr? r r3 r+ rB rA rD rm rc r rF r r r c g | ]
}|d S rH rH rI s r rL z'test_iter_iterrange.. r r )r r )rA r )r? r? )r rx )r r c g | ]
}|d S rH rH rI s r rL z'test_iter_iterrange.. '''ae'''r r rO r
r )rd r* r$ c g | ]
}|d S rH rH rI s r rL z'test_iter_iterrange.. r r c g | ]
}|d S rH rH rI s r rL z'test_iter_iterrange.. r r c h t j g d }| D ]}t j ||f }|S )Nr r+ )r. r concatenate)r valrK s r get_arrayz&test_iter_iterrange..get_array s? hr&&& + +A.#q**CC
r )r r
r )r rW ri r r r )r$ r4 a_fortr rr s r test_iter_iterranger si Jr &&q!Q//A
WW3W
Fq8*zl#% ' ' 'Ag&&&#####V,,,
B < < Q[!$$$''Q'''!QqT ):;;;;q8Z(:,c : 7 7 7Ag&&&#####V,,,
B < < Q[!$$$''Q'''!QqT ):;;;; q999 C : 7 7 7A g&&&1 v&&&
B 6 6 Q[!$$$YYq\\6!A$qt)#455556 6r c
g } | t j dd ddd j | t j dd t j d d
d d }d |_ t j d
d |d d < | | | t j dd dddd j | D ]}dD ]}g }t |ddgg dgdd| }|j
sft |d j |k | |d
| |j
ft t j | | d d S )Nr r] r+ rD rA rB r r$ )A r rO r3 r1 x r? )rO rD rA r? r r r1 r
r )rF r r! rb r )rd r) r$ r rc )r r. r rW rX r/ r% r r, r r r r r3 r r r ri )arraysr4 r$ rO r s r test_iter_bufferingr s
F
MM")B ! ! !!(Aq!1!1!LLNN88::O O O MM")Bd+++,,,
$'''+AAG
9Rt $ $ $AaaaD
MM!
MM")Ct,,,44Q1a@@BCCC
C C7 C CJDq:7 : : :;!$#*&0 2 2 2A
j
! Z/000 AaDIIKK(((
j
--qwwSw/A/A B B B B C C Cr c t j d ddd j } t
| dgg dgddd
}d }|5 |j s%||d < |d z
}| |j %d d d n # 1 swxY w Y t |
d
t j d d S )Nr rD rA rB r
r% r r! r rb r1 )r) rd r$ r rO rc ) r. r rW rX r/ r% r r r r ri r s r test_iter_write_bufferingr s2 "
aA&&(5577@@BBAq:,3334" " " "A
A
* AaD
FA
JJLLL *
s##RYr]]33333s 0-B))B-0B-c t j d } t j dd }t | |gg ddgdd }t |j t t d | t t d
| t t d | d }t t || | t |j t |j d
|5 t |d d d|d< t |dd ddg t d |D t t t d dgdz d d d d S # 1 swxY w Y d S )Nr# rO r$ r+ )r
r r r r% r' r( c | j S r r r s r r z3test_iter_buffering_delayed_alloc..1 r r c | d S r rH r s r r z3test_iter_buffering_delayed_alloc..2 s
qt r c | dd S )Nr rD rH r s r r z3test_iter_buffering_delayed_alloc..3 s q1v r c d| d< d S r rH r s r assign_iterz6test_iter_buffering_delayed_alloc..assign_iter5 s !r r r rD c F g | ] }|d d |d d g S )r rH rO rH rI s r rL z5test_iter_buffering_delayed_alloc..@ s- 666qqtBx1b*666r )
r. r r r has_delayed_bufallocr
r r r r r ziprS )r4 r r r s r !test_iter_buffering_delayed_allocr ' s ! A
!4 A1vQQQ M $ " $ $ $A A "###* 4 4a888*mmQ///*ooq111 *k1---GGIII & &''' %%%
T TQqT1 !QqsVaV$$$66A666SqA3q5=Q=Q8R8RSSS T T T T T T T T T T T T T T T T T Ts 6A???
"BM***7799BBDDAq:/3334& htnn - ! ! !A
A
cFFFaKFFFF Abi"-888899999sH /BBB#EE
E
3HH HK++K/2K/c g d} t j dd dz t ddgdgd |
}d |D }t |d d
d t |d d d t |d d dgdz gdz t |d d d t |d d
d t |d d d t |d d dgdz gdz t |d d d t |d j t j | g d} t j dd ddg dg dgdfd < ddg dg dgdfd< ddg dg dgdfd< t rt j d }t ddgdgd |
}d |D }t |d d
d t |d d d t |d d dgdz gdz t |d d d t |d d
d t |d d d t |d d dgdz gdz t |d d d t |d j t j | d gdz \ }}}t r(t t j d | dg} t j d d g| t ddgdgd d
}t d |D d d!g g d"} t j d#d$g| t t fd% g d"}g d&}t j d'd(g| t ddgdgd |
}t |d j t j | t d) |D t j d'| t j d(| g d S )*Nr4 r$ r rZ )r r\ r rC rE rA r$ r+ ?r
rF rF r' r( c 6 g | ]}t j | S rH r. r rI s r rL z;test_iter_buffered_cast_structured_type.. s # # #ABHQKK # # #r r r4 r r rD rC rO r r rE )r- r- r- )r r r r; r; r; r; c 6 g | ]}| S rH )r3 rI s r rL z;test_iter_buffered_cast_structured_type.. s AFFHH r r* ) @r r3 c g | ]
}|d S rH rH rI s r rL z;test_iter_buffered_cast_structured_type.. s %%%R"R&%%%r r? r r* r+ r, )r2 r test)r r r c 2 t ddgdgdd S )Nr
rF rF r' r3 r( )r r{ s r r z9test_iter_buffered_cast_structured_type.. s- q:y)J< r ))rC u2rS )r r r5 rB r? r# c 6 g | ]}t j | S rH r/ rI s r rL z;test_iter_buffered_cast_structured_type.. s +++2"(2,,+++r ) r. r r r r, r r r0 r1 r r
r )sdtr rO rN rK sdt1sdt2r4 s @r 'test_iter_buffered_cast_structured_typer= s F
E
EC
!4 3&Aq:y)J< $ ! # # #A $ # # # #Da s###a q!!!a # {1}---a s###a s###a q!!!a # {1}---a s###a
... F
E
EC
S!!!A 8# >AaD 8# >AaD 8# >AaD #
_QqT
"
"q:y)J< $ ! # # #A ! a Da s###a q!!!a # {1}---a s###a s###a q!!!a # {1}---a s###a
...JD!Q 0S_QqT**B/// -C
&$ s+++Aq:y)J< $ " $ $ $A %%1%%%1v... 1
0
0C
"K0<< > Xq!fE222
]9
%
% > > J?Z$;ff=M#t}
>
>
>
> > > > > > > > > > > > > > > > > >s A??B B c t j t 5 t j t j dd fdgddg d d d n # 1 swxY w Y t j t j dd fdgd gddg
} t j t 5 | 5 t
| }d|d < d d d n # 1 swxY w Y d d d d S # 1 swxY w Y d S ) Nr4 S1r+ r r' r
)r* r) rz rO r )r* op_flagsr) rz .)r? r@ r r. r r r )r8 bufs r test_buffered_cast_error_pathsrH s z " " 8 8
28Ct,,,.3%":, 8 8 8 88 8 8 8 8 8 8 8 8 8 8 8 8 8 8
BHQc***, (M8J<
Q
Q
QB z " "
r((CCH sG 0AA A+C'.C C'C C'C C''C+.C+zCannot start subprocesszPyPy seems to not hit this.c t j d } t j t j d| gt j d }d|v sJ d S )Na:
import numpy as np
it = np.nditer((np.array(1, dtype="i"),), op_dtypes=["S1"],
op_flags=["writeonly"], casting="unsafe", flags=["buffered"])
buf = next(it)
buf[...] = "a"
del buf, it # Flushing only happens during deallocate right now.
z-cT)stderrtextr )textwrapdedent
subprocess check_outputr0
executableSTDOUT)coder s r )test_buffered_cast_error_paths_unraisablerS s` ?
D
!3>4">)3): G G GC 3 r c dg} dg}t j d| }t j d |d< t |ddgd gd
| }t |d j t j | t
|t t d D ]0\ }}t t j
|d |k 1d
g} dg}t j d| }t j d |d d d d d f< t |ddgdgd
| }|5 t |d j t j | d }|D ]H}t t j
|d |k |d d xx dz
cc< |dz
}I d d d n # 1 swxY w Y t |d t j d ddd dz dg} dg}t j d| }t j d |d d d d d d f< t |ddgdgd
| }|5 t |d j t j | d }|D ]-}t |d | |dxx dz
cc< |dz
}. d d d n # 1 swxY w Y t |d t j d dddd t j d z dz dg} dg}t j d| }t j d |d d d d d d f< t |ddgd gd
| }t |d j t j | d }|D ] }t |d | |dz
} dg} dg}t j d| }t j d |d d d d d d f< t |ddgd gd
| }t |d j t j | d }|D ] }t |d | |dz
} dg} dg}t j d| }t j d dddd |d< t |ddgd gd
| }t |d j t j | d }|D ])}t |d || d |dz
}*dg} dg}t j d| }t j d dd |d< t |ddgd gd
| }t |d j t j | d }|D ]1}t |d || d d d |dz
}2dg} d g}t j d| }t j d dd |d< t |ddgd gd
| }t |d j t j | d }|D ]Q}t |d d d || d t |d dd g d |dz
}Rdg} d g}t j d| }t j d dd |d< t |ddgd gd
| }t |d j t j | d }|D ]W}t |d d || d t |d d || d |dz
}Xd g} d!g}t j d| }t j d ddd |d< t |ddgd gd
| }t |d j t j | d }|D ]}t |d d dd f || d d d d f t |d d ddf || d d d d f t |d dd d f d d g |dz
}d"g} d!g}t j d| }t j d ddd |d< t |ddgd gd
| }t |d j t j | d }|D ]}t |d d dd f || d d d d f t |d d ddf || d d d df t |d dd d f d d g |dz
}d S )#Nr* )r4 r rA rD rD r r+ r# r4 r
rF rF r' r( r )r4 rE r )r4 rE rU r% rD rO )r4 rE r )rO rA rD rD )r4 r$ r )r4 r$ rU H rA )r4 r r )r4 r$ r r )r4 r r )r4 r$ r r )r r r r )r4 r$ r )r4 r )rD rO )r4 r$ )rA rD )r4 r r )
r. r r r r r, r r rS r r rW r )r; r< r4 r rK counts r test_iter_buffered_cast_subarrayrX ! sK
=D " #D
T"""A
Yq\\AcFq:y)J< $ " $ $ $A 1RXd^^,,,4a>>** ) )5qv''((((
D ! "D
T"""AillAcF111a7Oq:y)K= $ " $ $ $A
QqTZ$ 000 A BF1S6U?++ , , ,
cF1IIINIII QJEE 31 --aA66q8999
" "D D
T"""A1 AcF111aA:q:y)K= $ " $ $ $A
QqTZ$ 000 A 3 ' ' '
cFFFaKFFF QJEE 31 --aAq99"',:O:OOPQQRRR
# #D D
T"""A1 AcF111aA:q:y)J< $ " $ $ $A 1RXd^^,,,
E
QsVU###
" "D D
T"""A1 AcF111aA:q:y)J< $ " $ $ $A 1RXd^^,,,
E
QsVU###
" "D " #D
T"""A
Yw
'
'1a
3
3AcFq:y)J< $ " $ $ $A 1RXd^^,,,
E
QsVQuXc]+++
D D
T"""A
Ys^^
#
#Aq
)
)AcFq:y)J< $ " $ $ $A 1RXd^^,,,
E
QsVQuXc]2A2 .///
D D
T"""A
Ys^^
#
#Aq
)
)AcFq:y)J< $ " $ $ $A 1RXd^^,,,
E
QsVBQBZ5# ///QsVABBZ ...
D D
T"""A
Ys^^
#
#Aq
)
)AcFq:y)J< $ " $ $ $A 1RXd^^,,,
E
QsVAY%
...QsVAY%
...
D D
T"""A
Ys^^
#
#Aq!
,
,AcFq:y)J< $ " $ $ $A 1RXd^^,,,
E
QsVBQBE]AeHSM!!!Q$$7888QsVBQBE]AeHSM!!!Q$$7888QsVAaaaC[1a&)))
D D
T"""A
Yu
%
%aA
.
.AcFq:y)J< $ " $ $ $A 1RXd^^,,,
E
QsVBQBE]AeHSM!!!Q$$7888QsVBQBE]AeHSM!!!Q$$7888QsVAaaaC[1a&)))
s& *A;F22F69F6A KK Kc t j d ddd } t j d ddd }t t t
| |gddgdgd ggd
t | |gddgd gd ggd
t j d ddd } t | |gg d
dgd ggd
t j d ddd } t t t
| |gddgdgd ggd
t j d ddd } t t t
| |gddgdgd ggd
d S )Nr# rD rA rO r r
r r% r rb rc rF r
r r rB )r. r rW r
r r rD s r test_iter_buffering_badwritebackr[ s ! Q1%%A
"
aA&&A*fq!f / =;-0 Aq6J0
L;- (
! Q1%%A
Aq6===
MK= )
! Q1%%A*fq!f / =;-0 ! Q1%%A*fq!f / =;-0 r c t j g dt j } t | j t j d t t t | dgdgd t | dgdgd }t |d d
t |d j t j d t j g dt j } t | j t j d t t t | dgdgd t | dgdgd
}t |d d t |d j t j d
d S )N)abcr4 abcdr+ S4r
rF S2r- S6r s abcU4U2U6r] ) r. r bytes_r r, r
r r str_r s r test_iter_buffering_stringrg s^
%%%RY777A"(4..))))VQ zl " " " "q:, ===A1v 1RXd^^,,,
%%%RW555A"(4..))))VQ zl "$ $ $ $q:, ===A1u 1RXd^^,,,,,r c t j d } t | g dd }t |d j | j d S )N )r
growinnerr r? r# r )r. r r r r r s r test_iter_buffering_growinnerrk sQ
"
Aq<<<&' ) ) )A 1AF#####r c
x t j d dd g d} ddg}ddgdd ggt j g}fd
} | D ]k\ }}}t j | d g|| || }|5 d |j d
d< | ||_ |D ]'\ }} | | t j
z
} (|j d
}
d d d n # 1 swxY w Y t d d D ]} t j |d g|| || | } | 5 d | j d
d< | || _ | D ]'\ }
}||
t j
z
}(| j d
}d d d n # 1 swxY w Y t ||
md S )Ni )r
r r r rF )rF )r% r r )r rO rE rr c 3 K t dd D ]f} t | d D ]S}D ]N}| j z |j z j f}t j j d| }dD ] }|||fV
OTgd S )Nir )rA rA rA r )rS r r. lib
stride_tricks
as_strided)xsysr r r skipr4 op_axes_lists r
get_paramsz3test_iter_buffered_reduce_reuse..get_params s x(( 1 1BB))
1
1 + 1 1G "AJQZ LG &.99!YPPC & 1 1 !7D 000001
1
1 1 1r )r rz rF r* r rE . )r rz rF r$ r* )
r. r floatr, r r3 r r r r int_rS r )rz rF r* rw r r ru nditer2a2_inb2_incomp_resbufsizenditer1a1_inb1_inr r4 rv s @@r test_iter_buffered_reduce_reuser s &$v+&A Q Q QE 89H
+i-EFL I
1
1
1
1
1
1 )jll . .Wd)SXXZZ .$+58&/1 1 1 , ,()G R % MMOOO $G ")
/
/ bg... '+H , , , , , , , , , , , , , , , Q~~
.
.Gid (/ux+2iI I IG
+
+,- $S)
$(!&- 3 3NUEU\\"' 2 22EE &r*
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
sH - - - -
. . .s& A#C22C6 9C6 -A#F F #F c
t j d ddd } t j d ddd }t j d dd }t | ||gg dd gdgdgg t t
t | ||gg dgdd gdgg t t
t | ||gg dgdgdd gg d S )
Nr rD rA rB r# rO r rF r )r. r rW r r
r )r4 r r s r test_iter_no_broadcastr 9 s
"
aA&&A
! Q1%%A
"
a##A
Aq!9b
(
L:, () ) ) *fq!Qi <*n!=
|LN N N*fq!Qi <*
N/KLN N N N Nr c > e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d S )
TestIterNestedc
t d ddd }t j |dgddgg \ }fd|D }t |g dg dg t j |ddgdgg \ }fd |D }t |ddgddgd
d gd d
gddgddgg t j |ddgdgg \ }fd|D }t |g dg dg dg dg d S )Nr rD rA r rO c . g | ]}t S rH r rJ _r s r rL z-TestIterNested.test_basic..O ###AQ###r r r# r r r r r c . g | ]}t S rH r r s r rL z-TestIterNested.test_basic..S r r rB r? r# r r r r r c . g | ]}t S rH r r s r rL z-TestIterNested.test_basic..W r r r rD rB rO rA r? r# r r r r r r rW r. nested_itersr selfr4 r rO r s @r
test_basiczTestIterNested.test_basicJ sO 2JJ q!Q ' 'qA3A-001#######T . . .0D0D0DEFFFqAq6A3-001#######TQFQFQFQFQFRHMNNNqAq6A3-001#######TIIIyyy***jjjIJJJJJr c
t d ddd }t j |dgddgg \ }fd|D }t |g dg dg t j |ddgdgg \ }fd |D }t |ddgddgd
d gd d
gddgddgg t j |ddgdgg \ }fd|D }t |g dg dg dg dg t j |dgddggd \ }fd|D }t |g dg dg t j |ddgdggd \ }fd |D }t |ddgd d
gddgddgd
d gddgg t j |ddgdggd \ }fd |D }t |g dg dg dg dg d S ) Nr rD rA r rO c . g | ]}t S rH r r s r rL z/TestIterNested.test_reorder..` r r r r c . g | ]}t S rH r r s r rL z/TestIterNested.test_reorder..d r r rB r? r# r r r r r c . g | ]}t S rH r r s r rL z/TestIterNested.test_reorder..h r r r r r r rb rc c . g | ]}t S rH r r s r rL z/TestIterNested.test_reorder..m r r r )r# r r r r r c . g | ]}t S rH r r s r rL z/TestIterNested.test_reorder..q r r c . g | ]}t S rH r r s r rL z/TestIterNested.test_reorder..u r r r r s @r test_reorderzTestIterNested.test_reorderZ s 2JJ q!Q ' ' qA3A-001#######T . . .0D0D0DEFFFqAq6A3-001#######TQFQFQFQFQFRHMNNNqAq6A3-001#######TIIIyyy***jjjIJJJ qA3A-s;;;1#######T . . .0D0D0DEFFFqAq6A3-s;;;1#######TQFQFQFQFQFRHMNNNqAq6A3-s;;;1#######TIIIzzz999jjjIJJJJJr c
t d ddd d d dd d dd d df }t j |dgddgg \ }fd|D }t |g dg d g t j |ddgdgg \ }fd
|D }t |ddgddgd d gd
dgddgddgg t j |ddgdgg \ }fd|D }t |g dg dg dg dg t j |dgddggd \ }fd|D }t |g dg d g t j |ddgdggd \ }fd |D }t |ddgddgdd
gd d gddgddgg t j |ddgdggd \ }fd |D }t |g d g d g d!g d"g d S )#Nr rD rA rE r rO c . g | ]}t S rH r r s r rL z1TestIterNested.test_flip_axes..~ r r r r c . g | ]}t S rH r r s r rL z1TestIterNested.test_flip_axes.. r r rB r? r# r r r r r c . g | ]}t S rH r r s r rL z1TestIterNested.test_flip_axes.. r r r r r r rb rc c . g | ]}t S rH r r s r rL z1TestIterNested.test_flip_axes.. r r )r r r r r r# r c . g | ]}t S rH r r s r rL z1TestIterNested.test_flip_axes.. r r c . g | ]}t S rH r r s r rL z1TestIterNested.test_flip_axes.. r r )r r r )r r r# )r? rA rO )rB rD r r r s @r test_flip_axesz TestIterNested.test_flip_axesx s 2JJ q!Q ' '"dddDDbD(8 9 qA3A-001#######T . . .0D0D0DEFFFqAq6A3-001#######TQFQFQFQFQFRHMNNNqAq6A3-001#######TIIIyyy***jjjIJJJ qA3A-s;;;1#######T 0 0 02D2D2DEFFFqAq6A3-s;;;1#######TRHq!fq!fq!fq!fq!fMNNNqAq6A3-s;;;1#######TJJJ
IIIyyyIJJJJJr c t d dd }t d dd }t j ||gdgdgg \ }fd|D }t |ddgddgddggddgddgddggg t j ||gdgdgg \ }fd|D }t |ddgddggddgddggddgddggg d S )NrD rO rA r c . g | ]}t S rH r r s r rL z1TestIterNested.test_broadcast.. r r c . g | ]}t S rH r r s r rL z1TestIterNested.test_broadcast.. r r r )r r4 r r rO r s @r test_broadcastz TestIterNested.test_broadcast s, 1II a # # 1II a # #1vaSz221#######TaVaVaV 41v1v1v6NOPPP1vaSz221#######TaVaV ,1v1v.>!Q!Q@PQRRRRRr c t dd dd }t j |dgdggdd gd
\ }t d j t j d
fd |D }t |g d
g dg d }t dd dd }t j |dgdggddgdd
\ }|5 5 t d j t j d
|D ]}D ]}|dxx dz
cc< t |g d
g dg d d d n # 1 swxY w Y d d d n # 1 swxY w Y t |g dg dg t dd dd }t j |dgdggddgdd
\ }t d j t j d
|D ]}D ]}|dxx dz
cc< t |g d
g dg | t |g dg dg d S )Nr# r3 r+ rD rA r rO rF r3 r )rF r* c . g | ]}t S rH r r s r rL z2TestIterNested.test_dtype_copy.. r r r )rA rB r? r$ r% r& r )rF r) r* .r5 r8 )r rW r. r r r, close)r r4 r rO rK rn r s @r test_dtype_copyz TestIterNested.test_dtype_copy sb
1D ! ! ! ) )!Q / /qA3*&0&%9&*, , ,1 QqTZ$ 000#######TIIIyyy1222
1D ! ! ! ) )!Q / /qA3*&1>%B$/&*, , ,1 4 4 4 4 1RXd^^ 4 4 4
AcFFFaKFFFF YYY 2 3 3 3 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 QIII.///
1D ! ! ! ) )!Q / /qA3*&1>%B$/&*, , ,1 QqTZ$ 000 A
#!
QIII./// QIII./////s7 $E('A E E(E E(E E((E,/E,c P t dd dd }t j |dgdggdgd gd
d \ }}t |d j t j d |D ]}|D ]}|d
xx dz
cc< t |g dg dg d S Nr# r$ r+ rD rA r rO r
r% r r )rz rF r) r* .r5 r8 r rW r. r r r, r r4 r r rK rn s r test_dtype_bufferedz"TestIterNested.test_dtype_buffered s
1D ! ! ! ) )!Q / /qA3*#-,&1]$/&* , , ,1
QqTZ$ 000 A
#!
QIII./////r c D t j d ddd }t j |g g dg \ }fd|D }t |g dg t j |g dg g \ }fd|D }t |dgd gdgdgd
gd gd gd
gdgdgdgdgg t j |ddgg d gg \ }}g }|D ]&}D ]!}| d |D "'t |g dg dg dg dg d S )Nr rD rA )rO r rD c . g | ]}t S rH r r s r rL z*TestIterNested.test_0d.. r r r c . g | ]}t S rH r r s r rL z*TestIterNested.test_0d.. r r r rO rB r? r# r r r r r c g | ]}|S rH rH )rJ zs r rL z*TestIterNested.test_0d.. s NNN1QNNNr r r r r )r. r rW r r r )r r4 r rO krK rn r s @r test_0dzTestIterNested.test_0d s IbMM ! !!Q * *q2yyy/221#######T B B BCDDDq999b/221#######TQC!qcA3aS1#sQC!rdUWTXYZZZ/!q!fb1#%6771a , ,A
,
, NNNNN++++
,TIIIyyy***jjjIJJJJJr c t dd dd }t j |dgdggdgd gd
d \ }}|5 |5 t |d j t j d |D ]}|D ]}|d
xx dz
cc< d d d n # 1 swxY w Y d d d n # 1 swxY w Y t |g dg dg d S r r r s r %test_iter_nested_iters_dtype_bufferedz4TestIterNested.test_iter_nested_iters_dtype_buffered s
1D ! ! ! ) )!Q / /qA3*#-,&1]$/&* , , ,1
1RXd^^ 4 4 4
AcFFFaKFFFF
QIII./////s7 B:AB# B:#B' 'B:*B' +B::B>B>N) r r r r r r r r r r r rH r r r r H s K K K K K K< K K K< S S S(0 (0 (0T
0
0
0 K K K"0 0 0 0 0r r c
, t j d } t t t | d gg dgddggdgdgg t j d dd } t t t | d gd
gdgddggdd gddgg d S ) Nr# rF r% r r rE rk rD rA r rO )r. r r
r r rW r{ s r test_iter_reduction_errorr s
! A*fq$i \K#<= S2$K) ) ) ) ! Q""A*fq$i/1B \K#<= Vb"X .0 0 0 0 0 0r c t j d } t | d gdgdgddggdgdgg }|5 d|j d d
< |D ]\ }}|d
xx |z
cc< t |j d j d t |j d t j | d d d n # 1 swxY w Y t j d d d } t | d gdd
gdgddggdd gddgg }|5 d|j d d
< t |d j d t |d j d |D ]:\ }}t t | D ]}||xx || z
cc< ;t |j d j d t |j d t j | d d d n # 1 swxY w Y t j d } t | d gdd
gdgddggd g dg }t | d gg ddgddggd g dgd }|5 |5 |j d
d |j d
d | |D ] }|d d
xx |d z
cc< |D ] }|d d
xx |d z
cc< t |j d |j d t |j d | j d d d n # 1 swxY w Y d d d d S # 1 swxY w Y d S )Nr# r rF r% r r rE rk rO .rD rA r r r )rD rA r? )r rE rO )r r r
r r )r r$ )r. r r r r r sumrW r[ r rS rT r fillr r )r4 r rK rn r it1r; s r test_iter_reductionr
s ! A4y;- \K#<= S2$K ) ) )A
/ /
1
c DAq
cFFFaKFFFFQZ]'+++QZ]BF1II.../ / / / / / / / / / / / / / / ! Q""A4y;8 \K#<= Vb"X . 0 0 0A
/ /
1
cQqTZ&&&QqT\4((( DAq 3q66]]
!!
QZ]'+++QZ]BF1II... / / / / / / / / / / / / / / /$ A
!T[/ : \K#<= !::: . 0 0 0C !T : : : \K#<= !::: .2 ? ? ?C
4 4c 4 4 Q Q Q Q A
aDIII1 IIII A
aDIII1 IIIIS\!_cl1o666S\!_((**AF333 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4s\ A6B88B<?B<CG,,G03G0M$C/M M$ M M$M M$$M(+M(c t j d } t j dd }t | |gddgdgdd ggd
gd gg }|5 t
|d
j t j d t |d
j |j k |D ]\ }}|dxx |z
cc< d d d n # 1 swxY w Y t
|t j | t j d
dd } t j d
d
gd }t | |gg ddgdd ggd
d
gd
d gg }|5 t
|d
j d t
|d
j d |D ]:\ }}t t | D ]}||xx || z
cc< ; d d d n # 1 swxY w Y t
|t j | d
t j d d
z }t j |d gg ddgddggd d
gd d ggd }|5 |j d
d
| t
|d
g d d d d n # 1 swxY w Y t j dt j ddd
d dd
d f d
dd
}t j |j
|j |d< t j d t j
dd }| } |d d dd d d f }t j ||gg d dgdgg }|5 |D ]\ } }| d d d d n # 1 swxY w Y t
|d
d d | d
d d t
|d d d d d S ) Nr# g r r+ r r
rF r% r r rE rk rO .rD rA r r r
r r )axis)r r r
r r r r )rO rD rO rD )r r r rB r r? r1 rZ )r. r r r% r/ r r r, r r rW r[ r rS rT r r r r rc transposer r3 )
r4 r r rK rn r rR r8 y_base y_base_copys
r test_iter_buffering_reductionr D
s ! A
4 ))++88::A1v Z0 \K#78 S2$K ) ) )A
QqTZ$ 000!
ag%&&& DAq
cFFFaKFFFF BF1II
! Q""A
!Qt$$$--//<<>>A1vAAA \K#78 VaW - / / /A
QqTZ&&&QqT\4((( DAq 3q66]]
!!
BF11%%%&&& ! qA At9 H H H\K4 5!Wr2h'
B
* *
AA
RULLL)))* * * * * * * * * * * * * * *
BG$$QqS2a!^4>>q!QGGA
Yqv
&
&qw
/
/AcF
Ys"'
*
*
*
2
21a
8
8F++--Ksss111TzA Aq6=== MJ<0
2
2B
DAq
FF1IIII 1 {14a4 0111!a sK 1A)C''C+.C+ A4H H!$H!
A
K&&K*-K*! P PPc J t j d } t j d }t j | |gg ddgdggd }|5 d |D }d d d n # 1 swxY w Y t |g d t t | | j d S )
N)rD r )rO r r rF r% r? )rz rF r$ c . g | ]\ }}|j d S r r rm s r rL zDtest_iter_buffering_reduction_reuse_reduce_loops..
s" ...41aAGAJ...r )r? rD r? rD )r. r r r r r )r4 r r8 bufsizess r 0test_iter_buffering_reduction_reuse_reduce_loopsr
s
A
A Aq6!K!K!K )l[M :
"
"
"B
/ /..2.../ / / / / / / / / / / / / / /<<<(((X '''''s
A$$A(+A(c
t j d } t j d }t j g dg dg }t j g d }t j g dd }t j g dd }t j g dd }t t t
| |gg d
d gd gg t t t
| |gg d d gd d
gg t t t
| |gg d gg dg t t t
| ||gg d
d gd d
gd d
gg t t t
| |gg d
gd d
gg t t t
| ||gdgd gd
d gd d
gg t j | ||gdgd gd
d gd d
gg t t t
| ||gdgd gd
d gd
d
gg t j | |gdgd
d gd d
ggd d gd t t t j | |gdgd
d gd d
ggd d gd t t t j | |gdgd
d gd d
ggd d gd d S )Nr r TTF)FTFr u1r+ r r$ r% writemaskedrF arraymask)r% r r r r
r r )r. r r r
r r r )r4 r r m2m3mbad1mbad2s r test_iter_writemasked_badinputr
s@
A
A
%%%';';';<==A %%% & &B )))4 ( ( (B HYYYd + + +E HYYYd + + +E *fq!fb!=1J<@B B B *fq!fb -0:{2KLN N N *fq!fb \#L#L#LMO O O *fq!Rj"!=1 +. +.01 1 1 *fq!fb!]Z$=>@ @ @ *fq!Qi+ \!=1 +.01 1 1
Iq!Rj;- \!=1 +.01 1 1
*fq!Rj;- \!=1!;/12 2 2 Iq"g
|!=1 +.0 #Tl ' ) ) ) ) )RYE
ZL!=1 +.0 #Tl ' ) ) ) ) )RYE
ZL!=1 +.0 #Tl ' ) ) ) ) ) )r c : | j n# t $ r Y dS w xY wdS )NTF)r r )iterators r _is_bufferedr
s: tt 5s
r4 r r r+ )& r rD )rB i8 | rA )r rA )r ri r? rE c | j }|d dz }t j |t }g d|z |d< t j | |gg ddgdd gg }|5 |D ]
\ }}d
|d< d d d n # 1 swxY w Y t | t j g d |z | t j | |gd gddgdd gg }d
}|5 |D ]!\ }}d|d< t j || rd}" d d d n # 1 swxY w Y |s)t | t j g d|z | n-t | t j g d|z | d| d< t j | |gd gddgdd ggdd gd }|5 |D ]
\ }}d|d< d d d n # 1 swxY w Y t | t j g d|z | d S )NrE rA r+ r .r% r rF r rO r r
Tr; Fr0 )r; r; rO rZ r' r )rA rA r; )r[ r. emptyboolr r broadcast_tomay_share_memory)r4 r[ repsmskr8 rK r is_buffereds r test_iter_writemaskedr
s7
GE 9>D
(5
%
%
%C"""T)CH
As8R }- k*,
-
-B
DAqAcFF BOIII$4e<<===
As8j\ }- k*,
-
-B K $ $ $ $DAqAcF"1a((
$ # $$ $ $ $ $ $ $ $ $ $ $ $ $ $ $ Q$(>FFGGGG Q
(>?????s6 A33A7:A7 %C;;C?C?F F#&F#mask mask_axesr r rO rB rH c t j d }dg}g d}ddg}|d }n|ddgg}t t 5 t j | |f|||g| d d d d S # 1 swxY w Y d S ) Nr@ r )r r% r r r r rO )rz rF r )r. r r
r r )r r r itflags
mask_flagsa_flagsr s r %test_iter_writemasked_broadcast_errorr # s
(6
CmG777JM*G q!f% z " " # #
4+W
G7L! # # # ## # # # # # # # # # # # # # # # # #s A))A-0A-c t j d d } | }t j ddd t }t j | |gddgdd gd
d ggd d
g }t }t rt j | }|D ] \ }}d|f|d<
~~~t r/t j | |z
t j | k sJ t | | || | | t j d|f| j k sJ ~ t rt j | |k sJ d S d S )Ni' z>i,Or rD )r r
rF r% r rF r z s
)E
!
!&
)
)CxxzzH
9 Q . . 5 5d ; ;D C;Y 7 -0 -/ $c]
,
,
,B I + ** " "
Xy>C Xr L y))E1R5Ed5K5KKKKKsD5z8TE?333 I1i.#)<< < A A C CCC C 3y))U22223 322r c t j t j d } g d}|D ] }t t t
| | d S )NrD )
r= r[ r r r r2 has_multi_index has_indexr' r nopr r r. r r r
AttributeErrordelattrr8 attrss r )test_iter_non_writable_attribute_deletionr ] s\ 271:: B 3 3 3D 6 6ngr155556 6r c t j t j d } g d}|D ] }t t t
| | d S )NrD )r r r r r r s r %test_iter_writable_attribute_deletionr g sT 271:: B > > >D
6 6ngr155556 6r c t j t j d } | d= | dd= d S # t $ r Y d S t $ r t
w xY w)NrA rO rD )r. r r r r r r8 s r test_iter_element_deletionr n sn 271:: B qE qsGGG
s 2
AAc 6 t j g dd fd dg } | D ]\ }}|dz
|d< |dz |d< t | j d ddgddgdd gg t j g dgd fd dgd
gd d g } | D ]\ }}|dz
|d< |dz |d< t | j d ddgddgdd gg t j d
g d } | D ]&\ }}}||z
|d< ||z |d< ||z |d< ||z |d< 't | j d ddgddgg d S )N)rO rA rx )r3 r r- rO r rD rB r )rE r )r* rz r )r rD N)NN)r3 r r r r r r r rx r? )r. r r r )r8 r4 r r s r test_iter_allocated_array_dtypesr
x s
JJJ%$ 1E F F FB 11u!1u!Q 1a&1a&2r(!;<<<
ZZZL$'D,3G% w
A
A
AB 11u!1u!Q 1a&1a&2r(!;<<<
=,H,H,H I I IB 1aa%$a%$a%$a%$Q 1b'B7!344444r c t ddgdgdggdz } t | j d t t | d t | j d t | j d t
t t | | t t | d t
t t | t t j
d dgdggdg } t | j d t t | d
t t j
d dgdggdgd } t | j d t t | d
t
t 5 t t j
d dgdggd d d d n # 1 swxY w Y g d
}t j
dd }t |ddgdgd| } t | }t |d d t |d d t |d dgdz gdz t |d d d S )NrD rA r rF r r rH r? rk rO r r r) r- r$ r+ r
rF r' r( r4 r r rC )r r r r r r r
StopIterationr r. r rT r r )r r: r4 rO s r test_0d_iterr s 1v
* q(899Aa&!!! ###a -q)))GGIIIa&!!!-q))) ry||m_
|nrdKKKAQry||m_
|ntr + + +AQ
z " " L Lry||m_
|nKKKKL L L L L L L L L L L L L L L F
E
EC
D!!!Aq:y)J< $ 5 5 5A 77DcC cA csWQYKM***cC s ()G G!$G!c t t d t j t j dz dd t t d d d df< t t fd G d d } t t t j j
t j | | gd d S ) Nc 4 t j dd d z S )N)ihB rD r$ r+ )r. r rH r r r z*test_object_iter_cleanup.. s RXj%E%E%E%L r r rE c b t j d d d d df d d d d df S )NrE )r. add)r oarrs r r z*test_object_iter_cleanup.. s7 RVDDDbDM3qqq$$B$w<%H%H r c e Zd Zd ZdS )#test_object_iter_cleanup..Tc t d )N Ambiguous)r )r s r __bool__z,test_object_iter_cleanup..T.__bool__ s K(( (r N)r r r r rH r r rX r s# ) ) ) ) )r rX rE r+ ) r
r r. r BUFSIZErW r r r
logical_orreducer )rX r r s @@r test_object_iter_cleanupr s ) L LMMM
)BJO
$
$
,
,R
4
4
;
;C
@
@C ::f DDBK) H H H H HIII) ) ) ) ) ) ) ) )R] 1 XqqssAACCj < < <> > > > >r c t j d dgddgd dgddgg d d d } t j t 5 t j | d d d d S # 1 swxY w Y d S )NrO rE rD )r. r r? r@ r r )r s r test_object_iter_cleanup_reducer s
(T1IBx$RH=
>
>sss
CC y ! !
s s A%%A),A)r )@ rB rD rm )r, rd c t j dt j }t j | dt | }t |t j ddt d S )Nr! r+ r )r r, outrB )r. r intpr r r full)r r# r s r %test_object_iter_cleanup_large_reducer& sY
'$bg
&
&
&C
&6S
9
9
9CsBGD!6 : : :;;;;;r c f t j t j j dz } t j j t j d | fd }t t t ||d d d f f t t t ||d d d f fdg d S )Nr
rO r r ry ) r. iinfor$ maxrp rq rr r r
r r )r r s r test_iter_too_larger* s 8BG D (D
&
)
)"(1++w
E
EC*fsC4L&9::: *fAAAtG %m_> > > > > >r c d d} d}| |z t j t j j k r/|dz
}| |z t j t j j k /ddg|z }g }t | D ];}|d d }d||dz < | t j |
>
>B
#
B /0 RXd^^,<
>
>
>B
; ;
#
j'2z::: ; ; ; ; ; ; ; ; ; ; ; ; ; ; ;
B /0 RXd^^,<
>
>
>B AAcFF
LE,&&&&&s B//B36B35D? 7.add_closeU s
Y1c{B \J<+j1IJL L IQ1 E!QA k!n
r c t j }t j | ||gg dgdgddgg }|5 |D ]\ }}} |||| |j d cd d d S # 1 swxY w Y d S r= )r. r r r )rK rn r# r> r8 r4 r r s r add_contextz*test_close_equivalent..add_context_ s
Y1c{B \J<+j1IJL L
" "
#
# Aqa""""";q> " " " " " " " " " " " " " " " " " "s $A A$'A$r? r r rD r )rS r )r? rA r s r test_close_equivalentrB R s " " " " %((E!HH%%AE!ROO$$$ E!HHeAhh''AE!ROO$$$$$r c " t j t j d } t t | d | t
t t | t
t t | d d S )NrA r r )
r. r r r r r r
r r r5 r s r test_close_raisesrD l sh 29Q<< B$r((A HHJJJ-r****gr:66666r c t j t j d } t t | j d d S )NrA rO )r. r r r
r r r s r test_close_parametersrF s s3 29Q<< B)RXq)))))r c
t j dd } | }t 5 }| t t j |g ddggdt j d g }~t |j
dk sJ d d d d S # 1 swxY w Y d S ) Nr# r$ r+ r% r& r r( rO ) r. r r% r/ r
recordRuntimeWarningr r, rT log)r4 r) r" r8 s r test_warn_nocloserK w s
!4 A
" " $ $B !
>"""
Yr2n = > 'BHTNN3CE E E 37||q ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! !s A%B>>CCr win32z!Errors with Python 3.9 on Windowsr buf_dtype))r rE rE r )i,Or zO,irO stepsr5 c d}t j t t j dz | | }t j |t j | gg dd }t | D ]}t | ~t j |t j | gg dd }t | D ]}| ~dS )a
Checks for reference counting leaks during cleanup. Using explicit
reference counts lead to occasional false positives (at least in parallel
test setups). This test now should still test leaks correctly when
run e.g. with pytest-valgrind or pytest-leaks
i @r; r r' r* rz r) N)
r. r% intr r r r, rS r r )r rM rQ r= r r8 steps r test_partial_iteration_cleanuprV s
E
'#bj3&''
/
/
6
6x
@
@C 328I#6#6"7:::H
N
N
NBe R
328I#6#6"7AAA8
U
U
UBe
r rN rP c R d}t j t t j dz | | }| dk r d |t t j dz < n%d |t t j dz d< t j | }t j |t j | gg dd }t j
t 5 t | t | d d d n # 1 swxY w Y |
t j
t 5 | | d d d n # 1 swxY w Y |t j | k sJ d S ) Nr4 r; rE r r r r' rS )r. r% rT r r r0 r1 r r, r? r@ r r r r )r rM r= r rW r8 s r test_partial_iteration_errorrX s
E
'#bj3&''
/
/
6
6x
@
@C3%)C
S ! !""+/C
S ! !"4( OE " "E 328I#6#6"7:::H
N
N
NB y ! !
R R HHJJJ y ! !
COE** * * * * * *s$ * DD D
)FF Fc d }t j dt j dd dddddf }t j d }t j ||fdd gd
d d gd
gdgg }| | j }| }t | t | k sJ t || D ].\ }}| | sJ /dS )a
Matches the expected output of a debug print with the actual output.
Note that the iterator dump should not be considered stable API,
this test is mainly to ensure the print does not crash.
Currently uses a subprocess to avoid dealing with the C level `printf`s.
a
------ BEGIN ITERATOR DUMP ------
| Iterator Address:
| ItFlags: BUFFER REDUCE REUSE_REDUCE_LOOPS
| NDim: 2
| NOp: 2
| IterSize: 50
| IterStart: 0
| IterEnd: 50
| IterIndex: 0
| Iterator SizeOf:
| BufferData SizeOf:
| AxisData SizeOf:
|
| Perm: 0 1
| DTypes:
| DTypes: dtype('float64') dtype('int32')
| InitDataPtrs:
| BaseOffsets: 0 0
| Operands:
| Operand DTypes: dtype('int64') dtype('float64')
| OpItFlags:
| Flags[0]: READ CAST ALIGNED
| Flags[1]: READ WRITE CAST ALIGNED REDUCE
|
| BufferData:
| BufferSize: 50
| Size: 5
| BufIterEnd: 5
| REDUCE Pos: 0
| REDUCE OuterSize: 10
| REDUCE OuterDim: 1
| Strides: 8 4
| Ptrs:
| REDUCE Outer Strides: 40 0
| REDUCE Outer Ptrs:
| ReadTransferFn:
| ReadTransferData:
| WriteTransferFn:
| WriteTransferData:
| Buffers:
|
| AxisData[0]:
| Shape: 5
| Index: 0
| Strides: 16 8
| Ptrs:
| AxisData[1]:
| Shape: 10
| Index: 0
| Strides: 80 0
| Ptrs:
------- END ITERATOR DUMP -------
r r+ r NrD g @rC r3 r' r r
rF r% )r* r) rz rF )
strip
splitlinesr. r r rW r debug_print
readouterrr# rT r
startswith)capfdr arr1arr2r8 r res_line
expected_lines r test_debug_printrd sQ 5j
k
n 9S ) ) ) 1 1"b 9 9!!!SSqS& AD
9R==D D$rq s4
8 8 8 8 8 8 8 8 8 , , , , , , , , , , , , $-EFF( ( GF( T0 0 06K K K6K K K6K K K60 0 0 $-EFF
)
) GF
) @ @ @X X X6*` *` *`X 0 0 06 6 6:#D #D #DJ 0 0 06 6 6:#D #D #DJ 2 2 2<$ $ $0" " " HD% D% D%L < < <>IS IS ISVK4 K4 K4Z & & & &< < < |< < < |*, *, *,X$ $ $L, , ,&%3 %3 %3N++ ++ ++Z>/ >/ >/@X X Xt$" $" $"L> > >4*9 *9 *9Z ",u"566 r|E':;; :;; + + <; <; 76 +B 7 7 7@ E E E@6 6 6 ) ) )6 6 66 6 6 6. . .& = = = ( ( (. . .,)< )< ) > >D'6 '6 '6RC C C:4 4 4$T T T6
1
1
1 5 5 5<0: 0: 0:dG4 G4 G4T> > >& G$=>> $-JKK LK ?> &e e eN D- - -$$ $ $ 0. 0. 0.f N N N s0 s0 s0 s0 s0 s0 s0 s0j
0
0
094 94 94v>! >! >!@( ( ( @) @) @)F $d # # # +T * * *33Q36 2 "$ / / /!QQQ!QQQ ?
$d # # #CCaC ( ,d + + +CCaC1H 5 "$ / / /!QQQ!TTrT0A B DE E9@ 9@E E9@x &+.
Aw & ' ' '!Q0 & ' ' '. !6 " " "RG, "F # # #b"X. "F # # #T* 1, - -# # - -#" 3 3 3>6 6 66 6 6 5 5 56$! $! $!L> > >, F+++AAAsssAAAI6 F#666qqq##A#qqqyA F+++AAAsssAAAI6;;C@@!B C C< < C C<
>
>
> O O OD?' ?' ?'B% % %47 7 7* * * $-EFF! ! GF! C$RaR(F2Ns|w7N> @ @*k2
))),, -, @ @ 4 $-EFF*k2
+ +
GF
+8N: N: N: N: N:r