�
�9Zc@`
sr
dZd
dl
mZmZmZ
d
dlZd
dlZd
dlZd
dl	j
Z
d
dlmZ
d
dlm
Z
mZ
d
dl	mZmZ
d
dlmZmZmZ
d
dlmZmZ
d
d	lmZmZmZmZmZmZ
d
d
lm Z m!Z!m"Z"
def
d��YZ#d
ef
d��YZ$def
d��YZ%d�Z&e'dkrn
e�
ndS(s^Tests suite for mrecords.

:author: Pierre Gerard-Marchant
:contact: pierregm_at_uga_dot_edu

i(tdivisiontabsolute_importtprint_functionN(
trecarray(tasbytestasbytes_nested(tmaskedtnomask(tTestCasetrun_module_suitettemppath(tfromrecordst
fromarrays(t
MaskedRecordst	mrecarrayRtfromtextfileRtaddfield(tassert_tassert_equaltassert_equal_recordstTestMRecordsc
B`
s�eZ
d�Zd
�Zd�Zd�Zd�Zd�Zd�Zd�Z	d�Z
d	�Zd
�Zd�Z
d�Zd
�Zd�Zd�Zd�ZRS(c
O`
s!tj
||
|�

|j�
dS(
N(Rt__init__tsetup(tselftargstkwds((sO/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyR"s

c
C`
s�d
ddddg}
dddd	d
g}tddd
ddg�
}dt
fdtfdg}dd
ddd
g}tjtt|
||��
d|d|�
|_dS(Ni
iiiig�������?g������
@gffffff
@g������@g@tonettwotthreetfourtfivet
at
bt
cs|S8itmasktdtype(R!s|S8(Rtinttfloattmatarraytlisttziptbase(RtilisttflisttslisttddtypeR"((sO/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyR&s





c
C`
s�|j}
|
j
t�
}t|j|
j�
t|j|
j�
tt|j	t
��

t|j	|
j	j
t
�
�
x#dD]}t|
|||�
qwWt|j
t�
|�
dS(NRR R!(RR R!(R*tviewRRt
recordmaskRt_maskRt
isinstancet_dataR(RR*tmbasetfield((sO/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyttest_byview0s	








c
C`
sg|jj
�}
|
jt�
}x=dD]5}tt||�||�
t|
|||�
q%W|d}tt|t��

t|j|j�
t|j	�ddt
d�
f�
t|jt�
t|j
j�tttf�
t|d
|d
d�
|d}tt|t��

t|j|j�
t|j	�d�
t|jt�
t|j
j�tttf�
t|d
|d
d�
t|d
tk�

|d	 }tt|t��

t|j|j�
t|jddg�
t|jtjtttftttfgd
|j
j�

