%PDF- %PDF-
Mini Shell

Mini Shell

Direktori : /usr/lib64/python3.9/site-packages/numpy/random/tests/__pycache__/
Upload File :
Create Path :
Current File : //usr/lib64/python3.9/site-packages/numpy/random/tests/__pycache__/test_random.cpython-39.pyc

a

z[ycd�@s�ddlZddlZddlZddlmZmZmZmZm	Z	m
Z
mZmZddlm
Z
ddlZGdd�d�ZGdd�d�ZGdd	�d	�ZGd
d�d�ZGdd
�d
�ZGdd�d�ZGdd�d�ZGdd�d�ZGdd�d�ZdS)�N)�assert_�
assert_raises�assert_equal�assert_warns�assert_no_warnings�assert_array_equal�assert_array_almost_equal�suppress_warnings)�randomc@s4eZdZdd�Zdd�Zdd�Zdd�Zd	d
�ZdS)�TestSeedcCs<tj�d�}t|�d�d�tj�d�}t|�d�d�dS)Nr��i����i�)�npr
�RandomStater�randint��self�s�r�D/usr/lib64/python3.9/site-packages/numpy/random/tests/test_random.py�test_scalarszTestSeed.test_scalarcCs�tj�td��}t|�d�d�tj�t�d��}t|�d�d�tj�dg�}t|�d�d�tj�dg�}t|�d�d�dS)N�
ri�ri�r
i	)rr
r�rangerr�arangerrrr�
test_arrayszTestSeed.test_arraycCs$tttjjd�tttjjd�dS)N������r�	TypeErrorrr
r�
ValueError�rrrr�test_invalid_scalar szTestSeed.test_invalid_scalarcCsbtttjjdg�tttjjdg�tttjjdg�tttjjgd��tttjjgd��dS)Nrr�)��r")r#���r"rr rrr�test_invalid_array%s
zTestSeed.test_invalid_arraycCsRtttjjtjgtjd��tttjjgd�g�tttjjgd�gd�g�dS)N��dtype�r#r$�)���)rrrr
r�array�int64r rrr�test_invalid_array_shape-s
��z!TestSeed.test_invalid_array_shapeN)�__name__�
__module__�__qualname__rrr!r&r0rrrrrs

rc@seZdZdd�Zdd�ZdS)�TestBinomialcCsBtjddd�}dD]*}tt�d|�dk�tt�||�|�qdS)Nr$�intr')r��?r#r)r�zerosrr
�binomialr)rr7�prrr�test_n_zero7szTestBinomial.test_n_zerocCstttjdtj�dS)Nr#)rrr
r8r�nanr rrr�
test_p_is_nan@szTestBinomial.test_p_is_nanN)r1r2r3r:r<rrrrr46s	r4c@s4eZdZdd�Zdd�Zdd�Zdd�Zd	d
�ZdS)�TestMultinomialcCst�dddg�dS)N�d皙�����?皙�����?�r
�multinomialr rrr�
test_basicFszTestMultinomial.test_basiccCst�dgd��dS)Nr>)r?r@�rDrDrAr rrr�test_zero_probabilityIsz%TestMultinomial.test_zero_probabilitycCsZtdt�dd�kodkn�t�ddd�}tt�d|k��tt�|dk��dS)N���rr,)rr
rr�all)r�xrrr�test_int_negative_intervalLs$z*TestMultinomial.test_int_negative_intervalcCs�ddg}ttj�d|t�d��jd�ttj�d|t�d��jd�ttj�d|t�d��jd�ttj�d|ddg�jd�ttj�d|d�jd�ttj�d|t�d��jd�tttjjd|t	d��dS)Nr6r#�r#r$r$�r$r$r$�r$r$)
rrr
rB�uint32�shaper.rr�float�rr9rrr�	test_sizeRs��zTestMultinomial.test_sizec	Cs�tttjjdddgg�tttjjddgdgg�tttjjddgdggdgdggg�tttjjdt�ddgddgg��dS)Nrrr#)rrrr
rBr.r rrr�test_multidimensional_pvals`s&z+TestMultinomial.test_multidimensional_pvalsN)r1r2r3rCrErIrQrRrrrrr=Es
r=c@s<eZdZdd�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
S)�TestSetStatecCs$d|_t�|j�|_|j��|_dS�Ni��I)�seedr
r�prng�	get_state�stater rrr�setup_methodhszTestSetState.setup_methodcCs<|j�d�}|j�|j�|j�d�}tt�||k��dS)N�)rVZtomaxint�	set_staterXrrrG�r�old�newrrrrCmszTestSetState.test_basiccCs@|jjdd�}|j�|j�|jjdd�}tt�||k��dS�Nr*��size)rV�standard_normalr[rXrrrGr\rrr�test_gaussian_resetssz TestSetState.test_gaussian_resetcCsR|j��|j��}|jjdd�}|j�|�|jjdd�}tt�||k��dSr_)rVrbrWr[rrrG)rrXr]r^rrr� test_gaussian_reset_in_media_reszs

z-TestSetState.test_gaussian_reset_in_media_rescCsz|jdd�}|jjdd�}|j�|�|jjdd�}|j�|j�|jjdd�}tt�||k��tt�||k��dS)Nr%rZr`)rXrVrbr[rrrG)rZ	old_stateZx1Zx2Zx3rrr�test_backwards_compatibility�sz)TestSetState.test_backwards_compatibilitycCs|j�dd�dS�Nr6)rV�negative_binomialr rrr�test_negative_binomial�sz#TestSetState.test_negative_binomialN)	r1r2r3rYrCrcrdrerhrrrrrSgsrSc	@s|eZdZejjZejejej	ej
ejejej
ejejg	Zdd�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Zdd�ZdS)�TestRandintcCstt|jdtd�dS�Nr#r')rr�rfuncrOr rrr�test_unsupported_type�sz!TestRandint.test_unsupported_typecCs�|jD]�}|tjurdn
t�|�j}|tjur2dnt�|�jd}tt|j|d||d�tt|j||d|d�tt|j|||d�tt|jdd|d�qdS)Nrr$r#r')	�ityper�bool_�iinfo�min�maxrrrk)r�dt�lbnd�ubndrrr�test_bounds_checking�s
z TestRandint.test_bounds_checkingcCs�|jD]�}|tjurdn
t�|�j}|tjur2dnt�|�jd}|d}t|j||dd|d�|�|}t|j||dd|d�|�||d}t|j||dd|d�|�qdS)Nrr$r#r�rar()rmrrnrorprqrrk)rrrrsrt�tgtrrr�test_rng_zero_and_extremes�s
z&TestRandint.test_rng_zero_and_extremescCs�|jD]�}|tjurdn
t�|�j}|tjur2dnt�|�jd}z|j|||d�Wqty�}ztdt	|���WYd}~qd}~00qdS)Nrr$r#r'zMNo error should have been raised, but one was with the following message:

