�
�9Zc@`
s
dZd
dl
mZmZmZ
d
dlZd
dlZd
dlmZm	Z	
d
dl
jjZd
dl
mZmZmZmZmZmZmZ
d
dljjZddlmZmZmZmZmZmZ
dd	d
ddd
ddddddddgZddddddgZeeZ e!ddd�Z"de!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-dd!e!d+�Z.dd!e!d,�Z/d!e!d-�Z0d!d.�
Z1dS(/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(tdivisiontabsolute_importtprint_functionN(tndarraytfloat_(tTestCasetassert_tassert_allclosetassert_array_almost_equal_nulpt
assert_raisest
build_err_msgtrun_module_suitei
(tmask_ortgetmasktmasked_arraytnomasktmaskedtfilledtalmosttapproxtassert_almost_equaltassert_array_almost_equaltassert_array_approx_equaltassert_array_comparetassert_array_equaltassert_array_lesstassert_closetassert_equaltassert_equal_recordstassert_mask_equaltassert_not_equaltfail_if_array_equalRRRRR	Rg�h㈵��>g:�0�yE>cC`
s�tt
|�
t
|
�
�}t|�
}t|
�
}|jjd
ksW|jjd
krmtj||�j�Stt|dt	d|�
|�j
t�
}tt|dt	d|�
d�j
t�
}	tj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
Otcopytmaski
(RR
RtdtypetchartnptequaltravelRtFalsetastypeRt
less_equaltumathtabsolute(t
at
bt
fill_valuetrtoltatolt
mtd1td2t
xt
yt
d((sE/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/testutils.pyR*s


$

*
*
0
ic
C`
s�tt
|�
t
|
�
�}t|�
}t|
�
}|jjd
ksW|jjd
krmtj||�j�Stt|dt	d|�
|�j
t�
}tt|dt	d|�
d�j
t�
}tjtj
||�
|�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
RR#R$R%R&R'RR(R)Rtaroundtabs(
R-R.tdecimalR/R2R3R4R5R6R7((sE/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/testutils.pyR@s


$

*
*
*
tcC`
s_tt
|�
t
|
�
|�
x<tt
|
�
�
D](}t|||
|d
||f�
q/WdS(s;
    Asserts the equality of two non-array sequences.

    s
item=%r
%sN(Rtlentrange(tactualtdesiredterr_msgt
k((sE/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/testutils.pyt_assert_equal_on_sequencesSs

&
cC`
s�t|j
|
j
�
xy|j
jD]k}tj||�tj|
|�}}|tk	r |tk	r ttj||�tj|
|��
q q Wd
S(sI
    Asserts that two records are equal.

    Pretty crude for now.

    N(RR#tnamestoperatortgetitemR(R-R.t
ftaftbf((sE/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/testutils.pyR^s

%

,
cC`
sBt|
t
�r�t|t
�s9ttt|�
�
�
�
ntt|�
t|
�
|�
xa|
j�D]S\}}||kr�td
||f�
�
nt|||
|d||f�
qbWdSt|
tt	f�r�t|tt	f�r�t
||
dd�
St|t�p
t|
t�sL
t||
g|�}|
|ksH
t|�
�
ndS|t
krd
|
t
k	s|
|
t
kr�
|t
k	r�
t||
g|dddd
�}t|�
�
ntj|d
tdt�
}tj|
d
tdt�
}
|j|
j}}|jdkr2|jdkr2t
|j�|
j�dd�
St||
|�S(s,
    Asserts that two items are equal.

    s%s not in %ss	key=%r
%sNR@R;theaderRCR5R6R!tsubokt
S(R5R6(t
isinstancetdicttAssertionErrortreprttypeRR<titemstlistttupleRBRR
Rt
ValueErrorR%tarrayR(tTrueR#R$ttolistR(R>R?R@RAt
itmsgtactual_dtypet
desired_dtype((sE/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/testutils.pyRms:






&
*

















cC`
s�
t|
t
�r�t|t
�s9ttt|�
�
�
�
ntt|�
t|
�
|�
x]|
j�D]O\}}||kr�tt|�
�
�
nt|||
|d
||f�
qbWdSt|
tt	f�rB
t|tt	f�rB
tt|�
t|
�
|�
x<t
t|
�
�
D](}t|||
|d||f�
q
WdSt|tj�sf
t|
tj�rv
t
||
|�St||
g|�}|
|ks�
t|�
�
ndS(s<
    Raises an assertion error if two items are equal.

    s	key=%r
%sNs
item=%r
%s(RLRMRNRORPt
fail_if_equalR<RQRRRSR=R%RRR
(R>R?R@RARXRY((sE/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/testutils.pyR\�s&






&

*


&

$



ic	C`
s�t|t
j�s$t|
t
j�rCt||
d
|d|d|�St||
gd|d|�
}tt|
|�
|�dks�t|�
�
ndS(s~
    Asserts that two items are almost equal.

    The test is equivalent to abs(desired-actual) < 0.5 * 10**(-decimal).

    R:R@tverboseiN(RLR%RRR
troundR9RN(R>R?R:R@R]RY((sE/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/testutils.pyR�s$






c	C`
s
tt
|
�
t
|�
�}t|
d
td|dtdt�
}
t|d
td|dtdt�
}|
tkr{|tk	s�|tkr�|
tk	r�t|
|gd|d|d|dd�
}t|�
�
ntj||
j	|�
|j	|�
d|d|d|�S(sn
    Asserts that comparison between two masked arrays is satisfied.

    The comparison is elementwise.

    R!R"t	keep_maskRJR@R]RIRCR5R6(R5R6(
RR
RR(RR
RTtutilsRR(	t
comparisonR5R6R@R]RIR/R2RY((sE/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/testutils.pyR�s	
$
$




	



c
C`
s)tt
j||
d
|d|dd�
dS(s@
    Checks the elementwise equality of two masked arrays.

    R@R]RIsArrays are not equalN(RRDt__eq__(R5R6R@R]((sE/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/testutils.pyR�s

c
C`
s/d
�}t|||
d|d|dd�
dS(sT
    Raises an assertion error if two masked arrays are not equal elementwise.

    cS`
stj
t||
��
S(
N(R%talltrueR(R5R6((sE/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/testutils.pytcompare�s
R@R]RIsArrays are not equalN(
R(R5R6R@R]Rd((sE/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/testutils.pyR�s	
c
`
s5�f
d
�}t|||
d|d|dd�
dS(s|
    Checks the equality of two masked arrays, up to given number odecimals.

    The equality is checked elementwise.

    c`
st||
d
d��
S(s<Returns the result of the loose comparison between x and y).R0g$@(
R(R5R6(
R:(sE/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/testutils.pyRd�sR@R]RIsArrays are not almost equalN(
R(R5R6R: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|dd�
dS(s|
    Checks the equality of two masked arrays, up to given number odecimals.

    The equality is checked elementwise.

    c`
st||
��S(
s<Returns the result of the loose comparison between x and y).(
R(R5R6(
R:(sE/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/testutils.pyRd
sR@R]RIsArrays are not almost equalN(
R(R5R6R:R@R]Rd((
R:sE/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/testutils.pyR
s
c
C`
s)tt
j||
d
|d|dd�
dS(s7
    Checks that x is smaller than y elementwise.

    R@R]RIsArrays are not less-orderedN(RRDt__lt__(R5R6R@R]((sE/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/testutils.pyR
s

cC`
sU|tkrt
|
tk�

n|
tkr>t
|tk�

nt||
d
|�

dS(s-
    Asserts the equality of two masks.

    R@N(RRR(tm1tm2R@((sE/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/testutils.pyR
s




(2t__doc__t
__future__RR
RRDtnumpyR%RRtnumpy.core.umathtcoreR+t
numpy.testingRRRRR	R
Rtnumpy.testing.utilsttestingR`RR
RRRRt
__all__maskedt__some__from_testingt__all__RVRRRBRRR\RRRRRRRRRR(((sE/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/testutils.pyt<module>sB



4
.

	
	
	

	'