https://t.me/RX1948
Server : Apache
System : Linux iad1-shared-b8-43 6.6.49-grsec-jammy+ #10 SMP Thu Sep 12 23:23:08 UTC 2024 x86_64
User : dh_edsupp ( 6597262)
PHP Version : 8.2.26
Disable Function : NONE
Directory :  /lib/python3/dist-packages/numpy/random/tests/__pycache__/

Upload File :
current_dir [ Writeable ] document_root [ Writeable ]

 

Current File : //lib/python3/dist-packages/numpy/random/tests/__pycache__/test_randomstate.cpython-310.pyc
o

6��al>�	@s�ddlZddlZddlZddlZddlZddlZddlmZm	Z	m
Z
mZmZm
Z
mZmZddlmZmZddlmZddddde�d	�d
fddd
d�Ze�e�jdkr`ddddddddd�Zndddddddd d�Zejd!ed"�d#d$��Zd%d&�ZGd'd(�d(�ZGd)d*�d*�ZGd+d,�d,�ZGd-d.�d.�Z Gd/d0�d0�Z!Gd1d2�d2�Z"Gd3d4�d4�Z#Gd5d6�d6�Z$Gd7d8�d8�Z%d9d:�Z&d;d<�Z'd=d>�Z(dS)?�N)�assert_�
assert_raises�assert_equal�assert_warns�assert_no_warnings�assert_array_equal�assert_array_almost_equal�suppress_warnings)�MT19937�PCG64)�random)gY@g333333�?)��?)�r�
r�g@)�dr
)�$@)�)�binomial�	geometric�hypergeometric�	logseries�multinomial�negative_binomial�poisson�zipf��@2fbead005fc63942decb5326d36a1f32fe2c9d32c904ee61e46866b88447c263�@23ead5dcde35d4cfd4ef2c105e4c3d43304b45dc1b1444b7823b9ee4fa144ebb�@0d764db64f5c3bad48c8c33551c13b4d07a1e7b470f77629bef6c985cac76fcf�@7b59bf2f1691626c5815cdcd9a49e1dd68697251d4521575219e4d2a1b8b2c67�@d754fa5b92943a38ec07630de92362dd2e02c43577fc147417dc5b9db94ccdd3�@8eb216f7cb2a63cf55605422845caaff002fddc64a7dc8b2d45acd477a49e824�@70c891d76104013ebd6f6bcf30d403a9074b886ff62e4e6b8eb605bf1a4673b7�@01f074f97517cd5d21747148ac6ca4074dde7fcb7acbaec0a936606fecacd93f)rrrrrrrr�@8626dd9d052cb608e93d8868de0a7b347258b199493871a1dc56e2a26cacb112�@8edd53d272e49c4fc8fbbe6c7d08d563d62e482921f3131d0a0e068af30f0db9�@83496cc4281c77b786c9b7ad88b74d42e01603a55c60577ebab81c3ba8d45657�@65878a38747c176bc00e930ebafebb69d4e1e16cd3a704e264ea8f5e24f548db�@7a984ae6dca26fd25374479e118b22f55db0aedccd5a0f2584ceada33db98605�@d636d968e6a24ae92ab52fe11c46ac45b0897e98714426764e820a7d77602a61�@956552176f77e7c9cb20d0118fc9cf690be488d790ed4b4c4747b965e61b0bb4�@f84ba7feffda41e606e20b28dfc0f1ea9964a74574513d4a4cbc98433a8bfa45�module)�scope�paramscCs|jt|jt|jfS�N)�param�	INT_FUNCS�INT_FUNC_HASHES)�request�r5�E/usr/lib/python3/dist-packages/numpy/random/tests/test_randomstate.py�int_func2s�r7cCsnt|d|d�t|dd|dd�t|dd|dd�t|d|d�t|d|d�dS)N�
bit_generator�state�key�pos�	has_gauss�gauss)rr)�a�br5r5r6�assert_mt19937_state_equal8s
r@c@sDeZdZdd�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	dS)�TestSeedcCs8t�d�}t|�d�d�t�d�}t|�d�d�dS)Nr��i����i�)r�RandomStater�randint��self�sr5r5r6�test_scalarAs

zTestSeed.test_scalarcCszt�td��}t|�d�d�t�t�d��}t|�d�d�t�dg�}t|�d�d�t�dg�}t|�d�d�dS)NrrBi�ri�rCi	)rrD�rangerrE�np�arangerFr5r5r6�
test_arrayGszTestSeed.test_arraycCs tttjd�tttjd�dS)N������r�	TypeErrorrrD�
ValueError�rGr5r5r6�test_invalid_scalarQ�zTestSeed.test_invalid_scalarcCsXtttjdg�tttjdg�tttjdg�tttjgd��tttjgd��dS)NrNrOr)�rr)rV���rrPrSr5r5r6�test_invalid_arrayVs
zTestSeed.test_invalid_arraycCsLtttjtjgtjd��tttjgd�g�tttjgd�gd�g�dS)N��dtype�rVr�)��r)rrRrrDrK�array�int64rSr5r5r6�test_invalid_array_shape^s�
�z!TestSeed.test_invalid_array_shapecCsFt�td��}tt��|�d�Wd�dS1swYdS)Nr��)rrDrrrQ�seed)rG�rsr5r5r6�test_cannot_seedfs
"�zTestSeed.test_cannot_seedcCstttjt�dSr0)rrRrrDr
rSr5r5r6�test_invalid_initializationksz$TestSeed.test_invalid_initializationN)
�__name__�
__module__�__qualname__rIrMrTrXrarerfr5r5r5r6rA@s
rAc@seZdZdd�Zdd�ZdS)�TestBinomialcCsBtjddd�}dD]}tt�d|�dk�tt�||�|�q	dS)Nr�intrY)rr
rVr)rK�zerosrrrr)rGrl�pr5r5r6�test_n_zerops
�zTestBinomial.test_n_zerocCstttjdtj�dS)NrV)rrRrrrK�nanrSr5r5r6�
test_p_is_nanyszTestBinomial.test_p_is_nanN)rgrhrirnrpr5r5r5r6rjos	rjc@sLeZdZdd�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	dd�Z
dS)�TestMultinomialcCst�dddg�dS)Nr皙�����?皙�����?�rrrSr5r5r6�
test_basic�zTestMultinomial.test_basiccCst�dgd��dS)Nr)rrrs�rwrwrtrSr5r5r6�test_zero_probability�rvz%TestMultinomial.test_zero_probabilitycCsZtdt�dd�kodkn�t�ddd�}tt�d|k��tt�|dk��dS)N���rOr^)rrrErK�all�rG�xr5r5r6�test_int_negative_interval�s$z*TestMultinomial.test_int_negative_intervalcCs�ddg}tt�d|t�d��jd�tt�d|t�d��jd�tt�d|t�d��jd�tt�d|ddg�jd�tt�d|d�jd�tt�d|t�d��jd�tttjd|t	d��dS)Nr
rV�rVrr�rrr�rr)
rrrrK�uint32�shaper_rrQ�float�rGrmr5r5r6�	test_size�s��zTestMultinomial.test_sizecCs,tttjdddg�tttjdddg�dS)Nr皙�����?rr皙��������������?�rrRrrrSr5r5r6�test_invalid_prob�sz!TestMultinomial.test_invalid_probcCstttjdddg�dS)NrOrsrrr�rSr5r5r6�test_invalid_n��zTestMultinomial.test_invalid_ncCstt�d�}|t�|ddd��}|ddd�}t�d�tjd|d�}t�d�tjdt�|�d�}t||�dS)Ng.@rVr\i۠iUr)�pvals)rKrL�sumrrcr�ascontiguousarrayr)rGrmr��
non_contig�contigr5r5r6�test_p_non_contiguous�s


z%TestMultinomial.test_p_non_contiguouscCsdtjgd�tjd�}||��}d}tjt|d��t�d|�Wd�dS1s+wYdS)N)
�G�z��?r���&�.>r�r�r�r�r�r�r�rYz-[\w\s]*pvals array is cast to 64-bit floating)�matchrV)	rKr_�float32r��pytest�raisesrRrr)rGr|r�r�r5r5r6�test_multinomial_pvals_float32�s
�"�z.TestMultinomial.test_multinomial_pvals_float32N)rgrhrirurxr}r�r�r�r�r�r5r5r5r6rq~s
rqc@sdeZdZdd�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	dd�Z
dd�Zdd�Zdd�Z
dS)�TestSetStatecCs$d|_t�|j�|_|j��|_dS�Ni��I)rcrrD�random_state�	get_stater9rSr5r5r6�setup�szTestSetState.setupcCs<|j�d�}|j�|j�|j�d�}tt�||k��dS)N�)r��tomaxint�	set_stater9rrKrz�rG�old�newr5r5r6ru�szTestSetState.test_basiccCs@|jjdd�}|j�|j�|jjdd�}tt�||k��dS�Nr\��size)r��standard_normalr�r9rrKrzr�r5r5r6�test_gaussian_reset�sz TestSetState.test_gaussian_resetcCsR|j��|j��}|jjdd�}|j�|�|jjdd�}tt�||k��dSr�)r�r�r�r�rrKrz)rGr9r�r�r5r5r6� test_gaussian_reset_in_media_res�s

z-TestSetState.test_gaussian_reset_in_media_rescCsz|jdd�}|jjdd�}|j�|�|jjdd�}|j�|j�|jjdd�}tt�||k��tt�||k��dS)NrWr�r�)r9r�r�r�rrKrz)rG�	old_state�x1�x2�x3r5r5r6�test_backwards_compatibility�sz)TestSetState.test_backwards_compatibilitycCs|j�dd�dS�Nr
)r�rrSr5r5r6�test_negative_binomial�sz#TestSetState.test_negative_binomialcCsxt�t��}t��)}|�t�}|��}tt|�dk�t	|t
�s"J�|ddks*J�Wd�dS1s5wYdS)NrVr8r)rrDrr	�record�RuntimeWarningr�r�len�
isinstance�dict)rGrd�sup�wr9r5r5r6�test_get_state_warning�s
"�z#TestSetState.test_get_state_warningcCsl|j��}d|dd�}tt|jj|�tt|jjtj|td��|jjdd�}|d=tt|jj|�dS)N)�UnknownrVrYF��legacyr8)	r�r�rrRr�rQrKr_�object)rGr9�	new_stater5r5r6�!test_invalid_legacy_state_setting�s

�z.TestSetState.test_invalid_legacy_state_settingcCsj|j�d�|j�d�|j��|jjdd�}t|dd�t�t�|j��}|jdd�}t	||�dS)NrrFr�r<rV)
r�rc�
random_sampler�r�r�pickle�loads�dumpsr@)rG�pickled�	rs_unpick�	unpickledr5r5r6�test_pickle�s
zTestSetState.test_picklecCs<|j��}|j��|j�|�|jjdd�}t||�dS)NFr�)r��__getstate__r��__setstate__r�r@)rG�
attr_stater9r5r5r6�test_state_setting�s


zTestSetState.test_state_settingcCst|j��d�s
J�dS)NzRandomState(MT19937))�reprr��
startswithrSr5r5r6�	test_reprr�zTestSetState.test_reprN)rgrhrir�rur�r�r�r�r�r�r�r�r�r5r5r5r6r��s	

r�c	@s�eZdZejZejejej	ej
ejejej
ejejg	Zdd�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zejje�d
�jdkdd�dd��Zdd�Zdd�ZdS)�TestRandintcCstt|jdtd�dS�NrVrY)rrQ�rfuncr�rSr5r5r6�test_unsupported_typesz!TestRandint.test_unsupported_typecCs�|jD]J}|tjurdnt�|�j}|tjurdnt�|�jd}tt|j|d||d�tt|j||d|d�tt|j|||d�tt|jdd|d�qdS)NrrrVrY)	�ityperK�bool_�iinfo�min�maxrrRr�)rG�dt�lbnd�ubndr5r5r6�test_bounds_checkings
�z TestRandint.test_bounds_checkingcCs�|jD]T}|tjurdnt�|�j}|tjurdnt�|�jd}|d}t|j||dd|d�|�|}t|j||dd|d�|�||d}t|j||dd|d�|�qdS)NrrrVrB�r�rZ)r�rKr�r�r�r�rr�)rGr�r�r��tgtr5r5r6�test_rng_zero_and_extremess
�z&TestRandint.test_rng_zero_and_extremescCs�|jD]<}|tjurdnt�|�j}|tjurdnt�|�jd}z
|j|||d�Wqty?}ztdt	|���d}~wwdS)NrrrVrYzMNo error should have been raised, but one was with the following message:

%s)
r�rKr�r�r�r�r��	Exception�AssertionError�str)rGr�r�r��er5r5r6�test_full_range,s
����zTestRandint.test_full_rangecCs�t��|jdd�D] }dD]}|jd|d|d�}t|��|k�t|��dk�qq|jdddtjd�}t|��dk�t|��dk�dS)NrV)r]�r�rir�r)	rrcr�r�rr�r�rKr�)rGr�r��valsr5r5r6�test_in_bounds_fuzz:s�zTestRandint.test_in_bounds_fuzzc
Cs�dddddddddd�	}|jdd�D]8}t�d�tjd	kr)|jd
dd|d
�}n|jd
dd|d
���}t�|�	t
j����}t
|t
�|�j|k�qt�d�|jd
ddtd
��	t
j�}t�|���}t
|t
�t�j|k�dS)N�@509aea74d792fb931784c4b0135392c65aec64beee12b0cc167548a2c3d31e71�@7b07f1a920e46f6d0fe02314155a2330bcfd7635e708da50e536c5ebb631a7d4�@e577bfed6c935de944424667e3da285012e741892dcb7051a8f1ce68ab05c92f�@0fbead0b06759df2cfb55e43148822d4a1ff953c7eb19a5b08445a63bb64fa9e�@001aac3a5acb935a9b186cbe14a1ca064b8bb2dd0b045d48abeacf74d0203404)	�bool�int16�int32r`�int8�uint16r��uint64�uint8rVrb�littlerrrBr�r)r�rrc�sys�	byteorderr��byteswap�hashlib�sha256�viewrKr��	hexdigestrrZ�namer�)rGr�r��val�resr5r5r6�test_repeatabilityIs*�



