Current File : //usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyc
�
E�`Qc@s�dZdZdZdZdZddlZddlmZm	Z	ddl
TddlTddlTd	efd
��YZ
defd��YZd
efd��YZdefd��YZdefd��YZdefd��YZdefd��YZdefd��YZdefd��YZdefd��YZdefd��YZdefd ��YZd!efd"��YZed#kr�e	�ndS($s�Tests suite for MaskedArray.
Adapted from the original test_ma by Pierre Gerard-Marchant

:author: Pierre Gerard-Marchant
:contact: pierregm_at_uga_dot_edu
:version: $Id: test_extras.py 3473 2007-10-29 15:18:13Z jarrod.millman $
s5Pierre GF Gerard-Marchant ($Author: jarrod.millman $)s1.0s$Revision: 3473 $s5$Date: 2007-10-29 17:18:13 +0200 (Mon, 29 Oct 2007) $i����N(tTestCasetrun_module_suite(t*tTestGenericcBs5eZd�Zd�Zd�Zd�Zd�ZRS(cCs4tddt�}tddgdddgdt�}t||�tjiddgd6ddgd	6�}tdd|�}tddgdddgd|�}t||�tdd|�}tddgddggdddgddggd|�}t||�tjddddgfg�}td d|�}tdd!fdd"fgddd#fdd$fgd|�}t||�td%d|�}tdd&fdd'fgddd(fdd)fgd|�}t||�td*d|�}tdd+fggddd,fggd|�}t||�d
S(-sTests masked_allitdtypeitmasktatbtnamestftformatsitbatbbN(i(i(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(RR	(RR	(RR	(i(ii(ii(ii(ii(i(ii(ii(ii(ii(ii(ii(ii(t
masked_alltfloattarraytassert_equaltnpR(tselfttesttcontroltdt((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyttest_masked_alls0$
)$
	
!!
!
0cCsYtddgdt�}t|�}tddgdddgdt�}t||�tjiddgd6ddgd	6�}tddgdddgd|�}t|�}tddgdddgd|�}t||�tjddddgfg�}tddfddfgdddfddfgd|�}t|�}t||�dS(sTests masked_alliiRRRRRR	R
ii
RRN(ii(ii(ii(ii(i
i
(i
i
(ii(ii(RR	(sbaR	(sbbR	(ii(ii(ii(ii(RRtmasked_all_likeRRR(RtbaseRRR((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyttest_masked_all_like6s$
)$$
!!cCsxttjd��}t|ddddddg<t|�}tdd�tdd	�tdd�g}t||�d
S(sTest clump_maskedi
iiiiii	iiN(tmasked_arrayRtarangetmaskedtclump_maskedtsliceR(RRRR((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyttest_clump_maskedJs
*cCslttjd��}t|ddddddg<t|�}tdd�td	d�g}t||�d
S(sTest clump_unmaskedi
iiiiii	iiN(RRRRtclump_unmaskedRR(RRRR((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyttest_clump_unmaskedSs
cCs�td�}t|�}t|td|j��t||dk|dkB|dkB<t|�}t|tdd�tdd�g�t|(t|�}t|d�dS(	sTest flatnotmasked_contiguousi
iiiiii	N(Rtflatnotmasked_contiguousRRtsizeRtNone(RRR((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyttest_flatnotmasked_contiguous[s$%(t__name__t
__module__RRRR!R%(((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyRs
					tTestAveragecBs2eZdZd�Zd�Zd�Zd�ZRS(s5Several tests of average. Why so many ? Good point...c	Cs�tddddgdttttg�}tdt|dd��tdt|dddddg��t|dddddgd	d
�\}}td|�|j|dk�t|(tt|dd�jtg�tddddgdttttg�}|jdd�}t|d
d
�d
f<tt|dd�ddg�tt|dd
�jdtg�tddgt|dd��t|ddd	d
�\}}t|ddg�d
S(sTest of average.gg�?g@g@Rtaxisitweightstreturnedig@iN(	RtTruetFalseRtaveraget
assertTrueRRtreshape(Rtotttresulttwts((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyttest_testAverage1ms *%*
*#cCs�ddddddg}ddddddgddddddgg}tddt�}tt|dd�d�tt|ddd|�d�ttddt�dtd�g�}tt|d�tjjtjd��d	d
�tt|dd�tjd�d	d�tt|dd�t|dd�t|dd�dg�tt|dd|�dd�tt|ddd|�d
ddd	ddg�tt|dd�t|dd�t|dd�dg�t	d�}ddddddg}ddddddgddddddgg}t
d�}ddddddg}	ttt||�dd�d�ttt||�dd�d�ttt||�dd�jt
g�ttt||	�dd�d
�tttt||�dd��d�t||�}
tt|
d�dd�tt|
dd�d
dddddg�tt|
dd�ddg�tt|
ddd|�d
dddddg�dS(sMore tests of average.iiiRR)g@R*g@g@g(@g4@g@gg�?g@g$@g�X@g@g@N(Rtfloat_RR.RR$RtaddtreducetzerostonesRRR,tcount(Rtw1tw2txtytm1tm2tm3tm4tm5tz((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyttest_testAverage2�s>0(3*) )0""("(+cCstd�}td�d}t||g||ggdddd�\}}tt|�t|��t|j|j�ttd�dddddgdd�\}}tt|�t|��ttd�dd�\}}tt|�t|��ttd�dtd�dd�\}}tt|�t|��tddgdd	ggt�}t|ttgt	tgg�}t|dd�}	t|	d
dg�t|dd�}
t|
ddg�t|dd�}
t|
d
d�t|dd�}
t|
ddg�dS(sYet more tests of average!iiR)iR+iiR*ig�?g@g�?g@g�?g@N(iii(iii(iii(iii(RR.RtshapeR9RRRR-R,R$(RRRtr1R;tr2R<ta2dta2dmta2data2dma((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyttest_testAverage3�s*00*!!cCsfttddg��}t|d�ttddddgdttttg��}t|d�dS(s"Test average on integers with maskiig�?iiRN(R.RRR-R,(RR((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyttest_onintegers_with_mask�s
0(R&R't__doc__R4RERMRN(((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyR(ks
		"	tTestConcatenatorcBs eZdZd�Zd�ZRS(s@
    Tests for mr_, the equivalent of r_ for masked arrays.
    cCs�ttd
tddddddg��td�}dddddg}t|d|�}t|dd|f}|jt|t�p�t|tj��t|ddddddddddddg�t|j	t|dd|f�d	S(sTests mr_ on 1D arrays.iiiiiiiRN(iiiiii(
tassert_array_equaltmr_RR9RR/t
isinstancetMaskedArraytcoreR(RRtmtdtc((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyttest_1d�s)(1cCs�tdd�}tdd�}tjtdd�d�}tjtdd�d�}t|d|�}t|d|�}td||f}|j|jdk�t|dd�dd�f|�t|dd�dd�f|�t|jtj	d||f�t||f}|j|jdk�t|dd�dd�f|�t|dd�dd�f|�t|jtj	||f�dS(	sTests mr_ on 2D arrays.iiRt1i
N(ii
(i
i(
trandRtround_RRRR/RFRQRtr_(Rta_1ta_2tm_1tm_2tb_1tb_2RW((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyttest_2d�s ## ##(R&R'RORYRd(((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyRP�s	t
TestNotMaskedcBs eZdZd�Zd�ZRS(s9
    Tests notmasked_edges and notmasked_contiguous.
    cCsJttjd�jdd�ddddddgdddddgdddddgdddddgdddddgg�}t|d�}t|ddg�t|d�}t|dd
dg�t|dddg�t|d�}t|dddg�t|dddg�t|jd�}t|ddg�t|jd�}t|dddg�t|dddg�t|jd
�}t|dddg�t|dddg�t|d<t|d�}t|dddg�t|ddd g�t|d
�}t|dd!d"g�t|dd#d$g�dS(%sTests unmasked_edgesiiRiiiiiii����i����N(iiiii(iiiii(iiiii(iiiii(iiiii(iiiii(iiiii(iiiii(iiiii(iiiii(iiiii(iiiii(iiiii(iiiii(iiiii(iiiii(iiiii(iiiii(iiiii(iiiii(iiii(iiii(iiii(iiii(	RRRR0tnotmasked_edgesR$RtdataR(RRgR((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt
test_edges�s:
c
Cs�ttjd�jdd�dddddddddgddddddddgddddddddgg�}t|d�}t|dtddd��t|d	td
dd��t|dtdd
d��t|d�}|jt	|d�dk�|j|d	dk�t|d|d�|jt	|d�dk�t|d�}t|ddtdd
d��|j|ddk�t|ddtddd��t|dd	tddd��dS(sTests notmasked_contiguousiiiRiii����ii����iii����iiiiN(
RRRR0tnotmasked_contiguousR$RRR/tlen(RRttmp((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyttest_contiguouss$$!!(R&R'RORhRl(((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyRe�s	#tTest2DFunctionscBs)eZdZd�Zd�Zd�ZRS(sTests 2D functionscCs�ttjd�jdd�ddddgdddgdddgg�}tt|�ddgdd	gg�tt|d�dddgd
dd	gg�tt|d�ddgddgdd	gg�t|jddddgdddgdddgg�}tt|�ddgd
d	gg�tt|d�dddgd
dd	gg�tt|d�ddgddgd
d	gg�t|jddddgdddgdddgg�}tt|�d	gg�tt|d�d
dd	gg�tt|d�dgdgd	gg�t|jddddgdddgdddgg�}tt|�jd�tt|d�jd�tt|d�jd�dS(
sTests compress2di	iRiiiiiiiiN(RRRR0Rtcompress_rowcolst_dataR#(RR=((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyttest_compress2d*s K%.19%.19"(9cCs�ttjd�jdd�ddddgdddgdddgg�}tt|�jdddgdddgdddgg�tt|d�jdddgdddgdddgg�tt|d�jdddgdddgdddgg�t|jddddgdddgdddgg�}tt|�jdddgdddgdddgg�tt|d�jdddgdddgdddgg�tt|d�jdddgdddgdddgg�t|jddddgdddgdddgg�}tt|�jdddgdddgdddgg�tt|d�jdddgdddgdddgg�tt|d�jdddgdddgdddgg�t|jddddgdddgdddgg�}|jt|�j	�t
k�|jt|d�j	�t
k�|jt|d�j	�t
k�|jt|�jj	��|jt|d�jj	��|jt|d�jj	��dS(sTests mask_rowcols.i	iRiiN(RRRR0Rtmask_rowcolsRRoR/tallR(RR=((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyttest_mask_rowcols=s&K:==9:==9:==9""c	CsGtjdd�}ddddddg}t|d|�jdd�}t|d|�jdd�}t||t�}t|jddgddgg�t||t�}t|jdddgdddgdddgg�t||t�}t|tj|j	d�|j	d���t||t�}t|tj|j	d�|j	d���ddddddg}t|d|�jdd�}t|d|�jdd�}t||t�}t|jddgddgg�t||t�}t|jdddgdddgdddgg�t||t�}t|tj|j	d�|j	d���t|t||��t||t�}t|tj|j	d�|j	d���ddddddg}t|d|�jdd�}t|d|�jdd�}t||�}t|jt
�t||�}t|jt
�t|dddddddg�jdd�}t|dddddddg�jdd�}t||t�}t|jddgddgg�t||t�}t|tj|j	d�|j	d���t||t�}t|jdddgdddgdddgg�t||t�}t|tj|j	d�|j	d���t|dddddddg�jdd�}t|dddddddg�jdd�}t||t�}t|jddgddgg�t||�}t|tj|j	d�|j	d���t||t�}t|jdddgdddgdddgg�t||t�}t|tj|j	d�|j	d���t|dddddddg�jdd�}t|dddddddg�jdd�}t||t�}t|jddgddgg�t||t�}t|tj|j	d�|j	d���t||t�}t|jdddgdddgdddgg�t||t�}t|tj|j	d�|j	d���dS(sTests dot productiiiRiiN(RRRR0tdotR,RRR-tfilledtnomask(RtnRVRRRX((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyttest_dotSsz"4++"4++00"+4+00"+4+00"+4(R&R'RORpRsRx(((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyRm(s		tTestApplyAlongAxiscBseZdZd�ZRS(sTests 2D functionscCsYtd�jddd�}d�}t|d|�}t|ddgddgg�dS(	Ng(@iicSs|dS(Ni((R((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pytmyfunc�siiii
(RR0tapply_along_axisR(RRRztxa((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyttest_3d�s	(R&R'ROR}(((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyRy�stTestApplyOverAxescBseZdZd�ZRS(sTests apply_over_axescCs�td�jddd�}ttj|ddg�}tjdgdgdggg�}t||�t||djtj	�<ttj|ddg�}tjd	gd
gdggg�}dS(Niiiiii<i\i|ii,(
RR0tapply_over_axesRtsumRRRtastypetbool(RRRtctrl((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt
test_basic�s$
(R&R'ROR�(((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyR~�st
TestMediancBs#eZd�Zd�Zd�ZRS(cCsKd\}}ttjdd|��}t|d*|d)ttj||fdt��}||dd�d	f<tjt|��}xAtd
|�D]0}tj	j
|�|||dd�|f<q�Wtt|dd�d	f�d	�tt|�d	�tt|dd	�tj
|��tt|jdd
�tj
|��dS(
sTests median w/ 2Dieig�g�?i
i����RNiiR)(iei(RRtlinspaceRtemptyRRRjtrangetrandomtshuffleRtmedianR8tT(RRwtpR=RDtidxti((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyRd�s!#"cCs�ttjd�jdd��}t|d*|d)tt|�d�tt|dd�ddd	g�tt|dd
�ddddddd
dddg
�tt|dd
�jd
d
d
ddddd
d
d
g
�dS(s-Tests median w/ 2D arrays and different axis.ii
ii����g-@R)ig+@g/@ii
iiN(RRRR0RRR�R(RR=((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt
test_2d_waxis�s!"7cCstjjd�jddd�}t||ddk<tt|d�ddgdd	gddgd
d	gg�d|_tt|d�ddgd
dgddgg�tjjd�jddd�}t||ddk<tt|d�ddgddgddgg�dS(sTests median w/ 3Diiiiiii	iiiici
ii
iiiiiN(iii(RtmaRR0RRR�RF(RR=((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyR}�s!:	1!(R&R'RdR�R}(((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyR��s			tTestCovcBs5eZd�Zd�Zd�Zd�Zd�ZRS(cCsttjjd��|_dS(Ni(RRR�R[Rg(R((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pytsetUp�scCs�|j}ttj|�t|��ttj|dt�t|dt��ttj|dtdt�t|dtdt��dS(s*Test cov on 1D variable w/o missing valuestrowvartbiasN(Rgtassert_almost_equalRtcovR-R,(RR=((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyttest_1d_wo_missing�s
	(cCs�|jjdd�}ttj|�t|��ttj|dt�t|dt��ttj|dtdt�t|dtdt��dS(s,Test cov on 1 2D variable w/o missing valuesiiR�R�N(RgR0R�RR�R-R,(RR=((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyttest_2d_wo_missing�s
(c	Cs�|j}t|d<||j�8}|j�}ttj|�t|��ttj|dt�t|dt��ttj|dtdt�t|dtdt��yt|dt�Wnt	k
r�nX|dd!}ttj||ddd��t||ddd���ttj||ddd�dt�t||ddd�dt��ttj||ddd�dtdt�t||ddd�dtdt��dS(s'Test cov 1 1D variable w/missing valuesi����R�R�tallow_maskediN(
RgRtmeant
compressedR�RR�R-R,t
ValueError(RR=tnx((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyttest_1d_w_missing�s$	
(

<%#+cCs�|j}t|d<|jdd�}tjt|��jt�}tj||j	�}||j
d�dd�dfjd�}t
t|�tj|�|jdd|d�t
t|dt�tj|dt�|jd|�tj|j	|�}||j
d�jd�}t
t|d	t�tj|d	t�|jdd|d�t
t|d	tdt�tj|d	tdt�|jd|�dS(
s(Test cov on 2D variable w/ missing valuei����iiiNig�?R�R�(RgRR0Rtlogical_nottgetmaskarrayR�tintRtR�R�R$RuR�R�RFR,R-(RR=tvalidtfractxf((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyttest_2d_w_missings	
,3%-(R&R'R�R�R�R�R�(((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyR��s
				tTestCorrcoefcBs>eZd�Zd�Zd�Zd�Zd�Zd�ZRS(cCsttjjd��|_dS(Ni(RRR�R[Rg(R((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyR�scCs5|j}ttj|dd�t|dd��dS(sTest ddof keywordtddofiN(RgR�Rtcorrcoef(RR=((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt	test_ddof!s	cCs�|j}ttj|�t|��ttj|dt�t|dt��ttj|dtdt�t|dtdt��dS(s*Test cov on 1D variable w/o missing valuesR�R�N(RgR�RR�R-R,(RR=((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyR�'s	cCs�|jjdd�}ttj|�t|��ttj|dt�t|dt��ttj|dtdt�t|dtdt��dS(s1Test corrcoef on 1 2D variable w/o missing valuesiiR�R�N(RgR0R�RR�R-R,(RR=((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyR�0sc	Cs�|j}t|d<||j�8}|j�}ttj|�t|��ttj|dt�t|dt��ttj|dtdt�t|dtdt��yt|dt�Wnt	k
r�nX|dd!}ttj||ddd��t||ddd���ttj||ddd�dt�t||ddd�dt��ttj||ddd�dtdt�t||ddd�dtdt��dS(s,Test corrcoef 1 1D variable w/missing valuesi����R�R�R�iN(
RgRR�R�R�RR�R-R,R�(RR=R�((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyR�9s$	
(

<%#+cCs}|j}t|d<|jdd�}t|�}tj|�}t|dd�dd�f|dd�dd�f�dS(s-Test corrcoef on 2D variable w/ missing valuei����iiN(RgRR0R�RR�(RR=RR((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyR�Qs	
(R&R'R�R�R�R�R�R�(((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyR�s							tTestPolynomialcBseZd�ZRS(cCs�tjjd�}tjjd�jdd�}tt||d�tj||d��|jt�}t|d<|jt�}t|d<|d<t||dd�dfddt	�\}}}}}tj|d	|d	d�dfj
�ddt	�\}}	}
}}xHt|||||f||	|
||f�D]\}
}t|
|�q;Wt||dd�dfddt	�\}}}}}tj|d	d!|d	d�dfddt	�\}}	}
}}xHt|||||f||	|
||f�D]\}
}t|
|�q�Wt||ddt	�\}}}}}tj|d	d!|d	d�dd�fddt	�\}}	}
}}xHt|||||f||	|
||f�D]\}
}t|
|�q�Wtjjd�d	}|j�}|d	d!}|d	d!}|d	d!}t||ddt	d
|�\}}}}}tj||ddt	d
|�\}}	}
}}t
||�xHt|||||f||	|
||f�D]\}
}t|
|�q�WdS(
s
Tests polyfiti
ii����iiiNtfullitw(ii(i����i����(RR�R[R0R�tpolyfittviewRTRR,R�tziptcopyR(RR=R>tCtRtKtStDRXtrtktsRWRta_R�twotxstystws((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyttest_polyfit_s:(
7D:7A:'G:


-0
:(R&R'R�(((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyR�]stTestArraySetOpscBs�eZd�Zd�Zd�Zd�Zd�Zd�Zd�Zd�Z	d�Z
d	�Zd
�Zd�Z
d�Zd
�ZRS(cCs�ddddddg}t|dtdt�}|jt|dt��t|dtdddgddddg��t|ddddg�t|dddddddg�d	S(
sTest unique on listiiitreturn_indextreturn_inverseiRiN(tuniqueR,R/RSRTRR(RRgR((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyttest_unique_onlist�s/c	Cs�tddddddgdddddddg�}t|dtdt�}t|dtddddgdddddg��t|dddd	dg�t|dddddddg�d|_tddddddgdddddddgd
d�}t|dtdt�}t|dtddddgdddddg��t|dddd	dg�t|dddddddg�dS(s*Test unique on masked data w/use_mask=TrueiiiRiR�R�i����it
fill_valueN(RR�R,RR�(RRgR((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyttest_unique_onmaskedarray�s65#	!5cCs�tdddgdt�}t|dtdt�}t|dtdgdtg��t|ddg�t|ddddg�t}t|dtdt�}t|dtt��t|ddg�t|ddg�dS(sTest all maskediRR�R�iiN(RR,R�RR(RRgR((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyttest_unique_allmasked�s#cCs�ttjd�ddddddg�}tddddgdddddg�}t|�}t||�t|j|j�t|j|j�dS(sTests mediff1diRiiiN(RRRRtediff1dRRgR(RR=RR((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyttest_ediff1d�s**
c
Cs-ttjd�ddddddg�}t|dt�}tdddddgddddddg�}t||�t|j|j�t|j|j�t|ddddg�}tdddddddgddddddddg�}t||�t|j|j�t|j|j�d	S(
sTest ediff1d w/ to_beginiRiitto_beginiiiN(	RRRR�RRRRgR(RR=RR((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyttest_ediff1d_tobegin�s*0
<
c
Cs-ttjd�ddddddg�}t|dt�}tdddddgddddddg�}t||�t|j|j�t|j|j�t|ddddg�}tdddddddgddddddddg�}t||�t|j|j�t|j|j�d	S(
sTest ediff1d w/ to_endiRiitto_endiiiN(	RRRR�RRRRgR(RR=RR((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyttest_ediff1d_toend�s*0
<
cCsEttjd�ddddddg�}t|dtdt�}tddddddgdddddddg�}t||�t|j|j�t|j|j�t|dddd	gdt�}tdddddddd	gdddddddddg�}t||�t|j|j�t|j|j�d
S(s#Test ediff1d w/ to_begin and to_endiRiiR�R�iiiN(	RRRR�RRRRgR(RR=RR((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyttest_ediff1d_tobegin_toend�s*6
!B
c	Cstjd�}t|�}tddddgdddddg�}t||�|jt|t��t|j|j�t|j	|j	�t|dt
dt
�}tddddddgdddddddg�}|jt|t��t|j|j�t|j	|j	�dS(sTest ediff1d w/ a ndarrayiiRiR�R�N(RRR�RRR/RSRTRgRR(RR=RR((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyttest_ediff1d_ndarray�s*
6cCs�tddddgdddddg�}tddddgdddddg�}t||�}tdddgddddg�}t||�dS(sTest intersect1diiRii����N(Rtintersect1dR(RR=R>RR((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyttest_intersect1d�s
**$c	Cs0tdddddgddddddg�}tdddd	ddgdddddddg�}t||�}t|tdd	dg��tdddddgddddddg�}dddd	dg}t||�}t|tdd	ddgdddddg��tdddg�}td
dd	g�}t||�}tt|t��t|dddd	dd
g�tddddgdddddg�}td
dd	dgdddddg�}t||�}tt|t��t|dddd	dd
g�tgtgg��dS(
s
Test setxor1diiiii����RiiiiiN(Rtsetxor1dRtassert_RSRTRQ(RRRR((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt
test_setxor1ds&0601**c	Cs*tdddddgddddddg�}tdddd	ddgdddddddg�}t||�}t|tttttg�tdddddgddddddg�}tdddgddddg�}t||�}t|tttttg�tgtgg��d
S(s	Test in1diiiii����RiiiN(Rtin1dRR,R-RQ(RRRR((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt	test_in1ds060$c
Cs�tddddddgdddddddg�}tdddd	ddgdddddddg�}t||�}tdddd	dddgddddddddg�}t||�tgtgg��d
S(sTest union1diiiii����RiiiN(Rtunion1dRRQ(RRRRR((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyttest_union1d*s66<
cCs�tddddddddgdddddddddg�}tddd	d	dddg�}t||�}t|tddd
gddddg��td�}td�}tt||�tdd
g��dS(sTest setdiff1diiiiiiRiii����i
ii	N(Rt	setdiff1dRR(RRRR((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyttest_setdiff1d5sB!+cCsVtjdddg�}tjdddg�}tt||�tjdg��dS(sTest setdiff1d_charrayRRRXR�N(RRRQR�(RRR((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyttest_setdiff1d_char_arrayAs(R&R'R�R�R�R�R�R�R�R�R�R�R�R�R�R�(((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyR��s																t
TestShapeBasecBseZd�ZRS(cCs�tdddgddddg�}t|�}t|jd�t|jj|jj�t|jd�t|jj|jj�dS(	sTest atleast_2diiiRiN(ii(i(Rt
atleast_2dRRFRRg(RRR((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyttest_atleast2dMs$(R&R'R�(((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyR�Kst__main__(ROt
__author__t__version__t__revision__t__date__tnumpyRt
numpy.testingRRtnumpy.ma.testutilst
numpy.ma.coretnumpy.ma.extrasRR(RPReRmRyR~R�R�R�R�R�R�R&(((s@/usr/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt<module>s0


VW%As(AA)