OwlCyberSecurity - MANAGER
Edit File: test_smoke.cpython-38.pyc
U �p�]�l � @ s d dl Z d dlZd dlmZ d dlZd dlZd dlmZm Z m Z d dl m Z m Z mZmZmZ ejdejejejejejejejejejf d�dd� �Zd d � Z d d d �Z dd� Z d dd�Z G dd� de �Z!G dd� de!�Z"G dd� de!�Z#G dd� de!�Z$G dd� de!�Z%G d d � d e!�Z&dS ) � N)�partial)� assert_equal�assert_�assert_array_equal)� Generator�MT19937�PCG64�Philox�SFC64�module)ZscopeZparamsc C s | j S �N)Zparam)Zrequest� r �?/usr/lib/python3/dist-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* r/ 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 r2 ? s � r2 c C s� |d krdt 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 �warmupP s rI 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 rG �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 rY )r r$ �vec_1d�vec_2dr% �mat� TypeError� seed_errorrU r r r rT k s zRNG._extra_setupc C s | j | jj_d S r )rR rG rN rQ ��selfr r r � _reset_stater s zRNG._reset_statec C sJ t | �� �}|jj}|�d� |�d� ||j_|jj}tt||�� d S )Nr )r rN rQ rA r r2 )r` rG rQ Z new_stater r r � test_initu 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 )NrO z Advance is not supported by {0}) rG rN rQ �hasattrrO r r2 � __class__�__name__�pytest�skip�format)r` rQ � 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||�� n0| j jjj }|dkr�t d| ��t � d� |�� d S )N�jumpedi.� )r zno "jumped" in %sz Jump is not supported by {0})rG rN rQ rc rk r r2 r@ rd re �AttributeErrorrf rg rh )r` rQ Zbit_gen2Z jumped_stateZbit_gen3Zrejumped_stateri 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 )rG �uniformr �len�all�r` �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 )Nrn r ro r rp r )rG rq r r& r rr rs rt r r r �test_uniform_array� s � zRNG.test_uniform_arrayc C s( t t| j�d��dk� t| jj� d S �Nr )r rr rG r@ r r_ r r r � test_random� s zRNG.test_randomc C s t t| j�d��dk� d S rx )r rr rG rA r_ 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 rx )r rr rG rA r r_ 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 rx )r rr rG rE r r& r/ r_ r r r �test_standard_gamma� s &zRNG.test_standard_gammac C s( t t| j�d��dk� t| jj� d S rx )r rr rG �standard_exponentialr r_ r r r �test_standard_exponential� s z RNG.test_standard_exponentialc C sH | j jddd�}tt|�dk� |jtjks0t�tt | j jdd�� d S )Nr rB r> � rG r} r rr r r rB �AssertionErrorr r �r` Zrandomsr r r � test_standard_exponential_float� s z#RNG.test_standard_exponential_floatc C sL | j jdddd�}tt|�dk� |jtjks2t�tt | j jddd�� d S )Nr rB �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 rx )r rr rG Zstandard_cauchyr r_ r r r �test_standard_cauchy� s zRNG.test_standard_cauchyc C s* t t| j�dd��dk� t| jj� d S rx )r rr rG Z standard_tr/ r_ 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 rG Zbinomialr_ r r r � test_binomial� s zRNG.test_binomialc C s< | j jj}| j �d�}|| j j_| j �d�}t||k� d S )Nl )rG rN rQ rC r )r` rQ Zint_1Zint_2r r r �test_reset_state� s zRNG.test_reset_statec C s4 t | �� �}t | �� �}tt|jj|jj� � d S r )r rN r r2 rQ �r` rG �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 rN rP r@ r r2 rQ 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 rN rP rA rQ r �r` rG rQ 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 r? �x r> r )r rN rP rC r �uint32rQ 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 )NrB r> r �r r )r rN rP r@ rQ r rs 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 rp �r r$ rG Z permutationr �any�r` ZoriginalZpermutedr 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 )rG Zbetar rr r r& r �r` Zvalsr r r � test_beta s &(z RNG.test_betac C s | j �d�}tt|�dk� d S rx )rG �bytesr rr r� r r r � test_bytes# s zRNG.test_bytesc C s. | j �dd�}tt|�dk� t| j j� d S �Nr r )rG Z chisquarer rr r/ r� r r r �test_chisquare' s zRNG.test_chisquarec C s. | j �dd�}tt|�dk� t| j j� d S r� )rG � exponentialr rr r/ r� r r r �test_exponential, s zRNG.test_exponentialc C s$ | j �ddd�}tt|�dk� d S )N� r� r )rG r r rr r� r r r �test_f1 s z RNG.test_fc C s$ | j �ddd�}tt|�dk� d S )Nr� r# r )rG Zgammar rr r� r r r � test_gamma5 s zRNG.test_gammac C s2 | j �dd�}tt|�dk� t| j jdd� d S )Nr! r T�r( )rG Z geometricr rr r/ r� r� r r r �test_geometric9 s zRNG.test_geometricc C s$ | j �ddd�}tt|�dk� d S r� )rG Zgumbelr rr r� r r r � test_gumbel> s zRNG.test_gumbelc C s$ | j �ddd�}tt|�dk� d S r� )rG Zlaplacer rr r� r r r � test_laplaceB s zRNG.test_laplacec C s$ | j �ddd�}tt|�dk� d S r� )rG Zlogisticr rr r� r r r � test_logiticF s zRNG.test_logiticc C s" | j �dd�}tt|�dk� d S )Nr! r )rG Z logseriesr rr r� r r r �test_logseriesJ s zRNG.test_logseriesc C s$ | j �ddd�}tt|�dk� d S �Nr 皙�����?)rG Znegative_binomialr rr r� r r r �test_negative_binomialN s zRNG.test_negative_binomialc C s$ | j �ddd�}tt|�dk� d S )Nr r# )rG Znoncentral_chisquarer rr r� r r r �test_noncentral_chisquareR 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 )rG Z noncentral_fr rr r r& r� r r r �test_noncentral_fV s zRNG.test_noncentral_fc C s$ | j �ddd�}tt|�dk� d S r� )rG Znormalr rr r� r r r � test_normal` s zRNG.test_normalc C s" | j �dd�}tt|�dk� d S )Nr r )rG Zparetor rr r� r r r � test_paretod s zRNG.test_paretoc C sV | j �dd�}tt|�dk� | j �t�dgd ��}tt|�dk� t| j j� d S rx )rG Zpoissonr rr r r& r/ r� r r r � test_poissonh s zRNG.test_poissonc C s" | j �dd�}tt|�dk� d S )Nr� r )rG Zpowerr rr r� r r r � test_powero s zRNG.test_powerc C s$ | j �ddd�}tt|�dk� d S )Nr r= )rG rC r rr r� r r r � test_integerss s zRNG.test_integersc C s2 | j �dd�}tt|�dk� t| j jdd� d S )Nr� r Tr� )rG Zrayleighr rr r/ r� r r r � test_rayleighw s zRNG.test_rayleighc C s$ | j �ddd�}tt|�dk� d S r� )rG Zvonmisesr rr r� r r r � test_vonmises| s zRNG.test_vonmisesc C s$ | j �ddd�}tt|�dk� d S �N� �?r )rG Zwaldr rr r� r r r � test_wald� s z RNG.test_waldc C s" | j �dd�}tt|�dk� d S r� )rG Zweibullr rr 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 rY )r rY rX )rG Zzipfr rr rZ 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 )rG 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 )rG 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 rY � )r� r# )rG Zmultivariate_normalr r r� )r` 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 )NrY gUUUUUU�?gUUUUUU�?)r# r r r" )rG 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� )rG Z dirichletr r )r` �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�dumpsrG �loadsr r3 r2 rN rQ )r` ZpickZunpickr r r � test_pickle� s � �zRNG.test_picklec C sX | j d kr"| jj}t�d�|�� | j dkr4tj}ntj}tj dg|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 {0}� r r> � r i� r# )rS rN re rf rg rh r r� rD r& rQ r r2 r$ �mod)r` ri r rP Zbgr7 r8 r r r �test_seed_array� sH � �� 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 )NrK r< r> ) r rN rI rQ r@ r rB r r r r r2 �r` rG rQ 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 rN rI rQ rE r rB r r r r r2 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 rN rI rQ rA r rB r r r r r2 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� ) rG rN rQ r �emptyrA r r rB )r` rG rQ 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� )rG rN rQ r r� r@ r rB �r` rG rQ r r� r� r r r �test_output_filling_uniform1 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� )rG rN rQ r r� r} r rB r� r r r � test_output_filling_exponentialC 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� )rG rN rQ r �zerosrE r rB r� r r r �test_output_filling_gammaU 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� ) rG rN rQ r r$ r� rE r rB )r` rG rQ r Zmur� r� r r r �#test_output_filling_gamma_broadcastg s z'RNG.test_output_filling_gamma_broadcastc C sp | j }d}t�|�}t�t�� |j|tjd� W 5 Q R X t�t�� |j|d d d� d� W 5 Q R X tj|tjd�}t�t�� |j|tj d� W 5 Q R X tj |tjd�}t�t�� |j d|tj d� W 5 Q R X t�t��" |j d|d d d� tjd� W 5 Q R X tj |tj d�}t�t�� |j d|tjd� W 5 Q R X t�t�� |j d|d d d� d� W 5 Q R X d S )Nr� r� r� r� r> r� ) rG r r� rf �raisesr] rA rB � ValueErrorrF r� rE )r` rG r r� r r r �test_output_fill_errorz s( & zRNG.test_output_fill_errorc C sh |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 j 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' �minra rG rC r r& �object) r` r �upper�lower�infor) r* r+ r, r- r r r �test_integers_broadcast� sH � � � zRNG.test_integers_broadcastc C s� t �dg�}t �dg�}| jj|||d�}|jdks8t�| jj|d ||d�}|jdks\t�| jj||d |d�}|jdks�t�d S )Nr r r> )r )r r& rG rC r r� )r` r ZhighZlowr� r r r �test_integers_numpy� s zRNG.test_integers_numpyc C s� |t jkrd}d}n t �|�}t|j�d }|j}t�t��" | j j ||d gd |d� W 5 Q R X t�t��" | j j |d |gd |d� W 5 Q R X t�t��$ | j j |d g|gd |d� W 5 Q R X t�t�� | j j dgdg|d� W 5 Q R X d S )Nr# r r r r> ) r r� r� r� r' r� rf r� r� rG rC )r` r r� r� r� r r r � test_integers_broadcast_errors� s & & ( z"RNG.test_integers_broadcast_errorsN)Jre � __module__� __qualname__� classmethodrW rT ra rb rj rm rv rw ry 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 rJ _ s� ' ! rJ c @ s e Zd Zedd� �Zdd� ZdS )� TestMT19937c C sJ t | _d | _dg| _t| j| j� �| _| jjj| _d| _| � � t | _ d S )Ni! ! r� ) r rN rO rP r rG rQ rR rS rT r� r^ rU r r r rW � 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 rQ r9 r# �pos) r r@ Z RandomStaterA Z get_staterG rN rQ r rs )r` ZnprgrQ r8 r r r �test_numpy_state� s z TestMT19937.test_numpy_stateN)re r� r� r� rW 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 rK rL ) r rN rO rP r rG rQ rR rS rT rU r r r rW � s zTestPhilox.setup_classN�re r� r� r� rW 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 )NrK � ) r rN rO rP r rG rQ rR rS rT rU r r r rW � 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 rM rU r r r rW s zTestPCG64.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 rN rO rP r r@ � default_rngrG rQ rR rS rT rU r r r rW s zTestDefaultRNG.setup_classc C s t t| jjt�� d S r )r r0 rG rN r r_ 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 5 Q R X t�t�� t j�ddg� W 5 Q R X d S )NrK r l -�tE+�I�6�%�Yl l ) ��~�eR :TI*�6� rp )r r@ r rf r� r� r_ r r r r� s � zTestDefaultRNG.test_seedN)re r� r� r� rW r r� r r r r r s r )F)N)'r� �time� functoolsr Znumpyr rf Z numpy.testingr r r Z numpy.randomr r r r r Zfixturer� Zint8Zint16Zint32Zint64Zuint8Zuint16r� rD r r r/ r2 rI r� rJ r� r� r r r r r r r �<module> s: �� ! v