| Current File : //usr/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyc |
�
E�`Qc @ s� d Z d Z d Z d Z d Z d d l Z d d l Td d l Td d l Td e j
f d � � YZ e Z d
e e
f d � � YZ e Z d e
e j f d
� � YZ e Z d e f d � � YZ e d k r� e � n d S( s� Tests suite for MaskedArray & subclassing.
:author: Pierre Gerard-Marchant
:contact: pierregm_at_uga_dot_edu
:version: $Id: test_subclassing.py 3473 2007-10-29 15:18:13Z jarrod.millman $
s5 Pierre GF Gerard-Marchant ($Author: jarrod.millman $)s 1.0s $Revision: 3473 $s5 $Date: 2007-10-29 17:18:13 +0200 (Mon, 29 Oct 2007) $i����N( t *t SubArrayc B s, e Z d Z i d � Z d � Z d � Z RS( s_ Defines a generic np.ndarray subclass, that stores some metadata
in the dictionary `info`.c C s% t j | � j | � } | | _ | S( N( t npt
asanyarrayt viewt info( t clst arrR t x( ( sE /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyt __new__ s c C s t | d i � | _ d S( NR ( t getattrR ( t selft obj( ( sE /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyt __array_finalize__ s c C sC t j j | | � } | j j i | j j d d � d d 6� | S( Nt addedi i ( R t ndarrayt __add__R t updatet pop( R t othert result( ( sE /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyR s *( t __name__t
__module__t __doc__R R
R ( ( ( sE /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyR s t MSubArrayc B s8 e Z i e d � Z d � Z d � Z e d e � Z RS( c C s: t | | � } t j | d | d | �} | j | _ | S( Nt datat mask( R t MaskedArrayR R ( R R R R t subarrt _data( ( sE /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyR $ s c C s$ t j | | � t j | | � d S( N( R R
R ( R R ( ( sE /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyR
) s c C s | j t � } t | _ | S( N( R R t Falset _sharedmask( R t _view( ( sE /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyt _get_series- s t fget( R R t nomaskR R
R! t propertyt _series( ( ( sE /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyR # s t MMatrixc B s5 e Z e d � Z d � Z d � Z e d e � Z RS( c C s. t j | � } t j | d | d | �} | S( NR R ( R t matrixR R ( R R R t matR ( ( sE /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyR 6 s c C s' t j j | | � t j | | � d S( N( R R' R
R ( R R ( ( sE /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyR
: s c C s | j t � } t | _ | S( N( R R R R ( R R ( ( sE /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyR! > s R" ( R R R# R R
R! R$ R% ( ( ( sE /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyR& 5 s t TestSubclassingc B sV e Z d Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z RS( s, Test suite for masked subclasses of ndarray.c C sC t j d � } t | d d d d d d g �} | | f | _ d S( Ni R i i ( R t aranget mmatrixR ( R R t mx( ( sE /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyt setUpI s !c C s� t j d � } d d d d d g } t | � } t | d | �} | j t | t � � t | j | � | j t | j t � � d S( s# Tests whether the subclass is kept.i i i R N( R R* R t masked_arrayt
assertTruet
isinstanceR t assert_equalR ( R R t mt xsubt xmsub( ( sE /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyt test_data_subclassingN s c C s/ | j \ } } | j t | j t j � � d S( s Tests subclassing MaskedArrayN( R R/ R0 R R R' ( R R R, ( ( sE /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyt test_maskedarray_subclassingX s c C sr | j \ } } t j d d � } z< | j t t | � t � � t t | � t j | � � Wd t j | � Xd S( s Tests masked_unary_operationt dividet ignoreN( R R t seterrR/ R0 t logR+ R1 ( R R R, t olderr( ( sE /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyt test_masked_unary_operations] s c C s� | j \ } } | j t t | | � t � � | j t t | | � t � � t t | | � | | � | j t t | | � j t j � � | j t t j | | � t � � | j t t
| | � t � � | j t t
| | � t � � d S( s Tests masked_binary_operationN( R R/ R0 t addR+ R1 R R R' t outert hypot( R R R, ( ( sE /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyt test_masked_binary_operationsg s %" c C s� | j \ } } t | j j � d | j �} | j t t | | � t � � | j t t | | � t � � t t | | � t | | � � d S( s& Tests domained_masked_binary_operationR N( R R. t __array__R R/ R0 R7 R+ R1 ( R R R, t xmx( ( sE /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyR@ t s
c C s� t t d � d d g d g d �} t t | � � } t | � } | d } | j t | t � � | j t | t � � | j t | j t
� � t | j j i � | d } | j t | t � � | j t | t � � | j t | j t
� � | j | j j d d k � | j
d d d d d g � t | j d d d d d g � | j j
d d d d d g � t | j d d d d d g � t | d i d d 6�} t | � } | j t | d � � t | j | j � d S(
Ni R i i i R R R t name( t arrayR* R. t subarrayt msubarrayR/ R0 R R R R R1 R t _set_maskt _maskR% t hasattr( R R t myt ymt zR3 t mxsub( ( sE /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyt test_attributepropagation| s* &
c C s� t j d � } d d d d d g } g t | | � D] \ } } | | f ^ q4 } t | d | d i | d 6�} t | d t �} | j t | t � � | j t | t � � t | j
| � t | � } | j t | t � � | j t | t � � t | j
| � t | d t �} | j t | t � � t | j
| j
� t | j
| j
� t | � } | j t | t � � t | j
| j
� t | j
| � d S( s= Checks that masked_array(...,subok=True) preserves the class.i i i R R R3 t subokN( R R* t zipR R. R R/ R0 R R1 RH t asarrayt TrueR R ( R R R2 t it jt xinfoR3 RM ( ( sE /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyt test_subclasspreservation� s( . (
R R R R- R5 R6 R< R@ RN RV ( ( ( sE /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyR) F s
t __main__( R t
__author__t __version__t __revision__t __date__t numpyR t
numpy.testingt numpy.ma.testutilst
numpy.ma.coreR R RE R R RF R' R& R+ t TestCaseR) R t run_module_suite( ( ( sE /usr/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyt <module> s"
n