Current File : //lib64/python3.9/site-packages/numpy/ma/tests/__pycache__/test_regression.cpython-39.opt-1.pyc
a
z[yc � @ s2 d dl Zd dlmZmZmZmZ G dd� d�ZdS )� N)�assert_�assert_array_equal�assert_allclose�suppress_warningsc @ st e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� ZdS ) �TestRegressionc C s4 t jjg d�g d�d�}tt j�|�g d�g� d S )N)r � � � r � � � )r r r r r r r r ��mask)r r r � )�np�ma� masked_arrayr Znonzero)�self�x� r �D/usr/lib64/python3.9/site-packages/numpy/ma/tests/test_regression.py�test_masked_array_create s �z'TestRegression.test_masked_array_createc C s t jjddgd� d S )Nr r
)r r �array�r r r r �test_masked_array s z TestRegression.test_masked_arrayc C s> ddl m}m} t�d�}t�|j|�}|||�}|| d S )Nr )� masked_where�MaskType)r r )Znumpy.mar r r �zeros�shape)r r r �a�b�cr r r �test_mem_masked_where s
z$TestRegression.test_mem_masked_wherec C s6 t j�d�}t jj|d< t �d�}|| || d S )N)r
r )r r )r
r )r r r Zmasked�r r r r r r �test_masked_array_multiply s
z)TestRegression.test_masked_array_multiplyc C s t jjdgdd��d� d S )Nr Fr
�
)r r r �repeatr r r r �test_masked_array_repeat" s z'TestRegression.test_masked_array_repeatc C s t tj�d�� d S )NZUnicode)�reprr r r r r r r � test_masked_array_repr_unicode& s z-TestRegression.test_masked_array_repr_unicodec C sF t jjg d�g d�d�}t �|�}t|jjdk� t|jjdk� d S )N)g g333333�?g @)FTFr
r r )r r r Z
atleast_2dr r �ndimr# r r r �test_atleast_2d* s
z TestRegression.test_atleast_2dc C s0 t jjg d�g d�d�}d|_t|jdk� d S )N)r r r! )r r r r
�X)r r r Z
fill_valuer )r r r r r � test_set_fill_value_unicode_py31 s z.TestRegression.test_set_fill_value_unicode_py3c C sD t jjt �d�dd�}t jjdtd�}|j|d� t|jdk� d S )Nr Tr
����Zdtype)�outr )r r r �arange�float�varr �_data)r r Zmoutr r r � test_var_sets_maskedarray_scalar7 s z/TestRegression.test_var_sets_maskedarray_scalarc C s� t j�g d�d�}t �g d��}t� �N}|�td� t jj||dd�}t jj||dd�}t|j |j � W d � n1 sz0 Y d S )N)r r r r
r r
)r g @g������@r r z bias and ddof have no effectr )Zddofr )
r r Z masked_equalr r �filter�DeprecationWarningZcorrcoefr �data)r r �y�supZr0Zr1r r r �test_ddof_corrcoef> s z!TestRegression.test_ddof_corrcoefc C sf t jjddgddgd�}t|jjdk� t �|d�}t|jjdk� t|jdk� t|jjdk� d S )N� �?� @Fr
)r )r r �r r )r r � MaskedArrayr r r Ztiler# r r r �test_mask_not_backmangledJ s z(TestRegression.test_mask_not_backmangledc C s. t jjg d�dd�}t|g |d d� � d S )N))r r< )r r= )r g @zi4,f4r/ r )r r r? r �r r r r r � test_empty_list_on_structuredT s z,TestRegression.test_empty_list_on_structuredc C s. t j�d��d�}t|jdd�|j�� � d S )Nr
r> �F)�order)r r r1 Zreshaper �tobytes�TrA r r r �!test_masked_array_tobytes_fortranY s z0TestRegression.test_masked_array_tobytes_fortranN)�__name__�
__module__� __qualname__r r r" r$ r' r) r+ r- r5 r; r@ rB rG r r r r r s
r )Znumpyr Z
numpy.testingr r r r r r r r r �<module> s
Mini Shell