o
    Eb                    @   s"
  d Z ddlZddlZddlZddlZ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mZ ddlZddlmZ ddlZddlZddlmZmZ ddlmZ ddlmZ ddlmZ dd	lm Z m!Z!m"Z"m#Z# ddl$m%Z% dd
l&m'Z' ddl(Zddl)m*Z*m+Z+m,Z, ddl-m.Z.m/Z/ ddl0m1Z1m2Z2 ddl3m4Z4m5Z5 ddl6m7Z7m8Z8 ddl9m:Z: ej;j<dkZ=ejdkoe> dkZ?ddgZ@dhddZAdd ZBdd ZCdd ZDdd ZEd d! ZFd"d# ZGd$d% ZHejIJd&g d'd(d) ZKd*d+ ZLd,d- ZMejIJd.g d/d0d1 ZNG d2d3 d3ZOG d4d5 d5ZPG d6d7 d7ZQG d8d9 d9ZRG d:d; d;ZSG d<d= d=ZTG d>d? d?ZUG d@dA dAZVG dBdC dCZWG dDdE dEZXG dFdG dGZYG dHdI dIZZG dJdK dKZ[G dLdM dMZ\G dNdO dOZ]G dPdQ dQZ^G dRdS dSZ_G dTdU dUZ`G dVdW dWZaG dXdY dYZbG dZd[ d[ZcG d\d] d]ZdG d^d_ d_ZeG d`da daZfG dbdc dcZgG ddde deZhG dfdg dgZiG dhdi diZjG djdk dkZkG dldm dmZlG dndo doZmG dpdq dqZnG drds dsZoG dtdu duZpdvdw Zqdxdy Zrdzd{ ZsG d|d} d}ZtG d~d dZuG dd dZvG dd dZwG dd dZxG dd dZyG dd dZzG dd dZ{G dd dZ|G dd dZ}G dd dZ~G dd dZG dd dZG dd dZG dd dZG dd dZG dd dZG dd dZG dd dZG dd dZdd ZG dd dZG dd dZG dd dZG dd dZG dd dZG dd dZG dd dZG dd dZG dd dZG dd dZG dd dZG dd dZG dd dZG dd dZG ddÄ dÃZG ddń dŃZG ddǄ dǃZddɄ Zdd˄ Zdd̈́ Zddτ ZejIje=ddэddӄ ZddՄ Zddׄ Zddل Zddۄ Zdd݄ Zdd߄ Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zdd ZejIJddeddgfd eddgfdeddgfdeddgfgd	d
 Zdd Zdd Zdd Zdd Zdd ZG dd de%jZG dd de%jZG dd de%jZG dd de%jZG dd dZG dd  d ZejIje=d!dэd"d# Zd$d% Zd&d' ZÐd(d) ZĐd*d+ ZŐd,d- ZƐd.d/ Zǐd0d1 ZȐd2d3 Zɐd4d5 Zʐd6d7 Zːd8d9 Z̐d:d; Z͐d<d= ZejIJd>d?d?g d@fdAdBg d@fdCdDg dEfgdFdG ZϐdHdI ZG dJdK dKZѐdLdM ZG dNdO dOZG dPdQ dQZG dRdS dSZՐdTdU ZejIJdVe2e/ dWdX ZאdYdZ Zؐd[d\ ZejIJd]d^d_gd`da ZejIJdbg dcddde Zېdfdg ZdS (i  z!
Test functions for stats module
    N)	assert_equalassert_array_equalassert_almost_equalassert_array_almost_equalassert_allcloseassert_assert_warnsassert_array_lesssuppress_warnings)raises)	typecodesarray)rec_append_fields)special)check_random_state)IntegrationWarningquad	trapezoidcumulative_trapezoid)
argsreduce)xlogy	polygammaentr)distcontinvdistcont   )distdiscreteinvdistdiscrete)FitDataError
_argus_phi)rootfmin)productdarwinZx86_64tukeylambdapearson3c                 C   s*   |d u r
d| |f }t t| ||d d S )Nz%s does not have attribute %smsg)r   hasattr)abr'    r+   F/usr/lib/python3/dist-packages/scipy/stats/tests/test_distributions.py_assert_hasattr3   s   r-   c                   C   s   t tjjd d S )NZf_gen)r-   scipystatsdistributionsr+   r+   r+   r,   test_api_regression9      r1   c                 C   s8   t j| ||d}t||||dtj |   d S )Nlocscale   )r/   vonmisesr   pdfnumpypikLsxvmr+   r+   r,   check_vonmises_pdf_periodic>   s   (rA   c                 C   s@   t j| ||d}t||d ||dtj |  d  d S )Nr3   r   r6   )r/   r7   r   cdfr9   r:   r;   r+   r+   r,   check_vonmises_cdf_periodicC   s   0rC   c                  C   s`   t tjjj} dd tD }dd tD }g d}|| | }t tdd |}| |ks.J d S )Nc                 S      g | ]}|d  qS r   r+   .0distr+   r+   r,   
<listcomp>J       z0test_distributions_submodule.<locals>.<listcomp>c                 S   rD   rE   r+   rF   r+   r+   r,   rI   K   rJ   )rv_discreterv_continuousrv_histogramentropytrapzc                 S   s   t | d S )N<)str
startswithr>   r+   r+   r,   <lambda>R       z.test_distributions_submodule.<locals>.<lambda>)setr.   r/   r0   __all__r   r   filter)actualZ
continuousZdiscreteotherexpectedr+   r+   r,   test_distributions_submoduleH   s   r\   c                  C   sx   dD ]7} ddt jddfD ],}t| dd| t| dd| t| dd| t| dd| t| dd| t| dd| qqd S )N)皙?r   e   r   r   
   d   )r9   r:   rA   rC   )r<   r?   r+   r+   r,   test_vonmises_pdf_periodicW   s   ra   c                   C   s&   t tjjtj  t tjjtj d S N)r   r/   Zvonmises_liner)   npr:   r*   r+   r+   r+   r,   test_vonmises_line_supportc   s   rd   c                  C   s   t d} t| dd d S )N   r         ?)r/   r7   r   rB   )r@   r+   r+   r,   test_vonmises_numericalh   s   
rg   zx, kappa, expected_pdf))r]   {Gz?g|65?)r]         9@g?U?)r]   re   g"?)       @rh   gDfI?)rj   ri   g1<)rj   re           c                 C   s    t j| |}t||dd d S )NV瞯<rtol)r/   r7   r8   r   )r?   kappaexpected_pdfr8   r+   r+   r,   test_vonmises_pdfw   s   rq   c                 K   sf   | j |fi |}tt| | j |fi |}|||}|||}||ks/tj||dds1J dS dS )aS  
    This utility function checks that the log-likelihood (computed by
    func) of the result computed using dist.fit() is less than or equal
    to the result computed using the generic fit method.  Because of
    normal numerical imprecision, the "equality" check is made using
    `np.allclose` with a relative tolerance of 1e-15.
    rl   rm   N)fitsupertyperc   allclose)rH   datafunckwdsZmle_analyticalZnumerical_optZll_mle_analyticalZll_numerical_optr+   r+   r,   _assert_less_or_close_loglike   s   

ry   c                 C   s  ddg}| j rt| j d}|g dd | 7 }tt|tt|}g d}tjt	dd | j
|fi | W d    n1 sDw   Y  tjt	dd | 
tjg W d    n1 sbw   Y  tjt	dd | 
tjg W d    n1 sw   Y  tjtd	d | j
|d
d W d    n1 sw   Y  tjtdd | j
|gdgt|d  R   W d    d S 1 sw   Y  d S )Nflocfscale,)f0f1f2r   r6      z3All parameters fixed. There is nothing to optimize.matchz#The data contains non-finite valueszUnknown keyword arguments:r6   )Zextra_keywordzToo many positional arguments.r   )shapeslensplitdictziprc   arangepytestr   RuntimeErrorrr   naninf	TypeError)rH   ZparamZnshapesZ	all_fixedrv   r+   r+   r,   assert_fit_warnings   s6   ""r   rH   )alpha	betaprimefatiguelifeinvgammainvgauss
invweibullZ	johnsonsblevylevy_llognormZgilbratZpowerlognormrayleighZwaldc                 C   s   t t}||  }tt| } t| j| jg|R  d t| j| jg|R  t	j
  t| j| jg|R  d t| j| jg|R  t	j
  dS )zgh-6235r   N)r   r   getattrr/   r   r8   r)   r   logpdfrc   r   r*   )rH   dctargsr+   r+   r,   test_support   s   
"r   c                   @   ,   e Zd Zdd Zdd Zdd Zdd Zd	S )
TestRandIntc                 C      t jd d S N  rc   randomseedselfr+   r+   r,   setup_method      zTestRandInt.setup_methodc                 C   s   t jjdddd}tt|dk t|dk@  tt|dk t jjdddd}tt|dk t|jj	t
d v  t jdd}t|dk|dk @  tt|tjtt|d	 t ddd
}t|jj	t
d v  d S )N      r`   sizer6   2   
AllInteger   .   r&   r   )r/   randintrvsr   r9   allr   shapedtypecharr   
isinstanceZ
ScalarTypereprrt   r   valsvalr+   r+   r,   test_rvs   s    zTestRandInt.test_rvsc                 C   sF   t jdd }t |dk|dk @ dd}tj|dd}t|| d S )Nr   $   r   r   {Gz?)r9   r_wherer/   r   pmfr   )r   r<   outr   r+   r+   r,   test_pdf   s   zTestRandInt.test_pdfc                 C   sd   t ddd}t|}t|dk|dkgd|d d d	 gd}tj|dd}t||d
d d S )Nr   r   r`   r   r         ?      @r   ri      decimal)	rc   linspacer9   ZfloorZselectr/   r   rB   r   )r   r?   r<   r   r   r+   r+   r,   test_cdf   s
   
*zTestRandInt.test_cdfN)__name__
__module____qualname__r   r   r   r   r+   r+   r+   r,   r      s
    r   c                   @   4   e Zd Zdd Zdd Zdd Zdd Zd	d
 ZdS )	TestBinomc                 C   r   r   r   r   r+   r+   r,   r      r   zTestBinom.setup_methodc                 C   s   t jjdddd}tt|dkt|dk@  tt|dk t|jjt	d v  t jdd}tt
|t t ddd}tt
|tj t|jjt	d v  d S Nr_         ?r   r   r   r   r   )r/   binomr   r   r9   r   r   r   r   r   r   intndarrayr   r+   r+   r,   r      s    zTestBinom.test_rvsc                 C   sD   t jddd}t jddd}t|dddd t|dddd d S )Nr`   r   r   r   rl   rn   atol)r/   r   r   r   )r   vals1vals2r+   r+   r,   test_pmf   s   zTestBinom.test_pmfc                 C   s|   t dd}tg d}tt|| }| }t|| t dd}| }t|d t dd}| }t|d d S )Nr6   rf   )      ?rf   r   rk   r   )	r/   r   rc   r   sumr   rN   r   r   )r   r*   
expected_p
expected_hhr+   r+   r,   test_entropy   s   

zTestBinom.test_entropyc                 C   sj   t  ' t dt ttjddd d ttjddd d W d    d S 1 s.w   Y  d S )Nerrorr6   r   )np)	warningscatch_warningssimplefilterRuntimeWarningr   r/   r   meanstdr   r+   r+   r,   test_warns_p0  s
   
"zTestBinom.test_warns_p0N)r   r   r   r   r   r   r   r   r+   r+   r+   r,   r      s    r   c                   @      e Zd Zdd ZdS )TestArcsinec                 C   s&   t jddg}t|tjtjg d S Nr   r   )r/   Zarcsiner8   r   rc   r   r   r   r+   r+   r,   test_endpoints  s   zTestArcsine.test_endpointsN)r   r   r   r   r+   r+   r+   r,   r   	      r   c                   @   $   e Zd Zdd Zdd Zdd ZdS )TestBernoullic                 C   r   r   r   r   r+   r+   r,   r     r   zTestBernoulli.setup_methodc                 C   s   t jjddd}tt|dkt|dk@  tt|dk t|jjt	d v  t jd}tt
|t t dd}tt
|tj t|jjt	d v  d S )Nr   r   r   r   r   r   r   )r/   	bernoullir   r   r9   r   r   r   r   r   r   r   r   r   r+   r+   r,   r     s    zTestBernoulli.test_rvsc                 C   st   t d}dtd dtd  }| }t|| t d}| }t|d t d}| }t|d d S )Nr         пr   rk   r   )r/   r   rc   logrN   r   r   )r   r*   r   r   r+   r+   r,   r   !  s   




zTestBernoulli.test_entropyN)r   r   r   r   r   r   r+   r+   r+   r,   r         r   c                   @   r   )TestBradfordc                 C   s:   d}t dd}tj||}tj||}t|| d S )Nr]   )rc   logspacer/   ZbradfordrB   ppfr   )r   cr?   qxxr+   r+   r,   test_cdf_ppf3  s
   zTestBradford.test_cdf_ppfN)r   r   r   r  r+   r+   r+   r,   r   1  r   r   c                   @   s,   e Zd ZdZdZdd Zdd Zdd Zd	S )
TestChignT9;g2 >h?@c                 C   s"   t jdd}t|| jdd d S )Nr_      rl   rm   )r/   chisfr   CHI_SF_10_4r   r>   r+   r+   r,   test_sfD     zTestChi.test_sfc                 C   s"   t j| jd}t|ddd d S )Nr  r_   rl   rm   )r/   r  isfr  r   r   r?   r+   r+   r,   test_isfH     zTestChi.test_isfc                 C   s"   t jjdd}t|| jdd d S )N  )df-q=rm   )r/   r  r   r   CHI_MEAN_1000r  r+   r+   r,   	test_meanL  r	  zTestChi.test_meanN)r   r   r   r  r  r  r  r  r+   r+   r+   r,   r  ;  s    r  c                   @   r   )
TestNBinomc                 C   r   r   r   r   r+   r+   r,   r   R  r   zTestNBinom.setup_methodc                 C   s   t jjdddd}tt|dk tt|dk t|jjt	d v  t jdd}tt
|t t ddd}tt
|tj t|jjt	d v  d S r   )r/   nbinomr   r   r9   r   r   r   r   r   r   r   r   r   r+   r+   r,   r   U  s   zTestNBinom.test_rvsc                 C   sH   t ttjdddtjddd tjjddd}t|d d S )Ni  i  gp=
ף?r   r   )	r   rc   expr/   r  logpmfr   r.   r   )r   r   r+   r+   r,   r   `  s
   zTestNBinom.test_pmfN)r   r   r   r   r   r   r+   r+   r+   r,   r  Q  r   r  c                   @   st   e Zd Zdd Zejjdd Zejjdd Zejjdd Z	ejjd	d
 Z
dd Zdd Zdd Zdd ZdS )TestGenInvGaussc                 C   r   r   r   r   r+   r+   r,   r   j  r   zTestGenInvGauss.setup_methodc                 C   :   t dd}t |jddd|j\}}t|dkd d S )Nffffff@      ?  r   r   random_state皙?Tr/   geninvgausskstestr   rB   r   r   gig_r   r+   r+   r,   test_rvs_with_mode_shiftm     z(TestGenInvGauss.test_rvs_with_mode_shiftc                 C   r  )N?r   r  r   r  r  Tr  r"  r+   r+   r,   test_rvs_without_mode_shiftt  r&  z+TestGenInvGauss.test_rvs_without_mode_shiftc                 C   r  )Nr]   皙?r  r   r  r  Tr  r"  r+   r+   r,   test_rvs_new_method{  r&  z#TestGenInvGauss.test_rvs_new_methodc                 C   s<   dd }t |ddd t |ddd t |ddd d S )Nc                 S   s0   t | |}|jddd}t ||jd dkS )Nr  r   r  r   r  )r/   r   r   r!  rB   )r   r*   r#  r   r+   r+   r,   my_ks_check  s   z4TestGenInvGauss.test_rvs_p_zero.<locals>.my_ks_checkr   r)  Tr'  r  )r   )r   r+  r+   r+   r,   test_rvs_p_zero  s   zTestGenInvGauss.test_rvs_p_zeroc                 C   s6   t tddjddddtddjddd  d S )N      r6   r_   r   r  r   r  )r   r/   r   r   r   r+   r+   r,   test_rvs_negative_p  s   z#TestGenInvGauss.test_rvs_negative_pc                 C   s   t jjddddd}tt j|ddgdd dkd	 d
tddd}}t jj|dd| |d}t|t 	|| t jj
|dd| |d}t|t 	|
| d S )Nr        r   r   )r   r   r*   r  r   r   333333?Tr`   rh   r_   )r   r*   r5   )r/   r   r   r   r!  rc   r   r8   r   r   rB   )r   Zigmur?   Zpdf_igZcdf_igr+   r+   r,   test_invgauss  s    zTestGenInvGauss.test_invgaussc                 C   s6   t g d}t ddd}t|tj|dd d S )N)
g;g8y8?g\}Z?ghܲ<?g&Khdd?g7!L?g	Cf?g_诱?g>0|?gRq?rh   r   r_   rf   r   )rc   r   r   r   r/   r   r8   )r   vals_Rr?   r+   r+   r,   
test_pdf_R  s   zTestGenInvGauss.test_pdf_Rc                 C   s0   t tjdddd t tjdddd d S )Nr   rf   g    >Ar   r6   )r   r/   r   r8   r   r+   r+   r,   test_pdf_zero  s   zTestGenInvGauss.test_pdf_zeroN)r   r   r   r   r   markslowr%  r(  r*  r,  r.  r3  r5  r6  r+   r+   r+   r,   r  i  s    




r  c                   @   T   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S )TestGenHyperbolicc                 C   r   r   r   r   r+   r+   r,   r     r   zTestGenHyperbolic.setup_methodc           
      C   l   t g d}d\}}}d\}}||| || f}tj|||d}t ddd}	t||	|ddd	 d S )
N)
gF^T=g{X'=gNunƢsy>g7!t?gɫzI?g̸?g?gq6.?gmPZz?g:PgN?r6   r6   r   rf   r  r3   r_   r   vIh%<=r   rn   )rc   r   r/   genhyperbolicr   r   r8   
r   r4  lmbdar   betar2  deltar   ghr?   r+   r+   r,   
test_pdf_r     
zTestGenHyperbolic.test_pdf_rc           
      C   r;  )
N)
g=R0W<=gEB=g#Nb>g_PG>g! 1Or?gUM?g u./?gfy?gl5\?gkV?r<  r=  r3   r>  r_   r   ư>r@  )rc   r   r/   rA  r   r   rB   rB  r+   r+   r,   
test_cdf_r  rH  zTestGenHyperbolic.test_cdf_rc                    s\   g d}d\}}}d\|| | f  fddt ddD }t||dd	d
 d S )N)g@쐧@g@쐧 @g܊B@gvGăi@r<  r=  c                    s$   g | ]}t j d |qS )r3   )r/   rA  moment)rG   ir   rE  r2  r+   r,   rI     s    z4TestGenHyperbolic.test_moments_r.<locals>.<listcomp>r   r   r   r?  r@  )ranger   )r   r4  rC  r   rD  Zvals_usr+   rM  r,   test_moments_r  s   	
z TestGenHyperbolic.test_moments_rc           
      C   sd   d\}}}d\}}||| || f}t j|||d}t |jddd|j\}}	t|	dkd d S )	Nr<  r=  r3   r  r   r  r  T)r/   rA  r!  r   rB   r   )
r   rC  r   rD  r2  rE  r   rF  r$  r   r+   r+   r,   r     s   
zTestGenHyperbolic.test_rvsc           	      C   s   t ddd}t |dt t jj d}}dt |}}| d ||f}tj|||d}t |	d|	dd	d d t j
f }t||tj||dd
d d S )Nr   r   r_   r6   r   r3   rh   Gz?r   rI  r@  )rc   r   float_powerfinfofloat32epssqrtr/   rA  r   newaxisr   r8   t)	r   r  r   rD  r2  rE  r   rF  r?   r+   r+   r,   
test_pdf_t  s    (
zTestGenHyperbolic.test_pdf_tc           	      C   s   dt t jjd}}}d\}}|||f}tj|||d}t |d|ddd d t jf }t	|
|tj
|ddd	 d S )
Nr/  r   )r   r   r3   rh   rP  r   rI  r@  )rc   rR  rS  rT  r/   rA  r   r   rV  r   r8   cauchy)	r   rC  r   rD  r2  rE  r   rF  r?   r+   r+   r,   test_pdf_cauchy  s   
(
z!TestGenHyperbolic.test_pdf_cauchyc           	      C   s   t ddd}t t jj}d\}}}||| || f}tj|||d}t dddd d t jf }t|	|tj
j	||ddd	d
d d S )Nr>  r_   )r   r   r   r3   r      r   r   r   dy=r@  )rc   r   rR  rS  rT  r/   rA  rV  r   r8   laplace)	r   r4   rE  rC  r   rD  r   rF  r?   r+   r+   r,   test_pdf_laplace(  s   

z"TestGenHyperbolic.test_pdf_laplacec           	   	   C   s   t dddt dddt dtd t dddt dddf\}}}}d	}||| || f}tj|||d
}t |d|ddd d t jf }t|	|tj
j	|||||dddd d S )Nr   r[  r_   r      r`   r/  r3   rh   rP  r   )r)   r*   r4   r5   r?  r@  )rc   r   rQ  rN  r/   rA  r   rV  r   r8   norminvgauss)	r   r   rD  rE  r2  rC  r   rF  r?   r+   r+   r,   test_pdf_norminvgauss=  s   (

z'TestGenHyperbolic.test_pdf_norminvgaussN)r   r   r   r   rG  rJ  rO  r   rX  rZ  r^  rc  r+   r+   r+   r,   r:    s    r:  c                   @   r   )TestNormInvGaussc                 C   r   r   r   r   r+   r+   r,   r   U  r   zTestNormInvGauss.setup_methodc                 C   @   t g d}t g d}tjj|ddd}t||dd d S )N)gp~٧>g @X>g}e?g{)߳?gwH?ir      r   r   rf   r)   r*   &.>r   )rc   r   r/   rb  rB   r   )r   Zr_cdfx_testvals_cdfr+   r+   r,   
test_cdf_RX  s   zTestNormInvGauss.test_cdf_Rc                 C   re  )N)gp϶>g\H55$?geСt&?go NiH?g]u>rf  r   rf   ri  rj  rk  )rc   r   r/   rb  r8   r   )r   Zr_pdfrl  Zvals_pdfr+   r+   r,   r5  d  s   zTestNormInvGauss.test_pdf_Rc                 C   s   d\}}t |d |d  }|| |d |d  d| |t |  ddd|d  |d    | f}t|tjj||dd d S )	Nr   rf   r6   r         @r   r  mvskmoments)rc   rU  r   r/   rb  )r   r)   r*   gammaZv_statsr+   r+   r,   
test_statsl  s   (zTestNormInvGauss.test_statsc                 C   s@   d\}}t g d}tj|||}t|tj||| d S )Nro  MbP?rf   +?)rc   r   r/   rb  r   r   rB   )r   r)   r*   rl  r   r+   r+   r,   test_ppfs  s   zTestNormInvGauss.test_ppfN)r   r   r   r   rn  r5  ru  ry  r+   r+   r+   r,   rd  T  s    rd  c                   @   sL   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S )TestGeomc                 C   r   r   r   r   r+   r+   r,   r   {  r   zTestGeom.setup_methodc                 C      t jjddd}tt|dk tt|dk t|jjt	d v  t jd}tt
|t t dd}tt
|tj t|jjt	d v  d S )Nr   r   r   r   r   r   )r/   geomr   r   r9   r   r   r   r   r   r   r   r   r   r+   r+   r,   r   ~     zTestGeom.test_rvsc                 C   s$   t jg dd}t|g d d S )Nr   rf   )rf   r         ?)r/   r|  r   r   r   r   r+   r+   r,   r     s   zTestGeom.test_pmfc                 C   sV   t tjg dd}tjg dd}t||ddd tjdd}t|d d S )Nr   rf   rl   r   r   r   rk   )rc   r   r/   r|  r   r  r   r   )r   r   r   r   r+   r+   r,   test_logpmf  s
   zTestGeom.test_logpmfc                 C   sL   t jg dd}t jg dd}tg d}t|| t|d|  d S )Nr   rf   rf   r   g      ?r   )r/   r|  rB   r  r   r   r   r   vals_sfr[   r+   r+   r,   test_cdf_sf  s
   
zTestGeom.test_cdf_sfc                 C   sV   t jg dd}t jg dd}tg d}t|t| t|t|  d S )Nr   rf   r  )	r/   r|  logcdflogsfr   r   rc   r   Zlog1pr  r+   r+   r,   test_logcdf_logsf  s
   zTestGeom.test_logcdf_logsfc                 C   ,   t jg dd}tg d}t|| d S )Nr  rf   )r   rj   rp  )r/   r|  r   r   r   r   r   r[   r+   r+   r,   ry       zTestGeom.test_ppfc                 C   s   t tjddddd d S )N#B;r   +=rk  )r   r/   r|  r   r   r+   r+   r,   test_ppf_underflow  s   zTestGeom.test_ppf_underflowN)r   r   r   r   r   r   r  r  r  ry  r  r+   r+   r+   r,   rz  z  s    
