관리-도구
편집 파일: 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 � less_equal�umath�absolute) �a�b� fill_value�rtolZatol�m�d1�d2�x�y�d� r; �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) r1 r2 �decimalr3 r5 r6 r7 r8 r9 r: r; r; 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�kr; r; r< �_assert_equal_on_sequencesS s rG 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 )r1 r2 �f�afZbfr; r; 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@ )rE )r8 r9 )�headerrH FT)r& �subok�S)� isinstance�dict�AssertionError�repr�typer rA �items�list�tuplerG r r r � ValueErrorr* �arrayr( r) �tolistr )rC rD rE rF �i�msgZ actual_dtypeZ desired_dtyper; r; 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)rP rQ rR rS rT � fail_if_equalrA rU rV rW rB r* r r" r )rC rD rE rF r[ r\ r; r; r< 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 r_ r N)rP r* r r r �roundr> rR )rC rD r? rE r_ r\ r; r; 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_maskrN )r8 r9 )rE r_ rM rH )rE r_ rM ) r r r r r rX �utilsr r ) Z comparisonr8 r9 rE r_ rM r3 r5 r\ r; r; r< r � s c C s t tj| |||dd� dS )z@ Checks the elementwise equality of two masked arrays. zArrays are not equal)rE r_ rM N)r rI �__eq__)r8 r9 rE r_ r; r; 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 )r8 r9 r; r; r< �compare� s z$fail_if_array_equal.<locals>.comparezArrays are not equal)rE r_ rM N)r )r8 r9 rE r_ rc r; 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 | |d� d�S )z<Returns the result of the loose comparison between x and y).g $@)r4 )r )r8 r9 )r? r; r< rc � s z*assert_array_approx_equal.<locals>.comparezArrays are not almost equal)rE r_ rM N)r )r8 r9 r? rE r_ rc r; )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 )r8 r9 )r? r; r< rc s z*assert_array_almost_equal.<locals>.comparezArrays are not almost equal)rE r_ rM N)r )r8 r9 r? rE r_ rc r; )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)rE r_ rM N)r rI �__lt__)r8 r9 rE r_ r; r; 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. )rE N)r r r )�m1�m2rE r; r; r< r s )Tr# r$ )r= T)r@ )r@ )r@ )r^ r@ T)r@ Tr@ T)r@ T)r@ T)r= r@ T)r= r@ T)r@ T)r@ )1�__doc__� __future__r r r rI �numpyr* r r Znumpy.core.umath�corer/ Z numpy.testingr r r r r r r Znumpy.testing.utils�testingra r r r r r r Z __all__maskedZ__some__from_testing�__all__r r rG r r r] r! r r r r r" r r r r r; r; r; r< �<module> sB $ '