Current File : //usr/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyo
�
E�`Qc@s#ddlTddlZddlmZmZmZmZmZmZm	Z	m
Z
mZmZm
Z
mZmZdefd��YZdefd��YZdefd	��YZd
efd��YZdefd
��YZd�Zd�Zd�Zd�Zd�Zedkre�ndS(i����(t*N(
tarraytonestr_tmgridt
unravel_indextzerostwheretndenumeratet
fill_diagonaltdiag_indicestdiag_indices_fromts_t	index_exptTestRavelUnravelIndexcBs#eZd�Zd�Zd�ZRS(cCs�ttjdd"�d#�ttjd$d%�d�ttjdd&�d'�ttjd(d)�d�tttjdd*�tttjd	d+�tttjd
d,�tttjd-d.�tttjd/d0�tttjd1d2�tttjd3d4�tttjd5d6�ttjd:d;�ddd
g�ttjddd
gd<�d@�tjdddgd
ddgg�}ttj|dA�dddg�ttj|dBdd�dddg�ttj|dCdd�dddg�ttj|dDddE�dddg�ttjdFdG�d!�ttjtjdddg�dH�dddgd
ddgg�ttjtjdddg�dIdd�dddgd
ddgg�ttjd!dJ�ddd
dg�dS(KNiiii�ii^iBi����g�?ii����g�������?giiiiii)i%tordertFii
tmodetclipiitwrapiii	iU(ii(ii(ii(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(iiii(iiii	(ii(ii(iiii	(tassert_equaltnpRtravel_multi_indext
assert_raisest
ValueErrort	TypeErrorR(tselftarr((sG/usr/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyt
test_basics4""'"(($*c
Cs(x!tjtjtjtjtjtjgD]�}tjddddddgddddddggd|�}d}d	|d|d}ttj	||�|�t|tj
||��|dd|d}ttj	||d
d�|�t|tj
||d
d��tjddddddgddddddgddddddggd|�}d}d
d	|d|d|d}ttj	||�|�t|tj
||��|dd|dd	|d}ttj	||d
d�|�t|tj
||d
d��q+WdS(NiiiiiitdtypeiiRRi	i
(ii(iii
(Rtint16tuint16tint32tuint32tint64tuint64RRRR(RRtcoordstshapetuncoords((sG/usr/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyttest_dtypes's&?K	""c	Cs�ttjddddgdd	d
�tjddddgd��ttjddddgdd	d�tjddddgd��tttjddddgd�dS(Niii����iiiiiRRitraiseRi(iiii(iiii(iiii(swrapR(sclipR((iiii(iiii(RRRRR(R((sG/usr/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyttest_clipmodes>s$(t__name__t
__module__RR'R)(((sG/usr/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyRs		tTestGridcBs#eZd�Zd�Zd�ZRS(cCs�tddd�}tddd�}t|jdk�t|jd
k�t|ddk�t|dd�t|ddk�t|d|ddd�t|d|ddd�t|d|dd
dd�dS(Ni����iy$@g�������?i
iiiig@g"@(i
(iggfffff�?(Rtassert_R%tassert_almost_equal(Rtatb((sG/usr/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyRHscCsPtjdddd�\}}t|dd�t|tddd�d�dS(	Nii
tretstepiig�H@yI@i
(RtlinspaceR.tassert_array_almost_equalR(Rtytst((sG/usr/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyttest_linspace_equivalenceTscCs�tddd�ddd�f}tddd�ddd�f}t|jd
k�t|jdk�t|d
d
dd�ftdd��t|ddd�d
fdtdd��t|d
ddd�ftdd�d�t|ddd�dfdtdd�d�t|d
ddd�f|d
d
dd�fdtd	d�d�t|ddd�df|ddd�d
fdtd	d�d�dS(Ni����iy$@i����ig�������?g�������?i
iitdi(ii
i
(iii(RR-R%tassert_array_equalRR3(RtcR7((sG/usr/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyttest_ndYs""+.-1G(R*R+RR6R:(((sG/usr/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyR,Gs		tTestConcatenatorcBs,eZd�Zd�Zd�Zd�ZRS(cCs�ttdtddddddg��td�}t|dd|f}t|ddddddddddddg�dS(	Niiiiiii(iiiiii(R8RRR(RR0R9((sG/usr/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyttest_1dgs)cCs-tddd�f}t|jdk�dS(Ng333333$@ii
tf8(RR-R(Rtg((sG/usr/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyttest_mixed_typemscCsEtdtdg�tdddg�df}t|jdk�dS(Ng333333$�iiiig$@R=(RRR-R(RR>((sG/usr/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyttest_more_mixed_typeqs.cCs�tdd�}tdd�}td||f}t|jdk�t|dd�dd�f|�t|dd�dd�f|�t||f}t|jdk�t|dd�dd�f|�t|dd�dd�f|�dS(Nit1i
(ii
(i
i(trandRR-R%R8(RR0R9R7((sG/usr/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyttest_2dus###(R*R+R<R?R@RC(((sG/usr/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyR;fs			tTestNdenumeratecBseZd�ZRS(cCsGtddgddgg�}ttt|��dd	dd
g�dS(Niiiii(ii((iii(ii((iii(ii((iii(ii((iii(RRtlistR(RR/((sG/usr/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyR�s(R*R+R(((sG/usr/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyRD�stTestIndexExpressioncBseZd�Zd�ZRS(cCsEtjd�}t|d |td �t|d |td �dS(Nii����(RtarangeRRR
(RR/((sG/usr/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyttest_regression_1�scCs�tjjddd�}t|dd�dd�ddgf|tdd�dd�ddgf�t|dd�dd�ddgf|tdd�dd�ddgf�dS(Niiiiii(RtrandomRBRR
R(RR/((sG/usr/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyt
test_simple_1�sO(R*R+RHRJ(((sG/usr/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyRF�s	c
Csotjtjdddgg�ddtjdddgg�f}t|ddddddddgg�dS(Niiiiiii(Rtc_RR(R/((sG/usr/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyttest_c_�sCccs�tdt�}t|d�t|tdddgdddgdddgg�fVtd	t�}t|d�t|tdddgdddgdddgdddgdddgdddgdddgdddgdddgdddgg
�fVtd
t�}t|dt�t|tdddgdddgdddgdddgdddgdddgdddgdddgdddgdddgg
�fVtdt�}t|d�t|tddddddddddg
ddddddddddg
ddddddddddg
g�fVtdt�}t|d�tdddg�}tt|dk�||||ffVdS(
Niiii
iii(ii(i
i(i
i(ii
(iiii(RtintR	R8RtTrueRR(R/ti((sG/usr/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyttest_fill_diagonal�sT


$!,
c
cstd�}tddddgddddgd	d
ddgd
dddgg�}d||<t|tddddgddddgd	d
ddgd
dddgg�fVtdd�}tdt�}d||<t|tddgddggddgddggg�fVdS(Niiiiiiiii	i
iii
iiiidi(iii(R
RR8RRM(tdiR/td3((sG/usr/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyttest_diag_indices�s&

	cCsTtjjd�}t|�\}}t|tjd��t|tjd��dS(Ni(ii(RRIRR8RG(txtrR9((sG/usr/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyttest_diag_indices_from�scCs�ttjddd��}gtjtjd��D]\}}|^q4}t||�ttjd��}t||�ttj��}t|dg�ttjd��}t|dg�dS(	Niii(iii(iii((((RERtndindexRRR8R(RTtixtetexpected((sG/usr/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyttest_ndindex�s1

t__main__(t
numpy.testingtnumpyRRRRRRRRRR	R
RRR
tTestCaseRR,R;RDRFRLRPRSRVR[R*trun_module_suite(((sG/usr/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyt<module>s
X@
		5