%s)
rmrrnrorprqrk�	Exception�AssertionError�str)rrrrsrt�errr�test_full_range�s
�zTestRandint.test_full_rangecCs�tj��|jdd�D]@}dD]6}|jd|d|d�}t|��|k�t|��dk�q q|jdddtjd�}t|��dk�t|��dk�dS)Nr#)r+�rZr$irvr)	rr
rUrmrkrrqrprn)rrrrtZvalsrrr�test_in_bounds_fuzz�s
zTestRandint.test_in_bounds_fuzzc
Cs�ddl}dddddddddd�	}|jdd�D]r}tj�d	�tjd
kr\|jddd|d
�}n|jddd|d
���}|�	|�
tj����}t
|t�|�j|k�q.tj�d	�|jdddtd
��
tj�}|�	|���}t
|t�t�j|k�dS)NrZ@509aea74d792fb931784c4b0135392c65aec64beee12b0cc167548a2c3d31e71Z@7b07f1a920e46f6d0fe02314155a2330bcfd7635e708da50e536c5ebb631a7d4Z@e577bfed6c935de944424667e3da285012e741892dcb7051a8f1ce68ab05c92fZ@0fbead0b06759df2cfb55e43148822d4a1ff953c7eb19a5b08445a63bb64fa9eZ@001aac3a5acb935a9b186cbe14a1ca064b8bb2dd0b045d48abeacf74d0203404)	�bool�int16�int32r/�int8�uint16rM�uint64�uint8r#i��littler-rrvr$)�hashlibrmrr
rU�sys�	byteorderrkZbyteswapZsha256�viewr�Z	hexdigestrr(�namer�)rr�rwrr�val�resrrr�test_repeatability�s,�

