3
���h � @ 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 sP t jjddddddddgddddddddgd�}tt jj|�dddd gg� d S )
Nr � � � � � � )�mask� )�np�ma� masked_arrayr Znonzero)�self�x� r �?/tmp/pip-build-5_djhm0z/numpy/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} tjd�}tj|j|�}|||�}|| d S )Nr )� masked_where�MaskTyper )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 jjd�}t jj|d< t jd�}|| || d S )Nr
r r r )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�jd� d S )Nr F)r
�
)r r r �repeat)r r r r �test_masked_array_repeat" s z'TestRegression.test_masked_array_repeatc C s t tjjd�� d S )NZUnicode)�reprr r r )r r r r � test_masked_array_repr_unicode& s z-TestRegression.test_masked_array_repr_unicodec C sJ t jjdddgdddgd�}t j|�}t|jjdk� t|jjdk� d S ) Ng g333333�?g @FT)r
r r )r r r Z
atleast_2dr r
�ndim)r r r r r r �test_atleast_2d* s
z TestRegression.test_atleast_2dc C s4 t jjdddgdddgd�}d|_t|jdk� d S )Nr 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 jd�dd�}t jjdtd�}|j|d� t|jdk� d S ) Nr T)r
r )�dtype)�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 jjdddddgd�}t jdddddg�}t� �D}|jtd� t jj||d d
�}t jj||dd
�}t|j |j � W d Q R X d S ) Nr r r r
r g @g������@z bias and ddof have no effectr )Zddof)
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 j|d�}t|jjd k� t|jd
k� t|jjd k� d S ) Ng �?g @F)r
r r )r )r r )r )r r )r r )r r � MaskedArrayr r
r Ztile)r r r r r r �test_mask_not_backmangledJ s z(TestRegression.test_mask_not_backmangledc C s0 t jjd
d d gdd�}t|g |d d � � d S )
Nr � �?r � @r � @zi4,f4)r+ r )r r; )r r< )r r= )r r r9 r )r r r r r � test_empty_list_on_structuredT s z,TestRegression.test_empty_list_on_structuredc C s. t jjd�jd�}t|jdd�|jj� � d S )Nr
r �F)�order)r r )r r r. Zreshaper �tobytes�T)r r 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* r2 r8 r: r> rC r r r r r s
r )Znumpyr Z
numpy.testingr r r r r r r r r �<module> s |