OwlCyberSecurity - MANAGER
Edit File: test_randomstate_regression.cpython-38.pyc
U �p�], � @ sX d dl Z d dlZd dlmZmZmZ d dlmZ d dlZ d dl m Z G dd� de �ZdS )� N)�assert_�assert_array_equal� assert_raises)�long)�mtrandc @ 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 d � Zejje�d �jd k d!d"�d#d$� �Zd%d&� Zd'd(� Zd)S )*�TestRegressionc C sL t �ddd�D ]8}t�|dd�}tt �|t j k�oBt �|t jk�� qd S )Ng �g @� � �2 )�npZlinspace�randomZvonmisesr �allZpi)�selfZmu�r� r �P/usr/lib/python3/dist-packages/numpy/random/tests/test_randomstate_regression.py�test_VonMises_range s z"TestRegression.test_VonMises_rangec C s� t t�tjddddd�dk �� t t�tjddddd�dk�� dg}tjd k}|rhtjd krh|�d � |D ]}t tj|� dk� qld S ) N� � � � ��size� r )� r r � 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 sx d}t �d� t jd|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 �seedZ logseriesr �sum�floatr )r �NZrvsnZfreq�msgr r r �test_logseries_convergence' s z)TestRegression.test_logseries_convergencec C s: t �d� t �d�}t �d� t �td��}t||� d S )N�� � )r r& � permutationr r )r �a�br r r �test_permutation_longs7 s z%TestRegression.test_permutation_longsc C st dddd gdddd gdddd gdddd gfD ]B}t �d� t|�}t �|� t||d |d |d |d g� q,d S ) Nr r% r )r r )r% r% )r r �90 r )r r& �list�shuffler )r �tZshuffledr r r � test_shuffle_mixed_dimension> s � z+TestRegression.test_shuffle_mixed_dimensionc C sl t �� }t�ddddddddddg �}td�D ]8}t �|� |�d � t|jd d t�d �d d �|� q.d S ) Nr � � r% r � r r i� r g $@)r �p) r � RandomStater �array�ranger& r �choiceZones)r �m�res�ir r r � test_call_within_randomstateI s z+TestRegression.test_call_within_randomstatec C sR t jdgdggdd� t jdgdggt�d�d� t jdgdggt�d�d� d S )Nr r r )r Zmultivariate_normalr Zint_�int64�r r r r �#test_multivariate_normal_size_typesS s z2TestRegression.test_multivariate_normal_size_typesc C s6 t �d� t jdddd�}tt�t�|�� d� d S )Ni��Ig-C ��6?�d r zNans in random.beta)r r& Zbetar r �anyZisnan)r �xr r r �test_beta_small_parameters[ s z)TestRegression.test_beta_small_parametersc C s~ t �d� dddg}dddg}tjtjtjfD ]J}tj||d�t|� }t j||d�}t ||k� t t t j||d d� q.d S ) Nr, r r% r r ��dtype)r: g�������?) r r& r Zfloat16Zfloat32Zfloat64r< r' r> r r � ValueError)r r/ ZcountsZdtZprobs�cr r r �"test_choice_sum_of_probs_toleranceb s z1TestRegression.test_choice_sum_of_probs_tolerancec C sD t �d� t�ddg�}td�D ]}t �|� q dd l}|�� d S )Nr, r/ A� aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaarF r )r r& r r< r= r4 �gc�collect�r r/ �_rO r r r �1test_shuffle_of_array_of_different_length_stringso s z@TestRegression.test_shuffle_of_array_of_different_length_stringsc C sP t �d� t�t�d�t�d�g�}td�D ]}t �|� q,dd l}|�� d S )Nr, r r r$ r ) r r& r r< �aranger= r4 rO rP rQ 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�d� t �d��|�}t�|�}t|t �dddg�� t|t �d��|�� G dd� dt �}t�d� |� }t�|�}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)r/ rD r r r � __array__� s z=TestRegression.test_permutation_subclass.<locals>.M.__array__N)rV rW rX r rT r/ rY r r r r �M� s rZ r r ) r Zndarrayr r& rT Zviewr. r r< �objectrY )r r) ZorigZpermrZ r? r r r �test_permutation_subclass� s z(TestRegression.test_permutation_subclassc C s@ t jdkrdnd}tjdd�� tjddd|d � W 5 Q R X d S ) NZbigz<i4z>i4znon-native byteorder is not)�matchr �� r )r rK )r � byteorder�pytestZdeprecated_callr �randint)r Zother_byteord_dtr r r �test_warns_byteorder� s z#TestRegression.test_warns_byteorderc C s: t j�d�}t jjdd�}|�dd�|�dd�ks6t�d S )Ni�[)r& r rF )r r r; ra �AssertionError)r Zrs1Zrs2r r r �"test_named_argument_initialization� s z1TestRegression.test_named_argument_initializationc C s� t jjddgd dd�}|jt �t�ks,t�t jjddgd ddd�}|jt �t�ksZt�t jjddd�}|jt �t�ks~t�t jjdddd�}|jt �t�ks�t�d S ) Nr g�������?r% )r: r F)r: �replacer r )re r )r r r> rK �intrc )r rM r r r �test_choice_retun_dtype� s z&TestRegression.test_choice_retun_dtype�lr z Cannot test with 32-bit C long)�reasonc C sH t �d� tjdddddddd d d g d d �}t jddd�}t||� d S )Nr l � � l /*�/ l u&-n l �LC0 l C3�4 l �RD7 l �v� l g}�1 i �tll �^C? rC rJ r r r )r r& r r< ra r )r �expectedZactualr r r �test_randint_117� s ��z TestRegression.test_randint_117c C s6 t j�d� tt�ddddddg�dddddg� d S )Nr2 r r � �?g �?g �?)r r r& r �binomialrD r r r �test_p_zero_stream� s �z!TestRegression.test_p_zero_streamc C sd t j�d� t �ddddddddddg ddddddddddg g�}ttjdgdggd d d �|� d S ) Ni�_� r r r r% r r r rl )r% r r )r r r& r< r rm )r rj r r r �test_n_zero_stream� s ��z!TestRegression.test_n_zero_streamN)rV rW rX r r# r+ r1 r6 rB rE rI rN rS rU r\ rb rd rg r` ZmarkZskipifr Ziinfo�maxrk rn ro r r r r r s* � r )r r` Z numpy.testingr r r Z numpy.compatr Znumpyr Z numpy.randomr r r[ r r r r r �<module> s