zTestRandint.test_repeatabilitycCs`tj}t�tj�j}t�t�tj�j�}t�t�tj�jd�}tjj|||d�}t||�dSrj)rr/rorqr�r
rr)rrrrwrsrt�actualrrr�test_int64_uint64_corner_case�s
z)TestRandint.test_int64_uint64_corner_casecCs�|jD]^}|tjurdn
t�|�j}|tjur2dnt�|�jd}|j|||d�}t|jt�|��qt	t
tjjfD]f}|t	ur�dn
t�|�j}|t	ur�dnt�|�jd}|j|||d�}t
t|d��tt|�|�qtdS)Nrr$r#r'r()rmrrnrorprqrkrr(r�r5�compatZlongr�hasattr�type)rrrrsrt�samplerrr�test_respect_dtype_singletons
z(TestRandint.test_respect_dtype_singletonN)r1r2r3rr
rrkrnr�r�r�r�r�rMr/r�rmrlrurxr}rr�r�r�rrrrri�s�	!ric@s�eZdZdd�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	dd�Z
dd�Zdd�Zdd�Z
dd�Zdd�Zdd�Zdd�Zdd �Zd!d"�Zd#d$�Zd%d&�Zej�d'ejej��ej��g�d(d)��Zej�d'ejej��ej��g�ej�d*d+d,g�d-d.���Zd/d0�Zd1d2�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��ZGd�d��ZHd�d��ZId�d��ZJd�d��ZKd�d��ZLd�d��ZMd�d��ZNd�d��ZOd�d��ZPd�d��ZQd�d��ZRd�S)��TestRandomDistcCs
d|_dSrT�rUr rrrrY%szTestRandomDist.setup_methodcCsJtj�|j�tj�dd�}t�ddgddgddgg�}t||d	d
�dS)Nr*r$�_�\�*��?琛�����?�<�p�?�,��o��?���H�?�Z������?���decimal)rr
rUZrandr.r�rr��desiredrrr�	test_rand(s
�zTestRandomDist.test_randcCsJtj�|j�tj�dd�}t�ddgddgddgg�}t||d	d
�dS)Nr*r$�!�<Oq�?�4�u,�?��3��?�L�I/DͿ��@��?@�HS�\@r�r�)rr
rUZrandnr.rr�rrr�
test_randn0s
�zTestRandomDist.test_randncCsJtj�|j�tjjdddd�}t�ddgddgd	d
gg�}t||�dS)N����c�r*r$r`�r*�����)�������)rr
rUrr.rr�rrr�test_randint8s
�zTestRandomDist.test_randintcCs�tj�|j�t��<}|�t�}tjjdddd�}tt|�dk�Wd�n1sV0Yt�	ddgdd	gd
dgg�}t
||�dS)Nr�r�r�r`r#r�r*r�r�r�r�)rr
rUr	�record�DeprecationWarning�random_integersr�lenr.r�r�sup�wr�r�rrr�test_random_integers@s
.
�z#TestRandomDist.test_random_integerscCsxt��H}|�t�}tj�t�d�jt�d�j�}tt	|�dk�Wd�n1sT0Yt�d�j}t
||�dS)N�lr#)r	r�r�rr
r�rorqrr�rr�rrr�test_random_integers_max_intKs

�.z+TestRandomDist.test_random_integers_max_intcCsrt���Vt�dt�tttjjt�d�j	�tttjjt�d�j	t�d�j	�Wd�n1sd0YdS)N�errorr�)
�warnings�catch_warnings�simplefilterr�rrr
r�rorqr rrr�test_random_integers_deprecatedZs

��z.TestRandomDist.test_random_integers_deprecatedcCsHtj�|j�tj�d�}t�ddgddgddgg�}t||dd	�dS)
Nr�r�r�r�r�r�r�r�r�)rr
rUr.rr�rrr�test_randomhs
�zTestRandomDist.test_randomcCs8tj�|j�tj�dd�}t�gd��}t||�dS)Nr+)r$r*r$r*�rr
rU�choicer.rr�rrr�test_choice_uniform_replacepsz*TestRandomDist.test_choice_uniform_replacecCs@tj�|j�tjjddgd�d�}t�gd��}t||�dS)Nr+)皙�����?r�皙�����?r��r9)r#r#r$r$r�r�rrr�test_choice_nonuniform_replacevsz-TestRandomDist.test_choice_nonuniform_replacecCs<tj�|j�tjjdddd�}t�gd��}t||�dS)Nr+r*F��replace)rr#r*r�r�rrr�test_choice_uniform_noreplace|sz,TestRandomDist.test_choice_uniform_noreplacecCsBtj�|j�tjjdddgd�d�}t�gd��}t||�dS)Nr+r*F)r�g333333�?r6r��r�r9)r$r*r#r�r�rrr� test_choice_nonuniform_noreplace�s�z/TestRandomDist.test_choice_nonuniform_noreplacecCs<tj�|j�tj�gd�d�}t�gd��}t||�dS)N)�a�b�c�dr+)r�r�r�r�r�r�rrr�test_choice_noninteger�sz%TestRandomDist.test_choice_nonintegercCs4tjj}tt|dd�tt|dd�tt|ddgddggd�tt|gd�tt|gd�dddgddggd	�tt|ddgdgd
�d	�tt|ddgdddgd	�tt|ddgdd
d
gd	�tt|gd�ddd�tt|gd�ddd�tt|gd�ddd�tt|gd�ddd�tt|gd�ddgd�d�dS)Nrr*g@r#r$r+)r#r$r*r+g�?r�)r�r�r?g�������?g��������r�r)Fr�r%)r)rr#)r#rrr�)rr
r�rr)rr�rrr�test_choice_exceptions�s$��z%TestRandomDist.test_choice_exceptionsc	Cs4ddg}tt�tjjddd���tt�tjjddd���tt�tjjdd|d���tt�tjjdd|d���tt�tjjddgdd���ttjjdgdd�du�t�ddg�}tjdtd	�}||d
<ttjj|dd�|u�t�}tt�tjjd|dd���tt�tjjd|dd���tt�tjjd|d|d���tt�tjjd|d|d���tt�tjjddg|dd���ttjjdg|dd�j	d
k�t�ddg�}tjdtd	�}||d
<ttjj||dd��
�|u�d}gd�}ttjjd
|dd�j|�ttjjd
|dd�j|�ttjjd
|d|d�j|�ttjjd
|d|d�j|�ttjjt�
d
�|dd�j|�ttjjd
d
dd�jd�ttjjd
dd
d�jd�ttjjddd
d�jd�ttjjd
d
d�jd�ttjjgdd�jd�ttjjddgdd�jd�tttjjgd�dS)Nr���������?r$Tr�Fr�r#r'r)r$r*)r�r�r�r�r�r?r-)r*rr+r`i����)rrr�r�)rrZisscalarr
r�r.�empty�object�tuple�ndim�itemrrNrrrr)rr9r��arrrrrr�test_choice_return_shape�sL  " �z'TestRandomDist.test_choice_return_shapecCs.t�gd��}gd�}tttjj||d�dS)N)�*r#r$)NNNr�)rr.rrr
r�)rr�r9rrr�test_choice_nan_probabilities�sz,TestRandomDist.test_choice_nan_probabilitiescCs,tj�|j�tj�d�}d}t||�dS)Nrs
�Ui���+Wf�)rr
rU�bytesrr�rrr�
test_bytes�szTestRandomDist.test_bytescCs�dd�dd�dd�dd�dd�dd�dd�d	d�d
d�dd�dd�fD]D}tj�|j�|gd
��}tj�|�|}|gd��}t||�qFdS)NcSs
t�g�S�N)rr.�rHrrr�<lambda>��z-TestRandomDist.test_shuffle.<locals>.<lambda>cSs|Sr�rr�rrrr��r�cSst�|��tj�Sr�)r�asarray�astyper�r�rrrr��r�cSst�|��tj�Sr�)rr�r��float32r�rrrr��r�cSst�|��tj�Sr�)rr�r�Z	complex64r�rrrr��r�cSst�|��t�Sr�)rr�r�r�r�rrrr��r�cSsdd�|D�S)NcSsg|]}||f�qSrr��.0�irrr�
<listcomp>�r��ATestRandomDist.test_shuffle.<locals>.<lambda>.<locals>.<listcomp>rr�rrrr��r�cSst�dd�|D��S)NcSsg|]}||g�qSrrr�rrrr��r�r�)rr�r�rrrr��r�cSst�||g�jSr�)rZvstack�Tr�rrrr��r�cSs*t�dd�|D�dtfdtfg��tj�S)NcSsg|]}||f�qSrrr�rrrr��r�r�r�r�)rr�r5r�Zrecarrayr�rrrr��s��cSs$t�dd�|D�dtfdtjfg�S)NcSsg|]}||f�qSrrr�rrrr��r�r�r�r�)rr�r�r�r�rrrr��s�)
r#r$r*r+r,r-�r~�	r)
rr#r�r-r$r+r,r~r�r*)rr
rU�shuffler)r�convZalistr�r�rrr�test_shuffle�s$�zTestRandomDist.test_shufflecCs�tj�t�td�d�ddd�}tj�t�d�ddd�}|��}|��}td�D]h}tj�|�t	t
|j|j�t
|j|j��tj�|�t	t
|j|j�t
|j|j��qVdS)N�)r,r+r*r#r�2)
rZmaZ
masked_valuesZreshaperr�copyr
r�r�sorted�data�mask)rr�r�Za_origZb_origr�rrr�test_shuffle_masked�s" � �z"TestRandomDist.test_shuffle_maskedr
cCsbdddddddd�}tjtd	d
��}|�|�Wd�n1sB0Yd|djvs^J�dS)Nrr#r$r*r+r,r-)rr#r$r*r+r,r-z!you are shuffling a 'dict' object��matchr�)�pytest�warns�UserWarningr��filename)rr
�valuesZrecrrr�test_shuffle_untyped_warnings�(z+TestRandomDist.test_shuffle_untyped_warning�use_array_likeTFcs�Gdd�dtj�}dt�dg�t�d�t�d�t�d�g}tj|td�}dd�|D��|rd|�|�}t�fd	d
�|D��s~J�|r�t|tjj	�s�t
jtdd��|�
|�Wd�q�1s�0Yn$|�
|�t�fd
d
�|D��s�J�dS)Nc@seZdZdS)z>TestRandomDist.test_shuffle_no_object_unpacking.<locals>.MyArrN)r1r2r3rrrr�MyArrsrr*rr�r'cSsh|]}t|��qSr��idr�rrr�	<setcomp>r�zBTestRandomDist.test_shuffle_no_object_unpacking.<locals>.<setcomp>c3s|]}t|��vVqdSr�rr��Zitem_idsrr�	<genexpr>r�zBTestRandomDist.test_shuffle_no_object_unpacking.<locals>.<genexpr>z#Shuffling a one dimensional array.*r�c3s|]}t|��vVqdSr�rr�r	rrr
&r�)r�ndarrayr.Zfloat64r�r�rG�
isinstancer
�	Generatorr�r�rr�)rr
rr�itemsr�rr	r� test_shuffle_no_object_unpacking
s$�
�*
z/TestRandomDist.test_shuffle_no_object_unpackingcCs�tj�|j�t�d�j}tj�|�tt�|�gd��tj�|j�}|�|�tt�|�gd��tj�	|j�}|�|�tt�|�gd��dS)Nr,)rr#r+r*r$)rr#r$r*r+)r+r#rr*r$)
rr
rUrr�r�rr�r�default_rng)rr��rngrrr�test_shuffle_memoryview(s

z&TestRandomDist.test_shuffle_memoryviewcCsPt�d�}d|j_tjtdd��tj�|�Wd�n1sB0YdS)Nr*Fz	read-onlyr�)	rr7�flagsZ	writeabler�Zraisesrr
r�)rr�rrr�test_shuffle_not_writeable7s
z)TestRandomDist.test_shuffle_not_writeablecCsNtj�|j�tjjdddd�}t�ddgddgd	d
gg�}t||dd�dS)
Nr�r�r�r`gA�Kč?g���hA?g �҇o�>gάXf�+q?gS��1�$?g'8zX� ?r�r�)rr
rU�betar.rr�rrr�	test_beta=s��zTestRandomDist.test_betacCsJtj�|j�tjjdddd�}t�ddgddgd	d
gg�}t||�dS)Nr>g�v��/�?r�r`�%�+r��0�.�-)rr
rUr8r.rr�rrr�
test_binomialFs
�zTestRandomDist.test_binomialcCsLtj�|j�tjjddd�}t�ddgddgdd	gg�}t||d
d�dS)Nr�r�r`g+Q�]u�O@g����+Q@gY7`�ZqP@g�e�(f�G@g�v�t�R@gEIȋR@�
r�)rr
rU�	chisquarer.rr�rrr�test_chisquareNs
�zTestRandomDist.test_chisquarecCsttj�|j�t�ddg�}tjjj|dd�}t�ddgddggd	d
gddggd
dgddggg�}t||dd�dS)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�)rr
rUr.�mtrand�	dirichletr)r�alphar�r�rrr�test_dirichletVs
����zTestRandomDist.test_dirichletcCs�t�ddg�}ttj�|t�d��jd�ttj�|t�d��jd�ttj�|t�d��jd�ttj�|ddg�jd�ttj�|d�jd�ttj�|t�d��jd�tttjj|t	d��dS)Nr r!r#rJr$rKrL)
rr.rr
r#rMrNrrrOrPrrr�test_dirichlet_sizebsz"TestRandomDist.test_dirichlet_sizecCs�t�ddg�}tttjjj|�tttjddgg�tttjdgdgg�tttjdgdggdgdggg�tttjt�ddgddgg��dS)NgH�z�G�?g��ؗ�Ҝ�r,r#)rr.rrr
r"r#)rr$rrr�test_dirichlet_bad_alphans"z'TestRandomDist.test_dirichlet_bad_alphacCsLtj�|j�tjjddd�}t�ddgddgdd	gg�}t||d
d�dS)Ng�8EGr��?r�r`gN��U�?gbI0��?g�VU\��@g�:��3�@g��U��?g��"qo�?r�r�)rr
rU�exponentialr.rr�rrr�test_exponentialys
�zTestRandomDist.test_exponentialcCs*ttjjdd�d�tttjjdd�dS�Nr��scale��)rrr
r(rrr rrr�test_exponential_0�sz!TestRandomDist.test_exponential_0cCsNtj�|j�tjjdddd�}t�ddgddgd	d
gg�}t||dd�dS)
N��Mr�r`g7���?g�`j���?g6�"+�?gܳ2>���?g�c�>+Y�?g�_}�S��?r�r�)rr
rU�fr.rr�rrr�test_f�s
�zTestRandomDist.test_fcCsNtj�|j�tjjdddd�}t�ddgddgd	d
gg�}t||dd�dS)
Nr,r*r�r`gԏGM�8@g���Ȍ<@g�	�":@g�v���#)@gZQ��?@gr��[��@@�r�)rr
rU�gammar.rr�rrr�
test_gamma�s
�zTestRandomDist.test_gammacCs.ttjjddd�d�tttjjddd�dS)Nr)rNr,r-)rrr
r4rrr rrr�test_gamma_0�szTestRandomDist.test_gamma_0cCsHtj�|j�tjjddd�}t�ddgddgddgg�}t||�dS)	N�_c97ݚ�?r�r`r~r��r,r/)rr
rU�	geometricr.rr�rrr�test_geometric�s
�zTestRandomDist.test_geometriccCsNtj�|j�tjjdddd�}t�ddgddgd	d
gg�}t||dd�dS)
Nr7�@r���locr,rag�ؕ��?g�s0��?g��#0��g�(�x���g-�K�D��?g�ůp`@�r�r�)rr
rU�gumbelr.rr�rrr�test_gumbel�s
�zTestRandomDist.test_gumbelcCs*ttjjdd�d�tttjjdd�dSr*)rrr
r>rrr rrr�
test_gumbel_0�szTestRandomDist.test_gumbel_0cCs�tj�|j�tjjddddd�}t�ddgddgddgg�}t||�tjjdddd	d�}t�gd
��}t||�tjjdddd	d�}t�gd
��}t||�tjjdddd	d�}t�gd��}t||�tjjdddd	d�}t�gd��}t||�dS)Nrr,r3r�r`r�rr*r+)r*r*r*r*r�r/)r/r/r/r/)rrrr)rr
rU�hypergeometricr.rr�rrr�test_hypergeometric�s&
�



z"TestRandomDist.test_hypergeometriccCsNtj�|j�tjjdddd�}t�ddgddgd	d
gg�}t||dd�dS)
Nr7r;r�r<g� b�O�?g�¸����?gpJm��	@g{K�*�t	@g�:^w-���g�Y~o���?r�r�)rr
rU�laplacer.rr�rrr�test_laplace�s
�zTestRandomDist.test_laplacecCs*ttjjdd�d�tttjjdd�dSr*)rrr
rCrrr rrr�test_laplace_0�szTestRandomDist.test_laplace_0cCsNtj�|j�tjjdddd�}t�ddgddgd	d
gg�}t||dd�dS)
Nr7r;r�r<g#��K-z�?g��J���?g�92��@gر�[@g�#+^�˿g��-��@r�r�)rr
rU�logisticr.rr�rrr�
test_logistic�s
�zTestRandomDist.test_logisticcCsNtj�|j�tjjdddd�}t�ddgddgd	d
gg�}t||dd�dS)
Nr7r;r�)�mean�sigmarag����Ɂ0@go�,+4FB@g���)ʭ6@g��5���?gn�xN�nP@g��)���U@rr�)rr
rU�	lognormalr.rr�rrr�test_lognormal�s
�zTestRandomDist.test_lognormalcCs*ttjjdd�d�tttjjdd�dS)Nr)rIr#r-)rrr
rJrrr rrr�test_lognormal_0�szTestRandomDist.test_lognormal_0cCsHtj�|j�tjjddd�}t�ddgddgddgg�}t||�dS)Ngƀ@���?r�)r9rar$r-r8r*)rr
rU�	logseriesr.rr�rrr�test_logseries�s
�zTestRandomDist.test_logseriescCshtj�|j�tjjddgddd�}t�gd�gd�ggd�gd	�ggd
�gd�gg�}t||�dS)Nr��UUUUUU�?r-r�r`)r+r*r,r+r$r$)r,r$r~r$r$r#)r*r+r*r-rr+)r$r#r+r*r-r+)r+r+r$r,r$r*)r+r*r+r$r*r+)rr
rUrBr.rr�rrr�test_multinomial�s
����zTestRandomDist.test_multinomialcCshtj�|j�d}ddgddgg}d}tj�|||�}t�ddgddggd	d
gddggd
dgddggg�}t||dd�tj�||�}t�ddg�}t||dd�ddg}ddgddgg}tttjj||�ttjj||dd�t	t
tjj||dd�tjddgddggtjd�}t��:}tj�||�|�
t�}t|�dk�sDJ�Wd�n1�sZ0YdS)N)r7rr#rr�g4�V�j�?g����y'@g�I1���?g�S�B�#@g���e<@g��4W(@g6�R¿��?g�9r
v"@g��с�?g{����#@g�4�x���g��"�f"@r�r�gN}�O6��?g��YM�W"@r$�ignore)Zcheck_valid�raiser�r')rr
rUZmultivariate_normalr.rr�RuntimeWarningrrrr�r	r�r�)rrHZcovrar�r�r�r�rrr�test_multivariate_normal�sB
������
z'TestRandomDist.test_multivariate_normalcCsJtj�|j�tjjdddd�}t�ddgddgd	d
gg�}t||�dS)Nr>g|�Pk��?r�)�nr9raiPiIi|icii�)rr
rUrgr.rr�rrrrh"s
�z%TestRandomDist.test_negative_binomialcCs�tj�|j�tjjdddd�}t�ddgddgdd	gg�}t||d
d�tjjdd
dd�}t�ddgddgddgg�}t||d
d�tj�|j�tjjdddd�}t�ddgddgddgg�}t||d
d�dS)Nr,r�)�df�noncrag��G�7@g!��ܴ*@g�1}�z9?@g�+驸�0@g26E_r#@g��I�!�1@r3r�r6r?g�����?g��T��D�?g��?T�?g�+�qnl�?g�~|�D�?g밬���?rg�{�0�1#@gp���rs'@g�~;���$@g"��dI�
@g����*@g�}�'�,@)rr
rU�noncentral_chisquarer.rr�rrr�test_noncentral_chisquare*s(
�
�
�z(TestRandomDist.test_noncentral_chisquarecCsPtj�|j�tjjddddd�}t�ddgdd	gd
dgg�}t||dd
�dS)Nr,r$r#r�)�dfnum�dfdenrWrag�^���~�?g�/+c���?g"��.�@g������@g�7���?g��E���?r3r�)rr
rU�noncentral_fr.rr�rrr�test_noncentral_f?s�
�z TestRandomDist.test_noncentral_fcCsNtj�|j�tjjdddd�}t�ddgddgd	d
gg�}t||dd�dS)
Nr7r;r�r<g�&&n@gO�G_�@g�~w��@gt��w]տgm&�5��@g��$�>�@r�r�)rr
rU�normalr.rr�rrr�test_normalHs
�zTestRandomDist.test_normalcCs*ttjjdd�d�tttjjdd�dSr*)rrr
r^rrr rrr�
test_normal_0PszTestRandomDist.test_normal_0cCsPtj�|j�tjjddd�}t�ddgddgdd	gg�}tjj||d
d�dS)Nr7r��r�rag�W�I�@g
�Ԩy�@gy��]�0�Ag1~��\�Ag����a@g�Y�
�7A�)Znulp)rr
rU�paretor.�testingZassert_array_almost_equal_nulpr�rrr�test_paretoTs��
zTestRandomDist.test_paretocCsHtj�|j�tjjddd�}t�ddgddgddgg�}t||�dS)Nr7r�)�lamrarr#)rr
rU�poissonr.rr�rrr�test_poissoncs
�zTestRandomDist.test_poissoncCs`t�d�j}d}tttjj|�tttjj|gd�tttjj|�tttjj|gd�dS)Nr�rr)rrorqrrr
rg)rZlambigZlamnegrrr�test_poisson_exceptionsksz&TestRandomDist.test_poisson_exceptionscCsLtj�|j�tjjddd�}t�ddgddgdd	gg�}t||d
d�dS)Nr7r�rag�;�'��?gܴ�w�*�?g��5��?gF�G�?g��D]?g��$`��?r�r�)rr
rU�powerr.rr�rrr�
test_powerss
�zTestRandomDist.test_powercCsLtj�|j�tjjddd�}t�ddgddgdd	gg�}t||d
d�dS)Nrr�)r,rag"�u���+@g�N$MB�*@g7v8B�4@g���55@ge��&@g~�ӣ�Z1@r3r�)rr
rU�rayleighr.rr�rrr�
test_rayleigh{s
�zTestRandomDist.test_rayleighcCs*ttjjdd�d�tttjjdd�dSr*)rrr
rlrrr rrr�test_rayleigh_0�szTestRandomDist.test_rayleigh_0cCsJtj�|j�tjjdd�}t�ddgddgddgg�}t||d	d
�dS)Nr�r`gMo�DL��?g6�([9�g�rF=��?g�M'	-��g�V���gVA:�Qy�?r�r�)rr
rUZstandard_cauchyr.rr�rrr�test_standard_cauchy�s
�z#TestRandomDist.test_standard_cauchycCsJtj�|j�tjjdd�}t�ddgddgddgg�}t||d	d
�dS)Nr�r`g���܁��?gc!z��?g&N��"�@g�NI��@gF��>���?g�~�ME�?r�r�)rr
rUZstandard_exponentialr.rr�rrr�test_standard_exponential�s
�z(TestRandomDist.test_standard_exponentialcCsLtj�|j�tjjddd�}t�ddgddgdd	gg�}t||d
d�dS)Nr*r�)rNrag)��@g��ľ��@gd]�,q�@gP�k�{@g9B#�1@g��v�� @r3r�)rr
rU�standard_gammar.rr�rrr�test_standard_gamma�s
�z"TestRandomDist.test_standard_gammacCs*ttjjdd�d�tttjjdd�dS)Nr)rNr-)rrr
rqrrr rrr�test_standard_gamma_0�sz$TestRandomDist.test_standard_gamma_0cCsJtj�|j�tjjdd�}t�ddgddgddgg�}t||d	d
�dS)Nr�r`r�r�r�r�r�r�r�r�)rr
rUrbr.rr�rrr�test_standard_normal�s
�z#TestRandomDist.test_standard_normalcCsLtj�|j�tjjddd�}t�ddgddgdd	gg�}t||d
d�dS)Nrr�)rVrag\V�H��?g�hU�%���g�H$�M�?g���L���gV�qdz�ǿg㖀���?r�r�)rr
rU�
standard_tr.rr�rrr�test_standard_t�s
�zTestRandomDist.test_standard_tcCsPtj�|j�tjjddddd�}t�ddgdd	gd
dgg�}t||dd
�dS)Ng{�G�z@g�(\��u$@gףp=
W4@r�)�left�mode�rightrag:�}��\)@ghرTj�(@g�ǨL�30@g6ڜ��A0@g6���sh&@g��<��,@r3r�)rr
rU�
triangularr.rr�rrr�test_triangular�s�
�zTestRandomDist.test_triangularcCsNtj�|j�tjjdddd�}t�ddgddgd	d
gg�}t||dd�dS)
N�G�z��?g�G�z%@r�)�low�highrag>#�H�@g��u��@g��#@g�}d]#@g���N��@g�w)*� @r�r�)rr
rU�uniformr.rr�rrr�test_uniform�s
�zTestRandomDist.test_uniformcCs�t�d�j}t�d�j}tjj}tt|tjd�tt|dtj�tt|||�tt|tjgdg�tt|dgtjg�tjjt�	|d�|dd�dS)NrOrr#g�؅W4vC)r}r~)
rZfinforprqr
rr�
OverflowError�infZ	nextafter)rZfminZfmax�funcrrr�test_uniform_range_bounds�sz(TestRandomDist.test_uniform_range_boundscCsnGdd�dtj�}t�d��|�}tttjj||�Gdd�dtj�}t�d��|�}tttjj|dd�dS)Nc@seZdZdd�ZdS)zGTestRandomDist.test_scalar_exception_propagation.<locals>.ThrowingFloatcSst�dSr��rr rrr�	__float__�szQTestRandomDist.test_scalar_exception_propagation.<locals>.ThrowingFloat.__float__N)r1r2r3r�rrrr�
ThrowingFloat�sr�g�?c@seZdZdd�ZeZdS)zITestRandomDist.test_scalar_exception_propagation.<locals>.ThrowingIntegercSst�dSr�r�r rrr�__int__�szQTestRandomDist.test_scalar_exception_propagation.<locals>.ThrowingInteger.__int__N)r1r2r3r��	__index__rrrr�ThrowingInteger�sr�r#)	rrr.r�rrr
rrA)rr�Zthrowing_floatr�Zthrowing_intrrr�!test_scalar_exception_propagation�s�z0TestRandomDist.test_scalar_exception_propagationcCsNtj�|j�tjjdddd�}t�ddgddgd	d
gg�}t||dd�dS)
Nr|�p=
ף�?r���mu�kapparagL�ZI@g��'N"@g����kr�?g�k=m�@g��ډ�?g�^�=�\�?r�r�)rr
rU�vonmisesr.rr�rrr�
test_vonmises�s
�zTestRandomDist.test_vonmisescCs:tj�|j�tjjdddd�}tj�t�|����dS)NrDgg5_P�G>i@Br�)rr
rUr�rdrZisfiniterG)r�rrrr�test_vonmises_small�sz"TestRandomDist.test_vonmises_smallcCsNtj�|j�tjjdddd�}t�ddgddgd	d
gg�}t||dd�dS)
Nr|r�r�)rHr,rag�Z���@g-�
g�@g�8
��m�?g��VP"�?g`V���?gx�^�&3�?r3r�)rr
rU�waldr.rr�rrr�	test_wald�s
�zTestRandomDist.test_waldcCsLtj�|j�tjjddd�}t�ddgddgdd	gg�}t||d
d�dS)Nr|r�rag@��6�?g���]A�?g(�1ڥR�?g��U��?g��2�_u�?g^}-�Q�?r�r�)rr
rU�weibullr.rr�rrr�test_weibulls
�zTestRandomDist.test_weibullcCs@tj�|j�ttjjddd�t�d��tttjjdd�dS)Nrr/rar-)r�)rr
rUrr�r7rrr rrr�test_weibull_0
szTestRandomDist.test_weibull_0cCsHtj�|j�tjjddd�}t�ddgddgddgg�}t||�dS)	Nr|r�ra�B�r#r*r)rr
rU�zipfr.rr�rrr�	test_zipfs
�zTestRandomDist.test_zipfN)Sr1r2r3rYr�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�ZmarkZparametrizerr
rrrrrrrrrr%r&r'r)r.r2r5r6r:r?r@rBrDrErGrKrLrNrPrTrhrYr]r_r`rerhrirkrmrnrorprrrsrtrvr{r�r�r�r�r�r�r�r�r�rrrrr�!s�.�
�	(		r�c@seZdZdd�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	dd�Z
dd�Zdd�Zdd�Z
dd�Zdd�Zdd�Zdd�Zdd �Zd!d"�Zd#d$�Zd%d&�Zd'd(�Zd)d*�Zd+d,�Zd-d.�Zd/d0�Zd1d2�Zd3d4�Zd5d6�Zd7d8�Zd9d:�Zd;d<�Z d=d>�Z!d?d@�Z"dAS)B�
TestBroadcastcCs
d|_dS)Ni�[r�r rrrrYszTestBroadcast.setup_methodcCstj�|j�dSr�)rr
rUr rrr�setSeedszTestBroadcast.setSeedcCsndg}dg}tjj}t�gd��}|��||d|�}t||dd�|��|||d�}t||dd�dS)Nrr#�g�l���?g_x
���?g=�*BN�?r*r3r�)rr
rr.r�r)rr}r~rr�r�rrrr�$szTestBroadcast.test_uniformcCs�dg}dg}dg}tjj}t�gd��}|��||d|�}t||dd�tt||d|�|��|||d�}t||dd�tt|||d�dS)Nrr#r)g-r��@g�Y�l�@g-�`�w�?r*r3r�)rr
r^r.r�rrr)rr=r,�	bad_scaler^r�r�rrrr_4szTestBroadcast.test_normalcCs�dg}dg}dg}dg}tjj}t�gd��}|��||d|�}t||dd�tt||d|�tt||d|�|��|||d�}t||dd�tt|||d�tt|||d�dS)	Nr#r$rr%)gn�QVf�?gV���KB�?gv�`k��?r*r3r�)rr
rr.r�rrr)rr�r��bad_aZbad_brr�r�rrrrGs zTestBroadcast.test_betacCsXdg}dg}tjj}t�gd��}|��||d�}t||dd�tt||d�dS�Nr#r)g�.Ug�Z�?giPr�q�?g��E
K��?r*r3r�)rr
r(r.r�rrr)rr,r�r(r�r�rrrr)]szTestBroadcast.test_exponentialcCsXdg}dg}tjj}t�gd��}|��||d�}t||dd�tt||d�dSr�)rr
rqr.r�rrr)rrN�	bad_shapeZ	std_gammar�r�rrrrrjsz!TestBroadcast.test_standard_gammacCs�dg}dg}dg}dg}tjj}t�gd��}|��||d|�}t||dd�tt||d|�tt||d|�|��|||d�}t||dd�tt|||d�tt|||d�dS)	Nr#r$rr%)g�.Ug�Z�?giPr�q�?g��E
K�?r*r3r�)rr
r4r.r�rrr)rrNr,r�r�r4r�r�rrrr5ws zTestBroadcast.test_gammacCs�dg}dg}dg}dg}tjj}t�gd��}|��||d|�}t||dd�tt||d|�tt||d|�|��|||d�}t||dd�tt|||d�tt|||d�dS)	Nr#r$rr%)gb��yʜ�?g��o���?gT��7�@r*r3r�)rr
r1r.r�rrr)rrZr[�	bad_dfnum�	bad_dfdenr1r�r�rrrr2�s zTestBroadcast.test_fc
Csddg}dg}dg}dg}dg}dg}tjj}t�gd��}|��||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�dS)
Nr$r*r+rrr%)g�r�^G"@g/�
*@g���ц�!@r3r�)rr
r\r.r�rrr)
rrZr[rWr�r��bad_noncZnonc_fr�r�rrrr]�s4zTestBroadcast.test_noncentral_fcCs<|��t�ddg�}tjjddddd�}t||dd�dS)Ng��a��z@gn4M;�%�?r�r$r`r3r�)r�rr.r
r\r)rr�r�rrr�test_noncentral_f_small_df�sz(TestBroadcast.test_noncentral_f_small_dfcCsXdg}dg}tjj}t�gd��}|��||d�}t||dd�tt||d�dS)Nr#r)g~�}�N?�?g�$F���?g'rS��?r*r3r�)rr
rr.r�rrr)rrV�bad_dfrr�r�rrrr�szTestBroadcast.test_chisquarecCs�dg}dg}dg}dg}tjj}t�gd��}|��||d|�}t||dd�tt||d|�tt||d|�|��|||d�}t||dd�tt|||d�tt|||d�dS)	Nr#r$rr%)g�� w�"@g2��WR@g���RY@r*r3r�)rr
rXr.r�rrr)rrVrWr�r�Znonc_chir�r�rrrrY�s z'TestBroadcast.test_noncentral_chisquarecCsXdg}dg}tjj}t�gd��}|��||d�}t||dd�tt||d�dS)Nr#r)gT
��@g~�F�l@gƚt��p�?r*r3r�)rr
rur.r�rrr)rrVr��tr�r�rrrrv�szTestBroadcast.test_standard_tcCs�dg}dg}dg}tjj}t�gd��}|��||d|�}t||dd�tt||d|�|��|||d�}t||dd�tt|||d�dS)Nr$r#r)g��W!�@g��OB���g�}�?�r*r3r�)rr
r�r.r�rrr)rr�r�Z	bad_kappar�r�r�rrrr��szTestBroadcast.test_vonmisescCsXdg}dg}tjj}t�gd��}|��||d�}t||dd�tt||d�dS)Nr#r)g&�8�?�?g�e��FX�?g�Ջː��?r*r3r�)rr
rcr.r�rrr)rr�r�rcr�r�rrrre
szTestBroadcast.test_paretocCsXdg}dg}tjj}t�gd��}|��||d�}t||dd�tt||d�dSr�)rr
r�r.r�rrr)rr�r�r�r�r�rrrr�szTestBroadcast.test_weibullcCsXdg}dg}tjj}t�gd��}|��||d�}t||dd�tt||d�dS)Nr#rr�r*r3r�)rr
rjr.r�rrr)rr�r�rjr�r�rrrrk'szTestBroadcast.test_powercCs�dg}dg}dg}tjj}t�gd��}|��||d|�}t||dd�tt||d|�|��|||d�}t||dd�tt|||d�dS)Nrr#r)g:��BKc�?g�3��k�?g�~q���?r*r3r�)rr
rCr.r�rrr)rr=r,r�rCr�r�rrrrD4szTestBroadcast.test_laplacecCs�dg}dg}dg}tjj}t�gd��}|��||d|�}t||dd�tt||d|�|��|||d�}t||dd�tt|||d�dS)Nrr#r)g�'��Zy�?g�dSO=�?g�3L�1��?r*r3r�)rr
r>r.r�rrr)rr=r,r�r>r�r�rrrr?GszTestBroadcast.test_gumbelcCs�dg}dg}dg}tjj}t�gd��}|��||d|�}t||dd�tt||d|�|��|||d�}t||dd�tt|||d�dS)Nrr#r)g“y��?g��J�R��?g�
~�&��?r*r3r�)rr
rFr.r�rrr)rr=r,r�rFr�r�rrrrGZszTestBroadcast.test_logisticcCs�dg}dg}dg}tjj}t�gd��}|��||d|�}t||dd�tt||d|�|��|||d�}t||dd�tt|||d�dS)Nrr#r)g����H"@g��ܮ�� @gvv	��:@r*r3r�)rr
rJr.r�rrr)rrHrIZ	bad_sigmarJr�r�rrrrKmszTestBroadcast.test_lognormalcCsXdg}dg}tjj}t�gd��}|��||d�}t||dd�tt||d�dS)Nr#r)gݱl�o��?g�/ӂ���?gz~�R�?r*r3r�)rr
rlr.r�rrr)rr,r�rlr�r�rrrrm�szTestBroadcast.test_rayleighcCs�dg}dg}dg}dg}tjj}t�gd��}|��||d|�}t||dd�tt||d|�tt||d|�|��|||d�}t||dd�tt|||d�tt|||d�tt|d	d�tt|dd	�dS)
Nr6r#rr%)g@m� �e�?g��`�I߿?g��|2��?r*r3r�rD)rr
r�r.r�rrr)rrHr,Zbad_meanr�r�r�r�rrrr��s$zTestBroadcast.test_waldcCsjdg}dg}dg}dg}dg}|d\}}tjj}t�gd��}	|��||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�dS)Nr#r*r$r+)g,�P;bD@g����%G@g,~S��@r3r�)rr
rzr.r�rrr)rrwryrxZbad_left_oneZbad_mode_oneZbad_left_twoZbad_mode_tworzr�r�rrrr{�s@��
�zTestBroadcast.test_triangularc	Cs�dg}dg}dg}dg}dg}tjj}t�gd��}|��||d|�}t||�tt||d|�tt||d|�tt||d|�|��|||d�}t||�tt|||d�tt|||d�tt|||d�dS)Nr#r6r��?�r#r#r#r*)rr
r8r.r�rrr)	rrUr9�bad_n�	bad_p_one�	bad_p_twoZbinomr�r�rrrr�s&

