[~k7 r d dl Z d dlmZ d dlZd dlZd dlmZmZm Z m
Z
d dl Z d dl m
Z
mZmZmZmZmZmZmZmZ d dlmZ d dlZdZn
# e$ r dZY nw xY w d dlZdZ n
# e$ r dZ Y nw xY wej j dk rdZe j e j ! e" Z#d Z$d
Z%d Z&d Z'd
Z(d Z)d Z*d Z+d Z,d Z-d Z.d Z/d Z0d Z1 G d d Z2 G d de2 Z3 G d d e2 Z4 G d d e2 Z5 G d d e2 Z6 G d! d"e2 Z7 G d# d$ Z8dS )% N)join) assert_equalassert_allcloseassert_array_equal
assert_raises) GeneratorMT19937PCG64 PCG64DXSMPhilox RandomState SeedSequenceSFC64 default_rng) interfaceFT c " | D ]}t | | t r t | | || :t | | t j r t | | || w| | || k sJ d S N)
isinstancedictassert_state_equalnpndarrayr )actualtargetkeys r/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/numpy/random/tests/test_direct.pyr r ' s . . fSk4 ( ( . vc{F3K 8 8 8 8
s RZ
0
0 . vc{F3K 8 8 8 8#;&+-----
. . c p | t j d z dz t j S )N g p>)r uint32astypefloat32)us r uint32_to_float32r% 1 s*
")A,,
; / 7 7
C CCr c p t j | } t j | t j d z t j }t j d }t j | |z t j }t j ||g }t
| S )N dtype )r uint64arrayr! column_stackravelr% )xupperlowerjoineds r uniform32_from_uint64r3 5 s
! A HQ")B--'ry 9 9 9E Ij ! !E HQYbi 0 0 0E
_eU^
,
,
2
2
4
4F V $ $$r c t j | t j d z } t j | t j d z } t | S )N r* )r r+ r! r% r/ s r uniform32_from_uint53r7 > sH
! "
%A
!bi
+++,,A Q r c t | S r )r% r6 s r uniform32_from_uint32r9 D s Q r c |dk rt | S |dk rt | S |dk rt | S t )N@ 5 r' )r3 r7 r9 NotImplementedErrorr/ bitss r uniform32_from_uintr@ H sM rzz$Q'''
$Q'''
$Q'''!!r c V |dv rt | S |dk rt | S d S )N)r; ? r< r' )uniform_from_uint64uniform_from_uint32r> s r uniform_from_uintrE S s9 |"1%%%
"1%%%
r c 6 | t j d z dz S )N g <)r r+ r6 s r rC rC Z s
2 #; <>> )]+++) 3444 " "E%u';R@@@ u{{2 2 % % % % % %r c j t j } | j j d }fdt
d D }d |D |k sJ | j d }fdt
dd D }d |D |k sJ | d }fdt
dd D }d |D }||k sJ |d
|d k sJ d S )
z? Test spawning new generators and bit_generators directly.
rJ c & g | ]
}j |fz S rm rk .0rP seqs r
z+test_generator_spawning.. s" <<. s ( ( (AAK ( ( (r c & g | ]
}j |fz S rm rz r{ s r r~ z+test_generator_spawning.. s" @@@aS]aT)@@@r rj c & g | ]}|j j S rm )seed_seqrk )r| bgs r r~ z+test_generator_spawning.. s 4 4 4bBK ! 4 4 4r c & g | ]
}j |fz S rm rz r{ s r r~ z+test_generator_spawning.. s" AAAaS]aT)AAAr c 0 g | ]}|j j j S rm )
bit_generatorr rk )r| rngs r r~ z+test_generator_spawning.. s! KKK3##,6KKKr r r N)r randomr r r ro rN uniform)r new_ss
expected_keysnew_bgsnew_rngs
found_keysr} s @r test_generator_spawningr sS
)
!
!C
$C
YYq\\F<<<<588<<.FakeSeedSequencec . t j || S )Nr( )r zeros)selfn_wordsr) s r rs z;test_non_spawnable..FakeSeedSequence.generate_state s 8G5111 1r N)__name__
__module__ __qualname__r r! rs rm r r FakeSeedSequencer s- 02 2 2 2 2 2 2r r zThe underlying SeedSequence)matchrJ ) rn rg registerr r r pytestraisesrr ro r )rg r r s r test_non_spawnabler s 8888882 2 2 2 2 2 2 2 +,,,
)
0 0 2 2
3
3C y(E F F F !
y(E F F F # # """# # # # # # # # # # # # # # # # # #s$ +B
BB3CC "C c 8 e Zd Zej Zi xZZed Z ed Z
d Z d Z d Z
d Zd Zd Zd Zd
Zd Zd Zd
Zd Zd Zej ed d Z ej e d d Z d Z dS )Basec t | _ d| _ t j | _ t | _ g | _ g | _ d S )Nr; )
r
r r? r r+ r) rr seed_error_typeinvalid_init_typesinvalid_init_valuesclss r setup_classzBase.setup_class s8 !I '!# "$ r c t | 5 }| }| d }d |dd D }g }|D ]P}| t | d d d Q|t
j || j dcd d d S # 1 swxY w Y d S )N,c R g | ]$}t | d %S )r )intstrip)r| ss r r~ z"Base._read_csv.. s* 888!C 1%%888r r r r( )seeddata) openreadlinesplitr\ r r r r, r) )r filenamecsvr r lines r _read_csvzBase._read_csv s%
(^^ Ks<<>>D::c??D88tABBx888DD
A
A C
3 3 9 9 ; ;Q ? ?@@@@ "(4sy*I*I*IJJ K K K K K K K K K K K K K K K K K Ks B0C
CCc | j | j d }| d }t || j d | j | j d }| }t || j d d | j | j d }| d }t || j d d S )Nr r r )r data1
random_rawr data2r r uintss r test_rawz
Base.test_raw s **DJv,>?
((..UDJv .///**DJv,>?
((**UDJv .q 1222**DJv,>?
((..UDJv ./////r c | j | j d }| d }|J | dd }|J d S )Nr F)outputr )r r r r s r test_random_rawzBase.test_random_raw s\ **DJv,>?
(((66}}} ((e(<<}}}}}r c d}t | j | j d }| | }t |t | j d || j t | j | j d }| d }t |t | j d || j d S )N r r )r
r r standard_normalr re r? r )r r] rsr_ s r test_gauss_invzBase.test_gauss_inv s
++TZ-?@
A
A""1%%'
6(:AtyII K K K ++TZ-?@
A
A""2&&'
6(:AtyII K K K K Kr c R t | j | j d }t | j d | j }| t
| }t || t |j t j t | j | j d }t | j d | j }| t
| }t || t |j t j d S )Nr r )
r r r rE r? r rM r r r) r float64r r r valsuniformss r test_uniform_doublezBase.test_uniform_double s
)t)4:f+=>
?
? F!3TY??99SYY''$'''X^RZ000
)t)4:f+=>
?
? F!3TY??99SYY''$'''X^RZ00000r c t | j | j d }t | j d | j }| t
| t j }t || t |j t j t | j | j d }t | j d | j }| t
| t j }t || t |j t j d S )Nr r r( )
r r r r@ r? r rM r r# r r r) r r s r test_uniform_floatzBase.test_uniform_float s
)t)4:f+=>
?
?"4:f#5tyAA99SYYbj999$'''X^RZ000
)t)4:f+=>
?
?"4:f#5tyAA99SYYbj999$'''X^RZ00000r c t | j | j d }dt | v sJ t | d dd t | v sJ d S Nr r z#xXr/ )r r r repridr0 replacer r s r test_reprzBase.test_repr s{
)t)4:f+=>
?
?d2hh&&&&R&&~~##%%--c377488CCCCCCr c T t | j | j d }dt | v sJ t | j j t | v sJ t | d dd t | vsJ d S r )r r r strr r r0 r r s r test_strz
Base.test_str s
)t)4:f+=>
?
?c"gg%%%%4%.//3r77::::R&&~~##%%--c377s2wwFFFFFFr c dd l } | j | j d }|j }| | }| | }|j }t
t | d t | d ||usJ t || t d }| | | }t |j |j d S )Nr r r d ) pickler r rp dumpsloadsr r r r r r ) r r r rp
bitgen_pklreloadedreloaded_statessaas r test_picklezBase.test_pickle s
**DJv,>?
#\\-00
<<
++! 9]33CCDII$X..>>tDD F F F H,,,,>5111
#
\\&,,r**
+
+RXrx(((((r c | j | j d }t j t 5 dh|_ d d d d S # 1 swxY w Y d S )Nr 1)r r r r rr rp r r s r test_invalid_state_typez Base.test_invalid_state_type- s **DJv,>?
]9
%
% ( (#&%M ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( (s AA A c | j | j d }|j }d|d< t j t
5 ||_ d d d d S # 1 swxY w Y d S )Nr otherBitGeneratorr )r r rp r r
ValueError)r r rp s r test_invalid_state_valuez Base.test_invalid_state_value2 s **DJv,>?
#!4o
]:
&
& ( ("'M ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( (s AAAc | j }| j D ]8}t j t 5 || d d d n # 1 swxY w Y 9d S r )r r r r rr r r sts r test_invalid_init_typezBase.test_invalid_init_type9 s *
) # #B y))
#
#
r""
#
#
#
#
#
#
#
#
#
#
#
#
#
#
# # #s <A A c | j }| j D ]?}t j t t
f 5 || d d d n # 1 swxY w Y @d S r )r r r r r
OverflowErrorr s r test_invalid_init_valuesz Base.test_invalid_init_values? s *
* # #B
M :;;
#
#
r""
#
#
#
#
#
#
#
#
#
#
#
#
#
#
# # #s AA
A c | j | j d }| d | dd t j t
5 | dd d d d d S # 1 swxY w Y d S )Nr r rV int32)r r
_benchmarkr r r r s r test_benchmarkzBase.test_benchmarkE s **DJv,>?
### H---
]:
&
& 1 1 $ $Q 0 0 0 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1s A>>BBzcffi not available)reasonc | j | j d }|j }t |t sJ |j }||u sJ d S Nr )r r cffir r )r r cffi_interfaceother_cffi_interfaces r test_cffizBase.test_cffiL sV **DJv,>?
&+.)44444 , 1 #~555555r zctypes not availablec | j | j d }|j }t |t sJ |j }||u sJ d S r )r r ctypesr r )r r ctypes_interfaceother_ctypes_interfaces r test_ctypeszBase.test_ctypesT sY **DJv,>?
(/*I66666!.!5 %)9999999r c | j | j d }|j }| }t || d S r )r r rp __getstate__r )r r rp alt_states r
test_getstatezBase.test_getstate\ sG **DJv,>?
#!..00 5),,,,,r N) r r r r r+ r) r r classmethodr r r r r r r r r r r r r r r r markskipif MISSING_CFFIr MISSING_CTYPESr r rm r r r r s IEEE% % [% K K [K 0 0 0
K
K
K 1 1 1 1 1 1D D D
G G G ) ) )"( ( (
( ( (# # # # # # 1 1 1 [ -ABB6 6 CB6 [/EFF: : GF:- - - - -r r c * e Zd Zed Zd ZdS )
TestPhiloxc N t | _ d| _ t j | _ | t t d | _ | t t d | _
t | _ g | _
ddd d ddz dz fg| _ d S ) Nr; z./data/philox-testset-1.csvz./data/philox-testset-2.csv)r Nr r rI i r )r r r? r r+ r) r r pwdr r rr r r r r s r r zTestPhilox.setup_classd s "I MM 3 4 46 6 MM 3 4 46 6 '!# #/tQ#XPQ\8R"S r c | j | j d }|j }| |d d |d d }t |j |j d S )Nr rp counterr )r r )r r rp r )r r rp keyeds r test_set_keyzTestPhilox.test_set_keyq sl **DJv,>?
#""5>)+D',W~e'< # > >=. <<<<
?
? &2
&&&++b//!&
C$ ///++b//!&
aft!3444++b//'!!!!'!!!!!!r c $ t | d }|j }|j d }d}|d |k sJ | t d dD |j d }d}|d |k sJ d S )N %@
rp r%4=
Kvu1/ c 3 K | ] }d |z V
dS rI Nrm r| rP s r z/TestPCG64.test_advange_large.. & CC1CCCCCCr ` r; r' r5 r rI r l @H}.|zKLfbe r r rp r# sumr r pcgrp
initial_stateadvanced_states r test_advange_largez TestPCG64.test_advange_large s
t)).99
:
: '"?
W~ .... CCC&BCCCCCDDD '"@W~ //////r Nr r r r r r* r= rm r r r r y K * * [*
"
"
" 0 0 0 0 0r r c 0 e Zd Zed Zd Zd ZdS )
TestPCG64DXSMc P t | _ d| _ t j | _ | t t d | _ | t t d | _
t t f| _
dd gfdg| _ dg| _ d S )Nr; z ./data/pcg64dxsm-testset-1.csvz ./data/pcg64dxsm-testset-2.csvr r r )r r r? r r+ r) r r r r r r rr r r r r s r r zTestPCG64DXSM.setup_class s %I MM$s,L"M"MNN MM$s,L"M"MNN )9 5"(D6)Y!? #(' r c t | j | j d }|j j }d}|j | | d }||j _ |j ddz |z | d }||j _ |j dddz z |z | d }||k sJ ||k sJ d S r r" r% s r r* z#TestPCG64DXSM.test_advance_symmetry r+ r c $ t | d }|j }|j }d}|d d |k sJ | t d dD |j d }d}|d |k sJ d S )Nr- r. rp c 3 K | ] }d |z V
dS r0 rm r1 s r r2 z3TestPCG64DXSM.test_advange_large.. r3 r r4 l oF [U mO1X9} r7 r9 s r r= z TestPCG64DXSM.test_advange_large s
t)).99
:
: ?
W~g&-7777 CCC&BCCCCCDDD '"@W~ //////r Nr> rm r r rA rA r? r rA c 0 e Zd Zed Zd Zd ZdS ) TestMT19937c 8 t | _ d| _ t j | _ | t t d | _ | t t d | _
t | _ g | _
dg| _ d S )Nr' z ./data/mt19937-testset-1.csvz ./data/mt19937-testset-2.csvr )r r r? r r! r) r r r r r r r r r r s r r zTestMT19937.setup_class sv #I MM$s,J"K"KLL MM$s,J"K"KLL (!# #(' r c | t t | j t j t j g t t | j t j t j g t t | j t j t j t j g t t | j t j dt j g t t | j t j g t t | j dt j g d S )Nr )r rr r r r, pi)r s r test_seed_float_arrayz!TestMT19937.test_seed_float_array s i!3RXrug5F5FGGGi!3RXvh5G5GHHHi!3RXrurufo5N5NOOOi!3RXq"%j5I5IJJJi!3beW===i!3aZ@@@@@r c t | j | j d }|j }|j }| d }|d |d d |d d f}||_ | d }t || |dz }||_ | d }t || d S )Nr i r rp r pos)r rY )r r r rp r$ r )r r r rp desiredtupr s r test_state_tuplez TestMT19937.test_state_tuple s
)t)4:f+=>
?
?(
#++g&&_%uW~e'<W~e$& !
W%%VW%%%Hn !
W%%VW%%%%%r N)r r r r r rK rP rm r r rG rG sN * * [*A A A
&
&
&
&
&r rG c $ e Zd Zed ZdS ) TestSFC64c P t | _ d| _ t j | _ | t t d | _ | t t d | _
t t f| _
dd gfdg| _ dg| _ d S )Nr; z./data/sfc64-testset-1.csvz./data/sfc64-testset-2.csvr r r )r r r? r r+ r) r r r r r r rr r r r r s r r zTestSFC64.setup_class s !I MM 2 3 35 5 MM 2 3 35 5 )9 5"(D6)Y!? #(' r N)r r r r r rm r r rR rR s-
*
* [
*
*
*r rR c e Zd Zd Zd ZdS )TestDefaultRNGc h dddddgffD ]'}t | }t |j t sJ (d S )Nrm r ) rW i. )r r r r
)r argsrgs r test_seedzTestDefaultRNG.test_seed sP 'T4L?; 7 7D d#B b .66 6 66 6 7 7r c t }t | }|j |u sJ t | }||u sJ |j |u sJ d S r )r r r )r r rY rg2s r test_passthroughz TestDefaultRNG.test_passthrough s\
XX
__ 2%%%%"oobyyyy B&&&&&&r N)r r r rZ r] rm r r rU rU s2 7 7 7
' ' ' ' 'r rU )9osos.pathr sysnumpyr
numpy.testingr r r r r numpy.randomr r r
r r r
r r r numpy.random._commonr r r ImportErrorr r flagsoptimizepathdirnameabspath__file__r r r% r3 r7 r9 r@ rE rC rD rW re rw r r r r r rA rG rR rU rm r r rl s
* * * * * * * * * * * *
+ * * * * *KKKLL LLLMMMNN NNN 9 L goobgooh//00. . .D D D% % % " " "& & &= = = # # # 2& & &&: : :,# # #$a- a- a- a- a- a- a- a-H= = = = = = = =,$0 $0 $0 $0 $0 $0 $0 $0N$0 $0 $0 $0 $0D $0 $0 $0N!& !& !& !& !&$ !& !& !&H * * * * * * * * ' ' ' ' ' ' ' ' ' 's$ A AAA A&%A&