1jt d dl Z d dlZd dlmZmZmZmZmZm Z m
Z
m Z m Z m
Z
mZ d dlmZmZmZmZ d dlmZmZmZmZmZmZmZ G d d Z G d d Z G d d
Z G d d Z G d
d Z G d d Z!d Z"e j# $ dd g e j# $ dg d e j# $ dg d d Z% G d d Z&d Z'dS ) N) arrayarange
atleast_1d
atleast_2d
atleast_3dblockvstackhstacknewaxis concatenatestack)_block_dispatcher _block_setup_block_concatenate_block_slicing)assert_
assert_raisesassert_array_equal assert_equalassert_raises_regex assert_warnsIS_PYPYc , e Zd Zd Zd Zd Zd Zd ZdS )
TestAtleast1dc t d }t d }t | t | g}t dg t dg g}t || d S N r r r selfabresdesireds t/builddir/build/BUILD/cloudlinux-venv-1.0.11/venv/lib64/python3.11/site-packages/numpy/core/tests/test_shape_base.py
test_0D_arrayzTestAtleast1d.test_0D_array s^ !HH !HH!}}jmm,!::uaSzz*3((((( c t ddg }t ddg }t | t | g}t ddg t ddg g}t || d S Nr r r r s r&
test_1D_arrayzTestAtleast1d.test_1D_array sj 1a&MM 1a&MM!}}jmm,!Q==%A--03(((((r( c t ddgddgg }t ddgddgg }t | t | g}||g}t || d S r* r r s r&
test_2D_arrayzTestAtleast1d.test_2D_array k Aq6Aq6" # # Aq6Aq6" # #!}}jmm,a&3(((((r( c t ddgddgg }t ddgddgg }t ||g }t ||g }t | t | g}||g}t || d S r* r r s r&
test_3D_arrayzTestAtleast1d.test_3D_array% Aq6Aq6" # # Aq6Aq6" # # 1a&MM 1a&MM!}}jmm,a&3(((((r( c : t t d j dk t t d j dk t t d j dk t t ddgddgg j dk d S )
zB Test to make sure equivalent Travis O's r1array function
r+ r @ @r r r N)r r shaper! s r& test_r1arrayzTestAtleast1d.test_r1array. s
1
#t+,,,
2$,---
3%-...
QFQF+,,2f<=====r( N)__name__
__module__ __qualname__r' r, r. r1 r< r( r& r r s_ ) ) )) ) )) ) )) ) )> > > > >r( r c , e Zd Zd Zd Zd Zd Zd ZdS )
TestAtleast2dc t d }t d }t | t | g}t dgg t dgg g}t || d S r r r r r s r& r' zTestAtleast2d.test_0D_array8 sb !HH !HH!}}jmm,1#<<u .3(((((r( c t ddg }t ddg }t | t | g}t ddgg t ddgg g}t || d S r* rD r s r& r, zTestAtleast2d.test_1D_array? sn 1a&MM 1a&MM!}}jmm,1a&??EAq6(OO43(((((r( c t ddgddgg }t ddgddgg }t | t | g}||g}t || d S r* rD r s r& r. zTestAtleast2d.test_2D_arrayF r/ r( c t ddgddgg }t ddgddgg }t ||g }t ||g }t | t | g}||g}t || d S r* rD r s r& r1 zTestAtleast2d.test_3D_arrayM r2 r( c t t d j dk t t ddg j dk t t ddgddggddgddggg j d k d
S ) zB Test to make sure equivalent Travis O's r2array function
r+ r r r5 r r r r7 r8 r r r r N)r r r: r; s r& test_r2arrayzTestAtleast2d.test_r2arrayV s
1
#v-...
B7##)V3444
aVaV ,1v1v.>?@@F)STTTTTr( N)r= r> r? r' r, r. r1 rL r@ r( r& rB rB 7 sd ) ) )) ) )) ) )) ) )U U U U Ur( rB c & e Zd Zd Zd Zd Zd ZdS )
TestAtleast3dc t d }t d }t | t | g}t dggg t dggg g}t || d S r r r r r s r& r' zTestAtleast3d.test_0D_array_ sf !HH !HH!}}jmm,A3%>>5A3%>>23(((((r( c t ddg }t ddg }t | t | g}t dgdggg t dgdggg g}t || d S r* rP r s r& r, zTestAtleast3d.test_1D_arrayf sz 1a&MM 1a&MM!}}jmm,A3* &&aSzl(;(;<3(((((r( c t ddgddgg }t ddgddgg }t | t | g}|d d d d t f |d d d d t f g}t || d S r* )r r r r r s r& r. zTestAtleast3d.test_2D_arraym s Aq6Aq6" # # Aq6Aq6" # #!}}jmm,QQQqqq'\?Aaaa7lO43(((((r( c t ddgddgg }t ddgddgg }t ||g }t ||g }t | t | g}||g}t || d S r* rP r s r& r1 zTestAtleast3d.test_3D_arrayt r2 r( N)r= r> r? r' r, r. r1 r@ r( r& rN rN ^ sP ) ) )) ) )) ) )) ) ) ) )r( rN c > e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d S )
TestHstackc < t t t d d S Nr )r TypeErrorr
r; s r& test_non_iterablez TestHstack.test_non_iterable i+++++r( c < t t t d d S Nr@ )r
ValueErrorr
r; s r& test_empty_inputzTestHstack.test_empty_input j&"-----r( c t d }t d }t ||g }t ddg }t || d S r r r
r r s r& r' zTestHstack.test_0D_array sL !HH !HHaVnnA--3(((((r( c t dg }t dg }t ||g }t ddg }t || d S r ra r s r& r, zTestHstack.test_1D_array sP 1#JJ 1#JJaVnnA--3(((((r( c t dgdgg }t dgdgg }t ||g }t ddgddgg }t || d S r ra r s r& r. zTestHstack.test_2D_array sk A3* A3* aVnn!Q!Q())3(((((r( c t j t d 5 t d t d D d d d n # 1 swxY w Y t j t d 5 t t d t
j d d d d d S # 1 swxY w Y d S )Narrays to stack must bematchc 3 > K | ]}t j d V dS r+ Nnpr .0_s r& z,TestHstack.test_generator.. * 33QBIaLL333333r( r c | S Nr@ )xs r& z+TestHstack.test_generator.. s r( )r+ r )pytestraisesrX r
rangemaprk onesr; s r& test_generatorzTestHstack.test_generator s,
]9,E
F
F
F 5 5 33%((333 4 4 4 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5
]9,E
F
F
F 6 6 3{{BGFOO44 5 5 5 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6s# 'AAA51B33B7:B7c t j g d }t j g d }t j ||fdt j }t j g d }t || d S )Nr r r+ g @g @g @unsafecastingdtyper r r+ r r+ r7 )rk r r
int64r r! r" r# r$ expected_ress r& test_casting_and_dtypez!TestHstack.test_casting_and_dtype sp HYYY H___ % %iAAAAx 2 2 233 3 -----r( c t j g d }t j g d }t j t 5 t ||fdt j d d d d S # 1 swxY w Y d S Nr| r} safer )rk r ru rv rX r
r r! r" r# s r& !test_casting_and_dtype_type_errorz,TestHstack.test_casting_and_dtype_type_error HYYY H___ % %
]9
%
% ; ; Aq66 : : : : ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; A22A69A6N) r= r> r? rY r^ r' r, r. rz r r r@ r( r& rU rU ~ s , , ,. . .) ) )) ) )) ) )6 6 6 . . .; ; ; ; ;r( rU c D e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
S )
TestVstackc < t t t d d S rW )r rX r r; s r& rY z TestVstack.test_non_iterable rZ r( c < t t t d d S r\ )r r] r r; s r& r^ zTestVstack.test_empty_input r_ r( c t d }t d }t ||g }t dgdgg }t || d S r r r r r s r& r' zTestVstack.test_0D_array sR !HH !HHaVnn!qc
##3(((((r( c t dg }t dg }t ||g }t dgdgg }t || d S r r r s r& r, zTestVstack.test_1D_array sV 1#JJ 1#JJaVnn!qc
##3(((((r( c t dgdgg }t dgdgg }t ||g }t dgdgdgdgg }t || d S r r r s r& r. zTestVstack.test_2D_array so A3* A3* aVnn!qcA3,--3(((((r( c t ddg }t ddg }t ||g }t ddgddgg }t || d S r r r s r& test_2D_array2zTestVstack.test_2D_array2 s_ 1a&MM 1a&MMaVnn!Q!Q())3(((((r( c t j t d 5 t d t d D d d d d S # 1 swxY w Y d S )Nre rf c 3 > K | ]}t j d V dS ri rj rl s r& ro z,TestVstack.test_generator.. rp r( r )ru rv rX r rw r; s r& rz zTestVstack.test_generator s
]9,E
F
F
F 5 5 33%((333 4 4 4 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5s 'AAAc t j g d }t j g d }t j ||fdt j }t j g dg dg }t || d S )Nr| r} r~ r r r+ r7 )rk r r r r r s r& r z!TestVstack.test_casting_and_dtype sv HYYY H___ % %iAAAAxIII 677 3 -----r( c t j g d }t j g d }t j t 5 t ||fdt j d d d d S # 1 swxY w Y d S r )rk r ru rv rX r r r s r& r z,TestVstack.test_casting_and_dtype_type_error r r N) r= r> r? rY r^ r' r, r. r rz r r r@ r( r& r r s , , ,. . .) ) )) ) )) ) )) ) )5 5 5. . .; ; ; ; ;r( r c
e Zd Zd Zd Zd Zd Zd Zej
e d d Z d Z
ej d
d d g ej d
g d ej dg d d Zej d
d d g ej dg d ej ddgfdgdgfd gdgdgfg d Zej d
d d g d Zej d
d d g d Zd S ) TestConcatenatec t j d }t j |g }d|d< |d |d k sJ d S Nr+ r )r r )rk eyer r s r& test_returns_copyz!TestConcatenate.test_returns_copy sH F1II NA3 $w!D'!!!!!!r( c dD ]}t j d|z }t j ||fd t t j t j ||f| t t j t j ||f|dz t t
t d t t
t t j d f t t
dt j t j d t j d f t j d }t j d }t t d
}t d
D ]}t j ||f|d t t
d | t j ||f|d t t
t j ||f|d t j |d
d }t j |d
d }|
| d t t
t d d S )Nr| r4 r axisr )r zall the input arrays must have same number of dimensions, but the array at index 0 has 1 dimension\(s\) and the array at index 1 has 2 dimension\(s\)rI r r r+ r+ zall the input array dimensions except for the concatenation axis must match exactly, but along dimension {}, the array at index 0 has size 1 and the array at index 1 has size 2r r@ )rk ry r r AxisErrorr] r r zeroslistrw formatmoveaxisappendpop)r! ndimr" r# r is r& test_exceptionsz TestConcatenate.test_exceptions s R RDT ""A NAq6 * * * * ",AT J J J J ",Atax[ Q Q Q Q Q j+t444j+ ~>>>
,
NRXa[["(6*:*: ; = = =
GI GI E!HH~~q % %A NAq6Q 0 0 0 0 I AT!W
6
6
6
6
*bnq!f47 K K K K Ar1%%A Ar1%%A KK $ $ $ $ j+r22222r( c \ t j dt j d }t t d }dg}t j ||fd }t |j |j t |j d t j ||fd }t |j
|j
t | z t |j |j t j |||fd d }t g d
}t || t j |j
t | z }t j ||fd }t j ||fd | }t ||u t || d S ) Nr7 r r9 r+ rs r r U)r r )z0.0z1.0z2.0z3.0012rs )r out)rk r float64reshaper rw r r r r sizelenr r r r )r! r" r# crdr routs r& test_concatenate_axis_Nonez*TestConcatenate.test_concatenate_axis_None s Iarz * * * 2 26 : : qNN
E NAq6 - - -QWag&&&QVQ NAq6 - - -QVQVc!ff_---QWag&&& NAq!94s ; ; ; ' ' '
(
(1a hqvA'' NAq6 - - -~q!f4S999t Q r( c t j dd }t j |d }t || t j |d }t || d S )Nr d )rk r r r )r! rs r s r& test_large_concatenate_axis_Nonez0TestConcatenate.test_large_concatenate_axis_None) s_
Ia N1d # #1a
N1c " "1a r( c t t d }t t |f | t t t | f | t t t | f | t t d }t t ||f ||z t t t | |f ||z t t t | |f ||z t t ||fd ||z t t ||fd ||z t g dg dg }t g dg }t g dg dg dg }t t ||f | t t ||fd | t t |j |j fd |j t t |j |j fd |j t t t |j |j fd t d
d
}|d d df }|d dd f }|d d d f }t t |||fd
| t t |||fd | t t |j |j |j fd |j | } t |||fd
| }
t | |
u t ||
d S )Nr7 r+ r r )
)
)r r r r * )r r+ . r r )r rw r r tupler Tr r] r r copyr r ) r! r4r3a23a13r$ a0a1a2r r s r& test_concatenatez TestConcatenate.test_concatenate4 s! %((^^;u--r222;b |44b999;b |44b999
%((^^;Bx00"r':::;b 277bAAA;b 277bAAA;Bx33R"W===;Bx44b2g>>>\\\<<<011YYYK \\\<<<;<<;Sz22C888;Sz155s;;;;su~q9935AAA;su~r::CEBBBj+su~qAAAY '' 22
bqb\
ac]
abb\;B|Q77===;B|R88#>>>;bdBD'91==suEEEhhjjBB<444t S$ r( z7PYPY handles sq_concat, nb_add differently than cpython)reasonc dd l }t ddg }t ddg }ddg}t g d }t t |j || t t |j || t t |j || t t |j |d t t |j d| d S )Nr r r r+ r7 )r r r+ r7 )operatorr r rX concat)r! r r" r# nr$ s r& test_operator_concatz$TestConcatenate.test_operator_concat^ s 1a&MM 1a&MM
qELLL!!i !Q777i !Q777i !Q777i !Q777i !Q77777r( c t ddg }t ddg }t t t ||ft j d t t t ||ft j d t t t ||ft j d t ||ft j d d S ) Nr r r+ r7 r8 r )r7 r )r r7 )r r r] r rk emptyr s r& test_bad_out_shapez"TestConcatenate.test_bad_out_shapek s 1a&MM 1a&MMj+1v28A;;GGGGj+1v28E??KKKKj+1v28E??KKKKQF ,,,,,,r( r Nr out_dtype)c8f4f8>f8i8S4r noequivr same_kindr~ c t j d| }t ddg t ddg f}t j |d || st t
5 t
|||| d d d n # 1 swxY w Y t t
5 t
||j ||
d d d n # 1 swxY w Y nNt
|||| }t
||j ||
}||u sJ t || |j |k sJ t t
5 t
|||| d d d d S # 1 swxY w Y d S ) Nr7 r g?g@gffffff
@g@r r r r r r r r r r r ) rk r r can_castr rX r r r )r! r r r r to_concatres_out res_dtypes r& test_out_and_dtypez"TestConcatenate.test_out_and_dtypet sv hq ***C:&&sCj(9(9: {9Q<GDDD
0 y))
L
LI3T7KKKK
L
L
L
L
L
L
L
L
L
L
L
L
L
L
L y))
8
8ISY!%w8 8 8 8
8
8
8
8
8
8
8
8
8
8
8
8
8
8
8 ")'+W> > >G#ISY)-w@ @ @Ic>>>> sI . . . ?i////
9
%
% H H s)$ G G G G H H H H H H H H H H H H H H H H H Hs6 *B
BB)CCC<E E!$E! string_dt)Sr S0U0arrsg r string ?c t j |||d }|j t j d | j k sJ d S )Nr~ )r r r r )rk r r r astype)r! r r r r$ s r& test_dtype_with_promotionz)TestConcatenate.test_dtype_with_promotion sN nTIxPPPyBHRLL / / : : @@@@@@@r( c 8 t j t 5 t j d gdgfd| d d d n # 1 swxY w Y t j t 5 t j d gdgfd| d d d d S # 1 swxY w Y d S )Nr r r r r ru rv rX rk r r! r s r& "test_string_dtype_does_not_inspectz2TestConcatenate.test_string_dtype_does_not_inspect s>
]9
%
% @ @ NTFQC=$ ? ? ? ? @ @ @ @ @ @ @ @ @ @ @ @ @ @ @
]9
%
% @ @ NTFQC=$ ? ? ? ? @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @s# AA A& BBBc t j t d 5 t j dgdgfd| d d d d S # 1 swxY w Y d S )Nz.*subarray dtyperf r z(2,)ir r r s r& test_subarray_errorz#TestConcatenate.test_subarray_error s
]9,>
?
?
? A A NQC!:W4 @ @ @ @ A A A A A A A A A A A A A A A A A As AA A )r= r> r? r r r r r ru markskipifr r r parametrizer r r r r@ r( r& r r s1 " " " '3 '3 '3R * ! ! !( ( ( T [(abb
8
8 cb
8- - - [ VdAY// [ [*O*O*OPP [ Y : : :< <H H< < QP 0/H. [ VdAY// [ [*@*@*@AA [ VdWtaSkQC(bT#: ;= =A A= = BA 0/A [ VdAY//@ @ 0/@ [ VdAY//A A 0/A A Ar( r c " t t t d dt j d t j d t j d gt j d t j d t j d gfD ]!} t
t | g d "t j g d }t j g d }t g dg dg }t
t j ||f | t
t j ||fd |j t
t j t ||g | t
t j t ||g | d t d D }g d }g d
}t || D ]-\ }}t t j || j | .t t j d t |d t t j d t |d d
t d D }g d}g d}t || D ]-\ }}t t j || j | .t t g g g g j dk t t g g g gd j dk t j | } t j ||f| t
| | t t" dt g t t" dt dt j d g t t" dt t j d dg t t" dt t j d dgd t t" dt t j d t j d gd t t" dt t j d t j d g t) j t d 5 t d t d D d d d n # 1 swxY w Y t j g d }t j g d }t j ||fddt j }
t j ddgddgdd gg } t
|
| t t 5 t ||ft j dd d d d d S # 1 swxY w Y d S ) Nr r| r r+ )r7 r8 r r c L g | ]!}t j d "S )r+ rk randomrandnrl s r&
z test_stack.. s&
4
4
4Qbiooa
4
4
4r( r )r r r )r r+ r+ r r r z
out of boundsc N g | ]"}t j d d #S r+ r7 r rl s r& r z test_stack.. s(
7
7
7biooa##
7
7
7r( )r r r r r r )r r+ r7 r+ r r7 r+ r7 r r r r )r+ r )r r+ r zneed at least one arrayzmust have the same shaper+ r+ re rf c 3 K | ]}|V d S rr r@ )rm rs s r& ro z test_stack.. s" ##Qq######r( r} r~ )r r r r7 r r )r rX r
rk int32r r r r rw zipr r: r r r
zeros_liker] r r ru rv r ) input_r" r# r1arraysaxesexpected_shapesr expected_shaper r$ r s r&
test_stackr' s )UA&&& HQKK!bhqkk:HQKK!bhqkk:< 5 5 5==)))4444
A
A 999% & &BrxA'',,,rxAQ///666rxaV --r222rxq!f
..333
4
4%))
4
4
4F >>D:::O #D/ : : C C nRXfd++1>BBBB ouf1MMMM ouf2NNNN
7
7U2YY
7
7
7F D; ; ;O #D/ : : C C nRXfd++1>BBBB E2r2, % /000 E2r2,Q ' ' ' - 7888
-
CHaV sB
$=ubIII
$> 29Q<< 02 2 2
$> ! a 02 2 2
$> ! a 0q: : : :
$> 0 0"(1++ >QH H H H
$> ! bill ;= = =
y(A B B B % %
##%((###$$$% % % % % % % % % % % % % % % A
!!A
(Aq6828
D
D
DC8aVaVaV 455LsL))) y ! ! > >
q!fBH1f====> > > > > > > > > > > > > > > > > >s$ 'R;;R?R? VV Vr r )r r r r r r r c h t ddg t ddg f}t ddgddgg }t j | }t j |d || s@t t
5 t
||| | d d d n # 1 swxY w Y nIt
||| | }t
||| | }||u sJ t || |j |k sJ t t
5 t
|||| d d d d S # 1 swxY w Y d S )
Nr r r+ r7 r r r r r ) r rk r r r rX r
r r )r r r r r$ r r r s r& test_stack_out_and_dtyper) s
1v q!f
.I
!Q!Q
!
!C
-
C
;y|Y @ @ @ ,
9
%
% . . )9W
.
.
.
. . . . . . . . . . . . . . . . s !74 4 4)9 #W6 6 6 #~~~~3 ***)++++ y ! ! > >
iS ====> > > > > > > > > > > > > > > > > >s$ =B B!$B!D''D+.D+c e Zd Z ej g d d Zd Zd Zd Zd Z d Z
d Z d
Z d Z
d Zd
Zd Zd Zd Zd Zd Zd Zd Zd Zd ZdS ) TestBlock)r force_concatenate
force_slicing)paramsc d }d }|j dk r|S |j dk r|S |j dk rt S t d )Nc L t | \ } }}}t | || S rr )r r r# list_ndim result_ndimrn s r& _block_force_concatenatez1TestBlock.block.._block_force_concatenate s* 0._block_force_slicing s* 0 t dg \ }}}}|dk sJ t dgg \ }}}}|dk sJ t ddgg \ }}}}|dk sJ t dgdgg \ }}}}|dk sJ t ddgddgg \ }}}}|dk sJ d S )Nr r r+ r7 )r )r! r rn
total_sizes r& test_block_total_size_estimatez(TestBlock.test_block_total_size_estimate& s *A3 / /1aQ *QC5 1 11aQ *QF8 4 41aQ *QC!: 6 61aQ *QFQF+; < <1aQr( c t j d }d|z }t j g dg dg } |||g }t || d S )Nr9 r )r r r r rk ry r r )r! r a_2db_2dr% results r& test_block_simple_row_wisez$TestBlock.test_block_simple_row_wise6 sd wv4x(LLL (LL * + +d|$$Wf%%%%%r( c t j d }d|z }t j ddgddgddgddgg } ||g|gg }t || d S Nr9 r r r> )r! r r? r@ expectedrA s r& test_block_simple_column_wisez'TestBlock.test_block_simple_column_wise> sy wv4x8aV V V V % & & '((Xv&&&&&r( c t j g d }t j g d }t j g d } |||g }t || d S Nr| r r rk r r r! r r" r# rE rA s r& "test_block_with_1d_arrays_row_wisez,TestBlock.test_block_with_1d_arrays_row_wiseH se HYYY HYYY 8 . . .//1v Xv&&&&&r( c t j g d }t j g d }t j g dg dg } |||g||gg }t || d S rH rI rJ s r& 'test_block_with_1d_arrays_multiple_rowsz1TestBlock.test_block_with_1d_arrays_multiple_rowsP s HYYY HYYY 8 / / / / / / 1 2 2AA'((Xv&&&&&r( c t j g d }t j g d }t j g dg dg } ||g|gg }t || d S )Nr| r rI )r! r a_1db_1drE rA s r& %test_block_with_1d_arrays_column_wisez/TestBlock.test_block_with_1d_arrays_column_wiseX sy x ""x ""8YYY &YY ( ) )'((Xv&&&&&r( c t j d }t j ddg } ||g|gg }t j ddgddgddgg }t || d S rD r> )r! r r? rP rA rE s r& test_block_mixed_1d_and_2dz$TestBlock.test_block_mixed_1d_and_2da s| wvxA '((8aV V V % & & Xv&&&&&r( c t j g dg }t j g dg }t j g dg }t j g d }t j d }t j g d }t j d }t j g dg dg dg d g d
g d
g } |||g|g|g||g|gg }
t |
| d S ) Nr r r rK )r+ r+ r+ r+ r+ r+ )r7 r7 r7 r7 r7 r7 r8 r r r r r r r r r r r r r r8 r r r r r r r r r r r rk r r r ) r! r one_2dtwo_2dthree_2dfour_1dfive_0dsix_1dzero_2drE rA s r& test_block_complicatedz TestBlock.test_block_complicatedj s$ 999+&&999+&&8 / / / 011(---..(1++///**(6""8 / / / / / / / / / / / / / / / / / / 1 2 2 (!
&) # $ $
VX&&&&&r( c t j g d }t j g dg dg dg }t j g d }t j g d }t j d }t j g d }t j d } | ||g|g|gg |g||g|gg } t j g dg d g d
g d g d g d g }
t | |
d S )
NrU rK r+ r+ r+ )r7 r7 r7 r8 rV rW rX )r+ r+ r+ r r r )r7 r7 r7 r r r rY rZ r[ ) r! r onetwothreefourfivesixzerorA rE s r& test_nestedzTestBlock.test_nested sC hyyy!!h 999iii899##x ""x{{h''x 576
3K
F
8 / / / / / / / / / / / / / / / / / / 1 2 2 VX&&&&&r( c t j dt dz }t j dt dz }t j dt dz }t j dt dz }t j d t d
z }t j d t d z }t j d
t dz }t j dt dz } |||g||gg||g|| ggg }
t g dg dg dg dg dgg dg dg dg dg dgg dg dg dg dg dgg dg dg dg dg dgg dg dg dg dg dgg } t |
| d S )NrK r )r+ r r r )r r+ r r+ r r7 )r r+ r+ r8 )r+ r r+ r )r+ r+ r r re )r r r7 r7 r7 )r+ r+ r8 r8 r8 )r r r r r )r r ro ro ro )rk ry intr r ) r! r a000a100a010a001a011a101a110a111rA rE s r& test_3dzTestBlock.test_3d s wy#&&*wy#&&*wy#&&*wy#&&*wy#&&*wy#&&*wy#&&*wy#&&*t t
t t
???*??*??*??*?? , +??*??*??*??*?? , +??*??*??*??*?? , +??*??*??*??*?? , +??*??*??*??*?? ,1 - . .< 68,,,,,r( c t j ddg }t j d }t t |||g t t |||g t j d t j d gt j d t j d gg}t t || d S )Nr r )r r+ r9 )rk r r r r] ry )r! r r" r# to_blocks r& test_block_with_mismatched_shapez*TestBlock.test_block_with_mismatched_shape s HaV F1IIj%!Q000j%!Q000WU^^RWU^^4WU^^RWU^^46j%22222r( c t |d t j d t |t j d t j d d S )Nr r+ )r rk r r r! r s r&
test_no_listszTestBlock.test_no_lists sR UU1XXrx{{333UU26!99%%rvayy11111r( c d}t t ||ddgg t t ||dg g t t ||dgdg t t ||g dg t t ||dgdggddggdgg d S )Nzdepths are mismatchedr r r+ r7 r8 r r] )r! r msgs r& test_invalid_nestingz TestBlock.test_invalid_nesting s %JUQH===JUQG<<<JUaS!H===JURG<<<JUS1#JVH
C5
r( c t t d|g t t d|g g t t d|dgg g d S )Nr r r r~ s r& test_empty_listszTestBlock.test_empty_lists sL J;;;J===J!b BBBBBr( c r t t d|ddgddgf t t d|ddg d S )Nr r r r+ r7 rJ r )r rX r~ s r&
test_tuplezTestBlock.test_tuple sB IwAA7GHHHIw7GHHHHHr( c d}dt j d z }dt j d z } ||||g }t j g dgg }t || d S )Nr r rJ r+ )r r r+ )r @r r6 r6 r6 r> r! r r" r# r rA rE s r& test_different_ndimsz TestBlock.test_different_ndims st
"" "1ay!!8 6 6 6 7 899VX&&&&&r( c d}dt j d z }dt j d z } |||g|gg }t j g dg dg dgg }t || d S )Nr r rJ r+ r| )r r r )r6 r6 r6 r> r s r& test_different_ndims_depthsz%TestBlock.test_different_ndims_depths s
"" "A}%%8lll *ll *ll , - . . VX&&&&&r( c t j dd }t j dd }||g||gg||g||ggg}||g||gg||g||ggg} || j d sJ || j d sJ t j dd }t j dd }||g||gg}||g||gg} || j d sJ || j d sJ d S )Nre C)orderF C_CONTIGUOUS F_CONTIGUOUSr )rk r flags)r! r arr_carr_fb_cb_fs r& test_block_memory_orderz!TestBlock.test_block_memory_order sT s+++s+++ !
!
uSzz ////uSzz ////s+++s+++u~u~ u~u~ uSzz ////uSzz //////r( N)r= r> r? ru fixturer r r< rB rF rK rM rQ rS rc rm ry r| r r r r r r r r@ r( r& r+ r+ sl V^JJJKKKX X LKX2" " " & & &' ' '' ' '' ' '' ' '' ' '' ' '0 ' ' '<5- 5- 5-n3 3 32 2 2
C C C
I I I' ' '
'
'
' 0 0 0 0 0r( r+ c * G d d } | } | } | }t t t | |g t t t |g |g t t t ||g ||g t t t |g||ggg |||g t t t ||f ||fg d S )Nc e Zd ZdS )(test_block_dispatcher..ArrayLikeN)r= r> r? r@ r( r& ArrayLiker / s r( r )r r r )r r" r# r s r& test_block_dispatcherr . s
A A A'**++aS111',,--s333'A//001a&999'!q1#h8899Aq!9EEE'A//00Aq6(;;;;;r( )(ru numpyrk
numpy.corer r r r r r r r
r r r
numpy.core.shape_baser r r r
numpy.testingr r r r r r r r rB rN rU r r r' r
r r) r+ r r@ r( r& r s,
G G G G G G G G G G G G %> %> %> %> %> %> %> %>P$U $U $U $U $U $U $U $UN ) ) ) ) ) ) ) )@-; -; -; -; -; -; -; -;`2; 2; 2; 2; 2; 2; 2; 2;l@A @A @A @A @A @A @A @AFA> A> A>H !%%&E&E&EFFGGGI I> >I I GF &%>,f0 f0 f0 f0 f0 f0 f0 f0R < < < <