zTestBroadcast.test_binomialc	Cs�dg}dg}dg}dg}dg}tjj}t�gd��}|��||d|�}t||�tt||d|�tt||d|�tt||d|�|��|||d�}t||�tt|||d�tt|||d�tt|||d�dS)Nr#r6rr�)r#rr#r*)rr
rgr.r�rrr)	rrUr9r�r�r�Z	neg_binomr�r�rrrrh�s&

z$TestBroadcast.test_negative_binomialcCsztj��j}dg}dg}|dg}tjj}t�gd��}|��||d�}t||�tt	||d�tt	||d�dS)Nr#rr$)r#r#rr*)
rr
rZ_poisson_lam_maxrgr.r�rrr)rZmax_lamrfZbad_lam_oneZbad_lam_tworgr�r�rrrrh�s

zTestBroadcast.test_poissoncCs�dg}dg}tjj}t�gd��}|��||d�}t||�tt||d�tjdd��2tt|tj	�tt|ddtj	g�Wd�n1s�0YdS)Nr$r)r$r$r#r*rQ)Zinvalid)
rr
r�r.r�rrrZerrstater;)rr�r�r�r�r�rrrr�s
zTestBroadcast.test_zipfcCsjdg}dg}dg}tjj}t�gd��}|��||d�}t||�tt||d�tt||d�dS)Nr6rr�rKr*)rr
r9r.r�rrr)rr9r�r�Zgeomr�r�rrrr:s
zTestBroadcast.test_geometriccCs�dg}dg}dg}dg}dg}dg}dg}tjj}t�gd��}	|��||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�dS)	Nr#r$rr%rr+r�r*)rr
rAr.r�rrr)rZngoodZnbadZnsampleZ	bad_ngoodZbad_nbadZbad_nsample_oneZbad_nsample_twoZ	hypergeomr�r�rrrrB!s<


