OwlCyberSecurity - MANAGER
Edit File: test_regression.cpython-38.pyc
U �p�]3 � @ sd d dl mZmZmZ d dlZd dlmZmZmZ d dl m Z d dl m Z d dl Z G dd� de�ZdS )� )�division�absolute_import�print_functionN)�assert_�assert_array_equal� assert_raises)�random)�longc @ sl e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� ZdS )�TestRegressionc C sN t �ddd�D ]:}tj�|dd�}tt �|t j k�oDt �|t jk�� qd S )Ng �g @� � �2 )�npZlinspacer ZmtrandZvonmisesr �allZpi)�selfZmu�r� r �D/usr/lib/python3/dist-packages/numpy/random/tests/test_regression.py�test_VonMises_range s z"TestRegression.test_VonMises_rangec C s� t t�tjjddddd�dk �� t t�tjjddddd�dk�� dg}tjd k}|rltjd krl|�d � |D ]}t tjj|� dk� qpd S ) N� � � � ��size� r )� r r l Zwin32)� ���r r ) r r r r Zhypergeometric�sys�maxsize�platform�append)r �argsZ is_64bits�argr r r �test_hypergeometric_range s ""� z(TestRegression.test_hypergeometric_rangec C s| d}t j�d� t jjd|d�}t �|dk�t|� }d| }t|dk|� t �|dk�t|� }d | }t|d k |� d S ) N�� r g�������?r r z"Frequency was %f, should be > 0.45g�������?� z"Frequency was %f, should be < 0.23gq= ףp�?)r r �seedZ logseries�sum�floatr )r �NZrvsnZfreq�msgr r r �test_logseries_convergence% s z)TestRegression.test_logseries_convergencec C sB t j�d� t j�d�}t j�d� t j�td��}t||� d S )N�� � )r r r' � permutationr r )r �a�br r r �test_permutation_longs5 s z%TestRegression.test_permutation_longsc C sv dddd gdddd gdddd gdddd gfD ]D}t j�d� t|�}t�|� t||d |d |d |d g� q,d S ) Nr r& r )r r )r& r& )r r i90 r )r r r' �list�shuffler )r �tZshuffledr r r � test_shuffle_mixed_dimension< s � z+TestRegression.test_shuffle_mixed_dimensionc C sp t j�� }t �ddddddddddg �}td�D ]:}t j�|� |�d � t|jd d t �d �d d �|� q0d S ) Nr � � r& r � r r i� r g $@)r �p) r r Z RandomState�array�ranger' r �choiceZones)r �m�res�ir r r � test_call_within_randomstateG s z+TestRegression.test_call_within_randomstatec C sX t jjdgdggdd� t jjdgdggt �d�d� t jjdgdggt �d�d� d S )Nr r r )r r Zmultivariate_normalZint_Zint64�r r r r �#test_multivariate_normal_size_typesQ s z2TestRegression.test_multivariate_normal_size_typesc C s: t j�d� t jjdddd�}tt �t �|�� d� d S )Ni��Ig-C ��6?�d r zNans in np.random.beta)r r r' Zbetar �anyZisnan)r �xr r r �test_beta_small_parametersY s z)TestRegression.test_beta_small_parametersc C s� t j�d� dddg}dddg}t jt jt jfD ]N}t j||d�t|� }t jj||d�}t ||k� t t t jj||d d� q0d S ) Nr- r r& r r )Zdtype)r: g�������?) r r r' Zfloat16Zfloat32Zfloat64r; r( r= r r � ValueError)r r0 ZcountsZdtZprobs�cr r r �"test_choice_sum_of_probs_tolerance` s z1TestRegression.test_choice_sum_of_probs_tolerancec C sH t j�d� t �ddg�}td�D ]}t j�|� q"dd l}|�� d S )Nr- r0 A� aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaarD r )r r r' r; r< r4 �gc�collect�r r0 �_rK r r r �1test_shuffle_of_array_of_different_length_stringsm s z@TestRegression.test_shuffle_of_array_of_different_length_stringsc C sT t j�d� t �t �d�t �d�g�}td�D ]}t j�|� q.dd l}|�� d S )Nr- r r r% r ) r r r' r; �aranger< r4 rK rL rM r r r � test_shuffle_of_array_of_objects| s z/TestRegression.test_shuffle_of_array_of_objectsc C s� G dd� dt j�}t j�d� t �d��|�}t j�|�}t|t �dddg�� t|t �d��|�� G dd� dt �}t j�d� |� }t j�|�}t|t �ddd ddg�� t|� � t �d �� d S ) Nc @ s e Zd ZdS )z3TestRegression.test_permutation_subclass.<locals>.NN)�__name__� __module__� __qualname__r r r r r* � s r* r r r r& c @ s e Zd Ze�d�Zdd� ZdS )z3TestRegression.test_permutation_subclass.<locals>.Mr c S s | j S )N)r0 rB r r r � __array__� s z=TestRegression.test_permutation_subclass.<locals>.M.__array__N)rR rS rT r rP r0 rU r r r r �M� s rV r r ) r Zndarrayr r' rP Zviewr/ r r; �objectrU )r r* ZorigZpermrV r> r r r �test_permutation_subclass� s z(TestRegression.test_permutation_subclassN)rR rS rT r r$ r, r2 r6 rA rC rG rJ rO rQ rX r r r r r s r )Z __future__r r r r Z numpy.testingr r r Znumpyr Z numpy.compatr r rW r r r r r �<module> s