o
    V=ºi5 ã                   @   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                   @   ó4   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   údC:\Users\psant\Downloads\PAF backend\school\venv\Lib\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   sb   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g d¢ƒ t ttjjg d¢ƒ d S )Nr   r   ì        )é   é   r#   )r$   éþÿÿÿr#   r   r!   r   r   r   Útest_invalid_array%   s
   zTestSeed.test_invalid_arrayc                 C   sR   t ttjjtjg tjd�ƒ t ttjjg d¢gƒ t ttjjg d¢g d¢gƒ d S )N©Údtype©r$   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'   r1   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   r8   Ú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
   r9   r   Únanr!   r   r   r   Útest_p_is_nan@   s   zTestBinomial.test_p_is_nanN)r2   r3   r4   r;   r=   r   r   r   r   r5   6   s    	r5   c                   @   r   )ÚTestMultinomialc                 C   s   t  dddg¡ d S )Néd   çš™™™™™É?çš™™™™™é?©r
   Úmultinomialr!   r   r   r   Ú
test_basicF   ó   zTestMultinomial.test_basicc                 C   s   t  dg d¢¡ d S )Nr?   )r@   rA   ç        rF   rF   rB   r!   r   r   r   Útest_zero_probabilityI   rE   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 )Nr7   r$   ©r$   r%   r%   ©r%   r%   r%   ©r%   r%   )
r   r   r
   rC   Úuint32Úshaper/   r   r   Úfloat©r   r:   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
   rC   r/   r!   r   r   r   Útest_multidimensional_pvals`   s   &(z+TestMultinomial.test_multidimensional_pvalsN)r2   r3   r4   rD   rG   rK   rS   rT   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   Úsetup_methodh   s   zTestSetState.setup_methodc                 C   s<   | j  d¡}| j  | j¡ | j  d¡}tt ||k¡ƒ d S )Né   )rX   ZtomaxintÚ	set_staterZ   r   r   rI   ©r   ÚoldÚnewr   r   r   rD   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)rX   Ústandard_normalr]   rZ   r   r   rI   r^   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 ra   )rX   rd   rY   r]   r   r   rI   )r   rZ   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&   r\   rb   )rZ   rX   rd   r]   r   r   rI   )r   Ú	old_stateÚx1Zx2Zx3r   r   r   Útest_backwards_compatibility…   s   z)TestSetState.test_backwards_compatibilityc                 C   s   | j  dd¡ d S ©Nr7   )rX   Únegative_binomialr!   r   r   r   Útest_negative_binomial‘   s   z#TestSetState.test_negative_binomialN)	r2   r3   r4   r[   rD   re   rf   ri   rl   r   r   r   r   rU   g   s    rU   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   ÚrfuncrQ   r!   r   r   r   Útest_unsupported_typeŸ   s   z!TestRandint.test_unsupported_typec                 C   s    | j D ]J}|tju rdnt |¡j}|tju rd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    ro   )r   ÚdtÚlbndÚubndr   r   r   Útest_bounds_checking¢   s   
úz TestRandint.test_bounds_checkingc                 C   s´   | j D ]T}|tju rdnt |¡j}|tju rd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   ©rc   r)   )rq   r   rr   rs   rt   ru   r   ro   )r   rv   rw   rx   Útgtr   r   r   Útest_rng_zero_and_extremes«   s   
õz&TestRandint.test_rng_zero_and_extremesc                 C   s„   | j D ]<}|tju rdnt |¡j}|tju rdnt |¡jd }z
| j|||d� W q ty? } ztdt	|ƒ ƒ‚d }~ww d S )Nr   r%   r$   r(   zMNo error should have been raised, but one was with the following message:

%s)
rq   r   rr   rs   rt   ru   ro   Ú	ExceptionÚAssertionErrorÚstr)r   rv   rw   rx   Úer   r   r   Útest_full_range¹   s   
þ€ÿúzTestRandint.test_full_rangec                 C   s’   t j ¡  | jdd … D ] }dD ]}| 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,   é   r\   r%   i   rz   r   )	r   r
   rW   rq   ro   r   ru   rt   rr   )r   rv   rx   Ú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 ]9}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Úint32r0   Úint8Úuint16rO   Úuint64Úuint8r$   iÒ  Úlittler.   r   rz   r%   )Úhashlibrq   r   r
   rW   ÚsysÚ	byteorderro   ZbyteswapÚsha256Úviewrˆ   Ú	hexdigestr   r)   Únamer…   )r   r�   r{   rv   Ú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 rn   )r   r0   rs   ru   rŠ   r
   r   r   )r   rv   r{   rw   rx   Úactualr   r   r   Útest_int64_uint64_corner_case÷   s   z)TestRandint.test_int64_uint64_corner_casec                 C   sà   | j D ]/}|tju rdnt |¡j}|tju rdnt |¡jd }| j|||d�}t|jt |¡ƒ qt	t
tjjfD ]3}|t	u rBdnt |¡j}|t	u rNdnt |¡jd }| j|||d�}tt|dƒ ƒ tt|ƒ|ƒ q:d S )Nr   r%   r$   r(   r)   )rq   r   rr   rs   rt   ru   ro   r   r)   r…   r6   ÚcompatÚlongr   ÚhasattrÚtype)r   rv   rw   rx   Úsampler   r   r   Útest_respect_dtype_singleton  s   
ùz(TestRandint.test_respect_dtype_singletonN)r2   r3   r4   r   r
   r   ro   rr   rˆ   r‹   r†   r‰   r‡   rO   r0   rŠ   rq   rp   ry   r|   r�   r„   r–   r˜   rž   r   r   r   r   rm   —   s    ÿ	!rm   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   ó
   d| _ d S rV   ©rW   r!   r   r   r   r[   %  ó   
zTestRandomDist.setup_methodc                 C   óJ   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
   rW   Zrandr/   r   ©r   r—   Údesiredr   r   r   Ú	test_rand(  ó   
þzTestRandomDist.test_randc                 C   r£   )Nr+   r%   çÿ!ã<Oqõ?ç4ð�u,Íû?çÁ³½3Ûû÷?çL«I/DÍ¿çà@ËÀŽ? @çÿHSÓ\@rª   r«   )r   r
   rW   Zrandnr/   r   r­   r   r   r   Ú
test_randn0  r°   zTestRandomDist.test_randnc                 C   óJ   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%   rb   é   r+   éÌÿÿÿé)   éÐÿÿÿé¾ÿÿÿ)r   r
   rW   r   r/   r   r­   r   r   r   Útest_randint8  ó   
þzTestRandomDist.test_randintc                 C   sŠ   t j | j¡ tƒ �}| t¡}t jjdddd�}tt|ƒdkƒ W d   ƒ n1 s+w   Y  t  	ddgdd	gd
dgg¡}t
||ƒ d S )Nr¹   rº   r»   rb   r$   r¼   r+   r½   r¾   r¿   rÀ   )r   r
   rW   r	   ÚrecordÚDeprecationWarningÚrandom_integersr   Úlenr/   r   ©r   ÚsupÚwr—   r®   r   r   r   Útest_random_integers@  s   
ý
þz#TestRandomDist.test_random_integersc                 C   sx   t ƒ �$}| t¡}tj t d¡jt d¡j¡}tt	|ƒdkƒ W d   ƒ n1 s*w   Y  t d¡j}t
||ƒ d S )NÚlr$   )r	   rÃ   rÄ   r   r
   rÅ   rs   ru   r   rÆ   r   rÇ   r   r   r   Útest_random_integers_max_intK  s   

ÿüz+TestRandomDist.test_random_integers_max_intc                 C   st   t  ¡ �, t  dt¡ tttjjt d¡j	ƒ tttjjt d¡j	t d¡j	ƒ W d   ƒ d S 1 s3w   Y  d S )NÚerrorrË   )
ÚwarningsÚcatch_warningsÚsimplefilterrÄ   r   r   r
   rÅ   rs   ru   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
   rW   r/   r   r­   r   r   r   Útest_randomh  s   
þzTestRandomDist.test_randomc                 C   s8   t j | j¡ t j dd¡}t  g d¢¡}t||ƒ d S )Nr,   )r%   r+   r%   r+   ©r   r
   rW   Úchoicer/   r   r­   r   r   r   Útest_choice_uniform_replacep  s   z*TestRandomDist.test_choice_uniform_replacec                 C   s@   t j | j¡ t jjddg d¢d�}t  g d¢¡}t||ƒ d S )Nr,   )çš™™™™™Ù?rÖ   çš™™™™™¹?r×   ©r:   )r$   r$   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  g d¢¡}t||ƒ d S )Nr,   r+   F©Úreplace)r   r$   r+   rÓ   r­   r   r   r   Útest_choice_uniform_noreplace|  ó   z,TestRandomDist.test_choice_uniform_noreplacec                 C   sB   t j | j¡ t jjdddg d¢d�}t  g d¢¡}t||ƒ d S )Nr,   r+   F)r×   g333333Ó?r7   r×   ©rÛ   r:   )r%   r+   r$   rÓ   r­   r   r   r   Ú test_choice_nonuniform_noreplace‚  s   ÿz/TestRandomDist.test_choice_nonuniform_noreplacec                 C   s<   t j | j¡ t j g d¢d¡}t  g d¢¡}t||ƒ d S )N)ÚaÚbÚcÚdr,   )râ   rã   râ   rã   rÓ   r­   r   r   r   Útest_choice_noninteger‰  rÝ   z%TestRandomDist.test_choice_nonintegerc                 C   s4  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|g d¢dddgddggd	� tt|ddgdg d
¢d	� tt|ddgdddgd	� tt|ddgdddgd	� tt|g d¢ddd� tt|g d¢ddd� tt|g d¢ddd� tt|g d¢ddd� tt|g d¢ddg d¢d� d S )Nr   r+   g      @r$   r%   r,   )r$   r%   r+   r,   g      Ð?rØ   )rÖ   rÖ   r@   gš™™™™™ñ?gš™™™™™¹¿rÖ   r*   FrÚ   r&   )r   )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   s4  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 u ƒ t ddg¡}tjdtd	�}||d
< t tjj|dd�|u ƒ 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� 
¡ |u ƒ d}g d¢}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.   )r+   r   r,   rb   iöÿÿÿ)r   r   rà   rá   )r   r   Zisscalarr
   rÔ   r/   ÚemptyÚobjectÚtupleÚndimÚitemr   rP   r   r   r   r    )r   r:   rà   Úarrr   r   r   r   Útest_choice_return_shape¢  sL     " ÿz'TestRandomDist.test_choice_return_shapec                 C   s.   t  g d¢¡}g d¢}ttt jj||d� d S )N)é*   r$   r%   )NNNrØ   )r   r/   r   r    r
   rÔ   )r   rà   r:   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
   rW   Ú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 ]"}t j | j¡ |g d¢ƒ}t j |¡ |}|g d¢ƒ}t||ƒ q#d S )Nc                 S   s
   t  g ¡S ©N)r   r/   ©rJ   r   r   r   Ú<lambda>Þ  s   
 z-TestRandomDist.test_shuffle.<locals>.<lambda>c                 S   s   | S rò   r   ró   r   r   r   rô   ß  s    c                 S   ó   t  | ¡ t j¡S rò   )r   ÚasarrayÚastyperˆ   ró   r   r   r   rô   à  ó    c                 S   rõ   rò   )r   rö   r÷   Úfloat32ró   r   r   r   rô   á  rø   c                 S   rõ   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ô   ã  ó    c                 S   s   dd„ | D ƒS )Nc                 S   ó   g | ]}||f‘qS r   r   ©Ú.0Úir   r   r   Ú
<listcomp>ä  ó    úATestRandomDist.test_shuffle.<locals>.<lambda>.<locals>.<listcomp>r   ró   r   r   r   rô   ä  s    c                 S   s   t  dd„ | D ƒ¡S )Nc                 S   s   g | ]}||g‘qS r   r   rü   r   r   r   rÿ   å  r   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   rû   r   r   rü   r   r   r   rÿ   è  r   r  rà   rá   )r   rö   r6   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   rû   r   r   rü   r   r   r   rÿ   ì  r   r  rà   rá   )r   rö   rè   r‡   ró   r   r   r   rô   ì  s    ÿ)
r$   r%   r+   r,   r-   r.   é   r‚   é	   r   )
r   r$   r  r.   r%   r,   r-   r‚   r  r+   )r   r
   rW   Ú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 ]4}t j |¡ t	t
|j|j  ƒt
|j|j  ƒƒ t j |¡ t	t
|j|j  ƒt
|j|j  ƒƒ q+d 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   sb   ddddddddœ}t jtd	d
��}| |¡ W d   ƒ n1 s!w   Y  d|d jv s/J ‚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  Úfilename)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 ƒ‰ |r2| |¡}t‡ fd	d
„|D ƒƒs?J ‚|rgt|t jj	ƒsgt
jtdd�� | |¡ W d   ƒ d S 1 s`w   Y  d S | |¡ t‡ fdd
„|D ƒƒsyJ ‚d S )Nc                   @   s   e Zd ZdS )z>TestRandomDist.test_shuffle_no_object_unpacking.<locals>.MyArrN)r2   r3   r4   r   r   r   r   ÚMyArr  s    r  r+   r   r  r(   c                 S   s   h | ]}t |ƒ’qS r   ©Úidrü   r   r   r   Ú	<setcomp>  r   zBTestRandomDist.test_shuffle_no_object_unpacking.<locals>.<setcomp>c                 3   ó   � | ]	}t |ƒˆ v V  qd S rò   r  rü   ©Zitem_idsr   r   Ú	<genexpr>  ó   € zBTestRandomDist.test_shuffle_no_object_unpacking.<locals>.<genexpr>z#Shuffling a one dimensional array.*r  c                 3   r  rò   r  rü   r  r   r   r   &  r!  )r   Úndarrayr/   Zfloat64rè   r‘   rI   Ú
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  |¡g d¢ƒ t j | j¡}| |¡ tt  |¡g d¢ƒ t j 	| j¡}| |¡ tt  |¡g d¢ƒ d S )Nr-   )r   r$   r,   r+   r%   )r   r$   r%   r+   r,   )r,   r$   r   r+   r%   )
r   r
   rW   r   r  r  r   rö   r   Údefault_rng)r   rà   Úrngr   r   r   Útest_shuffle_memoryview(  s   

z&TestRandomDist.test_shuffle_memoryviewc                 C   sR   t  d¡}d|j_tjtdd�� t j |¡ W d   ƒ d S 1 s"w   Y  d S )Nr+   Fz	read-onlyr  )	r   r8   ÚflagsZ	writeabler  Zraisesr    r
   r  )r   rà   r   r   r   Útest_shuffle_not_writeable7  s
   
"ÿz)TestRandomDist.test_shuffle_not_writeablec                 C   óN   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»   rb   gAàÁKÄ�?g‘ñØhA?g õÒ‡o¿>gÎ¬XfÈ+q?gSâë1Ã$?g'8zXŒ ?rª   r«   )r   r
   rW   Úbetar/   r   r­   r   r   r   Ú	test_beta=  s   þÿzTestRandomDist.test_betac                 C   r¸   )Nr?   gÉv¾Ÿ/Ý?r»   rb   é%   é+   rî   é0   é.   é-   )r   r
   rW   r9   r/   r   r­   r   r   r   Útest_binomialF  rÂ   zTestRandomDist.test_binomialc                 C   óL   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»   rb   g+Qð]uðO@gëòìÇ+Q@gY7`´ZqP@gÈe‡(fŒG@gþv¯t€R@gEIÈ‹R@é   r«   )r   r
   rW   Ú	chisquarer/   r   r­   r   r   r   Útest_chisquareN  ó   
þ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»   rb   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
   rW   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$   rL   r%   rM   rN   )
r   r/   r   r
   r=  rO   rP   r   r   rQ   rR   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   r5  )Ngï8EGrùñ?r»   rb   gN÷·Uñ?gbI0æð?gùVU\õº@gˆ:€ì3ù@g©ºUýå?g�Ÿ"qoû?rª   r«   )r   r
   rW   Úexponentialr/   r   r­   r   r   r   Útest_exponentialy  r9  zTestRandomDist.test_exponentialc                 C   ó*   t tjjdd�dƒ tttjjdd� d S ©Nr   ©Úscaleç       €)r   r   r
   rB  r   r    r!   r   r   r   Útest_exponential_0�  ó   z!TestRandomDist.test_exponential_0c                 C   r,  )Né   éM   r»   rb   g7¶„ó?gµ`j‹�ü?g6¶"+÷?gÜ³2>•‰ó?g‘cÍ>+Yð?g¡_}áS‚õ?rª   r«   )r   r
   rW   Úfr/   r   r­   r   r   r   Útest_f…  ó   
