Edit File: test_regression.cpython-37.pyc
B ��Fd � @ sj d dl mZmZmZ d dlZd dlZd dlmZm Z m Z m Z m Z m Z dZG dd� de �Zedkrfe � dS )� )�division�absolute_import�print_functionN)�assert_�TestCase�assert_array_equal�assert_allclose�run_module_suite�suppress_warnings� c @ sp e Zd Zefdd�Zefdd�Zefdd�Zefdd�Zefd d �Zd d � Z d d� Z dd� Z dd� Z dd� Z dS )�TestRegressionc C sP t jjddddddddgddddddddgd�}tt j�|�dddd gg� d S ) Nr r � � � � � )�mask� )�np�ma� masked_arrayr �nonzero)�self�level�x� r ��/builddir/build/BUILDROOT/alt-python37-numpy-1.13.3-8.el8.x86_64/opt/alt/python37/lib64/python3.7/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 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 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 r �test_masked_array_multiply s z)TestRegression.test_masked_array_multiplyc C s t jjdgdd��d� d S )Nr F)r � )r r r �repeat)r r 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 sJ t jjdddgdddgd�}t �|�}t|jjdk� t|jjdk� d S ) Ng g333333�?g @FT)r r r )r r r � atleast_2dr r �ndim)r r% r& r r r �test_atleast_2d0 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 � fill_valuer )r r% r r r � test_set_fill_value_unicode_py37 s z.TestRegression.test_set_fill_value_unicode_py3c C sD t jjt �d�dd�}t jjdtd�}|j|d� t|jdk� d S )Nr T)r ���)�dtype)�outr )r r r �arange�float�varr �_data)r r% �moutr r r � test_var_sets_maskedarray_scalar= s z/TestRegression.test_var_sets_maskedarray_scalarc C s� t j�dddddgd�}t �dddddg�}t� �D}|�td� 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 )�ddof) r r � masked_equalr r �filter�DeprecationWarning�corrcoefr �data)r r �y�sup�r0�r1r r r �test_ddof_corrcoefD s z!TestRegression.test_ddof_corrcoefN)�__name__� __module__� __qualname__�rlevelr r r( r* r- r/ r2 r5 r> rI r r r r r s r �__main__)� __future__r r r �warnings�numpyr Z numpy.testingr r r r r r rM r rJ r r r r �<module> s C