U
|
P ±d B ã @ 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 ek
|
r¤ dZY nX zd dlZdZ W n ek
|
rÎ dZ Y nX 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$e0Z1G d%d& d&e0Z2G d'd( d(e0Z3G d)d* d*e0Z4G d+d, d,e0Z5G 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 sh | D ]^}t | | tr*t| | || qt | | tjrNt| | || q| | || kstqd S ©N)Ú
|
isinstanceÚdictÚassert_state_equalÚnpZndarrayr ÚAssertionError)ÚactualÚtargetÚkey© r úUd:\z\workplace\vscode\pyvenv\venv\Lib\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 td S )Né@ é5 r" )r+ r. r/ ÚNotImplementedError©r( Úbitsr r r Úuniform32_from_uintH s r5 c C s$ |dkrt | S |dkr t| S d S )N)r0 é? r1 r" )Úuniform_from_uint64Úuniform_from_uint32r3 r r r Úuniform_from_uintS s r9 c C s | t d¡? d S )Né g <)r r& r- r r r r7 Z s r7 c C s` t t| d ¡}tdt| dD ]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 r8 ^ s r8 c C s | tj¡d S )Nç ð?)Úviewr Údoubler- r r r Úuniform_from_dsfmtg s rH c
|
C sÜ |dkrt | }n|dkr$t| }nt| }g }d}d }}t||k rÐd}|dks\|dkrd|| d }d||d d }|| || }|d7 }qLt d t |¡ | ¡} | | | ¡ | | | ¡ q<|d |
S )
|
N)r0 r6 r" r ç r; rE g @r g À)r7 r8 rH r? r ÚsqrtÚlogÚappend)
|
r( Únr4 ZdoublesÚgaussÚlocÚx1Zx2Zr2Úfr r r Úgauss_from_uintk s&
|
|
|
rR c C s ddl m} m}m} ttdddd}| d¡ tf |j}t|j|j t|j |j t
|
t | t
|
t | | }t
|
t |j d t| d¡dkstd S )Nr )Ú ISeedSequenceÚISpawnableSeedSequenceÚSeedlessSeedSequenceé
|
)r r; r= )Z spawn_keyZ pool_size)Znumpy.random.bit_generatorrS rT rU r r@ ÚspawnÚstater Zn_children_spawnedr Ú TypeErrorr2 Zgenerate_stater? r )rS rT rU Ús1Ús2Údummyr r r Útest_seedsequence s
|
|
|
r] 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 )Nr0 )
|
r Ú bit_generatorr4 r r& r$ rY Ú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|| jddW 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ÚsplitrL rg rh r r' r$ )rd ÚfilenameÚcsvrm rn Ú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 )Nrm éè rn r )r_ Údata1Ú
|
random_rawr Údata2©Úselfr_ Zuintsr r r Útest_raw¯ s
|
|
z Base.test_rawc C sF | j | jd }|jdd}|d ks(t|jddd}|d ksBtd S )Nrm F)Úoutputrv )r_ rw rx r rz 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é rm rn )r r_ rw Ústandard_normalr rR r4 ry )r{ rM ÚrsrN 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 )Nrm rn ) r r_ rw r9 r4 Úrandomr? r r r$ r Úfloat64ry ©r{ r ÚvalsZuniformsr 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 )Nrm rn r# ) r r_ rw r5 r4 r r? r r r r r$ ry r
r r r Útest_uniform_floatÜ s
|
|
zBase.test_uniform_floatc C sL t | j| jd }dt|ks$tt|d ¡ dd¡t|ksHtd S ©Nrm r z#xÚXr( )r r_ rw Úreprr Úidr) Úreplace©r{ r r r r Ú test_repré s zBase.test_reprc C sd t | j| jd }dt|ks$tt| jjt|ks<tt|d ¡ dd¡t|ks`td S r ) r r_ rw Ústrr Ú__name__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 rm rv éd ) Úpickler_ rw rX ÚdumpsÚloadsr r r r r r r ) r{ r r_ rX Z
|
bitgen_pklZreloadedZreloaded_stateÚssZaar r r Ú test_pickleô s
|
|
ÿ
|
zBase.test_picklec C s2 | j | jd }t t¡ dh|_W 5 Q R X d S )Nrm Ú1)r_ rw ÚpytestÚraisesrY rX ©r{ r_ 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 )Nrm ZotherBitGeneratorr_ )r_ rw rX r r Ú
|
ValueError)r{ r_ rX 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 )r_ ra r r rY ©r{ r_ Ú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 )r_ rb 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 )Nrm r rG Úint32)r_ rw Z
|
_benchmarkr r r r r r r Útest_benchmark s
|
|
zBase.test_benchmarkzcffi not available)Úreasonc C s: | j | jd }|j}t|ts$t|j}||ks6td S ©Nrm )r_ rw Úcffir r r )r{ r_ Zcffi_interfaceZother_cffi_interfacer r r Ú test_cffi$ s
|
zBase.test_cffizctypes not availablec C s: | j | jd }|j}t|ts$t|j}||ks6td S r¨ )r_ rw Úctypesr r r )r{ r_ Zctypes_interfaceZother_ctypes_interfacer r r Ú test_ctypes, s
|
zBase.test_ctypesc C s, | j | jd }|j}| ¡ }t|| d S r¨ )r_ rw rX Ú __getstate__r )r{ r_ rX Z alt_stater r r Ú test_getstate4 s zBase.test_getstateN) r Ú
|
__module__Ú __qualname__r r& r$ ry rw Ú classmethodre ru r| r~ r r r r r r r r r¢ r¤ r¦ r ÚmarkZskipifÚ MISSING_CFFIrª ÚMISSING_CTYPESr¬ r® r r r r r^ s0
|
|
|
|
|
r^ c @ s e Zd Zedd Zdd ZdS )Ú
|
TestPhiloxc C sb t | _d| _tj| _| ttd¡| _ | ttd¡| _
|
t | _ g | _ ddd d dd d fg| _d S ) Nr0 z./data/philox-testset-1.csvz./data/philox-testset-2.csv)r Nr ©rl r; i r )r r_ r4 r r& r$ ru r Úpwdrw ry rY r` ra rb rc r r r re < s ÿÿzTestPhilox.setup_classc C sF | j | jd }|j}| j |d d |d d d}t|j|j d S )Nrm rX Úcounterr )r¸ r )r_ rw rX r )r{ r_ rX Zkeyedr r r Ú test_set_keyI s
|
ÿzTestPhilox.test_set_keyN)r r¯ r° r± re 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 )Nr0 z./data/pcg64-testset-1.csvz./data/pcg64-testset-2.csv©g @©r Nr¶ )r r_ r4 r r& r$ ru r r· rw ry r rY r` ra rb rc r r r re R s
|
zTestPCG64.setup_classc C s¢ t | j| jd }|jj}d}|j |¡ | d¡}||j_|j dd | ¡ | d¡}||j_|j ddd | ¡ | d¡}||kst||kstd S ©Nrm l÷ÿÿÿ PAOi_n¼ rV r; é ©r r_ rw rX ÚadvanceÚintegersr ©r{ r rX ÚstepZval_negZval_posZval_bigr r r Útest_advance_symmetry] s
|
|
|
z TestPCG64.test_advance_symmetryc C sl t | d¡}|j}|jd }d}|d |ks2t| tdd dD ¡ |jd }d}|d |kshtd S )Nì ï%@
|
rX ì Ír%4= KÆv±uÅ1¹/Ø c s s | ]}d | V qdS ©r; Nr ©ri rB r r r Ú <genexpr>r s z/TestPCG64.test_advange_large.<locals>.<genexpr>©é` r0 r" r, r é r; r l ÿ@H}.|z³KLfbe ©r r_ rX r rÀ Úsum©r{ r ZpcgrX Z initial_stateZadvanced_stater r r Útest_advange_largel s
|
|
z TestPCG64.test_advange_largeN©r r¯ r° r± re 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 )Nr0 z ./data/pcg64dxsm-testset-1.csvz ./data/pcg64dxsm-testset-2.csvr» r¼ r¶ )r
|
r_ r4 r r& r$ ru r r· rw ry r rY r` ra rb rc r r r re y s
|
zTestPCG64DXSM.setup_classc C s¢ t | j| jd }|jj}d}|j |¡ | d¡}||j_|j dd | ¡ | d¡}||j_|j ddd | ¡ | d¡}||kst||kstd S r½ r¿ r r r r rÄ s
|
|
|
z#TestPCG64DXSM.test_advance_symmetryc C sl t | d¡}|j}|j}d}|d d |ks2t| tdd dD ¡ |jd }d}|d |kshtd S )NrÅ rÆ rX c s s | ]}d | V qdS rÇ r rÈ r r r rÉ s z3TestPCG64DXSM.test_advange_large.<locals>.<genexpr>rÊ 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 r_ r4 r r r$ ru r r· rw ry r r` ra rb rc r r r re 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 rY r_ r r' Úpi)r{ 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 )Nrm i r_ rX r Úpos)r rI )r r_ rw rX rÁ r )r{ r r_ rX ZdesiredÚtupr r r r Útest_state_tuple³ s
|
|
ÿ
|
|
|
z TestMT19937.test_state_tupleN)r r¯ r° r± re 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 )Nr0 z./data/sfc64-testset-1.csvz./data/sfc64-testset-2.csvr» r¼ r¶ )r r_ r4 r r& r$ ru r r· rw ry r rY r` ra rb rc r r r re Ä s ÿÿ
|
zTestSFC64.setup_classN)r r¯ r° r± re r r r r rÙ Ã s rÙ c @ s e Zd Zdd Zdd ZdS )ÚTestDefaultRNGc C s4 dddddgffD ] }t | }t|jtstqd S )Nr r )éÒ rÛ i. )r r r_ r r )r{ ÚargsÚrgr r r Ú test_seedÓ s zTestDefaultRNG.test_seedc C sB t }t|}|j|ks tt|}||ks0t|j|ks>td S r )r r r_ r )r{ ÚbgrÝ Zrg2r r r Útest_passthroughØ s z TestDefaultRNG.test_passthroughN)r r¯ r° rÞ rà r r r r rÚ Ò s rÚ )7ÚosÚos.pathr ÚsysÚnumpyr 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/ r5 r9 r7 r8 rH rR r] r^ rµ rº rÒ rÓ rÙ rÚ r r r r Ú<module> sP ,
|
|
|
%''$
|