�
t||
d	 jt�
�
x,d
D]$}tt||�|
d	 |�
q;WdS(NRR R!ii
g�������?Ri����iR#(RR R!(NNN(RR R!(R*tcopyR/RRtgetattrRR2R#ttolistRR0RR1titemtFalsetNonetTrueRRR"tnpR'(RR*R4R5tmbase_firstt
mbase_lasttmbase_sl((sO/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyttest_get<s>







"













	






c

C`
s5|jj
�}
|
jt�
}|j
�}d|_d|jj(t|djdddddg�
t|djdddddg�
d|_t|djdg
d�
tt	j
|d�
dg
d�
t|jtg
d�
t|jj
�tjdddddgdt�

�
t|_t|jjdg
d�
t|jjdg
d�
tt	j
|d	�
dg
d�
tt	j|d	�
td�
g
d�
t|jj
�tjdddddgdt�

�
|
jt�
j
�}d|jd
)t|jddd
ddg�
t|jjdddddg�
t|jd
)t|j|
d�
t|jjdddddg�
d dtfg}t	jd!d"d#gd|�

}|jt�
}t	jdddgddddg�

}tj��H
tjd�

||d<t|j|�
t|jjdddg�
WdQXdS($Ni?Bg@��x�DsN/AiRii
R#R!iiR talphas|S1tnumi
iiR"tignore(i?Bg@��x�DsN/A(iii(ii
i
(iii(iii(ii
i
(iii
(ii
i
(iii
(iii
(ii
i
(RCs|S1(Ri
(R i(R!i(R*R7R/Rt
fill_valueRR3RR1R&tgetmaskarrayR0R;R9R>R'tboolRR!R"tgetdataRR R$R
twarningstcatch_warningstsimplefilterRD(RR*R4tndtypetdatatrdatatval((sO/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyttest_set_fieldsds\


	

#
#	

!

	



	

	

!
'

	



	





"



"


'






c
C`
s�|jj
�}
|
jt�
}t|d
d<t|jdddddg�
t|jjdddddg�
tt	j
d�
t	jjd�
gd	d
t
fd
tfg�

}t|d
d<t|jdddddg�
t|jjdddddg�
dS(NRi����i
iiiiiR#R (R*R7R/RRRRR1RR>tarangetrandomtrandR$R%(RR*R4((sO/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyttest_set_fields_mask�s


"$



c
C`
s
|jj
�}
|
jt�
}t|_ttj|d
�
dg
d�
t|dj	|d
j	�
t|dj	|dj	�
t|j	j
�tjdg
ddt
�

�
t|_ttj|d�
dg
d�
t|j	j
�tjd	g
ddt
�

�
dS(
NR i
iRR!R#i(i
i
i
(iii(R*R7R/RRR"RR&RGR1R9R>R'RHR(RR*R4((sO/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyt
test_set_mask�s

	
!



	
!

c
C`
s
|jj
�}
|
jt�
}d
dddd
g|_t|jjd
dddd
g�
t|jjd
dddd
g�
t|jjd
dddd
g�
ddddd
g|_t|jjddddd
g�
t|jjddddd
g�
t|jjddddd
g�
dS(Ni
i(	R*R7R/RR"RRR R!(RR*R4((sO/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyttest_set_mask_fromarray�s


"
"
"
"
"
c
C`
sB
|jj
�jt�
}
tjddd	d
dgddtfdtfdtfg�

}||
_t|
j	jd
d
ddd
g�
t|
j
jddd
d
d
g�
t|
jjd
d
ddd
g�
t|
_||
_
t|
j	jd
d
ddd
g�
t|
j
jddd
d
d
g�
t|
jjd
d
ddd
g�
dS(Nii
R#RR R!(ii
i(ii
i(i
ii
(i
ii
(iii(R*R7R/RR>R'RHR"RRR R!R;t	fieldmask(RR4tnmask((sO/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyttest_set_mask_fromfields�s


$
	
"
"
"	
	
"
"
c

C`
sg|jj
�}
|
jt�
j
�}t|d
<t|jj�tj	dddddgdt
�

