File: //opt/imh-python/lib/python3.9/site-packages/numpy/ma/__pycache__/testutils.cpython-39.pyc
a
e�h�' � @ s4 d Z ddlZddlZddlmZmZ ddlm mZ ddl Zddl m
Z
m Z m Z m
Z
mZ ddlmZmZmZmZmZmZ g d�ZddlmZ g d �Zee Zd,d
d�Zd-dd�Zd.dd�Z dd� Z d/dd�Z d0dd�Z e Z d1d d �Z!e!Z"d2d d �Z#d3d d!�Z$d4d"d#�Z%d5d$d%�Z&d6d&d'�Z'd7d(d)�Z(d8d*d+�Z)dS )9z�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)�TestCase)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�umath�absolute) �a�b�
fill_value�rtolZatol�m�d1�d2�x�y�d� r7 ��/root/rpmbuild/BUILDROOT/imh-python39-modules-3.9.7-92.el8.x86_64/opt/imh-python/lib/python3.9/site-packages/numpy/ma/testutils.pyr ) s $r � 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! Fr" r � $@)
r
r r r% r& r' r( r) r r* r �around�abs)
r- r. �decimalr/ r1 r2 r3 r4 r5 r6 r7 r7 r8 r ? s r � 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 "rG c C sd t | j|j� | jjD ]H}t�| |�t�||� }}|tur|turt t�| |�t�||�� qdS )zI
Asserts that two records are equal.
Pretty crude for now.
N)r r% �names�operator�getitemr )r- r. �f�af�bfr7 r7 r8 r ] s r c C s� t |t�r�t | t�s$ttt| ����tt| �t|�|� |�� D ]B\}}|| vrbt|� d| � ���t| | || d|�d|� �� q@dS t |tt f�r�t | tt f�r�t
| |dd�S t | t �s�t |t �s�t | |g|�}|| ks�t|��dS | t
u r�|t
u�s|t
u �r*| t
u�r*t | |g|ddd�}t|��t�| �} t�|�}| j|j }}|jd k�rz|jd k�rzt
| �� |�� dd�S t| ||�S )
z,
Asserts that two items are equal.
z not in �key=r@ Nr> �rE �r4 r5 )�headerrH �S)�
isinstance�dict�AssertionError�repr�typer rA �items�list�tuplerG r r r �
ValueErrorr' Z
asanyarrayr% r& �tolistr )rC rD rE rF �i�msgZ actual_dtypeZ
desired_dtyper7 r7 r8 r l sD
" ���
�r c C s4 t |t�r�t | t�s$ttt| ����tt| �t|�|� |�� D ]<\}}|| vr\tt|���t| | || d|�d|� �� q@dS t |tt f�r�t | tt f�r�tt| �t|�|� t
t|��D ]$}t| | || d|�d|� �� q�dS t | t j ��st |t j ��rt
| ||�S t| |g|�}|| k�s0t|��dS )z<
Raises an assertion error if two items are equal.
rN r@ Nr? )rS rT rU rV rW �
fail_if_equalrA rX rY rZ rB r' r r r )rC rD rE rF r] r^ r7 r7 r8 r_ � s&
" "
r_ � 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= rE �verbose)rE ra r N)rS r' r r r �roundr< rU )rC rD r= rE ra r^ r7 r7 r8 r � s ��r c C s� t t|�t|��}t|d|ddd�}t|d|ddd�}|tu rF|tusV|tu rt|turtt||g|||dd�}t|��tjj| |� |�|� |�|||d�S )zn
Asserts that comparison between two masked arrays is satisfied.
The comparison is elementwise.
F)r# r$ Z keep_maskZsubokrP )rE ra rQ rH �rE ra rQ )
r
r r r r r[ r' Ztestingr r ) Z
comparisonr4 r5 rE ra rQ r/ r1 r^ r7 r7 r8 r � s$ �� ��r c C s t tj| |||dd� dS )z@
Checks the elementwise equality of two masked arrays.
�Arrays are not equalrc N)r rI �__eq__�r4 r5 rE ra r7 r7 r8 r � s
�r 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' �allr rP r7 r7 r8 �compare� s z$fail_if_array_equal.<locals>.comparerd rc N�r )r4 r5 rE ra rh r7 r7 r8 r � s �r 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� d�S )�<Returns the result of the loose comparison between x and y).r: )r0 )r rP �r= r7 r8 rh � s z*assert_array_approx_equal.<locals>.compare�Arrays are not almost equalrc Nri �r4 r5 r= rE ra rh r7 rl r8 r � s �r c s$ � fdd�}t || |||dd� dS )rj c s t | |� �S )rk )r rP rl r7 r8 rh s z*assert_array_almost_equal.<locals>.comparerm rc Nri rn r7 rl r8 r � s �r c C s t tj| |||dd� dS )z7
Checks that x is smaller than y elementwise.
zArrays are not less-orderedrc N)r rI �__lt__rf r7 r7 r8 r
s
�r c C s: | t u rt|t u � |t u r(t| t u � t| ||d� dS )z-
Asserts the equality of two masks.
rO N)r
r r )�m1�m2rE r7 r7 r8 r s
r )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__rI �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 rG r r r_ r r r r r r r r r r r7 r7 r7 r8 �<module> s:
'
