[>% 3 d dl Z d dlZd dlZd dlZd dlmZ d dlmZm Z m
Z
m Z m Z m
Z
mZmZmZmZ d dlmZmZmZmZ e e Zd ddddd d
dd d d
ddddddd g ddddddddd gZ ej dddg d Z G d d Z G d d Z G d! d" Z G d# d$ Z G d% d& Z G d' d( Z G d) d* Z! G d+ d, Z"ej# $ ed-. G d/ d0 Z% G d1 d2 Z&ej# ' d3e d4 Z(d5 Z)d6 Z*d7 Z+ej# ' d8d9d9gg ej# ' d:ddg d; Z,ej# ' dg d? Z-ej# ' dg ej# ' d@dAdBg ej# ' dCej. ej g dD Z/dE Z0dS )F N) LinAlgError)
assert_
assert_raises assert_equalassert_allclose assert_warnsassert_no_warningsassert_array_equalassert_array_almost_equalsuppress_warningsIS_WASM) GeneratorMT19937 SeedSequence RandomState
@bb1636883c2707b51c5b7fc26c6927af4430f2e0785a8c7bc886337f919f9edf )
key_sha256pos@ff682ac12bb140f2d72fba8d3506cf4e46817a0db27aae1683867629031d8d55iV )seedstepsinitialjumpedi$<i8 @16b791a1e04886ccbbb4d448d6ff791267dc458ae599475d08d5cced29d11614i7 @a0110a2cf23b56be0feaed8f787a7fc84bef0cb5623003d75b26bdfa1c18002ci )i|2iEm:iS43i5 i @d306cf01314d51bd37892d874308200951a35265ede54d200f1e065004c3e9eai @0e00ab449f01a5195a83b4aee0dfbc2ce8d46466a640b92e33977d2e42f777f8i moduleTF)scopeparamsc | j S N)param)requests }/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/numpy/random/tests/test_generator_mt19937.pyendpointr( & s
= c 2 e Zd Zd Zd Zd Zd Zd Zd ZdS )TestSeedc t t d }t | d d t t d }t | d d d S )Nr D )r r r integersselfss r' test_scalarzTestSeed.test_scalar, sh gajj ! !QZZ%%s+++ gj)) * *QZZ%%s+++++r) c @ t t t d }t | d d t t t j d }t | d d t t dg }t | d d t t dg }t | d d d S )Nr r- i r r. r/ r0 )r r ranger r1 nparanger2 s r'
test_arrayzTestSeed.test_array2 s geBii(( ) )QZZ%%s+++ gbimm,, - -QZZ%%s+++ gqcll # #QZZ%%s+++ gzl++ , ,QZZ%%s+++++r) c t t d }t | d d d S )Nr iz)r r r
random_rawr2 s r' test_seedsequencezTestSeed.test_seedsequence< s5 LOO $ $Q\\!__j11111r) c r t t t d t t t d d S )N r TypeErrorr
ValueErrorr3 s r' test_invalid_scalarz TestSeed.test_invalid_scalar@ s, i$///j'2.....r) c t t t dg t t t dg t t t g d d S )Nr@ rA )r< rB rE s r' test_invalid_arrayzTestSeed.test_invalid_arrayE sI i4&111j'B4000j'+>+>+>?????r) c F t t t t d S r$ )r rD r r rE s r' test_noninstantized_bitgenz#TestSeed.test_noninstantized_bitgenK s j)W55555r) N) __name__
__module__ __qualname__r5 r: r> rF rJ rL r) r' r+ r+ + sq , , , , , ,2 2 2/ / /
@ @ @ 6 6 6 6 6r) r+ c e Zd Zd Zd ZdS ) TestBinomialc t j dd }dD ]W}t t d| dk t t || | Xd S )N intdtype)r ?r< r )r8 zerosr randombinomialr
)r3 rY ps r' test_n_zerozTestBinomial.test_n_zeroP sw %((( A AA FOOAq))Q. / / / vua 8 8% @ @ @ @ A Ar) c \ t t t j dt j d S )Nr< )r rD rZ r[ r8 nanrE s r'
test_p_is_nanzTestBinomial.test_p_is_nanY s j&/1bf=====r) N)rM rN rO r] r` rP r) r' rR rR O s5 A A A> > > > >r) rR c > e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d S )
TestMultinomialc @ t dddg d S )Nd 皙?皙?rZ multinomialrE s r'
test_basiczTestMultinomial.test_basic_ s" 3c
+++++r) c @ t dg d d S )Nrd )re rf rk rk rg rE s r' test_zero_probabilityz%TestMultinomial.test_zero_probabilityb s% 3 9 9 9:::::r) c > t dt dd cxk odk nc t ddd }t t j d|k t t j |dk d S )NrA )r rZ r1 r8 allr3 xs r' test_int_negative_intervalz*TestMultinomial.test_int_negative_intervale s foob"--22222222333 OOBA & &rQw q2v r) c 4 ddg}t t d|t j d j d t t d|t j d j d t t d|t j d j d t t d|ddg j d t t d|d j d t t d|t j d j d t t t j d|t d d S )NrX r< r< rT rT rT rT rT rT rT )
r rZ rh r8 uint32shapearrayr rC floatr3 r\ s r' test_sizezTestMultinomial.test_sizek sH
#JV''1bill;;A6JJJV''1bill;;A6JJJV''1bill;;A6JJJV''1q!f55;YGGGV''1f55;YGGGV''1bhv.>.>??E i!3QAhh r) c t t t j dddg t t t j dddg d S )Nrd 皙?re 皙?r rD rZ rh rE s r' test_invalid_probz!TestMultinomial.test_invalid_proby s> j&"4cC:FFFj&"4cC:FFFFFr) c t t t j dddg t t t j dgdz ddg d S )NrA rf re r r rE s r' test_invalid_nz TestMultinomial.test_invalid_n} sD j&"4b3*EEEj&"4rdRi#sLLLLLr) c t j d }|t j |dd d z }|dd d }t t d }| d| }t t d }| dt j | }t || d S )Ng .@r< ۠iUrd )pvals)r8 r9 sumr r rh ascontiguousarrayr
)r3 r\ r rZ
non_contigcontigs r' test_p_non_contiguousz%TestMultinomial.test_p_non_contiguous s IcNN RVAaddG__ !$Q$7:..//''5'99
7:..//##Cr/CE/J/J#KK:v.....r) c H t j g dt j }|| z }t t d }d}t
j t | 5 | d| d d d d S # 1 swxY w Y d S )N)
Gz?r & .>r r r r r r r rV r z-[\w\s]*pvals array is cast to 64-bit floatingmatchr< )
r8 rz float32r r r pytestraisesrD rh )r3 rr r rZ r s r' test_multinomial_pvals_float32z.TestMultinomial.test_multinomial_pvals_float32 s H : : :AC
M
M
MAEEGG 7:..//@
]:U
3
3
3 ) ) q% ( ( ( ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) )s 3BB BN) rM rN rO ri rl rs r} r r r r rP r) r' rb rb ^ s , , ,; ; ; G G GM M M/ / /) ) ) ) )r) rb c P e Zd Zd Zd Zej dddg d Zej dg d ej dddg ej d g d
d Z d Z
d
Z d Z dS ) TestMultivariateHypergeometricc d| _ d S )Ni_ r rE s r' setup_methodz+TestMultivariateHypergeometric.setup_method s
r) c ^ t t t j dd t t t j g dd t t t j g dd t t t j g dd t t t j g d t t t j dd gd
dd t j t j }|j }||j j z }t t t j |d z
d gd
dd
d S )Nr )rT r r rA )rA rT r rT r< iɚ;e ro marginalsrd count)
r rD rZ multivariate_hypergeometricr8 iinfoint64maxrW itemsize)r3
int64_info max_int64max_int64_indexs r' test_argument_validationz7TestMultivariateHypergeometric.test_argument_validation s< j&"D! j&"D ii % % % j&"D jj! % % % j&"D ii % % % j&"D! j&"D #&1k ; ; ; Xbh''
N #z'7'@@j&"D&,c2Aq' C C C C Cr) methodr r c t t | j }| g dd| }t |g d | g d| }t |g | g dd| }t |t j dt
j | g dd| }t |g d | g d d
| }t |g d g d }| |t | | }t || | g d
dd
| }t |g d
gd
z d S )N)r r r r r r< sizer )r< r rV r< rT r )r r r r )r r r )r< r< r r< r< r r ro ) r r r r r
r8 emptyr r )r3 r rZ rr colorss r' test_edge_casesz.TestMultivariateHypergeometric.test_edge_cases s 749--.. . .yyy!F . K K1iii((( . .r1V . D D1b!!! . .r11V . L L1bhvRX > > >??? . .yyy!F . K K1iii((( . .yyy!F . K K1iii((( . .vs6{{6<
/
>
>1f%%% . .yyy"16<
/
>
>1yyyk!m,,,,,r) nsample) - 7 r )ro rT r iI c t t | j }t j g d }| |||| }t
|t r |f|j z }n
||j z }t |j | t |dk t ||k t |
d t j ||t t
|t rG|dk rCt | d ||z |
z dd
d S d S d S ) N)r ro r r r rA axis)
fill_valuerW 順 MbP?g{Gzt?)rtolatol)r r r r8 rz r
isinstancerU ry r r rp r
r fullr mean)r3 r r r rZ r sampleexpected_shapes r' test_typical_casesz1TestMultivariateHypergeometric.test_typical_cases s~ 749--..///**33FGT;A 4 C C dC 1 "Wv| 3NN !FL 0NV\>2221!!##$$$6!&&(()))6::2:.. 74G3GGG I I I dC 3TV^^
FKKQK / / #f ,vzz|| ;!%E
3
3
3
3
3
3 3 3^^r) c t t | j }| g dddd }t j g dg dg dg dg dg }t
|| d S )Nr ro r r )rT r< rT r< r< r rT r r r r r r r8 rz r
r3 rZ r expecteds r' test_repeatability1z2TestMultivariateHypergeometric.test_repeatability1 s 749--..33IIIqq;B 4 D D8YYY &YY &YY &YY &YY ( ) )
68,,,,,r) c t t | j }| g dddd }t j g dg dg dg dg d g }t
|| d S )
Nr 2 r ro r r )r )
r )r )r r r r s r' test_repeatability2z2TestMultivariateHypergeometric.test_repeatability2 s 749--..33LLL"9:;F 4 H H 8\\\ )\\ )\\ )\\ )\\ + , ,
68,,,,,r) c t t | j }| g dddd }t j g dg dg dg dg d g }t
|| d S )
Nr r ro r r )rT r r )ro r r )rT ro ro )r< ro r r s r' test_repeatability3z2TestMultivariateHypergeometric.test_repeatability3
s 749--..33LLL"9:;F 4 H H 8YYY &YY &YY &YY &YY ( ) )
68,,,,,r) N)
rM rN rO r r r mark parametrizer r r r r rP r) r' r r s C C CB [ X'=>>- - ?>-D [ Y88 [ X'=>> [ V%8%8%8993 3 :9 ?> 983, - - -
-
-
-
-
-
-
-
-r) r c & e Zd Zd Zd Zd Zd ZdS ) TestSetStatec d| _ t t | j | _ | j j | _ | j j | _ | j d | j d d | j d d f| _ d S )NI
bit_generatorstatekeyr )r r r rgr r legacy_staterE s r' r zTestSetState.setup_method st GDI . .// !W 2'-
!Z 8 !Z 0 7 !Z 0 7 9r) c | j d }| j | j _ | j d }t t j ||k d S Nr r )r standard_normalr r r r8 rp )r3 oldnews r' test_gaussian_resetz TestSetState.test_gaussian_reset$ sa g%%1%--#': g%%1%--scz""#####r) c | j | j j }| j d }|| j _ | j d }t t j ||k d S r )r r r r r r8 rp )r3 r r r s r' test_gaussian_reset_in_media_resz-TestSetState.test_gaussian_reset_in_media_res+ s
!!!"(g%%1%--#( g%%1%--scz""#####r) c < | j dd d S NrX )r negative_binomialrE s r' test_negative_binomialz#TestSetState.test_negative_binomial6 s"
!!#s+++++r) N)rM rN rO r r r r rP r) r' r r sP 9 9 9$ $ $ $ $ $, , , , ,r) r c d e Zd Zej Zeej ej ej
ej ej ej
ej ej g Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd
Zd Ze j d d
ej g d fd ej g d fd ej g d fg d Z d Z!d Z"d Z#d Z$d Z%d Z&d Z'e j j( e j dd d ej d fd d ej d fd!d"ej
d#fd$d%ej d&fg d' Z)d(S )) TestIntegersc L t t | j d|t d S Nr< r( rW )r rC rfuncr{ )r3 r( s r' test_unsupported_typez"TestIntegers.test_unsupported_typeC s# iQOOOOOOr) c " | j D ]}|t u rdnt j | j }|t u rdnt j | j dz }|r|dz
n|}t
t | j |dz
||| t
t | j ||dz || t
t | j |||| t
t | j dd|| t
t | j |dz
g||| t
t | j |g|dz g|| t
t | j |g|g|| t
t | j ddg|| t
t | j |dz g|g|| d S Nr rT r< r itypeboolr8 r minr r rD r r3 r( dtlbndubnds r' test_bounds_checkingz!TestIntegers.test_bounds_checkingF s * 7 7Bd
11 (8Dd
11 (81(
@
@
@ sQwi$ .T $ E EFI
K
K
K C SEC!G9)-R $ 9 9:=
?
?
? sQwi$ .b $ : :;>
@
@
@ sQwi$ .T $ E EFI
K
K
K $;1$C SEC!G9)-R $ 9 9:=
?
?
? sQwi$ .b $ : :;>
@
@
@ sQwi$ .T $ E EFI
K
K
K
K5 K Kr) c X | j D ]}|t u rdnt j | j }|t u rdnt j | j dz }|r|dz
n|} | |||| q# t $ r$}t dt | z d }~ww xY wd S Nr rT r< r zMNo error should have been raised, but one was with the following message:
%s
r r r8 r r r r ExceptionAssertionErrorstrr3 r( r r r es r' test_full_rangez TestIntegers.test_full_range s *
@
@Bd
11 (8Dd
11 (81(>
B, B''B,c t t }| j dd D ]n}dD ]i}| d||z
d|| }t | |k t | dk jo| dd|z
d|t }t | dk t | dk d S )Nr< )r r rT i r r )r r r r r r r r )r3 r( rZ r r valss r' test_in_bounds_fuzzz TestIntegers.test_in_bounds_fuzz s 799%%*QRR. ) )B"
)
)zz!TH_7+32 " ? ?
T)***
a((((
) zz!Q\( $ & &
Q
a r) c b | j D ]%}|t u rdnt j | j }|t u rdnt j | j dz }|r|dz
n|}d}t
t d }| ||||| }t
t d }| |g|g||| }t
t d }| |g|z |g|z ||| } t || t || 'd S )Nr rT r< r- r )
r r r8 r r r r r r1 r
)
r3 r( r r r r rZ scalar scalar_arrayrz s
r' test_scalar_array_equivz$TestIntegers.test_scalar_array_equiv sR * . .Bd
11 (8Dd
11 (81({l wT2L iJQ)iQ//itLol k )i\siJi)i=l }D l E9l )l
+x i#Jl b )i0i7Ol rn )ixiiAi.t)l q/q i;-/i)N)ro r r i90 rA r r< )r/ rI rH r )r8 rz r r r1 r
)r3 desiredr rZ rr s r' .test_repeatability_32bit_boundary_broadcastingz;TestIntegers.test_repeatability_32bit_boundary_broadcasting2 sS ( A A A A A A A A A C B A A A A A A A A C B A A A A A A A A C B A A A A A A A A C B A A A A A A A A C D E E & O OD wu~~..F"sQC 0 = = =%) + +A
qT-=''71: N N N N O Or) c
` t j dddz fdddft j di}|D ]}|| D ]y}|\ }}||z
}t j |gdz g }t j |gdz }t t
t j |||| t t
t j |||| t t
t j |||| t t
t j |||| t j |gdz gt
} t j |gdz t
}
t t
t j | ||| t t
t j ||
|| t t
t j | |
|| {d S ) Nr rT A )rA r r r r ))r l )l rR rS rT ) rU r r rV ) r8 r4 r rz r rD rZ r1 object) r3 r( configsrW configlowhighlow_ahigh_alow_ohigh_os r' &test_int64_uint64_broadcast_exceptionsz3TestIntegers.test_int64_uint64_broadcast_exceptionsI s 91b5z; H8 45 > >E!%.
>
> " Th3%(,,4&2+.. j&/3'/u> > > > j&/5$'/u> > > > j&/3'/u> > > > j&/5&'/u> > > > 3%(6:::4&2+V<<< j&/5$'/u> > > > j&/3'/u> > > > j&/5&'/u> > > > >)
> > >r) c t j }t j t j j }t j t j t j j }t j t j t j j dz |z
}t
|||| }t || d S r )r8 r r r r4 rZ r1 r )r3 r( r r r r actuals r' test_int64_uint64_corner_casez*TestIntegers.test_int64_uint64_corner_casee s Xhrx $x**.//y"(++/!3h>?? thb IIVS!!!!!r) c | j D ]}|t u rdnt j | j }|t u rdnt j | j dz }|r|dz
n|}|t u r t j n|}| |||| }t |j | t t t j j fD ]}|t u rdnt j | j }|t u rdnt j | j dz }|r|dz
n|}| |||| }t |d rJ t t | | d S )Nr rT r< r rW )r r r8 r r r rC r r rW rU compatlonghasattrtyper3 r( r r r r s r' test_respect_dtype_singletonz)TestIntegers.test_respect_dtype_singleton| s` * + +Bd
11 (8Dd
11 (81(>D"
$
$
$ Ca 88> E E E Ra 88> E E E E F Fr) c t j dk rdnd}t j t 5 t
ddd| d d d d S # 1 swxY w Y d S )Nbigzi4r r r )r7 r8 r r rD rZ r1 )r3 other_byteord_dts r' test_error_byteorderz!TestIntegers.test_error_byteorder s $'MU$:$:55
]:
&
& E E OOAs3CO D D D E E E E E E E E E E E E E E E E E Es AA !A z sample_size,high,dtype,chi2maxi@KL ro g @_@r g b@i i g ȩ@i g d@c t ||| }t j |d \ }}||z }||z
dz |z } | |k sJ d S )Nr T)
return_countsrT )rZ r1 r8 uniquer )
r3 sample_sizerZ rW chi2maxsamplesvaluescountsr chi2s
r' $test_integers_small_dtype_chisquaredz1TestIntegers.test_integers_small_dtype_chisquared sn //$[/FF7$??? %("Q&16688g~~~~~~r) N)*rM rN rO rZ r1 r r r8 r2 r5 r0 r3 r1 rx r r4 r r r
r r r r r r$ r) rA rE r r r rz rL rO r_ rb ri rk rp ru slowr rP r) r' r r < s OE 27BHbh
Xry"(BI
7EP P P7 7 72 7 7 7 4 K K K>
@
@
@
@
@
@ ! ! ! . . .*!1 !1 !1F, , ,. [
XRX C C C D D
E
? ? ? @ @
A
XRX C C C D D
E F K K K O O O.> > >8" " ".+ + +* + + + F F FE E E [ [ =
1bgu %
1bh &
D"(F +
D")V ,
r) r c 6 e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
e j
d ej ej fej ej fg d
Zd Zd Zd
Zd Zd Zd Zd Zd Zd Z d Z d Z d Z d Z d Z!d Z"d Z#d Z$d Z%d Z&d Z'd Z(d Z)d! Z*d" Z+d# Z,e j
d$e-e.g e j
d%d& ej/ g d'g d(g fd) ej/ g d*g d+g fd, ej/ g d-g d.g fg d/ Z0d0 Z1d1 Z2e j
d2d3d4g d5 Z3d6 Z4d7 Z5d8 Z6d9 Z7d: Z8d; Z9d< Z:d= Z;d> Z d@ Z?dA Z@dB ZAdC ZBdD ZCdE ZDdF ZEdG ZFdH ZGdI ZHdJ ZIdK ZJdL ZKdM ZLdN ZMdO ZNdP ZOdQ ZPe j
dR ejQ dSdT dUejR dVg dW ZSdX ZTe j U eVdYZ e j
d[g d\ d] ZWe j
d^d)gd_ggfd`gd,ggfg da ZXe j
d[g d\ db ZYdc ZZdd Z[de Z\df Z]dg Z^dh Z_di Z`dj Zadk Zbdl Zcdm Zddn Zedo Zfdp Zgdq Zhdr Zids Zjdt Zkdu Zldv Zmdw Zndx Zody Zpdz Zqd{ Zrd| Zsd} Ztd~ Zud Zvd Zwd Zxd Zyd Zzd Z{d Z|d Z}d Z~e j
dddg d Ze j
ddej dej dg e j
dg d d Zd Zd Zd Zd Zd&S )TestRandomDistc d| _ d S )Nr r rE s r' r zTestRandomDist.setup_method s
r) c t t | j }| ddd }t j ddgddgd d
gg }t
|| d S ) Nc r rT r ) % ir r r r1 r8 rz r
r3 rZ ra rN s r'
test_integersz TestRandomDist.test_integers sh 749--.. bv 66(S#JR3*=>>67+++++r) c t t | j }| dddt j }t j ddgddgd d
ggt j }t || d S ) Nr r r r r F D r r rV )r r r r1 r8 rx rz r
r s r' test_integers_maskedz#TestRandomDist.test_integers_masked su 749--.. BV29 EE(QGb"X2w7ryIII67+++++r) c t t | j }| dddd }t j ddgdd gd
d gg }t
|| d S ) Nr r r T)r r( r r r & r ir r s r' test_integers_closedz#TestRandomDist.test_integers_closed sj 749--.. bv EE(S#J"b C:>??67+++++r) c t t j d j t j d j d }t j d j }t || d S )NlTr( )rZ r1 r8 r r r )r3 ra rN s r' test_integers_max_intz$TestRandomDist.test_integers_max_int sa # !2BHSMM4E*. ! 0 0 (3--#VW%%%%%r) c j t t | j }| d }t j ddgddgddgg }t
||d t t | j }| }t
||d
d d S ) Nr gG0Ը?gæ?g 2
?gu A?gH
>H?g;dE?r decimalrT r r r rZ r8 rz r r s r' test_randomzTestRandomDist.test_random s 749--.. v&&( .0A B .0A B .0A B D E E "&'2>>>>749--.. !&'$-DDDDDDr) c t t | j }| d }t j ddgddgddgg }t
||d d S )
Nr CԸ?gP?gtX?g81
A?grO>H?g88dE?r r r r s r' test_random_floatz TestRandomDist.test_random_float st 749--.. v&&(Y 4 ' 4 ' 4 6 7 7 "&'1======r) c t t | j }| t j }d}t
||d d S )NrV r r r )r r r rZ r8 r r r s r' test_random_float_scalarz'TestRandomDist.test_random_float_scalar sN 749--.. RZ 00!&'1======r) zdtype, uint_view_typec t t | j }| d| }t j | | dz }d|cxk rdk sn J d S )Nr rV r< i$^ i,e )r r r rZ r8
count_nonzeror> )r3 rW uint_view_typerZ r num_ones_in_lsbs r' test_random_distribution_of_lsbz.TestRandomDist.test_random_distribution_of_lsb s~ 749--.. vU 33 *6;;~+F+F+JKK ............r) c H t t t j d d S )Nr1 rV )r rC rZ rE s r' test_random_unsupported_typez+TestRandomDist.test_random_unsupported_type s i g>>>>>>r) c t t | j }| dd }t j g dt j }t || d S )Nr )r r rT rT rV r r r choicer8 rz r r
r s r' test_choice_uniform_replacez*TestRandomDist.test_choice_uniform_replace s\ 749--.. q!$$(<<>>V]]C:I]>>D j&-R88888r) c ~ t j g d }g d}t t t j || d S )N)* r< rT )NNNr )r8 rz r rD rZ r )r3 r r\ s r' test_choice_nan_probabilitiesz,TestRandomDist.test_choice_nan_probabilities s? HZZZ j&-a888888r) c
t j d dz }d|dd d< t t | j }| dd|d d d }t t | j }| ddt j |d d d }t || d S )Nr ro r r< rT r r )r8 onesr r r r r r
)r3 r\ rZ r r s r' test_choice_p_non_contiguousz+TestRandomDist.test_choice_p_non_contiguous s GBKK!O!$Q$749--..]]1a1SSqS6]22
749--.. q!r';AcccF'C'C DD:v.....r) c t j d dz }t dd }|j t j k sJ t ddd }|j t j k sJ t dd| }|j t j k sJ t dd|d }|j t j k sJ d S )Nr g @rT Fr r )r\ r )r8 r rZ r rW r )r3 r\ ra s r' test_choice_return_typez&TestRandomDist.test_choice_return_type s GAJJO q!$$|rx'''' q!U 33|rx'''' q!q ))|rx'''' q!q% 88|rx''''''r) c p d}t t | j }| ddd }t j dk r| }t j | t j }t ||k d S )N@4266599d12bfcfb815213303432341c06b4349f5455890446578877bb322e222' rv Fr r6 )r r r r r7 r8 r9 r: r; r> r8 r2 r< r )r3 choice_hashrZ ra r@ s r' test_choice_large_samplez'TestRandomDist.test_choice_large_sample s X 749--.. udE :: =H $ $__&&FnV[[ 1 122<<>> s"#####r) c t t | j }| d }d}t || d S )Nr s
8%)r r r bytesr r s r'
test_byteszTestRandomDist.test_bytes sE 749--.. b!!5VW%%%%%r) c d d d d d d d d d d
d f D ]d}t t | j } |g d }| | |} |g d
}t || ed S )Nc * t j g S r$ )r8 rz rr s r' z-TestRandomDist.test_shuffle.. s rx|| r) c | S r$ rP r s r' r z-TestRandomDist.test_shuffle.. s q r) c d t j | t j S r$ )r8 asarrayastyper2 r s r' r z-TestRandomDist.test_shuffle.. s rz!}} 3 3BG < < r) c d t j | t j S r$ )r8 r r r r s r' r z-TestRandomDist.test_shuffle.. s rz!}} 3 3BJ ? ? r) c d t j | t j S r$ )r8 r r complex64r s r' r z-TestRandomDist.test_shuffle.. s rz!}} 3 3BL A A r) c Z t j | t S r$ )r8 r r rV r s r' r z-TestRandomDist.test_shuffle.. s rz!}} 3 3F ; ; r) c d | D S )Nc g | ]}||fS rP rP .0is r'
zATestRandomDist.test_shuffle.... s 2 2 21A 2 2 2r) rP r s r' r z-TestRandomDist.test_shuffle.. s 2 2 2 2 2 r) c > t j d | D S )Nc g | ]}||gS rP rP r s r' r zATestRandomDist.test_shuffle.... *=*=*=aAq6*=*=*=r) )r8 r r s r' r z-TestRandomDist.test_shuffle.. s rz*=*=1*=*=*= > > r) c 8 t j | | g j S r$ )r8 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 rP rP r s r' r zATestRandomDist.test_shuffle.... s +>+>+>qQF+>+>+>r) r r )r8 r rU r> recarrayr s r' r z-TestRandomDist.test_shuffle.. sG
+>+>A+>+>+>-0#Jc
+C!E !E!%bk!2!2 r) c n t j d | D dt dfdt j dfg S )Nc g | ]}||fS rP rP r s r' r zATestRandomDist.test_shuffle.... r r) r r< r )r8 r rV r1 r s r' r z-TestRandomDist.test_shuffle.. sA rz*=*=1*=*=*=,/+>,/4+@+B C C r)
r< rT r r ro r r r r r
r r< r r r ro r r rT r )r r r r r
)r3 convrZ alistra rN s r' test_shufflezTestRandomDist.test_shuffle s ,+ [<<??AA;;22>>224 4C C D 0 0D" wty 1 122FD77788E NN5 ! ! !Fd999::G vw / / / /- 0 0r) c t t | j }t j d d }| |d t j g dg dg dg dg }t || t t | j }t j d d }| |d t || d S )
Nr" r r r< r r r r< rT r r ro r r r r r r r r rA ) r r r r8 r9 reshaper rz r
r s r' test_shuffle_custom_axisz'TestRandomDist.test_shuffle_custom_axis s 749--..2 &&v..vA&&&( , , , , , , , , , , , , . / / 67+++749--..2 &&v..vB'''67+++++r) c < t t | j }t j g d }dD ]P}t j g d }| || t || Qd S )N)r r )r r< r )r r r r8 rz r r r
)r3 rZ rN r ra s r' test_shuffle_custom_axis_emptyz-TestRandomDist.test_shuffle_custom_axis_empty s 749--..(2,,&&v.. 0 0DXb\\))&11F NN6N - - - vw / / / / 0 0r) c t j d dd }| }t t | j }| |d t t | j }| |j t || d S )Nr rT r r< r )
r8 r9 r copyr r r r r r
)r3 y1y2rZ s r' test_shuffle_axis_nonsquarez*TestRandomDist.test_shuffle_axis_nonsquare s
Yr]]
"
"1b
)
)
WWYY749--..r"""749--..rt 2r"""""r) c t j t j t d d dz dz
d }t j t j d dz dz
d }| }| }t d D ]}t | t t |j |j t |j |j t | t t |j |j t |j |j d S )Nr )ro r r r< rA r )
r8 ma
masked_valuesr r7 r9 r rZ r r sorteddatamask)r3 r r a_origb_origr s r' test_shuffle_maskedz"TestRandomDist.test_shuffle_masked sC E
599f = = AA Er J J E "
1A 5r : :r L LA NN1 qvqvg'' V[L0I)J)J
L
L
L NN1 qvqvg'' V[L0I)J)J
L
L
L
L L Lr) c
t t | j }t j d }t t j |j |d t j d d }t t j |j |d t t |j |t ddd g dg dg}t t |j |d t j d }t t |j | t j
d }t t j |j |d d S )
Nr r< r r r r rT r )r ro r r )r r r r8 r9 r AxisErrorr r rC sliceNotImplementedErrorrz r r3 rZ r s r' test_shuffle_exceptionsz&TestRandomDist.test_shuffle_exceptions s 749--..immblFNC;;;ill""6**blFNC;;;i eAq$6G6GHHHyy)))$)6>3BBBhqkki 555gfooblFNC;;;;;r) c t t | j }t j d }d|j _ t j t d 5 |
| d d d d S # 1 swxY w Y d S )Nro F read-onlyr ) r r r r8 rY flags writeabler r rD r )r3 rZ r s r' test_shuffle_not_writeablez)TestRandomDist.test_shuffle_not_writeable s 749--.. HQKK !
]:[
9
9
9 NN1 s B BBc t t | j }g d}| | }g d}t || t t | j }t j g d j }| | }t |t j | j d}t t
j |j | d}t t
j |j | t t | j }d}g d}| | }t || d S )Nr r abcdg333333?r )
r r r r r r rT ro r< r )
r r r permutationr
r8
atleast_2dr r r$ ) r3 rZ r ra rN arr_2d bad_x_str bad_x_float integer_vals r' test_permutationz TestRandomDist.test_permutation s: 749--.....##E**00067+++749--.. = = =>>@##F++62=#9#9#;<<< blF$6 BBB blF$6 DDD749--.. 000##K0067+++++r) c t j d d }t j g dg dg dg dg }t t | j }| |d }t || t t | j }| |d }t || d S )
Nr" r
r r r
r r< r rA ) r8 r9 r rz r r r r0 r
)r3 r rN rZ ra s r' test_permutation_custom_axisz+TestRandomDist.test_permutation_custom_axis! s IbMM ! !& ) )( , , , , , , , , , , , , . / / 749--..##AA#..67+++749--..##AB#//67+++++r) c
t t | j }t j d }t t j |j |d t j d d }t t j |j |d t t |j |t ddd d S )Nr r< r r# r rT ) r r r r8 r9 r r$ r0 r rC r% r' s r' test_permutation_exceptionsz*TestRandomDist.test_permutation_exceptions. s 749--..immblF$6Q???ill""6**blF$6Q???i!3S%1d:K:KLLLLLr) rW zaxis, expectedN)r r r r r r )r r rT ro r< r r )r r< rT r r r )r r r r r ro r< )ro r r r rT r< )r r r r r r c . t t | j }t j d dd | }| ||| t || t t | j }t j d dd | }| || }|j |k sJ t || d S )Nr rT r r outr )
r r r r8 r9 r r permutedr
rW )r3 rW r r rZ rr ys r'
test_permutedz TestRandomDist.test_permuted6 s 749--.. IbMM ! !!Q ' ' . .u 5 5!,,,1h'''749--.. IbMM ! !!Q ' ' . .u 5 5 OOADO ) )w% 1h'''''r) c t t | j }t j d dd }| }|d d d d df }| |d| }t j g dg dg }t || ||d d d d df< t || d S ) N rT r r r< r< )r r r r )r r r r )
r r r r8 r9 r r r> rz r
)r3 rZ x0x1rr r? r s r' test_permuted_with_stridesz)TestRandomDist.test_permuted_with_stridesJ s 749--..
Yr]]
"
"1b
)
)
WWYY qqq##A#vJ OOAA1O - -8\\\ - - - / 0 01h''' 111ccc6
2r"""""r) c Z t g }t |g d S r$ )rZ r> r
)r3 r? s r' test_permuted_emptyz"TestRandomDist.test_permuted_emptyW s) OOB 1b!!!!!r) outshaper ro c t j g d }t j ||j }t j t d 5 t || d d d d S # 1 swxY w Y d S )Nr rV z
same shaper r= ) r8 rz rY rW r r rD rZ r> )r3 rH r r= s r' "test_permuted_out_with_wrong_shapez1TestRandomDist.test_permuted_out_with_wrong_shape[ s HYYY hxqw///
]:\
:
:
: ( ( OOA3O ' ' ' ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( (s
A77A;>A;c t j dt j }t j d }t j t d 5 t |d| d d d d S # 1 swxY w Y d S )N)r ro rV z Cannot castr r< r< ) r8 rY r1 r r r rC rZ r> )r3 r= rr s r' !test_permuted_out_with_wrong_typez0TestRandomDist.test_permuted_out_with_wrong_typeb s hvRX... GFOO
]9M
:
:
: 0 0 OOAA3O / / / 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0s A;;A?A?c t j d }d|j _ t j t d 5 t |d| d d d d S # 1 swxY w Y d S )N)rT ro Fr* r r< r< ) r8 rY r+ r, r r rD rZ r> rq s r' test_permuted_not_writeablez*TestRandomDist.test_permuted_not_writeableh s HV !
]:[
9
9
9 . . OOAA1O - - - . . . . . . . . . . . . . . . . . .s A''A+.A+c t t | j }| ddd }t j ddgddgd d
gg }t
||d d S )
Nr r r r gW<$=g^
=g/:Ћ?gG!Gc>g'`2?gU?r r )r r r betar8 rz r r s r' test_betazTestRandomDist.test_betan s~ 749--..R&11(#%:
;#%:
;#%:
;
=> > "&'2>>>>>>r) c d t t | j }| ddd }t j ddgddgdd gg }t
|| t t | j }| dd }d}t
|| d S )
Ng Zd;Y@gv/?r r r r 0 , r )r r r r[ r8 rz r
r s r'
test_binomialz TestRandomDist.test_binomialw s 749--.. $V <<(RH H H & ' ' 67+++749--.. $//67+++++r) c t t | j }| dd }t j ddgddgdd gg }t
||d
d S ) Nr r r gmE~@@g
1ςC@grӳL@gm%xL@g>>>>>r) c B t t | j }t j ddg }| |d }t j ddgddggd d
gd d ggd
dgddggg }t
||d t j ddg }t t |j | t t | j }t j ddg }| | }t
||d d d S )N I>>>Hgx 011 j&"2I>>>749--...0DEFF!!%((!&'$-DDDDDDr) c J t j ddg }t t |t j d j d t t |t j d j d t t |t j d j d t t |ddg j d t t |d j d t t |t j d j d t t t j |t d d S )Nr[ r\ r< ru rT rv rw )
r8 rz r rZ r_ rx ry r rC r{ r| s r' test_dirichlet_sizez"TestRandomDist.test_dirichlet_size s3 H*,@A B BV%%a1 66>>>>>r) c t t d d t t t j d d S Nr scale )r rZ rr r rD rE s r' test_exponential_0z!TestRandomDist.test_exponential_0 sA V''a'00!444j&"4C@@@@@@r) c t t | j }| ddd }t j ddgddgd d
gg }t
||d d S )
Nr M r r g}
(Ҍ?g
i?g=r?g]hDA?gjӨ?g8k*u?r r )r r r fr8 rz r r s r' test_fzTestRandomDist.test_f s 749--.."bv..( .0A B .0@ B .0A B D E E "&'2>>>>>>r) c t t | j }| ddd }t j ddgddgd d
gg }t
||d d S )
Nro r r r g,שn'@gt @g
>>>>>r) c t t dd d t t t j dd d S )Nr )ry rw rx )r rZ r r rD rE s r' test_gamma_0zTestRandomDist.test_gamma_0 sA V\\\33Q777j&,cEEEEEEr) c t t | j }| dd }t j ddgddgddgg }t
|| d S )N_c97ݚ?r r r< r r r )r r r geometricr8 rz r
r s r' test_geometricz TestRandomDist.test_geometric ss 749--..!!*6!::(QG G H & ' ' 67+++++r) c " t t t j d t t t j dgdz t t t j d t t t j dgdz t j d 5 t t t j t j t t t j t j gdz d d d d S # 1 swxY w Y d S )Nr r r ignoreinvalid)r rD rZ r r8 errstater_ rE s r' test_geometric_exceptionsz(TestRandomDist.test_geometric_exceptions s! j&"2C888j&"2SEBJ???j&"2D999j&"2TFRK@@@
[
*
*
* G G *f&6 ? ? ? *f&62
F F F G G G G G G G G G G G G G G G G G Gs ADD Dc t t | j }| ddd }t j ddgddgd d
gg }t
||d d S )
Nr @r locrw r gA7G@g1Fiҿgei@g_Cg0cPg`(_տr r )r r r gumbelr8 rz r r s r' test_gumbelzTestRandomDist.test_gumbel
s 749--.. :Sv FF( /1C D /1C D /1C D F G G "&'2>>>>>>r) c t t d d t t t j d d S ru )r rZ r r rD rE s r'
test_gumbel_0z TestRandomDist.test_gumbel_0 = V]]]++Q///j&-s;;;;;;r) c t t | j }| dddd }t j ddgddgddgg }t
|| | dd d
d }t j g d }t
|| | d
d dd }t j g d }t
|| | d dd
d }t j g d }t
|| | d d
dd }t j g d }t
|| d S )Ng333333$@g @r r r r r ro r r r )r r r r r r )r r r r )r r r r )r r r hypergeometricr8 rz r
r s r' test_hypergeometricz"TestRandomDist.test_hypergeometric sz 749--..&&tS"6&BB(aG G G % & & 67+++ &&q!QQ&77(<<<((67+++&&r1bq&99(+++,,67+++ &&q!QQ&77(<<<((67+++&&q"bq&99(<<<((67+++++r) c t t | j }| ddd }t j ddgddgd d
gg }t
||d d S )
Nr r r r g6@ ga@bDA"?g*[r{ g>K ?g(?g<?r r )r r r laplacer8 rz r r s r' test_laplacezTestRandomDist.test_laplace0 s 749--.. Jc GG( /2C D /2C D /2C D F G G "&'2>>>>>>r) c t t d d t t t j d d S ru )r rZ r r rD rE s r' test_laplace_0z TestRandomDist.test_laplace_08 s= V^^!^,,a000j&.<<<<<?g>&gQh @g%?n ?gĉ8|?r r )r r r logisticr8 rz r r s r'
test_logisticz TestRandomDist.test_logistic< s 749--.. Zs HH( /2C D .2C D /2C D F G G "&'2>>>>>>r) c t t | j }| ddd }t j ddgddgd d
gg }t
||d d S )
Nr r r )r sigmar gܒx?gk*+@g<Ҿ?g
@g5 a@g%jz(@r r )r r r lognormalr8 rz r r s r' test_lognormalz TestRandomDist.test_lognormalD s 749--..!!z6!JJ(o/? @ - @ -/? @ B C C "&'2>>>>>>r) c t t d d t t t j d d S )Nr )r r< rx )r rZ r r rD rE s r' test_lognormal_0z TestRandomDist.test_lognormal_0L sA V%%A%..222j&"2#>>>>>>r) c t t | j }| dd }t j ddgddgdd gg }t
|| d S )
Ngƀ@?r )r\ r r r r ro r< )r r r logseriesr8 rz r
r s r' test_logseriesz TestRandomDist.test_logseriesP ss 749--..!!JV!<<(RH G F $ % % 67+++++r) c ~ t t | j }| d dk sJ d S )Nr r< )r r r r r3 rZ s r' test_logseries_zeroz"TestRandomDist.test_logseries_zeroX s> 749--.. ""a''''''r) valuerk rA rj g @c t t | j }t j d 5 t j t 5 | | d d d n # 1 swxY w Y t j t 5 | t j |gdz d d d n # 1 swxY w Y t j t 5 | t j |gdz d d d d d d n # 1 swxY w Y d d d d S # 1 swxY w Y d S )Nr r r rT )
r r r r8 r r r rD r rz )r3 r rZ s r' test_logseries_exceptionsz(TestRandomDist.test_logseries_exceptions\ s( 749--..
[
*
*
* > > z**
(
( '''
(
(
(
(
(
(
(
(
(
(
(
(
(
(
( z**
9
9 5'B,!7!7888
9
9
9
9
9
9
9
9
9
9
9
9
9
9
9 z**
>
> 5'B,!7!7!!<===
>
>
>
>
>
>
>
>
>
>
>
>
>
>
>
> > > > > > > > > > > > > > > > > >s} E A3' E 3A7 7E :A7 ; E ,C E C E C E 35D4( E 4D8 8E ;D8 <E EEc t t | j }| ddgdz d }t j g dg dgg dg d gg d
g d gg }t
|| d S ) Nr UUUUUU?r r r )r< ro r< r r r )r rT r rT r rT )ro r rT r r r< )r r r rT r r )r r r< ro r rT )ro ro r r< rT r )r r r rh r8 rz r
r s r' test_multinomialz TestRandomDist.test_multinomiali s 749--..##B1 6#BB( / / / / / / 1 / / / / / / 1 / / / / / / 1 2 3 3 67+++++r) z fp errors don't work in wasmreasonr )svdeighcholeskyc t t | j }d}ddgddgg}d}| |||| }t j ddgdd ggd
d gd d
ggddgddggg }t
||d | ||| }t j ddg }t
||d ddg}ddgddgg}t t |j ||d ddgddgg}t t |j || t t |j ||d t t |j ||d t |j ||d t t |j ||d t t |j ||dd ddgddgg}|d v r8| ||d| }t
|d |d d! n t t |j ||d t j dd"gd"dggt j
# }t 5 } | ||| | t }
t! |
dk sJ d d d n # 1 swxY w Y t j d } t j d }t t |j ||d$ t t |j t j d% | t t |j | t j d t t |j | t j d& d S )'N)r r r< r r r gtg $&@g&بO=g!$@gDo ?g\&@g,R{?g
f!@g?ժ?gT^o#@g`R?gP'@r r gl?g$("@rT raise) check_validr r r )r r )r r r ).r ).r< r r rV other)rT r< r< r )r r r multivariate_normalr8 rz r r rD r RuntimeWarningr r r r recordrJ rY eyer ) r3 r rZ r covr ra rN r| supwmus r' test_multivariate_normalz'TestRandomDist.test_multivariate_normalt sW 749--.. 1v1v ++D#tF+KK( 03D G 13E G I 13E G 14E G I 14E G 13E G I J K K "&'2>>>> ++D#f+EE( -/@ABB!&'2>>>> 1v1v1v j&" 1 @ @G %gfowv./
1
1
1
1
1
+v'A4!+
-
-
-
- hC3(+2:>>>
C & &tS & @ @ @
>**Aq66Q;;;;;
Xa[[fQiij&"RV L L L *f&> 6(R-
)
)
) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) )s AA>>BBc t t 5 t dd }d d d d S # 1 swxY w Y d S Nr< r )r rD rZ r rq s r' #test_negative_binomial_p0_exceptionz2TestRandomDist.test_negative_binomial_p0_exception s
:
&
& / /((A..A / / / / / / / / / / / / / / / / / /s >AAc t j d 5 t t t j dd t t t j dgdg d d d d S # 1 swxY w Y d S )Nr r rR r )r8 r r rD rZ r rE s r' .test_negative_binomial_invalid_p_n_combinationz=TestRandomDist.test_negative_binomial_invalid_p_n_combination s [
*
*
* P P *f&>s K K K *f&># O O O P P P P P P P P P P P P P P P P P Ps AA((A,/A,c 4 t t | j }| ddd }t j ddgddgdd gg }t
||d
| d d
d }t j ddgddgddgg }t
||d
t t | j }| ddd }t j ddgddgddgg }t
||d
d S )Nro r )dfnoncr gs3J?gugو/@gNJm+@gA}-4@gq= &@gjhn
@r r rX re gZօDN?gE!X&?gt%T?gQ)V?g5Zlf?gΎ{?r gmK?gZ?g%yKY@g@W @gĈ{>@g&BB=@)r r r noncentral_chisquarer8 rz r r s r' test_noncentral_chisquarez(TestRandomDist.test_noncentral_chisquare se 749--..,,,GG( .0A B .0A B -1A B D E E "&'2>>>>,,&,II( 13G H 13G H 13G H J K K "&'2>>>>749--..,,,GG( -/? @ -/? @ -/? @ B C C "&'2>>>>>>r) c t t | j }| dddd }t j ddgdd gd
d gg }t
||d
d S )Nro rT r< r )dfnumdfdenr r g HP
?g n?g%f?gch?gzƎ?g]?:?r r )r r r noncentral_fr8 rz r r s r' test_noncentral_fz TestRandomDist.test_noncentral_f s 749--..$$1AA*0 % 2 2(^/? @ - @ -/? @ B C C "&'2>>>>>>r) c t t | j }| ddt j }t j | sJ d S )Nro rT )r r r )r r r r r8 r_ isnanr3 rZ ra s r' test_noncentral_f_nanz$TestRandomDist.test_noncentral_f_nan
sQ 749--..$$1ABF$CCx r) c t t | j }| ddd }t j ddgddgd d
gg }t
||d d S )
Nr r r r g?`x g\P@g/7u g0aj?gi%?gog@r r )r r r normalr8 rz r r s r' test_normalzTestRandomDist.test_normal s 749--.. :Sv FF( /2C D /2C D /2C D F G G "&'2>>>>>>r) c t t d d t t t j d d S ru )r rZ r r rD rE s r'
test_normal_0z TestRandomDist.test_normal_0 r r) c t t | j }| dd }t j ddgddgdd gg }t j ||d
d S ) Nr r r r g á?gƫh@gi`>?g0"&Agc@g=fAr )nulp)r r r paretor8 rz testing assert_array_almost_equal_nulpr s r' test_paretozTestRandomDist.test_pareto s 749--.. & 99( 35K L 35K L 35K L N O O
11&'1KKKKKr) c t t | j }| dd }t j ddgddgddgg }t
|| d S )Nr r )lamr r )r r r poissonr8 rz r
r s r' test_poissonzTestRandomDist.test_poisson) sp 749--.. JV <<(QF F F $ % % 67+++++r) c X t j d j }d}t t t
j | t t t
j |gdz t t t
j | t t t
j |gdz t j d 5 t t t
j t j t t t
j t j gdz d d d d S # 1 swxY w Y d S )Nr rA r r r ) r8 r r r rD rZ r r r_ )r3 lambiglamnegs r' test_poisson_exceptionsz&TestRandomDist.test_poisson_exceptions1 s2 '""&j&.&999j&.6(R-@@@j&.&999j&.6(R-@@@
[
*
*
* E E *fnbf = = = *fnrvhm D D D E E E E E E E E E E E E E E E E E Es 9AD D#&D#c t t | j }| dd }t j ddgddgdd gg }t
||d
d S ) Nr r r gZ >g?gÄ(=g҇?gŚE?gߢH?r r )r r r powerr8 rz r r s r'
test_powerzTestRandomDist.test_power< s 749--..
88( 24I J 24I J 24I J L M M "&'2>>>>>>r) c t t | j }| dd }t j ddgddgdd gg }t
||d
d S ) Nr r )rw r gry@gtQP0@gAI_
@g@u81@g G0@g" wc5@r r )r r r rayleighr8 rz r r s r'
test_rayleighz TestRandomDist.test_rayleighD s 749--.. r 77( -/@ A -/@ A .0A B D E E "&'2>>>>>>r) c t t d d t t t j d d S ru )r rZ r r rD rE s r' test_rayleigh_0z TestRandomDist.test_rayleigh_0L s= V__1_--q111j&/======r) c t t | j }| d }t j ddgddgddgg }t
||d
d S ) Nr r gH1gW|3
g\?gLeg5P gFA?r r )r r r standard_cauchyr8 rz r r s r' test_standard_cauchyz#TestRandomDist.test_standard_cauchyP s 749--..''V'44( /1C D /1C D /2C D F G G "&'2>>>>>>r) c t t | j }| dd }t j ddgddgdd gg }t
||d
d S ) Nr invr ga\5 ?g r7k?g8PC*?g5$[?gW3T?g* ?r r )r r r standard_exponentialr8 rz r r s r' test_standard_exponentialz(TestRandomDist.test_standard_exponentialX s 749--..,,&,GG( .0A B .0A B .0A B D E E "&'2>>>>>>r) c \ t t t j t j d S NrV )r rC rZ r r8 r1 rE s r' #test_standard_expoential_type_errorz2TestRandomDist.test_standard_expoential_type_error` s" i!>>>>>r) c t t | j }| dt j }d}t
||d d S )Nr rV g d@r r )r r r r r8 r r r s r' !test_standard_gammma_scalar_floatz0TestRandomDist.test_standard_gammma_scalar_floatk sS 749--..&&q
&;;$!&'1======r) c t t | j }| dd }t j ddgddgdd gg }t
||d
d S ) Nr r r gf&?gM
O?g1@g):@g
@g;i
@ro r r
r s r' test_standard_gamma_floatz(TestRandomDist.test_standard_gamma_floatq s{ 749--..&&QV&<<(Wf . $f . $g . 0 1 1 "&'1======r) c t j dt j }t t | j }| d|t j t j ddgddgd d
ggt j }t ||d t t | j }| d|dt j
t ||d d S )Nr rV $@r= rW gM»L$@ga { @gN]"@gPS")@g_+@gmB @ro r r= r rW ) r8 rY r r r r r rz r )r3 ra rZ rN s r' test_standard_gammma_float_outz-TestRandomDist.test_standard_gammma_float_outy s &
333749--.. dbj AAA(X 0 & 1 & 1 3:<*F F F "&'1====749--.. dV2: NNN!&'1======r) c J t t t j dd d S )Nrj r1 rV )r rC rZ r rE s r' test_standard_gamma_unknown_typez/TestRandomDist.test_standard_gamma_unknown_type s/ i!6 # % % % % % %r) c t j d }t t t j dd| t t t j dd| d S )Nr r r )r r= )r r< )r8 rY r rD rZ r )r3 r= s r' test_out_size_mismatchz%TestRandomDist.test_out_size_mismatch sc hrllj&"7B j&"7G r) c t t d d t t t j d d S )Nr )ry rx )r rZ r r rD rE s r' test_standard_gamma_0z$TestRandomDist.test_standard_gamma_0 sA V***33Q777j&"7sCCCCCCr) c t t | j }| d }t j ddgddgddgg }t
||d
d S ) Nr r g
bYg? ?gGH{Lgz\?gc`+?gT}?r r )r r r r r8 rz r r s r' test_standard_normalz#TestRandomDist.test_standard_normal s 749--..''V'44( /2B D /2C D /2C D F G G "&'2>>>>>>r) c \ t t t j t j d S r )r rC rZ r r8 r1 rE s r' %test_standard_normal_unsupported_typez4TestRandomDist.test_standard_normal_unsupported_type s" i!7rxHHHHHHr) c t t | j }| dd }t j ddgddgdd gg }t
||d
d S ) Nr r )r r gV1g |ڙ(?g-?g NhڿgMc?g!uLM r r )r r r
standard_tr8 rz r r s r' test_standard_tz TestRandomDist.test_standard_t s 749--..""bv"66( /2B D /1C D /1C D F G G "&'2>>>>>>r) c t t | j }| dddd }t j ddgdd gd
d gg }t
||d
d S )Ng{Gz@g(\u$@gףp=
W4@r )leftmoderightr g Epw @g4DC+@gŋ @g /,@gkuR*@gHNPcr+@r r )r r r
triangularr8 rz r r s r' test_triangularz TestRandomDist.test_triangular s 749--..""5(. # 0 0( .0@ B .0A B .0A B D E E "&'2>>>>>>r) c t t | j }| ddd }t j ddgddgd d
gg }t
||d d S )
NGz?gGz%@r )rY rZ r gm2@gJaD @g : @g
Q @g@T{e @gb @r r r r r uniformr8 rz r r s r' test_uniformzTestRandomDist.test_uniform s 749--.. Du6 BB( -0A B .0A B .0A B D E E "&'2>>>>>>r) c ( t j d j }t j d j }t j }t
t |t j d t
t |dt j t
t ||| t
t |t j gdg t
t |dgt j g t t j |d |dz d S )Nr{ r r< g W4vC)rY rZ )
r8 finfor r rZ r/ r
OverflowErrorinf nextafter)r3 fminfmaxfuncs r' test_uniform_range_boundsz(TestRandomDist.test_uniform_range_bounds s x $x $~mTBF7A666mT1bf555mT4666mTRVG9qc:::mTA3999
2<a00td{CCCCCr) c D t j } |dd }t |d |dt j gdt j g }t |dt j g |dgdggddg }t |dt j d z d S )N ?rk g
ףp= @rw )rZ r/ r r8 pirY )r3 r8 results r' test_uniform_zero_rangez&TestRandomDist.test_uniform_zero_range s ~c3 $$$sBElS"%L11be ---y7),w.@AA"(6*:*: :;;;;;r) c t j }t t |dd t t |ddgddg t t |ddgddggd d S )NrT r< r r )rZ r/ r rD )r3 r8 s r' test_uniform_neg_rangez%TestRandomDist.test_uniform_neg_range s_ ~j$1---j$!Q!Q888j$1a&!Q !<<<<.ThrowingFloatc t r$ rC rE s r' __float__zQTestRandomDist.test_scalar_exception_propagation..ThrowingFloat.__float__ r) N)rM rN rO rF rP r) r'
ThrowingFloatrC #
r) rH rj c e Zd Zd ZdS )ITestRandomDist.test_scalar_exception_propagation..ThrowingIntegerc t r$ rE rE s r' __int__zQTestRandomDist.test_scalar_exception_propagation..ThrowingInteger.__int__ rG r) N)rM rN rO rM rP r) r' ThrowingIntegerrK rI r) rN r< ) r8 ndarrayrz r> r rC rZ r/ r )r3 rH throwing_floatrN throwing_ints r' !test_scalar_exception_propagationz0TestRandomDist.test_scalar_exception_propagation s BJ # ++M::i $ & & & bj x{{''88 i!6 aKKKKKr) c t t | j }| ddd }t j ddgddgd d
gg }t
||d d S )
Nr- p=
ף?r r kappar gLA?gxyw@gzVR?gpE ?g8&Aпgv @r r )r r r vonmisesr8 rz r r s r'
test_vonmisesz TestRandomDist.test_vonmises s 749--.. D6 BB( /2C D /2C D /2C D F G G "&'2>>>>>>r) c t t | j }| ddd }t t j | d S )Nrk g g5_PG>i@B rU )r r r rW r r8 isfiniterp r3 rZ rs r' test_vonmises_smallz"TestRandomDist.test_vonmises_small sX 749--.. OOreO < < A""$$%%%%%r) c t t | j }| dt j }t
t j | d S )Nrk )r rV )r r r rW r8 r_ r r r[ s r' test_vonmises_nanz TestRandomDist.test_vonmises_nan sL 749--.. OOrO 0 0 r) rV g @ 4&k Cc d t t | j }t |j }|j j }| d|d }||j _ | d|d }|dk rt || d S t j ||k sJ d S )Nr r r g .A)
r r r r r r rW r r8 rp )r3 rV rZ rsr random_state_valsgen_valss r' test_vonmises_large_kappaz(TestRandomDist.test_vonmises_large_kappa s 749--..
-
.
.$* KK5rK : :%* "??1e"?55 3;; -x 8 8 8 8 86+x788 8 88 8 8r) r g gg @)r gư>r< g @@r` c t t | j }| ||d }t t j |t
j k o!t j |t
j k d S )Nr )r r r rW r r8 rp r< )r3 r rV rZ r\ s r' test_vonmises_large_kappa_rangez.TestRandomDist.test_vonmises_large_kappa_range sk 749--.. OOBr * *qBE6z""9rva25j'9'9:::::r) c t t | j }| ddd }t j ddgddgd d
gg }t
||d d S )
Nr- rT r )r rw r g
*س2?gt @g ;S@g\ F@gxF @g6b?r r )r r r waldr8 rz r r s r' test_waldzTestRandomDist.test_wald s 749--..$d@@( - @ -/? @ -/? @ B C C "&'2>>>>>>r) c t t | j }| dd }t j ddgddgdd gg }t
||d
d S ) Nr- r r gnJZ?g*/F?gs(RY?g]'%?g,qD ?g8 ?r r )r r r weibullr8 rz r r s r' test_weibullzTestRandomDist.test_weibull s 749--.. $V 44( .0A B .0A B .0A B D E E "&'2>>>>>>r) c t t | j }t | dd t j d t t |j d d S )Nr r r rx )r ) r r r r rl r8 rY r rD r s r' test_weibull_0z TestRandomDist.test_weibull_0& s^ 749--..V^^ab^1128B<<@@@j&.C888888r) c t t | j }| dd }t j ddgddgddgg }t
|| d S ) Nr- r r r< r ic ib rT )r r r zipfr8 rz r
r s r' test_zipfzTestRandomDist.test_zipf+ sp 749--..t&11(qAJ #J AJ ( ) ) 67+++++r) )rM rN rO r r r r r r r r r r r r8 r rx float64r4 r r r r r r r r r r r r r r r r r r r r r! r( r- r6 r8 r: rU rV rz r@ rE rG rK rM rO rR rV rY rb rd rf rh rl r rp rs ry r} r r r r r r r r r r r r r r r5 r_ r r skipifr
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+ r0 r9 r> r@ rR rX r] r_ re r< rg rj rm ro rr rP r) r' r r s , , , , , ,, , ,
&
&
&
E
E
E> > >> > > [ 4 !z29 5 !z29 5 78 8/ /8 8/? ? ?, , , , , , , , ,, , , , , , , , , , , , 2 2 2$,9 ,9 ,9\9 9 9
/ / /
(
(
($ $ $& & & 0 0 06 , , , 0 0 0# # # L L L
<
<
< , , ,2 , , ,M M M [ WsFm44 [ - #XRX/C/C/C/C/C/C/E &F &F G ("(,@,@,@,@,@,@,B #C #C D ("(,@,@,@,@,@,@,B #C #C D EF F
(
(F F 54
( # # #" " " [ Z&!55( ( 65( 0 0 0 . . . ? ? ? , , ,? ? ?E E E&
@
@
@ P P P6 6 6
@
@
@ [< < <? ? ?A A A? ? ?? ? ?F F F, , ,G G G? ? ?< < <, , ,4? ? ?= = =? ? ?? ? ?? ? ?, , ,( ( ( [ W|r|B';';R&LMM
>
> NM
> , , , [(FGG [ X'B'B'BCCG% G% DC HGG%R [ [QC4&?bTQC5M*JKK2 2 LK2
[ X'B'B'BCC
3
3 DC
3, , ,) ) ) / / /
P P P? ? ?*? ? ?
? ? ?< < < L L L , , , E E E? ? ?? ? ?> > >? ? ?? ? ?N N N? ? ?> > > > > > > > >% % % D D D? ? ?I I I? ? ?? ? ?? ? ?D D D < < <= = = L L L,? ? ?& & &
[ WsDk22 9 9 32 9 [ TC"%ruc#BCC [ W&@&@&@AA; ; BA DC;
? ? ?? ? ?9 9 9
, , , , ,r) r c p 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"e#j$ % d!d" e&j' d"d"g e&j' d"ggd"ggg g d# Z(d$ Z)d% Z*d&S )'
TestBroadcastc d| _ d S )Ni[r rE s r' r zTestBroadcast.setup_method7 s
r) c t t | j }dg}dg}|j }t j g d }t t | j }| |dz | }t
||d t t | j }| ||dz }t
||d d S )Nr r< )g4
7^?gF]
"?g .?r r r r. )r3 rZ rY rZ r/ rN ra s r' r0 zTestBroadcast.test_uniform; s 749--..cs.(QQQRR749--.. a..!&'2>>>>749--.. TAX..!&'2>>>>>>r) c " dg}dg}dg}t t | j }t j g d }t t | j }| |dz | }t
||d t t |j |dz | t t | j }|j } |||dz }t
||d t t |||dz d S )Nr r< rA )gw-Dؿgn0_x?gʊ;<.?r r r ) r r r r8 rz r r r rD )r3 r rw bad_scalerZ rN ra r s r' r zTestBroadcast.test_normalJ s cD 749--..(SSSTT749--.. sQw..!&'2>>>>j&-q)DDD749--.. UQY''!&'2>>>>j&#y1}=====r) c dg}dg}dg}dg}t j g d }t t | j }|j } ||dz | }t
||d t t ||dz | t t ||dz | t t | j }| ||dz }t
||d d S ) Nr< rT rA rH )gf8?g%eo?g$q??r r r ) r8 rz r r r rQ r r rD ) r3 r r bad_abad_brN rZ rQ ra s r' rR zTestBroadcast.test_beta\ s
C
C(QQQRR749--..{a!eQ !&'2>>>>j$ 1555j$Au555749--..QA&&!&'2>>>>>>r) c dg}dg}t j g d }t t | j }| |dz }t
||d t t |j |dz d S Nr< rA )g` ʄ?gE]?g
?r r r ) r8 rz r r r rr r r rD r3 rw rz rN rZ ra s r' rs z TestBroadcast.test_exponentialn s D (PPPQQ749--..##EAI..!&'2>>>>j&"4i!mDDDDDr) c dg}dg}t j g d }t t | j }|j } ||dz }t
||d t t ||dz d S r ) r8 rz r r r r r r rD )r3 ry bad_shaperN rZ std_gammara s r' r z!TestBroadcast.test_standard_gammax s D (PPPQQ749--..) 519%%!&'2>>>>j)Y];;;;;r) c > dg}dg}dg}dg}t j g d }t t | j }|j } ||dz | }t
||d t t ||dz | t t ||dz | t t | j }|j } |||dz }t
||d t t |||dz t t |||dz d S ) Nr< rT rA rH )g3 ʄ?gJE]?g
?r r r ) r8 rz r r r r r r rD ) r3 ry rw r rz rN rZ r ra s r' r zTestBroadcast.test_gamma s* D D (PPPQQ749--.. uqy%((!&'2>>>>j%Q >>>j%I>>>749--.. ueai((!&'2>>>>j%EAI>>>j% A
>>>>>r) c > dg}dg}dg}dg}t j g d }t t | j }|j } ||dz | }t
||d t t ||dz | t t ||dz | t t | j }|j } |||dz }t
||d t t |||dz t t |||dz d S ) Nr< rT rA rH )gM3B?gybi @g0?r r r ) r8 rz r r r r| r r rD ) r3 r r bad_dfnum bad_dfdenrN rZ r| ra s r' r} zTestBroadcast.test_f s* D D (QQQRR749--.. H519e$$!&'2>>>>j!Y]E:::j!UQY :::749--.. H5%!)$$!&'2>>>>j!Y :::j!UIM:::::r) c
z dg}dg}dg}dg}dg}dg}t j g d }t t | j }|j } | |dz || }
t
|
|d t j t j | ||t j gdz sJ t t | |dz || t t | |dz || t t | |dz || t t | j }|j } | ||dz | }
t
|
|d t t | ||dz | t t | ||dz | t t | ||dz | t t | j }|j } | |||dz }
t
|
|d t t | |||dz t t | |||dz t t | |||dz d S )
NrT r r r rA rH )gŜ,n1 @g!6)@g9?r r ) r8 rz r r r r r rp r r_ r rD ) r3 r r r r r bad_noncrN rZ nonc_fra s r' r z TestBroadcast.test_noncentral_f sI sC D 4(RRRSS749--..$ 5$//!&'2>>>>vbhvveURVHqL A ABBCCCCCj&)a-EEEj&%!)YEEEj&%!)UHEEE749--..$uqy$//!&'2>>>>j&)UQYEEEj&%Q EEEj&%HEEE749--..$udQh//!&'2>>>>j&)UD1HEEEj&%D1HEEEj&%1 EEEEEr) c t t | j }t j ddg }| dddd }t
||d d S )NgZ\#?gDqw?r rT r r r )r r r r8 rz r r )r3 rZ rN ra s r' test_noncentral_f_small_dfz(TestBroadcast.test_noncentral_f_small_df se 749--..( ,o>??$$S#qq$99!&'2>>>>>>r) c dg}dg}t j g d }t t | j }| |dz }t
||d t t |j |dz d S )Nr< rA )gC{?g7Q#?g%h1T@r r r ) r8 rz r r r rX r r rD r3 r bad_dfrN rZ ra s r' rY z TestBroadcast.test_chisquare s S(PPPQQ749--..!!"q&))!&'2>>>>j&"2FQJ?????r) c > dg}dg}dg}dg}t j g d }t t | j }|j } ||dz | }t
||d t t ||dz | t t ||dz | t t | j }|j } |||dz }t
||d t t |||dz t t |||dz d S ) Nr< rT rA rH )g-S?g[ @gr3/.?r r r ) r8 rz r r r r r r rD ) r3 r r r r rN rZ nonc_chira s r' r z'TestBroadcast.test_noncentral_chisquare s, Ss4(OOOPP749--..."q&$''!&'2>>>>j(FQJ===j(BFH===749--..."dQh''!&'2>>>>j(FD1H===j(B1 =====r) c dg}dg}t j g d }t t | j }| |dz }t
||d t t |j |dz d S )Nr< rA )g
>Qg{g,e?r r r ) r8 rz r r r r$ r r rD r s r' r% z TestBroadcast.test_standard_t s S(SSSTT749--..""26**!&'2>>>>j&"3VaZ@@@@@r) c dg}dg}dg}t j g d }t t | j }| |dz | }t
||d t t |j |dz | t t | j }| ||dz }t
||d t t |j ||dz d S )NrT r< rA )gD()@g ,)ʸ@gPor r r ) r8 rz r r r rW r r rD )r3 r rV bad_kapparN rZ ra s r' rX zTestBroadcast.test_vonmises s SD (RRRSS749--.. a//!&'2>>>>j&/269EEE749--.. UQY//!&'2>>>>j&/2y1}EEEEEr) c dg}dg}t j g d }t t | j }| |dz }t
||d t t |j |dz d S )Nr< rA )g/h?gcE?g/gV?r r r ) r8 rz r r r r r r rD r3 r r| rN rZ ra s r' r zTestBroadcast.test_pareto s
C(QQQRR749--.. q1u%%!&'2>>>>j&-;;;;;r) c dg}dg}t j g d }t t | j }| |dz }t
||d t t |j |dz d S r ) r8 rz r r r rl r r rD r s r' rm zTestBroadcast.test_weibull s
C(PPPQQ749--.. A&&!&'2>>>>j&.%!)<<<<>>>j&, :::::r) c dg}dg}dg}t j g d }t t | j }|j } ||dz | }t
||d t t ||dz | t t | j }|j } |||dz }t
||d t t |||dz d S )Nr r< rA )gh&Bg sgR#?r r r ) r8 rz r r r r r r rD )r3 r rw rz rN rZ r ra s r' r zTestBroadcast.test_laplace- s cD (SSSTT749--...q%((!&'2>>>>j'37I>>>749--...eai((!&'2>>>>j'3 A
>>>>>r) c dg}dg}dg}t j g d }t t | j }|j } ||dz | }t
||d t t ||dz | t t | j }|j } |||dz }t
||d t t |||dz d S )Nr r< rA )gs4?g$Cw%T?g տr r r ) r8 rz r r r r r r rD )r3 r rw rz rN rZ r ra s r' r zTestBroadcast.test_gumbel? s cD (RRRSS749--.. a''!&'2>>>>j&#'9===749--.. UQY''!&'2>>>>j&#y1}=====r) c 8 dg}dg}dg}t j g d }t t | j }| |dz | }t
||d t t |j |dz | t t | j }| ||dz }t
||d t t |j ||dz t | dd d d S )
Nr r< rA )g$DgXPg,?r r r rj rk )
r8 rz r r r r r r rD r )r3 r rw rz rN rZ ra s r' r zTestBroadcast.test_logisticQ s cD (QQQRR749--.. q%00!&'2>>>>j&/37IFFF749--.. eai00!&'2>>>>j&/3 A
FFFV__S#..44444r) c dg}dg}dg}t j g d }t t | j }|j } ||dz | }t
||d t t ||dz | t t | j }| ||dz }t t |j ||dz d S )Nr r< rA )g c?gu@gVQ?r r r ) r8 rz r r r r r r rD )r3 r r bad_sigmarN rZ r ra s r' r z TestBroadcast.test_lognormalb s sD (QQQRR749--..$ 4!8U++!&'2>>>>j)TAXyAAA749--..!!$ 22j&"2D)a-HHHHHr) c dg}dg}t j g d }t t | j }| |dz }t
||d t t |j |dz d S )Nr< rA )gs(?g4
mK?gVk +?r r r ) r8 rz r r r r r r rD r s r' r zTestBroadcast.test_rayleighr s D (
!
!
!
749--.. ++!&'2>>>>j&/9q=AAAAAr) c r dg}dg}dg}dg}t j g d }t t | j }| |dz | }t
||d t t |j |dz | t t |j |dz | t t | j }| ||dz }t
||d t t |j ||dz t t |j ||dz d S ) NrX r< r rH )gk7Z?g*}z9?gc- ?r r r ) r8 rz r r r ri r r rD )r3 r rw bad_meanrz rN rZ ra s r' rj zTestBroadcast.test_wald s( u3D (OOOPP749--..TAXu--!&'2>>>>j&+x!|UCCCj&+taxCCC749--..T519--!&'2>>>>j&+xCCCj&+tY]CCCCCr) c dg}dg}dg}dg}dg}|dz \ }}t j g d }t t | j } | j }
|
|dz || } t
| |d t t |
|dz || t t |
|dz || t t |
|dz || t t | j } | j }
|
||dz | } t
| |d t t |
||dz | t t |
||dz | t t |
||dz | t t | j } | j }
|
|||dz } t
| |d t t |
|||dz t t |
|||dz t t |
|||dz t t |
dd d
t t |
dd d
t t |
ddd d S ) Nr< r rT r )gR8T>?gK ?g;>h@r r r rk g 4@g 9@) r8 rz r r r r* r r rD ) r3 r' r) r( bad_left_one bad_mode_one bad_left_two bad_mode_tworN rZ r* ra s r' r+ z TestBroadcast.test_triangular s{ sss s %*QY" l(QQQRR749--..&
D1HdE22!&'2>>>>j*lQ.>eLLLj*dQh eLLLj*lQ.> 749--..&
D$(E22!&'2>>>>j*lD1HeLLLj*dL14DeLLLj*lL1>>>j*lD%!)LLLj*dL%!)LLLj*lLai ! ! ! j*c2s;;;j*c3<<<j*c3<<<<>>j)QUI>>>749--.., 1a!e$$67+++j)UAE:::j)Q A
>>>j)Q A
>>>>>r) c H dg}dg}t j g d }t t | j }|j }|dz g}|j } ||dz }t || t t ||dz t t ||dz d S )Nr< rA )r r r rT r )
r8 rz r r r _poisson_lam_maxr r
r rD ) r3 r bad_lam_onerN rZ max_lam bad_lam_twor ra s r' r zTestBroadcast.test_poisson s cd (999%%749--..) {m .q!!67+++j';?;;;j';?;;;;;r) c dg}dg}t j g d }t t | j }|j } ||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 )NrT r )r< r r< r r r ) r8 rz r r r rq r
r rD r r_ )r3 r r| rN rZ rq ra s r' rr zTestBroadcast.test_zipf s
C(999%%749--..{a!e67+++j$ 222
[
*
*
* < < *dBF 3 3 3 *dQ26N ; ; ; < < < < < < < < < < < < < < < < < F;c t t | j }dgdz dgdz g}t t |j d| t t |j dd d S )Nr r r r< rX )r r r r rD rh )r3 rZ r s r' test_invalid_pvals_broadcastz*TestBroadcast.test_invalid_pvals_broadcast sc 749--..1ugk*j&"4a???j&"4a=====r) c t t | j }| t j dd dgdz }|j dk sJ | dt j d }|j dk sJ | t j dd t j d }|j d k sJ d S )
N)r r r i8r r )r r r r r )r r r )r r r )r r r r )r r r rh r8 r ry r s r' test_empty_outputsz TestBroadcast.test_empty_outputs s 749--..##BHZ$>$>! LL|},,,,##B(=(=>>|{****##BHY$=$=$&H\$:$:< <||++++++r) N)+rM rN rO r r0 r rR rs r r r} r r rY r r% rX r rm r r r r r r rj r+ rV r r rr r r r r r r r r8 rz r r r rP r) r' rv rv 4 s
?
?
? > > >$? ? ?$E E E < < <? ? ?*; ; ;*!F !F !FF? ? ? @ @ @> > >*A A AF F F < < <= = =; ; ;? ? ?$> > >$5 5 5"I I I B B B D D D&&= &= &=P; ; ;.? ? ?0
<
<
< < < < < < <+H +H +HZ < < < , , ,> [ S2#+28RH#5#5#+28rdVrdV,<#=#=#%
A A
A.> > > , , , , ,r) rv zcan't start threadr c , e Zd Zd Zd Zd Zd Zd ZdS )
TestThreadc . t d | _ d S )Nr )r7 seedsrE s r' r zTestThread.setup_method s 1XX
r) c Z ddl m t j t | j f|z }t j t | j f|z }fdt
| j | D }d |D d |D t
| j | D ]+\ }} t t | | ,t j j
j dk r"t j
dk rt || d S t || d S )Nr )Threadc
h g | ].\ }} t t | |f /S ))targetargs)r r )r r4 or functions r' r z-TestThread.check_function.. sO
0
0
0A V89WQZZ+@+@!*D
E
E
E
0
0
0r) c 6 g | ]}| S rP )startr rr s r' r z-TestThread.check_function.. s q r) c 6 g | ]}| S rP )joinr s r' r z-TestThread.check_function.. s a r) r win32) threadingr r8 r rJ r zipr r intprW r r7 platformr r
) r3 r szout1out2tr4 r r s ` @r' check_functionzTestThread.check_function sI $$$$$$xTZ *R/00xTZ *R/00
0
0
0
0
0 TZ..
0
0
0 A 1
D)) / /DAq HYwqzz**A . . . . 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 )Nr r .)r r r= s r'
gen_randomz*TestThread.test_normal..gen_random s |||//CHHHr) )r r r r3 r s r' r zTestThread.test_normal s2 0 0 0
J844444r) c : d }| |d d S )Nc \ | t j d |d< d S )Nrd r- rv .)rr r8 r r s r' r z'TestThread.test_exp..gen_random s* ((rw{/C/C(DDCHHHr) r r r r s r' test_expzTestThread.test_exp s5 E E E
J;77777r) c : d }| |d d S )Nc D | ddgdz d |d< d S )Nr r r r r .)rh r s r' r z/TestThread.test_multinomial..gen_random s* ((fX\(FFCHHHr) )r r r r r s r' r zTestThread.test_multinomial s5 G G G
J:66666r) N)rM rN rO r r r r r rP 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 )NrT r r r )r8 rz argOneargTwoargThreetgtShaperE s r' r z$TestSingleEltArrayInput.setup_method s@ hsmm hsmm !
r) c t j t j t j t j t j t j t j t j t j t j
t j t j f }t j t j f}|D ]O}||v r |t j dg }n || j }t! |j | j Pd S r )rZ rr r rX r$ r rl r r r rq r r r8 rz r r ry r )r3 funcs probfuncsr8 r= s r' test_one_arg_funcsz*TestSingleEltArrayInput.test_one_arg_funcs s #V%:!6#4 v !6#3 5 %v'78 3 3Dy d28SE??++ d4;'' DM 2 2 2 2 3 3r) c t j t j t j t j t j t j t j t j t j t j
t j t j t j
t j f}t j
t 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 )NrX r )rZ r/ r rQ r r| r rW r r r r ri r[ r r8 rz r r r ry r )r3 r r r8 r r= s r' test_two_arg_funcsz*TestSingleEltArrayInput.test_two_arg_funcs s fl66 &. !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) c T t j t j t j t j t j t j t j t j t j g }t j }t j dg }t j dg }|D ]} ||||| }t |j | j ||d ||| }t |j | j |||d || }t |j | j d S )Nr< r r )r8 rC r2 r5 r0 r3 r1 rx r r4 rZ r1 rz r ry r )r3 r( r r8 rZ rY r r= s r' r z%TestSingleEltArrayInput.test_integers s 27BHbh 29bh ;x}}hsmm 3 3B$sD82>>>C DM 2 2 2$s1vthbAAAC DM 2 2 2$sDGhbAAAC DM 2 2 2 2 3 3r) c t j t j t 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 )
rZ r r* r r r r r ry r )r3 r r8 r= s r' test_three_arg_funcsz,TestSingleEltArrayInput.test_three_arg_funcs
s $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)rM rN rO r r r r r rP r) r' r r s_ 3 3 3&3 3 363 3 3" 3 3 3 3 3r) r rX c | d }| d }t | }| | |j d d }t j dk r| }t
j | }|j d d | d d k sJ | | d d k sJ | }|j d d }t j dk r| }t
j | }|j d d | d d k sJ | | d d k sJ d S )
Nr r r r rr r r r r )
r r= r r7 r8 r9 r: r; r< r )rX r r mt19937r r; r s r' test_jumpedr
sZ &>D 7OEdmmG u
-
'C
} llnn
^C
F = !% (F9,=e,D D D D D !2 highz low >= highrA )r r r r rD r1 )rZ r( r msgs r' !test_single_arg_integer_exceptionr v
s GAJJ
C
1**kC z - - - . . TH ---. . . . . . . . . . . . . . ."
5,,
C z - - - 2 2 R 1112 2 2 2 2 2 2 2 2 2 2 2 2 2 2 z - - - 4 4 bT4( 3334 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4s5 A""A&)A&B33B7:B7D DDrW f4f8c t j dd| }g d}t j t d 5 t
||| d d d n # 1 swxY w Y t j t d 5 t
|||j | d d d d S # 1 swxY w Y d S ) Nr F)orderrW r zSupplied output arrayr r r )r8 r r r rD rZ r ry )rW r= ry s r' test_c_contig_req_outr
sV
(6E
2
2
2C IIE z)@ A A A ; ; eE :::; ; ; ; ; ; ; ; ; ; ; ; ; ; ; z)@ A A A K K e39E JJJK K K K K K K K K K K K K K K K K Ks# A!!A%(A%$B88B<?B<r r Cdistc t j d|| } | || }||u sJ | |||j }||u sJ d S )Nr )rW r r )r= rW r )r8 r ry )r r rW r= variatess r' test_contig_req_outr
sd
(6e
4
4
4Ct5)))H s????t5sy999H s??????r) c R t j t j d } | d | \ }}}|d d dk sJ ||d d }||j _ |j j }||k sJ d S )Nr r< ) PCG64DXSM)r8 rZ r r r
__reduce__r r )r ctorr state_ar state_bs r' $test_generator_ctor_old_style_pickler
s RY 0 0 3 3 4 4Bq --//D$ 8~ % % % % d2A2hA #AOo#G g r) )1r7 r: r numpyr8 numpy.linalgr
numpy.testingr r r r r r r
r r r
numpy.randomr r r r rZ JUMP_TEST_DATAfixturer( r+ rR rb r r r r rv r rt r r r r r r r r r r r r rP r) r' r% s
$ $ $ $ $ $; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ;
G F F F F F F F F F F F 7799 "dmnoo!clopp "dmpqq!clopp 766"dmpqq!clopp * he}555 65!6 !6 !6 !6 !6 !6 !6 !6H > > > > > > > > 4) 4) 4) 4) 4) 4) 4) 4)lC- C- C- C- C- C- C- C-L , , , , , , , ,DE E E E E E E E P m, m, m, m, m, m, m, m,`#c , c , c , c , c , c , c , c ,L G$899+7 +7 +7 +7 +7 +7 +7 :9+7^R3 R3 R3 R3 R3 R3 R3 R3j >22@ @ 32@6(8 (8 (8V) ) ) "rd,,dE]33
4
4 43 -,
4 4,//K K 0/K 4,//3*--&"8&-!HII JI .- 0/
r)