U
    ÉZjA ã                   @   sÖ   d dl Z d dlZd dlZd dlmZmZmZmZm	Z	m
Z
mZmZ d dlmZ d dlZG dd„ dƒZG dd„ dƒZG dd	„ d	ƒZG d
d„ dƒZG dd„ dƒZG dd„ dƒZG dd„ dƒZG dd„ dƒZG dd„ dƒZdS )é    N)Úassert_Úassert_raisesÚassert_equalÚassert_warnsÚassert_no_warningsÚassert_array_equalÚassert_array_almost_equalÚsuppress_warnings)Úrandomc                   @   s4   e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd	d
„ Zd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   úa/var/www/html/TRUCKING_PROJECT/venv/lib/python3.8/site-packages/numpy/random/tests/test_random.pyÚtest_scalar   s    zTestSeed.test_scalarc                 C   s‚   t j tdƒ¡}t| d¡dƒ t j t  d¡¡}t| d¡dƒ t j dg¡}t| d¡dƒ t j dg¡}t| d¡dƒ d S )Né
   r   iÔ  r   iÍ  r   i	  )r   r
   r   Úranger   r   Úaranger   r   r   r   Ú
test_array   s    zTestSeed.test_arrayc                 C   s$   t ttjjdƒ t ttjjdƒ d S )Nç      à¿éÿÿÿÿ©r   Ú	TypeErrorr   r
   r   Ú
ValueError©r   r   r   r   Útest_invalid_scalar    s    zTestSeed.test_invalid_scalarc                 C   sf   t ttjjdgƒ t ttjjdgƒ t ttjjdgƒ t ttjjdddgƒ t ttjjdddgƒ d S )Nr   r   l        é   é   éþÿÿÿr   r    r   r   r   Útest_invalid_array%   s
    zTestSeed.test_invalid_arrayc                 C   sX   t ttjjtjg tjd�ƒ t ttjjdddggƒ t ttjjdddgdddggƒ 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 Zd Zdd„ Zdd„ ZdS )ÚTestBinomialc                 C   sB   t jddd�}dD ]*}tt d|¡dkƒ tt ||¡|ƒ qd S )Nr#   Úintr&   )r   ç      à?r"   r   )r   Úzerosr   r
   Úbinomialr   )r   r5   Úpr   r   r   Útest_n_zero7   s    zTestBinomial.test_n_zeroc                 C   s   t ttjdtjƒ d S )Nr"   )r   r   r
   r6   r   Únanr    r   r   r   Útest_p_is_nan@   s    zTestBinomial.test_p_is_nanN)r/   r0   r1   r8   r:   r   r   r   r   r2   6   s   	r2   c                   @   s4   e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd	d
„ ZdS )ÚTestMultinomialc                 C   s   t  dddg¡ d S )Néd   çš™™™™™É?çš™™™™™é?©r
   Úmultinomialr    r   r   r   Ú