rz  c                   @   r   )
TestPlanckc                 C   r   r   r   r   r+   r+   r,   r     r   zTestPlanck.setup_methodc                 C   r  )Nr   r   )g|dy?g8'\>ggrU!>)r/   planckr  r   r   r  r+   r+   r,   r    s   zTestPlanck.test_sfc                 C   r  )N)     @@     @@     p@r  )g    P.g    h>g    TF)r/   r  r  r   r   r  r+   r+   r,   
test_logsf  r  zTestPlanck.test_logsfN)r   r   r   r   r  r  r+   r+   r+   r,   r    s    r  c                   @      e Zd Zdd Zdd ZdS )TestGennormc                 C   0   g d}t j|d}t j|}t|| d S Nr   r   )r/   gennormr8   r]  r   r   pointspdf1pdf2r+   r+   r,   test_laplace     zTestGennorm.test_laplacec                 C   4   g d}t j|d}t jj|dd}t|| d S Nr   r6   g;f?r5   )r/   r  r8   normr   r  r+   r+   r,   	test_norm     zTestGennorm.test_normN)r   r   r   r  r  r+   r+   r+   r,   r        r  c                   @   r   )TestHalfgennormc                 C   r  r  )r/   halfgennormr8   exponr   r  r+   r+   r,   
test_expon  r  zTestHalfgennorm.test_exponc                 C   r  r  )r/   r  r8   Zhalfnormr   r  r+   r+   r,   test_halfnorm  r  zTestHalfgennorm.test_halfnormc                 C   s6   g d}t j|d}t j|d}t|d|  d S )Nr   g
(?r6   )r/   r  r8   r  r   r  r+   r+   r,   test_gennorm  s   zTestHalfgennorm.test_gennormN)r   r   r   r  r  r  r+   r+   r+   r,   r    s    r  c                   @   r   )TestLaplaceasymmetricc                 C   s6   t g d}tj|d}tj|}t|| d S r  )rc   r   r/   laplace_asymmetricr8   r]  r   r  r+   r+   r,   r    s   z"TestLaplaceasymmetric.test_laplacec                 C   sL   t g d}d}d| }tj||}tj||d  |}t|| d S )Nr   r6   r   )rc   r   r/   r  r8   r   )r   r  ro   Zkapinvr  r  r+   r+   r,   test_asymmetric_laplace_pdf  s   z1TestLaplaceasymmetric.test_asymmetric_laplace_pdfc              	   C   s   t t d t dg}d}tj||}tj||}tj||}t ddg}t ddg}t dd	g}tj||}	|}
tj	||}|}t
t ||||	|ft ||||
|f d S )
N   r_   r6   r]   gMbp?r)  gV-?皙?gMb`?)rc   r   r   r/   r  r8   rB   r  r   r
  r   concatenate)r   r  ro   r  Zcdf1Zsf1r  Zcdf2Zsf2Zppf1Zppf2Zisf1Zisf2r+   r+   r,   !test_asymmetric_laplace_log_10_16  s   z7TestLaplaceasymmetric.test_asymmetric_laplace_log_10_16N)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dZ
ejddd Zdd Zdd Zdd Zdd ZdS )TestTruncnormc                 C   r   r   r   r   r+   r+   r,   r     r   zTestTruncnorm.setup_methodc              	   C   J   t jjg ddddgd dd}ttjdd	dd
dtjg}t|| d S )Nr/  r   -C6?rf   gH.?r   r6         r   r      r6   r3   r   fO?l@r   )r/   	truncnormr   rc   r   r   r   r  r+   r+   r,   test_ppf_ticket1131  
   
z!TestTruncnorm.test_ppf_ticket1131c              	   C   r  )Nr  r  r   r   r  r6   r3   r   r  r  r   )r/   r  r
  rc   r   r   r   r  r+   r+   r,   test_isf_ticket1131  r  z!TestTruncnorm.test_isf_ticket1131c                 C   s   d\}}t jj||dddd}t||   k o"|   k o"|k n   d\}}t jj||dddd}t||   k oK|   k oK|k  d S    d S )N)ir>  r   r   r_   r   )r_      r/   r  r   r   minmaxr   lowhighr?   r+   r+   r,   test_gh_2477_small_values  s   .8z'TestTruncnorm.test_gh_2477_small_valuesc                 C   sH  d\}}t jj||dddd}t||   ko"|   ko"|kn  t|||gf d\}}t jj||dddd}t||   k oO|   k oO|k n   d\}}t jj||dddd}t||   k ou|   k ou|k n   d\}}t jj||dddd}t||   k o|   k o|k  d S    d S )	N)r`   r^   r   r   r_   r   )r    )'  i'  )i)r/   r  r   r   r  r  rQ   r  r+   r+   r,   test_gh_2477_large_values$  s   <..8z'TestTruncnorm.test_gh_2477_large_valuesc                 C   s  ddgddgfD ]\}}t t j ||t jg}|| d }tj|||}tj|||}tj|||}t g d}t g d}	t g d}
|d	k rVt g d
}
t|| t||	 t||
 tt 	|
d |
d  |d  t g d}tj
|||}t |t |d |g}t|| |d	k rttj|||d ttj|||d nttj|||d ttj|||d tj|||}tt 	||
d  |d d  qd S )Nr   r  r   rj   r   r   r   r   r   r   rk   rk   )r   d._MT
@BKg?r   r   )r   r  r  r   r   r6   rf   r   rf   r   gĖy	@gdv*?g,mj%V?r   )rc   r   r   r/   r  rB   r  r8   r   r   r   sign)r   r  r  xvalsxmidcdfssfspdfsexpected_cdfsexpected_sfsexpected_pdfspvalsppfsexpected_ppfsr8   r+   r+   r,   test_gh_9403_nontail_values7  sJ   



"z)TestTruncnorm.test_gh_9403_nontail_valuesc                 C   s  ddgddgfD ];\}}t t j ||t jg}|| d }tj|||}tj|||}tj|||}t g d}t g d}	t g d}
|d	k rWt g d
}
t|| t||	 t||
 tt 	|
d |
d  |d  t g d}tj
|||}t |t |d |g}t|| tj|||}t|| |d	k rttj|||d ttj|||d nttj|||d ttj|||d tj|||}tt 	||
d  |d d  t ||d}|d d d  }ttj|||tj|| | d d d  ttj|||tj|| | d d d  ttj|||tj|| | d d d  qd S )N'   (   rj   r  r  )r   pGC@Ff<r   r   )r   r  r  r   r   r6   rf   r  gEC@g i?gspXio)>r   r  r`  )rc   r   r   r/   r  rB   r  r8   r   r   r   r  r   )r   r  r  r  r  r  r  r  r  r  r  r  r  r  r8   Zxvals2r+   r+   r,   test_gh_9403_medium_tail_valuesZ  sd   




 z-TestTruncnorm.test_gh_9403_medium_tail_valuesr  c                 C   s^   |d d \}}}}t jj ||dd\}	}
}}t|	| t|
| t|||d t|| d S )Nr  rq  rr  r   )r/   r  r   )r   r)   r*   r[   	decimal_sZm0Zv0Zs0Zk0mvr>   r<   r+   r+   r,   _test_moments_one_range  s   

z%TestTruncnorm._test_moments_one_rangez&reduced accuracy with 32bit platforms.c                 C   s   |  ddg d |  ddg d |  ddg d |  d	d
g d |  dtjg d |  tj dg d |  ddg d |  ddg d |  ddg d |  ddg d | j ddg ddd |  ddg d |  dd g d! d S )"Nr   )r   r   rk   rk   r>  r_   r  r   )rk   gMFmz%?rk   gſr6   )rk   ghI}?rk   gHNMr   )gQ63E?nl$A?gv1"D?<8L?)gQ63Er  gv1"Dr  r`  )gU*? bn?gnA?!c3TXʿr   )gU*ҿr  gnAr  )gH$7"g͢xr?gN8`gp=#K@r   i)gif3g6oRf?gg+15|@i)go=gqXS?gCØsgm@   )r  r  r  )gGC&SԍuE?gynw@r  r  )gGC@r  gy?r  )r  rc   r   r   r+   r+   r,   test_moments  s   	zTestTruncnorm.test_momentsc                 C   s0   t jj dtjdd\}}t|d t|d d S )Nr   mvrr  g e3E?g:&A?)r/   r  rc   r   r   )r   r  r  r+   r+   r,   test_9902_moments  s   
zTestTruncnorm.test_9902_momentsc                 C   sV   d\}}t jj||dddd}t||   k o%|   k o%|k  d S    d S )N)r_   r   r   r   r_   r   r  r  r+   r+   r,   test_gh_1489_trac_962_rvs  s   8z'TestTruncnorm.test_gh_1489_trac_962_rvsc                 C   s   ddt j dt j t j dddddg}dddt jddddd	t jt jg}tjj||dt|fd
}t |dt|fks=J tt ||j	ddk tt |j
dd|k d S )Nr>  r_   rg  ir  r  r   r  -   r   r   )Zaxis)rc   r   r/   r  r   r   r   r   r   r  r  r  r+   r+   r,   test_gh_11299_rvs  s   & zTestTruncnorm.test_gh_11299_rvsc                 C   s.   t tjdrtjjdddtj d d S d S )Ndefault_rngr>  rg  r   r  )r(   rc   r   r/   r  r   r  r   r+   r+   r,   test_rvs_Generator  s
   
z TestTruncnorm.test_rvs_GeneratorN)r  )r   r   r   r   r  r  r  r  r  r  r  r   r7  Zxfail_on_32bitr  r  r  r  r  r+   r+   r+   r,   r  
  s    
#
/

7
r  c                   @   s&   e Zd Zejdg ddd ZdS )TestGenLogisticx, expected))ig4`n)ig4	ncg)r   gX
=)r`   gh@X)r  gh<c                 C   s$   d}t j||}t||dd d S )Nr  r?  rm   )r/   Zgenlogisticr   r   )r   r?   r[   r   logpr+   r+   r,   test_logpdf  s   zTestGenLogistic.test_logpdfN)r   r   r   r   r7  parametrizer  r+   r+   r+   r,   r    s    r  c                   @   r9  )TestHypergeomc                 C   r   r   r   r   r+   r+   r,   r     r   zTestHypergeom.setup_methodc                 C   s   t jjddddd}tt|dkt|dk@  tt|dk t|jjt	d v  t jddd}tt
|t t dddd}tt
|tj t|jjt	d v  d S )Nr[  r_   r   r   r   r   r   )r/   	hypergeomr   r   r9   r   r   r   r   r   r   r   r   r   r+   r+   r,   r     s   zTestHypergeom.test_rvsc                 C   s6   d}d}d}|}|}t jd|||}t|dd d S )Ni	  r     r6   gkㅒP?r  )r/   r  r   r   )r   Mr   NZtotZgoodZhgpmfr+   r+   r,   test_precision  s   zTestHypergeom.test_precisionc                 C   sl   t tjdddddd t tjdddddd t tjdddddd t tjdddddd d S )Nr   r6   r   r   r  rk   )r   r/   r  r   r   r+   r+   r,   	test_args  s   zTestHypergeom.test_argsc                 C   s4   t dtjdddd  kodk d S    d S )Nr   r   io i  if0  r   )r   r/   r  rB   r   r+   r+   r,   test_cdf_above_one  s   4z TestHypergeom.test_cdf_above_onec                    s   d dt g dd }d fdd|D }t g d}t||d	d
d g d}tj|   d}g d}t||d	d
d d S )Ng    +@g     @)r   ffffff@333333@r  gffffff@g@r        @     @c                    s"   g | ]}t j   |qS r+   )r/   r  r  )rG   ZeatenZorangesZpearsquantiler+   r,   rI   !  s    z1TestHypergeom.test_precision2.<locals>.<listcomp>)r   gpR(g㘍R2gJޱ p9g?Mf=
G?r   r   gƠ>r@  )g     @r  g     @g     @g     @)r   r  g)9g"1)rc   r   r   r/   r  r  )r   Zfruits_eatenresr[   Z	quantilesres2Z	expected2r+   r  r,   test_precision2  s   zTestHypergeom.test_precision2c                 C   sd   t ddd}| }tddg}tt|| }t|| t ddd}| }t|d d S )Nr  r   r   r   rk   )	r/   r  rN   rc   r   r   r   r   r   )r   Zhgr   r   r   r+   r+   r,   r   -  s   
zTestHypergeom.test_entropyc                 C   sl   d}d}d}d}t j||||}d}t||dd d}d	}d
}d}t j||||}d}t||dd d S )Nr
      cA    .A     j@gor   r   r   @  iX  ,  g_7	j$r   )r/   r  r  r   r   r<   r  r   r  resultr[   r+   r+   r,   r  9  s   zTestHypergeom.test_logsfc                 C   s   d}d}d}d}t j||||}d}t||dd d}d	}d
}d}t j||||}d}t||dd d}d	}d}d}t j||||}d}t||dd tg d}d	}d
}d}t j||||}tdd}t||dd d S )Nr   r  r  r  g)\Ur   r   r  r  r   r  g*@Vr   }      r  gר)r  r  r  )r/   r  r  r   rc   r   fullr  r+   r+   r,   test_logcdfQ  s8   zTestHypergeom.test_logcdfN)r   r   r   r   r   r  r  r  r  r   r  r  r+   r+   r+   r,   r     s    
	r   c                   @   s6   e Zd Zejdddgdd Zdd Zdd	 Zd
S )TestLoggammazx, c, sf)r  r  gQOu3;)r  r`   g\Z{0c                 C   s<   t j||}t||dd t j||}t||dd d S Nr  rm   )r/   loggammar  r   r
  )r   r?   r   r  r>   yr+   r+   r,   test_sf_isf  s   zTestLoggamma.test_sf_isfc                 C       t jdd}t|ddd d S )Nir6   g     @r  rm   )r/   r  r   r   )r   lpr+   r+   r,   r    s   zTestLoggamma.test_logpdfc                 C   sT   t g ddd}|D ]\}}}}}tjj|dd}t|||||gdd qd S )N)rf   g"~jg46<@g oŏ      @r   g	h"lxgQ?gZd;333333@g      (@g{/L@g??gгYҿgh|?5?r`  r   Zmsvkrr  r  r   )rc   r   reshaper/   r  r   )r   tabler   r   varskewZkurtcomputedr+   r+   r,   ru    s   zTestLoggamma.test_statsN)	r   r   r   r   r7  r  r!  r  ru  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ej	
dejdgdd Zej	
dejdgdd Zej	
dddgdd ZdS )TestLogisticc                 C   2   t dd}tj|}tj|}t|| d S Nr   r[  )rc   r   r/   logisticrB   r   r   r   r?   r   r   r+   r+   r,   r       zTestLogistic.test_cdf_ppfc                 C   r,  r-  )rc   r   r/   r.  r  r
  r   r/  r+   r+   r,   r!    r0  zTestLogistic.test_sf_isfc                 C   s4   d}d}t tjd| | t tj|| d S )Ng      <gg|EA@r   )r   r/   r.  r   r
  )r   r   Zdesiredr+   r+   r,   test_extreme_values  s   z TestLogistic.test_extreme_valuesc                 C   s*   t jg d}g d}t||dd d S )N)ir   r_   )gm\  .g9B.gO& $r?  rm   )r/   r.  r   r   )r   r  r[   r+   r+   r,   test_logpdf_basic  s   zTestLogistic.test_logpdf_basicc                 C   s"   t jddg}t|ddg d S )Nre   )r/   r.  r   r   r   r  r+   r+   r,   test_logpdf_extreme_values  s   z'TestLogistic.test_logpdf_extreme_valuesloc_rvs,scale_rvsr6   c                 C   sR   t jjd||d}dd }t|t j||fdj}t j|}t||dd d S )Nr`   r   r4   r5   c                 S   s   | \}}t |}tt|| | dt|| |   |d  }t|| | t|| | d t|| | d   | }||fS Nr   r6   )r   rc   r   r  )inputrv   r)   r*   r   x1x2r+   r+   r,   rw     s   z#TestLogistic.test_fit.<locals>.funcr0  gKH9rk  )r/   r.  r   r    	_fitstartr?   rr   r   )r   loc_rvs	scale_rvsrv   rw   Zexpected_solutionZ
fit_methodr+   r+   r,   test_fit  s   
zTestLogistic.test_fitc                 C   H   t jjd||d}|t j|fg}t j|i d }tt j|| d S Nr`   r7  r   )r/   r.  r   r<  _reduce_funcry   )r   r=  r>  rv   r   rw   r+   r+   r,   test_fit_comp_optimizer  s   z$TestLogistic.test_fit_comp_optimizer
testlogcdfTFc                 C   sH   t g d}|rtj|}ntj| }g d}t||dd d S )N)r  r3     r   r  )     g      g%h9fg?~T}%mgCx^&ْgV瞯<rm   )rc   r   r/   r.  r  r  r   )r   rD  r?   r   r[   r+   r+   r,   test_logcdfsf_tails  s   z TestLogistic.test_logcdfsf_tailsN)r   r   r   r  r!  r1  r2  r5  r   r7  r  rc   r   randr?  rC  rG  r+   r+   r+   r,   r+    s    

	r+  c                   @   r   )

TestLogserc                 C   r   r   r   r   r+   r+   r,   r     r   zTestLogser.setup_methodc                 C   r{  )Nr   r   r   r   r   r   )r/   logserr   r   r9   r   r   r   r   r   r   r   r   r   r+   r+   r,   r     r}  zTestLogser.test_rvsc                 C   s   t jdd}t|d d S )Nr  r  g&Y3)r/   rJ  r   r   r   r  r+   r+   r,   test_pmf_small_p	  s   zTestLogser.test_pmf_small_pc                 C      t jd}t|d d S )N:0yE>gW  ?)r/   rJ  r   r   rK  r+   r+   r,   test_mean_small_p  s   zTestLogser.test_mean_small_pN)r   r   r   r   r   rL  rO  r+   r+   r+   r,   rI    s
    rI  c                   @   sp   e Zd Zdd Zejdejej	gejde
jdgdd Zejdejd	fej	d
fgdd ZdS )TestGumbel_r_lc                 C   r   r   r   r   r+   r+   r,   r   #  r   zTestGumbel_r_l.setup_methodrH   r6  r6   c                 C   s@   |j d||d}|||fg}||i d }t||| d S rA  )r   r<  rB  ry   )r   rH   r=  r>  rv   r   rw   r+   r+   r,   rC  &  s   z&TestGumbel_r_l.test_fit_comp_optimizerz	dist, sgnr   r`  c                 C   s@   |t g d }||\}}t||d  t|ddd d S )N)r   r   r   r   r   r   r   gW  @g   @g3qtw>rI  rm   )rc   r   rr   r   )r   rH   Zsgnzr4   r5   r+   r+   r,   r?  2  s   zTestGumbel_r_l.test_fitN)r   r   r   r   r   r7  r  r/   gumbel_rgumbel_lrc   r   rH  rC  r?  r+   r+   r+   r,   rP  "  s    
rP  c                   @   s   e Zd Zdd Zdd Zejdejdddgejd	d
dgejdddgdd Z	ejdejdddgejd	d
dgejdddgdd Z
dd ZdS )
TestParetoc                 C   s  t  R t dt tjjddd\}}}}t|tj t|tj t|tj	 t|tj	 tjjddd\}}}}t|tj t|tj t|tj	 t|tj	 tjjddd\}}}}t|d t|tj t|tj	 t|tj	 tjjddd\}}}}t|d t|tj t|tj	 t|tj	 tjjd	dd\}}}}t
|d
 t
|d t|tj	 t|tj	 tjjddd\}}}}t
|d t
|d t|tj	 t|tj	 tjjddd\}}}}t
|d t
|d t
|dtd  t|tj	 tjjddd\}}}}t
|d t
|d t
|dtd  t|tj	 tjjddd\}}}}t
|d t
|d t
|dtd  t
|d W d    d S 1 s[w   Y  d S )Nr   rf   rq  rr  r   r  rp  rj         @g?grq@r         @gffffff?gK~?g      2@g۶m۶m?r$  UUUUUU?gqq?      $@      @g%I$I?gQ?gUUUUUU@grq?g88Nb@)r   r   r   r   r/   paretor   rc   r   r   r   rU  r   r  r  r>   r<   r+   r+   r,   ru  >  s`   











$zTestPareto.test_statsc                 C   s:   d}d}d}t jj||d|d}|| | }t|| d S )Ng    eAr6   r  r   r3   )r/   rZ  r  r   )r   r?   r*   r5   r   r[   r+   r+   r,   r  z  s   zTestPareto.test_sfz2ignore:invalid value encountered in double_scalars	rvs_shaper   r6   rvs_locr   	rvs_scaler   c                 C   s  t jjd|||d}t jj|dddd }t jj|dddd }t jj|dddd }||  kr;|  kr;dks>J  J t jjd|||d d}t jj|dd	\}}	}
t|
d |  t|d
d
t|d  tt	|d | d     t|	d d S )Nr`   r   r*   r5   r4   r   p=
ף?rz   r}   )rz   fix_b)rz   fbr6   rz   r   )
r/   rZ  r   rr   r   r  r   rc   r   r   )r   r\  r]  r^  rv   Zshape_mle_analytical1Zshape_mle_analytical2Zshape_mle_analytical3Zshape_mle_aZ	loc_mle_aZscale_mle_ar+   r+   r,   r?    s0   

zTestPareto.test_fitc                 C   s   t jjd|||d}|t j|fg}t j|i d }tt j|||d tt j||dd tt j||ddd tt j||d|d d	 d S )
Nr`   r_  r   rd  r   r  ra  r6   rz   r{   )r/   rZ  r   r<  rB  ry   )r   r\  r]  r^  rv   r   rw   r+   r+   r,   test_fit_MLE_comp_optimzer  s   
z%TestPareto.test_fit_MLE_comp_optimzerc                 C   s@   t tj tttjjg ddd tttjjg dddd d S )Nr   r6   rd  )r   r6   r   r   r   re  )r   r/   rZ  assert_raisesr   rr   r   r+   r+   r,   test_fit_warnings  s
   

zTestPareto.test_fit_warningsN)r   r   r   ru  r  r   r7  filterwarningsr  r?  rf  rh  r+   r+   r+   r,   rT  =  s    <

rT  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ejddg dfdddded ejgfdddded d gfd!g d"fgd#d$ Zd%d& Zd'S )(TestGenparetoc                 C   sl   dD ]}t |}tj|\}}t|d tt | qt d}tj|\}}t||gddg d S )Nr   rk   rk          rf   )	rc   asarrayr/   	genpareto_get_supportr   r   isposinfr   )r   r   r)   r*   r+   r+   r,   test_ab  s   


zTestGenpareto.test_abc                 C   s   t jdd}tddd}t||t j| t||t j| t||t j| tddd}t|	|t j	| d S )Nrk   r   r   rX  r   r   r_   )
r/   rn  rc   r   r   r8   r  rB   r  r   r   rvr?   r   r+   r+   r,   test_c0  s   zTestGenpareto.test_c0c                 C   s   t jdd}tddd}t||t j| t||t j| t||t j| tddd}t|	|t j	| t|
d	d d S )
Nr  rr  r   rX  r   rk   r   r_   r   )r/   rn  rc   r   r   r8   uniformrB   r  r   r   rs  r+   r+   r,   test_cm1  s   zTestGenpareto.test_cm1c                 C   s   t jdd}t|tj|tjgddg tt|	tj t jdd}t|tj|tjgddg tt|	tj t jdd}t|tj|tjgddg tt|	tj d S )Nr]   rr  rk   r   r  )
r/   rn  r   r8   rc   r   rB   r   isneginfr   r   rt  r+   r+   r,   
test_x_inf  s   """zTestGenpareto.test_x_infc                 C   s   t ddd}dD ]:}tj||}dD ]}tj||| }t||dd qtj||}dD ]}tj||| }t||dd q0q	d S )	Nr   r_   r   r   r`  r  g+r  rk  )r  r  )rc   r   r/   rn  r8   r   rB   )r   r?   r   Zpdf0dcZpdfcZcdf0Zcdfcr+   r+   r,   test_c_continuity  s   zTestGenpareto.test_c_continuityc              	   C      t jt jddddt jdddddd	t jdddd f }d
D ]}tj||}dD ]}tj||| }t||dd q*qd S Nr  rh   r]   baser   r   FZendpointr   )rk   r  r|  rk  )rc   r   r   r   r/   rn  r   r   )r   r   r   Zppf0r}  Zppfcr+   r+   r,   test_c_continuity_ppf     z#TestGenpareto.test_c_continuity_ppfc              	   C   r  r  )rc   r   r   r   r/   rn  r
  r   )r   r   r   Zisf0r}  Zisfcr+   r+   r,   test_c_continuity_isf  r  z#TestGenpareto.test_c_continuity_isfc              	   C   sj   t jt jddddt jdddddd	t jdddd f }d
