Y9= d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZm Z m
Z
m Z m Z m
Z
mZmZmZ d dlmZmZ d dlmZ ddddd ej d d
z fd d d
dZ ej e j dk r dddddddddZn dddd d d d d dZ ej d!e" d# Z ej d$ Z d% Z G d& d' Z G d( d) Z G d* d+ Z! G d, d- Z" G d. d/ Z# G d0 d1 Z$ G d2 d3 Z%ej& ' ed45 G d6 d7 Z( G d8 d9 Z)d: Z*d; Z+d< Z,d= Z-d> Z.d? Z/d@ Z0dA Z1dB Z2dS )C N) assert_
assert_raises assert_equal assert_warnsassert_no_warningsassert_array_equalassert_array_almost_equalsuppress_warningsIS_WASM)MT19937PCG64)random)g Y@g333333?) ?) r
r g @)d r ) $@) )binomial geometrichypergeometric logseries multinomialnegative_binomialpoissonzipf @2fbead005fc63942decb5326d36a1f32fe2c9d32c904ee61e46866b88447c263@23ead5dcde35d4cfd4ef2c105e4c3d43304b45dc1b1444b7823b9ee4fa144ebb@0d764db64f5c3bad48c8c33551c13b4d07a1e7b470f77629bef6c985cac76fcf@7b59bf2f1691626c5815cdcd9a49e1dd68697251d4521575219e4d2a1b8b2c67@d754fa5b92943a38ec07630de92362dd2e02c43577fc147417dc5b9db94ccdd3@8eb216f7cb2a63cf55605422845caaff002fddc64a7dc8b2d45acd477a49e824@70c891d76104013ebd6f6bcf30d403a9074b886ff62e4e6b8eb605bf1a4673b7@01f074f97517cd5d21747148ac6ca4074dde7fcb7acbaec0a936606fecacd93f)r r r r r r r r @8626dd9d052cb608e93d8868de0a7b347258b199493871a1dc56e2a26cacb112@8edd53d272e49c4fc8fbbe6c7d08d563d62e482921f3131d0a0e068af30f0db9@83496cc4281c77b786c9b7ad88b74d42e01603a55c60577ebab81c3ba8d45657@65878a38747c176bc00e930ebafebb69d4e1e16cd3a704e264ea8f5e24f548db@7a984ae6dca26fd25374479e118b22f55db0aedccd5a0f2584ceada33db98605@d636d968e6a24ae92ab52fe11c46ac45b0897e98714426764e820a7d77602a61@956552176f77e7c9cb20d0118fc9cf690be488d790ed4b4c4747b965e61b0bb4@f84ba7feffda41e606e20b28dfc0f1ea9964a74574513d4a4cbc98433a8bfa45module)scopeparamsc V | j t | j t | j fS N)param INT_FUNCSINT_FUNC_HASHES)requests w/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/numpy/random/tests/test_randomstate.pyint_funcr9 2 s$ M9W]3 GM * , , c # K t j } dV t j | dS )z:Ensures that the singleton bitgen is restored after a testN)npr get_bit_generatorset_bit_generator) orig_bitgens r8 restore_singleton_bitgenr@ 8 s@ )--//K EEEI ,,,,,r: c N t | d |d t | d d |d d t | d d |d d t | d |d t | d |d d S )N
bit_generatorstatekeypos has_gaussgauss)r r )abs r8 assert_mt19937_state_equalrJ @ s ?#Q%7888qz%(!G*U*;<<<qz%(!G*U*;<<<; ; 0007QwZ(((((r: c 8 e Zd Zd Zd Zd Zd Zd Zd Zd Z dS ) TestSeedc t j d }t | d d t j d }t | d d d S )Nr i i )r RandomStater randintselfss r8 test_scalarzTestSeed.test_scalarI s\ q ! !QYYt__c*** z * *QYYt__c*****r: c t j t d }t | d d t j t j d }t | d d t j dg }t | d d t j dg }t | d d d S )Nr rN i r i rO i )r rP ranger rQ r< arangerR s r8
test_arrayzTestSeed.test_arrayO s uRyy ) )QYYt__c*** ry}} - -QYYt__c*** s # #QYYt__c***
| , ,QYYt__c*****r: c t t t j d t t t j d d S )N r TypeErrorr rP
ValueErrorrS s r8 test_invalid_scalarz TestSeed.test_invalid_scalarY s2 i!3T:::j&"4b99999r: c T t t t j dg t t t j dg t t t j dg t t t j g d t t t j g d d S )Nr[ r\ r ) r r )rc r r] r` s r8 test_invalid_arrayzTestSeed.test_invalid_array^ s i!3dV<<<j&"4rd;;;j&"4zlCCCj&"46H6H6HIIIj&"46I6I6IJJJJJr: c t t t j t j g t j t t t j g dg t t t j g dg dg d S )Ndtyperc r ) r )r r_ r rP r< arrayint64r` s r8 test_invalid_array_shapez!TestSeed.test_invalid_array_shapef s j&"4bhrEGX7O 7O 7O P P Pj&"4yyykBBBj&"4yyy7@yy7B C C C C Cr: c t j t d }t t 5 | d d d d d S # 1 swxY w Y d S )Nr )r rP r
r r^ seed)rS rss r8 test_cannot_seedzTestSeed.test_cannot_seedn s
a
)
)
9
%
% GGDMMM s AA A c P t t t j t d S r3 )r r_ r rP r r` s r8 test_invalid_initializationz$TestSeed.test_invalid_initializations s j&"4g>>>>>r: N)
__name__
__module__ __qualname__rU rY ra re ro rt rv r: r8 rL rL H s + + + + + +: : :
K K KC C C
? ? ? ? ?r: rL c e Zd Zd Zd ZdS ) TestBinomialc t j dd }dD ]K}t t j d| dk t t j || | Ld S )Nr intrg )r r rc r )r< zerosr r r r )rS r ps r8 test_n_zerozTestBinomial.test_n_zerox ss %((( A AA FOAq))Q. / / / vua 8 8% @ @ @ @ A Ar: c \ t t t j dt j d S )Nrc )r r_ r r r< nanr` s r8
test_p_is_nanzTestBinomial.test_p_is_nan s j&/1bf=====r: N)rw rx ry r r rz r: r8 r| r| w s5 A A A> > > > >r: r| c > e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d S )
TestMultinomialc 4 t j dddg d S )Nr 皙?皙?r r r` s r8
test_basiczTestMultinomial.test_basic s 3c
+++++r: c 4 t j dg d d S )Nr )r r r r r r` s r8 test_zero_probabilityz%TestMultinomial.test_zero_probability 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\ rl )r r rQ r< allrS xs r8 test_int_negative_intervalz*TestMultinomial.test_int_negative_interval s fnR,,1111r1111222 N2r1 % %rQw q2v r: c 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 d|t d d S )Nr rc rc r r r r r r r )
r r r r< uint32shaperm r r^ floatrS r s r8 test_sizezTestMultinomial.test_size s6
#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 )Nr 皙?r 皙?r r_ r r r` s r8 test_invalid_probz!TestMultinomial.test_invalid_prob s> j&"4cC:FFFj&"4cC:FFFFFr: c L t t t j dddg d S )Nr\ r r r r` s r8 test_invalid_nz TestMultinomial.test_invalid_n s# j&"4b3*EEEEEr: c p t j d }|t j |dd d z }|dd d }t j d t j d| }t j d t j dt j | }t || d S )Ng .@rc rj i۠iUr )pvals)r< rX sumr rr r ascontiguousarrayr )rS r r
non_contigcontigs r8 test_p_non_contiguousz%TestMultinomial.test_p_non_contiguous s IcNN RVAaddG__ !$Q$ J '5999
J #Cr/CE/J/JKKK:v.....r: c t j g dt j }|| z }d}t j t | 5 t j d| d d d d S # 1 swxY w Y d S )N)
Gz?r & .>r r r r r r r rg z-[\w\s]*pvals array is cast to 64-bit floatingmatchrc ) r< rm float32r pytestraisesr_ r r )rS r r r s r8 test_multinomial_pvals_float32z.TestMultinomial.test_multinomial_pvals_float32 s H : : :AC
M
M
MAEEGG @
]:U
3
3
3 ) ) q% ( ( ( ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) )s A::A>A>N) rw rx ry r r r r r r r r rz r: r8 r r s , , ,; ; ; G G GF F F/ / /) ) ) ) )r: r c P e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
Z d Z
d S )
TestSetStatec d| _ t j | j | _ | j | _ d S NiI)rr r rP random_state get_staterC r` s r8 setup_methodzTestSetState.setup_method s8 " .ty 9 9&0022
r: c | j d }| j | j | j d }t t j ||k d S )N )r tomaxint set_staterC r r< r rS oldnews r8 r zTestSetState.test_basic si ((,, ##DJ/// ((,,scz""#####r: c | j d }| j | j | j d }t t j ||k d S Nrj size)r standard_normalr rC r r< r r s r8 test_gaussian_resetz TestSetState.test_gaussian_reset so //Q/77 ##DJ/// //Q/77scz""#####r: c T | j | j }| j d }| j | | j d }t t j ||k d S r )r r r r r r< r )rS rC r r s r8 test_gaussian_reset_in_media_resz-TestSetState.test_gaussian_reset_in_media_res s
))+++!++-- //Q/77 ##E*** //Q/77scz""#####r: c | j d d }| j d }| j | | j d }| j | j | j d }t t j ||k t t j ||k d S )Nrd r r )rC r r r r r< r )rS old_statex1x2x3s r8 test_backwards_compatibilityz)TestSetState.test_backwards_compatibility s JssO
.
.B
.
7
7 ##I...
.
.B
.
7
7 ##DJ///
.
.B
.
7
7rRx !!!rRx !!!!!r: c < | j dd d S Nr )r r r` s r8 test_negative_binomialz#TestSetState.test_negative_binomial s#
++C55555r: c | t j t }t 5 }| t
}| }t t | dk t |t sJ |d dk sJ d d d d S # 1 swxY w Y d S )Nrc rB r
) r rP r
r
recordRuntimeWarningr r len
isinstancedict)rS rs supwrC s r8 test_get_state_warningz#TestSetState.test_get_state_warning s
(
(
5C
>**ALLNNE CFFaK eT** * ** )W44444 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5s A4B11B58B5c | j }d|dd z }t t | j j | t t
| j j t
j |t | j d }|d= t t | j j | d S )N)Unknownrc rg FlegacyrB ) r r r r_ r r^ r< rm object)rS rC new_states r8 !test_invalid_legacy_state_settingz.TestSetState.test_invalid_legacy_state_setting s !++--!E!""I- j$"3"=yIIIi!2!<hy777 9 9 9!++5+99 / "j$"3"=uEEEEEr: c | j d | j d | j | j d }t |d d t
j t
j | j }| d }t || d S )Nr r Fr rF rc )
r rr
random_sampler r r pickleloadsdumpsrJ )rS pickled rs_unpick unpickleds r8 test_picklezTestSetState.test_pickle s q!!! '',,, ))+++#--U-;;W[)1---L d.?!@!@AA ''u'55 "7I66666r: c | j }| j | j | | j d }t || d S )NFr )r __getstate__r __setstate__r rJ )rS
attr_staterC s r8 test_state_settingz TestSetState.test_state_setting sq &3355
))+++ &&z222!++5+99":u55555r: c X t | j d sJ d S )NzRandomState(MT19937))reprr
startswithr` s r8 test_reprzTestSetState.test_repr s. D%&&112HIIIIIIIr: N)rw rx ry r r r r r r r r r r r rz r: r8 r r s 3 3 3
$ $ $ $ $ $ $ $ $
"
"
"6 6 6
5 5 5F F F7 7 76 6 6J J J J Jr: r c 4 e Zd Zej Zej ej ej ej
ej ej ej
ej ej g Zd Zd Zd Zd Zd Zd Zej ej d j dk d
d Z d Z d
Z dS ) TestRandintc J t t | j dt d S Nrc rg )r r^ rfuncr r` s r8 test_unsupported_typez!TestRandint.test_unsupported_type s! iQe<<<<<d< Z?d= Z@d> ZAd? ZBd@ ZCdA ZDdB ZEdC ZFdD ZGdE ZHdF ZIdG ZJdH ZKdI ZLdJ ZMdK ZNdL ZOdM ZPdN ZQdO ZRdP ZSdQ ZTdR ZUdS ZVdT ZWdU ZXdV ZYdW ZZdX Z[dYS )ZTestRandomDistc d| _ d S r rr r` s r8 r zTestRandomDist.setup_method s
r: c t j | j t j dd }t j ddgddgddgg }t ||d
d S ) Nrj r _\*?琛?<p?,o? H?Z? decimal)r rr randr< rm r rS r5 r2 s r8 test_randzTestRandomDist.test_rand sv DI Q""( 02E F 02E F /1C D F G G "&'2>>>>>>r: c t j | j t j }d}t ||d d S )NrE rK rL )r rr rN r rO s r8 test_rand_singletonz"TestRandomDist.test_rand_singleton s? DI %!&'2>>>>>>r: c H t j | j t j dd }t j ddgddgddgg }t ||d
t j | j t j }t ||d d
d S ) Nrj r !>>> DI !&'$-DDDDDDr: c t j | j t j ddd }t j ddgddgd d
gg }t || d S ) Nc rj r r rj ) )r rr rQ r< rm r rO s r8 test_randintzTestRandomDist.test_randint sl DI Rf555(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 || t j | j t 5 }| t }t j d d }t
t | dk d d d n # 1 swxY w Y t ||d
z d S )Nr_ r` ra r rc rb rj rc rd re rf r ) r rr r
r DeprecationWarningrandom_integersr r r< rm r )rS r r r5 r2 s r8 test_random_integersz#TestRandomDist.test_random_integers s DI
!C
-..A+C&AAAF CFFaK ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! (RG "I #J ( ) ) 67+++ DI
!C
-..A+Cf===F CFFaK ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! 67S=11111s% ABB
B
%AEE Ec ( t j | j t j | j }| d }t j t j dk r*t j ddgddgdd ggt j
}n)t j d d gd
dgddggt j
}t || | | j | }t ||d d S )Nra r ii4Oi˕+iRKiriFbqi rg l K+W+y l r&&J] l ; l nBz
l @XHT l D+l+ + rZ ) r rr rP r r< r r~ r rm rn r )rS rs r5 r2 s r8
test_tomaxintz TestRandomDist.test_tomaxint s DI
*
*&))
8C==
* *hi 8!+i 8!+i 8 :ACK K KGG h!46I J!46I J!46I J L &(X/ / /G
VW%%%
VWT]+++++r: c Z 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 || t 5 }| t }t j d j }t j |t j d j |t j d j }t t | dk d d d n # 1 swxY w Y t || d S )Nr. rc )
r
r rj r rk r< r r r r r rh r; )rS r r r5 r2 typers r8 test_random_integers_max_intz+TestRandomDist.test_random_integers_max_int s
!C
-..A+BHSMM,=,.HSMM,=? ?F CFFaK ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! (3--#VW%%%
!C
-..AHSMM&E+EE"(3--2C,D,D,1E"(3--2C,D,DF FF CFFaK ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! VW%%%%%s% A>BB B B)FFFc t j 5 t j dt t t t
j t j d j t t t
j t j d j t j d j d d d d S # 1 swxY w Y d S )Nerrorr. )
warningscatch_warnings simplefilterrj r r rk r< r r r` s r8 test_random_integers_deprecatedz.TestRandomDist.test_random_integers_deprecated s
$
&
& @ @ !'+= > > >
, 0 (3--+
-
-
-
, 0 (3--+RXc]]->
@
@
@ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @s B!CC Cc F t j | j t j d }t j ddgddgddgg }t ||d t j | j t j }t ||d
d d S ) Nra rE rF rG rH rI rJ rK rL rZ )r rr r r< rm r rO s r8 test_random_samplez!TestRandomDist.test_random_sample& s DI %f--( 02E F 02E F /1C D F G G "&'2>>>> DI %''!&'$-DDDDDDr: c t j | j t j dd }t j g d }t || d S )Nrk )r rj r rj r rr choicer< rm r rO s r8 test_choice_uniform_replacez*TestRandomDist.test_choice_uniform_replace2 sN DI q!$$(<<<((67+++++r: c t j | j t j ddg d }t j g d }t || d S )Nrk )皙?r 皙?r r )rc rc r r r{ rO s r8 test_choice_nonuniform_replacez-TestRandomDist.test_choice_nonuniform_replace8 sZ DI q!';';';<<<(<<<((67+++++r: c t j | j t j ddd }t j g d }t || d S )Nrk rj Freplace)r rc rj r{ rO s r8 test_choice_uniform_noreplacez,TestRandomDist.test_choice_uniform_noreplace> sS DI q!U333(999%%67+++++r: c t j | j t j dddg d }t j g d }t || d S )Nrk rj F)r 333333?r r r r )r rj rc r{ rO s r8 test_choice_nonuniform_noreplacez/TestRandomDist.test_choice_nonuniform_noreplaceD s\ DI q!U6J6J6JKKK(999%%67+++++r: c t j | j t j g dd }t j g d }t || d S )N)rH rI cdrk )r r r r r{ rO s r8 test_choice_nonintegerz%TestRandomDist.test_choice_nonintegerJ sX DI 3 3 3Q77(///0067+++++r: c t 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\ rj @rc r rk )rc r rj rk g ?r )r r r r r r ri Fr rd )r\ )r\ rc )rc r r r )r r| r r_ )rS r< s r8 test_choice_exceptionsz%TestRandomDist.test_choice_exceptionsP 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 g d d S )Nr r r Tr Fr rc rg r )r rj )r r r r r r r )rj r rk r i)r r rH rI )r r< isscalarr r| rm emptyr tuplendimitemr r rX rQ r r_ )rS r rH arrrT s r8 test_choice_return_shapez'TestRandomDist.test_choice_return_shapec s
#J FM!T : : :;;<<< FM!U ; ; ;<<=== FM!TQ ? ? ?@@AAA FM!Ua @ @ @AABBB FM1a&$ ? ? ?@@AAA
tfd333t;<<< HaV hq'''A
c4000A5666
GGBK
aD A A ABBBCCCBK
aE B B BCCCDDDBK
aDA F F FGGGHHHBK
aEQ G G GHHHIIIBK
q!fa F F FGGGHHH
tfa666;q@AAA HaV hq'''A
c1d33388::a?@@@
* * *V]1a666>>V]C:I>>>D j&-R88888r: c ~ t j g d }g d}t t t j || d S )N)* rc r )NNNr )r< rm r r_ r r| )rS rH r s r8 test_choice_nan_probabilitiesz,TestRandomDist.test_choice_nan_probabilities s? HZZZ j&-a888888r: c
l t j d dz }d|dd d< t j | j t j dd|d d d }t j | j t j ddt j |d d d }t
|| d S )Nr rl r rc r rj r )r< onesr rr r| r r )rS r r r s r8 test_choice_p_non_contiguousz+TestRandomDist.test_choice_p_non_contiguous s GBKK!O!$Q$ DI ]1a1SSqS6222
DI q!r';AcccF'C'CDDD:v.....r: c t j | j t j d }d}t || d S )Nr s
Ui+Wf)r rr bytesr rO s r8
test_byteszTestRandomDist.test_bytes s> DI b!!.VW%%%%%r: c d d d d d d d d d d
d f D ][}t j | j |g d }t j | |} |g d
}t || \d S )Nc * t j g S r3 )r< rm r s r8 z-TestRandomDist.test_shuffle.. s rx|| r: c | S r3 rz r s r8 r z-TestRandomDist.test_shuffle.. s q r: c d t j | t j S r3 )r< asarrayastyper r s r8 r z-TestRandomDist.test_shuffle.. s rz!}} 3 3BG < < r: c d t j | t j S r3 )r< r r r r s r8 r z-TestRandomDist.test_shuffle.. s rz!}} 3 3BJ ? ? r: c d t j | t j S r3 )r< r r complex64r s r8 r z-TestRandomDist.test_shuffle.. s rz!}} 3 3BL A A r: c Z t j | t S r3 )r< r r r r s r8 r z-TestRandomDist.test_shuffle.. s rz!}} 3 3F ; ; r: c d | D S )Nc g | ]}||fS rz rz .0is r8
zATestRandomDist.test_shuffle.... s 2 2 21A 2 2 2r: rz r s r8 r z-TestRandomDist.test_shuffle.. s 2 2 2 2 2 r: c > t j d | D S )Nc g | ]}||gS rz rz r s r8 r zATestRandomDist.test_shuffle.... *=*=*=aAq6*=*=*=r: )r< r r s r8 r z-TestRandomDist.test_shuffle.. s rz*=*=1*=*=*= > > r: c 8 t j | | g j S r3 )r< vstackTr s r8 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 rz rz r s r8 r zATestRandomDist.test_shuffle.... s +>+>+>qQF+>+>+>r: rH rI )r< r r~ r( recarrayr s r8 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 rz rz r s r8 r zATestRandomDist.test_shuffle.... r r: rH rc rI )r< r r r r s r8 r z-TestRandomDist.test_shuffle.. sA rz*=*=1*=*=*=,/+>,/4+@+B C C r:
rc r rj rk rl r r r
r rc r r r rk rl r r rj )r rr shuffler )rS convalistr5 r2 s r8 test_shufflezTestRandomDist.test_shuffle s ,+ [<<??AA;;22>>224 4C C D 0 0D"
K " " "D77788E N5 ! ! !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 }d S )Nr )rl rk rj rc r\ 2 c n t j d }t t t j | d S )Nrj )r< rm r r^ r r r s r8 test_shuffle_invalid_objectszHTestRandomDist.test_shuffle_masked..test_shuffle_invalid_objects s) A )V^Q 7 7 7 7 7r: )
r< ma
masked_valuesreshaperW rX copyr r r sorteddatamask)rS rH rI a_origb_origr r s r8 test_shuffle_maskedz"TestRandomDist.test_shuffle_masked sL E
599f = = AA Er J J E "
1A 5r : :r L LA N1 qvqvg'' V[L0I)J)J
L
L
L N1 qvqvg'' V[L0I)J)J
L
L
L
L 8 8 8 8 8r: c t j | j g d}t j | }g d}t || t j | j t j g d j }t j | }t |t j | j t j | j d}t t t j | t j | j d}t t t j | d}g d}t j | j t j | }t || d S )Nr r abcdg333333?r )
r r r rl rc rj rk r r r ) r rr permutationr r<
atleast_2dr r
IndexError)rS r r5 r2 arr_2d bad_x_str bad_x_float integer_vals r8 test_permutationz TestRandomDist.test_permutation sB DI ...#E**00067+++ DI = = =>>@#F++62=#9#9#;<<< DI j&"4i@@@ DI j&"4kBBB 000 DI #K0067+++++r: c t j | j t j ddd }t j ddgddgd d
gg }t ||d d S )
Nr r ra r gAKč?g hA?g ҇o>gάXf+q?gS1$?g'8zX ?rK rL )r rr betar< rm r rO s r8 test_betazTestRandomDist.test_beta s{ DI R&111()+BC)+BC)+BCEF F "&'2>>>>>>r: c @ t j | j t j ddd }t j ddgddgd d
gg }t || t j | j t j dd }d}t || d S ) Ng Zd;Y@gv/?ra r % + r 0 . - )r rr r r< rm r rO s r8
test_binomialz TestRandomDist.test_binomial s DI $V<<<(RH H H & ' ' 67+++ DI $//67+++++r: c t j | j t j dd }t j ddgddgdd gg }t ||d
d S ) Nr ra r g+Q]uO@g+Q@gY7`ZqP@ge(fG@gvtR@gEIȋR@
rL )r rr chisquarer< rm r rO s r8 test_chisquarez TestRandomDist.test_chisquare sz DI !"6222( 13G H 13G H 02F G I J J "&'2>>>>>>r: c ( t j | j t j ddg }t j |d }t j ddgddggd d
gd d ggd
dgddggg }t ||d t j ddg }t
t t j | t j | j t j ddg }t j | }t ||d d d S )N I>>>Hgx 011 j&"2I>>> DI .0DEFF!%((!&'$-DDDDDDr: 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 |t d d S )Nr r rc r r r r )
r< rm r r r r r r r^ r r s r8 test_dirichlet_sizez"TestRandomDist.test_dirichlet_size s! H*,@A B BV%a1 66>>>>>r: c t t j d d t t t j d d S Nr scale )r r r r r_ r` s r8 test_exponential_0z!TestRandomDist.test_exponential_0E s> V'a000!444j&"4C@@@@@@r: c t j | j t j ddd }t j ddgddgd d
gg }t ||d d S )
N M ra r g 7 ?g`j?g6"+?gܳ2>?gc>+Y?g_}S?rK rL )r rr fr< rm r rO s r8 test_fzTestRandomDist.test_fI s{ DI "bv...( 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 )
Nrl rj ra r gԏGM8@gȌ<@g ":@gv#)@gZQ?@gr[@@ rL )r rr gammar< rm r rO s r8
test_gammazTestRandomDist.test_gammaQ s{ DI a000( 13G H 13G H 13G H J K K "&'2>>>>>>r: c t t j dd d t t t j dd d S )Nr )r r r )r r r r r_ r` s r8 test_gamma_0zTestRandomDist.test_gamma_0Y s@ V\333Q777j&,cEEEEEEr: c t j | j t j dd }t j ddgddgddgg }t || d S ) N_c97ݚ?ra r r r rl r )r rr r r< rm r rO s r8 test_geometricz TestRandomDist.test_geometric] sk DI !*6:::(QF H G % & & 67+++++r: c H t t t j d t t t j dgdz t t t j d t t t j dgdz t 5 }| t 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 ) r r_ r r r
r r r< r rS r s r8 test_geometric_exceptionsz(TestRandomDist.test_geometric_exceptionse s( j&"2C888j&"2SEBJ???j&"2D999j&"2TFRK@@@
GC JJ~ & & & *f&6 ? ? ? *f&62
F F F G G G G G G G G G G G G G G G G G Gs A3DD Dc t j | j t j ddd }t j ddgddgd d
gg }t ||d d S )
Nr @ra locr r gؕ?gs0 ?g #0g(xg-KD?gůp`@rK rL )r rr gumbelr< rm r rO s r8 test_gumbelzTestRandomDist.test_gumbelo s{ DI :SvFFF( 02E F 02F G 02F G I J J "&'2>>>>>>r: c t t j d d t t t j d d S r )r r r! r r_ r` s r8
test_gumbel_0z TestRandomDist.test_gumbel_0w < V]+++Q///j&-s;;;;;;r: c t j | j t j dddd }t j ddgddgddgg }t || t j dd d
d }t j g d }t || t j d
d dd }t j g d }t || t j d dd
d }t j g d }t || t j d d
dd }t j g d }t || d S )Ng333333$@g @r ra r r r rl r rj rk )rj rj rj rj rK r )r r r r )r r r r )r rr r r< rm r rO s r8 test_hypergeometricz"TestRandomDist.test_hypergeometric{ sf DI &tS"6BBB(RH H F $ % % 67+++ &q!QQ777(<<<((67+++&r1bq999(+++,,67+++ &q!QQ777(<<<((67+++&q"bq999(<<<((67+++++r: c t j | j t j ddd }t j ddgddgd d
gg }t ||d d S )
Nr r ra r g bO?g¸?gpJm @g{K*t @g:^w-gY~o?rK rL )r rr laplacer< rm r rO s r8 test_laplacezTestRandomDist.test_laplace s{ DI JcGGG( 02E F 02E F 13F G I J J "&'2>>>>>>r: c t t j d d t t t j d d S r )r r r) r r_ r` s r8 test_laplace_0z TestRandomDist.test_laplace_0 s< V^!,,,a000j&.<<<<<>>>>>r: c t j | j t j ddd }t j ddgddgd d
gg }t ||d d S )
Nr r ra )meansigmar gɁ0@go,+4FB@g)ʭ6@g5 ?gnxNnP@g)U@r rL )r rr lognormalr< rm r rO s r8 test_lognormalz TestRandomDist.test_lognormal s| DI !z6JJJ( 13G H 13F G 13G H J K K "&'2>>>>>>r: c t t j d d t t t j d d S )Nr )r2 rc r )r r r3 r r_ r` s r8 test_lognormal_0z TestRandomDist.test_lognormal_0 s> V%A...222j&"2#>>>>>>r: c t j | j t j dd }t j ddgddgddgg }t || d S )Ngƀ@?ra )r r r r r rj )r rr r r< rm r rO s r8 test_logseriesz TestRandomDist.test_logseries sk DI !JV<<<(QF G F $ % % 67+++++r: c : t j d dk sJ d S )Nr rc )r r r` s r8 test_logseries_zeroz"TestRandomDist.test_logseries_zero s$ ""a''''''r: valuer r\ ?g @c j t j d 5 t j t 5 t j | d d d n # 1 swxY w Y t j t 5 t j t j |gdz d d d n # 1 swxY w Y t j t 5 t j 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 )Nignoreinvalidr r )r< errstater r r_ r r rm )rS r; s r8 test_logseries_exceptionsz(TestRandomDist.test_logseries_exceptions s
[
*
*
* > > 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| D(A D(A D(A D(5+B, D(,B0 0D(3B0 4 D(4D D(D D(D D((D,/D,c t j | j t j ddgdz d }t j g dg dgg dg d gg d
g d gg }t || d S ) Nr UUUUUU?r ra r )rk rj rl rk r r )rl r r r r rc )rj rk rj r r rk )r rc rk rj r rk )rk rk r rl r rj )rk rj rk r rj rk )r rr r r< rm r rO s r8 test_multinomialz TestRandomDist.test_multinomial s DI #B1 6BBB( / / / / / / 1 / / / / / / 1 / / / / / / 1 2 3 3 67+++++r: c 0 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 || t t j ||d t t t j ||d t j ddgddggt j }t 5 }t j || |
t }t | dk sJ d d d n # 1 swxY w Y t j d }t j d }t t t j ||d t t t j t j d | t t t j |t j d t t t j |t j d d S ) N)r r rc r ra g4Vj?gy'@gI1?gSB#@g e<@g 4W(@g6R¿?g9r
v"@gс?g{#@g4xg"f"@rK rL gN}O6?gYMW"@r r> ) check_validraiser rg other)r rc rc rj )r rr multivariate_normalr< rm r r r r r r_ r r
r r r eyer ) rS r1 covr r5 r2 r r mus r8 test_multivariate_normalz'TestRandomDist.test_multivariate_normal s DI 1v1v +D#t<<( /1B C /1B C E /1C D /1B C E /1B C 02C D F G H H "&'2>>>> +D#66( -/?@AA!&'2>>>> 1v1v1v ^V%?sKKK 65tS'/ 1 1 1 1 j&">>
C &tS 1 1 1
>**Aq66Q;;;;;
Xa[[fQiij&"RV L L L *f&> 6(R-
)
)
) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) )s A5BBBc 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 )Nrl ra )dfnoncr gG7@g! ܴ*@g1}z9?@g+驸0@g26E_r#@gI!1@r rL r r g ?gTD?g?T?g+qnl?g~|D?g밬?r g{01#@gprs'@g~;$@g"dI
@g*@g }',@)r rr noncentral_chisquarer< rm r rO s r8 test_noncentral_chisquarez(TestRandomDist.test_noncentral_chisquare sR DI ,GGG( 13G H 13G H 02F G I J J "&'2>>>>,&III( 03F G 03F G .1D E G H H "&'2>>>> DI ,GGG( .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 )Nrl r rc ra )dfnumdfdenrU r g^~?g/+c?g". @g @g7?gE?r rL )r rr noncentral_fr< rm r rO s r8 test_noncentral_fz TestRandomDist.test_noncentral_f. s DI $1AA*02 2 2( 02E F 02E F 02D E G H H "&'2>>>>>>r: c t j | j t j ddt j }t j | sJ d S )Nrl r )rY rZ rU )r rr r[ r< r isnan)rS r5 s r8 test_noncentral_f_nanz$TestRandomDist.test_noncentral_f_nan7 sI DI $1ABFCCCx r: c t j | j t j ddd }t j ddgddgd d
gg }t ||d d S )
Nr r ra r g&&n@gOG_ @g~w @gtw]տgm&5@g$>@rK rL )r rr normalr< rm r rO s r8 test_normalzTestRandomDist.test_normal< s{ DI :SvFFF( 02E F .0D E 02E F H I I "&'2>>>>>>r: c t t j d d t t t j d d S r )r r ra r r_ r` s r8
test_normal_0z TestRandomDist.test_normal_0D r% r: c t j | j t j dd }t j ddgddgdd gg }t j ||d
d S ) Nr ra rH r gW I@g
Ԩy@gy]0Ag1~\Aga@gY
7A )nulp)r rr paretor< rm testing assert_array_almost_equal_nulprO s r8 test_paretozTestRandomDist.test_paretoH s DI &999()+BC)+BC)+BCEF F
11&'1KKKKKr: c t j | j t j dd }t j ddgddgddgg }t || d S )Nr ra )lamr r rc )r rr r r< rm r rO s r8 test_poissonzTestRandomDist.test_poissonW sj DI JV<<<(QF F F $ % % 67+++++r: c ~ t j d j }d}t t t
j | t t t
j |gdz t t t
j | t t t
j |gdz t 5 }| t 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 ) r< r r r r_ r r r
r r r )rS lambiglamnegr s r8 test_poisson_exceptionsz&TestRandomDist.test_poisson_exceptions_ s7 # "j&.&999j&.6(R-@@@j&.&999j&.6(R-@@@
EC JJ~ & & & *fnbf = = = *fnrvhm D D D E E E E E E E E E E E E E E E E E Es 2A3D22D69D6c t j | j t j dd }t j ddgddgdd gg }t ||d
d S ) Nr ra rf g;'?gܴw*?g 5?gFG?g D ]?g$`?rK rL )r rr powerr< rm r rO s r8
test_powerzTestRandomDist.test_powerk sy DI
888( 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 ra )r r g"u+@gN$MB*@g7v8B4@g55@ge &@g~ӣZ1@r rL )r rr rayleighr< rm r rO s r8
test_rayleighz TestRandomDist.test_rayleighs sy DI r777( 02D E 13G H 13G H J K K "&'2>>>>>>r: c t t j d d t t t j d d S r )r r rx r r_ r` s r8 test_rayleigh_0z TestRandomDist.test_rayleigh_0{ s< V_1---q111j&/======r: c t j | j t j d }t j ddgddgddgg }t ||d
d S ) Nra r gMoDL?g6( [9grF=?gM' - gVgVA:Qy?rK rL )r rr standard_cauchyr< rm r rO s r8 test_standard_cauchyz#TestRandomDist.test_standard_cauchy sx DI 'V444( 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 ) Nra r g܁?gc!z?g&N"@gNI@gF>?g~ME?rK rL )r rr standard_exponentialr< rm r rO s r8 test_standard_exponentialz(TestRandomDist.test_standard_exponential sx DI ,&999( 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 ) Nrj ra )r r g)@gľ@gd],q@gPk{@g9B #1 @gv @r rL )r rr standard_gammar< rm r rO s r8 test_standard_gammaz"TestRandomDist.test_standard_gamma sz DI &QV<<<( 02E F 02E F 02C D F G G "&'2>>>>>>r: c t t j d d t t t j d d S )Nr )r r )r r r r r_ r` s r8 test_standard_gamma_0z$TestRandomDist.test_standard_gamma_0 s> V*333Q777j&"7sCCCCCCr: c t j | j t j d }t j ddgddgddgg }t ||d
d S ) Nra r rT rU rV rW rX rY rK rL )r rr r r< rm r rO s r8 test_standard_normalz#TestRandomDist.test_standard_normal sx DI 'V444( 02E F .0C D .0C D F G G "&'2>>>>>>r: c t j | j t j }t j d }t ||d d S )NrT rK rL r[ rO s r8 test_randn_singletonz#TestRandomDist.test_randn_singleton sJ DI (.//!&'2>>>>>>r: c t j | j t j dd }t j ddgddgdd gg }t ||d
d S ) Nr ra )rT r g\VH?ghU%gH$M?gLgVqdzǿg㖀?rK rL )r rr
standard_tr< rm r rO s r8 test_standard_tz TestRandomDist.test_standard_t sz DI "bv666( 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@ra )leftmoderightr g:}\)@ghرTj(@gǨL30@g6ڜA0@g6sh&@g<,@r rL )r rr
triangularr< rm r rO s r8 test_triangularz TestRandomDist.test_triangular s DI "5(.0 0 0( 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%@ra )lowhighr g>#H@g u@g#@g}d]#@gN@gw )* @rK rL )r rr uniformr< rm r rO s r8 test_uniformzTestRandomDist.test_uniform s{ DI Du6BBB( 02E F 02E F 02E F H I I "&'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 j t j |d |dz d S )Nr r rc g W4vC)r r )
r< finfor r r r r
OverflowErrorinf nextafter)rS fminfmaxfuncs r8 test_uniform_range_boundsz(TestRandomDist.test_uniform_range_bounds s x $x $~mTBF7A666mT1bf555mT4666mTRVG9qc:::mTA3999
2<a00td{CCCCCCr: c G d dt j }t j d | }t t
t j || G d dt j }t j d | }t t
t j |dd d S )Nc e Zd Zd ZdS )GTestRandomDist.test_scalar_exception_propagation..ThrowingFloatc t r3 r^ r` s r8 __float__zQTestRandomDist.test_scalar_exception_propagation..ThrowingFloat.__float__ r: N)rw rx ry r rz r: r8
ThrowingFloatr #
r: r r< c e Zd Zd ZdS )ITestRandomDist.test_scalar_exception_propagation..ThrowingIntegerc t r3 r r` s r8 __int__zQTestRandomDist.test_scalar_exception_propagation..ThrowingInteger.__int__ r r: N)rw rx ry r rz r: r8 ThrowingIntegerr r r: r rc ) r< ndarrayrm r( r r^ r r r )rS r throwing_floatr throwing_ints r8 !test_scalar_exception_propagationz0TestRandomDist.test_scalar_exception_propagation s BJ # ++M::i $ & & & bj x{{''88 i!6 aKKKKKr: c t j | j t j ddd }t j ddgddgd d
gg }t ||d d S )
Nr p=
ף?ra rM kappar gLZI@g'N"@gkr?gk=m@gډ?g^=\?rK rL r rr vonmisesr< rm r rO s r8
test_vonmisesz TestRandomDist.test_vonmises s{ DI D6BBB( 02E F 02E F 02E F H I I "&'2>>>>>>r: c t j | j t j ddd }t t j | d S )Nr g g5_PG>@B r )r rr r r r< isfiniter rS rs r8 test_vonmises_smallz"TestRandomDist.test_vonmises_small sR DI Ore < < < A""$$%%%%%r: c t j | j t j ddd }t j g d }t ||d d S )Nr g cArj r )g A^>?g @R7?g x.r rL r rO s r8 test_vonmises_largez"TestRandomDist.test_vonmises_large sg DI Bc:::( 4 4 4 5 5 "&'1======r: c t j | j t j dt j }t t j | d S )Nr )rM r )r rr r r< r r r^ r s r8 test_vonmises_nanz TestRandomDist.test_vonmises_nan sF DI Or 0 0 0 r: c t j | j t j ddd }t j ddgddgd d
gg }t ||d d S )
Nr r ra )r1 r r gZ@g-
g@g8
m?gVP"?g`V?gx^&3?r rL )r rr waldr< rm r rO s r8 test_waldzTestRandomDist.test_wald s{ DI $d@@@( 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 ra rf g@6?g]A?g(1ڥR?gU?g2_u?g^}-Q?rK rL )r rr weibullr< rm r rO s r8 test_weibullzTestRandomDist.test_weibull sy DI $V444( 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 d d S )Nr r rf r )rH )r rr r r r< r r r_ r` s r8 test_weibull_0z TestRandomDist.test_weibull_0 sX DI V^ab11128B<<@@@j&.C888888r: c t j | j t j dd }t j ddgddgddgg }t || d S ) Nr ra rf B rc rj r )r rr r r< rm r rO s r8 test_zipfzTestRandomDist.test_zipf sj DI t&111(RH F G % & & 67+++++r: N)\rw rx ry r rP rR r] rg rl rn rq rw ry 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/ r4 r6 r8 r: r r> parametrizer< r r rB rE rN r rR rW r\ r_ rb rd rl ro rs rv ry r{ r~ r r r r r r r r r r r r r r r r r r rz r: r8 rA rA sS ? ? ?? ? ?
E
E
E, , ,2 2 2$, , ,(& & &, @ @ @
E
E
E, , , , , , , , , , , , , , , 2 2 2&,9 ,9 ,9\9 9 9
/ / /& & & 0 0 068 8 8$, , ,6? ? ? , , ,? ? ?E E E&
@
@
@; ; ;
6 6 6? ? ?A A A? ? ?? ? ?F F F, , ,G G G? ? ?< < <, , ,4? ? ?= = =? ? ?? ? ?? ? ?, , ,( ( ( [ W|r|B';';R&LMM > > NM > , , ,1% 1% 1%f, , ,) ) )? ? ?*? ? ?
? ? ?< < <
L
L
L , , ,
E
E
E? ? ?? ? ?> > >? ? ?? ? ?? ? ?D D D? ? ?? ? ? ? ? ?? ? ?? ? ?D D D L L L,? ? ?& & & > > >
? ? ?? ? ?9 9 9
, , , , ,r: rA 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 )N[rC r` s r8 r zTestBroadcast.setup_method% s
r: c 8 t j | j d S r3 )r rr r` s r8 set_seedzTestBroadcast.set_seed( s DI r: c * dg}dg}t j }t j g d }| ||dz | }t ||d | |||dz }t ||d d S )Nr rc gl ?g_x
?g =*BN?rj r rL )r r r< rm r r )rS r r r r2 r5 s r8 r zTestBroadcast.test_uniform+ s cs.( 1 1 1 2 2
q$''!&'2>>>>
dQh''!&'2>>>>>>r: c dg}dg}dg}t j }t j g d }| ||dz | }t ||d t
t ||dz | | |||dz }t ||d t
t |||dz d S )Nr rc r\ )g-r@gYl@g-`w?rj r rL )r ra r< rm r r r r_ )rS r r bad_scalera r2 r5 s r8 rb zTestBroadcast.test_normal; s cD ( 0 0 0 1 1
a''!&'2>>>>j&#'9===
UQY''!&'2>>>>j&#y1}=====r: c dg}dg}dg}dg}t 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 ) Nrc r r\ rd )gnQVf?gVKB?g v`k?rj r rL )r r r< rm r r r r_ )rS rH rI bad_abad_br r2 r5 s r8 r zTestBroadcast.test_betaN s
C
C{( 1 1 1 2 2
a!eQ !&'2>>>>j$ 1555j$Au555
aQ !&'2>>>>j$q1u555j$51955555r: c dg}dg}t j }t j g d }| ||dz }t ||d t
t ||dz d S Nrc r\ )g.UgZ?gi Prq?gE
K?rj r rL )r r r< rm r r r r_ )rS r r r r2 r5 s r8 r z TestBroadcast.test_exponentiald s D ( ( 1 1 1 2 2
UQY''!&'2>>>>j+y1}=====r: c dg}dg}t j }t j g d }| ||dz }t ||d t
t ||dz d S r )r r r< rm r r r r_ )rS r bad_shape std_gammar2 r5 s r8 r z!TestBroadcast.test_standard_gammaq s D ) ( 1 1 1 2 2
519%%!&'2>>>>j)Y];;;;;r: c dg}dg}dg}dg}t 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 ) Nrc r r\ rd )g.UgZ?gi Prq?gE
K?rj r rL )r r r< rm r r r r_ )rS r r r r r r2 r5 s r8 r zTestBroadcast.test_gamma~ 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 }t j g d }| ||dz | }t ||d t
t ||dz | t
t ||dz | | |||dz }t ||d t
t |||dz t
t |||dz d S ) Nrc r r\ rd )gbyʜ?go?gT7@rj r rL )r r r< rm r r r r_ )rS rY rZ bad_dfnum bad_dfdenr r2 r5 s r8 r zTestBroadcast.test_f s D D H( 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 }t j g d }| ||dz || } t | |d t j t j |||t j gdz sJ 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 rj rk r r\ rd )gr^G"@g/
*@gц!@r rL ) r r[ r< rm r r r r^ r r r_ )
rS rY rZ rU r r bad_noncnonc_fr2 r5 s
r8 r\ z TestBroadcast.test_noncentral_f s' sC D 4$( 0 0 0 1 1
5$//!&'2>>>>vbhvveURVHqL A ABBCCCCCj&)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 r r rL )r r< rm r r[ r )rS r2 r5 s r8 test_noncentral_f_small_dfz(TestBroadcast.test_noncentral_f_small_df sY
( -/@ABB$S#qq999!&'2>>>>>>r: c dg}dg}t j }t j g d }| ||dz }t ||d t
t ||dz d S )Nrc r\ )g~}N??g$F?g'rS?rj r rL )r r r< rm r r r r_ )rS rT bad_dfr r2 r5 s r8 r z TestBroadcast.test_chisquare s S$ ( 0 0 0 1 1
26""!&'2>>>>j)VaZ88888r: c dg}dg}dg}dg}t 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 ) Nrc r r\ rd )g w "@g2 WR@gRY@rj r rL )r rV r< rm r r r r_ )rS rT rU r r nonc_chir2 r5 s r8 rW z'TestBroadcast.test_noncentral_chisquare s Ss4.( 0 0 0 1 1
"q&$''!&'2>>>>j(FQJ===j(BFH===
"dQh''!&'2>>>>j(FD1H===j(B1 =====r: c 6 dg}dg}t j }t j g d }| ||dz }t ||d t
t ||dz t
t t j |dz d S )Nrc r\ )gT
@g~Fl@gƚtp?rj r rL )r r r< rm r r r r_ )rS rT r tr2 r5 s r8 r z TestBroadcast.test_standard_t s S ( 0 0 0 1 1
26!&'2>>>>j!VaZ000j&"3VaZ@@@@@r: c dg}dg}dg}t j }t j g d }| ||dz | }t ||d t
t ||dz | | |||dz }t ||d t
t |||dz d S )Nr rc r\ )gW!@gOBg}?rj r rL )r r r< rm r r r r_ )rS rM r bad_kappar r2 r5 s r8 r zTestBroadcast.test_vonmises s SD ?( 1 1 1 2 2
"q&%((!&'2>>>>j(BFI>>>
"eai((!&'2>>>>j(B A
>>>>>r: c 6 dg}dg}t j }t j g d }| ||dz }t ||d t
t ||dz t
t t j |dz d S )Nrc r\ )g&8??geFX?gՋː?rj r rL )r ri r< rm r r r r_ )rS rH r ri r2 r5 s r8 rl zTestBroadcast.test_pareto s
C ( 0 0 0 1 1
A !&'2>>>>j&%!)444j&-;;;;;r: c 6 dg}dg}t j }t j g d }| ||dz }t ||d t
t ||dz t
t t j |dz d S r )r r r< rm r r r r_ )rS rH r r r2 r5 s r8 r zTestBroadcast.test_weibull% s
C.( 1 1 1 2 2
Q !&'2>>>>j'519555j&.%!)<<<<>>>j%333j&, :::::r: c dg}dg}dg}t j }t j g d }| ||dz | }t ||d t
t ||dz | | |||dz }t ||d t
t |||dz d S )Nr rc r\ )g:BKc?g3k?g~q?rj r rL )r r) r< rm r r r r_ )rS r r r r) r2 r5 s r8 r* zTestBroadcast.test_laplaceA s cD .( 2 2 2 3 3
q%((!&'2>>>>j'37I>>>
eai((!&'2>>>>j'3 A
>>>>>r: c dg}dg}dg}t j }t j g d }| ||dz | }t ||d t
t ||dz | | |||dz }t ||d t
t |||dz d S )Nr rc r\ )g'Zy?gdSO=?g3L1?rj r rL )r r! r< rm r r r r_ )rS r r r r! r2 r5 s r8 r" zTestBroadcast.test_gumbelT s cD ( 1 1 1 2 2
a''!&'2>>>>j&#'9===
UQY''!&'2>>>>j&#y1}=====r: c dg}dg}dg}t j }t j g d }| ||dz | }t ||d t
t ||dz | | |||dz }t ||d t
t |||dz t t j dd d d S )
Nr rc r\ )gy ?gJR?g
~&?rj r rL r< r ) r r. r< rm r r r r_ r )rS r r r r. r2 r5 s r8 r/ zTestBroadcast.test_logisticg s cD ?( 2 2 2 3 3
#'5))!&'2>>>>j(C!GY???
#uqy))!&'2>>>>j(CQ ???V_S#..44444r: c ( dg}dg}dg}t j }t j g d }| ||dz | }t ||d t
t ||dz | t
t t j |dz | | |||dz }t ||d t
t |||dz t
t t j ||dz d S )Nr rc r\ )gH"@gܮ @gvv :@rj r rL )r r3 r< rm r r r r_ )rS r1 r2 bad_sigmar3 r2 r5 s r8 r4 z TestBroadcast.test_lognormal{ s sD $ ( 0 0 0 1 1
4!8U++!&'2>>>>j)TAXyAAAj&"2D1HiHHH
4++!&'2>>>>j)T9q=AAAj&"2D)a-HHHHHr: c dg}dg}t j }t j g d }| ||dz }t ||d t
t ||dz d S )Nrc r\ )gݱlo?g/ӂ?gz~R?rj r rL )r rx r< rm r r r r_ )rS r r rx r2 r5 s r8 ry zTestBroadcast.test_rayleigh s D ?( 0 0 0 1 1
%!)$$!&'2>>>>j(IM:::::r: c dg}dg}dg}dg}t j }t j g d }| ||dz | }t ||d t
t ||dz | t
t ||dz | t
t t j |dz | t
t t j |dz | | |||dz }t ||d t
t |||dz t
t |||dz t
t |d d t
t |dd d S )
Nr rc r rd )g@m e?g`I߿?g|2?rj r rL r )r r r< rm r r r r_ )rS r1 r bad_meanr r r2 r5 s r8 r zTestBroadcast.test_wald sw u3D {( 0 0 0 1 1
dQh&&!&'2>>>>j$1 e<<<j$q)<<<j&+x!|UCCCj&+taxCCC
dEAI&&!&'2>>>>j$%!)<<<j$i!m<<<j$Q///j$S11111r: c 2 dg}dg}dg}dg}dg}|dz \ }}t 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 t
t |dd d
t
t |dd d
t
t |ddd d S ) Nrc rj r rk )g,P;bD @g%G @g,~S @r rL r r g 4@g 9@)r r r< rm r r r r_ ) rS r r r bad_left_one bad_mode_one bad_left_two bad_mode_twor r2 r5 s r8 r z TestBroadcast.test_triangular sY sss s %*QY" l&
( 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 ! ! ! j*c2s;;;j*c3<<<j*c3<<<<>>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 }t j g d }| ||dz }t || t t ||dz t t ||dz d S )Nrc r\ r )rc rc r rj )
r rP _poisson_lam_maxr r< rm r r r r_ )rS max_lamrn bad_lam_one bad_lam_twor r2 r5 s r8 ro zTestBroadcast.test_poisson
s $&&7cd {m .(999%%
q!!67+++j';?;;;j';?;;;;;r: c dg}dg}t 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 rc rj r> r? )
r r r< rm r r r r_ rA r )rS rH r r r2 r5 s r8 r zTestBroadcast.test_zipf s
C{(999%%
a!e67+++j$ 222
[
*
*
* < < *dBF 3 3 3 *dQ26N ; ; ; < < < < < < < < < < < < < < < < < > >&6 6 6, > > > < < <? ? ?,; ; ;,!F !F !FF? ? ? 9 9 9> > >, A A A ? ? ?& < < < = = = ; ; ; ? ? ?&> > >&5 5 5(I I I* ; ; ;2 2 24&= &= &=P; ; ;.? ? ?.
<
<
< < < < 7 7 7&9 &9 &9P < < < < .z sL
0
0
0A V86+=a+@+@!*D
E
E
E
0
0
0r: c 6 g | ]}| S rz )startr r s r8 r z-TestThread.check_function..| s q r: c 6 g | ]}| S rz )joinr/ s r8 r z-TestThread.check_function..} s a r: rk win32) threadingr' r< r r r% zipr rP intprh itemsizer# platformr r ) rS r, szout1out2r rT r+ r' s ` @r8 check_functionzTestThread.check_functions sF $$$$$$xTZ *R/00xTZ *R/00
0
0
0
0
0 TZ..
0
0
0 A 1
D)) / /DAq HV'**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 )N' r .)ra rC outs r8
gen_randomz*TestThread.test_normal..gen_random s |||//CHHHr: )r> r8 r; rS rA s r8 rb zTestThread.test_normal s2 0 0 0
J844444r: c : d }| |d d S )Nc \ | t j d |d< d S )Nr rN r .)r r< r r? s r8 rA z'TestThread.test_exp..gen_random s* ((rw{/C/C(DDCHHHr: rG rB rC rD s r8 test_expzTestThread.test_exp s5 E E E
J;77777r: c : d }| |d d S )Nc D | ddgdz d |d< d S )Nr rD r r> r .)r r? s r8 rA z/TestThread.test_multinomial..gen_random s* ((fX\(FFCHHHr: )r> r rB rC rD s r8 rE zTestThread.test_multinomial s5 G G G
J:66666r: N)rw rx ry r r; rb rH rE rz r: r8 r# r# m s_ + + +,5 5 5 8 8 8 7 7 7 7 7r: r# c & e Zd Zd Zd Zd Zd ZdS )TestSingleEltArrayInputc t j dg | _ t j dg | _ t j dg | _ d| _ d S )Nr rj rk r )r< rm argOneargTwoargThreetgtShaper` s r8 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 )r r r r r ri r ru rx r r r r r< rm rN r r rQ )rS funcs probfuncsr r@ s r8 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 )Nr r )r r ra r r r rV r r) r! r. r3 r r r r< rm rO rN r r rQ )rS rS rT r rO r@ s r8 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 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 )
r r[ r r rN rO rP r r rQ )rS rS r r@ s r8 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)rw rx ry r rU rW rZ rz r: r8 rL rL sP 3 3 3&3 3 36 3 3 3 3 3r: rL c t j d | \ }}}t t | } ||ddi}t |j t j d k d S )Nr r r r. )r rr getattrr rh r< )r9 fnamer* r' r r5 s r8 test_integer_dtyper^ sd
K "E4 A
Q
1
F FLBHSMM )*****r: c h t j d | \ }}}t t | } ||ddi}t j dk r| }t
j | t j
}t ||k d S )Nr r r r" )
r rr r\ r# r$ r% r& r' r( r< r r) r )r9 r] r* r' r r+ r, s r8 test_integer_repeatr` s
K "E4 A
!T
C
} llnn
."'**
+
+
5
5
7
7C C6Mr: c t j t 5 t j dddgd d d d n # 1 swxY w Y t j t 5 t j ddgdd d d d n # 1 swxY w Y t j t 5 t j ddgddgd d d d d S # 1 swxY w Y d S )Nrc r gffffff?)r rc r r )r r r_ r r rz r: r8 test_broadcast_size_errorrb s z " " 4 4C:F33334 4 4 4 4 4 4 4 4 4 4 4 4 4 4 z " " 2 2A&11112 2 2 2 2 2 2 2 2 2 2 2 2 2 2 z " " 9 9Ac
88889 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9s5 A AA$B
BB. CC Cc t j t d } | d | \ }}}|d d dk sJ ||d d }| | | d }t |d |d t |d d |d d t |d d |d d t |d
|d
t |d |d d S ) Nr rc )r Fr rB rC rD rE rF rG )
r< r rP r r
__reduce__r r r r )rs ctorr* state_arI state_bs r8 &test_randomstate_ctor_old_style_picklerh s2 wqzz * *Bq --//D$ 8| # # # # d2A2hAKKkkk''G )7?+CDDDww'.0@0GHHHww'.0@0GHHH%w{';<<<!77#344444r: c H t j d }|j }t j | t t j j j j t | sJ t j
}||u sJ d S rY ) r< r r rB r> r mtrand_rand_bit_generatorr; r= )r@ def_bgbg second_bgs r8
test_hot_swaprp s
Y
"
"1
%
%F BI ### bi&,;T"XX F FFF F ++--I
??????r: c t d }t j | t j d }t j d t j d }t
| t
| |d dk sJ |d d |d d k sJ |d d |d d k sJ d S ) Nr Fr rc rB r
rC inc)r
r< r r> r rr print)r@ rn rC r s r8 test_seed_alt_bit_genrt s qBI ### I u - -EINN1 ##5#11I %LLL ) !W , , , , >' "i&8&A A A A A >% Ig$6u$= = = = = = =r: c D t j }t d }t j | t j t d 5 t j | d d d d S # 1 swxY w Y d S )Nr zstate must be for a PCG64r ) r< r r r
r> r r r_ r )r@ rC rn s r8 test_state_error_alt_bit_genrv ) s I ! !E qBI ### z)D E E E # #
E"""# # # # # # # # # # # # # # # # # #s ( BB Bc t j d t j ddd }t d }|j }t j | t j d }|D ]}|| || k sJ t j d t j ddd }t j ||k rJ |j }|d d |d d k sJ |d d |d d k sJ d S ) Ń r @r Fr rC rr ) r< r rr rQ r
rC r> r r )r@ r rn rC state_directfieldpcg_valsr s r8 test_swap_workedr} 2 s9 INN5
9 Q , ,D qB HEI ###9&&e&44L 3 3U||E 222222INN5y GR00Hvdh&'''' 'I W g &%.*A A A A A W e $ '(:5(A A A A A A Ar: c F t j t d t j ddd }t j t d }| ddd }t
|| d S )Nrx r ry r )r< r r> r
rQ rP r )r@ singleton_valsrgnon_singleton_valss r8 %test_swapped_singleton_against_directr D s I e ---Y&&q'266N uU|| , ,BAw33#^44444r: )3r& r r# rt numpyr< r
numpy.testingr r r r r r r r
r numpy.randomr r
r r r5 r r~ r r6 fixturer9 r@ rJ rL r| r r r rA r r> r? r# rL r^ r` rb rh rp rt rv r} r rz r: r8 r s'
( ' ' ' ' ' ' ' % + c!1 2"+
28C==u #e$f$f)k&h,n"d a OO $f$f)k$f&h,n"d a O hy111, , 21,
- - -) ) ),? ,? ,? ,? ,? ,? ,? ,?^ > > > > > > > > 2) 2) 2) 2) 2) 2) 2) 2)jTJ TJ TJ TJ TJ TJ TJ TJn^+ ^+ ^+ ^+ ^+ ^+ ^+ ^+Bl
, l
, l
, l
, l
, l
, l
, l
,^H < H < H < H < H < H < H < H > > # # #B B B$5 5 5 5 5r: