| Current File : //proc/self/root/proc/self/root/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyc |
�
E�`Qc @ sM d Z d Z d Z d Z d d l Z d d l Z d d l m Z d d l m Z
m Z d d l
m Z m Z d d l Z d d l Td d l j Z d d
l m Z m Z d d l Z d d l m Z d d l m Z m Z m Z m Z m Z m Z d
e f d � � YZ d e f d � � YZ d e f d � � YZ e! d k rIe" � n d S( s] Tests suite for mrecords.
:author: Pierre Gerard-Marchant
:contact: pierregm_at_uga_dot_edu
s5 Pierre GF Gerard-Marchant ($Author: jarrod.millman $)s $Revision: 3473 $s5 $Date: 2007-10-29 17:18:13 +0200 (Mon, 29 Oct 2007) $i����N( t recarray( t fromrecordst
fromarrays( t asbytest asbytes_nested( t *( t maskedt nomask( t WarningManager( t
MaskedRecordst mrecarrayR t fromtextfileR t addfieldt TestMRecordsc B s� e Z d Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z
d � Z d
� Z d � Z
d � Z d
� Z d � Z d � Z d � Z d � Z RS( s! Base test class for MaskedArrays.c O s! t j | | | � | j � d S( N( t TestCaset __init__t setup( t selft argst kwds( ( sB /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyR " s c C s� d d d d d g } d d d d d
g } t d d d
d d g � } d t f d t f d g } d d d d d g } t j t t | | | � � d | d | �| _ d S( s
Generic setupi i i i i g�������?g������@gffffff
@g������@g @t onet twot threet fourt fivet at bt cs |S8i t maskt dtypeN( R s |S8( R t intt floatt mat arrayt listt zipt base( R t ilistt flistt slistt ddtypeR ( ( sB /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyR & s c C s� | j } | j t � } t | j | j � t | j | j � t t | j t
� � t | j | j j t
� � x# d D] } t | | | | � qw Wt | j t � | � d S( s Test creation by viewR R R N( R R R ( R$ t viewR
t assert_equalt
recordmaskt assert_equal_recordst _maskt assert_t
isinstancet _dataR ( R R$ t mbaset field( ( sB /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyt test_byview0 s
c C sg | j j � } | j t � } x= d D]5 } t t | | � | | � t | | | | � q% W| d } t t | t � � t | j | j � t | j � d d t
d � f � t | j t � t | j
j � t t t f � t | d | d d � | d } t t | t � � t | j | j � t | j � d
� t | j t � t | j
j � t t t f � t | d | d d � t | d t k � | d } t t | t � � t | j | j � t | j d d g � t | j t j t t t f t t t f g d
| j
j �� t | | d j t � � x, d D]$ } t t | | � | d | � q;Wd S( s Tests fields retrievalR R R i i g�������?R i����i R N( R R R ( NNN( R R R ( R$ t copyR) R
R* t getattrR. R/ R t tolistR R+ R R- t itemt Falset Nonet TrueR R, R t npR! ( R R$ R1 R2 t mbase_firstt
mbase_lastt mbase_sl( ( sB /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyt test_get< s<
"
$
c
C sG | j j � } | j t � } | j � } d | _ d | j j (t | d j d d d d d g � t | d j d d d d d g � d | _ t | d j d g d � t t j
| d � d g d � t | j t g d � t | j j
� t j d d d d d g d t �� t | _ t | j j d g d � t | j j d g d � t t j
| d � d g d � t t j | d � t d � g d � t | j j
� t j d d d d d g d t �� | j t � j � } d | j d
)t | j d d d
d d g � t | j j d d d d d g � t | j d
)t | j | d � t | j j d d d d d g � d! d t f g } t j d" d# d$ g d | �} | j t � } t j d d d g d d d d g �} t � } | j � zG t j d � | | d <t | j | � t | j j d d d g � Wd | j � Xd S(% s Tests setting fields.i?B g@��x �Ds N/Ai R i i R R i i R t alphas |S1t numi
i i R t ignoreN( i?B g@��x �Ds N/A( i i i ( i i i ( i i i ( i i i ( i i i ( i i i ( i i i ( i i i ( i i i ( i i i ( R@ s |S1( R i ( R i ( R i ( R$ R4 R) R
t
fill_valueR R0 R* R- R t getmaskarrayR+ R8 R6 R; R! t boolR R R t getdataR R R R R t __enter__t warningst simplefilterRA t __exit__( R R$ R1 t ndtypet datat rdatat valt warn_ctx( ( sB /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyt test_set_fieldsc sR
## !
!'
"
" '
c C s� | j j � } | j t � } t | d d <t | j d d d d d g � t | j j d d d d d g � t t j
d � t j j d � g d d t
f d
t f g �} t | d d <t | j d d d d d g � t | j j d d d d d g � d S( s" Tests setting the mask of a field.R i����i i i i i i R R N( R$ R4 R) R
R R* R R- R R; t aranget randomt randR R ( R R$ R1 ( ( sB /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyt test_set_fields_mask� s "$ c C s | j j � } | j t � } t | _ t t j | d � d g d � t | d j | d j � t | d j | d j � t | j j
� t j d g d d t
�� t | _ t t j | d � d g d � t | j j
� t j d g d d t
�� d S(
NR i i R R R i ( i i i ( i i i ( R$ R4 R) R
R R R* R RD R- R6 R; R! RE R ( R R$ R1 ( ( sB /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyt
test_set_mask� s ! !c C s | j j � } | j t � } d d d d d g | _ t | j j d d d d d g � t | j j d d d d d g � t | j j d d d d d g � d d d d d g | _ t | j j d d d d d g � t | j j d d d d d g � t | j j d d d d d g � d S( Ni i ( R$ R4 R) R
R R* R R R ( R R$ R1 ( ( sB /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyt test_set_mask_fromarray� s """""c C sB | j j � j t � } t j d d d d
d g d d t f d t f d t f g �} | | _ t | j j d d d d d g � t | j
j d d d d d g � t | j j d d d d d g � t | _ | | _
t | j j d d d d d g � t | j
j d d d d d g � t | j j d d d d d g � d S( Ni i R R R R ( i i i ( i i i ( i i i ( i i i ( i i i ( R$ R4 R) R
R; R! RE R R* R R R R8 t fieldmask( R R1 t nmask( ( sB /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyt test_set_mask_fromfields� s $ """ ""c
C sg | j j � } | j t � j � } t | d <t | j j � t j d d d d d g d t
�� t | j d d d d d g � | j t � j � } d | d *t | j j
d d d d d g � t | j j d d d d d g � t | j j
d d d
d d g � t | j j d d d d d g � t | j j
t d
d
d d d g � � t | j j d d d d d g � | j t � j � } t | d *t | j j
d d d d d g � t | j j d d d d d g � t | j j
d d d
d d g � t | j j d d d d d g � t | j j
t d d d d d g � � t | j j d d d d d g � d S( Ni����i i R i i i i g @gffffff
@g������@g @t 5R R R g�������?g������@R R ( i i i ( i i i ( i i i ( i i i ( i i i ( i i i ( R$ R4 R) R
R R* R- R6 R; R! RE R+ R R0 R R R ( R R$ R1 ( ( sB /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyt test_set_elements� s2
"""" "
"""" c C sa | j j � } | j t � } | j � y� d | d )t | j j d d d d d g � t | j j d d d d d g � t | j j t
d
d d d
d g � � t | j j d d d d d g � t | j j | j j � t | j j | j j � Wn$ t k
r n t
k
r � n
Xt d � � y d | d )Wn t t f k
rPn
Xt d � � d S( s! Tests setting slices w/ hardmask.i i����i i i g�������?g������@gffffff
@g @R R R RZ R i s) Flexible hard masks should be supported !s. Should have expected a readable buffer object!N( i i i ( R$ R4 R) R
t harden_maskR* R R0 R R R R- t NotImplementedErrort AssertionErrort Exceptiont TypeError( R R$ R1 ( ( sB /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyt test_setslices_hardmask� s,
"" "
c C s� | j j � } | j t � } | j � | j | j � t | _ t | j
| j
� | j � | j | j � t | _ t | j
t j
| j | j � � | j t j | d j
� t k � t | d j
| d j
� d S( s
Test hardmaskR R N( R$ R4 R) R
R\ t
assertTruet _hardmaskR R R, R- t soften_maskR t make_mask_nonet shapeR t make_maskR* ( R R$ R1 ( ( sB /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyt
test_hardmask s
#c C s� d d l } | j j � } | j t � } | j | � } | j | � } t | j | j � t | j
| j
� t | j | j � t | j | j � d S( s
Test picklingi����N( t cPickleR$ R4 R) R
t dumpst loadsR* R R, R0 R- ( R Ri R$ t mrect _t mrec_( ( sB /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyt
test_pickling# s c C s3 t j d d d g d d d d g d t �} t j d d d g d d d d g d t �} t j d
d d g d d d d g d d
�} d t f d t f d g } t | | | g d | d d �} | j � } t | d t j d d t �� t | d t j d d t �� t | d t j d d d
�� d S( s Test filling the arrayi i i R i R g�������?g������@gffffff
@R R R s |S8R R R RC i�� g �i�@s N/AN( R s |S8( i�� g �i�@s N/A( i i i�� ( g�������?g������@g �i�@( s ones twos N/A( R R! R R R t filledR* R; ( R t _at _bt _cR( Rl t
mrecfilled( ( sB /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyt test_filled/ s --- c C s t j d d d g d d d d g d t �} t j d d d g d d d d g d t �} t j d
d d g d d d d g d d
�} d t f d t f d g } t | | | g d | d d �} t | j � d d d t d � f d d t d � f g � d S( s Test tolist.i i i R i R g�������?g������@gffffff
@R R R s |S8R R R RC i�� g �i�@s N/AN( R s |S8( i�� g �i�@s N/A( i g�������?N( R R! R R R R* R6 R9 R ( R Rq Rr Rs R( Rl ( ( sB /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyt test_tolist< s --- c C s? t d d t d d �} d | d d <t | d d d � d S( s% Test the creation w/ format and namesi t formatst namesR$ i
i N( R
R R* ( R t x( ( sB /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyt test_withnamesK s c C s t d d d t f d d t f g �} t | d <t | j d � j � d t d � d f � t d d d
d d f g �} t | d <t | j d � j � t j d d | j
�j � � t d d d
�} t | d <d | d <| j d � t | j d � t j d d g d | j
�� d S( s1 Test that 'exotic' formats are processed properlyi R t it ss |S8t fi t 1g �?t f0s <f8i s i4, (2,3)float, floatN( R| s |S8( i i ( i ( i i i ( i i i ( i i i ( R
R R R R* Rp R7 R R; R! R R, ( R t easyt solot mult( ( sB /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyt test_exotic_formatsQ s '
+
( t __name__t
__module__t __doc__R R R3 R? RP RT RU RV RY R[ Ra Rh Ro Ru Rv Rz R� ( ( ( sB /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyR
s$
' 5
t TestViewc B s, e Z d � Z d � Z d � Z d � Z RS( c C s� t j d � t j j d � } } d t j f d t j f g } t j t | | � d | �} | j t j � } t j t | | � d | d d �} t
| | g d | d d
�} t t f | j
d <| | | | f | _ d S( Ni
R R R RC g "�g �X�i ( g "�g �X�( g "�g �X�( R; RQ RR RS R R! R# R) R R R R8 R: R RL ( R R R RK t arrt rect marrRl ( ( sB /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyt setUpf s " $ c C s[ | j \ } } } } | j � } | j t | t � � t | | � t | j | j � d S( N( RL R) Rb R/ R R, R- ( R Rl R R R� t test( ( sB /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyt test_view_by_itselfq s
c C s� | j \ } } } } t j d f } | j | � } | j t | t j � � t | t j t
| | � d t j �� | j | d t j k � d S( Ni R i i ( i i ( RL R; R R) Rb R/ R t MaskedArrayR* R! R# R ( R Rl R R R� t ntypeR� ( ( sB /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyt test_view_simple_dtypex s (c C s� | j \ } } } } d t j f d t j f g } | j | � } | j t | t � � t | | j | � � | j | d d t k � t | j
t j
| � � | j | j d k � d S( Nt At Bi (
RL R; R R) Rb R/ R R, R R* R t _fill_valueR9 ( R Rl R R R� t alttypeR� ( ( sB /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyt test_view_flexible_type� s ( R� R� R� R� R� R� ( ( ( sB /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyR� d s t TestMRecordsImportc B sM e Z d Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z RS( s! Base test class for MaskedArrays.c O s! t j | | | � | j � d S( N( R R R ( R R R ( ( sB /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyR � s c C s' t j d d d g d d d d g d t �} t j d d d g d d d d g d t �} t j t t t d
d d g � � d d d d g d d
�} d t f d t f d g } t | | | g d | d t d � t d � t d � f �} t | j | j | j f d | �} | | | f | _
d S( s
Generic setupi i i R i R g�������?g������@gffffff
@R R R s |S8R R R RC t 99999s 99999.s N/AN( R s |S8( R R! R R R" t mapR R t
recfromarraysR0 RL ( R Rq Rr Rs R( Rl t nrec( ( sB /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyR � s --$$c
C s) t j d d d g d d d d g d t �} t j d d d g d d d d g d t �} t j d
d d g d d d d g d d
�} | j \ } } } xB t d | | | f � D]( \ } } t t | | � j | j � q� Wt j d d d
g d d d d g �} t t
| d | j �| d � d S( Ni i i R i R g�������?g������@gffffff
@R R R s |S8R R R ( R R R ( R R! R R RL R# R* R5 R- R, R R (
R Rq Rr Rs Rl R� Rm R} t lt _x( ( sB /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyt test_fromarrays� s ---% 'c C su | j \ } } } d d d g } t | d
d �} t | d
d �} t | | � t | � } t | j | j � x6 | j j D]( } t t | | � t | j | � � q~ Wt | j � d
d �} t | j d
t
f d t f d g � x? t d d � D]. \ } }
t t | | � t | j |
� � q� Wt | � } t | j | j � t | j | j
� � t | j | j � d S( s Test construction from records.i t abcg ���
@i i t xyg ���@t g ����?Rx s c1, c2, c3, c4s c1,c2,c3t c1t c2t c3s |S5R R R N( i s abcg ���
@i ( i R� g ���@i ( i R� g ����?i ( R� s |S5( R� R� R� ( R R R ( RL t recfromrecordsR R, R* R Rx R5 R0 R6 R R R# Rp R- ( R Rl R� R( t palistt pat mpat _mrecR2 R} t n( ( sB /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyt test_fromrecords� s&
&% & c C sk | j \ } } } t | j � d | d d d d g �} t | j | j � t | j j � d d d g � t | j � d | d t �} t | j | j � t | j j � d d
d g � t | j � d | d | j �} t | j | j � t | j j � | j j � � t | j � d | d | j j � �} t | j | j � t | j j � | j j � � d S( s( Tests construction from records w/ mask.R R i i N( i i i ( i i i ( i i i ( i i i ( i i i ( i i i ( RL R R6 R, R0 R* R- R: ( R Rl R� R( R� ( ( sB /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyt test_fromrecords_wmask� s ' ! c C s� t d � } d d l } d d l } | j � \ } } | j | | � | j | � t | d d d d �} | j | � | j t | t
� � t | j d d d d g � t | j
j d d d d g � t | j d d d
d g � d S( s Tests reading from a text file.s� #
'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
i����Nt delimitort ,t varnamest ABCDEFGi i g �OAg�����|۽( R t ost tempfilet mkstempt writet closeR t removeRb R/ R R* t Ft ER- t C( R t fcontentR� R� t tmp_fdt tmp_flt mrectxt( ( sB /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyt test_fromtextfile� s
c C sv | j \ } } } d d d g d d d g } } t | t j | d | �� } t | j | � t | j j | � d S( s Tests addfieldid i� i, i i R N( RL R R R! R* t f3R- ( R Rl R� R( t dt m( ( sB /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyt
test_addfield� s
(
R� R� R� R R R� R� R� R� R� ( ( ( sB /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyR� � s
t __main__(# R� t
__author__t __revision__t __date__t syst numpyR; R t numpy.core.recordsR R� R R� t numpy.compatR R t numpy.ma.testutilst numpy.maR R R RH t numpy.testing.utilsR t numpy.ma.mrecordsR R
R R R R
R� R� R� t run_module_suite( ( ( sB /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyt <module> s*
.� E(o