OwlCyberSecurity - MANAGER
Edit File: test_direct.cpython-38.pyc
U �p�]�7 � @ s� d dl Z d dlmZ d dlZd dlmZmZmZm Z d dl Z d dl m Z m Z mZmZmZmZmZmZ d dlmZ zd dlZdZW n ek r� dZY nX zd dlZdZW n ek r� dZY nX e j� e j� e ��Z dd � Z d d � Z!d d � Z"dd� Z#dd� Z$dd� Z%dd� Z&dd� Z'dd� Z(dd� Z)d d � Z*G d d � d e+�Z,G d d!� d!e,�Z-G d"d#� d#e,�Z.G d$d%� d%e,�Z/G d&d'� d'e,�Z0G d(d)� d)e+�Z1dS )*� N)�join)� assert_equal�assert_allclose�assert_array_equal� assert_raises)� Generator�MT19937�PCG64�Philox� RandomState� SeedSequence�SFC64� default_rng)� interfaceFTc C sh | D ]^}t | | t�r*t| | || � qt | | tj�rNt| | || � q| | || kst�qd S �N)� isinstance�dict�assert_state_equal�npZndarrayr �AssertionError)�actual�target�key� r �@/usr/lib/python3/dist-packages/numpy/random/tests/test_direct.pyr s r c C sr t �| �} t j| t �d�? t jd�}t �d�}t j| |@ t jd�}t �||g��� }|t �d�? d }|�t j�S )N� ��dtype� �� � � �>)r �uint64�array�uint32Z column_stackZravel�astype�float32)�x�upper�lowerZjoined�outr r r �uniform32_from_uint64* s r* c C sF t �| �t �d�? } t �| t �d�@ �} | t �d�? d }|�t j�S )N� r r r )r r! r# r$ r% )r& r) r r r �uniform32_from_uint534 s r, c C s | t �d�? d S )Nr r )r r# �r& r r r �uniform32_from_uint32; s r. c C s8 |dkrt | �S |dkr t| �S |dkr0t| �S t�d S )N�@ �5 r )r* r, r. �NotImplementedError�r& �bitsr r r �uniform32_from_uint? s r4 c C s$ |dkrt | �S |dkr t| �S d S )N)r/ �? r0 r )�uniform_from_uint64�uniform_from_uint32r2 r r r �uniform_from_uintJ s r8 c C s | t �d�? d S )N� g �<)r r! r- r r r r6 Q s r6 c C s` t �t| �d �}tdt| �d�D ]8}| | d? }| |d d? }|d | d ||d < q"|S )N� r � � � g �Ag @C)r �empty�len�range)r& r) �i�a�br r r r7 U s r7 c C s | � tj�d S )N� �?)Zviewr �doubler- r r r �uniform_from_dsfmt^ s rF c C s� |dkrt | �}n|dkr$t| �}nt| �}g }d}d }}t|�|k r�d}|dks\|dkr�d|| d }d||d d }|| || }|d7 }qLt�d t�|� | �} |�| | � |�| | � q<|d |� S ) N)r/ r5 r r � r: rD g @r< g �)r6 r7 rF r? r Zsqrt�log�append) r& �nr3 Zdoubles�gaussZlocZx1Zx2Zr2�fr r r �gauss_from_uintb s& rM c C s� ddl m} m}m} ttd�ddd�}|�d� tf |j�}t|j|j� t|j |j � t t | � t t |� |� }t t |j d� t|�d��dks�t�d S )Nr )� ISeedSequence�ISpawnableSeedSequence�SeedlessSeedSequence� )r< r: r= )Z spawn_keyZ pool_size)Znumpy.random.bit_generatorrN rO rP r r@ Zspawn�stater Zn_children_spawnedr � TypeErrorr1 Zgenerate_stater? r )rN rO rP �s1�s2Zdummyr r r �test_seedsequencez s rV c @ s� e Zd ZejZi ZZedd� �Z edd� �Z dd� Z dd� Z d d � Z d d � Zd d� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zejjed d �d!d"� �Z ejje d#d �d$d%� �Z d&d'� Z d(S ))�Basec C s* t | _d| _tj| _t| _g | _g | _ d S )Nr/ ) r � bit_generatorr3 r r! r rS �seed_error_type�invalid_init_types�invalid_init_values��clsr r r � setup_class� s zBase.setup_classc C s� t |��x}|�� }|�d�}dd� |dd � D �}g }|D ]"}|�t|�d�d �� d�� q:|tj|| jd�d�W 5 Q R � S Q R X d S ) N�,c S s g | ]}t |�� d ��qS )r )�int�strip)�.0�sr r r � <listcomp>� s z"Base._read_csv.<locals>.<listcomp>r< ���r r )�seed�data) �open�readline�splitrI r` ra r r" r )r] �filenameZcsvrf rg �liner r r � _read_csv� s zBase._read_csvc C s� | j | jd � }|�d�}t|| jd � | j | jd � }|�� }t|| jd d � | j | jd � }|�d�}t|| jd � d S )Nrf �� rg r )rX �data1� random_rawr �data2��selfrX Zuintsr r r �test_raw� s z Base.test_rawc C sF | j | jd � }|jdd�}|d ks(t�|jddd�}|d ksBt�d S )Nrf F)�outputrn )rX ro rp r rr r r r �test_random_raw� s zBase.test_random_rawc C sx d}t | j| jd � �}|�|�}t|t| jd || j�� t | j| jd � �}|�d�}t|t| jd || j�� d S )N� rf rg )r rX ro �standard_normalr rM r3 rq )rs rJ �rsrK r r r �test_gauss_inv� s � �zBase.test_gauss_invc C s� t | j| jd � �}t| jd | j�}|�t|��}t||� t|j t j � t | j| j d � �}t| j d | j�}|�t|��}t||� t|j t j � d S )Nrf rg ) r rX ro r8 r3 �randomr? r r r r Zfloat64rq �rs ry ZvalsZuniformsr r r �test_uniform_double� s zBase.test_uniform_doublec C s� t | j| jd � �}t| jd | j�}|jt|�tjd�}t ||� t |j tj� t | j| j d � �}t| j d | j�}|jt|�tjd�}t ||� t |j tj� d S )Nrf rg r ) r rX ro r4 r3 r{ r? r r% r r r rq r| r r r �test_uniform_float� s zBase.test_uniform_floatc C sN t | j| jd � �}dt|�ks$t�d�t|���� �dd�t|�ksJt�d S �Nrf r z{:#x}�Xr& ) r rX ro �reprr �format�idr' �replace�rs ry r r r � test_repr� s zBase.test_reprc C sf t | j| jd � �}dt|�ks$t�t| jj�t|�ks<t�d�t|���� � dd�t|�ksbt�d S r ) r rX ro �strr �__name__r� r� r' r� r� r r r �test_str� s z Base.test_strc C s� dd l }| j| jd � }|j}|�|�}|�|�}|j}tt|��d�t|��d�� ||k sbt �t ||� t d�}|�|�|��}t |j|j� d S )Nr rf rn �d ) �picklerX ro rR �dumps�loadsr r rx r r r r ) rs r� rX rR Z bitgen_pklZreloadedZreloaded_stateZssZaar r r � test_pickle� s � zBase.test_picklec C s2 | j | jd � }t�t�� dh|_W 5 Q R X d S )Nrf �1)rX ro �pytest�raisesrS rR �rs rX r r r �test_invalid_state_type� s z Base.test_invalid_state_typec C s> | j | jd � }|j}d|d< t�t�� ||_W 5 Q R X d S )Nrf ZotherBitGeneratorrX )rX ro rR r� r� � ValueError)rs rX rR r r r �test_invalid_state_value s z Base.test_invalid_state_valuec C s4 | j }| jD ]"}t�t�� ||� W 5 Q R X q d S r )rX rZ r� r� rS �rs rX �str r r �test_invalid_init_type s zBase.test_invalid_init_typec C s8 | j }| jD ]&}t�ttf�� ||� W 5 Q R X q d S r )rX r[ r� r� r� � OverflowErrorr� r r r �test_invalid_init_values s z Base.test_invalid_init_valuesc C sL | j | jd � }|�d� |�dd� t�t�� |�dd� W 5 Q R X d S )Nrf r< rE Zint32)rX ro Z _benchmarkr� r� r� r� r r r �test_benchmark s zBase.test_benchmarkzcffi not available)�reasonc C s: | j | jd � }|j}t|t�s$t�|j}||ks6t�d S �Nrf )rX ro �cffir r r )rs rX Zcffi_interfaceZother_cffi_interfacer r r � test_cffi s zBase.test_cffizctypes not availablec C s: | j | jd � }|j}t|t�s$t�|j}||ks6t�d S r� )rX ro �ctypesr r r )rs rX Zctypes_interfaceZother_ctypes_interfacer r r � test_ctypes" s zBase.test_ctypesc C s, | j | jd � }|j}|�� }t||� d S r� )rX ro rR � __getstate__r )rs rX rR Z alt_stater r r � test_getstate* s zBase.test_getstateN) r� � __module__� __qualname__r r! r rq ro � classmethodr^ rm rt rv rz r} r~ r� r� r� r� r� r� r� r� r� ZmarkZskipif� MISSING_CFFIr� �MISSING_CTYPESr� r� r r r r rW � s0 rW c @ s e Zd Zedd� �Zdd� ZdS )� TestPhiloxc C sb t | _d| _tj| _| �ttd��| _ | �ttd��| _ t | _ g | _ ddd d dd d fg| _d S ) Nr/ z./data/philox-testset-1.csvz./data/philox-testset-2.csv)r< Nr< �re r: i r< )r rX r3 r r! r rm r �pwdro rq rS rY rZ r[ r\ r r r r^ 2 s ��zTestPhilox.setup_classc C sF | j | jd � }|j}| j |d d |d d d�}t|j|j� d S )Nrf rR �counterr )r� r )rX ro rR r )rs rX rR Zkeyedr r r � test_set_key? s �zTestPhilox.test_set_keyN)r� r� r� r� r^ r� r r r r r� 1 s r� c @ s e Zd Zedd� �Zdd� ZdS )� TestPCG64c C s^ t | _d| _tj| _| �ttd��| _ | �ttd��| _ t t f| _ dd gfdg| _dg| _d S )Nr/ z./data/pcg64-testset-1.csvz./data/pcg64-testset-2.csv�g������ @�r< Nr� )r rX r3 r r! r rm r r� ro rq r� rS rY rZ r[ r\ r r r r^ H s zTestPCG64.setup_classc C s� t | j| jd � �}|jj}d}|j�|� |�d�}||j_|j�dd | � |�d�}||j_|j�ddd | � |�d�}||ks�t�||ks�t�d S )Nrf l���� PAOi_n�� rQ r: � )r rX ro rR Zadvance�integersr )rs ry rR �stepZval_negZval_posZval_bigr r r �test_advance_symmetryS s z TestPCG64.test_advance_symmetryN)r� r� r� r� r^ r� r r r r r� G s r� c @ s( e Zd Zedd� �Zdd� Zdd� ZdS )� TestMT19937c C sP t | _d| _tj| _| �ttd��| _ | �ttd��| _ t | _ g | _ dg| _d S )Nr z ./data/mt19937-testset-1.csvz ./data/mt19937-testset-2.csvr� )r rX r3 r r# r rm r r� ro rq r� rY rZ r[ r\ r r r r^ d s zTestMT19937.setup_classc C s� t t| jt�tjg�� t t| jt�tj g�� t t| jt�tjtj g�� t t| jt�dtjg�� t t| jtjg� t t| jdtjg� d S )Nr )r rS rX r r" Zpi)rs r r r �test_seed_float_arrayo s z!TestMT19937.test_seed_float_arrayc C s� t | j| jd � �}|j}|j}|�d�}|d |d d |d d f}||_|�d�}t||� |d }||_|�d�}t||� d S )Nrf i rX rR r �pos)r rG )r rX ro rR r� r )rs ry rX rR Zdesired�tupr r r r �test_state_tuplew s � z TestMT19937.test_state_tupleN)r� r� r� r� r^ r� r� r r r r r� c s r� c @ s e Zd Zedd� �ZdS )� TestSFC64c C s^ t | _d| _tj| _| �ttd��| _ | �ttd��| _ t t f| _ dd gfdg| _dg| _d S )Nr/ z./data/sfc64-testset-1.csvz./data/sfc64-testset-2.csvr� r� r� )r rX r3 r r! r rm r r� ro rq r� rS rY rZ r[ r\ r r r r^ � s �� zTestSFC64.setup_classN)r� r� r� r� r^ r r r r r� � s r� c @ s e Zd Zdd� Zdd� ZdS )�TestDefaultRNGc C s4 dddddgffD ] }t |� }t|jt�st�qd S )Nr r )�� r� i. )r r rX r r )rs �args�rgr r r � test_seed� s zTestDefaultRNG.test_seedc C sB t � }t|�}|j|ks t�t|�}||ks0t�|j|ks>t�d S r )r r rX r )rs Zbgr� Zrg2r r r �test_passthrough� s z TestDefaultRNG.test_passthroughN)r� r� r� r� r� r r r r r� � s r� )2�os�os.pathr Znumpyr Z numpy.testingr r r r r� Z numpy.randomr r r r r r r r Znumpy.random.commonr r� r� � ImportErrorr� r� �path�dirname�abspath�__file__r� r r* r, r. r4 r8 r6 r7 rF rM rV �objectrW r� r� r� r� r� r r r r �<module> sF ( % $