U
|
P ±d" ã @ sê d dl Z d dlZd dlZd dlmZmZmZmZm Z m
|
Z
|
m Z m Z m Z d dlmZ d dlZG dd dZG dd dZG dd d ZG d
|
d d ZG d d d ZG dd dZG dd dZejje ddG dd dZG dd dZdS )é N) Úassert_Ú assert_raisesÚ assert_equalÚ assert_warnsÚassert_no_warningsÚassert_array_equalÚassert_array_almost_equalÚsuppress_warningsÚIS_WASM)Úrandomc @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd S ) ÚTestSeedc C s< t j d¡}t| d¡d t j d¡}t| d¡d d S )Nr éè i¬ ì ÿÿ i£ )Únpr Ú RandomStater Úrandint©ÚselfÚs© r úUd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/random/tests/test_random.pyÚ test_scalar s zTestSeed.test_scalarc C s t j td¡}t| d¡d t j t d¡¡}t| d¡d t j dg¡}t| d¡d t j dg¡}t| d¡d d S )Né
|
r iÔ r iÍ r i )r r r Úranger r Úaranger r r r Ú
|
test_array s zTestSeed.test_arrayc C s$ t ttjjd t ttjjd d S )Nç à¿éÿÿÿÿ©r Ú TypeErrorr r r Ú
|
ValueError©r r r r Útest_invalid_scalar s z TestSeed.test_invalid_scalarc C sf t ttjjdg t ttjjdg t ttjjdg t ttjjdddg t ttjjdddg d S )Nr r l é é éþÿÿÿr r! r r r Útest_invalid_array% s
|
zTestSeed.test_invalid_arrayc C sX t ttjjtjg tjd t ttjjdddgg t ttjjdddgdddgg d S )N©Údtyper# r$ é é é é )r r r r r ÚarrayÚint64r! r r r Útest_invalid_array_shape- s
|
ÿÿz!TestSeed.test_invalid_array_shapeN)Ú__name__Ú
|
__module__Ú __qualname__r r r" 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 )Nr$ Úintr' )r ç à?r# r )r Úzerosr r Úbinomialr )r r6 Úpr r r Ú test_n_zero7 s zTestBinomial.test_n_zeroc C s t ttjdtj d S )Nr# )r r r r7 r Únanr! r r r Ú test_p_is_nan@ s zTestBinomial.test_p_is_nanN)r0 r1 r2 r9 r; r r r r r3 6 s r3 c @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd S ) ÚTestMultinomialc C s t dddg¡ d S )Néd çÉ?çé?©r Ú multinomialr! r r r Ú
|
test_basicF s zTestMultinomial.test_basicc C s t ddddddg¡ d S )Nr= r> r? ç r@ r! r r r Útest_zero_probabilityI 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éûÿÿÿr r+ )r r r r Úall)r Úxr r r Útest_int_negative_intervalL s $z*TestMultinomial.test_int_negative_intervalc C sÐ ddg}t tj d|t d¡¡jd t tj d|t d¡¡jd t tj d|t d¡¡jd t tj d|ddg¡jd t tj d|d¡jd t tj d|t d¡¡jd tttjjd|t d d S )Nr5 r# ©r# r$ r$ ©r$ r$ r$ ©r$ r$ )
|
r r r rA Úuint32Úshaper- r r Úfloat©r r8 r r r Ú test_sizeR s ÿÿzTestMultinomial.test_sizec C s t ttjjdddgg t ttjjddgdgg t ttjjddgdggdgdggg t ttjjdt ddgddgg¡ d S )Nr r r# )r r r r rA r- r! r r r Útest_multidimensional_pvals` s &z+TestMultinomial.test_multidimensional_pvalsN)r0 r1 r2 rB rD rH rP rQ r r r r r< E s
|
r< c @ s< e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd S )Ú TestSetStatec C s$ d| _ t | j ¡| _| j ¡ | _d S ©NiÒI)Úseedr r ÚprngÚ get_stateÚstater! r r r Ú setup_methodh s zTestSetState.setup_methodc C s< | j d¡}| j | j¡ | j d¡}tt ||k¡ d S )Né )rU ZtomaxintÚ set_staterW r r rF ©r ÚoldÚnewr r r rB m s zTestSetState.test_basicc C s@ | j jdd}| j | j¡ | j jdd}tt ||k¡ d S ©Nr) ©Úsize)rU Ústandard_normalrZ rW r r rF r[ r r r Útest_gaussian_resets s z TestSetState.test_gaussian_resetc C sR | j ¡ | j ¡ }| j jdd}| j |¡ | j jdd}tt ||k¡ d S r^ )rU ra rV rZ r r rF )r rW r\ r] r r r Ú test_gaussian_reset_in_media_resz 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 )Nr% rY r_ )rW rU ra rZ r r rF )r Ú old_stateÚx1Zx2Zx3r r r Ú test_backwards_compatibility
s z)TestSetState.test_backwards_compatibilityc C s | j dd¡ d S ©Nr5 )rU Únegative_binomialr! r r r Útest_negative_binomial s z#TestSetState.test_negative_binomialN) r0 r1 r2 rX rB rb rc rf ri r r r r rR g s rR c @ s| e Zd ZejjZejejej ej
|
ej ej ej ejejg Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Zdd ZdS )Ú TestRandintc C s t t| jdtd d S ©Nr# r' )r r ÚrfuncrN r! 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$ r# r' ) Úityper Úbool_ÚiinfoÚminÚmaxr r rl )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$ r# r ©r` r( )rn r ro rp rq rr r rl )r rs rt ru Ú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$ r# r' zMNo error should have been raised, but one was with the following message:
|
|
%s)
|
rn r ro rp rq rr rl Ú ExceptionÚAssertionErrorÚstr)r rs rt ru Úer r r Útest_full_range¹ s
|
þzTestRandint.test_full_rangec C s t j ¡ | jdd
D ]@}dD ]6}| jd|d|d}t| ¡ |k t| ¡ dk q q| jdddt jd}t| ¡ dk t| ¡ dk d S )Nr# )r* é rY r$ i rw r ) r r rT rn rl r rr rq ro )r rs ru Úvalsr r r Útest_in_bounds_fuzzÇ s
|
z TestRandint.test_in_bounds_fuzzc
|
C sò dd l }dddddddddd }| jdd
D ]r}tj d ¡ tjd
|
kr\| jdd d |d }n| jdd d |d ¡ }| |
|
tj ¡¡ ¡ }t |t |¡j |k q.tj d ¡ | jddd td
|
tj ¡}| |¡ ¡ }t |t t¡j |k d S )Nr Z@509aea74d792fb931784c4b0135392c65aec64beee12b0cc167548a2c3d31e71Z@7b07f1a920e46f6d0fe02314155a2330bcfd7635e708da50e536c5ebb631a7d4Z@e577bfed6c935de944424667e3da285012e741892dcb7051a8f1ce68ab05c92fZ@0fbead0b06759df2cfb55e43148822d4a1ff953c7eb19a5b08445a63bb64fa9eZ@001aac3a5acb935a9b186cbe14a1ca064b8bb2dd0b045d48abeacf74d0203404) ÚboolÚint16Úint32r. Úint8Úuint16rL Úuint64Úuint8r# iÒ Úlittler, r rw r$ )Úhashlibrn r r rT ÚsysÚ byteorderrl ZbyteswapÚsha256Úviewr
Ú hexdigestr r( Únamer )r r rx rs ÚvalÚresr r r Útest_repeatabilityÖ s, ø
|
|
z TestRandint.test_repeatabilityc C s` t j}t t j¡j}t t t j¡j¡}t t t j¡jd ¡}t jj|||d}t|| d S rk )r r. rp rr r r r r )r rs rx rt ru Ú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$ r# r' r( )rn r ro rp rq rr rl r r( r r4 ÚcompatÚlongr ÚhasattrÚtype)r rs rt ru Úsampler r r Ú test_respect_dtype_singleton s
|
z(TestRandint.test_respect_dtype_singletonN)r0 r1 r2 r r r rl ro r
r r r r rL r. r rn rm rv ry r~ r r r r r r r r rj s ÿ !rj 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& Zej d'ejej ¡ ej ¡ g¡d(d) Z ej d'ejej ¡ ej ¡ g¡ej d*d+d,g¡d-d. Z d/d0 Z d1d2 Z d3d4 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 Z0dSdT Z1dUdV Z2dWdX Z3dYdZ Z4d[d\ Z5d]d^ Z6d_d` Z7dadb Z8dcdd Z9dedf Z:dgdh Z;didj Z<dkdl Z=dmdn Z>dodp Z?dqdr Z@dsdt ZAdudv ZBdwdx ZCdydz ZDd{d| ZEd}d~ ZFdd ZGdd ZHdd ZId
d ZJdd ZKdd ZLdd ZMdd ZNdd ZOdd ZPdd ZQdd ZRdS )ÚTestRandomDistc C s
|
d| _ d S rS ©rT r! r r r rX % s zTestRandomDist.setup_methodc C sJ t j | j¡ t j dd¡}t ddgddgddgg¡}t||d d
|
d S ) Nr) r$ ç_®\±*Íã?ç¬îâ?ç<îpì?ç®,º¥oì?ç ÉHÝ?çZ±°êæè?é ©Údecimal)r r rT Zrandr- r ©r r Údesiredr r r Ú test_rand( s
|
þzTestRandomDist.test_randc C sJ t j | j¡ t j dd¡}t ddgddgddgg¡}t||d d
|
d S ) Nr) r$ çÿ!ã<Oqõ?ç4ðu,Íû?çÁ³½3Ûû÷?çL«I/DÍ¿çà@ËÀ? @çÿ HSÓ\@r¤ r¥ )r r rT Zrandnr- r r§ r r r Ú
|
test_randn0 s
|
þzTestRandomDist.test_randnc C sJ t j | j¡ t jjdddd}t ddgddgd d
|
gg¡}t|| d S ) Néÿÿÿéc ©r) r$ r_ é r) éÌÿÿÿé) éÐÿÿÿé¾ÿÿÿ)r r rT r r- r r§ r r r Ú test_randint8 s
|
þzTestRandomDist.test_randintc C sv t j | j¡ t 2}| t¡}t jjdddd}tt|dk W 5 Q R X t ddgdd gd
|
d gg¡}t
|
|| d S ) Nr± r² r³ r_ r# r´ r) rµ r¶ r· r¸ ) r r rT r ÚrecordÚDeprecationWarningÚrandom_integersr Úlenr- r ©r ÚsupÚwr r¨ r r r Útest_random_integers@ s
|
|
þz#TestRandomDist.test_random_integersc C sd t >}| t¡}tj t d¡jt d¡j¡}tt |dk W 5 Q R X t d¡j}t
|
|| d S )NÚlr# ) r rº r» r r r¼ rp rr r r½ r r¾ r r r Ú test_random_integers_max_intK s
|
|
ÿ z+TestRandomDist.test_random_integers_max_intc C s^ t ¡ L t dt¡ tttjjt d¡j tttjjt d¡j t d¡j W 5 Q R X d S )NÚerrorr )
|
ÚwarningsÚcatch_warningsÚ simplefilterr» r r r r¼ rp rr r! r r r Ú test_random_integers_deprecatedZ s
|
|
þ
|
|
þz.TestRandomDist.test_random_integers_deprecatedc C sH t j | j¡ t j d¡}t ddgddgddgg¡}t||dd d S )
|
Nr³ r r r r¡ r¢ r£ r¤ r¥ )r r rT r- r r§ r r r Ú test_randomh s
|
þzTestRandomDist.test_randomc C s< t j | j¡ t j dd¡}t ddddg¡}t|| d S )Nr* r$ r) ©r r rT Úchoicer- r r§ r r r Útest_choice_uniform_replacep s z*TestRandomDist.test_choice_uniform_replacec C sH t j | j¡ t jjddddddgd}t ddddg¡}t|| d S )Nr* çÙ?ç¹?©r8 r# r$ rÊ r§ r r r Ú test_choice_nonuniform_replacev s z-TestRandomDist.test_choice_nonuniform_replacec C s> t j | j¡ t jjdddd}t dddg¡}t|| d S )Nr* r) F©Úreplacer r# rÊ r§ r r r Ú test_choice_uniform_noreplace| s z,TestRandomDist.test_choice_uniform_noreplacec C sH t j | j¡ t jjdddddddgd}t ddd g¡}t|| d S )
|
Nr* r) FrÎ g333333Ó?r5 ©rÒ r8 r$ r# rÊ r§ r r r Ú test_choice_nonuniform_noreplace s
|
ÿz/TestRandomDist.test_choice_nonuniform_noreplacec C sD t j | j¡ t j ddddgd¡}t ddddg¡}t|| d S )NÚaÚbÚcÚdr* rÊ r§ r r r Útest_choice_noninteger s z%TestRandomDist.test_choice_nonintegerc C sF t jj}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 )Nr r) g @r# r$ r* g Ð?rÏ rÍ r> gñ?g¹¿FrÑ r% )r )r r# r rÔ )r r rË r r )r r r r r Útest_choice_exceptions s& ÿ ÿz%TestRandomDist.test_choice_exceptionsc C s< ddg}t t tjjddd¡ t t tjjddd¡ t t tjjdd|d¡ t t tjjdd|d¡ t t tjjddgdd¡ t tjjd gddd k t ddg¡}tjdtd }||d
|
< t tjj|dd|k t }t t tjjd|dd¡ t t tjjd|dd¡ t t tjjd|d|d¡ t t tjjd|d|d¡ t t tjjddg|dd¡ t tjjd g|ddj d
|
k t ddg¡}tjdtd }||d
|
< t tjj||dd
|
¡ |k d }ddddd d g}t tjjd|ddj | t tjjd|ddj | t tjjd|d|dj | t tjjd|d|dj | t tjjt d¡|ddj | t tjjd
|
d
|
ddj d t tjjd
|
dd
|
dj d t tjjddd
|
dj d t tjjd
|
d
|
dj d t tjjg ddj d t tjjddgddj d tttjjg d d S )NrÎ çÍÌÌÌÌÌì?r$ TrÑ FrÔ r# r' r )r$ r) rÍ r> r, )r) r r* r_ iöÿÿÿ)r r rÖ r× )r r Zisscalarr rË r- ÚemptyÚobjectÚtupleÚndimÚitemr rM r r r r )r r8 rÖ Úarrr r r r Útest_choice_return_shape¢ sL " ÿz'TestRandomDist.test_choice_return_shapec C s2 t dddg¡}d d d g}ttt jj||d d S )Né* r# r$ rÏ )r r- r r r rË )r rÖ r8 r r r Ú test_choice_nan_probabilitiesÐ s
|
z,TestRandomDist.test_choice_nan_probabilitiesc C s, t j | j¡ t j d¡}d}t|| d S )Nr s
|
Uiÿ+Wf¥)r 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 ]d}t j | j¡ |d dddddddddg
|
}t j |¡ |}|dd ddddddddg
|
}t|| qFd S )Nc S s
|
t g ¡S ©N)r r- ©rG 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í Úfloat32ré 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 rò å s ró )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 rò è s ró rÖ r× )r rì r4 r Zrecarrayré r r r rê è s ÿþc S s$ t dd | D dtfdt jfg¡S )Nc S s g | ] }||fqS r r rï r r r rò ì s ró rÖ r× )r rì rÞ r ré r r r rê ì s ÿr# r$ r) r* r+ r, é 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 ]h}t j |¡ t t
|
|j |j t
|
|j |j t j |¡ t t
|
|j |j t
|
|j |j qVd S )Né )r+ r* r) r# r é2 ) r ÚmaZ masked_valuesZreshaper r Úcopyr r÷ r ÚsortedÚdataÚmask)r rÖ r× Za_origZb_origrñ r r r Útest_shuffle_maskedõ s " ÿ ÿz"TestRandomDist.test_shuffle_maskedr c C sN dddddddd}t jtd d
|
}| |¡ W 5 Q R X d |d jksJtd S ) Nr r# r$ r) r* r+ r, )r r# r$ r) r* r+ r, z!you are shuffling a 'dict' object©ÚmatchrÉ )ÚpytestÚwarnsÚ UserWarningr÷ Úfilenamer{ )r r ÚvaluesZrecr r r Ú test_shuffle_untyped_warning s ÿz+TestRandomDist.test_shuffle_untyped_warningÚuse_array_likeTFc sÞ G dd dt j}d t dg¡t d¡t d¡t d¡g}t j|td}dd |D |rd| |¡}t fd d
|
|D s~t|r¶t|t j j
|
s¶t j t d d | |¡ W 5 Q R X n$| |¡ t fd d
|
|D sÚtd S )Nc @ s e Zd ZdS )z>TestRandomDist.test_shuffle_no_object_unpacking.<locals>.MyArrN)r0 r1 r2 r r r r ÚMyArr s r r) r rõ r' c S s h | ] }t |qS r ©Úidrï r r r Ú <setcomp> s zBTestRandomDist.test_shuffle_no_object_unpacking.<locals>.<setcomp>c 3 s | ]}t | kV qd S rè r rï ©Zitem_idsr r Ú <genexpr> s zBTestRandomDist.test_shuffle_no_object_unpacking.<locals>.<genexpr>z#Shuffling a one dimensional array.*r c 3 s | ]}t | kV qd S rè r rï r r r r & s )r Úndarrayr- Úfloat64rÞ r rF r{ Ú
|
isinstancer Ú Generatorr r r r÷ )r r r r Úitemsrâ r r r Ú test_shuffle_no_object_unpacking s&
|
ÿ
|
ÿ
|
z/TestRandomDist.test_shuffle_no_object_unpackingc C s¨ t j | j¡ t d¡j}t j |¡ tt |¡dddddg t j | j¡}| |¡ tt |¡dddddg t j | j¡}| |¡ tt |¡dddddg d S )Nr+ r r# r* r) r$ )
|
r r rT r r r÷ r rì r Ú default_rng)r rÖ Úrngr r r Útest_shuffle_memoryview( s
|
|
z&TestRandomDist.test_shuffle_memoryviewc C s< t d¡}d|j_tjtdd t j |¡ W 5 Q R X d S )Nr) Fz read-onlyr ) r r6 ÚflagsZ writeabler Zraisesr r r÷ )r rÖ r r r Útest_shuffle_not_writeable7 s
|
z)TestRandomDist.test_shuffle_not_writeablec C sN t j | j¡ t jjdddd}t ddgddgd d
|
gg¡}t||d d d S ) NrÎ rÜ r³ r_ gAàÁKÄ?g ñØhA?g õÒo¿>gάXfÈ+q?gSâë1Ã$?g'8zX ?r¤ r¥ )r r rT Úbetar- r r§ r r r Ú test_beta= s þÿzTestRandomDist.test_betac C sJ t j | j¡ t jjdddd}t ddgddgd d
|
gg¡}t|| d S ) Nr= gÉv¾/Ý?r³ r_ é% é+ rä é0 é. é- )r r rT r7 r- r r§ r r r Ú test_binomialF s
|
þz TestRandomDist.test_binomialc C sL t j | j¡ t jjddd}t ddgddgdd gg¡}t||d
|
d d S ) Nrü r³ r_ g+Qð]uðO@gëòìÇ+Q@gY7`´ZqP@gÈe(fG@gþv¯tR@gEIÈR@é r¥ )r r rT Ú chisquarer- r r§ r r r Útest_chisquareN s
|
þz TestRandomDist.test_chisquarec C st t j | j¡ t ddg¡}t jjj|dd}t ddgddggd d
|
gd d ggd dgddggg¡}t||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¥ )r r rT r- ÚmtrandÚ dirichletr )r Úalphar r¨ r r r Útest_dirichletV s
|
ÿÿÿüz TestRandomDist.test_dirichletc C sÈ t ddg¡}tt j |t d¡¡jd tt j |t d¡¡jd tt j |t d¡¡jd tt j |ddg¡jd tt j |d¡jd tt j |t d¡¡jd ttt jj|t d d S )Nr( r) r# rI r$ rJ rK )
|
r r- r r r+ rL rM r r rN rO r r r Útest_dirichlet_sizeb s z"TestRandomDist.test_dirichlet_sizec C s t ddg¡}ttt jjj| tttjddgg tttjdgdgg tttjdgdggdgdggg tttjt ddgddgg¡ d S )NgHáz®Gá?g¼Ø²Ò¼r+ r# )r r- r r r r* r+ )r r, r r r Útest_dirichlet_bad_alphan s "z'TestRandomDist.test_dirichlet_bad_alphac C sL t j | j¡ t jjddd}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 r rT Ú exponentialr- r r§ r r r Útest_exponentialy s
|
þz TestRandomDist.test_exponentialc C s* t tjjddd tttjjdd d S ©Nr ©Úscaleç )r r r r0 r r r! r r r Útest_exponential_0 s z!TestRandomDist.test_exponential_0c C sN t j | j¡ t jjdddd}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 r rT Úfr- r r§ r r r Útest_f
s
|
þzTestRandomDist.test_fc C sN t j | j¡ t jjdddd}t ddgddgd d
|
gg¡}t||d d d S ) Nr+ r) r³ r_ gÔGMç8@g §âÈ<@g ´":@gÂvÀñÇ#)@gZQø·?@gr[¦@@é r¥ )r r rT Úgammar- r r§ r r r Ú
|
test_gamma s
|
þzTestRandomDist.test_gammac C s. t tjjdddd tttjjddd d S )Nr )rM r4 r5 )r r r r< r r r! r r r Ú test_gamma_0 s zTestRandomDist.test_gamma_0c C sH t j | j¡ t jjddd}t ddgddgddgg¡}t|| d S ) Nç_c97Ý¿?r³ r_ r rõ é r+ r7 )r r rT Ú geometricr- r r§ r r r Útest_geometric s
|
þz TestRandomDist.test_geometricc C sN t j | j¡ t jjdddd}t ddgddgd d
|
gg¡}t||d d d S ) Nr? ç @r³ ©Úlocr4 r` gåûØßÉ?g¨s0í Ö?gÊ ð#0÷¿gï(ôx÷¿g-¶KÇD´ñ?géůp`@æ¿r¤ r¥ )r r rT Úgumbelr- r r§ r r r Ú test_gumbel¡ s
|
þzTestRandomDist.test_gumbelc C s* t tjjddd tttjjdd d S r2 )r r r rF r r r! r r r Ú test_gumbel_0© s z TestRandomDist.test_gumbel_0c C s t j | j¡ t jjddddd}t ddgddgddgg¡}t|| t jjdddd d}t ddddg¡}t|| t jjd
|
dd d d}t d d d d g¡}t|| t jjdddd d}t ddddg¡}t|| t jjdd
|
d d d}t ddddg¡}t|| d S ) Nr r+ r; r³ r_ rö r r) r* r¤ r7 )r r rT Úhypergeometricr- r r§ r r r Útest_hypergeometric s&
|
þ
|
|
|
|
z"TestRandomDist.test_hypergeometricc C sN t j | j¡ t jjdddd}t ddgddgd d
|
gg¡}t||d d d S ) Nr? rC r³ rD gð¥ bÙOå?g¹Â¸ëÉçà?gpJmµú @g{KÏ*Ët @g:^w-«¿gY~oõûû?r¤ r¥ )r r rT Úlaplacer- r r§ r r r Ú test_laplaceÇ s
|
þzTestRandomDist.test_laplacec C s* t tjjddd tttjjdd d S r2 )r r r rK r r r! r r r Útest_laplace_0Ï s z TestRandomDist.test_laplace_0c C sN t j | j¡ t jjdddd}t ddgddgd d
|
gg¡}t||d d d S ) Nr? rC r³ rD g#K-zñ?gîJ¬ë?g92ÄÜ @gرã÷[@gê#+^ÑÀË¿g¢¥-óâ@r¤ r¥ )r r rT Úlogisticr- r r§ r r r Ú test_logisticÓ s
|
þz TestRandomDist.test_logisticc C sN t j | j¡ t jjdddd}t ddgddgd d
|
gg¡}t||d d d S ) Nr? rC r³ )ÚmeanÚsigmar` gòÚÉ0@goÎ,+4FB@gù)Ê6@g§5 éêæ?gnþxNnP@g
©)úµU@r% r¥ )r r rT Ú lognormalr- r r§ r r r Útest_lognormalÛ s
|
þz TestRandomDist.test_lognormalc C s* t tjjddd tttjjdd d S )Nr )rQ r# r5 )r r r rR r r r! r r r Útest_lognormal_0ã s z TestRandomDist.test_lognormal_0c C sH t j | j¡ t jjddd}t ddgddgddgg¡}t|| d S )NgÆ@õí?r³ )r8 r` r$ r, r@ r) )r r rT Ú logseriesr- r r§ r r r Útest_logseriesç s
|
þz TestRandomDist.test_logseriesc C s t j | j¡ t jjddgd dd}t ddddd d gdd d
|
d d d ggddddd dgd d ddddggddd dd dgdddd ddggg¡}t|| d S ) Nrû çUUUUUUÅ?r, r³ r_ r* r) r+ r$ r r# r )r r rT rA r- r r§ r r r Útest_multinomialï s ÿÿÿüz TestRandomDist.test_multinomialc C sR t j | j¡ d}ddgddgg}d}t j |||¡}t ddgddggd d
|
gd d ggd dgddggg¡}t||dd t j ||¡}t ddg¡}t||dd ddg}ddgddgg}ttt jj|| tt jj||dd t t
|
t jj||dd t jddgddggt j d}t 0}t j ||¡ | t¡}t|dksDtW 5 Q R X d S )N)r? r r# 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$ Úignore)Z check_validÚraiserÎ r' )r r rT Zmultivariate_normalr- r r ÚRuntimeWarningr r r rî r rº r½ r{ )r rP Zcovr` r r¨ r¿ rÀ r r r Útest_multivariate_normalú sB
|
ÿÿÿü ÿÿ
|
z'TestRandomDist.test_multivariate_normalc C sJ t j | j¡ t jjdddd}t ddgddgd d
|
gg¡}t|| d S ) Nr= g|ò°Pk¿?r³ )Únr8 r` iP iI i| ic i i )r r rT rh r- r r§ r r r ri " s
|
þz%TestRandomDist.test_negative_binomialc C sÔ t j | j¡ t jjdddd}t ddgddgdd gg¡}t||d
|
d t jjd d dd}t ddgddgddgg¡}t||d
|
d t j | j¡ t jjdddd}t ddgddgddgg¡}t||d
|
d d S )Nr+ r³ )ÚdfÚnoncr` gôûÖGë7@g!© ÈÜ´*@g1}z9?@g¹+驸0@g26E_r#@g¦I!ó1@r; r¥ r5 r> g÷ô ÷?gT¶DÃ?gÀõ?T?gÍ+òqnlð?gÓ~|òùDÕ?gë°¬òÇÃ?r g{¿0¾1#@gp§²rs'@gþ~;ïÑÓ$@g"dI @gö°Õöë÷*@g Ñ}'Á,@)r r rT Únoncentral_chisquarer- r r§ r r r Útest_noncentral_chisquare* s(
|
þ
|
þ
|
þz(TestRandomDist.test_noncentral_chisquarec C sP t j | j¡ t jjddddd}t ddgdd gd
|
d gg¡}t||d d d S )Nr+ r$ r# r³ )ÚdfnumÚdfdenr_ r` g^ýíå~ö?g/+c¢äÕ?g"¡®. @gü÷´ @g×7í§þÛ?g»E·Öò?r; r¥ )r r rT Ú noncentral_fr- r r§ r r r Útest_noncentral_f? s ÿ
|
þz TestRandomDist.test_noncentral_fc C sN t j | j¡ t jjdddd}t ddgddgd d
|
gg¡}t||d d d S ) Nr? rC r³ rD gí&&n@gO»G_Ê @gÜ~w ²ø@gtÁÜûw]Õ¿gm&¨5ú½@gñ$È>Û@r¤ r¥ )r r rT Únormalr- r r§ r r r Ú test_normalH s
|
þzTestRandomDist.test_normalc C s* t tjjddd tttjjdd d S r2 )r r r rf r r r! r r r Ú test_normal_0P s z TestRandomDist.test_normal_0c C sP t j | j¡ t jjddd}t ddgddgdd gg¡}t jj||d
|
d d S ) Nr? r³ ©rÖ r` gÅWò I£@g ùÔ¨y@gy£]Ó0Ag1~ÖÒ\AgåËíãa@gÑY
|
²7Aé )Znulp)r r rT Úparetor- ÚtestingZ assert_array_almost_equal_nulpr§ r r r Ú test_paretoT s þÿ
|
zTestRandomDist.test_paretoc C sH t j | j¡ t jjddd}t ddgddgddgg¡}t|| d S )Nr? r³ )Úlamr` r r# )r r rT Úpoissonr- r r§ r r r Ú test_poissonc s
|
þzTestRandomDist.test_poissonc C s` t d¡j}d}ttt jj| ttt jj|gd ttt jj| ttt jj|gd d S )Nr r r )r rp rr r r r ro )r ZlambigZlamnegr r r Útest_poisson_exceptionsk s z&TestRandomDist.test_poisson_exceptionsc C sL t j | j¡ t jjddd}t ddgddgdd gg¡}t||d
|
d d S ) Nr? r³ ri gõ;'û?gÜ´èwÞ*?g× 5Ø?gF¡GÙ?g ÆËD ]?gýÞ$`°ÉÀ?r¤ r¥ )r r rT Úpowerr- r r§ r r r Ú
|
test_powers s
|
þzTestRandomDist.test_powerc C sL t j | j¡ t jjddd}t ddgddgdd gg¡}t||d
|
d d S ) Nr r³ )r4 r` g"íu¨ÈÆ+@gîN$MBÄ*@g7v8Bô4@gêÿª55@geã &@g~Ó£ÌZ1@r; r¥ )r r rT Úrayleighr- r r§ r r r Ú test_rayleigh{ s
|
þz TestRandomDist.test_rayleighc C s* t tjjddd tttjjdd d S r2 )r r r rt r r r! r r r Útest_rayleigh_0 s z TestRandomDist.test_rayleigh_0c C sJ t j | j¡ t jjdd}t ddgddgddgg¡}t||d d
|
d S ) Nr³ r_ gMo³DL®è?g6Ï( [9ÀgðºrF=òí?gÉM' - ÀgóV¼ëûÀgVA:ìQyÇ?r¤ r¥ )r r rT Zstandard_cauchyr- r r§ r r r Útest_standard_cauchy s
|
þz#TestRandomDist.test_standard_cauchyc C sJ t j | j¡ t jjdd}t ddgddgddgg¡}t||d d
|
d S ) Nr³ r_ g°öÜÜî?gc!z¨ì?g&N¥"@g¬NIàÇ@gFÇï>úã?g¡~MEø?r¤ r¥ )r r rT Zstandard_exponentialr- r r§ r r r Útest_standard_exponential s
|
þz(TestRandomDist.test_standard_exponentialc C sL t j | j¡ t jjddd}t ddgddgdd gg¡}t||d
|
d d S ) Nr) r³ )rM r` g)§@g¾§Ä¾¤@gd],qÂ@gPkÌ{@g9B #1 @gÔÚv÷ @r; r¥ )r r rT Ústandard_gammar- r r§ r r r Útest_standard_gamma s
|
þz"TestRandomDist.test_standard_gammac C s* t tjjddd tttjjdd d S )Nr )rM r5 )r r r ry r r r! r r r Útest_standard_gamma_0 s z$TestRandomDist.test_standard_gamma_0c C sJ t j | j¡ t jjdd}t ddgddgddgg¡}t||d d
|
d S ) Nr³ r_ rª r« r¬ r r® r¯ r¤ r¥ )r r rT ra r- r r§ r r r Útest_standard_normal£ s
|
þz#TestRandomDist.test_standard_normalc C sL t j | j¡ t jjddd}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 r rT Ú
|
standard_tr- r r§ r r r Útest_standard_t« s
|
þz TestRandomDist.test_standard_tc C sP t j | j¡ t jjddddd}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í<áþ,@r; r¥ )r r rT Ú
|
triangularr- r r§ r r r Útest_triangular³ s ÿ
|
þz TestRandomDist.test_triangularc C sN t j | j¡ t jjdddd}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 r rT Úuniformr- r r§ r r r Ú test_uniform¼ s
|
þzTestRandomDist.test_uniformc C s t d¡j}t d¡j}t jj}tt|t j d tt|dt j tt||| tt|t j gdg tt|dgt jg t jjt |d¡|d d d S )NrN r r# g Ø
W4vC)r
r )
|
r Zfinforq rr r r r Ú OverflowErrorÚinfZ nextafter)r ZfminZfmaxÚfuncr r r Útest_uniform_range_boundsÄ s z(TestRandomDist.test_uniform_range_boundsc C sn G dd dt j}t d¡ |¡}ttt jj|| G dd dt j}t d¡ |¡}ttt jj|dd d S )Nc @ s e Zd Zdd ZdS )zGTestRandomDist.test_scalar_exception_propagation.<locals>.ThrowingFloatc S s t d S rè ©r r! r r r Ú __float__Ü s zQTestRandomDist.test_scalar_exception_propagation.<locals>.ThrowingFloat.__float__N)r0 r1 r2 r r r r r Ú ThrowingFloatÛ s r g ð?c @ s e Zd Zdd ZeZdS )zITestRandomDist.test_scalar_exception_propagation.<locals>.ThrowingIntegerc S s t d S rè r r! r r r Ú__int__ä s zQTestRandomDist.test_scalar_exception_propagation.<locals>.ThrowingInteger.__int__N)r0 r1 r2 r Ú __index__r r r r ÚThrowingIntegerã s r r# ) r r r- r r r r r rI )r r Zthrowing_floatr Z throwing_intr r r Ú!test_scalar_exception_propagationÔ s ÿz0TestRandomDist.test_scalar_exception_propagationc C sN t j | j¡ t jjdddd}t ddgddgd d
|
gg¡}t||d d d S ) Nr ç¤p=
|
×£ø?r³ ©ÚmuÚkappar` gL×üZI@g¦¢'N"@gÀ·
þkrØ?gk=m@g¹Úó?gÜ^Ì=\ý?r¤ r¥ )r r rT Úvonmisesr- r r§ r r r Ú test_vonmisesì s
|
þz TestRandomDist.test_vonmisesc C s: t j | j¡ t jjdddd}t j t |¡ ¡ ¡ d S )NrC g g5_PG>i@B r )r r rT r rl r ÚisfiniterF )r Úrr r r Útest_vonmises_smallô s z"TestRandomDist.test_vonmises_smallc C sN t j | j¡ t jjdddd}t ddgddgd d
|
gg¡}t||d d d S ) Nr r r³ )rP r4 r` gÉZ¥¢@g-ï g@g¦8
|
ÄÖmÖ?gÃÎVP"ø?g`Vô áÎ?gx©^½&3Ì?r; r¥ )r r rT Úwaldr- r r§ r r r Ú test_waldú s
|
þzTestRandomDist.test_waldc C sL t j | j¡ t jjddd}t ddgddgdd gg¡}t||d
|
d d S ) Nr r³ ri g@Ý6ï?g¦Óé]Aí?g(Ã1Ú¥Rþ?gù¤U þ?gÍ2§_uå?g^}-Qö?r¤ r¥ )r r rT Úweibullr- r r§ r r r Ú test_weibull s
|
þzTestRandomDist.test_weibullc C s@ t j | j¡ tt jjdddt d¡ ttt jjdd d S )Nr r7 ri r5 )rÖ )r r rT r r r6 r r r! r r r Útest_weibull_0
|
s z TestRandomDist.test_weibull_0c C sH t j | j¡ t jjddd}t ddgddgddgg¡}t|| d S ) Nr r³ ri éB é r# r) r% )r r rT Úzipfr- r r§ r r r Ú test_zipf s
|
þzTestRandomDist.test_zipfN)Sr0 r1 r2 rX r© r° r¹ rÁ rà rÈ rÉ rÌ rÐ rÓ rÕ rÚ rÛ rã rå rç rú r r ÚmarkZ parametrizer r r r r
|
r r r r r$ r' r- r. r/ r1 r6 r: r= r> rB rG rH rJ rL rM rO rS rT rV rX r\ ri ra re rg rh rm rp rq rs ru rv rw rx rz r{ r| r~ r r r r r r 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"dAS )BÚ TestBroadcastc C s
|
d| _ d S )NiÍ[r r! r r r rX s zTestBroadcast.setup_methodc C s t j | j¡ d S rè )r r rT r! r r r ÚsetSeed s zTestBroadcast.setSeedc C sp dg}dg}t jj}t dddg¡}| ¡ ||d |}t||dd | ¡ |||d }t||dd d S ) Nr r# çºlö×÷ á?ç_x ¤¥á?ç =å*BNà?r) r; r¥ )r r r r- r¨ r )r r
r r r¨ r r r r r $ s þzTestBroadcast.test_uniformc C s dg}dg}dg}t jj}t dddg¡}| ¡ ||d |}t||dd tt||d | | ¡ |||d }t||dd tt|||d d S )
|
Nr r# r g-rö´@g¦Y®lõ@g-µ`¦wý?r) r; r¥ )r r rf r- r¨ r r r )r rE r4 Ú bad_scalerf r¨ r r r r rg 4 s þzTestBroadcast.test_normalc C sÄ dg}dg}dg}dg}t jj}t dddg¡}| ¡ ||d |}t||d d
|
tt||d | tt||d | | ¡ |||d }t||d d
|
tt|||d tt|||d d S ) Nr# r$ r r% gnÍQVfÉ?gV©KB³?g v®`køÏ?r) r; r¥ )r r r r- r¨ r r r )r rÖ r× Úbad_aZbad_br r¨ r r r r r G s& þzTestBroadcast.test_betac C sZ dg}dg}t jj}t dddg¡}| ¡ ||d }t||dd tt||d d S © Nr# r gö.Ug¬Zè?gi Prqè?gçÅE KÌæ?r) r; r¥ )r r r0 r- r¨ r r r )r r4 r¬ r0 r¨ r r r r r1 ] s þ z TestBroadcast.test_exponentialc C sZ dg}dg}t jj}t dddg¡}| ¡ ||d }t||dd tt||d d S r® )r r ry r- r¨ r r r )r rM Ú bad_shapeZ std_gammar¨ r r r r rz j s þ z!TestBroadcast.test_standard_gammac C sÄ dg}dg}dg}dg}t jj}t dddg¡}| ¡ ||d |}t||d d
|
tt||d | tt||d | | ¡ |||d }t||d d
|
tt|||d tt|||d d S ) Nr# r$ r r% gö.Ug¬Zø?gi Prqø?gçÅE KÌö?r) r; r¥ )r r r< r- r¨ r r r )r rM r4 r¯ r¬ r< r¨ r r r r r= w s& þzTestBroadcast.test_gammac C sÄ dg}dg}dg}dg}t jj}t dddg¡}| ¡ ||d |}t||d d
|
tt||d | tt||d | | ¡ |||d }t||d d
|
tt|||d tt|||d d S ) Nr# r$ r r% gbyÊé?g»åoðÄë?gT¹æ7Í@r) r; r¥ )r r r9 r- r¨ r r r )r rb rc Ú bad_dfnumÚ bad_dfdenr9 r¨ r r r r r: s& þzTestBroadcast.test_fc
|
C sf dg}dg}dg}dg}dg}dg}t jj}t ddd g¡}| ¡ ||d ||} t| |d
|
d tt||d || tt||d || tt||d || | ¡ |||d |} t| |d
|
d tt|||d | tt|||d | tt|||d | | ¡ ||||d } t| |d
|
d tt||||d tt||||d tt||||d d S ) Nr$ r) r* r r r% gÈr±^G"@g/ *@g°Ñ!@r; r¥ )r r rd r- r¨ r r r )
|
r rb rc r_ r° r± Úbad_noncZnonc_fr¨ r r r r re £ s: þz TestBroadcast.test_noncentral_fc C s< | ¡ t ddg¡}tjjddddd}t||dd d S )Ng¬ªaz@gn4M;î%é?rÜ r$ r_ r; r¥ )r¨ r r- r rd r )r r¨ r r r r Útest_noncentral_f_small_dfÄ s z(TestBroadcast.test_noncentral_f_small_dfc C sZ dg}dg}t jj}t dddg¡}| ¡ ||d }t||dd tt||d d S ) Nr# r g~¼}ÐN?â?g$Fà?g'rSèÀ?r) r; r¥ )r r r& r- r¨ r r r )r r^ Úbad_dfr& r¨ r r r r r' Ê s þ z TestBroadcast.test_chisquarec C sÄ dg}dg}dg}dg}t jj}t dddg¡}| ¡ ||d |}t||d d
|
tt||d | tt||d | | ¡ |||d }t||d d
|
tt|||d tt|||d d S ) Nr# r$ r r% g©ç wÌ "@g2 øêWR@gâRY@r) r; r¥ )r r r` r- r¨ r r r )r r^ r_ r´ r² Znonc_chir¨ r r r r ra × s& þz'TestBroadcast.test_noncentral_chisquarec C sZ dg}dg}t jj}t dddg¡}| ¡ ||d }t||dd tt||d d S ) Nr# r gT Äò@g~Fl@gÆtøpð?r) r; r¥ )r r r} r- r¨ r r r )r r^ r´ Útr¨ r r r r r~ í s þ z TestBroadcast.test_standard_tc C s dg}dg}dg}t jj}t dddg¡}| ¡ ||d |}t||dd tt||d | | ¡ |||d }t||dd tt|||d d S )
|
Nr$ r# r g´®W!è@güOBº¦ÀgÈ}Ç?àý¿r) r; r¥ )r r r r- r¨ r r r )r r r Z bad_kappar r¨ r r r r r ú s þzTestBroadcast.test_vonmisesc C sZ dg}dg}t jj}t dddg¡}| ¡ ||d }t||dd tt||d d S ) Nr# r g&8¾?ò?geâFXò?gªÕËð?r) r; r¥ )r r rk r- r¨ r r r )r rÖ r rk r¨ r r r r rm s þ zTestBroadcast.test_paretoc C sZ dg}dg}t jj}t dddg¡}| ¡ ||d }t||dd tt||d d S r® )r r r r- r¨ r r r )r rÖ r r r¨ r r r r r s þ zTestBroadcast.test_weibullc C sZ dg}dg}t jj}t dddg¡}| ¡ ||d }t||dd tt||d d S ) Nr# r r© rª r« r) r; r¥ )r r rr r- r¨ r r r )r rÖ r rr r¨ r r r r rs ' s þ zTestBroadcast.test_powerc C s dg}dg}dg}t jj}t dddg¡}| ¡ ||d |}t||dd tt||d | | ¡ |||d }t||dd tt|||d d S )
|
Nr r# r g:¨×BKc±?gÖ3®k²?gî~qÉÁ?r) r; r¥ )r r rK r- r¨ r r r )r rE r4 r¬ rK r¨ r r r r rL 4 s þzTestBroadcast.test_laplacec C s dg}dg}dg}t jj}t dddg¡}| ¡ ||d |}t||dd tt||d | | ¡ |||d }t||dd tt|||d d S )
|
Nr r# r gÈ'£ZyÑ?g÷dSO=Ñ?gÏ3Lõ1³Õ?r) r; r¥ )r r rF r- r¨ r r r )r rE r4 r¬ rF r¨ r r r r rG G s þzTestBroadcast.test_gumbelc C s dg}dg}dg}t jj}t dddg¡}| ¡ ||d |}t||dd tt||d | | ¡ |||d }t||dd tt|||d d S )
|
Nr r# r gÂy ±ÕÀ?gªÀJõRÁ?gþ
|
~Æ&£?r) r; r¥ )r r rN r- r¨ r r r )r rE r4 r¬ rN r¨ r r r r rO Z s þzTestBroadcast.test_logisticc C s dg}dg}dg}t jj}t dddg¡}| ¡ ||d |}t||dd tt||d | | ¡ |||d }t||dd tt|||d d S )
|
Nr r# r gÄñòÏH"@g¼ùÜ®Í @gvv ý²:@r) r; r¥ )r r rR r- r¨ r r r )r rP rQ Z bad_sigmarR r¨ r r r r rS m s þz TestBroadcast.test_lognormalc C sZ dg}dg}t jj}t dddg¡}| ¡ ||d }t||dd tt||d d S ) Nr# r gݱlËo½ó?gé/Ó´Æó?gz~ßRó?r) r; r¥ )r r rt r- r¨ r r r )r r4 r¬ rt r¨ r r r r ru s þ zTestBroadcast.test_rayleighc C sà dg}dg}dg}dg}t jj}t dddg¡}| ¡ ||d |}t||d d
|
tt||d | tt||d | | ¡ |||d }t||d d
|
tt|||d tt|||d tt|d d tt|dd d S ) Nr5 r# r r% g@mò e¾?g¸`Iß¿?g¸|2í?r) r; r¥ rC )r r r r- r¨ r r r )r rP r4 Zbad_meanr¬ r r¨ r r r r r s* þzTestBroadcast.test_waldc C sl dg}dg}dg}dg}dg}|d \}}t jj}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 d S )
|
Nr# r) r$ r* g,âP;bD @gºÁ±È%G @g,~S²¨ @r; r¥ )r r r r- r¨ r r r ) r r r r Z bad_left_oneZ bad_mode_oneZ bad_left_twoZ bad_mode_twor r¨ r r r r r ¥ sF þÿÿ
|
ÿz TestBroadcast.test_triangularc C sæ dg}dg}dg}dg}dg}t jj}t dddg¡}| ¡ ||d |}t|| tt||d | tt||d | tt||d | | ¡ |||d }t|| tt|||d tt|||d tt|||d d S )Nr# r5 r ç ø?r) )r r r7 r- r¨ r r r ) r r] r8 Ú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 jj}t dddg¡}| ¡ ||d |}t|| tt||d | tt||d | tt||d | | ¡ |||d }t|| tt|||d tt|||d tt|||d d S )Nr# r5 r r¶ r r) )r r rh r- r¨ r r r ) r r] r8 r· r¸ r¹ Z neg_binomr¨ r r r r ri à s&
|
|
z$TestBroadcast.test_negative_binomialc C s| t j ¡ j}dg}dg}|d g}t jj}t dddg¡}| ¡ ||d }t|| tt ||d tt ||d d S )Nr# r r$ r r) )
|
r r r Z_poisson_lam_maxro r- r¨ r r r )r Zmax_lamrn Z bad_lam_oneZ bad_lam_tworo r¨ r r r r rp ÷ s
|
|
zTestBroadcast.test_poissonc C s dg}dg}t jj}t dddg¡}| ¡ ||d }t|| tt||d t jdd( tt|t j tt|ddt j g W 5 Q R X d S )Nr$ r r# r) rY )Úinvalid)
|
r r r¤ r- r¨ r r r Zerrstater: )r rÖ r r¤ r¨ r r r r r¥ s
|
zTestBroadcast.test_zipfc C sl dg}dg}dg}t jj}t dddg¡}| ¡ ||d }t|| tt||d tt||d d S )Nr5 r r¶ r$ r) )r r rA r- r¨ r r r )r r8 r¸ r¹ Zgeomr¨ r r r r rB s
|
z TestBroadcast.test_geometricc C s dg}dg}dg}dg}dg}dg}dg}t jj}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 d S )Nr# r$ r r% r r* r) )r r rI r- r¨ r r r ) r ZngoodZnbadZnsampleZ bad_ngoodZbad_nbadZbad_nsample_oneZbad_nsample_twoZ hypergeomr¨ r r r r rJ ! s<
|
|
|
z!TestBroadcast.test_hypergeometricc C sl dg}dg}dg}t jj}t dddg¡}| ¡ ||d }t|| tt||d tt||d d S )Nr5 r$ r r# r) )r r rU r- r¨ r r r )r r8 r¸ r¹ rU r¨ r r r r rV D s
|
z TestBroadcast.test_logseriesN)#r0 r1 r2 rX r¨ r rg r r1 rz r= r: re r³ r' ra r~ r rm r rs rL rG rO rS ru r r r$ ri rp r¥ rB rJ rV r r r r r§ s@ ! $ #r§ zcan't start thread)Úreasonc @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd S ) Ú
|
TestThreadc C s t d| _d S )Nr* )r Úseedsr! r r r rX U s zTestThread.setup_methodc sÌ ddl m t t| jf| ¡}t t| jf| ¡} fddt| j|D }dd |D dd |D t| j|D ]\}}tj |¡| q|t ¡ j
|
j dkr¾t j dkr¾t|| n
|
t|| d S ) Nr )ÚThreadc s( g | ] \}} t j |¡|fd qS ))ÚtargetÚargs)r r r )rð r Úo©r¾ Úfunctionr r rò _ s ÿz-TestThread.check_function.<locals>.<listcomp>c S s g | ] }| ¡ qS r )Ústart©rð rG r r r rò a s c S s g | ] }| ¡ qS r )ÚjoinrÅ r r r rò b s r* Úwin32)Ú threadingr¾ r rÝ r½ r½ Úzipr r Zintpr( Úitemsizer Úplatformr r )r rà ÚszZout1Zout2rµ r rÁ r r r Úcheck_functionX s
|
ÿ zTestThread.check_functionc C s dd }| j |dd d S )Nc S s | j dd|d< d S )Né' r_ .)rf ©rW Úoutr r r Ú
|
gen_randomo s z*TestThread.test_normal.<locals>.gen_random)rÎ ©rÌ ©rÍ ©r rÑ r r r rg n s zTestThread.test_normalc C s dd }| j |dd d S )Nc S s | j t d¡d|d< d S )N©r= r r3 .)r0 r ZonesrÏ r r r rÑ t s z'TestThread.test_exp.<locals>.gen_randomrÕ rÒ rÓ rÔ r r r Útest_exps s zTestThread.test_expc C s dd }| j |dd d S )Nc S s | j ddgd dd|d< d S )Nr rW r, rÎ r_ .)rA rÏ r r r rÑ y s z/TestThread.test_multinomial.<locals>.gen_random)rÎ r, rÒ rÓ rÔ r r r rX x s zTestThread.test_multinomialN)r0 r1 r2 rX rÍ rg rÖ rX r r r r r¼ R s
|
r¼ c @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd S ) ÚTestSingleEltArrayInputc C s4 t dg¡| _t dg¡| _t dg¡| _d| _d S )Nr$ r) r* )r# )r r- ÚargOneÚargTwoÚargThreeÚtgtShaper! r r r rX s z$TestSingleEltArrayInput.setup_methodc C s t jjt jjt jjt jjt jjt jjt jjt jj t jj
|
t jj t jj t jj f }t jj t jj f}|D ]6}||kr~|t dg¡}n
|
|| j}t|j| j q`d S rg )r r r0 ry r& r} rk r rr rt ro r¤ rA rU r- rØ r rM rÛ )r ÚfuncsÚ probfuncsr rÐ r r r Útest_one_arg_funcs s$ û
|
z*TestSingleEltArrayInput.test_one_arg_funcsc C sè t jjt jjt jjt jjt jjt jjt jjt jj t jj
|
t jj t jj t jj t jjt jjf}t jjt jjf}|D ]v}||krt dg¡}n| j}|| j|}t|j| j || jd |}t|j| j || j|d }t|j| j qld S )Nr5 r )r r r rf r r< r9 r` r rK rF rN rR r r7 rh r- rÙ rØ r rM rÛ )r rÜ rÝ r rÙ rÐ r r r Útest_two_arg_funcs s2 ú z*TestSingleEltArrayInput.test_two_arg_funcsc C s° t tjtjtjtjtjtjtjtj g }tj
|
j }t dg¡}t dg¡}|D ]`}||||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 rL r. r r r r- r rM rÛ )r rn r r r
rs rÐ r r r r¹ ´ s ÿ z$TestSingleEltArrayInput.test_randintc C s t jjt jjt jjg}|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 rd r rI rØ rÙ rÚ r rM rÛ )r rÜ r rÐ r r r Útest_three_arg_funcsÅ s ÿz,TestSingleEltArrayInput.test_three_arg_funcsN)r0 r1 r2 rX rÞ rß r¹ rà r r r r r× s
|
r× )rÅ r Únumpyr Z numpy.testingr r r r r r r r r
|
r r r r3 r< rR rj r r§ r¦ Zskipifr¼ r× r r r r Ú<module> s2 , '"0 | >,
|