zTestRandint.test_repeatability�lrzCannot test with 32-bit C long)�reasonc
Cs�t�gd�gd�gd�ggd�gd�gd�ggd�gd�gd	�ggd
�gd�gd�ggd
�gd�gd�gg�}dD]$}t�d�|jdgdgdgggd�|d�}t||durX|n|d�q9dS)N)l�Q�[ldm�"l�0)l;�ilvl�mliXp)i\L�_l�Z�NiwU�')l�<�~l�20liP�?)i�fi��wLl@
Vo)l�o�\i���wi�
1])i{�YSlr=�|i��)l�0ll�i�&�})lQZ�)l�(q
l'}�Q)lWqV1i�M
l'�t)inMl�b�"l))i��al��lS"f)i�d�i9
8l�nDk)l�Jci��-l�)i�Mi�'�*l�l�,)N)r^r\r\i90rOrrV)rCrlr�)rKr_rrcr�r)rG�desiredr�r|r5r5r6�.test_repeatability_32bit_boundary_broadcastingis8
������
��z:TestRandint.test_repeatability_32bit_boundary_broadcastingcCs^tj}t�tj�j}t�t�tj�j�}t�t�tj�jd�}tj|||d�}t||�dSr�)rKr`r�r�r�rrEr)rGr�r�r�r��actualr5r5r6�test_int64_uint64_corner_case�s
z)TestRandint.test_int64_uint64_corner_casecCs�|jD]/}|tjurdnt�|�j}|tjurdnt�|�jd}|j|||d�}t|jt�|��qt	t
tjjfD]3}|t	urBdnt�|�j}|t	urNdnt�|�jd}|j|||d�}t
t|d��tt|�|�q:dS)NrrrVrYrZ)r�rKr�r�r�r�r�rrZr�rk�compat�longr�hasattr�type)rGr�r�r��sampler5r5r6�test_respect_dtype_singleton�s
�z(TestRandint.test_respect_dtype_singletonN)rgrhrirrEr�rKr�r�r�r�r�r�r�r`r�r�r�r�r�r�r�r�r��mark�skipifr�r�rrr
r5r5r5r6r�
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"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^�Z1d_d`�Z2dadb�Z3dcdd�Z4dedf�Z5dgdh�Z6didj�Z7dkdl�Z8dmdn�Z9dodp�Z:dqdr�Z;dsdt�Z<dudv�Z=dwdx�Z>dydz�Z?d{d|�Z@d}d~�ZAdd��ZBd�d��ZCd�d��ZDd�d��ZEd�d��ZFd�d��ZGd�d��ZHd�d��ZId�d��ZJd�d��ZKd�d��ZLd�d��ZMd�d��ZNd�d��ZOd�d��ZPd�d��ZQd�d��ZRd�d��ZSd�S)��TestRandomDistcC�
d|_dSr��rcrSr5r5r6r���
zTestRandomDist.setupcCsFt�|j�t�dd�}t�ddgddgddgg�}t||d	d
�dS)Nr\r�_�\�*��?琛�����?�<�p�?�,��o��?���H�?�Z������?���decimal)rrc�randrKr_r�rGrrr5r5r6�	test_rand��
�zTestRandomDist.test_randcCs*t�|j�t��}d}t||dd�dS)Nrrr)rrcrrrr5r5r6�test_rand_singleton�sz"TestRandomDist.test_rand_singletoncCslt�|j�t�dd�}t�ddgddgddgg�}t||d	d
�t�|j�t��}t||dd	d
�dS)Nr\r�!�<Oq�?�4�u,�?��3��?�L�I/DͿ��@��?@�HS�\@rr�rr�rrc�randnrKr_rrr5r5r6�
test_randn�s
�zTestRandomDist.test_randncC�Ft�|j�tjdddd�}t�ddgddgd	d
gg�}t||�dS)N����c�r\rr��r\�����)�������)rrcrErKr_rrr5r5r6�test_randint��
�zTestRandomDist.test_randintcCs�t�|j�t��}|�t�}tjdddd�}tt|�dk�Wd�n1s)wYt�	ddgdd	gd
dgg�}t
||�t�|j�t��}|�t�}tjddd�}tt|�dk�Wd�n1siwYt
||d
�dS)Nr*r+r,r�rVr-r\r.r/r0r1��r)rrcr	r��DeprecationWarning�random_integersrr�rKr_r)rGr�r�rrr5r5r6�test_random_integers�s$
�
�

