U
|
P ±dáS ã @ sà d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZm Z m
|
Z
|
m Z m Z m Z mZmZmZ d dlmZmZ d dlmZ ddddde d ¡d
|
fd d d dZe e¡jdk rÄdddddddddZndddd d d d d dZejd!ed"d#d$ Z ejd%d& Z d'd( Z G d)d* d*Z G d+d, d,Z G d-d. d.Z!G d/d0 d0Z"G d1d2 d2Z#G d3d4 d4Z$G d5d6 d6Z%ej&j'ed7d8G d9d: d:Z(G d;d< d<Z)d=d> Z*d?d@ Z+dAdB Z,dCdD Z-dEdF Z.dGdH Z/dIdJ Z0dKdL Z1dMdN Z2dS )Oé N) Úassert_Ú assert_raisesÚ assert_equalÚ assert_warnsÚassert_no_warningsÚassert_array_equalÚassert_array_almost_equalÚsuppress_warningsÚIS_WASM)ÚMT19937ÚPCG64)Úrandom)g Y@g333333ã?)ç à?)é r é
|
r é g @)éd r )ç $@)é )ÚbinomialÚ geometricÚhypergeometricÚ logseriesÚ multinomialÚnegative_binomialÚpoissonÚzipfì Z@2fbead005fc63942decb5326d36a1f32fe2c9d32c904ee61e46866b88447c263Z@23ead5dcde35d4cfd4ef2c105e4c3d43304b45dc1b1444b7823b9ee4fa144ebbZ@0d764db64f5c3bad48c8c33551c13b4d07a1e7b470f77629bef6c985cac76fcfZ@7b59bf2f1691626c5815cdcd9a49e1dd68697251d4521575219e4d2a1b8b2c67Z@d754fa5b92943a38ec07630de92362dd2e02c43577fc147417dc5b9db94ccdd3Z@8eb216f7cb2a63cf55605422845caaff002fddc64a7dc8b2d45acd477a49e824Z@70c891d76104013ebd6f6bcf30d403a9074b886ff62e4e6b8eb605bf1a4673b7Z@01f074f97517cd5d21747148ac6ca4074dde7fcb7acbaec0a936606fecacd93f)r r r r r r r r Z@8626dd9d052cb608e93d8868de0a7b347258b199493871a1dc56e2a26cacb112Z@8edd53d272e49c4fc8fbbe6c7d08d563d62e482921f3131d0a0e068af30f0db9Z@83496cc4281c77b786c9b7ad88b74d42e01603a55c60577ebab81c3ba8d45657Z@65878a38747c176bc00e930ebafebb69d4e1e16cd3a704e264ea8f5e24f548dbZ@7a984ae6dca26fd25374479e118b22f55db0aedccd5a0f2584ceada33db98605Z@d636d968e6a24ae92ab52fe11c46ac45b0897e98714426764e820a7d77602a61Z@956552176f77e7c9cb20d0118fc9cf690be488d790ed4b4c4747b965e61b0bb4Z@f84ba7feffda41e606e20b28dfc0f1ea9964a74574513d4a4cbc98433a8bfa45Úmodule)ZscopeÚparamsc C s | j t| j t| j fS ©N)ÚparamÚ INT_FUNCSÚINT_FUNC_HASHES)Úrequest© r% úZd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/random/tests/test_randomstate.pyÚint_func2 s ÿr' c c s t j ¡ } dV t j | ¡ dS )z:Ensures that the singleton bitgen is restored after a testN)Únpr Úget_bit_generatorÚset_bit_generator)Z orig_bitgenr% r% r& Úrestore_singleton_bitgen8 s
|
r+ c C sn t | d |d t| d d |d d t| d d |d d t | d |d t | d |d d S )NÚ bit_generatorÚstateÚkeyÚposÚ has_gaussÚgauss)r r )ÚaÚbr% r% r& Úassert_mt19937_state_equal@ s
|
r4 c @ sD e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dS )ÚTestSeedc C s8 t d¡}t| d¡d t d¡}t| d¡d d S )Nr éè i¬ ì ÿÿ i£ )r Ú RandomStater Úrandint©ÚselfÚsr% r% r& Ú test_scalarI s
|
|
zTestSeed.test_scalarc C sz t td¡}t| d¡d t t d¡¡}t| d¡d t dg¡}t| d¡d t dg¡}t| d¡d d S )Nr r6 iÔ r iÍ r7 i )r r8 Úranger r9 r( Úaranger: r% r% r& Ú
|
test_arrayO s zTestSeed.test_arrayc C s t ttjd t ttjd d S )Nç à¿éÿÿÿÿ©r Ú TypeErrorr r8 Ú
|
ValueError©r; r% r% r& Útest_invalid_scalarY s z TestSeed.test_invalid_scalarc C s\ t ttjdg t ttjdg t ttjdg t ttjdddg t ttjdddg d S )NrA rB r é r éþÿÿÿrC rF r% r% r& Útest_invalid_array^ s
|
zTestSeed.test_invalid_arrayc C sR t ttjtjg tjd t ttjdddgg t ttjdddgdddgg d S )N©ÚdtyperH r é é é r )r rE r r8 r( ÚarrayÚint64rF r% r% r& Útest_invalid_array_shapef s ÿÿz!TestSeed.test_invalid_array_shapec C s0 t td¡}tt | d¡ W 5 Q R X d S )Nr éÒ )r r8 r r rD Úseed)r; Úrsr% r% r& Útest_cannot_seedn s
|
zTestSeed.test_cannot_seedc C s t ttjt d S r )r rE r r8 r rF r% r% r& Útest_invalid_initializations s z$TestSeed.test_invalid_initializationN)
|
Ú__name__Ú
|
__module__Ú __qualname__r= r@ rG rJ rR rV rW r% r% r% r& r5 H s
|
r5 c @ s e Zd Zdd Zdd ZdS )Ú TestBinomialc C sB t jddd}dD ]*}tt d|¡dk tt ||¡| qd S )Nr ÚintrK )r r rH r )r( Úzerosr r r r )r; r] Úpr% r% r& Ú test_n_zerox s zTestBinomial.test_n_zeroc C s t ttjdtj d S )NrH )r rE r r r( ÚnanrF r% r% r& Ú test_p_is_nan s zTestBinomial.test_p_is_nanN)rX rY rZ r_ ra r% r% r% r& r[ w 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 )Nr çÉ?çé?©r r rF r% r% r& Ú
|
test_basic s zTestMultinomial.test_basicc C s t ddddddg¡ d S )Nr rc rd ç re rF r% r% r& Útest_zero_probability 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éûÿÿÿrB rO )r r r9 r( Úall©r; Úxr% r% r& Útest_int_negative_interval 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 )Nr rH ©rH r r ©r r r ©r r )
|
r r r r( Úuint32ÚshaperP r rD Úfloat©r; r^ r% r% r& Ú test_size s ÿ ÿzTestMultinomial.test_sizec C s, t ttjdddg t ttjdddg d S )Nr çñ?rc 繿çÍÌÌÌÌÌì?©r rE r r rF r% r% r& Útest_invalid_prob¡ s z!TestMultinomial.test_invalid_probc C s t ttjdddg d S )NrB rd rc ry rF r% r% r& Útest_invalid_n¥ s z TestMultinomial.test_invalid_nc C st t d¡}|t |dd d
¡ }|dd d
}t d¡ tjd|d}t d¡ tjdt |¡d}t|| d S )Ng .@rH rM iÛ iUr )Úpvals)r( r? Úsumr rT r Úascontiguousarrayr )r; r^ r| Ú
|
non_contigÚcontigr% r% r& Útest_p_non_contiguous¨ s
|
|
|
z%TestMultinomial.test_p_non_contiguousc C s^ t jddddddddddg
|
t jd}|| ¡ }d}tjt|d t d|¡ W 5 Q R X d S )Ng®Gáz®ï?gÖ&è .>rK z-[\w\s]*pvals array is cast to 64-bit floating©ÚmatchrH ) r( rP Úfloat32r} ÚpytestÚraisesrE r r )r; rl r| r r% r% r& Ú test_multinomial_pvals_float32² s ÿÿ z.TestMultinomial.test_multinomial_pvals_float32N) rX rY rZ rf rh rm ru rz r{ r r r% r% r% r& rb s
|
rb c @ sd e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z dS )Ú TestSetStatec C s$ d| _ t | j ¡| _| j ¡ | _d S ©NiÒI)rT r r8 Ú random_stateÚ get_stater- rF r% r% r& Ú setup_method¼ s zTestSetState.setup_methodc C s< | j d¡}| j | j¡ | j d¡}tt ||k¡ d S )Né )r ÚtomaxintÚ set_stater- r r( rj ©r; ÚoldÚnewr% r% r& rf Á s zTestSetState.test_basicc C s@ | j jdd}| j | j¡ | j jdd}tt ||k¡ d S ©NrM ©Úsize)r Ústandard_normalr r- r r( rj r r% r% r& Útest_gaussian_resetÇ s z TestSetState.test_gaussian_resetc C sR | j ¡ | j ¡ }| j jdd}| j |¡ | j jdd}tt ||k¡ d S r )r r r r r r( rj )r; r- r r r% r% r& Ú test_gaussian_reset_in_media_resÎ s
|
|
z-TestSetState.test_gaussian_reset_in_media_resc C sz | j d d
}| jjdd}| j |¡ | jjdd}| j | j ¡ | jjdd}tt ||k¡ tt ||k¡ d S )NrI r r )r- r r r r r( rj )r; Ú old_stateÚx1Zx2Zx3r% r% r& Ú test_backwards_compatibilityÙ s z)TestSetState.test_backwards_compatibilityc C s | j dd¡ d S ©Nr )r r rF r% r% r& Útest_negative_binomialå s z#TestSetState.test_negative_binomialc C sb t t ¡}t F}| t¡}| ¡ }tt|dk t |t
|
sDt |d dksTt W 5 Q R X d S )NrH r, r ) r r8 r r ÚrecordÚRuntimeWarningr r ÚlenÚ
|
isinstanceÚdictÚAssertionError)r; rU ÚsupÚwr- r% r% r& Útest_get_state_warningê s
|
z#TestSetState.test_get_state_warningc C sl | j ¡ }d|dd
}tt| j j| tt| j jtj|td | j jdd}|d= tt| j j| d S )N)ÚUnknownrH rK F©Úlegacyr, ) r r r rE r rD r( rP Úobject)r; r- Ú new_stater% r% r& Ú!test_invalid_legacy_state_settingó s
|
|
ÿz.TestSetState.test_invalid_legacy_state_settingc C sj | j d¡ | j d¡ | j ¡ | j jdd}t|d d t t | j ¡¡}|jdd}t || d S )Nr r Fr¨ r0 rH )
|
r rT Ú random_sampler r r ÚpickleÚloadsÚdumpsr4 )r; ZpickledZ rs_unpickZ unpickledr% r% r& Ú test_pickleý s
|
zTestSetState.test_picklec C s< | j ¡ }| j ¡ | j |¡ | j jdd}t|| d S )NFr¨ )r Ú __getstate__r Ú __setstate__r r4 )r; Z
|
attr_stater- r% r% r& Útest_state_setting s
|
|
|
z TestSetState.test_state_settingc C s t | j d¡std S )NzRandomState(MT19937))Úreprr Ú
|
startswithr£ rF r% r% r& Ú test_repr s zTestSetState.test_reprN)rX rY rZ r rf r r r r r¦ r¬ r± r´ r· r% r% r% r& r » s
|
|
r c @ s e Zd ZejZejejej ej
|
ej ej ej ejejg Zdd Zdd Zdd Zdd Zd d
|
Zd d Zejje d ¡jdk dddd Z dd Z dd Z dS )Ú TestRandintc C s t t| jdtd d S ©NrH rK )r rD Úrfuncrs rF r% r% r& Útest_unsupported_type s z!TestRandint.test_unsupported_typec C s | j D ]}|tjkrdn
|
t |¡j}|tjkr2dnt |¡jd }tt| j|d ||d tt| j||d |d tt| j|||d tt| jdd|d qd S )Nr r rH rK ) Úityper( Úbool_ÚiinfoÚminÚmaxr rE rº )r; ÚdtÚlbndÚubndr% r% r& Útest_bounds_checking s
|
z TestRandint.test_bounds_checkingc C s´ | j D ]¨}|tjkrdn
|
t |¡j}|tjkr2dnt |¡jd }|d }t| j||d d|d| |}t| j||d d|d| || d }t| j||d d|d| qd S )Nr r rH r6 ©r rL )r¼ r( r½ r¾ r¿ rÀ r rº )r; rÁ r rà Útgtr% r% r& Útest_rng_zero_and_extremes& s
|
z&TestRandint.test_rng_zero_and_extremesc C s | j D ]}|tjkrdn
|
t |¡j}|tjkr2dnt |¡jd }z| j|||d W q tk
|
r } ztdt | W 5 d }~X Y qX qd S )Nr r rH rK zMNo error should have been raised, but one was with the following message:
|
|
%s)
|
r¼ r( r½ r¾ r¿ rÀ rº Ú Exceptionr£ Ústr)r; rÁ r rà Úer% r% r& Útest_full_range4 s
|
þzTestRandint.test_full_rangec C s t ¡ | jdd
D ]@}dD ]6}| jd|d|d}t| ¡ |k t| ¡ dk q q| jdddtjd}t| ¡ dk t| ¡ dk d S )NrH )rN é r r i rÅ r ) r rT r¼ rº r rÀ r¿ r( r½ )r; rÁ rà Úvalsr% r% r& Útest_in_bounds_fuzzB s z TestRandint.test_in_bounds_fuzzc
|
C sæ dddddddddd }| j dd
D ]p}t d¡ tjd krR| jd
|
d d |d }n| jd
|
d d |d ¡ }t | t
|
j ¡¡ ¡ }t |t
|
|¡j |k q&t d¡ | jd
|
dd td t
|
j ¡}t |¡ ¡ }t |t
|
t¡j |k d S )NZ@509aea74d792fb931784c4b0135392c65aec64beee12b0cc167548a2c3d31e71Z@7b07f1a920e46f6d0fe02314155a2330bcfd7635e708da50e536c5ebb631a7d4Z@e577bfed6c935de944424667e3da285012e741892dcb7051a8f1ce68ab05c92fZ@0fbead0b06759df2cfb55e43148822d4a1ff953c7eb19a5b08445a63bb64fa9eZ@001aac3a5acb935a9b186cbe14a1ca064b8bb2dd0b045d48abeacf74d0203404) ÚboolÚint16Úint32rQ Úint8Úuint16rq Úuint64Úuint8rH rS Úlittler r r6 rÅ r )r¼ r rT ÚsysÚ byteorderrº ÚbyteswapÚhashlibÚsha256Úviewr( rÒ Ú hexdigestr rL ÚnamerÏ )r; rÆ rÁ ÚvalÚresr% r% r& Útest_repeatabilityQ s* ø
|
|
|
|
z TestRandint.test_repeatabilityÚlr z Cannot test with 32-bit C long©Úreasonc 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 ]J}t d/¡ | jd0gd1gd2ggd3d4d5g|d6}t||d k rÐ|n|d1 qd S )7Nl áQö[ l dm¬" l ë0 l ;¡i l vlÔm l iXp i\L¹_l øZ N iwU¬'l Ï<Ô~ l È20l iPð?iÒfi¦ØwLl @
|
Vo l ðo\ i´wiý
|
1]i{YSl r=ï| ißl â0 l lÊ iû&ã}l QZî) l (q l '}åQ l WqV1 iêM l 'ßt i nMl øbþ" l ) iÄ al ³Ñ l S"f i¸di9 8l ÙnDk l ¬J c i¹á-l i¬Mi'ñ*l äl¼, )N)rO rM rM i90 rB r rH r7 r l r )r( rP r rT rº r )r; Údesiredr rl r% r% r& Ú.test_repeatability_32bit_boundary_broadcastingq s6 þþþþþô
|
ÿz:TestRandint.test_repeatability_32bit_boundary_broadcastingc C s^ t j}t t j¡j}t t t j¡j¡}t t t j¡jd ¡}tj|||d}t|| d S r¹ )r( rQ r¾ rÀ rÔ r r9 r )r; rÁ rÆ r rà Úactualr% r% r& Ú test_int64_uint64_corner_case s z)TestRandint.test_int64_uint64_corner_casec C sà | j D ]^}|tjkrdn
|
t |¡j}|tjkr2dnt |¡jd }| j|||d}t|jt |¡ qt t
|
tj j fD ]f}|t krdn
|
t |¡j}|t krdnt |¡jd }| j|||d}t t|d tt|| qtd S )Nr r rH rK rL )r¼ r( r½ r¾ r¿ rÀ rº r rL rÏ r\ ÚcompatÚlongr ÚhasattrÚtype)r; rÁ r rà Úsampler% r% r& Ú test_respect_dtype_singleton s
|
z(TestRandint.test_respect_dtype_singletonN) rX rY rZ r r9 rº r( r½ rÒ rÕ rÐ rÓ rÑ rq rQ rÔ r¼ r» rÄ rÇ rË rÎ rá r
ÚmarkÚskipifr¾ rÀ ræ rè rî r% r% r% r& r¸ s& ÿ ÿ
|
r¸ c @ s¼ 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"dAdB Z#dCdD Z$dEdF Z%dGdH Z&dIdJ Z'dKdL Z(dMdN Z)dOdP Z*dQdR Z+dSdT Z,dUdV Z-dWdX Z.dYdZ Z/d[d\ Z0d]d^ Z1e2j3 4d_e5 6d`da¡dbe5j7dcg¡ddde Z8dfdg Z9dhdi Z:djdk Z;dldm Z<dndo Z=dpdq Z>drds Z?dtdu Z@dvdw ZAdxdy ZBdzd{ ZCd|d} ZDd~d ZEdd ZFdd ZGdd
ZHdd ZIdd ZJdd ZKdd ZLdd ZMdd ZNdd ZOdd ZPdd ZQdd ZRdd ZSdd ZTdd ZUd d¡ ZVd¢d£ ZWd¤d¥ ZXd¦d§ ZYd¨d© ZZdªS )«ÚTestRandomDistc C s
|
d| _ d S r ©rT rF r% r% r& r · s zTestRandomDist.setup_methodc C sF t | j¡ t dd¡}t ddgddgddgg¡}t||d d
|
d S ) NrM r ç_®\±*Íã?ç¬îâ?ç<îpì?ç®,º¥oì?ç ÉHÝ?çZ±°êæè?é ©Údecimal)r rT Úrandr( rP r ©r; rç rå r% r% r& Ú test_randº s
|
þzTestRandomDist.test_randc C s* t | j¡ t ¡ }d}t||dd d S )Nró rù rú )r rT rü r rý r% r% r& Útest_rand_singleton s z"TestRandomDist.test_rand_singletonc C sl t | j¡ t dd¡}t ddgddgddgg¡}t||d d
|
t | j¡ t ¡ }t||d d d
|
d S ) NrM r çÿ!ã<Oqõ?ç4ðu,Íû?çÁ³½3Ûû÷?çL«I/DÍ¿çà@ËÀ? @çÿ HSÓ\@rù rú ©r r ©r rT Zrandnr( rP r rý r% r% r& Ú
|
test_randnÈ s
|
þ zTestRandomDist.test_randnc C sF t | j¡ t jdddd}t ddgddgd d
|
gg¡}t|| d S ) Néÿÿÿéc ©rM r r é rM éÌÿÿÿé) éÐÿÿÿé¾ÿÿÿ)r rT r9 r( rP r rý r% r% r& Ú test_randintÔ s
|
þzTestRandomDist.test_randintc C sÆ t | j¡ t 0}| t¡}t jdddd}tt|dk W 5 Q R X t ddgdd gd
|
d gg¡}t
|
|| t | j¡ t .}| t¡}t jd dd}tt|dk W 5 Q R X t
|
||d d S )Nr r
|
r r rH r rM r r r r éÆ r ) r rT r r ÚDeprecationWarningÚrandom_integersr r r( rP r )r; r¤ r¥ rç rå r% r% r& Útest_random_integersÜ s
|
|
þ
|
|
z#TestRandomDist.test_random_integersc C sª t | j¡ t | j¡}|jdd}t t¡jdkrXtjddgddgdd ggtj d
|
}n"tjd d gd dgddggtj d
|
}t
|
|| | | j¡ | ¡ }t
|
||d d S )Nr r iÿÿÿiÁª4OiÿË+i³RºKiriFbÀqiÀ rK l ÿK+W+¥y l r&&ÌJÒ] l ; l ÕnBz l @XÓHT l D+l+ +Ý r ) r rT r8 r r( r¾ r\ rÀ rP rQ r )r; rU rç rå r% r% r& Ú test_tomaxintî s(
|
þþ
|
þý
|
z TestRandomDist.test_tomaxintc C sÈ t <}| t¡}t t d¡jt d¡j¡}tt |dk W 5 Q R X t d¡j}t
|
|| t P}| t¡}t d¡j }t |t d¡j|t d¡j¡}tt |dk W 5 Q R X t
|
|| d S )Nrâ rH ) r r r r r r( r¾ rÀ r r r rL rì )r; r¤ r¥ rç rå Ztyperr% r% r& Ú test_random_integers_max_int s
|
|
ÿ
|
|
ÿz+TestRandomDist.test_random_integers_max_intc C sZ t ¡ H t dt¡ tttjt d¡j tttjt d¡j t d¡j W 5 Q R X d S )NÚerrorrâ )
|
ÚwarningsÚcatch_warningsÚ simplefilterr r r r r( r¾ rÀ rF r% r% r& Ú test_random_integers_deprecated s
|
|
þ
|
|
þz.TestRandomDist.test_random_integers_deprecatedc C sj t | j¡ t d¡}t ddgddgddgg¡}t||dd t | j¡ t ¡ }t||d
|
dd d S ) Nr ró rô rõ rö r÷ rø rù rú r )r rT r r( rP r rý r% r% r& Útest_random_sample& s
|
|
þ z!TestRandomDist.test_random_samplec C s8 t | j¡ t dd¡}t ddddg¡}t|| d S )NrN r rM ©r rT Úchoicer( rP r rý r% r% r& Útest_choice_uniform_replace2 s z*TestRandomDist.test_choice_uniform_replacec C sD t | j¡ t jddddddgd}t ddddg¡}t|| d S )NrN çÙ?ç¹?©r^ rH r r rý r% r% r& Ú test_choice_nonuniform_replace8 s z-TestRandomDist.test_choice_nonuniform_replacec C s: t | j¡ t jdddd}t dddg¡}t|| d S )NrN rM F©Úreplacer rH r rý r% r% r& Ú test_choice_uniform_noreplace> s z,TestRandomDist.test_choice_uniform_noreplacec C sD t | j¡ t jdddddddgd}t ddd g¡}t|| d S )
|
NrN rM Fr" ç333333Ó?r ©r& r^ r rH r rý r% r% r& Ú test_choice_nonuniform_noreplaceD s z/TestRandomDist.test_choice_nonuniform_noreplacec C s@ t | j¡ t ddddgd¡}t ddddg¡}t|| d S )Nr2 r3 ÚcÚdrN r rý r% r% r& Útest_choice_nonintegerJ s z%TestRandomDist.test_choice_nonintegerc C sD t j}tt|dd tt|dd tt|ddgddggd 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 )NrB rM ç @rH r rN g Ð?r# r! rc rv rw Fr% rI )rB )rB rH r r) )r r r rE )r; rí r% r% r& Útest_choice_exceptionsP 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 )Nr" rx r Tr% Fr) rH rK r )r rM r! rc r )rM r rN r iöÿÿÿ)r r r2 r3 )r r( Zisscalarr r rP Úemptyrª ÚtupleÚndimÚitemr rr r? r9 r rE )r; r^ r2 Zarrr< r% r% r& Útest_choice_return_shapec sL ÿz'TestRandomDist.test_choice_return_shapec C s0 t dddg¡}d d d g}tttj||d d S )Né* rH r r# )r( rP r rE r r )r; r2 r^ r% r% r& Ú test_choice_nan_probabilities s
|
z,TestRandomDist.test_choice_nan_probabilitiesc C s| t d¡d }d|dd d
< t | j¡ tjdd|d d d
d}t | j¡ tjddt |d d d
¡d}t|| d S )Nr rO r. rH r rM r# )r( Úonesr rT r r~ r )r; r^ r r r% r% r& Ú test_choice_p_non_contiguous s z+TestRandomDist.test_choice_p_non_contiguousc C s( t | j¡ t d¡}d}t|| d S )Nr s
|
Uiÿ+Wf¥)r rT Ú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 ]`}t | j¡ |d dddddddddg
|
}t |¡ |}|dd ddddddddg
|
}t|| qFd S )Nc S s
|
t g ¡S r )r( rP ©rl r% r% r& Ú<lambda>¨ ó z-TestRandomDist.test_shuffle.<locals>.<lambda>c S s | S r r% r; r% r% r& r< © r= c S s t | ¡ t j¡S r )r( ÚasarrayÚastyperÒ r; r% r% r& r< ª r= c S s t | ¡ t j¡S r )r( r> r? r r; r% r% r& r< « r= c S s t | ¡ t j¡S r )r( r> r? Z complex64r; r% r% r& r< ¬ r= c S s t | ¡ t¡S r )r( r> r? rª r; r% r% r& r< r= 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% r; r% r% r& r< ® r= c S s t dd | D ¡S )Nc S s g | ] }||gqS r% r% r@ r% r% r& rC ¯ s rD )r( r> r; r% r% r& r< ¯ r= c S s t | | g¡jS r )r( ZvstackÚTr; r% r% r& r< ° r= c S s* t dd | D dtfdtfg¡ t j¡S )Nc S s g | ] }||fqS r% r% r@ r% r% r& rC ² s rD r2 r3 )r( r> r\ rÜ Zrecarrayr; r% r% r& r< ² s ÿþc S s( t dd | D dtdfdt jdfg¡S )Nc S s g | ] }||fqS r% r% r@ r% r% r& rC ¶ s rD r2 ©rH r3 )r( r> rª rÑ r; r% r% r& r< ¶ s
|
ÿÿrH r rM rN rO r é rÌ é r )r rT Úshuffler )r; ÚconvÚalistrç rå r% r% r& Ú test_shuffle¥ s$ ò
|
zTestRandomDist.test_shufflec 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 qVdd }d S ) Nr )rO rN rM rH rB é2 c S s t d¡}tttj| d S )NrM )r( rP r rD r rI rk r% r% r& Ú test_shuffle_invalid_objectsÎ s
|
zHTestRandomDist.test_shuffle_masked.<locals>.test_shuffle_invalid_objects) r( ÚmaZ masked_valuesZreshaper> r? Úcopyr rI r ÚsortedÚdataÚmask)r; r2 r3 Za_origZb_origrB rN r% r% r& Útest_shuffle_maskedÀ s "
|
ÿ
|
ÿz"TestRandomDist.test_shuffle_maskedc C s t | j¡ ddddddddd d
|
g
|
}t |¡}d
|
dd dddddddg
|
}t|| t | j¡ t ddddddddd d
|
g
|
¡j}t |¡}t|t |¡j t | j¡ d }ttt j| t | j¡ d }ttt j| d }d d
|
ddddddddg
|
}t | j¡ t |¡}t|| d S )NrH r rM rN rO r rG rÌ rH r Zabcdg333333ó?r ) r rT Z permutationr r( Z
|
atleast_2drE r Ú
|
IndexError)r; rK rç rå Zarr_2dZ bad_x_strZ bad_x_floatZ integer_valr% r% r& Útest_permutationÒ s(
|
|
|
|
z TestRandomDist.test_permutationc C sJ t | j¡ t jdddd}t ddgddgd d
|
gg¡}t||d d d S ) Nr" rx r r gAàÁKÄ?g ñØhA?g õÒo¿>gάXfÈ+q?gSâë1Ã$?g'8zX ?rù rú )r rT Úbetar( rP r rý r% r% r& Ú test_betaí s þÿzTestRandomDist.test_betac C sl t | j¡ t jdddd}t ddgddgd d
|
gg¡}t|| t | j¡ t dd¡}d}t|| d S ) Ng Zd;ßY@gÉv¾/Ý?r r é% é+ r5 é0 é. é- )r rT r r( rP r rý r% r% r& Ú test_binomialö s
|
þ
|
z TestRandomDist.test_binomialc C sH t | j¡ t jddd}t ddgddgdd gg¡}t||d
|
d d S ) NrM r r g+Qð]uðO@gëòìÇ+Q@gY7`´ZqP@gÈe(fG@gþv¯tR@gEIÈR@é rú )r rT Ú chisquarer( rP r rý r% r% r& Útest_chisquare s
|
þz TestRandomDist.test_chisquarec C sÀ t | j¡ t ddg¡}t j|dd}t ddgddggd d
|
gd d ggd dgddggg¡}t||dd t ddg¡}ttt j| t | j¡ t ddg¡}t |¡}t||d dd d S )Nç I<ÝI@ç÷É EZßC@r r gÉi{ ßsá?gn, åAÝ?göÇ]^óã?gpDéBØ?gȧ#¼yªá?gr°¸ «Ü?g«1UÞâ?gÈѨUCÚ?g£
^,%÷â?g»ôB§µÚ?gÚOoX;â?gMì`!OÛ?rù rú çHáz®Gá?ç¼Ø²Ò¼r )r rT r( rP Ú dirichletr r rE )r; Ú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 )Nrb rc rH rn r ro rp )
|
r( rP r r rf rq rr r rD rs rt r% r% r& Útest_dirichlet_size s z"TestRandomDist.test_dirichlet_sizec C s t ddg¡}tttj| d S )Nrd re )r( rP r rE r rf )r; rg r% r% r& Útest_dirichlet_bad_alpha* s z'TestRandomDist.test_dirichlet_bad_alphac C sf t dddg¡}|d d d
}t | j¡ tj|dd}t | j¡ tjt |¡dd}t|| d S )Nrb g ð¿rc r r r )r( rP r rT rf r~ r )r; r2 rg r r r% r% r& Ú#test_dirichlet_alpha_non_contiguous/ s ÿz2TestRandomDist.test_dirichlet_alpha_non_contiguousc C sH t | j¡ t jddd}t ddgddgdd gg¡}t||d
|
d d S ) Ngï8EGrùñ?r r gN÷·Uñ?gbI0æð?gùVU\õº@g:ì3ù@g© ºUýå?g"qoû?rù rú )r rT Ú exponentialr( rP r rý r% r% r& Útest_exponential9 s
|
þz TestRandomDist.test_exponentialc C s& t tjddd tttjdd d S ©Nr ©Úscaleç )r r rl r rE rF r% r% r& Útest_exponential_0A s z!TestRandomDist.test_exponential_0c C sJ t | j¡ t jdddd}t ddgddgd d
|
gg¡}t||d d d S ) Né éM r r g 7¶ ó?gµ`jü?g6¶"+÷?gܳ2>ó?gcÍ>+Yð?g¡_}áSõ?rù rú )r rT Úfr( rP r rý r% r% r& Útest_fE s
|
þzTestRandomDist.test_fc C sJ t | j¡ t jdddd}t ddgddgd d
|
gg¡}t||d d d S ) NrO rM r r gÔGMç8@g §âÈ<@g ´":@gÂvÀñÇ#)@gZQø·?@gr[¦@@é rú )r rT Úgammar( rP r rý r% r% r& Ú
|
test_gammaM s
|
þzTestRandomDist.test_gammac C s* t tjdddd tttjddd d S )Nr )rr rp rq )r r rx r rE rF r% r% r& Ú test_gamma_0U s zTestRandomDist.test_gamma_0c C sD t | j¡ t jddd}t ddgddgddgg¡}t|| d S ) Nç_c97Ý¿?r r rÌ rG é rO rs )r rT r r( rP r rý r% r% r& Útest_geometricY s
|
þz TestRandomDist.test_geometricc C s t ttjd t ttjdgd t ttjd t ttjdgd t 6}| t¡ t ttjtj t ttjtjgd W 5 Q R X d S )Nrv r rw ) r rE r r r r r r( r` ©r; r¤ r% r% r& Útest_geometric_exceptionsa s
|
z(TestRandomDist.test_geometric_exceptionsc C sJ t | j¡ t jdddd}t ddgddgd d
|
gg¡}t||d d d S ) Nr{ ç @r ©Úlocrp r gåûØßÉ?g¨s0í Ö?gÊ ð#0÷¿gï(ôx÷¿g-¶KÇD´ñ?géůp`@æ¿rù rú )r rT Úgumbelr( rP r rý r% r% r& Ú test_gumbelk s
|
þzTestRandomDist.test_gumbelc C s& t tjddd tttjdd d S rn )r r r r rE rF r% r% r& Ú test_gumbel_0s s z TestRandomDist.test_gumbel_0c C s t | j¡ t jddddd}t ddgddgddgg¡}t|| t jdd d
|
d d}t d
|
d
|
d
|
d
|
g¡}t|| t jd d d d d}t d d d d g¡}t|| t jd dd
|
d d}t d d d d g¡}t|| t jd d d d d}t d d d d g¡}t|| d S )Ng333333$@g @rw r r r rH rO r rM rN rù rs )r rT r r( rP r rý r% r% r& Útest_hypergeometricw s&
|
þ
|
|
|
|
z"TestRandomDist.test_hypergeometricc C sJ t | j¡ t jdddd}t ddgddgd d
|
gg¡}t||d d d S ) Nr{ r r r gð¥ bÙOå?g¹Â¸ëÉçà?gpJmµú @g{KÏ*Ët @g:^w-«¿gY~oõûû?rù rú )r rT Úlaplacer( rP r rý r% r% r& Ú test_laplace s
|
þzTestRandomDist.test_laplacec C s& t tjddd tttjdd d S rn )r r r r rE rF r% r% r& Útest_laplace_0 s z TestRandomDist.test_laplace_0c C sJ t | j¡ t jdddd}t ddgddgd d
|
gg¡}t||d d d S ) Nr{ r r r g#K-zñ?gîJ¬ë?g92ÄÜ @gرã÷[@gê#+^ÑÀË¿g¢¥-óâ@rù rú )r rT Úlogisticr( rP r rý r% r% r& Ú test_logistic s
|
þz TestRandomDist.test_logisticc C sJ t | j¡ t jdddd}t ddgddgd d
|
gg¡}t||d d d S ) Nr{ r r )ÚmeanÚsigmar gòÚÉ0@goÎ,+4FB@gù)Ê6@g§5 éêæ?gnþxNnP@g
©)úµU@r_ rú )r rT Ú lognormalr( rP r rý r% r% r& Útest_lognormal¥ s
|
þz TestRandomDist.test_lognormalc C s& t tjddd tttjdd d S )Nr )r rH rq )r r r r rE rF r% r% r& Útest_lognormal_0 s z TestRandomDist.test_lognormal_0c C sD t | j¡ t jddd}t ddgddgddgg¡}t|| d S )NgÆ@õí?r )r^ r r r r| rM )r rT r r( rP r rý r% r% r& Útest_logseries± s
|
þz TestRandomDist.test_logseriesc C s t d¡dkstd S )Nr rH )r r r£ rF r% r% r& Útest_logseries_zero¹ s z"TestRandomDist.test_logseries_zeroÚvaluerg rB ç ð?g @c
|
C s t jdd t t¡ t |¡ W 5 Q R X t t¡ t t |gd ¡¡ W 5 Q R X t t¡& t t |gd ¡d d d
¡ W 5 Q R X W 5 Q R X d S )NÚignore©Úinvalidr r )r( Úerrstater
r rE r r rP )r; r r% r% r& Útest_logseries_exceptions¼ s z(TestRandomDist.test_logseries_exceptionsc C s t | j¡ t jddgd dd}t ddddd d gdd d
|
d d d ggddddd dgd d ddddggddd dd dgdddd ddggg¡}t|| d S ) Nr çUUUUUUÅ?r r r rN rM rO r rÌ rH r )r rT r r( rP r rý r% r% r& Útest_multinomialÈ s ÿÿÿüz TestRandomDist.test_multinomialc C s® t | j¡ d}ddgddgg}d}t |||¡}t ddgddggd d
|
gd d ggd dgddggg¡}t||dd t ||¡}t ddg¡}t||dd ddg}ddgddgg}ttt j|| tt j||dd t t
|
t j||dd tjddgddggtj d}t .}t ||¡ | t¡}t|dks6tW 5 Q R X t d¡}t d¡}t t
|
t j||dd t t
|
t jt d ¡| t t
|
t j|t d¡ t t
|
t j|t d ¡ d S ) N)r{ r rH r r g4¸Výj÷?g¾±¥y'@g÷I1õù?gSÙBï#@gû
ªe<@g ÒÔ4W(@g6R¿û?gò9r
|
v"@gÑæ?g{§×ïÂ#@gò4Íx¿gÇçú"öf"@rù rú gN}©O6¦ì?gÐÇYM÷W"@r r )Z check_validÚraiser" rK Úother)r rH rH rM )r rT Zmultivariate_normalr( rP r r r r r rE r r r r r£ r] Zeyer0 ) r; r Zcovr rç rå r¤ r¥ Úmur% r% r& Útest_multivariate_normalÓ sd
|
ÿÿÿü
|
ÿ ÿ
|
|
|
ÿ ÿ ÿ ÿz'TestRandomDist.test_multivariate_normalc C sF t | j¡ t jdddd}t ddgddgd d
|
gg¡}t|| d S ) Nr g|ò°Pk¿?r )Únr^ r iP iI i| ic i i )r rT r r( rP r rý r% r% r& r s
|
þz%TestRandomDist.test_negative_binomialc C sJ t :}| t¡ tttjdtj tttjdtjgd W 5 Q R X d S )Nr r ) r r r r rE r r r( r` r~ r% r% r& Ú!test_negative_binomial_exceptions s
|
|
|
ÿz0TestRandomDist.test_negative_binomial_exceptionsc C sÊ t | j¡ t jdddd}t ddgddgdd gg¡}t||d
|
d t jd d dd}t ddgddgddgg¡}t||d
|
d t | j¡ t jdddd}t ddgddgddgg¡}t||d
|
d d S )NrO r )ÚdfÚnoncr gôûÖGë7@g!© ÈÜ´*@g1}z9?@g¹+驸0@g26E_r#@g¦I!ó1@rw rú r rc g÷ô ÷?gT¶DÃ?gÀõ?T?gÍ+òqnlð?gÓ~|òùDÕ?gë°¬òÇÃ?r g{¿0¾1#@gp§²rs'@gþ~;ïÑÓ$@g"dI @gö°Õöë÷*@g Ñ}'Á,@)r rT Únoncentral_chisquarer( rP r rý r% r% r& Útest_noncentral_chisquare s(
|
þ
|
þ
|
þz(TestRandomDist.test_noncentral_chisquarec C sL t | j¡ t jddddd}t ddgdd gd
|
d gg¡}t||d d d S )NrO r rH r )ÚdfnumÚdfdenr£ r g^ýíå~ö?g/+c¢äÕ?g"¡®. @gü÷´ @g×7í§þÛ?g»E·Öò?rw rú )r rT Ú noncentral_fr( rP r rý r% r% r& Útest_noncentral_f* s
|
ÿ
|
þz TestRandomDist.test_noncentral_fc C s0 t | j¡ t jddtjd}t |¡s,td S )NrO r )r¦ r§ r£ )r rT r¨ r( r` Úisnanr£ )r; rç r% r% r& Útest_noncentral_f_nan3 s z$TestRandomDist.test_noncentral_f_nanc C sJ t | j¡ t jdddd}t ddgddgd d
|
gg¡}t||d d d S ) Nr{ r r r gí&&n@gO»G_Ê @gÜ~w ²ø@gtÁÜûw]Õ¿gm&¨5ú½@gñ$È>Û@rù rú )r rT Únormalr( rP r rý r% r% r& Ú test_normal8 s
|
þzTestRandomDist.test_normalc C s& t tjddd tttjdd d S rn )r r r¬ r rE rF r% r% r& Ú test_normal_0@ s z TestRandomDist.test_normal_0c C sL t | j¡ t jddd}t ddgddgdd gg¡}tjj||d
|
d d S ) Nr{ r ©r2 r gÅWò I£@g ùÔ¨y@gy£]Ó0Ag1~ÖÒ\AgåËíãa@gÑY
|
²7Aé )Znulp)r rT Úparetor( rP ZtestingZ assert_array_almost_equal_nulprý r% r% r& Ú test_paretoD s þÿ
|
zTestRandomDist.test_paretoc C sD t | j¡ t jddd}t ddgddgddgg¡}t|| d S )Nr{ r )Úlamr r rH )r rT r r( rP r rý r% r% r& Ú test_poissonS s
|
þzTestRandomDist.test_poissonc C s t d¡j}d}tttj| tttj|gd tttj| tttj|gd t 6}| t ¡ tttjt j
|
tttjt j
|
gd W 5 Q R X d S )Nrâ rB r ) r( r¾ rÀ r rE r r r r r r` )r; ZlambigZlamnegr¤ r% r% r& Útest_poisson_exceptions[ s
|
z&TestRandomDist.test_poisson_exceptionsc C sH t | j¡ t jddd}t ddgddgdd gg¡}t||d
|
d d S ) Nr{ r r¯ gõ;'û?gÜ´èwÞ*?g× 5Ø?gF¡GÙ?g ÆËD ]?gýÞ$`°ÉÀ?rù rú )r rT Úpowerr( rP r rý r% r% r& Ú
|
test_powerg s
|
þzTestRandomDist.test_powerc C sH t | j¡ t jddd}t ddgddgdd gg¡}t||d
|
d d S ) Nr r )rp r g"íu¨ÈÆ+@gîN$MBÄ*@g7v8Bô4@gêÿª55@geã &@g~Ó£ÌZ1@rw rú )r rT Úrayleighr( rP r rý r% r% r& Ú test_rayleigho s
|
þz TestRandomDist.test_rayleighc C s& t tjddd tttjdd d S rn )r r r¸ r rE rF r% r% r& Útest_rayleigh_0w s z TestRandomDist.test_rayleigh_0c C sF t | j¡ t jdd}t ddgddgddgg¡}t||d d
|
d S ) Nr r gMo³DL®è?g6Ï( [9ÀgðºrF=òí?gÉM' - ÀgóV¼ëûÀgVA:ìQyÇ?rù rú )r rT Zstandard_cauchyr( rP r rý r% r% r& Útest_standard_cauchy{ s
|
þz#TestRandomDist.test_standard_cauchyc C sF t | j¡ t jdd}t ddgddgddgg¡}t||d d
|
d S ) Nr r g°öÜÜî?gc!z¨ì?g&N¥"@g¬NIàÇ@gFÇï>úã?g¡~MEø?rù rú )r rT Zstandard_exponentialr( rP r rý r% r% r& Útest_standard_exponential s
|
þz(TestRandomDist.test_standard_exponentialc C sH t | j¡ t jddd}t ddgddgdd gg¡}t||d
|
d d S ) NrM r )rr r g)§@g¾§Ä¾¤@gd],qÂ@gPkÌ{@g9B #1 @gÔÚv÷ @rw rú )r rT Ústandard_gammar( rP r rý r% r% r& Útest_standard_gamma s
|
þz"TestRandomDist.test_standard_gammac C s& t tjddd tttjdd d S )Nr )rr rq )r r r½ r rE rF r% r% r& Útest_standard_gamma_0 s z$TestRandomDist.test_standard_gamma_0c C sF t | j¡ t jdd}t ddgddgddgg¡}t||d d
|
d S ) Nr r r r r r r r rù rú )r rT r r( rP r rý r% r% r& Útest_standard_normal s
|
þz#TestRandomDist.test_standard_normalc C s0 t | j¡ t ¡ }t d¡}t||dd d S )Nr rù rú r rý r% r% r& Útest_randn_singleton s
|
z#TestRandomDist.test_randn_singletonc C sH t | j¡ t jddd}t ddgddgdd gg¡}t||d
|
d d S ) Nr r )r¢ r g\VÔHÂï?gßhUÍ%¶¿gÿH$ðMÏõ?güøüL³á¿gV«qdz¥Ç¿gãÕýã?rù rú )r rT Ú
|
standard_tr( rP r rý r% r% r& Útest_standard_t¥ s
|
þz TestRandomDist.test_standard_tc C sL t | j¡ t jddddd}t ddgdd gd
|
d gg¡}t||d d d S )Ng{®Gáz@gö(\Âu$@g×£p=
|
W4@r )ÚleftÚmodeÚrightr g:§}Â\)@ghرTjÓ(@gǨL30@g6ÚÅA0@g6¤õsh&@gí<áþ,@rw rú )r rT Ú
|
triangularr( rP r rý r% r% r& Útest_triangular s
|
ÿ
|
þz TestRandomDist.test_triangularc C sJ t | j¡ t jdddd}t ddgddgd d
|
gg¡}t||d d d S ) Nç®Gáz®ó?g®Gáz%@r )ÚlowÚhighr g>#¥HÃö@g £¨uºó@gï§Ý#@g}d]#@gøN¶õ@gw )*ó @rù rú )r rT Úuniformr( rP 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 )Nrs r rH g Ø
W4vC)rÊ rË )
|
r( Zfinfor¿ rÀ r rÌ r Ú OverflowErrorÚinfÚ nextafter)r; ZfminZfmaxÚfuncr% r% r& Útest_uniform_range_bounds¾ s z(TestRandomDist.test_uniform_range_boundsc 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 ©rD rF r% r% r& Ú __float__Ö s zQTestRandomDist.test_scalar_exception_propagation.<locals>.ThrowingFloat.__float__N)rX rY rZ rÔ r% r% r% r& Ú ThrowingFloatÕ s rÕ r c @ s e Zd Zdd ZdS )zITestRandomDist.test_scalar_exception_propagation.<locals>.ThrowingIntegerc S s t d S r rÓ rF r% r% r& Ú__int__Þ s zQTestRandomDist.test_scalar_exception_propagation.<locals>.ThrowingInteger.__int__N)rX rY rZ rÖ r% r% r% r& ÚThrowingIntegerÝ s r× rH ) r( ZndarrayrP rÜ r rD r rÌ r )r; rÕ Zthrowing_floatr× Z throwing_intr% r% r& Ú!test_scalar_exception_propagationÎ s
|
ÿz0TestRandomDist.test_scalar_exception_propagationc C sJ t | j¡ t jdddd}t ddgddgd d
|
gg¡}t||d d d S ) NrÉ ç¤p=
|
×£ø?r ©r Úkappar gL×üZI@g¦¢'N"@gÀ·
þkrØ?gk=m@g¹Úó?gÜ^Ì=\ý?rù rú ©r rT Úvonmisesr( rP r rý r% r% r& Ú test_vonmisesä s
|
þz TestRandomDist.test_vonmisesc C s2 t | j¡ t jdddd}tt |¡ ¡ d S )Nrg g g5_PG>é@B rÚ )r rT rÝ r r( Úisfiniterj ©r; Úrr% r% r& Útest_vonmises_smallì s z"TestRandomDist.test_vonmises_smallc C s> t | j¡ t jdddd}t dddg¡}t||dd d S )
|
Nrg g ÐcArM rÚ g AÔü^>?g @ÌR7?g x¢.¿rÌ rú rÜ rý r% r% r& Útest_vonmises_largeò s þz"TestRandomDist.test_vonmises_largec C s. t | j¡ t jdtjd}tt |¡ d S )Nrg )r rÛ )r rT rÝ r( r` r rª rá r% r% r& Útest_vonmises_nanû s z TestRandomDist.test_vonmises_nanc C sJ t | j¡ t jdddd}t ddgddgd d
|
gg¡}t||d d d S ) NrÉ rÙ r )r rp r gÉZ¥¢@g-ï g@g¦8
|
ÄÖmÖ?gÃÎVP"ø?g`Vô áÎ?gx©^½&3Ì?rw rú )r rT Úwaldr( rP r rý r% r% r& Ú test_wald s
|
þzTestRandomDist.test_waldc C sH t | j¡ t jddd}t ddgddgdd gg¡}t||d
|
d d S ) NrÉ r r¯ g@Ý6ï?g¦Óé]Aí?g(Ã1Ú¥Rþ?gù¤U þ?gÍ2§_uå?g^}-Qö?rù rú )r rT Úweibullr( rP r rý r% r% r& Ú test_weibull s
|
þzTestRandomDist.test_weibullc C s: t | j¡ tt jdddt d¡ ttt jdd d S )Nr rs r¯ rq )r2 )r rT r rè r( r] r rE rF r% r% r& Útest_weibull_0 s z TestRandomDist.test_weibull_0c C sD t | j¡ t jddd}t ddgddgddgg¡}t|| d S ) NrÉ r r¯ éB é rH rM r_ )r rT r r( rP r rý r% r% r& Ú test_zipf s
|
þzTestRandomDist.test_zipfN)[rX rY rZ r rþ rÿ r r r r r r r r r$ r' r* r- r/ r4 r6 r8 r: rL rT rV rX r^ ra rh ri rj rk rm rr rv ry rz r} r r r
r r r r r r r r r
rï Z parametrizer( rÐ r` r r r r r¡ r¥ r© r« r r® r² r´ rµ r· r¹ rº r» r¼ r¾ r¿ rÀ rÁ rà rÈ rÍ rÒ rØ rÞ rã rä rå rç ré rê rí r% r% r% r& rñ ³ s¦ .
|
|
|
3 rñ c @ s 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"dAS )BÚ TestBroadcastc C s
|
d| _ d S )NéÍ[rò rF r% r% r& r ! s zTestBroadcast.setup_methodc C s t | j¡ d S r )r rT rF r% r% r& Úset_seed$ s zTestBroadcast.set_seedc C sn dg}dg}t j}t dddg¡}| ¡ ||d |}t||dd | ¡ |||d }t||dd d S ) Nr rH çºlö×÷ á?ç_x ¤¥á?ç =å*BNà?rM rw rú )r rÌ r( rP rð r )r; rÊ rË rÌ rå rç r% r% r& rÍ ' s þzTestBroadcast.test_uniformc C s dg}dg}dg}t j}t dddg¡}| ¡ ||d |}t||dd tt||d | | ¡ |||d }t||dd tt|||d d S )
|
Nr rH rB g-rö´@g¦Y®lõ@g-µ`¦wý?rM rw rú )r r¬ r( rP rð r r rE )r; r rp Ú bad_scaler¬ rå rç r% r% r& r 7 s þzTestBroadcast.test_normalc C s dg}dg}dg}dg}t j}t dddg¡}| ¡ ||d |}t||d d
|
tt||d | tt||d | | ¡ |||d }t||d d
|
tt|||d tt|||d d S ) NrH r rB rI gnÍQVfÉ?gV©KB³?g v®`køÏ?rM rw rú )r rW r( rP rð r r rE )r; r2 r3 Úbad_aZbad_brW rå rç r% r% r& rX J s& þzTestBroadcast.test_betac C sX dg}dg}t j}t dddg¡}| ¡ ||d }t||dd tt||d d S © NrH rB gö.Ug¬Zè?gi Prqè?gçÅE KÌæ?rM rw rú )r rl r( rP rð r r rE )r; rp rô rl rå rç r% r% r& rm ` s þ z TestBroadcast.test_exponentialc C sX dg}dg}t j}t dddg¡}| ¡ ||d }t||dd tt||d d S rö )r r½ r( rP rð r r rE )r; rr Ú bad_shapeZ std_gammarå rç r% r% r& r¾ m s þ z!TestBroadcast.test_standard_gammac C s dg}dg}dg}dg}t j}t dddg¡}| ¡ ||d |}t||d d
|
tt||d | tt||d | | ¡ |||d }t||d d
|
tt|||d tt|||d d S ) NrH r rB rI gö.Ug¬Zø?gi Prqø?gçÅE KÌö?rM rw rú )r rx r( rP rð r r rE )r; rr rp r÷ rô rx rå rç r% r% r& ry z s& þzTestBroadcast.test_gammac C s dg}dg}dg}dg}t j}t dddg¡}| ¡ ||d |}t||d d
|
tt||d | tt||d | | ¡ |||d }t||d d
|
tt|||d tt|||d d S ) NrH r rB rI gbyÊé?g»åoðÄë?gT¹æ7Í@rM rw rú )r ru r( rP rð r r rE )r; r¦ r§ Ú bad_dfnumÚ bad_dfdenru rå rç r% r% r& rv s& þzTestBroadcast.test_fc
|
C s dg}dg}dg}dg}dg}dg}t j}t ddd g¡}| ¡ ||d ||} t| |d
|
d t t |||tjgd ¡¡st t
|
t ||d || t
|
t ||d || t
|
t ||d || | ¡ |||d |} t| |d
|
d t
|
t |||d | t
|
t |||d | t
|
t |||d | | ¡ ||||d } t| |d
|
d t
|
t ||||d t
|
t ||||d t
|
t ||||d d S ) Nr rM rN r rB rI gÈr±^G"@g/ *@g°Ñ!@rw rú ) r r¨ r( rP rð r rj rª r` r£ r rE )
|
r; r¦ r§ r£ rø rù Úbad_noncZnonc_frå rç r% r% r& r© ¦ s< þ$z TestBroadcast.test_noncentral_fc C s: | ¡ t ddg¡}tjddddd}t||dd d S )Ng¬ªaz@gn4M;î%é?rx r r rw rú )rð r( rP r r¨ r )r; rå rç r% r% r& Útest_noncentral_f_small_dfÉ s z(TestBroadcast.test_noncentral_f_small_dfc C sX dg}dg}t j}t dddg¡}| ¡ ||d }t||dd tt||d d S ) NrH rB g~¼}ÐN?â?g$Fà?g'rSèÀ?rM rw rú )r r` r( rP rð r r rE )r; r¢ Úbad_dfr` rå rç r% r% r& ra Ï s þ z TestBroadcast.test_chisquarec C s dg}dg}dg}dg}t j}t dddg¡}| ¡ ||d |}t||d d
|
tt||d | tt||d | | ¡ |||d }t||d d
|
tt|||d tt|||d d S ) NrH r rB rI g©ç wÌ "@g2 øêWR@gâRY@rM rw rú )r r¤ r( rP rð r r rE )r; r¢ r£ rü rú Znonc_chirå rç r% r% r& r¥ Ü s& þz'TestBroadcast.test_noncentral_chisquarec C sj dg}dg}t j}t dddg¡}| ¡ ||d }t||dd tt||d ttt j|d d S ) NrH rB gT Äò@g~Fl@gÆtøpð?rM rw rú )r r r( rP rð r r rE )r; r¢ rü Útrå rç r% r% r& rà ò s þ z TestBroadcast.test_standard_tc C s dg}dg}dg}t j}t dddg¡}| ¡ ||d |}t||dd tt||d | | ¡ |||d }t||dd tt|||d d S )
|
Nr rH rB g´®W!è@güOBº¦ÀgÈ}Ç?àý¿rM rw rú )r rÝ r( rP rð r r rE )r; r rÛ Z bad_kapparÝ rå rç r% r% r& rÞ s þzTestBroadcast.test_vonmisesc C sj dg}dg}t j}t dddg¡}| ¡ ||d }t||dd tt||d ttt j|d d S ) NrH rB g&8¾?ò?geâFXò?gªÕËð?rM rw rú )r r± r( rP rð r r rE )r; r2 rõ r± rå rç r% r% r& r² s þ zTestBroadcast.test_paretoc C sj dg}dg}t j}t dddg¡}| ¡ ||d }t||dd tt||d ttt j|d d S rö )r rè r( rP rð r r rE )r; r2 rõ rè rå rç r% r% r& ré ! s þ zTestBroadcast.test_weibullc C sj dg}dg}t j}t dddg¡}| ¡ ||d }t||dd tt||d ttt j|d d S ) NrH rB rñ rò ró rM rw rú )r r¶ r( rP rð r r rE )r; r2 rõ r¶ rå rç r% r% r& r· / s þ zTestBroadcast.test_powerc C s dg}dg}dg}t j}t dddg¡}| ¡ ||d |}t||dd tt||d | | ¡ |||d }t||dd tt|||d d S )
|
Nr rH rB g:¨×BKc±?gÖ3®k²?gî~qÉÁ?rM rw rú )r r r( rP rð r r rE )r; r rp rô r rå rç r% r% r& r = s þzTestBroadcast.test_laplacec C s dg}dg}dg}t j}t dddg¡}| ¡ ||d |}t||dd tt||d | | ¡ |||d }t||dd tt|||d d S )
|
Nr rH rB gÈ'£ZyÑ?g÷dSO=Ñ?gÏ3Lõ1³Õ?rM rw rú )r r r( rP rð r r rE )r; r rp rô r rå rç r% r% r& r P s þzTestBroadcast.test_gumbelc C sª dg}dg}dg}t j}t dddg¡}| ¡ ||d |}t||dd tt||d | | ¡ |||d }t||dd tt|||d tt d
|
d ¡d
|
d S ) Nr rH rB gÂy ±ÕÀ?gªÀJõRÁ?gþ
|
~Æ&£?rM rw rú r rg ) r r r( rP rð r r rE r )r; r rp rô r rå rç r% r% r& r c s" þzTestBroadcast.test_logisticc C sÀ dg}dg}dg}t j}t dddg¡}| ¡ ||d |}t||dd tt||d | ttt j|d | | ¡ |||d }t||dd tt|||d ttt j||d d S )
|
Nr rH rB gÄñòÏH"@g¼ùÜ®Í @gvv ý²:@rM rw rú )r r r( rP rð r r rE )r; r r Z bad_sigmar rå rç r% r% r& r w s$ þz TestBroadcast.test_lognormalc C sX dg}dg}t j}t dddg¡}| ¡ ||d }t||dd tt||d d S ) NrH rB gݱlËo½ó?gé/Ó´Æó?gz~ßRó?rM rw rú )r r¸ r( rP rð r r rE )r; rp rô r¸ rå rç r% r% r& r¹ s þ zTestBroadcast.test_rayleighc C s dg}dg}dg}dg}t j}t dddg¡}| ¡ ||d |}t||d d
|
tt||d | tt||d | ttt j|d | ttt j|d | | ¡ |||d }t||d d
|
tt|||d tt|||d tt|d d tt|dd d S ) Nr rH r rI g@mò e¾?g¸`Iß¿?g¸|2í?rM rw rú rg )r ræ r( rP rð r r rE )r; r rp Zbad_meanrô ræ rå rç r% r% r& rç s. þzTestBroadcast.test_waldc C s dg}dg}dg}dg}dg}|d \}}t j}t dddg¡} | ¡ ||d ||}
|
t|
|
| dd tt||d || tt||d || tt||d || | ¡ |||d |}
|
t|
|
| dd tt|||d | tt|||d | tt|||d | | ¡ ||||d }
|
t|
|
| dd tt||||d tt||||d tt||||d tt|d
|
d d tt|d
|
d d tt|d
|
d
|
d
|
d S )NrH rM r rN g,âP;bD @gºÁ±È%G @g,~S²¨ @rw rú r rg g 4@g 9@)r rÇ r( rP rð r r rE ) r; rÄ rÆ rÅ Z bad_left_oneZ bad_mode_oneZ bad_left_twoZ bad_mode_tworÇ rå rç r% r% r& rÈ ³ sL þÿÿ
|
ÿz TestBroadcast.test_triangularc C sä dg}dg}dg}dg}dg}t j}t dddg¡}| ¡ ||d |}t|| tt||d | tt||d | tt||d | | ¡ |||d }t|| tt|||d tt|||d tt|||d d S )NrH r rB ç ø?rM )r r r( rP rð r r rE ) r; r r^ Úbad_nÚ bad_p_oneÚ bad_p_twoZbinområ rç r% r% r& r^ Û s&
|
|
zTestBroadcast.test_binomialc C sä dg}dg}dg}dg}dg}t j}t dddg¡}| ¡ ||d |}t|| tt||d | tt||d | tt||d | | ¡ |||d }t|| tt|||d tt|||d tt|||d d S )NrH r rB rþ r rM )r r r( rP rð r r rE ) r; r r^ rÿ r r Z neg_binområ rç r% r% r& r ò s&
|
|
z$TestBroadcast.test_negative_binomialc C sx t ¡ j}dg}dg}|d g}t j}t dddg¡}| ¡ ||d }t|| tt ||d tt ||d d S )NrH rB r r rM )
|
r r8 Z_poisson_lam_maxr r( rP rð r r rE )r; Zmax_lamr³ Z bad_lam_oneZ bad_lam_twor rå rç r% r% r& r´ s
|
|
|
zTestBroadcast.test_poissonc C s dg}dg}t j}t dddg¡}| ¡ ||d }t|| tt||d tjdd( tt|tj tt|ddtj g W 5 Q R X d S )Nr r rH rM r r )
|
r r r( rP rð r r rE r r` )r; r2 rõ r rå rç r% r% r& rí s
|
zTestBroadcast.test_zipfc C sj dg}dg}dg}t j}t dddg¡}| ¡ ||d }t|| tt||d tt||d d S )Nr rB rþ r rM )r r r( rP rð r r rE )r; r^ r r Zgeområ rç r% r% r& r} & s
|
z TestBroadcast.test_geometricc C sÚ dg}dg}dg}dg}dg}dg}dg}t j}t dddg¡} | ¡ ||d ||}
|
t|
|
| tt||d || tt||d || tt||d || tt||d || | ¡ |||d |}
|
t|
|
| tt|||d | tt|||d | tt|||d | tt|||d | | ¡ ||||d }
|
t|
|
| tt||||d tt||||d tt||||d tt||||d tt|ddd tt|ddd tt|ddd tt|ddd
|
d S ) NrH r rB rI r rN rM r r é )r r r( rP rð r r rE ) r; ZngoodZnbadZnsampleZ bad_ngoodZbad_nbadZbad_nsample_oneZbad_nsample_twoZ hypergeområ rç r% r% r& r 3 sD
|
|
|
z!TestBroadcast.test_hypergeometricc C sj dg}dg}dg}t j}t dddg¡}| ¡ ||d }t|| tt||d tt||d d S )Nr r rB rH rM )r r r( rP rð r r rE )r; r^ r r r rå rç r% r% r& r [ s
|
z TestBroadcast.test_logseriesN)#rX rY rZ r rð rÍ r rX rm r¾ ry rv r© 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î s@ # ( (rî zcan't start threadrã c @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd S ) Ú
|
TestThreadc C s t d| _d S )NrN )r> ÚseedsrF r% r% r& r l s zTestThread.setup_methodc sÊ ddl m t t| jf| ¡}t t| jf| ¡} fddt| j|D }dd |D dd |D t| j|D ]\}}t |¡| q|t ¡ j
|
j dkr¼t j dkr¼t|| n
|
t|| d S ) Nr )ÚThreadc s& g | ] \}} t |¡|fd qS ))ÚtargetÚargs)r r8 )rA r< Úo©r Úfunctionr% r& rC v s ÿz-TestThread.check_function.<locals>.<listcomp>c S s g | ] }| ¡ qS r% )Ústart©rA rl r% r% r& rC x s c S s g | ] }| ¡ qS r% )Újoinr r% r% r& rC y s rN Úwin32)Ú threadingr r( r0 r r Úzipr r8 ZintprL Úitemsizer× Úplatformr r )r; r
|
ÚszZout1Zout2rý r< r r% r r& Úcheck_functiono s
|
ÿ zTestThread.check_functionc C s dd }| j |dd d S )Nc S s | j dd|d< d S )Né' r .)r¬ ©r- Úoutr% r% r& Ú
|
gen_random s z*TestThread.test_normal.<locals>.gen_random)r ©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©r r6 ro .)rl r( r7 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 r r .)r r r% r% r& r s z/TestThread.test_multinomial.<locals>.gen_random)r r r r r r% r% r& r s zTestThread.test_multinomialN)rX rY rZ r r r r r r% r% r% r& r i s
|
r c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
|
ÚTestSingleEltArrayInputc C s4 t dg¡| _t dg¡| _t dg¡| _d| _d S )Nr rM rN rF )r( rP ÚargOneÚargTwoÚargThreeÚtgtShaperF 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 )r rl r½ r` r r± rè r¶ r¸ r r r r r( rP r r rr r" )r; ÚfuncsÚ probfuncsrÑ 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 )Nr r )r rÌ r¬ rW rx ru r¤ rÝ r r r r ræ r r r( rP r r r rr r" )r; r# r$ rÑ r r r% r% r& Útest_two_arg_funcs³ s2 ú z*TestSingleEltArrayInput.test_two_arg_funcsc 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 )
|
r r¨ rÇ r r r r! r rr r" )r; r# rÑ r r% r% r& Útest_three_arg_funcsÎ s ÿz,TestSingleEltArrayInput.test_three_arg_funcsN)rX rY rZ r r% r& r( r% r% r% r& r s r c C sD t d¡ | \}}}tt |}||ddi}t|jt d¡k d S )Nrï r r râ )r rT Úgetattrr rL r( )r' Úfnamer rÛ ru rç r% r% r& Útest_integer_dtypeÞ s
|
|
|
|
r+ c C sd t d¡ | \}}}tt |}||ddi}tjdkr>| ¡ }t | t j
|
¡¡ ¡ }t ||k d S )Nrï r rß rÖ ) r rT r) r× rØ rÙ rÚ rÛ rÜ r( rÒ rÝ r )r' r* r rÛ ru rß rà r% r% r& Útest_integer_repeatæ s
|
|
|
|
r, c C s t t¡ tjdddgdd W 5 Q R X t t¡ tjddgddd W 5 Q R X t t¡ tjddgddgdd W 5 Q R X d S )NrH r( gffffffæ?)r rH r r )r
r rE r r r% r% r% r& Útest_broadcast_size_errorñ s r- c C sÐ t j td¡} | d¡ | ¡ \}}}|d d
dks<t||d d
}| |¡ |jdd}t |d |d t
|
|d d |d d t
|
|d d |d d t |d
|
|d
|
t |d |d d S ) Nr rH )r Fr¨ r, r- r. r/ r0 r1 ) r( r r8 r r Ú
|
__reduce__r£ r r r r )rU Zctorr Zstate_ar3 Zstate_br% r% r& Ú&test_randomstate_ctor_old_style_pickleû s
|
|
r/ c C sR t j d¡}|j}t j |¡ tt jjjjt |s8t
|
t j ¡ }||ksNt
|
d S r' ) r( r Z default_rngr, r* r¡ ZmtrandZ_randZ_bit_generatorrì r£ r) )r+ Zdef_bgÚbgZ second_bgr% r% r& Ú test_hot_swap s
|
r1 c C s t d}tj |¡ tjjdd}tj d¡ tjjdd}t| t| |d dks\t|d d |d d ksxt|d d |d d kstd S ) Nr Fr¨ rH r, r r- Úinc)r r( r r* r rT Úprintr£ )r+ r0 r- r« r% r% r& Útest_seed_alt_bit_gen s r4 c C sH t j ¡ }td}t j |¡ tjtdd t j |¡ W 5 Q R X d S )Nr zstate must be for a PCG64r ) r( r r r r* r
r rE r )r+ r- r0 r% r% r& Ú test_state_error_alt_bit_gen% s
|
|
r5 c C sÒ t j d¡ t j ddd¡}td}|j}t j |¡ t jjdd}|D ]}|| || ksHtqHt j d¡ t j ddd¡}t ||k¡rt|j}|d d |d d ks²t|d d |d d ksÎtd S ) NéÍ r é @r Fr¨ r- r2 )
|
r( r rT r9 r r- r* r r£ rj )r+ rÍ r0 r- Z state_directÚfieldZpcg_valsr« r% r% r& Útest_swap_worked. s r9 c C sL t j td¡ t j ddd¡}t j td¡}| ddd¡}t|| d S )Nr6 r r7 r )r( r r* r r9 r8 r )r+ Zsingleton_valsZrgZnon_singleton_valsr% r% r& Ú%test_swapped_singleton_against_direct@ s
|
r: )3rÚ r® r× r Únumpyr( r
Z numpy.testingr r r r r r r r r
|
Z numpy.randomr r r r7 r" r¾ r\ rÀ r# Zfixturer' r+ r4 r5 r[ rb r r¸ rñ rî rï rð r r r+ r, r- r/ r1 r4 r5 r9 r: r% r% r% r& Ú<module> s , ù
|
ù
|
ù
|
|
/5W " q O/E
|
|
