y>>j")"7* FFFj")"79K9K9KLLLj")"79L9L9LMMMMMr c 2 t t t j j t j g t j t t t j j g dg t t t j j g dg dg d S )Ndtyper* r+ ) )r r% r r r arrayint64r& s r test_invalid_array_shapez!TestSeed.test_invalid_array_shape- s j")"7hr222 4 4 4j")"7)))EEEj")"7))):C)):E F F F F Fr N)__name__
__module__ __qualname__r r r' r- r8 r r r r sg + + + + + += = =
N N NF F F F Fr r c e Zd Zd Zd ZdS ) TestBinomialc t j dd }dD ]K}t t j d| dk t t j || | Ld S )Nr+ intr/ )r ?r* r )r zerosr r binomialr )r rB ps r test_n_zerozTestBinomial.test_n_zero7 ss %((( A AA FOAq))Q. / / / vua 8 8% @ @ @ @ A Ar c \ t t t j dt j d S )Nr* )r r% r rC r nanr& s r
test_p_is_nanzTestBinomial.test_p_is_nan@ s j&/1bf=====r N)r9 r: r; rE rH r< r r r> r> 6 s5 A A A> > > > >r r> c , e Zd Zd Zd Zd Zd Zd ZdS )TestMultinomialc 4 t j dddg d S )Nd 皙?皙?r multinomialr& s r
test_basiczTestMultinomial.test_basicF s 3c
+++++r c 4 t j dg d d S )NrL )rM rN rS rS rO r& s r test_zero_probabilityz%TestMultinomial.test_zero_probabilityI s" 3 9 9 9:::::r c & t dt j dd cxk odk nc t j ddd }t t j d|k t t j |dk d S )Nr" r4 )r r r r all)r xs r test_int_negative_intervalz*TestMultinomial.test_int_negative_intervalL s fnR,,1111r1111222 N2r1 % %rQw q2v r c z ddg}t t j d|t j d j d t t j d|t j d j d t t j d|t j d j d t t j d|ddg j d t t j d|d j d t t j d|t j d j d t t t j j d|t d d S )NrA r* r* r+ r+ r+ r+ r+ r+ r+ )
r r r rP uint32shaper6 r r$ floatr rD s r test_sizezTestMultinomial.test_sizeR sV
#JRY**1a1 >>DfMMMRY**1a1 >>DfMMMRY**1a1 >>DfMMMRY**1a!Q88> JJJRY**1a88> JJJRY**1a&1A1ABBH i!61Ahh r c
t t t j j dddgg t t t j j ddgdgg t t t j j ddgdggdgdggg t t t j j dt j ddgddgg d S )Nr r r* )r r% r r rP r6 r& s r test_multidimensional_pvalsz+TestMultinomial.test_multidimensional_pvals` s j")"7q!fXFFFj")"7qcA3ZHHHj")"7sQCjA3QRPS*=UVVVj")"7RX1vPQSTvFV=W=WXXXXXr N)r9 r: r; rQ rT rY rb rd r< r r rJ rJ E sd , , ,; ; ; Y Y Y Y Yr rJ c 2 e Zd Zd Zd Zd Zd Zd Zd ZdS ) TestSetStatec d| _ t j | j | _ | j | _ d S NiI)seedr r prng get_statestater& s r setup_methodzTestSetState.setup_methodh s6 &ty11 Y((**
r c | j d }| j | j | j d }t t j ||k d S )N )rj tomaxint set_staterl r r rW r oldnews r rQ zTestSetState.test_basicm sf i $$ DJ'''i $$scz""#####r c | j d }| j | j | j d }t t j ||k d S Nr2 size)rj standard_normalrq rl r r rW rr s r test_gaussian_resetz TestSetState.test_gaussian_resets sl i''Q'// DJ'''i''Q'//scz""#####r c T | j | j }| j d }| j | | j d }t t j ||k d S rv )rj ry rk rq r r rW )r rl rs rt s r test_gaussian_reset_in_media_resz-TestSetState.test_gaussian_reset_in_media_resz s
!!### ##%%i''Q'// E"""i''Q'//scz""#####r c | j d d }| j d }| j | | j d }| j | j | j d }t t j ||k t t j ||k d S )Nr, ro rw )rl rj ry rq r r rW )r old_statex1x2x3s r test_backwards_compatibilityz)TestSetState.test_backwards_compatibility s JssO
Y
&
&B
&
/
/ I&&&
Y
&
&B
&
/
/ DJ'''
Y
&
&B
&
/
/rRx !!!rRx !!!!!r c < | j dd d S NrA )rj negative_binomialr& s r test_negative_binomialz#TestSetState.test_negative_binomial s"
##C-----r N) r9 r: r; rm rQ rz r| r r r< r r rf rf g sn + + +
$ $ $ $ $ $ $ $ $
"
"
". . . . .r rf c e Zd Zej j Zej ej ej ej
ej ej ej
ej ej g Zd Zd Zd Zd Zd Zd Zd Zd Zd S )
TestRandintc J t t | j dt d S Nr* r/ )r r$ rfuncr` r& s r test_unsupported_typez!TestRandint.test_unsupported_type s! iQe<<<<<d: Z?d; Z@d< ZAd= ZBd> ZCd? ZDd@ ZEdA ZFdB ZGdC ZHdD ZIdE ZJdF ZKdG ZLdH ZMdI ZNdJ ZOdK ZPdL ZQdM ZRdNS )OTestRandomDistc d| _ d S rh ri r& s r rm zTestRandomDist.setup_method% s
r c t j | j t j dd }t j ddgddgddgg }t ||d
d S ) Nr2 r+ _\*?琛?<p?,o? H?Z? decimal)r r ri randr6 r r r desireds r test_randzTestRandomDist.test_rand( s~
ty!!! 1%%( 02E F 02E F /1C D F G G "&'2>>>>>>r c t j | j t j dd }t j ddgddgddgg }t ||d
d S ) Nr2 r+ !>>>>>r c t j | j t j ddd }t j ddgddgd d
gg }t || d S ) Nc r2 r+ rw r2 ) )r r ri r r6 r r s r test_randintzTestRandomDist.test_randint8 sv
ty!!!""3"88(RG "I #J ( ) ) 67+++++r c t j | j t 5 }| t
}t j ddd }t t | dk d d d n # 1 swxY w Y t j ddgdd gd
d gg }t || d S ) Nr r r rw r* r r2 r r r r ) r r ri r
recordDeprecationWarningrandom_integersr lenr6 r r supwr r s r test_random_integersz#TestRandomDist.test_random_integers@ s
ty!!!
!C
-..AY..sBV.DDF CFFaK ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! (RG "I #J ( ) ) 67+++++s A B B #B c t 5 }| t }t j t j d j t j d j }t t | dk d d d n # 1 swxY w Y t j d j }t || d S )Nlr* ) r
r r r r r r r r r r r s r test_random_integers_max_intz+TestRandomDist.test_random_integers_max_intK s
!C
-..AY..rx}}/@/1x}}/@B BF CFFaK ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! (3--#VW%%%%%s B B$$B(+B(c t j 5 t j dt t t t
j j t j d j t t t
j j t j d j t j d j d d d d S # 1 swxY w Y d S )Nerrorr )
warningscatch_warnings simplefilterr r r r r r r r& s r test_random_integers_deprecatedz.TestRandomDist.test_random_integers_deprecatedZ s
$
&
& @ @ !'+= > > >
, )3 (3--+
-
-
-
, )3 (3--+RXc]]->
@
@
@ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @s B+C CCc t j | j t j d }t j ddgddgddgg }t ||d d S )
Nr r r r r r r r r )r r ri r6 r r s r test_randomzTestRandomDist.test_randomh s~
ty!!!!!&))( 02E F 02E F /1C D F G G "&'2>>>>>>r c t j | j t j dd }t j g d }t || d S )Nr3 )r+ r2 r+ r2 r r ri choicer6 r r s r test_choice_uniform_replacez*TestRandomDist.test_choice_uniform_replacep sX
ty!!!!!!Q''(<<<((67+++++r c t j | j t j ddg d }t j g d }t || d S )Nr3 )皙?r 皙?r rD )r* r* r+ r+ r r s r test_choice_nonuniform_replacez-TestRandomDist.test_choice_nonuniform_replacev sd
ty!!!!!!Q*>*>*>!??(<<<((67+++++r c t j | j t j ddd }t j g d }t || d S )Nr3 r2 Freplace)r r* r2 r r s r test_choice_uniform_noreplacez,TestRandomDist.test_choice_uniform_noreplace| s]
ty!!!!!!Q!66(999%%67+++++r c t j | j t j dddg d }t j g d }t || d S )Nr3 r2 F)r g333333?rA r r rD )r+ r2 r* r r s r test_choice_nonuniform_noreplacez/TestRandomDist.test_choice_nonuniform_noreplace sl
ty!!!!!!Q$8$8$8 " : :(999%%67+++++r c t j | j t j g dd }t j g d }t || d S )N)abcdr3 )r r r r r r s r test_choice_nonintegerz%TestRandomDist.test_choice_noninteger sb
ty!!!!!"6"6"6::(///0067+++++r c t j j }t t |dd t t |dd t t |ddgddggd t t |g d t t |g ddddgddgg t t |ddgdg d
t t |ddgdd d g t t |ddgdd
d
g t t |g ddd t t |g ddd t t |g ddd t t |g ddd t t |g dddg d d S )Nr" r2 g @r* r+ r3 )r* r+ r2 r3 g ?r )r r rM g?gr r1 Fr r, )r" )r" r* )r* r r r )r r r r r% )r r s r test_choice_exceptionsz%TestRandomDist.test_choice_exceptions s !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 2r c
ddg}t t j t j dd t t j t j dd t t j t j dd| t t j t j dd| t t j t j ddgd t t j d gd d u t j ddg }t j dt }||d
< t t j |d |u t }t t j t j d|d t t j t j d|d t t j t j d|d| t t j t j d|d| t t j t j ddg|d t t j d g|d j d
k t j ddg }t j dt }||d
< t t j ||d
|u d }g d }t t j d
|d j | t t j d
|d j | t t j d
|d| j | t t j d
|d| j | t t j t j
d
|d j | t t j d
d
d j d t t j d
dd
j d t t j ddd
j d t t j d
d
j d t t j g d j d t t j ddgd j d t t t j j g d d S )Nr ?r+ Tr Fr r* r/ r )r+ r2 )r r r r r rM r5 )r2 r r3 rw i)r r r r )r r isscalarr r r6 emptyobjecttuplendimitemr r_ r r r r% )r rD r arrr s r test_choice_return_shapez'TestRandomDist.test_choice_return_shape s*
#J BI , ,Q , = =>>??? BI , ,Q , > >??@@@ BI , ,Q , B BCCDDD BI , ,Q , C CDDEEE BI , ,aVT , B BCCDDD $ 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.. s rx|| r c | S r+ r< r, s r r- z-TestRandomDist.test_shuffle.. s q r c d t j | t j S r+ )r asarrayastyper r, s r r- z-TestRandomDist.test_shuffle.. s rz!}} 3 3BG < < r c d t j | t j S r+ )r r0 r1 float32r, s r r- z-TestRandomDist.test_shuffle.. s rz!}} 3 3BJ ? ? r c d t j | t j S r+ )r r0 r1 complex64r, s r r- z-TestRandomDist.test_shuffle.. s rz!}} 3 3BL A A r c Z t j | t S r+ )r r0 r1 r r, s r r- z-TestRandomDist.test_shuffle.. s rz!}} 3 3F ; ; r c d | D S )Nc g | ]}||fS r< r< .0is r
zATestRandomDist.test_shuffle.... s 2 2 21A 2 2 2r r< r, s r r- z-TestRandomDist.test_shuffle.. s 2 2 2 2 2 r c > t j d | D S )Nc g | ]}||gS r< r< r9 s r r< zATestRandomDist.test_shuffle.... *=*=*=aAq6*=*=*=r )r r0 r, s r r- z-TestRandomDist.test_shuffle.. s rz*=*=1*=*=*= > > r c 8 t j | | g j S r+ )r vstackTr, s r r- z-TestRandomDist.test_shuffle.. s ry!Q 0 0 2 r c t j d | D dt fdt fg t j S )Nc g | ]}||fS r< r< r9 s r r< zATestRandomDist.test_shuffle.... s +>+>+>qQF+>+>+>r r r )r r0 r@ r recarrayr, s r r- z-TestRandomDist.test_shuffle.. sG
+>+>A+>+>+>-0#Jc
+C!E !E!%bk!2!2 r c j t j d | D dt fdt j fg S )Nc g | ]}||fS r< r< r9 s r r< zATestRandomDist.test_shuffle.... r? r r r )r r0 r r r, s r r- z-TestRandomDist.test_shuffle.. s7 rz*=*=1*=*=*=,/=3/*J L L r )
r* r+ r2 r3 r4 r5 r r )
r r* rI r5 r+ r3 r4 r rH r2 )r r ri shuffler )r convalistr r s r test_shufflezTestRandomDist.test_shuffle s ,+ [<<??AA;;22>>224 4L L M 0 0D
INN49 % % %D77788E I e $ $ $Fd999::G vw / / / /+ 0 0r c " t j t j t d d dz dz
d }t j t j d dz dz
d }| }| }t d D ]}t j | t t |j |j t |j |j t j | t t |j |j t |j |j d S )N )r4 r3 r2 r* r" 2 )
r ma
masked_valuesreshaper r copyr rJ r sorteddatamask)r r r a_origb_origr; s r test_shuffle_maskedz"TestRandomDist.test_shuffle_masked sK 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 Lr r c dddddddd}t j t d
5 } |j | d d d n # 1 swxY w Y d |d j v sJ d S ) Nr r* r+ r2 r3 r4 r5 )r r* r+ r2 r3 r4 r5 z!you are shuffling a 'dict' objectmatchr )pytestwarns UserWarningrJ filename)r r valuesrecs r test_shuffle_untyped_warningz+TestRandomDist.test_shuffle_untyped_warning s 1qQ1;;
\+9; ; ; #>A FN6 " " " # # # # # # # # # # # # # # # A//////s AA
Ause_array_likeTFc G d dt j }d t j dg t j d t j d t j d g}t j |t }d |D |r| | }t
fd|D sJ |ret |t j j sFt j t d
5 |j
| d d d d S # 1 swxY w Y d S |j
| t
fd |D sJ d S ) Nc e Zd ZdS )>TestRandomDist.test_shuffle_no_object_unpacking..MyArrN)r9 r: r; r< r r MyArrrh s Dr ri r2 r rH r/ c , h | ]}t | S r< idr9 s r zBTestRandomDist.test_shuffle_no_object_unpacking.. s )))aBqEE)))r c 3 : K | ]}t | v V d S r+ rk r: r; item_idss r zBTestRandomDist.test_shuffle_no_object_unpacking.. s. 222a55H$222222r z#Shuffling a one dimensional array.*r\ c 3 : K | ]}t | v V d S r+ rk ro s r rq zBTestRandomDist.test_shuffle_no_object_unpacking..& s. 66Qr!uu(666666r )r ndarrayr6 float64r r rW
isinstancer Generatorr^ r_ r` rJ )r r re ri itemsr! rp s @r test_shuffle_no_object_unpackingz/TestRandomDist.test_shuffle_no_object_unpacking
s BJ
"(A3--A bjmm
huF+++))5))) "((5//C 2222c22222222 7*VRY5H"I"I 7 k?A A A
$
$ s###
$
$
$
$
$
$
$
$
$
$
$
$
$
$
$
$
$
$
FN3 6666#66666 6 66 6 6s D D#&D#c z t j | j t j d j }t j | t
t j | g d t j | j }| | t
t j | g d t j | j }| | t
t j | g d d S )Nr4 )r r* r3 r2 r+ )r r* r+ r2 r3 )r3 r* r r2 r+ )
r r ri r rV rJ r r0 r default_rng)r r rngs r test_shuffle_memoryviewz&TestRandomDist.test_shuffle_memoryview( s ty!!! IaLL
! RZ]]OOO444i##DI.. ARZ]]OOO444i##DI.. ARZ]]OOO44444r c t j d }d|j _ t j t d 5 t j | d d d d S # 1 swxY w Y d S )Nr2 Fz read-onlyr\ ) r rB flags writeabler^ raisesr% r rJ )r r s r test_shuffle_not_writeablez)TestRandomDist.test_shuffle_not_writeable7 s HQKK !
]:[
9
9
9 ! ! I a ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! !s A))A-0A-c t j | j t j ddd }t j ddgddgd d
gg }t ||d d S )
Nr r r rw gAKč?g hA?g ҇o>gάXf+q?gS1$?g'8zX ?r r )r r ri betar6 r r s r test_betazTestRandomDist.test_beta= s
ty!!! BV 44()+BC)+BC)+BCEF F "&'2>>>>>>r c t j | j t j ddd }t j ddgddgd d
gg }t || d S ) NrL gv/?r rw % + r$ 0 . - )r r ri rC r6 r r s r
test_binomialz TestRandomDist.test_binomialF sv
ty!!!##CF#;;(RH H H & ' ' 67+++++r c t j | j t j dd }t j ddgddgdd gg }t ||d
d S ) NrP r rw g+Q]uO@g+Q@gY7`ZqP@ge(fG@gvtR@gEIȋR@
r )r r ri chisquarer6 r r s r test_chisquarez TestRandomDist.test_chisquareN s
ty!!!$$Rf$55( 13G H 13G H 02F G I J J "&'2>>>>>>r c < t j | j t j ddg }t j j |d }t j ddgddggd d
gd d ggd
dgddggg }t
||d d S )N I>>>>>r c t j ddg }t t j |t j d j d t t j |t j d j d t t j |t j d j d t t j |ddg j d t t j |d j d t t j |t j d j d t t t j j |t d d S )Nr r r* r[ r+ r\ r] )
r r6 r r r r^ r_ r r$ r` ra s r test_dirichlet_sizez"TestRandomDist.test_dirichlet_sizeb sA H*,@A B BRY((BIaLL99?HHHRY((BIaLL99?HHHRY((BIaLL99?HHHRY((QF3399EEERY((F3399EEERY((BHV,<,<==CYOOOi!4aqBBBBBr c t j ddg }t t t j j j | t t t j ddgg t t t j dgdgg t t t j dgdggdgdggg t t t j t j ddgddgg d S )NgHzG?gؗҜr4 r* )r r6 r r% r r r )r r s r test_dirichlet_bad_alphaz'TestRandomDist.test_dirichlet_bad_alphan s '8,--j")"2">>>>>r c t t j d d t t
t j j d d S Nr scale )r r r r r r% r& s r test_exponential_0z!TestRandomDist.test_exponential_0 sE RY***33Q777j")"7sCCCCCCr c t j | j t j ddd }t j ddgddgd d
gg }t ||d d S )
N M r rw g 7 ?g`j?g6"+?gܳ2>?gc>+Y?g_}S?r r )r r ri fr6 r r s r test_fzTestRandomDist.test_f s
ty!!!R&11( 02E F 02E F 02E F H I I "&'2>>>>>>r c t j | j t j ddd }t j ddgddgd d
gg }t ||d d S )
Nr4 r2 r rw gԏGM8@gȌ<@g ":@gv#)@gZQ?@gr[@@ r )r r ri gammar6 r r s r
test_gammazTestRandomDist.test_gamma s
ty!!! AF 33( 13G H 13G H 13G H J K K "&'2>>>>>>r c t t j dd d t t
t j j dd d S )Nr )r_ r r )r r r r r r% r& s r test_gamma_0zTestRandomDist.test_gamma_0 sE RY__1A_66:::j")/CHHHHHHr c t j | j t j dd }t j ddgddgddgg }t || d S ) N_c97ݚ?r rw r rH r4 r )r r ri geometricr6 r r s r test_geometricz TestRandomDist.test_geometric st
ty!!!$$Zf$==(QF H G % & & 67+++++r c t j | j t j ddd }t j ddgddgd d
gg }t ||d d S )
Nr @r locr rx gؕ?gs0 ?g #0g(xg-KD?gůp`@r r )r r ri gumbelr6 r r s r test_gumbelzTestRandomDist.test_gumbel s
ty!!!!!j&!II( 02E F 02F G 02F G I J J "&'2>>>>>>r c t t j d d t t
t j j d d S r )r r r r r r% r& s r
test_gumbel_0z TestRandomDist.test_gumbel_0 E RY%%A%..222j")"2#>>>>>>r c 6 t j | j t j dddd }t j ddgddgddgg }t || t j dddd }t j g d
}t || t j d dd d }t j g d
}t || t j dddd }t j g d }t || t j dd d d }t j g d }t || d S )Nr r4 r r rw rI r r2 r3 )r2 r2 r2 r2 r r )r r r r )r r r r )r r ri hypergeometricr6 r r s r test_hypergeometricz"TestRandomDist.test_hypergeometric s
ty!!!))"a&)AA(RH H F $ % % 67+++ ))!Q)::(<<<((67+++))"a!)<<(+++,,67+++ ))!Q)::(<<<((67+++))!R!)<<(<<<((67+++++r c t j | j t j ddd }t j ddgddgd d
gg }t ||d d S )
Nr r r r g bO?g¸?gpJm @g{K*t @g:^w-gY~o?r r )r r ri laplacer6 r r s r test_laplacezTestRandomDist.test_laplace s
ty!!!""z6"JJ( 02E F 02E F 13F G I J J "&'2>>>>>>r c t t j d d t t
t j j d d S r )r r r r r r% r& s r test_laplace_0z TestRandomDist.test_laplace_0 sE RY&&Q&//333j")"33??????r c t j | j t j ddd }t j ddgddgd d
gg }t ||d d S )
Nr r r r g#K-z?gJ?g92 @gر[@g#+^˿g-@r r )r r ri logisticr6 r r s r
test_logisticz TestRandomDist.test_logistic s
ty!!!##
#F#KK( 02D E 02E F 13F G I J J "&'2>>>>>>r c t j | j t j ddd }t j ddgddgd d
gg }t ||d d S )
Nr r r )meansigmarx gɁ0@go,+4FB@g)ʭ6@g5 ?gnxNnP@g)U@r r )r r ri lognormalr6 r r s r test_lognormalz TestRandomDist.test_lognormal s
ty!!!$$*Cf$MM( 13G H 13F G 13G H J K K "&'2>>>>>>r c t t j d d t t
t j j d d S )Nr )r r* r )r r r r r r% r& s r test_lognormal_0z TestRandomDist.test_lognormal_0 sE RY((q(111555j")"5SAAAAAAr c t j | j t j dd }t j ddgddgddgg }t || d S )Ngƀ@?r )rD rx r+ r5 r r2 )r r ri logseriesr6 r r s r test_logseriesz TestRandomDist.test_logseries st
ty!!!$$z$??(QF G F $ % % 67+++++r c t j | j t j ddgdz d }t j g dg dgg dg d gg d
g d gg }t || d S ) NrO UUUUUU?r5 r rw )r3 r2 r4 r3 r+ r+ )r4 r+ r r+ r+ r* )r2 r3 r2 r5 r r3 )r+ r* r3 r2 r5 r3 )r3 r3 r+ r4 r+ r2 )r3 r2 r3 r+ r2 r3 )r r ri rP r6 r r s r test_multinomialz TestRandomDist.test_multinomial s
ty!!!&&rD6!8&&AA( / / / / / / 1 / / / / / / 1 / / / / / / 1 2 3 3 67+++++r c t j | j d}ddgddgg}d}t j ||| }t j ddgddggd d
gd d ggd
dgddggg }t ||d t j || }t j ddg }t ||d ddg}ddgddgg}t
t t j j || t t j j ||d t t t j j ||d t j ddgddggt j }t 5 }t j || |
t }t | dk sJ d d d d S # 1 swxY w Y d S )N)r r r* r r g4Vj?gy'@gI1?gSB#@g e<@g 4W(@g6R¿?g9r
v"@gс?g{#@g4xg"f"@r r gN}O6?gYMW"@r+ ignore) check_validraiser r/ )r r ri multivariate_normalr6 r r RuntimeWarningr r r% r3 r
r r )r r covrx r r r r s r test_multivariate_normalz'TestRandomDist.test_multivariate_normal s_
ty!!! 1v1v ..tS$??( /1B C /1B C E /1C D /1B C E /1B C 02C D F G H H "&'2>>>> ..tS99( -/?@AA!&'2>>>> 1v1v1v ^RY%BD#NNN 298$'/ 1 1 1 1 j")"?s") + + + + hC3(+2:>>>
C I ) )$ 4 4 4
>**Aq66Q;;;;; s AG##G'*G'c t j | j t j ddd }t j ddgddgd d
gg }t || d S ) NrL g|Pk?r )nrD rx iP iI i| ic i i )r r ri r r6 r r s r r z%TestRandomDist.test_negative_binomial" sv
ty!!!,,sf6,JJ(S#J #J #J ( ) ) 67+++++r c | t j | j t j ddd }t j ddgddgdd gg }t ||d
t j d d
d }t j ddgddgddgg }t ||d
t j | j t j ddd }t j ddgddgddgg }t ||d
d S )Nr4 r )dfnoncrx gG7@g! ܴ*@g1}z9?@g+驸0@g26E_r#@gI!1@r r rA rM g ?gTD?g?T?g+qnl?g~|D?g밬?r g{01#@gprs'@g~;$@g"dI
@g*@g }',@)r r ri noncentral_chisquarer6 r r s r test_noncentral_chisquarez(TestRandomDist.test_noncentral_chisquare* si
ty!!!//116/JJ( 13G H 13G H 02F G I J J "&'2>>>>//2BV/LL( 03F G 03F G .1D E G H H "&'2>>>>
ty!!!//116/JJ( .0B C /1B C /1C D F G G "&'2>>>>>>r c t j | j t j dddd }t j ddgdd gd
d gg }t ||d
d S )Nr4 r+ r* r )dfnumdfdenr rx g^~?g/+c?g". @g @g7?gE?r r )r r ri noncentral_fr6 r r s r test_noncentral_fz TestRandomDist.test_noncentral_f? s
ty!!!''aqq-3 ( 5 5( 02E F 02E F 02D E G H H "&'2>>>>>>r c t j | j t j ddd }t j ddgddgd d
gg }t ||d d S )
Nr r r r g&&n@gOG_ @g~w @gtw]տgm&5@g$>@r r )r r ri normalr6 r r s r test_normalzTestRandomDist.test_normalH s
ty!!!!!j&!II( 02E F .0D E 02E F H I I "&'2>>>>>>r c t t j d d t t
t j j d d S r )r r r r r r% r& s r
test_normal_0z TestRandomDist.test_normal_0P r r c t j | j t j dd }t j ddgddgdd gg }t j ||d
d S ) Nr r r rx gW I@g
Ԩy@gy]0Ag1~\Aga@gY
7A )nulp)r r ri paretor6 testing assert_array_almost_equal_nulpr s r test_paretozTestRandomDist.test_paretoT s
ty!!!!!JV!<<()+BC)+BC)+BCEF F
11&'1KKKKKr c t j | j t j dd }t j ddgddgddgg }t || d S )Nr r )lamrx r r* )r r ri poissonr6 r r s r test_poissonzTestRandomDist.test_poissonc st
ty!!!""z"??(QF F F $ % % 67+++++r c t t j d j }d}t t t j j | t t t j j |gdz t t t j j | t t t j j |gdz d S )Nr r" r )r r r r r% r r )r lambiglamnegs r test_poisson_exceptionsz&TestRandomDist.test_poisson_exceptionsk s # "j")"3V<<<j")"3fXb[AAAj")"3V<<<j")"3fXb[AAAAAr c t j | j t j dd }t j ddgddgdd gg }t ||d
d S ) Nr r r g;'?gܴw*?g 5?gFG?g D ]?g$`?r r )r r ri powerr6 r r s r
test_powerzTestRandomDist.test_powers s
ty!!! :F ;;( 02E F 02E F 02E F H I I "&'2>>>>>>r c t j | j t j dd }t j ddgddgdd gg }t ||d
d S ) Nr r )r rx g"u+@gN$MB*@g7v8B4@g55@ge &@g~ӣZ1@r r )r r ri rayleighr6 r r s r
test_rayleighz TestRandomDist.test_rayleigh{ s
ty!!!##"6#::( 02D E 13G H 13G H J K K "&'2>>>>>>r c t t j d d t t
t j j d d S r )r r r r r r% r& s r test_rayleigh_0z TestRandomDist.test_rayleigh_0 sE RY''a'00!444j")"4C@@@@@@r c t j | j t j d }t j ddgddgddgg }t ||d
d S ) Nr rw gMoDL?g6( [9grF=?gM' - gVgVA:Qy?r r )r r ri standard_cauchyr6 r r s r test_standard_cauchyz#TestRandomDist.test_standard_cauchy s
ty!!!***77( 02F G 02F G 13F G I J J "&'2>>>>>>r c t j | j t j d }t j ddgddgddgg }t ||d
d S ) Nr rw g܁?gc!z?g&N"@gNI@gF>?g~ME?r r )r r ri standard_exponentialr6 r r s r test_standard_exponentialz(TestRandomDist.test_standard_exponential s
ty!!!//V/<<( 02E F /1D E /1D E G H H "&'2>>>>>>r c t j | j t j dd }t j ddgddgdd gg }t ||d
d S ) Nr2 r )r_ rx g)@gľ@gd],q@gPk{@g9B #1 @gv @r r )r r ri standard_gammar6 r r s r test_standard_gammaz"TestRandomDist.test_standard_gamma s
ty!!!)))??( 02E F 02E F 02C D F G G "&'2>>>>>>r c t t j d d t t
t j j d d S )Nr )r_ r )r r r r
r r% r& s r test_standard_gamma_0z$TestRandomDist.test_standard_gamma_0 sE RY--A-66:::j")":#FFFFFFr c t j | j t j d }t j ddgddgddgg }t ||d
d S ) Nr rw r r r r r r r r )r r ri ry r6 r r s r test_standard_normalz#TestRandomDist.test_standard_normal s
ty!!!***77( 02E F .0C D .0C D F G G "&'2>>>>>>r c t j | j t j dd }t j ddgddgdd gg }t ||d
d S ) Nr r )r rx g\VH?ghU%gH$M?gLgVqdzǿg㖀?r r )r r ri
standard_tr6 r r s r test_standard_tz TestRandomDist.test_standard_t s
ty!!!%%&%99( 02F G 02F G 13F G I J J "&'2>>>>>>r c t j | j t j dddd }t j ddgdd gd
d gg }t ||d
d S )Ng{Gz@g(\u$@gףp=
W4@r )leftmoderightrx g:}\)@ghرTj(@gǨL30@g6ڜA0@g6sh&@g<,@r r )r r ri
triangularr6 r r s r test_triangularz TestRandomDist.test_triangular s
ty!!!%%4e5+1 & 3 3( 13F G 13G H 13F G I J J "&'2>>>>>>r c t j | j t j ddd }t j ddgddgd d
gg }t ||d d S )
NGz?gGz%@r )lowhighrx g>#H@g u@g#@g}d]#@gN@gw )* @r r )r r ri uniformr6 r r s r test_uniformzTestRandomDist.test_uniform s
ty!!!""t%f"EE( 02E F 02E F 02E F H I I "&'2>>>>>>r c < t j d j }t j d j }t j j }t
t |t j d t
t |dt j t
t ||| t
t |t j gdg t
t |dgt j g t j t j |d |dz d S )Nr` r r* g W4vC)r r )
r finfor r r r r
OverflowErrorinf nextafter)r fminfmaxfuncs r test_uniform_range_boundsz(TestRandomDist.test_uniform_range_bounds s x $x $y mTBF7A666mTABF;;;mTDD999mTRVG9qc:::mTA3999
bl4 3 3$+FFFFFr c G d dt j }t j d | }t t
t j j || G d dt j }t j d | }t t
t j j |dd d S )Nc e Zd Zd ZdS )GTestRandomDist.test_scalar_exception_propagation..ThrowingFloatc t r+ r$ r& s r __float__zQTestRandomDist.test_scalar_exception_propagation..ThrowingFloat.__float__ r N)r9 r: r; r0 r< r r
ThrowingFloatr- s#
r r2 g ?c e Zd Zd ZeZdS )ITestRandomDist.test_scalar_exception_propagation..ThrowingIntegerc t r+ r/ r& s r __int__zQTestRandomDist.test_scalar_exception_propagation..ThrowingInteger.__int__ r1 r N)r9 r: r; r6 __index__r< r r ThrowingIntegerr4 s$
IIIr r8 r* ) r rs r6 r r r$ r r r )r r2 throwing_floatr8 throwing_ints r !test_scalar_exception_propagationz0TestRandomDist.test_scalar_exception_propagation s BJ # ++M::i!2N$ & & & bj x{{''88 i!9<ANNNNNr c t j | j t j ddd }t j ddgddgd d
gg }t ||d d S )
Nr p=
ף?r mukapparx gLZI@g'N"@gkr?gk=m@gډ?g^=\?r r )r r ri vonmisesr6 r r s r
test_vonmisesz TestRandomDist.test_vonmises s
ty!!!##t4f#EE( 02E F 02E F 02E F H I I "&'2>>>>>>r c t j | j t j ddd }t j t j | d S )NrS g g5_PG>i@B r> )r r ri rA r r isfiniterW )r rs r test_vonmises_smallz"TestRandomDist.test_vonmises_small sd
ty!!! I "F ? ?
2;q>>--//00000r c t j | j t j ddd }t j ddgddgd d
gg }t ||d d S )
Nr r= r )r r rx gZ@g-
g@g8
m?gVP"?g`V?gx^&3?r r )r r ri waldr6 r r s r test_waldzTestRandomDist.test_wald s
ty!!! TF CC( 02E F 02E F 02E F H I I "&'2>>>>>>r c t j | j t j dd }t j ddgddgdd gg }t ||d
d S ) Nr r r g@6?g]A?g(1ڥR?gU?g2_u?g^}-Q?r r )r r ri weibullr6 r r s r test_weibullzTestRandomDist.test_weibull s
ty!!!""T"77( 02E F 02E F 02E F H I I "&'2>>>>>>r c t j | j t t j dd t j d t
t t j j d d S )Nr r r r )r )r r ri r rK rB r r% r& s r test_weibull_0z TestRandomDist.test_weibull_0
se
ty!!!RY&&&44bhrllCCCj")"3s;;;;;;r c t j | j t j dd }t j ddgddgddgg }t || d S ) Nr r r B r* r2 r )r r ri zipfr6 r r s r test_zipfzTestRandomDist.test_zipf sq
ty!!! $V 44(RH F G % & & 67+++++r N)Sr9 r: r; rm r r r r r r r r r r
r r r r" r% r( rM rZ r^ mark parametrizer r r rz rd rx r| r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r! r* r; rB rF rI rL rN rS r< r r r r ! sG ? ? ?? ? ?, , , , , ,
&
&
& @ @ @ ? ? ?, , , , , , , , , , , ,, , , 2 2 2&,< ,< ,<\< < <
& & & 0 0 04 L L L [ X
Y --//1F1F1H1H IK K0 0K K0 [ X BI))++RY-B-B-D-DEG G [ -e}==7 7 >=G G70
5
5
5 ! ! ! ? ? ?, , ,? ? ?
?
?
?
C
C
C P P P? ? ?D D D? ? ?? ? ?I I I, , ,? ? ?? ? ?, , ,4? ? ?@ @ @? ? ?? ? ?B B B, , , , , ,& & & P, , ,? ? ?*? ? ?? ? ?? ? ?
L
L
L , , ,B B B? ? ?? ? ?A A A? ? ?? ? ?? ? ?G G G? ? ?? ? ?? ? ?? ? ?G G G O O O0? ? ?1 1 1 ? ? ?? ? ?< < <
, , , , ,r r c e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
Z d Z
d Zd
Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Z d Z d Z d Z d Z d Z!d Z"d!S )"
TestBroadcastc d| _ d S )Ni[r r& s r rm zTestBroadcast.setup_method s
r c N t j | j d S r+ )r r ri r& s r setSeedzTestBroadcast.setSeed s
ty!!!!!r c 4 dg}dg}t j j }t j g d }| ||dz | }t ||d | |||dz }t ||d d S )Nr r* gl ?g_x
?g =*BN?r2 r r )r r r r6 rZ r )r r r r r r s r r! zTestBroadcast.test_uniform$ s cs)#( 1 1 1 2 2
q$''!&'2>>>> dQh''!&'2>>>>>>r c dg}dg}dg}t j j }t j g d }| ||dz | }t ||d t
t ||dz | | |||dz }t ||d t
t |||dz d S )Nr r* r" )g-r@gYl@g-`w?r2 r r )r r r r6 rZ r r r% )r r r bad_scaler r r s r r zTestBroadcast.test_normal4 s cD !( 0 0 0 1 1
a''!&'2>>>>j&#'9=== UQY''!&'2>>>>j&#y1}=====r c dg}dg}dg}dg}t j j }t j g d }| ||dz | }t ||d t
t ||dz | t
t ||dz | | |||dz }t ||d t
t |||dz t
t |||dz d S ) Nr* r+ r" r, )gnQVf?gVKB?g v`k?r2 r r )r r r r6 rZ r r r% )r r r bad_abad_br r r s r r zTestBroadcast.test_betaG s
C
Cy~( 1 1 1 2 2
a!eQ !&'2>>>>j$ 1555j$Au555 aQ !&'2>>>>j$q1u555j$51955555r c dg}dg}t j j }t j g d }| ||dz }t ||d t
t ||dz d S Nr* r" )g.UgZ?gi Prq?gE
K?r2 r r )r r r r6 rZ r r r% )r r r^ r r r s r r z TestBroadcast.test_exponential] s D i+ ( 1 1 1 2 2
UQY''!&'2>>>>j+y1}=====r c dg}dg}t j j }t j g d }| ||dz }t ||d t
t ||dz d S rc )r r r
r6 rZ r r r% )r r_ bad_shape std_gammar r s r r z!TestBroadcast.test_standard_gammaj s D I, ( 1 1 1 2 2
519%%!&'2>>>>j)Y];;;;;r c dg}dg}dg}dg}t j j }t j g d }| ||dz | }t ||d t
t ||dz | t
t ||dz | | |||dz }t ||d t
t |||dz t
t |||dz d S ) Nr* r+ r" r, )g.UgZ?gi Prq?gE
K?r2 r r )r r r r6 rZ r r r% )r r_ r re r^ r r r s r r zTestBroadcast.test_gammaw s D D ( 0 0 0 1 1
uqy%((!&'2>>>>j%Q >>>j%I>>> ueai((!&'2>>>>j%EAI>>>j% A
>>>>>r c dg}dg}dg}dg}t j j }t j g d }| ||dz | }t ||d t
t ||dz | t
t ||dz | | |||dz }t ||d t
t |||dz t
t |||dz d S ) Nr* r+ r" r, )gbyʜ?go?gT7@r2 r r )r r r r6 rZ r r r% )r r r bad_dfnum bad_dfdenr r r s r r zTestBroadcast.test_f s D D IK( 0 0 0 1 1
519e$$!&'2>>>>j!Y]E:::j!UQY ::: 5%!)$$!&'2>>>>j!Y :::j!UIM:::::r c dg}dg}dg}dg}dg}dg}t j j }t j g d }| ||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 ||||dz t
t ||||dz t
t ||||dz d S )
Nr+ r2 r3 r r" r, )gr^G"@g/
*@gц!@r r )r r r r6 rZ r r r% )
r r r r ri rj bad_noncnonc_fr r s
r r z TestBroadcast.test_noncentral_f s sC D 4'( 0 0 0 1 1
5$//!&'2>>>>j&)a-EEEj&%!)YEEEj&%!)UHEEE uqy$//!&'2>>>>j&)UQYEEEj&%Q EEEj&%HEEE udQh//!&'2>>>>j&)UD1HEEEj&%D1HEEEj&%1 EEEEEr c | t j ddg }t j dddd }t ||d d S )Ngaz@gn4M;%?r r+ rw r r )rZ r r6 r r r )r r r s r test_noncentral_f_small_dfz(TestBroadcast.test_noncentral_f_small_df s^ ( -/@ABB''S!!'<<!&'2>>>>>>r c dg}dg}t j j }t j g d }| ||dz }t ||d t
t ||dz d S )Nr* r" )g~}N??g$F?g'rS?r2 r r )r r r r6 rZ r r r% )r r bad_dfr r r s r r z TestBroadcast.test_chisquare s SI' ( 0 0 0 1 1
26""!&'2>>>>j)VaZ88888r c dg}dg}dg}dg}t j j }t j g d }| ||dz | }t ||d t
t ||dz | t
t ||dz | | |||dz }t ||d t
t |||dz t
t |||dz d S ) Nr* r+ r" r, )g w "@g2 WR@gRY@r2 r r )r r r r6 rZ r r r% )r r r rq rl nonc_chir r s r r z'TestBroadcast.test_noncentral_chisquare s Ss491( 0 0 0 1 1
"q&$''!&'2>>>>j(FQJ===j(BFH=== "dQh''!&'2>>>>j(FD1H===j(B1 =====r c dg}dg}t j j }t j g d }| ||dz }t ||d t
t ||dz d S )Nr* r" )gT
@g~Fl@gƚtp?r2 r r )r r r r6 rZ r r r% )r r rq tr r s r r z TestBroadcast.test_standard_t s S I ( 0 0 0 1 1
26!&'2>>>>j!VaZ00000r c dg}dg}dg}t j j }t j g d }| ||dz | }t ||d t
t ||dz | | |||dz }t ||d t
t |||dz d S )Nr+ r* r" )gW!@gOBg}?r2 r r )r r rA r6 rZ r r r% )r r? r@ bad_kapparA r r s r rB zTestBroadcast.test_vonmises s SD 9%( 1 1 1 2 2
"q&%((!&'2>>>>j(BFI>>> "eai((!&'2>>>>j(B A
>>>>>r c dg}dg}t j j }t j g d }| ||dz }t ||d t
t ||dz d S )Nr* r" )g&8??geFX?gՋː?r2 r r )r r r r6 rZ r r r% )r r r` r r r s r r zTestBroadcast.test_pareto
s
C!( 0 0 0 1 1
A !&'2>>>>j&%!)44444r c dg}dg}t j j }t j g d }| ||dz }t ||d t
t ||dz d S rc )r r rK r6 rZ r r r% )r r r` rK r r s r rL zTestBroadcast.test_weibull s
C)#( 1 1 1 2 2
Q !&'2>>>>j'51955555r c dg}dg}t j j }t j g d }| ||dz }t ||d t
t ||dz d S )Nr* r" r\ r2 r r )r r r r6 rZ r r r% )r r r` r r r s r r zTestBroadcast.test_power' s
C ( 1 1 1 2 2
q1u !&'2>>>>j%33333r c dg}dg}dg}t j j }t j g d }| ||dz | }t ||d t
t ||dz | | |||dz }t ||d t
t |||dz d S )Nr r* r" )g:BKc?g3k?g~q?r2 r r )r r r r6 rZ r r r% )r r r r^ r r r s r r zTestBroadcast.test_laplace4 s cD )#( 2 2 2 3 3
q%((!&'2>>>>j'37I>>> eai((!&'2>>>>j'3 A
>>>>>r c dg}dg}dg}t j j }t j g d }| ||dz | }t ||d t
t ||dz | | |||dz }t ||d t
t |||dz d S )Nr r* r" )g'Zy?gdSO=?g3L1?r2 r r )r r r r6 rZ r r r% )r r r r^ r r r s r r zTestBroadcast.test_gumbelG s cD !( 1 1 1 2 2
a''!&'2>>>>j&#'9=== UQY''!&'2>>>>j&#y1}=====r c dg}dg}dg}t j j }t j g d }| ||dz | }t ||d t
t ||dz | | |||dz }t ||d t
t |||dz d S )Nr r* r" )gy ?gJR?g
~&?r2 r r )r r r r6 rZ r r r% )r r r r^ r r r s r r zTestBroadcast.test_logisticZ s cD 9%( 2 2 2 3 3
#'5))!&'2>>>>j(C!GY??? #uqy))!&'2>>>>j(CQ ?????r c dg}dg}dg}t j j }t j g d }| ||dz | }t ||d t
t ||dz | | |||dz }t ||d t
t |||dz d S )Nr r* r" )gH"@gܮ @gvv :@r2 r r )r r r r6 rZ r r r% )r r r bad_sigmar r r s r r z TestBroadcast.test_lognormalm s sD I' ( 0 0 0 1 1
4!8U++!&'2>>>>j)TAXyAAA 4++!&'2>>>>j)T9q=AAAAAr c dg}dg}t j j }t j g d }| ||dz }t ||d t
t ||dz d S )Nr* r" )gݱlo?g/ӂ?gz~R?r2 r r )r r r r6 rZ r r r% )r r r^ r r r s r r zTestBroadcast.test_rayleigh s D 9%( 0 0 0 1 1
%!)$$!&'2>>>>j(IM:::::r c l dg}dg}dg}dg}t j j }t j g d }| ||dz | }t ||d t
t ||dz | t
t ||dz | | |||dz }t ||d t
t |||dz t
t |||dz t
t |d d t
t |dd d S )
NrA r* r r, )g@m e?g`I߿?g|2?r2 r r rS )r r rH r6 rZ r r r% )r r r bad_meanr^ rH r r s r rI zTestBroadcast.test_wald sE u3D y~( 0 0 0 1 1
dQh&&!&'2>>>>j$1 e<<<j$q)<<< dEAI&&!&'2>>>>j$%!)<<<j$i!m<<<j$Q///j$S11111r c dg}dg}dg}dg}dg}|dz \ }}t j j }t j g d } | ||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 ||||dz t
t ||||dz t
t ||||dz d S )Nr* r2 r+ r3 )g,P;bD @g%G @g,~S @r r )r r r r6 rZ r r r% ) r r r r bad_left_one bad_mode_one bad_left_two bad_mode_twor r r s r r z TestBroadcast.test_triangular s sss s %*QY" lY)
( 0 0 0 1 1
D1HdE22!&'2>>>>j*lQ.>eLLLj*dQh eLLLj*lQ.>
D$(E22!&'2>>>>j*lD1HeLLLj*dL14DeLLLj*lL1>>>j*lD%!)LLLj*dL%!)LLLj*lLai ! ! ! ! !r c v dg}dg}dg}dg}dg}t j j }t j g d }| ||dz | }t || t
t ||dz | t
t ||dz | t
t ||dz | | |||dz }t || t
t |||dz t
t |||dz t
t |||dz d S )Nr* rA r" ?r* r* r* r2 )r r rC r6 rZ r r r% ) r r rD bad_n bad_p_one bad_p_twobinomr r s r r zTestBroadcast.test_binomial s8
C
ED E "(999%% q1ua 67+++j%A666j%Q :::j%Q ::: q!a% 67+++j%A666j%IM:::j%IM:::::r c v dg}dg}dg}dg}dg}t j j }t j g d }| ||dz | }t || t
t ||dz | t
t ||dz | t
t ||dz | | |||dz }t || t
t |||dz t
t |||dz t
t |||dz d S )Nr* rA r" r )r* r r* r2 )r r r r6 rZ r r r% ) r r rD r r r neg_binomr r s r r z$TestBroadcast.test_negative_binomial s<
C
ED E I/ (999%% 1q5!$$67+++j)UQY:::j)QUI>>>j)QUI>>> 1a!e$$67+++j)UAE:::j)Q A
>>>j)Q A
>>>>>r c z t j j }dg}dg}|dz g}t j j }t j g d }| ||dz }t || t t ||dz t t ||dz d S )Nr* r" r+ )r* r* r r2 )
r r r _poisson_lam_maxr r6 rZ r r r% )r max_lamr bad_lam_one bad_lam_twor r r s r r zTestBroadcast.test_poisson s )'')):cd {m )#(999%% q!!67+++j';?;;;j';?;;;;;r c dg}dg}t j j }t j g d }| ||dz }t || t
t ||dz t j d 5 t
t |t j t
t |ddt j g d d d d S # 1 swxY w Y d S )Nr+ r )r+ r+ r* r2 r )invalid)
r r rR r6 rZ r r r% errstaterG )r r r` rR r r s r rS zTestBroadcast.test_zipf s
Cy~(999%% a!e67+++j$ 222
[
*
*
* < < *dBF 3 3 3 *dQ26N ; ; ; < < < < < < < < < < < < < < < < < dg}dg}dg}dg}dg}dg}dg}t j j }t j g d } | ||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 ||||dz t
t ||||dz t
t ||||dz t
t ||||dz d S ) Nr* r+ r" r, r r3 r r2 )r r r r6 rZ r r r% ) r ngoodnbadnsample bad_ngoodbad_nbadbad_nsample_onebad_nsample_two hypergeomr r s r r z!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:MNNNNNr c . dg}dg}dg}t j j }t j g d }| ||dz }t || t
t ||dz t
t ||dz d S )NrA r+ r" r r2 )r r r r6 rZ r r r% )r rD r r r r r s r r z TestBroadcast.test_logseriesD s
EC D I' (999%% 1q5!!67+++j)Y];;;j)Y];;;;;r N)#r9 r: r; rm rZ r! r r r r r r r ro r r r rB r rL r r r r r r rI r r r r rS r r r r< r r rW rW s " " " ? ? ? > > >&6 6 6, > > > < < <? ? ?,; ; ;, F F FB? ? ? 9 9 9> > >, 1 1 1? ? ?& 5 5 5 6 6 6 4 4 4? ? ?&> > >&@ @ @&B B B& ; ; ;2 2 20"! "! "!H; ; ;.? ? ?.
<
<
< < < < 7 7 7!O !O !OF < < < < ._ sQ
0
0
0A V829+@+@+C+CQ*G
H
H
H
0
0
0r c 6 g | ]}| S r< )startr: rX s r r< z-TestThread.check_function..a s q r c 6 g | ]}| S r< )joinr s r r< z-TestThread.check_function..b s a r r3 win32) threadingr r r r r zipr r intpr0 itemsizer platformr r ) r r szout1out2ru r r r s ` @r check_functionzTestThread.check_functionX sK $$$$$$xTZ *R/00xTZ *R/00
0
0
0
0
0 TZ..
0
0
0 A 1
D)) 2 2DAq HRY**1--q 1 1 1 1 799? #q ( (S\W-D-D %dD 1 1 1 1 1 tT * * * * *r c : d }| |d d S )Nc 8 | d |d< d S )N' rw .)r rl outs r
gen_randomz*TestThread.test_normal..gen_randomo s |||//CHHHr )r r r r r s r r zTestThread.test_normaln s0 0 0 0 J844444r c : d }| |d d S )Nc \ | t j d |d< d S )NrL r r .)r r onesr s r r z'TestThread.test_exp..gen_randomt s* ((rw{/C/C(DDCHHHr r r r r s r test_expzTestThread.test_exps s3 E E E J;77777r c : d }| |d d S )Nc D | ddgdz d |d< d S )Nr r r5 r rw .)rP r s r r z/TestThread.test_multinomial..gen_randomy s* ((dVAXE(BBCHHHr )r r5 r r r s r r zTestThread.test_multinomialx s3 C C C J:66666r N)r9 r: r; rm r r r r r< r r r r R s_ + + +,5 5 5
8 8 8
7 7 7 7 7r r c , e Zd Zd Zd Zd Zd Zd ZdS )TestSingleEltArrayInputc t j dg | _ t j dg | _ t j dg | _ d| _ d S )Nr+ r2 r3 )r* )r r6 argOneargTwoargThreetgtShaper& s r rm z$TestSingleEltArrayInput.setup_method s@ hsmm hsmm !
r c r t j j t j j t j j t j j t j j t j j t j j t j j t j j
t j j t j j t j j
f }t j j t j j
f}|D ]O}||v r |t j dg }n || j }t! |j | j Pd S r )r r r r
r r r rK r r r rR r r r6 r r r_ r )r funcs probfuncsr) r s r test_one_arg_funcsz*TestSingleEltArrayInput.test_one_arg_funcs s & (@$bi&:!29#4 ")"4"BIN$bi&9 ; Y(")*=> 3 3Dy d28SE??++ d4;'' DM 2 2 2 2 3 3r c t t j j t j j t j j t j j t j j t j j t j j t j j t j j
t j j t j j t j j
t j j t j j f}t j j t j j f}|D ]}||v rt j dg }n| j } || j | }t' |j | j || j d | }t' |j | j || j |d }t' |j | j d S )NrA r )r r r r r r r r rA r r r r rH rC r r6 r r r r_ r )r r r r) r r s r test_two_arg_funcsz*TestSingleEltArrayInput.test_two_arg_funcs sC "BI$4 bi <#RY%6!29#5$bin#RY%@
B Y')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 3r c N t t j t j t j t j t j t j t j t j g }t j
j }t j dg }t j dg }|D ]} |||| }t |j | j ||d || }t |j | j |||d | }t |j | j d S )Nr* r r/ )r r r r r r r r^ r7 r r r r6 r r_ r )r r r) r r r r s r r z$TestSingleEltArrayInput.test_randint s rw"(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 3r c t j j t j j t j j g}|D ]} || j | j | j }t |j | j
|| j d | j | j }t |j | j
|| j | j d | j }t |j | j
d S )Nr ) r r r r r r r r r r_ r )r r r) r s r test_three_arg_funcsz,TestSingleEltArrayInput.test_three_arg_funcs s ')=)+ 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)r9 r: r; rm r r r r r< r r r r s_ 3 3 3&3 3 363 3 3" 3 3 3 3 3r r )r r^ numpyr
numpy.testingr r r r r r r r
r r r r r> rJ rf r r rW rT skipifr r r< r r r s|
$F $F $F $F $F $F $F $FN > > > > > > > > Y Y Y Y Y Y Y YD-. -. -. -. -. -. -. -.`G+ G+ G+ G+ G+ G+ G+ G+Tt , t , t , t , t , t , t , t ,nw< w< w< w< w< w< w< w