| Current File : //usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyc |
�
E�`Qc @ s� d Z d Z d d l Z d d l Z d d l Z d d l j j Z d d l m Z d d l
Td d l Z d d l Td d l m
Z
m Z d d l m Z e j Z d d l Z e j d d k r� d d
l m Z n d e f d � � YZ d
e f d � � YZ d e f d � � YZ d e f d � � YZ d e f d � � YZ d e f d � � YZ d e f d � � YZ d e f d � � YZ d e f d � � YZ d e f d � � YZ d e f d � � YZ! d! e f d" � � YZ" d# � Z# e$ d$ k r�e% � n d S(% sn Tests suite for MaskedArray & subclassing.
:author: Pierre Gerard-Marchant
:contact: pierregm_at_uga_dot_edu
s Pierre GF Gerard-Marchanti����N( t ndarray( t *( t asbytest asbytes_nested( t WarningManageri i ( t reducet TestMaskedArrayc B sm 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 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% d$ � Z& d% � Z' d& � Z( d' � Z) RS(( s! Base test class for MaskedArrays.c
C sn t j d d d d t d d d d d d d d g � } t j d d d d d
d d d d d d d g � } d } d d
d
d
d
d
d d
d
d
d
d
g } d
d
d d
d
d d d
d
d
d
d g } t | d | �} t | d | �} t j d d d d g � } t | d d
d d
d
g �} t j | d | � }
| j d � | | | | | | | | | |
f
| _ d S( s Base data definition.g �?g �g @g @g @g $�g $@g @g g �g �i i t maskg �g �?g�������?g@��x �DN( t npt arrayt pit masked_arrayt wheret set_fill_valuet d( t selft xt yt a10t m1t m2t xmt ymt zt zmt xf( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt setUp! s 73**
c C s� t d d d g � } t d d d g d d d d g �} t | j d � t | j d � t | j d � t | j d � t | j d � t | j d � d S( s" Tests some basic array attributes.i i i R i N( i ( i ( R t assert_equalt ndimt sizet shape( R t at b( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_basicattributes1 s $c C s� t d � } t t | � d � t d d t �} t t | � t t � � t d d t �} t t | � d � t d d d �} | j | j � j | j
j k � d S( s Checks masking a scalari t 0R i N( R R t strt Truet masked_print_optiont FalseR t
assertTruet filledt dtypet _data( R R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_basic0d= s c
C sa | j \
} } } } } } } } } }
| j t | � � | j t | � � | j | | j d � j � � t | j j t � | j j t � � | j } t
t j | � | � t
| j | � t
| j | j � t
| j | j � t
| j
t d � | � � t
t | � t | � t d � | � � t | |
� t t | d � |
� t | | � d S( s; Test of basic array creation and properties in 1 dimension.i c S s | | S( N( ( R R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt <lambda>T s c S s | | S( N( ( R R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyR, U s g@��x �DN( R R' t
isMaskedArrayR( t anyt
fail_if_equalR t astypet intR R R R) R R t countt lent assert_array_equal( R R R R R R R R R R R t s( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_basic1dH s ' % )
c
C s. | j \
} } } } } } } } } }
x d d
g D]� } | | _ | | _ | | _ | | _ | |
_ | j t | � � | j t | � � t t | � | � t | j | � t | j t d � | � � t t | � t | � t d � | � � t | |
� t t | d � |
� t | | � q4 Wd S( s< Test of basic array creation and properties in 2 dimensions.i i i i c S s | | S( N( ( R R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyR, i s c S s | | S( N( ( R R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyR, j s g@��x �DN( i i ( i i (
R R R' R- R R R R2 R3 R( ( R R R R R R R R R R R R5 ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_basic2d[ s ' )
c
C s� | j \
} } } } } } } } } }
t t j | | f � t | | f � � t t j | | f � t | | f � � t t j | | f � t | | f � � t t j | | | f � t | | | f � � d S( s Tests concatenations.N( R R R t concatenate( R R R R R R R R R R R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_concatenate_basico s
'(((c
C s� | j \
} } } } } } } } } }
d } | | _ | _ | _ | _ t | j t j | | � � t | j t j | | � � t | | f d � } t t j | | f d � | � t t j | j | j f d � | j � t d � } t t
d � d t t g �} t | | f � } t
| d d d d g � t
| j t t t t g � t | | f � } t
| d d d d g � t
| j t t t t g � d S( s Tests concatenations.i i i i R i N( i i ( R R R R R t reshapeR8 t _maskt zerosR t onesR& R$ R4 (
R R R R R R R R R R R R5 t xmym( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_concatenate_alongaxisx s ' ( c C sm t t t j j d � t j d � � d d t f d t f g �} t | d | d g � } t | | � d S( s+ Tests the concatenation on flexible arrays.i
R) R R i N(
R t zipR t randomt randt aranget floatR1 R8 t assert_equal_records( R t datat test( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_concatenate_flexible� s
c C sv t d d d g d d d d g d d �} t | j d � t | j d d d g g � t | j d d d g g � d S( s Check the use of ndmini i i R i t ndminN( i i ( R R R R* R; ( R R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_creation_ndmin� s * c C sg t d d d g � } t | d <t | d d d t �} t | j | j j � t | j | j j � d S( s5 Make sure we're not losing the original mask w/ ndmini i i i����RI R) N( R t maskedRD R R R; ( R R t xx( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt$ test_creation_ndmin_from_maskedarray� s
c C s� t d d t �} t | d d d g <t | � } t | j | j � t | � } t | j | j � t | d d d d d g d �} t | j | j � d S(
s@ Tests how masks are initialized at the creation of Maskedarrays.i R) i i i R i i N( RC RD RK t MaskedArrayR R R/ ( R RF t dma_1t dma_2t dma_3( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_creation_maskcreation� s "c C s! t t j d � d d d d d d g �} t | | d d d � f � } t | d d d d d g d d d d d g g � t | j d d d d d g d d d d d g g � t | _ t | | d d d � f � } t | d d d d d g d d d d d g g � | j | j t k � d S(
s; Tests creaating a masked array from alist of masked arrays.i R i i Ni����i i i ( R R RC R R; t nomaskR R' ( R R RF ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt' test_creation_with_list_of_maskedarrays� s * 14 1c
C s� | j \
} } } } } } } } } }
d | _ t | _ t | � } t | j | j � t | j | j � t | j | j � t | j | j � d S( Ni���( R t
fill_valueR$ t _hardmaskt asarrayR R* R; ( R R R R R R R R R R R t xmm( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_asarray� s ' c C s� t j � } z| t j d d � t t j d d g d d d d g �} t | � } t | j | j d d g � t | j d d d g � Wd t j | � Xd S( s Checks fix_invalid.t invalidt ignoreg g �?R i i N(
R t geterrt seterrR t nant fix_invalidR R* RU R; ( R t err_status_iniRF t
data_fixed( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_fix_invalid� s ' c C sd t d � } t | d <| j t t � d k � | j | d t k � t t | d d � d � d S( s Test of masked elementi i s --i N( RC RK R' R# R R( ( R R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_maskedelement� s
c C s� t d d t �} d
} | | d <t | d | � | j | d | k � d d l } | j j � } | | d <| j | d | k � d S( s" Tests setting elements with objecti R) i i i i i i����N( i i i i i ( t emptyt objectR R' t datetimet now( R R R Rf t dt( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_set_element_as_object� s
c C s� t j d d d d g � } t | d d d d d g �} t | d d d d d g �} t | � } t | � t | � } } t t j | � t | d t �� t t | d � t | d � k � t | d | d k � t | d t k � t | d | d � t | d d !| d d !� t | | � t | d | d � d | d <d | d <t | | � d
| d d +d
| d d +t | | � t | d <t | | � t | d d +t | | � | | (t | d <t t
t | � t d d d d g � � � t d d d d g d d d d g � | (t t
t | � t d d d d g � � � t d d d d g d d d d g � | (t t
t | � t d d d d g � � � t t
| t d d d d g � � � t j
d � d } t | d � } t | | � t t
t d d d d d g t � | j � � t d | j � t d d
d d g t � } t j d d
d d g t � } | d } | d } t t | � t � t t | � t � t | | � t | d d !j d k � d S( s Tests conversions and indexingi i i i R i t endwithi i ic g �?g @t helloN( i ( R R R# t reprR t sortR& t assert_t typeRK t allequalt getmaskR RC t
masked_valuest MaskTypeR RU Re R ( R t x1t x2t x3t x4t junkt garbaget s1t s2( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt
test_indexing� sX "$
+(+(+%
.
c
C s� d d d d d g } t | � } t | � } | j | | k � t | d d �} | j | | k � t j d � } t | d | �} t | j j | j � | j t | | j � � t | j
j | j � t | � } | j | j j | j j k � | j | j | j k � t | d | �} | j | j j | j k � | j | j
j | j k � | j | d t k � d | d <| j | d t k � | j | j
j | j k � | j t | j d � � t | d d | �} | j t
| � j | d j k � t d � }
t |
d <t |
d � } t t |
|
g � | � t t | � d d d d d d d d g � t |
d d d �} t | d d d d d d d d g � t |
d d d �}
t | |
� |
j d d d �} t | | � |
j d d � } t | | � |
j � } t | j |
j � t | j
|
j
� t d d d g d d d d g �} t | � } t | j j j | j j j � t | j
j j | j
j j � t | d t �} t | j j j | j j j � t | j
j j | j
j j � d
S( s2 Tests of some subtle points of copying and sizing.i i t copyi R i i g �?i i t axisi N( i ( i i i i ( i i i i ( t make_maskR' R RC R R R* t __array_interface__Rp RF R; R RK R( R) t resizeR8 Rq t repeatR} R t ctypesR$ t assert_not_equal( R t nt mR t m3Rt t y1t y1at y2t y3Rw t y4t y5t y6t y7t y8t y9R R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_copy s^
#
+%
$ c C s d d l m } t d d d g d t t t g �} | | � } t | j | j � t t | j � t | j � � d | d <t | j d d d g � t | j d d d g � | | � } t | j | j � t | j d <t | j d d d g � t | j d d d g � d S( Ni����( t deepcopyi i i R (
R} R� R R& R$ R R R� t idR; ( R R� R t copied( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt
test_deepcopy^ s $
c C s� d d l } t d � } t | d d d � <d | _ | j | j � � } t | j | j � t | j | j � t | j d � d S( s Tests picklingi����Ni
i i� ( t cPickleRC RK RU t loadst dumpsR R; R* ( R R� R t a_pickled( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt
test_picklingp s c C s� d d l } t t j t d � � d d d d d d g d �} | j | j � � } t | j | j � t | | � | j t
| j t j � � d S( s& Test pickling w/ a subclass of ndarrayi����Ni
R i i i ( R� R R t matrixt rangeR� R� R R; R' t
isinstanceR* ( R R� R R� ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_pickling_subbaseclass{ s 4
c C sj d d l } t j j } | j | j � � } t | j | j � t | j | j � t | j | j � d S( s Test pickling MaskedConstanti����N(
R� R t maRK R� R� R t
_baseclassR; R* ( R R� t mct
mc_pickled( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_pickling_maskedconstant� s c C s{ d d l } t d d
g d d d g d d
t f d t f g �} | j | j � � } t | j | j � t | | � d S( s" Tests pickling w/ structured arrayi����Ni g �?i g @R i R) R R ( i g �?( i g @( i i ( i i ( R� R R1 RD R� R� R R; ( R R� R R� ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_pickling_wstructured� s c C sS d d l } t d � } d | _ | j } | j | j | � � } t | | � d S( s% Tests pickling w/ F_CONTIGUOUS arraysi����Ni
i ( i����i ( R� RC R t TR� R� R ( R R� R R RG ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_pickling_keepalignment� s c C sy t d d d g � } t d d d g d d d d g �} t | d j d � t | d j d � t | d j d � d S(
s0 Tests single element subscripts of Maskedarrays.i i i R i N( ( ( ( R R R ( R R R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_single_element_subscript� s
$c s� t d t t d � � � t d t t d � � � t d t t d g g g � � � t d t t d g g � � � | j t t t d d g � � t � } | j � z? t j d t
� t t j
t t d g d d g �� � � Wd | j � Xt d d d g d d d d g �� | j t � f d � � t t � d
� d � | j t j
t � d � � � | j t t � � t t � d
� d � | j t � f d � � d S(
s, Tests some communication issues with Python.i g �?R[ R Ni i i c s
t � � S( N( RD ( ( R ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyR, � s i����g @c s t � d � S( Ni ( R1 ( ( R ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyR, � s ( R R1 R RD t assertRaisest TypeErrorR t __enter__t warningst simplefiltert UserWarningRn R t isnant __exit__R' t MAError( R t warn_ctx( ( R s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt
test_topython� s$ "
/ $ c C s� t d � } | j d d � } d | j d <t | d d k � | d | } t | j | � t | j d | � t | t | � j d | | � d
| j d <t d � } t | d <t t | d � t t � k � | d
k } t t
t | t t � � d k � t t t | t t � � | j k � t
| | � } t | j | j k � t | d t k � t | d t k � t | d t k � t | d
t k � t | d t k � t | | � d S( s Test of other odd featuresi i i i i i y $@i
ie g .i i i i N( i i ( RC R: t flatRn R t realt imagt conjugateRK R# R2 R R t masked_whereR) ( R R R t c( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_oddfeatures_1� s.
"
"%c C s� t d d d d d g � } t d d d d d g � } t | d <t | | | � } t | d d d d
d g � t | d <t | | | � } t | d d d d
d g � t | d t k � t | d t k � t | d t k � d S(
s Tests some more features.g �?g @g @g @g @i i i g g �i����N( R RK R R Rn ( R R R� R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_oddfeatures_2� s
c C sO t d g d t �} t d g � } | j } | | | | <t | d g � d S( s Tests some generic features.i
R i N( R R$ R R ( R t atestt btestt idx( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_oddfeatures_3� s
c C s� t d
g d d t f d d t f g �} t | d <t | j � t j t d � t d � t d � f g d | j �� t | d <t | j d � t j d g d | j �� d S( s Test filled w/ flexible dtypei R) t iR5 s |S8t fi R" g t 1g �?N( i i i ( R5 s |S8( i R� g �?( R R1 RD RK R R( R t default_fill_valueR) ( R t flexi( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_filled_w_flexible_dtype� s
c C s� d t f d t f g } t d d d
g d | �} | j � } t t | � d t d � f � | j d � } t t | � d � d | _ t t | j � � d � d S( s Test filled w/ mvoidR R i g @R i R) g �?i����i���N( i g @( i i ( i����i����( i i����( i���i���( i i���( R1 RD t mvoidR( R t tupleR� RU ( R t ndtypeR RG ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_filled_w_mvoid s c C s� d t f d d t f d t f g f g } t d d f d d f g d d d
f d d f g d | �} | j d � } t j d d f d d f g d | �} t | | � | d j d � } t j d d g d | d j �} t | | � d
S( s Test filled w/ nested dtypet At Bt BAt BBi i R i R) N( i i ( i i ( i i ( i i ( i i ( i i ( i i ( i i ( R1 R R( R R R) ( R R� R RG t control( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_filled_w_nested_dtype s *!'
"c C sm t d d d g d t �} d | j d <| j � } t | j d d � d | j d <t | j d d � d S( s5 Checks that _optinfo dictionary isn't back-propagatedi i i R) s ???t infos !!!N( R RD t _optinfoR} R ( R R R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_optinfo_propagation( s
c C s� t j d t f d d t f d t f g f g � } t d d f d d f g d d d f d d f g d
| �} d } t t | � | � d S( s, Test printing a masked array w/ fancy dtype.R R t tR5 i i g @i i g @R i R) s [(--, (2, --)) (4, (--, 6.0))]N( i g @( i g @( i i ( i i ( R R) R1 RD R R R# ( R t
fancydtypeRG R� ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_fancy_printoptions2 s 3 c C s� d t f d t f g } t j d d g d | �} t | � } t j d d g d d g g d t j �} t | | � t | j | j � t d d g d d d g d | �} t | � } t d d g d d g g d d d g d d g g d t j �} t | | � t | j | j � t | j | j � d t f d d
t f d t f g f g } t d d f d d f g d d d f d d f g d | �} t | � } t d d d g d d d
g g d d d d g d d d g g d t j �} t | | � t | j | j � t | j | j � d t f d t f g } t j d g d g g d | �} t | � } t j d d g g d d g g g d t j �} t | | � t | j | j � d S( s' Test flatten_structured_array on arraysR R i i R) g �?g @R i t bat bbg�������?g������@N( i i ( i i ( i i ( i i ( i i ( i i ( i g�������?( i g������@( i i ( i i ( i i ( i i ( R1 RD R R t flatten_structured_arrayR R) R ( R R� R RG R� ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_flatten_structured_array< s8 *
$ $
*! !*
! 0
c C s� d t f d t f g } t j d g d | �d } t | � } t t | t � � t d
g d d g d | �d } t t | t � � t d d
g d d d g d | �} t | j d | j d � } t t | t � � d S( s Test creating a mvoid objectR R i i R) i R N( i i ( i i ( i i ( i i ( i i ( i i ( i i ( R1 R R R� Rn R� R R* R; ( R R� R R� ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_void0da s "$ c C s� d t f d t f g } t d d g d d
d g d | �} | d } | j t | t j � � t | d | d f d � t | d d � | d } | j t | t � � | j | d t k � | j | d t k � t | d d � d
S( s Test mvoid.__getitem__R R i i i i R i R) N( i i ( i i ( i i ( i i ( i i ( R1 R R' R� R t voidR R� RK ( R R� R R� ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_mvoid_getitemo s $
c C sz d t f d t f g } t d d g d d
d g d | �} t t | d � d d g � t t | d � t d g � d
S( s Test iteration on __getitem__R R i i i i R i R) N( i i ( i i ( i i ( i i ( R1 R R t listRK ( R R� R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_mvoid_iter s $ c C s� t d d g d d t f d t f g �} t t | d � d � t | d d <t j } t j d � z2 t t | d � d � t t | d � d � Wd
t j | � Xd
S(
s Test printing a mvoidi i R) R R i s (1, 1)s -X-s (1, -X-)N( i i ( i i ( R R1 R R# RK R% t _displayt set_displayRl ( R t mxt ini_display( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_mvoid_print� s *
(* t __name__t
__module__t __doc__R R! R+ R6 R7 R9 R? RH RJ RM RR RT RY Rb Rc Ri R| R� R� R� R� R� R� 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_core.pyR sP
2 ?
% t TestMaskedArrayArithmeticc 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 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
C s� t j d d d d t d d d d d d d d g � } t j d d d d d
d d d d d d d g � } d } d d
d
d
d
d
d d
d
d
d
d
g } d
d
d d
d
d d d
d
d
d
d g } t | d | �} t | d | �} t j d d d d g � } t | d d
d d
d
g �} t j | d | � }
| j d � | | | | | | | | | |
f
| _ t j � | _ t j d d d d � d S( s Base data definition.g �?g �g @g @g @g $�g $@g @g g �g �i i R g �g �?g�������?g@��x �Dt divideR[ RZ N(
R R R
R R R
R R\ t
err_statusR] ( R R R R R R R R R R R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyR � s 73**
'c C s t j | j � d S( N( R R] R� ( R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt tearDown� s c
C s� | j \
} } } } } } } } } }
t d d g d d g g � } t | d d g d d g g � } t | | | | � t | | | | � t | | | | � x@d
d d g D]/}
| j |
� } | j |
� } | j |
� } | j |
� } |
j |
� }
t | | � t | | | | � t | | | | � t | | | | � t | | | | � t | | | | � t | | | | � t | | | | � t | | | | � t | | | | � t | | | | � t | | | | � t | | | | � t | d | d � t t | � d t | � d � t | | | | � t t j | | � t | | � � t t j | | � t | | � � t t j | | � t | | � � t t j
| | � t
| | � � q� Wd S(
s Test of basic arithmetic.i i i i i i i g @N( i ( i i ( i i ( R R R R R: t absR t addt subtractt multiplyR� ( R R R R R R R R R R R t a2dt a2dmR5 ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_basic_arithmetic� s@ ' !!"""c C sh t d d t �} d
| _ t d d t �} | | } t | d d d g d d d g g � t | j d d
d
g d d
d
g g � | | d d d � f } t | d d d g d d d g g � t | j d d
d
g d d
d
g g � t d d t �} | | d d � d f } t | d d d g d d d g g � t | j d d d g d
d
d
g g � d S( Ni R) i i g �g �?g @g @i i g @g @( i i ( RC RD R R R t None( R R R R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_divide_on_different_shapes� s
%(%(%c C sY t j d g � } t d g � } | j t | | t � � | j t | | t � � d S( s Tests mixed arithmetics.i N( R R R' R� RN ( R t naR� ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_mixed_arithmetic� s c C so t j t � j } t | d | d g � } t t | d � d d d g � t t d | � d d d g � d S( Ng �?g i i i ( R t finfoRD t tinyR R t getmaskarray( R R� R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_limits_arithmetic� s c C s� t d d d �} | j d t d � j � | j d | j � | j | j � | j t | | � j � | j t | | � j � d S( s. Tests some scalar arithmetics on MaskedArrays.i R i N( R R' R t maximumt minimum( R R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_masked_singleton_arithmetic� s c C s� t d d d g d d d d g �} t | d d k t k � t | d d k t k � t | d d k t � t | d d k t � d S( s& Tests (in)equality on masked snigletoni i i R i i����N( R Rn RK R R& R$ ( R R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_masked_singleton_equality� s
$c C s� t d d g � } | t } t | j | j � t | j t t g � | d t } t | t k � | t } t | j | j � t | j t t g � d S( s+ Checks that there's no collapsing to maskedi i i N( R RK R R R; R$ Rn ( R R R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt% test_arithmetic_with_masked_singleton s
c C sC t d g � } | t } t | j | j � t | j t g � d S( s4 Check that we're not losing the shape of a singletoni N( R RK R R R R$ ( R R R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt5 test_arithmetic_with_masked_singleton_on_1d_singleton s
c C sk t d d d �} t | j � j j | j j � t d � d } t | j d � t | j d d g � d S( Ni R g i i ( i i ( i ( R R R( R� RF R R ( R R R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_scalar_arithmetic s
c
C s3 | j \
} } } } } } } } } }
t t j | � t | � � t t j | � t | � � t t j | � t | � � t t j | � t | � � t t j | � t | � � t t j | � t | � � t t j t
| � � t | � � t t j t
| � � t | � � t t j t
| � � t | � � t t j
| � t
| � � t t j | � t | � � t t j | � t | � � t t j | � t | � � t t j | | � t | | � � t t j | � t | � � t t j | d | � t | d | � � t t j | d | d t �t | d | d t �� t t j | | � t | | � � t t j | | � t | | � � t t j | | � t | | � � t t j | | � t | | � � t t j | | � t | | � � t t j | | � t | | � � t t j | � t | � � d S( s( Test various functions such as sin, cos.y �?t degN( R R R t cost cosht sint sinht tant tanht sqrtR� t logt log10t expt arcsint arccost arctant arctan2t absolutet angleR$ t equalt not_equalt lesst greatert
less_equalt
greater_equalR� ( R R R R R R R R R R R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_basic_ufuncs& s2 ' """ " ,8""""""c C s� t d d d d g d d d d d g �} t j d d k r_ | j t t | � t j � � n | j t t | � t � � t d t | � � t d t d � � t d t d d d g �� | j
d � } t t t | d � t � � t j d d k rt t t | � t j � � n t t t | � t
j � � t d t | � � t t t | d � � t k � t d d g t | d � � d
S( s Tests countg g �?g @g @R i i i i N( i i ( R t syst version_infoR' R� R2 R t integerR1 R R: Rn R t typest IntTypeRq RS ( R t ott( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_count_funcC s *" c
C s� | j \
} } } } } } } } } }
t j | � } t | � } t t | � t | � � t t | � t | � � t t d d d g d d d g � d d d g � t t d d d g d d d g � d d d g � t d � } t d � d } t | d <t | d <t t | | � t
t | | � | | � � t t | | � t
t | | � | | � � t
t | � d k � t
t | � d k � t d � j d d � } t | d
<t t | � d � d S( s Tests minimum and maximum.i i i i i i i i����N( i����i����( R R t ravelR t maxR� t minR� RC RK R R R Rn R: (
R R R R R R R R R R R t xrt xmr( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_minmax_funcX s$ ' 11
++
c C s t j d � } t | | � } | j t | t � � t | t j | | � � t j | | � } | j t | t � � t | t j j | | � � t | | � } | j t | t � � t | t j | | � � t j | | � } | j t | t � � t | t j j | | � � d S( Ni ( i i ( R R= R� R' R� RN R t outerR� ( R R t aminimumt amaximum( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_minimummaximum_funcp s c C sG t d d d g d t t t g �} t j j | � } t | d � d S( s6 Test np.min/maximum.reduce on array w/ full False maski i i R N( R R& R R� R R ( R R R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_minmax_reduce� s $c C s; t j j d � j � } t t j j d d d � d | �} d | _ x� d D]� } t t | � } t t j j
| � } t j d d t �} y | | d
d d | �} Wn t
k
r� n Xt j d d t �} | | d
d d | �} | j | | k � | j d � | | d
d d | �} | j | | k � qL Wd
S( s1 Tests the min/max functions with explicit outputsi i i
R i i R" R! R) R~ t outi���N( i i ( s mins max( i ( i ( R RA RB t roundR t uniformR t getattrt numpyR� t coreRd R1 t MaskErrorRD R' t fill( R R R t funcnamet npfunct mafunct noutt result( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_minmax_funcs_with_output� s" $
c
C s� | j \
} } } } } } } } } } | j f | _ t | j � d � | j | d j � t k � | j | d j d � t k � | j | d j d � t k � t | j � d � | j | d j � t k � | j | d j d � t k � | j | d j d � t k � t | j � d � | j | d j � t k � | j | d j d � t k � | j | d j d � t k � t d d d g d t
�} | j | j � t k � | j | j � t k � | j | j � t k � d
S( s Additional tests on max/mini
i i����g $�g 4@i i i R N( R R R R R! R' RK R" t ptpR R$ ( R t _R R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_minmax_methods� s$ ' c
C sv | j \
} } } } } } } } } }
t t j j | � t j | � � t t j j | � t j | � � t d t t d � d d �� t d t t d � d d �� t t j | d d �t | d d �� t t j t | d � d d �t | d d �� t t j | d � t | d � � t t j | d d �t | d d �� t t j | d � t | d � � t t j t | d � d d �t | d d �� d } | | _
| _
| _
| _
t | � d k rrt t j | | f d � t | | f d � � t t j j | d � t j | d � � t t j | d � t | d � � t t j | d � t | d � � n d S( s Tests add, sum, product.i R~ i i i N( i i (
R R R R� R t
accumulatet sumR R( t productR R3 R8 ( R R R R R R R R R R R R5 ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_addsumprod� s$ '"" (1"("1 .("c C s� t d g d g d g g d t g t g t g g �} t d d g d d g d d g g � } | | } t d d g d d g d d g g d d d g d
d
g d
d
g g �} t | | � t | j | j � t | j | j � | | } t d d g d d g d d g g d d d g d
d
g d
d
g g �} t | | � t | j | j � t | j | j � t d g d g d g g � } t d d g d d g d d g g d d d g d d g d d
g g �} | | } t d d g d
d g d d g g d d d g d d g d d
g g �} t | | � t | j | j � t | j | j � | | } t d d g d
d g d d g g d d d g d d g d d
g g �} t | | � t | j | j � t | j | j � d S( s! Test binary operations on 2D datag �?g @g @R g @g @g @g @i i i i i i
i i N( R R& R$ R RF R ( R R R RG R� ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_binops_d2D� s: 6'
$$
$$
$$
$$
$$
c C s t d g d g d g g d t g t g t g g �} t d d g d d g d d g g � } | | } t d d d d g d d g d d g g d d d g d
d
g d
d
g g �} t | | � t | j | j � t | j | j � | | } t d d d d g d d g d d g g d d d g d
d
g d
d
g g �} t | | � t | j | j � t | j | j � t d g d g d g g � } t d d g d d g d d g g d d d g d d g d d
g g �} | | } t d d d d g d d
d d g d d d g g d d d g d d g d d
g g �} t | | � t | j | j � t | j | j � | | } t d d d d g d
d d d g d d d g g d d d g d d g d d
g g �} t | | � t | j | j � t | j | j � d S( s* Test domained binary operations on 2D datag �?g @g @R g @g @g @g @i i i i i i i i N( R R& R$ R RF R ( R R R RG R� ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_domained_binops_d2D� s: 6'
,$
,$
$$
8$
8$
c C s� t d d d g d t t t g d t �} | d } t | j d d d g � | d 7} t | j d d d g � | d } t | j d d d g � | d :} t | j d d d g � d S( s1 Check that we don't shrink a mask when not wantedg �?g @g @R t shrinki i N( R R& R R ( R R R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_noshrinking s *
c
C s� | j \
} } } } } } } } } }
t t | | � t | | � � t | | � } t | t j | | � � t | j t | j | j � � t | | � } t | t j | | � � t | j t t | j | j � | d k � � d S( s Tests modi N( R R t modR R t mask_or( R R R R R R R R R R R RG ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_mod) s ' c C s� t d � } t j d � } t | d d +| j d d d � } | j d d d � } t t j | d � t | d
� � t t j | d d � t | d d � � t t j t | d � t | d � � t | | � � t t j t | d � t | d � � t | | � � t
d d d
d d g t � } t | d <t | d d d g � } t | d d k � t | d d k � t | d d k � d S( s. Test of take, transpose, inner, outer productsi i i i i i i i t abct defN( i i i ( i i i ( i i i ( i i i (
RC R RK R: R t transposet taket innerR( R& R Re Rn ( R R R R� ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_TakeTransposeInnerOuter4 s"
"($$
c C s� t d d g d d d g �} t | j d d g � t | j j � d d g � t | j j | j j j � t | j d d g � t | j j � d d g � t | j j | j j j � d
S(
s
Check complexi y $@i y @R i i
i g@��x �DNy �? $@y 4@ @( R R R� R( R) R* R� ( R RL ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_imag_realI s c C s: t t j j d d d � � j d d � } t | d d � d f <| d <| d <d } x� | D]� } t t | � } t | | � } t j d d t �} | j d � | | d d d | �} t
| | k � t | | d d d | � � t d d t �} | d d d | � } t
| | k � t
| d t k � q] Wd S( Ni i
i i i i����R= t prodt vart stdR! R" R9 t meanR) i���R~ R+ ( i����i����( s sums prods vars stds maxs mins ptps mean(
R R RA R- R: RK R. Rd RD R2 Rn R R1 ( R R t funclistR3 R4 t xmmetht outputR7 ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_methods_with_outputT s *&
c C s' d t f d t f g } t d d
g d d d g d | �} | | k } t | t t g � t | j t t g � t d
d g d d d g d | �} | | k } t | t t g � t | j t t g � t d d g d d d g d | �} | | k } t | t t g � t | j t t g � d S( s& Test the equality of structured arraysR� R� i i R i R) N( i i ( i i ( i i ( i i ( i i ( i i ( i i ( i i ( i i ( i i ( i i ( i i ( R1 R R R$ R R& ( R R� R RG R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_eq_on_structuredk s $ $ $ c C s' d t f d t f g } t d d
g d d d g d | �} | | k } t | t t g � t | j t t g � t d
d g d d d g d | �} | | k } t | t t g � t | j t t g � t d d g d d d g d | �} | | k } t | t t g � t | j t t g � d S( s& Test the equality of structured arraysR� R� i i R i R) N( i i ( i i ( i i ( i i ( i i ( i i ( i i ( i i ( i i ( i i ( i i ( i i ( R1 R R R& R R$ ( R R� R RG R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_ne_on_structured| s $ $ $ c C s t d d g d d d g �} t | d k t � t | j d k t � t | j d k t � t | d k t � t d d g d t �} t | d k t � t | d k t � t d d g d t �} t | d k t � t | d k t � t } t | d k t � d S( Ni i R i ( R R R� R& RF R R$ RK ( R R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_eq_w_None� s c C s\ t d � } t | d k t � t | d k t � t | d k t � t | d k t � d S( Ni i ( R R R$ R& ( R R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_eq_w_scalar� s
c C s t d d d d d g d d d d d d g �} t t j t j d t j d � d g d d d d d d g �} t | � } t | | � t | j | j � t | j d d d d d g � t j | � } t | | � t | j | j � t | j d d d d d g � d S( sE Check that the mask is not back-propagated when using numpy functionsi����i i i i R N( R R R^ R R R ( R R R� RG ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_numpyarithmetics� s 0'
(" R� R� R� R R� R� R� R� R� R� R� R� R� R R R R% R) R* R8 R; R? R@ RA RC RF RL RM RU RV RW RX RY RZ ( ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyR� � s@ $
% % t TestMaskedArrayAttributesc B sG e Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z RS( c C s� t d d d g d d d d g �} t | � } t | j | j � t | d d d d g d t �} t | j d d d g � t | d d d d g d t �} t | j d d d g � t | d d d d g �} t | j d d d g � d S( s Tests the keep mask flagi i i R i t keep_maskN( R R R R& R$ ( R R R� ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt
test_keepmask� s $ !!c C s� t d � } d d d d d g } t | � } t | d | d t �} t | d | d t d t �} d d g | d d g <d d g | d d g <t | j d d d
d d g � t | j d d d
d d g � t | j d d d d d g � | j | j � | j | j � d d d
g | d d +d d d
g | d d +t | j d d d d d g � t | j d d d d
d g � t | j t
� t | d <t | d <t | j d d d d d g � t | j d d d d d g � d | (d | (t | j d d d d d g � t | j d d d d d g � t | j d d d d d g � t | j t
� | j � t d � | (t | j d d d
d d g � t | j t
� | j
� t | | d k <t | j d d d
d d g � t | j d d d d d g � d | t | d k t � <t | j d d d
d d g � t | j d d d d d g � t d d
g d d g g d d d g d d g g d t �} d | d <t | j d d g d d g g � t | j d d g d d g g � d | d <t | j d d g d d g g � t | j d d g d d g g � d
| t | d k t � <t | j d d
g d
d g g � t | j d d g d d g g � d S( s Test hard_maski i i R t hard_maskR} i
i( i i i i i i����N( i����i����( RC R R R$ R& R R* R R' RV RS RK t soften_maskt harden_maskR; R( ( R R R� R� t xht xs( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt
test_hardmask� s`
<
""
"""c C s� t d � } d d d d d g } t | � } t | d | d t �} d | d d +d | d d +t | j d d d d d g � d
S( s Another test of hardmaski i i R R^ i� i i i N( RC R R R$ R R* ( R R R� R� Ra ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_hardmask_again s
c C s� t d d d g d d d d g �} | j � } t | | � d | d <t | | � t | t d d d g d d d d g �� | j � } d | d <t | | � t | t d d d g d d d d g �� d S( s OK, yet another test of hardmaski i i R i N( R R` R R_ ( R R R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_hardmask_oncemore_yay
s $
+
c C ss t d � } t | d <d | d <t | j t � t d � } t | _ t | d <d | d <t | j t d � � d S( s" Checks the behaviour of _smallmaski
i N( RC RK R R; RS R& t
_smallmaskR< ( R R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_smallmask s
c C sQ t d d d g d d d d g �} | j � } t | | � t | j t � d S( s Tests .shrink_mask()i i i R i N( R t shrink_maskR R RS ( R R R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_shrink_mask) s $
c C s� t t j d d d g g � d d d d g �} t d d d g d d d d g �| _ t t j d d d g g � d d d d g �} t | | � t t j d d d g g � d d d d g �} | j } | d d d g | (t | | � d S( s Test flat on masked_matricesi i i R i N( R R R� R� R ( R RG R� t testflat( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_flat1 s 0'0
0 ( R� R� R] Rc Rd Re Rg Ri Rk ( ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyR[ � s
9
t TestFillingValuesc B sY e Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z
RS( c C s� t j j j } | d t � } t | d � | d t � } t | t d � � | d d � } t | t d � � | d d � } t | t d � � | d t � } t | t d � � d S( s Test _check_fill_valuei s |S3R" g@��x �DN( R R� R0 t _check_fill_valueR1 R R� R� R ( R Rm t fval( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_check_on_scalarA s
c C s� t j j j } d t f d t f d g } | d d d g | � } | j t | t � � t | j
� d d t d � g � | d
| � } | j t | t � � t | j
� t
d � t
d � t t
d
� � g � t j d d | �} | | | � } | j t | t � � t | j
� d d t d � g � t j d d d d t f d t f g �} | | | � } | j t | t � � t | j
� d d t d � g � t j d d t �} | | | � } | j t | t � � t | j
� d d t d � g � t j d � } | | | � } | j t | t � � t | j
� d d t d � g � d t f g } | d | � } | j t | t � � t | j
� d � d
S( s$ Tests _check_fill_value with recordsR R R� s |S3i���g����)�g�s ???i g R" R) i6e�N( R� s |S3( i���g����)�g�s ???( s ???i���g����)�g�( R� s |S3( i���g����)�g�s ???( i���g����)�g�s ???( i6e�( R R� R0 Rm R1 RD R' R� R R t itemR R� R� R Re ( R Rm R� Rn t fill_valRU ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_check_on_fieldsS s> " " """c C sJ t t d d d g � � } | j j i d d 6� t | d t �} t | j d d d g � t | j t d
� � t | d t �} t | j d d d g � t | j t d � � | j
t � } t | j d d d g � t | j t d
� � t | j d d � | j
d g � } t | d j | j � t | d j | j � d S( s2 Tests the behavior of fill_value during conversiont 3t 4t 5s updated!t commentR) i i i i g R s |S3N( R s |S3( R R R� t updateR1 R R* RU R� RD R0 ( R R R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_fillvalue_conversion� s c C s� t d d d g d d �} | d d d g } t | j | j � d t f d g } t d d t d f g d
| �} d | _ t | j j � d t d � g � t | d j d � t | d j t d � � d | _ t | j j � d t d � f � t | d j d � t | d j t d � � t d d d g � } d | _ t t j
| j � j t � t | j d � t | j t j d � � d S( s Yet more fun with the fill_valuei i i RU i���i R� R5 s |S3R R R
R) i� g 8�@t 999i s ???g������@N( R5 s |S3( i R ( i R ( i s ???( R R t _fill_valueRD R R
RU Rp R R RW R) ( R RF t seriest mtypeR ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_fillvalue� s$ ! " " c C s� t j j j } d t f d d t f g } t j t d � t d � t d � f d | �} t | d
| � | � d t d f g } t j t d � f d d t f g �j
| � } t | d
| � | � t j d d d t f g �j
| � } t | d | � | � t j d � } t j t d � t d � t d � f d d �j
| � } | d
| � } t | | � t j d d d �j
| � } t | d | � | � d
S( s% Tests yet more exotic flexible dtypesR� R5 s |S8R� i R" g R) t f0i s int, (2,3)float, floats int, float, floatN( R5 s |S8( i i ( i ( i i i ( R R� R0 Rm R1 RD R R� R R� R0 R) ( R Rm R� R� RG ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_fillvalue_exotic_dtype� s, '
c C s� t d d f d d f g d d t f d d
t f d t f g f g �} | j } t | d t | d � � t | d d
t | d d
� � t | d d t | d d � � t | � } t | d t | d � � t | d d t | d d
� � t | d d t | d d � � t | d t | d � � t | � } t | d t | d � � t | d d t | d d
� � t | d d t | d d � � t | d t | d � � d
S( s- Tests extremum fill values for flexible type.i i i i i i R) R� R� R� R� i N( i i ( i i ( R R1 RU R R� t minimum_fill_valuet maximum_fill_value( R R RG ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_extremum_fill_value� s - ## ## ##c C sR d t f d t f g } t t d d d g d d d g � d d d | �} | j } | d } | j d � t | j t j d � � t t | j � d � d
| j d <t t | j � d � t t d d d g d d d g � d d t f d t f g �} | d } | j d � t | j t j d � � t t | j � d t d � f � d S( s, Test setting fill_value on individual fieldsR R i i i i i i RU i���R) i
i����i N( i���i���( i
i���( i
i����(
R1 R R@ Rz R
R R R� RU R� ( R R� R R� t aaR� t tt( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_fillvalue_individual_fields� s $
?
c C s� d t f } d t f } t d d g d t f t f g d t j f d t j | g � �} t | j d | | g �} | d | d <| d j | d j � | j
d
} t t j | d � � t
| d t d � � d
S( s= Check that fill_value is always defined for structured arraysR R g �?g @R RU R) i i����N( g �?( g @( ( RD R R& R R^ R) Rd R R
RU Rz Rn R� R R� ( R R� t adtypeR R R� ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt( test_fillvalue_implicit_structured_array� s !!
c C s� t d d d �} t | j d � t d d d d t �} t | j d � t d d d d t �} t | j d � t d d d d t �} t | j d � d S( s6 Test adding a fill_value parameter to empty/ones/zerosi RU g 8�@R) g N( Rd R RU R= RD R< t complext identity( R R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_fillvalue_as_arguments s ( R� R� Ro Rr Rx R} R R� R� R� R� ( ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyRl ? s 0
t
TestUfuncsc B sD e Z d Z d � Z d � Z d � Z d � Z d � Z d � Z RS( s9 Test class for the application of ufuncs on MaskedArrays.c C s� t d d d t d g d d d d g d g d �t d d d t d g d d d d g d g d �f | _ t j � | _ t j d d d
d � d S( s Base data definition.g �?i i����i R i i R� R[ RZ N( R R
R R R\ R� R] ( R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyR s 4=c C s t j | j � d S( N( R R] R� ( R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyR� ! s c * C sA x:d d d d d d d d d d
d d d
d d d d d d d d d d d d d d d d d d d d! d" d# d$ d% d& d' d( d) d* g* D]� } y t t | � } Wn t k
r� t t | � } n Xt t j j | � } | j | j } | | � } | | � } t | j
d+ � | j
d+ � | � t | j | j d, | �q� Wd- S(. s! Tests new ufuncs on MaskedArrays.R R R
R R� R R R R R
R R R R t arcsinht arccosht arctanhR t fabst negativet floort ceilt logical_notR� R� R� R� t true_dividet floor_dividet remaindert fmodt hypotR R R R R R R t logical_andt
logical_ort logical_xori t err_msgN(
R. t umatht AttributeErrort fromnumericR/ R� R0 R t ninR R( t assert_mask_equalR ( R R� t uft mft argst urt mr( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_testUfuncRegression$ s4
"c C s� | j d } | j t | d d � � | j t | d d �� t t | d d d �d � t t | d d �d � t t j | � t � d S( s Tests reduce on MaskedArrays.i R~ i N(
R R' t alltruet sometrueR R= R> R� R R
( R R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_reduceD s
c C s� t d d � j d d � } t | d k | � } t | j � | j � � t | j � d � t | j d � | j d � � t | j d � d d d d g � | j | j d � d j � | j | j d � d j � d
S( s Tests extrema on MaskedArrays.i i
i i i i i i i N( RC R: R� R R! R" R' R ( R R t amask( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_minmaxM s " c C s� t d d d d d g d d d d d d g �} t j | � } t d d d t j d � d g d d d d d d g �} t | | � t | j | j � | j t | j t � � d S( sE Check that the mask of the result is a ndarray (not a MaskedArray...)i����i i i i R N( R R R R R R' R� RN ( R R RG R� ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_ndarray_maskX s 0!
( R� R� R� R R� R� R� R� R� ( ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyR� s t! TestMaskedArrayInPlaceArithmeticsc 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 d � Z d � Z RS( s Test MaskedArray Arithmeticsc C sq t d � } t d � } t d � } t | d <| | | f | _ | j t � | j t � | j t � f | _ d S( Ni
i ( RC RK t intdataR0 RD t floatdata( R R R R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyR g s
c C s� | j \ } } } t | d <| d 7} t | | d � | d 7} t | | d � | j \ } } } | j j j } | d 7} t | | j j j k � t | | d � d S( s Test of inplace additionsi i g �?N( R� RK R R� RF R� R* Rn ( R R R R R: t id1( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_inplace_addition_scalaro s
c C s� | j \ } } } | j } t d d t j �} t | d <| | 7} | | 7} t | | | � t | | | � t | j t | | j � � d S( s Test of inplace additionsi
R) i����N( R� R RC R t int16RK R RE ( R R R R R� R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_inplace_addition_array~ s
c C sL | j \ } } } | d 8} t | | d � | d 8} t | | d � d S( s Test of inplace subtractionsi N( R� R ( R R R R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_inplace_subtraction_scalar� s
c C s� | j \ } } } | j } t d d t �} t | d <| | 8} | | 8} t | | | � t | | | � t | j t | | j � � d S( s Test of inplace subtractionsi
R) i����N( R� R RC RD RK R RE ( R R R R R� R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_inplace_subtraction_array� s
c C sL | j \ } } } | d 9} t | | d � | d 9} t | | d � d S( s Test of inplace multiplicationg @i N( R� R ( R R R R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt" test_inplace_multiplication_scalar� s
c C s� | j \ } } } | j } t d d t �} t | d <| | 9} | | 9} t | | | � t | | | � t | j t | | j � � d S( s Test of inplace multiplicationi
R) i����N( R� R RC RD RK R RE ( R R R R R� R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt! test_inplace_multiplication_array� s
c C sn | j \ } } } t d � d } t d � d } t | d <| d } t | | � | d } t | | � d S( s Test of inplace divisioni
i N( R� RC RK R ( R R R R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_inplace_division_scalar_int� s
c C sT | j \ } } } | d :} t | | d � | t d � :} t | t d � � d S( s Test of inplace divisiong @i
N( i
( R� R RC R= ( R R R R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt" test_inplace_division_scalar_float� s
c C s� | j \ } } } | j } t d d t �} t | d <| | :} | | :} t | | | � t | | | � t | j t t | | j � | d k � � d S( s Test of inplace divisioni
R) i����i N( R� R RC RD RK R RE ( R R R R R� R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt! test_inplace_division_array_float� s
c C s� d d d d t d d d d d d d d g } d d d d d
d d d d d d d g } d d
d
d
d
d
d d
d
d
d
d
g } d
d
d d
d
d d d
d
d
d
d g } t | d | �} t | d | �} | | } t | j d d d d
d
d d d
d
d
d d g � t | j d d d d
t d d d d d d d d g � | j � } | | :} t | j d d d d
d
d d d
d
d
d d g � t | j d d d d
t d d d d d d d d g � d S( Ng �?g �g @g @g @g $�g $@g @g g �g �i i 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_core.pyt test_inplace_division_misc� s .***
49
4c C s� t d d d g d d d d g �} | d } t | j d d d g � t | j d d d g � | d 7} t | j d d d g � t | j d d d g � t d d d g d d d d g �} | t d d d g d d d d g �} t | j d d d g � t | j d d d g � t d d d g d d d d g �} | t d d d g d d d d g �7} t | j d d d g � t | j d d d g � d S( s' Test keeping data w/ (inplace) additioni i i R i i N( R R RF R ( R R RL ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_datafriendly_add� s $
$($(c C s� t d d d g d d d d g �} | d } t | j d d d g � t | j d d d g � t d d d g d d d d g �} | d 8} t | j d d d g � t | j d d d g � t d d d g d d d d g �} | t d d d g d d d d g �} t | j d d d g � t | j d d d g � t d d d g d d d d g �} | t d d d g d d d d g �8} t | j d d d g � t | j d d d g � d S( s* Test keeping data w/ (inplace) subtractioni i i R i N( R R RF R ( R R RL ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_datafriendly_sub� s $
$
$($(c C s� t d d d g d d d d g �} | d } t | j d d d g � t | j d d d g � t d d d g d d d d g �} | d 9} t | j d d d g � t | j d d d g � t d d d g d d d d g �} | t d d d g d d d d g �} t | j d d
d g � t | j d d d g � t d d d g d d d d g �} | t d d d g d d d d g �9} t | j d d
d g � t | j d d d g � d S( s- Test keeping data w/ (inplace) multiplicationi i i R i i i
i i i( N( R R RF R ( R R RL ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_datafriendly_mul s $
$
$($(c C s� t d d d g d d d d g �} | d } t | j d d d d d g � t | j d d d g � t d d d g d d d d g �} | d :} t | j d d d d d g � t | j d d d g � t d d d g d d d d g �} | t d d
d g d d d d g �} t | j d d d
d g � t | j d d d g � t d d d g d d d d g �} | t d d
d g d d d d g �:} t | j d d d
d g � t | j d d d g � d S(
s' Test keeping data w/ (inplace) divisioni i i R i g @g �?g @g $@g 4@g >@N( R R RF R ( R R RL ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_datafriendly_div- s $
!$
!$( $( c C s� t d d d g d d d d g �} | d } t | j d d
d g � t | j d d d g � | d C} t | j d d d g � t | j d d d g � d S( s$ Test keeping data w/ (inplace) powerg �?g @g @R i i g @i Ng�;f��@g�;f��@( R R RF R ( R R RL ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_datafriendly_powE s $
c C s$ t d d g d d g g � } t d d g d d d g �} | | 7} t | d d g d d g g � | j t k r� t | j d d g d d g g � n t d d g d d g g � } t d d g d d d g �} | | 7} t | d d g d d g g � t | j d d g d d g g � d S( Ni i R i i i ( R R R RS ( R R R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_datafriendly_add_arraysR s
%
c C s$ t d d g d d g g � } t d d g d d d g �} | | 8} t | d d g d d g g � | j t k r� t | j d d g d d g g � n t d d g d d g g � } t d d g d d d g �} | | 8} t | d d g d d g g � t | j d d g d d g g � d S( Ni i R i i ( R R R RS ( R R R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_datafriendly_sub_arraysa s
%
c C s$ t d d g d d g g � } t d d g d d d g �} | | 9} t | d d g d d g g � | j t k r� t | j d d g d d g g � n t d d g d d g g � } t d d g d d d g �} | | 9} t | d d g d d g g � t | j d d g d d g g � d S( Ni i R i ( R R R RS ( R R R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_datafriendly_mul_arraysp s
%
( R� R� R� R R� R� 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_core.pyR� d s(
t TestMaskedArrayMethodsc 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 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( s2 Test class for miscellaneous MaskedArrays methods.c % C st t j d d d d d d d d d d
d d d
d d d d d d d d d d d d d d d d d d d d! d" d# d$ g$ � } | j d% d% � } | j d& d' d' d& � } t j d( d) d( d) d( d( d) d( d) d) d( d) d( d( d( d) d( d) d( d( d( d) d) d) d) d( d( d) d( d( d( d( d) d( d) d( g$ � } t d* | d+ | � } t d* | d+ | j | j � � } t d* | d+ | j | j � � } t j d) d) d( d) d( d( d) d) d) d) d( d) d( d( d) d) d( d) d( d( d( d) d) d) d) d( d( d) d) d( d( d( d) d( d) d) g$ � } t d* | d+ | � } t d* | d+ | j | j � � }
t d* | d+ | j | j � � } | | | | | | | | |
| f
| _ d, S(- s Base data definition.g � @g�G�z. @g-���!@g !@g�&1� �?g��ʡE�@g\��(� @g ��Q� @g{�G�#@gP��n�@gh��|?�!@g!�rh��@gj�t�@gJ +�@g�z�Ga@g!�rh��@g�ʡE��@g-���@g)\��(@g�(\�B#@g +�� @gu�V @g�~j��@g�"�~�@g��K7�� @g
ףp=J#@g�G�z�@g��Mb�@g}?5^��#@g��/�$@g!�rh�m @g-���#@gJ +�@g���Mb@g��ʡ�@g-�����?i i i i i RF R N( R R R: R R ( R R t Xt XXR� R� t mXt mXXR t m2xt m2Xt m2XX( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyR � s6 !!!!c C s� t d d d g � } t d d d g d d d d g �} t | j � | j j � � t | j � | j j � � t | j � | j j � � t | j � | j j � � t | j d d d d d � | j j d d d d d � � t | j d d d g � | j j d d d g � � t | j � | j j � � t | j
� | j j
� � t d d g d d g g � } t | j � | j j � � t | j � | j j � � t | j
d d g � | j j
d d g � � t | j � | j j � � d S( s Tests some MaskedArray methods.i i i R i i N( R R R. R* t allt argmaxt argmint chooset compresst conjR� t diagonalR= RJ RI ( R R R R� ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_generic_methods� s $ :4 .c C s t j j d � } | t j j d � d } | j t | | � � t j | d <| j t | | � � t j | d <| j t | | � � t | � } t | d <| j t | | d t �� | j t | | d t � � | d 9} d | d <| j t | d d t �� d S( s Tests allclose on arraysi
g:�0�yE>i i����t masked_equalN(
R RA RB R' t allcloset infR RK R$ R& ( R R R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt
test_allclose� s
c C s) t j d d d g d d d g d d d g g � } t j t t t g t t t g t t t g g d
t j �} t | d | �} t j t t t g t t t g t t t g g d
t j �} | d k } | d k } t | j � t k � t | j � t k � t | j d
� t t t g � t | j d � t t t g � t | j d
� t t t g � t | j d � t t t g � t | j � t k � t | j � t k � t | j d
� t t t g � t | j d � t t t g � t | j d
� t t t g � t | j d � t t t g � d S( s% Checks the any/all methods/functions.g�p=
ף�?g�p=
ף�?g�������?g�Q� ���?g ��Q� �?g)\��(�?gףp=
��?gףp=
��?gffffff�?R) R g �?i i N(
R R R$ R& t bool_R Rn R� R. R ( R R R� R� t xbigt mxbigt mxsmall( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_allany� s0 c C sP t j d d d g d d d g d d d g g � } t j | � } t j t t t g t t t g t t t g g d
t j �} t | d | �} | d k } | d k } t | j � t k � t | j � t k � t
| j d
� t j t t t g � � t
| j d � t j t t t g � j � t
| j d
� t j t t t g � � t
| j d � t j t t t g � j � t | j � t k � t | j � t k � t
| j d
� t j t t t g � � t
| j d � t j t t t g � j � t
| j d
� t j t t t g � � t
| j d � t j t t t g � j � d S( Ng�p=
ף�?g�p=
ף�?g�������?g�Q� ���?g ��Q� �?g)\��(�?gףp=
��?gףp=
��?gffffff�?R) R g �?i i ( R R R� R$ R& R� R Rn R� R. R R� ( R R R� R� R� t mXbigt mXsmall( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_allany_onmatrices� s, (+(+(+(c C s� t d d t �} t d d d g d t �} | j | j � t k � | j d | � | j | � | j | j t � | j | t k � t d d t �} | j | j � t k � | j d | � | j | � | j | j t � | j | t k � d S(
s Some fun with all and anyR) i i i R R+ N( ( ( Rd t boolR R$ R' R� RK R; R. ( R t storet full( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_allany_oddities s
c
C s | j \
} } } } } } } } } }
t | j � d � t | j � d � t | j � d � t | j � d � t | j � d � t | j � d � t | j � d � t | j � d � t | j d � d d d d d d g � t | j d � d d d d d d g � t | j d � d d d d d d g � t | j d � d d d d d d g � t | j d � d d d d d d g � t | j d � d d d d d d g � t | j d � d d d d d d g � t | j d � d d d d d d g � d
S( s& Tests argmin & argmax on MaskedArrays.i# i i i i i i i i N( 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_core.pyt test_argmax_argmin s" '(((((((c % C sl t j d d d d d d d d d d
d d d
d d d d d d d d d d d d d d d d d d d d! d" d# d$ g$ � } t j d% d& d% d& d% d% d& d% d& d& d% d& d% d% d% d& d% d& d% d% d% d& d& d& d& d% d% d& d% d% d% d% d& d% d& d% g$ � } t | d' | �} | j d( d) � } t | j | j � t | j | j d( d) � � t | j | j j d( d) � � d* S(+ s Tests clip on MaskedArrays.g � @g�G�z. @g-���!@g !@g�&1� �?g��ʡE�@g\��(� @g ��Q� @g{�G�#@gP��n�@gh��|?�!@g!�rh��@gj�t�@gJ +�@g�z�Ga@g!�rh��@g�ʡE��@g-���@g)\��(@g�(\�B#@g +�� @gu�V @g�~j��@g�"�~�@g��K7�� @g
ףp=J#@g�G�z�@g��Mb�@g}?5^��#@g��/�$@g!�rh�m @g-���#@gJ +�@g���Mb@g��ʡ�@g-�����?i i R i i N( R R t clipR R R* ( R R R� R� t clipped( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_clip/ s *$- c C s= t d d d d d g d d �} | d k | d k @} t | j | � d d g � t | d
d g <| j | � } t | j d d g � t | j d d
g � t | j d � t | | | � | d k } | j | � } t | j d d d g � t | j d d d
g � t | j d � t | | | � t d d d g d d d g g d d d d
g d
d d g g �} | j | j � d k � } t | j d d d d g � t | j d
d
d d g � t j d d
d
g � } | j | d
k d d
�} t | j d d g d d g g � t | j d d
g d
d g g � d S( s
test compressg �?g @g @g @g @RU i' g �?g @i i i i i
i i i( i2 i< R i R~ N(
R R R� RK R* R; RU R R R ( R R t conditionR R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt
test_compressA s. ! B "c C s t d d d d g d d d d d g �} | j � } t | | � t | d <| j � } t | d d d g � t t j d d d d g � d d d d d g �} | j � } t | | � | j t | t j � � t | d <| j � } t | d d d g g � d S( s Tests compressedi i i i R i N( i i ( R t
compressedR RK R R� R' R� ( R R R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_compressed` s *
3
c C s� d t f d t f d g } t d d d g d | �} t t | j j � � t | � � t | � } t | j | j � t | j | j � t t | � d | �} t | j | j � t | j | j � d S( s Tests empty/likeR R R� s |S8i g�������?s 1.1i g������@s 2.2i gffffff
@s 3.3R) N( R� s |S8( i g�������?s 1.1( i g������@s 2.2( i gffffff
@s 3.3(
R1 RD R R R3 RU Rp t
empty_likeR Rd ( R t datatypeR R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt
test_emptyr s " c
C sC t d � } d d d d d g } t | � } t | d | �} | j | d t k � | j | d t k � d d g | d d g <| j | d t k � | j | d t k � t | d d d d
d g � t t d � d d d d d d g d �} d d d d g } | j | d d d d g � t | t d d d d d d d d d
d g
� � t | j d d d d d d d d d d g
� | j | t d d d d g d d d d g � � t
| d d d d d d d d d
d g
� t | j d d d d d d d d d d g
� t t d � d d d d d d g d �} t | | d d d d g � t | t d d d d d d d d d
d g
� � t | j d d d d d d d d d d g
� t | | t d d d d g d d d d g � � t
| d d d d d d d d d
d g
� t | j d d d d d d d d d d g
� d S( s
Tests put.i i i R i i i
i( i i����i i i i N( RC R R R' RK R R t putRW R R4 ( R R R� R� R R� ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_put� s2 + 1.1+.+ 1.1+c C s� t d � } d d d d d g } t | � } t | d d | d t d t �} | j d d d d d g d d d d d g � t | j d d d d d g � d
S( s Tests put on hardmaski i i R R^ R} i i i N( RC R R R$ R� R R* ( R R R� R� Ra ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_put_hardmask� s ".c C s& t d � d } t | d d d d d d d g �} d d d d d d g } | j � } t | | d � t | d d d d d d g � | j � } t | | d � t | j d d d d d d g � t | j d d d d d d g � t d d
d d d
d g d d d d d d d g �} | j � } t | | | � t | j d d d d d d g � t | j d d d d d d g � | j � } t | | | � t | j d d d d d d g � t | j d d d d d d g � | j � } | j � t | | | � t | d d d d d d g � d S( Ni i R i ic i i i i
i i i( i2 i< ( RC R R} t putmaskR R* R; R` ( R R R� R RL t mxxt values( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_putmask� s. $ ""6 "" ""
c C s� t d d d d d g g d d d d d d g g �} | j � } t | j j | j � t d d g d d d g �} | j � } t | j j | j � t t j d d d d d g � d d d d d d g g �} | j � } t | j d � t | j j | j � t d d d d g d d d d d g d t �} t | j � j d d d d g � d | _ d | _ | j � } t | j d d d d g � t | j d d d d g � t | j d � d
S(
s Tests raveli i i i i R i RB i����N( i i ( i i (
R R R R; R R R� R& RU R* ( R R t aravelt ar( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt
test_ravel� s$ 6 < 0" c C sr t d � } t | d <| j d d � } t | j d � t | j j d � t | j d � t | j j d � d S( s
Tests reshapei i i N( i i ( i i ( i ( i ( RC RK R: R R R; ( R R R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_reshape� s
c C s� t d d d d g d d d d d g d t j �} t | � } t | j d d d d g � t | j d d d d g � t | d t �} t | j d d d d g � t | j d d d d g � | j � t | j d d d d g � t | j d d d d g � t d d d d g d d d d d g d t j �} | j d t � t | j d d d d g � t | j d d d d g � d d d d g } t | � } | j t t
t � � t d d d d
d g d t d t j
�} t | d t �} t | j d
d d d d g � t d d d d
d g d d d d d d g d t j
�} t | d t �} t | j d d d
d d g � t | j d d d d d g � d S( s Test sorti i i i R i R) Rj i����i����N( R R t uint8Rm R R* R; R& R' R� t sortedRN RS t int8( R R t sortedx( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_sort� s0 3
3 * 9 c C sj t d d d g d d d g g � } | j d � t | d d d g d d d g g � t d d d g d d d g g � } | j d � t | d d d g d d d g g � t d d d g d d d g g d d d d g d d d g g �} | j d � t | d d d g d d d g g � t | j d d d g d d d g g � t d d d g d d d g g d d d d g d d d g g �} | j d � t | d d d g d d d g g � t | j d d d g d d d g g � t d d d g d d d
g d d d g g d d d g d d d g d d d
g g d d d g d d d g d d d
g g d d d
g d d d g d d d g g g � } t | | d d k <| j � } | j d � } | j d � | j d � t | | � | j � } | j d � } | j d � | j d � t | | � | j � } | j d � } | j d � | j d � t | | � d
S( s Check sort of 2D array.i i i i i i R i i i i ic N( R Rm R R; RK R} R( ( R R t amt an( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_sort_2d sD $
%$
%B
%(B
%(*''0
c C s^ t d d d
d d d d g d d d d d d d d g d d t f d t f g �} t | � } t d d d d d d d g d d d! d" d# d$ d% d& g d d t f d t f g �} t | | � t | j | j � t | d t �} t d' d( d) d* d+ d, d- g d d. d/ d0 d1 d2 d3 d4 g d d t f d t f g �} t | | � t | j | j � d
S(5 s Test sort on flexible dtype.i i i i R R) R� R� Rj N( 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 ( 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 ( 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 R1 Rm R R R& ( R R RG R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_sort_flexible9 s
c C sP t d d d d d g d d d d d d g �} t t j | � t | � � d S( s Test argsorti i i i i R i N( R R R t argsort( R R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_argsortM s 0c C s� t d d d g g � } t | j � d d d g � t d d d g g d d d d g g �} t | j � d d d g � t | j � j d d d g � t d g g d t �} | j | j � t k � d S( s
Check squeezei i i R N( R R t squeezeR; R$ R' RK ( R RF ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_squeezeS s * c % C s� t j d d d d d d d d d d
d d d
d d d d d d d d d d d d d d d d d d d d! d" d# d$ g$ � } t j d% d& d% d& d% d% d& d% d& d& d% d& d% d% d% d& d% d& d% d% d% d& d& d& d& d% d% d& d% d% d% d% d& d% d& d% g$ � } t | d' | �j d( d( � } | j d) d* d* d) � } | j d% d& � } t | d+ | d, d, � d+ f � | j d% d* � } t | j d- � d, S(. s Tests swapaxes on MaskedArrays.g � @g�G�z. @g-���!@g !@g�&1� �?g��ʡE�@g\��(� @g ��Q� @g{�G�#@gP��n�@gh��|?�!@g!�rh��@gj�t�@gJ +�@g�z�Ga@g!�rh��@g�ʡE��@g-���@g)\��(@g�(\�B#@g +�� @gu�V @g�~j��@g�"�~�@g��K7�� @g
ףp=J#@g�G�z�@g��Mb�@g}?5^��#@g��/�$@g!�rh�m @g-���#@gJ +�@g���Mb@g��ʡ�@g-�����?i i R i i i i����N( i i i i ( R R R: t swapaxesR R ( R R R� R� R� t mXswappedt
mXXswapped( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt
test_swapaxes^ s$ !c C s� t d d d d g d d d d g � } t | j d d d g � t d d d g d d d g � � t | j d d d g � | d d d g � t | j d d g d d g g � t d d g d d g g d d g d d g g � � t d d d g d d d g g d
d d d g d d d g g �} t | j d d g d d �t d d g d d g g d
d d g d d g g �� t t | d d g d d �t d d g d d g g d
d d g d d g g �� d
S( s
Tests takei
i i i( i i i i2 i< R i R~ N( R R RJ R ( R R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_takev s ':,!4B7c C s� t j d � } t j d � d d d � f t j d � d d � d f } t | d | t | � k �} t | | d
d �} t d d d g d d d g d d d g d d d g d d d g g � } t | | � t | | � } t d d d g d d d g d d d g d d d g d d d g g � } t | d <| d <| d <t | | � t | j | j � t d d d �} t | | � } t | d <| d <t | | � t | j | j � d S( s Test take w/ masked indicesi( i i% i i i Ni R t modeR� 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 R RC R� R3 RJ R RK R ( R R t indicest mindicesRG t ctrl( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_take_masked_indices� s0 <
c C s+ t t j d � � } t | d d g <| j � } | j | d d k � | j | d d k � d | _ | j � } d d d d g d d d d
g d d d d
g g } t | d d d d d g � t | d d d d d
g � t | d d d d d
g � t | | � t t d d d g d d d g d d d g � d d t
f d t f d g �} t | d <t | j � d d t d � f d d t d � f d g � t d d g d d! d" g d d t
f d t
f g �} | j � } t | d d g d d g g � | d } | j � } t | d d g � d S(# s
Tests to listi i i����i i i i i i i i i i g�������?g������@gffffff
@t onet twot thrR) R R R� s |S8i����R N( i i ( R� s |S8( NNN( i i ( i i ( i i ( i i (
R R RC RK t tolistR' R� R R R@ R1 RD R ( R R t xlistR R RG ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_tolist� s8 3
c C s� t d d
g d d t f d t f g �} x< | D]4 } x+ | j � D] } t t | t j � � qD Wq1 Wd | j d <x< | D]4 } x+ | j � D] } t t | t j � � q� Wq} Wd S( s? Test mvoid.tolist: make sure we return a standard Python objecti i i i R) R R N( i i ( i i ( i i ( R R1 R Rn R� R t genericR ( R R t entryRp ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_tolist_specialcase� s *
c C s t d � } | j � } t | d | j � t | d | j � t | d d d d g <| j � } t | d | j � t | d | j � d t f d d t f g } t g t t
j d � d t
j j d � � D] \ } } } | | | f ^ q� d
| �} t | d d d d g <| j � } t | d | j � t | d | j � t
j
d � } t g t t
j d � t
j j d � t
j j d � � D] \ } } } | | | f ^ q�d
| �} t | d d d d g <| j � } t | d | j � t | d | j � d S( s Test the conversion to recordsi
R* R; i i i i����R� R5 s |S3R� t
ABCDEFGHIJKLMR) s int, (2,3)float, floatN( R5 s |S3( RC t toflexR R* R; RK R1 RD R R@ R RA RB R) RE ( R RF t recordR� R� R5 R� t ff( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_toflex� s4 7 7 c C s t d d d g � } t | j � � } t | | � t | j | j � t d d d g d d d d g �} t | j � � } t | | � t | j | j � t d
d d g d d d d g d d
t f d t f g �} t | j � � } t | | � t | j | j � d S( s7 Test the reconstruction of a masked_array from a recordi i i R i g �?g @g @R) R� R� N( i g �?( i g @( i g @( i i ( i i ( i i ( R t fromflexR R R R1 RD RF ( R R RG ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt
test_fromflex� s
$
!
c C s� t d d d d d g g d d d d d d g �} t d g d g d g d g d g g d d d d d d g �} t | j | � t | j � | � t t j | j d � | j d � � d S( s! Test a _arraymethod w/ n argumenti i i i i R i N( R R R� RI RN t cumsum( R t marrayR� ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_arraymethod
s 3'( R� R� R� R R� R� R� R� R� R� R� R� R� R� R� R� 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_core.pyR� � s<
" ) $ ! t TestMaskedArrayMathMethodsc 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 RS(
c % C st t j d d d d d d d d d d
d d d
d d d d d d d d d d d d d d d d d d d d! d" d# d$ g$ � } | j d% d% � } | j d& d' d' d& � } t j d( d) d( d) d( d( d) d( d) d) d( d) d( d( d( d) d( d) d( d( d( d) d) d) d) d( d( d) d( d( d( d( d) d( d) d( g$ � } t d* | d+ | � } t d* | d+ | j | j � � } t d* | d+ | j | j � � } t j d) d) d( d) d( d( d) d) d) d) d( d) d( d( d) d) d( d) d( d( d( d) d) d) d) d( d( d) d) d( d( d( d) d( d) d) g$ � } t d* | d+ | � } t d* | d+ | j | j � � }
t d* | d+ | j | j � � } | | | | | | | | |
| f
| _ d, S(- s Base data definition.g � @g�G�z. @g-���!@g !@g�&1� �?g��ʡE�@g\��(� @g ��Q� @g{�G�#@gP��n�@gh��|?�!@g!�rh��@gj�t�@gJ +�@g�z�Ga@g!�rh��@g�ʡE��@g-���@g)\��(@g�(\�B#@g +�� @gu�V @g�~j��@g�"�~�@g��K7�� @g
ףp=J#@g�G�z�@g��Mb�@g}?5^��#@g��/�$@g!�rh�m @g-���#@gJ +�@g���Mb@g��ʡ�@g-�����?i i i i i RF R N( R 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_core.pyR
s6 !!!!c
C s� | j \
} } } } } } } } } }
| j d � } t | j | j d � j d � � | j d � } t | j | j d � j d � � | j d � } t | j | j d � j d � � | j d � } t | j | j d � j d � � d S( s' Tests cumsum & cumprod on MaskedArrays.i i N( R R" R R* R( t cumprod( R R R� R� R� R� R� R� R� R� R� t mXcp( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_cumsumprod:
s '"""c C s& t t j j d d d � � j d d � } t | d d � d f <| d <| d <x� d D]� } t t | � } t | | � } t j d d
t �} | j d � | | d d d
| �} | j
| | k � t | | d d d
| � � t d d
t �} | d d d
| � } | j
| | k � qW Wd S( s Tests cumsum/cumprod w/ outputi i
i i i Ni����R" R& R) i���R~ R+ ( i����i����( s cumsums cumprod( i i ( i i (
R R RA R- R: RK R. Rd RD R2 R' R R1 ( R R R3 R4 RS RT R7 ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_cumsumprod_with_outputH
s *&
c
C s | j \
} } } } } } } } } }
| j \ } } t | j � | j � j � � t j | t j � } t j | t j � }
x: t | � D], } | d d � | f j � j � |
| <q� Wx. t | � D] } | | j � j � | | <q� Wt | j d � |
� t | j d � | � d S( s Tests ptp on MaskedArrays.Ni i ( R R R R9 R� R R< RD R� ( R R R� R� R� R� R� R� R� R� R� R� t rowst colst k( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_ptp^
s ' * c C s� t d d d g d d d d g d t j �} t | j � d � t d d d g d d d g g d t �} t | j d
d � d d d g � d
S( s Test sum on object dtypei i i R i R) i i i R~ i i N( R R Re R R= ( R R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_sum_objectm
s -*c C s� t d d d g d d d d g d t j �} t | j � d � t d d d g d d d g g d t �} t | j d
d � d d d g � d
S( s Test prod on object dtypei i i R i R) i i i R~ i
i Ni ( R R Re R RN ( R R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_prod_objectt
s -*c C sQ t d d d g d t j �} t | j � d � t | j � d d d g � d S( s Test mean/anom on object dtypei i i R) i����i N( R R Re R RQ t anom( R R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_meananom_object{
s c
C s� | j \
} } } } } } } } } }
| j � } t | j � | j � j � j � � t | j � | j � t | j | j � d d �� d S( s Tests trace on MaskedArrays.R~ i N( R R� R t traceR� R= t assert_almost_equalR ( R R R� R� R� R� R� R� R� R� R� t mXdiag( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt
test_trace�
s
' % c
C sC | j \
} } } } } } } } } }
t | j d d � | j � j � � t | j d d � | j � j � � t | j d d d d � | j � j d d � � t | j d d d d � | j � j d d � � t | j d d � j | j d d � j � t | j � j | j � j � | j d d � | j d d � } } t | j d d d d � | j � j d d � � t | j d d d d � | j � j d d � � x� t d � D]� }
t | |
| |
j � j � � t | |
| d d � |
f j � j � � t t j
| |
� | d d � |
f j � j � � q�Wd S( s Tests var & std on MaskedArrays.R~ t ddofi i i i i N( R R3 RO R� R� RP R R R� R R ( R R R� R� R� R� R� R� R� R� R� t mXvar0t mXvar1R, ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_varstd�
s '%%+ %11!-c C s t j d d t �} t d d t �} t t d � d t �} x� d
D]� } t | | � } | j | � t k � | j | d � t k � | j | d � t k � | d | � } | j | t k � t | j
t � | d | � } | j t j | � � qF Wt t d � d t �} d
| d <x� d D]� } t | | � } | j | d d � t k � | j | d d d �t k � | j | d d d �t k � | d | d d � } | j | t k � t | j
t � | d | d d � } | j t j | � � q'Wd S( s Test a special case for vari R) i
R RO RP i i����R+ i R6 N( s vars std( s vars std( R Rd RD R RC R$ R. R' RK R R R� ( R R6 t moutR t
methodnamet methodR: ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_varstd_specialcases�
s2
c C s6 t d d d g d d d g g d d d d g d d d g g �} | j d d d d � } t | j d � d d d g � t | j d d d g � | j d d d d � } t | j d � d d d g � t | j d d d g � | j d d d d � } t | j d � d d d g � t | j d d d g � d S( Ni i R R~ R6 i ( R RP R R( R ( R R RG ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_varstd_ddof�
s B c C s� t d � j d � } t | d
<t j | � } t | d d d g � t | � } t | d d d g � t | j d d d g � t | � } t d d d g d d d g d d d g g d d d d g d d d g d d d g g �} t | | � d S( s Test diagi i i i i i R N( i i ( i i ( RC R: RK R t diagR R R ( R R R+ R� ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_diag�
s
--c C s� t d d d g d d d g g � } t | j d � d d d g � t | j d
� d d g � t | j d � d d g � t | j d � d d d
g � t | j d
� d d g � t | j d � d d g � t | j d � d d d g � t | j d
� d d g � t | j d � d d g � t | j d � d d d g � t | j d
� d d g � t | j d � d d g � d S( s- Test the combination nomask & methods w/ axisi i i i i i i i i i����i i
i ix N( R R R= RN R" R! ( R R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_axis_methods_nomask�
s $ ( R� R� R R( R) R- R. R/ R1 R5 R9 R= R> R@ RA ( ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyR%
s ! #
t! TestMaskedArrayMathMethodsComplexc B s e Z d Z d � Z d � Z RS( s2 Test class for miscellaneous MaskedArrays methods.c % C st t j d d d d d d d d d d
d d d
d d d d d d d d d d d d d d d d d d d d! d" d# d$ g$ � } | j d% d% � } | j d& d' d' d& � } t j d( d) d( d) d( d( d) d( d) d) d( d) d( d( d( d) d( d) d( d( d( d) d) d) d) d( d( d) d( d( d( d( d) d( d) d( g$ � } t d* | d+ | � } t d* | d+ | j | j � � } t d* | d+ | j | j � � } t j d) d) d( d) d( d( d) d) d) d) d( d) d( d( d) d) d( d) d( d( d( d) d) d) d) d( d( d) d) d( d( d( d) d( d) d) g$ � } t d* | d+ | � } t d* | d+ | j | j � � }
t d* | d+ | j | j � � } | | | | | | | | |
| f
| _ d, S(- s Base data definition.y � @y �G�z. @y -���!@y !@y �&1� �?g��ʡE�@g\��(� @g ��Q� @g{�G�#@gP��n�@gh��|?�!@g!�rh��@gj�t�@gJ +�@g�z�Ga@g!�rh��@g�ʡE��@g-���@g)\��(@g�(\�B#@g +�� @gu�V @g�~j��@y �"�~�@y ��K7�� @g
ףp=J#@g�G�z�@g��Mb�@g}?5^��#@g��/�$@g!�rh�m @g-���#@gJ +�@g���Mb@g��ʡ�@y -�����?i i i i i RF R N( R 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_core.pyR �
s6 !!!!c
C s� | j \
} } } } } } } } } }
t | j d d � | j � j � � t | j d d � | j � j � � t | j d d � j | j d d � j � t | j � j | j � j � | j d d � | j d d � } } t | j d d d d � | j � j d d � � t | j d d d d � | j � j d d � � x� t d � D]� }
t | |
| |
j � j � � t | |
| d d � |
f j � j � � t t j
| |
� | d d � |
f j � j � � qOWd S( s Tests var & std on MaskedArrays.R~ i i i R6 i i N( R R3 RO R� R� RP R R R� R R ( R R R� R� R� R� R� R� R� R� R� R7 R8 R, ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyR9 s '%%+ %11!-( R� R� R� R R9 ( ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyRB �
s !t TestMaskedArrayFunctionsc 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 d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z RS( s' Test class for miscellaneous functions.c
C sV t j d d d d t d d d d d d d d g � } t j d d d d d
d d d d d d d g � } d } d d
d
d
d
d
d d
d
d
d
d
g } d
d
d d
d
d d d
d
d
d
d g } t | d | �} t | d | �} t j d d d d g � } t | d d
d d
d
g �} t j | d | � }
| j d � | | f | _ d S( Ng �?g �g @g @g @g $�g $@g @g g �g �i i R g �g �?g�������?g@��x �D( R R 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_core.pyR , s 73**
c C sC d d g } t t | � } t | d d g � t | d d � d S( Ni i ( R� R& R ( R R R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_masked_where_bool: s c C s} d d d g } t | d � } t | | � t | j d d d g � t | d � } t | | � t | j d d d g � d S( Ni i i i ( R� R R; t masked_not_equal( R R R� ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_masked_equal_wlist@ s
c C sK d d d g } t | d � } t | j d d d g � t | j d � d S( Ni i i i ( R� R R; RU ( R R R� ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_masked_equal_fill_valueI s c C s� t d d d d d g � } t | d <t t t | d � | � t | d � � t t t | d � | � t | d � � t t t | d � | � t | d � � t t t
| d � | � t | d � � t t t | d � | � t
| d � � t t t | d � | � t | d � � t t t | d � | � t
| d � � t t d d d d d g d d d d
d g � d d d d
d g � d
S( s Tests masking functions.g �?g @g @g @g @i i i i i i ic N( R RK R R� R t masked_greaterR t masked_greater_equalR t masked_lessR t masked_less_equalR RE R R� ( R R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_masked_where_conditionO s
(((((((c C sD t d d t �} t | j t � } t | | � } t | | � d S( s Tests some generic features.i
R) N( i
i
i
( R= RD R< R Rs R� R ( R R� R� t ctest( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_masked_where_oddities] s c C s� t d � } y t d | � } Wn t k
r2 n
Xt d � � t | d � } t | j d d d d d d d d d d g
� d S( Ni
i s Should have failed...i ( RC R� t
IndexErrort AssertionErrorR R ( R R RG ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt" test_masked_where_shape_constrainte s
c
C s� t t t d � d d � d d d d d g � t t t d � d d � d d d d d g � t t t t d � d d d d d d g �d d � j d d d d d g � t t t t d � d d d d d d g �d d � j d d d d d g � t t t t d � d d d d d d g �d � j d d d d d g � t t t d d d d d g d d d d d d g �d � j d d d d d g � d S( Ni i i i i� i i R ( R t
masked_insideR� t masked_outsideR R R� RE ( R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_masked_otherfunctionsq s ..LLIc C s� t d d d d d g d d d d d d g �} t | j � d d
d d d
g � t | j d � d d d d d g � t | j d � d d d d d g � t | � } | j d | � t | d d
d d d
g � t d d
d d d g � } t d d d d d g � } t | d <t | | | � } t | d d
d d d g � t | d <t | | | � } t | d d
d d d g � t | d t k � t | d t k � t | d t k � d S( Ng82�����?g�� ���@g,��E|� @g����E@g�ŏ1�@R i i g �?g @g @g @g @g333333�?gffffff@g @gffffff@g������@i g�(\���?g�S㥛�@g-��� @g��ʡE@g���K�@R+ g @i g g �i����( R R R, R� RK R Rn ( R R R R R� R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt
test_roundz s& "%%
c C s t t j j d d d � � j d d � } t | d d � d f <| d <| d
<t j d d t �} | j d � t j | d
d d | �} | j
| | k � t | | j d
d d | � � t d d t �} | j d
d d | � } | j
| | k � d S( s% Testing round with an explicit outputi i
i i i Ni����R) i���t decimalsi R+ ( i����i����( i i ( i i ( R R RA R- R: RK Rd RD R2 R, R' R ( R R RT R7 ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_round_with_output� s *&
c C s< t d � } | j t | t � � t | t j d � � d S( Ni ( R� R' R� RN R R ( R R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt
test_identity� s c C s� d } t t | d � d � | j t | t � t k � t d d d d d g � } t d d d d d g d d d d d d
g �} t | | � } t | d d d d d g � t | j d
d d d d
g � t | _ t | | � } t | j d
d d d d
g � | | } t | j | j � t | | � t | j | j � | | C} t | j | j � t | | � t | j | j � d S( Ng�������g @g\��(\�?g�������?g g �?g �R i i gKTj����?(
R3 t powerR' RK R R R; RS R R* ( R R R R R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt
test_power� s& 0
c C s t j d d d g d d d g g � } t | d d d d g d d d g g �} t j d
d d g � } t | d d d d g �} t j | | g � } t | d d d d g d d d g g �} t d d d g d d d g g d d d d g d d d g g �} | | } t | | � t | j | j � | | } t | | � t | j | j � | | } t | | � t | j | j � t d d d g d d d g g d d d d g d d d g g �} | | } t | | � t | j | j � | | } t | | � t | j | j � d S( s Test power w/ broadcastingg �?g @g @g @g @g @R i i i i i i i Ni i i i iq i� i i i i i i ( R R R R ( R t a2t a2mt b1t b1mt b2t b2mR RG ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_power_w_broadcasting� s2 '**!!
!!
c C s t j d d d d t d d d d d d d d g � } t j d d d d d
d d d d d d d g � } d } d d
d
d
d
d
d d
d
d
d
d
g } d
d
d d
d
d d d
d
d
d
d g } t | d | �} t | d | �} t j d d d d g � } t | d d
d d
d
g �} t j | d | � }
| j d � t | d k | d � } t | d d d d d d d d d d d d g � t | j | j � t | d k d | � } t | d d d d d
d d d d d d d g � t | j d d
d d
d
d
d d
d
d
d
d
g � t | d k | t � } t | d d d d d d d d d d d d g � | j j � } t
| | d k j t
� <t | j | � | j t
� }
t |
d k |
t � } t | d d d d d d d d d d d d g � t | j |
j � d S( s Test the where functiong �?g �g @g @g @g $�g $@g @g g �g �i i R g �g �?g�������?g@��x �Di i����g "�i i
i N( R R R
R R R
R R; RK R} R$ R( R0 R1 R) ( R R R R R R R R R R R R t tmpt ixm( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt
test_where� s4 73**
11411c C sK t d � } t | d <| d k } t | | t � } t | j | j k � t | d t k � t | d t k � t | d t k � t | d t k � t | d t k � t | | � t | t | � } t | j | j k � t | d t k � t | d t k � t | d t k � t | d t k � t | d t k � d S( Ni
i i i i i ( RC RK R Rn R) R ( R R R� R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_where_with_masked_choice s$
c C s t d d d d d g � } t d d d d d g � } t | d <t | | | � } t | d d d d
d g � t | d <t | | | � } t | d d d d
d g � t | d t k � t | d t k � t | d t k � t d d � } t | d
<t d d � d } t | d <t d d d d d g d d d d d d g �} | j d � } t | | | � } t | | | � } t | | � t t | � t k � t | d d d d d g � t | t d � } t | d d d d d g � t | d t � } t | d d d d d g � d S( Ng �?g @g @g @g @i i i g g �i����i i����i
R i i( i2 ic ( R RK R R Rn RC R( Rq RS ( R R R� R R t cmR ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_where_with_masked_condition s4
0
c C s~ t j d d t j �} t j d d t j �d } t | d k | | � j } t j t j t j g g � } t | | � d S( s% Test the type conservation with wherei R) g������@g �?N( R RC t int32t float32R R) t find_common_typeR ( R R R RG R� ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_where_type4 s
c
C s* d d d d g d d d d g d d
d d g d
d d d g g } t d d d d g | � } t | t d d d d g � � t d d d d g | d d �} t | t d d d d g � � t d d d d g | d d �} t | t d d d d g � � t d d d d g d d d d d g �} t | | d d �} t | t d d d d g � � t | j d d d d g � t | d d d d d g d d d d g d d d d g d d d d g g �} d d d d g } t | | d d �} t | t d d d d g � � t | j d d d d g � d S( s Test choosei i i i i
i i i
i i i i i i i i! i R
R� t wrapR ic N( R� R R R ( R t choicest chosent indices_( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_choose= s$ $ ! ! * '' c
C s� d d d d g d d d d g d d
d d g d
d d d g g } t d d t �} t d d d d g | d | �} t | t d d d d g � � | j | | k � t d d t �} t d d d d g d d d d d g �} t | | d d d | �} t | t d d d d g � � t | j d d d d g � t | d d d d d g d d d d g d d d d g d d d d g g �} d d d d g } t d d t �j t � } t | | d d d | �} t | t d d d d g � � d S( s( Test choose with an explicit out keywordi i i i i
i i i
i i i i i i i i! i R) R+ R R
Rl ic i?B N( Rd R1 R� R R R' R t viewR ( R Rm R� Rn Ro ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_choose_with_outU s" $! * ''c C s[ t d � } t | d <| j d � } t | j d � | j | j d � | j d d � } t | j d
� | j | j d � | j d d d �} t | j d � | j | j d � | j d d d d �} t | j d
� | j | j d � t j | d � } | j t | t � � t | j d � | j | d t k � | j | j d � d S( Ni
i i i t Ct ordert F( i i ( i i ( i i ( i i ( i i ( i i ( i i ( i i ( i i (
RC RK R: R R R' t flagsR R� RN ( R R R R� ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyR� l s&
c C s� d t j f d t j f g } t | � } t | d t j f d t j f g � t j d f } t | � } t | t j d f � t j } t | � } t | t j t j � � d t j f d d t j f d t j f g f g } t | � } t j d d d
d g f g � } t | | � d t j d f f g } t | � } t | t j d t j d f f g � � d t f g } t | � } t | t j d
t f g � � d S( s Test make_mask_descrR R i R� R� R] R� N( R s b1( s bas b1( s bbs b1( R� R ( R� R ( R RD t make_mask_descrR R� R) ( R t ntypeRG R� ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_make_mask_descr� s& % 3 !
( c C s� d d g } t | � } t | j t � t | d d g � t j d d g d t j �} t | � } t | j t � t | d d g � d t j f d t j f g } t j d d g d | �} t | � } t | j t � t | d d g � d t j f d t j f g } t j d d
g d | �} t | d | j �} t | j | � t | | � d t j f d t j f g } d t j f d t j f g } t j d d g d | �} t | d | j �} t | j | � t | t j d
d g d | �� d S( s Test make_maski i R) R R N( i i ( i i ( i i ( i i ( i i ( i i ( i i ( i i ( R R R) Rs R R R� RD ( R R RG t mdtypet bdtype( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_make_mask� s0
c C s d t j f d t j f g } t j d
d d d
g d | �} t | t � } t | | � t t | � } t | | � t | t � } t | | � t j d d d d g d | �} t | | � } t j d d d d g d | �} t | | � d t j f d t j f g } t j d d d d g d | �} y t | | � } Wn t k
rVn Xd t j f d d t j f d t j f g f g } t j d d f d d f g d | �} t j d d f d d f g d | �} t j d d f d d f g d | �}
t t | | � |
� d S( NR R i i R) R� R� R� R� ( 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 ( i i ( i i ( i i ( i i ( i i ( i i ( i i ( R R� R RE RS R R& t
ValueError( R R| R RG t otherR� t othertypeR) R� t bmaskt cntrl( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_mask_or� s. !
!!
!
3'''c C s6 t j d d d g d t j �} t t | � | � t j d d
g d d t f d t f g �} t | � } t j d d d d g d t �} t | | � d t f d d t f d t f g f g } d d f d d f g } t j | d | �} t | � } t j d d d d d d g d t �} t | | � d S(
s Tests flatten maski i R) R R R� R� N( i i ( i i ( i i ( i i ( R R R� R t flatten_mask( R R RG R� Rz RF ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_flatten_mask� s !- !
* 'c C sx t j d d d d g � } t | d t �} t | � } t | | j � � t | d � } t | | j d d � � d S( s Test functions on ndarraysi i i i R N( i i ( R R R& R0 R R: ( R R R� RG ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_on_ndarray� s ( R� R� R� R RD RF RG RL RN RQ RT RU RW RX RZ Ra Rd Re Rg Rk Rp Rr R� Ry R| R� R� R� ( ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyRC ) s4 " ! $ t TestMaskedFieldsc B sP e Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z RS( c C s� d d d d d g } d d d d d
g } d d d
d d g } d t f d t f d g } d t f d t f d t f g } d d d d d g } t t | | | � d | d | �} t d | d | d | d | � | _ d S( Ni i i i i g�������?g������@gffffff
@g������@g @R R t threet fourt fiveR R R� s |S8i R R) t baset ddtypeRz ( R� s |S8( R1 RD R� R R@ t dictRF ( R t ilistt flistt slistR� Rz R R� ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyR
s !$c C s, | j d } | j d } t | _ t | j t j | j d | �� t | _ t | j t j | j d | �� t
| _ t | j t j | j d | �� t | _ t | j t j | j d | �� d d d d d g | _ t | j t j g d d d d d g D] } | | | f ^ qd | �� d S( NR� Rz R) i i (
RF RS R RE R; R R< R RK R= R& R$ R ( R R� Rz R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_set_records_masks
s
" " " " 7c C s� | j d } | d | d | d } } } t t d f | d <t | j t � t | j d d d d d
g � t | j t � t | j t d d d
d g � t | j d � t | j t d d d d d g � � d S( s% Check setting an element of a record)R� R R R� R
i i i i i g������@gffffff
@g������@g @s |S8R R� R� R� N( RF R
R R) R1 R* RD R ( R R� t base_at base_bt base_c( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_set_record_element&
s
c C s� | j d } | d | d | d } } } t t d f | d *t | j t � t | j d d d d d g � t | j t � t | j t t t d d
g � t | j d � t | j t d d d d d
g � � d S( NR� R R R� R
i i i g������@g @s |S8R� R� ( RF R
R R) R1 R* RD R ( R R� R� R� R� ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_set_record_slice6
s
c C s� | j d } | d | d | d } } } t | d <xL d D]D } t | | j d d d d d g � t | | j | j | � q> Wd S( s Check record accessR� R R R� i i N( R R R� ( RF RK R R R* ( R R� R� R� R� R� ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_mask_elementE
s
#c C s� d t f d t f g } t d d | �} t t | � t j d d
d g d d d
g �� t | (t t | � t j d d d g d d d g �� d S( s# Test getmaskarray on flexible dtypeR R i R) i s |b1i N( i i ( i i ( i i ( R s |b1( R s |b1( i i ( i i ( i i ( R s |b1( R s |b1( R1 RD Rd R R� R R RK ( R R� RG ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_getmaskarrayO
s c C s< t t j d � t j j d � � } t j | � } t | d d t f d t f g �} d | j d <t j d g d d g d t �} | j t � } t
| | j � � t
| j | � | j t d f � } t
| | � t
| j | j d d � � | j t d f t j
� } t
| | � | j t | t j
� � d
S( s Test view w/ flexible dtypei
R) R R i i i i i����N( i i ( R@ R RC RA RB R RD R R� Rq R R R: R� R' R� ( R t iteratorRF R t controlmaskRG ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_view[
s $$
#
c
C sx d t f d t f g } t t t j j d � t j d � � d | �} t j t d d d d d d d d d d g
d d d d d d d d d d g
� d d t f d t f g �| _ | j t
| d t j � � | j t
| d t � � t
| d j | j d � t
| d j | j d � | j t
| d t � � t
| d j | j d � t
| d j | j d � d S( NR R i
R) i i i����( RD R R@ R RA RB RC R� R R' R� R� RN RE R* R; ( R R� R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_getitemo
s 0*' (
R� R� R R� R� R� R� R� R� R� ( ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyR�
s
t TestMaskedViewc B sG e Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z RS( c C s� t t j d � t j j d � � } t j | � } t | d d t f d t f g �} d | j d <t j d g d d g d t �} | | | f | _ d S( Ni
R) R R i i i ( i i (
R@ R RC RA RB R RD R R� RF ( R R� RF R R� ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyR �
s $$
#c C s^ | j \ } } } | j � } | j t | t � � t | j | j � t | j | j � d S( N( RF Rq R' R� RN R R* R; ( R RF R R� RG ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_view_to_nothing�
s
c C sn | j \ } } } | j t j � } | j t | t � � t | | j � t | | j | j
� j � � d S( N( RF Rq R R R' R� RN R R* RE R) R ( R RF R R� RG ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_view_to_type�
s
c C s^ | j \ } } } | j t � } | j t | t � � t | | j � � t | j | � d S( N( RF Rq RD R' R� RN R R R ( R RF R R� RG ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_view_to_simple_dtype�
s
c C s{ | j \ } } } | j d t f d t f g � } t | j j j d � t | d | d � t | d | d � | d j d t f d t f g � } | j t | t � � t | j j j d � t | d | d d � t | d | d d � | d j d t f d t f g � } | j t | t � � t | j j d � t | d | d d � t | d | d d � d S(
NR� R� R R i i����( R� R� ( R� R� ( R� R� (
RF Rq RD R R R) t namesR' R� RN ( R RF R R� RG ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_view_to_flexible_dtype�
s !%%c C s� | j \ } } } | j t d f � } | j t | t � � t | | � t | j | j d d � � | d j t d f � } | j t | t � � t | | d � t | j d � | d j t d f � } | j t | t � � t | | d � d S( Ni i����i i ( i i ( RF Rq RD R' R� RN R R R: ( R RF R R� RG ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_view_to_subdtype�
s
c C sn | j \ } } } | j t d f t j � } t | | � | j t | t j � � | j t | t � � d S( Ni ( RF Rq RD R R� R R' R� RN ( R RF R R� RG ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_view_to_dtype_and_type�
s
( R� R� R R� R� R� R� R� R� ( ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyR� �
s c C sV t j j d d d d g d d d d d g �} t t j | � d g d g g � d S( Ni i i i R ( R R� R R t argwhere( R ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt test_masked_array�
s 0t __main__(& R� t
__author__R R� R/ R t numpy.core.fromnumericR0 R� R t numpy.ma.testutilst
numpy.ma.coret numpy.compatR R t numpy.testing.utilsR R
R R t functoolsR t TestCaseR R� R[ Rl R� R� R� R% RB RC R� R� R� R� t run_module_suite( ( ( s> /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_core.pyt <module> sL
� � {� � %��M� � � ��5� �{O