þzTestRandomDist.test_fc                 C   r,  )Nr-   r+   r»   rb   gÔ�GMçš8@g §â”ÈŒ<@g˜	´":@gÂvÀñÇ#)@gZQø·?@gr‚‚[•¦@@é   r«   )r   r
   rW   Úgammar/   r   r­   r   r   r   Ú
test_gamma�  rO  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   )rP   rG  rH  )r   r   r
   rQ  r   r    r!   r   r   r   Útest_gamma_0•  s   zTestRandomDist.test_gamma_0c                 C   óH   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»   rb   r‚   r  é   r-   rK  )r   r
   rW   Ú	geometricr/   r   r­   r   r   r   Útest_geometric™  ó   
þzTestRandomDist.test_geometricc                 C   r,  )NrU  ç       @r»   ©ÚlocrG  rc   gåûØ•ßÉ?g¨s0í Ö?gÊð#0÷¿gï(ôx”÷¿g-¶KÇD´ñ?géÅ¯p`@æ¿rª   r«   )r   r
   rW   Úgumbelr/   r   r­   r   r   r   Útest_gumbel¡  rO  zTestRandomDist.test_gumbelc                 C   rD  rE  )r   r   r
   r]  r   r    r!   r   r   r   Útest_gumbel_0©  rJ  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  g d
