bgsErddlZddlmZddlZddlZddlmZmZm Z m Z ddl Z ddl m Z mZmZmZmZmZmZmZmZddlmZ ddlZdZn #e$rdZYnwxYw ddlZdZn #e$rdZYnwxYwejjdkrdZej ej!e"Z#d Z$d Z%d Z&d Z'd Z(dZ)dZ*dZ+dZ,dZ-dZ.dZ/dZ0dZ1GddZ2Gdde2Z3Gdde2Z4Gdde2Z5Gdd e2Z6Gd!d"e2Z7Gd#d$Z8dS)%N)join) assert_equalassert_allcloseassert_array_equal assert_raises) GeneratorMT19937PCG64 PCG64DXSMPhilox RandomState SeedSequenceSFC64 default_rng) interfaceFTc"|D]}t||trt||||:t||tjrt ||||w||||ksJdSN) isinstancedictassert_state_equalnpndarrayr)actualtargetkeys q/builddir/build/BUILD/cloudlinux-venv-1.0.7/venv/lib64/python3.11/site-packages/numpy/random/tests/test_direct.pyrr's.. fSk4 ( ( . vc{F3K 8 8 8 8 s RZ 0 0 . vc{F3K 8 8 8 8#;&+----- ..cp|tjdz dztjS)Ngp>)ruint32astypefloat32)us ruint32_to_float32r%1s* ")A,, ; / 7 7 C CCrcptj|}tj|tjdz tj}tjd}tj||ztj}tj||g}t |S)N dtype)ruint64arrayr! column_stackravelr%)xupperlowerjoineds runiform32_from_uint64r35s ! A HQ")B--'ry 9 9 9E Ij ! !E HQYbi 0 0 0E _eU^ , , 2 2 4 4F V $ $$rctj|tjdz }tj|tjdz}t|S)Nr*)rr+r!r%r/s runiform32_from_uint53r7>sH !  " %A !bi +++,,A Q  rc t|Sr)r%r6s runiform32_from_uint32r9Ds Q  rc|dkrt|S|dkrt|S|dkrt|St)N@5r')r3r7r9NotImplementedErrorr/bitss runiform32_from_uintr@HsM rzz$Q''' $Q''' $Q'''!!rcV|dvrt|S|dkrt|SdS)N)r;?r<r')uniform_from_uint64uniform_from_uint32r>s runiform_from_uintrESs9 |"1%%% "1%%% rc6|tjdz dzS)N g<)rr+r6s rrCrCZs 2 #; <>>)]+++)3444 " "E%u';R@@@ u{{2  2 % % % % % %rcjtj}|jjd}fdt dD}d|D|ksJ|jd}fdt ddD}d|D|ksJ|d}fdt ddD}d |D}||ksJ|d |d ksJd S) z? Test spawning new generators and bit_generators directly. rJc&g|] }j|fzSrmrk.0rPseqs r z+test_generator_spawning..s"<<.s ( ( (AAK ( ( (rc&g|] }j|fzSrmrzr{s rr~z+test_generator_spawning..s"@@@aS]aT)@@@rrjc&g|]}|jjSrm)seed_seqrk)r|bgs rr~z+test_generator_spawning..s 4 4 4bBK ! 4 4 4rc&g|] }j|fzSrmrzr{s rr~z+test_generator_spawning..s"AAAaS]aT)AAArc0g|]}|jjjSrm) bit_generatorrrk)r|rngs rr~z+test_generator_spawning..s!KKK3##,6KKKrrrN)rrandomrrrrorNuniform)rnew_ss expected_keysnew_bgsnew_rngs found_keysr}s @rtest_generator_spawningrsS )   ! !C  $C YYq\\F<<<<588<<.FakeSeedSequencec.tj||S)Nr()rzeros)selfn_wordsr)s rrsz;test_non_spawnable..FakeSeedSequence.generate_states8G5111 1rN)__name__ __module__ __qualname__rr!rsrmrrFakeSeedSequencers-02  2 2 2 2 2 2rrzThe underlying SeedSequence)matchrJ) rnrgregisterrrrpytestraisesrrror)rgrrs rtest_non_spawnablers88888822222222+,,, )   0 0 2 2 3 3C y(E F F F !  y(E F F F## """##################s$+B  BB3CC"Cc8eZdZejZixZZedZ edZ dZ dZ dZ dZdZdZd Zd Zd Zd Zd ZdZdZejeddZejeddZdZdS)Basect|_d|_tj|_t |_g|_g|_ dS)Nr;) r rr?rr+r)rrseed_error_typeinvalid_init_typesinvalid_init_valuesclss r setup_classzBase.setup_classs8!I '!#"$rc t|5}|}|d}d|ddD}g}|D]P}|t |dddQ|t j||jdcdddS#1swxYwYdS)N,cRg|]$}t|d%S)r)intstrip)r|ss rr~z"Base._read_csv..s*888!C 1%%888rrrr()seeddata) openreadlinesplitr\rrrr,r))rfilenamecsvrrlines r _read_csvzBase._read_csvs% (^^ 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 KsB0C  CCc|j|jd}|d}t||jd|j|jd}|}t||jdd|j|jd}|d}t||jddS)Nrrr)rdata1 random_rawrdata2rruintss rtest_rawz Base.test_raws**DJv,>? ((..UDJv.///**DJv,>? ((**UDJv.q1222**DJv,>? ((..UDJv./////rc|j|jd}|d}|J|dd}|JdS)NrF)outputr)rrrrs rtest_random_rawzBase.test_random_raws\**DJv,>? (((66}}}((e(<<}}}}}rcd}t|j|jd}||}t |t |jd||jt|j|jd}|d}t |t |jd||jdS)Nrr)r rrstandard_normalrrer?r)rr]rsr_s rtest_gauss_invzBase.test_gauss_invs  ++TZ-?@ A A""1%%' 6(:AtyII K K K++TZ-?@ A A""2&&' 6(:AtyII K K K K KrcRt|j|jd}t|jd|j}|t |}t||t|j tj t|j|j d}t|j d|j}|t |}t||t|j tj dS)Nrr) rrrrEr?rrMrrr)rfloat64rrrvalsuniformss rtest_uniform_doublezBase.test_uniform_doubles )t)4:f+=> ? ? F!3TY??99SYY''$'''X^RZ000 )t)4:f+=> ? ? F!3TY??99SYY''$'''X^RZ00000rct|j|jd}t|jd|j}|t |tj}t||t|j tjt|j|j d}t|j d|j}|t |tj}t||t|j tjdS)Nrrr() rrrr@r?rrMrr#rrr)rrs rtest_uniform_floatzBase.test_uniform_floats )t)4:f+=> ? ?"4:f#5tyAA99SYYbj999$'''X^RZ000 )t)4:f+=> ? ?"4:f#5tyAA99SYYbj999$'''X^RZ00000rct|j|jd}dt|vsJt |dddt|vsJdSNrrz#xXr/)rrrrepridr0replacerrs r test_reprzBase.test_reprs{ )t)4:f+=> ? ?d2hh&&&&R&&~~##%%--c377488CCCCCCrcTt|j|jd}dt|vsJt|jjt|vsJt |dddt|vsJdSr)rrrstrrrr0rrs rtest_strz Base.test_strs )t)4:f+=> ? ?c"gg%%%%4%.//3r77::::R&&~~##%%--c377s2wwFFFFFFrcddl}|j|jd}|j}||}||}|j}t t|dt|d||usJt||td}|||}t|j|jdS)Nrrrd) picklerrrpdumpsloadsrrrrrr) rrrrp bitgen_pklreloadedreloaded_statessaas r test_picklezBase.test_pickles **DJv,>? #\\-00 << ++!9]33CCDII$X..>>tDD F F FH,,,,>5111 #   \\&,,r** + +RXrx(((((rc|j|jd}tjt5dh|_ddddS#1swxYwYdS)Nr1)rrrrrrrprrs rtest_invalid_state_typezBase.test_invalid_state_type-s**DJv,>? ]9 % % ( (#&%M  ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( (s AA  A c|j|jd}|j}d|d<tjt 5||_ddddS#1swxYwYdS)NrotherBitGeneratorr)rrrprr ValueError)rrrps rtest_invalid_state_valuezBase.test_invalid_state_value2s**DJv,>? #!4o ]: & & ( ("'M  ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( (sAAAc|j}|jD]8}tjt5||dddn #1swxYwY9dSr)rrrrrrrrsts rtest_invalid_init_typezBase.test_invalid_init_type9s* ) # #By)) # # r"" # # # # # # # # # # # # # # # # #s<A A c|j}|jD]?}tjtt f5||dddn #1swxYwY@dSr)rrrrr OverflowErrorrs rtest_invalid_init_valueszBase.test_invalid_init_values?s* * # #B M:;; # # r"" # # # # # # # # # # # # # # # # #sAA A c|j|jd}|d|ddtjt 5|ddddddS#1swxYwYdS)NrrrVint32)rr _benchmarkrrrrs rtest_benchmarkzBase.test_benchmarkEs**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 1sA>>BBzcffi not available)reasonc|j|jd}|j}t|tsJ|j}||usJdSNr)rrcffirr)rrcffi_interfaceother_cffi_interfaces r test_cffizBase.test_cffiLsV**DJv,>? &+.)44444,1#~555555rzctypes not availablec|j|jd}|j}t|tsJ|j}||usJdSr)rrctypesrr)rrctypes_interfaceother_ctypes_interfaces r test_ctypeszBase.test_ctypesTsY**DJv,>? (/*I66666!.!5%)9999999rc|j|jd}|j}|}t ||dSr)rrrp __getstate__r)rrrp alt_states r test_getstatezBase.test_getstate\sG**DJv,>? #!..00 5),,,,,rN) rrrrr+r)rr classmethodrrrrrrrrrrrrrrrrmarkskipif MISSING_CFFIrMISSING_CTYPESrr rmrrrrs IEEE%%[%KK[K 0 0 0 K K K 1 1 1 1 1 1DDD GGG )))"((( (((### ### 111 [ -ABB66CB6 [/EFF::GF:-----rrc*eZdZedZdZdS) TestPhiloxcNt|_d|_tj|_|ttd|_ |ttd|_ t|_ g|_ ddddddzdzfg|_dS) Nr;z./data/philox-testset-1.csvz./data/philox-testset-2.csv)rNrrrIir)r rr?rr+r)rrpwdrrrrrrrrs rrzTestPhilox.setup_classds"I MM 3 4 466 MM 3 4 466 '!##/tQ#XPQ\8R"Src|j|jd}|j}||dd|dd}t|j|jdS)Nrrpcounterr)rr)rrrpr)rrrpkeyeds r test_set_keyzTestPhilox.test_set_keyqsl**DJv,>? #""5>)+D',W~e'<#>>=. <<<< ? ? &2   &&&++b//!&   C$///++b//!&   aft!3444++b//'!!!!'!!!!!!rc$t|d}|j}|jd}d}|d|ksJ|t ddD|jd}d}|d|ksJdS)N%@ rp r%4= Kvu1/c3 K|] }d|zV dSrINrmr|rPs r z/TestPCG64.test_advange_large..&CC1CCCCCCr`r;r'r5r rIrl @H}.|zKLfberrrpr#sumrrpcgrp initial_stateadvanced_states rtest_advange_largezTestPCG64.test_advange_larges t)).99 : : '"? W~.... CCC&BCCCCCDDD '"@W~//////rNrrrr rr*r=rmrrrryK**[* " " " 0 0 0 0 0rrc0eZdZedZdZdZdS) TestPCG64DXSMcPt|_d|_tj|_|ttd|_ |ttd|_ ttf|_ ddgfdg|_dg|_dS)Nr;z./data/pcg64dxsm-testset-1.csvz./data/pcg64dxsm-testset-2.csvrrr)r rr?rr+r)rrrrrrrrrrrrs rrzTestPCG64DXSM.setup_classs%I MM$s,L"M"MNN MM$s,L"M"MNN )95"(D6)Y!?#('rct|j|jd}|jj}d}|j||d}||j_|jddz|z|d}||j_|jdddzz|z|d}||ksJ||ksJdSr r"r%s rr*z#TestPCG64DXSM.test_advance_symmetryr+rc$t|d}|j}|j}d}|dd|ksJ|t ddD|jd}d}|d|ksJdS)Nr-r.rpc3 K|] }d|zV dSr0rmr1s rr2z3TestPCG64DXSM.test_advange_large..r3rr4l oF [UmO1X9}r7r9s rr=z TestPCG64DXSM.test_advange_larges t)).99 : : ? W~g&-7777 CCC&BCCCCCDDD '"@W~//////rNr>rmrrrArAr?rrAc0eZdZedZdZdZdS) TestMT19937c8t|_d|_tj|_|ttd|_ |ttd|_ t|_ g|_ dg|_dS)Nr'z./data/mt19937-testset-1.csvz./data/mt19937-testset-2.csvr)r rr?rr!r)rrrrrrrrrrs rrzTestMT19937.setup_classsv#I MM$s,J"K"KLL MM$s,J"K"KLL (!##('rc|tt|jtjtjgtt|jtjtj gtt|jtjtjtj gtt|jtjdtjgtt|jtjgtt|jdtjgdS)Nr)rrrrrr,pi)rs rtest_seed_float_arrayz!TestMT19937.test_seed_float_arraysi!3RXrug5F5FGGGi!3RXvh5G5GHHHi!3RXrurufo5N5NOOOi!3RXq"%j5I5IJJJi!3beW===i!3aZ@@@@@rct|j|jd}|j}|j}|d}|d|dd|ddf}||_|d}t |||dz}||_|d}t ||dS)Nrirrprpos)rrY)rrrrpr$r)rrrrpdesiredtuprs rtest_state_tuplezTestMT19937.test_state_tuples )t)4:f+=> ? ?( #++g&&_%uW~e'<W~e$&! W%%VW%%%Hn! W%%VW%%%%%rN)rrrr rrKrPrmrrrGrGsN**[*AAA & & & & &rrGc$eZdZedZdS) TestSFC64cPt|_d|_tj|_|ttd|_ |ttd|_ ttf|_ ddgfdg|_dg|_dS)Nr;z./data/sfc64-testset-1.csvz./data/sfc64-testset-2.csvrrr)rrr?rr+r)rrrrrrrrrrrrs rrzTestSFC64.setup_classs!I MM 2 3 355 MM 2 3 355 )95"(D6)Y!?#('rN)rrrr rrmrrrRrRs- * *[ * * *rrRceZdZdZdZdS)TestDefaultRNGchdddddgffD]'}t|}t|jtsJ(dS)Nrmr)rWi.)rrrr )rargsrgs r test_seedzTestDefaultRNG.test_seedsP'T4L?; 7 7Dd#Bb.66 6 66 6 7 7rct}t|}|j|usJt|}||usJ|j|usJdSr)r rr)rrrYrg2s rtest_passthroughzTestDefaultRNG.test_passthroughs\ XX __2%%%%"oobyyyy B&&&&&&rN)rrrrZr]rmrrrUrUs2777 '''''rrU)9osos.pathrsysnumpyr numpy.testingrrrrr numpy.randomrr r r r r rrrnumpy.random._commonrrr ImportErrorrrflagsoptimizepathdirnameabspath__file__rrr%r3r7r9r@rErCrDrWrerwrrrrrrArGrRrUrmrrrls  ************ +*****KKKLLLLLMMMNNNNN9L goobgooh//00...DDD%%%      """&&&===###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$AAAAA&%A&