Current File : //lib64/python3.9/site-packages/numpy/random/tests/__pycache__/test_direct.cpython-39.opt-1.pyc
a
z[yc-@ � @ s� d dl Z d dlmZ d dlZd dlZd dlmZmZm Z m
Z
d dl Z d dl m
Z
mZmZmZmZmZmZmZmZ d dlmZ zd dlZdZW n ey� dZY n0 zd dlZdZ W n ey� dZ Y n0 ej j dkr�dZe j � e j �!e"��Z#d d
� Z$d d � Z%d
d� Z&dd� Z'dd� Z(dd� Z)dd� Z*dd� Z+dd� Z,dd � Z-d d � Z.d d � Z/G d!d"� d"�Z0G d#d$� d$e0�Z1G d%d&� d&e0�Z2G d'd(� d(e0�Z3G d)d*� d*e0�Z4G d+d,� d,e0�Z5G d-d.� d.�Z6dS )/� N)�join)� assert_equal�assert_allclose�assert_array_equal�
assert_raises) � Generator�MT19937�PCG64� PCG64DXSM�Philox� RandomState� SeedSequence�SFC64� default_rng)� interfaceFT� c C sT | D ]J}t | | t�r*t| | || � qt | | tj�rt| | || � qqd S �N)�
isinstance�dict�assert_state_equal�npZndarrayr )�actual�target�key� r �D/usr/lib64/python3.9/site-packages/numpy/random/tests/test_direct.pyr ' s r c C s | t �d�? d �t j�S )N� g p>)r �uint32Zastype�float32)�ur r r �uint32_to_float321 s r c C s\ t �| �} t j| t �d�? t jd�}t �d�}t j| |@ t jd�}t �||g��� }t|�S )N� ��dtype� �� )r �uint64�arrayr Z column_stackZravelr )�x�upper�lowerZjoinedr r r �uniform32_from_uint645 s
r* c C s0 t �| �t �d�? } t �| t �d�@ �} t| �S )N� r$ )r r% r r �r' r r r �uniform32_from_uint53> s r- c C s t | �S r )r r, r r r �uniform32_from_uint32D 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_uintH s r4 c C s$ |dv rt | �S |dkr t| �S d S )N)r/ �? r0 r! )�uniform_from_uint64�uniform_from_uint32r2 r r r �uniform_from_uintS s r8 c C s | t �d�? d S )N� g �<)r r% r, r r r r6 Z s r6 c C s` t �t| �d �}tdt| �d�D ]8}| | d? }| |d d? }|d | d ||d < q"|S )N� r � r � g �Ag @C)r �empty�len�range)r' �out�i�a�br r r r7 ^ s r7 c C s | � tj�d S )N� �?)�viewr �doubler, r r r �uniform_from_dsfmtg s rG c
C s� |dv rt | �}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 rG r> r Zsqrt�log�append)
r' �nr3 Zdoubles�gauss�locZx1Zx2Zr2�fr r r �gauss_from_uintk s&
rO c C s� ddl m} m}m} ttd�ddd�}|�d� tf i |j��}t|j|j� t|j |j � t
t | � t
t |� |� }t
t |j
d� d S )Nr )�
ISeedSequence�ISpawnableSeedSequence�SeedlessSeedSequence�
)r r: r< )Z spawn_keyZ pool_size)Znumpy.random.bit_generatorrP rQ rR r
r? Zspawn�stater Zn_children_spawnedr � TypeErrorr1 Zgenerate_state)rP rQ rR �s1�s2Zdummyr r r �test_seedsequence� s
rX 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# rU �seed_error_type�invalid_init_types�invalid_init_values��clsr r r � setup_class� s zBase.setup_classc C s� t |��z}|�� }|�d�}dd� |dd � D �}g }|D ]"}|�t|�d�d �� d�� q:|tj|| jd�d�W d � S 1 s�0 Y d S ) N�,c S s g | ]}t |�� d ��qS )r )�int�strip)�.0�sr r r �
<listcomp>� � z"Base._read_csv.<locals>.<listcomp>r ���r r" )�seed�data) �open�readline�splitrJ rb rc r r& r# )r_ �filenameZcsvri rj �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 )Nri �� rj r )rZ �data1�
random_rawr �data2��selfrZ Zuintsr r r �test_raw� s
z
Base.test_rawc C s. | j | jd � }|jdd�}|jddd�}d S )Nri F)�outputrq )rZ rr rs ru 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� ri rj )r rZ rr �standard_normalr rO r3 rt )rv rK �rsrL 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 )Nri rj )
r rZ rr r8 r3 �randomr> r r r# r Zfloat64rt �rv r| 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 )Nri rj r" )
r rZ rr r4 r3 r~ r> r r r r r# rt r r r r �test_uniform_float� s
zBase.test_uniform_floatc C s t | j| jd � �}d S �Nri �r rZ rr �rv r| r r r � test_repr� s zBase.test_reprc C s t | j| jd � �}d S 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�� t ||� t
d�}|�|�|��}t |j|j� d S )Nr ri rq �d ) �picklerZ rr rT �dumps�loadsr r r{ r r
r ) rv r� rZ rT Z
bitgen_pklZreloadedZreloaded_state�ssZaar r r � test_pickle� s
�
zBase.test_picklec C sF | j | jd � }t�t�� dh|_W d � n1 s80 Y d S )Nri �1)rZ rr �pytest�raisesrU rT �rv rZ r r r �test_invalid_state_type s z Base.test_invalid_state_typec C sR | j | jd � }|j}d|d< t�t�� ||_W d � n1 sD0 Y d S )Nri ZotherBitGeneratorrZ )rZ rr rT r� r� �
ValueError)rv rZ rT r r r �test_invalid_state_value
s
z Base.test_invalid_state_valuec C sH | j }| jD ]6}t�t�� ||� W d � q 1 s80 Y q d S r )rZ r\ r� r� rU �rv rZ �str r r �test_invalid_init_type s
zBase.test_invalid_init_typec C sL | j }| jD ]:}t�ttf�� ||� W d � q 1 s<0 Y q d S r )rZ r] r� r� r� �
OverflowErrorr� r r r �test_invalid_init_values s
z Base.test_invalid_init_valuesc C s` | j | jd � }|�d� |�dd� t�t�� |�dd� W d � n1 sR0 Y d S )Nri r rF Zint32)rZ rr Z
_benchmarkr� r� r� r� r r r �test_benchmark s
zBase.test_benchmarkzcffi not available)�reasonc C s | j | jd � }|j}|j}d S r� )rZ rr �cffi)rv rZ Zcffi_interfaceZother_cffi_interfacer r r � test_cffi$ s zBase.test_cffizctypes not availablec C s | j | jd � }|j}|j}d S r� )rZ rr �ctypes)rv rZ Zctypes_interfaceZother_ctypes_interfacer r r � test_ctypes, s zBase.test_ctypesc C s, | j | jd � }|j}|�� }t||� d S r� )rZ rr rT � __getstate__r )rv rZ rT Z alt_stater r r �
test_getstate4 s zBase.test_getstateN) �__name__�
__module__� __qualname__r r% r# rt rr � classmethodr` rp rw ry r} r� r� r� r� r� r� r� r� r� r� r� ZmarkZskipif� MISSING_CFFIr� �MISSING_CTYPESr� r� r r r r rY � s0
rY 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 �rh r: i r )r rZ r3 r r% r# rp r �pwdrr rt rU r[ r\ r] r^ r r r r` < s ��zTestPhilox.setup_classc C sF | j | jd � }|j}| j |d d |d d d�}t|j|j� d S )Nri rT �counterr )r� r )rZ rr rT r )rv rZ rT Zkeyedr r r � test_set_keyI s
�zTestPhilox.test_set_keyN)r� r� r� r� r` r� r r r r r� ; s
r� c @ s( e Zd Zedd� �Zdd� 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 rZ r3 r r% r# rp r r� rr rt r� rU r[ r\ r] r^ r r r r` R s
zTestPCG64.setup_classc C s� t | j| jd � �}|jj}d}|j�|� |�d�}||j_|j�dd | � |�d�}||j_|j�ddd | � |�d�}d S �Nri l���� PAOi_n�� rS r: � �r rZ rr rT �advance�integers�rv r| rT �stepZval_negZval_posZval_bigr r r �test_advance_symmetry] s
z TestPCG64.test_advance_symmetryc C sL t | �d��}|j}|jd }d}|�tdd� dD ��� |jd }d}d S )N� �%�@
rT � �r%4=
K�v�u�1�/� c s s | ]}d | V qdS �r: Nr �rd rA r r r � <genexpr>r rg z/TestPCG64.test_advange_large.<locals>.<genexpr>��` r/ r! r+ r � r: r l �@H}.|z�KLf�be �r rZ rT r� �sum�rv r| ZpcgrT Z
initial_stateZadvanced_stater r r �test_advange_largel s
z TestPCG64.test_advange_largeN�r� r� r� r� r` r� r� r r r r r� Q s
r� c @ s( e Zd Zedd� �Zdd� Zdd� ZdS )�
TestPCG64DXSMc C s^ t | _d| _tj| _| �ttd��| _ | �ttd��| _
t t f| _
dd gfdg| _dg| _d S )Nr/ z ./data/pcg64dxsm-testset-1.csvz ./data/pcg64dxsm-testset-2.csvr� r� r� )r
rZ r3 r r% r# rp r r� rr rt r� rU r[ r\ r] r^ r r r r` y s
zTestPCG64DXSM.setup_classc C s� t | j| jd � �}|jj}d}|j�|� |�d�}||j_|j�dd | � |�d�}||j_|j�ddd | � |�d�}d S r� r� r� r r r r� � s
z#TestPCG64DXSM.test_advance_symmetryc C sH t | �d��}|j}|j}d}|�tdd� dD ��� |jd }d}d S )Nr� r� c s s | ]}d | V qdS r� r r� r r r r� � rg z3TestPCG64DXSM.test_advange_large.<locals>.<genexpr>r� rT l oF� �[U mO1X�9}� r� r� r r r r� � s
z TestPCG64DXSM.test_advange_largeNr� r r r r r� x 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 rZ r3 r r r# rp r r� rr rt r� r[ r\ r] r^ r r r r` � 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 rU rZ r r& Zpi)rv r r r �test_seed_float_array� 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 )Nri i rZ rT r �pos)r rH )r rZ rr rT r� r )rv r| rZ rT Zdesired�tupr r r r �test_state_tuple� s
�
z TestMT19937.test_state_tupleN)r� r� r� r� r` r� r� r r r r r� � 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 rZ r3 r r% r# rp r r� rr rt r� rU r[ r\ 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 s$ dddddgffD ] }t |� }qd S )Nr r )�� r� i. )r )rv �args�rgr r r � test_seed� s zTestDefaultRNG.test_seedc C s t � }t|�}t|�}d S r )r r )rv �bgr� Zrg2r r r �test_passthrough� s z TestDefaultRNG.test_passthroughN)r� r� r� r� r� r r r r r� � s r� )7�osZos.pathr �sysZnumpyr Z
numpy.testingr r r r r� Z numpy.randomr r r r
r r r
r r Znumpy.random._commonr r� r� � ImportErrorr� r� �flags�optimize�path�dirname�abspath�__file__r� r r r* r- r. r4 r8 r6 r7 rG rO rX rY r� r� r� r� r� r� r r r r �<module> sP ,
%''$
Mini Shell