Current File : //lib64/python3.9/site-packages/numpy/random/tests/__pycache__/test_regression.cpython-39.pyc
a
z[yc? � @ sB d dl Z d dlmZmZmZ d dlmZ d dlZG dd� d�ZdS )� N)�assert_�assert_array_equal�
assert_raises)�randomc @ sd e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dS )�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 �H/usr/lib64/python3.9/site-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 �win32)� ���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�| }d|d�d�}t|d k|� t �|d
k�| }d|d�d �}t|d k |� d S )
N�� r g�������?r r zFrequency was �fz, should be > 0.45g�������?� z, should be < 0.23gq=
ףp�?)r
r �seedZ logseries�sumr )r �NZrvsnZfreq�msgr r r �test_logseries_convergence" s z)TestRegression.test_logseries_convergencec C s~ g d�g d�g d�g d�fD ]\}t j�d� t|�}t�|� t j|d |d |d |d gtd
�}tt j|td
�|� q d S ) N)r r$ r N)�r r �r$ r$ �r r N)r r+ r, N)r* r$ r Ni90 r r r r$ �Zdtype)r
r r% �list�shuffle�array�objectr )r �tZshuffledZexpectedr r r � test_shuffle_mixed_dimension2 s �
&z+TestRegression.test_shuffle_mixed_dimensionc C s` t j�� }t �g d��}td�D ]:}t j�|� |�d� t|jddt �d�d d�|� q d S )N)
r � � r$ r � r r5 r r r i� r g $@)r �p) r
r Z RandomStater0 �ranger% r �choiceZones)r �m�res�ir r r � test_call_within_randomstate> 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_typesH 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_parametersP s z)TestRegression.test_beta_small_parametersc C s� t j�d� g d�}g d�}t jt jt jfD ]N}t j||d�t|� }t jj||d�}t ||v � t
t t jj||d d� q,d S )N�� )r r$ r )r r r$ r- )r7 g�������?) r
r r% Zfloat16Zfloat32Zfloat64r0 r& r9 r r �
ValueError)r �aZcountsZdtZprobs�cr r r �"test_choice_sum_of_probs_toleranceW s z1TestRegression.test_choice_sum_of_probs_tolerancec C sH t j�d� t �ddg�}td�D ]}t j�|� q"dd l}|�� d S )NrD rF A� aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaar@ r )r
r r% r0 r8 r/ �gc�collect�r rF �_rI r r r �1test_shuffle_of_array_of_different_length_stringsd s z@TestRegression.test_shuffle_of_array_of_different_length_stringsc C sX t j�d� t jt �d�t �d�gtd�}td�D ]}t j�|� q2dd l}|� � d S )NrD r r r- r" r )
r
r r% r0 �aranger1 r8 r/ rI rJ rK r r r � test_shuffle_of_array_of_objectss s z/TestRegression.test_shuffle_of_array_of_objectsc C s� G dd� dt j�}t j�d� t �d��|�}t j�|�}t|t �g d��� t|t �d��|�� G dd� d�}t j�d� |� }t j�|�}t|t �g d��� 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$ r c @ s e Zd Ze�d�Zdd� ZdS )z3TestRegression.test_permutation_subclass.<locals>.Mr c S s | j S )N)rF r> r r r � __array__� s z=TestRegression.test_permutation_subclass.<locals>.M.__array__N)rP rQ rR r
rN rF rS r r r r �M� s
rT )r$ r r r r r )
r
Zndarrayr r% rN �viewZ permutationr r0 rS )r r' ZorigZpermrT r: r r r �test_permutation_subclass� s z(TestRegression.test_permutation_subclassN)rP rQ rR r r! r) r3 r= r? rC rH rM rO rV r r r r r s
r ) r Z
numpy.testingr r r Znumpyr r
r r r r r �<module> s
Mini Shell