Current File : //opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyc
�
� 9Zc @` sr d Z d d l m Z m Z m Z d d l Z d d l Z d d l Z d d l j
Z
d d l m Z d d l m
Z
m Z d d l m Z m Z d d l m Z m Z m Z d d l m Z m Z d d l m Z m Z m Z m Z m Z m Z d d
l m Z m! Z! m" Z" d e f d � � YZ# d
e f d � � YZ$ d e f d � � YZ% d � Z& e' d k rne � n d S( s^ Tests suite for mrecords.
:author: Pierre Gerard-Marchant
:contact: pierregm_at_uga_dot_edu
i ( t divisiont absolute_importt print_functionN( t recarray( t asbytest asbytes_nested( t maskedt nomask( t TestCaset run_module_suitet temppath( t fromrecordst
fromarrays( t
MaskedRecordst mrecarrayR t fromtextfileR t addfield( t assert_t assert_equalt assert_equal_recordst 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 RS( c O` s! t j | | | � | j � d S( N( R t __init__t setup( t selft argst kwds( ( sO /opt/alt/python27/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( Ni i i i i g�������?g������@gffffff
@g������@g @t onet twot threet fourt fivet at bt cs |S8i t maskt dtype( R! s |S8( R t intt floatt mat arrayt listt zipt base( R t ilistt flistt slistt ddtypeR" ( ( 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 � t t | j t
� � t | j | j j t
� � x# d D] } t | | | | � qw Wt | j t � | � d S( NR R R! ( R R R! ( R* t viewR R t
recordmaskR t _maskR t
isinstancet _dataR ( R R* t mbaset field( ( sO /opt/alt/python27/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( NR R R! i i g�������?R i����i R# ( R R R! ( NNN( R R R! ( R* t copyR/ R R t getattrR R2 R# t tolistR R0 R R1 t itemt Falset Nonet TrueR R R" t npR' ( R R* R4 R5 t mbase_firstt
mbase_lastt mbase_sl( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyt test_get< s>
"
c
C` s5 | 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 � �H t j d � | | d <t | j | � t | j j d d d g � Wd QXd S($ Ni?B g@��x �Ds N/Ai R i i R# R! i i R t alphas |S1t numi
i i R" t ignore( 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 ( RC s |S1( R i ( R i ( R! i ( R* R7 R/ R t
fill_valueR R3 R R1 R& t getmaskarrayR0 R; R9 R> R' t boolR R! R" t getdataR R R$ R
t warningst catch_warningst simplefilterRD ( R R* R4 t ndtypet datat rdatat val( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyt test_set_fieldsd s\
## !
!'
"
" '
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( NR i����i i i i i i R# R ( R* R7 R/ R R R R R1 R R> t aranget randomt randR$ R% ( R R* R4 ( ( sO /opt/alt/python27/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* R7 R/ R R R" R R& RG R1 R9 R> R' RH R ( R R* R4 ( ( sO /opt/alt/python27/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* R7 R/ R R" R R R R! ( R R* R4 ( ( sO /opt/alt/python27/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* R7 R/ R R> R' RH R" R R R R! R; t fieldmask( R R4 t nmask( ( sO /opt/alt/python27/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* R7 R/ R R R R1 R9 R> R' RH R0 R R3 R R! R ( R R* R4 ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyt test_set_elements� s4
"""" "
"""" 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( Ni i����i i i g�������?g������@gffffff
@g @R R R R[ R i s) Flexible hard masks should be supported !s. Should have expected a readable buffer object!( i i i ( R* R7 R/ R t harden_maskR R R3 R R! R R1 t NotImplementedErrort AssertionErrort Exceptiont TypeError( R R* R4 ( ( sO /opt/alt/python27/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( NR R ( R* R7 R/ R R] t
assertTruet _hardmaskR R" R R1 t soften_maskR& t make_mask_nonet shapeR# t make_maskR ( R R* R4 ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyt
test_hardmask s
#c C` s� | j j � } | j t � } t j | � } t j | � } t | j | j � t | j
| j
� t | j | j � t | j | j � d S( N( R* R7 R/ R t picklet dumpst loadsR R# R R3 R1 ( R R* t mrect _t mrec_( ( sO /opt/alt/python27/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( Ni i i R" i R# g�������?g������@gffffff
@R R R s |S8R R R! RF i�� g �i�@s N/A( R! s |S8( i�� g �i�@s N/A( i i i�� ( g�������?g������@g �i�@( R R s N/A( R& R' R$ R% R t filledR R> ( R t _at _bt _cR. Rm t
mrecfilled( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyt test_filled4 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( Ni i i R" i R# g�������?g������@gffffff
@R R R s |S8R R R! RF i�� g �i�@s N/A( R! s |S8( i�� g �i�@s N/A( i g�������?N( R& R' R$ R% R R R9 R< R ( R Rr Rs Rt R. Rm ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyt test_tolistC s --- c C` s? t d d t d d �} d | d d <t | d d d � d S( Ni t formatst namesR* i
i ( R R% R ( R t x( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyt test_withnamesP 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( Ni R# t it ss |S8t fi t 1g �?t f0s <f8i s i4, (2,3)float, float( R} s |S8( i i ( i ( i i i ( i i i ( i i i ( R R$ R% R R Rq R: R R> R' R# R ( R t easyt solot mult( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyt test_exotic_formatsV s '
+
( t __name__t
__module__R R R6 RB RQ RU RV RW RZ R\ Rb Ri Rp Rv Rw R{ R� ( ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyR s"
( 9
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 t | | � � d | �} t | | g d | d d �} t t
f | j d <| | | | f | _ d S(
Ni
R R R# RF g "�g �X�i ( g "�g �X�(
R> RR RS RT R% R' R( R) R R; R= R" RN ( R R R RM t arrRm ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyt setUpl s " $ c C` s[ | j \ } } } } | j � } | j t | t � � t | | � t | j | j � d S( N( RN R/ Rc R2 R
R R1 ( R Rm R R R� t test( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyt test_view_by_itselfu s
c C` s� | j \ } } } } t j d f } | j | � } | j t | t j � � t | t j t
t | | � � d t j �� | j | d t j k � d S( Ni R# i i ( i i (
RN R> R% R/ Rc R2 R& t MaskedArrayR R' R( R) R ( R Rm R R R� t ntypeR� ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyt test_view_simple_dtype| 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 (
RN R> R% R/ Rc R2 R
R R R R# t _fill_valueR< ( R Rm R R R� t alttypeR� ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyt test_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 t TestMRecordsImportc B` sG e Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z RS( c O` s! t j | | | � | j � d S( N( R R R ( R R R ( ( sO /opt/alt/python27/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( Ni i i R" i R# g�������?g������@gffffff
@R R R s |S8R R R! RF t 99999s 99999.s N/A( R! s |S8( R& R' R$ R% R( t mapR R t
recfromarraysR3 RN ( R Rr Rs Rt R. Rm t nrec( ( sO /opt/alt/python27/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% RN R) R R8 R1 R R R# (
R Rr Rs Rt Rm R� Rn R~ t lt _x( ( sO /opt/alt/python27/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( Ni t abcg ���
@i i t xyg ���@t g ����?Ry s c1, c2, c3, c4s c1,c2,c3t c1t c2t c3s |S5R R R! ( i R� g ���
@i ( i R� g ���@i ( i R� g ����?i ( R� s |S5( R� R� R� ( R R R! ( RN t recfromrecordsR R R R# Ry R8 R3 R9 R$ R% R) Rq R1 ( R Rm R� R. t palistt pat mpat _mrecR5 R~ t n( ( sO /opt/alt/python27/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( NR# R" i i ( i i i ( i i i ( i i i ( i i i ( i i i ( i i i ( RN R R9 R R3 R R1 R= ( R Rm R� R. R� ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyt test_fromrecords_wmask� s ' ! c C` s� d } t � �D } t | d � � } | j | � Wd QXt | d d d d �} Wd QX| 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( 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 ,t varnamest ABCDEFGi i g �OAg�����|۽( R
t opent writeR Rc R2 R
R t Ft ER1 t C( R t fcontentt pathR~ t mrectxt( ( sO /opt/alt/python27/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( Nid i� i, i i R" ( RN R R& R' R t f3R1 ( R Rm R� R. t dt m( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyt
test_addfield� s
( R� R� R R R� R� R� R� R� ( ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyR� � s
c C` sH t j d
d g d d d
g d d t f d t j f g �} | d d S( Ni t 2i t 4R" i R# R R ( i R� ( i R� ( i i ( i i ( R& t masked_arrayR$ R> t object( 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__R R R RJ Rj t numpyR> t numpy.maR& R t numpy.compatR R R R t
numpy.testingR R R
t numpy.core.recordsR R� R R� t numpy.ma.mrecordsR
R R R t numpy.ma.testutilsR R R R R� R� R� R� ( ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_mrecords.pyt <module> s& . � K&i
Mini Shell