yz"test_array_array..Z s rxRZ'H'H'H ) __array__ ?
)r/ )r/ r )r r/ )r/ r/ )
NNNNNNNNNNN)
r r r r r r r r r r )r ) typeobjectr& onesr( r r' r sys getrefcountnanr
bytesstrgetattr __builtins__r hasattr
memoryviewuint8dictr intfloatr$ datarangeMAXDIMSr
ValueErrorfloat32full)tobjones11tndarray old_refcountS2S3S5U2U3U5builtinsdatar nestedis r* test_array_arrayrV sl <x00
\3?8#<#<=== $bj111"&
3335 5 5 :t,,
RZ((((\3?4#8#8999 #RZ0002:...0 0 0 @rz22
#RZ000
CCCC\3?2:#>#>???
5!* B 5!* B 5!* B&
3332:...0 0 0&!!',,,&...4b999&+++RXe__===$b)))272R+@+@+@AAA
3( B 3( B 3( B%rz2222:...0 0 0% &+++%s+++12666%r***BHSYY,?,?@@@#R((("'"B*?*?*?@@@|Z >>H GHe $ $%%%
(:f%%RZ
8
8
8C((())) CINbj ()))
(:f%%
&
&Clll### CINbh &''' bj)))A S6)
a&;
<
<
< > >A!2:...222 #%67 s dE]T3K@ B B BA S6)
1#5
6
6
6 8 8A rx{{'((%--888 KS6)
H H
I
I
I K K M MA!2:...
0K0KLLL F
2:
HV *bh
CCCC 4&2+RZ888bj999; ; ;D6(R-rz:::"&
;;;= = =D6B;-rz:::"&
;;;= = =D6B;-",BJ???26<<<> > > 4&2+RZ888bj999; ; ;D6(R-rz:::"&
;;;= = =D6B;-rz:::"&
;;;= = =D6B;-",BJ???26<<<> > > 3%"*BJ777bj1113 3 3C5'B,bj999
3335 5 5C52:,bj999
3335 5 5C52:,+2:>>>4446 6 6 ,bj999bj999; ; ;7)b.
;;;"&
;;;= = =<.
;;;"&
;;;= = =<.2-RZ@@@26<<<> > > +RZ888bj1113 3 36(R-rz:::
3335 5 5;-rz:::
3335 5 5;-",BJ???4446 6 6 6 6r, r' TFc t dd }| rt j | }t t 5 t j |d d d d d S # 1 swxY w Y d S )Nr r M8r )r r& r' r TypeError)r' rts r* test_array_impossible_castsr[ s
!Q B
Xb\\ y ! ! ! !
4 ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! !s A A#&A#rS r )r r r r c t j 5 t j | }t || j |j j sJ d d d d S # 1 swxY w Y d S r0 )pytestdeprecated_callr& fastCopyAndTransposer Tflagsowndata)rS bs r* test_fastCopyAndTransposerf s
! ! #A & &Q w s 8AA !A c J t j dd dd } | d }t | | t |j t j d t | j |j | j d }t | j | t |j t j d t | j j |j | d }t | | t | |u | dd }t | |u | dd d
}t | | t | |u t |j j
| dd d
}t | | t | |u t |j j | d d
}t | | t |j t j d t t | j dd
| ddd }t | |u G d dt j }t j g dg dgd | } | ddd }t | |u | dd }t | | t t# | | | ddd }t | | t | |u t t# | |u t j dgd } | d }t | | t |j t j d t j dgd } | d }t | | t |j t j d t j d gd } | d }t | | t |j t j d t j d gd } | d }t | | t |j t j d t j d!d d } t% | t j d"d# t j d!d d } t% | t j d$d% t j d!gd d } t% | t j d"d# t j d!gd d } t% | t j d$d% t j d!d } t% | t j d"d# t j d!d } t% | t j d$d% t j d&d } t j | d'd( }t |j dk t j d)gd } t t | j d*d
t j d)d } t t | j d+d
t t | j d S ),Nr] f4r r r i4FcopyF)orderrk Cc8safecastingr )subokrk c e Zd ZdS )$test_array_astype..MyNDArrayN)__name__
__module__ __qualname__r r, r* MyNDArrayru s r, ry )r r r )r r Tsd aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaOSS100daaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaUU100s
aaaaaaaaaaS10
aaaaaaaaaaU10l
~ l!]c s 123456789012345678901234567890S30 123456789012345678901234567890U30u aŀuint32)bufferr
shapei S1U1)r& arangereshapeastyper r
stridesrb r rc f_contiguous c_contiguousr rY ndarrayr' viewr1 r size)rS re ry s r* test_array_astyper s
!4 ((A..A
AA"(4..)))AI&&&
4Aa"(4..)))ai((( AA a
E""A AFOOO
Su--AA a AG !!! Su--AA AFOOO AG !!!
v&&AA"(4..))))QXtV<<<<
QU++A AFOOO
BJ
)))YYY't44499)DDA
T..A AFOOO
E""AAa)$$$
U//AA a DGG9 $%%% (3'''A
AA"(6**+++
'#&&&A
AA"(6**+++ '#&&&A
AA"(5//***
&%%%A
AA"(5//***
/s;;;BB3GGAq"(#4EBBBCCC
/s;;;BB3GGAq"(#35AAABBB
01===DDSIIAq"(#4EBBBCCC
01===DDSIIAq"(#35AAABBB
/s;;;Aq"(#4EBBBCCC
/s;;;Aq"(#35AAABBB
#&&&A
!81555A AFaK
$t$$$A)QXtV<<<<
T"""A)QXtV<<<< )QX&&&&&r, dtr| r c R t j dgt }| | j j t j | d j k sJ t j || d sJ t j || d rJ t j d| d sJ d S )N r r unsaferq same_kindr{ )r& r' r2 r r
itemsizecan_castr arrs r* +test_array_astype_to_string_discovery_emptyr " s
(B4v
&
&
&C ::b>> (BHXXX,>,>,G G G G G ;sB 1 1 111 1{3K88888 8
;sB 1 1 111 111r, )dfS13U32c t j | } t j g | }| d j j | j k sJ d S )Nr V)r& r
r' r r r s r* test_array_astype_to_voidr 0 sL " B
(2R
C ::c?? )R[ 8 8 8 8 8 8r, c r t j g d d } | j dk sJ d S )Nr{ r r V8)r& r' r r
)r s r* test_object_array_astype_to_voidr 6 s?
(2S
!
!
!
(
(
-
-C 9 r, tuintr? r@ c t j dt j }t t j |j | d S )Nr/ r )r& r' complex_r ComplexWarningr )r rS s r* test_array_astype_warningr < s7
2;'''A"AHa00000r, r
out_dtypezS10,S9z?,?c t j g d| }t j g d| }t | | | dS )z
Currently, for `astype` strings are cast to booleans effectively by
calling `bool(int(string)`. This is not consistent (see gh-9875) and
will eventually be deprecated.
)10z10 z0 0r )TTFFN)r& r' r r )r
r r expecteds r* test_string_to_boolean_castr D s\
(3335
A
A
ACx222)DDDHszz),,h77777r, c dD ]Z}t j |g| }t t 5 | | ddd n # 1 swxY w Y [dS )zn
These currently error out, since cast to integers fails, but should not
error out in the future.
)FalseTruer z non-emptyr N)r& r' r rD r )r
r invalidr s r* "test_string_to_boolean_cast_errorsr R s < " "hy...
:
&
& " " JJy ! ! ! " " " " " " " " " " " " " " "" "s AA A str_type scalar_typec l |d } || dk sJ t j |gt | d dk sJ t j | | d dk sJ t j d| }||d< |d dk sJ d S )Ns 1+3jy ? @r r r r )r& r' r2 r zeros)r r valuer s r* test_string_to_complex_castr ` s
K E ;u % % % %
8UG6 * * * 1 1+ > >r Bd J J J J
8E?? ! !+ . .r 2d : : : :
(1K
(
(
(C
CF q6T>>>>>>r, AllFloatc t j d| }d |d< t j | d sJ t j t j d | d sJ t j t j d g| d sJ t j t j d | d sJ d S )Nr r r r )r& r isnanr' r )r
r s r* test_none_to_nan_castr l s
(1E
"
"
"C
CF
8C==
8BHT/// 0 0 444 4
8BHdV5111 2 21 555 5
8BHTNN))%00 1 1" 555 555r, c , t j dd dd } t j | d t | d t j | j d t | d t j g dg d gd
}t j | d | t | g d
g dg t j ddgddgddggd
}t j | j d| t | g dg dg d S )Nr] rh r r r r. @)r r r )r r r ? @where)r r r )r r r r r @)r r r )r r r )r& r r copytor rb r' )rS masks r* test_copyto_fromscalarr w s(
!4 ((A..A IaCIac3C 8YYY *# 6 6 6DIaD!!!!___ooo6777
8aVaVaV,C 8 8 8DIac3d####___ooo677777r, c t j dd dd } t j | g dg dg t | g dg dg t j | d d d df | d d dd d df t | g d
g d g t t t j | d t j | d d
t | d t j | dg d t | g dg dg t t t j | dg d t j | dd
g dg dg t | g dg dg t j | d d d df | d d dd d df dd gd d gg t | g dg dg t t t j g dg d d S ) Nr] ri r r r )r r r )r] r r r )r r] r r r r r. r rq )TFTr )r r r r r )r r r )r r r )rr r )r rz rz )rz r r r )rz r r r r r )r r rz )r& r r r r r rY )rS s r* test_copytor s-
!4 ((A..A Ia)))YYY'(((YYY *+++ Ia2A2h$$B$2+'''YYY *+++ )RY3/// Iah''''A Ia---....YYY *+++ )RY36I6I6IJJJJ Iahyyy))).DEEEEYYY *+++ Ia2A2h$$B$2+1v1v.>????YYY *+++ )RY 999=====r, c
d d} dg| z g dz }t j t | | z
}t j t | | z
}t j | | d }t j |||d d d d}t j | }t
d|z D ];t j | }fdt
| D }t j | }t j ||| t |d k | t | t | t j | }t j |||d d d t |d k |d d d t | t | t j | }t j |d d d |d d d |d d d t |d d d d k |d d d t |d d d t |d d d t j | }t j |d d d |d d d |d d d
t |d d d d k |d d d
t |d d d t |d d d
d D ]}t j | }t j | }t j |
t j }|||d k < t j ||| t |d k | t | t | =t j | }t j ||d t | |j t j | }t j | }t j ||d
t | |j d S )Ni T)TTTTr r r c " g | ] }|z d k S )r r ).0r) rU s r*
z&test_copyto_permut.. s! 0 0 0aa!e\ 0 0 0r, r ) r r F)
r& r lenr3 r' r rB r r sumr r= r ) padlrr r powercimaskrU s @r* test_copyto_permutr s
C
////A
Q A
As
A
8A;;stt DIa$ttt*%%%%
E
A
1e8__ * * HUOO 0 0 0 05<< 0 0 0x{{
!Qd####161%%%QUUWWc!ff%%% HUOO
!Qd44R4j))))161TTrT7+++QUUWWc!ff%%% HUOO
!CCaC&!CCaC&SSqS 22221SSqS6Q;##A#///QsssVZZ\\3q1v;;/// HUOO
!CCaC&!CCaC&TTrT
33331SSqS6Q;$$B$000QsssVZZ\\3q2w<<000) * *AA8A;;DHQKK$$RX..E E$!) Ia$ ' ' ' ' qAvq ) ) ) #a&& ) ) ) ) * AIa$ !&!!!
A
AIa% !&!!!!!r, c t j d dddd } | d }t j d dddd dd }dfd }t | j j t | j j t |j j t |j j t |j j t |j j | d
|| d dd | d
||d dd | d
||d dd t j | d
|| d dd t j |d
||d dd t j |d
||d dd | d || dd d | d ||dd d | d ||dd d t j | d || dd d t j |d ||dd d t j |d ||dd d | d
|| d dd | d
||dd d | d
||ddd t j | d
|| d dd t j |d
||dd d t j |d
||ddd d S )N r r r rz rl rm Fc t | |u t | | t j j | t j j | d S r0 )r r rc r r )r) yccontigfcontigr ress r* check_copy_resultz*test_copy_order..check_copy_result sU Q!V QSY+W555SY+W55555r, rn T)r r r K)F) r& r r rk swapaxesr rc r r )rS re r r r s @r* test_copy_orderr si
"
aAq))A SA
"
aAq))221a88A6 6 6 6 6 6 AG !!! $ $%%% $ $%%% AG !!! $ $%%% $ $%%% &&s&
Cc1dE4HHHH
&&s&
Cc1dE5IIII
&&s&
Cc1dE5IIII
'!3
Cc1dE4HHHH
'!3
Cc1dE5IIII
'!3
Cc1dE5IIII &&s&
Cc1eT5IIII
&&s&
Cc1eT4HHHH
&&s&
Cc1eT5IIII
'!3
Cc1eT5IIII
'!3
Cc1eT4HHHH
'!3
Cc1eT5IIII &&s&
Cc1dE4HHHH
&&s&
Cc1eT4HHHH
&&s&
Cc1eUDIIII
'!3
Cc1dE4HHHH
'!3
Cc1eT4HHHH
'!3
Cc1eUDIIIIIIr, c @ t j d d d dd d d d f } | j d d dz | _ t j d dd }d } || dd ||dd |t j d d
d
|t j d gdgggd
d
d
|t j d d
d |t j dd
dd
|t j | d dd |t j | dd d
d |t j | ddd dd
|| d d
d
|| d d d ddd f d
d
||d dd
||d d d d ddd d d d f d
d
|| d
d
|t j d d
d
d S )N)rz rz r r )i)r r r r r r rz c z t | j j |k t | j j |k d S r0 )r rc r r )rS r r s r* check_contigz+test_contiguous_flags..check_contig# s8 $/000$/00000r, F)r r r r r Tr rl r r r rj rn )rk rm )ndminrk rm r .)r r .r )r& r3 r r emptyr' ravelsqueeze)rS re r s r* test_contiguous_flagsr st
33Q3qqq7#A "1"
'AI
))!Q//A1 1 1
LE5!!!LE5!!!L/**D$777LQC!:,c222D$???L&!!4///L&,,,eT::: L!%(((%777L!%s333T5AAAL!15<< t j dgd } t j g dd }t j | | }t |d t j g dd t |d t j g dd d S )Nr zu4,u4,u4r )r )rz r r] )r\ r r )r r r r )r& r' broadcast_arraysr )rS re results r* test_broadcast_arraysr > s
)J///A
222*EEEA
A
&
&FBH%F%F%Fj Y Y YZZZBH%F%F%Fj Y Y YZZZZZr, )r
fill_valueexpected_outputr g @r )r r r g @c P t j | | }t || d S r0 )r& rF r )r r r outputs r* test_full_from_listr F s+ WUJ
'
'F )))))r, c t j dt j } | t j d }t j | | rJ | t j t j j }t j | | rJ | t j d }t j | | sJ | t j t j j }t j | | sJ | t j t j j }t j | | sJ | t j d }t || | t j t j j }t || t t | j t j t j j d S )Nr/ r Trj F)
r& r intpr may_share_memory _CopyModeALWAYS IF_NEEDEDNEVERr( r r rD )r res_true
res_always res_false
res_if_needed res_nevers r* test_astype_copyflagr M s
)Bbg
&
&
&Czz"'z--H"31111 1BG",*=>>J"3
3333 3
27
//I
sI . ... .JJrwR\-CJDDM
sM 2 222 2
27 );
<r! s
/ / / / / /
J6 J6 J6X 4-00! ! 10! ! Q"" n' n' n'` Sz** 2 2 +* 2 7 7 7889 9 989
JvE**RZ-@@ 1 1 1
';/
)RX
'28
"(8
hbhuoo . 01 18 8 1 18 ';/
)RX
'28
"(8
hbhuoo . 01 1" " 1 1" c5"'2;%GHH r}bn57 7 7 7 IH ",z":;;6 6 <;68 8 8""> "> ">H3" 3" 3"j;J ;J ;Jz ; ; ;@[ [ [ CCC
3+xrx#sc3Z(@ A A B
3+xrx#sc3Z#(M N N O QR R* *R R*+ + + + +r,