¢¡}t||ƒ t jjdddd	d�}t  g d¢¡}t||ƒ t jjdddd	d�}t  g d¢¡}t||ƒ t jjdddd	d�}t  g d¢¡}t||ƒ d S )Nr   r-   rP  r»   rb   r  r   r+   r,   )r+   r+   r+   r+   rª   rK  )rK  rK  rK  rK  )r   r   r   r   )r   r
   rW   Úhypergeometricr/   r   r­   r   r   r   Útest_hypergeometric­  s&   
þ



z"TestRandomDist.test_hypergeometricc                 C   r,  )NrU  rZ  r»   r[  gð¥ bÙOå?g¹Â¸ëÉçà?gpJmµú	@g{KÏ*Ët	@g•:^w-š«¿gˆY~oõûû?rª   r«   )r   r
   rW   Úlaplacer/   r   r­   r   r   r   Útest_laplaceÇ  rO  zTestRandomDist.test_laplacec                 C   rD  rE  )r   r   r
   rb  r   r    r!   r   r   r   Útest_laplace_0Ï  rJ  zTestRandomDist.test_laplace_0c                 C   r,  )NrU  rZ  r»   r[  g#€�K-zñ?g€îJš¬ë?g�92ÄÜ@gØ±ã÷[@gê#+^ÑÀË¿g¢¥-óâ@rª   r«   )r   r
   rW   Úlogisticr/   r   r­   r   r   r   Útest_logisticÓ  rO  zTestRandomDist.test_logisticc                 C   r,  )NrU  rZ  r»   )ÚmeanÚsigmarc   g›ŽòÚÉ�0@goÎ,+4FB@gù�ƒ)Ê­6@g†§5éêæ?gnþxN—nP@g…©)‡úµU@r6  r«   )r   r
   rW   Ú	lognormalr/   r   r­   r   r   r   Útest_lognormalÛ  rO  zTestRandomDist.test_lognormalc                 C   s*   t tjjdd�dƒ tttjjdd� d S )Nr   )rh  r$   rH  )r   r   r
   ri  r   r    r!   r   r   r   Útest_lognormal_0ã  rJ  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»   )r:   rc   r%   r.   rV  r+   )r   r
   rW   Ú	logseriesr/   r   r­   r   r   r   Útest_logseriesç  rY  zTestRandomDist.test_logseriesc                 C   sh   t j | j¡ t jjddgd dd�}t  g d¢g d¢gg d¢g d	¢gg d
¢g d¢gg¡}t||ƒ d S )Nr	  çUUUUUUÅ?r.   r»   rb   )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
   rW   rC   r/   r   r­   r   r   r   Útest_multinomialï  s   
ÿÿÿüzTestRandomDist.test_multinomialc                 C   sf  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ƒ �}t j ||¡ | t¡}t|ƒdks¡J ‚W d   ƒ d S 1 s¬w   Y  d S )N)rU  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
   rW   Zmultivariate_normalr/   r   r   ÚRuntimeWarningr   r   r    rù   r	   rÃ   rÆ   )r   rg  Zcovrc   r—   r®   rÈ   rÉ   r   r   r   Útest_multivariate_normalú  sD   
ÿÿÿüÿÿ
"ýz'TestRandomDist.test_multivariate_normalc                 C   r¸   )Nr?   g|ò°Pkš¿?r»   )Únr:   rc   iP  iI  i|  ic  i  i‡  )r   r
   rW   rk   r/   r   r­   r   r   r   rl   "  rÂ   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Únoncrc   gôûÖGë7@g!© ÈÜ´*@gž1}“z9?@g¹+é©¸™0@g26E_r#@g¦�I�!ó1@rP  r«   r7   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
   rW   Únoncentral_chisquarer/   r   r­   r   r   r   Útest_noncentral_chisquare*  s(   
þ
þ
þz(TestRandomDist.test_noncentral_chisquarec                 C   óP   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Údfdenrv  rc   gŒ^ýíå~ö?gƒ/+c¢äÕ?g"¡®.ž@gü†Œ÷Ž´@g×7í§ŸþÛ?g»ŒE·Öò?rP  r«   )r   r
   rW   Únoncentral_fr/   r   r­   r   r   r   Útest_noncentral_f?  ó   ÿ
þz TestRandomDist.test_noncentral_fc                 C   r,  )NrU  rZ  r»   r[  gíœ&&n@gO»G_Ê@gÜ~w²ø@gtÁÜûw]Õ¿gm&¨5ú½@gŒñ$È>Û@rª   r«   )r   r
   rW   Únormalr/   r   r­   r   r   r   Útest_normalH  rO  zTestRandomDist.test_normalc                 C   rD  rE  )r   r   r
   r  r   r    r!   r   r   r   Útest_normal_0P  rJ  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 )NrU  r»   ©rà   rc   gÅWò˜I£@gùÔ¨y–@gy£Ÿ]Ó0‰Ag1~ÖÒ\�AgåËíãša@gÑY„
