|
Server : Apache System : Linux iad1-shared-b8-43 6.6.49-grsec-jammy+ #10 SMP Thu Sep 12 23:23:08 UTC 2024 x86_64 User : dh_edsupp ( 6597262) PHP Version : 8.2.26 Disable Function : NONE Directory : /lib/python3/dist-packages/numpy/ma/__pycache__/ |
Upload File : |
o
6��a�' � @ s8 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 d.dd�Z e Z d/d d �Z!e!Z"
d0d d �Z#d1d d!�Z$d1d"d#�Z%d2d$d%�Z&d2d&d'�Z'd1d(d)�Z(d.d*d+�Z)dS )3z�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 |jjdkr%t�||��� 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 �
less_equal�umath�absolute) �a�b�
fill_value�rtol�atol�m�d1�d2�x�y�d� r9 �4/usr/lib/python3/dist-packages/numpy/ma/testutils.pyr ) s $r � c
C s� t t| �t|��}t| �}t|�}|jjdks |jjdkr%t�||��� 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/ �decimalr0 r3 r4 r5 r6 r7 r8 r9 r9 r: r ? s r � c C sJ t t| �t|�|� tt|��D ]}t | | || d|�d|� �� qdS )z;
Asserts the equality of two non-array sequences.
�item=�
N)r �len�range)�actual�desired�err_msg�kr9 r9 r: �_assert_equal_on_sequencesR s "rI c C sd t | j|j� | jjD ]$}t�| |�t�||�}}|tur/|tur/t t�| |�t�||�� q dS )zI
Asserts that two records are equal.
Pretty crude for now.
N)r r% �names�operator�getitemr )r. r/ �f�af�bfr9 r9 r: r ] s �r c C s| t |t�rDt | t�sttt| ����tt| �t|�|� |�� D ]!\}}|| vr1t|� d| � ���t| | || d|�d|� �� q dS t |tt f�rYt | tt f�rYt
| |dd�S t | t �stt |t �stt | |g|�}|| ksrt|��dS | t
u r||t
us�|t
u r�| t
ur�t | |g|ddd�}t|��t�| �} t�|�}| j|j}}|jd kr�|jd kr�t
| �� |�� dd�S t| ||�S )
z,
Asserts that two items are equal.
z not in �key=rB Nr@ �rG �r6 r7 )�headerrJ �S)�
isinstance�dict�AssertionError�repr�typer rC �items�list�tuplerI r r r �
ValueErrorr' �
asanyarrayr% r& �tolistr )rE rF rG rH �i�msg� actual_dtype�
desired_dtyper9 r9 r: r l s>
" �
� r c C s. t |t�rAt | t�sttt| ����tt| �t|�|� |�� D ] \}}|| vr.tt|���t| | || d|�d|� �� q dS t |tt f�rtt | tt f�rttt| �t|�|� t
t|��D ]}t| | || d|�d|� �� q_dS t | t j �s�t |t j �r�t
| ||�S t| |g|�}|| ks�t|��dS )z<
Raises an assertion error if two items are equal.
rP rB NrA )rU rV rW rX rY �
fail_if_equalrC rZ r[ r\ rD r' r r r )rE rF rG rH r` ra r9 r9 r: rd � s(
" " �rd � c C s^ t | tj�s t |tj�rt| ||||d�S t| |g||d�}tt|| �|�dks-t|��dS )z~
Asserts that two items are almost equal.
The test is equivalent to abs(desired-actual) < 0.5 * 10**(-decimal).
)r? rG �verbose)rG rf r N)rU r' r r r �roundr> rW )rE rF r? rG rf ra r9 r9 r: r � s ���r c C s� t t|�t|��}t|d|ddd�}t|d|ddd�}|tu r#|tus+|tu r:|tur:t||g|||dd�}t|��tjj| |� |�|� |�|||d�S )zn
Asserts that comparison between two masked arrays is satisfied.
The comparison is elementwise.
F)r# r$ � keep_mask�subokrR )rG rf rS rJ �rG rf rS )
r
r r r r r] r' �testingr r ) �
comparisonr6 r7 rG rf rS r0 r3 ra r9 r9 r: r � s ��r c C � t tj| |||dd� dS )z@
Checks the elementwise equality of two masked arrays.
�Arrays are not equalrj N)r rK �__eq__�r6 r7 rG rf r9 r9 r: r � �
�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' �alltruer rR r9 r9 r: �compare� s z$fail_if_array_equal.<locals>.comparern rj N�r )r6 r7 rG rf rs r9 r9 r: r � s
�r c �$ � 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< )r1 )r rR �r? r9 r: rs � s z*assert_array_approx_equal.<locals>.compare�Arrays are not almost equalrj Nrt �r6 r7 r? rG rf rs r9 rx r: r � �
�r c ru )rv c s t | |� �S )rw )r rR rx r9 r: rs s z*assert_array_almost_equal.<locals>.comparery rj Nrt rz r9 rx r: r � r{ r c C rm )z7
Checks that x is smaller than y elementwise.
zArrays are not less-orderedrj N)r rK �__lt__rp r9 r9 r: r
rq r c C s: | t u r
t|t u � |t u rt| t u � t| ||d� dS )z-
Asserts the equality of two masks.
rQ N)r
r r )�m1�m2rG r9 r9 r: r s
r )Tr r )r; T)r@ )re r@ T)r@ Tr@ T)r@ T)r; r@ T)*�__doc__rK �numpyr' r r �numpy.core.umath�corer, �
numpy.testingr r r r r r
r r r
r r �
__all__masked�unittestr �__some__from_testing�__all__r r rI r r rd r r r r r r r r r r r9 r9 r9 r: �<module> s<
'
