U
|
P ±d ) ã @ sR d Z ddlZddlZddlmZmZ ddlm mZ ddl Zddl m
|
Z
|
m Z m Z m Z mZ ddlmZmZmZmZmZmZ ddd d
|
d d d dddddddgZddlmZ dddddgZee Zd1d dZd2d dZd3d"d#Z d$d Z d4d%dZ d5d&d'Z e Z d6d)d Z!e!Z"d7d*d Z#d8d+d Z$d9d,dZ%d:d-d Z&d;d.d
|
Z'd<d/dZ(d=d0dZ)dS )>zÌ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 $
|
|
é N)ÚndarrayÚfloat_)Úassert_Úassert_allcloseÚ assert_array_almost_equal_nulpÚ assert_raisesÚ build_err_msgé )Úmask_orÚgetmaskÚ masked_arrayÚnomaskÚmaskedÚfilledÚalmostÚapproxÚassert_almost_equalÚassert_array_almost_equalÚassert_array_approx_equalÚassert_array_compareÚassert_array_equalÚassert_array_lessÚ assert_closeÚ assert_equalÚassert_equal_recordsÚassert_mask_equalÚassert_not_equalÚfail_if_array_equal)ÚTestCaser r r r r Tçñhãµøä>ç:0âyE>c C sª t t| t|}t| }t|}|jjdks:|jjdkrJt ||¡ ¡ S tt|d|d| t
|
¡}tt|d|dd t
|
¡} t t || ¡||t | ¡ ¡}
|
|
|
¡ 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.
|
|
ÚOF©ÚcopyÚmaskr )r
|
r r ÚdtypeÚcharÚnpÚequalÚravelr Úastyper Z
|
less_equalÚumathÚabsolute) ÚaÚbÚ
|
fill_valueÚrtolZatolÚmÚd1Úd2ÚxÚyÚd© r7 úId:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/ma/testutils.pyr ) s $é c
|
C s¦ t t| t|}t| }t|}|jjdks:|jjdkrJt ||¡ ¡ S tt|d|d| t
|
¡}tt|d|dd t
|
¡}t t || ¡|¡d| 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 ZaroundÚabs)
|
r- r. Údecimalr/ r1 r2 r3 r4 r5 r6 r7 r7 r8 r ? s Ú c C sJ t t| t|| tt|D ]$}t | | || d|d| q dS )z;
|
Asserts the equality of two non-array sequences.
|
|
úitem=Ú
|
N)r ÚlenÚrange)ÚactualÚdesiredÚerr_msgÚkr7 r7 r8 Ú_assert_equal_on_sequencesR s "rF c C sd t | j|j | jjD ]H}t | |¡t ||¡ }}|tk r|tk rt t | |¡t ||¡ qdS )zI
|
Asserts that two records are equal.
|
|
Pretty crude for now.
|
|
N)r r% ÚnamesÚoperatorÚgetitemr )r- r. ÚfÚafZbfr7 r7 r8 r ] s c C s t |trt | ts$ttt| tt| t|| | ¡ D ]B\}}|| krbt| d| t| | || d|d| q@dS t |tt fr²t | tt fr²t
|
| |ddS t | t sèt |t sèt | |g|}|| ksät|dS | t krú|t k s|t kr*| t k r*t | |g|ddd}t|t | ¡} t |¡}| j|j }}|jd krz|jd krzt
|
| ¡ | ¡ ddS t| ||S )
|
z,
|
Asserts that two items are equal.
|
|
z not in úkey=r? Nr= ©rD ©r4 r5 )ÚheaderrG ÚS)Ú
|
isinstanceÚdictÚAssertionErrorÚreprÚtyper r@ ÚitemsÚlistÚtuplerF r r r Ú
|
ValueErrorr' Z
|
asanyarrayr% r& Útolistr )rB rC rD rE ÚiÚmsgZ actual_dtypeZ desired_dtyper7 r7 r8 r l sH
|
|
" ÿÿ ÿ
|
|
þc C s4 t |trt | ts$ttt| tt| t|| | ¡ D ]<\}}|| kr\tt|t| | || d|d| q@dS t |tt frèt | tt frètt| t|| t
|
t|D ]$}t| | || d|d| q¾dS t | t j st |t j rt | ||S t| |g|}|| ks0t|dS )z<
|
Raises an assertion error if two items are equal.
|
|
rL r? Nr> )rQ rR rS rT rU Ú fail_if_equalr@ rV rW rX rA r' r r r )rB rC rD rE r[ r\ r7 r7 r8 r] s&
|
|
" "
|
r] é c C s^ t | tjst |tjr*t| ||||dS t| |g||d}tt|| |dksZt|dS )z~
|
Asserts that two items are almost equal.
|
|
The test is equivalent to abs(desired-actual) < 0.5 * 10**(-decimal).
|
|
)r< rD Úverbose)rD r_ r N)rQ r' r r r Úroundr; rS )rB rC r< rD r_ r\ r7 r7 r8 r ° s ÿ ÿc C s t t|t|}t|d|ddd}t|d|ddd}|tkrF|tk sV|tkrt|tk rtt||g|||dd}t|tjj| | |¡| |¡|||dS )zn
|
Asserts that comparison between two masked arrays is satisfied.
|
|
The comparison is elementwise.
|
|
F)r# r$ Z keep_maskZsubokrN )rD r_ rO rG ©rD r_ rO )
|
r
|
r r r r rY r' Ztestingr r ) Z
|
comparisonr4 r5 rD r_ rO r/ r1 r\ r7 r7 r8 r à s( ÿÿ ÿ üc C s t tj| |||dd dS )z@
|
Checks the elementwise equality of two masked arrays.
|
|
úArrays are not equalra N)r rH Ú__eq__©r4 r5 rD r_ r7 r7 r8 r Ü s
|
|
þc C s dd }t || |||dd dS )zT
|
Raises an assertion error if two masked arrays are not equal elementwise.
|
|
c S s t t| |¡ S )N)r' Zalltruer rN r7 r7 r8 Úcompareë s z$fail_if_array_equal.<locals>.comparerb ra N©r )r4 r5 rD r_ re r7 r7 r8 r æ s ÿc s$ fdd}t || |||dd dS )ú|
|
Checks the equality of two masked arrays, up to given number odecimals.
|
|
The equality is checked elementwise.
|
|
c s t | |d dS )ú<Returns the result of the loose comparison between x and y).r: )r0 )r rN ©r< r7 r8 re ø s z*assert_array_approx_equal.<locals>.compareúArrays are not almost equalra Nrf ©r4 r5 r< rD r_ re r7 ri r8 r ñ s ÿc s$ fdd}t || |||dd dS )rg c s t | | S )rh )r rN ri r7 r8 re s z*assert_array_almost_equal.<locals>.comparerj ra Nrf rk r7 ri r8 r ÿ s ÿc C s t tj| |||dd dS )z7
|
Checks that x is smaller than y elementwise.
|
|
zArrays are not less-orderedra N)r rH Ú__lt__rd r7 r7 r8 r s
|
|
þc C s: | t krt|t k |t kr(t| t k t| ||d dS )z-
|
Asserts the equality of two masks.
|
|
rM N)r r r )Úm1Úm2rD r7 r7 r8 r s
|
)Tr r )r9 T)r= )r= )r= )r^ r= T)r= Tr= T)r= T)r= T)r9 r= T)r9 r= T)r= T)r= )*Ú__doc__rH Únumpyr' r r Znumpy.core.umathÚcorer+ Z numpy.testingr r r r r r
|
r r r r r Z __all__maskedZunittestr Z__some__from_testingÚ__all__r r rF r r r] r r r r r r r r r r r7 r7 r7 r8 Ú<module> s` û þ
|
|
|
|
'
|
|
ÿ
|
|
|
|
|
|
|
