Current File : //usr/lib64/python3.9/site-packages/numpy/ma/__pycache__/testutils.cpython-39.opt-1.pyc
a
z[yc�' � @ 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 �8/usr/lib64/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 Zaround�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 "rF 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�afZbfr7 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= �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 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.
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 | 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< rD �verbose)rD r_ r N)rQ r' r r r �roundr; rS )rB rC r< rD r_ 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_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$ �� ��r 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
�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' 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 �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 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 �r 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 �r 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
�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.
rM N)r
r r )Zm1Zm2rD 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__rH Znumpyr' r r Znumpy.core.umathZcorer+ 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:
'
�
Mini Shell