Current File : //lib64/python3.9/site-packages/numpy/random/tests/__pycache__/test_smoke.cpython-39.opt-1.pyc
a
z[ycn � @ s( d dl Z d dlmZ d dlZd dlZd dlmZmZm Z d dl
m Z m Z m
Z
mZmZmZ ejdejejejejejejejejejf d�dd� �Zd d
� Z d d d
�Z dd� Z d!dd�Z G dd� d�Z G dd� de �Z!G dd� de �Z"G dd� de �Z#G dd� de �Z$G d d � d e �Z%G d d � d e �Z&dS )"� N)�partial)� assert_equal�assert_�assert_array_equal)� Generator�MT19937�PCG64� PCG64DXSM�Philox�SFC64�module)Zscope�paramsc C s | j S �N)Zparam)Zrequest� r �C/usr/lib64/python3.9/site-packages/numpy/random/tests/test_smoke.py�dtype s r c C sr | � }t t�|�� | d�}t |jdk� | d�}t |jdk� | d�}t |jdk� | dd�}t |jdk� d S )N�
�r �r r �r r r )� r ��size)r �np�isscalar�shape)�f�valr r r �params_0 s
r Fc C s d}t �dd�}t �dd��d�}t �dd��d�}t �ddg�}t �dd��d �}|r�d
}|d |�� }|d |�� }|d |�� }|d |�� }|d |�� }| |� | |dd � | |� | |� | |� | |d
d � | |dd � | |dd � d S )Ng @� @g (@� �Y@r g P�@r � @)� r r" � �?g �?r r �r � )r �arange�reshape�array�max)r �bounded�a�b�c�d�e�gr r r �params_1 s* r1 c C s� d}t | t�r0| D ]}|t| | || �M }qnzt| �t|�krV|t| �t|�kM }nTt | tttjf�r�t |tttjf�r�t| |�D ]\}}|t||�M }q�n || |kM }|S )NT) �
isinstance�dict�
comp_state�type�list�tupler Zndarray�zip)�state1�state2Z identical�key�s1�s2r r r r4 > s
� r4 c C s� |d u rdt j�dd� }| �|� | �|� | j|t jd� | j|t jd� | jdd|t jd� | jdd|t jd� | �d|� | jd|t jd� | j|t jd� | j|t jd� d S )N� r � �r � l g &@) r �randomZrandint�standard_normal�float32�integers�uint64�standard_gamma�float64)�rg�nr r r �warmupO s
rK c @ sD e Zd Zedd� �Zedd� �Zdd� Zdd� Zd d
� Zd d � Z d
d� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Z d3d4� Z d5d6� Z d7d8� Z d9d:� Z d;d<� Z!d=d>� Z"d?d@� Z#dAdB� Z$dCdD� Z%dEdF� Z&dGdH� Z'dIdJ� Z(dKdL� Z)dMdN� Z*dOdP� Z+dQdR� Z,dSdT� Z-dUdV� Z.dWdX� Z/dYdZ� Z0d[d\� Z1d]d^� Z2d_d`� Z3dadb� Z4dcdd� Z5dedf� Z6dgdh� Z7didj� Z8dkdl� Z9dmdn� Z:dodp� Z;dqdr� Z<dsdt� Z=dudv� Z>dwdx� Z?dydz� Z@d{d|� ZAd}d~� ZBdd�� ZCd�d�� ZDd�d�� ZEd�d�� ZFd�d�� ZGd�d�� ZHd�d�� ZId�S )��RNGc C sD t | _d | _dg| _t| j| j� �| _| jjj| _d| _| � � d S )N�90 �@ �
r �
bit_generator�advance�seedr rI �state�
initial_state�seed_vector_bits� _extra_setup��clsr r r � setup_class_ s zRNG.setup_classc C sH t �dd�| _t �dd�d d d �f | _t �ddd��d�| _t| _d S )Nr r g{�G�z�?��d r[ )r r& �vec_1d�vec_2dr' �mat� TypeError�
seed_errorrW r r r rV j s zRNG._extra_setupc C s | j | jj_d S r )rT rI rP rS ��selfr r r � _reset_stateq s zRNG._reset_statec C sJ t | �� �}|jj}|�d� |�d� ||j_|jj}tt||�� d S )Nr" )r rP rS rC r r4 )rb rI rS Z new_stater r r � test_initt s
z
RNG.test_initc C s` | j jj}t| j jd�r@| j j�| j� tt|| j jj� � n | j jjj}t �
d|� �� d S )NrQ z Advance is not supported by ) rI rP rS �hasattrrQ r r4 � __class__�__name__�pytest�skip)rb rS � bitgen_namer r r � test_advance} s
zRNG.test_advancec C s� | j jj}t| j jd�rr| j j�� }|j}tt||� � | j �d� || j j_| j j�� }|j}tt||�� n2| j jjj }|dvr�t
d|� ���t � d|� �� d S )N�jumpedi.� )r zno "jumped" in zJump is not supported by )
rI rP rS re rl r r4 rB rf rg �AttributeErrorrh ri )rb rS Zbit_gen2Z jumped_stateZbit_gen3Zrejumped_staterj r r r � test_jump� s
z
RNG.test_jumpc C sF | j jdddd�}tt|�dk� t|dk�� � t|dk�� � d S )N� �� r r ���r )rI �uniformr �len�all�rb �rr r r � test_uniform� s zRNG.test_uniformc C s� | j jt�dgd �ddd�}tt|�dk� t|dk�� � t|dk�� � | j jt�dgd �t�dgd �dd�}tt|�dk� t|dk�� � t|dk�� � | j jdt�dgd �dd�}tt|�dk� t|dk�� � t|dk�� � d S )Nro r rp r rq r )rI rr r r( r rs rt ru r r r �test_uniform_array� s � zRNG.test_uniform_arrayc C s( t t| j�d��dk� t| jj� d S �Nr )r rs rI rB r ra r r r � test_random� s zRNG.test_randomc C s t t| j�d��dk� d S ry )r rs rI rC ra r r r �test_standard_normal_zig� s z RNG.test_standard_normal_zigc C s( t t| j�d��dk� t| jj� d S ry )r rs rI rC r ra r r r �test_standard_normal� s zRNG.test_standard_normalc C sP t t| j�dd��dk� t t| j�t�dgd �d��dk� t| jj� d S ry )r rs rI rG r r( r1 ra r r r �test_standard_gamma� s &zRNG.test_standard_gammac C s( t t| j�d��dk� t| jj� d S ry )r rs rI �standard_exponentialr ra r r r �test_standard_exponential� s z RNG.test_standard_exponentialc C s8 | j jddd�}tt|�dk� tt| j jdd�� d S )Nr rD r@ �rI r~ r rs r r �rb Zrandomsr r r � test_standard_exponential_float� s z#RNG.test_standard_exponential_floatc C s< | j jdddd�}tt|�dk� tt| j jddd�� d S )Nr rD �inv)r �methodr� r� r r r �#test_standard_exponential_float_log� s
� �z'RNG.test_standard_exponential_float_logc C s( t t| j�d��dk� t| jj� d S ry )r rs rI Zstandard_cauchyr ra r r r �test_standard_cauchy� s zRNG.test_standard_cauchyc C s* t t| j�dd��dk� t| jj� d S ry )r rs rI Z
standard_tr1 ra r r r �test_standard_t� s zRNG.test_standard_tc C s0 t | j�dd�dk� t | j�dd�dk� d S )Nr r# r �� )r rI Zbinomialra r r r �
test_binomial� s zRNG.test_binomialc C s< | j jj}| j �d�}|| j j_| j �d�}t||k� d S )Nl )rI rP rS rE r )rb rS Zint_1Zint_2r r r �test_reset_state� s
zRNG.test_reset_statec C s4 t | �� �}t | �� �}tt|jj|jj� � d S r )r rP r r4 rS �rb rI �rg2r r r �test_entropy_init� s
�zRNG.test_entropy_initc C sJ t | j| j� �}t | j| j� �}|�� |�� tt|jj|jj�� d S r )r rP rR rB r r4 rS r� r r r � test_seed� s
z
RNG.test_seedc C sZ t | j| j� �}|�� |jj}|jdd�}t | �� �}||j_|jdd�}t||� d S )Nr r )r rP rR rC rS r �rb rI rS Zn1r� Zn2r r r �test_reset_state_gauss� s zRNG.test_reset_state_gaussc C sv t | j| j� �}|jdddtjd� |jj}|jdddtjd�}t | �� �}||j_|jdddtjd�}t||� d S )Nr rA �x r@ r )r rP rR rE r �uint32rS r r� r r r �test_reset_state_uint32� s zRNG.test_reset_state_uint32c C sh t | j| j� �}|jdd� |jj}|jddd�}t | �� �}||j_|jddd�}t||k�� � d S )NrD r@ r �r r )r rP rR rB rS r rt r� r r r �test_reset_state_float s zRNG.test_reset_state_floatc C s. t �ddd�}| j�|�}t||k�� � d S �N�� r rq �r r& rI Z permutationr �any�rb �originalZpermutedr r r � test_shuffle s zRNG.test_shufflec C s. t �ddd�}| j�|�}t||k�� � d S r� r� r� r r r �test_permutation s zRNG.test_permutationc C s� | j �ddd�}tt|�dk� | j �t�dgd �d�}tt|�dk� | j �dt�dgd ��}tt|�dk� | j �t�dgd �t�dgd ��}tt|�dk� | j �t�dgd �t�dggd ��}t|jdk� d S )Nr r r )rI Zbetar rs r r( r �rb Zvalsr r r � test_beta s &(z
RNG.test_betac C s | j �d�}tt|�dk� d S ry )rI �bytesr rs r� r r r �
test_bytes" s zRNG.test_bytesc C s. | j �dd�}tt|�dk� t| j j� d S �Nr r )rI Z chisquarer rs r1 r� r r r �test_chisquare& s zRNG.test_chisquarec C s. | j �dd�}tt|�dk� t| j j� d S r� )rI � exponentialr rs r1 r� r r r �test_exponential+ s zRNG.test_exponentialc C s$ | j �ddd�}tt|�dk� d S )N� r� r )rI r r rs r� r r r �test_f0 s z
RNG.test_fc C s$ | j �ddd�}tt|�dk� d S )Nr� r% r )rI Zgammar rs r� r r r �
test_gamma4 s zRNG.test_gammac C s2 | j �dd�}tt|�dk� t| j jdd� d S )Nr# r T�r* )rI Z geometricr rs r1 r� r� r r r �test_geometric8 s zRNG.test_geometricc C s$ | j �ddd�}tt|�dk� d S r� )rI Zgumbelr rs r� r r r � test_gumbel= s zRNG.test_gumbelc C s$ | j �ddd�}tt|�dk� d S r� )rI Zlaplacer rs r� r r r � test_laplaceA s zRNG.test_laplacec C s$ | j �ddd�}tt|�dk� d S r� )rI Zlogisticr rs r� r r r � test_logiticE s zRNG.test_logiticc C s" | j �dd�}tt|�dk� d S )Nr# r )rI Z logseriesr rs r� r r r �test_logseriesI s zRNG.test_logseriesc C s$ | j �ddd�}tt|�dk� d S �Nr 皙�����?)rI Znegative_binomialr rs r� r r r �test_negative_binomialM s zRNG.test_negative_binomialc C s$ | j �ddd�}tt|�dk� d S )Nr r% )rI Znoncentral_chisquarer rs r� r r r �test_noncentral_chisquareQ s z RNG.test_noncentral_chisquarec C s� | j �dddd�}tt|�dk� | j �t�dgd �dd�}tt|�dk� | j �dt�dgd �d�}tt|�dk� | j �ddt�dgd ��}tt|�dk� d S )Nr� r� r% r )rI Z noncentral_fr rs r r( r� r r r �test_noncentral_fU s zRNG.test_noncentral_fc C s$ | j �ddd�}tt|�dk� d S r� )rI Znormalr rs r� r r r � test_normal_ s zRNG.test_normalc C s" | j �dd�}tt|�dk� d S )Nr! r )rI Zparetor rs r� r r r � test_paretoc s zRNG.test_paretoc C sV | j �dd�}tt|�dk� | j �t�dgd ��}tt|�dk� t| j j� d S ry )rI Zpoissonr rs r r( r1 r� r r r � test_poissong s
zRNG.test_poissonc C s" | j �dd�}tt|�dk� d S )Nr� r )rI Zpowerr rs r� r r r �
test_powern s zRNG.test_powerc C s$ | j �ddd�}tt|�dk� d S )Nr r? )rI rE r rs r� r r r �
test_integersr s zRNG.test_integersc C s2 | j �dd�}tt|�dk� t| j jdd� d S )Nr� r Tr� )rI Zrayleighr rs r1 r� r r r �
test_rayleighv s zRNG.test_rayleighc C s$ | j �ddd�}tt|�dk� d S r� )rI Zvonmisesr rs r� r r r �
test_vonmises{ s zRNG.test_vonmisesc C s$ | j �ddd�}tt|�dk� d S �N� �?r )rI Zwaldr rs r� r r r � test_wald s z
RNG.test_waldc C s" | j �dd�}tt|�dk� d S r� )rI Zweibullr rs r� r r r � test_weibull� s zRNG.test_weibullc C sx | j �dd�}tt|�dk� | j �| j�}tt|�dk� | j �| j�}t|jdk� | j �| j�}t|jdk� d S )Nr r[ )r" r[ rZ )rI Zzipfr rs r\ r] r r^ r� r r r � test_zipf� s z
RNG.test_zipfc C sL | j �ddd�}tt�|�� | j �t�dgd �dd�}t|jdk� d S )N� r? r r )rI Zhypergeometricr r r r( r r� r r r �test_hypergeometric� s zRNG.test_hypergeometricc C sL | j �ddd�}tt�|�� | j �dt�dgd �d�}t|jdk� d S )N���r r r r )rI Z
triangularr r r r( r r� r r r �test_triangular� s zRNG.test_triangularc C s� ddg}ddgddgg}| j �||d�}t|jdk� | j �||d�}t|jdk� | j �||d�}t|jdk� t||k�� � d S )Nr r" r[ � )r� r% )rI Zmultivariate_normalr r r� )rb ZmeanZcov�xZx_zigZx_invr r r �test_multivariate_normal� s z RNG.test_multivariate_normalc C sH | j �dddg�}t|jdk� | j jdddgdd�}t|jdk� d S )Nr[ gUUUUUU�?gUUUUUU�?)r% r r r$ )rI Z multinomialr r r� r r r �test_multinomial� s zRNG.test_multinomialc C s | j �dd�}t|jdk� d S )N)r r r� r? )r? r� )rI Z dirichletr r )rb �sr r r �test_dirichlet� s zRNG.test_dirichletc C s� t �| j�}t �|�}tt| j�t|�k� tt| jjj|jj�� t �| j�}t �|�}tt| j�t|�k� tt| jjj|jj�� d S r ) �pickle�dumpsrI �loadsr r5 r4 rP rS )rb ZpickZunpickr r r � test_pickle� s
�
�zRNG.test_picklec C sX | j d u r"| jj}t�d|� �� | j dkr4tj}ntj}tjdg|d�}| �|�}|j }| �d�}|j }t
t ||�� tj d|d�}| �|�}|j }| �|d �}|j }t
t ||� � tj d|d�}| �|�}|j }| �|d �}|j }t
t ||� � dt�
tj d|d�| j d � d }| �|�}|j }| �|d �}|j }t
t ||� � d S ) Nz#Vector seeding is not supported by � r" r@ � r i� r% )rU rP rg rh ri r r� rF r( rS r r4 r& �mod)rb rj r rR �bgr9 r: r r r �test_seed_array� sD
��
zRNG.test_seed_arrayc C s� t | �d��}t|� |jj}|jdtjd�}t | �� �}t|� ||j_|jdtjd�}t||� t|j tj� t
t |jj|jj�� d S )NrM r> r@ ) r rP rK rS rB r rD r r r r r4 �rb rI rS Zr1r� Zr2r r r �test_uniform_float� s
zRNG.test_uniform_floatc C s� t | �� �}t|� |jj}|jddtjd�}t | �� �}t|� ||j_|jddtjd�}t||� t|j tj� t
t |jj|jj�� d S )Ng @r> r@ ) r rP rK rS rG r rD r r r r r4 r� r r r �test_gamma_floats� s
zRNG.test_gamma_floatsc C s� t | �� �}t|� |jj}|jdtjd�}t | �� �}t|� ||j_|jdtjd�}t||� t|j tj� t
t |jj|jj�� d S �Nr> r@ � r rP rK rS rC r rD r r r r r4 r� r r r �test_normal_floats� s
zRNG.test_normal_floatsc C s� t | �� �}t|� |jj}|jdtjd�}t | �� �}t|� ||j_|jdtjd�}t||� t|j tj� t
t |jj|jj�� d S r� r� r� r r r �test_normal_zig_floats s
zRNG.test_normal_zig_floatsc C s� | j }|jj}d}t�|�}||j_|j|d� ||j_|j|d�}t||� t�|�}||j_|j||jd� tj|tjd�}||j_|j|tjd� ||j_|j|tjd�}t||� d S )N�� � �a ��outr )r� r r@ �r� r r� ) rI rP rS r �emptyrC r r rD )rb rI rS r �existing�directZsizedr r r �test_output_fill s$
zRNG.test_output_fillc C s� | j }|jj}d}t�|�}||j_|j|d� ||j_|j|d�}t||� tj|tjd�}||j_|j|tjd� ||j_|j|tjd�}t||� d S �Nr� r� r r@ r� r� )rI rP rS r r� rB r rD �rb rI rS r r� r� r r r �test_output_filling_uniform/ s
z RNG.test_output_filling_uniformc C s� | j }|jj}d}t�|�}||j_|j|d� ||j_|j|d�}t||� tj|tjd�}||j_|j|tjd� ||j_|j|tjd�}t||� d S r� )rI rP rS r r� r~ r rD r� r r r � test_output_filling_exponentialA s
z#RNG.test_output_filling_exponentialc C s� | j }|jj}d}t�|�}||j_|jd|d� ||j_|jd|d�}t||� tj|tjd�}||j_|jd|tjd� ||j_|jd|tjd�}t||� d S )Nr� r� r� r r@ r� r� )rI rP rS r �zerosrG r rD r� r r r �test_output_filling_gammaS s
z RNG.test_output_filling_gammac C s� | j }|jj}d}t�d�d }t�|�}||j_|j||d� ||j_|j||d�}t||� tj|tjd�}||j_|j||tjd� ||j_|j||tjd�}t||� d S ) Nr� g @X@r� r� r r@ r� r� ) rI rP rS r r& r� rG r rD )rb rI rS r Zmur� r� r r r �#test_output_filling_gamma_broadcaste s
z'RNG.test_output_filling_gamma_broadcastc C s | j }d}t�|�}t�t�� |j|tjd� W d � n1 sD0 Y t�t��& |j|d d d� d� W d � n1 s�0 Y tj|tjd�}t�t�� |j|tj d� W d � n1 s�0 Y tj
|tjd�}t�t��" |j d|tj d� W d � n1 �s 0 Y t�t��, |j d|d d d� tjd� W d � n1 �sd0 Y tj
|tj d�}t�t��" |j d|tjd� W d � n1 �s�0 Y t�t��( |j d|d d d� d� W d � n1 �s�0 Y d S )Nr� r� r� r� r@ r� ) rI r r� rh �raisesr_ rC rD �
ValueErrorrH r� rG )rb rI r r� r r r �test_output_fill_errorx s(
. 4 . 2 < 2 zRNG.test_output_fill_errorc
C sf |t jkrd}d}n t �|�}t|j�d }|j}| �� | jj||gd |d�}| �� | jj|gd ||d�}t ||� | �� | jj||d|d�}t ||� | �� | jjt �
|gd �t j
|gt d�d|d�}t ||� | �� | jjt �
|gd �t �
|gd �d|d�} t || � | �� | jjd|d|d�}| �� | jj|gd |d�}t ||� d S )Nr% r r" r r@ r� ) r �bool_�iinfo�intr) �minrc rI rE r r( �object)
rb r �upper�lower�infor+ r, r- r. r/ r r r �test_integers_broadcast� sB
��
�
zRNG.test_integers_broadcastc C sZ t �dg�}t �dg�}| jj|||d�}| jj|d ||d�}| jj||d |d�}d S )Nr" r r@ )r r( rI rE )rb r ZhighZlowr� r r r �test_integers_numpy� s zRNG.test_integers_numpyc C sL |t jkrd}d}n t �|�}t|j�d }|j}t�t��, | j j
||d gd |d� W d � n1 sn0 Y t�t��, | j j
|d |gd |d� W d � n1 s�0 Y t�t��. | j j
|d g|gd |d� W d � n1 s�0 Y t�t��& | j j
dgdg|d� W d � n1 �s>0 Y d S )Nr% r r" r r@ ) r r� r� r� r) r� rh r� r� rI rE )rb r r� r� r� r r r � test_integers_broadcast_errors� s
: : < z"RNG.test_integers_broadcast_errorsN)Jrg �
__module__� __qualname__� classmethodrY rV rc rd rk rn rw rx rz 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� 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� r� r� r� r r r r rL ^ s�
&
!
rL c @ s e Zd Zedd� �Zdd� ZdS )� TestMT19937c C sJ t | _d | _dg| _t| j| j� �| _| jjj| _d| _| � � t
| _ d S )Ni! ! r� ) r rP rQ rR r rI rS rT rU rV r� r` rW r r r rY � s zTestMT19937.setup_classc C sh t j�� }|�d� |�� }|| jj_| jjj}t|d |d d k� � � t|d |d d k� d S )N�c r" rS r; r% �pos)
r rB Z RandomStaterC Z get_staterI rP rS r rt )rb ZnprgrS r: r r r �test_numpy_state� s
z TestMT19937.test_numpy_stateN)rg r� r� r� rY r r r r r r� � s
r� c @ s e Zd Zedd� �ZdS )�
TestPhiloxc C sD t | _d| _dg| _t| j| j� �| _| jjj| _d| _| � � d S �Nl rM rN )
r
rP rQ rR r rI rS rT rU rV rW r r r rY � s zTestPhilox.setup_classN�rg r� r� r� rY r r r r r � s r c @ s e Zd Zedd� �ZdS )� TestSFC64c C sD t | _d | _dg| _t| j| j� �| _| jjj| _d| _| � � d S )NrM � )
r rP rQ rR r rI rS rT rU rV rW r r r rY � s zTestSFC64.setup_classNr r r r r r � s r c @ s e Zd Zedd� �ZdS )� TestPCG64c C sD t | _d| _dg| _t| j| j� �| _| jjj| _d| _| � � d S r rO rW r r r rY � s zTestPCG64.setup_classNr r r r r r � s r c @ s e Zd Zedd� �ZdS )�
TestPCG64DXSMc C sD t | _d| _dg| _t| j| j� �| _| jjj| _d| _| � � d S r )
r rP rQ rR r rI rS rT rU rV rW r r r rY
s zTestPCG64DXSM.setup_classNr r r r r r s r c @ s( e Zd Zedd� �Zdd� Zdd� ZdS )�TestDefaultRNGc C sB t | _d| _dg| _tjj| j� | _| jjj| _ d| _
| � � d S r ) r rP rQ rR r rB � default_rngrI rS rT rU rV rW r r r rY s zTestDefaultRNG.setup_classc C s t t| jjt�� d S r )r r2 rI rP r ra r r r �test_default_is_pcg64" s z$TestDefaultRNG.test_default_is_pcg64c C s� t j�� t j�d � t j�d� t j�d� t j�d� t j�ddg� t�t�� t j�d� W d � n1 sv0 Y t�t�� t j�ddg� W d � n1 s�0 Y d S )NrM r l -�tE+�I�6�%�Yl l ) ��~�eR :TI*�6� rq )r rB r rh r� r� ra r r r r� ' s
� * zTestDefaultRNG.test_seedN)rg r� r� r� rY r
r� r r r r r s
r )F)N)'r� � functoolsr Znumpyr rh Z
numpy.testingr r r Z numpy.randomr r r r r
r Zfixturer� Zint8Zint16Zint32Zint64Zuint8Zuint16r� rF r r r1 r4 rK rL r� r r r r r r r r r �<module> s4 ��
!
u
Mini Shell