idQddlZddlZddlZddlmZmZmZmZm Z m Z m Z m Z m Z ddlmZddlZGddZGddZGdd ZGd d ZGd d ZGddZGddZeje dGddZGddZdS)N) assert_ assert_raises assert_equal assert_warnsassert_no_warningsassert_array_equalassert_array_almost_equalsuppress_warningsIS_WASM)randomc,eZdZdZdZdZdZdZdS)TestSeedctjd}t|ddtjd}t|dddS)Nrii)npr RandomStaterrandintselfss S/opt/cloudlinux/venv/lib/python3.11/site-packages/numpy/random/tests/test_random.py test_scalarzTestSeed.test_scalarsf I ! !! $ $QYYt__c*** I ! !* - -QYYt__c*****cXtjtd}t |ddtjtjd}t |ddtjdg}t |ddtjdg}t |dddS)N ririri )rr rrangerrarangers r test_arrayzTestSeed.test_arrays I ! !%)) , ,QYYt__c*** I ! !")B-- 0 0QYYt__c*** I ! !1# & &QYYt__c*** I ! !:, / /QYYt__c*****rctttjjdtt tjjddS)Nr TypeErrorrr r ValueErrorrs rtest_invalid_scalarzTestSeed.test_invalid_scalar s6i!6===j")"7<<<<>>j")"7*FFFj")"79K9K9KLLLj")"79L9L9LMMMMMrc2tttjjtjgtjtttjjgdgtttjjgdgdgdS)Ndtyper*r+))rr%rr rarrayint64r&s rtest_invalid_array_shapez!TestSeed.test_invalid_array_shape-sj")"7hr222 4 4 4j")"7)))EEEj")"7))):C)):E F F F F FrN)__name__ __module__ __qualname__rrr'r-r8rrrrsg+++ +++=== NNNFFFFFrrceZdZdZdZdS) TestBinomialctjdd}dD]K}ttjd|dkt tj|||LdS)Nr+intr/)r?r*r)rzerosrr binomialr)rrBps r test_n_zerozTestBinomial.test_n_zero7ss%((( A AA FOAq))Q. / / / vua88% @ @ @ @ A Arc\tttjdtjdS)Nr*)rr%r rCrnanr&s r test_p_is_nanzTestBinomial.test_p_is_nan@s j&/1bf=====rN)r9r:r;rErHr<rrr>r>6s5AAA>>>>>rr>c,eZdZdZdZdZdZdZdS)TestMultinomialc4tjdddgdS)Nd皙?皙?r multinomialr&s r test_basiczTestMultinomial.test_basicFs3c +++++rc4tjdgddS)NrL)rMrNrSrSrOr&s rtest_zero_probabilityz%TestMultinomial.test_zero_probabilityIs"3 9 9 9:::::rc&tdtjddcxkodknctjddd}ttjd|kttj|dkdS)Nr"r4)rr rrall)rxs rtest_int_negative_intervalz*TestMultinomial.test_int_negative_intervalLsfnR,,1111r1111222 N2r1 % %rQw   q2vrc zddg}ttjd|tjdjdttjd|tjdjdttjd|tjdjdttjd|ddgjdttjd|djdttjd|tjdjdtttjjd|tddS)NrAr*r*r+r+r+r+r+r+r+) rrr rPuint32shaper6rr$floatrrDs r test_sizezTestMultinomial.test_sizeRsV #JRY**1a1>>DfMMMRY**1a1>>DfMMMRY**1a1>>DfMMMRY**1a!Q88> JJJRY**1a88> JJJRY**1a&1A1ABBH    i!61Ahh     rc tttjjdddggtttjjddgdggtttjjddgdggdgdgggtttjjdtjddgddggdS)Nrrr*)rr%rr rPr6r&s rtest_multidimensional_pvalsz+TestMultinomial.test_multidimensional_pvals`sj")"7q!fXFFFj")"7qcA3ZHHHj")"7sQCjA3QRPS*=UVVVj")"7RX1vPQSTvFV=W=WXXXXXrN)r9r:r;rQrTrYrbrdr<rrrJrJEsd,,,;;;      YYYYYrrJc2eZdZdZdZdZdZdZdZdS) TestSetStatecd|_tj|j|_|j|_dSNiI)seedr rprng get_statestater&s r setup_methodzTestSetState.setup_methodhs6 &ty11 Y((** rc|jd}|j|j|jd}t t j||kdS)N)rjtomaxint set_staterlrrrWroldnews rrQzTestSetState.test_basicmsfi  $$ DJ'''i  $$scz""#####rc|jd}|j|j|jd}t t j||kdSNr2size)rjstandard_normalrqrlrrrWrrs rtest_gaussian_resetz TestSetState.test_gaussian_resetssli''Q'// DJ'''i''Q'//scz""#####rcT|j|j}|jd}|j||jd}t t j||kdSrv)rjryrkrqrrrW)rrlrsrts r test_gaussian_reset_in_media_resz-TestSetState.test_gaussian_reset_in_media_reszs !!### ##%%i''Q'// E"""i''Q'//scz""#####rc|jdd}|jd}|j||jd}|j|j|jd}t t j||kt t j||kdS)Nr,rorw)rlrjryrqrrrW)r old_statex1x2x3s rtest_backwards_compatibilityz)TestSetState.test_backwards_compatibilitysJssO Y & &B & / / I&&& Y & &B & / / DJ''' Y & &B & / /rRx  !!!rRx  !!!!!rc<|jdddSNrA)rjnegative_binomialr&s rtest_negative_binomialz#TestSetState.test_negative_binomials" ##C-----rN) r9r:r;rmrQrzr|rrr<rrrfrfgsn+++ $$$ $$$ $ $ $ " " ".....rrfc eZdZejjZejejej ej ej ej ej ejejg ZdZdZdZdZdZdZdZdZd S) TestRandintcJtt|jdtdSNr*r/)rr$rfuncr`r&s rtest_unsupported_typez!TestRandint.test_unsupported_types!iQe<<<<<d:Z?d;Z@d<ZAd=ZBd>ZCd?ZDd@ZEdAZFdBZGdCZHdDZIdEZJdFZKdGZLdHZMdIZNdJZOdKZPdLZQdMZRdNS)OTestRandomDistcd|_dSrhrir&s rrmzTestRandomDist.setup_method%s  rctj|jtjdd}tjddgddgddgg}t ||d dS) Nr2r+_\*?琛?<p?,o?H?Z?decimal)rr rirandr6r rrdesireds r test_randzTestRandomDist.test_rand(s~ ty!!!1%%(02EF02EF/1CDFGG "&'2>>>>>>rctj|jtjdd}tjddgddgddgg}t ||d dS) Nr2r+!>>>>>rctj|jtjddd}tjddgddgd d gg}t ||dS) Ncr2r+rwr2))rr rirr6rrs r test_randintzTestRandomDist.test_randint8sv ty!!!""3"88(RG "I #J()) 67+++++rctj|jt5}|t }tjddd}tt|dkdddn #1swxYwYtj ddgdd gd d gg}t||dS) Nrrrrwr*rr2rrrr) rr rir recordDeprecationWarningrandom_integersrlenr6rrsupwrrs rtest_random_integersz#TestRandomDist.test_random_integers@s ty!!!   !C -..AY..sBV.DDF CFFaK  ! ! ! ! ! ! ! ! ! ! ! ! ! ! !(RG "I #J()) 67+++++sABB #B ct5}|t}tjtjdjtjdj}tt|dkdddn #1swxYwYtjdj}t||dS)Nlr*) r rrrr rrrrrrrs rtest_random_integers_max_intz+TestRandomDist.test_random_integers_max_intKs   !C -..AY..rx}}/@/1x}}/@BBF CFFaK  ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! (3--#VW%%%%%sB B$$B(+B(c tj5tjdtt tt jjt jdj t tt jjt jdj t jdj ddddS#1swxYwYdS)Nerrorr) warningscatch_warnings simplefilterrrrr rrrr&s rtest_random_integers_deprecatedz.TestRandomDist.test_random_integers_deprecatedZs  $ & & @ @  !'+= > > > ,)3(3--+ - - - ,)3(3--+RXc]]-> @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @sB+C  CCctj|jtjd}tjddgddgddgg}t ||d dS) Nrrrrrrrrr)rr rir6r rs r test_randomzTestRandomDist.test_randomhs~ ty!!!!!&))(02EF02EF/1CDFGG "&'2>>>>>>rctj|jtjdd}tjgd}t ||dS)Nr3)r+r2r+r2rr richoicer6rrs rtest_choice_uniform_replacez*TestRandomDist.test_choice_uniform_replacepsX ty!!!!!!Q''(<<<((67+++++rctj|jtjddgd}tjgd}t ||dS)Nr3)皙?r皙?rrD)r*r*r+r+rrs rtest_choice_nonuniform_replacez-TestRandomDist.test_choice_nonuniform_replacevsd ty!!!!!!Q*>*>*>!??(<<<((67+++++rctj|jtjddd}tjgd}t ||dS)Nr3r2Freplace)rr*r2rrs rtest_choice_uniform_noreplacez,TestRandomDist.test_choice_uniform_noreplace|s] ty!!!!!!Q!66(999%%67+++++rctj|jtjdddgd}tjgd}t ||dS)Nr3r2F)rg333333?rArr rD)r+r2r*rrs r test_choice_nonuniform_noreplacez/TestRandomDist.test_choice_nonuniform_noreplacesl ty!!!!!!Q$8$8$8"::(999%%67+++++rctj|jtjgdd}tjgd}t ||dS)N)abcdr3)rrrrrrs rtest_choice_nonintegerz%TestRandomDist.test_choice_nonintegersb ty!!!!!"6"6"6::(///0067+++++rc tjj}tt|ddtt|ddtt|ddgddggdtt|gdtt|gddddgddgg tt|ddgdgd  tt|ddgdd d g tt|ddgdd d g tt|gdddtt|gdddtt|gdddtt|gdddtt|gdddgddS)Nr"r2g@r*r+r3)r*r+r2r3g?r)rrrMg?grr1Fr r,)r")r"r*)r*rrr)rr rrr%)rrs rtest_choice_exceptionsz%TestRandomDist.test_choice_exceptionss!j&"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 2rc  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 tjjgddS)Nr?r+Tr Frr*r/r)r+r2)rrrrrrMr5)r2rr3rwi)rrrr)rrisscalarr rr6emptyobjecttuplendimitemrr_rrrr%)rrDrarrrs rtest_choice_return_shapez'TestRandomDist.test_choice_return_shapes* #J BI,,Q,==>>??? BI,,Q,>>??@@@ BI,,Q,BBCCDDD BI,,Q,CCDDEEE BI,,aVT,BBCCDDD   $ 66$>??? HaV  hq'''A   d 33q8999 GGBK 0 0At 0 D DEEEFFFBK 0 0Au 0 E EFFFGGGBK 0 0Atq 0 I IJJJKKKBK 0 0Au 0 J JKKKLLLBK 0 0!QD 0 I IJJJKKK   $D 99>!CDDD HaV  hq'''A   a 66;;==BCCC  * * *RY%%aD%99?CCCRY%%aE%::@!DDDRY%%aDA%>>DaHHHRY%%aEQ%??EqIIIRY%%billAt%DDJANNN RY&&q!)&<<BINNNRY&&q#A&66z-TestRandomDist.test_shuffle..srx||rc|Sr+r<r,s rr-z-TestRandomDist.test_shuffle..sqrcdtj|tjSr+)rasarrayastyperr,s rr-z-TestRandomDist.test_shuffle..srz!}}33BG<<rcdtj|tjSr+)rr0r1float32r,s rr-z-TestRandomDist.test_shuffle..srz!}}33BJ??rcdtj|tjSr+)rr0r1 complex64r,s rr-z-TestRandomDist.test_shuffle..srz!}}33BLAArcZtj|tSr+)rr0r1rr,s rr-z-TestRandomDist.test_shuffle..srz!}}33F;;rcd|DS)Ncg|]}||fSr<r<.0is r zATestRandomDist.test_shuffle....s2221A222rr<r,s rr-z-TestRandomDist.test_shuffle..s22222rc>tjd|DS)Ncg|]}||gSr<r<r9s rr<zATestRandomDist.test_shuffle....*=*=*=aAq6*=*=*=r)rr0r,s rr-z-TestRandomDist.test_shuffle..srz*=*=1*=*=*=>>rc8tj||gjSr+)rvstackTr,s rr-z-TestRandomDist.test_shuffle..sry!Q002rctjd|DdtfdtfgtjS)Ncg|]}||fSr<r<r9s rr<zATestRandomDist.test_shuffle....s+>+>+>qQF+>+>+>rrr)rr0r@rrecarrayr,s rr-z-TestRandomDist.test_shuffle..sG +>+>A+>+>+>-0#Jc +C!E!E!%bk!2!2rcjtjd|DdtfdtjfgS)Ncg|]}||fSr<r<r9s rr<zATestRandomDist.test_shuffle....r?rrr)rr0rrr,s rr-z-TestRandomDist.test_shuffle..s7rz*=*=1*=*=*=,/=3/*J L Lr) r*r+r2r3r4r5r r) rr*rIr5r+r3r4rrHr2)rr rishuffler)rconvalistrrs r test_shufflezTestRandomDist.test_shuffles,+ [<<??AA;;22>>2244LLM 0 0D INN49 % % %D77788E I  e $ $ $Fd999::G vw / / / /+ 0 0rc"tjtjt dddzdz d}tjtjddzdz d}|}|}t dD]}tj|tt|j |j t|j |j tj|tt|j |j t|j |j dS)N)r4r3r2r*r"2) rma masked_valuesreshaperrcopyr rJrsorteddatamask)rrra_origb_origr;s rtest_shuffle_maskedz"TestRandomDist.test_shuffle_maskedsK E   599f = = AA Er J J E   "  1A 5r : :r L LA I  a qvqvg'' V[L0I)J)J L L L I  a qvqvg'' V[L0I)J)J L L L L  L Lrr cdddddddd}tjtd 5}|j|dddn #1swxYwYd |djvsJdS) Nrr*r+r2r3r4r5)rr*r+r2r3r4r5z!you are shuffling a 'dict' objectmatchr)pytestwarns UserWarningrJfilename)rr valuesrecs rtest_shuffle_untyped_warningz+TestRandomDist.test_shuffle_untyped_warnings1qQ1;; \+9;;; #>A FN6 " " " # # # # # # # # # # # # # # #A//////sAA Ause_array_likeTFcGddtj}dtjdgtjdtjdtjdg}tj|t}d|D|r||}t fd|DsJ|ret|tjj sFtj td 5|j |ddddS#1swxYwYdS|j |t fd |DsJdS) NceZdZdS)>TestRandomDist.test_shuffle_no_object_unpacking..MyArrN)r9r:r;r<rrMyArrrhs Drrir2rrHr/c,h|]}t|Sr<idr9s r zBTestRandomDist.test_shuffle_no_object_unpacking..s)))aBqEE)))rc3:K|]}t|vVdSr+rkr:r;item_idss r zBTestRandomDist.test_shuffle_no_object_unpacking..s.222a55H$222222rz#Shuffling a one dimensional array.*r\c3:K|]}t|vVdSr+rkros rrqzBTestRandomDist.test_shuffle_no_object_unpacking..&s.66Qr!uu(666666r)rndarrayr6float64rrrW isinstancer Generatorr^r_r`rJ)rr reriitemsr!rps @r test_shuffle_no_object_unpackingz/TestRandomDist.test_shuffle_no_object_unpacking s     BJ    "(A3--A bjmm huF+++))5)))  "((5//C2222c22222222  7*VRY5H"I"I 7k?AAA $ $s### $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ FN3   6666#66666 6 6 6 6 6sDD#&D#cztj|jtjdj}tj|t tj|gdtj|j}||t tj|gdtj |j}||t tj|gddS)Nr4)rr*r3r2r+)rr*r+r2r3)r3r*rr2r+) rr rirrVrJrr0r default_rng)rrrngs rtest_shuffle_memoryviewz&TestRandomDist.test_shuffle_memoryview(s ty!!! IaLL  !RZ]]OOO444i##DI.. ARZ]]OOO444i##DI.. ARZ]]OOO44444rctjd}d|j_t jt d5tj|ddddS#1swxYwYdS)Nr2Fz read-onlyr\) rrBflags writeabler^raisesr%r rJ)rrs rtest_shuffle_not_writeablez)TestRandomDist.test_shuffle_not_writeable7s HQKK! ]:[ 9 9 9 ! ! I  a  ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! !s A))A-0A-ctj|jtjddd}tjddgddgd d gg}t ||d dS) NrrrrwgAKč?ghA?g ҇o>gάXf+q?gS1$?g'8zX ?rr)rr ribetar6r rs r test_betazTestRandomDist.test_beta=s ty!!!BV44()+BC)+BC)+BCEFF "&'2>>>>>>rctj|jtjddd}tjddgddgd d gg}t ||dS) NrLgv/?rrw%+r$0.-)rr rirCr6rrs r test_binomialzTestRandomDist.test_binomialFsv ty!!!##CF#;;(RHHH&'' 67+++++rctj|jtjdd}tjddgddgdd gg}t ||d dS) NrPrrwg+Q]uO@g+Q@gY7`ZqP@ge(fG@gvtR@gEIȋR@ r)rr ri chisquarer6r rs rtest_chisquarezTestRandomDist.test_chisquareNs ty!!!$$Rf$55(13GH13GH02FGIJJ "&'2>>>>>>rc<tj|jtjddg}tjj|d}tjddgddggd d gd d ggd dgddggg}t ||ddS)NI>>>>>rctjddg}ttj|tjdjdttj|tjdjdttj|tjdjdttj|ddgjdttj|djdttj|tjdjdtttjj|tddS)Nrrr*r[r+r\r]) rr6rr rr^r_rr$r`ras rtest_dirichlet_sizez"TestRandomDist.test_dirichlet_sizebsA H*,@A B BRY((BIaLL99?HHHRY((BIaLL99?HHHRY((BIaLL99?HHHRY((QF3399EEERY((F3399EEERY((BHV,<,<==CYOOOi!4aqBBBBBrc tjddg}tttjjj|tttjddggtttjdgdggtttjdgdggdgdgggtttjtjddgddggdS)NgHzG?gؗҜr4r*)rr6rr%r rr)rrs rtest_dirichlet_bad_alphaz'TestRandomDist.test_dirichlet_bad_alphans'8,--j")"2">>>>>rcttjddt t tjjddSNrscale)rrr rrr%r&s rtest_exponential_0z!TestRandomDist.test_exponential_0sERY***33Q777j")"7sCCCCCCrctj|jtjddd}tjddgddgd d gg}t ||d dS) N Mrrwg7?g`j?g6"+?gܳ2>?gc>+Y?g_}S?rr)rr rifr6r rs rtest_fzTestRandomDist.test_fs ty!!!R&11(02EF02EF02EFHII "&'2>>>>>>rctj|jtjddd}tjddgddgd d gg}t ||d dS) Nr4r2rrwgԏGM8@gȌ<@g ":@gv#)@gZQ?@gr[@@r)rr rigammar6r rs r test_gammazTestRandomDist.test_gammas ty!!!AF33(13GH13GH13GHJKK "&'2>>>>>>rcttjdddt t tjjdddS)Nr)r_rr)rrr rrr%r&s r test_gamma_0zTestRandomDist.test_gamma_0sERY__1A_66:::j")/CHHHHHHrctj|jtjdd}tjddgddgddgg}t ||dS) N_c97ݚ?rrwrrHr4r)rr ri geometricr6rrs rtest_geometriczTestRandomDist.test_geometricst ty!!!$$Zf$==(QFHG%&& 67+++++rctj|jtjddd}tjddgddgd d gg}t ||d dS) Nr@rlocrrxgؕ?gs0?g#0g(xg-KD?gůp`@rr)rr rigumbelr6r rs r test_gumbelzTestRandomDist.test_gumbels ty!!!!!j&!II(02EF02FG02FGIJJ "&'2>>>>>>rcttjddt t tjjddSr)rrr rrr%r&s r test_gumbel_0zTestRandomDist.test_gumbel_0ERY%%A%..222j")"2#>>>>>>rc6tj|jtjdddd}tjddgddgddgg}t ||tjdddd }tjgd }t ||tjd dd d }tjgd }t ||tjdddd }tjgd}t ||tjdd d d }tjgd}t ||dS)Nrr4rrrwrIrr2r3)r2r2r2r2rr)rrrr)rrrr)rr rihypergeometricr6rrs rtest_hypergeometricz"TestRandomDist.test_hypergeometrics ty!!!))"a&)AA(RHHF$%% 67+++))!Q)::(<<<((67+++))"a!)<<(+++,,67+++))!Q)::(<<<((67+++))!R!)<<(<<<((67+++++rctj|jtjddd}tjddgddgd d gg}t ||d dS) Nrrrrg bO?g¸?gpJm @g{K*t @g:^w-gY~o?rr)rr rilaplacer6r rs r test_laplacezTestRandomDist.test_laplaces ty!!!""z6"JJ(02EF02EF13FGIJJ "&'2>>>>>>rcttjddt t tjjddSr)rrr rrr%r&s rtest_laplace_0zTestRandomDist.test_laplace_0sERY&&Q&//333j")"33??????rctj|jtjddd}tjddgddgd d gg}t ||d dS) Nrrrrg#K-z?gJ?g92@gر[@g#+^˿g-@rr)rr rilogisticr6r rs r test_logisticzTestRandomDist.test_logistics ty!!!## #F#KK(02DE02EF13FGIJJ "&'2>>>>>>rctj|jtjddd}tjddgddgd d gg}t ||d dS) Nrrr)meansigmarxgɁ0@go,+4FB@g)ʭ6@g5?gnxNnP@g)U@rr)rr ri lognormalr6r rs rtest_lognormalzTestRandomDist.test_lognormals ty!!!$$*Cf$MM(13GH13FG13GHJKK "&'2>>>>>>rcttjddt t tjjddS)Nr)rr*r)rrr rrr%r&s rtest_lognormal_0zTestRandomDist.test_lognormal_0sERY((q(111555j")"5SAAAAAArctj|jtjdd}tjddgddgddgg}t ||dS)Ngƀ@?r)rDrxr+r5rr2)rr ri logseriesr6rrs rtest_logserieszTestRandomDist.test_logseriesst ty!!!$$z$??(QFGF$%% 67+++++rc tj|jtjddgdzd}tjgdgdggdgd ggd gd gg}t ||dS) NrOUUUUUU?r5rrw)r3r2r4r3r+r+)r4r+rr+r+r*)r2r3r2r5rr3)r+r*r3r2r5r3)r3r3r+r4r+r2)r3r2r3r+r2r3)rr rirPr6rrs rtest_multinomialzTestRandomDist.test_multinomials ty!!!&&rD6!8&&AA(//////1//////1//////1 233 67+++++rctj|jd}ddgddgg}d}tj|||}tjddgddggd d gd d ggd dgddggg}t ||dtj||}tjddg}t ||dddg}ddgddgg}t ttjj||ttjj||dtttjj||dtjddgddggtj }t5}tj||| t}t|dksJ ddddS#1swxYwYdS)N)rrr*rrg4Vj?gy'@gI1?gSB#@g e<@g4W(@g6R¿?g9r v"@gс?g{#@g4xg"f"@rrgN}O6?gYMW"@r+ignore) check_validraiserr/)rr rimultivariate_normalr6r rRuntimeWarningrrr%r3r rr)rrcovrxrrrrs rtest_multivariate_normalz'TestRandomDist.test_multivariate_normals_ ty!!!1v1v..tS$??(/1BC/1BCE/1CD/1BCE/1BC02CDF GHH "&'2>>>>..tS99(-/?@AA!&'2>>>>1v1v1v^RY%BD#NNN 298$'/ 1 1 1 1 j")"?s") + + + +hC3(+2:>>>   C I ) )$ 4 4 4 >**Aq66Q;;;;;                  sAG##G'*G'ctj|jtjddd}tjddgddgd d gg}t ||dS) NrLg|Pk?r)nrDrxiPiIi|ici i)rr rirr6rrs rrz%TestRandomDist.test_negative_binomial"sv ty!!!,,sf6,JJ(S#J #J #J()) 67+++++rc|tj|jtjddd}tjddgddgdd gg}t ||d tjd d d}tjddgddgddgg}t ||d tj|jtjddd}tjddgddgddgg}t ||d dS)Nr4r)dfnoncrxgG7@g!ܴ*@g1}z9?@g+驸0@g26E_r#@gI!1@rrrArMg?gTD?g?T?g+qnl?g~|D?g밬?rg{01#@gprs'@g~;$@g"dI @g*@g }',@)rr rinoncentral_chisquarer6r rs rtest_noncentral_chisquarez(TestRandomDist.test_noncentral_chisquare*si ty!!!//116/JJ(13GH13GH02FGIJJ "&'2>>>>//2BV/LL(03FG03FG.1DEGHH "&'2>>>> ty!!!//116/JJ(.0BC/1BC/1CDFGG "&'2>>>>>>rctj|jtjdddd}tjddgdd gd d gg}t ||d dS)Nr4r+r*r)dfnumdfdenrrxg^~?g/+c?g". @g@g7?gE?rr)rr ri noncentral_fr6r rs rtest_noncentral_fz TestRandomDist.test_noncentral_f?s ty!!!''aqq-3(55(02EF02EF02DEGHH "&'2>>>>>>rctj|jtjddd}tjddgddgd d gg}t ||d dS) Nrrrrg&&n@gOG_ @g~w@gtw]տgm&5@g$>@rr)rr rinormalr6r rs r test_normalzTestRandomDist.test_normalHs ty!!!!!j&!II(02EF.0DE02EFHII "&'2>>>>>>rcttjddt t tjjddSr)rrr rrr%r&s r test_normal_0zTestRandomDist.test_normal_0Prrctj|jtjdd}tjddgddgdd gg}tj||d dS) NrrrrxgW I@g Ԩy@gy]0Ag1~\Aga@gY 7A)nulp)rr riparetor6testingassert_array_almost_equal_nulprs r test_paretozTestRandomDist.test_paretoTs ty!!!!!JV!<<()+BC)+BC)+BCEFF 11&'1KKKKKrctj|jtjdd}tjddgddgddgg}t ||dS)Nrr)lamrxrr*)rr ripoissonr6rrs r test_poissonzTestRandomDist.test_poissoncst ty!!!""z"??(QFFF$%% 67+++++rcttjdj}d}tttjj|tttjj|gdztttjj|tttjj|gdzdS)Nrr"r)rrrrr%r r)rlambiglamnegs rtest_poisson_exceptionsz&TestRandomDist.test_poisson_exceptionsks#"j")"3V<<<j")"3fXb[AAAj")"3V<<<j")"3fXb[AAAAArctj|jtjdd}tjddgddgdd gg}t ||d dS) Nrrrg;'?gܴw*?g5?gFG?gD]?g$`?rr)rr ripowerr6r rs r test_powerzTestRandomDist.test_powerss ty!!!:F;;(02EF02EF02EFHII "&'2>>>>>>rctj|jtjdd}tjddgddgdd gg}t ||d dS) Nrr)rrxg"u+@gN$MB*@g7v8B4@g55@ge&@g~ӣZ1@rr)rr rirayleighr6r rs r test_rayleighzTestRandomDist.test_rayleigh{s ty!!!##"6#::(02DE13GH13GHJKK "&'2>>>>>>rcttjddt t tjjddSr)rrr rrr%r&s rtest_rayleigh_0zTestRandomDist.test_rayleigh_0sERY''a'00!444j")"4C@@@@@@rctj|jtjd}tjddgddgddgg}t ||d dS) NrrwgMoDL?g6([9grF=?gM' -gVgVA:Qy?rr)rr ristandard_cauchyr6r rs rtest_standard_cauchyz#TestRandomDist.test_standard_cauchys ty!!!***77(02FG02FG13FGIJJ "&'2>>>>>>rctj|jtjd}tjddgddgddgg}t ||d dS) Nrrwg܁?gc!z?g&N"@gNI@gF>?g~ME?rr)rr ristandard_exponentialr6r rs rtest_standard_exponentialz(TestRandomDist.test_standard_exponentials ty!!!//V/<<(02EF/1DE/1DEGHH "&'2>>>>>>rctj|jtjdd}tjddgddgdd gg}t ||d dS) Nr2r)r_rxg)@gľ@gd],q@gPk{@g9B #1@gv @rr)rr ristandard_gammar6r rs rtest_standard_gammaz"TestRandomDist.test_standard_gammas ty!!!)))??(02EF02EF02CDFGG "&'2>>>>>>rcttjddt t tjjddS)Nr)r_r)rrr r rr%r&s rtest_standard_gamma_0z$TestRandomDist.test_standard_gamma_0sERY--A-66:::j")":#FFFFFFrctj|jtjd}tjddgddgddgg}t ||d dS) Nrrwrrrrrrrr)rr riryr6r rs rtest_standard_normalz#TestRandomDist.test_standard_normals ty!!!***77(02EF.0CD.0CDFGG "&'2>>>>>>rctj|jtjdd}tjddgddgdd gg}t ||d dS) Nrr)rrxg\VH?ghU%gH$M?gLgVqdzǿg㖀?rr)rr ri standard_tr6r rs rtest_standard_tzTestRandomDist.test_standard_ts ty!!!%%&%99(02FG02FG13FGIJJ "&'2>>>>>>rctj|jtjdddd}tjddgdd gd d gg}t ||d dS)Ng{Gz@g(\u$@gףp= W4@r)leftmoderightrxg:}\)@ghرTj(@gǨL30@g6ڜA0@g6sh&@g<,@rr)rr ri triangularr6r rs rtest_triangularzTestRandomDist.test_triangulars ty!!!%%4e5+1&33(13FG13GH13FGIJJ "&'2>>>>>>rctj|jtjddd}tjddgddgd d gg}t ||d dS) NGz?gGz%@r)lowhighrxg>#H@gu@g#@g}d]#@gN@gw )* @rr)rr riuniformr6r rs r test_uniformzTestRandomDist.test_uniforms ty!!!""t%f"EE(02EF02EF02EFHII "&'2>>>>>>rc<tjdj}tjdj}tjj}t t|tj dt t|dtjt t|||t t|tj gdgt t|dgtjgtjtj |d|dz dS)Nr`rr*g؅W4vC)rr) rfinforrr r r OverflowErrorinf nextafter)rfminfmaxfuncs rtest_uniform_range_boundsz(TestRandomDist.test_uniform_range_boundssx  $x  $y mTBF7A666mTABF;;;mTDD999mTRVG9qc:::mTA3999 bl433$+FFFFFrcGddtj}tjd|}t t tjj||Gddtj}tjd|}t t tjj|dddS)NceZdZdZdS)GTestRandomDist.test_scalar_exception_propagation..ThrowingFloatctr+r$r&s r __float__zQTestRandomDist.test_scalar_exception_propagation..ThrowingFloat.__float__rN)r9r:r;r0r<rr ThrowingFloatr-s#     rr2g?ceZdZdZeZdS)ITestRandomDist.test_scalar_exception_propagation..ThrowingIntegerctr+r/r&s r__int__zQTestRandomDist.test_scalar_exception_propagation..ThrowingInteger.__int__r1rN)r9r:r;r6 __index__r<rrThrowingIntegerr4s$    IIIrr8r*) rrsr6rrr$r r r)rr2throwing_floatr8 throwing_ints r!test_scalar_exception_propagationz0TestRandomDist.test_scalar_exception_propagations     BJ   #++M::i!2N$ & & &     bj   x{{''88 i!9<ANNNNNrctj|jtjddd}tjddgddgd d gg}t ||d dS) Nrp= ף?rmukapparxgLZI@g'N"@gkr?gk=m@gډ?g^=\?rr)rr rivonmisesr6r rs r test_vonmiseszTestRandomDist.test_vonmisess ty!!!##t4f#EE(02EF02EF02EFHII "&'2>>>>>>rctj|jtjddd}tjtj|dS)NrSg g5_PG>i@Br>)rr rirArrisfiniterW)rrs rtest_vonmises_smallz"TestRandomDist.test_vonmises_smallsd ty!!! I  "F  ? ? 2;q>>--//00000rctj|jtjddd}tjddgddgd d gg}t ||d dS) Nrr=r)rrrxgZ@g- g@g8 m?gVP"?g`V?gx^&3?rr)rr riwaldr6r rs r test_waldzTestRandomDist.test_walds ty!!!TFCC(02EF02EF02EFHII "&'2>>>>>>rctj|jtjdd}tjddgddgdd gg}t ||d dS) Nrrrg@6?g]A?g(1ڥR?gU?g2_u?g^}-Q?rr)rr riweibullr6r rs r test_weibullzTestRandomDist.test_weibulls ty!!!""T"77(02EF02EF02EFHII "&'2>>>>>>rctj|jttjddtjdt ttjjddS)Nrrrr)r)rr rirrKrBrr%r&s rtest_weibull_0zTestRandomDist.test_weibull_0 se ty!!!RY&&&44bhrllCCCj")"3s;;;;;;rctj|jtjdd}tjddgddgddgg}t ||dS) NrrrBr*r2r)rr rizipfr6rrs r test_zipfzTestRandomDist.test_zipfsq ty!!!$V44(RHFG%&& 67+++++rN)Sr9r:r;rmrrrrrrrrr r rrrr"r%r(rMrZr^mark parametrizerr rrzrdrxr|rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr rrrrrr!r*r;rBrFrIrLrNrSr<rrrr!sG??????,,, , , , & & & @ @ @???,,, ,,, ,,, ,,,,,, 222&,<,<,<\<<< &&& 0004 L L L [X Y --//1F1F1H1H IKK00KK0 [X BI))++RY-B-B-D-DEGG [-e}==77>=GG70 5 5 5!!! ???,,,??? ? ? ? C C C P P P???DDD??????III,,,??????,,,4???@@@??????BBB,,, , , ,&&&P,,,???*????????? L L L,,,BBB??????AAA?????????GGG????????????GGG OOO0???111 ??????<<< ,,,,,rrceZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZ dZ!d Z"d!S)" TestBroadcastcd|_dS)Ni[rr&s rrmzTestBroadcast.setup_methods  rcNtj|jdSr+)rr rir&s rsetSeedzTestBroadcast.setSeeds ty!!!!!rc4dg}dg}tjj}tjgd}|||dz|}t ||d||||dz}t ||ddS)Nrr*gl ?g_x ?g =*BN?r2rr)rr r r6rZr )rrrr rrs rr!zTestBroadcast.test_uniform$scs)#(11122 q$''!&'2>>>> dQh''!&'2>>>>>>rcdg}dg}dg}tjj}tjgd}|||dz|}t ||dt t||dz|||||dz}t ||dt t|||dzdS)Nrr*r")g-r@gYl@g-`w?r2rr)rr rr6rZr rr%)rrr bad_scalerrrs rrzTestBroadcast.test_normal4scD !(00011 a''!&'2>>>>j&#'9=== UQY''!&'2>>>>j&#y1}=====rcdg}dg}dg}dg}tjj}tjgd}|||dz|}t ||dt t||dz|t t||dz|||||dz}t ||dt t|||dzt t|||dzdS) Nr*r+r"r,)gnQVf?gVKB?g v`k?r2rr)rr rr6rZr rr%)rrrbad_abad_brrrs rrzTestBroadcast.test_betaGs C Cy~(11122 a!eQ!&'2>>>>j$ 1555j$Au555 aQ!&'2>>>>j$q1u555j$51955555rcdg}dg}tjj}tjgd}|||dz}t ||dt t||dzdSNr*r")g.UgZ?giPrq?gE K?r2rr)rr rr6rZr rr%)rrr^rrrs rrzTestBroadcast.test_exponential]sD i+ (11122 UQY''!&'2>>>>j+y1}=====rcdg}dg}tjj}tjgd}|||dz}t ||dt t||dzdSrc)rr r r6rZr rr%)rr_ bad_shape std_gammarrs rrz!TestBroadcast.test_standard_gammajsD I, (11122 519%%!&'2>>>>j)Y];;;;;rcdg}dg}dg}dg}tjj}tjgd}|||dz|}t ||dt t||dz|t t||dz|||||dz}t ||dt t|||dzt t|||dzdS) Nr*r+r"r,)g.UgZ?giPrq?gE K?r2rr)rr rr6rZr rr%)rr_rrer^rrrs rrzTestBroadcast.test_gammawsD D  (00011 uqy%((!&'2>>>>j%Q>>>j%I>>> ueai((!&'2>>>>j%EAI>>>j% A >>>>>rcdg}dg}dg}dg}tjj}tjgd}|||dz|}t ||dt t||dz|t t||dz|||||dz}t ||dt t|||dzt t|||dzdS) Nr*r+r"r,)gbyʜ?go?gT7@r2rr)rr rr6rZr rr%)rrr bad_dfnum bad_dfdenrrrs rrzTestBroadcast.test_fsD D IK(00011 519e$$!&'2>>>>j!Y]E:::j!UQY ::: 5%!)$$!&'2>>>>j!Y :::j!UIM:::::rcdg}dg}dg}dg}dg}dg}tjj}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) Nr+r2r3rr"r,)gr^G"@g/ *@gц!@rr)rr rr6rZr rr%) rrrrrirjbad_noncnonc_frrs rrzTestBroadcast.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 EEEEErc|tjddg}tjdddd}t ||ddS)Ngaz@gn4M;%?rr+rwrr)rZrr6r rr )rrrs rtest_noncentral_f_small_dfz(TestBroadcast.test_noncentral_f_small_dfs^ (-/@ABB''S!!'<<!&'2>>>>>>rcdg}dg}tjj}tjgd}|||dz}t ||dt t||dzdS)Nr*r")g~}N??g$F?g'rS?r2rr)rr rr6rZr rr%)rrbad_dfrrrs rrzTestBroadcast.test_chisquaresSI' (00011 26""!&'2>>>>j)VaZ88888rcdg}dg}dg}dg}tjj}tjgd}|||dz|}t ||dt t||dz|t t||dz|||||dz}t ||dt t|||dzt t|||dzdS) Nr*r+r"r,)g w"@g2 WR@gRY@r2rr)rr rr6rZr rr%)rrrrqrlnonc_chirrs rrz'TestBroadcast.test_noncentral_chisquaresSs491(00011 "q&$''!&'2>>>>j(FQJ===j(BFH=== "dQh''!&'2>>>>j(FD1H===j(B1 =====rcdg}dg}tjj}tjgd}|||dz}t ||dt t||dzdS)Nr*r")gT @g~Fl@gƚtp?r2rr)rr rr6rZr rr%)rrrqtrrs rrzTestBroadcast.test_standard_tsS I (00011 26!&'2>>>>j!VaZ00000rcdg}dg}dg}tjj}tjgd}|||dz|}t ||dt t||dz|||||dz}t ||dt t|||dzdS)Nr+r*r")gW!@gOBg}?r2rr)rr rAr6rZr rr%)rr?r@ bad_kapparArrs rrBzTestBroadcast.test_vonmisessSD 9%(11122 "q&%((!&'2>>>>j(BFI>>> "eai((!&'2>>>>j(B A >>>>>rcdg}dg}tjj}tjgd}|||dz}t ||dt t||dzdS)Nr*r")g&8??geFX?gՋː?r2rr)rr rr6rZr rr%)rrr`rrrs rrzTestBroadcast.test_pareto s C!(00011 A!&'2>>>>j&%!)44444rcdg}dg}tjj}tjgd}|||dz}t ||dt t||dzdSrc)rr rKr6rZr rr%)rrr`rKrrs rrLzTestBroadcast.test_weibulls C)#(11122 Q!&'2>>>>j'51955555rcdg}dg}tjj}tjgd}|||dz}t ||dt t||dzdS)Nr*r"r\r2rr)rr rr6rZr rr%)rrr`rrrs rrzTestBroadcast.test_power's C (11122 q1u!&'2>>>>j%33333rcdg}dg}dg}tjj}tjgd}|||dz|}t ||dt t||dz|||||dz}t ||dt t|||dzdS)Nrr*r")g:BKc?g3k?g~q?r2rr)rr rr6rZr rr%)rrrr^rrrs rrzTestBroadcast.test_laplace4scD )#(22233 q%((!&'2>>>>j'37I>>> eai((!&'2>>>>j'3 A >>>>>rcdg}dg}dg}tjj}tjgd}|||dz|}t ||dt t||dz|||||dz}t ||dt t|||dzdS)Nrr*r")g'Zy?gdSO=?g3L1?r2rr)rr rr6rZr rr%)rrrr^rrrs rrzTestBroadcast.test_gumbelGscD !(11122 a''!&'2>>>>j&#'9=== UQY''!&'2>>>>j&#y1}=====rcdg}dg}dg}tjj}tjgd}|||dz|}t ||dt t||dz|||||dz}t ||dt t|||dzdS)Nrr*r")g“y?gJR?g ~&?r2rr)rr rr6rZr rr%)rrrr^rrrs rrzTestBroadcast.test_logisticZscD 9%(22233 #'5))!&'2>>>>j(C!GY??? #uqy))!&'2>>>>j(CQ?????rcdg}dg}dg}tjj}tjgd}|||dz|}t ||dt t||dz|||||dz}t ||dt t|||dzdS)Nrr*r")gH"@gܮ @gvv :@r2rr)rr rr6rZr rr%)rrr bad_sigmarrrs rrzTestBroadcast.test_lognormalmssD I' (00011 4!8U++!&'2>>>>j)TAXyAAA 4++!&'2>>>>j)T9q=AAAAArcdg}dg}tjj}tjgd}|||dz}t ||dt t||dzdS)Nr*r")gݱlo?g/ӂ?gz~R?r2rr)rr rr6rZr rr%)rrr^rrrs rrzTestBroadcast.test_rayleighsD 9%(00011 %!)$$!&'2>>>>j(IM:::::rcldg}dg}dg}dg}tjj}tjgd}|||dz|}t ||dt t||dz|t t||dz|||||dz}t ||dt t|||dzt t|||dzt t|d dt t|dd dS) NrAr*rr,)g@m e?g`I߿?g|2?r2rrrS)rr rHr6rZr rr%)rrrbad_meanr^rHrrs rrIzTestBroadcast.test_waldsEu3D y~(00011 dQh&&!&'2>>>>j$1 e<<<j$q)<<< dEAI&&!&'2>>>>j$%!)<<<j$i!m<<<j$Q///j$S11111rcdg}dg}dg}dg}dg}|dz\}}tjj}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||||dzdS)Nr*r2r+r3)g,P;bD@g%G@g,~S@rr)rr rr6rZr rr%) rrrr bad_left_one bad_mode_one bad_left_two bad_mode_tworrrs rrzTestBroadcast.test_triangularssss s %*QY" lY) (00011 D1HdE22!&'2>>>>j*lQ.>eLLLj*dQh eLLLj*lQ.>     D$(E22!&'2>>>>j*lD1HeLLLj*dL14DeLLLj*lL1>>>j*lD%!)LLLj*dL%!)LLLj*lLai ! ! ! ! !rcvdg}dg}dg}dg}dg}tjj}tjgd}|||dz|}t ||t t||dz|t t||dz|t t||dz|||||dz}t ||t t|||dzt t|||dzt t|||dzdS)Nr*rAr"?r*r*r*r2)rr rCr6rZrrr%) rrrDbad_n bad_p_one bad_p_twobinomrrs rrzTestBroadcast.test_binomials8 C ED E  "(999%% q1ua67+++j%A666j%Q :::j%Q ::: q!a%67+++j%A666j%IM:::j%IM:::::rcvdg}dg}dg}dg}dg}tjj}tjgd}|||dz|}t ||t t||dz|t t||dz|t t||dz|||||dz}t ||t t|||dzt t|||dzt t|||dzdS)Nr*rAr"r)r*rr*r2)rr rr6rZrrr%) rrrDrrr neg_binomrrs rrz$TestBroadcast.test_negative_binomials< C ED E I/ (999%% 1q5!$$67+++j)UQY:::j)QUI>>>j)QUI>>> 1a!e$$67+++j)UAE:::j)Q A >>>j)Q A >>>>>rcztjj}dg}dg}|dzg}tjj}tjgd}|||dz}t||tt||dztt||dzdS)Nr*r"r+)r*r*rr2) rr r_poisson_lam_maxrr6rZrrr%)rmax_lamr bad_lam_one bad_lam_tworrrs rrzTestBroadcast.test_poissons)'')):cd {m )#(999%% q!!67+++j';?;;;j';?;;;;;rcdg}dg}tjj}tjgd}|||dz}t ||t t||dztjd5t t|tj t t|ddtj gddddS#1swxYwYdS)Nr+r)r+r+r*r2r)invalid) rr rRr6rZrrr%errstaterG)rrr`rRrrs rrSzTestBroadcast.test_zipfs Cy~(999%% a!e67+++j$ 222 [ * * * < < *dBF 3 3 3 *dQ26N ; ; ; < < < < < < < < < < < < < < < < < dg}dg}dg}dg}dg}dg}dg}tjj}tjgd} |||dz||} t | | t t||dz||t t||dz||t t||dz||t t||dz||||||dz|} t | | t t|||dz|t t|||dz|t t|||dz|t t|||dz||||||dz} t | | t t||||dzt t||||dzt t||||dzt t||||dzdS) Nr*r+r"r,rr3rr2)rr rr6rZrrr%) rngoodnbadnsample bad_ngoodbad_nbadbad_nsample_onebad_nsample_two hypergeomrrs rrz!TestBroadcast.test_hypergeometric!s,s#D 4##I, (999%% 519dG4467+++j)Y]D'JJJj)UQY'JJJj)UQYoNNNj)UQYoNNN 5$(G4467+++j)Yq'JJJj)UHqL'JJJj)UD1HoNNNj)UD1HoNNN 5$! 4467+++j)YgkJJJj)UHgkJJJj)UD/A:MNNNj)UD/A:MNNNNNrc.dg}dg}dg}tjj}tjgd}|||dz}t ||t t||dzt t||dzdS)NrAr+r"rr2)rr rr6rZrrr%)rrDrrrrrs rrzTestBroadcast.test_logseriesDs EC D I' (999%% 1q5!!67+++j)Y];;;j)Y];;;;;rN)#r9r:r;rmrZr!rrrrrrrrorrrrBrrLrrrrrrrIrrrrrSrrrr<rrrWrWs""" ??? >>>&666, > > > < < <???,;;;,FFFB??? 9 9 9>>>, 1 1 1???& 5 5 5 6 6 6 4 4 4???&>>>&@@@&BBB& ; ; ;2220"!"!"!H;;;.???. < < < < < < 7 7 7!O!O!OF < < < < ._sQ 0 0 0AV829+@+@+C+CQ*G H H H 0 0 0rc6g|]}|Sr<)startr:rXs rr<z-TestThread.check_function..as qrc6g|]}|Sr<)joinrs rr<z-TestThread.check_function..bs arr3win32) threadingrrrrrzipr rintpr0itemsizerplatformr r) rrszout1out2rurrrs ` @rcheck_functionzTestThread.check_functionXsK$$$$$$xTZ*R/00xTZ*R/00 0 0 0 0 0TZ.. 0 0 0A1 D)) 2 2DAq HRY**1--q 1 1 1 1 799? #q ( (S\W-D-D %dD 1 1 1 1 1 tT * * * * *rc:d}||ddS)Nc8|d|d<dS)N'rw.)rrlouts r gen_randomz*TestThread.test_normal..gen_randomos|||//CHHHr)rrrrrs rrzTestThread.test_normalns0 0 0 0 J844444rc:d}||ddS)Nc\|tjd|d<dS)NrLrr.)rronesrs rrz'TestThread.test_exp..gen_randomts*((rw{/C/C(DDCHHHrrrrrs rtest_expzTestThread.test_expss3 E E E J;77777rc:d}||ddS)NcD|ddgdzd|d<dS)Nrrr5rrw.)rPrs rrz/TestThread.test_multinomial..gen_randomys*((dVAXE(BBCHHHr)rr5rrrs rrzTestThread.test_multinomialxs3 C C C J:66666rN)r9r:r;rmrrrrr<rrrrRs_+++,555 888 77777rrc,eZdZdZdZdZdZdZdS)TestSingleEltArrayInputctjdg|_tjdg|_tjdg|_d|_dS)Nr+r2r3)r*)rr6argOneargTwoargThreetgtShaper&s rrmz$TestSingleEltArrayInput.setup_methods@hsmm hsmm !   rc rtjjtjjtjjtjjtjjtjjtjjtjj tjj tjj tjj tjj f }tjj tjj f}|D]O}||vr|tjdg}n||j}t!|j|jPdSr)rr rr rrrrKrrrrRrrr6rrr_r)rfuncs probfuncsr)rs rtest_one_arg_funcsz*TestSingleEltArrayInput.test_one_arg_funcss& (@$bi&:!29#4")"4"BIN$bi&9 ;Y(")*=>  3 3Dy  d28SE??++d4;'' DM 2 2 2 2 3 3rcttjjtjjtjjtjjtjjtjjtjjtjj tjj tjj tjj tjj tjjtjjf}tjjtjjf}|D]}||vrtjdg}n|j}||j|}t'|j|j||jd|}t'|j|j||j|d}t'|j|jdS)NrAr)rr r rrrrrrArrrrrHrCrr6rrrr_r)rrrr)rrs rtest_two_arg_funcsz*TestSingleEltArrayInput.test_two_arg_funcssC"BI$4bi<#RY%6!29#5$bin#RY%@ BY')DE  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 3rc Nttjtjtjtjtjtjtjtj g }tj j }tj dg}tj dg}|D]}||||}t|j|j||d||}t|j|j|||d|}t|j|jdS)Nr*rr/)rrrrrrrr^r7rr rr6rr_r)rrr)rrrrs rrz$TestSingleEltArrayInput.test_randintsrw"(BI29bh ;y x}}hsmm 3 3B$sD+++C DM 2 2 2$s1vt2...C DM 2 2 2$sDG2...C DM 2 2 2 2 3 3rctjjtjjtjjg}|D]}||j|j|j}t|j |j ||jd|j|j}t|j |j ||j|jd|j}t|j |j dS)Nr) rr rrrrrrrr_r)rrr)rs rtest_three_arg_funcsz,TestSingleEltArrayInput.test_three_arg_funcss')=)+ 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 3rN)r9r:r;rmrrrrr<rrrrs_ 333&3336333" 3 3 3 3 3rr)rr^numpyr numpy.testingrrrrrrr r r r rrr>rJrfrrrWrTskipifrrr<rrrs|                        $F$F$F$F$F$F$F$FN > > > > > > > >YYYYYYYYD-.-.-.-.-.-.-.-.`G+G+G+G+G+G+G+G+Tt ,t ,t ,t ,t ,t ,t ,t ,nw<w<w<w<w<w<w<w