3
���h;( � @ sL d Z ddlZddlZddlmZmZ ddljjZddl Zddl m
Z
m Z m Z m
Z
mZ ddlmZmZmZmZmZmZ ddd d
d d d
dddddddgZddlmZ dddddgZee Zd1d d�Zd2d d�Zd3d"d#�Z d$d� Z d4d%d�Z d5d&d'�Z e Z d6d)d �Z!e!Z"d7d*d �Z#d8d+d
�Z$d9d,d�Z%d:d-d �Z&d;d.d
�Z'd<d/d�Z(d=d0d�Z)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:|jjdkrJtj||�j� S tt|d|d�|�j t
�}tt|d|d�d�j t
�} tj t j
|| �||t j
| � �}
|
j� 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�umathZabsolute) �a�b�
fill_value�rtolZatol�m�d1�d2�x�y�d� r5 �3/tmp/pip-build-5_djhm0z/numpy/numpy/ma/testutils.pyr ) s $� c
C s� t t| �t|��}t| �}t|�}|jjdks:|jjdkrJtj||�j� S tt|d|d�|�j t
�}tt|d|d�d�j t
�}tj tj || �|�d| k} | j� 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! F)r" r# r g $@)
r
r r r$ r% r& r' r( r r) r Zaround�abs)
r+ r, �decimalr- r/ r0 r1 r2 r3 r4 r5 r5 r6 r ? s � c C sJ t t| �t|�|� x0tt|��D ] }t | | || d||f � q"W dS )z;
Asserts the equality of two non-array sequences.
z
item=%r
%sN)r �len�range)�actual�desired�err_msg�kr5 r5 r6 �_assert_equal_on_sequencesR s rA c C sh t | j|j� xT| jjD ]H}tj| |�tj||� }}|tk r|tk rt tj| |�tj||�� qW dS )zI
Asserts that two records are equal.
Pretty crude for now.
N)r r$ �names�operator�getitemr )r+ r, �f�afZbfr5 r5 r6 r ] s c C s� t |t�r�t | t�s$ttt| ����tt| �t|�|� xH|j� D ]<\}}|| krbtd|| f ��t| | || d||f � qBW dS t |tt f�r�t | tt f�r�t
| |dd�S t | t �p�t |t �s�t | |g|�}|| ks�t|��dS | t
ko�|t
k �s |t
k�r(| t
k �r(t | |g|dd
d�}t|��tj| d d
d �} tj|d d
d �}| j|j }}|jd k�r�|jd k�r�t
| j� |j� dd�S t| ||�S )z,
Asserts that two items are equal.
z %s not in %sz key=%r
%sNr: )r? r2 r3 )�headerrB FT)r" �subok�S)r2 r3 )�
isinstance�dict�AssertionError�repr�typer r; �items�list�tuplerA r r r �
ValueErrorr& �arrayr$ r% �tolistr )r= r> r? r@ �i�msgZ actual_dtypeZ
desired_dtyper5 r5 r6 r l s:
c C s4 t |t�r�t | t�s$ttt| ����tt| �t|�|� xD|j� D ]8\}}|| kr^tt|���t| | || d||f � qBW dS t |tt f�r�t | tt f�r�tt| �t|�|� x0t
t|��D ] }t| | || d||f � q�W dS t | t j ��st |t j ��rt
| ||�S t| |g|�}|| k�s0t|��dS )z<
Raises an assertion error if two items are equal.
z key=%r
%sNz
item=%r
%s)rJ rK rL rM rN �
fail_if_equalr; rO rP rQ r< r& r r r )r= r> r? r@ rU rV r5 r5 r6 rW � s&
rW � c C s^ t | tj�st |tj�r*t| ||||d�S 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).
)r9 r? �verbose)r? rY r N)rJ r& r r r �roundr8 rL )r= r> r9 r? rY rV r5 r5 r6 r � s
c C s� t t|�t|��}t|d|ddd�}t|d|ddd�}|tkrF|tk sV|tkrt|tk rtt||g|||dd�}t|��tjj| |j |�|j |�|||d�S )zn
Asserts that comparison between two masked arrays is satisfied.
The comparison is elementwise.
F)r" r# Z keep_maskrH r2 r3 )r? rY rG rB )r? rY rG )r2 r3 )
r
r r r r rR r& Ztestingr r ) Z
comparisonr2 r3 r? rY rG r- r/ rV r5 r5 r6 r � s
c C s t tj| |||dd� dS )z@
Checks the elementwise equality of two masked arrays.
zArrays are not equal)r? rY rG N)r rC �__eq__)r2 r3 r? rY r5 r5 r6 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 jt| |�� S )N)r& Zalltruer )r2 r3 r5 r5 r6 �compare� s z$fail_if_array_equal.<locals>.comparezArrays are not equal)r? rY rG N)r )r2 r3 r? rY r\ r5 r5 r6 r � s c s$ � fdd�}t || |||dd� dS )z|
Checks the equality of two masked arrays, up to given number odecimals.
The equality is checked elementwise.
c s t | |d� d�S )z<Returns the result of the loose comparison between x and y).g $@)r. )r )r2 r3 )r9 r5 r6 r\ � s z*assert_array_approx_equal.<locals>.comparezArrays are not almost equal)r? rY rG N)r )r2 r3 r9 r? rY r\ r5 )r9 r6 r � s c s$ � fdd�}t || |||dd� dS )z|
Checks the equality of two masked arrays, up to given number odecimals.
The equality is checked elementwise.
c s t | |� �S )z<Returns the result of the loose comparison between x and y).)r )r2 r3 )r9 r5 r6 r\ s z*assert_array_almost_equal.<locals>.comparezArrays are not almost equal)r? rY rG N)r )r2 r3 r9 r? rY r\ r5 )r9 r6 r � s c C s t tj| |||dd� dS )z7
Checks that x is smaller than y elementwise.
zArrays are not less-ordered)r? rY rG N)r rC �__lt__)r2 r3 r? rY r5 r5 r6 r
s
c C s: | t krt|t k� |t kr(t| t k� t| ||d� dS )z-
Asserts the equality of two masks.
)r? N)r
r r )�m1�m2r? r5 r5 r6 r s
)Tr r )r7 T)r: )r: )r: )rX r: T)r: Tr: T)r: T)r: T)r7 r: T)r7 r: T)r: T)r: )*�__doc__rC Znumpyr& 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 rA r r rW r r r r r r r r r r r5 r5 r5 r6 �<module> sB
'
|