3
���hR8 � @ s� d dl Z d dlmZ d dlZd dlZd dlmZmZm Z m
Z
d dl Z d dl m
Z
mZmZmZmZmZmZmZ d dlmZ yd dlZdZW n ek
r� dZY nX yd dlZdZW n ek
r� dZY nX ej j dkr�dZe j j e j 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-G d d � d �Z.G d!d"� d"e.�Z/G d#d$� d$e.�Z0G d%d&� d&e.�Z1G d'd(� d(e.�Z2G d)d*� d*�Z3dS )+� N)�join)� assert_equal�assert_allclose�assert_array_equal�
assert_raises)� Generator�MT19937�PCG64�Philox� RandomState� SeedSequence�SFC64� default_rng)� interfaceFT� c C sl xf| D ]^}t | | t�r,t| | || � qt | | tj�rPt| | || � q| | || kst�qW d S )N)�
isinstance�dict�assert_state_equal�npZndarrayr �AssertionError)�actual�target�key� r �?/tmp/pip-build-5_djhm0z/numpy/numpy/random/tests/test_direct.pyr ' s
r c C sr t j| �} t j| t jd�? t jd�}t jd�}t j| |@ t jd�}t j||g�j� }|t jd�? d }|jt j�S )
N� )�dtypel �� � g �?� � i � g �>)r �uint64�array�uint32Z column_stackZravel�astype�float32)�x�upper�lowerZjoined�outr r r �uniform32_from_uint641 s
r) c C sF t j| �t jd�? } t j| t jd�@ �} | t jd�? d }|jt j�S ) N� l �� r g �?r r i � g �>)r r r" r# r$ )r% r( r r r �uniform32_from_uint53; s r+ c C s | t jd�? d S )Nr g �?r r i � g �>)r r" )r% r r r �uniform32_from_uint32B 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_uintF s r1 c C s$ |dkrt | �S |dkr t| �S d S )Nr- �? r. r )r- r2 r. )�uniform_from_uint64�uniform_from_uint32)r% r0 r r r �uniform_from_uintQ s r5 c C s | t jd�? d S )N� g �?g @Cg �<)r r )r% r r r r3 X s r3 c C sd t jt| �d �}xLtdt| �d�D ]8}| | d? }| |d d? }|d | d ||d < q$W |S )Nr r � r � g �Ag @C)r �empty�len�range)r% r( �i�a�br r r r4 \ s r4 c C s | j tj�d S )Ng �?)�viewr �double)r% r r r �uniform_from_dsfmte s rA c
C s� |d
krt | �}n|dkr$t| �}nt| �}g }d}d }}x�t|�|k r�d}xP|dks`|dkr�d|| d }d||d d }|| || }|d7 }qPW tjd tj|� | �} |j| | � |j| | � q>W |d |� S ) Nr- r2 r r g r g �?g @r )r- r2 g �)r3 r4 rA r: r �sqrt�log�append)
r% �nr0 Zdoubles�gauss�locZx1Zx2Zr2�fr r r �gauss_from_uinti s&
rI c C s� ddl m} m}m} ttd�ddd�}|jd� tf |j�}t|j|j� t|j |j � t
t | � t
t |� |� }t
t |j
d� t|jd��dks�t�d S ) Nr )�
ISeedSequence�ISpawnableSeedSequence�SeedlessSeedSequence�
r r r8 )Z spawn_keyZ pool_size)r r )Znumpy.random.bit_generatorrJ rK rL r r; �spawn�stater Zn_children_spawnedr � TypeErrorr/ Zgenerate_stater: r )rJ rK rL �s1�s2�dummyr r r �test_seedsequence� s
rT 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_generatorr0 r r r rP �seed_error_type�invalid_init_types�invalid_init_values)�clsr r r � setup_class� s zBase.setup_classc C s� t |��p}|j� }|jd�}dd� |dd � D �}g }x*|D ]"}|jt|jd�d j� d�� q<W |tj|| jd�d�S Q R X d S ) N�,c S s g | ]}t |j� d ��qS )r )�int�strip)�.0�sr r r �
<listcomp>� s z"Base._read_csv.<locals>.<listcomp>r r )r )�seed�data���) �open�readline�splitrD r] r^ r r! r )rZ �filename�csvrb rc �liner r r � _read_csv� s
"zBase._read_csvc C s� | j | jd � }|jd�}t|| jd � | j | jd � }|j� }t|| jd d � | j | jd � }|jd�}t|| jd � d S )Nrb i� rc r )rV �data1�
random_rawr �data2)�selfrV �uintsr r r �test_raw� s
z
Base.test_rawc C sF | j | jd � }|jdd�}|d ks(t�|jddd�}|d ksBt�d S )Nrb F)�outputi� )rV rl rm r )ro rV rp r r r �test_random_raw� s
zBase.test_random_rawc C sx d}t | j| jd � �}|j|�}t|t| jd || j�� t | j| jd � �}|jd�}t|t| jd || j�� d S )N� rb rc )r rV rl �standard_normalr rI r0 rn )ro rE �rsrF r r r �test_gauss_inv� s
zBase.test_gauss_invc C s� t | j| jd � �}t| jd | j�}|jt|��}t||� t|j t
j � t | j| j d � �}t| j d | j�}|jt|��}t||� t|j t
j � d S )Nrb rc )
r rV rl r5 r0 �randomr: r r r r Zfloat64rn )ro rv �vals�uniformsr 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 )Nrb rc )r )
r rV rl r1 r0 rx r: r r$ r r r rn )ro rv ry rz r r r �test_uniform_float� s
zBase.test_uniform_floatc C sN t | j| jd � �}dt|�ks$t�djt|��j� jdd�t|�ksJt�d S )Nrb r z{:#x}�Xr% ) r rV rl �reprr �format�idr& �replace)ro rv r r r � test_repr� s zBase.test_reprc C sf t | j| jd � �}dt|�ks$t�t| jj�t|�ks<t�djt|��j� j dd�t|�ksbt�d S )Nrb r z{:#x}r} r% )
r rV rl �strr �__name__r r� r& r� )ro rv r r r �test_str� s z
Base.test_strc C s� dd l }| j| jd � }|j}|j|�}|j|�}|j}tt|�jd�t|�jd�� ||k sbt �t
||� t d�}|j|j|��}t |j|j� d S )Nr rb i� �d )
�picklerV rl rO �dumps�loadsr r ru r r r r ) ro r� rV rO Z
bitgen_pklZreloadedZreloaded_state�ssZaar r r � test_pickle� s
zBase.test_picklec C s2 | j | jd � }tjt�� dh|_W d Q R X d S )Nrb �1)rV rl �pytest�raisesrP rO )ro rV r r r �test_invalid_state_type s z Base.test_invalid_state_typec C s> | j | jd � }|j}d|d< tjt�� ||_W d Q R X d S )Nrb ZotherBitGeneratorrV )rV rl rO r� r� �
ValueError)ro rV rO r r r �test_invalid_state_value s
z Base.test_invalid_state_valuec
C s8 | j }x,| jD ]"}tjt�� ||� W d Q R X qW d S )N)rV rX r� r� rP )ro rV �str r r �test_invalid_init_type s zBase.test_invalid_init_typec
C s< | j }x0| jD ]&}tjttf�� ||� W d Q R X qW d S )N)rV rY r� r� r� �
OverflowError)ro rV r� r r r �test_invalid_init_values s z Base.test_invalid_init_valuesc
C sL | j | jd � }|jd� |jdd� tjt�� |jdd� W d Q R X d S )Nrb r r@ Zint32)rV rl Z
_benchmarkr� r� r� )ro rV 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 )Nrb )rV rl �cffir r r )ro rV 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 )Nrb )rV rl �ctypesr r r )ro rV Zctypes_interfaceZother_ctypes_interfacer r r � test_ctypes) s
zBase.test_ctypesc C s, | j | jd � }|j}|j� }t||� d S )Nrb )rV rl rO � __getstate__r )ro rV rO Z alt_stater r r �
test_getstate1 s zBase.test_getstateN) r� �
__module__� __qualname__r r r rn rl � classmethodr[ rk rq rs rw r{ r| r� r� r� r� r� r� r� r� r� �markZskipif� MISSING_CFFIr� �MISSING_CTYPESr� r� r r r r rU � s(
rU c @ s e Zd Zedd� �Zdd� ZdS )�
TestPhiloxc C sb t | _d| _tj| _| jttd��| _ | jttd��| _
t | _ g | _
dd d d dd d fg| _d S )
Nr- z./data/philox-testset-1.csvz./data/philox-testset-2.csvr r i )r Nr rd )rd )r
rV r0 r r r rk r �pwdrl rn rP rW rX rY )rZ r r r r[ 9 s zTestPhilox.setup_classc C sF | j | jd � }|j}| j |d d |d d d�}t|j|j� d S )Nrb rO �counterr )r� r )rV rl rO r )ro rV rO Zkeyedr r r � test_set_keyF s
zTestPhilox.test_set_keyN)r� r� r� r� r[ r� r r r r r� 8 s
r� c @ s e Zd Zedd� �Zdd� ZdS )� TestPCG64c C s^ t | _d| _tj| _| jttd��| _ | jttd��| _
t t f| _
dd gfdg| _d g| _d S )
Nr- z./data/pcg64-testset-1.csvz./data/pcg64-testset-2.csv皙���� @r )r� )r Nrd )rd )r rV r0 r r r rk r r� rl rn r� rP rW rX rY )rZ r r r r[ O s
zTestPCG64.setup_classc C s� t | j| jd � �}|jj}d}|jj|� |jd�}||j_|jjdd | � |jd�}||j_|jjddd | � |jd�}||ks�t�||ks�t�d S )Nrb l PAOi_n�� rM r � l���� PAOi_n�� )r rV rl rO Zadvance�integersr )ro rv rO �stepZval_negZval_posZval_bigr r r �test_advance_symmetryZ s
z TestPCG64.test_advance_symmetryN)r� r� r� r� r[ r� r r r r r� N s r� c @ s( e Zd Zedd� �Zdd� Zdd� ZdS )� TestMT19937c C sP t | _d| _tj| _| jttd��| _ | jttd��| _
t | _ g | _
dg| _d S )Nr z ./data/mt19937-testset-1.csvz ./data/mt19937-testset-2.csvr rd )rd )r rV r0 r r" r rk r r� rl rn r� rW rX rY )rZ r r r r[ k s zTestMT19937.setup_classc C s� t t| jtjtjg�� t t| jtjtj g�� t t| jtjtjtj g�� t t| jtjdtjg�� t t| jtjg� t t| jdtjg� d S )Nr )r rP rV r r! �pi)ro r r r �test_seed_float_arrayv s z!TestMT19937.test_seed_float_arrayc C s� t | j| jd � �}|j}|j}|jd
�}|d |d d |d d f}||_|jd �}t||� |d }||_|jd
�}t||� d S )Nrb r r* rV rO r �posr � i i )r r� i )r rV rl rO r� r )ro rv rV rO 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� j s r� c @ s e Zd Zedd� �ZdS )� TestSFC64c C s^ t | _d| _tj| _| jttd��| _ | jttd��| _
t t f| _
dd gfdg| _d g| _d S )
Nr- z./data/sfc64-testset-1.csvz./data/sfc64-testset-2.csv皙���� @r )r� )r Nrd )rd )r
rV r0 r r r rk r r� rl rn r� rP rW rX rY )rZ 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 s8 x2f ddddgfgD ] }t |� }t|jt�st�qW d S )N�� i. )N)r� )r r rV r r )ro �args�rgr r r � test_seed� s zTestDefaultRNG.test_seedc C sB t � }t|�}|j|ks t�t|�}||ks0t�|j|ks>t�d S )N)r
r rV r )ro �bgr� Zrg2r r r �test_passthrough� s z TestDefaultRNG.test_passthroughN)r� r� r� r� r� r r r r r� � s r� )4�os�os.pathr �sysZnumpyr Z
numpy.testingr r r r r� Z numpy.randomr 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, r1 r5 r3 r4 rA rI rT rU r� r� r� r� r� r r r r �<module> sL (
% $ |