Ñò
\ÐKc @ s– d Z d Z d Z d Z d Z d d k Z d d k Z d d k l Z l Z d d k
i i Z d d k
Td d k i i Z d d k l Z l Z l Z l Z l Z l Z l Z l 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 | ƒ } | i i d j p | i i d j o t i | | ƒ i ƒ St t | d t d | ƒ| ƒ i
t ƒ } t t | d t d | ƒd ƒ i
t ƒ } t i t
i | | ƒ | | t
i | ƒ ƒ }
|
i ƒ 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.6/site-packages/numpy/ma/testutils.pyt approx s &**0i c
C sø t t | ƒ t | ƒ ƒ } t | ƒ } t | ƒ } | i i d j p | i i d j o t i | | ƒ i ƒ St t | d t d | ƒ| ƒ i
t ƒ } t t | d t d | ƒd ƒ i
t ƒ } t i t i
| | ƒ | ƒ d | j } | i ƒ 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.6/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.6/site-packages/numpy/ma/testutils.pyt _assert_equal_on_sequences> s
&c C s– t | i | i ƒ x| | i i D]n } t i | | ƒ t i | | ƒ } } | t j o6 | t j o) t t i | | ƒ t i | | ƒ ƒ 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.6/site-packages/numpy/ma/testutils.pyt assert_equal_recordsE s
%-c
C sX t | t ƒ o³ t | t ƒ p t t t | ƒ ƒ ƒ ‚ n t t | ƒ t | ƒ | ƒ xc | i ƒ D]U \ } } | | j o t d | | f ƒ ‚ n t | | | | d | | f ƒ qe Wd St | t t f ƒ o* t | t t f ƒ o t
| | d d ƒSt | t ƒ p
t | t ƒ p7 t | | g | ƒ } | | j p t | ƒ ‚ n d S| t
j o
| t
j p | t
j o> | t
j o1 t | | g | d d d d
ƒ} t | ƒ ‚ n t i | d
t d t ƒ} t i | d
t d t ƒ} | i | i } } | i d j o0 | i d j o t
| i ƒ | i ƒ 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.6/site-packages/numpy/ma/testutils.pyR( O s<
&,
c C s¸ t | t ƒ o¯ t | t ƒ p t t t | ƒ ƒ ƒ ‚ n t t | ƒ t | ƒ | ƒ x_ | i ƒ D]Q \ } } | | j o t t | ƒ ƒ ‚ n t | | | | d | | f ƒ qe Wd St | t t f ƒ ov t | t t f ƒ o` t t | ƒ t | ƒ | ƒ x< t
t | ƒ ƒ D]( } t | | | | d | | f ƒ qWd St | t i ƒ p t | t i ƒ o t
| | | ƒ St | | g | ƒ } | | j p 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.6/site-packages/numpy/ma/testutils.pyRL | s*
&, &&
i c C s˜ t | t i ƒ p t | t i ƒ o t | | d | d | d | ƒSt | | g d | d | ƒ} t t | | ƒ | ƒ d j p 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.6/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 j o
| t j p | t j oG | t j o: t | | g d | d | d | d d ƒ} t | ƒ ‚ n t i | | i | ƒ | i | ƒ 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.6/site-packages/numpy/ma/testutils.pyRT ¥ s $$ c
C s) t t i | | 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.6/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 i t | | ƒ ƒ S( N( R t alltrueR" ( R R ( ( s8 /usr/lib64/python2.6/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.6/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.6/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.6/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.6/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.6/site-packages/numpy/ma/testutils.pyRO Ö s c
C s) t t i | | 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.6/site-packages/numpy/ma/testutils.pyt assert_array_lessà s c C s9 | t j o n | t j o n t | | d | ƒd S( s" Asserts the equality of two masks.R- N( R RG ( t m1t m2R- ( ( s8 /usr/lib64/python2.6/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.6/site-packages/numpy/ma/testutils.pyt s8
:
-