²7Aé   )Znulp)r   r
   rW   Ú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 )NrU  r»   )Úlamrc   r   r$   )r   r
   rW   Úpoissonr/   r   r­   r   r   r   Útest_poissonc  rY  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   rs   ru   r   r    r
   rˆ  )r   ZlambigZlamnegr   r   r   Útest_poisson_exceptionsk  s   z&TestRandomDist.test_poisson_exceptionsc                 C   r5  )NrU  r»   r‚  gõ;•'û”?gÜ´èwÞ*�?g×†5›Ø?gF¡GÙ?g ÆËD]?gýÞ$`°ÉÀ?rª   r«   )r   r
   rW   Úpowerr/   r   r­   r   r   r   Ú
test_powers  r9  zTestRandomDist.test_powerc                 C   r5  )Nr   r»   )rG  rc   g"íu¨ÈÆ+@gîN$MBÄ*@g7v8Bô4@gêÿª55@geãˆ&@g~ŽÓ£ÌZ1@rP  r«   )r   r
   rW   Úrayleighr/   r   r­   r   r   r   Útest_rayleigh{  r9  zTestRandomDist.test_rayleighc                 C   rD  rE  )r   r   r
   r�  r   r    r!   r   r   r   Útest_rayleigh_0ƒ  rJ  zTestRandomDist.test_rayleigh_0c                 C   óJ   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»   rb   gMo³DL®è?g6Ï([9ÀgðºrF=òí?gÉM'	-™ ÀgóV¼ëûÀgVA:ìQyÇ?rª   r«   )r   r
   rW   Zstandard_cauchyr/   r   r­   r   r   r   Útest_standard_cauchy‡  r°   z#TestRandomDist.test_standard_cauchyc                 C   r�  )Nr»   rb   g°ö–Ü�Üî?gc!z¨ì?g&N˜¥"�@g¬NIàŠÇ@gFÇï>ú’ã?g¡~‚MEø?rª   r«   )r   r
   rW   Zstandard_exponentialr/   r   r­   r   r   r   Útest_standard_exponential�  r°   z(TestRandomDist.test_standard_exponentialc                 C   r5  )Nr+   r»   )rP   rc   g)§ž@g¾§Ä¾¤„@gd]‰,qÂ@gP‡kÌ{@g9B#Œ1@gÔÚv÷‡ @rP  r«   )r   r
   rW   Ústandard_gammar/   r   r­   r   r   r   Útest_standard_gamma—  r9  z"TestRandomDist.test_standard_gammac                 C   rD  )Nr   )rP   rH  )r   r   r
   r“  r   r    r!   r   r   r   Útest_standard_gamma_0Ÿ  rJ  z$TestRandomDist.test_standard_gamma_0c                 C   r�  )Nr»   rb   r±   r²   r³   r´   rµ   r¶   rª   r«   )r   r
   rW   rd   r/   r   r­   r   r   r   Útest_standard_normal£  r°   z#TestRandomDist.test_standard_normalc                 C   r5  )Nr   r»   )ru  rc   g\VÔHÂï?gßhUÍ%›¶¿gÿH$ðMÏõ?güøüL›³á¿gV«qdz¥Ç¿gã–€Õý“ã?rª   r«   )r   r
   rW   Ú
standard_tr/   r   r­   r   r   r   Útest_standard_t«  r9  zTestRandomDist.test_standard_tc                 C   ry  )Ng{®Gáz@gö(\�Âu$@g×£p=
W4@r»   )ÚleftÚmodeÚrightrc   g:§}ŒÂ\)@ghØ±TjÓ(@g‡Ç¨L‰30@g6Úœ™ÅA0@g6¤õ—sh&@gœí<Šáþ,@rP  r«   )r   r
   rW   Ú
triangularr/   r   r­   r   r   r   Útest_triangular³  r~  zTestRandomDist.test_triangularc                 C   r,  )Nç®Gáz®ó?g®Gáz%@r»   )ÚlowÚhighrc   g>#¥HÃö@g£¨uºó@g•ï‡§Ý#@g„}d]#@gø†‰N¶õ@g­w)*ó @rª   r«   )r   r
   rW   Úuniformr/   r   r­   r   r   r   Útest_uniform¼  rO  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 )NrQ   r   r$   g  Ø…W4vC)rŸ  r   )