D ]}ttjtj||||dd qd S )Nr  rh   r]   r  r   r   Fr  r   )rN  gC]r2rl   gV瞯Ҽrl   rk  )	rc   r   r   r   r   r/   rn  rB   r   )r   r   r   r+   r+   r,   test_cdf_ppf_roundtrip  s   z$TestGenpareto.test_cdf_ppf_roundtripc                 C   s    t jdddd}t|d d S )Ng    _Brh   r   r   gpEȜ)r/   rn  r  r   r4  r+   r+   r,   r  "  s   zTestGenpareto.test_logsfzc, expected_statsr   )r   r   r6   r  r   rW  gqq@r_   r6   gqq?g      ?g$I$	?rq?r  gll0@r`  )rf   UUUUUU?r   333333c                 C   s$   t jj |dd}t||ddd d S )Nrq  rr  r?  rl   r   )r/   rn  r   )r   r   expected_statsr  r+   r+   r,   ru  (  s   zTestGenpareto.test_statsc                 C   s   t jd}t|ddd d S )NrN  gvǼ
  ?r?  rm   )r/   rn  r(  r   )r   r  r+   r+   r,   test_var2  s   zTestGenpareto.test_varN)r   r   r   rq  ru  rw  rz  r~  r  r  r  r  r   r7  r  rc   rU  r   ru  r  r+   r+   r+   r,   rj    s(    

	


rj  c                   @   <   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd ZdS )TestPearson3c                 C   r   r   r   r   r+   r+   r,   r   9  r   zTestPearson3.setup_methodc                 C   s   t jjddd}tt|dk t|jjtd v  t jd}tt	|t
 t dd}tt	|tj t|jjtd v  tt|dk d S )Nr]   r   r   ZAllFloatrf   r   )r/   r%   r   r   r9   r   r   r   r   r   floatr   r   r   r+   r+   r,   r   <  s   zTestPearson3.test_rvsc                 C   s|   t jdg d}t|tg ddd t jdd}t|tdgdd t jg d	d}t|tg d
dd d S )Nr6   rk   r]   r)  )gtT?g+q?g?rI  rk  r  r]   ND}i?r  r  r`  r   r   )r  gj?gdC?gs?g'M?)r/   r%   r8   r   rc   r   r  r+   r+   r,   r   G  s   
zTestPearson3.test_pdfc                 C   sp   t jdg d}t|tg ddd t jdd}t|dgdd t jg d	d}t|g d
dd d S )Nr6   r  )grE?gTp]0?g~?rI  rk  r  r]   g E&"J?r  )giZ*J?gh+8w?gcK?g|<x6?gG?)r/   r%   rB   r   rc   r   r  r+   r+   r,   r   Q  s   

zTestPearson3.test_cdfc                    s@   g d}dd t j|} fdd|D }t|| d S )N)r  r`  r   rf   rf   r  c                    s$   g | ]}t t|j d  qS rE   )r   r/   r%   r8   )rG   r)  Zneg_infZx_evalr+   r,   rI   b  s    z<TestPearson3.test_negative_cdf_bug_11186.<locals>.<listcomp>)r/   r%   rB   r   )r   Zskewsr  Zint_pdfsr+   r  r,   test_negative_cdf_bug_11186[  s   z(TestPearson3.test_negative_cdf_bug_11186c                 C   sP   t jdd}t|d tt|t t jdd}t|d tt|t d S )Nr   r6   r   rI  )r/   r%   rK  r   rt   r  )r   rK  r+   r+   r,   test_return_array_bug_11746f  s   

z(TestPearson3.test_return_array_bug_11746N)	r   r   r   r   r   r   r   r  r  r+   r+   r+   r,   r  8  s    

r  c                   @   sD   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S )
TestKappa4c                 C   sD   g d}d}dD ]}t j|||}t j|| }t|| qd S )N)rk   r]   r)  rf   r   )
gffffffr  r/  皙ɿ皙r]   r)  rf   r   ffffff?)r/   kappa4rB   rn  r   r   r?   r   r<   r   Z	vals_compr+   r+   r,   test_cdf_genparetos  s   zTestKappa4.test_cdf_genparetoc                 C   sL   t ddd}d}t ddd}tj|||}tj||}t|| d S )Nrg  r   r_   rk   r  r   )rc   r   r/   r  rB   
genextremer   r  r+   r+   r,   test_cdf_genextreme~  s   zTestKappa4.test_cdf_genextremec                 C   s@   t ddd}d}d}tj|||}tj|}t|| d S )Nr   r_   r   rk   )rc   r   r/   r  rB   r  r   r  r+   r+   r,   test_cdf_expon     zTestKappa4.test_cdf_exponc                 C   @   t ddd}d}d}tj|||}tj|}t|| d S )Nrg  r   r_   rk   )rc   r   r/   r  rB   rR  r   r  r+   r+   r,   test_cdf_gumbel_r  r  zTestKappa4.test_cdf_gumbel_rc                 C   s@   t ddd}d}d}tj|||}tj|}t|| d S )Nrg  r   r_   r  rk   )rc   r   r/   r  rB   r.  r   r  r+   r+   r,   test_cdf_logistic  r  zTestKappa4.test_cdf_logisticc                 C   r  )Nrg  r   r_   r   )rc   r   r/   r  rB   rv  r   r  r+   r+   r,   test_cdf_uniform  r  zTestKappa4.test_cdf_uniformc                 C   s   t dd d S r8  )r/   r  r   r+   r+   r,   test_integers_ctor  s   zTestKappa4.test_integers_ctorN)
r   r   r   r  r  r  r  r  r  r  r+   r+   r+   r,   r  r  s    					r  c                   @   r   )TestPoissonc                 C   r   r   r   r   r+   r+   r,   r     r   zTestPoisson.setup_methodc                 C   s@   t d}tjg d|}d|d |d d g}t|| d S )Nr6   r   r   r6   rf   r  )rc   r   r/   poissonr   r   )r   Zln2r   r[   r+   r+   r,   test_pmf_basic  s   
zTestPoisson.test_pmf_basicc                 C   s@   t jg dd}g d}t|| t jdd}t|d d S )Nr  r   )r   r   r   ffffff?r   r   )r/   r  r   r   intervalr   )r   r   r[   r  r+   r+   r,   test_mu0  s
   
zTestPoisson.test_mu0c                 C   r{  )Nrf   r   r   r   r   r   )r/   r  r   r   r9   r   r   r   r   r   r   r   r   r   r+   r+   r,   r     r}  zTestPoisson.test_rvsc                 C   s   d}t jj |dd}t|||td| d| g tg d}t jj |dd}||tjddtd gtjddgf}t|| d S )	Ng      0@rq  rr  r   )rk   r   rj   r   r6   rf   )r/   r  r   rc   rU  r   r   )r   r2  r  r[   r+   r+   r,   ru    s    &zTestPoisson.test_statsN)r   r   r   r   r  r  r   ru  r+   r+   r+   r,   r    s    	r  c                   @   l   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S )	TestKSTwoc                 C   r   r   r   r   r+   r+   r,   r     r   zTestKSTwo.setup_methodc                 C   s   dD ]W}t dd| d| ddd|  dg}d| | }tj|d }|dkr.t |nd}t dd|| ddtjd|  t	dd|  ddg}tj
||}t|| qd S )Nr   r6   r   r_   r`   r  r   rf   r   r   r6   rk   )rc   r   r.   r   gammalnr  r/   ksoner  r  kstworB   r   )r   r   r?   v1lgelgr[   rm  r+   r+   r,   r     s   &zTestKSTwo.test_cdfc                 C   s   t ddd}dD ]U}t dd| d| ddd|  dg}d| | }tj|d }|dkr5t |nd}t ddd||  dtj	d| t
d| ddg}tj	||}t|| q	d S )Nr   r   r  r  rf   r   r6   )rc   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   &
zTestKSTwo.test_sfc                 C   s\   t ddddd  }g d}|D ]}|t | }tj||}t |}t|d qd S )Nr   r6   r  r   )r   r`      i  r  i  rN  )rc   r   rU  r/   r  rB   Zdiffr	   )r   r?   nsZ_xxnprobsZdiffsr+   r+   r,   test_cdf_sqrtn  s   
zTestKSTwo.test_cdf_sqrtnc                 C   sF   t ddd}dD ]}tj||}tj||}t|d|  q	d S Nr   r   r  r  )rc   r   r/   r  rB   r  r   )r   r?   r   rm  r  r+   r+   r,   r    s   zTestKSTwo.test_cdf_sfc                 C   sT   t ddd}dD ]}|t | }tj||}tj||}t|d|  q	d S r  )rc   r   rU  r/   r  rB   r  r   )r   r?   r   r  rm  r  r+   r+   r,   test_cdf_sf_sqrtn  s   zTestKSTwo.test_cdf_sf_sqrtnc                 C   n   t ddd}dD ]+}||d| k }tj||}d|k |dk @ }tj||}t|| || dd q	d S )	Nr   r   r  r  rf   rP  r  rm   )rc   r   r/   r  rB   r   r   r   r?   r   r  rm  condr   r+   r+   r,   test_ppf_of_cdf  s   zTestKSTwo.test_ppf_of_cdfc                 C   r  )	Nr   r   r  r  rf   r   r  rm   )rc   r   r/   r  r
  r  r   )r   r?   r   r  Zvals_isfr  r   r+   r+   r,   test_isf_of_sf(  s   zTestKSTwo.test_isf_of_sfc                 C   t   t ddd}dD ].}|t | |d| k }tj||}d|k |dk @ }tj||}t|| ||  q	d S )Nr   r   r  r  rf   r   )rc   r   rU  r/   r  rB   r   r   r  r+   r+   r,   test_ppf_of_cdf_sqrtn1  s   zTestKSTwo.test_ppf_of_cdf_sqrtnc                 C   r  )Nr   r   r  r  rf   r  )rc   r   rU  r/   r  r  r
  r   )r   r?   r   r  r  r  r   r+   r+   r,   test_isf_of_sf_sqrtn:  s   zTestKSTwo.test_isf_of_sf_sqrtnc                 C   sJ   t ddddd  }dD ]}tj||}tj||}t|| qd S r  )rc   r   r/   r  r   rB   r   )r   r  r   r  rm  r+   r+   r,   ry  D  s   zTestKSTwo.test_ppfc              	   C   s   g d}t g d}t g dg dg dg dg dg dg}t|D ]%\}}|t d	 t t jd	 |  }tj||}t	||| d
d q&d S )N)r_   r   r`   r  r  r  )r   gUUUUUU?rf   r   r6   r   )gRT>gr	?gK5
?gn
"5?gU
?gk?)g&6#>gb
E>gI5?g\?g;?g%1?)g6>gL>g<?g䓀^?gO|6$?g qO?)g}+: >gwQ9>gH{?g]qv(?g)~/v?gmJ?)gV27J=gGk(>gWX?gY?g{0`?gZ
H?)g_R=g4>g̪?gT_@?g<e?g2?r6   h㈵>rm   )
rc   r   	enumerater   rU  r:   r/   r  rB   r   )r   r  Zratiosr[   idxr   r?   rm  r+   r+   r,   test_simard_lecuyer_table1K  s   "z$TestKSTwo.test_simard_lecuyer_table1N)r   r   r   r   r   r  r  r  r  r  r  r  r  ry  r  r+   r+   r+   r,   r    s    
		
r  c                   @   r   )TestZipfc                 C   r   r   r   r   r+   r+   r,   r   n  r   zTestZipf.setup_methodc                 C   r{  )Nr  r   r   r   r   r   )r/   zipfr   r   r9   r   r   r   r   r   r   r   r   r   r+   r+   r,   r   q  r}  zTestZipf.test_rvsc                 C   s\   t jj dd\}}tt| t|tj t jj ddd\}}tt||g   d S )Nffffff@r)   333333@skr)   rs  )r/   r  r   rc   isfiniter   r   r   r[  r+   r+   r,   r  |  s
   zTestZipf.test_momentsN)r   r   r   r   r   r  r+   r+   r+   r,   r  m  r   r  c                   @   r   )
TestDLaplacec                 C   r   r   r   r   r+   r+   r,   r     r   zTestDLaplace.setup_methodc                 C   s   t jjddd}tt|dk t|jjtd v  t jd}tt	|t
 t dd}tt	|tj t|jjtd v  tt jdd u d S )Nr  r   r   r   r   r  )r/   dlaplacer   r   r9   r   r   r   r   r   r   r   r   r+   r+   r,   r     s   zTestDLaplace.test_rvsc                 C   s   d}t |}| d\}}}}d}t| |d }||}	t|	|d  t|	|d  }
}t||fd t||f|
||
d  d fd	d
d d S )Nr   rq  %   r   r6   r  r  rp  r  rN  r@  )r/   r  rc   r   r   r   r   r   )r   r)   dlr  r  r>   r<   r  r   ppm2Zm4r+   r+   r,   ru    s   

&(zTestDLaplace.test_statsc                 C   sF   t d}t|}|d\}}}}t||fd t||fd d S )Nrj   rq  )rk   rk   )r$  g      
@)rc   r   r/   r  r   r   )r   r)   r  r  r  r>   r<   r+   r+   r,   test_stats2  s
   

zTestDLaplace.test_stats2N)r   r   r   r   r   ru  r  r+   r+   r+   r,   r    s
    r  c                       s   e Zd Zdd Zejddej	dd gdd Z
ejddej	dd g fd	d
Zdd Zdd Zdd Z  ZS )TestInvgaussc                 C   r   r   r   r   r+   r+   r,   r     r   zTestInvgauss.setup_methodzrvs_mu,rvs_loc,rvs_scale)r6   r   r   r   r_   c                 C   sB  t jjd|||d}t jj||d\}}}|| }t|}t|t|d |d   }	||	 }
t|
|ddd t|	|ddd t	|| t jjd|||d}t jj||d |d d\}}}t	|d | t	|d | t jj|d	d
d }t jj|d	dd }t jj|d	dd }||  kr|  krd	ksJ  J d S )Nr`   r   r2  r4   r5   rd  r`  rl   r@  r   re  r`  )fmur   )Zfix_mur}   )
r/   r   r   rr   rc   r   r   r   r   r   )r   rvs_mur]  r^  rv   r2  r4   r5   Zmu_temp	scale_mleZmu_mleZ
shape_mle1Z
shape_mle2Z
shape_mle3r+   r+   r,   r?    s.   



*zTestInvgauss.test_fitc           
         s  t jjd|||d}ttt jt jj}||}t j|}t|| ||ddd}t jj|ddd}t|| |t j|fg}t j|i d }	t	t j||	|d t
||d  dks`J t	t j||	|d d t	t j||	dd t	t j||	|t
jdd d d S )	Nr`   r  r   r6   )rz   r  r   rd  re  )r/   r   r   rs   rt   rr   r   r<  rB  ry   rc   r   r   rH  )
r   r  r]  r^  rv   Z	super_fitZsuper_fittedZinvgauss_fitr   rw   	__class__r+   r,   rf    s*   




z'TestInvgauss.test_fit_MLE_comp_optimzerc                 C   sN   t tj tt tjjg ddd W d    d S 1 s w   Y  d S )Nr   r6   rd  )r   r/   r   r   r   r   rr   r   r+   r+   r,   test_fit_raise_errors  s   
"z"TestInvgauss.test_fit_raise_errorsc                 C   s   g d}g d}t jjd|d}t|| t jjddd}t|d t jjddd}t|d	 t jd
d}t|d t jdd}t|d d S )N)g4wT;?gjdV&}?g{i\0>gDV_h?gqacX?)r   r   r   r   r   皙?r2  rw  ?gfFn   gr^:g9̗?r  gx_;:g!?gٷ
??)r/   r   rB   r   r   r  )r   r2  r[   rY   Z
cdf_actualZ	sf_actualr+   r+   r,   r    s   



zTestInvgauss.test_cdf_sfc                 C   sh   t jjddd}t|d t jdd}t|d t jjddd}t|d t jdd}t|d	 d S )
Nr  r  r  giJ.r  gr^ߺrw  gpgFgpQ^L)r/   r   r  r   r  )r   r  r  r+   r+   r,   r    s   	


zTestInvgauss.test_logcdf_logsf)r   r   r   r   r   r7  r  rc   r   rH  r?  rf  r  r  r  __classcell__r+   r+   r  r,   r    s    
&#r  c                       sz   e Zd Zejdg dejdg ddd Zejdg d fd	d
Zdd Zdd Z	dd Z
dd Z  ZS )TestLaplacer]  )rg  r   r   r6   r^  )r   r6   r   r_   c                 C   sL  t jjd||d}t|}tt|| t| }t j|\}}t	||ddd t	||ddd t jj||d\}}t	||ddd t jj||d\}}t	|| |d }tt|| t| }t jj||d\}}t
|| t jj||d\}}t
|| ttt jj|||d ttt jjtjg ttt jjtjg d S )	Nr`   r7  rl   r@  rd  r{   r6   re  )r/   r]  r   rc   medianr   absr   rr   r   r   rg  r   r   r   )r   r]  r^  rv   Zloc_mler  r4   r5   r+   r+   r,   r?  2  s*   



zTestLaplace.test_fitzrvs_scale,rvs_loc))r_   rg  )r   r_   )r)  rf   c                    s   t jjd||d}dd }t j|\}}ttt jt j|\}}||||}	||||}
|	|
k s>tj|	|
ddds@J d S d S )Nr  r7  c              	   S   s8   dt | td|  d| tt||     S )Nr`  r6   r   )r   rc   r   r   r  )r4   r5   rv   r+   r+   r,   llf  s   z2TestLaplace.test_fit_MLE_comp_optimzer.<locals>.llrl   r@  )r/   r]  r   rr   rs   rt   rc   ru   )r   r]  r^  rv   r  r4   r5   Zloc_optZ	scale_optZll_mleZll_optr  r+   r,   rf  _  s   
z&TestLaplace.test_fit_MLE_comp_optimzerc                 C   sZ   t g d}tjj|dd\}}t|dddd tjj|dd\}}t|dddd d S )N)r   r   rp  r          @g      ,@r  rd  r  rl   r@  r  )rc   r   r/   r]  rr   r   )r   rv   r4   r5   r+   r+   r,   test_fit_simple_non_random_datat  s
   z+TestLaplace.test_fit_simple_non_random_datac                 C   sl   d}t j| }|dksJ t j|}|dksJ t j|}|dks'J t j| }|dks4J d S )Nr  rk   r   )r/   r]  rB   r  )r   r?   Zp0p1r+   r+   r,   test_sf_cdf_extremes}  s   z TestLaplace.test_sf_cdf_extremesc                 C   s.   d}t j|}t|t| d dd d S )Nr  r6   r?  rm   )r/   r]  r  r   rc   r  )r   r?   r   r+   r+   r,   r       zTestLaplace.test_sfc                 C   s.   d}t j|}t|td|  dd d S )Ng}:r6   r?  rm   )r/   r]  r
  r   rc   r   )r   r   r?   r+   r+   r,   r    r  zTestLaplace.test_isf)r   r   r   r   r7  r  r?  rf  r  r  r  r  r  r+   r+   r  r,   r  1  s    +	r  c                   @   r   )TestInvGammac                 C   s   t  P t dt tjjddd}g d}t|| g d}tjj|dd}g dtjdd	gtj	d
dgtj	tj	dgf}t
||D ]	\}}t|| qBW d    d S 1 sWw   Y  d S )Nr   g(\O3@rq  r  )g~?gCX&?g*7gS?gBqFq @)皙?@gffffff@)rX  gEy?gў3oz?g^Yb?g,r?gOD@g2*Z@gFN8@)r   r   r   r   r/   r   r   rc   r   r   r   r   )r   rq  r[   r)   r?   r   r+   r+   r,   test_invgamma_inf_gh_1866  s    



"z&TestInvGamma.test_invgamma_inf_gh_1866c                 C   s6   t dd}tj|d}tj|d}t|| d S )Ngr   r   )rc   r   r/   r   rB   r   r   r/  r+   r+   r,   r    s   zTestInvGamma.test_cdf_ppfc                 C   sR   t jdkrtdd}ntdd}tj|d}tj|d}t||dd d S )Nl        r6   r`      r   r   rm   )	sysmaxsizerc   r   r/   r   r  r
  r   r/  r+   r+   r,   r!    s   
zTestInvGamma.test_sf_isfN)r   r   r   r  r  r!  r+   r+   r+   r,   r    s    r  c                   @   r   )
TestFc                 C   s\   t jddgg}|D ]\}}}|j|jg|R  }q	dd |D }dd |D }t|| d S )N)r6   r   r   c                 S   &   g | ]\}}}|j |jg|R  qS r+   r8   r)   rG   _f_argsr$  r+   r+   r,   rI        & z(TestF.test_endpoints.<locals>.<listcomp>c                 S      g | ]\}}}|qS r+   r+   rG   r  r  Z	_correct_r+   r+   r,   rI         )r/   fr8   r)   r   )r   rv   r  r  Z_correctanscorrectr+   r+   r,   r     s   zTestF.test_endpointsc                 C   sX   t jj dddd\}}}}tt| tt| tt| tt|  d S )Nr        @rq  rr  )r/   r  r   rc   r  r[  r+   r+   r,   test_f_moments  s
   zTestF.test_f_momentsc                 C   sV   t   t dt tjjdgd g ddd W d    d S 1 s$w   Y  d S )Nr   r  r  )r6   r  r  rh  rq  dfndfdrs  )r   r   r   r   r/   r  r   r+   r+   r,   test_moments_warnings  s   
"zTestF.test_moments_warningsc                 C   sD  t dgdgg}t ddg}tjj||dd\}}}}||d  gd }t|| d|d  || d  | |d d  |d  }t|| d| | d t d|d   |d	 t ||| d    }	t||	 d|d
| d  || d  |d |d d    }
||d	  |d  || d  }|
| }t|| d S )Nr   r  r   rq  r  r6   r  rh  r  r      )rc   r   r/   r  r   rU  )r   r  r	  r  r  r>   r<   r  Zv2s2Zk2numZk2denZk2r+   r+   r,   test_stats_broadcast  s"   
0
 
 zTestF.test_stats_broadcastN)r   r   r   r   r  r
  r  r+   r+   r+   r,   r    s
    
r  c                   C   s   t tjddgddg d S )Nr   r  g?g?)r   r/   rW  r   r+   r+   r+   r,   test_rvgeneric_std  s   r  c                   C   s   t tjjdddtjtjtjtjf t tjjddddtjtjtjf t tjjddddtjtjtjf t tjjdddddtjtjf t tjjd	d
dtjtjf t tjjdd
ddtjf t tjjdd
ddtjf t tjjdd
dd d S )Nr   rq  )r  rs  )\(?rk   r6   gGz @g     i@r   r  gGz@r  g
ףp=
@)rk   gq    @)r   r/   rW  rc   r   r   r+   r+   r+   r,   test_moments_t  s    r  c                  C   s*   g d} g d}t tj| |dd d S )N)r   r6      r`   )g?@gTM]?ggY?gU?r?  rm   )r   r/   rW  rN   )r  r[   r+   r+   r,   test_t_entropy	  s   r  c                   @   r  )TestRvDiscretec                 C   r   r   r   r   r+   r+   r,   r   	  r   zTestRvDiscrete.setup_methodc                 C   s   g d}g d}d}t jd||fd}|j|d}tt|tj t||D ]\}}ttt	||kt
| | dk  q&| }tt|t d S )N)r`  r   r   r6   r   r  )rk   333333?r  rk   r  rk   r  sample)namevaluesr   r  )r/   rK   r   r   r   r9   r   r   r  r   r  r   )r   ZstatesZprobabilitysamplesrr?   r>   r   r+   r+   r,   r   	  s   &zTestRvDiscrete.test_rvsc                 C   sr   t g d}tjg d|fd}tt|| }| }t|| tjg dg dfd}| }t|d d S )N)r   g?r  r  r  )r   r   r   rk   )	rc   r   r/   rK   r   r   rN   r   r   )r   r  r   r   r   r+   r+   r,   r    	  s   
zTestRvDiscrete.test_entropyc                 C   sT   g d}g d}t j||fd}ddgddgg}t||dd	gd
dggdd d S )Nr   r6   r  rf   r  r)  r  r   r$  rp  r6   rf   r)  rk   r  r  rk  )r/   rK   r   r   )r   xkpkrt  r?   r+   r+   r,   r   ,	  s   

zTestRvDiscrete.test_pmfc                    d   g d}g d}t j||fd g d}g d}t ||dd t fdd	|D |dd d S )
Nr  r  r  )r  r   r  r  rj   rp  r  r   )r   rf   rf   rf   r  r  r   r   r  rk  c                       g | ]}  |qS r+   )rB   )rG   r   rt  r+   r,   rI   A	  r  z+TestRvDiscrete.test_cdf.<locals>.<listcomp>)r/   rK   r   rB   )r   r  r  Zx_valuesr[   r+   r!  r,   r   7	     
zTestRvDiscrete.test_cdfc                    r  )
Nr  r  r  )r]   rf   333333?r  r'  r   )r   r   r6   r6   r  r  r  rk  c                    r   r+   )r   )rG   r   r!  r+   r,   rI   N	  r  z+TestRvDiscrete.test_ppf.<locals>.<listcomp>)r/   rK   r   r   )r   r  r  Zq_valuesr[   r+   r!  r,   ry  D	  r"  zTestRvDiscrete.test_ppfc                 C   sN   g dg df}t j|d}t|||jd d d |jdd   d S )N)r   r6   r  r  rh  )r]   r)  r  r  r]   r  r`  rN  r   )r/   rK   r   r   rB   r  )r   r   rt  r+   r+   r,   test_cdf_ppf_nextQ	  s
   z TestRvDiscrete.test_cdf_ppf_nextc                 C   s`   t dd}t g dg dg dg}tj||fd}t| t |j	|j
 dd d S )Nr   )r   r  )r]   r]   r1  r  )r]   r]   r  r  r  r  rk  )rc   r   r&  r   r/   rK   r   expectr   r  r  r   r  r  rt  r+   r+   r,   test_multidimensionY	  s   
$z"TestRvDiscrete.test_multidimensionc                 C   s   g d}ddg}t ttjfi t||fd g d}t ttjfi t||fd g d}g d}t ttjfi t||fd g d}g d}t ttjfi t||fd d S )Nr   rf   r  )rf   333333?gffffff濩r   r6   r   r  r   )r  r  r  r  r  )rg  
ValueErrorr/   rK   r   r   r  r  r+   r+   r,   test_bad_inputb	  s   "zTestRvDiscrete.test_bad_inputc                 C   s   t ddt dd}}tttjfi t||fd t ddt dd}}tttjfi t||fd t ddt dd}}t	tj||fd
dd d S )	Nr  r6   r6   )r6   r   gUUUUUU?r  r  )r   r6   r   )rc   r   r&  r  rg  r*  r/   rK   r   r   r   r+  r+   r+   r,   test_shape_rv_sampler	  s    z#TestRvDiscrete.test_shape_rv_samplec                 C   sD   g d}g d}t j||fd}t| t|j|j dd d S )N)r   r6   r  r  r  r  )r]   r)  r)  r)  r)  r]   r  r  rk  )r/   rK   r   r%  rc   r   r  r  r&  r+   r+   r,   test_expect1	  s   $zTestRvDiscrete.test_expect1c                 C   s   g d}g d}t j||fd}t| | dd t| tdd t||D dd t|dd	 td
d t||D dd d S )N)/g      i@g     r@g      y@g     @@g     @g     @      @g      @r  g     0@g     @g     P@g     @g     p@g      @g     @g      @g     @r  g     h@g     0@g     @g     @g     @g     P@g     @g     @g     @r  g     8@g      @g     ȩ@g     @g     X@g      @g     @g     @g     x@g     @@g     @g     h@g     ̰@g     0@g     @g     @g     \@g     @)/g-C6:?rk   gF%uk?g:vz?rk   rk   g.nr?g|гY?rk   rk   rk   g @6*? 4U0*C?g@+?gC6z?g nr?g ?gOjM?g n?g1w-!?g:pΈ?gףp=
?g _Le?rk   g^)p?g~jtx?gJ4q?gGzt?g J4q?g Mb@?g@Psׂ?g@^)ˀ?rk   rk   g N@s?g 1w-!_?r1  gൄ|г?rk   gH}}?rk   g
F%u?rk   rk   g ^)ˀ?g v?rk   r  r  rk  c                 s   s    | ]	\}}|| V  qd S rb   r+   rG   r  wr+   r+   r,   	<genexpr>	  s    z.TestRvDiscrete.test_expect2.<locals>.<genexpr>c                 S      | d S Nr6   r+   r?   r+   r+   r,   rT   	      z-TestRvDiscrete.test_expect2.<locals>.<lambda>c                 s   s     | ]\}}|d  | V  qdS )r6   Nr+   r2  r+   r+   r,   r4  	      )r/   rK   r   r%  r   r   r   )r   r   pyrt  r+   r+   r,   test_expect2	  s   
zTestRvDiscrete.test_expect2N)r   r   r   r   r   r   r   r   ry  r$  r'  r,  r.  r/  r;  r+   r+   r+   r,   r  	  s    	r  c                   @   r  )TestSkewCauchyc                 C   sl   t ddd}ttjj|ddtj| ttjj|ddtj| ttjj|ddtj| d S Nrg  r   r`   r   r  )	rc   r   r   r/   
skewcauchyr8   rY  rB   r   r  r+   r+   r,   test_cauchy	  s   


zTestSkewCauchy.test_cauchyc                 C   s   t jd t jdd d }t jdd d }g d}g d}ttj||| ttj||| ttj	||| d S )Nr   r_   r6   r   r   )
g& 45?gQʶ?g-4<??gJ?gw@v?gJ%?g>[u?g
a2?g	Rn?gѦ}cي?)
gN@Q?gPPz@	?g?g;
7?glg?g!?%?gQҙ?gC"?go?g@Sg?)
rc   r   r   rH  r   r/   r>  r8   rB   r   )r   r)   r?   r8   rB   r+   r+   r,   test_skewcauchy_R	  s   z TestSkewCauchy.test_skewcauchy_RN)r   r   r   r?  r@  r+   r+   r+   r,   r<  	  s    	r<  c                   @   r  )TestSkewNormc                 C   s   t d| _d S r   )r   rngr   r+   r+   r,   r   	     zTestSkewNorm.setup_methodc                 C   s0   t ddd}ttjj|ddtj| d S r=  )rc   r   r   r/   skewnormr8   r  r  r+   r+   r,   test_normal	  s   
zTestSkewNorm.test_normalc                 C   sH   d}t jjd|| jd}t||j t jjd|| jd}t||j d S )N)r   r  r   r   )r)   r   r  r  )r/   rD  r   rB  r   r   )r   r   r?   r+   r+   r,   r   	  s
   zTestSkewNorm.test_rvsc                 C   s   t jjdtddd| jd}t|t|t |t 	|g}t jj ddddd}t
||dd t jjd	tddd| jd}t|t|t |t 	|g}t jj d	dddd}t
||dd d S )
Nr  r  r   r6   )r)   r   r4   r5   r  rq  )r)   r4   r5   rs  r   r   )r/   rD  r   r   rB  rc   r   r(  r)  Zkurtosisr   )r   Xr[   r*  r+   r+   r,   r  	  s   $$zTestSkewNorm.test_momentsc                 C   sF   t jg dd}t|tddd t jdd}t|ddd d S )	N)r_   r[  r   r`  r   r  rm   r  rU  r   )r/   rD  rB   r   rc   onesr   r+   r+   r,   test_cdf_large_x	  s   zTestSkewNorm.test_cdf_large_xc                 C   sr   g dg dg dg dg dg}|D ]#\}}}t j||}t||dd t j| | }t||dd qd S )N)r   gzbLe9)r   r6   gn'/2;)r  r   g::)r  r`  g|<)r>  r   Oul2;rN  rm   )r/   rD  rB   r   r  )r   Zcdfvalsr?   r)   Zcdfvalr   r+   r+   r,   test_cdf_sf_small_values
  s   z%TestSkewNorm.test_cdf_sf_small_valuesN)	r   r   r   r   rE  r   r  rH  rK  r+   r+   r+   r,   rA  	  s    	rA  c                   @   r   )
	TestExponc                 C   s   t tjdd d S r   )r   r/   r  r8   r   r+   r+   r,   	test_zero
  s   zTestExpon.test_zeroc                 C   s0   t tjdd t tjtjdd d S )NgC]r2<r  )r   r/   r  rB   r
  r  r   r+   r+   r,   	test_tail
  s   zTestExpon.test_tailc                 C   ,   t dddddt jg}tttjj| d S No_?46@yX5ͻ@j+?UX@)rc   r   r   rg  r   r/   r  rr   r  r+   r+   r,   test_nan_raises_error
     zTestExpon.test_nan_raises_errorc                 C   rO  rP  )rc   r   r   rg  r   r/   r  rr   r  r+   r+   r,   test_inf_raises_error"
  rW  zTestExpon.test_inf_raises_errorN)r   r   r   rM  rN  rV  rX  r+   r+   r+   r,   rL  
  s
    rL  c                   @   r   )TestNormc                 C   rO  rP  )rc   r   r   rg  r   r/   r  rr   r  r+   r+   r,   rV  )
  rW  zTestNorm.test_nan_raises_errorc                 C   rO  rP  )rc   r   r   rg  r   r/   r  rr   r  r+   r+   r,   rX  .
  rW  zTestNorm.test_inf_raises_errorc                 C   s    g d}t ttjj|dd d S )Nr   shrimp)plate)rg  r   r/   r  rr   r  r+   r+   r,   test_bad_keyword_arg3
  s   zTestNorm.test_bad_keyword_argN)r   r   r   rV  rX  r\  r+   r+   r+   r,   rY  (
  s    rY  c                   @   s    e Zd ZdZdd Zdd ZdS )TestUniformgh-10300c                 C   rO  rP  )rc   r   r   rg  r   r/   rv  rr   r  r+   r+   r,   rV  :
  rW  z!TestUniform.test_nan_raises_errorc                 C   rO  rP  )rc   r   r   rg  r   r/   rv  rr   r  r+   r+   r,   rX  ?
  rW  z!TestUniform.test_inf_raises_errorN)r   r   r   __doc__rV  rX  r+   r+   r+   r,   r]  8
  s    r]  c                
   @   s   e Zd Zdd Zdd Zdd Zdd Zej	d	g d
dd Z
ej	dg dg dg dg dg dgdd Zej	dg dg dg dg dg dg dgdd ZdS )TestExponNormc                 C   s   dd }d\}}}d||  }t jj |||dd}t||||| d\}}}d||  }t jj |||dd}t||||| d\}}}d||  }t jj |||dd}t||||| d	\}}}d||  }t jj |||dd}t||||| d S )
Nc                 S   sh   dd| | d   }d| | d  |d  }dd| | d  d  }|d|   || d| |    ||gS )Nr   r   r6   r   r-        @r  r+   )lamsigr2  ZopK2Zexp_skewZexp_kurtr+   r+   r,   get_momsH
  s   $z,TestExponNorm.test_moments.<locals>.get_moms)r   r   r   r   rq  r4   r5   rs  )r  r6   r]   )r   r   r   )rg  r  rV  )r/   	exponnormr   )r   rd  r2  rc  rb  Kstsr+   r+   r,   r  F
  s"   



zTestExponNorm.test_momentsc                 C   2   t dddddt jg}tttjj|ddd d S 	NrQ  rR  rS  rT  rU  r   r   re  )rc   r   r   rg  r   r/   rf  rr   r  r+   r+   r,   rV  a
     z#TestExponNorm.test_nan_raises_errorc                 C   ri  rj  )rc   r   r   rg  r   r/   rf  rr   r  r+   r+   r,   rX  f
  rk  z#TestExponNorm.test_inf_raises_errorc                 C   sT   t tjddd t tjddd t tjddd t tjddd d S )Ni|r   rk     rh   )r   r/   rf  r8   r   r+   r+   r,   test_extremes_xk
  s   zTestExponNorm.test_extremes_xzx, K, expected))r[  rh   g6
N-)r   rh   g՜n+H?)r`  rh   gi?)r   rh   gbJI-)r_   r   gI8?)r_   r  gQ3|-0?c                 C   s   t tj|||dd d S r  )r   r/   rf  r8   )r   r?   rg  r[   r+   r+   r,   test_std_pdf}
  s   zTestExponNorm.test_std_pdfzx, K, scale, expected)r   rh   r   gVAҤ?)rg  {Gzt?r   g'^>)rF  rh   r`   rk   )rF  rh   r  gbr ;)r   rw  r   gVMe?c                 C   <   t jj|||d}|dkr|dksJ d S t||dd d S )Nr  rk   r?  rm   )r/   rf  rB   r   r   r?   rg  r5   r[   r   r+   r+   r,   test_cdf_small_K
  s   zTestExponNorm.test_cdf_small_K)r_   rh   r   ghG}$;)r6   ro  r   g]fJ?)r   ro  rf   g31"g#;)r_   ro  rf   gf
+-)r[  ro  rf   rk   )r  rw  r   ga9S?c                 C   rp  )Nr  rk   g-a=rm   )r/   rf  r  r   rq  r+   r+   r,   test_sf_small_K
  s   zTestExponNorm.test_sf_small_KN)r   r   r   r  rV  rX  rm  r   r7  r  rn  rr  rs  r+   r+   r+   r,   r`  E
  s8    

r`  c                   @   r   )TestGenExponc                 C   s@   ddl m} tjtdddddd}t||dddd d S )	Nr   )simpsr_   rh   rf   rj   )Zdxr   )scipy.integrateru  r/   genexponr8   r9   r   r   )r   ru  r   r+   r+   r,   test_pdf_unity_area
  s   z TestGenExpon.test_pdf_unity_areac                 C   s:   t jtdddddd}ttd|k|dk@  d S )Nr   r_   rh   rf   rj   r   )r/   rw  rB   r9   r   r   r   )r   rB   r+   r+   r,   test_cdf_bounds
  s   zTestGenExpon.test_cdf_boundsc                 C   s$   t jdddd}t|ddd d S )Nr   r   r6   r  gM <r?  rm   )r/   rw  r  r   r  r+   r+   r,   test_sf_tail
  s   zTestGenExpon.test_sf_tailN)r   r   r   rx  ry  rz  r+   r+   r+   r,   rt  
  s    rt  c                   @   r   )TestExponpowc                 C   s6   t tjddd t tjtjdddd d S )N绽|=rj   r  r   r  )r   r/   exponpowrB   r
  r  r   r+   r+   r,   rN  
  s   zTestExponpow.test_tailN)r   r   r   rN  r+   r+   r+   r,   r{  
      r{  c                   @   r  )TestSkellamc                 C   s@   t dd}d\}}t g d}ttj||||dd d S )Nr>  r   r_   r   )gYjP'?g$S?gQ`s2?gC/qF?gD<]Y?g/Xj?g86y?g]a¨?g:?g?gQ>?g?g:?g]a¨?g96?g0X?gA<]?gB/q?gQ`s?g$S?gYjP'?gX_?g+rx?gfSr?gJXx~?r   )r9   r   r   r   r/   skellamr   )r   r<   mu1mu2ZskpmfRr+   r+   r,   r   
     zTestSkellam.test_pmfc                 C   s@   t dd}d\}}t g d}ttj||||dd d S )Nr>  r   r  )gƃQ?gS`X'?gVy)>?g:	 S?gei.f?gȼJ]x?g)?g{^'+i?gLIg>?gnZ?g?rvͪ?gA?g0?g#J?gqSm
.?g~B?gd?gFBD?gQ9?gmF^?gcMy?gGJq?gC^c?g.){?gñݸ?r   r   )r9   r   r   r   r/   r  rB   )r   r<   r  r  ZskcdfRr+   r+   r,   r   
  r  zTestSkellam.test_cdfN)r   r   r   r   r   r+   r+   r+   r,   r  
  s    r  c                   @   r  )TestLognormc                 C   sZ   t   t dt tjg dd}t|g d W d    d S 1 s&w   Y  d S )Nr   r   rf   r   r   )rk   g~r?g e3E?)r   r   r   r   r/   r   r8   r   r   r8   r+   r+   r,   r   
  s
   
"zTestLognorm.test_pdfc                 C   sn   d\}}}t tjj|| |dtjt|| |  t tjj|| |dtjt|| |  d S )N)g(\5i@   gOn?rS   )r   r/   r   r  r  rc   r   r  )r   r;  r2  Zsigmar+   r+   r,   r    s   
zTestLognorm.test_logcdfN)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ejj	e
d	d
dd Zejj	e
d	d
dd Zejj	e
d	d
dd Zdd Zdd ZdS )TestBetac                 C   s:   t jddd}t|d t jddd}t|tj d S )Nr   r   rf   g;B.)r/   rD  r   r   rc   r   r   r   r+   r+   r,   r    s   
zTestBeta.test_logpdfc                 C   sV   d\}}t g d}t||}t|| d t||t || d S )Ni  i  r)  rf   r#  gƒ)	rc   r   r/   rD  r   r   r   r8   r  r   r   rD  r?   r*   r+   r+   r,   test_logpdf_ticket_1866#  s
    z TestBeta.test_logpdf_ticket_1866c                 C   s$   g d}t ttjj|dddd d S )Nr]   rf   r#  r   r   rZ  )rz   r{   r[  )rg  r   r/   rD  rr   r  r+   r+   r,   test_fit_bad_keyword_args*  s   
z"TestBeta.test_fit_bad_keyword_argsc                 C   s"   g d}t ttjj|ddd d S )Nr  rf   )fafix_a)rg  r*  r/   rD  rr   r  r+   r+   r,   #test_fit_duplicated_fixed_parameter/  s   z,TestBeta.test_fit_duplicated_fixed_parameterzOverflow, see gh-14901reasonc                 C   s$   d\}}}t tj|||d d S )N)g?g     R@g   0xAgx)>)r   r/   rD  r   )r   r   r)   r*   r+   r+   r,   test_issue_126355  s   
	zTestBeta.test_issue_12635c                 C   sh   t g d}t g d}d}tj||d d| }t|| tj||d d| }t|| d S )N)g@3@?g^?gNC?)r_   r`   r  r\  r   順 )rc   r   r/   rD  r
  r   r  )r   Zinv_RZ
count_listr   invr  r+   r+   r,   test_issue_12794A  s   
zTestBeta.test_issue_12794c           	      C   sb   d}t dd}d}d| |d || }}}tj|||}tj|||}t|d|  d S )Nh㈵>r   r[  r  )rc   r   r/   rD  r   rB   r   )	r   Zalpha_2Zcount_Znobsr   r)   r*   r  r  r+   r+   r,   test_issue_12796S  s   zTestBeta.test_issue_12796c                 C   sD   d\}}t tjd||tj d\}}t tjd||tj d S )Nro  r   )r)  r   r   )r   r/   rD  r8   rc   r   )r   r)   r*   r+   r+   r,   r   _  s   zTestBeta.test_endpointsc                 C   sJ   d\}}}t t tj||| W d    d S 1 sw   Y  d S )N)gףp=
?g   vH7Bg  @0B)r   Zwarnsr   r/   rD  r   )r   r   r)   r*   r+   r+   r,   test_boost_eval_issue_14606j  s   
"z$TestBeta.test_boost_eval_issue_14606N)r   r   r   r  r  r  r  r   r7  skipifMACOS_INTELr  r  r  r   r  r+   r+   r+   r,   r    s    


r  c                   @   r  )TestBetaPrimec                 C   sZ   d\}}t g d}t||}tt ||  t|	|t 
|| d S )Nr  r  )rc   r   r/   r   r   r  r   r   r   r8   r  r  r+   r+   r,   r  q  s
    zTestBetaPrime.test_logpdfc                    s   t jddd}t|d d\ tg d}t j| }tt|  t j	j
 fdd|D }t||dd	d
 d S )Nr   r)  r  rk   r  r  c                    s   g | ]
}t j| qS r+   )r/   r   )rG   r   r   rD  Zgen_cdfr+   r,   rI         z*TestBetaPrime.test_cdf.<locals>.<listcomp>g-=r@  )r/   r   rB   r   rc   r   r   r  r   rL   Z_cdf_singler   )r   r?   r  Zcdfs_gr+   r  r,   r   x  s   
zTestBetaPrime.test_cdfN)r   r   r   r  r   r+   r+   r+   r,   r  p  r  r  c                   @   r   )
	TestGammac                 C   s<   t jjdddd}t|d t jjdddd}t|d d S )	NZ   i  r)  r  g
CTb?r   r_   gBΖ?)r/   rt  r8   r   r  r+   r+   r,   r     s   
zTestGamma.test_pdfc                 C      t jdd}t|d d S r   )r/   rt  r   r   r  r+   r+   r,   r    s   zTestGamma.test_logpdfc                 C   s"   g d}t ttjj|ddd d S )Nr  r   rZ  )rz   r[  )rg  r   r/   rt  rr   r  r+   r+   r,   r    s   z#TestGamma.test_fit_bad_keyword_argsc                 C   s@   t jtjdddddsJ t jtjdddd	dsJ d S )
NFFg<r   gplC@r  rk  gJz5r`   g1}t@r?  )rc   Ziscloser/   rt  r
  r   r+   r+   r,   r    s   
zTestGamma.test_isfN)r   r   r   r   r  r  r  r+   r+   r+   r,   r    s
    r  c                   @   r  )TestChi2c                 C   s4   t tjddddd t tjddddd d S )Nr  gޞw1D?   r   r`   g6:֜?r   r/   chi2r8   r   r+   r+   r,   r    s   
zTestChi2.test_precisionc                 C   s|   d}t jd|}t|ddd t jd|}t|ddd d}t jd	|}t|d
dd t jd|}t|ddd d S )Nr  gλm:=6g'bd5 <r|  rm   rf   gQ8@   gܐ؆0guu[9=r]   g8-*@)r/   r  r   r   )r   r  r?   r+   r+   r,   ry    s   zTestChi2.test_ppfN)r   r   r   r  ry  r+   r+   r+   r,   r    s    r  c                   @   r   )TestGumbelLc                 C   r,  Nra  r   )rc   r   r/   rS  rB   r   r   r/  r+   r+   r,   r    r0  zTestGumbelL.test_cdf_ppfc                 C   sH   t dd}tj|}tj|}t |}t| }t	|| d S r  )
rc   r   r/   rS  r  r  r  r   expm1r   )r   r?   r   rQ  ur  r+   r+   r,   r    s   
zTestGumbelL.test_logcdf_logsfc                 C   r,  )Nr   r   )rc   r   r/   rS  r  r
  r   r/  r+   r+   r,   r!    r0  zTestGumbelL.test_sf_isfN)r   r   r   r  r  r!  r+   r+   r+   r,   r    s    r  c                   @   r  )TestGumbelRc                 C      t tjdddd d S )Nr   g?~T}%m;r  rm   )r   r/   rR  r  r   r+   r+   r,   r       
zTestGumbelR.test_sfc                 C   r  )Nr  r  r  rm   )r   r/   rR  r
  r   r+   r+   r,   r    r  zTestGumbelR.test_isfN)r   r   r   r  r  r+   r+   r+   r,   r    s    	r  c                   @   sf   e Zd Zdd Zejjdd Zejjdd Zdd Z	d	d
 Z