�z#TestRandomDist.test_random_integerscCs�t�|j�t�|j�}|jdd�}t�t�jdkr,tjddgddgdd	ggtj	d
�}ntjddgd
dgddggtj	d
�}t
||�|�|j�|��}t
||d�dS)Nr,r�i���i��4Oi�˕+i�R�KiriFb�qi���rYl�K+W+�ylr&&�J�]l��;	l��nBz
l�@X�H�TlD+l++�r%)rrcrDr�rKr�rkr�r_r`r)rGrdrrr5r5r6�
test_tomaxint�s(
��
��
zTestRandomDist.test_tomaxintcCs�t��#}|�t�}t�t�d�jt�d�j�}tt	|�dk�Wd�n1s)wYt�d�j}t
||�t��-}|�t�}t�d�j}t�|t�d�j�|t�d�j��}tt	|�dk�Wd�n1slwYt
||�dS)Nr�rV)
r	r�r5rr6rKr�r�rr�rrZr)rGr�r�rr�typerr5r5r6�test_random_integers_max_int�s$

��

��z+TestRandomDist.test_random_integers_max_intcCspt���*t�dt�tttjt�d�j	�tttjt�d�j	t�d�j	�Wd�dS1s1wYdS)N�errorr�)
�warnings�catch_warnings�simplefilterr5rrr6rKr�r�rSr5r5r6�test_random_integers_deprecateds

��"�z.TestRandomDist.test_random_integers_deprecatedcCsjt�|j�t�d�}t�ddgddgddgg�}t||dd	�t�|j�t��}t||d
dd	�dS)Nr,rrrrrrrrr%)rrcr�rKr_rrr5r5r6�test_random_samples

�z!TestRandomDist.test_random_samplecCs4t�|j�t�dd�}t�gd��}t||�dS)Nr])rr\rr\�rrc�choicerKr_rrr5r5r6�test_choice_uniform_replace*sz*TestRandomDist.test_choice_uniform_replacecCs<t�|j�tjddgd�d�}t�gd��}t||�dS)Nr])皙�����?rD皙�����?rE�rm)rVrVrrrArr5r5r6�test_choice_nonuniform_replace0sz-TestRandomDist.test_choice_nonuniform_replacecCs8t�|j�tjdddd�}t�gd��}t||�dS)Nr]r\F��replace)rrVr\rArr5r5r6�test_choice_uniform_noreplace6�z,TestRandomDist.test_choice_uniform_noreplacecCs>t�|j�tjdddgd�d�}t�gd��}t||�dS)Nr]r\F)rE�333333�?r
rE�rIrm)rr\rVrArr5r5r6� test_choice_nonuniform_noreplace<sz/TestRandomDist.test_choice_nonuniform_noreplacecCs8t�|j�t�gd�d�}t�gd��}t||�dS)N)r>r?�c�dr])rOrPrOrPrArr5r5r6�test_choice_nonintegerBrKz%TestRandomDist.test_choice_nonintegercCs2tj}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)NrOr\�@rVrr])rVrr\r]g�?rF)rDrDrrr�r�rDr[FrHrW)rO)rOrV)rVrrrM)rrBrrR)rGr	r5r5r6�test_choice_exceptionsHs$�
�z%TestRandomDist.test_choice_exceptionsc	Csddg}tt�tjddd���tt�tjddd���tt�tjdd|d���tt�tjdd|d���tt�tjddgdd���ttjdgdd�du�t�ddg�}tjdtd	�}||d
<ttj|dd�|u�t�}tt�tjd|dd���tt�tjd|dd���tt�tjd|d|d���tt�tjd|d|d���tt�tjddg|dd���ttjdg|dd�j	d
k�t�ddg�}tjdtd	�}||d
<ttj||dd��
�|u�d}gd�}ttjd
|dd�j|�ttjd
|dd�j|�ttjd
|d|d�j|�ttjd
|d|d�j|�ttjt�
d
�|dd�j|�ttjd
d
dd�jd�ttjd
dd
d�jd�ttjddd
d�jd�ttjd
d
d�jd�ttjgdd�jd�ttjddgdd�jd�tttjgd�dS)NrEr�rTrHFrMrVrYr)rr\)rErErErErDrrr)r\rr]r�i����)rrr>r?)rrK�isscalarrrBr_�emptyr��tuple�ndim�itemrr�rLrErrR)rGrmr>�arrrHr5r5r6�test_choice_return_shape[sL �z'TestRandomDist.test_choice_return_shapecCs,t�gd��}gd�}tttj||d�dS)N)�*rVr)NNNrF)rKr_rrRrrB)rGr>rmr5r5r6�test_choice_nan_probabilities�sz,TestRandomDist.test_choice_nan_probabilitiesc	Cs|t�d�d}d|ddd�<t�|j�tjdd|ddd�d�}t�|j�tjddt�|ddd��d�}t||�dS)Nrr^rRrVrr\rF)rK�onesrrcrBr�r)rGrmr�r�r5r5r6�test_choice_p_non_contiguous�s z+TestRandomDist.test_choice_p_non_contiguouscCs(t�|j�t�d�}d}t||�dS)Nrs
�Ui���+Wf�)rrc�bytesrrr5r5r6�
test_bytes�s
zTestRandomDist.test_bytescCs�dd�dd�dd�dd�dd�dd�dd�d	d�d
d�dd�dd�fD] }t�|j�|gd
��}t�|�|}|gd��}t||�q#dS)NcSs
t�g�Sr0)rKr_�r|r5r5r6�<lambda>�s
z-TestRandomDist.test_shuffle.<locals>.<lambda>cSs|Sr0r5rar5r5r6rb�scS�t�|��tj�Sr0)rK�asarray�astyper�rar5r5r6rb��cSrcr0)rKrdrer�rar5r5r6rb�rfcSrcr0)rKrdre�	complex64rar5r5r6rb�rfcSst�|��t�Sr0)rKrdrer�rar5r5r6rb��cSsdd�|D�S)NcS�g|]}||f�qSr5r5��.0�ir5r5r6�
<listcomp>���ATestRandomDist.test_shuffle.<locals>.<lambda>.<locals>.<listcomp>r5rar5r5r6rb�scSst�dd�|D��S)NcSsg|]}||g�qSr5r5rjr5r5r6rm�rnro)rKrdrar5r5r6rb�rncSst�||g�jSr0)rK�vstack�Trar5r5r6rb�rhcSs*t�dd�|D�dtfdtfg��tj�S)NcSrir5r5rjr5r5r6rm�rnror>r?)rKrdrkr��recarrayrar5r5r6rb�s�
cSs(t�dd�|D�dtdfdtjdfg�S)NcSrir5r5rjr5r5r6rm�rnror>�rVr?)rKrdr�r�rar5r5r6rb�s

���
rVrr\r]r^r�r��	r�
rrVrvrrr]r^r�rur\)rrc�shuffler)rG�conv�alistrrr5r5r6�test_shuffle�s&�
�zTestRandomDist.test_shufflecCs�tj�t�td�d�ddd�}tj�t�d�ddd�}|��}|��}td�D]2}t�|�t	t
|j|j�t
|j|j��t�|�t	t
|j|j�t
|j|j��q+dd�}dS)	Nr)r^r]r\rVrO�2cSst�d�}tttj|�dS)Nr\)rKr_rrQrrxr{r5r5r6�test_shuffle_invalid_objects�s
zHTestRandomDist.test_shuffle_masked.<locals>.test_shuffle_invalid_objects)
rK�ma�
masked_values�reshaperJrL�copyrrxr�sorted�data�mask)rGr>r?�a_orig�b_origrlr}r5r5r6�test_shuffle_masked�s"
 �
 �z"TestRandomDist.test_shuffle_maskedcCs�t�|j�gd�}t�|�}gd�}t||�t�|j�t�gd��j}t�|�}t|t�|�j�t�|j�d}tttj|�t�|j�d}tttj|�d}gd�}t�|j�t�|�}t||�dS)Nrtrw�abcdg333333�?r)
rvrr�r^rVr\r]rurr)	rrc�permutationrrK�
atleast_2drqr�
IndexError)rGrzrr�arr_2d�	bad_x_str�bad_x_float�integer_valr5r5r6�test_permutation�s(



zTestRandomDist.test_permutationcC�Jt�|j�tjdddd�}t�ddgddgd	d
gg�}t||dd�dS)
NrEr�r,r�gA�Kč?g���hA?g �҇o�>gάXf�+q?gS��1�$?g'8zX� ?rr)rrc�betarKr_rrr5r5r6�	test_beta�s��zTestRandomDist.test_betacCslt�|j�tjdddd�}t�ddgddgd	d
gg�}t||�t�|j�t�dd�}d}t||�dS)NgZd;�Y@g�v��/�?r,r��%�+r[�0�.�-)rrcrrKr_rrr5r5r6�
test_binomial�s
�
zTestRandomDist.test_binomialcC�Ht�|j�tjddd�}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)rrc�	chisquarerKr_rrr5r5r6�test_chisquare��
�zTestRandomDist.test_chisquarecCs�t�|j�t�ddg�}tj|dd�}t�ddgddggd	d
gddggd
dgddggg�}t||dd�t�ddg�}tttj|�t�|j�t�ddg�}t�|�}t||ddd�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��?rr�H�z�G�?缉ؗ�Ҝ�r%)rrcrKr_�	dirichletrrrR)rG�alpharr�	bad_alphar5r5r6�test_dirichlets(
����
zTestRandomDist.test_dirichletcCs�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��dS)Nr�r�rVr~rrr�)
rKr_rrr�r�r�rrQr�r�r5r5r6�test_dirichlet_sizesz"TestRandomDist.test_dirichlet_sizecCs t�ddg�}tttj|�dS)Nr�r�)rKr_rrRrr�)rGr�r5r5r6�test_dirichlet_bad_alpha"rUz'TestRandomDist.test_dirichlet_bad_alphacCsdt�gd��}|ddd�}t�|j�tj|dd�}t�|j�tjt�|�dd�}t||�dS)N)r�g�r�rr,r�)rKr_rrcr�r�r)rGr>r�r�r�r5r5r6�#test_dirichlet_alpha_non_contiguous's�z2TestRandomDist.test_dirichlet_alpha_non_contiguouscCr�)Ng�8EGr��?r,r�gN��U�?gbI0��?g�VU\��@g�:��3�@g��U��?g��"qo�?rr)rrc�exponentialrKr_rrr5r5r6�test_exponential1r�zTestRandomDist.test_exponentialcC�&ttjdd�d�tttjdd�dS�Nr��scale��)rrr�rrRrSr5r5r6�test_exponential_09�z!TestRandomDist.test_exponential_0cCr�)
N��Mr,r�g7���?g�`j���?g6�"+�?gܳ2>���?g�c�>+Y�?g�_}�S��?rr)rrc�frKr_rrr5r5r6�test_f=�
�zTestRandomDist.test_fcCr�)
Nr^r\r,r�gԏGM�8@g���Ȍ<@g�	�":@g�v���#)@gZQ��?@gr��[��@@�r)rrc�gammarKr_rrr5r5r6�
test_gammaEr�zTestRandomDist.test_gammacCs*ttjddd�d�tttjddd�dS)Nr)r�r�r�)rrr�rrRrSr5r5r6�test_gamma_0MszTestRandomDist.test_gamma_0cC�Dt�|j�tjddd�}t�ddgddgddgg�}t||�dS)	N�_c97ݚ�?r,r�r�ru�r^r�)rrcrrKr_rrr5r5r6�test_geometricQ�
�zTestRandomDist.test_geometriccCs�tttjd�tttjdgd�tttjd�tttjdgd�t��!}|�t�tttjtj�tttjtjgd�Wd�dS1sIwYdS)Nr�rr�)	rrRrrr	r�r�rKro�rGr�r5r5r6�test_geometric_exceptionsYs
"�z(TestRandomDist.test_geometric_exceptionscCr�)
Nr��@r,��locr�r�g�ؕ��?g�s0��?g��#0��g�(�x���g-�K�D��?g�ůp`@�rr)rrc�gumbelrKr_rrr5r5r6�test_gumbelcr�zTestRandomDist.test_gumbelcCr�r�)rrr�rrRrSr5r5r6�
test_gumbel_0kr�zTestRandomDist.test_gumbel_0cCs�t�|j�tjddddd�}t�ddgddgddgg�}t||�tjdd	d
dd�}t�gd��}t||�tjd
d	ddd�}t�gd��}t||�tjd	dd
dd�}t�gd��}t||�tjd	d
ddd�}t�gd��}t||�dS)Ng333333$@g@r�r,r�rrvr^rr\r])r\r\r\r\rr�)r�r�r�r�)rrrr)rrcrrKr_rrr5r5r6�test_hypergeometricos&
�



z"TestRandomDist.test_hypergeometriccCr�)
Nr�r�r,r�g� b�O�?g�¸����?gpJm��	@g{K�*�t	@g�:^w-���g�Y~o���?rr)rrc�laplacerKr_rrr5r5r6�test_laplace�r�zTestRandomDist.test_laplacecCr�r�)rrr�rrRrSr5r5r6�test_laplace_0�r�zTestRandomDist.test_laplace_0cCr�)
Nr�r�r,r�g#��K-z�?g��J���?g�92��@gر�[@g�#+^�˿g��-��@rr)rrc�logisticrKr_rrr5r5r6�
test_logistic�r�zTestRandomDist.test_logisticcCr�)
Nr�r�r,)�mean�sigmar�g����Ɂ0@go�,+4FB@g���)ʭ6@g��5���?gn�xN�nP@g��)���U@r�r)rrc�	lognormalrKr_rrr5r5r6�test_lognormal�r�zTestRandomDist.test_lognormalcCs&ttjdd�d�tttjdd�dS)Nr)r�rVr�)rrr�rrRrSr5r5r6�test_lognormal_0�r�zTestRandomDist.test_lognormal_0cCsDt�|j�tjddd�}t�ddgddgddgg�}t||�dS)Ngƀ@���?r,)rmr�rrr�r\)rrcrrKr_rrr5r5r6�test_logseries�r�zTestRandomDist.test_logseriescCs\t��!}|�t�tttjtj�tttjtjgd�Wd�dS1s'wYdS)Nr)	r	r�r�rrRrrrKror�r5r5r6�test_logseries_exceptions�s

