\ d dl Z d dlmZ d dlZd dlZd dlmZmZm Z d dl
m Z m Z m
Z
mZmZmZ ej dej ej ej ej ej ej ej ej ej f d Zd Z dd
Z d Z d d Z G d
d Z G d de Z! G d de Z" G d de Z# G d de Z$ G d de Z% G d de Z&dS ) N)partial) assert_equalassert_assert_array_equal) GeneratorMT19937PCG64 PCG64DXSMPhiloxSFC64module)scopeparamsc | j S N)param)requests q/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/numpy/random/tests/test_smoke.pydtyper s = c v | }t t j | | d }t |j dk | d }t |j dk | d }t |j dk | d }t |j dk d S )N
r r r r r r ) r size)r npisscalarshape)fvals r params_0r$ s
!##C BK
!B%%C CI
!H++C CI !"""
!L//C CI %&&&
!...C CI r Fc d}t j dd }t j dd d }t j dd d }t j ddg }t j dd d }|rd
}|d | z z }|d | z z }|d | z z }|d | z z }|d | z z } | | | |d | | | | | | | |d
| |d | |d d S )Ng @ @g (@ Y@r g P@r @) r r) ?g ?r r r )r arangereshapearraymax)r" boundedabcdegs r params_1r8 s A
#t A
#u %%h//A
#v &&|44A
#s A
#t $$Z00A
quuww
quuww
quuww
quuww
quuww AaDDDAahAaDDDAaDDDAaDDDAabMMMMAagAalr c d}t | t r%| D ]!}|t | | || z }"nt | t | k r$|t | t | k z }nt | t t
t j f rQt |t t
t j f r*t | | D ]\ }}|t || z }n || |k z }|S )NT)
isinstancedict
comp_statetypelisttupler ndarrayzip)state1state2 identicalkeys1s2s r r< r< > s I&$ * > >C F3K== =II >
f f % %T&\\T&\\11 veRZ 8 9 9 *jubj1?3 ?3 * ff--
0
0BZB / //
0
6) )I r c |#dt j dd z }| | | | | |t j | |t j | dd|t j | dd|t j | d| | d|t j | |t j | |t j d S )N r r l g &@) r randomrandintstandard_normalfloat32integersuint64standard_gammafloat64)rgns r warmuprW O s" y ""1b)) )qqq
+++q
+++KK7ARYK///KK7ARYK///dA dARZ000IIarzI"""IIarzI"""""r c e Zd Zed Zed Zd Zd Zd Zd Z d Z
d Z d Z d
Z
d Zd Zd
Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd 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& Z)d' Z*d( Z+d) Z,d* Z-d+ Z.d, Z/d- Z0d. Z1d/ Z2d0 Z3d1 Z4d2 Z5d3 Z6d4 Z7d5 Z8d6 Z9d7 Z:d8 Z;d9 Zd< Z?d= Z@d> ZAd? ZBd@ ZCdA ZDdB ZEdC ZFdD ZGdE ZHdF ZIdGS )HRNGc t | _ d | _ dg| _ t | j | j | _ | j j j | _ d| _ | d S )N90 @
r
bit_generatoradvanceseedr rU state
initial_stateseed_vector_bits _extra_setupclss r setup_classzRNG.setup_class_ sh " 7,3,ch788 F 0 6 ! r c t j dd | _ t j dd d d d f | _ t j ddd d | _ t | _ d S )Nr& r' g{Gz?d rj )r r- vec_1dvec_2dr. mat TypeError
seed_errorre s r rd zRNG._extra_setupj sc YsE**
YsE**473
)C--55jAA"r c 2 | j | j j _ d S r )rb rU r^ ra selfs r _reset_statezRNG._reset_stateq s &*&8 ###r c t | }|j j }| d | d ||j _ |j j }t t || d S )Nr) )r r^ ra rO r r< )rr rU ra new_states r test_initz
RNG.test_initt s
t))++
,
, &
1
1 !& $*
5),,-----r c \ | j j j }t | j j d rS| j j | j t t
|| j j j d S | j j j j }t j
d| d S )Nr_ z Advance is not supported by ) rU r^ ra hasattrr_ r r< __class____name__pytestskip)rr ra bitgen_names r test_advancezRNG.test_advance} s %+ 47() 4 4 F G ! ) )$, 7 7 7
5$'*?*EFFF G G G G G '/9BK KD{DD E E E E Er c F | j j j }t | j j d r| j j }|j }t t
|| | j d || j j _ | j j }|j }t t
|| d S | j j j j }|dvrt d| t j d| d S )Njumpedi. )r zno "jumped" in zJump is not supported by )
rU r^ ra rx r r r< rM ry rz AttributeErrorr{ r| )rr ra bit_gen2 jumped_statebit_gen3rejumped_stater} s r test_jumpz
RNG.test_jump s
%+ 47(( 3 3
Cw,3355H#>L
5,777 8 8 8 GNN7 8 8 8*/DG ! 'w,3355H %^N J|^<< = = = = = '/9BK*,,$%D{%D%DEEE KAKAA B B B B Br c | j ddd }t t | dk t |dk t |dk d S )N r r r )rU uniformr lenallrr rs r test_uniformzRNG.test_uniform sm GOOD#BO / /A" R a r c | j t j dgdz dd }t t | dk t |dk t |dk | j t j dgdz t j dgdz d }t t | dk t |dk t |dk | j dt j dgdz d }t t | dk t |dk t |dk d S )Nr r r r r r )rU r r r/ r r r r s r test_uniform_arrayzRNG.test_uniform_array s GOOBHdVb[ 1 13RO @ @A" R a GOOBHdVb[ 1 1 HcURZ 0 0r
;
;A" R a GOOD"(C52:"6"6RO @ @A" R a r c t t | j d dk t | j j d S Nr )r r rU rM r$ rq s r test_randomzRNG.test_random sC DGNN2&&''2-... r c v t t | j d dk d S r )r r rU rO rq s r test_standard_normal_zigz RNG.test_standard_normal_zig s3 DG++B//00B677777r c t t | j d dk t | j j d S r )r r rU rO r$ rq s r test_standard_normalzRNG.test_standard_normal F DG++B//00B6777()))))r c
H t t | j dd dk t t | j t j dgdz d dk t
| j j d S r )r r rU rS r r/ r8 rq s r test_standard_gammazRNG.test_standard_gamma s DG**2r2233r9:::DG**28RD2I+>+>CCDDJKKK'(((((r c t t | j d dk t | j j d S r )r r rU standard_exponentialr$ rq s r test_standard_exponentialz RNG.test_standard_exponential sF DG004455;<<<-.....r c | j dd }t t | dk |j t
j k sJ t t | j j d d S )Nr rP rK rU r r r r r rP r$ r rr randomss r test_standard_exponential_floatz#RNG.test_standard_exponential_float sp '..r.CCG "###}
****5YGGGHHHHHr c | j ddd }t t | dk |j t
j k sJ t t | j j dd d S )Nr rP inv)r methodr r s r #test_standard_exponential_float_logz'RNG.test_standard_exponential_float_log s '..r6; / = =G "###}
****5Y %' ' ' ( ( ( ( (r c t t | j d dk t | j j d S r )r r rU standard_cauchyr$ rq s r test_standard_cauchyzRNG.test_standard_cauchy r r c t t | j dd dk t | j j d S r )r r rU
standard_tr8 rq s r test_standard_tzRNG.test_standard_t sH DG&&r2..//25666#$$$$$r c t | j dd dk t | j dd dk d S )Nr r* r )r rU binomialrq s r
test_binomialzRNG.test_binomial sR R((A-... r**a/00000r c | j j j }| j d }|| j j _ | j d }t ||k d S )Nl )rU r^ ra rQ r )rr ra int_1int_2s r test_reset_statezRNG.test_reset_state s^ %+ ''&+ # '' r c t | }t | }t t |j j |j j d S r )r r^ r r< ra rr rU rg2s r test_entropy_initzRNG.test_entropy_init sr
t))++
,
,**,,--Jr / 5 " 0 68 8 8 9 9 9 9 9r c ( t | j | j }t | j | j }| | t t |j j |j j d S r )r r^ r` rM r r< ra r s r test_seedz
RNG.test_seed sz
)t)495
6
6**DI677
2+133D3JKKLLLLLr c P t | j | j }| |j j }| d }t | }||j _ | d }t || d S )Nr r )r r^ r` rO ra r rr rU ra n1r n2s r test_reset_state_gausszRNG.test_reset_state_gauss s
)t)495
6
6
&
R
(
(**,,--"'
b
)
)2r"""""r c t | j | j }| dddt j |j j }| dddt j }t | }||j _ | dddt j }t || d S )Nr rL x rK r )r r^ r` rQ r uint32ra r r s r test_reset_state_uint32zRNG.test_reset_state_uint32 s
)t)495
6
6
Aw29 555 &
[[GRry[
9
9**,,--"'
\\!Wb \
:
:2r"""""r c t | j | j }| d |j j }| dd }t | }||j _ | dd }t ||k d S )NrP rK r r r )r r^ r` rM ra r r r s r test_reset_state_floatzRNG.test_reset_state_float s
)t)495
6
6
""" &
YYBiY
0
0**,,--"'
ZZRyZ
1
1r !!!!!r c t j ddd }| j | }t ||k d S N r r r r- rU permutationr anyrr originalpermuteds r test_shufflezRNG.test_shuffle O 9S!R((7&&x00X%**,,-----r c t j ddd }| j | }t ||k d S r r r s r test_permutationzRNG.test_permutation r r c P | j ddd }t t | dk | j t j dgdz d }t t | dk | j dt j dgdz }t t | dk | j t j dgdz t j dgdz }t t | dk | j t j dgdz t j dggdz }t |j dk d S )Nr& r r )rU betar r r r/ r! rr valss r test_betaz
RNG.test_beta sJ w||Cb))D R w||BHcURZ 0 0#66D R w||C3%"*!5!566D R w||BHcURZ 0 0"(C52:2F2FGGD R w||BHcURZ 0 0"(SE7R<2H2HII
h&'''''r c z | j d }t t | dk d S r )rU bytesr r r s r
test_byteszRNG.test_bytes" s3 w}}R D R r c | j dd }t t | dk t | j j d S Nr& r )rU chisquarer r r8 r s r test_chisquarezRNG.test_chisquare& sJ w b))D R "#####r c | j dd }t t | dk t | j j d S r )rU exponentialr r r8 r s r test_exponentialzRNG.test_exponential+ sJ w""3++D R $%%%%%r c ~ | j ddd }t t | dk d S )N r r )rU r" r r r s r test_fz
RNG.test_f0 s7 wyyD"%%D R r c ~ | j ddd }t t | dk d S )Nr r, r )rU gammar r r s r
test_gammazRNG.test_gamma4 s7 w}}Q2&&D R r c | j dd }t t | dk t | j j d d S )Nr* r Tr1 )rU geometricr r r8 r r s r test_geometriczRNG.test_geometric8 sO w b))D R $d333333r c ~ | j ddd }t t | dk d S r )rU gumbelr r r s r test_gumbelzRNG.test_gumbel= s7 w~~c3++D R r c ~ | j ddd }t t | dk d S r )rU laplacer r r s r test_laplacezRNG.test_laplaceA s7 wsC,,D R r c ~ | j ddd }t t | dk d S r )rU logisticr r r s r test_logiticzRNG.test_logiticE s9 w S"--D R r c | | j dd }t t | dk d S )Nr* r )rU logseriesr r r s r test_logserieszRNG.test_logseriesI s7 w b))D R r c ~ | j ddd }t t | dk d S Nr 皙?)rU negative_binomialr r r s r test_negative_binomialzRNG.test_negative_binomialM s9 w((S"55D R r c ~ | j ddd }t t | dk d S )Nr r, )rU noncentral_chisquarer r r s r test_noncentral_chisquarez RNG.test_noncentral_chisquareQ s9 w++B266D R r c l | j dddd }t t | dk | j t j dgdz dd }t t | dk | j dt j dgdz d }t t | dk | j ddt j dgdz }t t | dk d S )Nr r r, r )rU noncentral_fr r r r/ r s r test_noncentral_fzRNG.test_noncentral_fU s w##AtQ33D R w##BHaS2X$6$6a@@D R w##Arx '<' t | }t | |j j }| ddt
j }t | }t | ||j _ | ddt
j }t || t |j t
j t t |j j |j j d S )Ng @rI rK ) r r^ rW ra rS r rP r r r r r< rK s r test_gamma_floatszRNG.test_gamma_floats s
t))++
,
,r
&
sBbj
9
9**,,--s "'
Rrz
:
:2r"""RXrz***
2+133D3JKKLLLLLr c : t | }t | |j j }| dt
j }t | }t | ||j _ | dt
j }t || t |j t
j t t |j j |j j d S NrI rK r r^ rW ra rO r rP r r r r r< rK s r test_normal_floatszRNG.test_normal_floats
t))++
,
,r
&
"*
5
5**,,--s "'
2:
6
62r"""RXrz***
2+133D3JKKLLLLLr c : t | }t | |j j }| dt
j }t | }t | ||j _ | dt
j }t || t |j t
j t t |j j |j j d S rR rS rK s r test_normal_zig_floatszRNG.test_normal_zig_floats rU r c | j }|j j }d}t j | }||j _ | | ||j _ | | }t
|| t j | }||j _ | ||j t j |t j }||j _ | |t j ||j _ | |t j }t
|| d S )N a outr )r^ r rK r^ r r ) rU r^ ra r emptyrO r r! rP )rr rU ra r existingdirectsizeds r test_output_fillzRNG.test_output_fill s#
W &8D>>!&
x(((!& ###..VX&&&!&
u5;7778D
333!&
xrz:::!& ##RZ#@@VX&&&&&r c | j }|j j }d}t j | }||j _ | | ||j _ | | }t
|| t j |t j }||j _ | |t j ||j _ | |t j }t
|| d S NrY r] r rK r_ r )rU r^ ra r r` rM r rP rr rU ra r ra rb s r test_output_filling_uniformz RNG.test_output_filling_uniform/ s
W &8D>>!&
h !& %%VX&&&8D
333!&
hbj 111!& BJ77VX&&&&&r c | j }|j j }d}t j | }||j _ | | ||j _ | | }t
|| t j |t j }||j _ | |t j ||j _ | |t j }t
|| d S rf )rU r^ ra r r` r r rP rg s r test_output_filling_exponentialz#RNG.test_output_filling_exponentialA s
W &8D>>!&
H ---!& ((d(33VX&&&8D
333!&
HBJ ???!& ((d"*(EEVX&&&&&r c | j }|j j }d}t j | }||j _ | d| ||j _ | d| }t
|| t j |t j }||j _ | d|t j ||j _ | d|t j }t
|| d S )NrY r r] r rK r_ r )rU r^ ra r zerosrS r rP rg s r test_output_filling_gammaz RNG.test_output_filling_gammaS s
W &8D>>!&
#8,,,!& ""3T"22VX&&&8D
333!&
#82:>>>!& ""3T"DDVX&&&&&r c J | j }|j j }d}t j d dz }t j | }||j _ | || ||j _ | || }t || t j |t j }||j _ | ||t j ||j _ | ||t j }t || d S ) NrY g @X@r r] r rK r_ r ) rU r^ ra r r- rl rS r rP )rr rU ra r mura rb s r #test_output_filling_gamma_broadcastz'RNG.test_output_filling_gamma_broadcaste s
W &
Yt__s
"8D>>!&
"(+++!& ""2D"11VX&&&8D
333!&
"("*===!& ""2D
"CCVX&&&&&r c | j }d}t j | }t j t
5 | |t j d d d n # 1 swxY w Y t j t 5 | |d d d d d d n # 1 swxY w Y t j |t j }t j t
5 | |t j d d d n # 1 swxY w Y t j
|t j }t j t
5 | d|t j d d d n # 1 swxY w Y t j t 5 | d|d d d t j d d d n # 1 swxY w Y t j
|t j }t j t
5 | d|t j d d d n # 1 swxY w Y t j t 5 | d|d d d d d d d S # 1 swxY w Y d S )NrY r_ r r] rK r ) rU r r` r{ raisesrn rO rP
ValueErrorrT rl rS )rr rU r ra s r test_output_fill_errorzRNG.test_output_fill_errorx s?
W8D>>
]9
%
% ? ? 82: > > > ? ? ? ? ? ? ? ? ? ? ? ? ? ? ?
]:
&
& 2 2 8CCaC= 1 1 1 2 2 2 2 2 2 2 2 2 2 2 2 2 2 28D
333
]9
%
% ? ? 82: > > > ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? 8D
333
]9
%
% C C cxrz B B B C C C C C C C C C C C C C C C
]:
&
& H H cx!}BJ G G G H H H H H H H H H H H H H H H8D
333
]9
%
% C C cxrz B B B C C C C C C C C C C C C C C C
]:
&
& 6 6 cx!} 5 5 5 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6s} "A%%A),A) B55B9<B99"D''D+.D++#FF !F >,G66G:=G::#I))I-0I-
!J;;J?J?c |t j k rd}d}n2t j | }t |j dz }|j }| | j ||gdz | }| | j |gdz || }t || | | j ||d| }t || | | j t j
|gdz t j
|gt d| }t || | | j t j
|gdz t j
|gdz d| } t || | | j d|d| }| | j |gdz | }t || d S )Nr, r r) r rK r ) r bool_iinfointr0 minrs rU rQ r r/ object)
rr r upperlowerinfor2 r3 r4 r5 r6 s
r test_integers_broadcastzRNG.test_integers_broadcast sK BH EEE8E??DMMA%EHE G UUGbL > > G eWr\5 > >Q G UE% @ @Q G RX
GbL 8UG6:::
Q G HeWr\ " "BHeWr\$:$:
Q G QBe < < G eWr\ 7 7Qr c n t j dg }t j dg }| j ||| }|j dk sJ | j |d || }|j dk sJ | j ||d | }|j dk sJ d S )Nr) r rK )r) )r r/ rU rQ r! )rr r highlowr^ s r test_integers_numpyzRNG.test_integers_numpy s x}}hsmmg sD 66yD g s1vt5 99yD g sDG5 99yD r c 6 |t j k rd}d}n2t j | }t |j dz }|j }t
j t 5 | j
||dz gdz | d d d n # 1 swxY w Y t
j t 5 | j
|dz
|gdz | d d d n # 1 swxY w Y t
j t 5 | j
|dz
g|gdz | d d d n # 1 swxY w Y t
j t 5 | j
dgdg| d d d d S # 1 swxY w Y d S )Nr, r r) r rK ) r rv rw rx r0 ry r{ rr rs rU rQ )rr r r{ r| r} s r test_integers_broadcast_errorsz"RNG.test_integers_broadcast_errors s BH EEE8E??DMMA%EHE
]:
&
& C C G UUQYK"$4E B B B C C C C C C C C C C C C C C C
]:
&
& C C G UQY" E B B B C C C C C C C C C C C C C C C
]:
&
& E E G eai[5'B,e D D D E E E E E E E E E E E E E E E
]:
&
& 4 4 G aS1#U 3 3 3 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4sH !%BBB6%C''C+.C+ &D==EE! FFFN)Jrz
__module__ __qualname__ classmethodrg rd rs rv r~ r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r! r$ r( r, r5 r8 r< rC rI rN rP rT rW rd rh rj rm rp rt r~ r r r r rY rY ^ s^ [ # # [# 9 9 9. . .F F FC C C"
! ! !8 8 8* * *) ) )
/ / /I I I ( ( (* * *% % %1 1 1 9 9 9 M M M# # ## # #" " ". . .
. . .
(
(
(! ! !$ $ $
& & &
! ! !! ! !4 4 4
! ! !! ! !! ! !! ! !! ! !! ! !! ! !! ! !! ! !" " "! ! !! ! !1 1 1
! ! !! ! !! ! !* * *% % % % % % ( ( (' ' ' $ $ $ 8 8 8$0 $0 $0L M M M M M M M M M M M M' ' ',' ' '$' ' '$' ' '$' ' '&6 6 6. B ! ! !4 4 4 4 4r rY c * e Zd Zed Zd ZdS ) TestMT19937c t | _ d | _ dg| _ t | j | j | _ | j j j | _ d| _ | t | _ d S )Ni! ! rE ) r r^ r_ r` r rU ra rb rc rd rs ro re s r rg zTestMT19937.setup_class sl # 23,3,ch788 F 0 6 ! #r c t j }| d | }|| j j _ | j j j }t |d |d d k t |d |d d k d S )Nc r) ra rE r, pos)
r rM RandomStaterO get_staterU r^ ra r r )rr nprgra rC s r test_numpy_statez TestMT19937.test_numpy_state s y$$&& R &+ #&,qVG_U 3388::;;;qVG_U 3355555r N)rz r r r rg r r r r r r s< $ $ [$6 6 6 6 6r r c $ e Zd Zed ZdS )
TestPhiloxc t | _ d| _ dg| _ t | j | j | _ | j j j | _ d| _ | d S Nl r[ r\ )
r r^ r_ r` r rU ra rb rc rd re s r rg zTestPhilox.setup_class sf "/ 7,3,ch788 F 0 6 ! r Nrz r r r rg r r r r r - [ r r c $ e Zd Zed ZdS ) TestSFC64c t | _ d | _ dg| _ t | j | j | _ | j j j | _ d| _ | d S )Nr[ )
r r^ r_ r` r rU ra rb rc rd re s r rg zTestSFC64.setup_class sf ! 7,3,ch788 F 0 6 " r Nr r r r r r r r r c $ e Zd Zed ZdS ) TestPCG64c t | _ d| _ dg| _ t | j | j | _ | j j j | _ d| _ | d S r r] re s r rg zTestPCG64.setup_class sf !/ 7,3,ch788 F 0 6 ! r Nr r r r r r r r r c $ e Zd Zed ZdS )
TestPCG64DXSMc t | _ d| _ dg| _ t | j | j | _ | j j j | _ d| _ | d S r )
r
r^ r_ r` r rU ra rb rc rd re s r rg zTestPCG64DXSM.setup_class
sf %/ 7,3,ch788 F 0 6 ! r Nr r r r r r r r r c 0 e Zd Zed Zd Zd ZdS )TestDefaultRNGc t | _ d| _ dg| _ t j j | j | _ | j j j | _ d| _
| d S r ) r r^ r_ r` r rM default_rngrU ra rb rc rd re s r rg zTestDefaultRNG.setup_class s_ "/ 7&1 F 0 6 ! r c ^ t t | j j t d S r )r r: rU r^ r rq s r test_default_is_pcg64z$TestDefaultRNG.test_default_is_pcg64" s'
470%8899999r c t j t j d t j d t j d t j d t j ddg t j t
5 t j d d d d n # 1 swxY w Y t j t
5 t j ddg d d d d S # 1 swxY w Y d S )Nr[ r l -tE+I6%Yl l ) ~eR :TI*6 r )r rM r r{ rr rs rq s r r zTestDefaultRNG.test_seed' s
d###
e$$$
a
DEEE
E F H I I I
]:
&
& & & I ! !" % % % & & & & & & & & & & & & & & &
]:
&
& / / I ! !5"+ . . . / / / / / / / / / / / / / / / / / /s$ DDD%"EEEN)rz r r r rg r r r r r r r sK [ : : :
/ / / / /r r )Fr )'r> functoolsr numpyr r{
numpy.testingr r r numpy.randomr r r r
r r fixturerv int8int16int32int64uint8uint16r rR r r$ r8 r< rW rY r r r r r r r r r r sc
C C C C C C C C C C N N N N N N N N N N N N N N N N h"'28RXrx")RY CD D D D D
!
!
! B " # # # # n 4 n 4 n 4 n 4 n 4 n 4 n 4 n 4b6 6 6 6 6# 6 6 6, C / / / / /S / / / / /r