ejjejdddgdd ZdS )TestLevyStablec                 C   s   g d}t j|\}}}}t|dddd t|dd t|dd	 t|d
d |g d }t j|\}}}	}
t|d t|d t|
dd	 t|	dd	 d S )N)wgr  rk   rk   rk   rk   羼 u?r  r  r  r  67',?r  r  r  r  tA}˜.?r  r  r  r  Gz?r   rh   r   )\(̿r6   gۧ1?r  g9]c?)r  r  r  r  r  r`  gv?r  )r/   levy_stabler<  r   r   r   )r   r?   Zalpha1Zbeta1loc1scale1r;  Zalpha2Zbeta2loc2scale2r+   r+   r,   r?    s   

zTestLevyStable.test_fitc                    s  t jt jt jtd}t|}tjj	j
|jdd}ttdr&tjntj g dddd fd	d
gddddd
 gddddd
 gddddd
 gddddd
 gddddd
 gddddd
 gg}t|D ]{\}\}}}}|tj_|tj_|dur||| n|}	t T}
|
td |
td tjj|	d |	d |	d ddd}t|	d|}t||	d  }||dd |   kt|B  }t||	d |d!|||f d"d# W d   n1 sw   Y  qedS )$a!   Test pdf values against Nolan's stablec.exe output
            see - http://fs2.american.edu/jpnolan/www/stable/stable.html

            There's a known limitation of Nolan's executable for alpha < 0.2.

            Repeat following with beta = -1, -.5, 0, .5 and 1
                stablec.exe <<
                1 # pdf
                1 # Nolan S equivalent to S0 in scipy
                .25,2,.25 # alpha
                -1,-1,0 # beta
                -10,10,1 # x
                1,0 # gamma, delta
                2 # output file
        zdata/stable-pdf-sample-data.npyx,p,alpha,betanamesisin)ZbestNrh  N
quadratureNrh  c                    s   | d dk | d g d@ S )Nr   r   r?   )r>  rg  r   r   r_   r+   r  Znpisinr+   r,   rT   2  s   
 z7TestLevyStable.test_pdf_nolan_samples.<locals>.<lambda>	zolotarevc                 S   s   | d dkS Nr   r   r+   r  r+   r+   r,   rT   6      c                 S   s   | d dk| d dk@ S Nr   r   rD  r   r+   r  r+   r+   r,   rT   8      r   c                 S      | d dk| d dk@ S r  r+   r  r+   r+   r,   rT   :  r  fftr   r  c                 S      | d dkS r  r+   r  r+   r+   r,   rT   >  r  r   c                 S   s   | d dk | d dk@ S )Nr   r   r   r+   r  r+   r+   r,   rT   ?  r  c                 S   r  )Nr   r   r?   r   r+   r  r+   r+   r,   rT   A  r  z6Density calculation unstable for alpha=1 and beta!=0.*z2Density calculations experimental for FFT method.*r?   r   rD  r5   r4   calcr   r  rX  z&pdf test %s failed with method '%s'
%sFverbose)ospathabspathjoindirname__file__rc   loadcorerecords
fromarraysTr(   r  Zin1dr  r/   r  pdf_default_methodpdf_fft_min_points_thresholdr
   recordr   r8   r   r  isnanr   r   fnrv   testsZixdefault_methodfft_min_pointsdecimal_placesZfilter_funcZsubdatasupr   Zsubdata2ZpadiffZfailuresr+   r  r,   test_pdf_nolan_samples  sj   


z%TestLevyStable.test_pdf_nolan_samplesc              
   C   sL  t jt jt jtd}t|}tjj	j
|jdd}g dddddd	 gg}t|D ]u\}\}}}}|tj_|tj_|d
urH||| n|}	t N}
|
td tjj|	d |	d |	d ddd}t|	d|}t||	d  }||dd|   kt|B  }t||	d |d|||f dd W d
   n1 sw   Y  q.d
S )a!   Test cdf values against Nolan's stablec.exe output
            see - http://fs2.american.edu/jpnolan/www/stable/stable.html

            There's a known limitation of Nolan's executable for alpha < 0.2.

            Repeat following with beta = -1, -.5, 0, .5 and 1
                stablec.exe <<
                2 # cdf
                1 # Nolan S equivalent to S0 in scipy
                .25,2,.25 # alpha
                -1,-1,0 # beta
                -10,10,1 # x
                1,0 # gamma, delta
                2 # output file
        zdata/stable-cdf-sample-data.npyr  r  )r  Nrh  Nr  r   r6   c                 S   r  r  r+   r  r+   r+   r,   rT   w  r  z7TestLevyStable.test_cdf_nolan_samples.<locals>.<lambda>NzXFFT method is considered experimental for cumulative distribution function evaluations.*r?   r   rD  r   r  r  r   r  rX  z&cdf test %s failed with method '%s'
%sFr  )r  r  r  r  r  r  rc   r  r  r  r  r  r  r/   r  r  r  r
   r  r   rB   r   r  r  r   r  r+   r+   r,   test_cdf_nolan_samples[  sF   

z%TestLevyStable.test_cdf_nolan_samplesc           
   
   C   s   t g d}t g d}t g d}g dg dg}t jddM t 1}|jtdd	 |D ]\}}}|tj_|tj_	tjj
|d
|d
dd}	t|	||| q1W d   n1 s[w   Y  W d   dS W d   dS 1 ssw   Y  dS )z
        sample points extracted from Tables and Graphs of Stable Probability
        Density Functions - Donald R Holt - 1973 - p 187.
        )r   r   r   r   r   r   r   r   r6   r6   r6   r6   r   r   r   r   r  r  r  r  )gtF_?gܵ|?gQ?g@?gec]?gHP?gPkw?g!rh?gTN?g?gDioɴ?g r?gH?g+?gHPs?gX ?g&?g0*?gv/?ga2U0*?)r   r   rf   r   r   r   rf   r   r   r   rf   r   r   r   rf   r   r   r   rf   r   )r  Nr  )r  Nr   ignorer   zDensity calculation unstable.*)categorymessager   r   r  N)rc   r   errstater
   rX   r   r/   r  r  r  r8   r   )
r   ZxsZdensityZbetasr  r  r  r  r  r8   r+   r+   r,   'test_pdf_alpha_equals_one_beta_non_zero  s&   Pz6TestLevyStable.test_pdf_alpha_equals_one_beta_non_zeroc                 C   s`   ddt jt jt jfgddgg}|D ]\}}tjj|d |d |d |d dd	}t|| qd S )
N)r  r  r   r   r   )r6   r'  r_   r  )r_   rY  r   r   r   r6   r   rq  re  )rc   r   ZNaNr/   r  r   )r   Z
param_setsr   Z	exp_statsZ
calc_statsr+   r+   r,   ru    s   zTestLevyStable.test_statsrD  rf   r   c                 C   sZ   t jd d}d}d}tjj||||dd}tj|d||||fd\}}|d	ks+J d S )
Nh:r   rf   r  i  r4   r5   r   r  r0  rh   )rc   r   r   r/   r  r   r!  )r   rD  r   r4   r5   r?   statr   r+   r+   r,   test_rvs_alpha1  s   

zTestLevyStable.test_rvs_alpha1N)r   r   r   r?  r   r7  r8  r  r  r  ru  r  r  r+   r+   r+   r,   r    s    
K
2$r  c                   @   r  )TestArrayArgumentc                 C   r   r   r   r   r+   r+   r,   r     r   zTestArrayArgument.setup_methodc                 C   s.   t jjtdtddd}t|jd d S )Nr   r  r  )r/   r  r   rc   r   rG  r   r   r   r   r+   r+   r,   test_noexception  s   z"TestArrayArgument.test_noexceptionN)r   r   r   r   r  r+   r+   r+   r,   r        r  c                   @   r  )TestDocstringc                 C   sH   t jjd urtdt jj v  t jjd ur"tdt jj v  d S d S )Nr   r   )r/   r   r_  r   lowerr   r   r+   r+   r,   test_docstrings  s
   zTestDocstring.test_docstringsc                 C   s   t   t   d S rb   )r/   rL   rK   r   r+   r+   r,   test_no_name_arg  s   zTestDocstring.test_no_name_argN)r   r   r   r  r  r+   r+   r+   r,   r    r  r  c                  C   s   t g d} t| dk| d\}}t|g d t|g d tddk| d\}}t|| d  t|dg t| dk| d\}}t||  t|dgt|   d S )N)r   r   r6   r   r6   r   r   r   r6   )r   r6   r6   r   r   )r6   r6   r6   r6   r6   r   )r   r   r   r9   r   r)   r*   r   r+   r+   r,   TestArgsreduce  s   
r  c                   @   s   e Zd Zg dZdd Zg dZejde	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ddgdd Zejdddgdd ZdS )TestFitMethod)ncfr  r  c                 C   r   r   r   r   r+   r+   r,   r     r   zTestFitMethod.setup_method)r  r  rv  z	dist,argsc                 C   s~   || j v rtd|  tdddddtjg}tdddddtjg}tt|}t	t
|j|ddd	 t	t
|j|ddd	 d
S )r^  z"%s fit known to fail or deprecatedrQ  rR  rS  rT  rU  r   r   re  N)fitSkipNonFiniter   skiprc   r   r   r   r   r/   rg  r   rr   )r   rH   r   r?   r   Zdistfuncr+   r+   r,   !test_fit_w_non_finite_data_values  s   

z/TestFitMethod.test_fit_w_non_finite_data_valuesc              	   C   s   t jd t jdd8 tjjddddd}t t |t d d	 	 }t
t tjj|d
dd|d
dgdd W d    d S 1 sGw   Y  d S )Ni90  r  r  r   rk         4@r[  r   r6   r   re  rN  rk  )rc   r   r   r  r/   r   r   rU  r   r   r   r   rr   )r   r?   Zexpected_shaper+   r+   r,   test_fix_fit_2args_lognorm  s   "
"z(TestFitMethod.test_fix_fit_2args_lognormc                 C   s   t dd}tj|\}}t|d t|t d tjj|dd\}}t|d t|t d tjj|dd\}}t|d t|d d S )Nr   r  r   r6   rd  r  )rc   r   r/   r  rr   r   rU  r   r   r?   r4   r5   r+   r+   r,   test_fix_fit_norm  s   


zTestFitMethod.test_fix_fit_normc                 C   sn  t dd}t | }d}tjj||d\}}}t | | }tt |t	| |dd t
|| t|| | dd d}d}tjj|||d\}}}t
|| t
|| t|| | dd d	}d}tjj|||d\}}}t
|| t
|| t|| | dd d}d	}	tjj|||	d
\}}}t
|| t
||	 |t |	 }
tt	||
 d S )Nr   r  r   rd  r   r   rh  r}   rz   r6   re  )rc   r   r   r   r/   rt  rr   r   r   Zdigammar   )r   r?   Zmeanlogrz   r)   r4   r5   r>   r}   r{   r   r+   r+   r,   test_fix_fit_gamma  s6   






z TestFitMethod.test_fix_fit_gammac              	   C   s  dd }t g d}tjj|ddd\}}}}t|d t|d t||||ddgdd t g d}tjj|d	ddd
\}}}}t|d	 t|d t|d ||||\}}t|ddd d| }	tjj|	d	ddd\}
}}}t|d	 t|d t|d ||
||	\}}t|ddd t|
| tt	tjj|ddd t g d}tt	tjj|ddd tt	tjj|ddd	d tt	tjj|ddd	d tt	tjj|ddd	dd d S )Nc                 S   sj   t |}t| }td|  }t| | }||| t|    ||| t|   g}|S Nr   )r   rc   r   r   r   Zpsi)r)   r*   r?   r   s1r  Zpsiabrw   r+   r+   r,   mlefuncH  s   z0TestFitMethod.test_fix_fit_beta.<locals>.mlefunc)r~  r   rf   r   r   re  rI  rk  r6   )r}   rz   r{   r  )r~   rz   r{   rf   r  )rz   r{   r}   )rz   r{   r~   r   )r}   r~   rz   r{   )
rc   r   r/   rD  rr   r   r   r   rg  r*  )r   r  r?   r)   r*   r4   r5   daZdbr;  a2b2r  r  r   r+   r+   r,   test_fix_fit_betaE  s:   









zTestFitMethod.test_fix_fit_betac                 C   s   t g d}tj|\}}t|d t|d tjj|dd\}}t|d t|d tjj|dd\}}t|d t|d d S )N)r6   r6   r  r  r  r  r  rh  r6   r   r  r   rd  r  )rc   r   r/   r  rr   r   r  r+   r+   r,   test_expon_fit|  s   




zTestFitMethod.test_expon_fitc                 C   s   t g d}t |d }tjj|dd\}}}t|| dd t|d t|t 	|
 dd tjj|ddd\}}}t|t |t d d 
 dd t|d t|d tjj|dd	d
\}}}t|d	 t|d t|t 	|
 dd d S )N)r  r   r_   r      ;   r   rd  r  rm   r  re  r6   r   )rz   Zfix_s)rc   r   r   r/   r   rr   r   r   r   r  r   rU  )r   r?   Zlnxm1r   r4   r5   r+   r+   r,   test_lognorm_fit  s    




zTestFitMethod.test_lognorm_fitc                 C   s   t g d}tj|\}}t||  t||  tjj|dd\}}t|d t||  tjj|dd\}}t|d t|d t	t
tjj|dd t	t
tjj|dd d S )N)r   r  r(        "@r   rd  r_   r  rj   r   )rc   r   r/   rv  rr   r   r  Zptpr  rg  r*  r  r+   r+   r,   test_uniform_fit  s   


zTestFitMethod.test_uniform_fitmethodZMLEZMMc              
   C   sl  d\}}t jj||ddd}t jj|d|d}t jj|d|d}t||ddd	 t jj|d|d
}t||ddd	 t jj|d|d}t jj|d|d}t||ddd	 t jj|d|d}t||ddd	 ttt jj|dd|d ttt jj|dddd|d t jj|ddd|d}	|	\}
}}}t|
||gg d d}t jj|dd}t jj|||d\}
}}t|
| d S )N)rp  r$  r`   r   r  rp  )r}   r  )r  r  r  r@  )r  r  r$  )r~   r  )rc  r  )rb  r  r   r6   )r  r}   r  r   r   )r  r~   rz   r{   r  )r  rz   r{   r  )rp  r   r   r   )	r/   rD  r   rr   r   rg  r*  r   rt  )r   r  r)   r*   r?   Zres_1Zres_2Zres_3Zres_4Zres_5ZaaZbbr  ssrv   r+   r+   r,   test_fshapes  s.   zTestFitMethod.test_fshapesc                 C   s@   t j}|jddd}tdd}tt|j|fi |d|i d S )Nr6   r`   )rg  r   i)Zenikibenikir  )r/   rf  r   r   rg  r   rr   )r   r  rH   rv   r   r+   r+   r,   test_extra_params  s   
"zTestFitMethod.test_extra_paramsN)r   r   r   r  r   r   r   r7  r  r   r  r  r  r  r  r  r  r  r  r  r+   r+   r+   r,   r    s"    

	&7
&r  c                   @   r9  )
TestFrozenc                 C   r   r   r   r   r+   r+   r,   r     r   zTestFrozen.setup_methodc                 C   s  t j}t jddd}|d}|jdddd}t|| |d}|jdddd}t|| |d}|jdddd}t|| |d}|jdddd}t|| |d}|jdddd}t|| | }|jddd}t|| |	 }|j	ddd}t|| |
 }|j
ddd}t|| | }|jddd}t|| | }|jddd}t|| |d}|jdddd}t|| t|j|j t|j|j d S )NrX  rp  r3   r  r   r6   )r/   r  r8   r   rB   r   r
  r  r  r   r(  r   rN   rK  r)   r*   )r   rH   frozenresult_fr  r+   r+   r,   r    sJ   
















zTestFrozen.test_normc                 C   s  d}t j}t |}|d}|d|}t|| |d}|d|}t|| |d}|d|}t|| |d}|d|}t|| |d}|d|}t|| | }||}t|| |	 }|	|}t|| |
 }|
|}t|| | }||}t|| | }||}t|| |d}|d|}t|| t|j|jj t|j|jj d S )Nrj   r  r   rX  r6   )r/   rt  r8   r   rB   r   r
  r  r  r   r(  r   rN   rK  r)   rH   r*   )r   r)   rH   r  r  r  r+   r+   r,   
test_gamma  sL   






















zTestFrozen.test_gammac                 C   s8   t d}|d}|j dd |d}t|| d S )Nr   r6   rq  rr  )r/   r   rK  r   )r   r  m1r  r+   r+   r,   test_regression_ticket_1293Q  s
   


z&TestFrozen.test_regression_ticket_1293c                 C   sR  d}t j|d}|j|\}}t||gddg d}t jjd|d t|j|dtjg d}t j|d}|j|\}}t||gddg d}t jd| t|jj|jj	ft j| t jdd}t
|j|ju dD ]6}t|}t j|d}|j|j	}}t|d t
t| td}t j|\}}t||gdd	g qpd S )
Nr  rr  rk   rX  r]   r   rk  rl  rf   )r/   rn  rH   ro  r   r8   rc   r   r)   r*   r   rm  rp  r   )r   r   rt  r)   r*   Zrv1r+   r+   r,   rq  `  s4   



zTestFrozen.test_abc                 C   s   t ttjd d S )NZ	rv_frozen)r   r(   r/   r0   r   r+   r+   r,   test_rv_frozen_in_namespace  s   z&TestFrozen.test_rv_frozen_in_namespacec                 C   sV   t  }tt|d d|_t|j tj	d  tj	d}|j
d|d d S )Nr  *   r   rh  r  )r/   r  r   r(   r  r   Z	get_staterc   r   ZRandomStater   )r   r  Zrndmr+   r+   r,   test_random_state  s   
zTestFrozen.test_random_statec           
      C   s   t dd}t d}t jg dg dfd}|||fD ]H}d|_|jdd	 t|}|jdd	}t|}|jdd	}t	|| |
d
|
d
g}	t	|	d |	d  t	||	d ||	d  qd S )Ngdsz@g@e?rp  )r   r   r6   r   )r]   r)  r  r  r  r   rh  r   rf   r   r   )r/   rD  r  rK   r  r   pickledumpsloadsr   r   rB   )
r   rD  Zpoissr  distfnr>   Zr0Z	unpickledZr1Zmediansr+   r+   r,   test_pickling  s&   




zTestFrozen.test_picklingc              	   C   s   dd }t jdddd}|j|dddd	}t jj|d
dddddd}t|| t jddd}||}t jj|ddd}t|| d S )Nc                 S      | S rb   r+   r7  r+   r+   r,   rw        z$TestFrozen.test_expect.<locals>.funcr6   r   r  )r)   r4   r5   r   T)lbubconditionalr6   r   r4   r5   r*  r+  r,  r4   )r   r   r4   )r/   rt  r%  r   r  )r   rw   ZgmZgm_valZ	gamma_valr   Zp_valZpoisson_valr+   r+   r,   test_expect  s   

zTestFrozen.test_expectN)r   r   r   r   r  r  r  rq  r   r"  r'  r1  r+   r+   r+   r,   r    s    34,r  c                   @   r  )
TestExpectc                 C   s   t jjdd ddd}t|ddd t jjd	d ddd}t|ddd t jjd
ddd}t jjdddd}t jjdd dd||d}t|ddd t jjdd dd||dd}t|ddd d S )Nc                 S      | d | d  S )Nr   r+   r7  r+   r+   r,   rT     rU   z&TestExpect.test_norm.<locals>.<lambda>r   r6   r3   r  r  r   c                 S   r(  rb   r+   r7  r+   r+   r,   rT         r  r  c                 S      dS r	  r+   r7  r+   r+   r,   rT     r4  )r4   r5   r*  r+  r'  c                 S   r5  r	  r+   r7  r+   r+   r,   rT     r4  T)r4   r5   r*  r+  r,  r   )r/   r  r%  r   r   )r   r  r  r*  r+  prob90prob90cr+   r+   r,   r    s   zTestExpect.test_normc              	   C   s   t jjdd dddd}t|ddd	 t jjd
d dddd}t|ddd	 t jjdddddd}t jjdddddd}t jjdd ddd||dd}t|ddd	 t jjdd ddd||dd}t|ddd	 d S )Nc                 S   r3  )NUUUUUU@r+   r7  r+   r+   r,   rT     rU   z&TestExpect.test_beta.<locals>.<lambda>r  r   r6   )r   r4   r5   qq?r  r   c                 S   r(  rb   r+   r7  r+   r+   r,   rT     r4  r   rj   r8  r  r_   r3   r  c                 S   r5  )Nr   r+   r7  r+   r+   r,   rT     r4  )r_   r_   Fr.  r'  c                 S   r5  r	  r+   r7  r+   r+   r,   rT     r4  Tr   )r/   rD  r%  r   r   )r   r  r  r+  r*  r6  r7  r+   r+   r,   	test_beta  s    zTestExpect.test_betac           
      C   s&  t jj ddddd\}}t jjdd ddd	}t||d
d t jjdd ddd	}t||dd t jjdd dddd
d}t||dd dt jjdd
gddddd  }t jjdd ddddd}t||d
d t jjdd dddddd}t|ddd t jjdd dddd}	t|	dd
d d S )Nr[  r_   rh  r   r/  c                 S   r(  rb   r+   r7  r+   r+   r,   rT     r4  z+TestExpect.test_hypergeom.<locals>.<lambda>)r[  r_   rh  r0  r  r   c                 S      | d d S Nr  r6   r+   r7  r+   r+   r,   rT     r  r  c                 S   r;  r<  r+   r7  r+   r+   r,   rT     r  r   )r   r4   r*  r+  r   c                 S   r5  r	  r+   r7  r+   r+   r,   rT     r4  r  r   c                 S   r5  r	  r+   r7  r+   r+   r,   rT   	  r4  T)r   r4   r*  r+  r,  c                 S   r5  r	  r+   r7  r+   r+   r,   rT     r4  r   )r   r*  r+  )r/   r  r%  r   r   r   )
r   Zm_trueZv_truer  r  Zv_boundsZ	prob_trueprob_boundsZprob_bcZprob_br+   r+   r,   test_hypergeom  s2   "zTestExpect.test_hypergeomc                 C   sb   t jjdd dddd}dt jdd }t||d	d
 t jjdd dddd}t|dd	d
 d S )Nc                 S   r5  r	  r+   r7  r+   r+   r,   rT     r4  z)TestExpect.test_poisson.<locals>.<lambda>r-  r   F)r   r*  r,  r   r6   r  r   c                 S   r5  r	  r+   r7  r+   r+   r,   rT     r4  T)r/   r  r%  rB   r   )r   r=  Zprob_b_trueZprob_lbr+   r+   r,   test_poisson  s   zTestExpect.test_poissonc                 C   s<   t j}|jdd}|jdd |jdd}t||dd d S )N)r  r0  )rf   r  r   )r/   Zgenhalflogisticr%  r   )r   Zhalflogres1r  r+   r+   r,   test_genhalflogistic  s
   zTestExpect.test_genhalflogisticc                 C   sv   t ttjdd t ttjjdd dd t ttjjdd dd t ttjjdd dd d S )	Ni  Gz?c                 S   r5  r	  r+   r7  r+   r+   r,   rT   ,  r4  z/TestExpect.test_rice_overflow.<locals>.<lambda>)rB  r0  c                 S   r5  r6  r+   r7  r+   r+   r,   rT   -  r4  c                 S   r5  )Nr   r+   r7  r+   r+   r,   rT   .  r4  )r   rc   r  r/   ricer8   r%  r   r+   r+   r,   test_rice_overflow'  s   "zTestExpect.test_rice_overflowc                 C   sp   d\}}t jjdd |fd}t|||d  td|  dd t jjdd |f|d	}t||| dd d S )
N)r  r   c                 S   r(  rb   r+   r<   r+   r+   r,   rT   3  r4  z(TestExpect.test_logser.<locals>.<lambda>r0  r   rl   rk  c                 S   r(  rb   r+   rE  r+   r+   r,   rT   9  r4  r0  )r/   rJ  r%  r   rc   r   )r   r   r4   Zres_0Zres_lr+   r+   r,   test_logser0  s   zTestExpect.test_logserc                 C   sh   d\}}t jjdd ||fd}t jjdd ||fd}t||| dd t||d  || dd d S )	N)r  r  c                 S   r(  rb   r+   r7  r+   r+   r,   rT   @  r4  z)TestExpect.test_skellam.<locals>.<lambda>r0  c                 S   r5  r6  r+   r7  r+   r+   r,   rT   A  r8  r  rk  r6   )r/   r  r%  r   )r   r  p2r  r  r+   r+   r,   test_skellam<  s
   zTestExpect.test_skellamc                 C   sL   d\}}t jdd ||f}t|tdd t||D ||  dd d S )N)r   q   c                 S   r(  rb   r+   r7  r+   r+   r,   rT   I  r4  z)TestExpect.test_randint.<locals>.<lambda>c                 s   s    | ]}|V  qd S rb   r+   rG   r$  r+   r+   r,   r4  K  s    z*TestExpect.test_randint.<locals>.<genexpr>rl   rk  )r/   r   r%  r   r   rN  )r   lohir  r+   r+   r,   test_randintE  s
    
zTestExpect.test_randintc                 C   s   t ttjjdd d d S )Nc                 S   r5  r6  r+   r7  r+   r+   r,   rT   P  r8  z&TestExpect.test_zipf.<locals>.<lambda>r-  )r   r   r/   r  r%  r   r+   r+   r,   	test_zipfM  s   
zTestExpect.test_zipfc                 C   s@   t jjdd dd}t jjdd ddddd	}t||d
d d S )Nc                 S   r5  r	  r+   r7  r+   r+   r,   rT   T  r4  z/TestExpect.test_discrete_kwds.<locals>.<lambda>r-  r0  c                 S   r5  r	  r+   r7  r+   r+   r,   rT   U  r4  r      rN  )r   ZmaxcountZ	chunksizeZ	tolerancer  r   )r/   r  r%  r   )r   Zn0Zn1r+   r+   r,   test_discrete_kwdsR  s
   zTestExpect.test_discrete_kwdsc                 C   s6   dd }dD ]}t jd|}t|||dd qd S )Nc                 S   s0   | d d| d   d| d   d| d   |  S )Nr   r_   r  r  r   r   r6   r+   r  r+   r+   r,   poiss_moment5\  s   0z-TestExpect.test_moment.<locals>.poiss_moment5)r   r  r   r|  rm   )r/   r  rK  r   )r   rQ  r2  Zm5r+   r+   r,   test_momentY  s
   zTestExpect.test_momentN)r   r   r   r  r:  r>  r?  rA  rD  rF  rH  rM  rN  rP  rR  r+   r+   r+   r,   r2    s    "
		r2  c                   @   r   )TestNctc                 C   s@   t dd}t|dd t dd}t|dddd d S )Nr   r   rf   r`  g K?r_   r   )r/   nctr   rB   r   ry  r+   r+   r,   test_nc_parametere  s   zTestNct.test_nc_parameterc              	   C   sZ   t jdtddd d d f tddd}tg dg dg dg}t||d	d
 d S )Nr   r  r  r]   r   )gMp^j?g6].ftv?gȳ'ς?gu?)g!Y<a?g]d1p?go|?g?)gHdY?g2Ԥg?g}o/u?g .mr?r  rm   )r/   rT  r8   rc   r   r   r   r   )r   r  r[   r+   r+   r,   test_broadcastingm  s   zTestNct.test_broadcastingc                 C   s   t dd}t| d d S )Nr  r   rj   )r/   rT  r   r(  ry  r+   r+   r,   test_variance_gh_issue_2401u  s   z#TestNct.test_variance_gh_issue_2401c                 C   s   t jj dddd\}}}}t||||gtjtjtjtjg t jj dddd\}}}}tt| t|||gtjtjtjg t jj dddd\}}}}tt|||g  t|tj d S )Nr'  r  rq  )r  ncrs  r  r  )r/   rT  r   rc   r   r   r  r   r[  r+   r+   r,   test_nct_inf_moments}  s   "zTestNct.test_nct_inf_momentsc                 C   s   t jdd}t j dd}ddg}t||d dd t||dd t jdd}t j dd}d	d