�
t|jdddddg�
|
jt�
j
�}d|d*t|jj
dddddg�
t|jjdddddg�
t|jj
d	d	d
ddg�
t|jjdddddg�
t|jj
td
d
dddg�
�
t|jjdddddg�
|
jt�
j
�}t|d*t|jj
dddddg�
t|jjdddddg�
t|jj
ddd
ddg�
t|jjdddddg�
t|jj
tdddddg�
�
t|jjdddddg�
dS(Ni����ii
R#iiiig@gffffff
@g������@g@t
5RRRg�������?g������
@RR(iii(i
i
i
(iii(i
i
i
(i
i
i
(iii(R*R7R/RRRR1R9R>R'RHR0RR3R R!R(RR*R4((sO/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyttest_set_elements�s4










"
"
"
"


"


"
"
"
"


c
C`
sa
|jj
�}
|
jt�
}|j�
y�d|d)t|jjdddd
d
g�
t|jjdddd
d	g�
t|j	jt
d
ddd
dg�
�
t|jjdddddg�
t|jj|jj�
t|jj|j	j�
Wn$tk
r	



n t
k
r



�n
Xtd�
�
yd|d)Wnttfk
rP



n
Xtd�
�
dS(Nii����i
iig�������?g������
@gffffff
@g@RRRR[Ris)Flexible hard masks should be supported !s.Should have expected a readable buffer object!(iii(R*R7R/Rtharden_maskRRR3R R!RR1tNotImplementedErrortAssertionErrort	Exceptiont	TypeError(RR*R4((sO/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyttest_setslices_hardmask�s,






"
"


"









c
C`
s�|jj
�}
|
jt�
}|j�
|j|j�

t|_t	|j
|
j
�
|j�
|j|j�

t|_t	|j
tj
|
j|
j��
|jtj|d
j
�
tk�

t|dj
|d
j
�
dS(NR R(R*R7R/RR]t
assertTruet	_hardmaskRR"RR1tsoften_maskR&tmake_mask_nonetshapeR#t	make_maskR(RR*R4((sO/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyt
test_hardmask
s




	




		

#
c
C`
s�|jj
�}
|
jt�
}tj|�
}tj|�
}t|j|j�
t	|j
|j
�
t|j|j�
t	|j|j�
dS(
N(R*R7R/RtpickletdumpstloadsRR#RR3R1(RR*tmrect
_tmrec_((sO/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyt
test_pickling)
s






c
C`
s3
tj
d
ddgdddd
gdt�
}
tj
ddd	gdddd
gdt�
}tj
d
ddgdddd
gdd
�
}dtfdtfdg}t|
||gd|dd�
}|j�}t|dtj
ddt�

�
t|dtj
ddt�

�
t|dtj
ddd
�

�
dS(Ni
iiR"iR#g�������?g������
@gffffff
@RRRs|S8RR R!RFi��
g�i�@sN/A(R!s|S8(i��
g�i�@sN/A(i
ii��
(g�������?g������
@g�i�@(RRsN/A(R&R'R$R%RtfilledRR>(Rt_at_bt_cR.Rmt
mrecfilled((sO/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyttest_filled4
s-
-
-


	

 




c
C`
s

tj
d
ddgdddd
gdt�
}
tj
ddd	gdddd
gdt�
}tj
d
ddgdd
ddgdd
�
}dtfdtfdg}t|
||gd|dd�
}t|j�dddtd�
fddtd�
fg�
dS(Ni
iiR"iR#g�������?g������
@gffffff
@RRRs|S8RR R!RFi��
g�i�@sN/A(R!s|S8(i��
g�i�@sN/A(i
g�������?N(	R&R'R$R%RRR9R<R(RRrRsRtR.Rm((sO/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyttest_tolistC
s-
-
-


	

c
C`
s?td
dt
dd�
}
d|
dd<t|
ddd�
dS(Ni
tformatstnamesR*i
i(RR%R(Rt
x((sO/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyttest_withnamesP
s

c
C`
s
td
ddt
fddtfg�

}
t|
d<t|
jd
�
j�d
td�
d	f�
td
dd
ddfg
�

}t|d<t|jd
�
j�tj	dd|j
�

j��
tddd
�

}t|d<d|d
<|jd�

t|jd�
tj	ddgd|j
�

�
dS(Ni
R#t
it
ss|S8t
fit
1g�?tf0s<f8isi4, (2,3)float, float(R}s|S8(ii(
i
(i
i
i
(iii(i
i
i
(RR$R%RRRqR:RR>R'R#R(Rteasytsolotmult((sO/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyttest_exotic_formatsV
s'


+












(t__name__t
__module__RRR6RBRQRURVRWRZR\RbRiRpRvRwR{R�(((sO/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyR s"		
		(	9					 					
	tTestViewc
B`
s,eZ
d�Zd
�Zd�Zd�ZRS(c
C`
s�tj
d
�
tjjd
�
}
}dtjfdtjfg}tjtt|
|��
d|�

}t|
|gd|dd	�
}t	t
f|jd<||
||f|_dS(
Ni
RR R#RFg"�g�X�i(g"�g�X�(
R>RRRSRTR%R'R(R)RR;R=R"RN(RRR RMtarrRm((sO/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pytsetUpl
s
"

$

c
C`
s[|j\}
}}}|
j
�}|jt|t��

t||
�
t|j|
j�
dS(
N(RNR/RcR2R
RR1(RRmRR R�ttest((sO/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyttest_view_by_itselfu
s






c
C`
s�|j\}
}}}t
jd
f}|
j|�
}|jt|tj��

t|t
j	t
t||��
dt
j�

�
|j|dtjk�

dS(NiR#ii
(ii
(
RNR>R%R/RcR2R&tMaskedArrayRR'R(R)R(RRmRR R�tntypeR�((sO/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyttest_view_simple_dtype|
s




.
c
C`
s�|j\}
}}}d
t
jfdt
jfg}|
j|�
}|jt|t��

t||j|�
�
|j|ddtk�

t	|j
t
j
|�
�
|j|jdk�

dS(Nt
At
Bi(
RNR>R%R/RcR2R
RRRR#t_fill_valueR<(RRmRR R�talttypeR�((sO/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyttest_view_flexible_type�
s







(R�R�R�R�R�R�(((sO/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyR�j
s				tTestMRecordsImportc
B`
sGeZ
d�Zd
�Zd�Zd�Zd�Zd�Zd�ZRS(c
O`
s!tj
||
|�

|j�
dS(
N(RRR(RRR((sO/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyR�
s

c
	C`
s'
tj
d
ddgdddd
gdt�
}
tj
ddd	gdddd
gdt�
}tj
tttd
ddg��
dddd
gdd
�
}dtfdtfdg}t|
||gd|dtd�
td�
td�
f�
}t|
j	|j	|j	fd|�

}|||f|_
dS(Ni
iiR"iR#g�������?g������
@gffffff
@RRRs|S8RR R!RFt99999s99999.sN/A(R!s|S8(R&R'R$R%R(tmapRRt
recfromarraysR3RN(RRrRsRtR.Rmtnrec((sO/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyR�
s-
-
$





$
c

C`
s)
tj
d
ddgdddd
gdt�
}
tj
ddd	gdddd
gdt�
}tj
d
ddgdddd
gdd
�
}|j\}}}xBtd|
||f�D](\}}tt||�j|j�
q�Wtj
d
dd
gdd
ddg�

}	t	t
|	d|j�

|d�
dS(Ni
iiR"iR#g�������?g������
@gffffff
@RRRs|S8RR R!(RR R!(R&R'R$R%RNR)RR8R1RRR#(
RRrRsRtRmR�RnR~t
lt_x((sO/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyttest_fromarrays�
s
-
-
-

%
 '
c
C`
su
|j\}
}}dddg}t
|d
d�

}t|d
d�

}t||�
t|�
}t|j|
j�
x6|jjD](}tt||�t|
j|��
q~Wt|j	�d
d�

}t|jd
t
fdtfdg�
x?tdd�D].\}	}
tt||	�t|
j|
��
q�Wt|
�
}t|j|
j�
t|j|
j
��
t|j|
j�
dS(Ni
tabcg���
@iitxyg���@t
 g����?Rysc1, c2, c3, c4sc1,c2,c3tc1tc2tc3s|S5RR R!(i
R�g���
@i(iR�g���@i
(iR�g����?i(R�s|S5(R�R�R�(RR R!(RNtrecfromrecordsRRRR#RyR8R3R9R$R%R)RqR1(RRmR�R.tpalisttpatmpat_mrecR5R~t
n((sO/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyttest_fromrecords�
s&

	






&
%

&


c
C`
sk
|j\}
}}t
|j�d
|ddddg�
}t|j|
j�
t|jj�dddg�
t
|j�d
|dt�
}t|j|
j�
t|jj�dd	d
g�
t
|j�d
|d|
j�
}t|j|
j�
t|jj�|
jj��
t
|j�d
|d|
jj��
}t|j|
j�
t|jj�|
jj��
dS(NR#R"ii
(iii(i
i
i
(iii(i
i
i
(i
i
i
(i
i
i
(RNRR9RR3RR1R=(RRmR�R.R�((sO/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyttest_fromrecords_wmask�
s'



!




c
C`
s�d
}
t��D}t
|d��}|j|
�

WdQXt|dddd�
}WdQX|jt|t��

t|jddddg�
t|j	j
ddddg�
t|jddd	d
g�
dS(Ns�#
'One (S)','Two (I)','Three (F)','Four (M)','Five (-)','Six (C)'
'strings',1,1.0,'mixed column',,1
'with embedded "double quotes"',2,2.0,1.0,,1
'strings',3,3.0E5,3,,1
'strings',4,-1e-10,,,1
t
wt	delimitort
,tvarnamestABCDEFGi
ig�OAg�����|۽(R
topentwriteRRcR2R
Rt
Ft
ER1t
C(RtfcontenttpathR~tmrectxt((sO/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyttest_fromtextfile�
s	







c
C`
sv|j\}
}}d
ddgdddg}}t
|
tj|d|�

�}
t|
j|�
t|
jj|�
dS(Nidi�i,
i
iR"(RNRR&R'Rtf3R1(RRmR�R.t
dt
m((sO/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyt
test_addfield�
s




(	R�R�RRR�R�R�R�R�(((sO/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyR��
s		
				c
C`
sHtj
d
dgddd
gddtfd	tjfg�
}|d

dS(Ni
t
2it
4R"iR#RR (i
R�(iR�(ii(ii
(R&tmasked_arrayR$R>tobject(
t
y((sO/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyt#test_record_array_with_object_field�
s



t__main__((t__doc__t
__future__RR
RRJRjtnumpyR>tnumpy.maR&Rtnumpy.compatRRRRt
numpy.testingRR	R
tnumpy.core.recordsRR�RR�tnumpy.ma.mrecordsR
RRRtnumpy.ma.testutilsRRRRR�R�R�R�(((sO/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyt<module>s&







.�K&i