K im ddlZddlmZddlZddlZddlmZmZm Z m Z m Z m Z ddl mZmZmZej"dej$ej&ej(ej*ej,ej.ej0ej2ej4f dZdZdd Zd Zdd ZGd d Z Gdde Z!Gdde Z"Gdde Z#Gdde Z$Gdde Z%Gdde Z&y)N)partial)MT19937PCG64 PCG64DXSMSFC64 GeneratorPhilox)assert_assert_array_equal assert_equalmodule)scopeparamsc|jSN)param)requests c/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/numpy/random/tests/test_smoke.pydtyper s ==cP|}ttj||d}t|jdk(|d}t|jdk(|d}t|jdk(|d}t|jdk(y)N rrrrrr)rsize)r npisscalarshape)fvals rparams_0r$s #C BKK  B%C CII  H+C CII !" L/C CII %& .C CII  rcd}tjdd}tjddjd}tjddjd}tjddg}tjddjd }|rpd }|d |j zz }|d |j zz }|d |j zz }|d |j zz }|d |j zz }||||d ||||||||d ||d ||d y)Ng@@g(@Y@rgP@r@)rr)?g?rrr)rarangereshapearraymax)r"boundedabcdegs rparams_1r8s+ A #tA #u%%h/A #v&&|4A #sA #t$$Z0A  quuw  quuw  quuw  quuw  quuw aDahaDaDaDabMagalrcd}t|tr|D]}|t||||z}|St|t|k7r|t|t|k(z}|St|tt t jfrJt|tt t jfr%t||D]\}}|t||z}|S|||k(z}|S)NT) isinstancedict comp_statetypelisttuplerndarrayzip)state1state2 identicalkeys1s2s rr<r<@sI&$ >C F3K= =I >  ff %T&\T&\11   VdE2::6 7J T5"**-=/&&) ,FB B+ +I ,  Vv%% rc|#dtjjddz}|j||j||j|tj|j|tj|j dd|tj |j dd|tj |jd||jd|tj|j|tj|j|tjy)N rrlg&@) rrandomrandintstandard_normalfloat32integersuint64standard_gammafloat64)rgns rwarmuprWPsy ""1b) )qqq +q +KK7ARYYK/KK7ARYYK/dAdARZZ0IIarzzI"IIarzzI"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@ZCdAZDdBZEdCZFdDZGdEZHdFZIyG)HRNGct|_d|_dg|_t |j|j|_|j jj |_d|_|jy)N90@ r bit_generatoradvanceseedrrUstate initial_stateseed_vector_bits _extra_setupclss r setup_classzRNG.setup_class`sf" 7,3,,chh78FF0066! rctjdd|_tjdddddf|_tjdddj d|_t |_y)Nr&r'g{Gz?drj)rr-vec_1dvec_2dr.mat TypeError seed_errorres rrdzRNG._extra_setupksUYYsE* YYsE*473 ))C-55jA"rcN|j|jj_yr)rbrUr^raselfs r _reset_statezRNG._reset_staters&*&8&8#rct|j}|jj}|jd|jd||j_|jj}t t ||y)Nr))rr^rarOr r<)rrrUra new_states r test_initz RNG.test_initusp t))+ ,  && 1 1!&$$**  5),-rc|jjj}t|jjdrd|jjj |jt t ||jjj y|jjjj}tjd|y)Nr_zAdvance is not supported by ) rUr^rahasattrr_r r< __class____name__pytestskip)rrra bitgen_names r test_advancezRNG.test_advance~s%%++ 477(() 4 GG ! ! ) )$,, 7  5$''*?*?*E*EFF G''//99BBK KK6{mD Erc|jjj}t|jjdr|jjj }|j}t t || |jjd||jj_|jjj }|j}t t ||y|jjjj}|dvrtd|tjd|y)Njumpedi.)rzno "jumped" in zJump is not supported by ) rUr^rarxrr r<rMryrzAttributeErrorr{r|)rrrabit_gen2 jumped_statebit_gen3rejumped_stater}s r test_jumpz RNG.test_jumps%%++ 477((( 3ww,,335H#>>L  5,77 8 GGNN7 8*/DGG ! ! 'ww,,335H%^^N J|^< =''//99BBK*,${m%DEE KK3K=A Brc|jjddd}tt|dk(t|dkDj t|dkj y)Nrrr)rUuniformr lenallrrrs r test_uniformzRNG.test_uniformsN GGOOD#BO /A" R a rcJ|jjtjdgdzdd}t t |dk(t |dkDj t |dkj |jjtjdgdztjdgdzd}t t |dk(t |dkDj t |dkj |jjdtjdgdzd}t t |dk(t |dkDj t |dkj y)Nrrrrrr)rUrrr/r rrrs rtest_uniform_arrayzRNG.test_uniform_arrays( GGOOBHHdVb[13RO @A" R a  GGOOBHHdVb[1HHcURZ0r  ;A" R a  GGOOD"((C52:"6RO @A" R a rctt|jjddk(t |jjyNr)r rrUrMr$rqs r test_randomzRNG.test_randoms1DGGNN2&'2-. rcdtt|jjddk(yr)r rrUrOrqs rtest_standard_normal_zigzRNG.test_standard_normal_zigs#DGG++B/0B67rctt|jjddk(t |jjyr)r rrUrOr$rqs rtest_standard_normalzRNG.test_standard_normal5DGG++B/0B67(()rc 4tt|jjdddk(tt|jjt j dgdzddk(t |jjyr)r rrUrSrr/r8rqs rtest_standard_gammazRNG.test_standard_gammasiDGG**2r23r9:DGG**288RD2I+>CDJK''(rctt|jjddk(t |jjyr)r rrUstandard_exponentialr$rqs rtest_standard_exponentialzRNG.test_standard_exponentials5DGG0045;<--.rc|jjdd}tt|dk(|jt j k(sJtt|jjdy)NrrPrK rUrr rrrrPr$rrrrandomss rtest_standard_exponential_floatz#RNG.test_standard_exponential_floats[''..r.CG "#}} ***55YGHrc|jjddd}tt|dk(|jt j k(sJtt|jjddy)NrrPinv)rmethodrrs r#test_standard_exponential_float_logz'RNG.test_standard_exponential_float_logsi''..r6;/=G "#}} ***55Y %' (rctt|jjddk(t |jjyr)r rrUstandard_cauchyr$rqs rtest_standard_cauchyzRNG.test_standard_cauchyrrctt|jjdddk(t |jjyr)r rrU standard_tr8rqs rtest_standard_tzRNG.test_standard_ts7DGG&&r2./256##$rct|jjdddk\t|jjdddk\y)Nrr*r)r rUbinomialrqs r test_binomialzRNG.test_binomials>  R(A-.  r*a/0rc|jjj}|jjd}||jj_|jjd}t ||k(y)Nl)rUr^rarQr )rrraint_1int_2s rtest_reset_statezRNG.test_reset_states]%%++  '&+#  'rct|j}t|j}tt|jj|jj yr)rr^r r<rarrrUrg2s rtest_entropy_initzRNG.test_entropy_initsY t))+ ,**,-Jr//55"006688 9rcFt|j|j}t|j|j}|j|jt t |jj |jj yr)rr^r`rMr r<rars r test_seedz RNG.test_seedss )t))4995 6***DII67   2++1133D3D3J3JKLrcHt|j|j}|j|jj}|jd}t|j}||j_|jd}t ||y)Nrr)rr^r`rOrar rrrUran1rn2s rtest_reset_state_gausszRNG.test_reset_state_gausss )t))4995 6   &&   R  (**,-"'  b )2r"rct|j|j}|jdddtj |jj }|jdddtj }t|j}||j_|jdddtj }t||y)NrrLxrKr)rr^r`rQruint32rar rs rtest_reset_state_uint32zRNG.test_reset_state_uint32s )t))4995 6 Aw299 5  && [[GRryy[ 9**,-"' \\!Wb \ :2r"rcpt|j|j}|jd|jj}|jdd}t|j}||j_|jdd}t ||k(j y)NrPrKrrr)rr^r`rMrar rrs rtest_reset_state_floatzRNG.test_reset_state_floats )t))4995 6 "  && YYBiY 0**,-"' ZZRyZ 1r !rctjddd}|jj|}t ||k7j yNrrrr-rU permutationr anyrroriginalpermuteds r test_shufflezRNG.test_shuffle >99S!R(77&&x0X%**,-rctjddd}|jj|}t ||k7j yrrrs rtest_permutationzRNG.test_permutationrrc|jjddd}tt|dk(|jjt j dgdzd}tt|dk(|jjdt j dgdz}tt|dk(|jjt j dgdzt j dgdz}tt|dk(|jjt j dgdzt j dggdz}t|j dk(y)Nr&rr)rUbetar rrr/r!rrvalss r test_betaz RNG.test_betasww||Cb)D R ww||BHHcURZ0#6D R ww||C3%"*!56D R ww||BHHcURZ0"((C52:2FGD R ww||BHHcURZ0"((SE7R<2HI h&'rch|jjd}tt|dk(yr)rUbytesr rrs r test_byteszRNG.test_bytes#s$ww}}R D R rc|jjdd}tt|dk(t |jjyNr&r)rU chisquarer rr8rs rtest_chisquarezRNG.test_chisquare's:ww  b)D R ""#rc|jjdd}tt|dk(t |jjyr)rU exponentialr rr8rs rtest_exponentialzRNG.test_exponential,s:ww""3+D R $$%rcl|jjddd}tt|dk(y)Nrr)rUr"r rrs rtest_fz RNG.test_f1s(wwyyD"%D R rcl|jjddd}tt|dk(y)Nrr,r)rUgammar rrs r test_gammazRNG.test_gamma5s(ww}}Q2&D R rc|jjdd}tt|dk(t |jj dy)Nr*rTr1)rU geometricr rr8rrs rtest_geometriczRNG.test_geometric9s<ww  b)D R $$d3rcl|jjddd}tt|dk(yr)rUgumbelr rrs r test_gumbelzRNG.test_gumbel>s(ww~~c3+D R rcl|jjddd}tt|dk(yr)rUlaplacer rrs r test_laplacezRNG.test_laplaceBs(wwsC,D R rcl|jjddd}tt|dk(yr)rUlogisticr rrs r test_logiticzRNG.test_logiticFs*wwS"-D R rcj|jjdd}tt|dk(y)Nr*r)rU logseriesr rrs rtest_logserieszRNG.test_logseriesJs(ww  b)D R rcl|jjddd}tt|dk(yNr皙?)rUnegative_binomialr rrs rtest_negative_binomialzRNG.test_negative_binomialNs*ww((S"5D R rcl|jjddd}tt|dk(y)Nrr,)rUnoncentral_chisquarer rrs rtest_noncentral_chisquarezRNG.test_noncentral_chisquareRs*ww++B26D R rc0|jjdddd}tt|dk(|jjt j dgdzdd}tt|dk(|jjdt j dgdzd}tt|dk(|jjddt j dgdz}tt|dk(y)Nrrr,r)rU noncentral_fr rrr/rs rtest_noncentral_fzRNG.test_noncentral_fVsww##AtQ3D R ww##BHHaS2X$6a@D R ww##Arxx '!& x(!&###.VX&!& u5;;788D 3!& xrzz:!&##RZZ#@VX&rcD|j}|jj}d}tj|}||j_|j |||j_|j |}t ||tj|tj}||j_|j |tj||j_|j |tj}t ||yNrYr]rrKr_r)rUr^rarr`rMr rPrrrUrarrarbs rtest_output_filling_uniformzRNG.test_output_filling_uniform0s WW  &&88D>!& h !&%VX&88D 3!& hbjj 1!&BJJ7VX&rcD|j}|jj}d}tj|}||j_|j |||j_|j |}t ||tj|tj}||j_|j |tj||j_|j |tj}t ||yrf)rUr^rarr`rr rPrgs rtest_output_filling_exponentialz#RNG.test_output_filling_exponentialBs WW  &&88D>!& H-!&((d(3VX&88D 3!& HBJJ?!&((d"**(EVX&rcL|j}|jj}d}tj|}||j_|j d|||j_|j d|}t ||tj|tj}||j_|j d|tj||j_|j d|tj}t ||y)NrYrr]rrKr_r)rUr^rarzerosrSr rPrgs rtest_output_filling_gammazRNG.test_output_filling_gammaTs WW  &&88D>!& #8,!&""3T"2VX&88D 3!& #82::>!&""3T"DVX&rc||j}|jj}d}tjddz}tj |}||j_|j ||||j_|j ||}t||tj |tj}||j_|j ||tj||j_|j ||tj}t||y) NrYg@X@rr]rrKr_r) rUr^rarr-rlrSr rP)rrrUrarmurarbs r#test_output_filling_gamma_broadcastz'RNG.test_output_filling_gamma_broadcastfs WW  && YYt_s "88D>!& "(+!&""2D"1VX&88D 3!& "("**=!&""2D "CVX&rcr|j}d}tj|}tjt 5|j |tjdddtjt5|j |ddddddtj|tj}tjt 5|j |tjdddtj|tj}tjt 5|jd|tjdddtjt5|jd|dddtjdddtj|tj}tjt 5|jd|tjdddtjt5|jd|ddddddy#1swYxYw#1swYxYw#1swY|xYw#1swY xYw#1swYxYw#1swYxYw#1swYyxYw)NrYr_rr]rKr) rUrr`r{raisesrnrOrP ValueErrorrTrlrS)rrrUrras rtest_output_fill_errorzRNG.test_output_fill_errorys WW88D> ]]9 % ?   82::  > ? ]]: & 2   8CaC=  1 288D 3 ]]9 % ?   82::  > ?88D 3 ]]9 % C   cxrzz  B C ]]: & H   cx!}BJJ  G H88D 3 ]]9 % C   cxrzz  B C ]]: & 6   cx!}  5 6 6! ? ? 2 2 ? ? C C H H C C 6 6sS"I!I."I;#J )J:#J!>J-!I+.I8;JJJ!J*-J6c~|tjk(rd}d}n9tj|}t|jdz}|j }|j |jj||gdz|}|j |jj|gdz||}t|||j |jj||d|}t|||j |jjtj|gdztj|gtd|}t|||j |jjtj|gdztj|gdzd|} t|| |j |jjd|d|}|j |jj|gdz|}t||y)Nr,rr)rrKr) rbooliinfointr0minrsrUrQr r/object) rrrupperlowerinfor2r3r4r5r6s rtest_integers_broadcastzRNG.test_integers_broadcasts BGG EE88E?DMA%EHHE  GG  UUGbL  >  GG  eWr\5  >Q  GG  UE%  @Q  GG  RXX GbL88UG6:   Q  GG   HHeWr\ "BHHeWr\$:   Q  GG  QBe  <  GG  eWr\  7Qrctjdg}tjdg}|jj|||}|jdk(sJ|jj|d||}|jdk(sJ|jj||d|}|jdk(sJy)Nr)rrK)r))rr/rUrQr!)rrrhighlowr^s rtest_integers_numpyzRNG.test_integers_numpysxx}hhsmggsD6yyD   ggs1vt59yyD   ggsDG59yyD   rc6|tjk(rd}d}n9tj|}t|jdz}|j }t jt5|jj||dzgdz|dddt jt5|jj|dz |gdz|dddt jt5|jj|dz g|gdz|dddt jt5|jjdgdg|dddy#1swYxYw#1swYxYw#1swYdxYw#1swYyxYw)Nr,rr)rrK) rrvrwrxr0ryr{rrrsrUrQ)rrrr{r|r}s rtest_integers_broadcast_errorsz"RNG.test_integers_broadcast_errorss^ BGG EE88E?DMA%EHHE ]]: & C GG  UUQYK"$4E  B C ]]: & C GG  UQY" E  B C ]]: & E GG  eai[5'B,e  D E ]]: & 4 GG  aS1#U  3 4 4  C C C C E E 4 4s0+&E+2&E79'F!F+E47FF FN)Jrz __module__ __qualname__ classmethodrgrdrsrvr~rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr rrrrrrr!r$r(r,r5r8r<rCrIrNrPrTrWrdrhrjrmrprtr~rrrrrYrY_s## 9.FC"  !8*) /I (*%1 9 M##". . (!$ & !!4 !!!!!!!!!"!!1 !!!*% % (' $ 8$0L M M M M','$'$'$'&6.B !4rrYc"eZdZedZdZy) TestMT19937ct|_d|_dg|_t |j|j|_|j jj |_d|_|jt|_ y)Ni!!rE) rr^r_r`rrUrarbrcrdrsrores rrgzTestMT19937.setup_classsl# 23,3,,chh78FF0066! #rcptjj}|jd|j }||j j _|j j j}t|d|ddk(jt|d|ddk(y)Ncr)rarEr,pos) rrM RandomStaterO get_staterUr^rar r)rrnprgrarCs rtest_numpy_statezTestMT19937.test_numpy_statesyy$$& R  &+#&&,,qVG_U3388:;aF7OE223rN)rzrrrrgrrrrrrs$$4rrceZdZedZy) TestPhiloxct|_d|_dg|_t |j|j|_|j jj |_d|_|jyNlr[r\) r r^r_r`rrUrarbrcrdres rrgzTestPhilox.setup_classsd"/ 7,3,,chh78FF0066! rNrzrrrrgrrrrrrrceZdZedZy) TestSFC64ct|_d|_dg|_t |j|j|_|j jj |_d|_|jy)Nr[) rr^r_r`rrUrarbrcrdres rrgzTestSFC64.setup_classsd! 7,3,,chh78FF0066" rNrrrrrrrrrceZdZedZy) TestPCG64ct|_d|_dg|_t |j|j|_|j jj |_d|_|jyrr]res rrgzTestPCG64.setup_classsd!/ 7,3,,chh78FF0066! rNrrrrrrrrrceZdZedZy) TestPCG64DXSMct|_d|_dg|_t |j|j|_|j jj |_d|_|jyr) rr^r_r`rrUrarbrcrdres rrgzTestPCG64DXSM.setup_class sd%/ 7,3,,chh78FF0066! rNrrrrrr rrrc(eZdZedZdZdZy)TestDefaultRNGct|_d|_dg|_t j j |j|_|jjj|_ d|_ |jyr) rr^r_r`rrM default_rngrUrarbrcrdres rrgzTestDefaultRNG.setup_classsb"/ 7&&1FF0066! rc^tt|jjtyr)r r:rUr^rrqs rtest_default_is_pcg64z$TestDefaultRNG.test_default_is_pcg64#s  47700%89rctjjtjjdtjjdtjjdtjjdtjjddgtjt 5tjjddddtjt 5tjjddgdddy#1swYMxYw#1swYyxYw)Nr[rl -tE+I6%Yl l ) ~eR :TI*6r)rrMrr{rrrsrqs rrzTestDefaultRNG.test_seed(s  d# e$ a  DE EFH I ]]: & & II ! !" % & ]]: & / II ! !5"+ . / / & & / /s E"E E  EN)rzrrrrgrrrrrrrs  : /rr)Fr)'r> functoolsrr{numpyr numpy.randomrrrrrr numpy.testingr r r fixturervint8int16int32int64uint8uint16rrRrr$r8r<rWrYrrrrrrrrrrs  LLCCh"((BHHbhh"))RYY CDD !B  #n 4n 4b4#4,       C /S/r