g}t||d dd t||dd d S )Nr  r6   g@ @g*dn?r   r|  rm   r  gqI  @gL) ?rj  )r/   rT  r   r   )r   Znct_mean_df_1000Znct_stats_df_1000Zexpected_stats_df_1000Znct_meanZ	nct_statsr  r+   r+   r,   test_nct_stats_large_df_values  s   
z&TestNct.test_nct_stats_large_df_valuesN)r   r   r   rU  rV  rW  rY  rZ  r+   r+   r+   r,   rS  d  s    rS  c                   @   r   )
TestRecipInvGaussc                 C   s   t jdd}|dksJ d S )Nr   r#  rk   )r/   recipinvgaussr8   r   r+   r+   r,   test_pdf_endpoint  s   z#TestRecipInvGauss.test_pdf_endpointc                 C   s"   t jdd}|tj ksJ d S )Nr   r#  )r/   r\  r   rc   r   r4  r+   r+   r,   test_logpdf_endpoint  r	  z&TestRecipInvGauss.test_logpdf_endpointc                 C   s$   t jdd}d}t||dd d S )Nr  rf   g9V9s;r  rm   )r/   r\  rB   r   r   r   r[   r+   r+   r,   test_cdf_small_x  s   z"TestRecipInvGauss.test_cdf_small_xc                 C   s"   t jdd}d}t||d d S )NP   rf   glH<g+<)r/   r\  r  r   r_  r+   r+   r,   test_sf_large_x  s   z!TestRecipInvGauss.test_sf_large_xN)r   r   r   r]  r^  r`  rb  r+   r+   r+   r,   r[    s
    r[  c                   @   r   )TestRicec                 C   s   g d}t ttjj|dd  t ttjj|dd  t ttjj|dd  t ttjj	|dd  g d}t ttjj
|dd  tjjddd}t t|  d}ttj|dtj|||dd	 d S )
N)r)  r   r   rk   r*   )r]   r]   rf   r'  r   rq  rr  rN  r@  )r   rc   r  r/   rC  r8   r   r   rB   r  r   r   )r   r?   r   rq  r*   r+   r+   r,   test_rice_zero_b  s   
zTestRice.test_rice_zero_bc                 C   s2   t jj}t|ddjd t|dddjd d S )Nrp  rd  r   )r   r   )r*   r   )r/   rC  r   r   r   r   r  r+   r+   r,   test_rice_rvs  s   zTestRice.test_rice_rvsc                 C   s   t jtdddtddd}g d}t|| tddd}t jj|ddd}g d	}t|| tj jd
tddd}g d}t|| d S )Nr_      )g/Ƹ?g[jǊ\?g 	?g9LJ?g#O?gXx?glO?g٥a%?gVzӯ?gTQ+xQ?g?`J?go?gbu=?gIo|?g"Z6?r]   r   g     @_@r  r  )	g<~@gwd
@gC?@gb
0@g` A@@g'NxP@g33-a@g4pv@g$F@rf      )gJ-$@g+f4@gce*D>@g$D@gnHGI@grN@g<
uQ@giff T@g9r[V@g3bXQ Y@gqxJ[@g`*DD ^@g@`@gdAa@)r/   rC  rB   rc   r   r   r   r.   )r   rB   Zcdf_expZprobabilitiesr   Zppf_expr+   r+   r,   test_rice_gh9836  s   "

zTestRice.test_rice_gh9836N)r   r   r   re  rf  ri  r+   r+   r+   r,   rc    s    rc  c                   @   r   )
TestErlangc                 C   r   r   r   r   r+   r+   r,   r     r   zTestErlang.setup_methodc              	   C   s   t  6 t dt tttjjddddd g d}tjj|dd}tj	j|dd}t
||d	d
 W d    d S 1 s=w   Y  d S )Nr   ?r   r   r  r  )rf   r   rj   r$  r  rw  rm   )r   r   r   r   rg  r/   erlangr   rr   rt  r   )r   rv   Zresult_erlangZresult_gammar+   r+   r,   test_erlang_runtimewarning  s   
"z%TestErlang.test_erlang_runtimewarningc                 C   s.   t tjjdddgdtjjdddgd d S )Nrf   r   r`  r  )r   r/   rl  r8   rt  r   r+   r+   r,   test_gh_pr_10949_argcheck"  s   z$TestErlang.test_gh_pr_10949_argcheckN)r   r   r   r   rm  rn  r+   r+   r+   r,   rj    s    rj  c                   @   sr   e Zd Zdd Zdd Zdd Zejde	j
dgd	d
 Zejdddge	j
dgdd Zdd ZdS )TestRayleighc                 C   r   )Nr  r   r   r+   r+   r,   r   (  r   zTestRayleigh.setup_methodc                 C   rM  )Nr   g)Zx)r/   r   r   r   r   r   r+   r+   r,   r  ,     zTestRayleigh.test_logpdfc                 C   rM  )Nr   i)r/   r   r  r   rp  r+   r+   r,   r  0  rq  zTestRayleigh.test_logsfzrvs_loc,rvs_scaler6   c                 C   s   t jjd||d}dd }|||}t jj||d\}}t|| t|| t jj|dd\}}t|d t j|\}}t|||| d S )Nr  r7  c                 S   s"   t | | d dt|   d S )Nr6   rf   )rc   r   r   )rv   rz   r+   r+   r,   r  8  s   "z(TestRayleigh.test_fit.<locals>.scale_mlerd  r#  r  )r/   r   r   rr   r   )r   r]  r^  rv   r  Zscale_expectr4   r5   r+   r+   r,   r?  4  s   



zTestRayleigh.test_fitrB  rh   c                 C   r@  )Nr  r7  r   )r/   r   r   r<  rB  ry   )r   r]  r^  rv   r   rw   r+   r+   r,    test_fit_comparison_super_methodL  s   z-TestRayleigh.test_fit_comparison_super_methodc                 C   s   t tj d S rb   )r   r/   r   r   r+   r+   r,   rh  Y  rC  zTestRayleigh.test_fit_warningsN)r   r   r   r   r  r  r   r7  r  rc   r   rH  r?  rr  rh  r+   r+   r+   r,   ro  '  s    


ro  c                   @   r   )TestExponWeibc                 C   sB   d}d}d}t j|||}t j|||}t||gddg d S )Nr]   r   g      Y@g:23W+g~`9k)r/   	exponweibr8   r   r   )r   r?   r)   r   r   r  r+   r+   r,   test_pdf_logpdf_  s   zTestExponWeib.test_pdf_logpdfc                 C   sj   t ddd}d}d}tj|||}tj||}t|| tj|||}tj||}t|| d S )Nr   r`  r  r   r`   )rc   r   r/   rt  r8   weibull_minr   r   r   r?   r)   r   r   r[   r  r+   r+   r,   test_a_is_1j  s   
zTestExponWeib.test_a_is_1c                 C   sf   t ddd}d}d}tj|||}tj|}t|| tj|||}tj|}t|| d S )NrI  r   r_   )rc   r   r/   rt  r8   r  r   r   rw  r+   r+   r,   test_a_is_1_c_is_1z  s   
z TestExponWeib.test_a_is_1_c_is_1N)r   r   r   ru  rx  ry  r+   r+   r+   r,   rs  ]  s    rs  c                   @   r  )TestFatigueLifec                 C   r"  )Nr0  rU  X9r?  rm   )r/   r   r  r   r  r+   r+   r,   rz    s   zTestFatigueLife.test_sf_tailc                 C   s$   d}t j|d}t|ddd d S )Nr{  rU  r0  r?  rm   )r/   r   r
  r   )r   r   r   r+   r+   r,   test_isf_tail  s   zTestFatigueLife.test_isf_tailN)r   r   r   rz  r|  r+   r+   r+   r,   rz    s    rz  c                   @   r  )TestWeibullc                 C   r  r   )r/   rv  r   r   rp  r+   r+   r,   r       zTestWeibull.test_logpdfc           
      C   sP  d}d}d}t jj|||d}t|tdd  t jj|||d}t|dtd  t jj|||d}t|t	
d  t jj|||d}t|tt	
d  t jj|||d}t|td t jj|||d}	t|	d t jjdddd}t|td	 t jjdddd}	t|	d	 d
}t jj|||d}t|tdd  t jj|||d}t|dtd  t jj|||d}t|td t jj|||d}t|d t jj|||d}t|t	
d  t jj|||d}	t|	tt	
d  t jjdddd}t|t	
d  t jjdddd}	t|	tt	
d  d S )Nr  rj   rp  r  r   r   r   r6   ra  r-  g&.gCe )r/   rv  r8   r   rc   r  r   r   rB   r   r  r  r  r  Zweibull_max)
r   r?   r)   r*   r   r#  r   Zlcr>   Zlsr+   r+   r,   test_with_maxima_distrib  sH   


z$TestWeibull.test_with_maxima_distribN)r   r   r   r  r  r+   r+   r+   r,   r}    s    r}  c                   @   r  )	TestRdistc                 C   s0   t j}g d}t|||dd|dd d S )Nrv  g     @r   r   )r/   rdistr   rB   r   )r   r&  r  r+   r+   r,   test_rdist_cdf_gh1285  s
   
zTestRdist.test_rdist_cdf_gh1285c                 C   sN   t ddd}d}tdt|d |d |d d  t|| d S )NgGzrP  r_   g@rf   r6   r   )rc   r   r   r/   rD  r8   r  )r   r?   r   r+   r+   r,   test_rdist_beta  s
   &zTestRdist.test_rdist_betaN)r   r   r   r  r  r+   r+   r+   r,   r    r  r  c                   @   r  )TestTrapezoidc                 C   s`   g d}|D ]'}d|dg}t tj|||tj|| t tj|||tj|| qd S )N)r   r  rf   r   r   r   )r   r/   r   r8   triangrB   )r   Zmodesmoder?   r+   r+   r,   test_reduces_to_triang  s   
z$TestTrapezoid.test_reduces_to_triangc                 C   sN   t ddd}ttj|ddtj| ttj|ddtj| d S Nr   r   r_   )rc   r   r   r/   r   r8   rv  rB   r  r+   r+   r,   test_reduces_to_uniform  s   "z%TestTrapezoid.test_reduces_to_uniformc                 C   s  t tjdddd t tjdddd t tjdddd t tjdddd t tjd	ddd
 t tjdddd t tjdddd
 t tjd	ddd t tjdddd t tjdddd t tjdddd t tjdddd d S )Nr   r6   r   rf   r  r  r)  r   r]   g      ?      ?r'  g      ?r~  g      ?)r   r/   r   r8   rB   r   r+   r+   r,   
test_cases  s    zTestTrapezoid.test_casesc           	         sf  d\              f\}}}}d      fdd}|d}|d|d  }d          t d      }ttj|||||dd ttj|||||dd ttj|||||dd ttjd	d	d
dddd ttjd	dd
dd	dd ttjd	dd
dddd d S )N)r  r`  r6   r   r6   c                    sT   | d  | d     | d   | d        | d  | d  S )Nr6   r   r+   )r   r)   r*   r   dr   r+   r,   rK  6  s    z6TestTrapezoid.test_moments_and_entropy.<locals>.momentr   rf   r  r   r   r  r  r`  r   )rc   r   r   r/   r   r   r(  rN   )	r   r  rG  r4   r5   rK  r   r(  rN   r+   r  r,   test_moments_and_entropy/  s&   0> z&TestTrapezoid.test_moments_and_entropyc                 C   s`  t g d}t ddgd d d f }t g d}tj|||}t |||\}}}t j|j|jd}t 	|j}	t
|	| | | D ]\}
}}}tj|||||
< qIt|||jdd t tjj||dd	}t ||\}}t |jd
f}t 	|j}	t
|	| | D ]\}
}}tjj||dd	||
< qt||j|jdd d S )N)r]   r)  r  rf   r#  )r1  r   r'  )r   rl   rk  rq  rr  r  )rc   r   r/   r   r8   Zbroadcast_arraysemptyr   r   r   r   ravelr   r&  r   rm  r  )r   r   r  r?   r  Zccddr   r  ZindrL  r:  c1Zd1r+   r+   r,   test_trapezoid_vectJ  s"   &z!TestTrapezoid.test_trapezoid_vectc                 C   s0   t ddd}ttj|ddtj| d S r  )rc   r   r   r/   rO   r8   rv  r  r+   r+   r,   
test_trapzd  s   "zTestTrapezoid.test_trapzN)	r   r   r   r  r  r  r  r  r  r+   r+   r+   r,   r    s    	r  c                   @   r   )
TestTriangc                 C   s"  t jdd ttjddd ttjddd ttjddd ttjddd ttjddd ttjddd	 ttjddd ttjddd
 ttjddd ttjddd ttjddd ttjddd W d    d S 1 sw   Y  d S )Nraiser  r   rj   rf   r   r   rk   r6   r   r   )rc   r  r   r/   r  r8   rB   r   r+   r+   r,   test_edge_casesk  s   "zTestTriang.test_edge_casesN)r   r   r   r  r+   r+   r+   r,   r  j  r~  r  c                   @   r  )
TestMielkec                 C   s\   d\}}t t||dtj t t|ddtj ttt|dd d S )N)g rh@gv?r   r   r  )r   r/   mielkerK  rc   r   r   r  )r   r<   r>   r+   r+   r,   r    s    zTestMielke.test_momentsc                 C   s@   t ddd}d\}}ttj|||| tj||| d S )Nrh   r`   r   )g@gHzG@)rc   r   r   r/   burrr8   r  )r   r?   r<   r>   r+   r+   r,   test_burr_equivalence  s   *z TestMielke.test_burr_equivalenceN)r   r   r   r  r  r+   r+   r+   r,   r  ~  r  r  c                   @   r   )TestBurrc              	   C   s   t jddgt jddgt jddgt jddgt jddgt jddgt jd	d
gg}dd |D }dd |D }t|| dd |D }dd |D }t|| d S )N)r   r   )rf   r6   )r   r   )r6   rf   ro  rf   r   )r   r6   rj   c                 S   r  r+   r  r  r+   r+   r,   rI     r  z0TestBurr.test_endpoints_7491.<locals>.<listcomp>c                 S   r  r+   r+   r   r+   r+   r,   rI     r  c                 S   r  r+   )r   r)   r  r+   r+   r,   rI     r  c                 S   s   g | ]
\}}}t |qS r+   )rc   r   r   r+   r+   r,   rI     r  )r/   Zfiskr  burr12r   )r   rv   r  r  r+   r+   r,   test_endpoints_7491  s   






	
zTestBurr.test_endpoints_7491c                 C   s<   d\}}t ||  \}}d\}}t|| t|| d S )N)r   r   )g4U?g&MI?)r/   r  r   )r   r   r  r   varianceZmean_hcZvariance_hcr+   r+   r,   test_burr_stats_9544  s
   
zTestBurr.test_burr_stats_9544c           	      C   sF  d\}}t ||  \}}tt| tt| d\}}t ||  \}}tt| tt| d\}}t jtg d||\}}}}tt| tt| tt| tt| d\}}t jg d||\}}}}tt| tt| tt| tt| d\}}t jg d||\}}}}tt| tt| tt| tt| d\}}t jg d||\}}}}tt| tt| tt| tt| d\}}t jg d||\}}}}tt| tt| tt| tt| d S )N)rf   r   )r  r   )r   r6   r   r  )rU  r   )rV  r   )rY  r   )r/   r  r   rc   r  r  _munpr   )	r   r   r  r   r  Ze1Ze2Ze3Ze4r+   r+   r,   test_burr_nan_mean_var_9544  sL   "z$TestBurr.test_burr_nan_mean_var_9544N)r   r   r   r  r  r  r+   r+   r+   r,   r    s    r  c                   @   s  e Zd Zg dZg dZg dZeeeefZg dZ	ddddd	ej
gZg d
Zeee	eeeZddddej
dfddej
dfddej
dfgZdd Zejjdd ZejeZdZeejeedZe eZ!W d   n1 svw   Y  ej"de!d dd Z#ej"de!d dd Z$ejjej"de!d d d! Z%d"d# Z&ejj'd$d% Z(ej"d&ed'd( Z)ejjd)d* Z*ejj'd+d, Z+dS )-TestStudentizedRange)gQ1@g33333F@g
ףp=*K@gHzM@g/$@g~j!@g33333$@g{Gz&@gh|?5	@gQ8@g&1@g|?5@g@gOn@g!rhm@gB`"@r  gʡEs@g|?5^@g/$@gV-@g/$@gF@g?5^I@)gRV@gffffffl@gp@g     r@gʡ @gHzG/@gQ82@gQ3@g!rh@g     @g+@gʡEs @gjt@gB`"[@g@gnJ@gV-@gFx@gsh|?@g+N@g/$@g @g@gGz@)gfffff"@i  i
  i  gHzG2@g(\A@gQC@gfffffE@g @gNbX9"@gHz$@g(\&@g-@gn@@gw/@g=
ףp @gJ+@g-'@gn@gHz@gV-@gnJ@g$@g/ݤ@)r  rP  rx  r   r   r_   r[  x   )r6   rh  r  r[  )r]   r   i)#  gupm1f?)r   r_   r  g?<-=A?gv"?r  gd暕?ga-A?c                 C   s<   | j D ]\}}|\}}}tj|||}t||dd qd S Nr  rm   )rv   r/   studentized_rangerB   r   )r   pvkr   Z
p_expectedr  r<   Zres_pr+   r+   r,   test_cdf_against_tables  s
   
z,TestStudentizedRange.test_cdf_against_tablesc                 C   s.   | j D ]\}}tjj| }t||dd qd S r  )rv   r/   r  r   r   )r   r  Z
q_expectedZres_qr+   r+   r,   test_ppf_against_tables  s   z,TestStudentizedRange.test_ppf_against_tablesz&data/studentized_range_mpmath_ref.jsonr  Ncase_resultZcdf_datac                 C   N   |d }|d }|d |d |d f}t jj| }t|||d |d d d S 	Nsrc_case	mp_resultr   r<   r  expected_atolexpected_rtolr@  r/   r  rB   r   r   r  r  r  Zqkvr  r+   r+   r,   test_cdf_against_mp     
z(TestStudentizedRange.test_cdf_against_mpZpdf_datac                 C   r  r  )r/   r  r8   r   r  r+   r+   r,   test_pdf_against_mp  r  z(TestStudentizedRange.test_pdf_against_mpZmoment_datac                 C   r  )	Nr  r  r  r<   r  r  r  r@  )r/   r  rK  r   )r   r  r  r  Zmkvr  r+   r+   r,   test_moment_against_mp)  s   
z+TestStudentizedRange.test_moment_against_mpc                 C   s4   d\}}t tjjdtj||fd}t|d d d S )Nr   r_   r   r0  r   )r   r/   r  r8   rc   r   r   )r   r<   r  r  r+   r+   r,   test_pdf_integration5  s   z)TestStudentizedRange.test_pdf_integrationc                 C   s\   d\}}t jdddd}tj|||dd  }tj|||}t||}t||dd d S )	Nr  r   r_   rh   )stepr   r  rm   )rc   r   r/   r  rB   r8   r   r   )r   r<   r  r?   Zy_cdfZ	y_pdf_rawZy_pdf_cumulativer+   r+   r,   test_pdf_against_cdf;  s   
z)TestStudentizedRange.test_pdf_against_cdfr_case_resultc                 C   s*   |\}}}}t j|||}t|| d S rb   r  )r   r  r   r<   r  Zr_resr  r+   r+   r,   test_cdf_against_rK  s   z'TestStudentizedRange.test_cdf_against_rc                 C   st   t jddgddgddg}t|jd tjtdd	 t jdddgg d
 W d    d S 1 s3w   Y  d S )Nr   r6   r  r   r_   r  r-  z...could not be broadcast...r   )r_   r  r   )r/   r  r  r   r   r   r   r*  rK  r+   r+   r,   test_moment_vectorizationR  s
   "z.TestStudentizedRange.test_moment_vectorizationc              	   C   s   t  /}tjdd |t tjg d\}}}}W d    n1 s&w   Y  W d    n1 s5w   Y  ttj	|| d S )Nr  Zinvalidr   )
r
   rc   r  rX   r   r/   r  r<  r   Z	_argcheck)r   r  r<   r  r$  r+   r+   r,   test_fitstart_valid\  s   
 z(TestStudentizedRange.test_fitstart_valid),r   r   r   Zq05Zq01Zq001rc   r  ZqsZpsr   Zvsksr   r"   rv   Zr_datar  r   r7  r8  r  r  r  r  r  Zpath_prefixZrelative_pathopenr  filejsonr  Zpregenerated_datar  r  r  r  r  Zxslowr  r  r  r  r+   r+   r+   r,   r    sN    








	r  c                   C   sR   t tjddddd t tjddddd t tjjdd	d
