y>>R)'8<>>j""6HHHj""6GGGj""6HHHj""6GGGi!5y&IIIR%oyAA999MMM I 6 6 I I I h 999-..R)#v66 EEE fC 8 8 8,,, H H H f6 : : :LLL J J JR)#v > > XqM E E Es L L Lj )hqj!_C I I I I I IrT c & t j t j t j t j t j t j fD ]}t j g dg dg| }d}d|d z |d z }t t j || | t |t j
|| |d d|d z z }t t j ||d
| t |t j
||d
t j g dg dg d g| }d }d
d|d z |d z z |d z }t t j || | t |t j
|| |d d|d d|d z z z z }t t j ||d
| t |t j
||d
d S )N)r r r r r) r% )r r* r r) r r rL )r+ r= r= r r r+ r0 r1 )r r) r r r> r+ )r+ r= rJ rJ r ) r? int16uint16int32uint32r[ uint64rD r rB r@ )rE rM coordsr uncoordss rG test_dtypesz!TestRavelUnravelIndex.test_dtypesr s5 h 28i296 O OEX###%7%7%78G G GFE{6!9,H -fe<>>
J
J
J !1(E!M!M!M N N NX###%7%7%79K9K9KL F E1VAY;vay01&);H -fe<>>
J
J
J !1(E!M!M!M N N N N/ O OrT c < t t j g ddd t j g dd t t j g ddd t j g dd t t t j g dd d S )N)r+ r r$ r )r% r) r, r< r; r6 )r r r* r )r; raiser5 rk )r r r r )r r? rB r rC rE s rG test_clipmodesz$TestRavelUnravelIndex.test_clipmodes s
F K K K } = = ? ? ? R)--- 0PQ Q Q ),,,
FF H H H ,mmm] L L L L LrT c t j g dd \ }}t |j j t |j j d S )Nr r r) )r% r+ )r? r@ r flags writeable)rE xys rG test_writeabilityz'TestRavelUnravelIndex.test_writeability sH 6221!"""!"""""rT c t j dd }t |d t t dt j dgd t t dt j dgd d S )Nr rK z0d arrayz
out of boundsr )r? r@ r r rC rE rr s rG test_0dz TestRavelUnravelIndex.test_0d sp Q # #QJ
B4Dqc2NNN )9A3 D D D D DrT r7 )r5 r; rk c N t j t j dt j d| }|j dk sJ t t 5 t j t j dt j d| d d d d S # 1 swxY w Y d S )N)r) r rL r r r r6 r )r) r )r? rB zerosrZ r r rC )rE r7 ress rG test_empty_array_ravelz,TestRavelUnravelIndex.test_empty_array_ravel s "HV27333YTK K KyD
:
&
& K K HV27333YT
K
K
K
K K K K K K K K K K K K K K K K K K Ks 6BB !B c R t j t j dt j d }t | dk sJ t d |D sJ t
t 5 t j dgd d d d d S # 1 swxY w Y d S )Nr rL ry r) c 3 , K | ]}|j d k V dS )rz N)r ).0as rG zATestRavelUnravelIndex.test_empty_array_unravel.. s( 00q17d?000000rT r )r? r@ r{ rZ lenallr rC )rE r| s rG test_empty_array_unravelz.TestRavelUnravelIndex.test_empty_array_unravel s rx 9 9 99EE3xx1}}}}00C00000110
:
&
& - - aS) , , , - - - - - - - - - - - - - - - - - -s 8B B #B N)__name__
__module__ __qualname__rH rS r` ri rm rt rw pytestmark parametrizer} r rK rT rG r r s 5I 5I 5In" " "(I I I0O O O6
L
L
L# # # D D D [ V%>%>%>??K K @?K- - - - -rT r c e Zd Zd Zd Zd Zd Zej dddg d Z
d Z d
Z d Z
d S )
TestGridc t ddd }t ddd }t |j dk t |j dk t |d dk t |d d t |d dk t |d |d z
dd t |d |d d z d t |d |d z
d
d d S ) Nr$ r $@r' )rJ ) r ggfffff?g q q?)r
r r r )rE r bs rG rH zTestGrid.test_basic s "Qs(O "Qs(O5 !!!5 !!!!
AbE1%%%!
AaD1IsB///AbE1Q4;333AaD1Iw33333rT c t j ddd \ }}t |d t |t ddd d d S )Nr rJ T)retstepg9/?y I@r4 )r? linspacer r r
)rE rs sts rG test_linspace_equivalencez"TestGrid.test_linspace_equivalence sN Ar40002B'''!!U1R8_b99999rT c d t ddddddf }t ddddddf }t |j dk t |j d k t |d
d
d d f t j d d t |d d d d
f dt j d d z t
|d
dd d f t j d d d
t
|d d d df dt j d d z d
t
|d
dd d f |d
d
d d f z
dt j dd z d
t
|dd d df |dd d d
f z
dt j dd z d
d S )Nr$ r r r r' g?)r rJ rJ )r r r r rJ dr r )r
r r r r? onesr )rE cr s rG test_ndzTestGrid.test_nd s "Qs(BqH$ % "Qs(BqH$ %;&''';&'''1Q4111:C(8(8'89991Q41:r"'"c*:*:':;;;!!A$r111u+rwr3/?/?DDD!!A$qqq"u+qS1A1A/A2FFF!!Aq!!!G*qAqqqz"9"%bgb#&6&6"6 < < <!!Aqqq!G*qAAAqz"9"%bgb#&6&6"6 < < < <