z!TestBroadcast.test_hypergeometriccCsjdg}dg}dg}tjj}t�gd��}|��||d�}t||�tt||d�tt||d�dS)Nr6r$rr�r*)rr
rMr.r�rrr)rr9r�r�rMr�r�rrrrNDs
zTestBroadcast.test_logseriesN)#r1r2r3rYr�r�r_rr)rrr5r2r]r�rrYrvr�rer�rkrDr?rGrKrmr�r{rrhrhr�r:rBrNrrrrr�s@

!





$
#r�c@s4eZdZdd�Zdd�Zdd�Zdd�Zd	d
�ZdS)�
TestThreadcCstd�|_dS)Nr+)r�seedsr rrrrYTszTestThread.setup_methodcs�ddlm�t�t|j�f|�}t�t|j�f|�}��fdd�t|j|�D�}dd�|D�dd�|D�t|j|�D]\}}�tj�|�|�q|t�	�j
jdkr�tj
dkr�t||�n
t||�dS)	Nr)�Threadcs(g|] \}}��tj�|�|fd��qS))�target�args)rr
r)r�r�o�r��functionrrr�^s�z-TestThread.check_function.<locals>.<listcomp>cSsg|]}|���qSr)�start�r�rHrrrr�`r�cSsg|]}|���qSr)�joinr�rrrr�ar�r+�win32)�	threadingr�rr�r�r��zipr
rZintpr(�itemsizer��platformrr)rr��szZout1Zout2r�rr�rr�r�check_functionWs
�zTestThread.check_functioncCsdd�}|j|dd�dS)NcSs|jdd�|d<dS)N�'r`.)r^�rX�outrrr�
gen_randomnsz*TestThread.test_normal.<locals>.gen_random)r��r��r��rr�rrrr_mszTestThread.test_normalcCsdd�}|j|dd�dS)NcSs|jt�d�d�|d<dS)N�r>rr+.)r(rZonesr�rrrr�ssz'TestThread.test_exp.<locals>.gen_randomr�r�r�r�rrr�test_exprszTestThread.test_expcCsdd�}|j|dd�dS)NcSs|jddgddd�|d<dS)NrrOr-r�r`.)rBr�rrrr�xsz/TestThread.test_multinomial.<locals>.gen_random)r�r-r�r�r�rrrrPwszTestThread.test_multinomialN)r1r2r3rYr�r_r�rPrrrrr�Rs
r�c@s,eZdZdd�Zdd�Zdd�Zdd�Zd	S)
�TestSingleEltArrayInputcCs4t�dg�|_t�dg�|_t�dg�|_d|_dS)Nr$r*r+)r#)rr.�argOne�argTwo�argThree�tgtShaper rrrrYsz$TestSingleEltArrayInput.setup_methodcCs�tjjtjjtjjtjjtjjtjjtjjtjj	tjj
tjjtjjtjj
f}tjjtjj
f}|D]6}||vr~|t�dg��}n
||j�}t|j|j�q`dSrf)rr
r(rqrrurcr�rjrlrgr�r9rMr.r�rrNr�)r�funcs�	probfuncsr�r�rrr�test_one_arg_funcs�s�
z*TestSingleEltArrayInput.test_one_arg_funcscCs�tjjtjjtjjtjjtjjtjjtjjtjj	tjj
tjjtjjtjj
tjjtjjf}tjjtjjf}|D]v}||vr�t�dg�}n|j}||j|�}t|j|j�||jd|�}t|j|j�||j|d�}t|j|j�qldS)Nr6r)rr
rr^rr4r1rXr�rCr>rFrJr�r8rgr.r�r�rrNr�)rr�r�r�r�r�rrr�test_two_arg_funcs�s&�z*TestSingleEltArrayInput.test_two_arg_funcscCs�tjjtjjtjjg}|D]l}||j|j|j�}t|j	|j
�||jd|j|j�}t|j	|j
�||j|jd|j�}t|j	|j
�qdS)Nr)rr
r\rzrAr�r�r�rrNr�)rr�r�r�rrr�test_three_arg_funcs�s�z,TestSingleEltArrayInput.test_three_arg_funcsN)r1r2r3rYr�r�r�rrrrr�~s-r�)r�r�ZnumpyrZ
numpy.testingrrrrrrrr	r
r�rr4r=rSrir�r�r�r�rrrr�<module>s0('"0|>,

Zerion Mini Shell 1.0