r   Zfinfort   ru   r
   r¡  r   ÚOverflowErrorÚinfÚ	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   ó   t ‚rò   ©r   r!   r   r   r   Ú	__float__Ü  ó   zQTestRandomDist.test_scalar_exception_propagation.<locals>.ThrowingFloat.__float__N)r2   r3   r4   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   r¨  rò   r©  r!   r   r   r   Ú__int__ä  r«  zQTestRandomDist.test_scalar_exception_propagation.<locals>.ThrowingInteger.__int__N)r2   r3   r4   r­  Ú	__index__r   r   r   r   ÚThrowingIntegerã  s    r¯  r$   )	r   r"  r/   r‘   r   r   r
   r¡  r`  )r   r¬  Zthrowing_floatr¯  Zthrowing_intr   r   r   Ú!test_scalar_exception_propagationÔ  s   ÿz0TestRandomDist.test_scalar_exception_propagationc                 C   r,  )Nrž  ç¤p=
×£ø?r»   ©ÚmuÚkapparc   gL×üZI@g¦¢'N"@gÀ·…þkrØ?g”k=mœ@g�¹Ú‰ó?gÜ^Ì=�\ý?rª   r«   )r   r
   rW   Úvonmisesr/   r   r­   r   r   r   Útest_vonmisesì  rO  zTestRandomDist.test_vonmisesc                 C   s:   t j | j¡ t jjdddd�}t j t  |¡ ¡ ¡ d S )NrF   gg5_PŸG>i@B r²  )r   r
   rW   rµ  r…  r   ÚisfiniterI   )r   Úrr   r   r   Útest_vonmises_smallô  s   z"TestRandomDist.test_vonmises_smallc                 C   r,  )Nrž  r±  r»   )rg  rG  rc   gÉZ¥ƒ¢@g-ïŸg†@g¦8
ÄÖmÖ?gÃÎVP"ø?g`Vô áÎ?gx©^½&3Ì?rP  r«   )r   r
   rW   Úwaldr/   r   r­   r   r   r   Ú	test_waldú  rO  zTestRandomDist.test_waldc                 C   r5  )Nrž  r»   r‚  g@–Ý6ï?g¦Óé]Aí?g(Ã1Ú¥Rþ?gù¤U þ?g“Í2§_uå?g^}-­Qö?rª   r«   )r   r
   rW   Úweibullr/   r   r­   r   r   r   Útest_weibull  r9  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   rK  r‚  rH  )rà   )r   r
   rW   r   r¼  r8   r   r    r!   r   r   r   Útest_weibull_0
  s   zTestRandomDist.test_weibull_0c                 C   rT  )	Nrž  r»   r‚  éB   é   r$   r+   r6  )r   r
   rW   Úzipfr/   r   r­   r   r   r   Ú	test_zipf  rY  zTestRandomDist.test_zipfN)Sr2   r3   r4   r[   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.  r4  r8  r?  r@  rA  rC  rI  rN  rR  rS  rX  r^  r_  ra  rc  rd  rf  rj  rk  rm  ro  rs  rl   rx  r}  r€  r�  r†  r‰  rŠ  rŒ  rŽ  r�  r‘  r’  r”  r•  r–  r˜  r�  r¢  r§  r°  r¶  r¹  r»  r½  r¾  rÂ  r   r   r   r   rŸ   !  s¢    .ÿ
ÿ	(		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   r    )NiÍ[r¡   r!   r   r   r   r[     r¢   zTestBroadcast.setup_methodc                 C   s   t j | j¡ d S rò   )r   r
   rW   r!   r   r   r   ÚsetSeed  s   zTestBroadcast.setSeedc                 C   sn   dg}dg}t jj}t  g d¢¡}|  ¡  ||d |ƒ}t||dd� |  ¡  |||d ƒ}t||dd� d S )Nr   r$   ©gºlö×÷á?g_x¤¥á?g=å*BNà?r+   rP  r«   )r   r
   r¡  r/   rÅ  r   )r   rŸ  r   r¡  r®   r—   r   r   r   r¢  $  s   zTestBroadcast.test_uniformc                 C   ó˜   dg}dg}dg}t jj}t  g d¢¡}|  ¡  ||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+   rP  r«   )r   r
   r  r/   rÅ  r   r   r    )r   r\  rG  Ú	bad_scaler  r®   r—   r   r   r   r€  4  ó   zTestBroadcast.test_normalc                 C   óÂ   dg}dg}dg}dg}t jj}t  g d¢¡}|  ¡  ||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+   rP  r«   )r   r
   r-  r/   rÅ  r   r   r    )r   rà   rá   Úbad_aZbad_br-  r®   r—   r   r   r   r.  G  ó    zTestBroadcast.test_betac                 C   óX   dg}dg}t jj}t  g d¢¡}|  ¡  ||d ƒ}t||dd� tt||d ƒ d S ©Nr$   r   )gö.Ug¬Zè?gi Pr�qè?gçÅEKÌæ?r+   rP  r«   )r   r
   rB  r/   rÅ  r   r   r    )r   rG  rÈ  rB  r®   r—   r   r   r   rC  ]  ó   zTestBroadcast.test_exponentialc                 C   rÍ  rÎ  )r   r
   r“  r/   rÅ  r   r   r    )r   rP   Ú	bad_shapeZ	std_gammar®   r—   r   r   r   r”  j  rÏ  z!TestBroadcast.test_standard_gammac                 C   rÊ  )	Nr$   r%   r   r&   )gö.Ug¬Zø?gi Pr�qø?gçÅEKÌö?r+   rP  r«   )r   r
   rQ  r/   rÅ  r   r   r    )r   rP   rG  rÐ  rÈ  rQ  r®   r—   r   r   r   rR  w  rÌ  zTestBroadcast.test_gammac                 C   rÊ  )	Nr$   r%   r   r&   )gb—�yÊœé?g»åoðÄë?gT¹æ7Í@r+   rP  r«   )r   r
   rM  r/   rÅ  r   r   r    )r   rz  r{  Ú	bad_dfnumÚ	bad_dfdenrM  r®   r—   r   r   r   rN  �  rÌ  zTestBroadcast.test_fc           
      C   sd  dg}dg}dg}dg}dg}dg}t jj}t  g 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 ƒ}	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‹°�Ñ†š!@rP  r«   )r   r
   r|  r/   rÅ  r   r   r    )