ddddd d S )Ng2g?r_   test_540_567)r   err_msgg2gk?g;B.?gͣh?gHŎ*?r3   g.y?)r   r/   r  rB   r+   r+   r+   r,   r  e  s   
r  c                   C   s   t jjdd d S )Nrt  r  )r/   Z_continuous_distnsZ	gamma_genr+   r+   r+   r,   test_regression_ticket_1316q  s   r  c                   C   s   t tjdddd d S )Nrk   r6   rf   r  r  r+   r+   r+   r,   test_regression_ticket_1326w  s   r  c               	   C   s  t ddd} t jddE dddt dgdgdggfD ]}tj| |}t|dk  tt 	|   qt dgdgd	gg}tj| |}W d    n1 sUw   Y  tt 	|   t|d
 dk  t|d dk  t|d dk
  t|d dk
  d S )N      r   r^   r  Zdividerk   r  rl  rj   r   r   r6   )rc   r   r  r   r/   r$   r8   r   r   r  any)r?   rb  r   r+   r+   r,   test_regression_tukey_lambda|  s   "	r  zdocstrings strippedr  c                   C   s$   t dtjjv t dtjjv  d S )Nzpdf(x, mu, loc=0, scale=1)zpmf(x,)r   r/   r  r_  r+   r+   r+   r,   test_regression_ticket_1421  s   r  c                   C   s  t jdd tt tjdt j tt tjdt j tt tj	dt j tt tj
dt j tt tjdt j tt tjdt j tt tjdt j tt tjdt j tt tjt jd tt tjt jd tt tj	t jd tt tj
t jd tt tjt jd tt tjt jd tt tjt jd tt tjt jd W d    d S 1 sw   Y  d S )Nr  r  r   rf   )rc   r  r   r  r/   rW  r  r   rB   r  r  r8   r   r   r
  r   r   r  r+   r+   r+   r,    test_nan_arguments_gh_issue_1362  s$   "r  c                  C   sj  t jd t g d} tjj| d | d | d dd}t jdd	 t tjj|d
d}W d    n1 s9w   Y  t	|| dd t tjj|ddd}t	|| dd t tjj|ddd}t	|| dd t tjj|ddd}t	|| dd t jd d}d}tj
j|ddd}t tj
j||d}t |t || d  g}t	||dd d S )Ni.  )r   rk   rf   r   r   r6   r`   r   r  r  rk   rd  r   rf   )r{   r4   r   )r}   r4   r  r'  rj   r  )rc   r   r   r   r/   r   r   r  rr   r   r  rU  r   )truer?   paramsr4   rz   r[   r+   r+   r,   test_frozen_fit_ticket_1536  s(     r  c                  C   s<   t jd tjjdd} tj| }d}t||dd d S )Ni	 r`   r   )g
ףp=
?gʡE?r   r   )rc   r   r   r/   rY  r   rr   r   )r   r  r[   r+   r+   r,   test_regression_ticket_1530  s
   r  c                  C   sV   t jd t jd} dD ]}tj| | \}}t||dd t|ddd qd S )Nr   r!  )r
  g   6Ar   rk  r#  )rc   r   r   Zrandnr/   rY  rr   r   )r?   offsetr4   r5   r+   r+   r,   test_gh_pr_4806  s   r  c                  C   s   t jj ddd} dtjd d ddg}t| |dd t jj d	dd} g d
}t| |dd t jj ddd} g d}t| |dd d S )Nr   rq  rr  r6   r   r(  r_   r   g
ףp=
	@)r   gz5v?r   gRTgQ?)r   gNg @r   gJFٻ)r/   r$   rc   r:   r   )r  r[   r+   r+   r,   "test_tukeylambda_stats_ticket_1545  s   r  c                   C   s   t ttjdd d S )Nr  r  )r   rc   r  r/   r  r  r+   r+   r+   r,   test_poisson_logpmf_ticket_1436  s   r  c                  C   s4   ddg} | D ]\}}t jj |dd}t|| qdS )a  Test the powerlaw stats function.

    This unit test is also a regression test for ticket 1548.

    The exact values are:
    mean:
        mu = a / (a + 1)
    variance:
        sigma**2 = a / ((a + 2) * (a + 1) ** 2)
    skewness:
        One formula (see https://en.wikipedia.org/wiki/Skewness) is
            gamma_1 = (E[X**3] - 3*mu*E[X**2] + 2*mu**3) / sigma**3
        A short calculation shows that E[X**k] is a / (a + k), so gamma_1
        can be implemented as
            n = a/(a+3) - 3*(a/(a+1))*a/(a+2) + 2*(a/(a+1))**3
            d = sqrt(a/((a+2)*(a+1)**2)) ** 3
            gamma_1 = n/d
        Either by simplifying, or by a direct calculation of mu_3 / sigma**3,
        one gets the more concise formula:
            gamma_1 = -2.0 * ((a - 1) / (a + 3)) * sqrt((a + 2) / a)
    kurtosis: (See https://en.wikipedia.org/wiki/Kurtosis)
        The excess kurtosis is
            gamma_2 = mu_4 / sigma**4 - 3
        A bit of calculus and algebra (sympy helps) shows that
            mu_4 = 3*a*(3*a**2 - a + 2) / ((a+1)**4 * (a+2) * (a+3) * (a+4))
        so
            gamma_2 = 3*(3*a**2 - a + 2) * (a+2) / (a*(a+3)*(a+4)) - 3
        which can be rearranged to
            gamma_2 = 6 * (a**3 - a**2 - 6*a + 2) / (a*(a+3)*(a+4))
    )r   )rf   r  rk   r  )rj   )gUUUUUU?r9  g^cQg333333rq  rr  N)r/   powerlawr   )casesr)   Z
exact_mvskrq  r+   r+   r,   test_powerlaw_stats  s   r  c                  C   s   t jdd} t| d d S )Nr   r   rk   )r/   r  r   r   r   r+   r+   r,   test_powerlaw_edge  r~  r  c                  C   sh   t jdd} t| d t jdg d} t| tjddg t jdg d} t| tjdtj g d S )Nr   r   rk   )r   r   r  r   )r/   r}  r   r   r8   rc   r   r  r+   r+   r,   test_exponpow_edge  s   
r  c                  C   sR   t jddd} t| d t jddd} t| d t jddd} t| d d S )	Nr   r   r   r  r  g>r_   gqq?)r/   Zgengammar8   r   r  r   r  r+   r+   r,   test_gengamma_edge+  s   

r  c               	   C   s   t g d} t jdd6 t }|td |td tj	|  W d    n1 s.w   Y  W d    d S W d    d S 1 sFw   Y  d S )N)0g>X*ȿgPix#?g=?g&^?goϿg뵏˿gO_ZQ?g\?g~?g)/?gRܶ?g핚οgH߃[g?gS"gm_ пg7k?g#BF?gKU=?gT?gZg[Qs?g2?g6Hwƭ8?g*-
?ga]?g'?g޻?g?J_ ?g~sw?g0FfͿgg"@ӿg?%:˿g&zM?g?Gg	F?g~,g HgW2ҿgTzy?gwmǿg @gW=ֿg	't?g".v1glg.Hꮿg,?g*˿r  r  z:The maximum number of subdivisions .50. has been achieved.z-floating point number truncated to an integer)
rc   r   r  r
   rX   r   r   r/   r  rr   )r  r  r+   r+   r,   test_ksone_fit_freeze8  s    "r  c                  C   s   t ttddd } g d}tt | |dd tt | d j|dd tj| d d	 j	}t 
tj| tj|  }t||d	d d S )
Nr   r  r  )gDB.g8_$g%`OAgjhwHRgWMB@v`g7VY}igۦx\Arg	xg#g/dg3$gegd)gNq{3gيGǓgr4g37'
gfFg?dJgضg2,
gm%gӅJgz;pgɿ{gk2gij<%giVSͶgBR4g
Q/MrN  rk  y        +=y        |=r|  )rc   rm  listrN  r   r/   r  r  realimagr  r   )r?   r[   ZderivZderiv_expectedr+   r+   r,   test_norm_logcdfP  s   	r  c                  C   sT   t g d} t g d}tj| }t||dd tj|}t|| dd d S )N)r  r   rf   r]   rh   rw  )g~V1?g|N?gba"?g܄4ťY?rJ  g$%+Ir|  rm   r?  )rc   r   r/   r   rB   r   r   )r?   r[   r   r   r+   r+   r,   test_levy_cdf_ppfi  s   r  c                  C   :   t g d} t g d}tj| }t||dd d S )N)g  4&kCg(,* EgMraB3Ggd~QJ)gep[>g}XEQ=gѭEG<g?_%~:r  rm   )rc   r   r/   r   r  r   r?   r[   r   r+   r+   r,   test_levy_sf     r  c                  C   r  )N)gMbg{Gzg{Gztg~jtX)g#]<gOul2;gޢا6g̅7=!r?  rm   )rc   r   r/   r   r  r   r  r+   r+   r,   test_levy_l_sf  r  r  c                  C   s8   t g d} tj| }tj|}t|| dd d S )N)g [n<r   rP  gvIh%,=rm   )rc   r   r/   r   r
  r  r   )r   r?   r   r+   r+   r,   test_levy_l_isf  s   r  c                   C   s|   t tjddddd t tjddddd t tjddddd t tjd	d
d
dd t tjdd
d
dd d S )Nr  i i  i  )g      c@g     h@g=
ףp=?Gz?)g      c@g     h@{Gz?r`   rh  r   )r   r/   r  r  r   r+   r+   r+   r,   test_hypergeom_interval_1802  s   r  c                  C   sr  t jd t jdddd} tttjj| dddd	 tttjj| ddd
dd	 tttjj| ddd
d tttjj| ddddd tttjj	ddddd tttjj
| ddddd tttjj| ddddd tttjjddddd tttjjddddd tttjj| ddddd tj| dd tj| ddd
 tjdd tjddd
 tjddd
d tj	ddd
d tjtjj	dddd tjj| ddd	 tttjj| ddd
 tttjj| ddd
d	 tttjj| ddd	 tttjj| dd
ddd	 tttjj| dd
ddd tttjj| dd
dddd	 tttjj| dd
ddddd	 tj| dd
ddd d S )Nr   r]   gffffff?r   )Znumr6   r   r   r/  r  rf   r3   rj   r  r  r   r  r  )rc   r   r   r   rg  r   r/   rt  r8   r   rB   r   rN   rr   r|  r   r  rt  r  r7  r+   r+   r,   test_distribution_too_many_args  s8   r  c                  C   sH   t jtddddd} t jtddddd}t| |ddd d S )	Nr[  r  r)  r6   gYZ@rw  r   r   )r/   ncx2rB   rc   r   Z_cdfvecr   ri  r+   r+   r,   test_ncx2_tails_ticket_955  s   r  c               	   C   s   t  ) t dt ttjdt	dddd tj
dt	ddd} W d    n1 s0w   Y  tt|   t  % t dt ttjddd	d ttj
ddd	d
 W d    d S 1 sjw   Y  d S )Nr   r   iT  i^  r6   r   r  r   r   gsq6)r   r   r   r   r   r/   r  r8   rc   r   r   r   rx  r   r   )Zlogvalr+   r+   r,   test_ncx2_tails_pdf  s   

"r  zmethod, expectedrB   gu%>g	;Q=r8   g<x>gRz\R>r   gʎK$/gԆX1r   gfHu@ge:@c                 C   s,   t tj| dddgdd}t||dd d S )Nr]   r   r  r_   )rX  r  rl   rk  )r   r/   r  r   )r  r[   r  r+   r+   r,   test_ncx2_zero_nc  s   r  c                  C   s4   t jjdddd} t jjddd}t| |dd d S )Nr_   r   r   )r  rX  r  )r  r  rl   rk  )r/   r  r   r  r   )r  r[   r+   r+   r,   test_ncx2_zero_nc_rvs  s   r  c                  C   s,   dt dd } ttjjdd| dd d S )Nr_   r   r
  r   r  rX  r   )rc   r   r   r/   r  rB   )rX  r+   r+   r,   test_ncx2_gh12731  s   r  c                  C   sB   t g d} d\}}tjj| ||d}g d}t||dd d S )N)gю]	@gj%@gb	y/7@g H@gIZ@gXl@g'/2?8@gݰmА@gU@gy]`'@g@g*\@g48c@)r[  gh ?8@r  )r   r   r   r   r   g?gy[qD?rk   rk   rk   rk   rk   rk   r  rk  )rc   r   r/   r  r  r   )r?   nurb  r  Zsf_expectedr+   r+   r,   test_ncx2_gh8665  s
   
r  c                  C   s"   t jddd} t| dd d S )Nr   r   r  )r/   Zfoldnormr   rB   r!  r+   r+   r,   test_foldnorm_zero.  s   r  c                  C   s   t j g ddd} t j ddgdd}tdd |D }t||  t j g d} t j dd	g}td
d |D }t||  d S )N)rk   rf   r   r   rf   r   c                 s   s     | ]}t jt j|f V  qd S rb   rc   r   r   rJ  r+   r+   r,   r4  9  r9  z-test_stats_shapes_argcheck.<locals>.<genexpr>)r6   r%  r`  r6   r%  c                 s   s     | ]}t j|t jf V  qd S rb   r  rJ  r+   r+   r,   r4  ?  r9  )r/   r   tupler   r   )Zmv3Zmv2Zmv2_augmentedr+   r+   r,   test_stats_shapes_argcheck4  s   
r  c                   @   r   )
_distr_genc                 C   r5  Nr!  r+   r   r?   r)   r+   r+   r,   _pdfJ  r)  z_distr_gen._pdfNr   r   r   r  r+   r+   r+   r,   r  I  r~  r  c                   @   r   )_distr2_genc                 C      d| | S r  r+   r  r+   r+   r,   _cdfO     z_distr2_gen._cdfN)r   r   r   r  r+   r+   r+   r,   r   N  r~  r   c                   @   r  )_distr3_genc                 C   s   || S rb   r+   r   r?   r)   r*   r+   r+   r,   r  T  s   z_distr3_gen._pdfc                 C   r  r  r+   r  r+   r+   r,   r  W  s   z_distr3_gen._cdfNr   r   r   r  r  r+   r+   r+   r,   r  S  r  r  c                   @   r  )_distr6_genc                 C   s   || | S rb   r+   r  r+   r+   r,   r  _  r  z_distr6_gen._pdfc                 C   r  r  r+   r  r+   r+   r,   r  b  r  z_distr6_gen._cdfNr  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#S )$TestSubclassingExplicitShapesc                 C   s$   t ddd}t|jdddd d S )Ndummyr)   r  r   r   r  r!  r  r   r8   r   dummy_distrr+   r+   r,   test_correct_shapesi  s   z1TestSubclassingExplicitShapes.test_correct_shapesc                 C   s,   t ddd}tt|jdfi tdd d S )Nr	  Ar
  r   r  )r  rg  r   r8   r   r  r+   r+   r,   test_wrong_shapes_1m  s    z1TestSubclassingExplicitShapes.test_wrong_shapes_1c                 C   s4   t ddd}tdddd}tt|jdfi | d S )Nr	  za, b, cr
  r   r6   r   r  )r  r   rg  r   r8   )r   r  r   r+   r+   r,   test_wrong_shapes_2q  s   z1TestSubclassingExplicitShapes.test_wrong_shapes_2c                 C   "   t ddd}tttfi | d S )Nr	  r!  r
  )r   rg  r   r  r   r   r+   r+   r,   test_shapes_stringv     z0TestSubclassingExplicitShapes.test_shapes_stringc                 C   r  )Nr	  z(!)r
  r   rg  SyntaxErrorr  r  r+   r+   r,   test_shapes_identifiers_1{  r  z7TestSubclassingExplicitShapes.test_shapes_identifiers_1c                 C   r  )Nr	  Z4chanr
  r  r  r+   r+   r,   test_shapes_identifiers_2     z7TestSubclassingExplicitShapes.test_shapes_identifiers_2c                 C   r  )Nr	  zm(fti)r
  r  r  r+   r+   r,   test_shapes_identifiers_3  r  z7TestSubclassingExplicitShapes.test_shapes_identifiers_3c                 C   r  )Nr	  za=2r
  r  r  r+   r+   r,   "test_shapes_identifiers_nodefaults  r  z@TestSubclassingExplicitShapes.test_shapes_identifiers_nodefaultsc                 C   r  )Nr	  z*argsr
  r  r  r+   r+   r,   test_shapes_args  r  z.TestSubclassingExplicitShapes.test_shapes_argsc                 C   r  )Nr	  z**kwargsr
  r  r  r+   r+   r,   test_shapes_kwargs  r  z0TestSubclassingExplicitShapes.test_shapes_kwargsc                 C   r  )Nr	  za, b, c, lambdar
  r  r  r+   r+   r,   test_shapes_keywords  r  z2TestSubclassingExplicitShapes.test_shapes_keywordsc                 C   s@   G dd dt j}|dd}t|jdddt jdd  d S )Nc                   @   r   )zFTestSubclassingExplicitShapes.test_shapes_signature.<locals>._dist_genc                 S      t j|| S rb   r/   r  r  r  r+   r+   r,   r    r   zKTestSubclassingExplicitShapes.test_shapes_signature.<locals>._dist_gen._pdfNr  r+   r+   r+   r,   	_dist_gen  r~  r"  r)   r   rf   r6   r  r/   rL   r   r8   r  r   r"  rH   r+   r+   r,   test_shapes_signature  s   
$z3TestSubclassingExplicitShapes.test_shapes_signaturec                 C   s>   G dd dt j}|dd}tt|jdfi tddd d S )	Nc                   @   r   )zSTestSubclassingExplicitShapes.test_shapes_signature_inconsistent.<locals>._dist_genc                 S   r   rb   r!  r  r+   r+   r,   r    r   zXTestSubclassingExplicitShapes.test_shapes_signature_inconsistent.<locals>._dist_gen._pdfNr  r+   r+   r+   r,   r"    r~  r"  a, br#  rf   r   r6   ri  )r/   rL   rg  r   r8   r   r%  r+   r+   r,   "test_shapes_signature_inconsistent  s   
"z@TestSubclassingExplicitShapes.test_shapes_signature_inconsistentc                 C   sz   G dd dt j}|dd}t|jdddt jdd  t|ddt jdd  tt|jdfi tdd d S )	Nc                   @   r   )z?TestSubclassingExplicitShapes.test_star_args.<locals>._dist_genc                 W   s   |d }t j|| S Nr   r!  )r   r?   r   extra_kwargr+   r+   r,   r    s   zDTestSubclassingExplicitShapes.test_star_args.<locals>._dist_gen._pdfNr  r+   r+   r+   r,   r"    r~  r"  r*  r#  rf   !   r*  )Zxxx)r/   rL   r   r8   r  rg  r   r   r%  r+   r+   r,   test_star_args  s
   
  z,TestSubclassingExplicitShapes.test_star_argsc                 C   sj   G dd dt j}|dd}t|jddddt jdd d  t|dddt jdd d  d S )	Nc                   @   r   )zATestSubclassingExplicitShapes.test_star_args_2.<locals>._dist_genc                 W   s   |d }t j|| | S r)  r!  )r   r?   r  r   r*  r+   r+   r,   r    s   zFTestSubclassingExplicitShapes.test_star_args_2.<locals>._dist_gen._pdfNr  r+   r+   r+   r,   r"    r~  r"  zoffset, extra_kwargr#  rf   o   r+  )r  r*  r$  r%  r+   r+   r,   test_star_args_2  s   
z.TestSubclassingExplicitShapes.test_star_args_2c                 C   s<   G dd dt j}|dd}t|jdddt jd d S )Nc                   @   r   )zBTestSubclassingExplicitShapes.test_extra_kwarg.<locals>._distr_genc                 _   s   | dd}tj|| S )Nr*  r   )popr/   r  r  )r   r?   r   kwargsr*  r+   r+   r,   r    s   zGTestSubclassingExplicitShapes.test_extra_kwarg.<locals>._distr_gen._pdfNr  r+   r+   r+   r,   r    r~  r  r*  r#  r   r   r,  r$  )r   r  rH   r+   r+   r,   test_extra_kwarg  s   
 z.TestSubclassingExplicitShapes.test_extra_kwargc                 C   s8   G dd dt j}|dd}t|dt jd d S )Nc                   @   r   )zDTestSubclassingExplicitShapes.shapes_empty_string.<locals>._dist_genc                 S   s   t j|S rb   )r/   r  r8   r  r+   r+   r,   r    r  zITestSubclassingExplicitShapes.shapes_empty_string.<locals>._dist_gen._pdfNr  r+   r+   r+   r,   r"    r~  r"   r#  rf   r$  r%  r+   r+   r,   shapes_empty_string  s   
z1TestSubclassingExplicitShapes.shapes_empty_stringN)r   r   r   r  r  r  r  r  r  r  r  r  r  r  r&  r(  r-  r/  r2  r4  r+   r+   r+   r,   r  f  s$    		r  c                   @   sl   e Zd Zdd Zdd Zejjedddd Z	ejjeddd	d
 Z
dd Zdd Zdd Zdd ZdS )TestSubclassingNoShapesc                 C   s"   t dd}t|jdddd d S )Nr	  r  r   r  r!  r  r  r+   r+   r,   test_only__pdf  s   
z&TestSubclassingNoShapes.test_only__pdfc                 C   s"   t dd}t|jdddd d S )Nr	  r  r   r  )r   r   r8   r  r+   r+   r,   test_only__cdf  s   
z&TestSubclassingNoShapes.test_only__cdfdocstring strippedr  c                 C   sD   t dd}t|jd t|jd td|j}tt|dk d S )Nr	  r  r   r)   zlogpdf\(x, a, loc=0, scale=1\))	r  r   numargsr   refindallr_  r   r   r   r  r  r+   r+   r,   test_signature_inspection  s   
z1TestSubclassingNoShapes.test_signature_inspectionc                 C   sD   t dd}t|jd t|jd td|j}tt|dk d S )Nr	  r  r6   r'  z!logpdf\(x, a, b, loc=0, scale=1\)r   )	r  r   r9  r   r:  r;  r_  r   r   r<  r+   r+   r,   test_signature_inspection_2args  s   
z7TestSubclassingNoShapes.test_signature_inspection_2argsc                 C   s   t ttdd d S )Nr	  r  )rg  r   r  r   r+   r+   r,   0test_signature_inspection_2args_incorrect_shapes  r2   zHTestSubclassingNoShapes.test_signature_inspection_2args_incorrect_shapesc                 C   .   G dd dt j}tt|fi tdd d S )Nc                   @   s   e Zd ZdddZdS )z>TestSubclassingNoShapes.test_defaults_raise.<locals>._dist_genr!  c                 S   r5  r  r+   r  r+   r+   r,   r    r)  zCTestSubclassingNoShapes.test_defaults_raise.<locals>._dist_gen._pdfN)r!  r  r+   r+   r+   r,   r"    s    r"  r	  r  r/   rL   rg  r   r   r   r"  r+   r+   r,   test_defaults_raise     z+TestSubclassingNoShapes.test_defaults_raisec                 C   r@  )Nc                   @   r   )z>TestSubclassingNoShapes.test_starargs_raise.<locals>._dist_genc                 W   r5  r  r+   )r   r?   r)   r   r+   r+   r,   r    r)  zCTestSubclassingNoShapes.test_starargs_raise.<locals>._dist_gen._pdfNr  r+   r+   r+   r,   r"    r~  r"  r	  r  rA  rB  r+   r+   r,   test_starargs_raise
  rD  z+TestSubclassingNoShapes.test_starargs_raisec                 C   r@  )Nc                   @   r   )z<TestSubclassingNoShapes.test_kwargs_raise.<locals>._dist_genc                 [   r5  r  r+   )r   r?   r)   r1  r+   r+   r,   r    r)  zATestSubclassingNoShapes.test_kwargs_raise.<locals>._dist_gen._pdfNr  r+   r+   r+   r,   r"    r~  r"  r	  r  rA  rB  r+   r+   r,   test_kwargs_raise  rD  z)TestSubclassingNoShapes.test_kwargs_raiseN)r   r   r   r6  r7  r   r7  r  DOCSTRINGS_STRIPPEDr=  r>  r?  rC  rE  rF  r+   r+   r+   r,   r5    s    


	r5  r8  c                  C   sT   g d} t jD ] }tt |}t|t jt jfr'| D ]}tt||j	d u  qqd S )N)z,\s*,z\(\s*,z^\s*:)
r/   rW   r   r   rK   rL   r   r:  searchr_  )ZbadonesdistnamerH   Zregexr+   r+   r,   r    s   

r  c                   C   s4   t tjtjddd t tjtjddd d S )Nr_   r  r   rh  r]   r   )r   r/   r  r  rc   r   r  r  r+   r+   r+   r,   test_infinite_input#  s   rJ  c                  C   &   t jt jddd} t| d d S Ng0.++r   )r/   Zlomaxr   rB   r   r  r+   r+   r,   test_lomax_accuracy(     rM  c                  C   rK  rL  )r/   Zgompertzr   rB   r   r  r+   r+   r,   test_gompertz_accuracy.  rN  rO  c                  C   rK  rL  )r/   Z
truncexponr   rB   r   r  r+   r+   r,   test_truncexpon_accuracy4  rN  rP  c                  C   s*   t jt jddd} t| ddd d S )N	   r   r  r   r   )r/   r   r
  r  r   r  r+   r+   r,   test_rayleigh_accuracy:  s   rR  c                  C   s   t jdd5} t d tjdd tjdd tjdd tjt	j
 d t| }t|d W d   dS 1 s>w   Y  dS )zregression test for gh-6219T)r  alwaysrf   r   rk   N)r   r   r   r/   r  rB   r8   r   r   rc   r   r   r   )r3  Znumber_of_warnings_thrownr+   r+   r,    test_genextreme_give_no_warnings@  s   
"rT  c                  C   s   d} t jd}t|d|  d dd t jd}t|| d dd t jd}t|d t jjd	d
d}t|| d td
 d dd t jd
}t|d|  d dd t jd}t|d|  d dd d S )Ngox?r  r6   r   r  rm   r   r   rl  r_   r  r   r  r>  r  )r/   r  rN   r   r   rc   r   )Zeuler_gammar   r+   r+   r,   test_genextreme_entropyN  s   
 rU  c                  C   s   d} t j| d}t|d t j|d}t||  d} t j| d}t|d t j|d}t||  d} t j| d}t|d t j|d}t||  d S )	Ng    חAg      g'b%4gQ@r~  gؗҜ;r   g.l6?)r/   r  r  r   r
  )r?   r>   r;  r+   r+   r,   test_genextreme_sf_isfe  s   




rV  c                  C   s"   d} t j| dd}t|d d S )NgؗҜ<r6   r   gH&8>)r/   r  r   r   )Zprobr  r+   r+   r,   test_burr12_ppf_small_arg  s   	rW  c                  C   s6  t ddddd } tjj| ddd}t g d	}t||d
d tjj| ddd}t g d}t||d
d tjj| ddddd}t g d}t||d
d tjj| ddd}t g d}t||d
d tjj| ddd}t g d}t||d
d tjj| ddddd}t g d}t||d
d dS )z
    All values are calculated using the independent implementation of the
    ROOT framework (see https://root.cern.ch/).
    Corresponding ROOT code is given in the comments.
    r  r      Nr`  r   rj   )rD  r  )gC,d	Ɣ?gH㸘?g5?gM"M#w?gɫs^?g @?gƴ?g@w?ɫs^?|?q ?g C?rZ  rY  gJó?g,BV\ۤ?g;?g#ek?g%ǓE?g(Ȍx?g>J>rw  rm   rp  )g#{`?gdf?g Jq?gSS){?g'	w?uL? T)?cAJ?GTn?R1#?gA]P?r_  r^  r]  r\  gvӢ?gKq?gL?g
	N<!?gAw>rf   )rD  r  r4   r5   )g=aO}?gI8?g#T?aW)?ǓI?