test_basicF   s    zTestMultinomial.test_basicc                 C   s   t  ddddddg¡ d S )Nr<   r=   r>   ç        r?   r    r   r   r   Útest_zero_probabilityI   s    z%TestMultinomial.test_zero_probabilityc                 C   sZ   t dt dd¡  kodk n  ƒ t ddd¡}t t d|k¡ƒ t t |dk ¡ƒ 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Ð   ddg}t tj d|t d¡¡jdƒ t tj d|t d¡¡jdƒ t tj d|t d¡¡jdƒ t tj d|ddg¡jdƒ t tj d|d¡jdƒ t tj d|t d¡¡jdƒ tttjjd|t	dƒƒ d S )Nr4   r"   ©r"   r#   r#   ©r#   r#   r#   ©r#   r#   )
r   r   r
   r@   Úuint32Úshaper,   r   r   Úfloat©r   r7   r   r   r   Ú	test_sizeR   s    ÿÿzTestMultinomial.test_sizec              	   C   s€   t ttjjdddggƒ t ttjjddgdggƒ t ttjjddgdggdgdgggƒ t ttjjdt ddgddgg¡ƒ d S )Nr   r   r"   )r   r   r   r
   r@   r,   r    r   r   r   Útest_multidimensional_pvals`   s    &z+TestMultinomial.test_multidimensional_pvalsN)r/   r0   r1   rA   rC   rG   rO   rP   r   r   r   r   r;   E   s
   r;   c                   @   s<   e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd	d
„ Zdd„ ZdS )ÚTestSetStatec                 C   s$   d| _ t | j ¡| _| j ¡ | _d S ©NiÒ–I)Úseedr
   r   ÚprngÚ	get_stateÚstater    r   r   r   Úsetuph   s    zTestSetState.setupc                 C   s<   | j  d¡}| j  | j¡ | j  d¡}tt ||k¡ƒ d S )Né   )rT   ZtomaxintÚ	set_staterV   r   r   rE   ©r   ÚoldÚnewr   r   r   rA   m   s    zTestSetState.test_basicc                 C   s@   | j jdd�}| j  | j¡ | j jdd�}tt ||k¡ƒ d S ©Nr(   ©Úsize)rT   Ústandard_normalrY   rV   r   r   rE   rZ   r   r   r   Útest_gaussian_resets   s    z TestSetState.test_gaussian_resetc                 C   sR   | j  ¡  | j  ¡ }| j jdd�}| j  |¡ | j jdd�}tt ||k¡ƒ d S r]   )rT   r`   rU   rY   r   r   rE   )r   rV   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… }| jjdd�}| j |¡ | jjdd�}| j | j ¡ | jjdd�}tt ||k¡ƒ tt ||k¡ƒ d S )Nr$   rX   r^   )rV   rT   r`   rY   r   r   rE   )r   Ú	old_stateÚx1Zx2Zx3r   r   r   Útest_backwards_compatibility…   s    z)TestSetState.test_backwards_compatibilityc                 C   s   | j  dd¡ d S ©Nr4   )rT   Únegative_binomialr    r   r   r   Útest_negative_binomial‘   s    z#TestSetState.test_negative_binomialN)	r/   r0   r1   rW   rA   ra   rb   re   rh   r   r   r   r   rQ   g   s   rQ   c                	   @   s|   e Zd ZejjZejejej	ej
ejejejejejg	Zdd„ Zdd„ Zdd„ Zdd„ Zd	d
„ Zdd„ Zdd„ Zdd„ ZdS )ÚTestRandintc                 C   s   t t| jdtd� d S ©Nr"   r&   )r   r   ÚrfuncrM   r    r   r   r   Útest_unsupported_typeŸ   s    z!TestRandint.test_unsupported_typec                 C   s    | j D ]”}|tjkrdn
t |¡j}|tjkr2dnt |¡jd }tt| j|d ||d� tt| j||d |d� tt| j|||d� tt| jdd|d� qd S )Nr   r#   r"   r&   )	Úityper   Úbool_ÚiinfoÚminÚmaxr   r   rk   )r   ÚdtÚlbndÚubndr   r   r   Útest_bounds_checking¢   s    
z TestRandint.test_bounds_checkingc                 C   s´   | j D ]¨}|tjkrdn
t |¡j}|tjkr2dnt |¡jd }|d }t| j||d d|d�|ƒ |}t| j||d d|d�|ƒ || d }t| j||d d|d�|ƒ qd S )Nr   r#   r"   r   ©r_   r'   )rm   r   rn   ro   rp   rq   r   rk   )r   rr   rs   rt   Útgtr   r   r   Útest_rng_zero_and_extremes«   s    
z&TestRandint.test_rng_zero_and_extremesc                 C   s�   | j D ]„}|tjkrdn
t |¡j}|tjkr2dnt |¡jd }z| j|||d� W q tk
rˆ } ztdt	|ƒ ƒ‚W 5 d }~X Y qX qd S )Nr   r#   r"   r&   zMNo error should have been raised, but one was with the following message:

%s)
rm   r   rn   ro   rp   rq   rk   Ú	ExceptionÚAssertionErrorÚstr)r   rr   rs   rt   Úer   r   r   Útest_full_range¹   s    
þzTestRandint.test_full_rangec                 C   s’   t j ¡  | jdd … D ]@}dD ]6}| jd|d|d�}t| ¡ |k ƒ t| ¡ dkƒ q q| jdddt jd�}t| ¡ dk ƒ t| ¡ dkƒ d S )Nr"   )r)   é   rX   r#   i   rv   r   )	r   r
   rS   rm   rk   r   rq   rp   rn   )r   rr   rt   Úvalsr   r   r   Útest_in_bounds_fuzzÇ   s    
zTestRandint.test_in_bounds_fuzzc              
   C   sò   dd l }ddddddddddœ	}| jdd … D ]r}tj d	¡ tjd
kr\| jddd|d�}n| jddd|d� ¡ }| 	| 
tj¡¡ ¡ }t|t |¡j |kƒ q.tj d	¡ | jdddtd� 
tj¡}| 	|¡ ¡ }t|t t¡j |kƒ d S )Nr   Z@509aea74d792fb931784c4b0135392c65aec64beee12b0cc167548a2c3d31e71Z@7b07f1a920e46f6d0fe02314155a2330bcfd7635e708da50e536c5ebb631a7d4Z@e577bfed6c935de944424667e3da285012e741892dcb7051a8f1ce68ab05c92fZ@0fbead0b06759df2cfb55e43148822d4a1ff953c7eb19a5b08445a63bb64fa9eZ@001aac3a5acb935a9b186cbe14a1ca064b8bb2dd0b045d48abeacf74d0203404)	ÚboolÚint16Úint32r-   Úint8Úuint16rK   Úuint64Úuint8r"   iÒ  Úlittler+   r   rv   r#   )Úhashlibrm   r   r
   rS   ÚsysÚ	byteorderrk   ÚbyteswapÚsha256Úviewr„   Ú	hexdigestr   r'   Únamer�   )r   r‰   rw   rr   Ú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¡jd ¡}t jj|||d�}t||ƒ d S rj   )r   r-   ro   rq   r†   r
   r   r   )r   rr   rw   rs   rt   Úactualr   r   r   Útest_int64_uint64_corner_case÷   s    z)TestRandint.test_int64_uint64_corner_casec                 C   sà   | j D ]^}|tjkrdn
t |¡j}|tjkr2dnt |¡jd }| j|||d�}t|jt |¡ƒ qt	t
tjjfD ]f}|t	kr„dn
t |¡j}|t	krœdnt |¡jd }| j|||d�}tt|dƒ ƒ tt|ƒ|ƒ qtd S )Nr   r#   r"   r&   r'   )rm   r   rn   ro   rp   rq   rk   r   r'   r�   r3   ÚcompatÚlongr   ÚhasattrÚtype)r   rr   rs   rt   Úsampler   r   r   Útest_respect_dtype_singleton  s    
z(TestRandint.test_respect_dtype_singletonN)r/   r0   r1   r   r
   r   rk   rn   r„   r‡   r‚   r…   rƒ   rK   r-   r†   rm   rl   ru   rx   r}   r€   r“   r•   r›   r   r   r   r   ri   —   s      ÿ	!ri   c                   @   sª  e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd	d
„ Zdd„ Zdd„ Z	dd„ Z
dd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd „ Zd!d"„ Zd#d$„ Zd%d&„ Zej d'ejej ¡ ej ¡ g¡d(d)„ ƒZej d'ejej ¡ ej ¡ g¡ej 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~„ 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—S )˜ÚTestRandomDistc                 C   s
   d| _ d S rR   ©rS   r    r   r   r   rW   %  s    zTestRandomDist.setupc                 C   sJ   t j | j¡ t j dd¡}t  ddgddgddgg¡}t||d	d
� d S )Nr(   r#   ç_®\±*Íã?ç�›˜¬”îâ?ç<î�pì?ç®,º¥oˆì?çˆÂÉHÝ?çZ±­°êæè?é   ©Údecimal)r   r
   rS   Zrandr,   r   ©r   r”   Údesiredr   r   r   Ú	test_rand(  s    
þzTestRandomDist.test_randc                 C   sJ   t j | j¡ t j dd¡}t  ddgddgddgg¡}t||d	d
� d S )Nr(   r#   çÿ!ã<Oqõ?ç4ð�u,Íû?çÁ³½3Ûû÷?çL«I/DÍ¿çà@ËÀŽ? @çÿHSÓ\@r¤   r¥   )r   r
   rS   Zrandnr,   r   r§   r   r   r   Ú
test_randn0  s    
þzTestRandomDist.test_randnc                 C   sJ   t j | j¡ t jjdddd�}t  ddgddgd	d
gg¡}t||ƒ d S )Né�ÿÿÿéc   ©r(   r#   r^   é   r(   éÌÿÿÿé)   éÐÿÿÿé¾ÿÿÿ)r   r
   rS   r   r,   r   r§   r   r   r   Útest_randint8  s    
þzTestRandomDist.test_randintc              	   C   sv   t j | j¡ tƒ �2}| t¡}t jjdddd�}tt|ƒdkƒ W 5 Q R X t  	ddgdd	gd
dgg¡}t
||ƒ d S )Nr±   r²   r³   r^   r"   r´   r(   rµ   r¶   r·   r¸   )r   r
   rS   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¡}tj t d¡jt d¡j¡}tt	|ƒdkƒ W 5 Q R X t d¡j}t
||ƒ d S )NÚlr"   )r	   rº   r»   r   r
   r¼   ro   rq   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  dt¡ tttjjt d¡j	ƒ tttjjt d¡j	t d¡j	ƒ W 5 Q R X d S )NÚerrorrÂ   )
ÚwarningsÚcatch_warningsÚsimplefilterr»   r   r   r
   r¼   ro   rq   r    r   r   r   Útest_random_integers_deprecatedZ  s    

þ
 
þz.TestRandomDist.test_random_integers_deprecatedc                 C   sH   t j | j¡ t j d¡}t  ddgddgddgg¡}t||dd	� d S )
Nr³   rž   rŸ   r    r¡   r¢   r£   r¤   r¥   )r   r
   rS   r,   r   r§   r   r   r   Útest_randomh  s    
þzTestRandomDist.test_randomc                 C   s<   t j | j¡ t j dd¡}t  ddddg¡}t||ƒ d S )Nr)   r#   r(   ©r   r
   rS   Úchoicer,   r   r§   r   r   r   Útest_choice_uniform_replacep  s    z*TestRandomDist.test_choice_uniform_replacec                 C   sH   t j | j¡ t jjddddddgd�}t  ddddg¡}t||ƒ d S )Nr)   çš™™™™™Ù?çš™™™™™¹?©r7   r"   r#   rÊ   r§   r   r   r   Útest_choice_nonuniform_replacev  s    z-TestRandomDist.test_choice_nonuniform_replacec                 C   s>   t j | j¡ t jjdddd�}t  dddg¡}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 jjdddddddgd�}t  ddd	g¡}t||ƒ d S )
Nr)   r(   FrÎ   g333333Ó?r4   ©rÒ   r7   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 ddddgd¡}t  ddddg¡}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 jj}tt|ddƒ tt|ddƒ tt|ddgddggdƒ tt|g dƒ tt|ddddgdddgddggd� tt|ddgdd	d	d
gd� tt|ddgdddgd� tt|ddgdd	d	gd� tt|dddgddd� tt|dddgddd� tt|dddgddd� tt|dddgddd� tt|dddgdddddgd� 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<  ddg}t t tjjddd�¡ƒ t t tjjddd�¡ƒ t t tjjdd|d�¡ƒ t t tjjdd|d�¡ƒ t t tjjddgdd�¡ƒ t tjjd gdd�d kƒ t ddg¡}tjdtd	�}||d
< t tjj|dd�|kƒ tƒ }t t tjjd|dd�¡ ƒ t t tjjd|dd�¡ ƒ t t tjjd|d|d�¡ ƒ t t tjjd|d|d�¡ ƒ t t tjjddg|dd�¡ ƒ t tjjd g|dd�j	d
kƒ t ddg¡}tjdtd	�}||d
< t tjj||dd� 
¡ |kƒ d}ddddddg}ttjjd|dd�j|ƒ ttjjd|dd�j|ƒ ttjjd|d|d�j|ƒ ttjjd|d|d�j|ƒ ttjjt d¡|dd�j|ƒ ttjjd
d
dd�jdƒ ttjjd
dd
d�jdƒ ttjjddd
d�jdƒ ttjjd
d
d�jdƒ ttjjg dd�jdƒ ttjjddgdd�jdƒ tttjjg 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   Zisscalarr
   rË   r,   ÚemptyÚobjectÚtupleÚndimÚitemr   rL   r   r   r   r   )r   r7   rÖ   Úarrr   r   r   r   Útest_choice_return_shape¢  sL      " ÿz'TestRandomDist.test_choice_return_shapec                 C   s2   t  dddg¡}d d d g}ttt jj||d� d S )Né*   r"   r#   rÏ   )r   r,   r   r   r
   rË   )r   rÖ   r7   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
   rS   Úbytesr   r§   r   r   r   Ú
test_bytesÕ  s    zTestRandomDist.test_bytesc                 C   s°   dd„ dd„ dd„ dd„ dd„ dd„ dd„ d	d„ d
d„ dd„ dd„ fD ]d}t j | j¡ |ddddddddddg
ƒ}t j |¡ |}|ddddddddddg
ƒ}t||ƒ qFd S )Nc                 S   s
   t  g ¡S ©N)r   r,   ©rF   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   dd„ | D ƒS )Nc                 S   s   g | ]}||f‘qS 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  dd„ | D ƒ¡S )Nc                 S   s   g | ]}||g‘qS r   r   rï   r   r   r   rò   å  s     ró   )r   rì   ré   r   r   r   rê   å  rë   c                 S   s   t  | | g¡jS rè   )r   ZvstackÚTré   r   r   r   rê   æ  rë   c                 S   s*   t  dd„ | D ƒdtfdtfg¡ t j¡S )Nc                 S   s   g | ]}||f‘qS r   r   rï   r   r   r   rò   è  s     ró   rÖ   r×   )r   rì   r3   rŽ   Zrecarrayré   r   r   r   rê   è  s   ÿþc                 S   s$   t  dd„ | D ƒdtfdt jfg¡S )Nc                 S   s   g | ]}||f‘qS 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
   rS   Úshuffler   )r   ÚconvÚalistr”   r¨   r   r   r   Útest_shuffleÛ  s$    òzTestRandomDist.test_shufflec                 C   sÄ   t j t  tdƒd¡d d d¡}t j t  d¡d d d¡}| ¡ }| ¡ }tdƒD ]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_valuesZreshaper   r   Úcopyr
   r÷   r   ÚsortedÚdataÚmask)r   rÖ   r×   Za_origZb_origrñ   r   r   r   Útest_shuffle_maskedõ  s    " ÿ ÿz"TestRandomDist.test_shuffle_maskedr
   c              	   C   sN   ddddddddœ}t jtd	d
��}| |¡ W 5 Q R X d|d jksJt‚d 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÷   Úfilenamerz   )r   r
   ÚvaluesZrecr   r   r   Útest_shuffle_untyped_warning  s    ÿz+TestRandomDist.test_shuffle_untyped_warningÚuse_array_likeTFc              	      sÞ   G dd„ dt jƒ}d t  dg¡t  d¡t  d¡t  d¡g}t j|td�}dd„ |D ƒ‰ |rd| |¡}t‡ fd	d
„|D ƒƒs~t‚|r¶t|t j	j
ƒs¶tjtdd�� | |¡ W 5 Q R X n$| |¡ t‡ fdd
„|D ƒƒsÚt‚d S )Nc                   @   s   e Zd ZdS )z>TestRandomDist.test_shuffle_no_object_unpacking.<locals>.MyArrN)r/   r0   r1   r   r   r   r   ÚMyArr  s   r  r(   r   rõ   r&   c                 S   s   h | ]}t |ƒ’qS r   ©Úidrï   r   r   r   Ú	<setcomp>  s     zBTestRandomDist.test_shuffle_no_object_unpacking.<locals>.<setcomp>c                 3   s   | ]}t |ƒˆ kV  qd S rè   r  rï   ©Zitem_idsr   r   Ú	<genexpr>  s     zBTestRandomDist.test_shuffle_no_object_unpacking.<locals>.<genexpr>z#Shuffling a one dimensional array.*r  c                 3   s   | ]}t |ƒˆ kV  qd S rè   r  rï   r  r   r   r  &  s     )r   Úndarrayr,   Zfloat64rÞ   rŽ   rE   rz   Ú
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 |¡ tt  |¡dddddgƒ t j | j¡}| |¡ tt  |¡dddddgƒ t j 	| j¡}| |¡ tt  |¡dddddgƒ d S )Nr*   r   r"   r)   r(   r#   )
r   r
   rS   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_tjtdd�� t j |¡ W 5 Q R X d S )Nr(   Fz	read-onlyr  )	r   r5   ÚflagsZ	writeabler  Zraisesr   r
   r÷   )r   rÖ   r   r   r   Útest_shuffle_not_writeable7  s    
z)TestRandomDist.test_shuffle_not_writeablec                 C   sN   t j | j¡ t jjdddd�}t  ddgddgd	d
gg¡}t||dd� d S )NrÎ   rÜ   r³   r^   gAàÁKÄ�?g‘ñØhA?g õÒ‡o¿>gÎ¬XfÈ+q?gSâë1Ã$?g'8zXŒ ?r¤   r¥   )r   r
   rS   Úbetar,   r   r§   r   r   r   Ú	test_beta=  s    þÿzTestRandomDist.test_betac                 C   sJ   t j | j¡ t jjdddd�}t  ddgddgd	d
gg¡}t||ƒ d S )Nr<   gÉv¾Ÿ/Ý?r³   r^   é%   é+   rä   é0   é.   é-   )r   r
   rS   r6   r,   r   r§   r   r   r   Útest_binomialF  s    
þzTestRandomDist.test_binomialc                 C   sL   t j | j¡ t jjddd�}t  ddgddgdd	gg¡}t||d
d� d S )Nrü   r³   r^   g+Qð]uðO@gëòìÇ+Q@gY7`´ZqP@gÈe‡(fŒG@gþv¯t€R@gEIÈ‹R@é   r¥   )r   r
   rS   Ú	chisquarer,   r   r§   r   r   r   Útest_chisquareN  s    
þzTestRandomDist.test_chisquarec                 C   st   t j | j¡ t  ddg¡}t jjj|dd�}t  ddgddggd	d
gddggddgddggg¡}t||dd� d S )Nç“­I<ÝI@ç÷ÉEZßC@r³   r^   gÉi{ßsá?gn,	åAÝ?göÇ]‹^óã?gpDéBØ?gÈ§#¼yªá?gr°¸‡«Ü?g—«1UÞâ?gÈÑ¨œUCÚ?g£…^,%÷â?g»ôB§µÚ?gÚ‰OoX;â?gMì`!O‰Û?r¤   r¥   )r   r
   rS   r,   ÚmtrandÚ	dirichletr   )r   Úalphar”   r¨   r   r   r   Útest_dirichletV  s    
ÿÿÿüzTestRandomDist.test_dirichletc                 C   sÈ   t  ddg¡}tt j |t  d¡¡jdƒ tt j |t  d¡¡jdƒ tt j |t  d¡¡jdƒ tt j |ddg¡jdƒ tt j |d¡jdƒ tt j |t  d¡¡jdƒ ttt jj|t	dƒƒ d S )Nr'  r(  r"   rH   r#   rI   rJ   )
r   r,   r   r
   r*  rK   rL   r   r   rM   rN   r   r   r   Útest_dirichlet_sizeb  s    z"TestRandomDist.test_dirichlet_sizec                 C   s�   t  ddg¡}ttt jjj|ƒ tttjddggƒ tttjdgdggƒ tttjdgdggdgdgggƒ tttjt  ddgddgg¡ƒ 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 jjddd�}t  ddgddgdd	gg¡}t||d
d� d S )Ngï8EGrùñ?r³   r^   gN÷·Uñ?gbI0æð?gùVU\õº@gˆ:€ì3ù@g©ºUýå?g�Ÿ"qoû?r¤   r¥   )r   r
   rS   Úexponentialr,   r   r§   r   r   r   Útest_exponentialy  s    
þzTestRandomDist.test_exponentialc                 C   s*   t tjjdd�dƒ tttjjdd� d S ©Nr   ©Úscaleç       €)r   r   r
   r/  r   r   r    r   r   r   Útest_exponential_0�  s    z!TestRandomDist.test_exponential_0c                 C   sN   t j | j¡ t jjdddd�}t  ddgddgd	d
gg¡}t||dd� d S )Né   éM   r³   r^   g7¶„ó?gµ`j‹�ü?g6¶"+÷?gÜ³2>•‰ó?g‘cÍ>+Yð?g¡_}áS‚õ?r¤   r¥   )r   r
   rS   Úfr,   r   r§   r   r   r   Útest_f…  s    
þzTestRandomDist.test_fc                 C   sN   t j | j¡ t jjdddd�}t  ddgddgd	d
gg¡}t||dd� d S )Nr*   r(   r³   r^   gÔ�GMçš8@g §â”ÈŒ<@g˜	´":@gÂvÀñÇ#)@gZQø·?@gr‚‚[•¦@@é   r¥   )r   r
   rS   Úgammar,   r   r§   r   r   r   Ú
test_gamma�  s    
þzTestRandomDist.test_gammac                 C   s.   t tjjddd�dƒ tttjjddd� d S )Nr   )rL   r3  r4  )r   r   r
   r;  r   r   r    r   r   r   Útest_gamma_0•  s    zTestRandomDist.test_gamma_0c                 C   sH   t j | j¡ t jjddd�}t  ddgddgddgg¡}t||ƒ d S )	Nç_c97Ýš¿?r³   r^   r~   rõ   é   r*   r6  )r   r
   rS   Ú	geometricr,   r   r§   r   r   r   Útest_geometric™  s    
þzTestRandomDist.test_geometricc                 C   sN   t j | j¡ t jjdddd�}t  ddgddgd	d
gg¡}t||dd� d S )Nr>  ç       @r³   ©Úlocr3  r_   gåûØ•ßÉ?g¨s0í Ö?gÊð#0÷¿gï(ôx”÷¿g-¶KÇD´ñ?géÅ¯p`@æ¿r¤   r¥   )r   r
   rS   Úgumbelr,   r   r§   r   r   r   Útest_gumbel¡  s    
þzTestRandomDist.test_gumbelc                 C   s*   t tjjdd�dƒ tttjjdd� d S r1  )r   r   r
   rE  r   r   r    r   r   r   Útest_gumbel_0©  s    zTestRandomDist.test_gumbel_0c                 C   s  t j | j¡ t jjddddd�}t  ddgddgddgg¡}t||ƒ t jjdddd	d�}t  ddddg¡}t||ƒ t jjd
ddd	d�}t  ddddg¡}t||ƒ t jjdddd	d�}t  ddddg¡}t||ƒ t jjdd
dd	d�}t  ddddg¡}t||ƒ d S )Nr   r*   r:  r³   r^   rö   r   r(   r)   r¤   r6  )r   r
   rS   Úhypergeometricr,   r   r§   r   r   r   Útest_hypergeometric­  s&    
þ



z"TestRandomDist.test_hypergeometricc                 C   sN   t j | j¡ t jjdddd�}t  ddgddgd	d
gg¡}t||dd� d S )Nr>  rB  r³   rC  gð¥ bÙOå?g¹Â¸ëÉçà?gpJmµú	@g{KÏ*Ët	@g•:^w-š«¿gˆY~oõûû?r¤   r¥   )r   r
   rS   Úlaplacer,   r   r§   r   r   r   Útest_laplaceÇ  s    
þzTestRandomDist.test_laplacec                 C   s*   t tjjdd�dƒ tttjjdd� d S r1  )r   r   r
   rJ  r   r   r    r   r   r   Útest_laplace_0Ï  s    zTestRandomDist.test_laplace_0c                 C   sN   t j | j¡ t jjdddd�}t  ddgddgd	d
gg¡}t||dd� d S )Nr>  rB  r³   rC  g#€�K-zñ?g€îJš¬ë?g�92ÄÜ@gØ±ã÷[@gê#+^ÑÀË¿g¢¥-óâ@r¤   r¥   )r   r
   rS   Úlogisticr,   r   r§   r   r   r   Útest_logisticÓ  s    
þzTestRandomDist.test_logisticc                 C   sN   t j | j¡ t jjdddd�}t  ddgddgd	d
gg¡}t||dd� d S )Nr>  rB  r³   )ÚmeanÚsigmar_   g›ŽòÚÉ�0@goÎ,+4FB@gù�ƒ)Ê­6@g†§5éêæ?gnþxN—nP@g…©)‡úµU@r$  r¥   )r   r
   rS   Ú	lognormalr,   r   r§   r   r   r   Útest_lognormalÛ  s    
þzTestRandomDist.test_lognormalc                 C   s*   t tjjdd�dƒ tttjjdd� d S )Nr   )rP  r"   r4  )r   r   r
   rQ  r   r   r    r   r   r   Útest_lognormal_0ã  s    zTestRandomDist.test_lognormal_0c                 C   sH   t j | j¡ t jjddd�}t  ddgddgddgg¡}t||ƒ d S )NgÆ€@õŒí?r³   )r7   r_   r#   r+   r?  r(   )r   r
   rS   Ú	logseriesr,   r   r§   r   r   r   Útest_logseriesç  s    
þzTestRandomDist.test_logseriesc                 C   s˜   t j | j¡ t jjddgd dd�}t  ddddd	d	gdd	d
d	d	dggddddddgd	dddddggddd	dd	dgdddd	ddggg¡}t||ƒ d S )Nrû   çUUUUUUÅ?r+   r³   r^   r)   r(   r*   r#   r~   r"   r   )r   r
   rS   r@   r,   r   r§   r   r   r   Útest_multinomialï  s    ÿÿÿüzTestRandomDist.test_multinomialc              	   C   sR  t j | j¡ d}ddgddgg}d}t j |||¡}t  ddgddggd	d
gddggddgddggg¡}t||dd� t j ||¡}t  ddg¡}t||dd� ddg}ddgddgg}ttt jj||ƒ tt jj||dd� t	t
t jj||dd� t jddgddggt jd�}tƒ �0}t j ||¡ | t¡}t|ƒdk�sDt‚W 5 Q R X d S )N)r>  r   r"   r   r³   g4¸Výj÷?g¾±Ž¥y'@g÷I1‰õù?g—SÙBïŠ#@gû…ªe<@g ÒÔ4W(@g6ŠRÂ¿„û?gò9r
v"@g‹–Ñ�æ?g{§×ïÂ#@gò4Íx›—¿gÇçú"öf"@r¤   r¥   gN}©O6¦ì?gÐÇYM÷W"@r#   Úignore)Zcheck_validÚraiserÎ   r&   )r   r
   rS   Zmultivariate_normalr,   r   r   ÚRuntimeWarningr   r   r   rî   r	   rº   r½   rz   )r   rO  Zcovr_   r”   r¨   r¿   rÀ   r   r   r   Útest_multivariate_normalú  sB    
ÿÿÿüÿÿ
z'TestRandomDist.test_multivariate_normalc                 C   sJ   t j | j¡ t jjdddd�}t  ddgddgd	d
gg¡}t||ƒ d S )Nr<   g|ò°Pkš¿?r³   )Únr7   r_   iP  iI  i|  ic  i  i‡  )r   r
   rS   rg   r,   r   r§   r   r   r   rh   "  s    
þz%TestRandomDist.test_negative_binomialc                 C   sÔ   t j | j¡ t jjdddd�}t  ddgddgdd	gg¡}t||d
d� t jjdddd�}t  ddgddgddgg¡}t||d
d� t j | j¡ t jjdddd�}t  ddgddgddgg¡}t||d
d� d S )Nr*   r³   )ÚdfÚnoncr_   gôûÖGë7@g!© ÈÜ´*@gž1}“z9?@g¹+é©¸™0@g26E_r#@g¦�I�!ó1@r:  r¥   r4   r=   g�÷ô‹÷?g’šT¶ˆDÃ?gÀõ?Tƒ?gÍ+òqnlð?gÓ~|òùDÕ?gë°¬òÇÃ?r   g›{¿0¾1#@gpŽ§²rs'@gþ~;ïÑÓ$@g"‹†dIŽ@gö°Õöë÷*@gÑ}Ÿ'Á,@)r   r
   rS   Únoncentral_chisquarer,   r   r§   r   r   r   Útest_noncentral_chisquare*  s(    
þ
þ
þz(TestRandomDist.test_noncentral_chisquarec                 C   sP   t j | j¡ t jjddddd�}t  ddgdd	gd
dgg¡}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
   rS   Únoncentral_fr,   r   r§   r   r   r   Útest_noncentral_f?  s    ÿ
þz TestRandomDist.test_noncentral_fc                 C   sN   t j | j¡ t jjdddd�}t  ddgddgd	d
gg¡}t||dd� d S )Nr>  rB  r³   rC  gíœ&&n@gO»G_Ê@gÜ~w²ø@gtÁÜûw]Õ¿gm&¨5ú½@gŒñ$È>Û@r¤   r¥   )r   r
   rS   Únormalr,   r   r§   r   r   r   Útest_normalH  s    
þzTestRandomDist.test_normalc                 C   s*   t tjjdd�dƒ tttjjdd� d S r1  )r   r   r
   re  r   r   r    r   r   r   Útest_normal_0P  s    zTestRandomDist.test_normal_0c                 C   sP   t j | j¡ t jjddd�}t  ddgddgdd	gg¡}t jj||d
d� d S )Nr>  r³   ©rÖ   r_   gÅWò˜I£@gùÔ¨y–@gy£Ÿ]Ó0‰Ag1~ÖÒ\�AgåËíãša@gÑY„
²7Aé   )Znulp)r   r
   rS   Úparetor,   ÚtestingZassert_array_almost_equal_nulpr§   r   r   r   Útest_paretoT  s    þÿ
zTestRandomDist.test_paretoc                 C   sH   t j | j¡ t jjddd�}t  ddgddgddgg¡}t||ƒ d S )Nr>  r³   )Úlamr_   r   r"   )r   r
   rS   Úpoissonr,   r   r§   r   r   r   Útest_poissonc  s    
þzTestRandomDist.test_poissonc                 C   s`   t  d¡j}d}ttt jj|ƒ ttt jj|gd ƒ ttt jj|ƒ ttt jj|gd ƒ d S )NrÂ   r   r   )r   ro   rq   r   r   r
   rn  )r   ZlambigZlamnegr   r   r   Útest_poisson_exceptionsk  s    z&TestRandomDist.test_poisson_exceptionsc                 C   sL   t j | j¡ t jjddd�}t  ddgddgdd	gg¡}t||d
d� d S )Nr>  r³   rh  gõ;•'û”?gÜ´èwÞ*�?g×†5›Ø?gF¡GÙ?g ÆËD]?gýÞ$`°ÉÀ?r¤   r¥   )r   r
   rS   Úpowerr,   r   r§   r   r   r   Ú
test_powers  s    
þzTestRandomDist.test_powerc                 C   sL   t j | j¡ t jjddd�}t  ddgddgdd	gg¡}t||d
d� d S )Nr   r³   )r3  r_   g"íu¨ÈÆ+@gîN$MBÄ*@g7v8Bô4@gêÿª55@geãˆ&@g~ŽÓ£ÌZ1@r:  r¥   )r   r
   rS   Úrayleighr,   r   r§   r   r   r   Útest_rayleigh{  s    
þzTestRandomDist.test_rayleighc                 C   s*   t tjjdd�dƒ tttjjdd� d S r1  )r   r   r
   rs  r   r   r    r   r   r   Útest_rayleigh_0ƒ  s    zTestRandomDist.test_rayleigh_0c                 C   sJ   t j | j¡ t jjdd�}t  ddgddgddgg¡}t||d	d
� d S )Nr³   r^   gMo³DL®è?g6Ï([9ÀgðºrF=òí?gÉM'	-™ ÀgóV¼ëûÀgVA:ìQyÇ?r¤   r¥   )r   r
   rS   Zstandard_cauchyr,   r   r§   r   r   r   Útest_standard_cauchy‡  s    
þz#TestRandomDist.test_standard_cauchyc                 C   sJ   t j | j¡ t jjdd�}t  ddgddgddgg¡}t||d	d
� d S )Nr³   r^   g°ö–Ü�Üî?gc!z¨ì?g&N˜¥"�@g¬NIàŠÇ@gFÇï>ú’ã?g¡~‚MEø?r¤   r¥   )r   r
   rS   Zstandard_exponentialr,   r   r§   r   r   r   Útest_standard_exponential�  s    
þz(TestRandomDist.test_standard_exponentialc                 C   sL   t j | j¡ t jjddd�}t  ddgddgdd	gg¡}t||d
d� d S )Nr(   r³   )rL   r_   g)§ž@g¾§Ä¾¤„@gd]‰,qÂ@gP‡kÌ{@g9B#Œ1@gÔÚv÷‡ @r:  r¥   )r   r
   rS   Ústandard_gammar,   r   r§   r   r   r   Útest_standard_gamma—  s    
þz"TestRandomDist.test_standard_gammac                 C   s*   t tjjdd�dƒ tttjjdd� d S )Nr   )rL   r4  )r   r   r
   rx  r   r   r    r   r   r   Útest_standard_gamma_0Ÿ  s    z$TestRandomDist.test_standard_gamma_0c                 C   sJ   t j | j¡ t jjdd�}t  ddgddgddgg¡}t||d	d
� d S )Nr³   r^   rª   r«   r¬   r­   r®   r¯   r¤   r¥   )r   r
   rS   r`   r,   r   r§   r   r   r   Útest_standard_normal£  s    
þz#TestRandomDist.test_standard_normalc                 C   sL   t j | j¡ t jjddd�}t  ddgddgdd	gg¡}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
   rS   Ú
standard_tr,   r   r§   r   r   r   Útest_standard_t«  s    
þzTestRandomDist.test_standard_tc                 C   sP   t j | j¡ t jjddddd�}t  ddgdd	gd
dgg¡}t||dd� d S )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   r
   rS   Ú
triangularr,   r   r§   r   r   r   Útest_triangular³  s    ÿ
þzTestRandomDist.test_triangularc                 C   sN   t j | j¡ t jjdddd�}t  ddgddgd	d
gg¡}t||dd� d S )Nç®Gáz®ó?g®Gáz%@r³   )ÚlowÚhighr_   g>#¥HÃö@g£¨uºó@g•ï‡§Ý#@g„}d]#@gø†‰N¶õ@g­w)*ó @r¤   r¥   )r   r
   rS   Úuniformr,   r   r§   r   r   r   Útest_uniform¼  s    
þzTestRandomDist.test_uniformc                 C   sš   t  d¡j}t  d¡j}t jj}tt|t j dƒ tt|dt jƒ tt|||ƒ tt|t j gdgƒ tt|dgt jgƒ t jjt  	|d¡|d d� d S )NrM   r   r"   g  Ø…W4vC)r„  r…  )
r   Zfinforp   rq   r
   r†  r   ÚOverflowErrorÚinfZ	nextafter)r   ZfminZfmaxÚfuncr   r   r   Útest_uniform_range_boundsÄ  s    z(TestRandomDist.test_uniform_range_boundsc                 C   sn   G dd„ dt jƒ}t  d¡ |¡}ttt jj||ƒ G dd„ dt jƒ}t  d¡ |¡}ttt jj|ddƒ d S )Nc                   @   s   e Zd Zdd„ ZdS )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)r/   r0   r1   r�  r   r   r   r   ÚThrowingFloatÛ  s   rŽ  g      ð?c                   @   s   e Zd Zdd„ ZeZdS )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)r/   r0   r1   r�  Ú	__index__r   r   r   r   ÚThrowingIntegerã  s   r‘  r"   )	r   r  r,   rŽ   r   r   r
   r†  rH  )r   rŽ  Zthrowing_floatr‘  Zthrowing_intr   r   r   Ú!test_scalar_exception_propagationÔ  s    ÿz0TestRandomDist.test_scalar_exception_propagationc                 C   sN   t j | j¡ t jjdddd�}t  ddgddgd	d
gg¡}t||dd� d S )Nrƒ  ç¤p=
×£ø?r³   ©ÚmuÚkappar_   gL×üZI@g¦¢'N"@gÀ·…þkrØ?g”k=mœ@g�¹Ú‰ó?gÜ^Ì=�\ý?r¤   r¥   )r   r
   rS   Úvonmisesr,   r   r§   r   r   r   Útest_vonmisesì  s    
þzTestRandomDist.test_vonmisesc                 C   s:   t j | j¡ t jjdddd�}t j t  |¡ ¡ ¡ d S )NrB   gg5_PŸG>i@B r”  )r   r
   rS   r—  rk  r   ÚisfiniterE   )r   Úrr   r   r   Útest_vonmises_smallô  s    z"TestRandomDist.test_vonmises_smallc                 C   sN   t j | j¡ t jjdddd�}t  ddgddgd	d
gg¡}t||dd� d S )Nrƒ  r“  r³   )rO  r3  r_   gÉZ¥ƒ¢@g-ïŸg†@g¦8
ÄÖmÖ?gÃÎVP"ø?g`Vô áÎ?gx©^½&3Ì?r:  r¥   )r   r
   rS   Úwaldr,   r   r§   r   r   r   Ú	test_waldú  s    
þzTestRandomDist.test_waldc                 C   sL   t j | j¡ t jjddd�}t  ddgddgdd	gg¡}t||d
d� d S )Nrƒ  r³   rh  g@–Ý6ï?g¦Óé]Aí?g(Ã1Ú¥Rþ?gù¤U þ?g“Í2§_uå?g^}-­Qö?r¤   r¥   )r   r
   rS   Úweibullr,   r   r§   r   r   r   Útest_weibull  s    
þzTestRandomDist.test_weibullc                 C   s@   t j | j¡ tt jjddd�t  d¡ƒ ttt jjdd� d S )Nr   r6  rh  r4  )rÖ   )r   r
   rS   r   rž  r5   r   r   r    r   r   r   Útest_weibull_0
  s    zTestRandomDist.test_weibull_0c                 C   sH   t j | j¡ t jjddd�}t  ddgddgddgg¡}t||ƒ d S )	Nrƒ  r³   rh  éB   é   r"   r(   r$  )r   r
   rS   Úzipfr,   r   r§   r   r   r   Ú	test_zipf  s    
þzTestRandomDist.test_zipfN)Sr/   r0   r1   rW   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.  r0  r5  r9  r<  r=  rA  rF  rG  rI  rK  rL  rN  rR  rS  rU  rW  r[  rh   r`  rd  rf  rg  rl  ro  rp  rr  rt  ru  rv  rw  ry  rz  r{  r}  r‚  r‡  r‹  r’  r˜  r›  r�  rŸ  r   r¤  r   r   r   r   rœ   !  s    .ÿ
ÿ	(		rœ   c                   @   s  e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd	d
„ Zdd„ Zdd„ Z	dd„ Z
dd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd „ Zd!d"„ Zd#d$„ Zd%d&„ Zd'd(„ Zd)d*„ Zd+d,„ Z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"dAS )BÚTestBroadcastc                 C   s
   d| _ d S )NiÍ[r�   r    r   r   r   rW     s    zTestBroadcast.setupc                 C   s   t j | j¡ d S rè   )r   r
   rS   r    r   r   r   ÚsetSeed  s    zTestBroadcast.setSeedc                 C   sp   dg}dg}t jj}t  dddg¡}|  ¡  ||d |ƒ}t||dd� |  ¡  |||d ƒ}t||dd� 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    þzTestBroadcast.test_uniformc                 C   sš   dg}dg}dg}t jj}t  dddg¡}|  ¡  ||d |ƒ}t||dd	� tt||d |ƒ |  ¡  |||d ƒ}t||dd	� tt|||d ƒ d S )
Nr   r"   r   g-rö´@g¦Y®lõ@g-µ`¦wý?r(   r:  r¥   )r   r
   re  r,   r§  r   r   r   )r   rD  r3  Ú	bad_scalere  r¨   r”   r   r   r   rf  4  s     þzTestBroadcast.test_normalc                 C   sÄ   dg}dg}dg}dg}t jj}t  dddg¡}|  ¡  ||d |ƒ}t||d	d
� tt||d |ƒ tt||d |ƒ |  ¡  |||d ƒ}t||d	d
� tt|||d ƒ tt|||d ƒ d S )Nr"   r#   r   r$   gnÍQVfÉ?gV©‹šKB³?gv®`køÏ?r(   r:  r¥   )r   r
   r  r,   r§  r   r   r   )r   rÖ   r×   Úbad_aZbad_br  r¨   r”   r   r   r   r  G  s&    þzTestBroadcast.test_betac                 C   sZ   dg}dg}t jj}t  dddg¡}|  ¡  ||d ƒ}t||dd� tt||d ƒ d S ©	Nr"   r   gö.Ug¬Zè?gi Pr�qè?gçÅEKÌæ?r(   r:  r¥   )r   r
   r/  r,   r§  r   r   r   )r   r3  r«  r/  r¨   r”   r   r   r   r0  ]  s    þzTestBroadcast.test_exponentialc                 C   sZ   dg}dg}t jj}t  dddg¡}|  ¡  ||d ƒ}t||dd� tt||d ƒ d S r­  )r   r
   rx  r,   r§  r   r   r   )r   rL   Ú	bad_shapeZ	std_gammar¨   r”   r   r   r   ry  j  s    þz!TestBroadcast.test_standard_gammac                 C   sÄ   dg}dg}dg}dg}t jj}t  dddg¡}|  ¡  ||d |ƒ}t||d	d
� tt||d |ƒ tt||d |ƒ |  ¡  |||d ƒ}t||d	d
� tt|||d ƒ tt|||d ƒ d S )Nr"   r#   r   r$   gö.Ug¬Zø?gi Pr�qø?gçÅEKÌö?r(   r:  r¥   )r   r
   r;  r,   r§  r   r   r   )r   rL   r3  r®  r«  r;  r¨   r”   r   r   r   r<  w  s&    þzTestBroadcast.test_gammac                 C   sÄ   dg}dg}dg}dg}t jj}t  dddg¡}|  ¡  ||d |ƒ}t||d	d
� tt||d |ƒ tt||d |ƒ |  ¡  |||d ƒ}t||d	d
� tt|||d ƒ tt|||d ƒ d S )Nr"   r#   r   r$   gb—�yÊœé?g»åoðÄë?gT¹æ7Í@r(   r:  r¥   )r   r
   r8  r,   r§  r   r   r   )r   ra  rb  Ú	bad_dfnumÚ	bad_dfdenr8  r¨   r”   r   r   r   r9  �  s&    þzTestBroadcast.test_fc           
      C   sf  dg}dg}dg}dg}dg}dg}t jj}t  ddd	g¡}|  ¡  ||d ||ƒ}	t|	|d
d� tt||d ||ƒ tt||d ||ƒ tt||d ||ƒ |  ¡  |||d |ƒ}	t|	|d
d� tt|||d |ƒ tt|||d |ƒ tt|||d |ƒ |  ¡  ||||d ƒ}	t|	|d
d� tt||||d ƒ tt||||d ƒ tt||||d ƒ d S )Nr#   r(   r)   r   r   r$   gÈr±^G"@g/�*@g‹°�Ñ†š!@r:  r¥   )r   r
   rc  r,   r§  r   r   r   )
r   ra  rb  r^  r¯  r°  Úbad_noncZnonc_fr¨   r”   r   r   r   rd  £  s:    þzTestBroadcast.test_noncentral_fc                 C   s<   |   ¡  t ddg¡}tjjddddd�}t||dd� d S )Ng¬ªaŒ‚z@gn4M;î%é?rÜ   r#   r^   r:  r¥   )r§  r   r,   r
   rc  r   )r   r¨   r”   r   r   r   Útest_noncentral_f_small_dfÄ  s    z(TestBroadcast.test_noncentral_f_small_dfc                 C   sZ   dg}dg}t jj}t  dddg¡}|  ¡  ||d ƒ}t||dd� tt||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Ä   dg}dg}dg}dg}t jj}t  dddg¡}|  ¡  ||d |ƒ}t||d	d
� tt||d |ƒ tt||d |ƒ |  ¡  |||d ƒ}t||d	d
� tt|||d ƒ tt|||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±  Znonc_chir¨   r”   r   r   r   r`  ×  s&    þz'TestBroadcast.test_noncentral_chisquarec                 C   sZ   dg}dg}t jj}t  dddg¡}|  ¡  ||d ƒ}t||dd� tt||d ƒ d S )	Nr"   r   gTÄò�@g~‹Fžl@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š   dg}dg}dg}t jj}t  dddg¡}|  ¡  ||d |ƒ}t||dd	� tt||d |ƒ |  ¡  |||d ƒ}t||dd	� tt|||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     þzTestBroadcast.test_vonmisesc                 C   sZ   dg}dg}t jj}t  dddg¡}|  ¡  ||d ƒ}t||dd� tt||d ƒ d S )	Nr"   r   g&ƒ8¾?ò?g�e�âFXò?gªÕ‹Ë�Ÿð?r(   r:  r¥   )r   r
   rj  r,   r§  r   r   r   )r   rÖ   r¬  rj  r¨   r”   r   r   r   rl    s    þzTestBroadcast.test_paretoc                 C   sZ   dg}dg}t jj}t  dddg¡}|  ¡  ||d ƒ}t||dd� tt||d ƒ d S r­  )r   r
   rž  r,   r§  r   r   r   )r   rÖ   r¬  rž  r¨   r”   r   r   r   rŸ    s    þzTestBroadcast.test_weibullc                 C   sZ   dg}dg}t jj}t  dddg¡}|  ¡  ||d ƒ}t||dd� tt||d ƒ d S )	Nr"   r   r¨  r©  rª  r(   r:  r¥   )r   r
   rq  r,   r§  r   r   r   )r   rÖ   r¬  rq  r¨   r”   r   r   r   rr  '  s    þzTestBroadcast.test_powerc                 C   sš   dg}dg}dg}t jj}t  dddg¡}|  ¡  ||d |ƒ}t||dd	� tt||d |ƒ |  ¡  |||d ƒ}t||dd	� tt|||d ƒ d S )
Nr   r"   r   g:¨×BKc±?gÖ3®šk²?gî~qÉÁ“?r(   r:  r¥   )r   r
   rJ  r,   r§  r   r   r   )r   rD  r3  r«  rJ  r¨   r”   r   r   r   rK  4  s     þzTestBroadcast.test_laplacec                 C   sš   dg}dg}dg}t jj}t  dddg¡}|  ¡  ||d |ƒ}t||dd	� tt||d |ƒ |  ¡  |||d ƒ}t||dd	� tt|||d ƒ d S )
Nr   r"   r   gÈ'�£ZyÑ?g÷dSO=Ñ?gÏ3Lõ1³Õ?r(   r:  r¥   )r   r
   rE  r,   r§  r   r   r   )r   rD  r3  r«  rE  r¨   r”   r   r   r   rF  G  s     þzTestBroadcast.test_gumbelc                 C   sš   dg}dg}dg}t jj}t  dddg¡}|  ¡  ||d |ƒ}t||dd	� tt||d |ƒ |  ¡  |||d ƒ}t||dd	� tt|||d ƒ d S )
Nr   r"   r   gÂ“y±ÕÀ?gªÀJõR�Á?gþ
~Æ&‘£?r(   r:  r¥   )r   r
   rM  r,   r§  r   r   r   )r   rD  r3  r«  rM  r¨   r”   r   r   r   rN  Z  s     þzTestBroadcast.test_logisticc                 C   sš   dg}dg}dg}t jj}t  dddg¡}|  ¡  ||d |ƒ}t||dd	� tt||d |ƒ |  ¡  |||d ƒ}t||dd	� tt|||d ƒ d S )
Nr   r"   r   gÄñò�ÏH"@g¼ùÜ®ƒÍ @gvv	ý²:@r(   r:  r¥   )r   r
   rQ  r,   r§  r   r   r   )r   rO  rP  Z	bad_sigmarQ  r¨   r”   r   r   r   rR  m  s     þzTestBroadcast.test_lognormalc                 C   sZ   dg}dg}t jj}t  dddg¡}|  ¡  ||d ƒ}t||dd� tt||d ƒ d S )	Nr"   r   gÝ±lËo½ó?gé/Ó‚´Æó?gz~ßRó?r(   r:  r¥   )r   r
   rs  r,   r§  r   r   r   )r   r3  r«  rs  r¨   r”   r   r   r   rt  €  s    þzTestBroadcast.test_rayleighc                 C   sà   dg}dg}dg}dg}t jj}t  dddg¡}|  ¡  ||d |ƒ}t||d	d
� tt||d |ƒ tt||d |ƒ |  ¡  |||d ƒ}t||d	d
� tt|||d ƒ tt|||d ƒ tt|ddƒ tt|ddƒ d S )Nr4   r"   r   r$   g@mò ‰e¾?g¸�`ŸIß¿?g¸‡|2‹í?r(   r:  r¥   rB   )r   r
   rœ  r,   r§  r   r   r   )r   rO  r3  Zbad_meanr«  rœ  r¨   r”   r   r   r   r�  �  s*    þzTestBroadcast.test_waldc                 C   sl  dg}dg}dg}dg}dg}|d \}}t jj}t  dddg¡}	|  ¡  ||d ||ƒ}
t|
|	dd	� tt||d ||ƒ tt||d ||ƒ tt||d ||ƒ |  ¡  |||d |ƒ}
t|
|	dd	� tt|||d |ƒ tt|||d |ƒ tt|||d |ƒ |  ¡  ||||d ƒ}
t|
|	dd	� tt||||d ƒ tt||||d ƒ tt||||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æ   dg}dg}dg}dg}dg}t jj}t  dddg¡}|  ¡  ||d |ƒ}t||ƒ tt||d |ƒ tt||d |ƒ tt||d |ƒ |  ¡  |||d ƒ}t||ƒ tt|||d ƒ tt|||d ƒ tt|||d ƒ d S )Nr"   r4   r   ç      ø?r(   )r   r
   r6   r,   r§  r   r   r   )	r   r\  r7   Úbad_nÚ	bad_p_oneÚ	bad_p_twoZbinomr¨   r”   r   r   r   r#  É  s&    

zTestBroadcast.test_binomialc           	      C   sæ   dg}dg}dg}dg}dg}t jj}t  dddg¡}|  ¡  ||d |ƒ}t||ƒ tt||d |ƒ tt||d |ƒ tt||d |ƒ |  ¡  |||d ƒ}t||ƒ tt|||d ƒ tt|||d ƒ tt|||d ƒ d S )Nr"   r4   r   rµ  r   r(   )r   r
   rg   r,   r§  r   r   r   )	r   r\  r7   r¶  r·  r¸  Z	neg_binomr¨   r”   r   r   r   rh   à  s&    

z$TestBroadcast.test_negative_binomialc                 C   s|   t j ¡ j}dg}dg}|d g}t jj}t  dddg¡}|  ¡  ||d ƒ}t||ƒ tt	||d ƒ tt	||d ƒ d S )Nr"   r   r#   r   r(   )
r   r
   r   Z_poisson_lam_maxrn  r,   r§  r   r   r   )r   Zmax_lamrm  Zbad_lam_oneZbad_lam_tworn  r¨   r”   r   r   r   ro  ÷  s    

zTestBroadcast.test_poissonc              	   C   s�   dg}dg}t jj}t  dddg¡}|  ¡  ||d ƒ}t||ƒ tt||d ƒ t jdd��( tt|t j	ƒ tt|ddt j	gƒ W 5 Q R X d S )Nr#   r   r"   r(   rX  )Úinvalid)
r   r
   r£  r,   r§  r   r   r   Zerrstater9   )r   rÖ   r¬  r£  r¨   r”   r   r   r   r¤    s    
zTestBroadcast.test_zipfc                 C   sl   dg}dg}dg}t jj}t  dddg¡}|  ¡  ||d ƒ}t||ƒ tt||d ƒ tt||d ƒ d S )Nr4   r   rµ  r#   r(   )r   r
   r@  r,   r§  r   r   r   )r   r7   r·  r¸  Zgeomr¨   r”   r   r   r   rA    s    
zTestBroadcast.test_geometricc                 C   sœ  dg}dg}dg}dg}dg}dg}dg}t jj}t  dddg¡}	|  ¡  ||d ||ƒ}
t|
|	ƒ tt||d ||ƒ tt||d ||ƒ tt||d ||ƒ tt||d ||ƒ |  ¡  |||d |ƒ}
t|
|	ƒ tt|||d |ƒ tt|||d |ƒ tt|||d |ƒ tt|||d |ƒ |  ¡  ||||d ƒ}
t|
|	ƒ tt||||d ƒ tt||||d ƒ tt||||d ƒ tt||||d ƒ d S )Nr"   r#   r   r$   r   r)   r(   )r   r
   rH  r,   r§  r   r   r   )r   ZngoodZnbadZnsampleZ	bad_ngoodZbad_nbadZbad_nsample_oneZbad_nsample_twoZ	hypergeomr¨   r”   r   r   r   rI  !  s<    


z!TestBroadcast.test_hypergeometricc                 C   sl   dg}dg}dg}t jj}t  dddg¡}|  ¡  ||d ƒ}t||ƒ tt||d ƒ tt||d ƒ d S )Nr4   r#   r   r"   r(   )r   r
   rT  r,   r§  r   r   r   )r   r7   r·  r¸  rT  r¨   r”   r   r   r   rU  D  s    
zTestBroadcast.test_logseriesN)#r/   r0   r1   rW   r§  r‡  rf  r  r0  ry  r<  r9  rd  r²  r&  r`  r}  r˜  rl  rŸ  rr  rK  rF  rN  rR  rt  r�  r‚  r#  rh   ro  r¤  rA  rI  rU  r   r   r   r   r¦    s@   !$#r¦  c                   @   s4   e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd	d
„ ZdS )Ú
TestThreadc                 C   s   t dƒ| _d S )Nr)   )r   Úseedsr    r   r   r   rW   T  s    zTestThread.setupc                    sÌ   ddl m‰  t t| jƒf| ¡}t t| jƒf| ¡}‡ ‡fdd„t| j|ƒD ƒ}dd„ |D ƒ dd„ |D ƒ t| j|ƒD ]\}}ˆtj |¡|ƒ q|t 	¡ j
jdkr¾tjdkr¾t||ƒ n
t||ƒ d S )	Nr   )ÚThreadc                    s(   g | ] \}}ˆ ˆt j |¡|fd �‘qS ))ÚtargetÚargs)r   r
   r   )rð   r   Úo©r¼  Úfunctionr   r   rò   ^  s   ÿz-TestThread.check_function.<locals>.<listcomp>c                 S   s   g | ]}|  ¡ ‘qS r   )Ústart©rð   rF   r   r   r   rò   `  s     c                 S   s   g | ]}|  ¡ ‘qS r   )ÚjoinrÃ  r   r   r   rò   a  s     r)   Úwin32)Ú	threadingr¼  r   rÝ   r½   r»  Úzipr
   r   Zintpr'   ÚitemsizerŠ   Úplatformr   r   )r   rÁ  ÚszZout1Zout2r´  r   r¿  r   rÀ  r   Úcheck_functionW  s    
ÿzTestThread.check_functionc                 C   s   dd„ }| j |dd� d S )Nc                 S   s   | j dd�|d< d S )Né'  r^   .)re  ©rV   Úoutr   r   r   Ú
gen_randomn  s    z*TestThread.test_normal.<locals>.gen_random)rÌ  ©rÊ  ©rË  ©r   rÏ  r   r   r   rf  m  s    zTestThread.test_normalc                 C   s   dd„ }| j |dd� d S )Nc                 S   s   | j t d¡d�|d< d S )N©r<   r   r2  .)r/  r   ZonesrÍ  r   r   r   rÏ  s  s    z'TestThread.test_exp.<locals>.gen_randomrÓ  rÐ  rÑ  rÒ  r   r   r   Útest_expr  s    zTestThread.test_expc                 C   s   dd„ }| j |dd� d S )Nc                 S   s   | j ddgd dd�|d< d S )Nr   rV  r+   rÌ  r^   .)r@   rÍ  r   r   r   rÏ  x  s    z/TestThread.test_multinomial.<locals>.gen_random)rÌ  r+   rÐ  rÑ  rÒ  r   r   r   rW  w  s    zTestThread.test_multinomialN)r/   r0   r1   rW   rË  rf  rÔ  rW  r   r   r   r   rº  R  s
   rº  c                   @   s,   e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd	S )
ÚTestSingleEltArrayInputc                 C   s4   t  dg¡| _t  dg¡| _t  dg¡| _d| _d S )Nr#   r(   r)   )r"   )r   r,   ÚargOneÚargTwoÚargThreeÚtgtShaper    r   r   r   rW     s    zTestSingleEltArrayInput.setupc                 C   sœ   t jjt jjt jjt jjt jjt jjt jjt jj	t jj
t jjt jjt jjf}t jjt jjf}|D ]6}||kr~|t  dg¡ƒ}n
|| jƒ}t|j| jƒ q`d S rf   )r   r
   r/  rx  r%  r|  rj  rž  rq  rs  rn  r£  r@  rT  r,   rÖ  r   rL   rÙ  )r   ÚfuncsÚ	probfuncsrŠ  rÎ  r   r   r   Útest_one_arg_funcs…  s$         û
z*TestSingleEltArrayInput.test_one_arg_funcsc                 C   sè   t jjt jjt jjt jjt jjt jjt jjt jj	t jj
t jjt jjt jjt jjt jjf}t jjt jjf}|D ]v}||kr†t  dg¡}n| j}|| j|ƒ}t|j| jƒ || jd |ƒ}t|j| jƒ || j|d ƒ}t|j| jƒ qld S )Nr4   r   )r   r
   r†  re  r  r;  r8  r_  r—  rJ  rE  rM  rQ  rœ  r6   rg   r,   r×  rÖ  r   rL   rÙ  )r   rÚ  rÛ  rŠ  r×  rÎ  r   r   r   Útest_two_arg_funcs˜  s2          úz*TestSingleEltArrayInput.test_two_arg_funcsc                 C   sŒ   t jjt jjt jjg}|D ]l}|| j| j| jƒ}t|j	| j
ƒ || jd | j| jƒ}t|j	| j
ƒ || j| jd | jƒ}t|j	| j
ƒ qd S )Nr   )r   r
   rc  r�  rH  rÖ  r×  rØ  r   rL   rÙ  )r   rÚ  rŠ  rÎ  r   r   r   Útest_three_arg_funcsÅ  s    ÿz,TestSingleEltArrayInput.test_three_arg_funcsN)r/   r0   r1   rW   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   r2   r;   rQ   ri   rœ   r¦  rº  rÕ  r   r   r   r   Ú<module>   s0   ('"0      |    >,