r   rz  r{  rv  rÑ  rÒ  Úbad_noncZnonc_fr®   r—   r   r   r   r}  £  s4   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%   rb   rP  r«   )rÅ  r   r/   r
   r|  r   )r   r®   r—   r   r   r   Útest_noncentral_f_small_dfÄ  s   z(TestBroadcast.test_noncentral_f_small_dfc                 C   rÍ  )Nr$   r   )g~¼}ÐN?â?g‘$FŽŸà?g'rS�èÀ?r+   rP  r«   )r   r
   r7  r/   rÅ  r   r   r    )r   ru  Úbad_dfr7  r®   r—   r   r   r   r8  Ê  rÏ  zTestBroadcast.test_chisquarec                 C   rÊ  )	Nr$   r%   r   r&   )g©ç wÌ "@g2øêWR@g�ˆâRY@r+   rP  r«   )r   r
   rw  r/   rÅ  r   r   r    )r   ru  rv  rÕ  rÓ  Znonc_chir®   r—   r   r   r   rx  ×  rÌ  z'TestBroadcast.test_noncentral_chisquarec                 C   rÍ  )Nr$   r   )gTÄò�@g~‹Fžl@gÆštø�pð?r+   rP  r«   )r   r
   r—  r/   rÅ  r   r   r    )r   ru  rÕ  Útr®   r—   r   r   r   r˜  í  rÏ  zTestBroadcast.test_standard_tc                 C   rÇ  )Nr%   r$   r   )g´®W!è@gü�OBº¦ÀgÈ}Ç?àý¿r+   rP  r«   )r   r
   rµ  r/   rÅ  r   r   r    )r   r³  r´  Z	bad_kapparµ  r®   r—   r   r   r   r¶  ú  rÉ  zTestBroadcast.test_vonmisesc                 C   rÍ  )Nr$   r   )g&ƒ8¾?ò?g�e�âFXò?gªÕ‹Ë�Ÿð?r+   rP  r«   )r   r
   r„  r/   rÅ  r   r   r    )r   rà   rË  r„  r®   r—   r   r   r   r†    rÏ  zTestBroadcast.test_paretoc                 C   rÍ  rÎ  )r   r
   r¼  r/   rÅ  r   r   r    )r   rà   rË  r¼  r®   r—   r   r   r   r½    rÏ  zTestBroadcast.test_weibullc                 C   rÍ  )Nr$   r   rÆ  r+   rP  r«   )r   r
   r‹  r/   rÅ  r   r   r    )r   rà   rË  r‹  r®   r—   r   r   r   rŒ  '  rÏ  zTestBroadcast.test_powerc                 C   rÇ  )Nr   r$   r   )g:¨×BKc±?gÖ3®šk²?gî~qÉÁ“?r+   rP  r«   )r   r
   rb  r/   rÅ  r   r   r    )r   r\  rG  rÈ  rb  r®   r—   r   r   r   rc  4  rÉ  zTestBroadcast.test_laplacec                 C   rÇ  )Nr   r$   r   )gÈ'�£ZyÑ?g÷dSO=Ñ?gÏ3Lõ1³Õ?r+   rP  r«   )r   r
   r]  r/   rÅ  r   r   r    )r   r\  rG  rÈ  r]  r®   r—   r   r   r   r^  G  rÉ  zTestBroadcast.test_gumbelc                 C   rÇ  )Nr   r$   r   )gÂ“y±ÕÀ?gªÀJõR�Á?gþ
~Æ&‘£?r+   rP  r«   )r   r
   re  r/   rÅ  r   r   r    )r   r\  rG  rÈ  re  r®   r—   r   r   r   rf  Z  rÉ  zTestBroadcast.test_logisticc                 C   rÇ  )Nr   r$   r   )gÄñò�ÏH"@g¼ùÜ®ƒÍ @gvv	ý²:@r+   rP  r«   )r   r
   ri  r/   rÅ  r   r   r    )r   rg  rh  Z	bad_sigmari  r®   r—   r   r   r   rj  m  rÉ  zTestBroadcast.test_lognormalc                 C   rÍ  )Nr$   r   )gÝ±lËo½ó?gé/Ó‚´Æó?gz~ßRó?r+   rP  r«   )r   r
   r�  r/   rÅ  r   r   r    )r   rG  rÈ  r�  r®   r—   r   r   r   rŽ  €  rÏ  zTestBroadcast.test_rayleighc                 C   sÞ   dg}dg}dg}dg}t jj}t  g d¢¡}|  ¡  ||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 )
