idL ddlZddlZddlZddlZddlZddlZddlmZm Z m Z m Z m Z m Z mZmZmZddlmZmZddlmZdddddejd d z fd d d dZejejdkr dddddddddZn dddddddd dZejd!e"d#Zejd$Zd%ZGd&d'ZGd(d)Z Gd*d+Z!Gd,d-Z"Gd.d/Z#Gd0d1Z$Gd2d3Z%ej&'ed45Gd6d7Z(Gd8d9Z)d:Z*d;Z+d<Z,d=Z-d>Z.d?Z/d@Z0dAZ1dBZ2dS)CN) assert_ assert_raises assert_equal assert_warnsassert_no_warningsassert_array_equalassert_array_almost_equalsuppress_warningsIS_WASM)MT19937PCG64)random)gY@g333333?)?)r rg@)dr)$@))binomial geometrichypergeometric logseries multinomialnegative_binomialpoissonzipf@2fbead005fc63942decb5326d36a1f32fe2c9d32c904ee61e46866b88447c263@23ead5dcde35d4cfd4ef2c105e4c3d43304b45dc1b1444b7823b9ee4fa144ebb@0d764db64f5c3bad48c8c33551c13b4d07a1e7b470f77629bef6c985cac76fcf@7b59bf2f1691626c5815cdcd9a49e1dd68697251d4521575219e4d2a1b8b2c67@d754fa5b92943a38ec07630de92362dd2e02c43577fc147417dc5b9db94ccdd3@8eb216f7cb2a63cf55605422845caaff002fddc64a7dc8b2d45acd477a49e824@70c891d76104013ebd6f6bcf30d403a9074b886ff62e4e6b8eb605bf1a4673b7@01f074f97517cd5d21747148ac6ca4074dde7fcb7acbaec0a936606fecacd93f)rrrrrrrr@8626dd9d052cb608e93d8868de0a7b347258b199493871a1dc56e2a26cacb112@8edd53d272e49c4fc8fbbe6c7d08d563d62e482921f3131d0a0e068af30f0db9@83496cc4281c77b786c9b7ad88b74d42e01603a55c60577ebab81c3ba8d45657@65878a38747c176bc00e930ebafebb69d4e1e16cd3a704e264ea8f5e24f548db@7a984ae6dca26fd25374479e118b22f55db0aedccd5a0f2584ceada33db98605@d636d968e6a24ae92ab52fe11c46ac45b0897e98714426764e820a7d77602a61@956552176f77e7c9cb20d0118fc9cf690be488d790ed4b4c4747b965e61b0bb4@f84ba7feffda41e606e20b28dfc0f1ea9964a74574513d4a4cbc98433a8bfa45module)scopeparamscV|jt|jt|jfSN)param INT_FUNCSINT_FUNC_HASHES)requests X/opt/cloudlinux/venv/lib/python3.11/site-packages/numpy/random/tests/test_randomstate.pyint_funcr92s$ M9W]3 GM * ,,c#Ktj}dVtj|dS)z:Ensures that the singleton bitgen is restored after a testN)nprget_bit_generatorset_bit_generator) orig_bitgens r8restore_singleton_bitgenr@8s@)--//K EEEI ,,,,,r:cNt|d|dt|dd|ddt|dd|ddt|d|dt|d|ddS)N bit_generatorstatekeypos has_gaussgauss)rr)abs r8assert_mt19937_state_equalrJ@s?#Q%7888qz%(!G*U*;<<<qz%(!G*U*;<<<;;0007QwZ(((((r:c8eZdZdZdZdZdZdZdZdZ dS) TestSeedctjd}t|ddtjd}t|dddS)Nrii)r RandomStaterrandintselfss r8 test_scalarzTestSeed.test_scalarIs\  q ! !QYYt__c***  z * *QYYt__c*****r:ctjtd}t|ddtjt jd}t|ddtjdg}t|ddtjdg}t|dddS)NrrNirirOi )rrPrangerrQr<arangerRs r8 test_arrayzTestSeed.test_arrayOs  uRyy ) )QYYt__c***  ry}} - -QYYt__c***  s # #QYYt__c***   | , ,QYYt__c*****r:ctttjdtttjddS)Nr TypeErrorrrP ValueErrorrSs r8test_invalid_scalarzTestSeed.test_invalid_scalarYs2i!3T:::j&"4b99999r:cTtttjdgtttjdgtttjdgtttjgdtttjgddS)Nr[r\r)rr)rcrr]r`s r8test_invalid_arrayzTestSeed.test_invalid_array^si!3dV<<<j&"4rd;;;j&"4zlCCCj&"46H6H6HIIIj&"46I6I6IJJJJJr:ctttjt jgtjtttjgdgtttjgdgdgdS)Ndtypercr)r)rr_rrPr<arrayint64r`s r8test_invalid_array_shapez!TestSeed.test_invalid_array_shapefsj&"4bhrEGX7O7O7O P P Pj&"4yyykBBBj&"4yyy7@yy7B C C C C Cr:ctjtd}tt5|dddddS#1swxYwYdS)Nr)rrPr rr^seed)rSrss r8test_cannot_seedzTestSeed.test_cannot_seedns  a ) ) 9 % %   GGDMMM                  sAA AcPtttjtdSr3)rr_rrPr r`s r8test_invalid_initializationz$TestSeed.test_invalid_initializationssj&"4g>>>>>r:N) __name__ __module__ __qualname__rUrYrarerortrvr:r8rLrLHs+++ +++::: KKKCCC ?????r:rLceZdZdZdZdS) TestBinomialctjdd}dD]K}ttjd|dkt tj|||LdS)Nrintrg)rrrcr)r<zerosrrrr)rSrps r8 test_n_zerozTestBinomial.test_n_zeroxss%((( A AA FOAq))Q. / / / vua88% @ @ @ @ A Ar:c\tttjdtjdSNrc)rr_rrr<nanr`s r8 test_p_is_nanzTestBinomial.test_p_is_nans j&/1bf=====r:N)rwrxryrrrzr:r8r|r|ws5AAA>>>>>r:r|c>eZdZdZdZdZdZdZdZdZ dZ d S) TestMultinomialc4tjdddgdS)Nr皙?皙?rrr`s r8 test_basiczTestMultinomial.test_basics3c +++++r:c4tjdgddS)Nr)rrrrrr`s r8test_zero_probabilityz%TestMultinomial.test_zero_probabilitys"3 9 9 9:::::r:c&tdtjddcxkodknctjddd}ttjd|kttj|dkdS)Nr\rl)rrrQr<allrSxs r8test_int_negative_intervalz*TestMultinomial.test_int_negative_intervalsfnR,,1111r1111222 N2r1 % %rQw   q2vr:c ddg}ttjd|tjdjdttjd|tjdjdttjd|tjdjdttjd|ddgjdttjd|djdttjd|tjdjdtttjd|tddS)Nrrcrcrrrrrrr) rrrr<uint32shapermrr^floatrSrs r8 test_sizezTestMultinomial.test_sizes6 #JV'1bill;;A6JJJV'1bill;;A6JJJV'1bill;;A6JJJV'1q!f55;YGGGV'1f55;YGGGV'1bhv.>.>??E    i!3QAhh     r:ctttjdddgtttjdddgdS)Nr皙?r皙?rr_rrr`s r8test_invalid_probz!TestMultinomial.test_invalid_probs>j&"4cC:FFFj&"4cC:FFFFFr:cLtttjdddgdS)Nr\rrrr`s r8test_invalid_nzTestMultinomial.test_invalid_ns#j&"4b3*EEEEEr:cptjd}|tj|dddz}|ddd}tjdtjd|}tjdtjdtj|}t||dS)Ng.@rcrji۠iUr)pvals)r<rXsumrrrrascontiguousarrayr)rSrr non_contigcontigs r8test_p_non_contiguousz%TestMultinomial.test_p_non_contiguouss IcNN RVAaddG__!$Q$ J'5999  J#Cr/CE/J/JKKK:v.....r:ctjgdtj}||z }d}t jt |5tjd|ddddS#1swxYwYdS)N) Gz?r& .>rrrrrrrrgz-[\w\s]*pvals array is cast to 64-bit floatingmatchrc) r<rmfloat32rpytestraisesr_rr)rSrrrs r8test_multinomial_pvals_float32z.TestMultinomial.test_multinomial_pvals_float32s H:::AC M M MAEEGG @ ]:U 3 3 3 ) )  q% ( ( ( ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) )sA::A>A>N) rwrxryrrrrrrrrrzr:r8rrs,,,;;;      GGGFFF///)))))r:rcPeZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d S) TestSetStatecd|_tj|j|_|j|_dSNiI)rrrrP random_state get_staterCr`s r8 setup_methodzTestSetState.setup_methods8 ".ty99&0022 r:c|jd}|j|j|jd}t t j||kdS)N)rtomaxint set_staterCrr<rrSoldnews r8rzTestSetState.test_basicsi((,, ##DJ///((,,scz""#####r:c|jd}|j|j|jd}t t j||kdSNrjsize)rstandard_normalrrCrr<rrs r8test_gaussian_resetz TestSetState.test_gaussian_resetso//Q/77 ##DJ/////Q/77scz""#####r:cT|j|j}|jd}|j||jd}t t j||kdSr)rrrrrr<r)rSrCrrs r8 test_gaussian_reset_in_media_resz-TestSetState.test_gaussian_reset_in_media_ress ))+++!++--//Q/77 ##E***//Q/77scz""#####r:c|jdd}|jd}|j||jd}|j|j|jd}t t j||kt t j||kdS)Nrdrr)rCrrrrr<r)rS old_statex1x2x3s r8test_backwards_compatibilityz)TestSetState.test_backwards_compatibilitysJssO   . .B . 7 7 ##I...   . .B . 7 7 ##DJ///   . .B . 7 7rRx  !!!rRx  !!!!!r:c<|jdddSNr)rrr`s r8test_negative_binomialz#TestSetState.test_negative_binomials# ++C55555r:c0tjt}t5}|t }|}tt|dkddddS#1swxYwYdSr) rrPr r recordRuntimeWarningrrlen)rSrssupwrCs r8test_get_state_warningz#TestSetState.test_get_state_warnings   ( (   5C >**ALLNNE CFFaK  5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5sAB  BBc|j}d|ddz}tt|jj|tt |jjt j|t|jd}|d=tt|jj|dS)N)UnknownrcrgFlegacyrB) rrrr_rr^r<rmobject)rSrC new_states r8!test_invalid_legacy_state_settingz.TestSetState.test_invalid_legacy_state_settings!++--!E!""I- j$"3"=yIIIi!2!<hy777 9 9 9!++5+99 / "j$"3"=uEEEEEr:c|jd|jd|j|jd}t |ddt jt j|j}|d}t||dS)NrrFrrFrc) rrr random_samplerrrpickleloadsdumpsrJ)rSpickled rs_unpick unpickleds r8 test_picklezTestSetState.test_pickles q!!! '',,, ))+++#--U-;;W[)1---Ld.?!@!@AA ''u'55 "7I66666r:c|j}|j|j||jd}t ||dS)NFr)r __getstate__r __setstate__rrJ)rS attr_staterCs r8test_state_settingzTestSetState.test_state_settingsq&3355  ))+++ &&z222!++5+99":u55555r:cdSr3rzr`s r8 test_reprzTestSetState.test_reprr:N)rwrxryrrrrrrrrrrrrzr:r8rrs333 $$$ $$$ $ $ $ " " "666 555FFF777666JJJJJr:rc 4eZdZejZejejej ej ej ej ej ejejg ZdZdZdZdZdZdZejejdjdkd d Zd Zd ZdS) TestRandintcJtt|jdtdSNrcrg)rr^rfuncrr`s r8test_unsupported_typez!TestRandint.test_unsupported_types!iQe<<<<<d<Z?d=Z@d>ZAd?ZBd@ZCdAZDdBZEdCZFdDZGdEZHdFZIdGZJdHZKdIZLdJZMdKZNdLZOdMZPdNZQdOZRdPZSdQZTdRZUdSZVdTZWdUZXdVZYdWZZdXZ[dYS)ZTestRandomDistcd|_dSrrrr`s r8rzTestRandomDist.setup_methods  r:ctj|jtjdd}tjddgddgddgg}t ||d dS) Nrjr_\*?琛?<p?,o?H?Z?decimal)rrrrandr<rmr rSr3r0s r8 test_randzTestRandomDist.test_randsv DIQ""(02EF02EF/1CDFGG "&'2>>>>>>r:ctj|jtj}d}t||ddS)NrCrIrJ)rrrrLr rMs r8test_rand_singletonz"TestRandomDist.test_rand_singletons? DI%!&'2>>>>>>r:cHtj|jtjdd}tjddgddgddgg}t ||d tj|jtj}t ||d d dS) Nrjr!>>> DI!&'$-DDDDDDr:ctj|jtjddd}tjddgddgd d gg}t ||dS) Ncrjrrrj))rrrrQr<rmrrMs r8 test_randintzTestRandomDist.test_randintsl DIRf555(RG "I #J()) 67+++++r:ctj|jt5}|t}tjddd}t t|dkdddn #1swxYwYtj ddgdd gd d gg}t||tj|jt5}|t}tjd d}t t|dkdddn #1swxYwYt||d zdS)Nr]r^r_rrcr`rjrarbrcrdr) rrrr rDeprecationWarningrandom_integersrrr<rmr)rSrrr3r0s r8test_random_integersz#TestRandomDist.test_random_integerss DI   !C -..A+C&AAAF CFFaK  ! ! ! ! ! ! ! ! ! ! ! ! ! ! !(RG "I #J()) 67+++ DI   !C -..A+Cf===F CFFaK  ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! 67S=11111s%ABB  B %AEE Ec(tj|jtj|j}|d}t jt jdkr*t jddgddgdd ggtj }n)t jd d gd dgddggtj }t||||j|}t||ddS)Nr_rii4Oi˕+iRKiriFbqirglK+W+ylr&&J]l; lnBz l@XHTlD+l++rX) rrrrPrr<rr~rrmrnr)rSrsr3r0s r8 test_tomaxintzTestRandomDist.test_tomaxints DI   * *&)) 8C==  * *hi 8!+i 8!+i 8 :ACKKKGGh!46I J!46I J!46I J L&(X///G VW%%%  VWT]+++++r:c Zt5}|t}tjt jdjt jdj}tt|dkdddn #1swxYwYt jdj}t||t5}|t}t j dj }tj|t jdj|t jdj}tt|dkdddn #1swxYwYt||dS)Nr,rc) r rrhrrir<rrrrrrhr9)rSrrr3r0typers r8test_random_integers_max_intz+TestRandomDist.test_random_integers_max_ints   !C -..A+BHSMM,=,.HSMM,=??F CFFaK  ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! (3--#VW%%%   !C -..AHSMM&E+EE"(3--2C,D,D,1E"(3--2C,D,DFFF CFFaK  ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! VW%%%%%s%A>BB BB)FFFc tj5tjdtt tt jtjdj t tt jtjdj tjdj ddddS#1swxYwYdS)Nerrorr,) warningscatch_warnings simplefilterrhrrrir<rrr`s r8test_random_integers_deprecatedz.TestRandomDist.test_random_integers_deprecateds  $ & & @ @  !'+= > > > , 0(3--+ - - - , 0(3--+RXc]]-> @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @sB!CC CcFtj|jtjd}tjddgddgddgg}t ||d tj|jtj}t ||d d dS) Nr_rCrDrErFrGrHrIrJrX)rrrrr<rmr rMs r8test_random_samplez!TestRandomDist.test_random_sample&s DI%f--(02EF02EF/1CDFGG "&'2>>>> DI%''!&'$-DDDDDDr:ctj|jtjdd}tjgd}t ||dS)Nrk)rrjrrjrrrchoicer<rmrrMs r8test_choice_uniform_replacez*TestRandomDist.test_choice_uniform_replace2sN DIq!$$(<<<((67+++++r:ctj|jtjddgd}tjgd}t ||dS)Nrk)皙?r}皙?r~r)rcrcrrryrMs r8test_choice_nonuniform_replacez-TestRandomDist.test_choice_nonuniform_replace8sZ DIq!';';';<<<(<<<((67+++++r:ctj|jtjddd}tjgd}t ||dS)NrkrjFreplace)rrcrjryrMs r8test_choice_uniform_noreplacez,TestRandomDist.test_choice_uniform_noreplace>sS DIq!U333(999%%67+++++r:ctj|jtjdddgd}tjgd}t ||dS)NrkrjF)r~333333?rr~rr)rrjrcryrMs r8 test_choice_nonuniform_noreplacez/TestRandomDist.test_choice_nonuniform_noreplaceDs\ DIq!U6J6J6JKKK(999%%67+++++r:ctj|jtjgdd}tjgd}t ||dS)N)rHrIcdrk)rrrrryrMs r8test_choice_nonintegerz%TestRandomDist.test_choice_nonintegerJsX DI333Q77(///0067+++++r:c tj}tt|ddtt|ddtt|ddgddggdtt|gdtt|gddddgddgg tt|ddgdgd  tt|ddgdd d g tt|ddgdd d g tt|gdddtt|gdddtt|gdddtt|gdddtt|gdddgddS)Nr\rj@rcrrk)rcrrjrkg?r)r}r}rrrr}riFrrd)r\)r\rc)rcrrr)rrzrr_)rSr:s r8test_choice_exceptionsz%TestRandomDist.test_choice_exceptionsPsj&"a000j&"a000j&Aq6Aq6*:A>>>j&"a000j&,,,d|4 6 6 6 6j&1a&!GGGGj&1a&!T{CCCCj&1a&!SzBBBBj&)))QFFFFj&)))RGGGGj&)))UEJJJJj&)))WeLLLLj&)))Q#yyy 2 2 2 2 2 2r:c  ddg}ttjtjddttjtjddttjtjdd|ttjtjdd|ttjtjddgdttjdgddutjddg}tjdt }||d <ttj|d|ut}ttjtjd|d ttjtjd|d ttjtjd|d| ttjtjd|d| ttjtjddg|d ttjdg|dj d ktjddg}tjdt }||d <ttj||d |ud }gd }ttjd |dj |ttjd |dj |ttjd |d|j |ttjd |d|j |ttjtj d |dj |ttjd d dj dttjd dd j dttjddd j dttjd d j dttjgdj dttjddgdj dtt tjgddS)Nr~rrTrFrrcrgr)rrj)r~r~r~r~r}rr)rjrrkri)rrrHrI)rr<isscalarrrzrmemptyrtuplendimitemrrrXrQrr_)rSrrHarrrTs r8test_choice_return_shapez'TestRandomDist.test_choice_return_shapecs #J FM!T:::;;<<< FM!U;;;<<=== FM!TQ???@@AAA FM!Ua@@@AABBB FM1a&$???@@AAA tfd333t;<<< HaV  hq'''A c4000A5666 GGBK aD A A ABBBCCCBK aE B B BCCCDDDBK aDA F F FGGGHHHBK aEQ G G GHHHIIIBK q!fa F F FGGGHHH tfa666;q@AAA HaV  hq'''A c1d33388::a?@@@  * * *V]1a666>>V]C:I>>>D   j&-R88888r:c~tjgd}gd}tttj||dS)N)*rcr)NNNr)r<rmrr_rrz)rSrHrs r8test_choice_nan_probabilitiesz,TestRandomDist.test_choice_nan_probabilitiess? HZZZ    j&-a888888r:c ltjddz }d|ddd<tj|jtjdd|ddd}tj|jtjddtj|ddd}t ||dS)Nrrlrrcrrjr)r<onesrrrrzrr)rSrrrs r8test_choice_p_non_contiguousz+TestRandomDist.test_choice_p_non_contiguouss GBKK!O!$Q$ DI]1a1SSqS6222  DIq!r';AcccF'C'CDDD:v.....r:ctj|jtjd}d}t||dS)Nrs Ui+Wf)rrrbytesrrMs r8 test_byteszTestRandomDist.test_bytess> DIb!!.VW%%%%%r:c ddddddddd d d f D][}tj|j|gd }tj||}|gd }t||\dS)Nc*tjgSr3)r<rmrs r8z-TestRandomDist.test_shuffle..srx||r:c|Sr3rzrs r8rz-TestRandomDist.test_shuffle..sqr:cdtj|tjSr3)r<asarrayastyperrs r8rz-TestRandomDist.test_shuffle..srz!}}33BG<<r:cdtj|tjSr3)r<rrrrs r8rz-TestRandomDist.test_shuffle..srz!}}33BJ??r:cdtj|tjSr3)r<rr complex64rs r8rz-TestRandomDist.test_shuffle..srz!}}33BLAAr:cZtj|tSr3)r<rrrrs r8rz-TestRandomDist.test_shuffle..srz!}}33F;;r:cd|DS)Ncg|]}||fSrzrz.0is r8 zATestRandomDist.test_shuffle....s2221A222r:rzrs r8rz-TestRandomDist.test_shuffle..s22222r:c>tjd|DS)Ncg|]}||gSrzrzrs r8rzATestRandomDist.test_shuffle....*=*=*=aAq6*=*=*=r:)r<rrs r8rz-TestRandomDist.test_shuffle..srz*=*=1*=*=*=>>r:c8tj||gjSr3)r<vstackTrs r8rz-TestRandomDist.test_shuffle..sry!Q002r:ctjd|DdtfdtfgtjS)Ncg|]}||fSrzrzrs r8rzATestRandomDist.test_shuffle....s+>+>+>qQF+>+>+>r:rHrI)r<rr~r&recarrayrs r8rz-TestRandomDist.test_shuffle..sG +>+>A+>+>+>-0#Jc +C!E!E!%bk!2!2r:cntjd|DdtdfdtjdfgS)Ncg|]}||fSrzrzrs r8rzATestRandomDist.test_shuffle....rr:rHrcrI)r<rrrrs r8rz-TestRandomDist.test_shuffle..sArz*=*=1*=*=*=,/+>,/4+@+B C Cr: rcrrjrkrlrr r rrcrrrrkrlrrrj)rrrshuffler)rSconvalistr3r0s r8 test_shufflezTestRandomDist.test_shuffles,+ [<<??AA;;22>>2244CCD 0 0D" K " " "D77788E N5 ! ! !Fd999::G vw / / / /- 0 0r:ctjtjt dddzdz d}tjtjddzdz d}|}|}t dD]}tj|tt|j |j t|j |j tj|tt|j |j t|j |j d}dS)Nr)rlrkrjrcr\2cntjd}tttj|dS)Nrj)r<rmrr^rrrs r8test_shuffle_invalid_objectszHTestRandomDist.test_shuffle_masked..test_shuffle_invalid_objectss) A )V^Q 7 7 7 7 7r:) r<ma masked_valuesreshaperWrXcopyrrrsorteddatamask)rSrHrIa_origb_origrrs r8test_shuffle_maskedz"TestRandomDist.test_shuffle_maskedsL E   599f = = AA Er J J E   "  1A 5r : :r L LA N1    qvqvg'' V[L0I)J)J L L L N1    qvqvg'' V[L0I)J)J L L L L 8 8 8 8 8r:ctj|jgd}tj|}gd}t||tj|jt jgdj}tj|}t|t j|jtj|jd}tttj|tj|jd}tttj|d}gd}tj|jtj|}t||dS)Nrrabcdg333333?r) rrrrlrcrjrkrrr) rrr permutationrr< atleast_2drr IndexError)rSrr3r0arr_2d bad_x_str bad_x_float integer_vals r8test_permutationzTestRandomDist.test_permutationsB DI...#E**00067+++ DI===>>@#F++62=#9#9#;<<< DI j&"4i@@@ DI j&"4kBBB 000 DI#K0067+++++r:ctj|jtjddd}tjddgddgd d gg}t ||d dS) Nr~rr_rgAKč?ghA?g ҇o>gάXf+q?gS1$?g'8zX ?rIrJ)rrrbetar<rmr rMs r8 test_betazTestRandomDist.test_betas{ DIR&111()+BC)+BC)+BCEFF "&'2>>>>>>r:c@tj|jtjddd}tjddgddgd d gg}t ||tj|jtjdd}d}t ||dS) NgZd;Y@gv/?r_r%+r0.-)rrrrr<rmrrMs r8 test_binomialzTestRandomDist.test_binomials DI$V<<<(RHHH&'' 67+++ DI$//67+++++r:ctj|jtjdd}tjddgddgdd gg}t ||d dS) Nrr_rg+Q]uO@g+Q@gY7`ZqP@ge(fG@gvtR@gEIȋR@ rJ)rrr chisquarer<rmr rMs r8test_chisquarezTestRandomDist.test_chisquaresz DI!"6222(13GH13GH02FGIJJ "&'2>>>>>>r:c(tj|jtjddg}tj|d}tjddgddggd d gd d ggd dgddggg}t ||dtjddg}t ttj|tj|jtjddg}tj|}t ||dddS)NI>>>Hgx011 j&"2I>>> DI.0DEFF!%((!&'$-DDDDDDr:ctjddg}ttj|tjdjdttj|tjdjdttj|tjdjdttj|ddgjdttj|djdttj|tjdjdtttj|tddS)Nrrrcrrrr) r<rmrrrrrrr^rrs r8test_dirichlet_sizez"TestRandomDist.test_dirichlet_sizes! H*,@A B BV%a166>>>>>r:cttjddtttjddSNrscale)rrrrr_r`s r8test_exponential_0z!TestRandomDist.test_exponential_0Es>V'a000!444j&"4C@@@@@@r:ctj|jtjddd}tjddgddgd d gg}t ||d dS) N Mr_rg7?g`j?g6"+?gܳ2>?gc>+Y?g_}S?rIrJ)rrrfr<rmr rMs r8test_fzTestRandomDist.test_fIs{ DI"bv...(02EF02EF02EFHII "&'2>>>>>>r:ctj|jtjddd}tjddgddgd d gg}t ||d dS) Nrlrjr_rgԏGM8@gȌ<@g ":@gv#)@gZQ?@gr[@@rJ)rrrgammar<rmr rMs r8 test_gammazTestRandomDist.test_gammaQs{ DIa000(13GH13GH13GHJKK "&'2>>>>>>r:cttjdddtttjdddS)Nr)rrr)rrrrr_r`s r8 test_gamma_0zTestRandomDist.test_gamma_0Ys@V\333Q777j&,cEEEEEEr:ctj|jtjdd}tjddgddgddgg}t ||dS) N_c97ݚ?r_rrrrlr )rrrrr<rmrrMs r8test_geometriczTestRandomDist.test_geometric]sk DI!*6:::(QFHG%&& 67+++++r:cHtttjdtttjdgdztttjdtttjdgdzt 5}|t tttjtjtttjtjgdzddddS#1swxYwYdS)Nrrr) rr_rrr rrr<rrSrs r8test_geometric_exceptionsz(TestRandomDist.test_geometric_exceptionses(j&"2C888j&"2SEBJ???j&"2D999j&"2TFRK@@@   GC JJ~ & & & *f&6 ? ? ? *f&62 F F F G G G G G G G G G G G G G G G G G GsA3DDDctj|jtjddd}tjddgddgd d gg}t ||d dS) Nr@r_locrrgؕ?gs0?g#0g(xg-KD?gůp`@rIrJ)rrrgumbelr<rmr rMs r8 test_gumbelzTestRandomDist.test_gumbelos{ DI:SvFFF(02EF02FG02FGIJJ "&'2>>>>>>r:cttjddtttjddSr)rrrrr_r`s r8 test_gumbel_0zTestRandomDist.test_gumbel_0w<V]+++Q///j&-s;;;;;;r:ctj|jtjdddd}tjddgddgddgg}t ||tjdd d d }tjgd }t ||tjd d dd }tjgd}t ||tjd dd d }tjgd}t ||tjd d dd }tjgd}t ||dS)Ng333333$@g@rr_rrrrlrrjrk)rjrjrjrjrIr )r r r r )rrrr)rrrrr<rmrrMs r8test_hypergeometricz"TestRandomDist.test_hypergeometric{sf DI&tS"6BBB(RHHF$%% 67+++&q!QQ777(<<<((67+++&r1bq999(+++,,67+++&q!QQ777(<<<((67+++&q"bq999(<<<((67+++++r:ctj|jtjddd}tjddgddgd d gg}t ||d dS) Nrrr_rg bO?g¸?gpJm @g{K*t @g:^w-gY~o?rIrJ)rrrlaplacer<rmr rMs r8 test_laplacezTestRandomDist.test_laplaces{ DIJcGGG(02EF02EF13FGIJJ "&'2>>>>>>r:cttjddtttjddSr)rrr'rr_r`s r8test_laplace_0zTestRandomDist.test_laplace_0s<V^!,,,a000j&.<<<<<>>>>>r:ctj|jtjddd}tjddgddgd d gg}t ||d dS) Nrrr_)meansigmargɁ0@go,+4FB@g)ʭ6@g5?gnxNnP@g)U@rrJ)rrr lognormalr<rmr rMs r8test_lognormalzTestRandomDist.test_lognormals| DI!z6JJJ(13GH13FG13GHJKK "&'2>>>>>>r:cttjddtttjddS)Nr)r0rcr)rrr1rr_r`s r8test_lognormal_0zTestRandomDist.test_lognormal_0s>V%A...222j&"2#>>>>>>r:ctj|jtjdd}tjddgddgddgg}t ||dS)Ngƀ@?r_)rrrrrrj)rrrrr<rmrrMs r8test_logserieszTestRandomDist.test_logseriessk DI!JV<<<(QFGF$%% 67+++++r:cdSr3rzr`s r8test_logseries_zeroz"TestRandomDist.test_logseries_zerorr:valuerr\?g@cjtjd5tjt5t j|dddn #1swxYwYtjt5t jtj|gdzdddn #1swxYwYtjt5t jtj|gdzddddddn #1swxYwYddddS#1swxYwYdS)Nignoreinvalidrr)r<errstaterrr_rrrm)rSr9s r8test_logseries_exceptionsz(TestRandomDist.test_logseries_exceptionss [ * * * > >z** ( ( ''' ( ( ( ( ( ( ( ( ( ( ( ( ( ( (z** 9 9 5'B,!7!7888 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9z** > > 5'B,!7!7!!<=== > > > > > > > > > > > > > > >  > > > > > > > > > > > > > > > > > >s|D(A D(A D(A D(5+B, D(,B0 0D(3B0 4D(4D D(D D(D D((D,/D,ctj|jtjddgdzd}tjgdgdggdgd ggd gd gg}t ||dS) NrUUUUUU?rr_r)rkrjrlrkrr)rlrrrrrc)rjrkrjrrrk)rrcrkrjrrk)rkrkrrlrrj)rkrjrkrrjrk)rrrrr<rmrrMs r8test_multinomialzTestRandomDist.test_multinomials DI#B1 6BBB(//////1//////1//////1 233 67+++++r:ctj|jd}ddgddgg}d}tj|||}tjddgddggd d gd d ggd dgddggg}t ||dtj||}tjddg}t ||dddg}ddgddgg}t ttj||ttj||dtttj||dtjddgddggtj }t5}tj||| t}dddn #1swxYwYtjd}tjd}tttj||dtttjtjd|tttj|tjdtttj|tjddS)N)rrrcrr_g4Vj?gy'@gI1?gSB#@g e<@g4W(@g6R¿?g9r v"@gс?g{#@g4xg"f"@rIrJgN}O6?gYMW"@rr<) check_validraiser~rgother)rrcrcrj)rrrmultivariate_normalr<rmr rrrrr_rr rreyer) rSr/covrr3r0rrmus r8test_multivariate_normalz'TestRandomDist.test_multivariate_normals DI1v1v+D#t<<(/1BC/1BCE/1CD/1BCE/1BC02CDF GHH "&'2>>>>+D#66(-/?@AA!&'2>>>>1v1v1v^V%?sKKK 65tS'/ 1 1 1 1 j&">>   C  &tS 1 1 1 >**A                Xa[[fQiij&"RV L L L *f&>6(R- ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) )sA5BBBctj|jtjddd}tjddgddgdd gg}t ||d tjd d d}tjddgddgddgg}t ||d tj|jtjddd}tjddgddgddgg}t ||d dS)Nrlr_)dfnoncrgG7@g!ܴ*@g1}z9?@g+驸0@g26E_r#@gI!1@rrJrrg?gTD?g?T?g+qnl?g~|D?g밬?rg{01#@gprs'@g~;$@g"dI @g*@g }',@)rrrnoncentral_chisquarer<rmr rMs r8test_noncentral_chisquarez(TestRandomDist.test_noncentral_chisquaresR DI,GGG(13GH13GH02FGIJJ "&'2>>>>,&III(03FG03FG.1DEGHH "&'2>>>> DI,GGG(.0BC/1BC/1CDFGG "&'2>>>>>>r:ctj|jtjdddd}tjddgdd gd d gg}t ||d dS)Nrlrrcr_)dfnumdfdenrSrg^~?g/+c?g". @g@g7?gE?rrJ)rrr noncentral_fr<rmr rMs r8test_noncentral_fz TestRandomDist.test_noncentral_f.s DI$1AA*0222(02EF02EF02DEGHH "&'2>>>>>>r:cztj|jtjddtj}dS)Nrlr)rWrXrS)rrrrYr<r)rSr3s r8test_noncentral_f_nanz$TestRandomDist.test_noncentral_f_nan7s1 DI$1ABFCCCr:ctj|jtjddd}tjddgddgd d gg}t ||d dS) Nrrr_rg&&n@gOG_ @g~w@gtw]տgm&5@g$>@rIrJ)rrrnormalr<rmr rMs r8 test_normalzTestRandomDist.test_normal<s{ DI:SvFFF(02EF.0DE02EFHII "&'2>>>>>>r:cttjddtttjddSr)rrr^rr_r`s r8 test_normal_0zTestRandomDist.test_normal_0Dr#r:ctj|jtjdd}tjddgddgdd gg}tj||d dS) Nrr_rHrgW I@g Ԩy@gy]0Ag1~\Aga@gY 7A)nulp)rrrparetor<rmtestingassert_array_almost_equal_nulprMs r8 test_paretozTestRandomDist.test_paretoHs DI&999()+BC)+BC)+BCEFF 11&'1KKKKKr:ctj|jtjdd}tjddgddgddgg}t ||dS)Nrr_)lamrrrc)rrrrr<rmrrMs r8 test_poissonzTestRandomDist.test_poissonWsj DIJV<<<(QFFF$%% 67+++++r:c~tjdj}d}ttt j|ttt j|gdzttt j|ttt j|gdzt5}|tttt jtj ttt jtj gdzddddS#1swxYwYdS)Nr,r\r) r<rrrr_rrr rrr)rSlambiglamnegrs r8test_poisson_exceptionsz&TestRandomDist.test_poisson_exceptions_s7#"j&.&999j&.6(R-@@@j&.&999j&.6(R-@@@   EC JJ~ & & & *fnbf = = = *fnrvhm D D D E E E E E E E E E E E E E E E E E Es2A3D22D69D6ctj|jtjdd}tjddgddgdd gg}t ||d dS) Nrr_rcg;'?gܴw*?g5?gFG?gD]?g$`?rIrJ)rrrpowerr<rmr rMs r8 test_powerzTestRandomDist.test_powerksy DI 888(02EF02EF02EFHII "&'2>>>>>>r:ctj|jtjdd}tjddgddgdd gg}t ||d dS) Nrr_)rrg"u+@gN$MB*@g7v8B4@g55@ge&@g~ӣZ1@rrJ)rrrrayleighr<rmr rMs r8 test_rayleighzTestRandomDist.test_rayleighssy DIr777(02DE13GH13GHJKK "&'2>>>>>>r:cttjddtttjddSr)rrrurr_r`s r8test_rayleigh_0zTestRandomDist.test_rayleigh_0{s<V_1---q111j&/======r:ctj|jtjd}tjddgddgddgg}t ||d dS) Nr_rgMoDL?g6([9grF=?gM' -gVgVA:Qy?rIrJ)rrrstandard_cauchyr<rmr rMs r8test_standard_cauchyz#TestRandomDist.test_standard_cauchysx DI'V444(02FG02FG13FGIJJ "&'2>>>>>>r:ctj|jtjd}tjddgddgddgg}t ||d dS) Nr_rg܁?gc!z?g&N"@gNI@gF>?g~ME?rIrJ)rrrstandard_exponentialr<rmr rMs r8test_standard_exponentialz(TestRandomDist.test_standard_exponentialsx DI,&999(02EF/1DE/1DEGHH "&'2>>>>>>r:ctj|jtjdd}tjddgddgdd gg}t ||d dS) Nrjr_)rrg)@gľ@gd],q@gPk{@g9B #1@gv @rrJ)rrrstandard_gammar<rmr rMs r8test_standard_gammaz"TestRandomDist.test_standard_gammasz DI&QV<<<(02EF02EF02CDFGG "&'2>>>>>>r:cttjddtttjddS)Nr)rr)rrrrr_r`s r8test_standard_gamma_0z$TestRandomDist.test_standard_gamma_0s>V*333Q777j&"7sCCCCCCr:ctj|jtjd}tjddgddgddgg}t ||d dS) Nr_rrRrSrTrUrVrWrIrJ)rrrrr<rmr rMs r8test_standard_normalz#TestRandomDist.test_standard_normalsx DI'V444(02EF.0CD.0CDFGG "&'2>>>>>>r:ctj|jtj}tjd}t ||ddS)NrRrIrJrYrMs r8test_randn_singletonz#TestRandomDist.test_randn_singletonsJ DI(.//!&'2>>>>>>r:ctj|jtjdd}tjddgddgdd gg}t ||d dS) Nrr_)rRrg\VH?ghU%gH$M?gLgVqdzǿg㖀?rIrJ)rrr standard_tr<rmr rMs r8test_standard_tzTestRandomDist.test_standard_tsz DI"bv666(02FG02FG13FGIJJ "&'2>>>>>>r:ctj|jtjdddd}tjddgdd gd d gg}t ||d dS)Ng{Gz@g(\u$@gףp= W4@r_)leftmoderightrg:}\)@ghرTj(@gǨL30@g6ڜA0@g6sh&@g<,@rrJ)rrr triangularr<rmr rMs r8test_triangularzTestRandomDist.test_triangulars DI"5(.000(13FG13GH13FGIJJ "&'2>>>>>>r:ctj|jtjddd}tjddgddgd d gg}t ||d dS) NGz?gGz%@r_)lowhighrg>#H@gu@g#@g}d]#@gN@gw )* @rIrJ)rrruniformr<rmr rMs r8 test_uniformzTestRandomDist.test_uniforms{ DIDu6BBB(02EF02EF02EFHII "&'2>>>>>>r:ctjdj}tjdj}tj}t t|tj dt t|dtjt t|||t t|tj gdgt t|dgtjgt jtj |d|dz dS)Nrrrcg؅W4vC)rr) r<finforrrrr OverflowErrorinf nextafter)rSfminfmaxfuncs r8test_uniform_range_boundsz(TestRandomDist.test_uniform_range_boundssx  $x  $~mTBF7A666mT1bf555mT4666mTRVG9qc:::mTA3999 2<a00td{CCCCCCr:cGddtj}tjd|}t t t j||Gddtj}tjd|}t t t j|dddS)NceZdZdZdS)GTestRandomDist.test_scalar_exception_propagation..ThrowingFloatctr3r^r`s r8 __float__zQTestRandomDist.test_scalar_exception_propagation..ThrowingFloat.__float__r:N)rwrxryrrzr:r8 ThrowingFloatr#     r:rr:ceZdZdZdS)ITestRandomDist.test_scalar_exception_propagation..ThrowingIntegerctr3rr`s r8__int__zQTestRandomDist.test_scalar_exception_propagation..ThrowingInteger.__int__rr:N)rwrxryrrzr:r8ThrowingIntegerrrr:rrc) r<ndarrayrmr&rr^rrr)rSrthrowing_floatr throwing_ints r8!test_scalar_exception_propagationz0TestRandomDist.test_scalar_exception_propagations     BJ   #++M::i$ & & &     bj   x{{''88 i!6 aKKKKKr:ctj|jtjddd}tjddgddgd d gg}t ||d dS) Nrp= ף?r_rKkappargLZI@g'N"@gkr?gk=m@gډ?g^=\?rIrJrrrvonmisesr<rmr rMs r8 test_vonmiseszTestRandomDist.test_vonmisess{ DID6BBB(02EF02EF02EFHII "&'2>>>>>>r:ctj|jtjddd}tt j|dS)Nrg g5_PG>@Br)rrrrrr<isfiniterrSrs r8test_vonmises_smallz"TestRandomDist.test_vonmises_smallsR DI Ore < < < A""$$%%%%%r:ctj|jtjddd}tjgd}t ||ddS)NrgcArjr)gA^>?g@R7?gx.rrJrrMs r8test_vonmises_largez"TestRandomDist.test_vonmises_largesg DIBc:::(44455 "&'1======r:ctj|jtjdtj}t tj|dS)Nr)rKr)rrrrr<rrisnanrs r8test_vonmises_nanz TestRandomDist.test_vonmises_nansF DI Or 0 0 0 r:ctj|jtjddd}tjddgddgd d gg}t ||d dS) Nrrr_)r/rrgZ@g- g@g8 m?gVP"?g`V?gx^&3?rrJ)rrrwaldr<rmr rMs r8 test_waldzTestRandomDist.test_walds{ DI$d@@@(02EF02EF02EFHII "&'2>>>>>>r:ctj|jtjdd}tjddgddgdd gg}t ||d dS) Nrr_rcg@6?g]A?g(1ڥR?gU?g2_u?g^}-Q?rIrJ)rrrweibullr<rmr rMs r8 test_weibullzTestRandomDist.test_weibull sy DI$V444(02EF02EF02EFHII "&'2>>>>>>r:ctj|jttjddt jdt ttjddS)Nrr rcr)rH)rrrrrr<rrr_r`s r8test_weibull_0zTestRandomDist.test_weibull_0sX DIV^ab11128B<<@@@j&.C888888r:ctj|jtjdd}tjddgddgddgg}t ||dS) Nrr_rcBrcrjr)rrrrr<rmrrMs r8 test_zipfzTestRandomDist.test_zipfsj DIt&111(RHFG%&& 67+++++r:N)\rwrxryrrNrPr[rerjrlrorurwr{rrrrrrrrrrrrrrrrrrrrrrr rrrrr r"r%r(r*r-r2r4r6r8rr< parametrizer<rrr@rCrLrrPrUrZr\r_rarirlrprsrvrxr{r~rrrrrrrrrrrrrrrrrrzr:r8r?r?sS?????? E E E,,,222$,,,(&&&, @ @ @ E E E,,, ,,, ,,, ,,, ,,, 222&,9,9,9\999 ///&&& 0006888$,,,6??? , , ,???EEE& @ @ @;;; 666???AAA??????FFF,,,GGG???<<<,,,4???===?????????,,,((( [W|r|B';';R&LMM > >NM > , , ,1%1%1%f,,,)))???*???   ???<<< L L L,,, E E E??????>>>?????????DDD?????? ?????????DDD LLL,???&&& >>> ??????999 ,,,,,r:r?ceZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZ dZ!d Z"d!S)" TestBroadcastcd|_dS)N[rAr`s r8rzTestBroadcast.setup_method%s  r:c8tj|jdSr3)rrrr`s r8set_seedzTestBroadcast.set_seed(s DIr:c*dg}dg}tj}tjgd}|||dz|}t ||d||||dz}t ||ddS)Nrrcgl ?g_x ?g =*BN?rjrrJ)rrr<rmrr )rSrrrr0r3s r8rzTestBroadcast.test_uniform+scs.(11122 q$''!&'2>>>> dQh''!&'2>>>>>>r:cdg}dg}dg}tj}tjgd}|||dz|}t ||dt t||dz|||||dz}t ||dt t|||dzdS)Nrrcr\)g-r@gYl@g-`w?rjrrJ)rr^r<rmrr rr_)rSrr bad_scaler^r0r3s r8r_zTestBroadcast.test_normal;scD (00011 a''!&'2>>>>j&#'9=== UQY''!&'2>>>>j&#y1}=====r:cdg}dg}dg}dg}tj}tjgd}|||dz|}t ||dt t||dz|t t||dz|||||dz}t ||dt t|||dzt t|||dzdS) Nrcrr\rd)gnQVf?gVKB?g v`k?rjrrJ)rrr<rmrr rr_)rSrHrIbad_abad_brr0r3s r8rzTestBroadcast.test_betaNs C C{(11122 a!eQ!&'2>>>>j$ 1555j$Au555 aQ!&'2>>>>j$q1u555j$51955555r:cdg}dg}tj}tjgd}|||dz}t ||dt t||dzdSNrcr\)g.UgZ?giPrq?gE K?rjrrJ)rrr<rmrr rr_)rSrrrr0r3s r8rzTestBroadcast.test_exponentialdsD ( (11122 UQY''!&'2>>>>j+y1}=====r:cdg}dg}tj}tjgd}|||dz}t ||dt t||dzdSr)rrr<rmrr rr_)rSr bad_shape std_gammar0r3s r8rz!TestBroadcast.test_standard_gammaqsD ) (11122 519%%!&'2>>>>j)Y];;;;;r:cdg}dg}dg}dg}tj}tjgd}|||dz|}t ||dt t||dz|t t||dz|||||dz}t ||dt t|||dzt t|||dzdS) Nrcrr\rd)g.UgZ?giPrq?gE K?rjrrJ)rrr<rmrr rr_)rSrrrrrr0r3s r8rzTestBroadcast.test_gamma~sD D  (00011 uqy%((!&'2>>>>j%Q>>>j%I>>> ueai((!&'2>>>>j%EAI>>>j% A >>>>>r:cdg}dg}dg}dg}tj}tjgd}|||dz|}t ||dt t||dz|t t||dz|||||dz}t ||dt t|||dzt t|||dzdS) Nrcrr\rd)gbyʜ?go?gT7@rjrrJ)rr r<rmrr rr_)rSrWrX bad_dfnum bad_dfdenr r0r3s r8r zTestBroadcast.test_fsD D H(00011 519e$$!&'2>>>>j!Y]E:::j!UQY ::: 5%!)$$!&'2>>>>j!Y :::j!UIM:::::r:cdg}dg}dg}dg}dg}dg}tj}tjgd}|||dz||} t | |d t t||dz||t t||dz||t t||dz||||||dz|} t | |d t t|||dz|t t|||dz|t t|||dz||||||dz} t | |d t t||||dzt t||||dzt t||||dzdS) Nrrjrkrr\rd)gr^G"@g/ *@gц!@rrJ)rrYr<rmrr rr_) rSrWrXrSrrbad_noncnonc_fr0r3s r8rZzTestBroadcast.test_noncentral_fssC D 4$(00011  5$//!&'2>>>> j&)a-EEEj&%!)YEEEj&%!)UHEEE uqy$//!&'2>>>>j&)UQYEEEj&%QEEEj&%HEEE udQh//!&'2>>>>j&)UD1HEEEj&%D1HEEEj&%1 EEEEEr:c|tjddg}tjdddd}t ||ddS)Ngaz@gn4M;%?rrrrrJ)rr<rmrrYr )rSr0r3s r8test_noncentral_f_small_dfz(TestBroadcast.test_noncentral_f_small_dfsY (-/@ABB$S#qq999!&'2>>>>>>r:cdg}dg}tj}tjgd}|||dz}t ||dt t||dzdS)Nrcr\)g~}N??g$F?g'rS?rjrrJ)rrr<rmrr rr_)rSrRbad_dfrr0r3s r8rzTestBroadcast.test_chisquaresS$ (00011 26""!&'2>>>>j)VaZ88888r:cdg}dg}dg}dg}tj}tjgd}|||dz|}t ||dt t||dz|t t||dz|||||dz}t ||dt t|||dzt t|||dzdS) Nrcrr\rd)g w"@g2 WR@gRY@rjrrJ)rrTr<rmrr rr_)rSrRrSrrnonc_chir0r3s r8rUz'TestBroadcast.test_noncentral_chisquaresSs4.(00011 "q&$''!&'2>>>>j(FQJ===j(BFH=== "dQh''!&'2>>>>j(FD1H===j(B1 =====r:c6dg}dg}tj}tjgd}|||dz}t ||dt t||dzt ttj|dzdS)Nrcr\)gT @g~Fl@gƚtp?rjrrJ)rrr<rmrr rr_)rSrRrtr0r3s r8rzTestBroadcast.test_standard_tsS  (00011 26!&'2>>>>j!VaZ000j&"3VaZ@@@@@r:cdg}dg}dg}tj}tjgd}|||dz|}t ||dt t||dz|||||dz}t ||dt t|||dzdS)Nrrcr\)gW!@gOBg}?rjrrJ)rrr<rmrr rr_)rSrKr bad_kapparr0r3s r8rzTestBroadcast.test_vonmisessSD ?(11122 "q&%((!&'2>>>>j(BFI>>> "eai((!&'2>>>>j(B A >>>>>r:c6dg}dg}tj}tjgd}|||dz}t ||dt t||dzt ttj|dzdS)Nrcr\)g&8??geFX?gՋː?rjrrJ)rrfr<rmrr rr_)rSrHrrfr0r3s r8rizTestBroadcast.test_paretos C(00011 A!&'2>>>>j&%!)444j&-;;;;;r:c6dg}dg}tj}tjgd}|||dz}t ||dt t||dzt ttj|dzdSr)rrr<rmrr rr_)rSrHrrr0r3s r8rzTestBroadcast.test_weibull%s C.(11122 Q!&'2>>>>j'519555j&.%!)<<<<>>>j%333j&, :::::r:cdg}dg}dg}tj}tjgd}|||dz|}t ||dt t||dz|||||dz}t ||dt t|||dzdS)Nrrcr\)g:BKc?g3k?g~q?rjrrJ)rr'r<rmrr rr_)rSrrrr'r0r3s r8r(zTestBroadcast.test_laplaceAscD .(22233 q%((!&'2>>>>j'37I>>> eai((!&'2>>>>j'3 A >>>>>r:cdg}dg}dg}tj}tjgd}|||dz|}t ||dt t||dz|||||dz}t ||dt t|||dzdS)Nrrcr\)g'Zy?gdSO=?g3L1?rjrrJ)rrr<rmrr rr_)rSrrrrr0r3s r8r zTestBroadcast.test_gumbelTscD (11122 a''!&'2>>>>j&#'9=== UQY''!&'2>>>>j&#y1}=====r:cdg}dg}dg}tj}tjgd}|||dz|}t ||dt t||dz|||||dz}t ||dt t|||dzttjdd ddS) Nrrcr\)g“y?gJR?g ~&?rjrrJr:r) rr,r<rmrr rr_r)rSrrrr,r0r3s r8r-zTestBroadcast.test_logisticgscD ?(22233 #'5))!&'2>>>>j(C!GY??? #uqy))!&'2>>>>j(CQ???V_S#..44444r:c(dg}dg}dg}tj}tjgd}|||dz|}t ||dt t||dz|t ttj|dz|||||dz}t ||dt t|||dzt ttj||dzdS)Nrrcr\)gH"@gܮ @gvv :@rjrrJ)rr1r<rmrr rr_)rSr/r0 bad_sigmar1r0r3s r8r2zTestBroadcast.test_lognormal{ssD $ (00011 4!8U++!&'2>>>>j)TAXyAAAj&"2D1HiHHH 4++!&'2>>>>j)T9q=AAAj&"2D)a-HHHHHr:cdg}dg}tj}tjgd}|||dz}t ||dt t||dzdS)Nrcr\)gݱlo?g/ӂ?gz~R?rjrrJ)rrur<rmrr rr_)rSrrrur0r3s r8rvzTestBroadcast.test_rayleighsD ?(00011 %!)$$!&'2>>>>j(IM:::::r:cdg}dg}dg}dg}tj}tjgd}|||dz|}t ||dt t||dz|t t||dz|t ttj|dz|t ttj|dz|||||dz}t ||dt t|||dzt t|||dzt t|d dt t|dd dS) Nrrcrrd)g@m e?g`I߿?g|2?rjrrJr)rrr<rmrr rr_)rSr/rbad_meanrrr0r3s r8rzTestBroadcast.test_waldswu3D {(00011 dQh&&!&'2>>>>j$1 e<<<j$q)<<<j&+x!|UCCCj&+taxCCC dEAI&&!&'2>>>>j$%!)<<<j$i!m<<<j$Q///j$S11111r:c2dg}dg}dg}dg}dg}|dz\}}tj}tjgd} |||dz||} t | | dt t||dz||t t||dz||t t||dz||||||dz|} t | | dt t|||dz|t t|||dz|t t|||dz||||||dz} t | | dt t||||dzt t||||dzt t||||dzt t|dd d t t|dd d t t|ddddS) Nrcrjrrk)g,P;bD@g%G@g,~S@rrJrrg4@g9@)rrr<rmrr rr_) rSrrr bad_left_one bad_mode_one bad_left_two bad_mode_tworr0r3s r8rzTestBroadcast.test_triangularsYsss s %*QY" l& (00011 D1HdE22!&'2>>>>j*lQ.>eLLLj*dQh eLLLj*lQ.>     D$(E22!&'2>>>>j*lD1HeLLLj*dL14DeLLLj*lL1>>>j*lD%!)LLLj*dL%!)LLLj*lLai ! ! ! j*c2s;;;j*c3<<<j*c3<<<<>>j)QUI>>> 1a!e$$67+++j)UAE:::j)Q A >>>j)Q A >>>>>r:cZtjj}dg}dg}|dzg}tj}t jgd}|||dz}t||tt||dztt||dzdS)Nrcr\r)rcrcrrj) rrP_poisson_lam_maxrr<rmrrrr_)rSmax_lamrk bad_lam_one bad_lam_tworr0r3s r8rlzTestBroadcast.test_poisson s$&&7cd {m .(999%% q!!67+++j';?;;;j';?;;;;;r:cdg}dg}tj}tjgd}|||dz}t ||t t||dztjd5t t|tj t t|ddtj gddddS#1swxYwYdS)Nrr)rrrcrjr<r=) rrr<rmrrrr_r?r)rSrHrrr0r3s r8rzTestBroadcast.test_zipfs C{(999%% a!e67+++j$ 222 [ * * * < < *dBF 3 3 3 *dQ26N ; ; ; < < < < < < < < < < < < < < < < < >>&666, > > > < < <???,;;;,!F!F!FF??? 9 9 9>>>, A A A???& < < < = = = ; ; ;???&>>>&555(III* ; ; ;2224&=&=&=P;;;.???. < < < < < < 7 7 7&9&9&9P < < < < .zsL 0 0 0AV86+=a+@+@!*D E E E 0 0 0r:c6g|]}|Srz)startrrs r8rz-TestThread.check_function..|s qr:c6g|]}|Srz)joinr-s r8rz-TestThread.check_function..}s ar:rkwin32) threadingr%r<rrr#ziprrPintprhitemsizer!platformr r) rSr*szout1out2rrTr)r%s ` @r8check_functionzTestThread.check_functionssF$$$$$$xTZ*R/00xTZ*R/00 0 0 0 0 0TZ.. 0 0 0A1 D)) / /DAq HV'**A . . . . 799? #q ( (S\W-D-D %dD 1 1 1 1 1 tT * * * * *r:c:d}||ddS)Nc8|d|d<dS)N'r.)r^rCouts r8 gen_randomz*TestThread.test_normal..gen_randoms|||//CHHHr:)r<r6r9rSr?s r8r_zTestThread.test_normals2 0 0 0 J844444r:c:d}||ddS)Nc\|tjd|d<dS)NrrNr.)rr<rr=s r8r?z'TestThread.test_exp..gen_randoms*((rw{/C/C(DDCHHHr:rEr@rArBs r8test_expzTestThread.test_exps5 E E E J;77777r:c:d}||ddS)NcD|ddgdzd|d<dS)NrrBrr<r.)rr=s r8r?z/TestThread.test_multinomial..gen_randoms*((fX\(FFCHHHr:)r<rr@rArBs r8rCzTestThread.test_multinomials5 G G G J:66666r:N)rwrxryrr9r_rFrCrzr:r8r!r!ms_+++,555 888 77777r:r!c&eZdZdZdZdZdZdS)TestSingleEltArrayInputctjdg|_tjdg|_tjdg|_d|_dS)Nrrjrkr)r<rmargOneargTwoargThreetgtShaper`s r8rz$TestSingleEltArrayInput.setup_methods@hsmm hsmm !   r:c tjtjtjtjtjtjtjtjtj tj tj tj f }tj tj f}|D]O}||vr|tjdg}n||j}t!|j|jPdSr)rrrrrrfrrrrurrrrr<rmrLrrrO)rSfuncs probfuncsrr>s r8test_one_arg_funcsz*TestSingleEltArrayInput.test_one_arg_funcss#V%:!6#4v!6#3 5%v'78  3 3Dy  d28SE??++d4;'' DM 2 2 2 2 3 3r:ctjtjtjtjtjtjtjtjtj tj tj tj tj tjf}tj tjf}|D]}||vrtjdg}n|j}||j|}t'|j|j||jd|}t'|j|j||j|d}t'|j|jdS)Nrr)rrr^rrr rTrr'rr,r1rrrr<rmrMrLrrrO)rSrQrRrrMr>s r8test_two_arg_funcsz*TestSingleEltArrayInput.test_two_arg_funcssfl66&.!6;&": <_f&>?  3 3Dy  3%$t{F++C DM 2 2 2$t{1~v..C DM 2 2 2$t{F1I..C DM 2 2 2 2 3 3r:ctjtjtjg}|D]}||j|j|j}t|j|j ||jd|j|j}t|j|j ||j|jd|j}t|j|j dSNr) rrYrrrLrMrNrrrO)rSrQrr>s r8test_three_arg_funcsz,TestSingleEltArrayInput.test_three_arg_funcss$f&7&( 3 3D$t{DK??C DM 2 2 2$t{1~t{DMBBC DM 2 2 2$t{DKNDMBBC DM 2 2 2 2 3 3r:N)rwrxryrrSrUrXrzr:r8rJrJsP 333&3336 3 3 3 3 3r:rJctjd|\}}}tt|}||ddi}t|jt jdkdS)Nrrrr,)rrrgetattrrrhr<)r9fnamer(r%r r3s r8test_integer_dtyper\sd K "E4A Q 1  F FLBHSMM )*****r:chtjd|\}}}tt|}||ddi}tjdkr|}t j|tj  }t||kdS)Nrrrr ) rrrrZr!r"r#r$r%r&r<rr'r)r9r[r(r%r r)r*s r8test_integer_repeatr^s K "E4A !T  C }  llnn ."'** + + 5 5 7 7C C6Mr:ctjt5tjdddgddddn #1swxYwYtjt5tjddgdddddn #1swxYwYtjt5tjddgddgdddddS#1swxYwYdS)Nrcrgffffff?)rrcrr)rrr_rrrzr:r8test_broadcast_size_errorr`s z " "44C:F3333444444444444444 z " "22A&1111222222222222222 z " "99Ac 8888999999999999999999s5AAA$B  BB.CCCcptjtd}|d|\}}}||dd}|||d}t|d|dt|dd|ddt|dd|ddt|d |d t|d |d dS) NrrcFrrBrCrDrErFrG) r<rrPr r __reduce__rrrr)rsctorr(state_arIstate_bs r8&test_randomstate_ctor_old_style_picklerfs   wqzz * *Bq--//D$ d2A2hAKKkkk''G)7?+CDDDww'.0@0GHHHww'.0@0GHHH%w{';<<<!77#344444r:ctjd}|j}tj|tj}dSrW)r<rrrBr>r=)r@def_bgbg second_bgs r8 test_hot_swaprksN Y " "1 % %F  BI### ++--IIIr:c\td}tj|tjd}tjdtjd}t |t |dS)NrFrrc)r r<rr>rrrprint)r@rirCrs r8test_seed_alt_bit_genrns qBI### I  u  - -EINN1 ##5#11I %LLL )r:cDtj}td}tj|t jtd5tj|ddddS#1swxYwYdS)Nrzstate must be for a PCG64r) r<rrr r>rrr_r)r@rCris r8test_state_error_alt_bit_genrp)s I   ! !E qBI### z)D E E E## E"""##################s( BBBctjdtjddd}t d}|j}tj|tjd}|D]}tjdtjddd}|j}dS)Ńr@rFr)r<rrrrQr rCr>r)r@rrirC state_directfieldpcg_valsrs r8test_swap_workedrw2sINN5 9  Q , ,D qB HEI###9&&e&44L33INN5y  GR00HIIIr:cFtjtdtjddd}tjtd}|ddd}t ||dS)Nrrrrsr)r<rr>r rQrPr)r@singleton_valsrgnon_singleton_valss r8%test_swapped_singleton_against_directr|DsIe ---Y&&q'266N   uU|| , ,BAw33#^44444r:)3r$rr!rrnumpyr<r numpy.testingrrrrrrr r r numpy.randomr r rrr5rr~rr6fixturer9r@rJrLr|rrrr?rr<r=r!rJr\r^r`rfrkrnrprwr|rzr:r8rs'                        ('''''''%+c!12"+    28C==u#e$f$f)k&h,n"daOO$f$f)k$f&h,n"daOhy111,,21, ---))),?,?,?,?,?,?,?,?^ > > > > > > > >2)2)2)2)2)2)2)2)jTJTJTJTJTJTJTJTJn^+^+^+^+^+^+^+^+Bl ,l ,l ,l ,l ,l ,l ,l ,^H <H <H <H <H <H <H <H  > >###BBB$55555r: