File: //opt/alt/python37/lib64/python3.7/site-packages/numpy/ma/__pycache__/testutils.cpython-37.pyc
B
<�Fd�( � @ sn d Z ddlmZmZmZ ddlZddlZddlmZm Z ddl
m m Z ddl
mZmZmZmZmZmZmZ ddlm mZ ddl mZmZmZmZm Z m Z dd d
d d d
ddddddddgZ ddddddgZ e e Z d2d d �Z!d3d!d�Z"d4d#d$�Z#d%d� Z$d5d&d�Z%d6d'd(�Z&e&Z'd7d*d
�Z(e(Z)d8d+d
�Z*d9d,d�Z+d:d-d�Z,d;d.d �Z-d<d/d �Z.d=d0d�Z/d>d1d�Z0dS )?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 $
� )�division�absolute_import�print_functionN)�ndarray�float_)�TestCase�assert_�assert_allclose� assert_array_almost_equal_nulp�
assert_raises�
build_err_msg�run_module_suite� )�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_equalr r 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|d�d�� 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�umathZabsolute) �a�b�
fill_value�rtolZatol�m�d1�d2�x�y�d� r9 �E/opt/alt/python37/lib64/python3.7/site-packages/numpy/ma/testutils.pyr * s $� c
C s� t t| �t|��}t| �}t|�}|jjdks:|jjdkrJt�||��� S tt|d|d�|�� t
�}tt|d|d�d�� 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% F)r& r' r g $@)
r r r r( r) r* r+ r, r r- r Zaround�abs)
r/ r0 �decimalr1 r3 r4 r5 r6 r7 r8 r9 r9 r: 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�kr9 r9 r: �_assert_equal_on_sequencesS s rE c C sh t | j|j� xT| jjD ]H}t�| |�t�||� }}|tk r|tk rt t�| |�t�||�� qW dS )zI
Asserts that two records are equal.
Pretty crude for now.
N)r r( �names�operator�getitemr )r/ r0 �fZafZbfr9 r9 r: r ^ s c C s� t |t�r�t | t�s$ttt| ����tt| �t|�|� xH|�� D ]<\}}|| krbtd|| f ��t| | || d||f � qBW dS t |tt f�r�t | tt f�r�t
| |dd�S t | t �s�t |t �s�t | |g|�}|| ks�t|��dS | t
kr�|t
k �s |t
k�r(| t
k �r(t | |g|ddd�}t|��tj| dd d
�} tj|dd d
�}| j|j }}|jd k�r�|jd k�r�t
| �� |�� dd�S t| ||�S ) z,
Asserts that two items are equal.
z %s not in %sz key=%r
%sNr> )rC )r6 r7 )�headerrF FT)r& �subok�S)�
isinstance�dict�AssertionError�repr�typer r? �items�list�tuplerE r r r �
ValueErrorr* Zarrayr( r) �tolistr )rA rB rC rD �i�msgZ actual_dtypeZ
desired_dtyper9 r9 r: r m s:
c C s4 t |t�r�t | t�s$ttt| ����tt| �t|�|� xD|�� 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)rM rN rO rP rQ �
fail_if_equalr? rR rS rT r@ r* r r" r )rA rB rC rD rW rX r9 r9 r: rY � s&
rY � 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).
)r= rC �verbose)rC r[ r N)rM r* r r r �roundr< rO )rA rB r= rC r[ rX r9 r9 r: r � s
c C s� t t|�t|��}t|d|ddd�}t|d|ddd�}|tkrF|tk sV|tkrt|tk rtt||g|||dd�}t|��tj| |�|�|�|�|||d�S )zn
Asserts that comparison between two masked arrays is satisfied.
The comparison is elementwise.
F)r& r' Z keep_maskrK )r6 r7 )rC r[ rJ rF )rC r[ rJ ) r r r r r rU �utilsr r ) Z
comparisonr6 r7 rC r[ rJ r1 r3 rX r9 r9 r: r � s
c C s t tj| |||dd� dS )z@
Checks the elementwise equality of two masked arrays.
zArrays are not equal)rC r[ rJ N)r rG �__eq__)r6 r7 rC r[ r9 r9 r: 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 )r6 r7 r9 r9 r: �compare� s z$fail_if_array_equal.<locals>.comparezArrays are not equal)rC r[ rJ N)r )r6 r7 rC r[ r_ r9 r9 r: 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 $@)r2 )r )r6 r7 )r= r9 r: r_ � s z*assert_array_approx_equal.<locals>.comparezArrays are not almost equal)rC r[ rJ N)r )r6 r7 r= rC r[ r_ r9 )r= r: 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 )r6 r7 )r= r9 r: r_ s z*assert_array_almost_equal.<locals>.comparezArrays are not almost equal)rC r[ rJ N)r )r6 r7 r= rC r[ r_ r9 )r= r: r s c C s t tj| |||dd� dS )z7
Checks that x is smaller than y elementwise.
zArrays are not less-ordered)rC r[ rJ N)r rG �__lt__)r6 r7 rC r[ r9 r9 r: r s
c C s: | t krt|t k� |t kr(t| t k� t| ||d� dS )z-
Asserts the equality of two masks.
)rC N)r r r )Zm1Zm2rC r9 r9 r: r s
)Tr# r$ )r; T)r> )r> )r> )rZ r> T)r> Tr> T)r> T)r> T)r; r> T)r; r> T)r> T)r> )1�__doc__Z
__future__r r r rG Znumpyr* r r Znumpy.core.umathZcorer. Z
numpy.testingr r r r
r r r
Znumpy.testing.utilsZtestingr] r r r r r r Z
__all__maskedZ__some__from_testing�__all__r r rE r r rY r! r r r r r" r r r r r9 r9 r9 r: �<module> sB $
'