Nr7   r$   r   r&   )g@mò ‰e¾?g¸�`ŸIß¿?g¸‡|2‹í?r+   rP  r«   rF   )r   r
   rº  r/   rÅ  r   r   r    )r   rg  rG  Zbad_meanrÈ  rº  r®   r—   r   r   r   r»  �  s$   zTestBroadcast.test_waldc                 C   sj  dg}dg}dg}dg}dg}|d \}}t jj}t  g 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 ƒ}
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²¨ @rP  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�  ¥  s@   ÿÿ
ÿzTestBroadcast.test_triangularc           	      C   óä   dg}dg}dg}dg}dg}t jj}t  g d¢¡}|  ¡  ||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$   r7   r   ç      ø?©r$   r$   r$   r+   )r   r
   r9   r/   rÅ  r   r   r    )	r   rt  r:   Úbad_nÚ	bad_p_oneÚ	bad_p_twoZbinomr®   r—   r   r   r   r4  É  ó&   

zTestBroadcast.test_binomialc           	      C   r×  )Nr$   r7   r   rØ  )r$   r   r$   r+   )r   r
   rk   r/   rÅ  r   r   r    )	r   rt  r:   rÚ  rÛ  rÜ  Z	neg_binomr®   r—   r   r   r   rl   à  rÝ  z$TestBroadcast.test_negative_binomialc                 C   sz   t j ¡ j}dg}dg}|d g}t jj}t  g d¢¡}|  ¡  ||d ƒ}t||ƒ tt	||d ƒ tt	||d ƒ d S )Nr$   r   r%   )r$   r$   r   r+   )
r   r
   r   Z_poisson_lam_maxrˆ  r/   rÅ  r   r   r    )r   Zmax_lamr‡  Zbad_lam_oneZbad_lam_tworˆ  r®   r—   r   r   r   r‰  ÷  s   

zTestBroadcast.test_poissonc                 C   s¤   dg}dg}t jj}t  g d¢¡}|  ¡  ||d ƒ}t||ƒ tt||d ƒ t jdd�� tt|t j	ƒ tt|ddt j	gƒ W d   ƒ d S 1 sKw   Y  d S )Nr%   r   )r%   r%   r$   r+   rp  )Úinvalid)
r   r
   rÁ  r/   rÅ  r   r   r    Zerrstater<   )r   rà   rË  rÁ  r®   r—   r   r   r   rÂ    s   
"þzTestBroadcast.test_zipfc                 C   ój   dg}dg}dg}t jj}t  g d¢¡}|  ¡  ||d ƒ}t||ƒ tt||d ƒ tt||d ƒ d S )Nr7   r   rØ  rM   r+   )r   r
   rW  r/   rÅ  r   r   r    )r   r:   rÛ  rÜ  Zgeomr®   r—   r   r   r   rX    ó   
zTestBroadcast.test_geometricc                 C   sš  dg}dg}dg}dg}dg}dg}dg}t jj}t  g 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 ƒ}
t|
|	ƒ tt||||d ƒ tt||||d ƒ tt||||d ƒ tt||||d ƒ d S )	Nr$   r%   r   r&   r   r,   rÙ  r+   )r   r
   r`  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   ra  !  s<   


z!TestBroadcast.test_hypergeometricc                 C   rß  )Nr7   r%   r   rÙ  r+   )r   r
   rl  r/   rÅ  r   r   r    )r   r:   rÛ  rÜ  rl  r®   r—   r   r   r   rm  D  rà  zTestBroadcast.test_logseriesN)#r2   r3   r4   r[   rÅ  r¢  r€  r.  rC  r”  rR  rN  r}  rÔ  r8  rx  r˜  r¶  r†  r½  rŒ  rc  r^  rf  rj  rŽ  r»  r�  r4  rl   r‰  rÂ  rX  ra  rm  r   r   r   r   rÄ    sB    !$#rÄ  c                   @   r   )Ú
TestThreadc                 C   s   t dƒ| _d S )Nr,   )r   Úseedsr!   r   r   r   r[   T  s   zTestThread.setup_methodc                    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||ƒ d S 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   ó   g | ]}|  ¡ ‘qS r   )Ústart©rý   rJ   r   r   r   rÿ   `  r   c                 S   ré  r   )Újoinrë  r   r   r   rÿ   a  r   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   ó   dd„ }| j |dd� d S )Nc                 S   s   | j dd�|d< d S )Né'  rb   .)r  ©rZ   Úoutr   r   r   Ú
gen_randomn  rE   z*TestThread.test_normal.<locals>.gen_random)rõ  ©rò  ©ró  ©r   rø  r   r   r   r€  m  ó   zTestThread.test_normalc                 C   rô  )Nc                 S   s   | j t d¡d�|d< d S )N©r?   r   rF  .)rB  r   Zonesrö  r   r   r   rø  s  s   z'TestThread.test_exp.<locals>.gen_randomrý  rù  rú  rû  r   r   r   Útest_expr  rü  zTestThread.test_expc                 C   rô  )Nc                 S   s   | j ddgd dd�|d< d S )Nr   rn  r.   rõ  rb   .)rC   rö  r   r   r   rø  x  s   z/TestThread.test_multinomial.<locals>.gen_random)rõ  r.   rù  rú  rû  r   r   r   ro  w  rü  zTestThread.test_multinomialN)r2   r3   r4   r[   ró  r€  rþ  ro  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   r[     s   
z$TestSingleEltArrayInput.setup_methodc                 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 ]}||v r?|t  dg¡ƒ}n|| jƒ}t|j| jƒ q0d S rj   )r   r
   rB  r“  r7  r—  r„  r¼  r‹  r�  rˆ  rÁ  rW  rl  r/   r   r   rP   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 rCt  dg¡}n| j}|| j|ƒ}t|j| jƒ || jd |ƒ}t|j| jƒ || j|d ƒ}t|j| jƒ q6d S )Nr7   r   )r   r
   r¡  r  r-  rQ  rM  rw  rµ  rb  r]  re  ri  rº  r9   rk   r/   r  r   r   rP   r  )r   r  r  r¦  r  r÷  r   r   r   Útest_two_arg_funcs˜  s(   úòz*TestSingleEltArrayInput.test_two_arg_funcsc                 C   sŒ   t jjt jjt jjg}|D ]6}|| 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
   r|  rœ  r`  r   r  r  r   rP   r  )r   r  r¦  r÷  r   r   r   Útest_three_arg_funcsÅ  s   ÿøz,TestSingleEltArrayInput.test_three_arg_funcsN)r2   r3   r4   r[   r  r  r  r   r   r   r   rÿ  ~  s
    -rÿ  )rÎ   r  Znumpyr   Znumpy.testingr   r   r   r   r   r   r   r	   r
   rŽ   r   r5   r>   rU   rm   rŸ   rÄ  rá  rÿ  r   r   r   r   Ú<module>   s2    ('"0      |    >,