�
�9Zc@`
ss
dd
lm
Z
mZmZ
ddlZddlmZmZm	Z	m
Z
mZmZm
Z
mZ
ddlmZmZmZmZmZmZmZmZmZmZ
def
d��YZdef
d��YZd	ef
d
��YZdef
d��YZd
ef
d��YZdef
d��YZd�Z d�Z!d�Z"d�Z#d�Z$e%dkro
e�
ndS(i(tdivisiontabsolute_importtprint_functionN(trun_module_suitetTestCasetassert_tassert_equaltassert_array_equaltassert_almost_equaltassert_array_almost_equalt
assert_raises(
tmgridtndenumeratet
fill_diagonaltdiag_indicestdiag_indices_fromt	index_exptndindextr_ts_tix_tTestRavelUnravelIndexc
B`
s#eZ
d�Zd
�Zd�ZRS(c
C`
s�tt
jd
d"�d#�
tt
jd$d%�d
�
tt
jdd&�d'�
tt
jd(d)�d�
ttt
jdd*�
ttt
jd	d+�
ttt
jd
d,�
ttt
jd-d.�
ttt
jd/d0�
ttt
jd1d2�
ttt
jd3d4�
ttt
jd5d6�
tt
jd:d;�d
dd
g�
tt
jd
dd
gd<�d@�
t
jdddgd
ddgg�
}
tt
j|
dA�dddg�
tt
j|
dBdd�
dddg�
tt
j|
dCdd�
dddg�
tt
j|
dDddE�
dddg�
tt
jdFdG�d!�
tt
jt
jdddg�
dH�dddgd
ddgg�
tt
jt
jdddg�
dIdd�
dddgd
ddgg�
tt
jd!dJ�ddd
dg�
dS(KNii
ii�ii^iBi����g�?ii����g�������?giiiiii)i%tordert
Fii
tmodetclipiitwrapiii	iU(ii(i
i(i
i(ii(ii^(iiB(iiB(ii^(ii(ii(ii(i����i
(ii(ii
(ii(ii����(ii(ii(ii(g�������?g(iiiii*i.(iii(iiiiii*i.(ii(ii(ii(ii(RR(ii
ii
(iiii	(ii(ii(iiii	(Rtnpt
unravel_indextravel_multi_indexR
t
ValueErrort	TypeErrortarray(tselftarr((sT/opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyt
test_basics<











"

'
"

%

%


$


'

c

C`
s(x!tj
tjtjtjtjtjgD]�
}
tjd
dd
dddgd
dd
dddggd|
�

}d}d	|d|d
}ttj	||�|�
t|tj
||��
|dd|d
}ttj	||d
d�
|�
t|tj
||d
d�
�
tjd
dd
dddgd
dd
dddgd
dd
dddggd|
�

}d}d
d	|d|d
|d}ttj	||�|�
t|tj
||��
|dd|d
d	|d}ttj	||d
d�
|�
t|tj
||d
d�
�
q+WdS(Ni
iiiiitdtypeiiRRi	i
(ii(iii
(Rtint16tuint16tint32tuint32tint64tuint64R RRR(R!R$tcoordstshapetuncoords((sT/opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyttest_dtypes2s.


9








E
	

"


"


c

	C`
s�tt
jd
dddgdd	d
�
t
jddddgd��
tt
jd
dddgdd	d�
t
jddddgd��
ttt
jd
dddgd�
dS(Nii
i����iiiiiRRitraiseRi(iiii(iiii(iiii(RR/RR/(iiii(iiii(RRRR
R(
R!((sT/opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyttest_clipmodesMs
!




(t__name__t
__module__R#R.R0(((sT/opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyRs
	#	tTestGridc
B`
s#eZ
d�Zd
�Zd�ZRS(c
C`
s�td
dd�}
td
dd�}t
|
jdk�

t
|jd
k�

t
|
dd
k�

t|
d
d�
t
|dd
k�

t|d|ddd�
t|d
|ddd�
t|
d|
ddd�
dS(Ni����i
y$@g�������?i
iiiig@g"@(
i
(
iggfffff�?g�q�q�?(RRR,R(R!t
at
b((sT/opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyR#[s









c
C`
sLtj
d
ddd�
\}
}t|d	�
t|
td
dd�d�
dS(
Nii
tretstepi
ig�H@yI@i
g9/����?(RtlinspaceRR	R(R!t
ytst((sT/opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyttest_linspace_equivalencegs



c
C`
s�
td
dd�ddd�f}
td
dd�ddd�f}t
|
jd
k�

t
|jdk�

t|
d
d
dd�ftjdd��
t|
ddd�d
fdtjdd��
t|
d
d
dd�ftjdd�d�
t|
ddd�d
fdtjdd�d�
t|d
ddd�f|d
d
dd�fdtjd	d�d�
t|ddd�df|ddd�d
fdtjd	d�d�
dS(Ni����i
y$@i����ig�������?g�������?i
iit
di(ii
i
(iii(RRR,RRtonesR	(R!t
cR;((sT/opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyttest_ndls
"
"


.
1
0
4
0

0
(R1R2R#R:R>(((sT/opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyR3Zs
		tTestConcatenatorc
B`
s,eZ
d�Zd
�Zd�Zd�ZRS(c
C`
s�tt
dtjd
dddddg�
�
tjd�
}
t
|
dd|
f}t|d
d
d
d
d
ddd
d
d
d
d
g�
dS(	Ni
iiiiii(i
iiiii(RRRR R<(R!R5R=((sT/opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyttest_1d|s
,


c
C`
s-td
dd�f}
t
|
jdk�

dS(Ng333333$@i
i
tf8(RRR$(R!t
g((sT/opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyttest_mixed_type�s

c
C`
sKtd
t
jdg
�
t
jdddg�
df}
t|
jdk�

dS(Ng333333$�i
iiig$@RA(RRR RR$(R!RB((sT/opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyttest_more_mixed_type�s
4
c
C`
s
tj
jd
d
�}
tj
jd
d
�}td|
|f}t|jdk�

t|dd�dd
�f|
�
t|dd�d
d�f|�
t|
|f}t|jdk�

t|dd
�dd�f|
�
t|d
d�dd�f|�
dS(Nit
1i
(ii
(i
i(RtrandomtrandRRR,R(R!R5R=R;((sT/opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyttest_2d�s




#
#


#
(R1R2R@RCRDRH(((sT/opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyR?{s
			tTestNdenumeratec
B`
seZ
d�ZRS(
c
C`
sJtj
d
dgddgg�
}
ttt|
�
�
dd	dd
g�
dS(Ni
iiii(ii((iii
(ii
((ii
i(i
i((i
ii(i
i
((i
i
i(RR RtlistR(R!R4((sT/opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyR#�s
!

(R1R2R#(((sT/opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyRI�s
tTestIndexExpressionc
B`
seZ
d�Zd
�ZRS(c
C`
sEtj
d
�
}
t|
d |
td �
t|
d |
td �
dS(Nii����(RtarangeRRR(R!R4((sT/opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyttest_regression_1�s

c
C`
s�tj
jd
dd�}
t|
dd�dd�ddgf|
tdd�dd�ddgf�
t|
dd�dd�ddgf|
tdd�dd�ddgf�
dS(Niiiii
i(RRFRGRRR(R!R4((sT/opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyt
test_simple_1�s
O
(R1R2RMRN(((sT/opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyRK�s
	tTestIx_c
B`
s5eZ
d�Zd
�Zd�Zd�Zd�ZRS(c
C`
sIxBtt
jfD]1}
t
j|
d
�
�
\
}t|jt
j�
qWdS(Ni(trangeRRLRRR$tintp(R!tfuncR4((sT/opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyRM�s

c
`
s�d}
x�tt
jfD]�}t
jg|
D]}||�
^q)�}xtt||
��
D]h\�\}}t|j�|�
tt	�f
d�t|j�
D�
�
�

tt
j
|jt��

qZWqWdS(Niiiic
3`
s-|]#\}
}|
�kr|dkV
qd
S(i
N((t.0t
jtsh(
t
k(sT/opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pys	<genexpr>�s(iiii(
RPRRLRt	enumeratetzipRR,Rtallt
issubdtypeR$tint(R!tsizesRRtsztarraysR4((
RVsT/opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyttest_shape_and_dtype�s

(
(

)
c
C`
sBtt
ttg}
tj|
�
\
}ttj|
�
d
|�
dS(Ni(tTruetFalseRtnonzeroRR(R!tbool_atint_a((sT/opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyt	test_bool�s


c
C`
s5d
ddgdddgg}
tt
tj|
�
dS(Ni
iiiii(R
RRR(R!tidx2d((sT/opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyttest_1d_only�s

c
C`
sod
}
tj
|
�
}t||�}t|dj|
df�
t|djd|
f�
t|j|
f
�
dS(Niii
(RRLRRR,(R!tlength_of_vectort
xtout((sT/opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyttest_repeated_input�s




(R1R2RMR_ReRgRk(((sT/opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyRO�s

		
		c

C`
sotj
tjd
ddgg
�
ddtjdddgg
�
f}t|d
dddddddgg
�
dS(Ni
iiiiii(Rtc_R R(
R4((sT/opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyttest_c_�s
C
cc`
s�tj
dt�}t|d�
t|tjdddgdddgdddgg�
fV
tj
d	t�}t|d�
t|tjdddgdddgdddgdddgdddgdddgdddgdddgdddgdddgg
�
fV
tj
d
t�}t|dt�
t|tjdddgdddgdddgdddgdddgdddgdddgdddgdddgdddgg
�
fV
tj
dt�}t|d�
t|tjddddddddddg
ddddddddddg
ddddddddddg
g�
fV
tj
dt�}t|d�
tjdddg�
}
ttj|dk�
|
|
|
|
ffV
dS(
Niiii
ii
i(ii(i
i(i
i(ii
(iiii(	RtzerosR[R
RR R`Rtwhere(R4t
i((sT/opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyttest_fill_diagonal�sT



































'
!
,



c
c`
s$
td
�
}t
jdddd
gddddgd	d
ddgd
dddgg�
}
d|
|<t|
t
jdddd
gddddgd	d
ddgd
dddgg�
fV
tdd�}t
jdt�}
d|
|<t|
t
jddgddggddgddggg�
fV
dS(Nii
iiiiiii	i
iii
iiiidi(iii(RRR RRnR[(tdiR4td3((sT/opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyttest_diag_indices
s&
















	
cC`
sTtj
j
d�
}t|�
\}
}t|
tjd
�
�
t|tjd
�
�
dS(Ni(ii(RRFRRRL(Rit
rR=((sT/opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyttest_diag_indices_from&
s



cC`
s
tt
d
dd��
}gttjd�
�
D]\}
}|
^q.}t||�
tt
d�
�
}t||�
tt
d�
�
}t|tt
d�
�
�
tt
��
}t|dg
�
tt
d	�
�
}t|d
g
�
tt
dg
��
}t|g�
dS(Ni
iii(i
ii(i
ii(
i((((RJRRRRnRR(Ritixt
etexpected((sT/opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyttest_ndindex-
s

.







t__main__(&t
__future__RR
RtnumpyRt
numpy.testingRRRRRRR	R
tnumpy.lib.index_tricksRRR
RRRRRRRRR3R?RIRKRORmRqRtRvRzR1(((sT/opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyt<module>
s 
:FL!%		5