"�z(TestRandomDist.test_logseries_exceptionscCsdt�|j�tjddgddd�}t�gd�gd�ggd�gd	�ggd
�gd�gg�}t||�dS)Nr�UUUUUU�?rr,r�)r]r\r^r]rr)r^rr�rrrV)r\r]r\rrr])rrVr]r\rr])r]r]rr^rr\)r]r\r]rr\r])rrcrrKr_rrr5r5r6�test_multinomial�s
����zTestRandomDist.test_multinomialc	Cs�t�|j�d}ddgddgg}d}t�|||�}t�ddgddggd	d
gddggd
dgddggg�}t||dd�t�||�}t�ddg�}t||dd�ddg}ddgddgg}tttj||�ttj||dd�t	t
tj||dd�tjddgddggtjd�}t��}t�||�|�
t�}t|�dks�J�Wd�n1s�wYt�d�}t�d�}t	t
tj||dd�t	t
tjt�d�|�t	t
tj|t�d��t	t
tj|t�d��dS)N)r�rrVrr,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"@rrgN}�O6��?g��YM�W"@r�ignore)�check_valid�raiserErY�other)rrVrVr\)rrc�multivariate_normalrKr_rrr�rrrRr�r	r�r�rl�eyerU)	rGr��covr�rrr�r��mur5r5r6�test_multivariate_normal�s`
����
��
�

�
�
�
�z'TestRandomDist.test_multivariate_normalcCr))Nrg|�Pk��?r,)�nrmr�iPiIi|icii�)rrcrrKr_rrr5r5r6r��r3z%TestRandomDist.test_negative_binomialcCs`t��#}|�t�tttjdtj�tttjdtjgd�Wd�dS1s)wYdS)Nrr)	r	r�r�rrRrrrKror�r5r5r6�!test_negative_binomial_exceptions�s


