Edit File: test_regression.cpython-37.pyc
B ��Fd}" � @ s� d dl mZmZmZ d dlZd dlZd dlZd dlm Z m Z m Z m Z m Z mZmZ d dlmZ d dlmZ dZG dd� de �Zed kr�e � dS ) � )�division�absolute_import�print_functionN)�run_module_suite�TestCase�assert_� assert_equal�assert_array_equal�assert_array_almost_equal� assert_raises)�_assert_valid_refcount)�unicode� c @ s$ e Zd Zefdd�Zefdd�Zefdd�Zefdd�Zefd d �Zefd d �Z efd d�Z efdd�Z efdd�Z efdd�Z efdd�Zefdd�Zdd� Zdd � Zd d � Zd d � Zd!d"� Zefd#d$�Zefd%d&�Zefd'd(�Zefd)d*�Zd+d,� Zd-d.� Zd/d0� Zd1d2� Z d3d4� Z d5d6� Z d7S )8�TestRegressionc C s. t t�dg�t�ddg� t�ddg�� d S )Nr r ���)r �np�poly1d)�self�level� r ��/builddir/build/BUILDROOT/alt-python37-numpy-1.13.3-8.el8.x86_64/opt/alt/python37/lib64/python3.7/site-packages/numpy/lib/tests/test_regression.py� test_poly1d s zTestRegression.test_poly1dc C s> t j�d�}|�� }t j|dd� t j|dd� t||� d S )N)� r r )�rowvarr )r �random�copy�covr )r r �x�yr r r �test_cov_parameters s z"TestRegression.test_cov_parametersc C sH xBt d�D ]6}t�ddddgddg� t�dddddgddg� q W d S )N�d r � r � r )�ranger �digitize)r r �ir r r �test_mem_digitize s z TestRegression.test_mem_digitizec C s t g t�t�g ��� d S )N)r r �unique�array)r r r r r �test_unique_zero_sized% s z%TestRegression.test_unique_zero_sizedc C sV t �dd� �}|t �d�t �d�t �d�� |t �d�t �d�t �d�t �d�� d S )Nc W s | S )Nr )�argsr r r �<lambda>+ � z3TestRegression.test_mem_vectorise.<locals>.<lambda>)r r! r )r! r r )r r r! )r! r! )r � vectorize�zeros)r r �vtr r r �test_mem_vectorise) s z!TestRegression.test_mem_vectorisec C s0 t tjddd� dg� t tjdd� g � d S )Nr y �?)r r �mgrid)r r r r r �test_mgrid_single_element0 s z(TestRegression.test_mgrid_single_elementc C s dd� }t �|�}t|� d S )Nc S s dS )N�{ r )r r r r r �p7 s z1TestRegression.test_refcount_vectorize.<locals>.p)r r- r )r r r4 �vr r r �test_refcount_vectorize5 s z&TestRegression.test_refcount_vectorizec C s0 t jt jt jdgdd�}| �t jjt|d� d S )Nr r )�rr7 )r r �nan� assertRaises�linalg� LinAlgError�getattr)r r r4 r r r �test_poly1d_nan_roots< s z$TestRegression.test_poly1d_nan_rootsc C s t �g dg� d S )Ng �?)r �polymul)r r r r r �test_mem_polymulA s z TestRegression.test_mem_polymulc C s t �g �}t �|d� d S )Nz asdasd asdasd)r r( �append)r r r r r r �test_mem_string_concatE s z%TestRegression.test_mem_string_concatc C sJ t �dddg�}t �dddddg�}t �||�\}}t|| | |� d S )Nr r! r r" � )r r �polydivr )r r �ur5 �qr7 r r r � test_poly_divJ s z TestRegression.test_poly_divc C s: t �dddg�}t �ddg�}t||k� t||k� d S )Nr r! r r" )r r r )r r r r r r r � test_poly_eqQ s zTestRegression.test_poly_eqc Q C sv dddddg}dddd d d d d ddddddddddddddd d d d d d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;d<d=d>d?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVgQ}dWdXdYdZdZd[d\d]dWd[d]d^d\d^dWd[d\d_d\d_dYd\d\d\d`dZdWd]dZd^d[d^dYdYd\dadYdYdWdZdad\d\dZdXdWd[d[d\dYdWd_d\d[dZdYdYd_d[d[d\dXd[dZd^dYd[d\d\dZdadWd[d[d[d\dYdZdZd^dYgQ}t �||db�}t||� d S )cNg1$���ͱ�g�H P�B?g�{p#��g�p�I�#@g K<�l�s��Z �[ �\ �] �^ �_ �` �a �b �c r �e �f �g �h �i �j �k �l �m �n �o �p �q �r �s �t �u �v �w �x �y �z r3 �| �} �~ � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � g "@g @g @g @g @g @g &@g @g $@g *@g (@r" )r �polyfitr )r �refr r Ztestedr r r �test_polyfit_buildX s$ z!TestRegression.test_polyfit_buildc C sp d}t jdt jd�}t �||�\}}t|jt jk|� d}t jdt jd�}t �||�\}}t|jt jk|� d S )Nz Wrong type, should be complexr )�dtypezWrong type, should be float)r �ones�complexrC r r� �int�float)r �msgr rE r7 r r r �test_polydiv_typem s z TestRegression.test_polydiv_typec C s t ttjt�d�dd� d S )N)r � i )�bins)r � ValueErrorr � histogramddr� )r r r r � test_histogramdd_too_many_binsx s z-TestRegression.test_histogramdd_too_many_binsc C s\ d}t jdt jd�}tt �|�jt jk|� d}t jdt jd�}tt �|�jt jk|� d S )Nz Wrong type, should be complexr )r� zWrong type, should be float)r r� r� r �polyintr� r� r� )r r� r r r r �test_polyint_type| s z TestRegression.test_polyint_typec C s t t�t�g g��� d S )N)�listr � ndenumerater( )r r r r �test_ndenumerate_crash� s z%TestRegression.test_ndenumerate_crashc C s t t�tjg�t�d g�� d S )N)r r r( r8 �asfarray)r r r r r �test_asfarray_none� s z!TestRegression.test_asfarray_nonec sZ t �t j�jd }td| d d �� � fdd�}� fdd�}| �t|� | �t|� d S ) N� r! g�������?r c s@ d} t �| fd �}t jjd| � d�}d|t �|||||�< d S )Nr rB r )�size)r r� r �randint�ix_)�n�ar% )�thesizer r �dp� s z4TestRegression.test_large_fancy_indexing.<locals>.dpc s@ d} t �| fd �}t jjd| � d�}|t �|||||� d S )Nr rB r )r� )r r� r r� r� )r� r� r% )r� r r �dp2� s z5TestRegression.test_large_fancy_indexing.<locals>.dp2)r r� �intp�itemsizer� r9 r� )r r �nbitsr� r� r )r� r �test_large_fancy_indexing� s z(TestRegression.test_large_fancy_indexingc C s6 t �ddg�}t �d|�}tt j||f j|k� d S )N)r� �f4)�b�i4)r )r r� r. r �r_)r r �dtr r r r �test_void_coercion� s z!TestRegression.test_void_coercionc C sn dd l }dd l}|j}t|jd�|_z2yt�dt�d�i� W n td��Y nX W d |j� � ||_X d S )Nr �w�foor z ticket #1243) �os�sys�stdout�open�devnullr �whor( �AssertionError�close)r r r� r� Z oldstdoutr r r �test_who_with_0dim_array� s z'TestRegression.test_who_with_0dim_arrayc C s: t �� g}x*|D ]"}tt|ttf�� t|dk� qW d S )N� )r � get_includer � isinstance�strr )r � include_dirs�pathr r r �test_include_dirs� s z TestRegression.test_include_dirsc C sx t tt�t�dg�d�tj�� t tt�dgd�tj�� t tt�t�dg�d�tj�� t tt�dgd�tj�� d S )Nr r )r r� r �polyderr �ndarray)r r r r �test_polyder_return_type� s z'TestRegression.test_polyder_return_typec C sx ddl m} tjdddgtjd�}ddd g}t�d��tj�}tjtjtjg}y|||||� W n t� �Y nX d S ) Nr )� append_fieldsr r! r )r� r� r� �c) �numpy.lib.recfunctionsr� r r( �int32�eye�astype�float64r� )r r� �base�names�data�dlistr r r � test_append_fields_dtype_list� s z,TestRegression.test_append_fields_dtype_listc C s� t jd dkr ddlm} n ddlm} ddg}tj|d�|d�}t|tjdg|d�� d d d gfg}tj|d�|d�}t|tjd g|d�� d g}tj|d�|d�}t|tjdg|d�� dg}tj|d�|d�}dg}t|tj||d�� d S )Nr r )�StringIO)r� �u1r! )r� r� r! z0 1 2 3)r� ))r r )r! r r� )r� r� )r r )r� r� )))r r r! r )r� r� )r! r! )))r r )r! r )r� r� )r! r r! z0 1 2 3 4 5 6 7 8 9 10 11))))r r )r! r )r" rB ))� � )r� � )r� � )r� � version_info�ior� r �loadtxtr r( )r r� r� r r� r r r � test_loadtxt_fields_subarrays� s z,TestRegression.test_loadtxt_fields_subarraysc C s6 t jdt jd�}yt �|� W n t� �Y nX d S )Nr! )r� )r r. �bool�nansumr� )r r� r r r �test_nansum_with_boolean� s z'TestRegression.test_nansum_with_booleanc C s^ G dd� d�}t tjd�}z4ytj|� |d� W n tk rJ t� �Y nX W d |�� X d S )Nc @ s e Zd ZdZdS )z)TestRegression.test_py3_compat.<locals>.Cz3Old-style class in python2, normal class in python3N)�__name__� __module__� __qualname__�__doc__r r r r �C� s r� r� )�output)r� r� r� r �info�AttributeErrorr� r� )r r� �outr r r �test_py3_compat� s z TestRegression.test_py3_compatN) r� r� r� �rlevelr r r& r) r0 r2 r6 r= r? rA rF rG r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r s6 r �__main__)� __future__r r r r� r� �numpyr Z numpy.testingr r r r r r r Znumpy.testing.utilsr � numpy.compatr r� r r� r r r r �<module> s $ t