�
�9Zc@`
s�dZd
dl
mZmZmZ
d
dlZd
dlmZm	Z	m
Z
mZ
d
dlm
Z
mZmZmZmZmZmZ
de	f
d��YZedkr�e�
ndS(	s-Test functions for 1D array set operations.

i(tdivisiontabsolute_importtprint_functionN(trun_module_suitetTestCasetassert_array_equaltassert_equal(tediff1dtintersect1dtsetxor1dtunion1dt	setdiff1dtuniquetin1dt
TestSetOpsc
B`
steZ
d�Zd
�Zd�Zd�Zd�Zd�Zd�Zd�Z	d�Z
d	�Zd
�Zd�Z
RS(c
C`
s�d
�}
dddddddgd}ddddg}ddddg}dddddddgd}tj
ddddgd�}g}|jtjd	�

|jtjd
�

|jd�

|jd�

xK|D]C}tj||�}	tj||�}
|
|	|
||||�
q�Wd
}tjt|�
|�}	||	(tjt|�
|�}
||
(|
|	|
||||�
ddg}tjtt	||��
|�}	tjtt	||��
|�}
|
|	|
||||�
dddg}	t
tj|	�
ddg�
dddg}dddg}ddddddg}t|�
}
t
|
|�
t|dt�

\}}t
||�
t
||�
tj
d�
}d|d<tj|dt�

\}}t
|tjd�
�
dS(Nc
S`
s�d
}|jd|�}t
|�
}t||
|�
|jd|�}t
|ddd�\}}	t||
|�
t|	||�
|jd|�}t
|ddd�\}}	t||
|�
t|	||�
|jd|�}t
|ddd�\}}	t||
|�
t|	||�
|jd|�}t
|ddd�\}}
}t||
|�
t|
||�
t|||�
|jd	|�}t
|ddd�\}}
}t||
|�
t|
||�
t|||�
|jd
|�}t
|ddd�\}}
}t||
|�
t|
||�
t|||�
|jd|�}t
|ddd�\}}
}}t||
|�
t|
||�
t|||�
t|||�
dS(Nscheck {0} failed for type {1}tvaluestreturn_indexi
itreturn_inverset
return_countssreturn_index and return_inversesreturn_index and return_countss return_inverse and return_countss.return_index, return_inverse and return_counts(tformatRR(
t
at
bti1ti2t
ctdttbase_msgtmsgt
vt
jtj1tj2tj3((sS/opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_arraysetops.pyt	check_allsL























	
	
!



iii
ii
iit
AllIntegertAllFloats
datetime64[D]stimedelta64[D]t
Ott
ig�?yy�?R.(R%R&(R%R&y�?y�?�y�?�y�?(i
i(i
i(ii(tnptmultiplytextendt	typecodestappendtarraytemptytlentlisttzipRRtTruet	chararraytzeros(tselfR!RRRRRttypesRtaatbbtunqtinvta1ta2ta2_inv((sS/opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_arraysetops.pyttest_uniquesN	0




















!
!















c
C`
s
tj
d
dddg�
}
tj
ddddd
g�
}tj
ddd
g�
}t|
|dt�
}t||�
tj
d
d
dddg�
}
tj
ddddddd
g�
}tj
ddd
g�
}t|
|�}t||�
tgtgg��
dS(Niii
iiit
assume_unique(R'R,RR1R(R4RRtecRted((sS/opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_arraysetops.pyttest_intersect1dvs




$


c
C`
sg
tj
d
dddg�
}
tj
ddddd
g�
}tj
dddg�
}t|
|�}t||�
tj
dddg�
}
tj
dd
dg�
}tj
ddddd
dg�
}t|
|�}t||�
tj
ddddg�
}
tj
dd
ddg�
}tj
ddddd
dg�
}t|
|�}t||�
tgtgg��
dS(	Niii
iiiii(R'R,R	R(R4RRR?R((sS/opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_arraysetops.pyt
test_setxor1d�s 





!



!


c
C`
s�tj
g�
}
tj
d
g
�
}tj
d
dg�
}tgt|
�
�
tdg
t|
dd�

�
tdg
t|
dd�

�
tddgt|
dddd�
�
tgt|�
�
td
g
t|�
�
dS(Ni
iitto_begintto_endi����(R'R,RR(R4t	zero_elemtone_elemttwo_elem((sS/opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_arraysetops.pyttest_ediff1d�s





%

c
C`
s�xYdD]Q}
ddd
dg}dddd
dg|
}tj
ttttg�
}t||dt�
}t||�
d	|d
<tj
ttttg�
}t||dt�
}t||�
d\|d
<|d<tj
ttttg�
}t||dt�
}t||�
tj
dddddddddddd
ddg�
}dddg|
}ttttttttttttttg}t||�}t||�
|dddg|
}ttttttttttttttg}t||�}t||�
tj
ddd
dg�
}tj
dddd
dg|
�
}tj
ttttg�
}t||�}t||�
tj
ddd
d
dg�
}tj
ddddd
dg|
�
}tj
tttttg�
}t||�}t||�
tj
ddg�
}tj
ddg|
�
}tj
ttg�
}t||�}t||�
qWtj
dg
�
}tj
dg
�
}tj
tg
�
}t||�}t||�
ttgg�g�
dS(
Ni
i
iiiiiR>ii(i
i
(ii(R'R,R1tFalseR
R(R4tmultRRR?R((sS/opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_arraysetops.pyt	test_in1d�s\














9











"




%












c
	C`
s�tj
d
dddddddg�
}
tj
d
dg�
}tj
ttttttttg�
}t|
|�}t||�
dS(NRRRt
dt
e(R'R,R1RIR
R(R4RRR?R((sS/opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_arraysetops.pyttest_in1d_char_array�s

'
'
c
C`
s�x�d	D]�}
tj
dddddddddddd
ddg�
}dddg|
}ttjt||��
t||dt�
�
qWdS(
sTest in1d's invert parameteri
i
iiiitinvertN(i
i
(R'R,RROR
R1(R4RJRR((sS/opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_arraysetops.pyttest_in1d_invert�s

9

c
C`
s�tj
d
�
jdd�}
tj
dd�jdd�}tj
dd�jdd�}tjttttttg�
}tt|
|dt�
|�
tt|
|dt�
|�
tt|
|dt�
|�
tt|
|dt�
|�
dS(Niiii	i?iR>(R'tarangetreshapeR,RIR1RR
(R4RRtlong_bR?((sS/opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_arraysetops.pyttest_in1d_ravel�s


!


c
C`
s�tj
d
ddddg�
}
tj
ddddddd
g�
}tj
ddddd
dg�
}t|
|�}t||�
tgtgg��
dS(Niiii
ii(R'R,R
R(R4RRR?R((sS/opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_arraysetops.pyttest_union1d
s

$!


c
	C`
s
tj
d
dddddddg�
}
tj
dddddddg�
}tj
d
dg�
}t|
|�}t||�
tjd�
}
tjd	�
}tj
d	d
g�
}t|
|�}t||�
tgtgg��
tj
dtj�}
tt|
g�jtj�
dS(Niiiii
iiiii((R'R,RRRQtuint32Rtdtype(R4RRR?R((sS/opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_arraysetops.pyttest_setdiff1d
s
'
$









c
C`
sVtj
d
ddg�
}
tj
d
ddg�
}tt|
|�tj
dg
�
�
dS(NRRRt
s(R'R,RR(R4RR((sS/opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_arraysetops.pyttest_setdiff1d_char_array$
s


c
C`
s�tj
d
ddddg�
}
tj
ddddddd
g�
}t|
|�}t|
|�}t|
|�}t||�}t||�
dS(	Niii
iii	ii(R'R,R	RR
RR(R4RRtc1taux1taux2tc2((sS/opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_arraysetops.pyt
test_manyways)
s

$



(t__name__t
__module__R=RARBRHRKRNRPRTRURXRZR_(((sS/opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_arraysetops.pyRs	e				<					
	
		t__main__(t__doc__t
__future__RR
RtnumpyR't
numpy.testingRRRRtnumpy.lib.arraysetopsRRR	R
RRR
RR`(((sS/opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_arraysetops.pyt<module>s

"4�&