�"�z0TestRandomDist.test_negative_binomial_exceptionscCs�t�|j�tjdddd�}t�ddgddgdd	gg�}t||d
d�tjdd
dd�}t�ddgddgddgg�}t||d
d�t�|j�tjdddd�}t�ddgddgddgg�}t||d
d�dS)Nr^r,)�df�noncr�g��G�7@g!��ܴ*@g�1}�z9?@g�+驸�0@g26E_r#@g��I�!�1@r�rr
rrg�����?g��T��D�?g��?T�?g�+�qnl�?g�~|�D�?g밬���?rg�{�0�1#@gp���rs'@g�~;���$@g"��dI�
@g����*@g�}�'�,@)rrc�noncentral_chisquarerKr_rrr5r5r6�test_noncentral_chisquares(
�
�
�z(TestRandomDist.test_noncentral_chisquarecC�Lt�|j�tjddddd�}t�ddgdd	gd
dgg�}t||dd
�dS)Nr^rrVr,)�dfnum�dfdenr�r�g�^���~�?g�/+c���?g"��.�@g������@g�7���?g��E���?r�r)rrc�noncentral_frKr_rrr5r5r6�test_noncentral_f�
�
�z TestRandomDist.test_noncentral_fcCs0t�|j�tjddtjd�}t�|�sJ�dS)Nr^r)r�r�r�)rrcr�rKro�isnan)rGrr5r5r6�test_noncentral_f_nan"sz$TestRandomDist.test_noncentral_f_nancCr�)
Nr�r�r,r�g�&&n@gO�G_�@g�~w��@gt��w]տgm&�5��@g��$�>�@rr)rrc�normalrKr_rrr5r5r6�test_normal'r�zTestRandomDist.test_normalcCr�r�)rrr�rrRrSr5r5r6�
test_normal_0/r�zTestRandomDist.test_normal_0cCsLt�|j�tjddd�}t�ddgddgdd	gg�}tjj||d
d�dS)Nr�r,�r>r�g�W�I�@g
�Ԩy�@gy��]�0�Ag1~��\�Ag����a@g�Y�
�7A�)�nulp)rrc�paretorKr_�testing�assert_array_almost_equal_nulprr5r5r6�test_pareto3s��
zTestRandomDist.test_paretocCsDt�|j�tjddd�}t�ddgddgddgg�}t||�dS)Nr�r,)�lamr�rrV)rrcrrKr_rrr5r5r6�test_poissonBr�zTestRandomDist.test_poissoncCs�t�d�j}d}tttj|�tttj|gd�tttj|�tttj|gd�t��!}|�t	�tttjtj
�tttjtj
gd�Wd�dS1sQwYdS)Nr�rOr)rKr�r�rrRrrr	r�r�ro)rG�lambig�lamnegr�r5r5r6�test_poisson_exceptionsJs
"�z&TestRandomDist.test_poisson_exceptionscCr�)Nr�r,r�g�;�'��?gܴ�w�*�?g��5��?gF�G�?g��D]?g��$`��?rr)rrc�powerrKr_rrr5r5r6�
test_powerVr�zTestRandomDist.test_powercCr�)Nrr,)r�r�g"�u���+@g�N$MB�*@g7v8B�4@g���55@ge��&@g~�ӣ�Z1@r�r)rrc�rayleighrKr_rrr5r5r6�
test_rayleigh^r�zTestRandomDist.test_rayleighcCr�r�)rrrrrRrSr5r5r6�test_rayleigh_0fr�zTestRandomDist.test_rayleigh_0cC�Ft�|j�tjdd�}t�ddgddgddgg�}t||d	d
�dS)Nr,r�gMo�DL��?g6�([9�g�rF=��?g�M'	-��g�V���gVA:�Qy�?rr)rrc�standard_cauchyrKr_rrr5r5r6�test_standard_cauchyjrz#TestRandomDist.test_standard_cauchycCr)Nr,r�g���܁��?gc!z��?g&N��"�@g�NI��@gF��>���?g�~�ME�?rr)rrc�standard_exponentialrKr_rrr5r5r6�test_standard_exponentialrrz(TestRandomDist.test_standard_exponentialcCr�)Nr\r,)r�r�g)��@g��ľ��@gd]�,q�@gP�k�{@g9B#�1@g��v�� @r�r)rrc�standard_gammarKr_rrr5r5r6�test_standard_gammazr�z"TestRandomDist.test_standard_gammacCr�)Nr)r�r�)rrrrrRrSr5r5r6�test_standard_gamma_0�r�z$TestRandomDist.test_standard_gamma_0cCr)Nr,r�rr r!r"r#r$rr)rrcr�rKr_rrr5r5r6�test_standard_normal�rz#TestRandomDist.test_standard_normalcCs0t�|j�t��}t�d�}t||dd�dS)Nrrrr&rr5r5r6�test_randn_singleton�s
z#TestRandomDist.test_randn_singletoncCr�)Nrr,)r�r�g\V�H��?g�hU�%���g�H$�M�?g���L���gV�qdz�ǿg㖀���?rr)rrc�
standard_trKr_rrr5r5r6�test_standard_t�r�zTestRandomDist.test_standard_tcCr�)Ng{�G�z@g�(\��u$@gףp=
W4@r,)�left�mode�rightr�g:�}��\)@ghرTj�(@g�ǨL�30@g6ڜ��A0@g6���sh&@g��<��,@r�r)rrc�
triangularrKr_rrr5r5r6�test_triangular�r�zTestRandomDist.test_triangularcCr�)
N�G�z��?g�G�z%@r,)�low�highr�g>#�H�@g��u��@g��#@g�}d]#@g���N��@g�w)*� @rr)rrc�uniformrKr_rrr5r5r6�test_uniform�r�zTestRandomDist.test_uniformcCs�t�d�j}t�d�j}tj}tt|tjd�tt|dtj�tt|||�tt|tjgdg�tt|dgtjg�tjt�	|d�|dd�dS)Nr�rrVg�؅W4vC)rr)
rK�finfor�r�rrr�
OverflowError�inf�	nextafter)rG�fmin�fmax�funcr5r5r6�test_uniform_range_bounds�sz(TestRandomDist.test_uniform_range_boundscCsjGdd�dtj�}t�d��|�}tttj||�Gdd�dtj�}t�d��|�}tttj|dd�dS)Nc@�eZdZdd�ZdS)zGTestRandomDist.test_scalar_exception_propagation.<locals>.ThrowingFloatcS�t�r0�rQrSr5r5r6�	__float__��zQTestRandomDist.test_scalar_exception_propagation.<locals>.ThrowingFloat.__float__N)rgrhrir$r5r5r5r6�
ThrowingFloat��r&��?c@r!)zITestRandomDist.test_scalar_exception_propagation.<locals>.ThrowingIntegercSr"r0r#rSr5r5r6�__int__�r%zQTestRandomDist.test_scalar_exception_propagation.<locals>.ThrowingInteger.__int__N)rgrhrir)r5r5r5r6�ThrowingInteger�r'r*rV)	rK�ndarrayr_r�rrQrrr)rGr&�throwing_floatr*�throwing_intr5r5r6�!test_scalar_exception_propagation�s
�z0TestRandomDist.test_scalar_exception_propagationcCr�)
Nr�p=
ף�?r,�r��kappar�gL�ZI@g��'N"@g����kr�?g�k=m�@g��ډ�?g�^�=�\�?rr�rrc�vonmisesrKr_rrr5r5r6�
test_vonmises�r�zTestRandomDist.test_vonmisescCs2t�|j�tjdddd�}tt�|����dS)Nrwgg5_P�G>�@Br0)rrcr3rrK�isfiniterz�rG�rr5r5r6�test_vonmises_small�sz"TestRandomDist.test_vonmises_smallcCs<t�|j�tjdddd�}t�gd��}t||dd�dS)Nrwg�cAr\r0)g�A�^>?g@��R7?gx�.�r�rr2rr5r5r6�test_vonmises_large�sz"TestRandomDist.test_vonmises_largecCs.t�|j�tjdtjd�}tt�|��dS)Nrw)r�r1)rrcr3rKrorr�r7r5r5r6�test_vonmises_nan�sz TestRandomDist.test_vonmises_nancCr�)
Nrr/r,)r�r�r�g�Z���@g-�
g�@g�8
��m�?g��VP"�?g`V���?gx�^�&3�?r�r)rrc�waldrKr_rrr5r5r6�	test_wald�r�zTestRandomDist.test_waldcCr�)Nrr,r�g@��6�?g���]A�?g(�1ڥR�?g��U��?g��2�_u�?g^}-�Q�?rr)rrc�weibullrKr_rrr5r5r6�test_weibull�r�zTestRandomDist.test_weibullcCs:t�|j�ttjddd�t�d��tttjdd�dS)Nrr�r�r�)r>)rrcrr>rKrlrrRrSr5r5r6�test_weibull_0�szTestRandomDist.test_weibull_0cCr�)	Nrr,r��B�rVr\r�)rrcrrKr_rrr5r5r6�	test_zipfr�zTestRandomDist.test_zipfN)Trgrhrir�rrr(r2r7r8r:r?r@rCrGrJrNrQrSrZr\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�rrrrr	r
rrrrrr r.r4r9r:r;r=r?r@rCr5r5r5r6r
�s�.		


