File: //usr/local/lib/python3.10/dist-packages/numpy/ma/tests/__pycache__/test_regression.cpython-310.pyc
o
���g� � @ s2 d dl Zd dlmZmZmZmZ G dd� d�ZdS )� N)�assert_�assert_array_equal�assert_allclose�suppress_warningsc @ s| e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd S ) �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 �nonzero)�self�x� r �I/usr/local/lib/python3.10/dist-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 )�numpy.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 �masked�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 )N�Unicode)�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 �
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 �
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
�����dtype)�outr )r r r �arange�float�varr �_data)r r! �moutr 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� �(}|�td� t jj||dd�}t jj||dd�}t|j |j � W d � d S 1 s>w Y d S )N)r r r r
r r
)r g @g������@r r z bias and ddof have no effectr )�ddofr )
r r � masked_equalr r �filter�DeprecationWarning�corrcoefr �data)r r �y�sup�r0�r1r 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 �tiler& 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 rI )r rJ )r g @zi4,f4r5 r )r r rL 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
rK �F)�order)r r r8 �reshaper �tobytes�TrO r r r �!test_masked_array_tobytes_fortranY s z0TestRegression.test_masked_array_tobytes_fortranc C s2 t jjddt jfddt jfdt jfgfgd� d S )N)r )� rW r rD �i�jr5 )r r r �int_�voidr r r r �test_structured_array] s
�
�z$TestRegression.test_structured_arrayN)�__name__�
__module__� __qualname__r r r$ r' r* r- r0 r3 r= rH rN rP rV r\ r r r r r s
r )�numpyr �
numpy.testingr r r r r r r r r �<module> s