| Current File : //usr/lib64/python2.7/site-packages/numpy/ma/testutils.pyo |
�
E�`Qc @ s� d Z d Z d Z d Z d Z d d l Z d d l Z d d l m Z m Z d d l
j j Z d d l
Td d l j j Z d d l m Z m Z m Z m Z m Z m Z m Z m Z e d
d d � Z d
e d � Z d d � Z d � Z d d � Z d d � Z e Z d d e d � Z! e! Z" d e d e d � Z# d e d � Z$ d e d � Z% d
d e d � Z& d
d e d � Z' d e d � Z( d d � Z) d S( s� Miscellaneous functions for testing masked arrays and subclasses
:author: Pierre Gerard-Marchant
:contact: pierregm_at_uga_dot_edu
:version: $Id: testutils.py 3529 2007-11-13 08:01:14Z jarrod.millman $
s5 Pierre GF Gerard-Marchant ($Author: jarrod.millman $)s 1.0s $Revision: 3529 $s5 $Date: 2007-11-13 10:01:14 +0200 (Tue, 13 Nov 2007) $i����N( t ndarrayt float_( t *( t mask_ort getmaskt masked_arrayt nomaskt maskedt filledt equalt lessg�h㈵��>g:�0�yE>c C s� t t | � t | � � } t | � } t | � } | j j d k sW | j j d k rm t j | | � j � St t | d t d | �| � j
t � } t t | d t d | �d � j
t � } t j t
j | | � | | t
j | � � }
|
j � S( ss Returns true if all components of a and b are equal subject to given tolerances.
If fill_value is True, masked values considered equal. Otherwise, masked values
are considered unequal.
The relative error rtol should be positive and << 1.0
The absolute error atol comes into play for those elements of b that are very
small or zero; it says how small a must be also.
t Ot copyt maski ( R R R t dtypet chart npR t ravelR t Falset astypeR t
less_equalt umatht absolute( t at bt
fill_valuet rtolt atolt mt d1t d2t xt yt d( ( s8 /usr/lib64/python2.7/site-packages/numpy/ma/testutils.pyt approx s $**0i c
C s� t t | � t | � � } t | � } t | � } | j j d k sW | j j d k rm t j | | � j � St t | d t d | �| � j
t � } t t | d t d | �d � j
t � } t j t j
| | � | � d | k } | j � S( s� Returns True if a and b are equal up to decimal places.
If fill_value is True, masked values considered equal. Otherwise, masked values
are considered unequal.
R R R
i g $@( R R R R R R R R R R R R t aroundt abs(
R R t decimalR R R R R R R! ( ( s8 /usr/lib64/python2.7/site-packages/numpy/ma/testutils.pyt almost- s $***t c C s_ t t | � t | � | � x< t t | � � D]( } t | | | | d | | f � q/ Wd S( s0 Asserts the equality of two non-array sequences.s
item=%r
%sN( t assert_equalt lent range( t actualt desiredt err_msgt k( ( s8 /usr/lib64/python2.7/site-packages/numpy/ma/testutils.pyt _assert_equal_on_sequences> s &c C s� t | j | j � xy | j j D]k } t j | | � t j | | � } } | t k r | t k r t t j | | � t j | | � � q q Wd S( s9 Asserts that two records are equal. Pretty crude for now.N( R( R t namest operatort getitemR ( R R t ft aft bf( ( s8 /usr/lib64/python2.7/site-packages/numpy/ma/testutils.pyt assert_equal_recordsE s %,c
C sB t | t � r� t | t � s9 t t t | � � � � n t t | � t | � | � xa | j � D]S \ } } | | k r� t d | | f � � n t | | | | d | | f � qb Wd St | t t f � r� t | t t f � r� t
| | d d �St | t � pt | t � sLt | | g | � } | | k sHt | � � n d S| t
k rd| t
k s|| t
k r�| t
k r�t | | g | d d d d
�} t | � � n t j | d
t d t �} t j | d
t d t �} | j | j } } | j d k r2| j d k r2t
| j � | j � d d �St | | | � S( s! Asserts that two items are equal.s %s not in %ss key=%r
%sNR- R' t headerR0 R R R t subokt S( R R ( t
isinstancet dictt AssertionErrort reprt typeR( R) t itemst listt tupleR/ R t
build_err_msgR t
ValueErrorR t arrayR t TrueR R t tolistt assert_array_equal( R+ R, R- R. t it msgt actual_dtypet
desired_dtype( ( s8 /usr/lib64/python2.7/site-packages/numpy/ma/testutils.pyR( O s: &* c C s� t | t � r� t | t � s9 t t t | � � � � n t t | � t | � | � x] | j � D]O \ } } | | k r� t t | � � � n t | | | | d | | f � qb Wd St | t t f � rBt | t t f � rBt t | � t | � | � x< t
t | � � D]( } t | | | | d | | f � qWd St | t j � sft | t j � rvt
| | | � St | | g | � } | | k s�t | � � n d S( s6 Raises an assertion error if two items are equal.
s key=%r
%sNs
item=%r
%s( R: R; R< R= R> t
fail_if_equalR) R? R@ RA R* R R t fail_if_array_equalRB ( R+ R, R- R. RH RI ( ( s8 /usr/lib64/python2.7/site-packages/numpy/ma/testutils.pyRL { s& &* &$ i c
C s� t | t j � s$ t | t j � rC t | | d | d | d | �St | | g d | d | �} t t | | � | � d k s� t | � � n d S( sv Asserts that two items are almost equal.
The test is equivalent to abs(desired-actual) < 0.5 * 10**(-decimal)
R% R- t verbosei N( R: R R t assert_array_almost_equalRB t roundR$ R< ( R+ R, R% R- RN RI ( ( s8 /usr/lib64/python2.7/site-packages/numpy/ma/testutils.pyt assert_almost_equal� s $
c C s t t | � t | � � } t | d t d | d t d t �} t | d t d | d t d t �} | t k r{ | t k s� | t k r� | t k r� t | | g d | d | d | d d �} t | � � n t j | | j | � | j | � d | d | d | �S( sZ Asserts that a comparison relation between two masked arrays is satisfied
elementwise.R R
t keep_maskR8 R- RN R7 R0 R R ( R R (
R R R R R RB RC t utilst assert_array_compareR ( t
comparisonR R R- RN R7 R R RI ( ( s8 /usr/lib64/python2.7/site-packages/numpy/ma/testutils.pyRT � s $$ c
C s) t t j | | d | d | d d �d S( s5 Checks the elementwise equality of two masked arrays.R- RN R7 s Arrays are not equalN( RT R1 t __eq__( R R R- RN ( ( s8 /usr/lib64/python2.7/site-packages/numpy/ma/testutils.pyRG � s c
C s/ d � } t | | | d | d | d d �d S( sK Raises an assertion error if two masked arrays are not equal (elementwise).c S s t j t | | � � S( N( R t alltrueR" ( R R ( ( s8 /usr/lib64/python2.7/site-packages/numpy/ma/testutils.pyt compare� s R- RN R7 s Arrays are not equalN( RT ( R R R- RN RX ( ( s8 /usr/lib64/python2.7/site-packages/numpy/ma/testutils.pyRM � s c
s5 � f d � } t | | | d | d | d d �d S( s[ Checks the elementwise equality of two masked arrays, up to a given
number of decimals.c s t | | d d � �S( s< Returns the result of the loose comparison between x and y).R g $@( R" ( R R ( R% ( s8 /usr/lib64/python2.7/site-packages/numpy/ma/testutils.pyRX � s R- RN R7 s Arrays are not almost equalN( RT ( R R R% R- RN RX ( ( R% s8 /usr/lib64/python2.7/site-packages/numpy/ma/testutils.pyt assert_array_approx_equal� s c
s5 � f d � } t | | | d | d | d d �d S( s[ Checks the elementwise equality of two masked arrays, up to a given
number of decimals.c s t | | � � S( s< Returns the result of the loose comparison between x and y).( R& ( R R ( R% ( s8 /usr/lib64/python2.7/site-packages/numpy/ma/testutils.pyRX � s R- RN R7 s Arrays are not almost equalN( RT ( R R R% R- RN RX ( ( R% s8 /usr/lib64/python2.7/site-packages/numpy/ma/testutils.pyRO � s c
C s) t t j | | d | d | d d �d S( s, Checks that x is smaller than y elementwise.R- RN R7 s Arrays are not less-orderedN( RT R1 t __lt__( R R R- RN ( ( s8 /usr/lib64/python2.7/site-packages/numpy/ma/testutils.pyt assert_array_less� s c C sU | t k r t | t k � n | t k r> t | t k � n t | | d | �d S( s" Asserts the equality of two masks.R- N( R t assert_RG ( t m1t m2R- ( ( s8 /usr/lib64/python2.7/site-packages/numpy/ma/testutils.pyt assert_mask_equal� s
(* t __doc__t
__author__t __version__t __revision__t __date__R1 t numpyR R R t numpy.core.umatht coreR t
numpy.testingt numpy.testing.utilst testingRS R R R R R R R R
RE R" R& R/ R6 R( RL t assert_not_equalRQ t assert_closeRT RG RM RY RO R[ R_ ( ( ( s8 /usr/lib64/python2.7/site-packages/numpy/ma/testutils.pyt <module> s8
:
,