3			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�
TestBroadcastcCr)N��[rrSr5r5r6r�rzTestBroadcast.setupcCst�|j�dSr0)rrcrSr5r5r6�set_seedszTestBroadcast.set_seedcCsldg}dg}tj}t�gd��}|��||d|�}t||dd�|��|||d�}t||dd�dS)NrrV�g�l���?g_x
���?g=�*BN�?r\r�r)rrrKr_rFr)rGrrrrrr5r5r6rszTestBroadcast.test_uniformcC�dg}dg}dg}tj}t�gd��}|��||d|�}t||dd�tt||d|�|��|||d�}t||dd�tt|||d�dS)NrrVrO)g-r��@g�Y�l�@g-�`�w�?r\r�r)rr�rKr_rFrrrR)rGr�r��	bad_scaler�rrr5r5r6r�&�zTestBroadcast.test_normalcC�dg}dg}dg}dg}tj}t�gd��}|��||d|�}t||dd�tt||d|�tt||d|�|��|||d�}t||dd�tt|||d�tt|||d�dS)	NrVrrOrW)gn�QVf�?gV���KB�?gv�`k��?r\r�r)rr�rKr_rFrrrR)rGr>r?�bad_a�bad_br�rrr5r5r6r�9� zTestBroadcast.test_betacC�Vdg}dg}tj}t�gd��}|��||d�}t||dd�tt||d�dS�NrVrO)g�.Ug�Z�?giPr�q�?g��E
K��?r\r�r)rr�rKr_rFrrrR)rGr�rIr�rrr5r5r6r�O�zTestBroadcast.test_exponentialcCrOrP)rrrKr_rFrrrR)rGr��	bad_shape�	std_gammarrr5r5r6r	\rQz!TestBroadcast.test_standard_gammacCrK)	NrVrrOrW)g�.Ug�Z�?giPr�q�?g��E
K�?r\r�r)rr�rKr_rFrrrR)rGr�r�rRrIr�rrr5r5r6r�irNzTestBroadcast.test_gammacCrK)	NrVrrOrW)gb��yʜ�?g��o���?gT��7�@r\r�r)rr�rKr_rFrrrR)rGr�r��	bad_dfnum�	bad_dfdenr�rrr5r5r6r�rNzTestBroadcast.test_fc
	Cs�dg}dg}dg}dg}dg}dg}tj}t�gd��}|��||d||�}	t|	|dd	�t�t�|||tjgd���sAJ�t	t
