U
|
P ±dìÊ ã @ sL d dl Z d dlZd dlZd dlZd dlmZ d dlmZm Z m
|
Z
|
m Z m Z m Z mZmZmZmZ d dlmZmZmZmZ ee Zd ddddd d
|
dd d d ddddddd ddddgdddddddd gZejdd d gd d d ZG d!d" d"ZG d#d$ d$Z G d%d& d&Z G d'd( d(Z G d)d* d*Z G d+d, d,Z G d-d. d.Z!G d/d0 d0Z"ej#j$ed1d2G d3d4 d4Z%G d5d6 d6Z&ej# 'd7e¡d8d9 Z(d:d; Z)d<d= Z*d>d? Z+ej# 'd@dAdAgg¡ej# 'd d d g¡dBdC Z,ej# 'dDdEdFg¡dGdH Z-ej# 'dDdEdFg¡ej# 'dIdJdKg¡ej# 'dLej.ejg¡dMdN Z/dOdP Z0dS )Qé N)Ú LinAlgError)
|
Úassert_Ú assert_raisesÚ assert_equalÚassert_allcloseÚ assert_warnsÚassert_no_warningsÚassert_array_equalÚassert_array_almost_equalÚsuppress_warningsÚIS_WASM)Ú GeneratorÚMT19937Ú SeedSequenceÚ RandomStateé
|
Z@bb1636883c2707b51c5b7fc26c6927af4430f2e0785a8c7bc886337f919f9edfé )Ú
|
key_sha256ÚposZ@ff682ac12bb140f2d72fba8d3506cf4e46817a0db27aae1683867629031d8d55iV )ÚseedÚstepsÚinitialÚjumpedi$<ñi8 Z@16b791a1e04886ccbbb4d448d6ff791267dc458ae599475d08d5cced29d11614i7 Z@a0110a2cf23b56be0feaed8f787a7fc84bef0cb5623003d75b26bdfa1c18002ci i|Ï2iàEm:iS43i5 iÿ Z@d306cf01314d51bd37892d874308200951a35265ede54d200f1e065004c3e9eaiþ Z@0e00ab449f01a5195a83b4aee0dfbc2ce8d46466a640b92e33977d2e42f777f8iÛ ÚmoduleTF)ZscopeÚparamsc C s | j S ©N)Úparam)Úrequest© r ú`d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/random/tests/test_generator_mt19937.pyÚendpoint& s r c @ s< e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd S )ÚTestSeedc C s< t td}t| d¡d t td}t| d¡d d S )Nr éè éß ì ÿÿ éD )r r r Úintegers©ÚselfÚsr r r Ú test_scalar, s zTestSeed.test_scalarc C s t ttd}t| d¡d t tt d¡}t| d¡d t tdg}t| d¡d t tdg}t| d¡d d S )Nr r" iÑ r r# r$ r% )r r Úranger r& ÚnpÚaranger' r r r Ú
|
test_array2 s zTestSeed.test_arrayc C s t td}t| d¡d d S )Nr é iê´z)r r r Ú
|
random_rawr' r r r Útest_seedsequence< s zTestSeed.test_seedsequencec C s t ttd t ttd d S )Nç à¿éÿÿÿÿ©r Ú TypeErrorr Ú
|
ValueError©r( r r r Útest_invalid_scalar@ s z TestSeed.test_invalid_scalarc C s2 t ttdg t ttdg t ttdddg d S )Nr2 r3 r/ éþÿÿÿì r4 r7 r r r Útest_invalid_arrayE s zTestSeed.test_invalid_arrayc C s t ttt d S r )r r6 r r r7 r r r Útest_noninstantized_bitgenK s z#TestSeed.test_noninstantized_bitgenN) Ú__name__Ú
|
__module__Ú __qualname__r* r. r1 r8 r; r< r r r r r! + s
|
r! c @ s e Zd Zdd Zdd ZdS )Ú TestBinomialc C sB t jddd}dD ]*}tt d|¡dk tt ||¡| qd S )Né Úint©Údtype)r ç à?r/ r )r, Úzerosr ÚrandomÚbinomialr )r( rF Úpr r r Ú test_n_zeroP s zTestBinomial.test_n_zeroc C s t ttjdtj d S )Nr/ )r r6 rG rH r, Únanr7 r r r Ú test_p_is_nanY s zTestBinomial.test_p_is_nanN)r= r> r? rJ rL r r r r r@ O s r@ c @ sL e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dS )ÚTestMultinomialc C s t dddg¡ d S )Néd çÉ?çé?©rG Ú multinomialr7 r r r Ú
|
test_basic_ s zTestMultinomial.test_basicc C s t ddddddg¡ d S )NrN rO rP ç rQ r7 r r r Útest_zero_probabilityb s z%TestMultinomial.test_zero_probabilityc C sZ t dt dd¡ ko dk n t ddd¡}t t d|k¡ t t |dk ¡ d S )Néûÿÿÿr3 é )r rG r& r, Úall©r( Úxr r r Útest_int_negative_intervale s $z*TestMultinomial.test_int_negative_intervalc C s ddg}t t d|t d¡¡jd t t d|t d¡¡jd t t d|t d¡¡jd t t d|ddg¡jd t t d|d¡jd t t d|t d¡¡jd tttjd|t d d S )NrE r/ ©r/ rA rA ©rA rA rA ©rA rA )
|
r rG rR r, Úuint32ÚshapeÚarrayr r5 Úfloat©r( rI r r r Ú test_sizek s ÿ ÿzTestMultinomial.test_sizec C s, t ttjdddg t ttjdddg d S )NrN çñ?rO 繿çÍÌÌÌÌÌì?©r r6 rG rR r7 r r r Útest_invalid_proby s z!TestMultinomial.test_invalid_probc C s2 t ttjdddg t ttjdgd ddg d S )Nr3 rP rO r rh r7 r r r Útest_invalid_n} s z TestMultinomial.test_invalid_nc C sx t d¡}|t |dd d
¡ }|dd d
}ttd}|jd|d}ttd}|jdt |¡d}t|| d S )Ng .@r/ é éÛ iUrN )Úpvals)r, r- Úsumr r rR Úascontiguousarrayr )r( rI rm rG Ú
|
non_contigÚcontigr r r Útest_p_non_contiguous s
|
z%TestMultinomial.test_p_non_contiguousc C sj t jddddddddddg
|
t jd}|| ¡ }ttd}d}tjt|d | d|¡ W 5 Q R X d S )Ng®Gáz®ï?çÖ&è .>rC rl z-[\w\s]*pvals array is cast to 64-bit floating©Úmatchr/ )
|
r, ra Úfloat32rn r r ÚpytestÚraisesr6 rR )r( rZ rm rG ru r r r Ú test_multinomial_pvals_float32 s ÿÿ z.TestMultinomial.test_multinomial_pvals_float32N) r= r> r? rS rU r[ rd ri rj rr ry r r r r rM ^ s
|
rM c @ s e Zd Zdd Zdd Zej dddg¡dd Zej d
|
d d d dg¡ej dddg¡ej ddddg¡dd Z dd Z
|
dd Z dd Z dS ) Ú TestMultivariateHypergeometricc C s
|
d| _ d S )Nií_ ©r r7 r r r Ú setup_method s z+TestMultivariateHypergeometric.setup_methodc C s¸ t ttjdd t ttjdddgd t ttjdddgd t ttjdddgd t ttjg d t ttjddgd dd
|
t tj¡}|j}||jj }t ttj|d dgd dd d S ) Nr é rA rk r3 r/ iÿÉ;ée rW Ú marginalsrN Úcount)
|
r r6 rG Úmultivariate_hypergeometricr, ÚiinfoÚint64ÚmaxrD Úitemsize)r( Z
|
int64_infoZ max_int64Zmax_int64_indexr r r Útest_argument_validation sF ÿ ÿ ÿ ÿ ÿ ÿ
|
ÿz7TestMultivariateHypergeometric.test_argument_validationÚmethodr r c C s t t| j}|jdddgd|d}t|dddg |jg d|d}t|g |jg dd|d}t|tjdtjd |jdddgd|d}t|dddg |jd ddgd|d}t|dddg dddddg}|j|t||d}t|| |jdd
|
d gd d|d}t|dd
|
d ggd d S ) Nr ©r r/ ©Úsizer )r/ r rC rA rk r r} rW é ) r r r r r r, Úemptyr rn )r( r rG rZ Úcolorsr r r Útest_edge_casesº s(
|
ÿ
|
ÿz.TestMultivariateHypergeometric.test_edge_casesÚnsampleé é é- é7 r rW ©rA rk iðI c C sØ t t| j}t ddddg¡}|j||||d}t|trJ|f|j }n
|
||j }t |j| t
|
|dk ¡ t
|
||k ¡ t |j ddtj||td t|trÔ|d
|
krÔt|jdd|| | ¡ d d d d S )Nr rW é r r r r3 ©Úaxis)Z
|
fill_valuerD é çü©ñÒMbP?g{®Gázt?)ÚrtolZatol)r r r r, ra r Ú
|
isinstancerB r` r r rX r rn Úfullr Úmean)r( r r r rG r ÚsampleÚexpected_shaper r r Útest_typical_casesÝ s(
|
ÿ
|
|
ÿ þz1TestMultivariateHypergeometric.test_typical_casesc C sf t t| j}|jdddgdddd}t dddgdddgdddgdddgdddgg¡}t|| d S ) Nrk r} rW r r rA r/ r ©r r r r r, ra r ©r( rG r Úexpectedr r r Útest_repeatability1ö s ÿ üz2TestMultivariateHypergeometric.test_repeatability1c C sf t t| j}|jdddgdddd}t ddd gd
|
d dgdd d gddd gddd gg¡}t|| d S )Nr é é2 rW r r r é é é é é é r é r¡ r¢ r r r Útest_repeatability2 s þ üz2TestMultivariateHypergeometric.test_repeatability2c C sf t t| j}|jdddgdddd}t dd d
|
gdd d gdddgdd d gd dd gg¡}t|| d S )Nr r¥ r¦ r rW r r rA rk r© r} r/ é r¡ r¢ r r r Útest_repeatability3 s þ üz2TestMultivariateHypergeometric.test_repeatability3N) r= r> r? r| r rw ÚmarkÚ parametrizer r r¤ r® r° r r r r rz s !
|
" rz c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
|
Ú TestSetStatec C sT d| _ tt| j | _| jj| _| jj| _| jd | jd d | jd d f| _d S )NéÒIÚ bit_generatorÚstateÚkeyr )r r r Úrgrµ r¶ Z legacy_stater7 r r r r| s
|
|
þzTestSetState.setup_methodc C s< | j jdd}| j| j_| j jdd}tt ||k¡ d S ©Nrk ©r )r¸ Ústandard_normalr¶ rµ r r, rX )r( ÚoldÚnewr r r Útest_gaussian_reset$ s
|
z TestSetState.test_gaussian_resetc C sL | j ¡ | jj}| j jdd}|| j_| j jdd}tt ||k¡ d S r¹ )r¸ r» rµ r¶ r r, rX )r( r¶ r¼ r½ r r r Ú test_gaussian_reset_in_media_res+ s
|
z-TestSetState.test_gaussian_reset_in_media_resc C s | j dd¡ d S ©NrE )r¸ Únegative_binomialr7 r r r Útest_negative_binomial6 s z#TestSetState.test_negative_binomialN)r= r> r? r| r¾ r¿ r r r r r r³ s r³ c @ sz e Zd ZejZeejej ej
|
ej ej ej ejejg Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Zdd Zdd Zdd Zdd Ze j dde dddd d d d g¡fd e d!d"d#d$d%d&d'g¡fd(e d!d#d)d*d+d,d-g¡fg¡d.d/ Z d0d1 Z!d2d3 Z"d4d5 Z#d6d7 Z$d8d9 Z%d:d; Z&d<d= Z'e j j(e j d>d?d@ejdAfd?dBej dCfdDdEej
|
dFfdGdHej dIfg¡dJdK Z)dLS )MÚ TestIntegersc C s t t| jd|td d S ©Nr/ ©r rD )r r5 Úrfuncrb )r( r r r r Útest_unsupported_typeC s z"TestIntegers.test_unsupported_typec C s" | j D ]}|tkrdn
|
t |¡j}|tkr0dnt |¡jd }|rL|d n|}tt| j|d |||d tt| j||d ||d tt| j||||d tt| jdd||d tt| j|d g|||d tt| j|g|d g||d tt| j|g|g||d tt| jddg||d qd S ©Nr rA r/ rÅ © ÚitypeÚboolr, r Úminr r r6 rÆ ©r( r ÚdtÚlbndÚubndr r r Útest_bounds_checkingF sF ÿ ÿ ÿÿ ÿ ÿ ÿ ÿz!TestIntegers.test_bounds_checkingc C sÊ | j D ]¾}|tkrdn
|
t |¡j}|tkr.dnt |¡j| }tt| j|d gd |gd ||d tt| j|gd |d gd ||d tt| j||gd ||d tt| jdgd d||d qd S rÈ rÉ rÍ r r r Útest_bounds_checking_array] s(
|
ÿ ÿ ÿ ÿz'TestIntegers.test_bounds_checking_arrayc C s0 | j D ]"}|tkrdn
|
t |¡j}|tkr0dnt |¡jd }|rL|d n|}| }|d }t| j||| d||d| t| j|g|| d||d| |}t| j||| d||d| t| j||| gd||d| || d }t| j||| d||d| t| j|g|| gd||d| qd S )Nr rA r/ r" ©r r rD ©rÊ rË r, r rÌ r r rÆ )r( r rÎ rÏ rÐ Zis_openÚtgtr r r Útest_rng_zero_and_extremesk sZ ÿÿ ÿÿ ÿÿ ÿÿ ÿÿ ÿþz'TestIntegers.test_rng_zero_and_extremesc C s¼ d}| j D ]ª}|tkr dn
|
t |¡j}|tkr4dnt |¡jd }|rP|d n|}|d }t| j|g|d g||d| t| j|g| |d g| |d| t| j|g| |d g| ||d| |}t| j|g|d g||d| t| j|g| |d g| |d| t| j|g| |d g| ||d| || d }t| j|g|d g||d| t| j|g| |d g| |d| t| j|g| |d g| ||d| q
|
d S )Nr" r rA r/ ©r rD rC rÔ )r( r r rÎ rÏ rÐ rÕ r r r Ú test_rng_zero_and_extremes_array
s ÿÿ ÿÿ ÿÿ ÿÿ ÿÿ ÿÿ ÿÿ ÿÿ ÿÿz-TestIntegers.test_rng_zero_and_extremes_arrayc C s | j D ]}|tkrdn
|
t |¡j}|tkr.dnt |¡jd }|rJ|d n|}z| j||||d W q tk
|
r } ztdt | W 5 d }~X Y qX qd S ©Nr rA r/ rÅ zMNo error should have been raised, but one was with the following message:
|
|
%s©
|
rÊ rË r, r rÌ r rÆ Ú ExceptionÚAssertionErrorÚstr©r( r rÎ rÏ rÐ Úer r r Útest_full_range¤ s
|
þz TestIntegers.test_full_rangec C s¦ | j D ]}|tkrdn
|
t |¡j}|tkr.dnt |¡jd }|rJ|d n|}z | j|gd |g||d W q tk
|
r } ztdt | W 5 d }~X Y qX qd S rÙ rÚ rÞ r r r Útest_full_range_array³ s
|
þz"TestIntegers.test_full_range_arrayc C s t t }| jdd
D ]F}dD ]<}| jd|| d||d}t| ¡ |k t| ¡ dk q q| jdd| d|td}t| ¡ dk t| ¡ dk d S )Nr/ )r} r é rA i rÓ r )r r rÊ rÆ r r rÌ rË )r( r rG rÎ rÐ Úvalsr r r Útest_in_bounds_fuzz s
|
ÿÿz TestIntegers.test_in_bounds_fuzzc
|
C sÜ | j D ]Ð}|tkrdn
|
t |¡j}|tkr.dnt |¡jd }|rJ|d n|}d}ttd}|j|||||d}ttd}|j|g|g|||d}ttd}|j|g| |g| |||d} t || t || qd S )Nr rA r/ r" éÒ rÓ )
|
rÊ rË r, r rÌ r r r r& r )
|
r( r rÎ rÏ rÐ r rG ZscalarZ scalar_arrayra r r r Útest_scalar_array_equivÒ s0
|
ÿ ÿ ÿ ÿ
|
z$TestIntegers.test_scalar_array_equivc
|
C sô dddddddddd }| j dd
D ]v}ttd}tjd krZ|jd
|
d | d ||d }n |jd
|
d | d ||d ¡ }t |¡ ¡ }t
|
|t |¡j |k q&ttd}|jd
|
d| d |td t j¡}t |¡ ¡ }t
|
|t t¡j |k d S )NZ@053594a9b82d656f967c54869bc6970aa0358cf94ad469c81478459c6a90eee3Z@54de9072b6ee9ff7f20b58329556a46a447a8a29d67db51201bf88baa6e4e5d4Z@d3a0d5efb04542b25ac712e50d21f39ac30f312a5052e9bbb1ad3baa791ac84bZ@14e224389ac4580bfbdccb5697d6190b496f91227cf67df60989de3d546389b1Z@0e203226ff3fbbd1580f15da4621e5f7164d0d8d6b51696dd42d004ece2cbec1) rË Úint16Úint32r Úint8Úuint16r_ Úuint64Úuint8r/ rå Úlittler r¯ r" rÓ rA )rÊ r r ÚsysÚ byteorderr& ÚbyteswapÚhashlibÚsha256Ú hexdigestr r, rD ÚnamerË Úviewré )r( r rÕ rÎ rG ÚvalÚresr r r Útest_repeatabilityç s: ø
|
|
ÿÿ
|
ÿÿz TestIntegers.test_repeatabilityc C sâ | j D ]Ö}|ttjfkr dn
|
t |¡j}|ttjfkr:dnt |¡jd }|rV|d n|}ttd}|j ||d||d}ttd}|j |gd |||d}t
|
|| ttd}|j |gd |gd ||d}t
|
|| qd S )Nr rA r/ rå r" rÓ rÅ ) rÊ rË r, Úbool_r rÌ r r r r& r )r( r rÎ rÏ rÐ rG rö Zval_bcr r r Ú test_repeatability_broadcasting
|
s&
|
" ÿ ÿ
|
ÿz,TestIntegers.test_repeatability_broadcastingzbound, expectedr$ i uÑ iÉ,YQiåùXgi$#°Pl bA i{òÈEl /YÄi r: i uÑ iÊ,YQiæùXgi%#°Pl bA i|òÈEl 0YÄi ì l bA l 1YÄi i÷{,miù
uHl îHM c C sJ d t |fD ]8}ttd}|j||d}t||d k r:|n|d q d S )Nrå rº r )Úlenr r r& r )r( Úboundr£ r rG rZ r r r Ú!test_repeatability_32bit_boundary! s
|
z.TestIntegers.test_repeatability_32bit_boundaryc C sâ t dddgdddgddd ggd
|
d d gd ddgdddggdddgdddgdddggd d d gd d d!gd"d#d$ggd%d&d'gd(d)d*gd+d,d-ggg¡}d.D ]L}ttd/}|jd0gd1gd2ggd3d4d5g|d6}t||d k rÒ|n|d1 qd S )7NiÜ
|
¼`l ÓY¡/ iü0èbi)H{Ti«P-iTÿÒql C iZ'l !Îg l ©k40 l n2c l ÅDBJ iÒ|à(iy +l à_&A ip@ýGiG^l J|xK l ü
|
f l Dmv l ©E"g iÐ>{l wT2L iñJQiQÓ//itÉLol Øk¦ i\si¼ÿJiÑiéÿ=l }D l E9l l
|
+x i#J¨l bË iºð¼0i7ùOl rn ix¦iiµ½Ai.¸®tl q/q i;-/i
á)N)rW rk rk i90 r3 r r/ r$ r: rû rº )r, ra r r r& r )r( Údesiredr rG rZ r r r Ú.test_repeatability_32bit_boundary_broadcasting0 s8 þþþþþô þz;TestIntegers.test_repeatability_32bit_boundary_broadcastingc
|
C s4 t jddd fdddft jdi}|D ]}|| D ]ú}|\}}|| }t |gd g¡}t |gd ¡}tttj||||d tttj||||d tttj||||d tttj||||d t j|gd gtd
|
} t j|gd td
|
}
|
tttj| |||d tttj||
|
||d tttj| |
|
||d q2q$d S ) Nr rA éA )r3 ì ©r r ©r r ))r l )lûÿÿÿ r r r )ìûÿÿÿ r r rÅ rC ) r, rë r ra r r6 rG r& Úobject) r( r ZconfigsrD ÚconfigÚlowÚhighZlow_aZhigh_aZlow_oZhigh_or r r Ú&test_int64_uint64_broadcast_exceptionsG sP ÿ
|
ÿ ÿ ÿ ÿ ÿ ÿ ÿz3TestIntegers.test_int64_uint64_broadcast_exceptionsc C sd t j}t t j¡j}t t t j¡j¡}t t t j¡jd | ¡}tj||||d}t|| d S rÄ )r, r r r rë rG r& r )r( r rÎ rÕ rÏ rÐ Úactualr r r Ú test_int64_uint64_corner_casec s z*TestIntegers.test_int64_uint64_corner_casec C s
|
| j D ]x}|tkrdn
|
t |¡j}|tkr.dnt |¡jd }|rJ|d n|}|tkr\tjn|}| j||||d}t|j | qtt
|
tj j fD ]v}|tkrdn
|
t |¡j}|tkr¶dnt |¡jd }|rÒ|d n|}| j||||d}t |dröttt|| qd S )Nr rA r/ rÅ rD )rÊ rË r, r rÌ r rù rÆ r rD rB ÚcompatÚlongÚhasattrrÜ Útype©r( r rÎ rÏ rÐ r r r r Ú test_respect_dtype_singletonz s
|
z)TestIntegers.test_respect_dtype_singletonc C s² | j D ]¦}|tkrdn
|
t |¡j}|tkr.dnt |¡jd }|rJ|d n|}|tkr\tjn|}| j|g|g||d}t|j | | j|gd |gd ||d}t|j | qd S rÈ )
|
rÊ rË r, r rÌ r rù rÆ r rD r r r r Útest_respect_dtype_array s
|
ÿz%TestIntegers.test_respect_dtype_arrayc C s¦ | j D ]}| jddd||d}|jdks,t|j|ks:t| jddd||djdksXtttjddddjd ttjddddjd ttjddddjd qd S )Nr ©rk r r} rÅ éöÿÿÿ©r rº r )rÊ rÆ r` rÜ rD r rG r& )r( r rÎ r r r r Útest_zero_size s
|
ÿÿÿzTestIntegers.test_zero_sizec C s> t jdkrdnd}t t¡ tjddd|d W 5 Q R X d S )NÚbigz<i4z>i4r éÈ r r× )rî rï rw rx r6 rG r& )r( Zother_byteord_dtr r r Útest_error_byteorderª s z!TestIntegers.test_error_byteorderz sample_size,high,dtype,chi2maxi@KL rW g @_@r© g Àb@i iÄ g È©@iðúé g d¹@c
|
C sN t j|||d}tj|dd\}}|| }|| d | ¡ } | |k sJtd S )Nr× T)Z return_countsrA )rG r& r, Úuniquern rÜ )
|
r( Z sample_sizer rD Zchi2maxÚsamplesÚvaluesÚcountsr£ Zchi2r r r Ú$test_integers_small_dtype_chisquared° s
|
|
z1TestIntegers.test_integers_small_dtype_chisquaredN)*r= r> r? rG r& rÆ rË r, ré rì rç rê rè r_ r rë rÊ rÇ rÑ rÒ rÖ rØ rà rá rä ræ rø rú rw r± r² ra rþ r r
|
r r r r r Úslowr r r r r rà < sl ÿ # ÿ ÿ ÿüþ
|
ýÿrà c @ sT e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
e j dejejfejejfg¡dd Zdd Zdd Zdd Zdd Zd d Zd d Zd d! Zd"d# Zd$d% Z d&d' Z d(d) Z d*d+ Z d,d- Z d.d/ Z!d0d1 Z"d2d3 Z#d4d5 Z$d6d7 Z%d8d9 Z&d:d; Z'd<d= Z(d>d? Z)d@dA Z*dBdC Z+dDdE Z,e j dFe-e.g¡e j dGdHe /dIdJdKdLdMdNgdOdPdQdRdSdTgg¡fdKe /dTdSdQdLdMdNgdKdJdOdIdPdRgg¡fdSe /dRdIdPdKdQdSgdNdLdMdTdOdJgg¡fg¡dUdV Z0dWdX Z1dYdZ Z2e j d[d\dRg¡d]d^ Z3d_d` Z4dadb Z5dcdd Z6dedf Z7dgdh Z8didj Z9dkdl Z:dmdn Z;dodp Z<dqdr Z=e j j>dsdt Z?dudv Z@dwdx ZAdydz ZBd{d| ZCd}d~ ZDdd ZEdd ZFdd ZGd
d ZHdd ZIdd ZJdd ZKdd ZLdd ZMdd ZNdd ZOdd ZPe j de Qdd¡dejRdg¡dd ZSdd ZTe j jUeVd d¡e j d¢d£d¤d¥g¡d¦d§ ZWe j d¨dKgd©ggfdªgdSggfg¡d«d¬ ZXe j d¢d£d¤d¥g¡dd® ZYd¯d° ZZd±d² Z[d³d´ Z\dµd¶ Z]d·d¸ Z^d¹dº Z_d»d¼ Z`d½d¾ Zad¿dÀ ZbdÁd ZcdÃdÄ ZddÅdÆ ZedÇdÈ ZfdÉdÊ ZgdËdÌ ZhdÍdÎ ZidÏdÐ ZjdÑdÒ ZkdÓdÔ ZldÕdÖ Zmd×dØ ZndÙdÚ ZodÛdÜ ZpdÝdÞ Zqdßdà Zrdádâ Zsdãdä Ztdådæ Zudçdè Zvdédê Zwdëdì Zxdídî Zydïdð Zzdñdò Z{dódô Z|dõdö Z}d÷dø Z~e j dùdúdûg¡düdý Ze j dþdÿej d ejdg¡e j dùdddSddûg¡dd Zdd Zd d
|
Zd d Zd d Z
dHS ( ÚTestRandomDistc C s
|
d| _ d S )Nr´ r{ r7 r r r r| Æ s zTestRandomDist.setup_methodc C sH t t| j}|jdddd}t ddgddgd d
|
gg¡}t|| d S ) Néÿÿÿéc ©rk rA rº é°ÿÿÿéÈÿÿÿé) é% éÿÿÿiðÿÿÿ©r r r r& r, ra r ©r( rG r rÿ r r r Ú test_integersÉ s z TestRandomDist.test_integersc C sR t t| j}|jdddtjd}tjddgddgd d
|
ggtjd }t|| d S ) Nr r$ r% r× r é éF éD r r( rC )r r r r& r, r_ ra r r, r r r Útest_integers_maskedÏ s "z#TestRandomDist.test_integers_maskedc C sJ t t| j}|jddddd}t ddgdd gd
|
d gg¡}t|| d S ) Nr# r$ r% T)r r r& r' r( é& r* iñÿÿÿr+ r, r r r Útest_integers_closed× s z#TestRandomDist.test_integers_closedc C s: t jt d¡jt d¡jdd}t d¡j}t|| d S )NÚlT©r )rG r& r, r r r )r( r rÿ r r r Útest_integers_max_intÝ s
|
ÿ z$TestRandomDist.test_integers_max_intc C sn t t| j}| d¡}t ddgddgddgg¡}t||dd t t| j}| ¡ }t||d
|
dd d S ) Nr% gG0ðÔ¸?g±Ã¦û£æ?g 2 ïµ?gu Aè?gòH
|
Ô>Hå?g;dEåæ?r« ©Údecimalr ©r r r rG r, ra r
|
r, r r r Ú test_randomé s
|
|
þzTestRandomDist.test_randomc C sF t t| j}| d¡}t ddgddgddgg¡}t||dd d S )
|
Nr% çCø¼ðÔ¸?gP¨û£æ?gtXÈùîµ?g81â
|
Aè?gÚrOÓ>Hå?g×88dEåæ?r© r7 r9 r, r r r Útest_random_floatõ s
|
|
þz TestRandomDist.test_random_floatc C s2 t t| j}|jtjd}d}t||dd d S )NrC r; r© r7 )r r r rG r, rv r
|
r, r r r Útest_random_float_scalarý s z'TestRandomDist.test_random_float_scalarzdtype, uint_view_typec C sN t t| j}|jd|d}t | |¡d@ ¡}d| k rDdk sJn td S )Nr rC r/ i$^ i,e )r r r rG r, Z count_nonzerorõ rÜ )r( rD Zuint_view_typerG r Znum_ones_in_lsbr r r Ú test_random_distribution_of_lsb s z.TestRandomDist.test_random_distribution_of_lsbc C s t ttjdd d S )Nrè rC )r r5 rG r7 r r r Ú test_random_unsupported_type s z+TestRandomDist.test_random_unsupported_typec C s@ t t| j}| dd¡}tjddddgtjd}t|| d S )Nr} r rA rC ©r r r Úchoicer, ra r r r, r r r Útest_choice_uniform_replace s z*TestRandomDist.test_choice_uniform_replacec C sL t t| j}|jddddddgd}tjddddgtjd}t|| d S )Nr} çÙ?ç¹?©rI r r/ rC r@ r, r r r Ú test_choice_nonuniform_replace s z-TestRandomDist.test_choice_nonuniform_replacec C sn t t| j}|jdddd}tjdddgtjd}t|| |jddddd}tjdtjd}t|| d S ) Nr} rk F©ÚreplacerA r rC )rH Úshuffle) r r r rA r, ra r r r- r, r r r Ú test_choice_uniform_noreplace s
|
z,TestRandomDist.test_choice_uniform_noreplacec C sL t t| j}|jdddddddgd}tjdd dgtjd
|
}t|| d S ) Nr} rk FrD ç333333Ó?rE ©rH rI r rA rC r@ r, r r r Ú test_choice_nonuniform_noreplace' s z/TestRandomDist.test_choice_nonuniform_noreplacec C sB t t| j}| ddddgd¡}t ddddg¡}t|| d S )NÚaÚbÚcÚdr} ©r r r rA r, ra r r, r r r Útest_choice_noninteger- s z%TestRandomDist.test_choice_nonintegerc C s\ t t| j}| ddgddgddgddggd¡}t ddgddgddgg¡}t|| d S ) Nr r/ rA rk r} rW r¯ r© rR r, r r r Ú)test_choice_multidimensional_default_axis3 s $ z8TestRandomDist.test_choice_multidimensional_default_axisc C s^ t t| j}|jddgddgddgddggddd }t dgdgdgdgg¡}t|| d S )
|
Nr r/ rA rk r} rW r¯ r© r rR r, r r r Ú(test_choice_multidimensional_custom_axis9 s (z7TestRandomDist.test_choice_multidimensional_custom_axisc C s* t j}tt|dd tt|dd tt|g d tt|ddddgdddgddggd tt|ddgdd d d
|
gd tt|ddgdd d gd tt|ddgdd d gd tt|dddgdd d tt|dddgdd d tt|dddgdd d tt|dddgdd d tt|dddgdd dddgd d S )Nr3 rk ç @r/ rA r} ç Ð?rE rC rO re rf FrG r9 )r3 )r3 r/ r rL )rG rA r r6 )r( r r r r Útest_choice_exceptions? s$ ÿ ÿz%TestRandomDist.test_choice_exceptionsc C s ddg}t t tjddd¡ t t tjddd¡ t t tjdd|d¡ t t tjdd|d¡ t t tjddgdd¡ t tjd gddd k t ddg¡}tjdtd }||d
|
< t tj|dd|k t }t t tjd|dd¡ t t tjd|dd¡ t t tjd|d|d¡ t t tjd|d|d¡ t t tjddg|dd¡ t tjd g|ddj d
|
k t ddg¡}tjdtd }||d
|
< t tj||dd
|
¡ |k d }ddddd d g}t tjd|ddj | t tjd|ddj | t tjd|d|dj | t tjd|d|dj | t tjt d¡|ddj | t tjd
|
d
|
ddj d t tjd
|
dd
|
dj d t tjddd
|
dj d t tjd
|
d
|
dj d t tjg ddj d t tjddgddj d tttjg d d S )NrD rg rA TrG FrL r/ rC r r rC rO r¯ r rº r r r rN rO )r r, ZisscalarrG rA ra r r ÚtupleÚndimÚitemr r` r- r& r r6 )r( rI rN Úarrr) r r r Útest_choice_return_shapeQ sL ÿz'TestRandomDist.test_choice_return_shapec C s0 t dddg¡}d d d g}tttj||d d S )Né* r/ rA rE )r, ra r r6 rG rA )r( rN rI r r r Ú test_choice_nan_probabilities s
|
z,TestRandomDist.test_choice_nan_probabilitiesc C s t d¡d }d|dd d
< tt| j}|jdd|d d d
d}tt| j}|jddt |d d d
¡d}t|| d S )Nr rW rV r/ rA rk rE )r, Úonesr r r rA ro r )r( rI rG rp rq r r r Ú test_choice_p_non_contiguous s z+TestRandomDist.test_choice_p_non_contiguousc C s t d¡d }t dd¡}|jt jks*ttjdddd}|jt jksJttjdd|d}|jt jksjttjdd|dd}|jt jkstd S )Nr} g @rA FrG rE )rI rH )r, r` rG rA rD r rÜ )r( rI r r r r Útest_choice_return_type s z&TestRandomDist.test_choice_return_typec C sZ d}t t| j}|jdddd}tjdkr4| ¡ }t | t
|
j ¡¡ ¡ }t ||k d S )NZ@4266599d12bfcfb815213303432341c06b4349f5455890446578877bb322e222é' r FrG rí )r r r rA rî rï rð rñ rò rõ r, ré ró r )r( Z choice_hashrG r r÷ r r r Útest_choice_large_sample s
|
z'TestRandomDist.test_choice_large_samplec C s* t t| j}| d¡}d}t|| d S )Nr s
|
ðÔá 8%Ý)r r r Úbytesr r, r r r Ú
|
test_bytes¢ s
|
zTestRandomDist.test_bytesc C s® dd dd dd dd dd dd dd d d d
|
d d d d d f D ]b}t t| j}|d dddddddddg
|
}| |¡ |}|dd ddddddddg
|
}t|| qFd S )Nc S s
|
t g ¡S r )r, ra ©rZ r r r Ú<lambda>« ó z-TestRandomDist.test_shuffle.<locals>.<lambda>c S s | S r r rg r r r rh ¬ ri c S s t | ¡ t j¡S r )r, ÚasarrayÚastyperé rg r r r rh ri c S s t | ¡ t j¡S r )r, rj rk rv rg r r r rh ® ri c S s t | ¡ t j¡S r )r, rj rk Z complex64rg r r r rh ¯ ri c S s t | ¡ t¡S r )r, rj rk r rg r r r rh ° ri c S s dd | D S )Nc S s g | ] }||fqS r r ©Ú.0Úir r r Ú
|
<listcomp>± s úATestRandomDist.test_shuffle.<locals>.<lambda>.<locals>.<listcomp>r rg r r r rh ± ri c S s t dd | D ¡S )Nc S s g | ] }||gqS r r rl r r r ro ² s rp )r, rj rg r r r rh ² ri c S s t | | g¡jS r )r, ÚvstackÚTrg r r r rh ³ ri c S s* t dd | D dtfdtfg¡ t j¡S )Nc S s g | ] }||fqS r r rl r r r ro µ s rp rN rO )r, rj rB rõ Zrecarrayrg r r r rh µ s ÿþc S s( t dd | D dtdfdt jdfg¡S )Nc S s g | ] }||fqS r r rl r r r ro ¹ s rp rN ©r/ rO )r, rj r rè rg r r r rh ¹ s
|
ÿÿr/ rA rk r} rW r¯ r© r r r )r r r rI r )r( ÚconvrG Úalistr rÿ r r r Ú test_shuffle¨ s$ ò
|
zTestRandomDist.test_shufflec C s¢ t t| j}t d¡ d¡}|j|dd t ddddgdd d
|
d gd d ddgddddgg¡}t|| t t| j}t d¡ d¡}|j|dd t|| d S )Nrâ ©r} r} r/ r r rk rA r} r© rW r¯ r é r r r r« rª r r3 ) r r r r, r- ÚreshaperI ra r r, r r r Útest_shuffle_custom_axisà s
|
|
|
ý
|
z'TestRandomDist.test_shuffle_custom_axisc C sT t t| j}t g ¡ d¡}dD ],}t g ¡ d¡}|j||d t|| q"d S )N)r r¯ )r r/ r )r r r r, ra ry rI r )r( rG rÿ r r r r r Ú test_shuffle_custom_axis_emptyÑ s z-TestRandomDist.test_shuffle_custom_axis_emptyc C s^ t d¡ dd¡}| ¡ }tt| j}|j|dd tt| j}| |j¡ t || d S )Nr rA r r/ r )
|
r, r- ry Úcopyr r r rI rr r )r( Úy1Úy2rG r r r Útest_shuffle_axis_nonsquareÙ s z*TestRandomDist.test_shuffle_axis_nonsquarec C sÀ t j t tdd¡d d d¡}t j t d¡d d d¡}| ¡ }| ¡ }tdD ]d}t |¡ t t
|
|j |j t
|
|j |j t |¡ t t
|
|j |j t
|
|j |j qVd S )Nr )rW r} rk r/ r3 r¦ ) r, ÚmaZ masked_valuesry r+ r- r| rG rI r ÚsortedÚdataÚmask)r( rN rO Za_origZb_origrn r r r Útest_shuffle_maskedâ s "
|
ÿ
|
ÿz"TestRandomDist.test_shuffle_maskedc C sÀ t t| j}t d¡}ttj|j|d t d¡ d¡}ttj|j|d tt |j|t
|
ddd dddgddd gg}tt |j|d t d¡}tt |j| t d
|
¡}ttj|j|d d S ) Nr r/ r ©rk rk rk rA r} rW r¯ r% )r r r r, r- r Ú AxisErrorrI ry r5 ÚsliceÚNotImplementedErrorra r` ©r( rG r\ r r r Útest_shuffle_exceptionsð s
|
|
|
z&TestRandomDist.test_shuffle_exceptionsc C sH t t| j}t d¡}d|j_tjt dd |
|
|¡ W 5 Q R X d S )NrW Fú read-onlyrt ) r r r r, rF ÚflagsÚ writeablerw rx r6 rI )r( rG rN r r r Útest_shuffle_not_writeableÿ s
|
|
z)TestRandomDist.test_shuffle_not_writeablec C s t t| j}ddddddddd d
|
g
|
}| |¡}ddd dd
|
dddddg
|
}t|| t t| j}t ddddddddd d
|
g
|
¡j}| |¡}t|t |¡j d }ttj |j| d }ttj |j| t t| j}d }dd
|
ddd dddddg
|
}| |¡}t|| d S )Nr/ rA rk r} rW r¯ r© r r r Zabcdg333333ó?r )
|
r r r Ú permutationr r, Z
|
atleast_2drr r r ) r( rG ru r rÿ Zarr_2dZ bad_x_strZ bad_x_floatZ integer_valr r r Útest_permutation s$
|
|
|
|
z TestRandomDist.test_permutationc C s t d¡ d¡}t ddddgddd d
|
gd d d dgddddgg¡}tt| j}|j|dd}t|| tt| j}|j|dd}t|| d S )Nrâ rw r rk r/ rA r} r© rW r¯ r rx r r r r« rª r r r3 ) r, r- ry ra r r r r r )r( rN rÿ rG r r r r Ú test_permutation_custom_axis s
|
|
|
ý
|
z+TestRandomDist.test_permutation_custom_axisc C sh t t| j}t d¡}ttj|j|d t d¡ d¡}ttj|j|d tt |j|t
|
ddd d S )Nr r/ r r
rk rA ) r r r r, r- r r r ry r5 r r r r r Útest_permutation_exceptions, s
|
z*TestRandomDist.test_permutation_exceptionsrD zaxis, expectedNrk r© r r r rx r r} rA rW r/ r¯ c C s t t| j}t d¡ dd¡ |¡}|j|||d t|| t t| j}t d¡ dd¡ |¡}|j||d}|j |kst
|
t|| d S )Nr rA r¯ ©r Úoutr ) r r r r, r- ry rk Úpermutedr rD rÜ )r( rD r r£ rG rZ Úyr r r Ú test_permuted4 s
|
z TestRandomDist.test_permutedc C s t t| j}t d¡ dd¡}| ¡ }|d d
d d d
f }|j|d|d}t dddd gd
|
d dd gg¡}t || ||d d
d d d
f< t || d S ) Né rA rx rk r/ r r r r¯ r r r§ )
|
r r r r, r- ry r| r ra r )r( rG Zx0Úx1rZ r r£ r r r Útest_permuted_with_stridesH s
|
ÿ
|
z)TestRandomDist.test_permuted_with_stridesc C s t g ¡}t|g d S r )rG r r )r( r r r r Útest_permuted_emptyU s
|
z"TestRandomDist.test_permuted_emptyÚoutshaper c C sL t dddg¡}t j||jd}tjtdd tj||d W 5 Q R X d S )Nr/ rA rk rC z
|
same shapert ©r ) r, ra rF rD rw rx r6 rG r )r( r rN r r r r Ú"test_permuted_out_with_wrong_shapeY s z1TestRandomDist.test_permuted_out_with_wrong_shapec C sH t jdt jd}t d¡}tjtdd tj|d|d W 5 Q R X d S )N)rk rW rC z Cannot castrt r/ r ) r, rF rè r` rw rx r5 rG r )r( r rZ r r r Ú!test_permuted_out_with_wrong_type` s
|
z0TestRandomDist.test_permuted_out_with_wrong_typec C s@ t d¡}d|j_tjtdd tj|d|d W 5 Q R X d S )N)rA rW Fr rt r/ r ) r, rF r r rw rx r6 rG r rY r r r Útest_permuted_not_writeablef s
|
z*TestRandomDist.test_permuted_not_writeablec C sL t t| j}|jdddd}t ddgddgd d
|
gg¡}t||d d d S ) NrD rg r% rº gWâ<$ÅÝ=g¾^
|
ðº=gð/:Ð?gÿñÖG!Gc>g©À'ë`2?gåÂUÆ?r« r7 )r r r Úbetar, ra r
|
r, r r r Ú test_betal s þÿzTestRandomDist.test_betac C sp t t| j}|jdddd}t ddgddgdd gg¡}t|| t t| j}| dd¡}d}t|| d S )
|
Ng Zd;ßY@gÉv¾/Ý?r% rº r^ r( é0 é, r¦ )r r r rH r, ra r r, r r r Ú test_binomialu s
|
þ
|
z TestRandomDist.test_binomialc C sJ t t| j}|jddd}t ddgddgdd gg¡}t||d
|
d d S ) Nr¦ r% rº gmÉE~@@g¹
|
1ÏC@gÚrÓ³L@gìm%x¬L@g<kÿÝQ¶K@gótªâµK@rª r7 )r r r Ú chisquarer, ra r
|
r, r r r Útest_chisquare s
|
þz TestRandomDist.test_chisquarec C sÄ t t| j}t ddg¡}|j|dd}t ddgddggd d
|
gd d ggd dgddggg¡}t||dd t ddg¡}tt|j| t t| j}t ddg¡}| |¡}t||d dd d S )Nç I<ÝI@ç÷É EZßC@r% rº g28\há?gòÉG/Ý?g ©Aëpâá?gæ|) ;Ü?gùLXËâ?gfOáhôÛ?góë·Wá?g (ÞPãÜ?gz$Q¿ùß?gÄmWt à?gæKÂà?g4ÞÄhézÞ?r« r7 çHáz®Gá?ç¼Ø²Ò¼r ) r r r r, ra Ú dirichletr
|
r r6 )r( rG Úalphar rÿ Z bad_alphar r r Útest_dirichlet s(
|
ÿÿÿü
|
z TestRandomDist.test_dirichletc C sº t ddg¡}tt |t d¡¡jd tt |t d¡¡jd tt |t d¡¡jd tt |ddg¡jd tt |d¡jd tt |t d¡¡jd tttj|t d d S )Nr¨ r© r/ r\ rA r] r^ )
|
r, ra r rG r¬ r_ r` r r5 rb rc r r r Útest_dirichlet_size s z"TestRandomDist.test_dirichlet_sizec C s t ddg¡}tttj| tttjddgg tttjdgdgg tttjdgdggdgdggg tttjt ddgddgg¡ d S )Nrª r« rW r/ )r, ra r r6 rG r¬ )r( r r r r Útest_dirichlet_bad_alpha© s "z'TestRandomDist.test_dirichlet_bad_alphac C sj t dddg¡}|d d d
}tt| j}|j|dd}tt| j}|jt |¡dd}t|| d S )Nr¨ g ð¿r© rA r% rº )r, ra r r r r¬ ro r
|
)r( rN r rG rp rq r r r Ú#test_dirichlet_alpha_non_contiguous´ s ÿz2TestRandomDist.test_dirichlet_alpha_non_contiguousc C sx d}|t ddg¡ }tt| j}|j|dd}t ddgddggddgddggddgddggg¡}t||dd d S ) Nrs ç ð?r r% rº rT r« r7 )r, ra r r r r¬ r
|
)r( Zepsr rG r r£ r r r Útest_dirichlet_small_alpha¾ s ÿÿÿûz)TestRandomDist.test_dirichlet_small_alphac C sV t dddg¡}|| ¡ }tt| j}|j|dd}|jdd}t||dd d S )
|
Ng{®Gáz?g{®Gáz¤?g¸
ëQ¸?i -1rº r r r )r ) r, ra rn r r r r¬ r r )r( r Z
|
exact_meanrG r Z sample_meanr r r Ú%test_dirichlet_moderately_small_alphaÍ s z4TestRandomDist.test_dirichlet_moderately_small_alphac C sJ t t| j}|jddd}t ddgddgdd gg¡}t||d
|
d d S ) Ngï8EGrùñ?r% rº gvüïM¹?gùíì×øø?g#è±ïc³?gQü¹:Nü?gªAÜæPÑ÷?gE_9ù@r« r7 )r r r Ú exponentialr, ra r
|
r, r r r Útest_exponential× s
|
þz TestRandomDist.test_exponentialc C s& t tjddd tttjdd d S ©Nr ©Úscaleç )r rG rµ r r6 r7 r r r Útest_exponential_0ß s z!TestRandomDist.test_exponential_0c C sL t t| j}|jdddd}t ddgddgd d
|
gg¡}t||d d d S ) Nr éM r% rº g} (ÒÝ?g½ iñ?g=rûñ?g]hÍÇDAí?gjÓ¨¦à?g8·Ñk*uê?r« r7 )r r r Úfr, ra r
|
r, r r r Útest_fã s
|
þzTestRandomDist.test_fc C sL t t| j}|jdddd}t ddgddgd d
|
gg¡}t||d d d S ) NrW rk r% rº g,שËn'@gtô²± @g Ý<åe½2@g£1Äa3@g3r
|
yÑ-2@gs«1-2@r r7 )r r r Úgammar, ra r
|
r, r r r Ú
|
test_gammaë s
|
þzTestRandomDist.test_gammac C s* t tjdddd tttjddd d S )Nr )r` r¹ rº )r rG r¿ r r6 r7 r r r Ú test_gamma_0ó s zTestRandomDist.test_gamma_0c C sF t t| j}|jddd}t ddgddgddgg¡}t|| d S )Nç_c97Ý¿?r% rº r/ rx r r§ )r r r Ú geometricr, ra r r, r r r Útest_geometric÷ s
|
þz TestRandomDist.test_geometricc C s t ttjd t ttjdgd t ttjd t ttjdgd tjdd, t ttjtj t ttjtjgd W 5 Q R X d S )Nre r rf Úignore©Úinvalid)r r6 rG rà r, ÚerrstaterK r7 r r r Útest_geometric_exceptionsÿ s z(TestRandomDist.test_geometric_exceptionsc C sL t t| j}|jdddd}t ddgddgd d
|
gg¡}t||d d d S ) Nr ç @r% ©Úlocr¹ r gÈA7GëÀ@gþ«1FiÒ¿gei¹ì@g¸¹_Cä¿g0cÙPí ¬¿g`(¤_Õ¿r« r7 )r r r Úgumbelr, ra r
|
r, r r r Ú test_gumbel s
|
þzTestRandomDist.test_gumbelc C s& t tjddd tttjdd d S r· )r rG rÍ r r6 r7 r r r Ú test_gumbel_0 s z TestRandomDist.test_gumbel_0c C s t t| j}|jddddd}t ddgddgddgg¡}t|| |jdd d
|
d d}t d
|
d
|
d
|
d
|
g¡}t|| |jd d d d d}t d d d d g¡}t|| |jd dd
|
d d}t d d d d g¡}t|| |jd d d d d}t d d d d g¡}t|| d S )Ng333333$@g @r r% rº r r rW r rk r} r« r )r r r Úhypergeometricr, ra r r, r r r Útest_hypergeometric s&
|
þ
|
|
|
|
z"TestRandomDist.test_hypergeometricc C sL t t| j}|jdddd}t ddgddgd d
|
gg¡}t||d d d S ) Nr rÊ r% rË gÍÕ6@ Àga@bDA"ó?g*[rÑ{ Àg³>K ú?g«À(í?g<ý©ô?r« r7 )r r r Úlaplacer, ra r
|
r, r r r Ú test_laplace. s
|
þzTestRandomDist.test_laplacec C s& t tjddd tttjdd d S r· )r rG rÒ r r6 r7 r r r Útest_laplace_06 s z TestRandomDist.test_laplace_0c C sL t t| j}|jdddd}t ddgddgd d
|
gg¡}t||d d d S ) Nr rÊ r% rË gbýíµZÀgHÊg$>þ?g>ÔÑ&øÀgÐQhÊ @g%?n í÷?g¯ÄÒ8|ÿ?r« r7 )r r r Úlogisticr, ra r
|
r, r r r Ú test_logistic: s
|
þz TestRandomDist.test_logisticc C sL t t| j}|jdddd}t ddgddgd d
|
gg¡}t||d d d S ) Nr rÊ r% )r Úsigmar gÜÐx?gúÅÕk*è+@gð<¡Ò¾?g á
|
ð@g5 Ìaþ@gá%jzÇ(@rª r7 )r r r Ú lognormalr, ra r
|
r, r r r Útest_lognormalB s
|
þz TestRandomDist.test_lognormalc C s& t tjddd tttjdd d S )Nr )r× r/ rº )r rG rØ r r6 r7 r r r Útest_lognormal_0J s z TestRandomDist.test_lognormal_0c C sF t t| j}|jddd}t ddgddgdd gg¡}t|| d S )
|
NgÆ@õí?r% )rI r r r§ rk é rW r/ )r r r Ú logseriesr, ra r r, r r r Útest_logseriesN s
|
þz TestRandomDist.test_logseriesc C s$ t t| j}| d¡dks td S )Nr r/ )r r r rÜ rÜ ©r( rG r r r Útest_logseries_zeroV s z"TestRandomDist.test_logseries_zeroÚvaluerT r3 r² g @c
|
C s¬ t t| j}tjdd t t¡ | |¡ W 5 Q R X t t¡ | t |gd ¡¡ W 5 Q R X t t¡& | t |gd ¡d d d
¡ W 5 Q R X W 5 Q R X d S )NrÅ rÆ r rA )
|
r r r r, rÈ rw rx r6 rÜ ra )r( rà rG r r r Útest_logseries_exceptionsZ s z(TestRandomDist.test_logseries_exceptionsc C s t t| j}|jddgd dd}t dddddd gdd
|
dd
|
dd
|
ggdd d
|
dd dgddd d
|
d d ggdd ddd d
|
gddd dd
|
dggg¡}t|| d S ) Nr çUUUUUUÅ?r¯ r% rº r/ rW r} rk rA r r© )r r r rR r, ra r r, r r r Útest_multinomialg s ÿÿÿüz TestRandomDist.test_multinomialz fp errors don't work in wasm©Úreasonr ÚsvdÚeighÚcholeskyc C sv t t| j}d}ddgddgg}d}|j||||d}t ddgdd ggd
|
d gd d ggddgddggg¡}t||dd |j|||d}t ddg¡}t||dd ddg}ddgddgg}tt|j||dd ddgddgg}t t
|
|j|| t t
|
|j||dd tt |j||dd t |j||dd tt|j||dd tt|j||ddd ddgddgg}|d kr|j||d|d }t|d |d d!d ntt |j||dd tjdd"gd"dggtj d#}t 2} |j|||d | t
|
¡}
|
t|
|
dksþtW 5 Q R X t d¡} t d¡}tt|j||d$d tt|jt d%¡| tt|j| t d¡ tt|j| t d&¡ d S )'N)r r r/ r r% r gºt¦«õû¿g¨ãÒ $&@g&بO=åï¿g!Ðæ¯$@gÒDÝo¼ é?g¿ïº\&@gÀ,£Rü{ì?gÜÁó
|
f¿!@g?ÕªìÑæ?gTâ^o#@g`¾RïÎæ?gPÁû'@r« r7 gýùlÜÍ?g$Çû(Ù"@rA Úraise)Ú check_validrç rè rÅ )rê r )ræ rç r ).r ).r/ r¯ rD rC Úother)rA r/ r/ rk )r r r Úmultivariate_normalr, ra r
|
r r6 r ÚRuntimeWarningr r rv r Úrecordrü rÜ rF Zeyer ) r( r rG r Úcovr r rÿ r ÚsupÚwÚmur r r Útest_multivariate_normalr s
|
ÿÿÿü ÿ ÿ ÿ
|
ÿ ÿ ÿ
|
|
ÿÿ ÿ
|
|
|
ÿ ÿ ÿ ÿz'TestRandomDist.test_multivariate_normalz mean, covy ð? ð?y c C s8 t t| j}tjtdd | ||¡ W 5 Q R X d S )Nzmust not be complexrt )r r r rw rx r5 rì )r( r rï rG r r r Ú)test_multivariate_normal_disallow_complex½ s z8TestRandomDist.test_multivariate_normal_disallow_complexc C s¢ t t| j}d}t ddg¡}t ddgddgg¡}|j|||f|d}|| }|j| |d }t t | d¡¡dk ¡st
|
t t || ¡dk ¡st
|
d S )Nr" r/ rA r r9 rD rO ) r r r r, ra rì rr rX Úabsr rÜ ) r( r rG Zn_sr rï r) Zs_centerZcov_empr r r Ú$test_multivariate_normal_basic_statsà s z3TestRandomDist.test_multivariate_normal_basic_statsc C sH t t| j}|jdddd}t ddgddgd d
|
gg¡}t|| d S ) NrN g|ò°Pk¿?r% )ÚnrI r i i× i iø iX i¢ )r r r rÁ r, ra r r, r r r r Рs
|
þz%TestRandomDist.test_negative_binomialc C sF t jdd0 tttjdt j tttjdt jgd W 5 Q R X d S )NrÅ rÆ rN r )r, rÈ r r6 rG rÁ rK r7 r r r Ú!test_negative_binomial_exceptionsØ s
|
|
|
ÿz0TestRandomDist.test_negative_binomial_exceptionsc C s$ t t t dd¡}W 5 Q R X d S ©Nr/ r )r r6 rG rÁ rY r r r Ú#test_negative_binomial_p0_exceptionÞ s
|
z2TestRandomDist.test_negative_binomial_p0_exceptionc C s@ t jdd* tttjdd tttjdgdg W 5 Q R X d S )NrÅ rÆ r rD )r, rÈ r r6 rG rÁ r7 r r r Ú.test_negative_binomial_invalid_p_n_combinationã s z=TestRandomDist.test_negative_binomial_invalid_p_n_combinationc C sÎ t t| j}|jdddd}t ddgddgdd gg¡}t||d
|
d |jd d dd}t ddgddgddgg¡}t||d
|
d t t| j}|jdddd}t ddgddgddgg¡}t||d
|
d d S )NrW r% )ÚdfÚnoncr gsØ3Jû?gugÙò/@gNJÁþm+@gàA}¸-4@gq=È ¹&@gj
©ïhn @r r7 rE rO gZÖ
DÙN?gÒÔE!X&?gt÷ô¦%Tò?gQ§)»¤V?g5
Zlfþ?gÎáÎ{ó?r gmK°ê?gÓ©Z
Òü?g%yKY@gÅ@¥W @gĺµ{>@g&ÆBB=@)r r r Únoncentral_chisquarer, ra r
|
r, r r r Útest_noncentral_chisquareê s(
|
þ
|
þ
|
þz(TestRandomDist.test_noncentral_chisquarec C sN t t| j}|jddddd}t ddgdd gd
|
d gg¡}t||d d d S )NrW rA r/ r% )ÚdfnumÚdfdenrý r g HP
|
á®?gõ ÀnÎ?g²%¬fËë?gc×hÑ?gzÆÆëÍ?gÞ]Ô?:þ?r r7 )r r r Ú noncentral_fr, ra r
|
r, r r r Útest_noncentral_fÿ s
|
ÿ
|
þz TestRandomDist.test_noncentral_fc C s2 t t| j}|jddtjd}t |¡s.td S )NrW rA )r r rý )r r r r r, rK ÚisnanrÜ ©r( rG r r r r Útest_noncentral_f_nan s z$TestRandomDist.test_noncentral_f_nanc C sL t t| j}|jdddd}t ddgddgd d
|
gg¡}t||d d d S ) Nr rÊ r% rË gÕ?`xò Àg\èPà÷@g/7uï Àg0aj¸Öé?g®i%÷?go¶g®â@r« r7 )r r r Únormalr, ra r
|
r, r r r Ú test_normal s
|
þzTestRandomDist.test_normalc C s& t tjddd tttjdd d S r· )r rG r r r6 r7 r r r Ú test_normal_0 s z TestRandomDist.test_normal_0c C sN t t| j}|jddd}t ddgddgdd gg¡}tjj||d
|
d d S ) Nr r% ©rN r g ìáð?g¢Æ«hÕò@gi`®>ç?gü0"&Ag¯ÿcæ@g=¸fAr¥ )Znulp)r r r Úparetor, ra ZtestingZ assert_array_almost_equal_nulpr, r r r Ú test_pareto s
|
þ zTestRandomDist.test_paretoc C sF t t| j}|jddd}t ddgddgddgg¡}t|| d S )Nr r% )Úlamr r )r r r Úpoissonr, ra r r, r r r Ú test_poisson' s
|
þzTestRandomDist.test_poissonc C s t d¡j}d}tttj| tttj|gd tttj| tttj|gd t jdd, tttjt j tttjt jgd W 5 Q R X d S )Nr r3 r rÅ rÆ ) r, r r r r6 rG r rÈ rK )r( ZlambigZlamnegr r r Útest_poisson_exceptions/ s z&TestRandomDist.test_poisson_exceptionsc C sJ t t| j}|jddd}t ddgddgdd gg¡}t||d
|
d d S ) Nr r% r
|
g¯²¶Zý >g«¡µú¹?gÃî(ñ=gÒö¾Ã?gÅÿEö´?gûߢHÙ?r« r7 )r r r Úpowerr, ra r
|
r, r r r Ú
|
test_power: s
|
þzTestRandomDist.test_powerc C sJ t t| j}|jddd}t ddgddgdd gg¡}t||d
|
d d S ) Nr r% )r¹ r grêyÇ@gßtQÒP«0@gAI¼ð_ @g@âu8¿1@g¹æ ¾G0@g" wæc5@r r7 )r r r Úrayleighr, ra r
|
r, r r r Ú test_rayleighB s
|
þz TestRandomDist.test_rayleighc C s& t tjddd tttjdd d S r· )r rG r r r6 r7 r r r Útest_rayleigh_0J s z TestRandomDist.test_rayleigh_0c C sH t t| j}|jdd}t ddgddgddgg¡}t||d d
|
d S ) Nr% rº g¹H1µ¼Ô÷¿gW|ÿ3
|
Àgôå×\ìá?gLeõÈå¿g5P â õ¿gÛûFAïÒ?r« r7 )r r r Zstandard_cauchyr, ra r
|
r, r r r Útest_standard_cauchyN s
|
þz#TestRandomDist.test_standard_cauchyc C sJ t t| j}|jddd}t ddgddgdd gg¡}t||d
|
d d S ) Nr% Úinvr ga\5 º?g rô7k«ó?g8ªPC*¶?g5ñè$[÷?gW3ºTñ?g*ðÁê ô?r« r7 )r r r Ústandard_exponentialr, ra r
|
r, r r r Útest_standard_exponentialV s
|
þz(TestRandomDist.test_standard_exponentialc C s t ttjtjd d S ©NrC )r r5 rG r r, rè r7 r r r Ú#test_standard_expoential_type_error^ s z2TestRandomDist.test_standard_expoential_type_errorc C sJ t t| j}|jddd}t ddgddgdd gg¡}t||d
|
d d S ) Nrk r% ©r` r g ÎÌ&ä?gT»Ï¿ó?g1n2ÿ1@gDb<zË@gÊBváÿ @gc^þ @r r7 ©r r r Ústandard_gammar, ra r
|
r, r r r Útest_standard_gammaa s
|
þz"TestRandomDist.test_standard_gammac C s4 t t| j}|jdtjd}d}t||dd d S )Nrk rC g ÀÊd@r¯ r7 )r r r r r, rv r
|
r, r r r Ú!test_standard_gammma_scalar_floati s z0TestRandomDist.test_standard_gammma_scalar_floatc C sJ t t| j}|jddd}t ddgddgdd gg¡}t||d
|
d d S ) Nrk r% r gfÚþ&ä?gM O¯ó?g½Þý1@gÇ):Ë@g«²ïàÿ @g;ÂiÁþ @rW r7 r r, r r r Útest_standard_gamma_floato s
|
þz(TestRandomDist.test_standard_gamma_floatc C s t jdt jd}tt| j}|jd|t jd t jddgddgd d
|
ggt jd}t||d d tt| j}|jd|dt jd t||d d d S )Nr% rC ç $@©r rD gM¹Â»L$@gÐa¾¼ { @g°N]ùì"@gPSËÖú")@g«Ø_¦+@gmÊÞåB @rW r7 ©r r rD ) r, rF rv r r r r ra r
|
)r( r rG rÿ r r r Ú test_standard_gammma_float_outw s
|
þþz-TestRandomDist.test_standard_gammma_float_outc C s t ttjddd d S )Nr² rè rC )r r5 rG r r7 r r r Ú test_standard_gamma_unknown_type s
|
ÿz/TestRandomDist.test_standard_gamma_unknown_typec C s6 t d¡}tttjdd|d tttjdd|d d S )Nr r" r )r r )r r/ )r, rF r r6 rG r )r( r r r r Útest_out_size_mismatch s
|
ÿ ÿz%TestRandomDist.test_out_size_mismatchc C s& t tjddd tttjdd d S )Nr )r` rº )r rG r r r6 r7 r r r Útest_standard_gamma_0 s z$TestRandomDist.test_standard_gamma_0c C sH t t| j}|jdd}t ddgddgddgg¡}t||d d
|
d S ) Nr% rº gð
|
bYïý¿g? ö ô?gGH{Lìñ¿gzÃ\ãÕ?gcöÈ`+å?gTÌü}×åò?r« r7 )r r r r» r, ra r
|
r, r r r Útest_standard_normal s
|
þz#TestRandomDist.test_standard_normalc C s t ttjtjd d S r )r r5 rG r» r, rè r7 r r r Ú%test_standard_normal_unsupported_type s z4TestRandomDist.test_standard_normal_unsupported_typec C sJ t t| j}|jddd}t ddgddgdd gg¡}t||d
|
d d S ) Nr r% )rü r g
V1Á÷¿g |Ú(Ó?göÒË-èð?gÍ NàhÚ¿gMäcìºÀ?g!uLòîM Àr« r7 )r r r Ú
|
standard_tr, ra r
|
r, r r r Útest_standard_t s
|
þz TestRandomDist.test_standard_tc C sN t t| j}|jddddd}t ddgdd gd
|
d gg¡}t||d d d S )Ng{®Gáz@gö(\Âu$@g×£p=
|
W4@r% )ÚleftÚmodeÚrightr g E©pw @g 4àDC+@gÍÅá¹ @g¿ /¹,@gk£éïuR*@gHNPcr+@r r7 )r r r Ú
|
triangularr, ra r
|
r, r r r Útest_triangular¦ s
|
ÿ
|
þz TestRandomDist.test_triangularc C sL t t| j}|jdddd}t ddgddgd d
|
gg¡}t||d d d S ) Nç®Gáz®ó?g®Gáz%@r% )r r r gmª2@gJçaD @g ª¼: @g
|
ö§×QÁ @g@T{eê¯ @gÚbìþ @r« r7 ©r r r Úuniformr, ra r
|
r, r r r Ú test_uniform¯ s
|
þzTestRandomDist.test_uniformc C s t d¡j}t d¡j}tj}tt|t j d tt|dt j tt||| tt|t j gdg tt|dgt jg tjt |d¡|d d d S )Nrb r r/ g Ø
W4vC)r r )
|
r, ZfinforÌ r rG r4 r Ú OverflowErrorÚinfÚ nextafter)r( ZfminZfmaxÚfuncr r r Útest_uniform_range_bounds· s z(TestRandomDist.test_uniform_range_boundsc C sn t j}|dd}t|d |dtjgdtjg}t|dtjg |dgdggddg}t|dt d¡ d S )Nç ø?rT g
|
×£p=Â @r^ )rG r4 r r, ÚpirF )r( r9 Úresultr r r Útest_uniform_zero_rangeÇ s
|
|
z&TestRandomDist.test_uniform_zero_rangec C sH t j}tt|dd tt|ddgddg tt|ddgddggd d S )NrA r/ r rk )rG r4 r r6 )r( r9 r r r Útest_uniform_neg_rangeÐ s z%TestRandomDist.test_uniform_neg_rangec C sj G dd dt j}t d¡ |¡}tttj|| G dd dt j}t d¡ |¡}tttj|dd d S )Nc @ s e Zd Zdd ZdS )zGTestRandomDist.test_scalar_exception_propagation.<locals>.ThrowingFloatc S s t d S r ©r5 r7 r r r Ú __float__Þ s zQTestRandomDist.test_scalar_exception_propagation.<locals>.ThrowingFloat.__float__N)r= r> r? rA r r r r Ú ThrowingFloatÝ s rB r² c @ s e Zd Zdd ZdS )zITestRandomDist.test_scalar_exception_propagation.<locals>.ThrowingIntegerc S s t d S r r@ r7 r r r Ú__int__æ s zQTestRandomDist.test_scalar_exception_propagation.<locals>.ThrowingInteger.__int__N)r= r> r? rC r r r r ÚThrowingIntegerå s rD r/ ) r, Zndarrayra rõ r r5 rG r4 rÐ )r( rB Zthrowing_floatrD Z throwing_intr r r Ú!test_scalar_exception_propagationÖ s
|
ÿz0TestRandomDist.test_scalar_exception_propagationc C sL t t| j}|jdddd}t ddgddgd d
|
gg¡}t||d d d S ) Nr2 ç¤p=
|
×£ø?r% ©rò Úkappar gLþÏAºñ?gûxÞyw»@g·ÉzÞVRý?g¡pE Ñ ÿ?g8&A¦Ð¿géÿ
áv @r« r7 )r r r Úvonmisesr, ra r
|
r, r r r Ú test_vonmisesì s
|
þz TestRandomDist.test_vonmisesc C s4 t t| j}|jdddd}tt |¡ ¡ d S )NrT g g5_PG>i@B rG )r r r rI r r, ÚisfiniterX ©r( rG Úrr r r Útest_vonmises_smallô s z"TestRandomDist.test_vonmises_smallc C s0 t t| j}|jdtjd}tt |¡ d S )NrT )rò rH )r r r rI r, rK r r rL r r r Útest_vonmises_nanú s z TestRandomDist.test_vonmises_nanrH g Ã@g 4&õk Cc C sr t t| j}t|j}|jj}|jd|dd}||j_|jd|dd}|dk r\t|| nt ||k¡snt
|
d S )Nr r rº g .A) r r r r rµ r¶ rI r r, rX rÜ )r( rH rG Úrsr¶ Zrandom_state_valsZgen_valsr r r Útest_vonmises_large_kappaÿ s
|
z(TestRandomDist.test_vonmises_large_kapparò g ÀgÍÌÌÌÌÌÀg @rs gíµ ÷Æ°>g @@c C sF t t| j}| ||d¡}tt |tj k¡o>t |tjk¡ d S )Nr¦ )r r r rI r r, rX r< )r( rò rH rG rM r r r Ú test_vonmises_large_kappa_range s z.TestRandomDist.test_vonmises_large_kappa_rangec C sL t t| j}|jdddd}t ddgddgd d
|
gg¡}t||d d d S ) Nr2 rF r% )r r¹ r g *س©2Ñ?gçtÌïÉ @g ;S @g\ÄÞ ÔF@g«xèÓF @g ¿6bç?r r7 )r r r Úwaldr, ra r
|
r, r r r Ú test_wald s
|
þzTestRandomDist.test_waldc C sJ t t| j}|jddd}t ddgddgdd gg¡}t||d
|
d d S ) Nr2 r% r
|
gnJZ˾Á?g*Ô/Fçô?gsÄ(RY¼?gÿÛ]'%÷?g,×qãD ô?gçÿö8 þ?r« r7 )r r r Úweibullr, ra r
|
r, r r r Ú test_weibull s
|
þzTestRandomDist.test_weibullc C s< t t| j}t|jdddt d¡ tt|jdd d S )Nr r r
|
rº )rN ) r r r r rU r, rF r r6 rÞ r r r Útest_weibull_0$ s z TestRandomDist.test_weibull_0c C sF t t| j}|jddd}t ddgddgddgg¡}t|| d S ) Nr2 r% r
|
r/ r ic ib rA )r r r Úzipfr, ra r r, r r r Ú test_zipf) s
|
þzTestRandomDist.test_zipf)r= r> r? r| r- r1 r3 r6 r: r< r= rw r± r² r, rv r_ Úfloat64rë r> r? rB rF rJ rM rS rT rU rX r] r_ ra rb rd rf rv rz r{ r r r r r r r rB r ra 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ß r8 rK 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, r1 r5 r: r> r? rE rJ rN rO rQ r< rR rT rV rW rY r r r r r"  s
|
|
ÿÿ
|
. ÿÿÿüÿ
|
|
|
|
I$
|
|
|
r" c @ sT e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z dd Zdd Zdd Zd d Zd d Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zd+d, Zd-d. Zd/d0 Zd1d2 Zd3d4 Z d5d6 Z d7d8 Z d9d: Z d;d< Z d=d> Z!d?d@ Z"e#j$ %dAdBe& 'dBdBg¡e& 'dBggdBggg¡g¡dCdD Z(dEdF Z)dGdH Z*dIS )JÚ TestBroadcastc C s
|
d| _ d S )NiÍ[r{ r7 r r r r| 5 s zTestBroadcast.setup_methodc C s t t| j}dg}dg}|j}t dddg¡}t t| j}| |d |¡}t||dd t t| j}| ||d ¡}t||dd d S ) Nr r/ gî4Ô
|
7^Å?gÌF]
|
"É?gñØ .è?rk r r7 r3 )r( rG r r r4 rÿ r r r r r5 9 s zTestBroadcast.test_uniformc C s¶ dg}dg}dg}t t| j}t dddg¡}t t| j}| |d |¡}t||dd tt|j|d | t t| j}|j}|||d }t||dd tt|||d d S )
|
Nr r/ r3 gw-D¨ÊØ¿gn§0_xé?gºÊ;<.?rk r r7 ) r r r r, ra r r
|
r r6 )r( rÌ r¹ Ú bad_scalerG rÿ r r r r r r H s zTestBroadcast.test_normalc C s¬ dg}dg}dg}dg}t dddg¡}tt| j}|j}||d |}t||d d
|
tt||d | tt||d | tt| j}| ||d ¡}t||d d
|
d S ) Nr/ rA r3 r9 gfý8ñóõÇ?gü%eoç?g¬$q?ÙòÆ?rk r r7 ) r, ra r r r r¡ r
|
r r6 ) r( rN rO Úbad_aZbad_brÿ rG r¡ r r r r r¢ Z s zTestBroadcast.test_betac C s\ dg}dg}t dddg¡}tt| j}| |d ¡}t||dd tt|j|d d S © Nr/ r3 g` ±Êå?gþEÑðõ]Ë?g° øæ?rk r r7 ) r, ra r r r rµ r
|
r r6 ©r( r¹ r] rÿ rG r r r r r¶ l s z TestBroadcast.test_exponentialc C s^ dg}dg}t dddg¡}tt| j}|j}||d }t||dd tt||d d S r_ ) r, ra r r r r r
|
r r6 )r( r` Ú bad_shaperÿ rG Z std_gammar r r r r v s z!TestBroadcast.test_standard_gammac C sÔ dg}dg}dg}dg}t dddg¡}tt| j}|j}||d |}t||d d
|
tt||d | tt||d | tt| j}|j}|||d }t||d d
|
tt|||d tt|||d d S ) Nr/ rA r3 r9 g3 ±Êõ?gJEÑðõ]Û?g° øö?rk r r7 ) r, ra r r r r¿ r
|
r r6 ) r( r` r¹ ra r] rÿ rG r¿ r r r r rÀ s" zTestBroadcast.test_gammac C sÔ dg}dg}dg}dg}t dddg¡}tt| j}|j}||d |}t||d d
|
tt||d | tt||d | tt| j}|j}|||d }t||d d
|
tt|||d tt|||d d S ) Nr/ rA r3 r9 g
M3Bèà³?gybi¶ë @g0ñ»ðò?rk r r7 ) r, ra r r r r½ r
|
r r6 ) r( r r Ú bad_dfnumÚ bad_dfdenrÿ rG r½ r r r r r¾ s" zTestBroadcast.test_fc C s¦ dg}dg}dg}dg}dg}dg}t ddd g¡}tt| j}|j} | |d ||}
|
t|
|
|d
|
d t t | ||t j gd ¡¡st
|
t t | |d || t t | |d || t t | |d || tt| j}|j} | ||d |}
|
t|
|
|d
|
d t t | ||d | t t | ||d | t t | ||d | tt| j}|j} | |||d }
|
t|
|
|d
|
d t t | |||d t t | |||d t t | |||d d S ) NrA rk r} r r3 r9 gÅ,nÚ1 @g¨!ò°6Ö)@góðË9çó?r r7 ) r, ra r r r r r
|
rX r rK rÜ r r6 ) r( r r rý rb rc Úbad_noncrÿ rG Znonc_fr r r r r « s: $z TestBroadcast.test_noncentral_fc C s@ t t| j}t ddg¡}|jddddd}t||dd d S )NgZ\ÇáÞ#¨?gDqìwº¿?rg rA rº r r7 )r r r r, ra r r
|
)r( rG rÿ r r r r Útest_noncentral_f_small_dfÎ s z(TestBroadcast.test_noncentral_f_small_dfc C s\ dg}dg}t dddg¡}tt| j}| |d ¡}t||dd tt|j|d d S ) Nr/ r3 g¤éC{¬?g7Qð#÷?g%hÅ1T@rk r r7 ) r, ra r r r r¦ r
|
r r6 ©r( rü Úbad_dfrÿ rG r r r r r§ Ô s z TestBroadcast.test_chisquarec C sÔ dg}dg}dg}dg}t dddg¡}tt| j}|j}||d |}t||d d
|
tt||d | tt||d | tt| j}|j}|||d }t||d d
|
tt|||d tt|||d d S ) Nr/ rA r3 r9 g-¯èS½³?g×[ø @grª3/õ.ä?rk r r7 ) r, ra r r r rþ r
|
r r6 ) r( rü rý rg rd rÿ rG Znonc_chir r r r rÿ Þ s" z'TestBroadcast.test_noncentral_chisquarec C s\ dg}dg}t dddg¡}tt| j}| |d ¡}t||dd tt|j|d d S ) Nr/ r3 g äð>ßQö¿gçý{°ó¿g©,³eÆ?rk r r7 ) r, ra r r r r+ r
|
r r6 rf r r r r, ó s z TestBroadcast.test_standard_tc C s¦ dg}dg}dg}t dddg¡}tt| j}| |d |¡}t||dd tt|j|d | tt| j}| ||d ¡}t||dd tt|j||d d S )
|
NrA r/ r3 g½Dê()@g ,)ʸÝ@gPo»Àrk r r7 ) r, ra r r r rI r
|
r r6 )r( rò rH Z bad_kapparÿ rG r r r r rJ ý s zTestBroadcast.test_vonmisesc C s\ dg}dg}t dddg¡}tt| j}| |d ¡}t||dd tt|j|d d S ) Nr/ r3 g/h¼°î?gcÀEÎ?g/gÊVÌð?rk r r7 ) r, ra r r r r r
|
r r6 ©r( rN r^ rÿ rG r r r r r s zTestBroadcast.test_paretoc C s\ dg}dg}t dddg¡}tt| j}| |d ¡}t||dd tt|j|d d S r_ ) r, ra r r r rU r
|
r r6 rh r r r rV s zTestBroadcast.test_weibullc C s\ dg}dg}t dddg¡}tt| j}| |d ¡}t||dd tt|j|d d S ) Nr/ r3 gïÕÃTß?gþÁ Ê¥£È?gKêÜ®cà?rk r r7 ) r, ra r r r r r
|
r r6 rh r r r r ! s zTestBroadcast.test_powerc C sª dg}dg}dg}t dddg¡}tt| j}|j}||d |}t||dd tt||d | tt| j}|j}|||d }t||dd tt|||d d S )
|
Nr r/ r3 gh&Bñ¿g åséí¿gRÙÀ#ßèæ?rk r r7 ) r, ra r r r rÒ r
|
r r6 )r( rÌ r¹ r] rÿ rG rÒ r r r r rÓ + s zTestBroadcast.test_laplacec C sª dg}dg}dg}t dddg¡}tt| j}|j}||d |}t||dd tt||d | tt| j}|j}|||d }t||dd tt|||d d S )
|
Nr r/ r3 gs¡4û?g$Cw%Tø?g åòÕ¿rk r r7 ) r, ra r r r rÍ r
|
r r6 )r( rÌ r¹ r] rÿ rG rÍ r r r r rÎ = s zTestBroadcast.test_gumbelc C s¸ dg}dg}dg}t dddg¡}tt| j}| |d |¡}t||dd tt|j|d | tt| j}| ||d ¡}t||dd tt|j||d t | d
|
d ¡d
|
d S ) Nr r/ r3 g¡$ÆõD¸ù¿gXØëPö¿g,Ûò?rk r r7 r² rT )
|
r, ra r r r rÕ r
|
r r6 r )r( rÌ r¹ r] rÿ rG r r r r rÖ O s zTestBroadcast.test_logisticc C s dg}dg}dg}t dddg¡}tt| j}|j}||d |}t||dd tt||d | tt| j}| ||d ¡}tt|j||d d S )
|
Nr r/ r3 gõ cÚ¹å?g °u»@gVóé
Qð?rk r r7 ) r, ra r r r rØ r
|
r r6 )r( r r× Z bad_sigmarÿ rG rØ r r r r rÙ ` s z TestBroadcast.test_lognormalc C s\ dg}dg}t dddg¡}tt| j}| |d ¡}t||dd tt|j|d d S ) Nr/ r3 gsó·(ò?g4 mKçìä?g
Vk ¢+ó?rk r r7 ) r, ra r r r r r
|
r r6 r` r r r r p s þÿzTestBroadcast.test_rayleighc C sÔ dg}dg}dg}dg}t dddg¡}tt| j}| |d |¡}t||d d
|
tt|j|d | tt|j|d | tt| j}| ||d ¡}t||d d
|
tt|j||d tt|j||d d S ) NrE r/ r r9 gÓk7¥ZØ?g´*}z9à?gcûµÛ- ß?rk r r7 ) r, ra r r r rS r
|
r r6 )r( r r¹ Zbad_meanr] rÿ rG r r r r rT ~ s zTestBroadcast.test_waldc C s¸ dg}dg}dg}dg}dg}|d \}}t dddg¡}tt| j} | j}
|
|
|
|d ||} t| |dd tt|
|
|d || tt|
|
|d || tt|
|
|d || tt| j} | j}
|
|
|
||d |} t| |dd tt|
|
||d | tt|
|
||d | tt|
|
||d | tt| j} | j}
|
|
|
|||d } t| |dd tt|
|
|||d tt|
|
|||d tt|
|
|||d tt|
|
d
|
d d tt|
|
d
|
d d tt|
|
d
|
d
|
d
|
d S )Nr/ rk rA r} gR8Tµ¿>ù?gK Ìú?g;ôñ>h@r r7 r" rT g 4@g 9@) r, ra r r r r0 r
|
r r6 ) r( r- r/ r. Z bad_left_oneZ bad_mode_oneZ bad_left_twoZ bad_mode_tworÿ rG r0 r r r r r1 sJ ÿÿ
|
ÿz TestBroadcast.test_triangularc
|
C sò dg}dg}dg}dg}dg}t dddg¡}tt| j}|j}||d |} t| | tt||d | tt||d | tt||d | tt| j}| ||d ¡} t| | tt|||d tt|||d tt|||d d S )Nr/ rE r3 r; r rk ) r, ra r r r rH r r r6 )
|
r( r÷ rI Úbad_nÚ bad_p_oneÚ bad_p_tworÿ rG Zbinomr r r r r¥ ¹ s&
|
|
zTestBroadcast.test_binomialc
|
C sü dg}dg}dg}dg}dg}t jdddgt jd}tt| j}|j}||d |} t| | tt ||d | tt ||d | tt ||d | tt| j}|j}|||d } t| | tt |||d tt |||d tt |||d d S ) Nr/ rE r3 r; r rA rC rk )
|
r, ra r r r r rÁ r r r6 )
|
r( r÷ rI ri rj rk rÿ rG Z neg_binomr r r r r Рs(
|
|
z$TestBroadcast.test_negative_binomialc C sz dg}dg}t dddg¡}tt| j}|j}|d g}|j}||d }t|| tt ||d tt ||d d S )Nr/ r3 r rk rA )
|
r, ra r r r Z_poisson_lam_maxr r r r6 ) r( r Z bad_lam_onerÿ rG Zmax_lamZ bad_lam_twor r r r r r è s
|
|
zTestBroadcast.test_poissonc C s dg}dg}t dddg¡}tt| j}|j}||d }t|| tt||d t j dd( tt|t j
|
tt|ddt j
|
g W 5 Q R X d S )NrA r r/ r rk rÅ rÆ ) r, ra r r r rX r r r6 rÈ rK )r( rN r^ rÿ rG rX r r r r rY ÷ s
|
zTestBroadcast.test_zipfc C sp dg}dg}dg}t dddg¡}tt| j}|j}||d }t|| tt||d tt||d d S )NrE r3 r; r/ rk ) r, ra r r r rà r r r6 )r( rI rj rk rÿ rG rà r r r r rÄ s
|
z TestBroadcast.test_geometricc C s4 dg}dg}dg}dg}dg}dg}dg}t dddg¡}tt| j} | |d ||¡}
|
t|
|
| tt| j|d || tt| j|d || tt| j|d || tt| j|d || tt| j} | ||d |¡}
|
t|
|
| tt| j||d | tt| j||d | tt| j||d | tt| j||d | tt| j} | j} | |||d }
|
t|
|
| tt| |||d tt| |||d tt| |||d tt| |||d tt| ddd tt| ddd tt| ddd tt| ddd
|
tt| d dd tt| d d d tt| d dd gd d S )Nr/ rA r3 r9 r} r rk r r r i @iç l r¦ i r" ) r, ra r r r rÐ r r r6 ) r( ZngoodZnbadr Z bad_ngoodZbad_nbadZbad_nsample_oneZbad_nsample_tworÿ rG r Z hypergeomr r r rÑ sJ
|
|
|
z!TestBroadcast.test_hypergeometricc C sp dg}dg}dg}t dddg¡}tt| j}|j}||d }t|| tt||d tt||d d S )NrE rA r3 r/ rk ) r, ra r r r rÜ r r r6 )r( rI rj rk rÿ rG rÜ r r r r rÝ ? s
|
z TestBroadcast.test_logseriesc
|
C sÞ t t| j}|jddgdgd dd}tjdddd ddgdd
|
dd dd
|
ggd dd ddd gd ddd d d ggdddd ddgd
|
dd
|
d
|
d dgggtjd }t|| t t| j}| ddgdgd ¡}tjdddd ddgdd
|
dd dd
|
ggtjd }t|| t t| j}| ddgdgd gd ¡}tjdddd ddgdd
|
dd dd
|
ggtjd }t|| t t| j}| dgdggdgd gd ¡}tjdddd ddgdddd d d ggd dd
|
d
|
dd
|
gd ddd d d gggtjd }t|| d S )NrW r râ r¯ r% rº r rA r/ rk r} r© rC )r r r rR r, ra r r r, r r r rã L sV ÿÿÿüû
|
ÿÿ
|
ÿÿ
|
ÿÿþýz TestBroadcast.test_multinomialr÷ r c C s4 t t| j}t dgd ¡}| ||¡}t|tr:t n|j }|d }|j |ksVt
|
t ||g¡}| ||¡}t ||j d d
¡d }|j |kst
|
t |g|gg¡}| ||¡}t ||j d d
¡}|j |d ksÞt
|
|j||d| d}|j d| d ks
|
t
|
t t¡ |j||dd}W 5 Q R X d S )NrW r} )r} r3 r% rº )r/ r/ r/ r/ r/ r/ )r r r r, ra rR r rB rY r` rÜ rq Zbroadcast_shapesrw rx r6 )r( r÷ rG rm r Zn_shaper r r r Ú test_multinomial_pval_broadcastk s$ z-TestBroadcast.test_multinomial_pval_broadcastc C sF t t| j}dgd dgd g}tt|jd| tt|jdd d S )Nrâ r¯ rW r/ rE )r r r r r6 rR )r( rG rm r r r Ú test_invalid_pvals_broadcast s z*TestBroadcast.test_invalid_pvals_broadcastc C s t t| j}| t dd¡dgd ¡}|jdks6t| dt d¡¡}|jdksVt| t dd¡t d ¡¡}|jd ks~td S )
|
N)r r r¯ Úi8râ r¯ )r r r¯ r¯ r )r r r )rk r r© )rk r r© r} )r r r rR r, r r` rÜ r r r r Útest_empty_outputs s ÿz TestBroadcast.test_empty_outputsN)+r= r> r? r| r5 r r¢ r¶ r rÀ r¾ r re r§ rÿ r, rJ r rV r rÓ rÎ rÖ rÙ r rT r1 r¥ r r rY rÄ rÑ rÝ rã rw r± r² r, ra rl rm ro r r r r r\ 2 sN
|
#
|
|
|
|
|
( -
|
þ
|
r\ zcan't start threadrä c @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd S ) Ú
|
TestThreadc C s t d| _d S )Nr} )r+ Úseedsr7 r r r r| s zTestThread.setup_methodc sÌ ddl m t t| jf| ¡}t t| jf| ¡} fddt| j|D }dd |D dd |D t| j|D ]\}}tt|| q|t ¡ j
|
j dkr¾t j dkr¾t|| n
|
t|| d S ) Nr )ÚThreadc s( g | ] \}} t t||fd qS ))ÚtargetÚargs)r r )rm r) Úo©rr Úfunctionr r ro ¥ s ÿz-TestThread.check_function.<locals>.<listcomp>c S s g | ] }| ¡ qS r )Ústart©rm rZ r r r ro § s c S s g | ] }| ¡ qS r )Újoinry r r r ro ¨ s r} Úwin32)Ú threadingrr r, r rü rq Úzipr r ZintprD r
rî Úplatformr
|
r )r( rw ÚszZout1Zout2Útr) ru r rv r Úcheck_function s
|
ÿ zTestThread.check_functionc C s dd }| j |dd d S )Nc S s | j dd|d< d S )Nrc rº .)r ©r¶ r r r r Ú
|
gen_randomµ s z*TestThread.test_normal.<locals>.gen_random)rc ©r ©r ©r( r r r r r ´ s zTestThread.test_normalc C s dd }| j |dd d S )Nc S s | j t d¡d|d< d S )N©rN r" r¸ .)rµ r, r` r r r r r » s z'TestThread.test_exp.<locals>.gen_randomr r r
r r r r Útest_expº s zTestThread.test_expc C s dd }| j |dd d S )Nc S s | j ddgd dd|d< d S )Nr râ r¯ rc rº .)rR r r r r r Á s z/TestThread.test_multinomial.<locals>.gen_random)rc r¯ r r
r r r r rã À s zTestThread.test_multinomialN)r= r> r? r| r r r rã r r r r rp s
|
rp c @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd S ) ÚTestSingleEltArrayInputc C s4 t dg¡| _t dg¡| _t dg¡| _d| _d S )NrA rk r} rs )r, ra ÚargOneÚargTwoÚargThreeÚtgtShaper7 r r r r| É s z$TestSingleEltArrayInput.setup_methodc C s t jt jt jt jt jt jt jt jt j t j
|
t j t j f }t j t j f}|D ]6}||krb|t dg¡}n
|
|| j}t|j| j qDd S rÀ )rG rµ r r¦ r+ r rU r r r rX rà rÜ r, ra r r r` r )r( ÚfuncsÚ probfuncsr9 r r r r Útest_one_arg_funcsÏ s$ û
|
z*TestSingleEltArrayInput.test_one_arg_funcsc C sÈ t jt jt jt jt jt jt jt jt j t j
|
t j t j t j t jf}t j t jf}|D ]v}||krft dg¡}n| j}|| j|}t|j| j || jd |}t|j| j || j|d }t|j| j qLd S )NrE r )rG r4 r r¡ r¿ r½ rþ rI rÒ rÍ rÕ rØ rS rH rÁ r, ra r r r r` r )r( r r r9 r r r r r Útest_two_arg_funcsâ s2 ú z*TestSingleEltArrayInput.test_two_arg_funcsc C s¶ t jt jt jt jt jt jt jt jt j g }t
|
j }t dg¡}t dg¡}|D ]f}|||||d}t |j| j ||d |||d}t |j| j |||d ||d}t |j| j qJd S )Nr/ r rÅ )r, rù ré rì rç rê rè r_ r rë rG r& ra r r` r )r( r rÊ r9 r r rÎ r r r r r- ý s ÿ z%TestSingleEltArrayInput.test_integersc C s t jt jt jg}|D ]l}|| j| j| j}t|j| j || jd | j| j}t|j| j || j| jd | j}t|j| j qd S )Nr )
|
rG r r0 rÐ r r r r r` r )r( r r9 r r r r Útest_three_arg_funcs
|
s ÿz,TestSingleEltArrayInput.test_three_arg_funcsN)r= r> r? r| r r r- r r r r r r È s
|
r r c C sî | d }| d }t |}| |¡ |jd d }tjdkrB| ¡ }t |¡}|jd d | d d ksjt| ¡ | d d kst|
|
¡ }|jd d }tjdkrª| ¡ }t |¡}|jd d | d d ksÒt| ¡ | d d ksêtd S )
|
Nr r r¶ r· r r r r r ) r r0 r¶ rî rï rð rñ rò rÜ ró r )r r r Zmt19937r· rò r r r r Ú test_jumped
|
s"
|
|
|
|
|
r c C sl t d¡} t d¡}d}tj| |ddjdks2tt t¡ tj| ||d W 5 Q R X t t¡ tj| |dd W 5 Q R X t t¡ tj| |dd W 5 Q R X t d¡}t t¡ tj ||d W 5 Q R X t t¡ tj |dd W 5 Q R X t t¡ tj |dd W 5 Q R X t
|
|¡}t t¡ tj ||d W 5 Q R X t t¡ tj d
|
d d gd d W 5 Q R X t t¡ tj d
|
dgd d d W 5 Q R X t t¡ tj d
|
dgd d gd d W 5 Q R X t t¡ tj ddgd d gd d W 5 Q R X tj ddd}tj ddd}tj ddd}t |||¡jdks<ttjtdd tj|||dd W 5 Q R X d S )Nrk )r} rk )r r} rA )rW r} rk rº )r/ rk )r} r/ r/ )rk r r/ rK gffffffæ?)rA r/ rA rW z Output size \(6, 5, 1, 1\) isrt )r¯ rW r/ r/ )r, r` rG r r` rÜ rw rx r6 r r rH rR r¦ r )rò r× r r` r rN rO rP r r r Útest_broadcast_size_error9
|
sD
|
|
|
|
" "r c C sN t d¡} t d¡}tj| |dd t t¡ tj| |dd W 5 Q R X d S )Nrk rº rA )r, r` rG r rw rx r6 )rò r× r r r Útest_broadcast_size_scalard
|
s
|
|
|
r c C s8 g g dg} t td}t|j| | dg g gks4td S rù )r r r rI rÜ )ÚseqÚgenr r r Útest_ragged_shufflel
|
s
|
r r r9 c C s¦ t td}|rdnd}tjt|d |j| |d W 5 Q R X |rHdnd}tjt|d |jd| |d W 5 Q R X tjt|d |jdg| |d W 5 Q R X d S ) Nr zhigh < 0z high <= 0rt r5 z
|
low > highz low >= highr3 )r r rw rx r6 r& )r r r Úmsgr r r Ú!test_single_arg_integer_exceptiont
|
s r rD Zf4Zf8c C sv t jdd| d}dddg}tjtdd tj||| d W 5 Q R X tjtdd tj|||j| d
|
W 5 Q R X d S ) Nr ÚF)ÚorderrD r/ rA rk zSupplied output arrayrt r# r$ )r, r rw rx r6 rG r r` )rD r r` r r r Útest_c_contig_req_out
|
s
|
r r r ÚCÚdistc C sH t jd||d}| ||d}||ks(t| |||jd}||ksDtd S )Nr )rD r r# )r rD r )r, r rÜ r` )r r rD r Zvariatesr r r Útest_contig_req_out
|
s
|
r c C sp t j t j d¡¡} | d¡ | ¡ \}}}|d d
dks@t||d d
}||j_|jj}||ksltd S )Nr r/ )Ú PCG64DXSM) r, rG r r¡ r» Ú
|
__reduce__rÜ rµ r¶ )r¸ Zctorrt Zstate_arO Zstate_br r r Ú$test_generator_ctor_old_style_pickle
|
s
|
r£ )1rî rñ rw Únumpyr, Z numpy.linalgr Z numpy.testingr r r r r r r r
|
r r Z numpy.randomr r r r rG ZJUMP_TEST_DATAZfixturer r! r@ rM rz r³ rà r" r\ r± r[ rp r r² r r r r r r r» r r£ r r r r Ú<module> s 0
|
üü
|
üó
|
$6 " x j/U
|
+
|
|
|