J?E?d۳?GTn?8?R1#?kP?g(*T?rg  rf  re  rd  rc  rb  ra  r`  )gD
)?g:3P?g9?g?S[?rY  gOqx?gD
)?g9?gɫs^?gK?gX?g3<֌?gM+?gyS
c?gI`s?g辜?g{i ?g^0?g?#c?)g%r?gc@v?ghtj}?g(˔^?gHK4U?r[  ğ@ _?y}D?焂Rr/?N?YjhG?a0L?4f?<@?@CH?g	ސF?g5+-#?g?g?q ?rh  )g}'v?r[  g*V?ri  gr/0+?rj  g[8	?rk  gΥ?rl  gY?rm  g#)a?rn  g?̔?ro  g]ؚ?rp  g|c ?rq  )rc   r   r/   crystalballr8   r   r   rB   )rF  Z
calculatedr[   r+   r+   r,   test_crystalball_function  s&   rs  c                  C   s  t g d} t g d}t g d}tjd| |}t||dd t g d}t dd	t jd
dg}|| }tjd| |}t||dd t t jt jt jddg}|| }tjd| |}	t||	dd t t jt jt jt jdg}|| }
tjd| |}t|
|dd t t jt jt jt jdg}|| }tjd| |}t||dd t t jt jt jt jdg}|| }tjd| |}t||dd dS )zg
    All values are calculated using the pdf formula and the integrate function
    of Mathematica
    )rj   r   rp  rj   rp  )rp  rp  rj   r$  r  )r   r   r   r   r   r   rw  rm   )gԚh@g[&@gͪ@g2dQ@gn6-@g?ޫV&̿g BgQCRgQF\ jr   g
@gn(@r6   g?Җ)r   gMg'#@r  g('UHr   N)rc   r   r/   rr  r  r   r   )rD  r  Zexpected_0th_momentZcalculated_0th_momentr  r)   Zexpected_1th_momentZcalculated_1th_momentZexpected_2th_momentZcalculated_2th_momentZexpected_3th_momentZcalculated_3th_momentZexpected_4th_momentZcalculated_4th_momentZexpected_5th_momentZcalculated_5th_momentr+   r+   r,   !test_crystalball_function_moments  s4   rt  c                  C   sR   t dd} |  }d\}}}t|||}tt| ||}t||dd d S )Nr6   r   )ir   i@ Hz>rm   )	r/   rr  rN   rc   r   r   r   r8   r   )cbr@  rK  rL  r  r?   r  r+   r+   r,   test_crystalball_entropy  s   
rw  c                  C   s\   ddd} t g d}tjj|d| d\}}}t|ddd	 |dks%J t|d
dd	 dS )a?  
    Test fitting invweibull to data.

    Here is a the same calculation in R:

    > library(evd)
    > library(fitdistrplus)
    > x = c(1, 1.25, 2, 2.5, 2.8,  3, 3.8, 4, 5, 8, 10, 12, 64, 99)
    > result = fitdist(x, 'frechet', control=list(reltol=1e-13),
    +                  fix.arg=list(loc=0), start=list(shape=2, scale=3))
    > result
    Fitting of the distribution ' frechet ' by maximum likelihood
    Parameters:
          estimate Std. Error
    shape 1.048482  0.2261815
    scale 3.099456  0.8292887
    Fixed parameters:
        value
    loc     0

    r+   r   c                 S   s   t | |||dddS )Nr  )r   dispZxtolZftol)r!   )rw   x0r   rx  r+   r+   r,   	optimizer6  s   z"test_invweibull.<locals>.optimizer)r   r  r6   rU  r  r   r  r  r   rh  r_   r   @   c   )rz   rz  g`?r  rm   g [@N)r+   r   )rc   r   r/   r   rr   r   )rz  r?   r   r4   r5   r+   r+   r,   test_invweibull  s   
r}  z	df1,df2,xr6   )r/  r)  r   r  r  r  r  rE  r)  c                 C   sh   d}t j|| |}t j|| ||}t||dd t j|| |}t j|| ||}t||dd d S )Nr   r  rm   rI  )r/   r  rB   r  r   r8   )Zdf1Zdf2r?   rX  Zexpected_cdfZcalculated_cdfrp   Zcalculated_pdfr+   r+   r,   test_ncf_edge_case@  s   
r~  c                  C   s"   t jddd} t| ddd d S )Nr6   r  r  g     `E@r  rm   )r/   r  r(  r   )r  r+   r+   r,   test_ncf_varianceV     r  c                   @   r  )TestHistogramc                 C   s\   t jd t jg ddd}t|| _tjjddddd	}t j|d
d}t|| _	d S )Nr   )r   r6   r6   r   r   r   r  r  r  r  r   r   r   r   r   r  r  r  r  r  r  r  rh  rh  rQ  rh  Zbinsr   rU  r  {   )r4   r5   r   r  r   )
rc   r   r   	histogramr/   rM   templater  r   norm_template)r   r  rv   Znorm_histogramr+   r+   r,   r   b  s   
zTestHistogram.setup_methodc                 C   s   t g d}t g d}t| j|| t| jdd t| jdd t| jdd t| jdd t d	d
d}t| j|t	j
j|ddddd d S )Nrk   rf   r   r  rj   rU  rp  rV  r$  rY  r         @ra  r        @      @r        !@r  g      #@)rk   rk   r   r   {Gz?r  Q?r  {Gz?r  r)  r)  r  r  r  r  r  r  rk   rk   r  r  r  r  rk   rX  r  r6   r_   r   rU  r3   r]   rm   )rc   r   rm  r   r  r8   r   r   r  r/   r  )r   r  Z
pdf_valuesr?   r+   r+   r,   r   r  s   
zTestHistogram.test_pdfc                 C   s   t g d}t g d}t| j|| t| j|dd |dd  t ddd}t| j| j|| t ddd}t| j| j|| t d	dd
}t| j|t	j
j|ddddd d S )Nr  )rk   rk   rk   r  r   r  r  
ףp=
?Q?{Gz?r  rf   r#  (\?RQ?=
ףp=?)\(?r  r   r   r6   r`  r   r  r`   rk   r  r_   rU  r3   r]   rm   )rc   r   rm  r   r  rB   r   r   r  r/   r  )r   r  Z
cdf_valuesr?   r+   r+   r,   r    s   "
zTestHistogram.test_cdf_ppfc                 C   s   d}| j j|dd}tt|dk d tt|dkd| dd	 tt|d
kd| dd	 tt|dkd| dd	 tt|dkd| dd	 tt|dkd| dd	 tt|dkd| dd	 tt|dkd| dd	 tt|dkd| dd	 tt|dkd| dd	 tt|dkd| dd	 tt|dkd| dd	 tt|d kd!| dd	 tt|d"kd#| dd	 tt|d$kd%| dd	 tt|d&kd| dd	 tt|d&kd| dd	 tt|d&kd d S )'Nr  r  r  r   rk   rj   r   r)  rm   rU  r  rp  r  r]   rV  r  r$  r  rY  r  r   r  r  r  rf   ra  r#  r  r  r  r  r  r  r  r  r  r  r  )r  r   r   rc   r   r   )r   r  r  r+   r+   r,   r     s(   zTestHistogram.test_rvsc                 C   s6   t dD ]}t| j|tdd|dd qd S )Nr  r   rU  r  rm   )rN  r   r  r  r/   r  rK  )r   r   r+   r+   r,   	test_munp  s
   zTestHistogram.test_munpc                 C   s$   t | j tjjddddd d S )Nr   rU  r3   r  rm   )r   r  rN   r/   r  r   r+   r+   r,   r     s   

zTestHistogram.test_entropyN)	r   r   r   r   r   r  r   r  r   r+   r+   r+   r,   r  a  s    r  c                  C   s   t dd} | jddd}tjt|dd\}}d|   kr-|   kr-d	ks0J  J tt	|d
 dks>J d S )Nrw  r   r  r!  r  r_   r  rl  iL  r  )
r/   Z
loguniformr   rc   r  Zlog10r  r  r  r  )rt  r   r   r$  r+   r+   r,   test_loguniform  s
   . r  c                   @   s  e Zd Zdd Zejdddgddgdd	ggd
d Zejdddgdd Zejdg ddd Z	ejdg ddd Z
ejdg ddd Zejdg ddd  Zejdg d!d"d# Zejdg d$d%d& Zejdg d'd(d) Zd*S )+	TestArgusc                 C   s2   t jjdddd}tt d | dd d S )Nr   r  E  r  r  r   )r/   argusr   r   r   r  r+   r+   r,   test_argus_rvs_large_chi  s    z"TestArgus.test_argus_rvs_large_chizchi, random_stater]   r  rk     rV     c                 C   s6   t jj|d|d}t |d|f\}}t|dk d S )Nr  r  r  r  r/   r  r   r!  r   )r   r  r  r?   r$  r   r+   r+   r,   r     s   zTestArgus.test_rvsr  rj  rI  c                 C   s6   t jj|ddd}t |dd \}}t|dk d S )Nr  ie r  c                 S   s   dd| d  d  S )Nr   r6   r  r+   r7  r+   r+   r,   rT     rJ   z.TestArgus.test_rvs_small_chi.<locals>.<lambda>r  r  )r   r  r  r$  r   r+   r+   r,   test_rvs_small_chi  s   zTestArgus.test_rvs_small_chizchi, expected_mean))r   g щi?)r_   g.憃?)r  g\paP?)<   g	?)r|  g,6?c                 C   s"   t jj|dd}t||dd d S )Nr   r  r?  rm   )r/   r  r   r   )r   r  Zexpected_meanr  r+   r+   r,   r    r  zTestArgus.test_meanzchi, expected_var, rtol))r   gLH'B?r?  )r_   go$?r\  )r  g
=>rN  )r  g2W>rN  )r|  gb~P>rN  c                 C   s"   t jj|dd}t|||d d S )Nr   r  rm   )r/   r  r(  r   )r   r  Zexpected_varrn   r  r+   r+   r,   r    r  zTestArgus.test_varzchi, expected, rtol))r'  gwM?r  )rf   gj?r  )r]   gHEA`!?r  )rh   gcف>rl   )rw  gF=r  )r  g߹Q#B=r  )rI  g0ޟ<r  )rj  g*W%:rl   c                 C   s   t t|||d d S )Nrm   )r   r   )r   r  r[   rn   r+   r+   r,   test_argus_phi_small_chi  s   
z"TestArgus.test_argus_phi_small_chizchi, expected))rf   )g-\/?g+p7c?g5Ϧ\?)r)  )g
f?gi{?ٸ?gZ@7?)r]   )gB	?g]?gR{?)rh   )glͅw?g,?g[Z=?)rw  )gz,?g]?gѪ1?)r  )g8<?gE-?gI?)rI  )g~7?g7B.?g9?)rj  )g;?g9B.?g}5?c                 C   *   t g d}ttj|||dd d S )Nr]   rf   r'  r?  rm   )rc   r   r   r/   r  r8   r   r  r[   r?   r+   r+   r,   test_pdf_small_chi     zTestArgus.test_pdf_small_chi))rf   )gQ߹?e?grJ,?g򺗄?)r)  )gΎd?g?gh?)r]   )g|
o?gvb?g+@?)rh   )g̈́On?ge?gC|[3?)rw  )g!?m?g{jiH?gu53?)r  )g~m?g0-.?g3?)rI  )g~m?g7;B.?gŗ3?)rj  )g~m?g9B.?gŗ3?c                 C   r  )Nr  r  rm   )rc   r   r   r/   r  r  r  r+   r+   r,   test_sf_small_chi  r  zTestArgus.test_sf_small_chi))rf   )g+&?gk?gμm/?)r)  )g?CL\f?g^N?g]AͭR?)r]   )ga>$?g;Lf?gaW?)rh   )gx?g 43n?gwY?)rw  )go7🤎?g
+-oGn?gIQ9Y?)r  )gT?gGn?g@̋Y?)rI  )gU?g{Gn?giG͋Y?)rj  )gDU?g {Gn?gH͋Y?c                 C   r  )Nr  r  rm   )rc   r   r   r/   r  rB   r  r+   r+   r,   test_cdf_small_chi&  r  zTestArgus.test_cdf_small_chi))rf   )gQ)?g6{?r  )gB`"۹?)g?g8%?r\  )r]   )g388?gV%?r?  )rh   )g*j?gRS%?r?  )rw  )g!|?gZ%?r?  )r  )gy[|?gG%?r?  )rI  )g"3|?gW%?r?  )rj  )g!3|?gW%?r?  c                 C   s"   t jj |dd}t|||d d S )Nr  rr  rm   )r/   r  r   )r   r  r[   rn   r   r+   r+   r,   test_stats_small_chi5  s   zTestArgus.test_stats_small_chiN)r   r   r   r  r   r7  r  r   r  r  r  r  r  r  r  r  r+   r+   r+   r,   r    sT    

	


	






r  c                   @   s   e Zd Zdd Zdd Zejdg dejdg dejd	g d
dd Zejdg dejd	g d
dd Z	dS )TestNakagamic                 C   s$   d}d}t j||}t|d d S )NrU  r  gM+)r/   nakagamir   r   )r   r  r?   r  r+   r+   r,   r  F  s   zTestNakagami.test_logpdfc                 C   sD   d}d}t j||}t|ddd t j||}t||dd d S )NrU  r   g"S+:r?  rm   )r/   r  r  r   r
  )r   r  ry  r  r:  r+   r+   r,   r!  W  s   zTestNakagami.test_sf_isfr  )g?rU  r	  r4   )ri   r_   #   r5   )r  r   r[  c           
         s   d t jj |||ddt j\}}}t||dd t||dd t||dd fdd} fdd	} fd
d}	t||||ddd t||||ddd t|	|||ddd d S )Nr`   9  r   r  r4   r5   r  r)  rm   c                    s<   d|  d t d |   d|  |d  t  |   S )Nr  r   r6   rc   r   r  r4   r5   )r  r+   r,   	dlogl_dnux  s   z(TestNakagami.test_fit.<locals>.dlogl_dnuc                    sN    dt |  td|   dt t | |   t | | d  S )Nr   r   r6   )rc   r   r   r   r  r  r  r+   r,   
dlogl_dloc|  s
   z)TestNakagami.test_fit.<locals>.dlogl_dlocc                    s2   d  |  | d|  |d  t | d   S )Nr  r6   r   r  r  r  r+   r,   dlogl_dscale  s    z+TestNakagami.test_fit.<locals>.dlogl_dscaler   rw  rk  )r/   r  r   rr   r   )
r   r  r4   r5   nu_estloc_est	scale_estr  r  r  r+   r  r,   r?  j  s   zTestNakagami.test_fitc                 C   s   d}d}t jj||||dd}t jj||d\}}}t|}	tt|| d }
t||dd t||	dd t||
dd d S )	Nrf   r`   r  r  r  r6   ru  rm   )	r/   r  r   rr   rc   r  rU  r   r   )r   r4   r5   r  r   r  r  r  r  Zloc_theoZ
scale_theor+   r+   r,   test_fit_nu  s   
zTestNakagami.test_fit_nuN)
r   r   r   r  r!  r   r7  r  r?  r  r+   r+   r+   r,   r  D  s    r  c                   @   r   )TestWrapCauchyc                 C   sr   t ddgddgg}t dgdgg}tj||}|jdks"J dd	 t ||fD }t| |d
d d S )NQ?r   rf   r   r   r$  r-  c                 S   s   g | ]\}}t j||qS r+   )r/   
wrapcauchyrB   )rG   r:  r  r+   r+   r,   rI     s    z>TestWrapCauchy.test_cdf_shape_broadcasting.<locals>.<listcomp>r?  rm   )	rc   r   r/   r  rB   r   Znditerr   r  )r   r   r?   r   Zscalar_valuesr+   r+   r,   test_cdf_shape_broadcasting  s   z*TestWrapCauchy.test_cdf_shape_broadcastingc                 C   s"   t jtjd}t|ddd d S )Nr  rf   r  rm   )r/   r  rB   rc   r:   r   r   r+   r+   r,   test_cdf_center  r  zTestWrapCauchy.test_cdf_centerc                 C   s   d}d}d}t j||g|}d| d|  }t|d t|t|d  tj  t|d dt|ttj|d   tj   d S )Nr   r$  r   r   r   r6   )r/   r  rB   r   rc   ZarctanZtanr:   )r   r:  r;  r   r   Zcrr+   r+   r,   r     s   (6zTestWrapCauchy.test_cdfN)r   r   r   r  r  r   r+   r+   r+   r,   r    s    r  c                  C   sT   G dd dt j} | dd}ttj |  W d    d S 1 s#w   Y  d S )Nc                   @   r   )z1test_rvs_no_size_warning.<locals>.rvs_no_size_genc                 S   r5  r	  r+   r   r+   r+   r,   _rvs  r)  z6test_rvs_no_size_warning.<locals>.rvs_no_size_gen._rvsN)r   r   r   r  r+   r+   r+   r,   rvs_no_size_gen  r~  r  rvs_no_sizer  )r/   rL   r   rc   ZVisibleDeprecationWarningr   )r  r  r+   r+   r,   test_rvs_no_size_warning  s
   

"r  zdistname, argsc                 C   s   | t v rtd|  d tt| }t|tjrQt|dkr1|j| \}}t	|t
j t	|t
j d\}}|jg |||R  \}}t	|t
j t	|t
j d S |j| \}	}
t	|	t
j t	|
t
j d S )Nz6skipping test for the support method for distribution .r   r{  )$skip_test_support_gh13294_regressionr   r  r   r/   r   rL   r   supportr   rc   r   )rI  r   rH   a0b0r  r  a1b1r)   r*   r+   r+   r,   test_support_gh13294_regression  s"   

r  c                  C   sX  t jg dg d\} }ttj tj tj tjg}ttjtjtjtjg}t| | t|| | j|jks<J |j|jksDJ t jg g \}}tg tg }}t|| t|| |j|jksjJ |j|jksrJ t jg ddg\}}	tdtjg }
tdtjg }t||
 t|	| |j|
jksJ |	j|jksJ d S )N)r   r   r   r   )r   r   r   r`  r`  r  )	r/   r  r  rc   r   r   r   r   r   )r  r  Zex_a0Zex_b0r  r  Zex_a1Zex_b1r  r  Zex_a2Zex_b2r+   r+   r,   ,test_support_broadcasting_gh13294_regression  s(    





r  c                  C   sn   ddg} dgdgdgg}t tj| |ddgddgddgg td} td}tj| |jdks5J d S )	Nrk   r   rj   rp  r$  r  rE   )r   r   )r   r/   r  r(  rc   r  r   r3   r+   r+   r,   *test_stats_broadcasting_gh14953_regression  s   &

r  r  )gn!	g:I"<)gQ	@gE?c                 C   s*   t tj| | t tj|  | d S rb   )r   r/   cosinerB   r  )r?   r[   r+   r+   r,   test_cosine_cdf_sf  s   r  zp, expected))rI  gkM6O)r  g0ӭ!	)g333333?g;'u(@c                 C   s*   t tj| | t tj| |  d S rb   )r   r/   r  r   r
  )r   r[   r+   r+   r,   test_cosine_ppf_isf  s   r  c                  C   sT   dd t D } dd tD }| |ksJ dd tD }dd tD }||ks(J d S )Nc                 S   s   h | ]\}}t |tr|qS r+   )r   rQ   rG   r  r$  r+   r+   r,   	<setcomp>  s    
z*test_distr_params_lists.<locals>.<setcomp>c                 S      h | ]\}}|qS r+   r+   r  r+   r+   r,   r    rJ   c                 S   r  r+   r+   r  r+   r+   r,   r     rJ   c                 S   r  r+   r+   r  r+   r+   r,   r  !  rJ   )r   r   r   r   )Zdiscrete_distnamesZinvdiscrete_distnamesZcont_distnamesZinvcont_distnamesr+   r+   r,   test_distr_params_lists  s   r  rb   )r_  r   r:  r  r#  r  r  platformZnumpy.testingr   r   r   r   r   r   r   r	   r
   r   r   rg  r9   rc   r   r   Znumpy.lib.recfunctionsr   r.   r   Zscipy._lib._utilr   rv  r   r   r   r   Zscipy.statsr/   Z!scipy.stats._distn_infrastructurer   Zscipy.stats.distributionsZscipy.specialr   r   r   Zscipy.stats._distr_paramsr   r   Ztest_discrete_basicr   r   Zscipy.stats._continuous_distnsr   r   Zscipy.optimizer    r!   	itertoolsr"   flagsoptimizerG  machiner  r  r-   r1   rA   rC   r\   ra   rd   rg   r7  r  rq   ry   r   r   r   r   r   r   r   r  r  r  r:  rd  rz  r  r  r  r  r  r  r   r  r+  rI  rP  rT  rj  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r<  rA  rL  rY  r]  r`  rt  r{  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r2  rS  r[  rc  rj  ro  rs  rz  r}  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  rL   r  r   r  r  r  r5  r  rJ  rM  rO  rP  rR  rT  rU  rV  rW  rs  rt  rw  r}  r~  r  r  r  r  r  r  r  r  r  r  r  r  r  r+   r+   r+   r,   <module>   s   ,

.	
H $&5# \ 'S( w:?* $ l)+ !-:v,U* V
 o h =%E6,\^K 
	
&(
	x;	,G.!a	yY

