File: //lib/python3/dist-packages/numpy/random/tests/__pycache__/test_regression.cpython-310.pyc
o
6��aM � @ 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�o"t �|t jk�� qd S )Ng �g @� � �2 )�np�linspacer �mtrand�vonmisesr �all�pi)�self�mu�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}|r6tjd
kr6|�d � |D ] }t tjj|� dk� q8d S ) N� � � �
��size� r )� r r l �win32)� ���r r ) r r
r r �hypergeometric�sys�maxsize�platform�append)r �args� 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|d�d�}t|d k|� t �|d
k�t|� }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 �seed� logseries�sum�floatr )r �N�rvsn�freq�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
�|� qd S ) N)r r+ r N)�r r �r+ r+ �r r N)r r6 r7 N)r5 r+ r Ni90 r r r r+ ��dtype)r
r r, �list�shuffle�array�objectr )r �t�shuffled�expectedr 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�|� qd S )N)
r � � r+ r � r rC r r r i� r g $@)r �p) r
r � RandomStater< �ranger, r �choice�ones)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 �multivariate_normal�int_�int64�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, �betar �any�isnan)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 ]'}t j||d�t|� }t jj||d�}t ||v � t
t t jj||d d� qd S )N�� )r r+ r )r r r+ r8 )rE g�������?) r
r r, �float16�float32�float64r< r. rH r r �
ValueError)r �a�counts�dt�probs�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�|� qdd l}|�� d S )NrY r^ �� aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaarS r )r
r r, r< rG r; �gc�collect�r r^ �_re 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�|� qdd l}|� � d S )NrY r r r8 r) r )
r
r r, r< �aranger= rG r; re rf rg 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 r0 � s r0 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^ rQ r r r � __array__� s z=TestRegression.test_permutation_subclass.<locals>.M.__array__N)rl rm rn r
rj r^ ro r r r r �M� s
rp )r+ r r r r r )
r
�ndarrayr r, rj �view� permutationr r< ro )r r0 �orig�permrp rJ r r r �test_permutation_subclass� s z(TestRegression.test_permutation_subclassN)rl rm rn r r( r4 rA rM rR rX rc ri rk rv r r r r r s
r ) r! �
numpy.testingr r r �numpyr r
r r r r r �<module> s