F d Z ddlZddlZddlmZmZ ddlmc mZ ddl Zddl m
Z
m Z m Z m
Z
mZ ddlmZmZmZmZmZmZ g dZddlmZ g d Zeez Zd d
Zd dZd dZ d Z d dZ d dZ e Z d!dZ!e!Z" d"dZ#d#dZ$d#dZ%d$dZ&d$dZ'd#d Z(d d Z)dS )%zMiscellaneous 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 $
N)ndarrayfloat_)assert_assert_allclose assert_array_almost_equal_nulp
assert_raises
build_err_msg )mask_orgetmask masked_arraynomaskmaskedfilled)almostapproxassert_almost_equalassert_array_almost_equalassert_array_approx_equalassert_array_compareassert_array_equalassert_array_less assert_close assert_equalassert_equal_recordsassert_mask_equalassert_not_equalfail_if_array_equal)TestCase)r r r r r Th㈵>:0yE>c t t | t | }t | }t | }|j j dk s|j j dk r't j || S t t |d| | t }t t |d| d t } t j t j
|| z
||t j
| z z }
|
S )a
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.
OFcopymaskr
)r r r dtypecharnpequalravelr
astyper
less_equalumathabsolute) ab
fill_valuertolatolmd1d2xyds f/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/numpy/ma/testutils.pyr r ) s
GAJJ''A B B x}rx} 3 3xB %%'''|BU333Z@@GGOOA|BU333Q77>>vFFA
enQU++TD5>!;L;L4L-LMMA 7799 c t t | t | }t | }t | }|j j dk s|j j dk r't j || S t t |d| | t }t t |d| d t }t j t j ||z
| d| z k } | S )z
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# Fr$ r
$@)
r r r r' r( r) r* r+ r
r, r aroundabs)
r0 r1 decimalr2 r5 r6 r7 r8 r9 r: s
r; r r ? s
GAJJ''A B B x}rx} 3 3xB %%'''|BU333Z@@GGOOA|BU333Q77>>vFFA
"&Q--))Twh-??A 7799r< c t t | t | | t t | D ]%}t | | || d|d| &dS )z;
Asserts the equality of two non-array sequences.
item=
N)r lenrange)actualdesirederr_msgks r; _assert_equal_on_sequencesrM R st
Vc'llG444
3w<<
F FVAY
,DA,D,D7,D,DEEEE
Fr< c < t | j |j | j j D ]t}t j | | t j || }}|t
ur?|t
ur6t t j | | t j || udS )zI
Asserts that two records are equal.
Pretty crude for now.
N)r r' namesoperatorgetitemr )r0 r1 fafbfs r; r r ] s !'"""
W] I I $Q**H,t |t s)t | |g| }|| k st | dS | t u r |t us|t u r-| t ur$t | |g|dd }t | t j | } t j | }| j |j }}|j d k rA|j d k r6t | | d S t) | || S )
z,
Asserts that two items are equal.
z not in key=rF NrC rK r8 r9 )headerrO S)
isinstancedictAssertionErrorreprtyper rG itemslisttuplerM r r r
ValueErrorr)
asanyarrayr' r( tolistr )rI rJ rK rL imsg actual_dtype
desired_dtypes r; r r l sT '4 &$'' 5 d6ll!3!344 4S[[#g,,888MMOO I IDAq$%;%;6%;%;<<< GAJ0Gq0G0Gg0G0G H H H H'D%=)) Gj$ .O.O G)&'2FFFF vw ' ' :gw+G+G VW -w88& %% % 6 Gv$5$5 &F*:*:VW - #BjB B Boo
]6
"
"FmG$$G%+\7==\C M$6#$=$=)&--//*1..*:*:246 6 6 6 fgw 7 77r< c t |t rt | t s)t t t | t t
| t
| | | D ]H\ }}|| vr t t | t | | || d|d| IdS t |t t f rt | t t f rot t
| t
| | t t
| D ]%}t | | || d|d| &dS t | t j st |t j rt | || S t | |g| }|| k st | dS )z<
Raises an assertion error if two items are equal.
rV rF NrE )r[ r\ r] r^ r_
fail_if_equalrG r` ra rb rH r) r r r )rI rJ rK rL rf rg s r; rk rk s
'4 &$'' 5 d6ll!3!344 4c&kk3w<<999MMOO J JDAq$T!WW--- &)WQZ1H1H1Hw1H1H I I I I'D%=)) j$ .O.O c&kk3w<<999s7||$$ K KA &)WQZ1I1I1I1I1I J J J J&"*%% =GRZ)H)H ="67G<<<
)7
3
3C f S!!! r< c $ t | t j st |t j rt | |||| S t | |g|| }t t
|| z
| dk st | dS )z~
Asserts that two items are almost equal.
The test is equivalent to abs(desired-actual) < 0.5 * 10**(-decimal).
)rB rK verbose)rK rn r N)r[ r) r r r roundrA r] )rI rJ rB rK rn rg s r; r r s &"*%% KGRZ)H)H K('18'K K K K
) ' : : :C Wv%&& 0 0A 5 5 S!!! 6 5r< c t t | t | }t |d|dd }t |d|dd }|t u r |t us|t u r.|t ur%t ||g|||d }t | t j | | | | | ||| S )zn
Asserts that comparison between two masked arrays is satisfied.
The comparison is elementwise.
F)r% r& keep_masksubokrX )rK rn rY rO rK rn rY )
r r r
r r rc r) testingr r )
comparisonr8 r9 rK rn rY r2 r5 rg s r; r r s
GAJJ''AQUe5IIIAQUe5IIIA
fqF{{6kkAKKQFGW#)= = =oo
: * *:+,88J+?+?+,88J+?+?3:3:6 + K K Kr< c D t t j | |||d dS )z@
Checks the elementwise equality of two masked arrays.
Arrays are not equalrs N)r rP __eq__r8 r9 rK rn s r; r r s5
!Q!(' 68 8 8 8 8 8r< c 6 d }t || |||d dS )zT
Raises an assertion error if two masked arrays are not equal elementwise.
c H t j t | | S )N)r) allr rX s r; comparez$fail_if_array_equal..compare s F6!Q<<((()r< rw rs Nr )r8 r9 rK rn r} s r; r r s?
* * *!Q 68 8 8 8 8 8r< c < fd}t || |||d dS )|
Checks the equality of two masked arrays, up to given number odecimals.
The equality is checked elementwise.
c 0 t | |d z S ).compare s a 1111r< Arrays are not almost equalrs Nr~ r8 r9 rB rK rn r} s ` r; r r sL 2 2 2 2 2 !Q =? ? ? ? ? ?r< c < fd}t || |||d dS )r c & t | | S )r )r r s r; r} z*assert_array_almost_equal..compare s aG$$$r< r rs Nr~ r s ` r; r r sL % % % % % !Q =? ? ? ? ? ?r< c D t t j | |||d dS )z7
Checks that x is smaller than y elementwise.
zArrays are not less-orderedrs N)r rP __lt__ry s r; r r
s5
!Q!(' =? ? ? ? ? ?r< c | t u rt |t u |t u rt | t u t | || dS )z-
Asserts the equality of two masks.
rW N)r r r )m1m2rK s r; r r sU
V||f V||f r2w//////r< )Tr r! )r= T)rC )rl rC T)rC TrC T)rC T)r= rC T)*__doc__rP numpyr) r r numpy.core.umathcorer.
numpy.testingr r r r r r r r
r r r
__all__maskedunittestr __some__from_testing__all__r r rM r r rk r r r r r r r r r r r< r; r s] ! ! ! ! ! ! ! ! I H H H H H H H H H H H H H H H
.
. , & $8 $8 $8 $8N" " " "4 !
"
"
"
" # MO$(K K K K28 8 8 88 8 8 8 ? ? ? ? ? ? ? ? ? ? ? ? 0 0 0 0 0 0r<