||d||�t	t
||d||�t	t
||d||�|��|||d|�}	t|	|dd	�t	t
|||d|�t	t
|||d|�t	t
|||d|�|��||||d�}	t|	|dd	�t	t
||||d�t	t
||||d�t	t
||||d�dS)
Nrr\r]rrOrW)g�r�^G"@g/�
*@g���ц�!@r�r)rr�rKr_rFrrzr�rorrR)
rGr�r�r�rTrU�bad_nonc�nonc_frrr5r5r6r��s6$zTestBroadcast.test_noncentral_fcCs:|��t�ddg�}tjddddd�}t||dd�dS)Ng��a��z@gn4M;�%�?r�rr�r�r)rFrKr_rr�r)rGrrr5r5r6�test_noncentral_f_small_df�sz(TestBroadcast.test_noncentral_f_small_dfcCrO)NrVrO)g~�}�N?�?g�$F���?g'rS��?r\r�r)rr�rKr_rFrrrR)rGr��bad_dfr�rrr5r5r6r��rQzTestBroadcast.test_chisquarecCrK)	NrVrrOrW)g�� w�"@g2��WR@g���RY@r\r�r)rr�rKr_rFrrrR)rGr�r�rYrV�nonc_chirrr5r5r6r��rNz'TestBroadcast.test_noncentral_chisquarecC�hdg}dg}tj}t�gd��}|��||d�}t||dd�tt||d�tttj|d�dS)NrVrO)gT
��@g~�F�l@gƚt��p�?r\r�r)rr
rKr_rFrrrR)rGr�rY�trrr5r5r6r��zTestBroadcast.test_standard_tcCrH)NrrVrO)g��W!�@g��OB���g�}�?�r\r�r)rr3rKr_rFrrrR)rGr�r1�	bad_kappar3rrr5r5r6r4�rJzTestBroadcast.test_vonmisescCr[)NrVrO)g&�8�?�?g�e��FX�?g�Ջː��?r\r�r)rr�rKr_rFrrrR)rGr>rLr�rrr5r5r6r�r]zTestBroadcast.test_paretocCr[rP)rr>rKr_rFrrrR)rGr>rLr>rrr5r5r6r?r]zTestBroadcast.test_weibullcCr[)NrVrOrGr\r�r)rr�rKr_rFrrrR)rGr>rLr�rrr5r5r6r�r]zTestBroadcast.test_powercCrH)NrrVrO)g:��BKc�?g�3��k�?g�~q���?r\r�r)rr�rKr_rFrrrR)rGr�r�rIr�rrr5r5r6r�,rJzTestBroadcast.test_laplacecCrH)NrrVrO)g�'��Zy�?g�dSO=�?g�3L�1��?r\r�r)rr�rKr_rFrrrR)rGr�r�rIr�rrr5r5r6r�?rJzTestBroadcast.test_gumbelcCs�dg}dg}dg}tj}t�gd��}|��||d|�}t||dd�tt||d|�|��|||d�}t||dd�tt|||d�tt�dd	�d�dS)
NrrVrO)g“y��?g��J�R��?g�
~�&��?r\r�rr(rw)	rr�rKr_rFrrrRr)rGr�r�rIr�rrr5r5r6r�RszTestBroadcast.test_logisticcCs�dg}dg}dg}tj}t�gd��}|��||d|�}t||dd�tt||d|�tttj|d|�|��|||d�}t||dd�tt|||d�tttj||d�dS)NrrVrO)g����H"@g��ܮ�� @gvv	��:@r\r�r)rr�rKr_rFrrrR)rGr�r��	bad_sigmar�rrr5r5r6r�fszTestBroadcast.test_lognormalcCrO)NrVrO)gݱl�o��?g�/ӂ���?gz~�R�?r\r�r)rrrKr_rFrrrR)rGr�rIrrrr5r5r6r{rQzTestBroadcast.test_rayleighcCsdg}dg}dg}dg}tj}t�gd��}|��||d|�}t||dd�tt||d|�tt||d|�tttj|d|�tttj|d|�|��|||d�}t||dd�tt|||d�tt|||d�tt|d	d�tt|dd	�dS)
Nr
rVrrW)g@m� �e�?g��`�I߿?g��|2��?r\r�rrw)rr<rKr_rFrrrR)rGr�r��bad_meanrIr<rrr5r5r6r=�s(zTestBroadcast.test_waldcCs�dg}dg}dg}dg}dg}|d\}}tj}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�tt|dd	d
�tt|ddd
�tt|ddd�dS)NrVr\rr])g,�P;bD@g����%G@g,~S��@r�rrrwg4@g9@)rrrKr_rFrrrR)rGrrr�bad_left_one�bad_mode_one�bad_left_two�bad_mode_tworrrr5r5r6r�sF��
�zTestBroadcast.test_triangularc	C��dg}dg}dg}dg}dg}tj}t�gd��}|��||d|�}t||�tt||d|�tt||d|�tt||d|�|��|||d�}t||�tt|||d�tt|||d�tt|||d�dS)NrVr
rO��?�rVrVrVr\)rrrKr_rFrrrR)	rGr�rm�bad_n�	bad_p_one�	bad_p_two�binomrrr5r5r6r���&

zTestBroadcast.test_binomialc	Cre)NrVr
rOrf)rVrrVr\)rrrKr_rFrrrR)	rGr�rmrhrirj�	neg_binomrrr5r5r6r��rlz$TestBroadcast.test_negative_binomialcCsvt��j}dg}dg}|dg}tj}t�gd��}|��||d�}t||�tt	||d�tt	||d�dS)NrVrOr)rVrVrr\)
rrD�_poisson_lam_maxrrKr_rFrrrR)rG�max_lamr��bad_lam_one�bad_lam_tworrrr5r5r6r��s


zTestBroadcast.test_poissoncCs�dg}dg}tj}t�gd��}|��||d�}t||�tt||d�tjdd��tt|tj	�tt|ddtj	g�Wd�dS1sJwYdS)Nrr)rrrVr\r�)�invalid)
rrrKr_rFrrrR�errstatero)rGr>rLrrrr5r5r6rCs
"�zTestBroadcast.test_zipfcC�hdg}dg}dg}tj}t�gd��}|��||d�}t||�tt||d�tt||d�dS)Nr
rOrfrr\)rrrKr_rFrrrR)rGrmrirj�geomrrr5r5r6r��
zTestBroadcast.test_geometriccCs�dg}dg}dg}dg}dg}dg}dg}tj}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�tt|dd	d
�tt|d	dd
�tt|d	d	d�tt|d	d	d�dS)NrVrrOrWrr]rgr\rr�)rrrKr_rFrrrR)rG�ngood�nbad�nsample�	bad_ngood�bad_nbad�bad_nsample_one�bad_nsample_two�	hypergeomrrr5r5r6r�"sD


z!TestBroadcast.test_hypergeometriccCrt)Nr
rrOrgr\)rrrKr_rFrrrR)rGrmrirjrrrr5r5r6r�JrvzTestBroadcast.test_logseriesN)#rgrhrir�rFrr�r�r�r	r�r�r�rXr�r�rr4r�r?r�r�r�r�r�rr=rr�r�r�rCr�r�r�r5r5r5r6rD
sB

#

(
(rDc@s4eZdZdd�Zdd�Zdd�Zdd�Zd	d
�ZdS)�
TestThreadcCstd�|_dS)Nr])rJ�seedsrSr5r5r6r�ZszTestThread.setupcs�ddlm�t�t|j�f|�}t�t|j�f|�}��fdd�t|j|�D�}dd�|D�dd�|D�t|j|�D]\}}�t�|�|�q>t�	�j
jdkr_tj
dkr_t||�dSt||�dS)	Nr)�Threadcs&g|]\}}��t�|�|fd��qS))�target�args)rrD)rkrH�o�r��functionr5r6rmds�z-TestThread.check_function.<locals>.<listcomp>cS�g|]}|���qSr5)�start�rkr|r5r5r6rmfrncSr�r5)�joinr�r5r5r6rmgrnr]�win32)�	threadingr�rKrUr�r��ziprrD�intprZ�itemsizer��platformrr)rGr��sz�out1�out2r\rHr�r5r�r6�check_function]s
�zTestThread.check_functioncC�dd�}|j|dd�dS)NcSs|jdd�|d<dS)N�'r�.)r��r9�outr5r5r6�
gen_randomtrvz*TestThread.test_normal.<locals>.gen_random)r��r��r��rGr�r5r5r6r�s�zTestThread.test_normalcCr�)NcSs|jt�d�d�|d<dS)N�rrBr�.)r�rKr]r�r5r5r6r�zsz'TestThread.test_exp.<locals>.gen_randomr�r�r�r�r5r5r6�test_expyr�zTestThread.test_expcCr�)NcSs|jddgddd�|d<dS)Nrr�rr�r�.)rr�r5r5r6r��sz/TestThread.test_multinomial.<locals>.gen_random)r�rr�r�r�r5r5r6r�r�zTestThread.test_multinomialN)rgrhrir�r�r�r�r�r5r5r5r6r�Xsr�c@s,eZdZdd�Zdd�Zdd�Zdd�Zd	S)
�TestSingleEltArrayInputcCs4t�dg�|_t�dg�|_t�dg�|_d|_dS)Nrr\r]rs)rKr_�argOne�argTwo�argThree�tgtShaperSr5r5r6r��s
zTestSingleEltArrayInput.setupcCs�tjtjtjtjtjtjtjtjtj	tj
tjtjf}tjtjf}|D]}||vr1|t
�dg��}n||j�}t|j|j�q"dSr�)rr�rr�r
r�r>r�rrrrrrKr_r�rr�r�)rG�funcs�	probfuncsrr�r5r5r6�test_one_arg_funcs�s�
�z*TestSingleEltArrayInput.test_one_arg_funcscCs�tjtjtjtjtjtjtjtjtj	tj
tjtjtj
tjf}tj
tjf}|D];}||vr3t�dg�}n|j}||j|�}t|j|j�||jd|�}t|j|j�||j|d�}t|j|j�q&dS)Nr
r)rrr�r�r�r�r�r3r�r�r�r�r<rrrKr_r�r�rr�r�)rGr�r�rr�r�r5r5r6�test_two_arg_funcs�s(��z*TestSingleEltArrayInput.test_two_arg_funcscCs�tjtjtjg}|D]6}||j|j|j�}t|j|j	�||jd|j|j�}t|j|j	�||j|jd|j�}t|j|j	�q
dS)Nr)
rr�rrr�r�r�rr�r�)rGr�rr�r5r5r6�test_three_arg_funcs�s��z,TestSingleEltArrayInput.test_three_arg_funcsN)rgrhrir�r�r�r�r5r5r5r6r��s
r�cCsDt�d�|\}}}tt|�}||ddi�}t|jt�d�k�dS)NrEr�rr�)rrc�getattrrrZrK)r7�fnamer�r�r�rr5r5r6�test_integer_dtype�s



r�cCsdt�d�|\}}}tt|�}||ddi�}tjdkr|��}t�|�t	j
����}t||k�dS)NrEr�r5r�)
rrcr�r�r�r�r�r�r�rKr�r�r)r7r�r�r�r�r�r�r5r5r6�test_integer_repeat�s



r�cCs�t�t��tjdddgdd�Wd�n1swYt�t��tjddgddd�Wd�n1s9wYt�t��tjddgddgdd�Wd�dS1s[wYdS)NrVrLgffffff�?)rrVr�r)r�r�rRrrr5r5r5r6�test_broadcast_size_error�s��"�r�))r�r�r�r<�numpyrKr��
numpy.testingrrrrrrrr	�numpy.randomr
rrr]r2r�rkr�r3�fixturer7r@rArjrqr�r�r
rDr�r�r�r�r�r5r5r5r6�<module>s~(�
�
�
/5W"hO/E

https://t.me/RX1948 - 2025