idn ddlZddlmZddlZddlZddlmZmZm Z ddl m Z m Z m Z mZmZmZejdejejejejejejejejejf dZdZdd Zd Zdd ZGd dZ Gdde Z!Gdde Z"Gdde Z#Gdde Z$Gdde Z%Gdde Z&dS)N)partial) assert_equalassert_assert_array_equal) GeneratorMT19937PCG64 PCG64DXSMPhiloxSFC64module)scopeparamsc|jSN)param)requests R/opt/cloudlinux/venv/lib/python3.11/site-packages/numpy/random/tests/test_smoke.pydtyper s  =cv|}ttj||d}t|jdk|d}t|jdk|d}t|jdk|d}t|jdkdS)N rrrrrr)rsize)rnpisscalarshape)fvals rparams_0r$s !##C BK   !B%%C CI  !H++C CI !""" !L//C CI %&&& !...C CI      rFcd}tjdd}tjddd}tjddd}tjddg}tjddd }|rd }|d |zz }|d |zz }|d |zz }|d |zz }|d |zz }||||d ||||||||d ||d ||d dS)Ng@@g(@Y@rgP@r@)rr)?g?rrr)rarangereshapearraymax)r"boundedabcdegs rparams_1r8s A #tA #u%%h//A #v&&|44A #sA #t$$Z00A  quuww  quuww  quuww  quuww  quuww AaDDDAahAaDDDAaDDDAaDDDAabMMMMAagAalrcd}t|tr%|D]!}|t||||z}"nt|t|kr$|t|t|kz}nt|tt t jfrQt|tt t jfr*t||D]\}}|t||z}n |||kz}|S)NT) isinstancedict comp_statetypelisttuplerndarrayzip)state1state2 identicalkeys1s2s rr<r<>sI&$ * > >C F3K== =II > ff % %T&\\T&\\11 veRZ8 9 9 *jubj1?3?3 *ff-- 0 0BZB///  0 6) )I rc|#dtjddz}||||||tj||tj|dd|tj|dd|tj|d||d|tj||tj||tjdS)N rrlg&@) rrandomrandintstandard_normalfloat32integersuint64standard_gammafloat64)rgns rwarmuprWOs"y ""1b)) )qqq +++q +++KK7ARYK///KK7ARYK///dAdARZ000IIarzI"""IIarzI"""""rceZdZedZedZdZdZdZdZ dZ dZ d Z d Z d Zd Zd ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZ dZ!dZ"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/Z2d0Z3d1Z4d2Z5d3Z6d4Z7d5Z8d6Z9d7Z:d8Z;d9Zd<Z?d=Z@d>ZAd?ZBd@ZCdAZDdBZEdCZFdDZGdEZHdFZIdGS)HRNGct|_d|_dg|_t |j|j|_|jjj|_d|_| dS)N90@ r bit_generatoradvanceseedrrUstate initial_stateseed_vector_bits _extra_setupclss r setup_classzRNG.setup_class_sh" 7,3,ch788F06! rctjdd|_tjdddddf|_tjdddd|_t |_dS)Nr&r'g{Gz?drj)rr-vec_1dvec_2dr.mat TypeError seed_errorres rrdzRNG._extra_setupjscYsE** YsE**473 )C--55jAA"rc2|j|jj_dSr)rbrUr^raselfs r _reset_statezRNG._reset_stateqs&*&8###rct|}|jj}|d|d||j_|jj}t t ||dS)Nr))rr^rarOrr<)rrrUra new_states r test_initz RNG.test_initts t))++ , , & 1 1!&$*  5),,-----rc\|jjj}t|jjdrS|jj|jt t ||jjj dS|jjjj}tj d|dS)Nr_zAdvance is not supported by ) rUr^rahasattrr_rr< __class____name__pytestskip)rrra 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 ErcF|jjj}t|jjdr|jj}|j}t t || |jd||jj_|jj}|j}t t ||dS|jjjj }|dvrtd|tj d|dS)Njumpedi.)r zno "jumped" in zJump is not supported by ) rUr^rarxrrr<rMryrzAttributeErrorr{r|)rrrabit_gen2 jumped_statebit_gen3rejumped_stater}s r test_jumpz RNG.test_jumps %+ 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 Brc|jddd}tt|dkt|dkt|dkdS)Nrrr)rUuniformrlenallrrrs r test_uniformzRNG.test_uniformsm GOOD#BO / /A" R a rc|jtjdgdzdd}t t |dkt |dkt |dk|jtjdgdztjdgdzd}t t |dkt |dkt |dk|jdtjdgdzd}t t |dkt |dkt |dkdS)Nrrrrrr)rUrrr/rrrrs rtest_uniform_arrayzRNG.test_uniform_arrays GOOBHdVb[113RO @ @A" R a  GOOBHdVb[11HcURZ00r  ; ;A" R a  GOOD"(C52:"6"6RO @ @A" R a rctt|jddkt |jjdSNr)rrrUrMr$rqs r test_randomzRNG.test_randomsCDGNN2&&''2-...     rcvtt|jddkdSr)rrrUrOrqs rtest_standard_normal_zigzRNG.test_standard_normal_zigs3DG++B//00B677777rctt|jddkt |jjdSr)rrrUrOr$rqs rtest_standard_normalzRNG.test_standard_normalFDG++B//00B6777()))))rc Htt|jdddktt|jt jdgdzddkt |jjdSr)rrrUrSrr/r8rqs rtest_standard_gammazRNG.test_standard_gammasDG**2r2233r9:::DG**28RD2I+>+>CCDDJKKK'(((((rctt|jddkt |jjdSr)rrrUstandard_exponentialr$rqs rtest_standard_exponentialzRNG.test_standard_exponentialsFDG004455;<<<-.....rc|jdd}tt|dkt t |jjddS)NrrPrKrUrrrr$rrrrandomss rtest_standard_exponential_floatz#RNG.test_standard_exponential_floats['..r.CCG "###5YGGGHHHHHrc|jddd}tt|dkt t |jjdddS)NrrPinv)rmethodrrs r#test_standard_exponential_float_logz'RNG.test_standard_exponential_float_logsw'..r6;/==G "###5Y %''' ( ( ( ( (rctt|jddkt |jjdSr)rrrUstandard_cauchyr$rqs rtest_standard_cauchyzRNG.test_standard_cauchyrrctt|jdddkt |jjdSr)rrrU standard_tr8rqs rtest_standard_tzRNG.test_standard_tsHDG&&r2..//25666#$$$$$rct|jdddkt|jdddkdS)Nrr*r)rrUbinomialrqs r test_binomialzRNG.test_binomialsR  R((A-...  r**a/00000rc|jjj}|jd}||jj_|jd}t ||kdS)Nl)rUr^rarQr)rrraint_1int_2s rtest_reset_statezRNG.test_reset_states^%+  ''&+#  ''rct|}t|}tt|jj|jj dSr)rr^rr<rarrrUrg2s rtest_entropy_initzRNG.test_entropy_initsr t))++ , ,**,,--Jr/5"06888 9 9 9 9 9rc(t|j|j}t|j|j}||t t |jj|jjdSr)rr^r`rMrr<rars r test_seedz RNG.test_seedsz )t)495 6 6**DI677   2+133D3JKKLLLLLrcPt|j|j}||jj}|d}t|}||j_|d}t ||dS)Nrr)rr^r`rOrarrrrUran1rn2s rtest_reset_state_gausszRNG.test_reset_state_gausss )t)495 6 6  &   R  ( (**,,--"'  b ) )2r"""""rct|j|j}|dddtj|jj}|dddtj}t|}||j_|dddtj}t||dS)NrrLxrKr)rr^r`rQruint32rarrs rtest_reset_state_uint32zRNG.test_reset_state_uint32s )t)495 6 6 Aw29 555 & [[GRry[ 9 9**,,--"' \\!Wb \ : :2r"""""rct|j|j}|d|jj}|dd}t|}||j_|dd}t ||kdS)NrPrKrrr)rr^r`rMrarrrs rtest_reset_state_floatzRNG.test_reset_state_floats )t)495 6 6 """ & YYBiY 0 0**,,--"' ZZRyZ 1 1r  !!!!!rctjddd}|j|}t ||kdSNrrrr-rU permutationranyrroriginalpermuteds r test_shufflezRNG.test_shuffle O9S!R((7&&x00X%**,,-----rctjddd}|j|}t ||kdSrrrs rtest_permutationzRNG.test_permutationrrcP|jddd}tt|dk|jt jdgdzd}tt|dk|jdt jdgdz}tt|dk|jt jdgdzt jdgdz}tt|dk|jt jdgdzt jdggdz}t|jdkdS)Nr&rr)rUbetarrrr/r!rrvalss r test_betaz RNG.test_betasJw||Cb))D R   w||BHcURZ00#66D R   w||C3%"*!5!566D R   w||BHcURZ00"(C52:2F2FGGD R   w||BHcURZ00"(SE7R<2H2HII h&'''''rcz|jd}tt|dkdSr)rUbytesrrrs r test_byteszRNG.test_bytes"s3w}}R  D R     rc|jdd}tt|dkt |jjdSNr&r)rU chisquarerrr8rs rtest_chisquarezRNG.test_chisquare&sJw  b))D R   "#####rc|jdd}tt|dkt |jjdSr)rU exponentialrrr8rs rtest_exponentialzRNG.test_exponential+sJw""3++D R   $%%%%%rc~|jddd}tt|dkdS)Nrr)rUr"rrrs rtest_fz RNG.test_f0s7wyyD"%%D R     rc~|jddd}tt|dkdS)Nrr,r)rUgammarrrs r test_gammazRNG.test_gamma4s7w}}Q2&&D R     rc|jdd}tt|dkt |jjddS)Nr*rTr1)rU geometricrrr8rrs rtest_geometriczRNG.test_geometric8sOw  b))D R   $d333333rc~|jddd}tt|dkdSr)rUgumbelrrrs r test_gumbelzRNG.test_gumbel=s7w~~c3++D R     rc~|jddd}tt|dkdSr)rUlaplacerrrs r test_laplacezRNG.test_laplaceAs7wsC,,D R     rc~|jddd}tt|dkdSr)rUlogisticrrrs r test_logiticzRNG.test_logiticEs9wS"--D R     rc||jdd}tt|dkdS)Nr*r)rU logseriesrrrs rtest_logserieszRNG.test_logseriesIs7w  b))D R     rc~|jddd}tt|dkdSNr皙?)rUnegative_binomialrrrs rtest_negative_binomialzRNG.test_negative_binomialMs9w((S"55D R     rc~|jddd}tt|dkdS)Nrr,)rUnoncentral_chisquarerrrs rtest_noncentral_chisquarezRNG.test_noncentral_chisquareQs9w++B266D R     rcl|jdddd}tt|dk|jt jdgdzdd}tt|dk|jdt jdgdzd}tt|dk|jddt jdgdz}tt|dkdS)Nrrr,r)rU noncentral_frrrr/rs rtest_noncentral_fzRNG.test_noncentral_fUsw##AtQ33D R   w##BHaS2X$6$6a@@D R   w##Arx '<'t|}t||jj}|ddt j}t|}t|||j_|ddt j}t||t|j t jtt|jj|jjdS)Ng@rIrK) rr^rWrarSrrPrrrrr<rKs rtest_gamma_floatszRNG.test_gamma_floatss t))++ , ,r  &   sBbj  9 9**,,--s "'   Rrz  : :2r"""RXrz*** 2+133D3JKKLLLLLrc:t|}t||jj}|dt j}t|}t|||j_|dt j}t||t|j t jtt|jj|jjdSNrIrK rr^rWrarOrrPrrrrr<rKs rtest_normal_floatszRNG.test_normal_floats t))++ , ,r  &   "*  5 5**,,--s "'  2: 6 62r"""RXrz*** 2+133D3JKKLLLLLrc:t|}t||jj}|dt j}t|}t|||j_|dt j}t||t|j t jtt|jj|jjdSrRrSrKs rtest_normal_zig_floatszRNG.test_normal_zig_floats rUrc|j}|jj}d}tj|}||j_||||j_||}t ||tj|}||j_|||jtj|tj}||j_||tj||j_||tj}t ||dS)Naoutr)r^rrKr^rr) rUr^raremptyrOrr!rP)rrrUrarexistingdirectsizeds rtest_output_fillzRNG.test_output_fills# W &8D>>!& x(((!&###..VX&&&!& u5;7778D 333!& xrz:::!&##RZ#@@VX&&&&&rc|j}|jj}d}tj|}||j_||||j_||}t ||tj|tj}||j_||tj||j_||tj}t ||dSNrYr]rrKr_r)rUr^rarr`rMrrPrrrUrarrarbs rtest_output_filling_uniformzRNG.test_output_filling_uniform/s W &8D>>!& h !&%%VX&&&8D 333!& hbj 111!&BJ77VX&&&&&rc|j}|jj}d}tj|}||j_||||j_||}t ||tj|tj}||j_||tj||j_||tj}t ||dSrf)rUr^rarr`rrrPrgs rtest_output_filling_exponentialz#RNG.test_output_filling_exponentialAs W &8D>>!& H---!&((d(33VX&&&8D 333!& HBJ???!&((d"*(EEVX&&&&&rc|j}|jj}d}tj|}||j_|d|||j_|d|}t ||tj|tj}||j_|d|tj||j_|d|tj}t ||dS)NrYrr]rrKr_r)rUr^rarzerosrSrrPrgs rtest_output_filling_gammazRNG.test_output_filling_gammaSs W &8D>>!& #8,,,!&""3T"22VX&&&8D 333!& #82:>>>!&""3T"DDVX&&&&&rcJ|j}|jj}d}tjddz}tj|}||j_|||||j_|||}t||tj|tj}||j_|||tj||j_|||tj}t||dS) NrYg@X@rr]rrKr_r) rUr^rarr-rlrSrrP)rrrUrarmurarbs r#test_output_filling_gamma_broadcastz'RNG.test_output_filling_gamma_broadcastes  W & Yt__s "8D>>!& "(+++!&""2D"11VX&&&8D 333!& "("*===!&""2D "CCVX&&&&&rc|j}d}tj|}tjt 5||tjdddn #1swxYwYtjt5||ddddddn #1swxYwYtj|tj}tjt 5||tj dddn #1swxYwYtj |tj}tjt 5| d|tj dddn #1swxYwYtjt5| d|dddtjdddn #1swxYwYtj |tj }tjt 5| d|tjdddn #1swxYwYtjt5| d|dddddddS#1swxYwYdS)NrYr_rr]rKr) rUrr`r{raisesrnrOrP ValueErrorrTrlrS)rrrUrras rtest_output_fill_errorzRNG.test_output_fill_errorxs? 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|tjkrd}d}n2tj|}t|jdz}|j}||j||gdz|}||j|gdz||}t||||j||d|}t||||jtj |gdztj |gtd|}t||||jtj |gdztj |gdzd|} t|| ||jd|d|}||j|gdz|}t||dS)Nr,rr)rrKr) rbool_iinfointr0minrsrUrQrr/object) rrrupperlowerinfor2r3r4r5r6s rtest_integers_broadcastzRNG.test_integers_broadcastsK BH  EEE8E??DMMA%EHE  G  UUGbL  > >  G  eWr\5  > >Q  G  UE%  @ @Q  G  RX GbL8UG6:::    Q  G   HeWr\ " "BHeWr\$:$:    Q  G  QBe  < <  G  eWr\  7 7Qrc tjdg}tjdg}|j|||}|j|d||}|j||d|}dS)Nr)rrK)rr/rUrQ)rrrhighlowr^s rtest_integers_numpyzRNG.test_integers_numpysx}}hsmmgsD66gs1vt599gsDG599rc6|tjkrd}d}n2tj|}t|jdz}|j}t jt5|j ||dzgdz|dddn #1swxYwYt jt5|j |dz |gdz|dddn #1swxYwYt jt5|j |dz g|gdz|dddn #1swxYwYt jt5|j dgdg|ddddS#1swxYwYdS)Nr,rr)rrK) rrvrwrxr0ryr{rrrsrUrQ)rrrr{r|r}s rtest_integers_broadcast_errorsz"RNG.test_integers_broadcast_errorss 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__ classmethodrgrdrsrvr~rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr rrrrrrr!r$r(r,r5r8r<rCrIrNrPrTrWrdrhrjrmrprtr~rrrrrYrY^s^[##[# 999...FFFCCC"      !!!888***))) ///III (((***%%%111   999 MMM######"""... ... ( ( (!!!$$$ &&& !!!!!!444 !!!!!!!!!!!!!!!!!!!!!!!!!!!"""!!!!!!111 !!!!!!!!!***%%% %%% ( ( (''' $$$ 8 8 8$0$0$0L M M M M M M M M M M M M''','''$'''$'''$'''&666.B ! ! !44444rrYc*eZdZedZdZdS) TestMT19937ct|_d|_dg|_t |j|j|_|jjj|_d|_| t|_ dS)Ni!!rE) rr^r_r`rrUrarbrcrdrsrores rrgzTestMT19937.setup_classsl# 23,3,ch788F06! #rctj}|d|}||jj_|jjj}t|d|ddk t|d|ddkdS)Ncr)rarEr,pos) rrM RandomStaterO get_staterUr^rarr)rrnprgrarCs rtest_numpy_statezTestMT19937.test_numpy_statesy$$&& R     &+#&,qVG_U3388::;;;qVG_U3355555rN)rzrrrrgrrrrrrs<$$[$66666rrc$eZdZedZdS) TestPhiloxct|_d|_dg|_t |j|j|_|jjj|_d|_| dSNlr[r\) r r^r_r`rrUrarbrcrdres rrgzTestPhilox.setup_classsf"/ 7,3,ch788F06! rNrzrrrrgrrrrr-[rrc$eZdZedZdS) TestSFC64ct|_d|_dg|_t |j|j|_|jjj|_d|_| dS)Nr[) r r^r_r`rrUrarbrcrdres rrgzTestSFC64.setup_classsf! 7,3,ch788F06" rNrrrrrrrrrc$eZdZedZdS) TestPCG64ct|_d|_dg|_t |j|j|_|jjj|_d|_| dSrr]res rrgzTestPCG64.setup_classsf!/ 7,3,ch788F06! rNrrrrrrrrrc$eZdZedZdS) TestPCG64DXSMct|_d|_dg|_t |j|j|_|jjj|_d|_| dSr) r r^r_r`rrUrarbrcrdres rrgzTestPCG64DXSM.setup_class sf%/ 7,3,ch788F06! rNrrrrrr rrrc0eZdZedZdZdZdS)TestDefaultRNGct|_d|_dg|_t jj|j|_|jjj|_ d|_ | dSr) r r^r_r`rrM default_rngrUrarbrcrdres rrgzTestDefaultRNG.setup_classs_"/ 7&1F06! rc^tt|jjtdSr)rr:rUr^r rqs rtest_default_is_pcg64z$TestDefaultRNG.test_default_is_pcg64"s'  470%8899999rctjtjdtjdtjdtjdtjddgtjt 5tjddddn #1swxYwYtjt 5tjddgddddS#1swxYwYdS)Nr[rl -tE+I6%Yl l ) ~eR :TI*6r)rrMrr{rrrsrqs rrzTestDefaultRNG.test_seed's  d### e$$$ a    DEEE EFH I I I ]: & & & & I ! !" % % % & & & & & & & & & & & & & & & ]: & & / / I ! !5"+ . . . / / / / / / / / / / / / / / / / / /s$ DDD%"EEEN)rzrrrrgrrrrrrrsK  [ ::: / / / / /rr)Fr)'r> functoolsrnumpyrr{ numpy.testingrrr numpy.randomrrr r r r fixturervint8int16int32int64uint8uint16rrRrr$r8r<rWrYrrrrrrrrrrsc  CCCCCCCCCCNNNNNNNNNNNNNNNNh"'28RXrx")RY CDDDDD ! ! !B" # # # #n 4n 4n 4n 4n 4n 4n 4n 4b66666#666,                             C   /////S/////r