Current File : //proc/self/root/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/testutils.pyo
�
� 9Zc @` s
d Z d d l m Z m Z m 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
m Z m Z m Z m Z m Z m Z m Z d d l j j Z d d l m Z m Z m Z m Z m Z m Z d d d
d d d
d d d d d d d d g Z d d d d d d g Z e e 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- � Z0 d! d. � Z1 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 $
i ( t divisiont absolute_importt print_functionN( t ndarrayt float_( t TestCaset assert_t assert_allcloset assert_array_almost_equal_nulpt
assert_raisest
build_err_msgt run_module_suitei ( t mask_ort getmaskt masked_arrayt nomaskt maskedt filledt almostt approxt assert_almost_equalt assert_array_almost_equalt assert_array_approx_equalt assert_array_comparet assert_array_equalt assert_array_lesst assert_closet assert_equalt assert_equal_recordst assert_mask_equalt assert_not_equalt fail_if_array_equalR R R R R R g�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( s�
Returns true if all components of a and b are equal 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 npt equalt ravelR t Falset astypeR t
less_equalt umatht absolute( t at bt
fill_valuet rtolt atolt mt d1t d2t xt yt d( ( sE /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/testutils.pyR * 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/ R2 R3 R4 R5 R6 R7 ( ( sE /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/testutils.pyR @ s $***t c C` s_ t t | � t | � | � x< t t | � � D]( } t | | | | d | | f � q/ Wd S( s;
Asserts the equality of two non-array sequences.
s
item=%r
%sN( R t lent range( t actualt desiredt err_msgt k( ( sE /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/testutils.pyt _assert_equal_on_sequencesS 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( sI
Asserts that two records are equal.
Pretty crude for now.
N( R R# t namest operatort getitemR ( R- R. t ft aft bf( ( sE /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/testutils.pyR ^ 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 headerRC R5 R6 R! t subokt S( R5 R6 ( t
isinstancet dictt AssertionErrort reprt typeR R< t itemst listt tupleRB R R
R t
ValueErrorR% t arrayR( t TrueR# R$ t tolistR ( R> R? R@ RA t it msgt actual_dtypet
desired_dtype( ( sE /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/testutils.pyR m 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( s<
Raises an assertion error if two items are equal.
s key=%r
%sNs
item=%r
%s( RL RM RN RO RP t
fail_if_equalR< RQ RR RS R= R% R R R
( R> R? R@ RA RX RY ( ( sE /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/testutils.pyR\ � 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( s~
Asserts that two items are almost equal.
The test is equivalent to abs(desired-actual) < 0.5 * 10**(-decimal).
R: R@ t verbosei N( RL R% R R R
t roundR9 RN ( R> R? R: R@ R] RY ( ( sE /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/testutils.pyR � 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( sn
Asserts that comparison between two masked arrays is satisfied.
The comparison is elementwise.
R! R" t keep_maskRJ R@ R] RI RC R5 R6 ( R5 R6 (
R R
R R( R R
RT t utilsR R ( t
comparisonR5 R6 R@ R] RI R/ R2 RY ( ( sE /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/testutils.pyR � s $$ c
C` s) t t j | | d | d | d d �d S( s@
Checks the elementwise equality of two masked arrays.
R@ R] RI s Arrays are not equalN( R RD t __eq__( R5 R6 R@ R] ( ( sE /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/testutils.pyR � s c
C` s/ d � } t | | | d | d | d d �d S( sT
Raises an assertion error if two masked arrays are not equal elementwise.
c S` s t j t | | � � S( N( R% t alltrueR ( R5 R6 ( ( sE /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/testutils.pyt compare� s R@ R] RI s Arrays are not equalN( R ( R5 R6 R@ R] Rd ( ( sE /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/testutils.pyR � s c
` s5 � f d � } t | | | d | d | d d �d S( s|
Checks the equality of two masked arrays, up to given number odecimals.
The equality is checked elementwise.
c ` s t | | d d � �S( s< Returns the result of the loose comparison between x and y).R0 g $@( R ( R5 R6 ( R: ( sE /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/testutils.pyRd � s R@ R] RI s Arrays are not almost equalN( R ( R5 R6 R: R@ R] Rd ( ( R: sE /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/testutils.pyR � s c
` s5 � f d � } t | | | d | d | d d �d S( s|
Checks the equality of two masked arrays, up to given number odecimals.
The equality is checked elementwise.
c ` s t | | � � S( s< Returns the result of the loose comparison between x and y).( R ( R5 R6 ( R: ( sE /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/testutils.pyRd s R@ R] RI s Arrays are not almost equalN( R ( R5 R6 R: R@ R] Rd ( ( R: sE /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/testutils.pyR s c
C` s) t t j | | d | d | d d �d S( s7
Checks that x is smaller than y elementwise.
R@ R] RI s Arrays are not less-orderedN( R RD t __lt__( R5 R6 R@ R] ( ( sE /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/testutils.pyR 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 R R ( t m1t m2R@ ( ( sE /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/testutils.pyR s
(2 t __doc__t
__future__R R R RD t numpyR% R R t numpy.core.umatht coreR+ t
numpy.testingR R R R R R
R t numpy.testing.utilst testingR` R R
R R R R t
__all__maskedt __some__from_testingt __all__RV R R RB R R R\ R R R R R R R R R R ( ( ( sE /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/testutils.pyt <module> sB 4.
'
Mini Shell