
    _MhI                       d Z ddlZddlZddlZddlZddlm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mZmZ ddlZddlm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,m-Z- ddl.Zddl/m0Z0m1Z1m2Z2 ddl3m4Z4m5Z5 ddl6m7Z7m8Z8 ddl9m:Z:m;Z; ddl<m=Z=m>Z>m?Z? ddl@mAZA ejB        jC        dk    ZDej	        dk    o e	jE                    dk    ZFddgZGddZHd ZId ZJ G d d          ZKddZLd ZMejN        O                    d g d!          d"             ZP G d# d$          ZQ G d% d&          ZR G d' d(          ZS G d) d*          ZT G d+ d,          ZU G d- d.          ZV G d/ d0          ZW G d1 d2          ZX G d3 d4          ZY G d5 d6          ZZ G d7 d8          Z[ G d9 d:          Z\ G d; d<          Z] G d= d>          Z^ G d? d@          Z_ G dA dB          Z` G dC dD          Za G dE dF          Zb G dG dH          Zc G dI dJ          Zd G dK dL          Ze G dM dN          Zf G dO dP          Zg G dQ dR          Zh G dS dT          Zi G dU dV          Zj G dW dX          Zk G dY dZ          Zl G d[ d\          Zm G d] d^          Zn G d_ d`          Zo G da db          Zp G dc dd          Zq G de df          Zr G dg dh          Zs G di dj          Zt G dk dl          Zu G dm dn          Zv G do dp          Zw G dq dr          Zx G ds dt          Zy G du dv          Zz G dw dx          Z{ G dy dz          Z| G d{ d|          Z} G d} d~          Z~ G d d          Z G d d          Z G d d          Z G d d          Z G d d          Z G d d          Z G d d          Z G d d          Zdd ej        ddej        z  z            z  dd ej        ddej        z  z            z  dd ej        ddej        z  z            z  dd ej        ddej        z  z            z  dd ej        ddej        z  z            z  g	fd ej        ddej        z  z            dd ej        ddej        z  z            z  dd ej        ddej        z  z            z  dz  dd ej        ddej        z  z            z  dz  dd ej        ddej        z  z            z  dz  g	fdd ej        dej        z            z  ddd ej        dej        z            z  z  ddd ej        dej        z            z  z  ddd ej        dej        z            z  z  ddd ej        dej        z            z  z  ddd ej        dej        z            z  z  ddd ej        dej        z            z  z  dgfgZ G d d          Z G d d          Z G d d          Z G d d          Z G d d          Z G d d          Z G d d          Z G d d          Z G d d¦          Z G dÄ dĦ          Z G dń dƦ          Z G dǄ dȦ          Z G dɄ dʦ          Zd˄ Z G d̄ dͦ          Z G d΄ dϦ          Z G dЄ dѦ          Z G d҄ dӦ          Z G dԄ dզ          Z G dք dצ          Z G d؄ d٦          Zdڄ Z G dۄ dܦ          Z G d݄ dަ          Z G d߄ d          Z G d d          Z G d d          Z G d d          Z G d d          Z G d d          Z G d d          Z G d d          Z G d d          Z G d d          Z G d d          Z G d d          Z G d d          Z G d d          Z G d d          Z G d d          Z G d d           Z G d d          Z G d d          Z G d d          Z G d d          Zd	 ZejN                            eDd
          d             Zd Zd Zd Zd Zd Zd Zd Zd Zd ZejN        O                    dg d          d             ZÐd ZĐd ZŐd ZƐd Zǐd ZȐd Zɐd Zʐd  Zːd! Z̐d" ZejN        O                    d#d$ ej        d%d&g          fd' ej        d(d)g          fd* ej        d+d,g          fd- ej        d.d/g          fg          d0             Zΐd1 Zϐd2 ZАd3 Zѐd4 ZejN        O                    d5g d6          d7             ZejN        O                    d8g d9          d:             ZejN        O                    d;g d<          d=             ZՐd> ZejN        O                    d5g d?          d@             ZאdA Z G dB dCe)j                  Z G dD dEe)j                  Z G dF dGe)j                  Z G dH dIe)j                  Z G dJ dK          Z G dL dM          ZejN                            eDdN          dO             ZdP ZdQ ZdR ZdS ZdT ZdU ZdV ZdW ZdX ZdY ZejN        O                    d5g dZ          d[             ZejN        O                    d\d]d^g          d_             ZejN        O                    d`ddg dafdbdcg dafdddeg dffg          dg             Zdh Zdi Zdj Z G dk dl          Zdm Z G dn do          Z G dp dq          Z G dr ds          Z G dt du          Zdv ZejN        O                    dwe8e5z             dx             Zdy Zdz ZejN        O                    d8d{d|g          d}             ZejN        O                    d;g d~          d             Zd Zd Zd Z G d d          Z  G d d          Z G d d          Z G d d          Z G d d          Z G d d          ZejN        O                    dg d          d             ZdS (  z!
Test functions for stats module
    N)Path)assert_equalassert_array_equalassert_almost_equalassert_array_almost_equalassert_allcloseassert_assert_warnsassert_array_lesssuppress_warningsassert_array_max_ulpIS_PYPY)raises)	typecodesarray)rec_append_fields)special)check_random_state)IntegrationWarningquad	trapezoidcumulative_trapezoid)
argsreduce)_XMAX)xlogy	polygammaentr)distcontinvdistcont   )distdiscreteinvdistdiscrete)FitDataError
_argus_phi)rootfmindifferential_evolution)productdarwinx86_64tukeylambdapearson3c                 V    ||  d| }t          t          | |          |           d S )Nz does not have attribute msg)r	   hasattr)abr/   s      d/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/scipy/stats/tests/test_distributions.py_assert_hasattrr4   5   s;    
{00Q00GAqMMs######    c                  D    t          t          j        j        d           d S )Nf_gen)r4   scipystatsdistributions r5   r3   test_api_regressionr<   ;   s    EK-w77777r5   c                      t          t          j        j        j                  } d t
          D             }d t          D             }g d}||z   |z   }t          t          d |                    }| |k    sJ d S )Nc                     g | ]
}|d          S r   r;   .0dists     r3   
<listcomp>z0test_distributions_submodule.<locals>.<listcomp>B   s    ///d$q'///r5   c                     g | ]
}|d          S r?   r;   r@   s     r3   rC   z0test_distributions_submodule.<locals>.<listcomp>C   s    111DQ111r5   )rv_discreterv_continuousrv_histogramentropytrapzc                 H    t          |                               d           S )N<)str
startswithss    r3   <lambda>z.test_distributions_submodule.<locals>.<lambda>J   s    A(9(9#(>(>$> r5   )setr8   r9   r:   __all__r   r!   filter)actual
continuousdiscreteotherexpecteds        r3   test_distributions_submodulerY   @   s    *233F//h///J11L111H! ! !EH$u,H 6>>IIJJHXr5   c            	       v   e Zd Zej                            dg d          ej                            dddej        ddg          d                         Zd	 Z	d
 Z
ej                            dg d          d             Zej                            dg d          d             Zd Zej                            dg d          d             Zd Zej        j        ej                            dddg          ej                            dg d          ej                            dddg          ej                            dddg          d                                                             Zej        j        d             Zej                            d d!dg          d"             Zd# Zd$ Zd% Zd&S )'TestVonMisesk)皙?r    e   xr   r    
   d   c                     d }d } ||dd|            ||dd|            ||dd|            ||dd|            ||dd|            ||dd|           d S )Nc                     t          j        | ||          }t          |                    |          |                    |dt          j        z  |z  z                       d S )Nlocscale   )r9   vonmisesr   pdfnppir\   LrO   r_   vms        r3   check_vonmises_pdf_periodiczHTestVonMises.test_vonmises_periodic.<locals>.check_vonmises_pdf_periodicS   sS    q222Bq		266!q25y1}2E+F+FGGGGGr5   c                     t          j        | ||          }t          |                    |          dz  |                    |dt          j        z  |z  z            dz             d S )Nrd   r    rg   )r9   rh   r   cdfrj   rk   rl   s        r3   check_vonmises_cdf_periodiczHTestVonMises.test_vonmises_periodic.<locals>.check_vonmises_cdf_periodicW   sl    q222Bq		A "qAIM': ; ;a ?A A A A Ar5   r   r    r`   r;   )selfr\   r_   ro   rr   s        r3   test_vonmises_periodicz#TestVonMises.test_vonmises_periodicP   s    	H 	H 	H	A 	A 	A
 	$#Aq!Q///##Aq!Q///##Aq"a000##Aq!Q///##Aq!Q///##Aq"a00000r5   c                     t          t          j        j        t          j                    t          t          j        j        t          j                   d S N)r   r9   vonmises_liner1   rj   rk   r2   rs   s    r3   test_vonmises_line_supportz'TestVonMises.test_vonmises_line_supportd   s:    U(*RUF333U(*BE22222r5   c                 t    t          j        d          }t          |                    d          d           d S )N   r         ?)r9   rh   r   rq   )rs   rn   s     r3   test_vonmises_numericalz$TestVonMises.test_vonmises_numericalh   s1    ^C  BFF1IIs+++++r5   zx, kappa, expected_pdf))r]   {Gz?g|65?)r]         9@g?U?)r]   r{   g"?)       @r~   gDfI?)r   r   g1<)r   r{           c                 j    t           j                            ||          }t          ||d           d S NV瞯<rtol)r9   rh   ri   r   )rs   r_   kappaexpected_pdfri   s        r3   test_vonmises_pdfzTestVonMises.test_vonmises_pdfv   s6     n  E**\666666r5   zkappa, expected_entropy))r    gA	?)   g,eޞ?)ra   gBb9d22)  g
G )  gc                 h    t           j                            |          }t          ||d           d S NvIh%<=r   )r9   rh   rH   r   )rs   r   expected_entropyrH   s       r3   test_vonmises_entropyz"TestVonMises.test_vonmises_entropy   s5     .((//!1>>>>>>r5   c                 h   d}t           j                            |          }t           j                            |          }t           j                            |          }t          j        ddd                              |          }t          j        ddt           j        z  d                              |          }t          j        dddt           j        z  t          |          z  dz                                 |          }t          ||d           t          ||d           d S )	Ni@}r    r   rd   random_staterg   r   atol)	rj   randomdefault_rngr9   rh   rvsrk   absr   )rs   seedrng1rng2rng3rvs1rvs2rvs3s           r3   test_vonmises_rvs_gh4598z%TestVonMises.test_vonmises_rvs_gh4598   s   y$$T**y$$T**y$$T**~aQa00044$4GG~aQruWA666:::MM~aQ%&ruWSYY%6q%8; ; ;;>3D3;Q;Q 	d////d//////r5   zx, kappa, expected_logpdf))r]   r~   gEJ?)r]   r   goι?)r]   r{   g0,)r   r~   gdty)r   r   gʹw[A)r   r{   gc                 j    t           j                            ||          }t          ||d           d S r   )r9   rh   logpdfr   )rs   r_   r   expected_logpdfr   s        r3   test_vonmises_logpdfz!TestVonMises.test_vonmises_logpdf   s6     &&q%00e<<<<<<r5   c                 :   t           j                            d          }|                    d          dz  \  }}}t          j        ||                              d           }t          |d           t          j        |j        t           j	                  sJ ||dt           j
        z  z   f} t          j        ||          j        d g|R  }t          |d           t          j        |j        t           j	                  sJ ||dt           j
        z  z   f} t          j        ||          j        d	 g|R d
di}t          t          j        |          |dt           j
        z  z             t          j        |j        t           j                  sJ dS )a   
        Test that the vonmises expectation values are
        computed correctly.  This test checks that the
        numeric integration estimates the correct normalization
        (1) and mean angle (loc).  These expectations are
        independent of the chosen 2pi interval.
           kD +xNn    r`   re   r   c                     dS Nr    r;   r_   s    r3   rP   z3TestVonMises.test_vonmises_expect.<locals>.<lambda>       A r5   r    rg   c                     dS r   r;   r   s    r3   rP   z3TestVonMises.test_vonmises_expect.<locals>.<lambda>   r   r5   c                 0    t          j        d| z            S )Ny              ?)rj   expr   s    r3   rP   z3TestVonMises.test_vonmises_expect.<locals>.<lambda>   s    BF2a4LL r5   complex_funcN)rj   r   r   r9   rh   expectr   
issubdtypedtypefloatingrk   anglecomplexfloating)rs   rngre   r   lbresboundss          r3   test_vonmises_expectz!TestVonMises.test_vonmises_expect   s    i##$788A+UBnE22299++FFQ}SY44444R!be)^#9enE2229++OOOOQ}SY44444R!be)^#9enE2229:P:P S;AS S SPQS Ssag777}SY(:;;;;;;;r5   rvs_locrg   	rvs_shape)r    ra       חAfix_locTF	fix_shapec                 .   |r|rt          j        d           t          j                            d          }t
          j                            |d||          }ddi}|r||d<   |r||d<   t          t
          j        |t
          j        j	        fi | d S )	NzNothing to fit.r   r   )sizere   r   fscaler    flocf0)
pytestskiprj   r   r   r9   rh   r   _assert_less_or_close_loglikennlf)rs   r   r   r   r   r   datakwdss           r3   test_fit_MLE_comp_optimizerz(TestVonMises.test_fit_MLE_comp_optimizer   s      	+ 	+K)***i##$788~!!)$G/2 " 4 4 !} 	#"DL 	#"DJ%end&+n&9	C 	C=A	C 	C 	C 	C 	Cr5   c                 @   g d}t          j        |          }dt           j        z  }t          j                            ||          \  }}}|t          j        t                    j        k    sJ t          t          j        |t          j        j
        d|           d S )N)g!*Kg4,%OԿgZ۵?g
ge_@g-8gPghn?g<t?g.6@gjb?g0]o@g݅ZgSsgx7W?g%H?g65Jg:m2?g@g;,gM^lsF      ࿩r   r    )r   r   )rj   asarrayrk   r9   rh   fitfinfofloattinyr   r   )rs   r   re   	kappa_fitloc_fit	scale_fits         r3   test_vonmises_fit_bad_flocz'TestVonMises.test_vonmises_fit_bad_floc   s    M M M z$RUl(-(:(:4c(:(J(J%	7IBHUOO00000%end&+n&9!#	O 	O 	O 	O 	O 	Or5   signc                    t           j                            d          }t          j        |dz  t           j        z  d                              d|          }|dt           j        z  z   }t          j                            |          \  }}}t          j                            |          \  }}	}
t          ||	           t          ||           |dk    sJ t           j         |cxk     rt           j        k     sn J d S )	Nr   r|   r`   r   順 r      r    )	rj   r   r   r9   rh   rk   r   r   r   )rs   r   r   r   shifted_datar   r   r   kappa_fit_shiftedloc_fit_shifted_s              r3    test_vonmises_fit_unwrapped_dataz-TestVonMises.test_vonmises_fit_unwrapped_data   s    i##$788~$s(25.;;;??MP @ R Rag~(-(:(:4(@(@%	7I050B0B<0P0P-?A111	#4555A~~~~v''''"%''''''''r5   c                    t          j        d          }t          |                    d          ddt          j        z  z  d           t          |                    t          j        dz            dd           t          |                    t          j         dz            dd           t          |                    d          t          j        dz  d           t          |	                                dd	           t          |
                                dd	           t	          j        t	          j        |                    d
d                    t          j        k              sJ d S )Nr   r    rg   r   r         ??皙?r   r`     r   r   )r9   rh   r   ri   rj   rk   rq   sfppfmeanr   allr   r   )rs   rB   s     r3   test_vonmises_kappa_0_gh18166z*TestVonMises.test_vonmises_kappa_0_gh18166   s&   ~a  Q!be)_5AAAAq))4e<<<<q))4e<<<<ruSyu====		QU3333qu5555vbfTXX2DXAABBbeKLLLLLLLr5   c                 v    t           j                            dg          \  }}}|dk    r|dk    r|dk    sJ d S )Nr   g 7yACr    )r9   rh   r   )rs   r   re   rf   s       r3   test_vonmises_fit_equal_dataz)TestVonMises.test_vonmises_fit_equal_data  sA    !N..s33sE}}eqjjjjjjr5   c                     t           j        j                            ddgd           t           j        j                            t          j        dz  dz  gd           d S )Nr   gl=c>r   rg   gd?)r8   r9   rh   r   rj   rk   rx   s    r3   test_vonmises_fit_boundsz%TestVonMises.test_vonmises_fit_bounds  sZ     	  !WA 666 	  "%'9"5!6Q ?????r5   N)__name__
__module____qualname__r   markparametrizerj   rk   rt   ry   r}   r   r   r   r   r   xslowr   slowr   r   r   r   r   r;   r5   r3   r[   r[   O   s       [S---00[S1aC"8991 1 :9 101$3 3 3, , , [5/ / /0 07 70 07 [6; ; ;< <? ?< <?0 0 0  [8B B BC C= =C C=< < <4 [[YA//[[---88[Yu66[[4-88C C 98 76 98 0/ 
C$ [
O 
O 
O [Vb!W--
( 
( .-
(	M 	M 	M9 9 9
@ @ @ @ @r5   r[   Fc                    || j         } | j        |fi |} t          t          |           |           j        |fi |}|st	          j        ||k              sJ  |||          } |||          }||k    st	          j        ||d          sJ d|v r|d         |d         k    sJ d|v r|d         |d         k    sJ dS dS )a^  
    This utility function checks that the negative log-likelihood function
    (or `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.
    Nr   r   r   r   r   )r   r   supertyperj   anyallclose)	rB   r   funcmaybe_identicalr   mle_analyticalnumerical_optll_mle_analyticalll_numerical_opts	            r3   r   r     s!    |yTXd++d++N/E$t**d++/====M
  7vn566666^T22tM400!111K)+;%HHH 211
 ~~b!T&\11114b!T(^3333 33r5   c           
         ddg}| j         r6t          | j                             d                    }|g dd |         z  }t          t	          |t          j        t          |                                        }g d}t          j        t          d          5   | j
        |fi | d d d            n# 1 swxY w Y   t          j        t          d          5  | 
                    t
          j        g           d d d            n# 1 swxY w Y   t          j        t          d          5  | 
                    t
          j        g           d d d            n# 1 swxY w Y   t          j        t          d	          5  | 
                    |d
           d d d            n# 1 swxY w Y   t          j        t          d          5   | j
        |gdgt          |          dz
  z  R   d d d            d S # 1 swxY w Y   d S )Nr   r   ,)r   f1f2r    rg   r   3All parameters fixed. There is nothing to optimize.matchz#The data contains non-finite valueszUnknown keyword arguments:rg   )extra_keywordzToo many positional arguments.r    )shapeslensplitdictziprj   aranger   r   RuntimeErrorr   
ValueErrornaninf	TypeError)rB   paramnshapes	all_fixedr   s        r3   assert_fit_warningsr  ;  sL   XE{ .dk'',,--###HWH--S	#e** 5 56677I99D	|&
' 
' 
' $ $ 	#####$ $ $ $ $ $ $ $ $ $ $ $ $ $ $ 
zB
D 
D 
D  "&               
zB
D 
D 
D  "&               
y(D	E	E	E ( (Q'''( ( ( ( ( ( ( ( ( ( ( ( ( ( (	y(H	I	I	I . .-SZZ!^,----. . . . . . . . . . . . . . . . . .sZ   B88B<?B<!DDD1!EE"%E"F((F,/F,#G>>HHrB   )alpha	betaprimefatiguelifeinvgammainvgauss
invweibull	johnsonsblevylevy_llognormgibratpowerlognormrayleighwaldc                    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   getattrr9   r   ri   r1   r   r   rj   r  r2   )rB   dctargss      r3   test_supportr1  R  s     x..Ct9D5$D/$///333TV+d+++bfW555/$///333TV+d+++bfW55555r5   c                   &    e Zd Zd Zd Zd Zd ZdS )TestRandIntc                 D    t           j                            d           d S Nr   rj   r   r   rx   s    r3   setup_methodzTestRandInt.setup_methode      
	tr5   c                 v   t           j                            ddd          }t          t	          j        |dk               t	          j        |dk              z             t          t          |          dk               t           j                            ddd          }t          t	          j        |          dk               t          |j        j	        t          d         v            t           j                            dd          }t          |dk    |dk     z             t          t          |t          j                  t          t          |                    	           t          j        dd                              d
          }t          |j        j	        t          d         v            d S )Nr      ra   r   rg   2   
AllInteger   .   r.   r   )r9   randintr   r	   rj   r   r  shaper   charr   
isinstance
ScalarTypereprr   rs   valsvals      r3   test_rvszTestRandInt.test_rvsh  sW   }  BS 11tby!!BF419$5$55666D		S !!!}  BW 55')***
9\#::;;;mB''sRx()))
3..DcOODDDDmB##''**	)L"99:::::r5   c                     t           j        dd         }t          j        |dk    |dk     z  dd          }t          j                            |dd          }t          ||           d S )Nr   $   r   r:  {Gz?)rj   r_wherer9   rA  pmfr   )rs   r\   outrH  s       r3   test_pdfzTestRandInt.test_pdfu  s`    E!B$KhQ1r6*J::}  Ar**!$,,,,,r5   c                    t          j        ddd          }t          j        |          }t          j        |dk    |dk    gd|dz
  dz   d	z  gd          }t          j                            |dd          }t          ||d
           d S )Nr   rL  ra   r:  r         ?      @r    r      decimal)rj   linspacefloorselectr9   rA  rq   r   )rs   r_   r\   rQ  rH  s        r3   test_cdfzTestRandInt.test_cdf{  s    K2s##HQKKib!q&)C!C%'F1C+DaHH}  Ar**!$R888888r5   N)r   r   r   r7  rJ  rR  r\  r;   r5   r3   r3  r3  d  sP          ; ; ;- - -9 9 9 9 9r5   r3  c                   D    e Zd Zd Zd Zd Zd Zd Zd Zd Z	d Z
d	 Zd
S )	TestBinomc                 D    t           j                            d           d S r5  r6  rx   s    r3   r7  zTestBinom.setup_method  r8  r5   c                    t           j                            ddd          }t          t	          j        |dk              t	          j        |dk              z             t          t	          j        |          dk               t          |j        j        t          d         v            t           j                            dd          }t          t          |t                               t          j        dd                              d          }t          t          |t          j                             t          |j        j        t          d         v            d S Nr`   r   r<  r;  r   r>  r   )r9   binomr   r	   rj   r   rB  r   rC  r   rD  intndarrayrG  s      r3   rJ  zTestBinom.test_rvs  s	   {r4g66tqy!!BF42:$6$66777')***
9\#::;;;koob$''
3$$%%%k"d##''**
3
++,,,	)L"99:::::r5   c                     t           j                            ddd          }t           j                            ddd          }t          |ddd           t          |ddd           d S )Nra   r    r   rT  r   r   r   r9   rb  rP  r   )rs   vals1vals2s      r3   test_pmfzTestBinom.test_pmf  sd    S!,,3**sQ7777sQ777777r5   c                    t          j        dd          }t          j        g d          }t	          t          ||                     }|                                }t          ||           t          j        dd          }|                                }t          |d           t          j        dd          }|                                }t          |d           d S )Nrg   r|   )      ?r|   rl  r   rT  )	r9   rb  rj   r   sumr   rH   r   r   )rs   r2   
expected_p
expected_hhs        r3   test_entropyzTestBinom.test_entropy  s    K3X///00
%
J77888
IIKK:&&&K3IIKKQK3IIKKQr5   c                 j   t          j                    5  t          j        dt                     t	          t          j        dd                                          d           t	          t          j        dd                                          d           d d d            d S # 1 swxY w Y   d S )Nerrorrg   r   np)	warningscatch_warningssimplefilterRuntimeWarningr   r9   rb  r   stdrx   s    r3   test_warns_p0zTestBinom.test_warns_p0  s    $&& 	9 	9!'>:::qA...3355q999qA...2244a888	9 	9 	9 	9 	9 	9 	9 	9 	9 	9 	9 	9 	9 	9 	9 	9 	9 	9s   BB((B,/B,c                 Z    d}t           j                            d|d          |k    sJ d S )Nr   333333?rT  )qru  rv  )r9   rb  r   rs   ru  s     r3   test_ppf_p1zTestBinom.test_ppf_p1  s2    {S11Q666666r5   c                     d}d}t          j        d          }t          j                            |||          }t          j                            |||z            }t          ||d           d S )NglyCgq]\	T3<r   rt  缉ؗҜ<r   )rj   r  r9   rb  rP  poissonr   )rs   ru  rv  r_   r   refs         r3   test_pmf_poissonzTestBinom.test_pmf_poisson  sh    "IaLLkooa1o**m1q5))Su------r5   c                     d}d}d}t           j                            |||          }t           j                            |||          }t	          ||d           d S )NgJ-DgO
;r   r  r   )r9   rb  rP  rq   r   )rs   ru  rv  rr   r  s         r3   test_pmf_cdfzTestBinom.test_pmf_cdf  sZ    kooaA&&kooaA&&Su------r5   c                 l    t           j                            ddd          }t          |dd           d S )Nr   r   +?r   r  r   rg  )rs   r   s     r3   test_pmf_gh15101zTestBinom.test_pmf_gh15101  s4    kooau--QU++++++r5   N)r   r   r   r7  rJ  rj  rq  r|  r  r  r  r  r;   r5   r3   r^  r^    s          	; 	; 	;8 8 8   9 9 97 7 7
. . .. . ., , , , ,r5   r^  c                       e Zd Zd ZdS )TestArcsinec                     t           j                            ddg          }t          |t          j        t          j        g           d S Nr   r    )r9   arcsineri   r   rj   r  rs   rv  s     r3   test_endpointszTestArcsine.test_endpoints  s<     Mq!f%%Q()))))r5   N)r   r   r   r  r;   r5   r3   r  r    s#        * * * * *r5   r  c                        e Zd Zd Zd Zd ZdS )TestBernoullic                 D    t           j                            d           d S r5  r6  rx   s    r3   r7  zTestBernoulli.setup_method  r8  r5   c                    t           j                            dd          }t          t	          j        |dk              t	          j        |dk              z             t          t	          j        |          dk               t          |j        j        t          d         v            t           j                            d          }t          t          |t                               t          j        d                              d          }t          t          |t          j                             t          |j        j        t          d         v            d S )Nr   r<  r;  r   r    r>  r   )r9   	bernoullir   r	   rj   r   rB  r   rC  r   rD  rc  rd  rG  s      r3   rJ  zTestBernoulli.test_rvs  s   ""4g"66tqy!!BF419$5$55666')***
9\#::;;;o!!$''
3$$%%%od##''**
3
++,,,	)L"99:::::r5   c                    t          j        d          }dt          j        d          z  dt          j        d          z  z
  }|                                }t          ||           t          j        d          }|                                }t          |d           t          j        d          }|                                }t          |d           d S )Nrl        пr   r   rT  )r9   r  rj   logrH   r   r   )rs   r2   ro  rp  s       r3   rq  zTestBernoulli.test_entropy  s    OD!!26$<<'$rvd||*;;
IIKK:&&&OC  IIKKQOC  IIKKQr5   N)r   r   r   r7  rJ  rq  r;   r5   r3   r  r    sA          	; 	; 	;    r5   r  c                       e Zd Zd ZdS )TestBradfordc                     d}t          j        dd          }t          j                            ||          }t          j                            ||          }t          ||           d S )Nr]   )rj   logspacer9   bradfordrq   r   r   )rs   cr_   r  xxs        r3   test_cdf_ppfzTestBradford.test_cdf_ppf  s\    KR  Nq!$$^1%%2r5   N)r   r   r   r  r;   r5   r3   r  r    s#            r5   r  c                   F   e Zd Zd Zej                            dddddddd	d
ddedfg          d             Zej                            dg d          ej                            de	j
        j        dfe	j
        j        dfg          d                         Zej                            dg d          d             Zej                            dg d          ej                            de	j
        j        dfe	j
        j        dfg          d                         ZdS )
TestCauchyc                 T    t           j                            d          }|dk    sJ d S )NZbtir   )r9   cauchyri   r  s     r3   test_pdf_no_overflow_warningz'TestCauchy.test_pdf_no_overflow_warning  s*     LU##Cxxxxxxr5   x, ref)r   罡HP)g       r  )gIL8r  )gOd@<r  ):0yU>gHP)Mb@?gP)r]   g*6y)      ?g8@(J)g NgmCg'K@$[U)r  gonьgs32c                 r    t           j                            || g          }t          |||gd           d S r   )r9   r  r   r   )rs   r_   r  logps       r3   test_logpdfzTestCauchy.test_logpdf	  s>     |""Ar7++sCju555555r5   ))g 7y1gu	lY<)g^?)r   rl  )r   r|   )r    r   )r   g)nE?)g 7y1C?zmethod, sgnr    r   c                 F     |||z            }t          ||d           d S r   r   )rs   r_   r  methodsgnrv  s         r3   test_cdf_sfzTestCauchy.test_cdf_sf  s0      F3q5MM3U++++++r5   ))gRyXsg!;v+͋)(,* EgjLҳ)      $@g
 )r   9B.)g      $gCį)g:@sgg>Z)g 5g*\Bc                     t           j                            |          }t          ||d           t           j                            |           }t          ||d           d S N+<r   )r9   r  logcdfr   logsfrs   r_   r  r  r  s        r3   test_logcdf_logsfzTestCauchy.test_logcdf_logsf/  s`     $$Q''%0000""A2&&s//////r5   zp, ref))#B;g8)&.>g/)rl        )r|   r   )r   rT  )!?gqmA)g?gqJRBc                 F    | ||          z  }t          ||d           d S r   r  )rs   rv  r  r  r  r_   s         r3   test_ppf_isfzTestCauchy.test_ppf_isf>  s0     q		M3U++++++r5   N)r   r   r   r  r   r   r   r   r  r9   r  rq   r   r  r  r   isfr  r;   r5   r3   r  r    s          [	#	&	%	'	$	#	#	#	#	$
$	%
	'( (6 6( (6
 [	% 	% 	%	 	 [
,
A	
,/2		  
, , 	 	,
 [X; ; ;< <0 0< <0 [	/ 	/ 	/	 	 [
,
A	
,
B		!" ", ,	" "	 	, , ,r5   r  c                       e Zd ZdZdZd Zd Zd Zd Ze	j
                            ddefd	g          d
             Ze	j
                            dg d          d             ZdS )TestChignT9;g2 >h?@c                 t    t           j                            dd          }t          || j        d           d S )Nr`   r   r   r   )r9   chir   r   CHI_SF_10_4rs   rO   s     r3   test_sfzTestChi.test_sfZ  s5    ILLQ4+%888888r5   c                 t    t           j                            | j        d          }t	          |dd           d S )Nr   r`   r   r   )r9   r  r  r  r   rs   r_   s     r3   test_isfzTestChi.test_isf^  s5    IMM$*A..2E******r5   c                 r    d}d}t           j                            ||          }t          |dd           d S )Nr  r?  gar  r   )r9   r  r  r   )rs   r_   dfr  s       r3   test_logcdfzTestChi.test_logcdfb  s?    !!!R(( 6UCCCCCCr5   c                 r    d}d}t           j                            ||          }t          |dd           d S )Nr~   r?  @y`r  r   )r9   r  r  r   )rs   r_   r  r  s       r3   
test_logsfzTestChi.test_logsfi  s=    	2&&5EBBBBBBr5   df, ref     @@)g  ļBgcAc                 d    t          t          j                            |          |d           d S N-q=r   )r   r9   r  r   rs   r  r  s      r3   	test_meanzTestChi.test_meanv  s-     		r**Ce<<<<<<r5   ))-C6?gQ(݂)r    ńjɮ9?)r  gWr4'?)    _Bgr$h(?)}Ô%ITgs$h(?c                 r    t          t          j        |                                          |d           d S r   )r   r9   r  rH   r  s      r3   rq  zTestChi.test_entropy  s3     		"--//5AAAAAAr5   N)r   r   r   r  CHI_MEAN_1000r  r  r  r  r   r   r   r  rq  r;   r5   r3   r  r  Q  s         'K $M9 9 9+ + +D D DC C C [Y"M279: := =: := [Y; ; ;< <B B< <B B Br5   r  c                   r    e Zd Zd Zd Zej                            dg d          d             Zd Z	d Z
dS )	TestCrystalBallc                    t          j        ddd          dd         }t          j                            |dd          }t          j        g d	          }t          ||d
           t          j                            |dd          }t          j        g d          }t          ||d
           t          j                            |dddd          }t          j        g d          }t          ||d
           dS )
        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.
              rU     Nr   rT  r   betam)gF6Ɣ?g9nZ丘?g9b?g"i"w?g|^?g>?gG6ƴ?g9b?|^?߈܁r ?gTC?r  r  gm?gFZۤ?g
HB?gTXek?g?bȓE?gLZ;x?gJ>+=r         @)g2c}`?g>t\f?g˂q?gg'T){?gmw?g>t\?h'T)?7-h	J?u8n?A#?g:Q?r  r  r  r  gZrԢ?g߄q?g)aL?gr'O<!?g99{-w>r|   r  r  re   rf   )g3c}?g>t\?g˂?f'T)?#n?5-h	J??u8n?vL1??#?||P?g:Q?r  r  r   r  r  r  r  r  )rj   rY  r9   crystalballri   r   r   rs   X
calculatedrX   s       r3   rR  zTestCrystalBall.test_pdf  s;    Kc2&&ss+ &**13#*>>
8 	K 	K 	K 	L 	L 	*59999 &**13#*>>
8 	J 	J 	J 	K 	K 	*59999 &**13#3c*RR
8 	G 	G 	G 	H 	H 	*5999999r5   c                    t          j        ddd          dd         }t          j                            |dd          }t          j        g d	          }t          ||d
           t          j                            |dd          }t          j        g d          }t          ||d
           t          j                            |dddd          }t          j        g d          }t          ||d
           dS )r  r  rU  r  Nr   rT  r   r  )ghQ|
)?gj,,?gd}9?gH6?g|^?gEv?glQ|
)?gd}9?g|^?go[?gsX?g&Ռ?g~M+?gZRc?gtE_s?gi?gB?gEg?gvP
?gx훟?r   r   r  )g pr?g t\v?ge}?gݠn^?gCȘU?at\?ݠn^?~[n|D?gS4s/?2s?`qfhG?IgL?荩s?rA?IH?guk?ghi"?gP ?g.q?gs?r|   r  )go?r  ge?r	  gPU,?r
  gjݗ	?r  g?r  g_!Y?r  ga?r  gR?r  g(?r  g܉x?r  )rj   rY  r9   r  rq   r   r   r  s       r3   r\  zTestCrystalBall.test_cdf  s;    Kc2&&ss+ &**13#*>>
8 	E 	E 	E 	F 	F 	*59999 &**13#*>>
8 	E 	E 	E 	F 	F 	*59999 &**13#3c*RR
8 	E 	E 	E 	F 	F 	*5999999r5   zx, beta, m, rootref))      (@rT  r   gI9)      "@      @      ?gy
 <)   r]   gjt?g+,)g      r   r  gGA*?)g      >r|   rU  g^v'?)gd~Qr  皙?g眭?c                 n    t           j                            |||          }t          ||d           d S )Nr  r   r   )r9   r  r   r   )rs   r_   r  r  rootrefr   s         r3   r  zTestCrystalBall.test_sf  s<     !!!$!!44G%000000r5   c                    t          j        g d          }t          j        g d          }t          j        g d          }t          j                            d||          }t          ||d           t          j        g d          }t          j        dd	t           j        d
dg          }||z  }t          j                            d||          }t          ||d           t          j        t           j        t           j        t           j        ddg          }||z  }	t          j                            d||          }
t          |	|
d           t          j        t           j        t           j        t           j        t           j        dg          }||z  }t          j                            d||          }t          ||d           t          j        t           j        t           j        t           j        t           j        dg          }||z  }t          j                            d||          }t          ||d           t          j        t           j        t           j        t           j        t           j        dg          }||z  }t          j                            d||          }t          ||d           dS )zs
        All values are calculated using the pdf formula and the integrate function
        of Mathematica
        )r   rT  r  r   r  )r  r  r   r  r  )rT  rT  rT  rT  rT  r   MbP?r   )gԚh@g[&@gͪ@g2dQ@gn6-@g?ޫV&̿g BgQCRgQF\ jr    g
@gn(@rg   g?Җ)r   gMg'#@r   g('UHr   N)rj   r   r9   r  _munpr   r  )rs   r  r  expected_0th_momentcalculated_0th_momentnormr1   expected_1th_momentcalculated_1th_momentexpected_2th_momentcalculated_2th_momentexpected_3th_momentcalculated_3th_momentexpected_4th_momentcalculated_4th_momentexpected_5th_momentcalculated_5th_moments                    r3   test_momentszTestCrystalBall.test_moments  sg    x11122H...// !h'@'@'@AA % 1 7 74 C C+-BOOOO xHHHIIHh"&)YGHH$h % 1 7 74 C C+-BOOOOHbfbfbffh?@@$h % 1 7 74 C C+-BOOOOHbfbfbfbfjABB$h % 1 7 74 C C+-BOOOOHbfbfbfbfg>??$h % 1 7 74 C C+-BOOOOHbfbfbfbfh?@@$h % 1 7 74 C C+-BOOOOOOr5   c                    t          j        dd          }|                                }d\  }}}t          j        |||          }t          t          |                    |                    |          }t          ||d           d S )Nrg   r   )ir:  @ Hz>r   )	r9   r  rH   rj   rY  r   r   ri   r   )rs   cbres1lohiNr_   res2s           r3   rq  zTestCrystalBall.test_entropyL  sy    q!$$zz||&	BKB""bffQii!,,d......r5   N)r   r   r   rR  r\  r   r   r   r  r*  rq  r;   r5   r3   r  r    s        <: <: <:|<: <: <:D [	0 	0 	0 1 1 1+P +P +PZ/ / / / /r5   r  c                   &    e Zd Zd Zd Zd Zd ZdS )
TestNBinomc                 D    t           j                            d           d S r5  r6  rx   s    r3   r7  zTestNBinom.setup_methodX  r8  r5   c                    t           j                            ddd          }t          t	          j        |dk                         t          t	          j        |          dk               t          |j        j        t          d         v            t           j                            dd          }t          t          |t                               t          j        dd                              d          }t          t          |t          j                             t          |j        j        t          d         v            d S ra  )r9   nbinomr   r	   rj   r   rB  r   rC  r   rD  rc  rd  rG  s      r3   rJ  zTestNBinom.test_rvs[  s    |Dw77tqy!!"""')***
9\#::;;;lr4((
3$$%%%l2t$$((++
3
++,,,	)L"99:::::r5   c           	      2   t          t          j        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   rj   r   r9   r8  logpmfrP  r8   r   )rs   rI  s     r3   rj  zTestNBinom.test_pmff  s    u|223TBBCC((c488	: 	: 	: k ''1a00S!r5   c                     t           j                            g ddd          }t          j        t           j                            g ddd                    }t          ||           d S )N)r   r   r   r   333333@?rt  )r9   r8  r  rj   r  rq   r   rs   rH  r  s      r3   test_logcdf_gh16159zTestNBinom.test_logcdf_gh16159n  sb    |""<<<3$"??fU\%%lllcT%BBCCc"""""r5   N)r   r   r   r7  rJ  rj  r?  r;   r5   r3   r5  r5  W  sP          	; 	; 	;  # # # # #r5   r5  c                       e Zd Zd Zej        j        d             Zej        j        d             Zej        j        d             Z	ej        j        d             Z
d Zd Zd Zd	 Zd
S )TestGenInvGaussc                 D    t           j                            d           d S r5  r6  rx   s    r3   r7  zTestGenInvGauss.setup_methodv  r8  r5   c                     t          j        dd          }t          j        |                    dd          |j                  \  }}t          |dk    d           d S )Nffffff@r    r   r   皙?Tr9   geninvgausskstestr   rq   r   rs   gigr   rv  s       r3   test_rvs_with_mode_shiftz(TestGenInvGauss.test_rvs_with_mode_shifty  W     S))|CGGDGAA37KK1QXt$$$$$r5   c                     t          j        dd          }t          j        |                    dd          |j                  \  }}t          |dk    d           d S )Nr   r   rE  r   r   rF  TrG  rJ  s       r3   test_rvs_without_mode_shiftz+TestGenInvGauss.test_rvs_without_mode_shift  sW     T**|CGGDGAA37KK1QXt$$$$$r5   c                     t          j        dd          }t          j        |                    dd          |j                  \  }}t          |dk    d           d S )Nr]   皙?rE  r   r   rF  TrG  rJ  s       r3   test_rvs_new_methodz#TestGenInvGauss.test_rvs_new_method  rM  r5   c                     d }t           |dd          d           t           |dd          d           t           |dd          d           d S )Nc                     t          j        | |          }|                    dd          }t          j        ||j                  d         dk    S )NrE  r   r   r    rF  )r9   rH  r   rI  rq   )rv  r2   rK  r   s       r3   my_ks_checkz4TestGenInvGauss.test_rvs_p_zero.<locals>.my_ks_check  sG    #Aq))C''t$'77C<SW--a0477r5   r   rQ  Tr   r  )r   )rs   rU  s     r3   test_rvs_p_zerozTestGenInvGauss.test_rvs_p_zero  sm    	8 	8 	8
 	[[C(($///[[C(($///[[C(($/////r5   c                     t          t          j        dd                              dd          dt          j        dd                              dd          z             d S )N      rg   r`   r   r   r    r  )r   r9   rH  r   rx   s    r3   test_rvs_negative_pz#TestGenInvGauss.test_rvs_negative_p  sn    !$**..BT.JJE%c1--11r1MMM	O 	O 	O 	O 	Or5   c                 H   t           j                            dddd          }t          t          j        |ddg          d         dk    d	           d
t          j        ddd          }}t           j                            |dd|z  |          }t          |t          j	        |                              |                     t           j        
                    |dd|z  |          }t          |t          j	        |          
                    |                     d S )NrE  r   r    r   )r   rv  r2   r   r#  r0  333333?Tra   r~   r`   )rv  r2   rf   )r9   rH  r   r   rI  rj   rY  ri   r   r#  rq   )rs   igmur_   pdf_igcdf_igs         r3   test_invgausszTestGenInvGauss.test_invgauss  s   """MMU\"js;;;A>EtLLLR[q"--A"&&qDAF"&EEr 2 2 6 6q 9 9:::"&&qDAF"&EEr 2 2 6 6q 9 9:::::r5   c                     t          j        g d          }t          j        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?r~   r   r`   r|   r    )rj   r   rY  r   r9   rH  ri   )rs   vals_Rr_   s      r3   
test_pdf_RzTestGenInvGauss.test_pdf_R  sd      , , , - - Ka$$ 1 5 5aa @ @AAAAAr5   c                     t          t          j                            ddd          d           t          t          j                            ddd          d           d S )Nr   r|   g    >Ar=  rg   )r   r9   rH  ri   rx   s    r3   test_pdf_zerozTestGenInvGauss.test_pdf_zero  sT    U&**1c377;;; 	U&**3A66:::::r5   N)r   r   r   r7  r   r   r   rL  rO  rR  rV  rY  ra  rd  rf  r;   r5   r3   rA  rA  u  s           [% % % [% % % [% % % [0 0 0O O O	; 	; 	;	B 	B 	B; ; ; ; ;r5   rA  c                       e Zd Zd Zd Zd Zej                            dg d          d             Z	ej                            dg d          d             Z
d	 Zd
 Zd Zd Zd Zd ZdS )TestGenHyperbolicc                 D    t           j                            d           d S r5  r6  rx   s    r3   r7  zTestGenHyperbolic.setup_method  r8  r5   c                     t          j        g d          }d\  }}}d\  }}|||z  ||z  f}t          j        |||d}t          j        ddd          }	t          |                    |	          |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?rg   rg   r    r|   r  rd   r`   r   r   r   r   )rj   r   r9   genhyperbolicrY  r   ri   
rs   rc  lmbdar  r  r^  deltar0  ghr_   s
             r3   
test_pdf_rzTestGenHyperbolic.test_pdf_r  s           %ud	EuU{DJ/ $Be<<<KR$$q		6>>>>>>r5   c                     t          j        g d          }d\  }}}d\  }}|||z  ||z  f}t          j        |||d}t          j        ddd          }	t          |                    |	          |dd	           d S )
N)
g=R0W<=gEB=g#Nb>g_PG>g! 1Or?gUM?g u./?gfy?gl5\?gkV?rk  rl  rd   rm  r`   r   ư>rn  )rj   r   r9   ro  rY  r   rq   rp  s
             r3   
test_cdf_rzTestGenHyperbolic.test_cdf_r  s           %ud	EuU{DJ/ $Be<<<KR$$q		6======r5   zx, p, a, b, loc, scale, ref)
)rg   r   r  r|   r  g8ɹvF(;)rx  r`   r  rl  r    r   g[Ϡ?)rx  r`   r  g      ?r   r    g ✝`D;)rx        ?r  ):?r   r    gdL;)r   ry  r  rz  r   r    gzK^7?)r         r  rz  r   r    gj?)r   r{  r     r   r    ga=vG:)r  r{  r   r|  r   r    g?)(   r{  r   r|  r   r    g1bV?)<   r{  r   r|  r   r    g?c                 t    t           j                            ||||||          }t          ||d           d S Nrd   dy=r   )r9   ro  rq   r   )	rs   r_   rv  r1   r2   re   rf   r  rq   s	            r3   test_cdf_mpmathz!TestGenHyperbolic.test_cdf_mpmath  sB     !%%aAqc%GGSu------r5   ))r   rv  rV  r   r   r    gC!,?)r   r         @      @r    r   g,Ћf?)r  r   r  r  r    r   rT  )   rg   r   r  r|   r  gm"چ>),  r`   r  rl  r    r   g:;)r~  r{  r   r|  r   r    gw=)K   r{  r   r|  r   r    g3Z0&K<c                 t    t           j                            ||||||          }t          ||d           d S r  )r9   ro  r   r   )	rs   r_   rv  r1   r2   re   rf   r  r   s	            r3   test_sf_mpmathz TestGenHyperbolic.test_sf_mpmath  sB      ##Aq!QCu#EECe,,,,,,r5   c                     g d}d\  }}}d\  ||z  |z  ffdt          dd          D             }t          ||dd	           d S )
N)g@쐧@g@쐧 @g܊B@gvGăi@rk  rl  c                 Z    g | ]'}t          j        d                     |          (S )rd   )r9   ro  moment)rA   ir0  rr  r^  s     r3   rC   z4TestGenHyperbolic.test_moments_r.<locals>.<listcomp>$  sI        2U;;;BB1EE  r5   r    r   r   r   rn  )ranger   )	rs   rc  rq  r  r  vals_usr0  rr  r^  s	         @@@r3   test_moments_rz TestGenHyperbolic.test_moments_r  s    5 5 5 %ud	EuU{DJ/     1a[[  
 	ae<<<<<<r5   c                     d\  }}}d\  }}|||z  ||z  f}t          j        |||d}t          j        |                    dd          |j                  \  }}	t          |	dk    d           d S )	Nrk  rl  rd   rE  r   r   rF  T)r9   ro  rI  r   rq   r   )
rs   rq  r  r  r^  rr  r0  rs  r   rv  s
             r3   rJ  zTestGenHyperbolic.test_rvs+  s     %ud	EuU{DJ/ $Be<<<|BFF4F@@"&II1QXt$$$$$r5   c                 2   t          j        ddd          }t          j        |d          t          j        t           j                  j        z  d}}dt          j        |          }}| dz  ||f}t          j        |||d}t          j        |	                    d          |	                    d          d	          d d t           j
        f         }t          |                    |          t          j                            ||          dd
           d S )Nr    r:  r`   rg   r   rd   r~   Gz?r=  rv  rn  )rj   rY  float_powerr   float32epssqrtr9   ro  r   newaxisr   ri   t)	rs   r  r  r  r^  rr  r0  rs  r_   s	            r3   
test_pdf_tzTestGenHyperbolic.test_pdf_t8  s    [B## nR++BHRZ,@,@,DDatrwr{{EAud# $Be<<<KtbffTllB772:FFF1IIuw{{1b))	 	 	 	 	 	r5   c                    dt          j        t           j                  j        d}}}d\  }}|||f}t	          j        |||d}t          j        |                    d          |                    d          d          d d t           j        f         }t          |
                    |          t          j        
                    |          dd	           d S )
Nr   r   )r   r    rd   r~   r  r=  rv  rn  )rj   r   r  r  r9   ro  rY  r   r  r   ri   r  )	rs   rq  r  r  r^  rr  r0  rs  r_   s	            r3   test_pdf_cauchyz!TestGenHyperbolic.test_pdf_cauchyJ  s    
 "28BJ#7#7#;Qdu	Eud# $Be<<<KtbffTllB772:FFF1IIu|''**	 	 	 	 	 	r5   c                    t          j        ddd          }t          j        t           j                  j        }d\  }}}|||z  ||z  f}t          j        |||d}t          j        ddd          d d t           j        f         }t          |	                    |          t
          j
        	                    ||d          d	d
           d S )Nrm  r`   )r    r    r   rd   r  r  r=  r    r   dy=rn  )rj   rY  r   r  r  r9   ro  r  r   ri   laplace)	rs   re   rr  rq  r  r  r0  rs  r_   s	            r3   test_pdf_laplacez"TestGenHyperbolic.test_pdf_laplace[  s    k#r2&& $$($uduU{DJ/  $Cu===KR$$QQQ
]3FF1IIu}((1(==	 	 	 	 	 	r5   c           
      p   t          j        ddd          t          j        ddd          t          j        dt          d                    z  t          j        ddd          t          j        ddd          f\  }}}}d	}|||z  ||z  f}t	          j        |||d
}t          j        |                    d          |                    d          d          d d t           j        f         }t          |	                    |          t          j
        	                    |||||          dd           d S )Nr    r  r`   r      r   ra   r   rd   r~   r  r=  )r1   r2   re   rf   r   rn  )rj   rY  r  r  r9   ro  r   r  r   ri   norminvgauss)	rs   r  r  rr  r^  rq  r0  rs  r_   s	            r3   test_pdf_norminvgaussz'TestGenHyperbolic.test_pdf_norminvgaussp  s/    Ar2&&Ar2&&r~b%))'D'DDAq"%%D#r**	"tUB uu}dUl3 $Be<<<KtbffTllB772:FFF1IIu)--Ud% . 9 9	 	 	 	 	 	r5   N)r   r   r   r7  rt  rw  r   r   r   r  r  r  rJ  r  r  r  r  r;   r5   r3   rh  rh    s"         ? ? ?,> > >4 [%		< 		< 		< . . . [%	@ 	@ 	@	 	- -	 	-= = =.% % %  $  "  *    r5   rh  c                       e Zd Zej                            dddg          d             Zej                            dddg          d             Zd	 Zd
S )TestHypSecantzx, reference)r:  gA0=)r=  g!b;c                 h    t           j                            |          }t          ||d           d S r  )r9   	hypsecantr   r   )rs   r_   	referencer   s       r3   r  zTestHypSecant.test_sf  s4     _""IE222222r5   zp, reference)rv  g9bT*@)r  gTW-;@c                 h    t           j                            |          }t          ||d           d S r  )r9   r  r  r   )rs   rv  r  r_   s       r3   r  zTestHypSecant.test_isf  s4     O""95111111r5   c                     d}d}t           j                            |          }t          ||d           t           j                            |           }t          ||d           d S )N      I@g!br  r   )r9   r  r  r   r  r  s        r3   r  zTestHypSecant.test_logcdf_logsf  sh    %''**%0000%%qb))s//////r5   N)	r   r   r   r   r   r   r  r  r  r;   r5   r3   r  r    s        
 [^9:<= =3 3= =3 [^89;< <2 2< <20 0 0 0 0r5   r  c                       e Zd Zd Zd Zd Zej                            dg d          d             Z	d Z
d Zd	 Zd
 ZdS )TestNormInvGaussc                 D    t           j                            d           d S r5  r6  rx   s    r3   r7  zTestNormInvGauss.setup_method  r8  r5   c                     t          j        g d          }t          j        g d          }t          j                            |dd          }t          ||d           d S )N)gp~٧>g @X>g}e?g{)߳?gwH?ir  r      r?  r    r|   r1   r2   r  r   )rj   r   r9   r  rq   r   )rs   r_cdfx_testvals_cdfs       r3   
test_cdf_RzTestNormInvGauss.test_cdf_R  sr      < < < = =,,,--%))&A)==%d333333r5   c                     t          j        g d          }t          j        g d          }t          j                            |dd          }t          ||d           d S )N)gp϶>g\H55$?geСt&?go NiH?g]u>r  r    r|   r  r  r   )rj   r   r9   r  ri   r   )rs   r_pdfr  vals_pdfs       r3   rd  zTestNormInvGauss.test_pdf_R  sp     < < < = =,,,--%))&A)==%d333333r5   zx, a, b, sf, rtol))r   r    r   !7?r   )r  r    r   ?=r  )r    r   rX  gy5`?r  )r`   r   rX  gMLC9r  c                     t           j                            |||          }t          |||           t           j                            |||          }t          |||           d S Nr   r9   r  r   r   r  )rs   r_   r1   r2   r   r   rO   r  s           r3   test_sf_isf_mpmathz#TestNormInvGauss.test_sf_isf_mpmath  sh     !!!Q**2D))))""2q!,,14((((((r5   c                     ddg}ddg}d}ddg}t           j                            |||          }t          ||dd	           t           j                            |||          }t          ||d
           d S )Nr   r  r    r   r  r  r   r  rf  rv  r   r  )rs   r_   r1   r2   r   rO   r  s          r3   test_sf_isf_mpmath_vectorizedz.TestNormInvGauss.test_sf_isf_mpmath_vectorized  s    HF "89!!!Q**2E6666""2q!,,14((((((r5   c                     t          j        dd          }t          j        ddd          }|                    |          }|                    |          }t          ||           d S )Nr    r   r  rg   )r9   r  rj   r  r   r  r   )rs   dstr_   r   r  s        r3   test_gh8718zTestNormInvGauss.test_gh8718  s[     A&&IaQVVAYYggbkkQr5   c                 .   d\  }}t          j        |dz  |dz  z
            }||z  |dz  |dz  z  d|z  |t          j        |          z  z  ddd|dz  z  |dz  z  z   z  |z  f}t          |t          j                            ||d                     d S )	Nr    r|   rg   r   r  r    r   mvskmoments)rj   r  r   r9   r  )rs   r1   r2   gammav_statss        r3   
test_statszTestNormInvGauss.test_stats  s    11q!t$$u9adUAXosQw!bgenn:L/M!a!Q$hAo-.68We066q!V6LLMMMMMr5   c                     d\  }}t          j        g d          }t          j                            |||          }t          |t          j                            |||                     d S )Nr  r  r|   r  )rj   r   r9   r  r   r   rq   )rs   r1   r2   r  rH  s        r3   test_ppfzTestNormInvGauss.test_ppf  sf    1---..!%%fa33 2 6 6tQ B BCCCCCr5   N)r   r   r   r7  r  rd  r   r   r   r  r  r  r  r  r;   r5   r3   r  r    s          
4 
4 
44 4 4 [0J J JK K
) )K K
)	) 	) 	)     N N ND D D D Dr5   r  c                   P    e Zd Zd Zd Zd Zd Zd Zd Zd Z	d Z
d	 Zd
 Zd ZdS )TestGeomc                 D    t           j                            d           d S r5  r6  rx   s    r3   r7  zTestGeom.setup_method  r8  r5   c                    t           j                            dd          }t          t	          j        |dk                         t          t	          j        |          dk               t          |j        j        t          d         v            t           j                            d          }t          t          |t                               t          j        d                              d          }t          t          |t          j                             t          |j        j        t          d         v            d S )Nr   r<  r;  r   r>  r   )r9   geomr   r	   rj   r   rB  r   rC  r   rD  rc  rd  rG  s      r3   rJ  zTestGeom.test_rvs  s    z~~d~11tqy!!"""')***
9\#::;;;jnnT""
3$$%%%j""1%%
3
++,,,	)L"99:::::r5   c                 P   t           j                            d          }t          j                            t          j        d          d|          }|j        t           j        k    sJ t          j	        |t          j
        t           j                  j        k              sJ d S )Nl   A: ir   r   )rj   r   r   r9   r  r   r   r   int64r   iinfoint32max)rs   r   r   s      r3   test_rvs_9313zTestGeom.test_rvs_9313  s     i##O44jnnRVC[[qsnCCyBH$$$$vcBHRX..223333333r5   c                 n    t           j                            g dd          }t          |g d           d S )Nr  r|   )r|   rl  ry  )r9   r  rP  r   rs   rH  s     r3   rj  zTestGeom.test_pmf  s8    z~~iii--!$(:(:(:;;;;;r5   c                 8   t          j        t          j                            g dd                    }t          j                            g dd          }t          ||dd           t          j                            dd          }t          |d           d S )Nr  r|   r   r   rf  r    r   )rj   r  r9   r  rP  r:  r   r   )rs   rh  ri  rI  s       r3   test_logpmfzTestGeom.test_logpmf	  s    uz~~iii5566
!!)))S11u5q9999 j1%%S#r5   c                     t           j                            g dd          }t           j                            g dd          }t	          g d          }t          ||           t          |d|z
             d S )Nr  r|   r|   r         ?r    )r9   r  rq   r   r   r   rs   rH  vals_sfrX   s       r3   r  zTestGeom.test_cdf_sf  sr    z~~iii--*--			3//+++,,!$111!'1X:66666r5   c                 :   t           j                            g dd          }t           j                            g dd          }t	          g d          }t          |t          j        |                     t          |t          j        |                      d S )Nr  r|   r  )	r9   r  r  r  r   r   rj   r  log1pr  s       r3   r  zTestGeom.test_logcdf_logsf  s    z  C00*""999c22+++,,!$x(8(8999!'28XI+>+>?????r5   c                     t           j                            g dd          }t          g d          }t	          ||           d S )Nr  r|   )rT  r   r  )r9   r  r   r   r   rs   rH  rX   s      r3   r  zTestGeom.test_ppf!  sE    z~~000#66))!$11111r5   c                 f    t          t          j                            dd          dd           d S )Nr  rT  r  r   )r   r9   r  r   rx   s    r3   test_ppf_underflowzTestGeom.test_ppf_underflow&  s-    
ue44cFFFFFFr5   c                 v    t          j        d                                          }t          |dd           d S )Ng<,Ԛ?g۩@r   r   )r9   r  rH   r   )rs   rp  s     r3   test_entropy_gh18226zTestGeom.test_entropy_gh18226*  s;     Jv&&((,5999999r5   c                     t           j                            d          }t          j                            dd|          dk                                    sJ d S )NiKH9r`   r   r   )rj   r   RandomStater9   r  r   r   )rs   r   s     r3   test_rvs_gh18372zTestGeom.test_rvs_gh183720  sP     y,,Y77
u2LIIAMRRTTTTTTTr5   N)r   r   r   r7  rJ  r  rj  r  r  r  r  r  r  r  r;   r5   r3   r  r    s          	; 	; 	;4 4 4< < <  7 7 7@ @ @2 2 2
G G G: : :U U U U Ur5   r  c                        e Zd Zd Zd Zd ZdS )
TestPlanckc                 D    t           j                            d           d S r5  r6  rx   s    r3   r7  zTestPlanck.setup_method8  r8  r5   c                     t           j                            g dd          }t          g d          }t	          ||           d S )Nr  rU  )g|dy?g8'\>ggrU!>)r9   planckr   r   r   r  s      r3   r  zTestPlanck.test_sf;  sQ    |yyy"-- 2 2 2 3 3 	"$11111r5   c                     t           j                            g dd          }t          g d          }t	          ||           d S )N)r       @@     p@r  )g    P.g    h>g    TF)r9   r  r  r   r   r  s      r3   r  zTestPlanck.test_logsfB  sJ    |!!"7"7"7??:::;;!$11111r5   N)r   r   r   r7  r  r  r;   r5   r3   r  r  7  sA          2 2 22 2 2 2 2r5   r  c                   &    e Zd Zd Zd Zd Zd ZdS )TestGennormc                     g d}t           j                            |d          }t           j                            |          }t	          ||           d S Nr  r    )r9   gennormri   r  r   rs   pointspdf1pdf2s       r3   test_laplacezTestGennorm.test_laplaceI  sM    }  ++}  ((D$'''''r5   c                     g d}t           j                            |d          }t           j                            |d          }t	          ||           d S Nr  rg   g;f?rf   )r9   r  ri   r  r   r  s       r3   	test_normzTestGennorm.test_normP  sO    }  ++z~~fF~33D$'''''r5   c                    t           j                            d           t          j        d          }|                    d          }t          j        ||j                  j        dk    sJ t          j        d          }|                    d          }t          j	                            d          }t          j
        ||          j        dk    sJ t          j        d          }|                    d          }t          j                            dd	          }t          j
        ||          j        dk    sJ d S )
Nr   r|   r   r;  r]   r    rg   g;f?rf   r   )rj   r   r   r9   r  r   rI  rq   pvaluer  ks_2sampr  )rs   rB   r   rvs_laplacervs_norms        r3   rJ  zTestGennorm.test_rvsW  s   
	q}S!!hhDh!!|C**1C7777}QhhDh!!m''T'22~c;//6<<<<}QhhDh!!:>>t><<~c8,,3c999999r5   c                    t           j                            d           t          j        ddgddgg          }|                    g d          }t          j        |d d ddf         t          j        d          j                  d         d	k    sJ t          j        |d d ddf         t          j        d          j                  d         d	k    sJ t          j        |d d ddf         t          j        d          j                  d         d	k    sJ t          j        |d d ddf         t          j        d          j                  d         d	k    sJ d S )
Nr   r|   rT  r   rU  )r   rg   rg   r;  r    r]   )rj   r   r   r9   r  r   rI  rq   )rs   rB   r   s      r3   test_rvs_broadcastingz!TestGennorm.test_rvs_broadcastingh  sC   
	q}sBi"b233hhLLLh))|C1aL%-*<*<*@AA!DsJJJJ|C1aL%-*<*<*@AA!DsJJJJ|C1aL%-*<*<*@AA!DsJJJJ|C1aL%-*<*<*@AA!DsJJJJJJr5   N)r   r   r   r  r  rJ  r  r;   r5   r3   r  r  H  sU        ( ( (( ( (: : :"K K K K Kr5   r  c                   Z    e Zd Zej                            dddg          d             ZdS )
TestGibratx, sfx)r  gD.?)  gJ]b<c                     t          t          j                            |          |d           t          t          j                            |          |d           d S N+=r   )r   r9   r)  r   r  rs   r_   sfxs      r3   test_sf_isfzTestGibrat.test_sf_isf  sR     	**Ce<<<<((--qu======r5   N)r   r   r   r   r   r   r  r;   r5   r3   r  r  r  sV         [X(A(E(G H H> >H H> > >r5   r  c                       e Zd Zd Zej                            dg d          d             Zd Zd Z	ej                            dg d          d	             Z
d
S )TestGompertzc                     t           j                            t           j                            dd          d          }t	          |d           d S N0.++r    )r9   gompertzr   rq   r   r  s     r3   test_gompertz_accuracyz#TestGompertz.test_gompertz_accuracy  sA    Nu~11&!<<a@@6"""""r5   z	x, c, sfx))r    r  g|t?)r   r  g(fMr;)rF  r   g[?)      @r   gg[<c                     t          t          j                            ||          |d           t          t          j                            ||          |d           d S Nr  r   )r   r9   r  r   r  )rs   r_   r  r  s       r3   r  zTestGompertz.test_sf_isf  sX    
 	))!Q//5AAAA**322AEBBBBBBr5   c                 v    d}d}d}t           j                            ||          }t          ||d           d S )N       @r]   g_r  r   )r9   r  r  r   )rs   r_   r  r  r  s        r3   r  zTestGompertz.test_logcdf  sC    %&&q!,,%000000r5   c                 v    d}d}d}t           j                            ||          }t          ||d           d S )Ng"ul/rV  gH7Wr  r   )r9   r  r  r   )rs   r_   r  r  r  s        r3   r  zTestGompertz.test_logsf  sC    $$Q**s//////r5   c, ref))r  geUT8?)r    gtqq?)r   gs{)r  gN*6c                 d    t          t          j                            |          |d           d S r#  )r   r9   r  rH   rs   r  r  s      r3   rq  zTestGompertz.test_entropy  s/    
 	..q113UCCCCCCr5   N)r   r   r   r   r   r   r   r  r  r  rq  r;   r5   r3   r  r    s        # # #" [[ +M +M +M N NC C	N NC1 1 10 0 0 [X (E (E (E F FD D	F FD D Dr5   r  c                   Z    e Zd Zej                            dddg          d             ZdS )TestFoldNorm	x, c, ref)r  :0yE>g:0N<?)r  r  g:?c                 f    t          t          j                            ||          |d           d S r   )r   r9   foldnormrq   rs   r_   r  r  s       r3   r\  zTestFoldNorm.test_cdf  s1     	**1a00#EBBBBBBr5   N)r   r   r   r   r   r   r\  r;   r5   r3   r+  r+    s[         [[+N+M+O P PC CP PC C Cr5   r+  c                      e Zd Zej                            dddg          d             Zej                            dg d          d             Zej                            dd	d
g          ej                            dg d          ej                            dddg          ej                            dddg          d                                                 Zd Z	dS )TestHalfNormr  )r    |N?)r`   Oul2;c                     t          t          j                            |          |d           t          t          j                            |          |d           d S r#  )r   r9   halfnormr   r  r  s      r3   r  zTestHalfNorm.test_sf_isf  sT     	))!,,c>>>>**3//??????r5   r  ))Ww'&l7g;G<^͛7)C]r2<gE~o-<)r  g?c                 d    t          t          j                            |          |d           d S r   )r   r9   r6  rq   rs   r_   r  s      r3   r\  zTestHalfNorm.test_cdf  s/     	**1--s??????r5   r   h㈵>r  	rvs_scaler~   ra   r   r   TF	fix_scalec                    t           j                            d          }t          j                            ||d|          }|r[|rYd}t          j        t          |          5  t          j	        
                    |||           d d d            n# 1 swxY w Y   d S i }|r||d<   |r||d<   t          t          j        |fi |d	d
i d S Nr   r   re   rf   r   r   r  r  r   r   r   r   r  T)rj   r   r   r9   r6  r   r   r   r  halflogisticr   r   	rs   r   r<  r   r>  r   r   	error_msgr   s	            r3   r   z(TestHalfNorm.test_fit_MLE_comp_optimizer  s^    i##$788~!!gYT/2 " 4 4  	y 	%I|9=== M M"&&t')&LLLM M M M M M M M M M M M M M MF 	#"DL 	'&DN 	&end 	< 	<d 	< 	<6:	< 	< 	< 	< 	< 	<   $#BBBc                     t          j        t                    5  t          j                            g dd           d d d            d S # 1 swxY w Y   d S Nr  rg   r   )r   r   r#   r9   r6  r   rx   s    r3   test_fit_errorzTestHalfNorm.test_fit_error  s    ]<(( 	2 	2Nyyyq111	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2   $AAAN)
r   r   r   r   r   r   r  r\  r   rI  r;   r5   r3   r2  r2    sK        [X(?(C(E F F@ @F F@ [X (A (A (A B B@ @B B@ [Yt55[[*:*:*:;;[Yu66[[4-88< < 98 76 <; 65<22 2 2 2 2r5   r2  c                   2   e Zd Zej                            dddg          ej                            dddg          ej                            ddd	g          ej                            d
dd	g          d                                                 Zd ZdS )TestHalfCauchyr   r;  r  r<  r~   r   r   TFr>  c                    t           j                            d          }t          j                            ||d|          }|r[|rYd}t          j        t          |          5  t          j	        
                    |||           d d d            n# 1 swxY w Y   d S i }|r||d<   |r||d<   t          t          j	        |fi | d S )	Nr   r   rA  r  r  rB  r   r   )rj   r   r   r9   r6  r   r   r   r  
halfcauchyr   r   rD  s	            r3   r   z*TestHalfCauchy.test_fit_MLE_comp_optimizer  s=    i##$788~!!gYT/2 " 4 4  	y 	%I|9=== K K $$T	$JJJK K K K K K K K K K K K K K KF 	#"DL 	'&DN%e&6EEEEEEErF  c                     t          j        t                    5  t          j                            g dd           d d d            d S # 1 swxY w Y   d S rH  )r   r   r#   r9   rN  r   rx   s    r3   rI  zTestHalfCauchy.test_fit_error.  s    ]<(( 	4 	4   333	4 	4 	4 	4 	4 	4 	4 	4 	4 	4 	4 	4 	4 	4 	4 	4 	4 	4rJ  N)r   r   r   r   r   r   r   rI  r;   r5   r3   rL  rL    s        [Yt55[[4+66[Yu66[[4-88F F 98 76 76 65F,4 4 4 4 4r5   rL  c                      e Zd Zej                            dddg          d             Zej                            dg d          d             Zd Zd	 Z	ej                            d
ddg          ej                            dg d          ej                            dddg          ej                            dddg          d                                                 Z
d ZdS )TestHalfLogisticr  )ra   ]6)   L-c                 d    t          t          j                            |          |d           d S r   )r   r9   rC  r   r:  s      r3   r  zTestHalfLogistic.test_sf<  s0     	*--a00#EBBBBBBr5   q, ref))rR  ra   )rT  rS  )硏v?g   .!>)g?g      <c                 d    t          t          j                            |          |d           d S r   )r   r9   rC  r  rs   r  r  s      r3   r  zTestHalfLogistic.test_isfH  s0    
 	*..q113UCCCCCCr5   c                 p    d}d}t           j                            |          }t          ||d           d S )Ng      >@guVJr  r   )r9   rC  r  r   rs   r_   r  r  s       r3   r  zTestHalfLogistic.test_logcdfO  s=    $#**1--%000000r5   c                 p    d}d}t           j                            |          }t          ||d           d S )Nr  g+r  r   )r9   rC  r  r   rs   r_   r  r  s       r3   r  zTestHalfLogistic.test_logsfV  s=    $"((++s//////r5   r   r;  r  r<  r=  r   TFr>  c                    t           j                            d          }t          j                            ||d|          }i }|r[|rYd}t          j        t          |          5  t          j        	                    |||           d d d            n# 1 swxY w Y   d S |r||d<   |r||d<   t          t          j        |fi |d	d
i d S r@  )rj   r   r   r9   rC  r   r   r   r  r   r   )	rs   r   r<  r   r>  r   r   r   rE  s	            r3   r   z,TestHalfLogistic.test_fit_MLE_comp_optimizer]  s`    i##$788!%%'36 & 8 8  	y 	%I|9=== M M"&&t')&LLLM M M M M M M M M M M M M M MF 	#"DL 	'&DN 	&e&8$ 	< 	<$ 	< 	<6:	< 	< 	< 	< 	< 	<s   &#BBBc                     d}t          t          |          5  t          j                            g dd           d d d            d S # 1 swxY w Y   d S )Nz; Maximum likelihood estimation with 'halflogistic' requiresr  )r   rg   r   r    r   )assert_raisesr#   r9   rC  r   rs   r/   s     r3   test_fit_bad_flocz"TestHalfLogistic.test_fit_bad_flocz  s    L<s333 	6 	6""9991"555	6 	6 	6 	6 	6 	6 	6 	6 	6 	6 	6 	6 	6 	6 	6 	6 	6 	6s   $A

AAN)r   r   r   r   r   r   r  r  r  r  r   rb  r;   r5   r3   rQ  rQ  4  si        [X(D(D(F G GC CG GC [X (K (K (K L LD D	L LD1 1 10 0 0 [Yt55[[*:*:*:;;[Yu66[[4-88< < 98 76 <; 65<26 6 6 6 6r5   rQ  c                        e Zd Zd Zd Zd ZdS )TestHalfgennormc                     g d}t           j                            |d          }t           j                            |          }t	          ||           d S r  )r9   halfgennormri   exponr   r  s       r3   
test_exponzTestHalfgennorm.test_expon  sL     $$VQ//{v&&D$'''''r5   c                     g d}t           j                            |d          }t           j                            |d          }t	          ||           d S r  )r9   rf  ri   r6  r   r  s       r3   test_halfnormzTestHalfgennorm.test_halfnorm  sS     $$VQ//~!!&!77D$'''''r5   c                     g d}t           j                            |d          }t           j                            |d          }t	          |d|z             d S )Nr  g
(?rg   )r9   rf  ri   r  r   r  s       r3   test_gennormzTestHalfgennorm.test_gennorm  sT     $$VW55}  11D!D&)))))r5   N)r   r   r   rh  rj  rl  r;   r5   r3   rd  rd    sA        ( ( (( ( (* * * * *r5   rd  c                        e Zd Zd Zd Zd ZdS )TestLaplaceasymmetricc                     t          j        g d          }t          j                            |d          }t          j                            |          }t          ||           d S r  )rj   r   r9   laplace_asymmetricri   r  r   r  s       r3   r  z"TestLaplaceasymmetric.test_laplace  sX    )))$$'++FA66}  ((d#####r5   c                     t          j        g d          }d}d|z  }t          j                            ||          }t          j                            ||dz  z  |          }t          ||           d S )Nr  rg   r    )rj   r   r9   rp  ri   r   )rs   r   r   kapinvr  r  s         r3   test_asymmetric_laplace_pdfz1TestLaplaceasymmetric.test_asymmetric_laplace_pdf  sr    )))$$5'++FE::'++FE1H,=vFFd#####r5   c           
         t          j        t          j        d           t          j        d          g          }d}t          j                            ||          }t          j                            ||          }t          j                            ||          }t          j        ddg          }t          j        ddg          }t          j        dd	g          }t          j                            ||          }	|}
t          j        	                    ||          }|}t          t          j        ||||	|f          t          j        ||||
|f                     d S )
N   r`   rg   r]   gMbp?rQ  V-?r   gMb`?)rj   r   r  r9   rp  ri   rq   r   r   r  r   concatenate)rs   r   r   r  cdf1sf1r  cdf2sf2ppf1ppf2isf1isf2s                r3   !test_asymmetric_laplace_log_10_16z7TestLaplaceasymmetric.test_asymmetric_laplace_log_10_16  s9   BF2JJ;r

344'++FE::'++FE::&))&%88xu&&xi())hU|$$'++D%88'++C77dCt'DEEdCt'DEE	G 	G 	G 	G 	Gr5   N)r   r   r   r  rs  r  r;   r5   r3   rn  rn    sF        $ $ $$ $ $G G G G Gr5   rn  c                      e Zd Zd Zej                            dg d          d             Zej                            dg d          d             Zd Z	d Z
d	 Zd
 Zd Zd Zd Zg dg dg dg ddej        ddddgej         dddddgg dg dg dg	Z ej        e          Zej                            de          d             Zd Zd Zd  Zd! Zd" Zd# Zd$S )%TestTruncnormc                 D    t           j                            d           d S r5  r6  rx   s    r3   r7  zTestTruncnorm.setup_method  r8  r5   	a, b, ref))r   ra   r  )333333?ffffff?gܲl)rv  g>g+c                 f    t          t          j                            ||          |d           d S N绽|=r   r   r9   	truncnormrH   rs   r1   r2   r  s       r3   rq  zTestTruncnorm.test_entropy  s1    $ 	//155sGGGGGGr5   ))r  r  ghiɮ9?)r  r  r  )g0.+r  r  )}Ô%Ir  Z_2?c                 f    t          t          j                            ||          |d           d S r#  r  r  s       r3   test_extreme_entropyz"TestTruncnorm.test_extreme_entropy  s1      	//155sGGGGGGr5   c           	          t           j                            g ddddgdz  d          }t          j        t          j        dd	dd
dt          j        g          }t          ||           d S )Nr   r   r  r|   H.?r    rg   r  rT  r      rg   rd   r    fO?l@r   )r9   r  r   rj   r   r  r   r  s      r3   test_ppf_ticket1131z!TestTruncnorm.test_ppf_ticket1131  n    ""#E#E#EsB()s1uA # 7 78RVQ
Az1bfMNN!$11111r5   c           	          t           j                            g ddddgdz  d          }t          j        t          j        dd	dd
dt          j        g          }t          ||           d S )Nr  r  rT  r   r  rg   rd   r   r  r  r    )r9   r  r  rj   r   r  r   r  s      r3   test_isf_ticket1131z!TestTruncnorm.test_isf_ticket1131  r  r5   c                    d\  }}t           j                            ||ddd          }t          ||                                cxk     o|                                cxk     o|k     nc            d\  }}t           j                            ||ddd          }t          ||                                cxk     o|                                cxk     o|k     nc            d S )N)irm  r   r    r`   r;  )r`      r9   r  r   r	   minr  rs   lowhighr_   s       r3   test_gh_2477_small_valuesz'TestTruncnorm.test_gh_2477_small_values  s    	TOT1ab99aeegg........$....///	TOT1ab99aeegg........$..../////r5   c                    d\  }}t           j                            ||ddd          }t          ||                                cxk    o|                                cxk    o|k    nc           t          |||g          f d\  }}t           j                            ||ddd          }t          ||                                cxk     o|                                cxk     o|k     nc            d\  }}t           j                            ||ddd          }t          ||                                cxk     o|                                cxk     o|k     nc            d\  }}t           j                            ||ddd          }t          ||                                cxk     o|                                cxk     o|k     nc            d S )	N)ra   r^   r   r    r`   r;  )r     )'  i'  )i)r9   r  r   r	   r  r  rL   r  s       r3   test_gh_2477_large_valuesz'TestTruncnorm.test_gh_2477_large_values  s   	TOT1ab99quuww1111!%%''1111T111122CdA4G4GGG 	TOT1ab99aeegg........$..../// 	TOT1ab99aeegg........$....///"	TOT1ab99aeegg........$..../////r5   c                    ddgddgfD ]\  }}t          j        t           j         ||t           j        g          }||z   dz  }t          j                            |||          }t          j                            |||          }t          j                            |||          }t          j        g d          }t          j        g d          }	t          j        g d          }
|d	k     rt          j        g d
          }
t          ||           t          ||	           t          ||
           t          t          j	        |
d         |
d         z            |dz              t          j        g d          }t          j        
                    |||          }t          j        |t          j        |          dz  |g          }t          ||           |d	k     r_t          t          j                            |||          d           t          t          j                            |||          d           n^t          t          j                            |||          d           t          t          j                            |||          d           t          j                            |||          }t          t          j	        ||
d         z            |dz   dz             d S )Nr   r   r  r   r   r   r    r    rT  rT  r   r   )r   d._MT
@BKg?r   r   )r   r  r  r   r    rg   r|   r   r|   rT  gĖy	@gdv*?g,mj%V?rl  )rj   r   r  r9   r  rq   r   ri   r   r  r   r   )rs   r  r  xvalsxmidcdfssfspdfsexpected_cdfsexpected_sfsexpected_pdfspvalsppfsexpected_ppfsri   s                  r3   test_gh_9403_nontail_valuesz)TestTruncnorm.test_gh_9403_nontail_values  s   a&2r(+  	M  	MICHrvgsD"&9::EHc>D?&&uc488D/$$UC66C?&&uc488DH\\\22M8$8$8$899LH%A%A%ABBMQww ")E)E)E F Fm444\222m444}Q'7a8H'H I I #c	+ + +H]]]++E?&&uc488DHc273<<	+A4%HIIMm444Qww#EO$6$6tS$$G$G$68 8 8#EO$7$7c4$H$H$68 8 8 8 $EO$7$7c4$H$H$68 8 8#EO$6$6tS$$G$G$68 8 8/%%dC66Cs=+;'; < <tDy!mLLLLA 	M  	Mr5   c           	         ddgddgfD ]F\  }}t          j        t           j         ||t           j        g          }||z   dz  }t          j                            |||          }t          j                            |||          }t          j                            |||          }t          j        g d          }t          j        g d          }	t          j        g d          }
|d	k     rt          j        g d
          }
t          ||           t          ||	           t          ||
           t          t          j	        |
d         |
d         z            |dz              t          j        g d          }t          j        
                    |||          }t          j        |t          j        |          dz  |g          }t          ||           t          j                            |||          }t          ||           |d	k     r_t          t          j                            |||          d           t          t          j                            |||          d           n^t          t          j                            |||          d           t          t          j                            |||          d           t          j                            |||          }t          t          j	        ||
d         z            |dz   dz             t          j        ||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                    Hd S )N'   r}  ir   r  r  )r   pGC@Ff<r   r   )r   r  r  r   r    rg   r|   r  gEC@g i?gspXio)>rl  r  r   )rj   r   r  r9   r  rq   r   ri   r   r  r   r   rY  )rs   r  r  r  r  r  r  r  r  r  r  r  r  r  ri   xvals2s                   r3   test_gh_9403_medium_tail_valuesz-TestTruncnorm.test_gh_9403_medium_tail_values0  s   r(S#J/ ,	P ,	PICHrvgsD"&9::EHc>D?&&uc488D/$$UC66C?&&uc488DH\\\22M8$8$8$899LH%K%K%KLLMQww " *= *= *= !> !>m444\222m444}Q'7a8H'H I I #c	+ + +H]]]++E?&&uc488DHc273<<+CT%JKKMm444?&&tS$77De,,,Qww#EO$6$6tS$$G$G$68 8 8#EO$7$7c4$H$H$9; ; ; ; $EO$7$7c4$H$H$68 8 8#EO$6$6tS$$G$G$9; ; ;/%%dC66Cs=+;'; < <tDy!mLLLKT2..EDDbDk\F 3 3E3 E E % 2 26D53$ G G" MO O O 2 25#t D D % 3 3FTEC4 H H2 NP P P 3 3E3 E E % 3 3FTEC4 H H2 NP P P PW,	P ,	Pr5   c                     t          t          j        dd                              d          d           t          t          j        dt          j                                      d          d           d S )Ng      *@      .@      ,@g T?r  g @gR?)r   r9   r  rq   rj   r  rx   s    r3   test_cdf_tail_15110_14753z'TestTruncnorm.test_cdf_tail_15110_14753_  sr    
 	S1155c::*	, 	, 	,262266s;;*	, 	, 	, 	, 	,r5   )r:  r   rT  r   r   )rm  r`   r   rT  r   gLFu)r  r   r   gMFmz%?r   gſ)r   rg   r   ghI}?r   gHNMr   gQ63E?gnl$A?gr1"D?gK8L?gQ63Egr1"D)r   r   gU*? bn?gnA??c3TXʿ)r  r    gU*ҿr  gnAr  )rm  g"$7"gOxr?gp7`g'>K@casec                     |\  }}}}}}t           j                             ||d          \  }}	}
}t          ||	|
|g||||gd           d S )Nr  r  FFg<r   )r9   r  r   )rs   r  r1   r2   m0v0s0k0r  vrO   r\   s               r3   r*  zTestTruncnorm.test_moments  sd    #1b"b"_**1a*@@
1aAq!r2r2&6UCCCCCCr5   c                     t           j                             dt          j        d          \  }}t	          |d           t	          |d           d S )Nr   mvr  g e3E?g:&A?)r9   r  rj   r  r   )rs   r  r  s      r3   test_9902_momentszTestTruncnorm.test_9902_moments  sJ    $$Q$==1Az***Az*****r5   c                     d\  }}t           j                            ||ddd          }t          ||                                cxk     o|                                cxk     o|k     nc            d S )N)r`   r?  r   r    r`   r;  r  r  s       r3   test_gh_1489_trac_962_rvsz'TestTruncnorm.test_gh_1489_trac_962_rvs  sy    	TOT1ab99aeegg........$..../////r5   c                 P   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                            ||dt          |          f
          }t          j        |          dt          |          fk    sJ t          t          j        ||	                    d          k                         t          t          j        |
                    d          |k                         d S )Nrm  r`   r  ir}  r  r   r  -   r;  r   )axis)rj   r  r9   r  r   r  rB  r	   r   r  r  r  s       r3   test_gh_11299_rvszTestTruncnorm.test_gh_11299_rvs  s     BbfWrvgsCS"MB262sBRHOTCHH>>x{{q#c((m++++saeeemm+,,---quu!u}},--.....r5   c                     t          t          j        d          rAt          j                            dddt          j                                                   d S d S )Nr   rm  r  r   r   )r0   rj   r   r9   r  r   r   rx   s    r3   test_rvs_Generatorz TestTruncnorm.test_rvs_Generator  sh    29m,, 	FORa-/Y-B-B-D-D   F F F F F	F 	Fr5   c                    t          j        t           j         t           j         dt           j         dg          }t          j        t           j        t           j        ddt           j        g          }t          j        g d          }g d}t          t	          j        ||                              |          |           t          t	          j        | |                               |           |           d S )Nr`   r  )r`         @r  	   r  )gOul"gځ=!gL!g>EG	f g^K[X)rj   r   r  r   r9   r  r  r  )rs   r1   r2   r_   rX   s        r3   test_logcdf_gh17064z!TestTruncnorm.test_logcdf_gh17064  s    HrvgwRVGR899HbfbfaRV455H***+++ + + 	1--44Q77BBBQB//55qb998DDDDDr5   c                 z    t          j        dd                              d          }d}t          ||           d S )Nr   r   r   gl).0S?)r9   r  r  r   )rs   r   r  s      r3   test_moments_gh18634z"TestTruncnorm.test_moments_gh18634  s@     ob!$$++A..  S!!!!!r5   N)r   r   r   r7  r   r   r   rq  r  r  r  r  r  r  r  r  rj   r  _truncnorm_stats_datar   r*  r  r  r  r  r  r  r;   r5   r3   r  r    su          [[B B BC CH H	C CH  [[D D DE E
H HE E
H2 2 22 2 20 0 00 0 0&!M !M !MF-P -P -P^, , ,	 	 		1 	1 	1	= 	= 	=	< 	< 	<	
BF						
 &!						
	 	 	
	 	 	
	 	 	;"F %BH%:;;[V%:;;D D <;D
+ + +
0 0 0/ / /F F F	E 	E 	E" " " " "r5   r  c                      e Zd Zej                            dg d          d             Zej                            dg d          d             Zej                            ddd	g          d
             Zej                            dddg          d             Z	ej                            dddg          d             Z
ej                            dddg          d             ZdS )TestGenLogisticx, expected))ig4`n)ig4	ncg)r   gX
=)ra   gh@X)r   gh<c                 n    d}t           j                            ||          }t          ||d           d S )Nr  r   r   )r9   genlogisticr   r   )rs   r_   rX   r  r  s        r3   r  zTestGenLogistic.test_logpdf  s<      ''1--hU333333r5   r'  ))r  gTKEl@)r  gJk$@)r   gsF<?)r  7~F<?c                 d    t          t          j                            |          |d           d S r  )r   r9   r  rH   r)  s      r3   rq  zTestGenLogistic.test_entropy  s0    
 	)11!44cFFFFFFr5   r,  )rS  r`   g8.)  r  gpT	oc                 f    t          t          j                            ||          |d           d S r#  )r   r9   r  r   r0  s       r3   r  zTestGenLogistic.test_sf  s2     	),,Q22CeDDDDDDr5   	q, c, ref)r~   rS  g^ #@)r  rg   gGFf@c                 f    t          t          j                            ||          |d           d S NV瞯<r   )r   r9   r  r  rs   r  r  r  s       r3   r  zTestGenLogistic.test_isf  2     	)--a33SuEEEEEEr5   )r|   rS  gİ@)r  r  g1a@c                 f    t          t          j                            ||          |d           d S r  )r   r9   r  r   r  s       r3   r  zTestGenLogistic.test_ppf  r  r5   )ra   {Gz?g(?)r=  r  g'Tn7c                 f    t          t          j                            ||          |d           d S r   )r   r9   r  r  r0  s       r3   r  zTestGenLogistic.test_logcdf  s2     	)00A66%HHHHHHr5   N)r   r   r   r   r   r   r  rq  r  r  r  r  r;   r5   r3   r  r    s        [] -I -I -I J J
4 4J J
4 [X (E (E (E F FG G	F FG& [[+L+L+N O OE EO OE [[+I+H+J K KF FK KF [[+I+H+J K KF FK KF [[+N+K+M N NI IN NI I Ir5   r  c                   t    e Zd Zd Zd Zd Zd Zd Zd Zd Z	d Z
d	 Zd
 Zej        j        d             ZdS )TestHypergeomc                 D    t           j                            d           d S r5  r6  rx   s    r3   r7  zTestHypergeom.setup_method	  r8  r5   c                 L   t           j                            dddd          }t          j        |dk              t          j        |dk              z  sJ t          j        |          dk    sJ |j        j        t          d         v sJ t           j                            ddd          }t          |t                    sJ t          j        ddd                              d          }t          |t          j                  sJ |j        j        t          d         v sJ d S )Nr  r`   r   r<  r;  r   r>  )r9   	hypergeomr   rj   r   rB  r   rC  r   rD  rc  rd  rG  s      r3   rJ  zTestHypergeom.test_rvs  s   ""2r17";;vdai  26$!)#4#44444x~~((((z)L"99999o!!"a,,#s#####ob!R((,,Q//#rz*****y~<!8888888r5   c                     d}d}d}|}|}t           j                            d|||          }t          |dd           d S )Ni	  r=  r  rg   gkㅒP?r  )r9   r  rP  r   )rs   Mru  r2  totgoodhgpmfs          r3   test_precisionzTestHypergeom.test_precision  sO    ##AsD!44E#8"=====r5   c                    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   rg   r    rT  r  r   )r   r9   r  rP  rx   s    r3   	test_argszTestHypergeom.test_args!  s     	EO//1a;;S"EEEEO//1a;;S"EEEEO//1a;;S"EEEEO//1a;;S"EEEEEr5   c           	          t          dt          j                            dddd          cxk    odk    nc            d S )Nr   r:  io i  if0  rT  )r	   r9   r  rq   rx   s    r3   test_cdf_above_onez TestHypergeom.test_cdf_above_one*  sH    U_((XtUCCJJJJsJJJJKKKKKr5   c                 <  	 ddt          j        g d          dz  }d		fd|D             }t          j        g d          }t          ||dd	
           g d}t          j                            |z   d          }g d}t          ||dd	
           d S )Ng    +@g     @)r   ffffff@333333@r   gffffff@@r        @     @c                 Z    g | ]'}t           j                            z   |          (S r;   )r9   r  r   )rA   eatenorangespearsquantiles     r3   rC   z1TestHypergeom.test_precision2.<locals>.<listcomp>5  sC     * * * !!(GeOWeLL * * *r5   )r   gpR(g㘍R2gJޱ p9g?Mf=
G?r    r   gƠ>rn  )g     @r  g     @g     @g     @)r    r  g)9g"1)rj   r   r   r9   r  r   )
rs   fruits_eatenr   rX   	quantilesr3  	expected2r  r  r  s
          @@@r3   test_precision2zTestHypergeom.test_precision2.  s     x = = =>>D* * * * * *(* * *8 9 9 9 : :XAD9999 0//	!!)Wu_guMM>>>	iad;;;;;;r5   c                 `   t          j        ddd          }|                                }t          j        ddg          }t          j        t          ||                     }t          ||           t          j        ddd          }|                                }t          |d           d S )Nr   r    r   rl  r   )	r9   r  rH   rj   r   rm  r   r   r   )rs   hgrp  rn  ro  s        r3   rq  zTestHypergeom.test_entropyA  s    _Q1%%JJLLXtTl++
fU:z::;;;
:&&&_Q1%%JJLLQr5   c                     d}d}d}d}t           j                            ||||          }d}t          ||d           d}d	}d
}d}t           j                            ||||          }d}t          ||d           d S )Nr
      cA    .A     j@gor   rW  r    @  X  r  g_7	j$r?  )r9   r  r  r   rs   r\   r  ru  r2  resultrX   s          r3   r  zTestHypergeom.test_logsfM  s     &&q!Q22FHa8888&&q!Q22 FHb999999r5   c                 D   d}d}d}d}t           j                            ||||          }d}t          ||d           d}d	}d
}d}t           j                            ||||          }d}t          ||d           d}d	}d}d}t           j                            ||||          }d}t          ||d           t	          j        g d          }d	}d
}d}t           j                            ||||          }t	          j        dd          }t          ||d           d S )Nr    r  r  r  g)\Ur   rW  r}  r  r=  r  g*@Vr?  }      r  gר)r}  r}  r}  )r9   r  r  r   rj   r   fullr  s          r3   r  zTestHypergeom.test_logcdfe  sM    ''1a33FHa8888 ''1a33&FHb9999''1a33 FHb9999
 H\\\""''1a3371122FHb999999r5   c                     d}d}d}t           j                            |||          }||z  |z  }t          ||           d S )Nip iP i.  )r9   r  r   r   )rs   r  ru  r2  hmrms         r3   test_mean_gh18511zTestHypergeom.test_mean_gh18511  sO     _!!!Q**UQYBr5   c                    d}d}t          j        dd          }d|z  }t          j                            |dz
  |||          }t          j        |dk              sJ t          j        t          j        |          dk               sJ d S )Nr`   r   r   r?  r  r    r   )rj   r  r9   r  r   r   diff)rs   ru  r2  r  population_sizerv  s         r3   test_sf_gh18506zTestHypergeom.test_sf_gh18506  s     Iaq&Oq1uoq!<<va!e}}vbgajj1n%%%%%%%r5   N)r   r   r   r7  rJ  r  r  r  r  rq  r  r  r'  r   r   r   r+  r;   r5   r3   r  r    s          	9 	9 	9> > >F F FL L L< < <&
 
 
: : :0,: ,: ,:\	  	  	  [	& 	& 	& 	& 	&r5   r  c                   V   e Zd Zej                            dg d          d             Zej                            dg d          d             Zd Zd Z	d	 Z
d
 Zej                            dddg          d             Zej                            dg d          d             ZdS )TestLoggammaz	x, c, cdf))r    rg   g;Y%?)r      gRٲa6_<)gHr  gi|e?)r  g6l?)i+r]   g֬CCf9)ir   g ֺlc                     t           j                            ||          }t          ||d           t           j                            ||          }t          ||d           d S r   )r9   loggammarq   r   r   )rs   r_   r  rq   rv  ys         r3   r  zTestLoggamma.test_cdf_ppf  sb     Nq!$$3U++++NsA&&15))))))r5   zx, c, sf))r   r  gQOu3;)   ra   g\Z{0)r/  r  geɐ?)ig{Gzd?gr0T?c                     t           j                            ||          }t          ||d           t           j                            ||          }t          ||d           d S r   )r9   r1  r   r   r  )rs   r_   r  r   rO   r2  s         r3   r  zTestLoggamma.test_sf_isf  sb     Na##2E****Nr1%%15))))))r5   c                 j    t           j                            dd          }t          |dd           d S )Nrg   g     @r  r   )r9   r1  r   r   )rs   lps     r3   r  zTestLoggamma.test_logpdf  s6    
 ^""4++G%000000r5   c                 v    d}d}t           j                            ||          }d}t          ||d           d S )Nr        @gI.r  r   )r9   r1  r  r   )rs   r_   r  r  r  s        r3   r  zTestLoggamma.test_logcdf  sC    &&q!,,%%000000r5   c                 v    d}d}t           j                            ||          }d}t          ||d           d S )Ng      9      @g!{1+ҷr  r   )r9   r1  r  r   )rs   r_   r  r  r  s        r3   r  zTestLoggamma.test_logsf  sC    $$Q**#s//////r5   c                     t          j        g d                              dd          }|D ]?\  }}}}}t          j                            |d          }t          |||||gd           @d S )N)r|   g"~jg46<@g oŏr  rT  g	h"lxgQ?gZd;333333@r  g{/L@g??gгYҿgh|?5?r   r   msvkr  r   rW  )rj   r   reshaper9   r1  r   )rs   tabler  r   varskewkurtcomputeds           r3   r  zTestLoggamma.test_stats  s          
 wr1~~ 	 ). 	1 	1$AtS$~++Av+>>H%hsD$0G./1 1 1 1 1	1 	1r5   r  r]   r  c                    t           j                            |d          }t          j        |                                          sJ t           j                            |          }t          j        t          j        ||k               t          |                    }|
                    d          }|j        dcxk     r|j        k     sn J d S )Nr   r;  r  )confidence_levelr|   )r9   r1  r   rj   isfiniter   median	binomtestcount_nonzeror  proportion_cir  r  )rs   r  r_   medbtestcis         r3   rJ  zTestLoggamma.test_rvs  s     Nqv.. {1~~!!##### n##A&& 0S 9 93q66BB  % 88v%%%%bg%%%%%%%%r5   r'  ))r-  g3H໱k3@)r    r  )r
  g8,[H^}	)r  g90$)r  gtum\c                 d    t          t          j                            |          |d           d S r#  )r   r9   r1  rH   r)  s      r3   rq  zTestLoggamma.test_entropy	  s/     	..q113UCCCCCCr5   N)r   r   r   r   r   r   r  r  r  r  r  r  rJ  rq  r;   r5   r3   r-  r-    sX        [[E E EF F* *F F* [ZB B BC C
* *C C
*1 1 11 1 10 0 01 1 1 [S3,//& & 0/& [X< < <= =	D 	D= =	D 	D 	Dr5   r-  c                   ^    e Zd Zg dZej                            de          d             ZdS )TestJohnsonsu))r6  r    r    g?r-  )r   r    r    gH';vIh%,=)r   r    r    g?x7rR  r  c                     |\  }}}}}t          t          j                            |||          |d           t          t          j                            |||          ||           d S NrR  r   )r   r9   	johnsonsur   r  rs   r  r_   r1   r2   r   tols          r3   r  zTestJohnsonsu.test_sf_isf%	  j    1aS**1a33ReDDDD++B155qsCCCCCCr5   Nr   r   r   casesr   r   r   r  r;   r5   r3   rQ  rQ  	  sX        ; ; ;E [VU++D D ,+D D Dr5   rQ  c                   ^    e Zd Zg dZej                            de          d             ZdS )TestJohnsonb))r  r    r    r  r-  )r  r    r    g0jA:rR  )r  r    r    gQbF5rR  r  c                     |\  }}}}}t          t          j                            |||          |d           t          t          j                            |||          ||           d S )NrR  r   r   )r   r9   r%  r   r  rV  s          r3   r  zTestJohnsonb.test_sf_isf;	  rX  r5   NrY  r;   r5   r3   r\  r\  ,	  sX        = = =E [VU++D D ,+D D Dr5   r\  c                       e Zd Zd Zd Zd Zd Zd Zej	        
                    dddg          d	             Zd
 Zej	        
                    dddg          d             Zd ZdS )TestLogisticc                     t          j        dd          }t          j                            |          }t          j                            |          }t          ||           d S Nr  r  )rj   rY  r9   logisticrq   r   r   rs   r_   r2  r  s       r3   r  zTestLogistic.test_cdf_ppfD	  sS    KR  Nq!!^""2r5   c                     t          j        dd          }t          j                            |          }t          j                            |          }t          ||           d S ra  )rj   rY  r9   rb  r   r  r   rc  s       r3   r  zTestLogistic.test_sf_isfJ	  sS    KR  Na  ^""2r5   c                     d}d}t          t          j                            d|z
            |           t          t          j                            |          |           d S )Ng      <gg|EA@r    )r   r9   rb  r   r  )rs   rv  desireds      r3   test_extreme_valuesz TestLogistic.test_extreme_valuesP	  sV    !#**1q5117;;;**1--w77777r5   c                 t    t           j                            g d          }g d}t          ||d           d S )N)rx  r   r`   )gm\  .g9B.gO& $r   r   )r9   rb  r   r   )rs   r  rX   s      r3   test_logpdf_basiczTestLogistic.test_logpdf_basicW	  sI    ~$$\\\22) ) ) 	hU333333r5   c                 l    t           j                            ddg          }t          |ddg           d S )Nr{   r/  )r9   rb  r   r   rs   r  s     r3   test_logpdf_extreme_valuesz'TestLogistic.test_logpdf_extreme_values_	  s7    ~$$c4[11 	TD$<(((((r5   zloc_rvs,scale_rvs)g9x&?gi '?)gB"?gs^&?c                    t           j                            d||          }d }t          |t           j                            |          |f          j        }t           j                            |          }t          ||d           d S )Nra   r   re   rf   c                    | \  }}t          |          }t          j        t          j        ||z
  |z            dt          j        ||z
  |z            z   z            |dz  z
  }t          j        ||z
  |z  t          j        ||z
  |z            dz
  t          j        ||z
  |z            dz   z  z            |z
  }||fS Nr    rg   )r  rj   rm  r   )inputr   r1   r2   ru  x1x2s          r3   r  z#TestLogistic.test_fit.<locals>.funck	  s    DAqD		AqA~..RVTAXN3335 6 689A>B$(a&$(a0014&$(a001467 8 8:;<B r6Mr5   r[  r  r   )r9   rb  r   r%   	_fitstartr_   r   r   )rs   loc_rvs	scale_rvsr   r  expected_solution
fit_methods          r3   test_fitzTestLogistic.test_fite	  s     ~!!sy!II	 	 	 !u~'?'?'E'EM    	^''--
 	
$5EBBBBBBr5   c                     t           j                            ddd          }t          t           j        |           t          t           j        |d           t          t           j        |d           d S )Nra   r|   rg   rn  r    r   r   )r9   rb  r   r   rs   r   s     r3   test_fit_comp_optimizerz$TestLogistic.test_fit_comp_optimizer}	  sd    ~!!s1!==%end;;;%endCCCC%end1EEEEEEr5   
testlogcdfTFc                     t          j        g d          }|r t          j                            |          }n t          j                            |           }g d}t          ||d           d S )N)r  r/     r=  r  )     g      g%h9fg?~T}%mgCx^&ْV瞯<r   )rj   r   r9   rb  r  r  r   )rs   r~  r_   r2  rX   s        r3   test_logcdfsf_tailsz TestLogistic.test_logcdfsf_tails	  s     H00011 	)%%a((AA$$aR((AF F F8%000000r5   c                     t          j        g ddgdz  z   dgdz  z             }t          t          j        |           d S )N)	i5%   +   r  r  0   6   7   :   ;   r   =   r  )rj   r   r   r9   rb  r|  s     r3   test_fit_gh_18176zTestLogistic.test_fit_gh_18176	  sT     x>>>$(#&(TAX. / / 	&end;;;;;r5   N)r   r   r   r  r  rg  ri  rl  r   r   r   ry  r}  r  r  r;   r5   r3   r_  r_  B	  s            8 8 84 4 4) ) ) [03J3K3M N NC CN NC,F F F [\D%=991 1 :91< < < < <r5   r_  c                   &    e Zd Zd Zd Zd Zd ZdS )
TestLogserc                 D    t           j                            d           d S r5  r6  rx   s    r3   r7  zTestLogser.setup_method	  r8  r5   c                    t           j                            dd          }t          j        |dk              sJ t          j        |          dk    sJ |j        j        t          d         v sJ t           j                            d          }t          |t                    sJ t          j        d                              d          }t          |t          j                  sJ |j        j        t          d         v sJ d S )Nr   r<  r;  r    r>  r   )r9   logserr   rj   r   rB  r   rC  r   rD  rc  rd  rG  s      r3   rJ  zTestLogser.test_rvs	  s    |733vdai     x~~((((z)L"99999lt$$#s#####l4  $$Q''#rz*****y~<!8888888r5   c                 f    t           j                            dd          }t          |d           d S )Nr   r  g&Y3)r9   r  rP  r   rs   r  s     r3   test_pmf_small_pzTestLogser.test_pmf_small_p	  s1    LQ&& 	7#####r5   c                 d    t           j                            d          }t          |d           d S )Nr-  gW  ?)r9   r  r   r   r  s     r3   test_mean_small_pzTestLogser.test_mean_small_p	  s/    Ld## 	;'''''r5   N)r   r   r   r7  rJ  r  r  r;   r5   r3   r  r  	  sP          	9 	9 	9$ $ $( ( ( ( (r5   r  c            	          e Zd Z ej        d          d             Zej                            dej	        ej
        g          ej                            dg d          ej                            dg d          ej                            d	d
dgdd
gf          d                                                 Zej                            dej	        dfej
        dfg          d             ZdS )TestGumbel_r_lfunctionscopec                 @    t           j                            d          S r5  rj   r   r   rx   s    r3   r   zTestGumbel_r_l.rng	      y$$T***r5   rB   ru  r   r   r    rv  r]   r    r   zfix_loc, fix_scaleTFc                     |                     d|||          }t                      }|r|dz  |d<   |r|dz  |d<   t          ||fi | d S )Nra   r   re   rf   r   rg   r   r   )r   r  r   )	rs   rB   ru  rv  r   r>  r   r   r   s	            r3   r}  z&TestGumbel_r_l.test_fit_comp_optimizer	  s|     xxSgY%(  * * vv  	'"Q;DL 	+&]DN 	&dD99D99999r5   z	dist, sgnr    r   c                     |t          j        g d          z  }|                    |          \  }}t          ||dz             t          |dd           d S )N)r   r   r   r   r   r   r   gW  @g   @g3qtw>rv  r   )rj   r   r   r   )rs   rB   r  zre   rf   s         r3   ry  zTestGumbel_r_l.test_fit	  sf     :::;;;XXa[[
U 	S!334445DAAAAAAr5   N)r   r   r   r   fixturer   r   r   r9   gumbel_rgumbel_lr}  ry  r;   r5   r3   r  r  	  s=       V^*%%%+ + &%+ [Venen%EFF[Y


33[[***55[1#UmeT];= =: := = 65 43 GF
:  [[ENA+>,1NB+?+A B BB BB BB B Br5   r  c                      e Zd Zd Zd Z ej        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                                                 Zej        
                    ddd	g          ej        
                    d
dd	g          ej        
                    dddg          ej        
                    dd  eddgd          D                        ej        d          d                                                             Z ej        d          d             Zd Zd ZdS )
TestParetoc                 	   t          j                    5  t          j        dt                     t          j                            dd          \  }}}}t          |t          j                   t          |t          j                   t          |t          j	                   t          |t          j	                   t          j                            dd          \  }}}}t          |t          j                   t          |t          j                   t          |t          j	                   t          |t          j	                   t          j                            dd          \  }}}}t          |d           t          |t          j                   t          |t          j	                   t          |t          j	                   t          j                            dd          \  }}}}t          |d           t          |t          j                   t          |t          j	                   t          |t          j	                   t          j                            d	d          \  }}}}t          |d
           t          |d           t          |t          j	                   t          |t          j	                   t          j                            dd          \  }}}}t          |d           t          |d           t          |t          j	                   t          |t          j	                   t          j                            dd          \  }}}}t          |d           t          |d           t          |dt          j        d          z             t          |t          j	                   t          j                            dd          \  }}}}t          |d           t          |d           t          |dt          j        d          z             t          |t          j	                   t          j                            dd          \  }}}}t          |d           t          |d           t          |dt          j        d          z             t          |d           d d d            d S # 1 swxY w Y   d S )Nrs  r|   r  r  rT  r  r  r   r  g?grq@r   r;  gffffff?gK~?g      2@g۶m۶m?r  UUUUUU?gqq?r  r9  g%I$I?gQ?gUUUUUU@grq?g88Nb@)rw  rx  ry  rz  r9   paretor   rj   r  r  r   r  rs   r  r  rO   r\   s        r3   r  zTestPareto.test_stats	  sk    $&& 7	N 7	N!'>:::++C+@@JAq!QBF###BF###BF###BF###++C+@@JAq!QBF###BF###BF###BF###++C+@@JAq!QC   BF###BF###BF###++C+@@JAq!QC   BF###BF###BF###++C+@@JAq!QAy)))A2333BF###BF###++C+@@JAq!QAs###At$$$BF###BF###++C+@@JAq!QAy)))A2333A	277+;+;;<<<BF###++C+@@JAq!QAy)))Az***A027;3G3GGHHHBF###++C+@@JAq!QAy)))A2333A	RWW-=-==>>>ALMMMo7	N 7	N 7	N 7	N 7	N 7	N 7	N 7	N 7	N 7	N 7	N 7	N 7	N 7	N 7	N 7	N 7	N 7	Ns   R8SS Sc                     d}d}d}t           j                            ||d|          }||z  |z  }t          ||           d S )Ng    eArg   r  r   rd   )r9   r  r   r   )rs   r_   r2   rf   rv  rX   s         r3   r  zTestPareto.test_sf$
  sO    LOOAqauO55!Ga<8$$$$$r5   r  r  c                 @    t           j                            d          S r5  r  rx   s    r3   r   zTestPareto.rng,
  r  r5   z2ignore:invalid value encountered in double_scalarsr   r    rg   r   r   r<  r   c                 ,   t           j                            d||||          }t           j                            |dd          d         }t           j                            |dd          d         }t           j                            |dd          d         }||cxk    r|cxk    rdk    sn J t           j                            d|||dz   |          }t           j                            |d	          \  }	}
}t	          |dz   |                                           |dz
  }|j        d         }t	          |	|t          j        t          j	        ||                                z                      z             t	          |
d           d S )
Nra   r   r2   rf   re   r   r   p=
ף?)r   r   )r   fix_b)r   fbrg   r   )
r9   r  r   r   r   r  rB  rj   rm  r  )rs   r   r   r<  r   r   shape_mle_analytical1shape_mle_analytical2shape_mle_analytical3shape_mle_a	loc_mle_ascale_mle_a
data_shiftndatas                 r3   ry  zTestPareto.test_fit0
  s    |SIY$+#   ? ? !& 0 0A$ 0 G G J % 0 0AT 0 J J1 M % 0 0A$ 0 G G J%)> . . . .%. . . .)-. . . . . . |SIY%,q[   E E.3l.>.>t!.>.L.L+Y[1_dhhjj111AX
 #[RVBF:jnn6F6F+F$G$GHHH	J 	J 	JY"""""r5   r]   fix_shape, fix_loc, fix_scalec                     g | ]}d |v |	S Fr;   rA   rv  s     r3   rC   zTestPareto.<listcomp>P
  (     - - -A!&!  !+r5   TFr   repeatignoreinvalidc                     t           j                            d||||          }i }	|r||	d<   |r||	d<   |r||	d<   t          t           j        |fi |	 d S )Nra   r  r   r   r   )r9   r  r   r   
rs   r   r   r<  r   r   r>  r   r   r   s
             r3   r   z&TestPareto.test_fit_MLE_comp_optimizerL
  s     |SIY$+#   ? ?  	#"DJ 	#"DL 	'&DN%elDAADAAAAAr5   c           	          d\  }}}t           j                            |||dt          j                            d                    }t          t           j        |           d S )N)r    r   r    ra   ið& r   )r9   r  r   rj   r   r   r   )rs   rB  locationrf   r   s        r3   test_fit_known_bad_seedz"TestPareto.test_fit_known_bad_seedb
  sa    
 ")x|xS-/Y-B-B7-K-K   M M%elD99999r5   c                     t          t          j                   t          t          t          j        j        g dd           t          t          t          j        j        g ddd           d S )Nr  rg   r   )r   rg   r   r    r   rB  )r  r9   r  r`  r#   r   rx   s    r3   test_fit_warningszTestPareto.test_fit_warningsl
  sg    EL)))lEL$4iiiaHHHHlEL$4iiia	  	  	  	  	  	 r5   c                     t           j                            ddd|          }t          |d           t           j                            |          }d S )Ni~r    ra   )re   r2   r   r   r   )r9   r  r   r   r   )rs   r   r   r   s       r3   test_negative_datazTestPareto.test_negative_datat
  sK    |DACcJJ$"""
 LT""r5   N)r   r   r   r  r  r   r  r   r   filterwarningsr   ry  r(   rj   errstater   r  r  r  r;   r5   r3   r  r  	  s%       :N :N :Nx% % % V^*%%%+ + &%+ [ !1 2 2[[1a&11[YA//[[1a&11# # 21 0/ 212 2
#. [[2q'22[YA//[[1a&11[<- -$q)I)I)I - - -. . R["""B B #". . 21 0/ 32B R[""": : #":     # # # # #r5   r  c                      e Zd Zd Zd Zd Zd Zd Zd Zd Z	d Z
d	 Zej                            d
dg dfdddd ej        d          z  ej        gfdddd ej        d          z  dgfdg dfg          d             Zd ZdS )TestGenparetoc                 r   dD ]i}t          j        |          }t          j                            |          \  }}t          |d           t          t          j        |                     jt          j        d          }t          j                            |          \  }}t          ||gddg           d S )NrT  r   r          r|   )	rj   r   r9   	genpareto_get_supportr   r	   isposinfr   )rs   r  r1   r2   s       r3   test_abzTestGenpareto.test_ab
  s     	$ 	$A
1A?//22DAqBBKNN#### JsOO++A..1AS	*****r5   c                    t          j        d          }t          j        ddd          }t	          |                    |          t           j                            |                     t	          |                    |          t           j                            |                     t	          |                    |          t           j                            |                     t          j        ddd          }t	          |	                    |          t           j        	                    |                     d S )Nr   r  r   r  r:  rT  r`   )
r9   r  rj   rY  r   ri   rg  rq   r   r   rs   rvr_   r  s       r3   test_c0zTestGenpareto.test_c0
  s     _r"""K3##q		5;??1#5#5666q		5;??1#5#5666a%+.."3"3444KB##q		5;??1#5#566666r5   c                    t          j        d          }t          j        ddd          }t	          |                    |          t           j                            |                     t	          |                    |          t           j                            |                     t	          |                    |          t           j                            |                     t          j        ddd          }t	          |	                    |          t           j        	                    |                     t	          |
                    d	          d           d S )
Nr  r  r   r  r:  r   rT  r`   r    )r9   r  rj   rY  r   ri   uniformrq   r   r   r   r  s       r3   test_cm1zTestGenpareto.test_cm1
  s   _s###K3##q		5=#4#4Q#7#7888q		5=#4#4Q#7#7888a%-"2"21"5"5666KB##q		5=#4#4Q#7#7888 			!a(((((r5   c                    t          j        d          }t          |                    t          j                  |                    t          j                  gddg           t          t	          j        |	                    t          j                                       t          j        d          }t          |                    t          j                  |                    t          j                  gddg           t          t	          j        |	                    t          j                                       t          j        d          }t          |                    t          j                  |                    t          j                  gddg           t          t	          j        |	                    t          j                                       d S )Nr]   r  r   rT  r  )
r9   r  r   ri   rj   r  rq   r	   isneginfr   rs   r  s     r3   
test_x_infzTestGenpareto.test_x_inf
  sC   _s###82r(CCCBIIbf--..///_r"""82r(CCCBIIbf--..///_s###82r(CCCBIIbf--../////r5   c                    t          j        ddd          }dD ]}t          j                            ||          }dD ]7}t          j                            |||z             }t          ||d           8t          j                            ||          }dD ]7}t          j                            |||z             }t          ||d           8d S )	Nr   r`   r:  r   r   r  g+r  r   )r  r  )rj   rY  r9   r  ri   r   rq   )rs   r_   r  pdf0dcpdfccdf0cdfcs           r3   test_c_continuityzTestGenpareto.test_c_continuity
  s    K2r"" 		8 		8A?&&q!,,D% 8 8**1a"f55d77777?&&q!,,D$ 8 8**1a"f55d777778		8 		8r5   c           	      t   t           j        t          j        ddd          t          j        dddd          d	t          j        ddd          z
  f         }d
D ]\}t          j                            ||          }dD ]7}t          j                            |||z             }t          ||d           8]d S Nr  r~   r]   baser    r:  FendpointrT  )r   r  r  r   )rj   rN  r  rY  r9   r  r   r   )rs   r  r  ppf0r  ppfcs         r3   test_c_continuity_ppfz#TestGenpareto.test_c_continuity_ppf
      E"+eT444+dArE:::r{5$S9999: ;  	8 	8A?&&q!,,D% 8 8**1a"f55d777778	8 	8r5   c           	      t   t           j        t          j        ddd          t          j        dddd          d	t          j        ddd          z
  f         }d
D ]\}t          j                            ||          }dD ]7}t          j                            |||z             }t          ||d           8]d S r  )rj   rN  r  rY  r9   r  r  r   )rs   r  r  isf0r  isfcs         r3   test_c_continuity_isfz#TestGenpareto.test_c_continuity_isf
  r  r5   c           	      \   t           j        t          j        ddd          t          j        dddd          d	t          j        ddd          z
  f         }d
D ]P}t	          t
          j                            t
          j                            ||          |          |d           Qd S )Nr  r~   r]   r  r    r:  Fr  rT  )r-  gC]r2r   gV瞯Ҽr   r   )	rj   rN  r  rY  r   r9   r  rq   r   )rs   r  r  s      r3   test_cdf_ppf_roundtripz$TestGenpareto.test_cdf_ppf_roundtrip
  s    E"+eT444+dArE:::r{5$S9999: ; / 	+ 	+AEO//0C0CAq0I0I1MME+ + + + +	+ 	+r5   c                 j    t           j                            dddd          }t          |d           d S )Nr  r~   r   r    gpEȜ)r9   r  r  r   rk  s     r3   r  zTestGenpareto.test_logsf
  s4    $$T3155122222r5   zc, expected_statsr   )r    r    rg   r3  rl  r  gqq@r`   rg   gqq?g      ?g$I$	?rq?r  gll0@r   )r|   UUUUUU?r   333333c                 n    t           j                             |d          }t          ||dd           d S )Nr  r  r   r   rf  )r9   r  r   )rs   r  expected_statsr  s       r3   r  zTestGenpareto.test_stats
  s;     &&q&&99UGGGGGGr5   c                 h    t           j                            d          }t          |dd           d S )Nr-  gvǼ
  ?r   r   )r9   r  rA  r   )rs   r  s     r3   test_varzTestGenpareto.test_var
  s3    O%%,5999999r5   N)r   r   r   r  r  r  r  r  r  r  r   r  r   r   r   rj   r  r  r  r  r;   r5   r3   r  r  ~
  sA       + + +
7 
7 
7) ) )0 0 08 8 88 8 88 8 8+ + +3 3 3 [
\\\	
T2gbgajj="&1	2
]T7271::$5v>	?
"""	#	%& &H H& &H: : : : :r5   r  c                   >    e Zd Zd Zd Zd Zd Zd Zd Zd Z	d Z
d	S )
TestPearson3c                 D    t           j                            d           d S r5  r6  rx   s    r3   r7  zTestPearson3.setup_method
  r8  r5   c                    t           j                            dd          }t          j        |          dk    sJ |j        j        t          d         v sJ t           j                            d          }t          |t                    sJ t          j        d                              d          }t          |t          j
                  sJ |j        j        t          d         v sJ t          |          dk    sJ d S )Nr]   r<  r;  AllFloatr|   r   )r9   r,   r   rj   rB  r   rC  r   rD  r   rd  r  rG  s      r3   rJ  zTestPearson3.test_rvs
  s    ~!!#G!44x~~((((z)J"77777n  %%#u%%%%%nS!!%%a((#rz*****y~:!666663xx1}}}}}}r5   c                    t           j                            dg d          }t          |t	          j        g d          d           t           j                            dd          }t          |t	          j        dg          d           t           j                            g d	d          }t          |t	          j        g d
          d           d S )Nrg   r   r]   rQ  )gtT?g+q?g?rv  r   r  r]   ND}i?r  r   r   r   r    )r  gj?gdC?gs?g'M?)r9   r,   ri   r   rj   r   r  s     r3   rR  zTestPearson3.test_pdf  s    ~!!!___55bh'K'K'KLL!	# 	# 	# 	#~!!"c**bh
|444@@@@~!!"4"4"4c::bh (@ (@ (@ A AGK	M 	M 	M 	M 	M 	Mr5   c                 h   t           j                            dg d          }t          |t	          j        g d          d           t           j                            dd          }t          |dgd           t           j                            g d	d          }t          |g d
d           d S )Nrg   r  )grE?gTp]0?g~?rv  r   r  r]   g E&"J?r  )giZ*J?gh+8w?gcK?g|<x6?gG?)r9   r,   rq   r   rj   r   r  s     r3   r\  zTestPearson3.test_cdf  s    ~!!!___55bh'K'K'KLL!	# 	# 	# 	#~!!"c**zl6666~!!"4"4"4c:: ? ? ?EI	K 	K 	K 	K 	K 	Kr5   c                     g d}ddt           j                            |          }fd|D             }t          ||           d S )Nr  r   r   r|   r|   r  c                 l    g | ]0}t          t          j        |          j                  d          1S r?   )r   r9   r,   ri   )rA   rB  neg_infx_evals     r3   rC   z<TestPearson3.test_negative_cdf_bug_11186.<locals>.<listcomp>  sG     ' ' ' --17FCCAF ' ' 'r5   )r9   r,   rq   r   )rs   skewsr  int_pdfsr  r  s       @@r3   test_negative_cdf_bug_11186z(TestPearson3.test_negative_cdf_bug_11186  su     !  ~!!&%00' ' ' ' ' %' ' 'h'''''r5   c                 6   t           j                            dd          }t          |d           t	          |t
          j                  sJ t           j                            dd          }t          |d           t	          |t
          j                  sJ d S )Nr    rg   r   rv  )r9   r,   r  r   rD  rj   number)rs   r  s     r3   test_return_array_bug_11746z(TestPearson3.test_return_array_bug_11746#  s     &&q!,,VQ&"),,,,,&&q(33VQ&"),,,,,,,r5   c                    g d}d}t           j                            t           j                            ||          |          }t	          ||           t          j        dgdgg          }t          j        dd          }t	          t           j                            ||          t           j                            | |                      t	          t           j                            ||          t           j        	                    | |                      t	          t           j                            ||          t           j        
                    ||                       d S )Nr  r|   r   r  r   rg   )r9   r,   r   rq   r   rj   r   rY  ri   r   r  )rs   r  r  r   rB  r_   s         r3   test_ppf_bug_17050zTestPearson3.test_ppf_bug_17050.  s?    !  n  !3!3FE!B!BEJJV$$$ x$#((KA**1d33**A2u55	7 	7 	7**1d33))1"te44	6 	6 	6**1d33++Au555	7 	7 	7 	7 	7r5   c                    g d}g d}g d}t          t          j                            ||          |d           t          t          j                            |d          t          j                            |          d           d S )N)r]   r|   rT  皙)rU  r  r  r%  )g7>g~׼=gp'+7g;r  r   r   )r   r9   r,   r   r  )rs   rB  r_   r  s       r3   r  zTestPearson3.test_sfA  s     %$$"""> > >))!T22CeDDDD))!Q//q1A1ANNNNNNr5   N)r   r   r   r7  rJ  rR  r\  r  r  r  r  r;   r5   r3   r
  r
  
  s          	 	 	M M MK K K	( 	( 	(	- 	- 	-7 7 7&	O 	O 	O 	O 	Or5   r
  c                   8    e Zd Zd Zd Zd Zd Zd Zd Zd Z	dS )	
TestKappa4c                     g d}d}dD ]T}t           j                            |||          }t           j                            ||           }t	          ||           Ud S )N)r   r]   rQ  r|   rT  )
gffffffr  r   皙ɿr!  r]   rQ  r|   rT  ffffff?)r9   kappa4rq   r  r   rs   r_   rp  r\   rH  	vals_comps         r3   test_cdf_genparetozTestKappa4.test_cdf_genparetoN  sr        	- 	-A<##Aq!,,D++Ar22ID),,,,	- 	-r5   c                    t          j        ddd          }d}t          j        ddd          }t          j                            |||          }t          j                            ||          }t          ||           d S )Nr  r   r`   r   r  r   )rj   rY  r9   r'  rq   
genextremer   r(  s         r3   test_cdf_genextremezTestKappa4.test_cdf_genextremeY  st    KAr""KAr""|1a(($((A..	i(((((r5   c                     t          j        ddd          }d}d}t          j                            |||          }t          j                            |          }t          ||           d S )Nr   r`   rT  r   )rj   rY  r9   r'  rq   rg  r   r(  s         r3   test_cdf_exponzTestKappa4.test_cdf_exponb  sa    K2r""|1a((KOOA&&	i(((((r5   c                     t          j        ddd          }d}d}t          j                            |||          }t          j                            |          }t          ||           d S )Nr  r   r`   r   )rj   rY  r9   r'  rq   r  r   r(  s         r3   test_cdf_gumbel_rzTestKappa4.test_cdf_gumbel_rk  sc    KAr""|1a((N&&q))	i(((((r5   c                     t          j        ddd          }d}d}t          j                            |||          }t          j                            |          }t          ||           d S )Nr  r   r`   r  r   )rj   rY  r9   r'  rq   rb  r   r(  s         r3   test_cdf_logisticzTestKappa4.test_cdf_logistict  sc    KAr""|1a((N&&q))	i(((((r5   c                     t          j        ddd          }d}d}t          j                            |||          }t          j                            |          }t          ||           d S )Nr  r   r`   rT  )rj   rY  r9   r'  rq   r  r   r(  s         r3   test_cdf_uniformzTestKappa4.test_cdf_uniform}  sc    KAr""|1a((M%%a((	i(((((r5   c                 0    t          j        dd           d S rp  )r9   r'  rx   s    r3   test_integers_ctorzTestKappa4.test_integers_ctor  s     	Qr5   N)
r   r   r   r*  r-  r/  r1  r3  r5  r7  r;   r5   r3   r#  r#  M  s}        	- 	- 	-) ) )) ) )) ) )) ) )) ) )    r5   r#  c                   ,    e Zd Zd Zd Zd Zd Zd ZdS )TestPoissonc                 D    t           j                            d           d S r5  r6  rx   s    r3   r7  zTestPoisson.setup_method  r8  r5   c                     t          j        d          }t          j                            g d|          }d|dz  |dz  dz  g}t          ||           d S )Nrg   r   r    rg   r|   r   )rj   r  r9   r  rP  r   )rs   ln2rH  rX   s       r3   test_pmf_basiczTestPoisson.test_pmf_basic  sX    fQii}  C00QQq)h'''''r5   c                     t           j                            g dd          }g d}t          ||           t           j                            dd          }t          |d           d S )Nr<  r   )r    r   r   ffffff?r   r   )r9   r  rP  r   intervalr   )rs   rH  rX   rB  s       r3   test_mu0zTestPoisson.test_mu0  sc    }  A..994***=))$22Xv&&&&&r5   c                    t           j                            dd          }t          j        |dk              sJ t          j        |          dk    sJ |j        j        t          d         v sJ t           j                            d          }t          |t                    sJ t          j        d                              d          }t          |t          j                  sJ |j        j        t          d         v sJ d S )Nr|   r<  r;  r   r>  r   )r9   r  r   rj   r   rB  r   rC  r   rD  rc  rd  rG  s      r3   rJ  zTestPoisson.test_rvs  s    }  7 33vdai     x~~((((z)L"99999m$$#s#####mC  $$Q''#rz*****y~<!8888888r5   c           	         d}t           j                             |d          }t          |||t          j        d|z            d|z  g           t          j        g d          }t           j                             |d          }||t          j        ddt          j        d          z  gt          j        ddgf}t          ||           d S )	Ng      0@r  r  rT  )r   rT  r   r    rg   r|   )r9   r  r   rj   r  r   r  )rs   r^  r  rX   s       r3   r  zTestPoisson.test_stats  s    $$R$88RR#b& ABBBXooo&&$$R$88RVQ"'!**537GH)))))r5   N)r   r   r   r7  r>  rC  rJ  r  r;   r5   r3   r9  r9    s_          ( ( (' ' '	9 	9 	9* * * * *r5   r9  c                   V    e Zd Zd Zd Zd Zd Zd Zd Zd Z	d Z
d	 Zd
 Zd Zd ZdS )	TestKSTwoc                 D    t           j                            d           d S r5  r6  rx   s    r3   r7  zTestKSTwo.setup_method  r8  r5   c                    dD ]}t          j        dd|z  d|z  ddd|z  z
  dg          }d|z  |z  }t          j                            |dz             }|dk    rt          j        |          nd}t          j        dd||z  ddt          j                            d|          z  z
  t          dd|z  z
  d          dg          }t          j
                            ||          }t          ||           d S )Nr    rg   r   r`   ra   r   r   r|   r    rT  rg   r   )rj   r   r8   r   gammalnr   r9   ksoner   r  kstworq   r   )rs   ru  r_   v1lgelgrX   r  s           r3   r\  zTestKSTwo.test_cdf  s   ) 	0 	0A !SUAaCaAgq9::Aa%!B&&qs++B!#q26":::aCxArCx!"Qu{~~c1'='=%=!=!$Q2Xs!3!3!$!& ' 'H {q!,,HHh////#	0 	0r5   c                    t          j        ddd          }dD ]}t          j        dd|z  d|z  ddd|z  z
  dg          }d|z  |z  }t          j                            |dz             }|dk    rt          j        |          nd}t          j        ddd||z  z
  dt          j        	                    d|          z  t          d|z  d          dg          }t          j        	                    ||          }t          ||           d S )Nr   r    r  rJ  r|   rT  rg   )rj   rY  r   r8   r   rK  r   r9   rL  r   r  rM  r   )rs   r_   ru  rN  rO  rP  rX   r  s           r3   r  zTestKSTwo.test_sf  s   K1b!!) 	/ 	/A!SUAaCaAgq9::Aa%!B&&qs++B!#q26":::aCxc!"R#X!"5;>>#q#9#9!9!$QrT3!4 5 5H knnQ**GGX....	/ 	/r5   c                 
   t          j        ddd          dd          }g d}|D ]]}|t          j        |          z  }t          j                            ||          }t          j        |          }t          |d           ^d S )Nr   rg   r  r    )r=  ra   rS  i  r   r   r-  )rj   rY  r  r9   rM  rq   r)  r   )rs   r_   ns_xxnprobsdiffss          r3   test_cdf_sqrtnzTestKSTwo.test_cdf_sqrtn  s    
 K1b!!!""%,,, 	+ 	+Bbgbkk!BKOOB++EGENNEeT****		+ 	+r5   c                     t          j        ddd          }dD ]U}t          j                            ||          }t          j                            ||          }t          |d|z
             Vd S Nr   r    r  rJ  )rj   rY  r9   rM  rq   r   r   )rs   r_   ru  r  r  s        r3   r  zTestKSTwo.test_cdf_sf  so    K1b!!) 	= 	=A{q!,,HknnQ**G%hG<<<<	= 	=r5   c                    t          j        ddd          }dD ]l}|t          j        |          z  }t          j                            ||          }t          j                            ||          }t          |d|z
             md S rZ  )rj   rY  r  r9   rM  rq   r   r   )rs   r_   ru  rU  r  r  s         r3   test_cdf_sf_sqrtnzTestKSTwo.test_cdf_sf_sqrtn  s    K1b!!) 	= 	=ARWQZZB{r1--HknnR++G%hG<<<<		= 	=r5   c                 0   t          j        ddd          }dD ]|}||d|z  k             }t          j                            ||          }d|k     |dk     z  }t          j                            ||          }t          ||         ||         d           }d S )	Nr   r    r  rJ  r|   r  r  r   )rj   rY  r9   rM  rq   r   r   rs   r_   ru  rU  r  condrH  s          r3   test_ppf_of_cdfzTestKSTwo.test_ppf_of_cdf  s    K1b!!) 	= 	=A1s1u9B{r1--HLX_5D;??8Q//DDJ4t<<<<<	= 	=r5   c                 0   t          j        ddd          }dD ]|}||d|z  k             }t          j                            ||          }d|k     |dk     z  }t          j                            ||          }t          ||         ||         d           }d S )	Nr   r    r  rJ  r|   rT  r  r   )rj   rY  r9   rM  r  r   r   )rs   r_   ru  rU  vals_isfr_  rH  s          r3   test_isf_of_sfzTestKSTwo.test_isf_of_sf  s    K1b!!) 	= 	=A1s1u9B{r1--HLX^4D;>>(A..DDJ4t<<<<<	= 	=r5   c                 V   t          j        ddd          }dD ]}|t          j        |          z  |d|z  k             }t          j                            ||          }d|k     |dk     z  }t          j                            ||          }t          ||         ||                    d S )Nr   r    r  rJ  r|   rT  )rj   rY  r  r9   rM  rq   r   r   r^  s          r3   test_ppf_of_cdf_sqrtnzTestKSTwo.test_ppf_of_cdf_sqrtn  s    K1b!!) 	2 	2Abgajj.!c!e),B{r1--HLX^4D;??8Q//DDJ41111	2 	2r5   c                 V   t          j        ddd          }dD ]}|t          j        |          z  |d|z  k             }t          j                            ||          }d|k     |dk     z  }t          j                            ||          }t          ||         ||                    d S )Nr   r    r  rJ  r|   r@  )rj   rY  r  r9   rM  r   r  r   )rs   r_   ru  rU  r  r_  rH  s          r3   test_isf_of_sf_sqrtnzTestKSTwo.test_isf_of_sf_sqrtn  s    K1b!!) 	2 	2Abgajj.!c!e),BknnR++GKGdN3D;??7A..DDJ41111	2 	2r5   c                     t          j        ddd          dd          }dD ]R}t          j                            ||          }t          j                            ||          }t          ||           Sd S rZ  )rj   rY  r9   rM  r   rq   r   )rs   rV  ru  rU  r  s        r3   r  zTestKSTwo.test_ppf  st    Aq"%%abb)) 	- 	-A**B{r1--HHe,,,,	- 	-r5   c           	         g d}t          j        g d          }t          j        g dg dg dg dg dg dg          }t          |          D ]y\  }}|t          j        d	          z  t          j        t           j        d	z  |z            z  }t          j                            ||          }t          |||         d
           zd S )N)r`   r=  ra   rS  r  r   )rl  gUUUUUU?r|   r    rg   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?rg   r;  r   )
rj   r   	enumerater  r  rk   r9   rM  rq   r   )rs   rS  ratiosrX   idxru  r_   r  s           r3   test_simard_lecuyer_table1z$TestKSTwo.test_simard_lecuyer_table1&  s    +**888998- - -- - -- - -- - -- - -- - -
    mm 	@ 	@FC"RWRU1WQY%7%77A{q!,,HHhsm$?????	@ 	@r5   N)r   r   r   r7  r\  r  rX  r  r\  r`  rc  re  rg  r  rm  r;   r5   r3   rG  rG    s          0 0 0(/ / /+ + += = == = == = == = =2 2 22 2 2- - -@ @ @ @ @r5   rG  c                        e Zd Zd Zd Zd ZdS )TestZipfc                 D    t           j                            d           d S r5  r6  rx   s    r3   r7  zTestZipf.setup_methodI  r8  r5   c                    t           j                            dd          }t          j        |dk              sJ t          j        |          dk    sJ |j        j        t          d         v sJ t           j                            d          }t          |t                    sJ t          j        d                              d          }t          |t          j                  sJ |j        j        t          d         v sJ d S )Nr  r<  r;  r    r>  r   )r9   zipfr   rj   r   rB  r   rC  r   rD  rc  rd  rG  s      r3   rJ  zTestZipf.test_rvsL  s    z~~c~00vdai     x~~((((z)L"99999jnnS!!#s#####joo!!!$$#rz*****y~<!8888888r5   c                 v   t           j                             d          \  }}t          t          j        |                     t          |t          j                   t           j                             dd          \  }}t          t          j        ||g                                                      d S )Nffffff@r1   r<  skr1   r  )r9   rr  r	   rj   rG  r   r  r   r  s        r3   r*  zTestZipf.test_momentsW  s    z#&&1AQz#t441BKA''++---.....r5   N)r   r   r   r7  rJ  r*  r;   r5   r3   ro  ro  H  sA          	9 	9 	9/ / / / /r5   ro  c                   &    e Zd Zd Zd Zd Zd ZdS )TestDLaplacec                 D    t           j                            d           d S r5  r6  rx   s    r3   r7  zTestDLaplace.setup_methodb  r8  r5   c                    t           j                            dd          }t          j        |          dk    sJ |j        j        t          d         v sJ t           j                            d          }t          |t                    sJ t          j        d                              d          }t          |t          j
                  sJ |j        j        t          d         v sJ t           j                            d          J d S )Nr  r<  r;  r>  r   r   )r9   dlaplacer   rj   rB  r   rC  r   rD  rc  rd  rG  s      r3   rJ  zTestDLaplace.test_rvse  s    ~!!#G!44x~~((((z)L"99999n  %%#s#####nS!!%%a((#rz*****y~<!88888~!!#&&22222r5   c                    d}t          j        |          }|                     d          \  }}}}d}t          j        | |dz             }|                    |          }	t          j        |	|dz  z            t          j        |	|dz  z            }}
t          ||fd           t          ||f|
||
dz  z  dz
  fd	d
           d S )NrT  r  r  r    rg   r   rA  r  r  r-  rn  )r9   r|  rj   r  rP  rm  r   r   )rs   r1   dlr  r  rO   r\   r2  r  ppm2m4s               r3   r  zTestDLaplace.test_statsp  s    ^AXXf%%
1aYr1Q3VVBZZ2q5!!26"RU(#3#3BaVV$$$ARAX] 3%dKKKKKKr5   c                     t          j        d          }t          j        |          }|                    d          \  }}}}t	          ||fd           t          ||fd           d S )Nr   r  )r   r   )r        
@)rj   r  r9   r|  r   r   )rs   r1   r~  r  r  rO   r\   s          r3   test_stats2zTestDLaplace.test_stats2}  se    F2JJ^AXXf%%
1aaVX&&&A
+++++r5   N)r   r   r   r7  rJ  r  r  r;   r5   r3   ry  ry  a  sS          	3 	3 	3L L L, , , , ,r5   ry  c                       e Zd Zd Zej                            dddg          d             Zej                            dddg           fd            Zd Z	d	 Z
d
 Zej                            dg d          d             Z xZS )TestInvgaussc                 D    t           j                            d           d S r5  r6  rx   s    r3   r7  zTestInvgauss.setup_method  r8  r5   zrvs_mu,rvs_loc,rvs_scale)rg   r   r    )g
ףp=@g r@gʡE6@c                    t           j                            d|||          }t           j                            ||          \  }}}||z
  }t	          j        |          }t          |          t	          j        |dz  |dz  z
            z  }	||	z  }
t          |
|dd           t          |	|dd           t          ||           t           j                            d|||          }t           j                            ||dz
  |dz             \  }}}t          |dz   |           t          |dz
  |           t           j                            |d	
          d         }t           j                            |d	          d         }t           j                            |d	          d         }||cxk    r|cxk    rd	k    sn J d S )Nra   )r   r^  re   rf   r   r   r   rn  r    rB  r  )fmur   )fix_mur   )
r9   r#  r   r   rj   r   r  rm  r   r   )rs   rvs_mur   r<  r   r^  re   rf   mu_temp	scale_mlemu_mle
shape_mle1
shape_mle2
shape_mle3s                 r3   ry  zTestInvgauss.test_fit  s    ~!!sv&-Y " @ @ ++Dw+??Cg~'$--IIr
Wr](B!C!CD	" 	U;;;;	5u5AAAAS'"""~!!sv&-Y " @ @ ++Dw{3<q= , B BCY]E***Wq[#&&& ^''$'77:
^''T'::1=
^'''66q9
Z====:============r5   )gX9v>@g	@gGz@c                 \   t           j                            d          }t          j                            d||||          }t          t          t          j                  t          j                  j        } ||          }t          j                            |          }t          ||            ||dd          }t          j                            |dd          }t          ||           t          t          j        ||           t          j        ||dz
  z
  dk              sJ t          t          j        ||dz
             t          t          j        |d           t          t          j        ||t           j                            d          d         	           d S )
Nr   ra   )r   r^  re   rf   r   r   rg   )r   r  r   r    rB  )rj   r   r  r9   r#  r   r   r   r   r   r   r   rand)
rs   r  r   r<  r   r   	super_fitsuper_fittedinvgauss_fit	__class__s
            r3   r   z(TestInvgauss.test_fit_MLE_comp_optimizer  s    i##D))~!!sv&-YS " R R $u~..??C	 y~))$//\<000 !yA1555~))$QA)>>\<000 	&endIIII vtw{+q011111%end1MMMM 	&endCCCC 	&end-/Y^^A->->q-A	C 	C 	C 	C 	C 	Cr5   c                     t          t          j                   t          j        t
                    5  t          j                            g dd           d d d            d S # 1 swxY w Y   d S rH  )r  r9   r#  r   r   r#   r   rx   s    r3   test_fit_raise_errorsz"TestInvgauss.test_fit_raise_errors  s    EN+++]<(( 	2 	2Nyyyq111	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2s   $A$$A(+A(c                    g d}g d}t           j                            d|          }t          ||           t           j                            dd          }t	          |d           t           j                            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    皙?r^  r  ?gfFn   gr^:g9̗?r  gx_;:g!?gٷ
??)r9   r#  rq   r   r   r   )rs   r^  rX   rT   
cdf_actual	sf_actuals         r3   r  zTestInvgauss.test_cdf_sf  s    . . ."??##CB#//Xv&&& ^''$'77

$9:::N%%cd%33		#7888 ##GV44 3444 ##Hf55 122222r5   c                    t           j                            dd          }t          |d           t           j                            dd          }t          |d           t           j                            dd          }t          |d           t           j                            dd          }t          |d	           d S )
Nr  r  r  giJ.r  gr^ߺr  gpgFgpQ^L)r9   r#  r  r   r  )rs   r  r  s      r3   r  zTestInvgauss.test_logcdf_logsf  s     &&v$&77 1222&&sD11 5666$$Ut$445666$$S$//011111r5   zmu, ref))r  g,9)r  g%!)r~   g40)r   gK'
@)r  g`p
@c                 d    t          t          j                            |          |d           d S rT  )r   r9   r#  rH   )rs   r^  r  s      r3   rq  zTestInvgauss.test_entropy  s/     	..r22CeDDDDDDr5   )r   r   r   r7  r   r   r   ry  r   r  r  r  rq  __classcell__r  s   @r3   r  r    s;          [7')>?A A> >A A>: [7')>?A A C  C  C  CA A CD2 2 2!3 !3 !3F2 2 2. [Y )D )D )D E E
E EE E
E E E E Er5   r  c                   `    e Zd Zej                            dg d          d             Zd ZdS )
TestLandauname)ri   rq   r   r   r  c                    |dv rt          j        ddd          f}nt          j        ddd          f}t          t          j        |          }t          t          j        |          } || } |g |ddR  }t          ||d	           d S )
N>   r  r   r]   r   r   r   r`   r    r  r   )rj   rY  r.  r9   landaulevy_stabler   )rs   r  r_   landau_methodlevy_methodr   r  s          r3   test_landau_levy_agreementz%TestLandau.test_landau_levy_agreement  s    
 >!!Ca(()AAB2&&'Ad33e/66mQk#1#a####Su------r5   c                     t          t          j                            d          t          j        fdz             t          t          j                            d          t          j                   d S )Nr  r  r   r   )r   r9   r  rj   r  r  rx   s    r3   r*  zTestLandau.test_moments&  sU    U\'''77"&1EEEU\((++RV44444r5   N)r   r   r   r   r   r   r  r*  r;   r5   r3   r  r    sW        [V%G%G%GHH. . IH.5 5 5 5 5r5   r  c                       e Zd Zej                            dg d          ej                            dg d          d                         Zej                            dg d           fd            Zd	 Zd
 Z	d Z
d Zd Z xZS )TestLaplacer   )r  r   r    rg   r<  )r    rg   r   r`   c                    t           j                            d          }t          j                            d|||          }t          j        |          }t          j        t          j        ||z
                      t          |          z  }t          j        
                    |          \  }}t          ||dd           t          ||dd           t          j        
                    ||          \  }}t          ||dd           t          j        
                    ||          \  }}t          ||           |dz  }t          j        t          j        ||z
                      t          |          z  }t          j        
                    ||          \  }}t          ||           t          j        
                    ||          \  }}t          ||           t          t          t          j        j
        |||	           t          t          t          j        j
        t           j        g           t          t          t          j        j
        t           j        g           d S )
Nr   ra   r  r   rn  r   r{  rg   rB  )rj   r   r  r9   r  r   rH  rm  r   r  r   r   r   r`  r  r  r  r  )	rs   r   r<  r   r   loc_mler  re   rf   s	            r3   ry  zTestLaplace.test_fit-  s   
 i##D))}  cwi.1 ! 3 3 )D//F26$.1122SYY>	 ]&&t,,
UW5u====yu5AAAA ]&&t'&::
Uyu5AAAA]&&tI&>>
UW%%% kF26$*--..T:	 ]&&t#&66
UY&&& ]&&tI&>>
UWc""" 	lEM$5t'&	( 	( 	( 	( 	j%-"3bfX>>>j%-"3bfX>>>>>r5   rvs_loc,rvs_scale))r  r`   r`   r   )r|   rQ  c                    t           j                            d          }t          j                            d|||          }d }t          j                            |          \  }}t          t          t          j                  t          j                                      |          \  }}	 ||||          }
 |||	|          }|
|k     st          j	        |
|dd          sJ d S d S )Nr   r   r  c           	          dt          |           t          j        d|z            z  d|z  t          j        t          j        || z
                      z  z
  z  S )Nr   rg   r    )r  rj   r  rm  r   )re   rf   r   s      r3   llz3TestLaplace.test_fit_MLE_comp_optimizer.<locals>.lle  sS    CII%8E'26"&*<*<#=#==> ? ?r5   r   rn  )
rj   r   r  r9   r  r   r   r   r   r   )rs   r   r<  r   r   r  re   rf   loc_opt	scale_optll_mlell_optr  s               r3   r   z'TestLaplace.test_fit_MLE_comp_optimizer\  s     i##D))}  dy.1 ! 3 3	? 	? 	? ]&&t,,
U"4#6#6#(=2 225#d)) 	C%%GY--"+ff38u#F #F #Fr5   c                    t          j        g d          }t          j                            |d          \  }}t          |ddd           t          j                            |d          \  }}t          |ddd           d S )N)rT  rT  r  rU  r%  r  r3  r   r   r   rn  r{  )rj   r   r9   r  r   r   )rs   r   re   rf   s       r3   test_fit_simple_non_random_dataz+TestLaplace.test_fit_simple_non_random_datas  s    x77788]&&t!&44
Uqu59999]&&tA&66
UQU777777r5   c                 F   d}t           j                            |           }|dk    sJ t           j                            |          }|dk    sJ t           j                            |          }|dk    sJ t           j                            |           }|dk    sJ d S )Nr   r   rT  )r9   r  rq   r   )rs   r_   p0p1s       r3   test_sf_cdf_extremesz TestLaplace.test_sf_cdf_extremes|  s    ]r"" Syyyy ]q!!Syyyy]a   Syyyy ]qb!!Syyyyyyr5   c                     d}t           j                            |          }t          |t	          j        |           dz  d           d S )NrS  rg   r   r   )r9   r  r   r   rj   r   )rs   r_   rv  s      r3   r  zTestLaplace.test_sf  sE    MQ261"::a<e444444r5   c                     d}t           j                            |          }t          |t	          j        d|z             d           d S )Ng}:rg   r   r   )r9   r  r  r   rj   r  )rs   rv  r_   s      r3   r  zTestLaplace.test_isf  sE    Ma  BF1Q3KK<e444444r5   c                     d}d}t           j                            |          }t          ||           t           j                            |           }t          ||d           d S )Nr}  g$ICr  r   )r9   r  r  r   r  r  s        r3   r  zTestLaplace.test_logcdf_logsf  sc    %%%a(($$$##QB''s//////r5   )r   r   r   r   r   r   ry  r   r  r  r  r  r  r  r  s   @r3   r  r  ,  s       [Y66[[---88+? +? 98 76+?Z [0 3? 3? 3? @ @F F F F@ @F(8 8 8  *5 5 5
5 5 5
0 0 0 0 0 0 0r5   r  c                      e Zd Zd Zd Zej                            dg d          d             Zej                            dg d          ej                            dd	g          ej                            d
ddg          ej                            dddg          d                                                 Z	dS )TestLogLaplacec                     t          j        g d          }t          j        g d          }g d}t          t          j                            ||          |d           d S )N)r   r  rU  )r;  r    4&kC)g ?gKH9gXrL0r   r   )rj   r   r   r9   
loglaplacer   )rs   r  r_   r  s       r3   r  zTestLogLaplace.test_sf  sf     H___%%H'''((+++(++Aq113UCCCCCCr5   c                 z    d}g d}g d}t          t          j                            ||          |d           d S )Nr  )r   r]   r  r  r7  )gHhlh#?gH]9@?g"fj#@g11Ags!8wBr  r   )r   r9   r  r  )rs   r  r  r  s       r3   r  zTestLogLaplace.test_isf  sS     +++6 6 6(,,Q22CeDDDDDDr5   r  r    rg   r   r   c                 L   d|dz
           }t          j        d|dz   d          }t          t          j                            ||          t           j                   t          j        t          j        t          j                            ||                              rJ d S )Nr  r    r|   r  )	rj   r  r   r9   r  r  r  r   rG  )rs   r  momr  s       r3   test_moments_statsz!TestLogLaplace.test_moments_stats  s    QUmIc1s7C(( 	(//155rv>>> 6"+e&6&<&<Q&<&L&LMMNNNNNNNr5   r  )r|   rT  r   z
loc, scale)r  g@fix_cTFr>  c                    t           j                            d          }t          j                            |||d|          }d|i}|r||d<   |r||d<   dt          |          z
  }	|	dk    rUd	}
t          j        t          t          f|

          5  t          j        j        |fi | d d d            n# 1 swxY w Y   d S t          t          j        |fi | d S )Nr   ra   rA  r   fcr   r   r   r  r  )rj   r   r   r9   r  r   r  r   r   r  r  r   r   )rs   r  re   rf   r  r>  r   r   r   nfreerE  s              r3   test_fit_analytic_mlez$TestLogLaplace.test_fit_analytic_mle  s=    i##$788##A3e#14 $ 6 6 } 	DJ 	#"DNCIIA::MIj9KKK 3 3 $T22T2223 3 3 3 3 3 3 3 3 3 3 3 3 3 3F%e&6EEEEEEEs   B66B:=B:N)
r   r   r   r  r  r   r   r   r  r  r;   r5   r3   r  r    s       D D DE E E [S,,,//O O 0/O [S///22[\L>::[WtUm44[[4-88F F 98 54 ;: 32F F Fr5   r  c                      e Zd Zej                            dddg          d             Z ej        d          d             Zej                            dg d	          ej                            d
g d          ej                            dg d          ej                            dd  e	ddgd          D                       d                                                 Z
d Zd Zd ZdS )TestPowerlawzx, a, sf)rl  r   g      ?)g     ?g      p?gSQ>c                 f    t          t          j                            ||          |d           d S r   )r   r9   powerlawr   )rs   r_   r1   r   s       r3   r  zTestPowerlaw.test_sf  s1     	))!Q//%@@@@@@r5   r  r  c                 @    t           j                            d          S r5  r  rx   s    r3   r   zTestPowerlaw.rng  r  r5   r   )r]   r|   r   r    rg   r   r  r<  r  r  c                     g | ]}d |v |	S r  r;   r  s     r3   rC   zTestPowerlaw.<listcomp>  r  r5   TFr   r  c                 ,   t           j                            d||||          }t                      }	|r||	d<   |r5t	          j        |                                t          j                   |	d<   |r||	d<   t          t           j        |fi |	ddi d S )Nr"  )r   r1   re   rf   r   r   r   r   r  T)	r9   r  r   r  rj   	nextafterr  r  r   r  s
             r3   r   z(TestPowerlaw.test_fit_MLE_comp_optimizer  s     ~!!siW(1 " E E vv 	#"DJ 	=<

RVG<<DL 	'&DN 	&end 	< 	<d 	< 	<6:	< 	< 	< 	< 	< 	<r5   c           	          d}d}d}t           j                            |||dt          j                            d                    }dt          j        |          dz  i}t          t           j        |fi | d S )	Ng`- @r   grߟA@ra   r   )r1   re   rf   r   r   r   rg   )r9   r  r   rj   r   r   ptpr   )rs   r1   r  rf   r   r   s         r3   test_problem_casezTestPowerlaw.test_problem_case  s     #"~!!A85s/1y/D/DQ/G/G " I I "&,,*+%endCCdCCCCCr5   c                 z   t          t          j                   d}t          t          |          5  t          j                            g ddd           d d d            n# 1 swxY w Y   d}t          t          |          5  t          j                            g dd           d d d            n# 1 swxY w Y   d}t          t          |          5  t          j                            g dd	           d d d            n# 1 swxY w Y   d
}t          t          |          5  t          j                            g dd           d d d            n# 1 swxY w Y   d}t          t          |          5  t          j                            g dd           d d d            d S # 1 swxY w Y   d S )Nz7 Maximum likelihood estimation with 'powerlaw' requiresr  r    rg   r   r   r   rB  rg   r   r    z$Negative or zero `fscale` is outsider  r{  z0`fscale` must be greater than the range of data.)r  r9   r  r`  r#   r   r  ra  s     r3   r  zTestPowerlaw.test_fit_warnings  s   EN+++H<s333 	< 	<Nyyyq;;;	< 	< 	< 	< 	< 	< 	< 	< 	< 	< 	< 	< 	< 	< 	< I<s333 	2 	2Nyyyq111	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 I<s333 	2 	2Nyyyq111	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 6:S111 	5 	5Nyyy444	5 	5 	5 	5 	5 	5 	5 	5 	5 	5 	5 	5 	5 	5 	5 B:S111 	4 	4Nyyy333	4 	4 	4 	4 	4 	4 	4 	4 	4 	4 	4 	4 	4 	4 	4 	4 	4 	4sY   %A##A'*A'$B66B:=B:$D		DD,$EE #E ?$F00F47F4c                     g d}t           j        }t          j        d          5  t	          ||           d d d            d S # 1 swxY w Y   d S )N)r   r    rg   rg   r   r   r   r   r   r   r   r3  r  over)r9   r  rj   r  r   )rs   r   rB   s      r3   test_minimum_data_zero_gh17801z+TestPowerlaw.test_minimum_data_zero_gh17801-  s     433~[h''' 	6 	6)$555	6 	6 	6 	6 	6 	6 	6 	6 	6 	6 	6 	6 	6 	6 	6 	6 	6 	6s   AAAN)r   r   r   r   r   r   r  r  r   r(   r   r  r  r  r;   r5   r3   r  r    s{        [Z1HJK KA AK KA V^*%%%+ + &%+ [[*=*=*=>>[Y


33[[***55[<- -$q)I)I)I - - -. .< <. . 65 43 ?><$D D D4 4 466 6 6 6 6r5   r  c                   >   e Zd Zej                            dg d          d             Zej                            dddg          d             Zej                            dg d          d	             Zej                            dg d
          d             Z	dS )TestPowerLogNormzx, c, s, ref))ra   r  r    ga!R()r  r  r    g;?)r  r  r    gE?)MDr  r    g=c                 h    t          t          j                            |||          |d           d S r   )r   r9   r*  r   rs   r_   r  rO   r  s        r3   r  zTestPowerLogNorm.test_sfG  s4     	*--aA66%HHHHHHr5   zq, c, s, ref)g?r  r    r~   )gR4-r  r    r   c                 h    t          t          j                            |||          |d           d S )Ng|=r   )r   r9   r*  r  )rs   r  r  rO   r  s        r3   r  zTestPowerLogNorm.test_isfQ  s4     	*..q!Q775IIIIIIr5   ))r  r  r    g?)rv  r  r    gv6)rv  rS  r    g+	94|7)r~  rS  r    g?c                 h    t          t          j                            |||          |d           d S )Ngt =r   )r   r9   r*  rq   r  s        r3   r\  zTestPowerLogNorm.test_cdfW  s4     	*..q!Q775IIIIIIr5   ))r  r  r    gS^Ռe9)@xDr  r    g0g;)g\)c=Hr  r    g7H7c                 h    t          t          j                            |||          |d           d S )NgAfc=r   )r   r9   r*  ri   r  s        r3   rR  zTestPowerLogNorm.test_pdfj  s4    
 	*..q!Q775IIIIIIr5   N)
r   r   r   r   r   r   r  r  r\  rR  r;   r5   r3   r  r  6  s<       " [^E E EF F
I IF F
I
 [^ACEF FJ JF FJ [^A A AB B
J JB B
J [^G G GH HJ J	H HJ J Jr5   r  c                       e Zd Zej                            dg d          d             Zej                            dg d          d             Zej                            dg d          d             Zd	S )
TestPowerNormr,  ))r  r    g| <)r  rg   ge()ra   r  g%!@6)rS  r~   gT;-c                 f    t          t          j                            ||          |d           d S r   )r   r9   	powernormr   r0  s       r3   r  zTestPowerNorm.test_sfz  s1     	**1a00#EBBBBBBr5   r  ))r;  r  gĿ)gwJ?ra   g!X)r  r  g缂U)rF  r  gͫ1@)r8  rg   gD@)Jz5r   gU!zYr@c                 f    t          t          j                            ||          |d           d S Nr  r   )r   r9   r  r  r  s       r3   r  zTestPowerNorm.test_isf  s1     	++Aq113UCCCCCCr5   ))ir  g.u	T9)rg   r  ?)r  r  gsK-)r  r  gk8>)r  r  g^u1-c                 f    t          t          j                            ||          |d           d S rT  )r   r9   r  rq   r0  s       r3   r\  zTestPowerNorm.test_cdf  s1     	++Aq113UCCCCCCr5   N)	r   r   r   r   r   r   r  r  r\  r;   r5   r3   r  r  r  s         [[C C CD D
C CD D
C [[= = => >D D> >D [[B B BC CD DC CD D Dr5   r  c                   x    e Zd Zd Zd Zd Zd Zd Zej	        
                    dddg          d	             Zd
S )TestInvGammac                    t          j                    5  t          j        dt                     t          j                            dd          }g d}t          ||           g d}t          j                            |d          }g dt          j        dd	gt          j	        d
dgt          j	        t          j	        dgf}t          ||          D ]\  }}t          ||           	 d d d            d S # 1 swxY w Y   d S )Nrs  g(\O3@r  rw  )g~?gCX&?g*7gS?gBqFq @)r  @gffffff@)r  gEy?gў3oz?g^Yb?g,r?gOD@g2*Z@gFN8@)rw  rx  ry  rz  r9   r"  r   rj   r  r  r  r   )rs   r  rX   r1   r_   r2  s         r3   test_invgamma_inf_gh_1866z&TestInvGamma.test_invgamma_inf_gh_1866  sU    $&& 	* 	*!'>:::>''%'@@D% % %HD(+++A>''!V'<<D888}=k:57H D(++ * *1#Aq))))*	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	*s   CC66C:=C:c                     t          j        dd          }t          j                            |d          }t          j                            |d          }t          ||           d S )Ngr   r    )rj   r  r9   r"  rq   r   r   rc  s       r3   r  zTestInvGamma.test_cdf_ppf  sW    Ka  Nq!$$^1%%2r5   c                     t           j        dk    rt          j        dd          }nt          j        dd          }t          j                            |d          }t          j                            |d          }t          ||d           d S )Nl        rg   ra      r    rT  r   )	sysmaxsizerj   r  r9   r"  r   r  r   rc  s       r3   r  zTestInvGamma.test_sf_isf  s~    ;As##AA Ar""ANa##^1%%2C((((((r5   c                 v    d}d}d}t           j                            ||          }t          ||d           d S )Nr  r!  gM$r  r   )r9   r"  r  r   rs   r_   r1   r  r  s        r3   r  zTestInvGamma.test_logcdf  sC    #&&q!,,%000000r5   c                 v    d}d}d}t           j                            ||          }t          ||d           d S )Nr~   r;  grطr  r   )r9   r"  r  r   rs   r_   r1   r  r  s        r3   r  zTestInvGamma.test_logsf  sC    $$$Q**s//////r5   a, ref)r   g70K6:)r  gVMOuc                 d    t          t          j                            |          |d           d S r   )r   r9   r"  rH   rs   r1   r  s      r3   test_large_entropyzTestInvGamma.test_large_entropy  s/     	..q113UCCCCCCr5   N)r   r   r   r  r  r  r  r  r   r   r   r	  r;   r5   r3   r  r    s        * * *&  ) ) )1 1 10 0 0 [X?:<= =	D 	D= =	D 	D 	Dr5   r  c                   &    e Zd Zd Zd Zd Zd ZdS )TestFc                     t           j        ddgg}|D ]\  }}} |j        |j        g|R  }d |D             }d |D             }t	          ||           d S )N)rg   r    rT  c                 <    g | ]\  }}} |j         |j        g|R  S r;   ri   r1   rA   _f_argsr   s       r3   rC   z(TestF.test_endpoints.<locals>.<listcomp>  4    ===E1vrvbd#U###===r5   c                     g | ]\  }}}|	S r;   r;   rA   r  r  	_correct_s       r3   rC   z(TestF.test_endpoints.<locals>.<listcomp>      >>>!5UI9>>>r5   )r9   fri   r1   r   )rs   r   r  r  _correctanscorrects          r3   r  zTestF.test_endpoints  s    &#&'#' 	' 	'Bx"&&&&&CC=====>>>>>!#w/////r5   c                 ^   t           j                             ddd          \  }}}}t          t          j        |                     t          t          j        |                     t          t          j        |                     t          t          j        |                      d S )Nr        @r  r  )r9   r  r	   rj   rG  r  s        r3   test_f_momentszTestF.test_f_moments  s    W]]2sF];;
1aAAABKNN"#####r5   c                     t          j                    5  t          j        dt                     t          j                            dgdz  g dd           d d d            d S # 1 swxY w Y   d S )Nrs  r  r   )rg   r   r3  r  r  dfndfdr  )rw  rx  ry  rz  r9   r  rx   s    r3   test_moments_warningszTestF.test_moments_warnings  s    $&& 	H 	H!'>:::GMMrd1f,,,MGGG	H 	H 	H 	H 	H 	H 	H 	H 	H 	H 	H 	H 	H 	H 	H 	H 	H 	Hs   AA$$A(+A(c                    t          j        dgdgg          }t          j        ddg          }t          j                            ||d          \  }}}}||dz
  z  gdz  }t	          ||           d|dz  z  ||z   dz
  z  |z  |dz
  dz  z  |dz
  z  }t	          ||           d|z  |z   dz
  t          j        d|dz
  z            z  |d	z
  t          j        |||z   dz
  z            z  z  }	t	          ||	           d|d
|z  dz
  z  ||z   dz
  z  |dz
  |dz
  dz  z  z   z  }
||d	z
  z  |dz
  z  ||z   dz
  z  }|
|z  }t	          ||           d S )Nr   r  rV  r  r  rg   r   r  r3  r      )rj   r   r9   r  r   r  )rs   r   r!  r  r  rO   r\   r  v2s2k2numk2denk2s                r3   test_stats_broadcastzTestF.test_stats_broadcast  s   hbT{##hBx  W]]sV]DD
1aS1Woq 2aZ39q=)C/37Q,>#'J2us{Q"'!S1W+"6"66a273c	A#677792cQsURZ(C#IM:Qw37Q,./ 0sQw37+sSy1}=U]2r5   N)r   r   r   r  r  r"  r*  r;   r5   r3   r  r    sS        0 0 0$ $ $H H H    r5   r  c                      e Zd Zd Zd Zd Zej                            dddg          d             Z	ej                            dg d	          ej                            d
ddgddgddgg dg dgddgddggdgdggg          d                         Z
ej                            d
ddgddgddgg dg dgddgddggdgdggg          d             Zd Zej                            dg d          d             ZdS )TestStudentTc                 h    t          t          j                            ddg          ddg           d S )Nr   r3  g?g?)r   r9   r  r{  rx   s    r3   test_rvgeneric_stdzTestStudentT.test_rvgeneric_std  s/    !%'++q!f"5"5
J7OPPPPPr5   c                 b   t          t          j                            dd          t          j        t          j        t          j        t          j        f           t          t          j                            dd          dt          j        t          j        t          j        f           t          t          j                            dd          dt          j        t          j        t          j        f           t          t          j                            dd          ddt          j        t          j        f           t          t          j                            d	d
          t          j        t          j        f           t          t          j                            dd
          dt          j        f           t          t          j                            dd
          dt          j        f           t          t          j                            dd
          d           d S )Nr    r  )r  r  )\(?r   rg   gGz @g     i@r   rv  gGz@r   g
ףp=
@)r   gq    @)r   r9   r  rj   r  r  rx   s    r3   test_moments_tzTestStudentT.test_moments_t  s   UW]]a]88VRVRVRV4	6 	6 	6UW]]dF];;"&"&"&1	3 	3 	3UW]]a]88"&"&"&1	3 	3 	3UW]]dF];;/2626:	< 	< 	<UW]]a]668HIIIUW]]dD]99C=IIIUW]]a]66bfFFFUW]]dD]99;RSSSSSr5   c                 t    g d}g d}t          t          j                            |          |d           d S )N)r    rg   r  ra   )g?@gTM]?ggY?U?r   r   r   r9   r  rH   )rs   r  rX   s      r3   test_t_entropyzTestStudentT.test_t_entropy+  sE    __; ; ;++XEBBBBBBr5   zv, ref)ra   r3  )r  r  c                 d    t          t          j                            |          |d           d S r#  r4  )rs   r  r  s      r3   test_t_extreme_entropyz#TestStudentT.test_t_extreme_entropy2  s-     	**Ce<<<<<<r5   methname)ri   r   rq   r   r   r  
df_infmaskr   r    )r   r    r   )r    r    r    c                    t           j                            d           t          j        |t                    }t           j                            dd|j                  }t          j        j        |j         }t           j        ||<   t          j
        |dd          }t          j
        ||          dd          }t          j        dd          }t          ||          }t          ||          }	t          ||          }
 ||          }t          ||          |
||                              t          ||           |	||                               d S )	Nr   r   r`   r;  r   r    r  re   rf   rd   )rj   r   r   r   boolr  rB  randnr  r9   r  r  r.  r   )rs   r8  r9  r  r_   t_dist
t_dist_ref	norm_distt_meth
t_meth_ref	norm_methr   s               r3   test_t_inf_dfzTestStudentT.test_t_inf_dfC  s6    		qZ
$777
Yq":+;<<IOZ-.:BAQ///WJ;Qa@@@
J1A...	**Z22
Ix00	fQiiS_ii*&>&>???S*%zz!ZK.'A'ABBBBBr5   c                    t           j                            d           t          j        |t                    }t           j                            dd|j                  }t           j        ||<   t          j	                            |ddd          }t          j
                            ddd	          }t          j	                            ||          ddd          }t          d
          D ]G}t          ||         |         ||                    t          ||         |          ||                    Ht          j	                            |dd          }t          j
                            dd          }t          j	                            ||          dd          }t          ||         |           t          ||          |           d S )Nr   r;  r`   r;  r   r    r  )r  re   rf   r  re   rf   r  r   r<  rd   )rj   r   r   r   r=  r  rB  r  r9   r  r  r  r   rH   )rs   r9  r  r   
res_ex_infres_ex_noinfr  s          r3   test_t_inf_df_stats_entropyz(TestStudentT.test_t_inf_df_stats_entropyY  s   
 		qZ
$777
Yq":+;<<:gmmrq6mBBZ%%!1f%EE
w}}J;Qa,2 % 4 4q 	? 	?AQ
+Z];;;Q,l1o>>>>goo!o44Z''AQ'77
w"j[/qJJS_j111S*%|44444r5   c                     g d}g d}g d}g d}t          t          j                            ||          |d           t          t          j                            ||          |d           d S )N)r    r  r`   r    )r  d~QJr  r    )gZ_2g9g!ǟuIgdg)g73?r   gմFhAW;gm0_?r  r   )r   r9   r  r   ri   )rs   r_   r  
logpdf_refpdf_refs        r3   test_logpdf_pdfzTestStudentT.test_logpdf_pdfp  s     OO###@ @ @
> > >q"--zFFFFAr**G%@@@@@@r5   
x, df, ref))g     Rr?  guSaG)r   r?  r  )     R@r?  gGn`?c                     t           j                            ||          }t          ||d           t           j                            | |          }t          ||d           d S r  )r9   r  r  r   r  )rs   r_   r  r  r  r  s         r3   r  zTestStudentT.test_logcdf_logsf~  s`    
 2&&%0000qb"%%s//////r5   N)r   r   r   r.  r1  r5  r   r   r   r7  rE  rJ  rO  r  r;   r5   r3   r,  r,    s       Q Q QT T TC C C [X68:; ;= =; ;= [Z *< *< *< = =[\QFQFQF-6YY			,B./VaV,<./S1#J,8 9 9C C	9 9= =C  [\QFQFQF-6YY			,B./VaV,<./S1#J,8 9 95 5	9 95&
A 
A 
A [\C C CD D0 0	D D0 0 0r5   r,  c                   V    e Zd Zd Zd Zd Zd Zd Zd Zd Z	d Z
d	 Zd
 Zd Zd ZdS )TestRvDiscretec                 D    t           j                            d           d S r5  r6  rx   s    r3   r7  zTestRvDiscrete.setup_method  r8  r5   c                    g d}g d}d}t          j        d||f          }|                    |          }t          |t          j                  sJ t          ||          D ]>\  }}t          t          ||k              t          |          z  |z
            dk     sJ ?|                                }t	          j
        t          |          t          j                  sJ d S )N)r   r   r    rg   r   r   )r   r~  r  r   r~  r   r   sample)r  valuesr;  rF  )r9   rE   r   rD  rj   rd  r  r   rm  r   r   r   integer)rs   statesprobabilitysamplesr  r_   rO   rv  s           r3   rJ  zTestRvDiscrete.test_rvs  s    $$$4448V[4IJJJEEwE!RZ(((((,, 	> 	>DAqs16{{5>>1A566=====EEGG}T!WWbj1111111r5   c                 f   t          j        g d          }t          j        g d|f          }t	          t          ||                     }|                                }t          ||           t          j        g dg df          }|                                }t          |d           d S )N)rl  r=  r~  r<  rX  )rT  r   r   r   )	rj   r   r9   rE   rm  r   rH   r   r   )rs   r  rv  ro  rp  s        r3   rq  zTestRvDiscrete.test_entropy  s    ***++iii%7888%u--...
IIKK:&&&iii%=>>>IIKKQr5   c                     g d}g d}t          j        ||f          }ddgddgg}t          |                    |          dd	gd
dggd           d S )Nr  r|   r~  rQ  r^  rT  r  r  rg   r|   rQ  r   r~  r  r   )r9   rE   r   rP  )rs   xkpkr  r_   s        r3   rj  zTestRvDiscrete.test_pmf  s    YY__r2h///"X!Wq		sc$*/	1 	1 	1 	1 	1 	1r5   c                     g d}g d}t          j        ||f          g d}g d}t                              |          |d           t          fd|D             |d           d S )	Nr  r`  r^  )r   rT  r  r  r   r  r   r   )r   r|   r|   r|   r   r   r    r    r  r   c                 :    g | ]}                     |          S r;   )rq   )rA   r  r  s     r3   rC   z+TestRvDiscrete.test_cdf.<locals>.<listcomp>  s#    777777r5   )r9   rE   r   rq   )rs   ra  rb  x_valuesrX   r  s        @r3   r\  zTestRvDiscrete.test_cdf  s    YY__r2h///555555x(((???? 	7777h777 u	. 	. 	. 	. 	. 	.r5   c                     g d}g d}t          j        ||f          g d}g d}t                              |          |d           t          fd|D             |d           d S )	Nr  r`  r^  )r]   r|   r  r   r   rT  )r    r    rg   rg   r   r   r  r   c                 :    g | ]}                     |          S r;   )r   )rA   r  r  s     r3   rC   z+TestRvDiscrete.test_ppf.<locals>.<listcomp>  s#    555q555r5   )r9   rE   r   r   )rs   ra  rb  q_valuesrX   r  s        @r3   r  zTestRvDiscrete.test_ppf  s    YY__r2h///000%%%x(((???? 	5555H555 u	. 	. 	. 	. 	. 	.r5   c           	          g dg df}t          j        |          }t          |                    |                    |j        d d                   dz             |j        dd                     d S )N)r    rg   r   r  r  )r]   rQ  r~  r~  r]   r^  r   r-  r    )r9   rE   r   r   rq   ra  )rs   rH  r  s      r3   test_cdf_ppf_nextz TestRvDiscrete.test_cdf_ppf_next  s|    !:!:!:;d+++266"&&ss"4"4t";<<59	& 	& 	& 	& 	&r5   c                 B   t          j        d                              d          }t          j        g dg dg dg          }t	          j        ||f          }t          |                                t          j        |j	        |j
        z            d           d S )NrV  )r   r   )r]   r]   r\  rF  )r]   r]   rF  rF  r^  r  r   )rj   r  r?  r   r9   rE   r   r   rm  ra  rb  rs   ra  rb  r  s       r3   test_multidimensionz"TestRvDiscrete.test_multidimension  s    Yr]]""6**X---------/ 0 0 r2h///		RVBEBEM%:%:GGGGGGr5   c                    g d}ddg}t          t          t          j        fi t	          ||f           g d}t          t          t          j        fi t	          ||f           g d}g d}t          t          t          j        fi t	          ||f           g d}g d}t          t          t          j        fi t	          ||f           ddg}ddg}t          t          t          j        fi t	          ||f           d S )Nr  r|   r^  )r|   333333?gffffff濩r    rg   r   r   r   )r~  r~  r~  r~  r%  r    )r`  r  r9   rE   r  rs   ra  rb  s      r3   test_bad_inputzTestRvDiscrete.test_bad_input  s/   YY3Zj%"3MMtB87L7L7LMMMYYj%"3MMtB87L7L7LMMMYYj%"3MMtB87L7L7LMMM__'''j%"3MMtB87L7L7LMMMV3Zj%"3MMtB87L7L7LMMMMMr5   c                    t          j        d                              d          t          j        dd          }}t	          t
          t          j        fi t          ||f           t          j        d                              d          t          j        dd          }}t	          t
          t          j        fi t          ||f           t          j        d                              d          t          j        dd          }}t          t          j        ||f          
                    d          d           d S )	Nr   rg   rg   rg   r   gUUUUUU?r^  r3  r   rg   r   )rj   r  r?  r#  r`  r  r9   rE   r  r   rP  rq  s      r3   test_shape_rv_samplez#TestRvDiscrete.test_shape_rv_sample  s    1%%f--rwvs/C/CBj%"3MMtB87L7L7LMMM 1%%f--rwvs/C/CBj%"3MMtB87L7L7LMMM 1%%f--rwvs/C/CBU&r2h777;;A>>DDDDDr5   c                     g d}g d}t          j        ||f          }t          |                                t	          j        |j        |j        z            d           d S )N)r    rg   r   r3  r  r  )r]   rQ  rQ  rQ  rQ  r]   r^  r  r   )r9   rE   r   r   rj   rm  ra  rb  rl  s       r3   test_expect1zTestRvDiscrete.test_expect1  sd       +++r2h///		RVBEBEM%:%:GGGGGGr5   c           
         g d}g d}t          j        ||f          }t          |                                |                                d           t          |                                t          d t          ||          D                       d           t          |                    d           t          d t          ||          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:?r   gF%uk?g:vz?r   r   g.nr?g|гY?r   r   r   g @6*? 4U0*C?g@+?gC6z?g nr?g ?gOjM?g n?g1w-!?g:pΈ?gףp=
?g _Le?r   g^)p?g~jtx?gJ4q?gGzt?g J4q?g Mb@?g@Psׂ?g@^)ˀ?r   r   g N@s?g 1w-!_?r|  gൄ|г?r   gH}}?r   g
F%u?r   r   g ^)ˀ?g v?r   r^  r  r   c              3   &   K   | ]\  }}||z  V  d S rv   r;   rA   r  ws      r3   	<genexpr>z.TestRvDiscrete.test_expect2.<locals>.<genexpr>'  s*      99daAE999999r5   c                     | dz  S Nrg   r;   r   s    r3   rP   z-TestRvDiscrete.test_expect2.<locals>.<lambda>*  
    AqD r5   c              3   ,   K   | ]\  }}|d z  |z  V  dS )rg   Nr;   r~  s      r3   r  z.TestRvDiscrete.test_expect2.<locals>.<genexpr>+  s.      <<AAqD1H<<<<<<r5   )r9   rE   r   r   r   rm  r  )rs   r2  pyr  s       r3   test_expect2zTestRvDiscrete.test_expect2  s
   = = =? ? ?  q"g... 			RWWYYU;;;;		99c!Rjj99999	G 	G 	G 	G 			..11<<Q<<<<<5	J 	J 	J 	J 	J 	Jr5   N)r   r   r   r7  rJ  rq  rj  r\  r  rj  rm  rr  rw  ry  r  r;   r5   r3   rT  rT    s          2 2 2
 
 
	1 	1 	1. . .. . .& & &H H HN N N(E E EH H H#J #J #J #J #Jr5   rT  c                       e Zd Zd Zd ZdS )TestSkewCauchyc                    t          j        ddd          }t          t          j                            |d          t          j                            |                     t          t          j                            |d          t          j                            |                     t          t          j                            |d          t          j                            |                     d S Nr  r   ra   r   ru  )	rj   rY  r   r9   
skewcauchyri   r  rq   r   r  s     r3   test_cauchyzTestSkewCauchy.test_cauchy/  s    KAs##!%"2"6"6qA"6">">"',"2"21"5"5	7 	7 	7!%"2"6"6qA"6">">"',"2"21"5"5	7 	7 	7!%"2"6"6qA"6">">"',"2"21"5"5	7 	7 	7 	7 	7r5   c                    t           j                            d           t           j                            d          dz  dz
  }t           j                            d          dz  dz
  }g d}g d}t	          t
          j                            ||          |           t	          t
          j                            ||          |           t	          t
          j        	                    ||          |           d S )Nr   r`   rg   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?)
rj   r   r   r  r   r9   r  ri   rq   r   )rs   r1   r_   ri   rq   s        r3   test_skewcauchy_Rz TestSkewCauchy.test_skewcauchy_R8  s    & 		qINN2"Q&INN2#a'% % %$ $ $ 	(,,Q22C888(,,Q22C888(,,S!44a88888r5   N)r   r   r   r  r  r;   r5   r3   r  r  .  s2        7 7 7 9  9  9  9  9r5   r  c                       e Zd Zd Zej        d             Zej                            dg d          d             Z	dS )TestJFSkewTc                 *   dx}}|dz  }g d}g d}t          j        ||          }t          j        |          }t          |                    |          |                    |                     t          |                    |          |                    |                     t          |                    |          |                    |                     t          |                     d          |                     d                     d S )Nr   rg   )r  r   rT  r   )r   r]   rl  r   r   rT  r  )r9   	jf_skew_tr  r   ri   rq   r   )rs   r1   r2   r  r_   r  jfr  s           r3   test_compare_tzTestJFSkewT.test_compare_t\  s     	AU!!!---_Q""GBKKq		15588,,,q		15588,,,q		15588,,,((!''&//:::::r5   c                     t          j        t          t                    j        dz            }t           j                            |d          S )a  Sample data points computed using the `ST5` distribution from the
        GAMLSS package in R. The pdf has been calculated for (a,b)=(2,3),
        (a,b)=(8,4), and (a,b)=(12,13) for x in `np.linspace(-10, 10, 41)`.

        N.B. the `ST5` distribution in R uses an alternative parameterization
        in terms of nu and tau, where:
            - nu = (a - b) / (a * b * (a + b)) ** 0.5
            - tau = 2 / (a + b)
        z"data/jf_skew_t_gamlss_pdf_data.npyz	x,pdf,a,bnames)rj   loadr   __file__parentrec
fromarraysr|  s     r3   gamlss_pdf_datazTestJFSkewT.gamlss_pdf_datal  sD     wNN!$HH
 
 v  [ 999r5   za,b)ru  )r  r   )rV     c                     ||d         |k    |d         |k    z           }|d         |d         }}t          |t          j        ||                              |          d           dS )zCompare the pdf with a table of reference values. The table of
        reference values was produced using R, where the Jones and Faddy skew
        t distribution is available in the GAMLSS package as `ST5`.
        r1   r2   r_   ri   r  r   N)r   r9   r  ri   )rs   r  r1   r2   r   r_   ri   s          r3   test_compare_with_gamlss_rz&TestJFSkewT.test_compare_with_gamlss_r|  sr     S!Q&?3+?1+DE
 cDK3U_Q2266q99FFFFFFr5   N)
r   r   r   r  r   r  r  r   r   r  r;   r5   r3   r  r  [  s{        ; ; ;  ^: : ^: [U$>$>$>??	G 	G @?	G 	G 	Gr5   r  rg   r   g@r   gףp=
1@r?  gRZ@i   gS㥛Ԋ@r]   r^   g&@iQ i'  i0ybi l   HO1Z i3r  ir`   ira   ikrS  i1r   i  i4i N  i(  lEI. r,  i c                       e Zd Zd Zd Zd Zd Zd Zd Zd Z	e
j                            de          d	             Zd
 Zd Zd ZdS )TestSkewNormc                 .    t          d          | _        d S r5  )r   r   rx   s    r3   r7  zTestSkewNorm.setup_method  s    %d++r5   c                     t          j        ddd          }t          t          j                            |d          t          j                            |                     d S r  )rj   rY  r   r9   skewnormri   r  r  s     r3   test_normalzTestSkewNorm.test_normal  sY    KAs##!%."4"4Q!"4"<"<"'*.."3"3	5 	5 	5 	5 	5r5   c                     d}t           j                            d|| j                  }t	          ||j                   t           j                            d|| j                  }t	          ||j                   d S )N)r   r   r   r   )r1   r   r   r  )r9   r  r   r   r   rB  )rs   rB  r_   s      r3   rJ  zTestSkewNorm.test_rvs  sm    NEIIUAG$$$N%dhGGUAG$$$$$r5   c                    t           j                            dt          d          dd| j                  }t          j        |          t          j        |          t          j        |          t          j	        |          g}t           j                             dddd          }t          ||d           t           j                            d	t          d          dd| j                  }t          j        |          t          j        |          t          j        |          t          j	        |          g}t           j                             d	ddd          }t          ||d           d S )
Nr   r  r   rg   )r1   r   re   rf   r   r  )r1   re   rf   r  rW  r  )r9   r  r   rc  r   rj   r   rA  rB  kurtosisr   )rs   r  rX   rD  s       r3   r*  zTestSkewNorm.test_moments  s,   NSq,0H  6 6GAJJq		5:a==%.:K:KL>''!!V'LL!(Ha@@@@N#c((,0H  6 6GAJJq		5:a==%.:K:KL>''"!1f'MM!(Ha@@@@@@r5   c                     g dg dg dg dg dg dg dg}|D ]8\  }}}t           j                            ||          }t          ||d	           9d S )
N)r}  r   gEAV)r}  r   gV|@$a)r}  r   g0rY)r}  r|   Zk)r  r   r  )r   r  ļ)rg   g    cr  r-  r   )r9   r  r   r   )rs   
logpdfvalsr_   r1   	logpdfvalr  s         r3   test_pdf_large_xzTestSkewNorm.test_pdf_large_x  s     211333000222444555555

  * 	8 	8OAq)>((A..DD)$77777	8 	8r5   c                     t           j                            g dd          }t          |t	          j        d          d           t           j                            dd          }t          |dd           d S )	N)r`   r  r:  r   r   r  r   r  r  rT  )r9   r  rq   r   rj   onesr  s     r3   test_cdf_large_xzTestSkewNorm.test_cdf_large_x  sn     N|||R00271::E2222Nr3''3U++++++r5   c                    g dg dg dg dg dg}|D ]l\  }}}t           j                            ||          }t          ||d           t           j                            | |           }t          ||d           md S )N)r  r    gzbLe9)r  rg   gn'/2;)r   r   g::)r  r   g|<)rm  r  r4  r-  r   )r9   r  rq   r   r   )rs   cdfvalsr_   r1   cdfvalrv  s         r3   test_cdf_sf_small_valuesz%TestSkewNorm.test_cdf_sf_small_values  s     .--+++,,,...///
 $ 	2 	2LAq&""1a((AAvD1111!!1"qb))AAvD11111	2 	2r5   z
a, momentsc                     t          |d          D ]7\  }}t          j                            ||          }t	          ||d           8d S )Nr    )startr  r   )rj  r9   r  r  r   )rs   r1   r  orderrX   r  s         r3   test_noncentral_momentsz$TestSkewNorm.test_noncentral_moments  s[    (::: 	7 	7OE8.''q11CC66666	7 	7r5   c           	      t   t           j                            d          }d\  }}}t          j        |||          }|                    d|          }t          j                            |dd          \  }}}	t          j                            |dd          \  }
}}||cxk    rdk    sn J ||
k    sJ t          j                            |ddd	
          \  }}}|dk    sJ t          j        |||          }|                    d          }t          j        |          t          j        |          f}t          ||           t          j
                            dd|          }t          j                            |          }t          j        t          j        |                    sJ t          j                            |d	          \  }}}t          j        |          sJ t          j        |          t          j        |          }}t          |||t          j        dt           j        z            z  z              t          ||dz  ddt           j        z  z
  z             t          j                            |d          \  }}}t          j                            | d          \  }}}t          |||g| | |g           t          j                            |d	          \  }} }!t          j                            | d	          \  }"}#}$t          |"|#|$g| |  |!g           d S )Nl   #ke )r   r;  r|   ra   r   rX  r   r   gmmr   r  msr  r    r  rg   mle)rj   r   r   r9   r  r   r   r   rB  r   r  r   rG  isinfrA  r  rk   )%rs   r   r1   re   rf   rB   r   a2loc2scale2a3loc3scale3a4loc4scale4dist4r   r  r   a5loc5scale5r  r  a6ploc6pscale6pa6mloc6mscale6ma7ploc7pscale7pa7mloc7mscale7ms%                                        r3   ry  zTestSkewNorm.test_fit  s   i##$788$3~ae,,hhCch22 !>--c4a-@@D& >--c4a-@@D&t    q      Rxxxx !>--c1Qt-LLD&{{{{r400kk$k''gcllEJsOO+S!!! |#>> n  &&vbk#&&''''' !>--d4-@@D&x|| wt}}bfTll14&271RU7+;+;";;<<<619AI6777 $n00U0CCUG#n00#e0DDUGeW-ufg/FGGG#n00T0BBUG#n00#d0CCUGeW-ufg/FGGGGGr5   c                 F  	 t          j        g dddgz  z   dgz             }t          j                            |          }t          j                            ||          }t          j                            |d          }t          j                            ||          }||dz
  k     sJ t           j                            d          	d	d
dd}	fd}t          j        t          j        |||          }t           j        	                    ||j
        d           d S )N)r  r   r;  rV  r    r   T)superfitr|   l   y
ND )r  r   )rm  r`   )r  r`   r1   re   rf   c                 (    t          | |          S )N)r   )r'   )funr   r   s     r3   	optimizerz0TestSkewNorm.test_fit_gh19332.<locals>.optimizerH  s    )#v3????r5   )r  r  r   )rj   r   r9   r  r   r   r   r   testingr   params)
rs   r_   r  r   params_superr  r   r  
fit_resultr   s
            @r3   test_fit_gh19332zTestSkewNorm.test_fit_gh193323  s$   
 H***R1#X5;<<##A&&n!!&!,,
 ~))!d);;n!!,22S3Y i##$788y;GG	@ 	@ 	@ 	@ 	@ Yu~q&INNN


""6:+<4"HHHHHr5   c                 f    t          t          j                            dd          dd           d S )Nr~   r  gi}#%?r   r   )r   r9   r  r   rx   s    r3   r  zTestSkewNorm.test_ppfN  s2     	**4557HuUUUUUUr5   N)r   r   r   r7  r  rJ  r*  r  r  r  r   r   r   _skewnorm_noncentral_momentsr  ry  r  r  r;   r5   r3   r  r    s        , , ,5 5 5% % %A A A8 8 8 , , ,2 2 2" [\+GHH7 7 IH7
.H .H .H`I I I6V V V V Vr5   r  c                   &    e Zd Zd Zd Zd Zd ZdS )	TestExponc                 `    t          t          j                            d          d           d S r  )r   r9   rg  ri   rx   s    r3   	test_zerozTestExpon.test_zeroV  s&    U[__Q''+++++r5   c                     t          t          j                            d          d           t          t          j                            t          j                            d                    d           d S )Nr8  r}  )r   r9   rg  rq   r  r   rx   s    r3   	test_tailzTestExpon.test_tailY  sR    U[__U++U333U[__U[^^B%7%788"=====r5   c                     t          j        dddddt           j        g          }t          t          t
          j        j        |           d S No_?46@yX5ͻ@j+?UX@)rj   r   r  r`  r  r9   rg  r   r  s     r3   test_nan_raises_errorzTestExpon.test_nan_raises_error]  <    HfffffbfEFFj%+/155555r5   c                     t          j        dddddt           j        g          }t          t          t
          j        j        |           d S r  )rj   r   r  r`  r  r9   rg  r   r  s     r3   test_inf_raises_errorzTestExpon.test_inf_raises_errorb  r  r5   N)r   r   r   r  r  r  r  r;   r5   r3   r  r  U  sP        , , ,> > >6 6 6
6 6 6 6 6r5   r  c                   l    e Zd Zd Zd Zd Zej                            dddg          d             Z	dS )	TestNormc                     t          j        dddddt           j        g          }t          t          t
          j        j        |           d S r  )rj   r   r  r`  r  r9   r  r   r  s     r3   r  zTestNorm.test_nan_raises_errori  <    HfffffbfEFFj%*.!44444r5   c                     t          j        dddddt           j        g          }t          t          t
          j        j        |           d S r  )rj   r   r  r`  r  r9   r  r   r  s     r3   r  zTestNorm.test_inf_raises_errorn  r  r5   c                 \    g d}t          t          t          j        j        |d           d S )Nr  shrimp)plate)r`  r  r9   r  r   r  s     r3   test_bad_keyword_argzTestNorm.test_bad_keyword_args  s,    IIi(CCCCCCr5   re   r   r    c                     d}t           j                            d|z   d|z   |          }t          ||d           t           j                            d|z    d|z    |           }t          ||d           d S )Ngg	{F.:r  rV  re   r   r   )r9   r  
_delta_cdfr   )rs   re   rX   rr  s       r3   test_delta_cdfzTestNorm.test_delta_cdfw  s     )
%%bfbf#%>>xe4444
%%3i2c6%EExe444444r5   N)
r   r   r   r  r  r  r   r   r   r  r;   r5   r3   r  r  h  sv        5 5 5
5 5 5
D D D [UQF++
5 
5 ,+
5 
5 
5r5   r  c                       e Zd ZdZd Zd ZdS )TestUniformgh-10300c                     t          j        dddddt           j        g          }t          t          t
          j        j        |           d S r  )rj   r   r  r`  r  r9   r  r   r  s     r3   r  z!TestUniform.test_nan_raises_error  =    HfffffbfEFFj%-"3Q77777r5   c                     t          j        dddddt           j        g          }t          t          t
          j        j        |           d S r  )rj   r   r  r`  r  r9   r  r   r  s     r3   r  z!TestUniform.test_inf_raises_error  r  r5   N)r   r   r   __doc__r  r  r;   r5   r3   r  r    s7        N8 8 8
8 8 8 8 8r5   r  c            
       D   e Zd Zd Zd Zd Zd Zej        	                    dg d          d             Z
ej        	                    dg d	g d
g dg dg dg          d             Zej        	                    dg dg dg dg dg dg dg          d             ZdS )TestExponNormc                 l   d }d\  }}}d||z  z  }t           j                             |||d          }t          | ||||                     d\  }}}d||z  z  }t           j                             |||d          }t          | ||||                     d\  }}}d||z  z  }t           j                             |||d          }t          | ||||                     d\  }}}d||z  z  }t           j                             |||d          }t          | ||||                     d S )	Nc                     dd| |z  dz  z  z   }d| |z  dz  z  |dz  z  }dd| |z  dz  z   dz  z  }|d| z  z   ||z  d| | z  z  z   ||gS )NrT  r    rg   r   rX        @r   r;   )lamsigr^  opK2exp_skewexp_kurts         r3   get_momsz,TestExponNorm.test_moments.<locals>.get_moms  sz     c#g\))DC#I>)D4L8Ha39q.0B77H3JC#s3w- 78LLr5   )r   r    r    rT  r  rG  )r  rg   r]   )r   r   r    )r  r  r;  )r9   	exponnormr   )rs   r  r^  r  r  Kstss          r3   r*  zTestExponNorm.test_moments  sp   	M 	M 	M C39o##A2S&#IIC#sB!7!7888!C39o##A2S&#IIC#sB!7!7888C39o##A2S&#IIC#sB!7!7888"C39o##A2S&#IIC#sB!7!788888r5   c                     t          j        dddddt           j        g          }t          t          t
          j        j        |dd           d S 	Nr  r  r  r  r  r   r    rB  )rj   r   r  r`  r  r9   r  r   r  s     r3   r  z#TestExponNorm.test_nan_raises_error  D    HfffffbfEFFj%/"5qqKKKKKKr5   c                     t          j        dddddt           j        g          }t          t          t
          j        j        |dd           d S r  )rj   r   r  r`  r  r9   r  r   r  s     r3   r  z#TestExponNorm.test_inf_raises_error  r  r5   c                 v   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    r     r~   )r   r9   r  ri   rx   s    r3   test_extremes_xzTestExponNorm.test_extremes_x  s    EO//a88#>>>EO//a88#>>>EO//d;;SAAAEO//d;;SAAAAAr5   zx, K, expected))r  r~   g6
N-)r    r~   g՜n+H?)r   r~   gi?)r  r~   gbJI-)r`   r    gI8?)r`   r  gQ3|-0?c                 f    t          t          j                            ||          |d           d S r  )r   r9   r  ri   )rs   r_   r  rX   s       r3   test_std_pdfzTestExponNorm.test_std_pdf  s1     	++Aq118%HHHHHHr5   zx, K, scale, expected)r   r~   r    gVAҤ?)r  {Gzt?r    g'^>)r  r~   ra   r   )r  r~   r   gbr ;)r   r  r    gVMe?c                     t           j                            |||          }|dk    r
|dk    sJ d S t          ||d           d S )Nr  r   r   r   )r9   r  rq   r   rs   r_   r  rf   rX   rv  s         r3   test_cdf_small_KzTestExponNorm.test_cdf_small_K  sU     O1E22s??888888Axe444444r5   )r`   r~   r    ghG}$;)rg   r%  r    g]fJ?)r   r%  r|   g31"g#;)r`   r%  r|   gf
+-)r  r%  r|   r   )r  r  r    ga9S?c                     t           j                            |||          }|dk    r
|dk    sJ d S t          ||d           d S )Nr  r   -a=r   )r9   r  r   r   r'  s         r3   test_sf_small_KzTestExponNorm.test_sf_small_K  sU     Oq!511s??888888Axe444444r5   N)r   r   r   r*  r  r  r"  r   r   r   r$  r(  r+  r;   r5   r3   r  r    s       9 9 96L L L
L L L
B B B$ [-B B BC CI IC CI [4>>>DDD444GGG???	AB B5 5B B5& [4BBB@@@DDDFFF333@@@BC C5 5C C5 5 5r5   r  c                       e Zd Zd Zd Zej                            dg d          d             Zej                            dg d          d             Z	dS )	TestGenExponc                     ddl m} t          j                            t          j        ddd          ddd          }t           ||d          dd           d S )	Nr   )simpsonr`   r~   r|   r   )dxr    )scipy.integrater/  r9   genexponri   rj   r  r   )rs   r/  rv  s      r3   test_pdf_unity_areaz TestGenExpon.test_pdf_unity_area	  sd    ++++++NryB55sCEEGGA$///A66666r5   c                     t           j                            t          j        ddd          ddd          }t          j        d|k    |dk    z            sJ d S )Nr   r`   r~   r|   r   r    )r9   r2  rq   rj   r  r   )rs   rq   s     r3   test_cdf_boundszTestGenExpon.test_cdf_bounds  sT    n  1b$!7!7c3GGvqCxC1H-.......r5   zx, p, a, b, c))r?  gM <r    rg   r  )rl  g%DYY?r|   rg   r   )rl  gS4z?      #@rg   r|   )r~   ga^5?r  rl  r|   )r  gs%)?r  rl  r|   )ry  gdm?rl  r   r|   c                     t           j                            ||||          }t          ||d           t           j                            ||||          }t          ||d           d S r  )r9   r2  r   r   r  )rs   r_   rv  r1   r2   r  r   r  s           r3   r  zTestGenExpon.test_sf_isf  sj     ^q!Q**AE****n  Aq!,,QU++++++r5   ))rl  g;\i?r|   rg   r   )rl  gu0?r6  rg   r|   )r~   g-%I?r  rl  r|   )r  g]d?r  rl  r|   )ry  g泩'?rl  r   r|   c                     t           j                            ||||          }t          ||d           t           j                            ||||          }t          ||d           d S r  )r9   r2  rq   r   r   )rs   r_   rv  r1   r2   r  rq   r   s           r3   r  zTestGenExpon.test_cdf_ppf-  sj     n  Aq!,,QU++++n  Aq!,,QU++++++r5   N)
r   r   r   r3  r5  r   r   r   r  r  r;   r5   r3   r-  r-    s        7 7 7/ / /  [_I I IJ J, ,J J, [_J J JK K, ,K K, , ,r5   r-  c                       e Zd Zd ZdS )TestTruncexponc                     ddg}ddg}ddg}t          t          j                            ||          |d           t          t          j                            ||          |d	           d S )
Nr  ra   g_93@gBX@g,M<gǃ@֫5gLa㧝=r   r  )r   r9   
truncexponr   r  )rs   r2   r_   r  s       r3   r  zTestTruncexpon.test_sf_isf<  sx     I	"%'=>(++Aq113WEEEE(,,S!44aeDDDDDDr5   N)r   r   r   r  r;   r5   r3   r:  r:  :  s(        E E E E Er5   r:  c                       e Zd Zd ZdS )TestExponpowc                     t          t          j                            dd          d           t          t          j                            t          j                            dd          d          d           d S )Nr  r   r  r   r   )r   r9   exponpowrq   r  r   rx   s    r3   r  zTestExponpow.test_tailG  si    EN..ub995AAAEN..u~/@/@B/G/GLL	 	 	 	 	r5   N)r   r   r   r  r;   r5   r3   r>  r>  F  s#            r5   r>  c                        e Zd Zd Zd Zd ZdS )TestSkellamc                     t          j        dd          }d\  }}t          j        g d          }t          t          j                            |||          |d           d S )Nrm  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~?rW  )rj   r  r   r   r9   skellamrP  )rs   r\   mu1mu2skpmfRs        r3   rj  zTestSkellam.test_pmfN  sm    Ic2S- - -. . 	EM--ac::FBOOOOOOr5   c                     t          j        dd          }d\  }}t          j        g d          }t          t          j                            |||          |d           d S )Nrm  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   rW  )rj   r  r   r   r9   rD  rq   )rs   r\   rE  rF  skcdfRs        r3   r\  zTestSkellam.test_cdfc  sm    Ic2S- - -. . 	EM--ac::FANNNNNNr5   c                     d\  }}}t          t          j                            |||          dd           t          t          j                            |||          dd           d S )N)r   r    gs1Cr   r  r   r    )r   r9   rD  rP  rq   )rs   r_   rE  rF  s       r3   test_extreme_mu2zTestSkellam.test_extreme_mu2x  se    .3))!S#66FFFF))!S#66FFFFFFr5   N)r   r   r   rj  r\  rK  r;   r5   r3   rB  rB  M  sL        P P P*O O O*G G G G Gr5   rB  c                      e Zd Zd Zd Z ej        d          d             Zej        	                    dddg          ej        	                    d	g d
          ej        	                    dg d          ej        	                    dd  e
dd          D                        ej        d          d                                                             Zd ZdS )TestLognormc                     t          j                    5  t          j        dt                     t          j                            g dd          }t          |g d           d d d            d S # 1 swxY w Y   d S )Nrs  r   r|   r    r    )r   g~r?g e3E?)rw  rx  ry  rz  r9   r(  ri   r   rs   ri   s     r3   rR  zTestLognorm.test_pdf  s     $&& 	J 	J!'>:::-##KKK33C%c+H+H+HIII	J 	J 	J 	J 	J 	J 	J 	J 	J 	J 	J 	J 	J 	J 	J 	J 	J 	Js   AA00A47A4c           	         d\  }}}t          t          j                            ||z
  |          t          j                            t          j        ||z
            |z                       t          t          j                            ||z
  |          t          j                            t          j        ||z
            |z                       d S )N)g(\5i@   gOn?rN   )r   r9   r(  r   r  rj   r  r  )rs   rs  r^  sigmas       r3   r  zTestLognorm.test_logcdf  s    *B((B%(88
bfRUmmE&9::	< 	< 	<++BrEU+;;
((2u)<==	? 	? 	? 	? 	?r5   r  r  c                 @    t           j                            d          S r5  r  rx   s    r3   r   zTestLognorm.rng  r  r5   r   r]   rg   r   )r   r   rg   r<  )rQ  r    r   r  c                     g | ]}d |v |	S r  r;   )rA   es     r3   rC   zTestLognorm.<listcomp>  r  r5   )FTr   r  r  r  c                     t           j                            d||||          }i }	|r||	d<   |r||	d<   |r||	d<   t          t           j        |fi |	ddi d S )Nra   )r   rO   rf   re   r   r   r   r   r  T)r9   r(  r   r   r  s
             r3   r   z'TestLognorm.test_fit_MLE_comp_optimizer  s     }  cYi%,3 ! @ @  	#"DJ 	#"DL 	'&DN 	&emT 	< 	<T 	< 	<6:	< 	< 	< 	< 	< 	<r5   c                 z    d}g d}g d}t          t          j                            ||          |d           d S )NgI+?)r]   绽|=gҶOɃ;gk3;9"7)g?y+@g~K8ax@gƶ@g9]Ar  r   )r   r9   r(  r  )rs   rO   r  r  s       r3   r  zTestLognorm.test_isf  sR    
 &&&# # #))!Q//5AAAAAAr5   N)r   r   r   rR  r  r   r  r   r   r   r(   rj   r  r   r  r;   r5   r3   rM  rM    sP       J J J? ? ? V^*%%%+ + &%+ [[2q'22[Y


33[[***55[<- -q)I)I)I - - -. . R["""< < #". . 65 43 32<$	B 	B 	B 	B 	Br5   rM  c                   >   e Zd Zd Zd Zd Zd Zej        	                    e
d          d             Zej        	                    e
d          d             Zej        	                    e
d          d	             Zd
 Zej                            ed          d             Zej                            dej        j        ej        j        g          ej                            dddg          d                         Zej                            dddg          d             Zej                            dddg          d             Zej                            dg d          d             Zej                            dg d          d             ZdS ) TestBetac                     t           j                            ddd          }t          |d           t           j                            ddd          }t          |t          j                   d S )Nr   r    r|   g;B.)r9   r  r   r   rj   r  rs   r   s     r3   r  zTestBeta.test_logpdf  s\    ""1a--FN333""1c1--FBF+++++r5   c                 `   d\  }}t          j        g d          }t          j        ||          }t	          |                    |                                          d           t	          |                    |          t          j        |                    |                               d S )Ni  i  rQ  r|   r  gƒ)	rj   r   r9   r  r   r   rm  ri   r   rs   r  r  r_   r2   s        r3   test_logpdf_ticket_1866z TestBeta.test_logpdf_ticket_1866  s    tH___%%Jud##))+=>>>a"&!"5"566666r5   c                 `    g d}t          t          t          j        j        |ddd           d S )Nr]   r|   r  r   r    r   )r   r   r  )r`  r  r9   r  r   r  s     r3   test_fit_bad_keyword_argsz"TestBeta.test_fit_bad_keyword_args  s=    OOi1$	& 	& 	& 	& 	& 	&r5   c                 ^    g d}t          t          t          j        j        |dd           d S )Nrd  r|   )fafix_a)r`  r  r9   r  r   r  s     r3   #test_fit_duplicated_fixed_parameterz,TestBeta.test_fit_duplicated_fixed_parameter  s0     OOj%*.!3GGGGGGr5   zOverflow, see gh-14901reasonc                 p    d\  }}}t          t          j                            |||          d           d S )N)g?rQ  g   0xAgx)>)r   r9   r  r   )rs   rv  r1   r2   s       r3   test_issue_12635zTestBeta.test_issue_12635  s8     71a
q!Q//1FGGGGGr5   c                 >   t          j        g d          }t          j        g d          }d}t          j                            ||dz   d|z
            }t          ||           t          j                            ||dz   d|z
            }t          ||           d S )N)g@3@?g^?gNC?)r`   ra   r   r  r    r   )rj   r   r9   r  r  r   r   )rs   inv_R
count_listrv  invr   s         r3   test_issue_12794zTestBeta.test_issue_12794  s      1 1 1 2 2 Xooo..
jnnQ
Q0CDDU###jmmCa*1DEEQr5   c                     d}t          j        dd          }d}d|z
  |dz   ||z
  }}}t          j                            |||          }t          j                            |||          }t          |d|z
             d S )Nh㈵>r    r  r   )rj   r  r9   r  r   rq   r   )	rs   alpha_2count_nobsr  r1   r2   rq  r   s	            r3   test_issue_12796zTestBeta.test_issue_12796  s     1b!!g+vz4&=a1jnnQ1%%jnnS!Q''Q[)))))r5   c                    d\  }}t          t          j                            d||          t          j                   d\  }}t          t          j                            d||          t          j                   d\  }}t          t          j                            d||          d           t          t          j                            d||          d           d\  }}t          t          j                            d||          d           t          t          j                            d	||          d           d S )
Nr  r    )rQ  r   r   )r    r   r   +ph  )r   r    rT  )r   r9   r  ri   rj   r  )rs   r1   r2   s      r3   r  zTestBeta.test_endpoints  s    1UZ^^Aq!,,bf555 1UZ^^Aq!,,bf555 1UZ^^Aq!,,a000UZ^^FAq111555 1UZ^^Aq!,,a000UZ^^Ha33Q77777r5   zDoes not convert boost warningc                     d\  }}}t          j        t                    5  t          j                            |||           d d d            d S # 1 swxY w Y   d S )N)gףp=
?g   vH7B  @0B)r   warnsrz  r9   r  r   )rs   r  r1   r2   s       r3   test_boost_eval_issue_14606z$TestBeta.test_boost_eval_issue_14606  s    '1a\.)) 	$ 	$JNN1a###	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$s   "AAAr  a, b)rz        )@)r  rz  c                 H    d}	  ||||           d S # t           $ r Y d S w xY w)Nr   )OverflowError)rs   r  r1   r2   rv  s        r3    test_beta_ppf_with_subnormal_a_bz)TestBeta.test_beta_ppf_with_subnormal_a_b  sG    
 
	F1aOOOOO 	 	 	 DD	s    
!!zx, a, b, ref)r  r  r  g>׵NlIq)&.!>r  r  g'rOc                 l    t           j                            |||          }t          ||d           d S r  )r9   r  r  r   rs   r_   r1   r2   r  r  s         r3   r  zTestBeta.test_logcdf3  s8     ""1a++%000000r5   )r  r  r  gp}^U)rv  r  r  g
1*c                 j    t           j                            |||          }t          ||d           d S )Nr  )r9   r  r  r   rs   r_   r1   r2   r  r  s         r3   r  zTestBeta.test_logsf;  s5     
  Aq))sE*****r5   r  ))r|   r|   gο)r  r    g )r    r  g&׾k )r   r   g*)Mc                 p    t          t          j        ||                                          |           d S rv   r   r9   r  rH   r  s       r3   rq  zTestBeta.test_entropyL  s2     	
1a((0022C88888r5   za, b, ref, tol))r    r`   gCpr  )r`   r  g8EOr   )g    NAg    NAg(4Fr  )g    SAg   SAg8TAUr-  )r  g   _Bg[VVuD&r  )rL  rL  gR#Lr   )rg   r  g?r5r  )rg   r  <Fr  )rg   rL  guRTc\r  )r   r  gL4r  )r   r  6Er  )r   rL  g.fq>\r  )r`   r  gnQc}4r  )r`   r  g{C Er  )r`   rL  %\r  )r  rg   r  r  )r  r   r  r  )rL  r`   r  r  c                 t    t          t          j        ||                                          ||           d S r  r  )rs   r1   r2   r  rW  s        r3   r  zTestBeta.test_extreme_entropyT  s8    H 	
1a((0022CcBBBBBBr5   N)r   r   r   r  rb  re  ri  r   r   skipifMACOS_INTELrm  rr  rx  r  xfailr   r~  r   r9   r  r   r  r  r  r  rq  r  r;   r5   r3   r[  r[    s       , , ,7 7 7& & &
H H H [,DEE	H 	H FE	H [,DEE    FE " [,DEE	* 	* FE	*8 8 8. [w'GHH$ $ IH$
 [X

'GHH[Vnn%EFF  GF IH. [^GCEF F1 1F F1
 [^GEGH H+ +H H+ [[D D DE E
9 9E E
9 [	
 	
 	
 .C C/ .C C Cr5   r[  c                   x   e Zd Zg dZd Zd Zej                            dg d          d             Z	ej                            de          d             Z
d	 Zej                            d
eg dz             d             Zej                            d
ddg          d             Zd Zd Zd Zd Zd ZdS )TestBetaPrime))r        Y@rF  D>?)r  r  rF  m嗽?)r   rF  r]   gPK?)r   r  rF  g%p?)r  rF  r]   g#|/?)r  r  r  gsM
<)r  rF  r  $5Є*?)^ 9^;rF  r]   gompP?)r  r  r  g7j"!9)r  rF  r  z^J?)r  rF  r]   gܷ"+>)r  r  r  grNs: )r  rF  r  gK>c                    d\  }}t          j        g d          }t          j        ||          }t	          t          j        |                    |                                                               t          |	                    |          t          j
        |                    |                               d S )Nr_  r`  )rj   r   r9   r   r	   rG  r   r   r   ri   r   ra  s        r3   r  zTestBetaPrime.test_logpdf  s    tH___%%OE4((AHHQKK((,,..///a"&!"5"566666r5   c                    t           j                            ddd          }t          |d           d\  t	          j        g d          }t           j                            |          }t          t	          j        |                                                     t           j	        j
        fd|D             }t          ||dd	           d S )
Nr   rQ  r~  r   r_  r`  c                 @    g | ]} t           j        |          S r;   )r9   r   )rA   rI  r  r  gen_cdfs     r3   rC   z*TestBetaPrime.test_cdf.<locals>.<listcomp>  s+    JJJ''%/3t<<JJJr5   -=rn  )r9   r   rq   r   rj   r   r	   rG  r   rF   _cdf_singler   )rs   r_   r  cdfs_gr  r  r  s       @@@r3   r\  zTestBetaPrime.test_cdf  s     O3,,QtH___%%""1eT22D!!%%''((( %1JJJJJJJJJf15999999r5   zp, a, b, expected))r~   r  r  gV)?)r  r  r  gWJB9*=)r8  r  r  gOG J<)r  rl        @gΣ01)      ?rl  r  g8ǘ`?)>q?rF  r]   gMDc                 l    t           j                            |||          }t          ||d           d S r#  )r9   r   r   r   )rs   rv  r1   r2   rX   r_   s         r3   r  zTestBetaPrime.test_ppf  s8     O1a((8%000000r5   z
x, a, b, pc                 h    t          t          j                            |||          |d           d S r  )r   r9   r   r   )rs   r_   r1   r2   rv  s        r3   test_ppf_gh_17631zTestBetaPrime.test_ppf_gh_17631  s1    ++Aq!44aeDDDDDDr5   c                     t          j        d          }t          j        d          }t          j        d          }t          t          j                            |||          dd           d S )NrT  r|   r  r   )rj   r   r   r9   r   _ppf)rs   r1   r2   rv  s       r3   	test__ppfzTestBetaPrime.test__ppf  sX    HSMMHSMMHSMM,,Q155sGGGGGGr5   zx, a, b, expected))r  r  r  r  )r  rF  r]   gV[?)r  rF  r]   r  c                 h    t          t          j                            |||          |d           d S r#  )r   r9   r   rq   )rs   r_   r1   r2   rX   s        r3   test_cdf_gh_17631zTestBetaPrime.test_cdf_gh_17631  s3     	++Aq!44hUKKKKKKr5   )rL  rF  r]   g%?)rL  r  rF  goXr?c                 |    t           j                            |||          }|dk     sJ t          ||d           d S )NrT  gh㈵>r   )r9   r   rq   r   )rs   r_   r1   r2   rX   r2  s         r3   test_cdf_extreme_tailsz$TestBetaPrime.test_cdf_extreme_tails  sE     O1a((3wwww8$//////r5   c                     g d}g d}g d}g d}t           j                            |||          }t          ||d           d S )N)r   r   rg   rF  rF  rF  rF  r  r  rF  rF  rF  r  r  )r   rg   r    r]   r]   r]   r]   rF  rF  r  r  r  r  r  )r  r  ꌠ9Y>)Fr  r  r  r  r  r  r  r  r  r  r  )g/:g  U/7gKH9g<![a?gw 4?g)J?g5?r  r  r  r  g%?gsM
<r  r  r   )r9   r   r   r   )rs   r1   r2   r_   r  	sf_valuess         r3   r  zTestBetaPrime.test_sf  s|        0 0 0; ; ;
 O&&q!Q//		3U333333r5   c                 |    d}d}d}d}t           j                            |||          }t          ||d           d S )Nr{   r|   rU  gg%`ʼr  r   )r9   r   r  r   r  s         r3   r  zTestBetaPrime.test_logcdf   sJ    $''1a00%000000r5   c                 |    d}d}d}d}t           j                            |||          }t          ||d           d S )Nr-  r9  r|   g9;ЋVr  r   )r9   r   r  r   r  s         r3   r  zTestBetaPrime.test_logsf  sJ    %%%aA..s//////r5   c                     t           j                            g ddd           t          j        dd                               d           d S )N)r]   rl  r~  ro  皙?r   r    rB  r  r  )r9   r   r   rx   s    r3   test_fit_stats_gh18274z$TestBetaPrime.test_fit_stats_gh18274  sQ     	666QqIII!q!!!''/////r5   c                     t           j        dg}t          j        dddg                              d          }t          ||           d S )NgooB?rg   r	  gffffff@r   )rj   r  r9   r   r  r   )rs   r  r   s      r3   test_moment_gh18634z!TestBetaPrime.test_moment_gh18634  sJ     v()oa#s,,33A66S!!!!!r5   N)r   r   r   cdf_valsr  r\  r   r   r   r  r  r  r  r  r  r  r  r  r  r;   r5   r3   r  r  {  s         H 7 7 7: : :> [	B 	B 	B 1 1 1 [\844E E 54EH H H [ 
 
 
 	
 L L L [	.	/	12 20 0	2 204 4 4*1 1 10 0 00 0 0	" 	" 	" 	" 	"r5   r  c            
       t   e Zd Zd Zd Zd Zd Zd Zd Ze	j
                            ddd	g          d
             Ze	j
                            dg d          d             Ze	j
                            dg d          e	j
                            dg d          e	j
                            dg d          e	j
                            dddg          e	j
                            dddg          e	j
                            dddg          d                                                                         ZdS )	TestGammac                     t           j                            ddd          }t          |d           t           j                            ddd          }t          |d           d S )	NZ   i  rQ  r  g
CTb?r   r`   gBΖ?)r9   r  ri   r   rP  s     r3   rR  zTestGamma.test_pdf%  sZ    koob#To22C---kooa4o00C+++++r5   c                 f    t           j                            dd          }t          |d           d S r  )r9   r  r   r   r]  s     r3   r  zTestGamma.test_logpdf-  s1     ##Aq))FA&&&&&r5   c                 ^    g d}t          t          t          j        j        |dd           d S )Nrd  r   r   )r   r  )r`  r  r9   r  r   r  s     r3   re  z#TestGamma.test_fit_bad_keyword_args3  s.    OOi!!8LLLLLLr5   c                     t          j        t          j                            dd          dd          sJ t          j        t          j                            dd          dd	          sJ d S )
Nr  r    gplC@r  r   r  ra   g1}t@r   )rj   iscloser9   r  r  rx   s    r3   r  zTestGamma.test_isf7  s    " z%+//%33+%9 9 9 	9 	9 	9z%+//%55+%9 9 9 	9 	9 	9 	9 	9r5   c                 v    d}d}d}t           j                            ||          }t          ||d           d S )NP   r  gޟ򑁺r  r   )r9   r  r  r   r  s        r3   r  zTestGamma.test_logcdfM  sC    $##Aq))%000000r5   c                 v    d}d}d}t           j                            ||          }t          ||d           d S )Nr  r  gDmr  r   )r9   r  r  r   r  s        r3   r  zTestGamma.test_logsfT  sC    %!!!Q''s//////r5   rf   rT  rU  c                 |    t           j                            |dz  |dz  d|          }t          |dd           d S )N   r"  r   r  g>S*r   r   )r9   r  r  r   rs   rf   rr  s      r3   r  zTestGamma.test_delta_cdf[  sF     &&uSy%)Qe&LL6UCCCCCCr5   za, ref, rtol))r  gt.r   )rg   r  r   )ra   gX>k-־@r   )r
  gwM7@r   )g NgmCg%og$6@r   )r  oܘ#]@r   c                 d    t          t          j                            |          ||           d S r  )r   r9   r  rH   )rs   r1   r  r   s       r3   rq  zTestGamma.test_entropyi  s/     	++A..$??????r5   r1   )r~   r    r  re   )r~   r   r  rh  TFr   r>  c                 H   t           j                            d          }t          j                            |||d|          }i }	|r||	d<   |r||	d<   |r||	d<   dt          |	          z
  }
|
dk    rPd	}t          j        t          |
          5  t          j        j
        |fddi|	 d d d            n# 1 swxY w Y   d S t          j        j
        |fddi|	}t          j        | }|
dk    r4t          |                                t          j        |                     |
dk    r8t          |                    d          t          j        |dz                       |
dk    r:t          |                    d          t          j        |dz                       d S d S )Nr   ra   rA  rg  r   r   r   r   r  r  r  r  r    rg   )rj   r   r   r9   r  r   r  r   r   r  r   r   r   r  )rs   r1   re   rf   rh  r   r>  r   r   r   r  rE  thetarB   s                 r3   test_fit_mmzTestGamma.test_fit_mmz  s    i##$788{qcS,/  1 1  	DJ 	DL 	#"DNCIIA::MIz;;; ; ;::T:T:::; ; ; ; ; ; ; ; ; ; ; ; ; ; ;F::T:T::{E"A::DIIKK777A::DKKNNBGD!G,<,<===A::DKKNNBGD!G,<,<===== :s   B66B:=B:N)r   r   r   rR  r  re  r  r  r  r   r   r   r  rq  r  r;   r5   r3   r  r  $  s       , , ,' ' 'M M M9 9 9,1 1 10 0 0 [WsCj11D D 21D [^B B BC C@ @C C@ [S...11[UNNN33[Wnnn55[WtUm44[Yu66[[4-88> > 98 76 54 65 43 21> > >r5   r  c                      d} t          j        d| dz  z  d| dz  z  dd          }t          |                    d          d           d S )	Nr  g;f?rg   r    r   rd   rT  gV{q@)r9   r  r   ri   )jitterZs     r3   test_pdf_overflow_gh19616r    sT     F6FAI-q619}!1MMMAAEE#JJ 122222r5   c                       e Zd Zd Zej                            dg d          d             Zej                            dg d          d             Zej                            dg d          d	             Z	d
 Z
dS )
TestDgammac                    t           j                            d          }d}|                    d|          }|                    d|          }t
          j                            ||          }t
          j                            t          j	        |          |          dz  }t          ||           t          j        |          }t          |                    |          |d           d S )Nl   OP($ r`   r	  )r  r   rg   r  r   )rj   r   r   normalr  r9   dgammari   r  r   r   )rs   r   r   r_   r1   r   r  rB   s           r3   rR  zTestDgamma.test_pdf  s    i##$788JJRdJ++KKRdK++lq!$$koobfQii++a/S!!!|ASu555555r5   zx, a, expected))r  r    ggrU>)r  r    g$IC<)ir   g2Wj<)ry  g~)A.=)r   r    gf?c                    t           j                            ||          }t          ||d           t           j                            ||          }t          ||d           t           j                            | |          }t          ||d           t           j                            ||          }t          || d           d S r  )r9   r  rq   r   r   r   r  )rs   r_   r1   rX   rq   r   r   r  s           r3   test_cdf_ppf_sf_isf_tailz#TestDgamma.test_cdf_ppf_sf_isf_tail  s     lq!$$XE2222lx++QU++++\__aR##H51111lx++aRe,,,,,,r5   r  ))r  gmqn @)?g?)r  gx?c                 d    t          t          j                            |          |d           d S r#  r   r9   r  rH   r  s      r3   rq  zTestDgamma.test_entropy  s/      	,,Q//5AAAAAAr5   ))r  r  )r  g>_)r;  g5i)r
  go@)r  g4Fa3@)r  grO]@c                 d    t          t          j                            |          |d           d S r  r  r  s      r3   test_entropy_entreme_valuesz&TestDgamma.test_entropy_entreme_values  s/    " 	,,Q//5AAAAAAr5   c                    t          j        g d          }t          j                            |          }t          t          |                    D ]3}||         t          j                            ||                   k    sJ 4d S )N)r    r   r  r;  )rj   r   r9   r  rH   r  r  )rs   r_   r2  r  s       r3   test_entropy_array_inputz#TestDgamma.test_entropy_array_input  s~    H'''((L  ##s1vv 	6 	6AQ45<//!5555555	6 	6r5   N)r   r   r   rR  r   r   r   r  rq  r  r  r;   r5   r3   r  r    s        6 6 6, [-: : :; ;- -; ;- [X8 8 89 9B B	9 9B [X: : :; ;
B 
B; ;
B6 6 6 6 6r5   r  c                      e Zd Zd Zej                            dg d          d             Zej                            dg d          d             Zd Z	ej                            dg d	          d
             Z
d ZdS )TestChi2c                     t          t          j                            dd          dd           t          t          j                            dd          dd           d S )Nr   gޞw1D?r.  rW  ra   g6:֜?r   r9   chi2ri   rx   s    r3   r  zTestChi2.test_precision   sj    EJNN4668N$&	( 	( 	( 	(EJNN3446J$&	( 	( 	( 	( 	( 	(r5   rP  ))g     p@r   g~#e)g      ^@r?  gtA)r  r  gzĀ׿c                 j    t           j                            ||          }t          ||d           d S r  )r9   r  r  r   )rs   r_   r  r  r  s        r3   r  zTestChi2.test_logcdf  s6     ""1b))%000000r5   ))r  r?  r  )r  r}  gc)r  r`   g]3c                 j    t           j                            ||          }t          ||d           d S r  )r9   r  r  r   )rs   r_   r  r  r  s        r3   r  zTestChi2.test_logsf  s6     
  B''s//////r5   c                    d}t           j                            d|          }t          |dd           t           j                            d|          }t          |dd           d}t           j                            d	|          }t          |d
d           t           j                            d|          }t          |dd           d S )Nr<  gλm:=6g'bd5 <r  r   r|   gQ8@r  gܐ؆0guu[9=r]   g8-*@)r9   r  r   r   )rs   r  r_   s      r3   r  zTestChi2.test_ppf  s    JNN5"%%<5IIIIJNN3##7eDDDDJNN5"%%:GGGGJNN3##8uEEEEEEr5   r  ))r  gܫ>)r    gƯʉ?)ra   gn>@)   g͔v0@)r  g9:3@c                 r    t          t          j        |                                          |d           d S r   )r   r9   r  rH   r  s      r3   rq  zTestChi2.test_entropy4  s3     	
2..00#EBBBBBBr5   c                 d    t          t          j                            dd          dd           d S )Nr   rg   r|   r.  r  rx   s    r3   test_regression_ticket_1326z$TestChi2.test_regression_ticket_1326=  s*    EJNN322C<<<<<r5   N)r   r   r   r  r   r   r   r  r  r  rq  r  r;   r5   r3   r  r    s       ( ( ( [	+ 	+ 	+ 1 1 1
 [	+ 	+ 	+ 0 0 0F F F0 [Y: : :; ;C C; ;C= = = = =r5   r  c                   l    e Zd Zd Zd Zd Zej                            dddg          d             Z	dS )	TestGumbelLc                     t          j        dd          }t          j                            |          }t          j                            |          }t          ||           d S Nr  r  )rj   rY  r9   r  rq   r   r   rc  s       r3   r  zTestGumbelL.test_cdf_ppfD  sS    Kb!!Nq!!^""2r5   c                    t          j        dd          }t          j                            |          }t          j                            |          }t          j        |          }t          j        |           }t          ||           d S r  )
rj   rY  r9   r  r  r  r   r   expm1r   )rs   r_   r2  r  ur  s         r3   r  zTestGumbelL.test_logcdf_logsfJ  sr    Kb!!N!!!$$N  ##F1II]11r5   c                     t          j        dd          }t          j                            |          }t          j                            |          }t          ||           d S )Nr  r   )rj   rY  r9   r  r   r  r   rc  s       r3   r  zTestGumbelL.test_sf_isfR  sS    KQNa  ^""2r5   re   r   r    c                     t           j                            d|          }t           j                            ||          \  }}t	          ||           d S )Nra   )r   re   r   )r9   r  r   r   r   )rs   re   r   
fitted_locr   s        r3   test_fit_fixed_paramz TestGumbelL.test_fit_fixed_paramX  sR     ~!!s!44**4c*::
AZ%%%%%r5   N)
r   r   r   r  r  r  r   r   r   r  r;   r5   r3   r  r  B  ss               [URG,,& & -,& & &r5   r  c                       e Zd Zd Zd ZdS )TestGumbelRc                 d    t          t          j                            d          dd           d S )Nr=  ?~T}%m;r  r   )r   r9   r  r   rx   s    r3   r  zTestGumbelR.test_sfc  s=     	))"--/E"	$ 	$ 	$ 	$ 	$ 	$r5   c                 d    t          t          j                            d          dd           d S )Nr  r  r  r   )r   r9   r  r  rx   s    r3   r  zTestGumbelR.test_isfl  s=     	**5113D"	$ 	$ 	$ 	$ 	$ 	$r5   N)r   r   r   r  r  r;   r5   r3   r  r  a  s2        $ $ $$ $ $ $ $r5   r  c                   	   e Zd Z ej        d          d             Zej        d             Zej        d             Zej        d             Zej	        j
        ej	                            d ej        d           ej        d	ej	        j
        
          g          ej	                            dddg          ej	                            dg d          ej	                            dddg          d                                                             Zej	        j        ej	                            dddg          d                         Zd Zej	                            d          ej	                            dddg          ej	                            dddg          d                                     Zd  Zd! Zd" Zej	        j
        ej	                            d# ej        g d$g d%g d&           ej        g d'g d(g d)ej	        j
        
           ej        g d* ej        d+d,d-           ej        d.dd/          ej	        j        
          g          d0                         Zej	                            d# ej        g d$g d%g d&           ej        g d'g d(g d)ej	        j
        
           ej        g d* ej        d+d,d-           ej        d.dd/          ej	        j        
          g          d1             Zej	                            d2d3dg          ej	                            d4d5d6g          d7                         Zej	                            d8d9d:gd;d:gg          d<             Zej	                            d=d>d3ej        ej        ej        fgd?d@gg          dA             Zej	                            dBg dC          ej	                            dDej         j!        dE ej        dFd3dG          dHfej         j"        dE ej        dFd3dG          dHfej         j!        dI ej        d3dJdG          dEfej         j"        dI ej        d3dJdG          dHfg          dK                         Z#ej	                            dLg dM          dN             Z$ej	        j        ej	                            dLg dO          dP                         Z%dQ Z&dR Z'dSS )TTestLevyStableT)autousec                     dt           j        _        dt           j        _        dt           j        _        t           j        j        t           j        _        dS )z2Setup default parameters for levy_stable generatorS1	piecewiseN)r9   r  parameterizationcdf_default_methodpdf_default_method_levy_stable	_QUAD_EPSquad_epsrx   s    r3   reset_levy_stable_paramsz'TestLevyStable.reset_levy_stable_paramsw  s<     .2*/:,/:,%*%7%A"""r5   c                     t          j        t          t                    j        dz            }t           j                            |j        d          }|S )a  Sample data points for pdf computed with Nolan's stablec

        See - http://fs2.american.edu/jpnolan/www/stable/stable.html

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

        The data table loaded below is generated from Nolan's stablec
        with the following parameter space:

            alpha = 0.1, 0.2, ..., 2.0
            beta = -1.0, -0.9, ..., 1.0
            p = 0.01, 0.05, 0.1, 0.25, 0.35, 0.5,
        and the equivalent for the right tail

        Typically inputs for stablec:

            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/levy_stable/stable-Z1-pdf-sample-data.npyx,p,alpha,beta,pctr  rj   r  r   r  r  r  r  Tr|  s     r3   nolan_pdf_sample_dataz$TestLevyStable.nolan_pdf_sample_data  sM    6 wNN!<=
 
 v  /C DDr5   c                     t          j        t          t                    j        dz            }t           j                            |j        d          }|S )a#  Sample data points for cdf computed with Nolan's stablec

        See - http://fs2.american.edu/jpnolan/www/stable/stable.html

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

        The data table loaded below is generated from Nolan's stablec
        with the following parameter space:

            alpha = 0.1, 0.2, ..., 2.0
            beta = -1.0, -0.9, ..., 1.0
            p = 0.01, 0.05, 0.1, 0.25, 0.35, 0.5,

        and the equivalent for the right tail

        Ideally, Nolan's output for CDF values should match the percentile
        from where they have been sampled from. Even more so as we extract
        percentile x positions from stablec too. However, we note at places
        Nolan's stablec will produce absolute errors in order of 1e-5. We
        compare against his calculations here. In future, once we less
        reliant on Nolan's paper we might switch to comparing directly at
        percentiles (those x values being produced from some alternative
        means).

        Typically inputs for stablec:

            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/levy_stable/stable-Z1-cdf-sample-data.npyr
  r  r  r|  s     r3   nolan_cdf_sample_dataz$TestLevyStable.nolan_cdf_sample_data  sN    J wNN!<=
 
 v  /C DDr5   c                 b    t          j        t          t                    j        dz            }|S )a&  Sample data where loc, scale are different from 0, 1

        Data extracted in similar way to pdf/cdf above using
        Nolan's stablec but set to an arbitrary location scale of
        (2, 3) for various important parameters alpha, beta and for
        parameterisations S0 and S1.
        z1data/levy_stable/stable-loc-scale-sample-data.npy)rj   r  r   r  r  r|  s     r3   nolan_loc_scale_sample_dataz*TestLevyStable.nolan_loc_scale_sample_data  s1     wNN!?@
 
 r5   sample_sizer=  rE  marksr  S0r   z
alpha,beta))rT  r   )rT  r   )r  r   )r&  r|   zgamma,deltar    r   rv  c                     |t           j        _        t          j        ||||          }t          j        |                    |d          |j                  \  }}	|	dk    sJ d S )N)r  r  rf   re   r   r   rF  )r9   r  r  rI  r   rq   )
rs   r  r  r  r  rr  r  lsr   rv  s
             r3   rJ  zTestLevyStable.test_rvs  sp    ( .>*d%U
 
 
 |FF$F77
 
1 4xxxxxxr5   r  r|   r    c                     t           j                            d           d}d}d}t          j                            ||||d          }t          j        |d||||f          \  }}|d	k    sJ d
S )z3Additional test cases for rvs for alpha equal to 1.h:rT  r|   r  r  re   rf   r   r  r[  r~   N)rj   r   r   r9   r  r   rI  )rs   r  r  re   rf   r_   statrv  s           r3   test_rvs_alpha1zTestLevyStable.test_rvs_alpha1  s     		y!!!!!%3e'+ " - -,q-%*D#u$=? ? ?a4xxxxxxr5   c                    g d}t           j                            |          \  }}}}t          |ddd           t	          |dd           t	          |dd	           t	          |d
d           |g dz   }t           j                            |          \  }}}	}
t          |d           t          |d           t	          |
dd	           t	          |	dd	           d S )N)wgr  r   r   r   r   羼 u?r   r   r   r   67',?r!  r!  r!  r!  tA}˜.?r"  r"  r"  r"  Gz?r   r~   rf  )\(̿rg   gۧ1?r   g9]c?)r"  r"  r"  r"  r"  r   gv?r!  )r9   r  rt  r   r   r   rs   r_   alpha1beta1loc1scale1rs  alpha2beta2r  r  s              r3   ry  zTestLevyStable.test_fit  s   
 
 

 ',&7&A&A!&D&D#tV148888E4+++FGQ///'1	
 	
 	

 9999&+&7&A&A"&E&E#tVVQUBFFA...D&!,,,,,r5   zUnknown problem with fitstart.rj  zalpha,beta,delta,gamma)r  r  rg   r   )rT  r  rg   r   parametrizationc                     |t           j        _        t           j                            ||||dd          }t           j                            |          }|\  }}	}
}t          ||||g||	|
|gd           dS )z7Test that fit agrees with rvs for each parametrization.r  r   rA  r~   r   N)r9   r  r,  r   rt  r   )rs   r  r  rr  r  r,  r   r   	alpha_obsbeta_obs	delta_obs	gamma_obss               r3   test_fit_rvszTestLevyStable.test_fit_rvs  s     -<) $$4U%e$ % 
 
 ))$//471	8Y	D%')Y7	
 	
 	
 	
 	
 	
r5   c                 x   t          j        g d          }t          j                            |          \  }}}}t          j                            |           \  }}}}	t          |d           |dk    sJ t          ||           t          ||            t          ||            t          |	|           d S )Nr    r    r   r   r`   r`   r`   r:  r:  ra   ra   r    r   )rj   r   r9   r  rt  r   r   )
rs   r_   r&  r'  r(  r)  r*  r+  r  r  s
             r3   test_fit_beta_flipz!TestLevyStable.test_fit_beta_flip8  s    H???@@&+&7&A&A!&D&D#tV&+&7&A&A1"&E&E#tVUAqyyyyFF+++EE6***D4%(((FF+++++r5   c                 V   d}t          j        g d          }t          j                            |           \  }}}}t          j                            | |z             \  }}}	}
t          ||           t          ||           t          |	||z              t          |
|           d S )Nr    r4  )rj   r   r9   r  rt  r   )rs   SHIFTr_   r&  r'  r(  r)  r*  r+  r  r  s              r3   test_fit_delta_shiftz#TestLevyStable.test_fit_delta_shiftD  s    H???@@&+&7&A&A1"&E&E#tV&+&7&A&A1"u*&M&M#tVFF+++E5)))D$,///FF+++++r5   c                    g d}t           j                            |          \  }}}}|dk     sJ d|             |t          |          k     sJ dt          |           d|             g d}t           j                            |          \  }}}	}
|dk    sJ d|             |	t	          |          k    sJ dt	          |           d|	             d S )	N)r    r    r   r   r`   r`   r`   r:  r:     r:  r    zExpected alpha < 1, got zExpected loc < z, got )r    r    r   r   r`   r`   r`   r:  r:     r;  zExpected alpha > 1, got zExpected loc > )r9   r  rt  r  r  r%  s              r3   test_fit_loc_extrapz"TestLevyStable.test_fit_loc_extrapO  s    666&+&7&A&A!&D&D#tVzzz>f>>zzzc!ff}}}DADDdDD}}}777&+&7&A&A"&E&E#tVzzz>f>>zzzc"gg~~~FRFFFF~~~~~r5   z pct_range,alpha_range,beta_range)r~   r|   r  )r]   r    rg   )r   r   r   )r~   rF  r|   r@  r  )r]   r|   r    r  rg   )gr   r   r~  r  r    )r~   rF  r]   rl  ffffff?r|   ?r   r   r@  r  r]   rg   r  r   r  c                 t   |}t          j                    }|j        dk    o
|j        dk    d                    |j        |j        |j        g          }ddfdgddfd	gddfd
gddfdgddfdgddfdgg}t          |          D ]\  }	\  }
}}|
t          j        _	        || ||                   n|}t                      5 }|                    t          d           t          j                            |d         |d         |d         dd          }t          j        d          5  t!          |g d|t          j        ||d         z
            t          j        ||d         z
            t          j        |d                   z  g          }ddd           n# 1 swxY w Y   ||d         |k    t          j        |          z           }d|	 d |
 d!| d"|j        j         d#| 
}t+          ||d         ||d$%           ddd           n# 1 swxY w Y   dS )&z2Test pdf values against Nolan's stablec.exe outputLinuxi686/dnir-  c           	      r   t          j        | d                   t          j        | d                   z  t          j        | d                   z  | d         dk    | d         dk    z  | d         dk    | d         dk    z  | d         dk    z  z  | d         dk    t          j        | d         d	d
g          z  z  | d         dk    t          j        | d         ddg          z  z  | d         dk    t          j        | d         ddg          z  z  | d         dk    t          j        | d         ddg          z  t          j        t          j        | d                   g d          z  z  | d         dk    t          j        | d         dg          z  t          j        t          j        | d                   dg          z  z  | d         dk    t          j        | d         ddg          z  t          j        t          j        | d                   g d          z  z  | d         dk    | d         dk    z  | d         dk    z  z  | d         dk    | d         dk    z  | d         dk    z  z  | d         dk    | d         dk    z  | d         dk    z  z  | d         dk    t          j        | d         dg          z  t          j        t          j        | d                   g d          z  z  | d         dk    t          j        | d         ddg          z  t          j        t          j        | d                   g d          z  z  | d         dk    t          j        | d         d	d
g          z  t          j        t          j        | d                   ddg          z  z  | d         dk    z   z  S )Npctr  r  r   r|   r   r  r  r~   r  r~  rF  r@  rQ  r]   rl  r   )r|   r  r  r=  r>  )gٿ333333ӿr~  r  r|   rF  rT  r]   rQ  r~  r  )r   r   rT  r!  r  )rj   isinr   r  alpha_range
beta_range	pct_ranges    r3   rP   z7TestLevyStable.test_pdf_nolan_samples.<locals>.<lambda>  s   GAeHi00GAgJ445GAfIz223
 vY!^uX_. vY#-wZ3.0uX_. wZ3.GAeHsCj99: wZ3.GAeHsCj99:$ wZ3.GAeHr2h778%, wZ3.GAeHsCj99:GBF1V9$5$5|||DDE-6 wZ3.GAeHrd334GBF1V9$5$5t<<=7@ wZ3.GAeHsCj99:GBF1V9$5$57M7M7MNNOA#J wZ3.vY#-/uX-/K(T wZ3.vY$.0uX-/U-^ wZ3.vY#-/uX-/_2h wZ2-GAeHrd334GBF1V9$5$57G7G7GHHIi7r wZ2-GAeHsCj99:GBF1V9$5$5|||DDEs<| wZ2-GAeHsCj99:GBF1V9$5$5RyAAB}AF 7s*GC,EH r5   r  r  c                     t          j        | d                   t          j        | d                   z  t          j        | d                   z  | d         dk    z  | d         dk    z  S )NrE  r  r  rQ  rT  rj   rH  rI  s    r3   rP   z7TestLevyStable.test_pdf_nolan_samples.<locals>.<lambda>  sq    GAeHi00GAgJ445GAfIz223 wZ#%' wZ2%	' r5   c                     | d         dk     z  t          j        | d                   z  dv z  t          j        | d                   z  S )Nr  rT  rE  r  rN  )r  rJ  rK  is_linux_32rL  s    r3   rP   z7TestLevyStable.test_pdf_nolan_samples.<locals>.<lambda>  s_    wZ2%$_&GAeHi001 ;&( GAfIz22	3 r5   g&.=c                     t          j        | d                   t          j        | d                   z  t          j        | d                   z  | d         dk    z  S )NrE  r  r  rQ  rN  rI  s    r3   rP   z7TestLevyStable.test_pdf_nolan_samples.<locals>.<lambda>  s^    GAeHi00GAgJ445GAfIz223 wZ3&( r5   fft-simpsonr;  c                     | d         dk    t          j        | d                   z  t          j        | d                   z  t          j        | d                   z  S )Nr  r&  rE  r  rN  rI  s    r3   rP   z7TestLevyStable.test_pdf_nolan_samples.<locals>.<lambda>  s^    wZ3&GAeHi001GAgJ445 GAfIz223 r5   rv  c                     t          j        | d                   t          j        | d                   z  t          j        | d                   z  | d         dk    z  | d         dk     z  S )NrE  r  r  r    r&  rN  rI  s    r3   rP   z7TestLevyStable.test_pdf_nolan_samples.<locals>.<lambda>  sp    GAeHi00GAgJ445GAfIz223 wZ!^% wZ#%	' r5   Nz2Density calculations experimental for FFT method.*r_   r  r  r    r   rf   re   r  r  calcabserrrelerrrv  rY  z	pdf test  failed with method 'z' [platform: z]

Ferr_msgverbose)platformunamesystemmachinejoin	processorrj  r9   r  r  r   recordrz  ri   rj   r  r   r   isnanr   r  r   )rs   r  rL  rJ  rK  r   r`  platform_desctestsixdefault_methodr   filter_funcsubdatasuprv  subdata2failuresmessagerP  s     ```              @r3   test_pdf_nolan_samplesz%TestLevyStable.test_pdf_nolan_samples[  s   4 %   lg-I%-62I\5=%/:< < t J J J J J JL^ U % % % % % % U % % % % % % % W ' ' ' ' ' ' t & & & & & & t & & & & & &kC
L "+5!1!1*	 *	 B 3AE0 + 7 ;;t,,  =A "$$ %

"H   %))CLG$FO *   [h/// 	 	0444F1ws|#344F1ws|#344rvgcl7K7KK   H	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 $H%-(1++
W W W W W"/W W4<N4HW WLTW W   CL#!   ?% % % % % % % % % % % % % % %*	 *	s9   A'H+A(F8,H+8F<<H+?F< AH++H/	2H/	c                    |}ddfdgddfdgddfdgdd	fd
gddfdgddfdgg}t          |          D ]\  }\  }}	}
|t          j        _        |
| |
|                   n|}t	                      5 }|                    t          d           t          j                            |d         |d         |d         dd          }t          j	        d          5  t          |g d|t          j        ||d         z
            t          j        ||d         z
            t          j        |d                   z  g          }ddd           n# 1 swxY w Y   ||d         |	k    t          j        |          z           }d| d| d|j        j         d| }t          ||d         |	|d !           ddd           n# 1 swxY w Y   dS )"z4 Test cdf values against Nolan's stablec.exe output.r  r  c                    t          j        | d                   t          j        | d                   z  t          j        | d                   z  | d         dk    t          j        | d         g d          z  | d         dk    z  | d         dk    t          j        | d         g d          z  | d         dk    z  z   z  S 	NrE  r  r  rT  )rF  r%  r!  r~   r]   rQ  r~  r  rN  rI  s    r3   rP   z7TestLevyStable.test_cdf_nolan_samples.<locals>.<lambda>P  s    GAeHi00GAgJ445GAfIz223
 wZ2-GAfI/A/A/ABBCuX-/
 wZ2-GAfI??@uX-/	 r5   rF  c                    t          j        | d                   t          j        | d                   z  t          j        | d                   z  | d         dk    t          j        | d         g d          z  | d         dk    z  z  | d         dk    t          j        | d         g d          z  | d         dk    z  z  S rt  rN  rI  s    r3   rP   z7TestLevyStable.test_cdf_nolan_samples.<locals>.<lambda>e  s    GAeHi00GAgJ445GAfIz223 7r)&	+=+=+=>>?5T)+	 7r)&	???;;<5T)+ r5   rR  r;  c                     t          j        | d                   t          j        | d                   z  t          j        | d                   z  | d         dk    z  S )NrE  r  r  333333?rN  rI  s    r3   rP   z7TestLevyStable.test_cdf_nolan_samples.<locals>.<lambda>w  s^    GAeHi00GAgJ445GAfIz223 wZ#%' r5   r  c                     t          j        | d                   t          j        | d                   z  t          j        | d                   z  | d         dk    z  | d         dk    z  S )NrE  r  r  r  rx  rN  rI  s    r3   rP   z7TestLevyStable.test_cdf_nolan_samples.<locals>.<lambda>  q    GAeHi00GAgJ445GAfIz223 wZ#%' wZ3&	( r5   r  c                     t          j        | d                   t          j        | d                   z  t          j        | d                   z  | d         dk    z  | d         dk    z  S )NrE  r  r  r  r  rN  rI  s    r3   rP   z7TestLevyStable.test_cdf_nolan_samples.<locals>.<lambda>  rz  r5   r~   c                     t          j        | d                   t          j        | d                   z  t          j        | d                   z  | d         dk    z  | d         dk    z  S )NrE  r  r  rT  r  rN  rI  s    r3   rP   z7TestLevyStable.test_cdf_nolan_samples.<locals>.<lambda>  rz  r5   Nz[Cumulative density calculations experimental for FFT method. Use piecewise method instead.*r_   r  r  r    r   rU  r  r  rV  rv  rY  z	cdf test rZ  z'
r[  Fr\  )rj  r9   r  r  r   re  rz  rq   rj   r  r   r   rf  r   r  r   )rs   r  rL  rJ  rK  r   rh  ri  rj  r   rk  rl  rm  rv  rn  ro  rp  s     ```            r3   test_cdf_nolan_samplesz%TestLevyStable.test_cdf_nolan_samples3  s   2 % U % % % % % %, T $ $ $ $ $ $& t & & & & & & t & & & & & & t & & & & & & t & & & & & &GL
\ "+5!1!1(	 (	 B 3AE0 + 7 ;;t,,  =A "$$ #

"@  
 %))CLG$FO *   [h/// 	 	0444F1ws|#344F1ws|#344rvgcl7K7KK   H	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 $H%-(1++Ar A A A A&n2A A6>A ACL#!   ;# # # # # # # # # # # # # # #(	 (	s9   A'G2A(E&G&E**G-E*.AGG	G	r  r   r  ri   rq   c                    t          j                    }|j        dk    odt          j                    d         v }|r|dk    rt	          j        d           |}dt          j        _        dt          j        _	        ||d         |k             }d| t          j        _
        |d	v sJ |dk    rt          j        j        nt          j        j        } ||d
         |d         |d         dd          }	t          |	||         d           dS )zGTests for pdf and cdf where loc, scale are different from 0, 1
        r@  32bitr   ri   z4Test unstable on some platforms; see gh-17839, 17859r  r  S)ri   rq   r_   r  r  rg   r   rU  r;  N)r_  r`  ra  architecturer   r   r9   r  r  r  r  ri   rq   r   )
rs   r  r  r  r`  rP  r   rl  r  rN  s
             r3   test_location_scalez"TestLevyStable.test_location_scale  s      lg-W'X=R=T=TUV=W2W  	P45==KNOOO* 0;,/:,tG}-.-8[[*~%%%%%)U]]E!!8I8M 	 XCL''*GFO1!
 
 
 	GDM400000r5   zmethod,decimal_placesrC  r   r  c           	         t          j        g d          }t          j        g d          }t          j        g d          }t          j        d          5  t                      5 }|                    t
          d           |t          j        _        t          j        	                    |d|dd	
          }t          ||||           ddd           n# 1 swxY w Y   ddd           dS # 1 swxY w 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    rg   rg   rg   rg   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   rl  r|   r    r   rl  r|   r    r   rl  r|   r    r   rl  r|   r    r   rl  r|   r    r  r   zDensity calculation unstable.*)categoryrp  r    r   rU  N)rj   r   r  r   rS   rz  r9   r  r  ri   r   )rs   r  decimal_placesxsdensitybetasrm  ri   s           r3   'test_pdf_alpha_equals_one_beta_non_zeroz6TestLevyStable.test_pdf_alpha_equals_one_beta_non_zero  s    XHHH
 
 (  
 
   
 
 [X&&& 
	 
	(9(;(; 
	sJJ'8     4:E0#''AuA1'EECWnf  
	 
	 
	 
	 
	 
	 
	 
	 
	 
	 
	 
	 
	 
	 
	 
	 
	 
	 
	 
	 
	 
	 
	 
	 
	 
	 
	 
	 
	 
	 
	 
	 
	s7   C/'A$CC/C	C/C	C//C36C3zparams,expected)r#  r$  r   r    )rg   r   r`   r  )r`   r9  r   r   c                     t           j                             |d         |d         |d         |d         d          }t          ||           d S )Nr   r    rg   r   r  rG  )r9   r  r   )rs   r  rX   observeds       r3   r  zTestLevyStable.test_stats  sU     $**1IvayfQivay + 
 
 	Hh/////r5   r  )rl  r|   r   zfunction,beta,points,expectedrT  r  r`   r   r  r  c                     d|cxk     rdk     sn J t           ||||          t          j        t          |          |                     dS )a\  Ensure the pdf/cdf routines do not return nan outside support.

        This distribution's support becomes truncated in a few special cases:
            support is [mu, infty) if alpha < 1 and beta = 1
            support is (-infty, mu] if alpha < 1 and beta = -1
        Otherwise, the support is all reals. Here, mu is zero by default.
        r   r    r  r  N)r   rj   r#  r  )rs   r  r  r  r   rX   s         r3   !test_distribution_outside_supportz0TestLevyStable.test_distribution_outside_support  se    P 5}}}}1}}}}}}HV5t444GCKK**	
 	
 	
 	
 	
r5   zx,alpha,beta,expected))r   /iZ?s0?gz5F?)r   ~H?|/mg>m?)r   .d5?L,ٿg깃\?)r   >?ЀjoؿgA]4?)r   pa2?Յ пgqQ?c                     dt           j        _        t          t           j                            |||          |           dS )a  Test pdf for x equal to zeta.

        With S1 parametrization: x0 = x + zeta if alpha != 1 So, for x = 0, x0
        will be close to zeta.

        When case "x equal zeta" is not handled properly and quad_eps is not
        low enough: - pdf may be less than 0 - logpdf is nan

        The points from the parametrize block are found randomly so that PDF is
        less than 0.

        Reference values taken from MATLAB
        https://www.mathworks.com/help/stats/stable-distribution.html
        Afc=r  Nr9   r  r  r   ri   rs   r_   r  r  rX   s        r3   test_x_equal_zetaz TestLevyStable.test_x_equal_zetaI  sJ    Z &-"!!!5t!<<	
 	
 	
 	
 	
r5   )
)r  r  r  gz¶?)r  r  r  gV?)r  r  r  gRE)3\?)r  r  r  gB?)r  r  r  g/^?)-C6r  r  gH?)r  r  r  gn?)r  r  r  gx\?)r  r  r  gVM%?)r  r  r  gm?c                     dt           j        _        t          t           j                            |||          |           dS )a  Test pdf for x near zeta.

        With S1 parametrization: x0 = x + zeta if alpha != 1 So, for x = 0, x0
        will be close to zeta.

        When case "x near zeta" is not handled properly and quad_eps is not
        low enough: - pdf may be less than 0 - logpdf is nan

        The points from the parametrize block are found randomly so that PDF is
        less than 0.

        Reference values taken from MATLAB
        https://www.mathworks.com/help/stats/stable-distribution.html
        r  r  Nr  r  s        r3   test_x_near_zetazTestLevyStable.test_x_near_zeta}  sJ    D &-"!!!5t!<<	
 	
 	
 	
 	
r5   c                     t           j        j        }t          dddd          }t          j        }t	          j        di |}d|_        d|_         |j        di |d |d	          d
}|                    d |d	          
          }t          j        ||k              rJ d|_        d|_         |j        di |d |d	          d
}|                    d |d	          
          }t          ||           t          ||           d S )Nr&  r]   r   rT  r  r  re   rf   r  r   r`   
r   r;   )
rj   r   r   r  r9   r  r  r   r   r   	rs   r   r  unfrozenfrozen
unfrozen_afrozen_a
unfrozen_bfrozen_bs	            r3   $test_frozen_parameterization_gh20821z3TestLevyStable.test_frozen_parameterization_gh20821  s5    i#Ccs#>>>$",,V,,$(!"&!X\QQFQ##i..QQQQ
::2CC	NN:CC6(j011111$(!"&!X\QQFQ##i..QQQQ
::2CC	NN:CCXz***Z*****r5   c                    t           j        j        }t          dddd          }t          j        }d|_        t	          j        di |} |j        di |d |d          d	}|                    d |d          	          }t          ||           d
|_        t	          j        di |} |j        di |d |d          d	}|                    d |d          	          }t          ||           d S )Nr&  r]   r   rT  r  r  r`   r  r   r   r;   )	rj   r   r   r  r9   r  r  r   r   r  s	            r3   %test_frozen_parameterization_gh20821bz4TestLevyStable.test_frozen_parameterization_gh20821b  s     i#Ccs#>>>$$(!",,V,,!X\QQFQ##i..QQQQ
::2CC	NN:CCXz***$(!",,V,,!X\QQFQ##i..QQQQ
::2CC	NN:CCXz*****r5   N)(r   r   r   r   r  r  r  r  r  r   r   r   r  rJ  r   r  ry  r  r2  r5  r8  r<  rj   rY  rq  r}  r  r  r  r  r  r9   r  rq   ri   r  r  r  r  r  r;   r5   r3   r  r  v  s       V^D!!!B B "!B ^  ^B ^) ) ^)V ^  ^ [[FLlfl4v{7GHHH
 
 [/$>>[CCC  []VV,<==  >=  ?>  $ [[Vc1X..
 
 /. 
- - -0 [>??[ 	
  [D$< 
 
   @?

, 
, 
,	, 	, 	,
G 
G 
G [[*FL

 
 FL(((###(((k&	   FLDDDCB''B2&&k'	  -
 *@ @+  ,@D [*FL

 
 FL(((###(((k&	   FLDDDCB''B2&&k'	  -
 *z z+ *zx [Wq!f--[VeU^441 1 54 .-1@ [AJ!	
   @ [!RVRVRV!<=/	
 0 0 0 [W&7&7&788[' !%CB''	 !%CB''	 !%Ar2&&	 !%Ar2&&	'	
 :
 
;  98<
  [&	
 	
 	
) 8
 
9 8
0 [[	
 	
 	
  
 
!  "
0+ + +,+ + + + +r5   r  c                       e Zd Zd Zd ZdS )TestArrayArgumentc                 D    t           j                            d           d S r5  r6  rx   s    r3   r7  zTestArrayArgument.setup_method  r8  r5   c                     t           j                            t          j        d          t          j        d          d          }t          |j        d           d S )Nr   r  r  )r9   r  r   rj   r  r  r   rB  rs   r   s     r3   test_noexceptionz"TestArrayArgument.test_noexception  sK    jnn")A,,rwqzz")  + +SY(((((r5   N)r   r   r   r7  r  r;   r5   r3   r  r    s2          ) ) ) ) )r5   r  c                       e Zd Zd Zd ZdS )TestDocstringc                    t           j        j        2t          dt           j        j                                        v            t           j        j        4t          dt           j        j                                        v            d S d S )Nr+  r  )r9   r+  r  r	   lowerr  rx   s    r3   test_docstringszTestDocstring.test_docstrings  sn    >!-J%."8">">"@"@@AAA?".K5?#:#@#@#B#BBCCCCC /.r5   c                 R    t          j                     t          j                     d S rv   )r9   rF   rE   rx   s    r3   test_no_name_argzTestDocstring.test_no_name_arg  s&    r5   N)r   r   r   r  r  r;   r5   r3   r  r    s5        D D D    r5   r  c                     t          g d          } t          | dk    | d          \  }}t          |g d           t          |dg           t          ddk    | d          \  }}t          ||            t          |dgt          j        |           z             t          | dk    | d          \  }}t          ||            t          |dgt          j        |           z             d S )N)r    r   rg   r    rg   r   r   r    rg   )r   rg   rg   r   r   r   )r   r   r   rj   r   r1   r2   r  s      r3   test_args_reducer    s    ###$$Aa!eQ""DAqq///***q1#a!eQ""DAqq!q1#

*+++a!eQ""DAqq!q1#

*+++++r5   c                   R   e Zd Zg dZd Zg dZej                            de	          d             Z
d Zd Zd Zd	 Zd
 Zd Zd Zej        j        ej                            dddg          d                         Zej                            dddg          d             ZdS )TestFitMethod)ncfrL  rM  	irwinhallc                 D    t           j                            d           d S r5  r6  rx   s    r3   r7  zTestFitMethod.setup_method  r8  r5   )rg  r  r  r  z	dist,argsc                 t   || j         v rt          j        | d           t          j        dddddt          j        g          }t          j        dddddt          j        g          }t          t          |          }t          t          |j        |d           t          t          |j        |d           d	S )
r	  z  fit known to fail or deprecatedr  r  r  r  r  r    r{  N)fitSkipNonFiniter   r   rj   r   r  r  r.  r9   r`  r  r   )rs   rB   r0  r_   r2  distfuncs         r3   !test_fit_w_non_finite_data_valuesz/TestFitMethod.test_fit_w_non_finite_data_values  s     4(((K4AAABBBHfffffbfEFFHfffffbfEFF5$''j(,!<<<<j(,!<<<<<<r5   c           
         t           j                            d           t          j        d          5  t          j                            dddd          }t          j        t          j        |          t          j        d          z
  d	z  	                                          }t          t          j        t          j                            |d
d                    |d
dgd           d d d            d S # 1 swxY w Y   d S )Ni90  r  r  rl  r         4@r  r;  rg   r   rB  r-  r   )rj   r   r   r  r9   r(  r   r  r  r   r   r   r   )rs   r_   expected_shapes      r3   test_fix_fit_2args_lognormz(TestFitMethod.test_fix_fit_2args_lognorm  s>   
	u[X&&& 	@ 	@!!$Dr!::AWrvayy26"::'=&A%G%G%I%IJJNBHU]%6%6qq%6%L%LMM+Q3$@ @ @ @	@ 	@ 	@ 	@ 	@ 	@ 	@ 	@ 	@ 	@ 	@ 	@ 	@ 	@ 	@ 	@ 	@ 	@s   B;C==DDc                    t          j        dd          }t          j                            |          \  }}t          |d           t          |t          j        d                     t          j                            |d          \  }}t          |d           t          |t          j        d                     t          j                            |d          \  }}t          |d           t          |d           d S )Nr    r3  r   rg   r   r{  )rj   r  r9   r  r   r   r  r   rs   r_   re   rf   s       r3   test_fix_fit_normzTestFitMethod.test_fix_fit_norm  s    IaOOZ^^A&&
UC###E271::...Z^^AA^..
US!UBGAJJ'''Z^^Aa^00
UC###UAr5   c                    t          j        dd          }t          j        |                                          }d}t          j                            ||          \  }}}t          j        |                                          |z
  }t          t          j        |          t          j	        |          z
  |d           t          ||           t          ||                                |z  d           d}d}t          j                            |||          \  }}}t          ||           t          ||           t          ||                                |z  d           d	}d}t          j                            |||          \  }}}t          ||           t          ||           t          ||                                |z  d           d}d	}	t          j                            |||	
          \  }}}t          ||           t          ||	           |t          j        |	          z
  }
t          t          j	        |          |
           d S )Nr    r3  r   r   r   rW  r  r   r   rg   rB  )rj   r  r  r   r9   r  r   r   r   digammar   )rs   r_   meanlogr   r1   re   rf   rO   r   r   r  s              r3   test_fix_fit_gammaz TestFitMethod.test_fix_fit_gamma%  s   IaOO&)).."" 553F16688w&BF1II(:(::AqIIIIS$E16688A:q9999
 bt<<3QS$E16688A:q9999bt<<3QS$E16688A:q9999 VDD3S$UF###bfVnn$GOA..22222r5   c           	         d }t          j        g d          }t          j                            |dd          \  }}}}t          |d           t          |d           t           ||||          ddgd           t          j        g d          }t          j                            |ddd	          \  }}}}t          |d           t          |d           t          |d            ||||          \  }}t          |dd
           d|z
  }	t          j                            |	ddd          \  }
}}}t          |d           t          |d           t          |d            ||
||	          \  }}t          |dd
           t          |
|           t          t          t          j        j        |dd           t          j        g d          }t          t          t          j        j        |dd           t          t          t          j        j        |ddd           t          t          t          j        j        |ddd           t          t          t          j        j        |dddd           d S )Nc                 h   t          |          }t          j        |                                          }t          j        d|z
                                            }t	          j        | |z             }||| t	          j        |           z   z  z
  ||| t	          j        |          z   z  z
  g}|S r   )r  rj   r  rm  r   psi)r1   r2   r_   ru  s1r&  psiabr  s           r3   mlefuncz0TestFitMethod.test_fix_fit_beta.<locals>.mlefuncN  s     AAB!""BKA&&Eufw{1~~566ufw{1~~5668DKr5   )ry  rl  r|   r   r    rB  rv  r   rg   )r   r   r   r;  )r	  r   r   r|   rO  )r   r   r   )r   r   r	  r   )r   r	  r   r   )
rj   r   r9   r  r   r   r   r   r`  r  )rs   r  r_   r1   r2   re   rf   dadbrs  r  b2r  r  r2  s                  r3   test_fix_fit_betazTestFitMethod.test_fix_fit_betaK  s   		 		 		 H'''(( :>>!!A>>>1c5S!UA1a((1a&t<<<< H'''(( :>>!!>DD1c5QS!UAAq!!BAD)))) U$z~~bQQq~IIBfRT1VQR$$BAD))))B""" 	j%*.!#aHHHHHZZZ  j%*.!!AFFFFj%*.!!A!LLLLj%*.!!A!LLLL 	j%*.!aQ	( 	( 	( 	( 	( 	(r5   c                    t          j        g d          }t          j                            |          \  }}t          |d           t          |d           t          j                            |d          \  }}t          |d           t          |d           t          j                            |d          \  }}t          |d           t          |d           d S )N)rg   rg   r   r   r   r   r   r  rg   r   r{  r   r   r   )rj   r   r9   rg  r   r   r  s       r3   test_expon_fitzTestFitMethod.test_expon_fit  s    H---..[__Q''
US!UA[__Qq_11
US!UA[__QQ_//
US!UAr5   c           	         t          j        g d          }t          j        |dz
            }t          j                            |d          \  }}}t          ||                                d           t          |d           t          |t          j	        |
                                          d           t          j                            |dd          \  }}}t          |t          j        |t          j        d          z
  dz  
                                          d           t          |d           t          |d           t          j                            |dd	
          \  }}}t          |d	           t          |d           t          |t          j	        |
                                          d           d S )N)r  r   r`   r?     r  r    r   r  r   r3  rB  rg   r   )r   fix_s)rj   r   r  r9   r(  r   r   r{  r   r   r   r  )rs   r_   lnxm1rB  re   rf   s         r3   test_lognorm_fitzTestFitMethod.test_lognorm_fit  s   H---..q1u!M--aa-88sEuyy{{7777S!rvejjll33%@@@@!M--aa-BBsErw):Q(>'D'D'F'FGG"	$ 	$ 	$ 	$S!UA!M--aat-DDsEUD!!!S!rvejjll33%@@@@@@r5   c                    t          j        g d          }t          j                            |          \  }}t          ||                                           t          |t          j        |                     t          j                            |d          \  }}t          |d           t          ||                                           t          j                            |d          \  }}t          |d           t          |d           t          t          t          j        j        |d           t          t          t          j        j        |d           d S )N)rT  r  ro  r  r   r   r`   r{  r   rU  )rj   r   r9   r  r   r   r  r  r  r`  r  r  s       r3   test_uniform_fitzTestFitMethod.test_uniform_fit  s#   H)))**]&&q))
US!%%''"""UBF1II&&&]&&qq&11
US!UAEEGG$$$]&&q&44
US!UBj%-"3QSAAAAj%-"3QsCCCCCCr5   r  MLEMMc           
      `   d\  }}t           j                            ||dd          }t           j                            |d|          }t           j                            |d|          }t	          ||dd	           t           j                            |d|
          }t	          ||dd	           t           j                            |d|          }t           j                            |d|          }t	          ||dd	           t           j                            |d|          }t	          ||dd	           t          t          t           j        j        |dd|           t          t          t           j        j        |dddd|           t           j                            |ddd|          }	|	\  }
}}}t          |
||gg d           d}t           j                            |d          }t           j                            |||          \  }
}}t          |
|           d S )N)r  r  ra   r   r   r  )r   r  )rg  r  r  rn  )rh  r  r  )r	  r  )r  r  )r  r  r    rg   )rg  r   r  r   r   )rg  r	  r   r   r  )rg  r   r   r  )r  r   r    r;  )	r9   r  r   r   r   r`  r  r   r  )rs   r  r1   r2   r_   res_1res_2res_3res_4res_5aabbr  ssr   s                  r3   test_fshapeszTestFitMethod.test_fshapes  s   
 1JNN1acN==
qR77
qR77u5u====
q6::u5u====
qR77
qR77u5u====
q6::u5u==== 	j%*.!aOOOO 	j%*.!aQv	7 	7 	7 	7
 
qRa&IIBBb"b\:::... {qs++[__Ta_??
BRr5   c                     t           j        }|                    dd          }t          d          }t	          t
          |j        |fi |d|i d S )Nrg   ra   )r  r   i)enikibenikir  )r9   r  r   r  r`  r  r   )rs   r  rB   r   r/  s        r3   test_extra_paramszTestFitMethod.test_extra_params  s]     xx!#x&&t$$$i4FF3FFvFFFFFFr5   N)r   r   r   r   r7  r  r   r   r   r   r  r  r  r  r  r  r  r  r   r  r  r;   r5   r3   r  r    sW       111D   A@@[[(33= = 43=@ @ @  $3 $3 $3L5( 5( 5(n  A A A(D D D$ [[Xt}55$ $ 65 $L [Xt}55G G 65G G Gr5   r  c                   D    e Zd Zd Zd Zd Zd Zd Zd Zd Z	d Z
d	 Zd
S )
TestFrozenc                 D    t           j                            d           d S r5  r6  rx   s    r3   r7  zTestFrozen.setup_method  r8  r5   c                    t           j        }t          j        dd          }|                    d          }|                    ddd          }t          ||           |                    d          }|                    ddd          }t          ||           |                    d          }|                    ddd          }t          ||           |                    d          }|                    ddd          }t          ||           |                    d          }|                    ddd          }t          ||           |                                }|                    dd          }t          ||           |	                                }|	                    dd          }t          ||           |
                                }|
                    dd          }t          ||           |                                }|                    dd          }t          ||           |                                }|                    dd          }t          ||           |                    d          }|                    ddd          }t          ||           t          |j        |j                   t          |j        |j                   d S )Nr  r  rd   r  rl  rg   )r9   r  ri   r   rq   r   r  r   rH  r   rA  r{  rH   r  r1   r2   )rs   rB   r  result_fr  s        r3   r  zTestFrozen.test_norm  s   zC000::d##$D44Xv&&&::d##$D44Xv&&&::d##$D44Xv&&&::d##$D44Xv&&&99T??4s33Xv&&&==??S11Xv&&&;;==t3//Xv&&&::<<d#..Xv&&&::<<d#..Xv&&&>>##$c22Xv&&&==##QD44Xv&&&VXtv&&&VXtv&&&&&r5   c                    d}t           j        }t          j        |          }|                    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 )Nr   r  rl  r  rg   )r9   r  ri   r   rq   r   r  r   rH  r   rA  r{  rH   r  r1   rB   r2   )rs   r1   rB   r  r  r  s         r3   
test_gammazTestFrozen.test_gamma$  sa   {Q::d##$""Xv&&&::d##$""Xv&&&::d##$""Xv&&&::d##$""Xv&&&99T??q!!Xv&&&==??QXv&&&;;==1Xv&&&::<<!Xv&&&::<<!Xv&&&>>##aXv&&&==##Q""Xv&&&VXv{}---VXv{}-----r5   c                     t          j        d          }|                    d          }|                     d           |                    d          }t          ||           d S )Nr    rg   r  r  )r9   r(  r  r   )rs   r  m1r  s       r3   test_regression_ticket_1293z&TestFrozen.test_regression_ticket_1293X  sa    q!!]]1V$$$ ]]1 	Rr5   c                 l   d}t          j        |          }|j                            |          \  }}t	          ||gddg           d}t           j                            d|           t	          |j                            |          dt          j        g           d}t          j        |          }|j                            |          \  }}t	          ||gddg           d}t           j                            d|           t	          |j        j        |j        j	        ft           j                            |                     t          j        d          }t          |j        |j        u           dD ]}t          j        |          }t          j        |          }|j        |j	        }}t	          |d           t          t          j        |                     t          j        d          }t           j                            |          \  }}t          ||gdd	g           d S )
Nr!  r  r   r  r]   r   r  r  r|   )r9   r  rB   r  r   ri   rj   r  r1   r2   r	   r   r  r   )rs   r  r  r1   r2   rv1s         r3   r  zTestFrozen.test_abg  s    _q!!!w##A&&1aVb#Y'''A###RW))!,,q"&k:::_q!!!w##A&&1aVb#Y'''Aq!!!bgi+U_-I-I!-L-LMMMo$$$'(((  
	/ 
	/A
1A1%%%B4qABBKNN### 
3A?//22DAqQFRI....
	/ 
	/r5   c                 T    t          t          t          j        d                     d S )N	rv_frozen)r	   r0   r9   r:   rx   s    r3   test_rv_frozen_in_namespacez&TestFrozen.test_rv_frozen_in_namespace  s#    +[99:::::r5   c                    t          j                    }t          t          |d                     d|_        t          |j                                        t          j        	                    d                                                     t          j        	                    d          }|
                    d|           d S )Nr   *   r   r  r   )r9   r  r	   r0   r   r   	get_staterj   r   r  r   )rs   r  rndms      r3   test_random_statezTestFrozen.test_random_state  s    //000 !V(2244Y**2..88::	< 	< 	< y$$T**


-----r5   c                    t          j        dd          }t          j        d          }t          j        g dg df          }|||fD ]}d|_        |                    d	           t          j        |          }|                    d	          }t          j        |          }|                    d	          }t          ||           |
                    d
          |
                    d
          g}	t          |	d         |	d                    t          |                    |	d                   |                    |	d                              d S )Ngdsz@g@e?r  )r   r    rg   r   rG  r^  r   r  r;  r|   r   r    )r9   r  r  rE   r   r   pickledumpsloadsr   r   rq   )
rs   r  poissrW  distfnrO   r0	unpickledr1medianss
             r3   test_picklingzTestFrozen.test_pickling  s[    z,.ABBb!!"<<<+?+?+?+A B B B UF+ 	4 	4F"&FJJAJV$$A##BQIA&&BR    zz#	c(:(:;GWQZ000GAJ//"wqz224 4 4 4	4 	4r5   c           
         d }t          j        ddd          }t          j        dd          5  |                    |ddd	
          }t           j                            |dddddd	          }d d d            n# 1 swxY w Y   t          ||           t          j        dd          }|                    |          }t           j                            |dd          }t          ||           d S )Nc                     | S rv   r;   r   s    r3   r  z$TestFrozen.test_expect.<locals>.func  s    Hr5   rg   r   r   r  r  )r  divider    T)r   ubconditionalrg   r0  re   rf   r   r	  r	  r  )r   r0  re   )r9   r  rj   r  r   r   r  )rs   r  gmgm_val	gamma_valrv  p_valpoisson_vals           r3   test_expectzTestFrozen.test_expect  sQ   	 	 	 [1!1---[(;;; 	I 	IYYtaTYBBF**4d./A4 + I II	I 	I 	I 	I 	I 	I 	I 	I 	I 	I 	I 	I 	I 	I 	I 		***M!###m**4d*BB{+++++s   A A==BBN)r   r   r   r7  r  r  r  r  r	  r		  r	  r"	  r;   r5   r3   r  r    s          1' 1' 1'f2. 2. 2.h  */ */ */X; ; ;. . .4 4 40, , , , ,r5   r  c                   b    e Zd Zd Zd Zd Zd Zd Zd Zd Z	d Z
d	 Zd
 Zd Zd Zd Zd ZdS )
TestExpectc                 @   t           j                            d dd          }t          |dd           t           j                            d dd          }t          |dd           t           j                            d	dd          }t           j                            d
dd          }t           j                            d dd||          }t          |dd           t           j                            d dd||d          }t          |dd           d S )Nc                     | dz
  | dz
  z  S )Nr   r;   r   s    r3   rP   z&TestExpect.test_norm.<locals>.<lambda>  s    1qs r5   r   rg   rd   r   r.  rW  c                     | S rv   r;   r   s    r3   rP   z&TestExpect.test_norm.<locals>.<lambda>  s     r5   rF  r@  c                     dS r   r;   r   s    r3   rP   z&TestExpect.test_norm.<locals>.<lambda>      Q r5   re   rf   r   r	  r   c                     dS r   r;   r   s    r3   rP   z&TestExpect.test_norm.<locals>.<lambda>      a r5   Tre   rf   r   r	  r	  rT  )r9   r  r   r   r   )rs   r  r  r   r	  prob90prob90cs          r3   r  zTestExpect.test_norm  s   J33!DDAq"----Jmm!<<Aq"----Z^^Daq^11Z^^Daq^11"";;AQ2""MMFC4444*##KKQaB204 $ 6 6GR444444r5   c           	      R   t           j                            d ddd          }t          |dd           t           j                            d	 dd
d          }t          |dd           t           j                            ddddd          }t           j                            ddddd          }t           j                            d dd
d||d          }t          |dd           t           j                            d ddd||d          }t          |dd           d S )Nc                     | dz
  | dz
  z  S )NUUUUUU@r;   r   s    r3   rP   z&TestExpect.test_beta.<locals>.<lambda>  s    51U7(; r5   r  r   rg   )r0  re   rf   qq?r  rW  c                     | S rv   r;   r   s    r3   rP   z&TestExpect.test_beta.<locals>.<lambda>  s     r5   rU  r   r2	  r@  r`   rd   rF  c                     dS )NrT  r;   r   s    r3   rP   z&TestExpect.test_beta.<locals>.<lambda>  s    R r5   )r`   r`   Fr	  r   c                     dS r   r;   r   s    r3   rP   z&TestExpect.test_beta.<locals>.<lambda>  r,	  r5   TrT  )r9   r  r   r   r   )rs   r  r  r	  r   r.	  r/	  s          r3   	test_betazTestExpect.test_beta  s=   J;;'"#1  . .Avr2222JkkRrJJAub1111Z^^D"baq^99Z^^D"baq^99""<<hB)+ru # N NFC4444*##KKhA*+rt $ M MGR444444r5   c                 T   t           j                             dddd          \  }}t           j                            d dd          }t          ||d	
           t           j                            d dd          }t          ||d
           t           j                            d dddd	          }t          ||d
           dt           j                            dd	gdddd                                          z
  }t           j                            d dddd          }t          ||d	
           t           j                            d ddddd          }t          |dd
           t           j                            d ddd          }	t          |	dd	
           d S )Nr  r`   r  rU  r  c                     | S rv   r;   r   s    r3   rP   z+TestExpect.test_hypergeom.<locals>.<lambda>  r)	  r5   )r  r`   r  r	  r  rW  c                     | dz
  dz  S Nr  rg   r;   r   s    r3   rP   z+TestExpect.test_hypergeom.<locals>.<lambda>   s    adQY r5   r.  c                     | dz
  dz  S r;	  r;   r   s    r3   rP   z+TestExpect.test_hypergeom.<locals>.<lambda>  s    QrTAI r5   r   )r0  re   r   r	  r    c                     dS r   r;   r   s    r3   rP   z+TestExpect.test_hypergeom.<locals>.<lambda>  s    q r5   r3  rV  c                     dS r   r;   r   s    r3   rP   z+TestExpect.test_hypergeom.<locals>.<lambda>  s    1 r5   T)r0  re   r   r	  r	  c                     dS r   r;   r   s    r3   rP   z+TestExpect.test_hypergeom.<locals>.<lambda>  s    ! r5   r   )r0  r   r	  )r9   r  r   r   rP  rm  )
rs   m_truev_truer  r  v_bounds	prob_trueprob_boundsprob_bcprob_bs
             r3   test_hypergeomzTestExpect.test_hypergeom  s    ..r2qb.AAO"";;[b"IIAvr2222O""#6#6[') # + +Avr2222 ?))*=*=/:.0Q2 * ? ? 	Hfb9999 eo))1b'2r1!)DDHHJJJ	o,,[[{13b - B BKB???? /((;B,-"$ ) H HGQ3333 ''++, ( 4 4FAr222222r5   c                 $   t           j                            d ddd          }dt           j                            dd          z
  }t	          ||d	           t           j                            d
 ddd          }t	          |dd	           d S )Nc                     dS r   r;   r   s    r3   rP   z)TestExpect.test_poisson.<locals>.<lambda>  s    Q r5   r	  r   F)r0  r   r	  r    rg   r.  rW  c                     dS r   r;   r   s    r3   rP   z)TestExpect.test_poisson.<locals>.<lambda>!  s     r5   T)r9   r  r   rq   r   )rs   rD	  prob_b_trueprob_lbs       r3   test_poissonzTestExpect.test_poisson  s    m**;;Ta7< + > >))!Q///KbAAAA-&&{{!37 ' 9 9GQ333333r5   c                     t           j        }|                    d          }|                    d           |                    d          }t          ||d           d S )N)r  r[  )r|   r.  rW  )r9   genhalflogisticr   r   )rs   halflogr/  r3  s       r3   test_genhalflogisticzTestExpect.test_genhalflogistic%  sa     '~~6~**F###~~6~**D$333333r5   c                 
   t          t          j        t          j                            dd                               t          t          j        t          j                            d d                               t          t          j        t          j                            d d                               t          t          j        t          j                            d d                               d S )Nr|  Gz?c                     dS r   r;   r   s    r3   rP   z/TestExpect.test_rice_overflow.<locals>.<lambda>4       r5   )rS	  r[  c                     dS r  r;   r   s    r3   rP   z/TestExpect.test_rice_overflow.<locals>.<lambda>5  rU	  r5   c                     dS Nr   r;   r   s    r3   rP   z/TestExpect.test_rice_overflow.<locals>.<lambda>6  rU	  r5   )r	   rj   rG  r9   riceri   r   rx   s    r3   test_rice_overflowzTestExpect.test_rice_overflow/  s     	EJNN35566777EJ--kk-HHIIJJJEJ--kk-HHIIJJJEJ--kk-HHIIJJJJJr5   c                 (   d\  }}t           j                            d |f          }t          |||dz
  z  t	          j        d|z
            z  d           t           j                            d |f|          }t          |||z   d           d S )	N)r~  r   c                     | S rv   r;   r\   s    r3   rP   z(TestExpect.test_logser.<locals>.<lambda>;  r,	  r5   r[  rT  r   r   c                     | S rv   r;   r]	  s    r3   rP   z(TestExpect.test_logser.<locals>.<lambda>A  r,	  r5   r	  )r9   r  r   r   rj   r  )rs   rv  re   res_0res_ls        r3   test_logserzTestExpect.test_logser8  s    3##KKqd#;;QVrvb1f~~5E	C 	C 	C 	C ##KKqd#DDus{777777r5   c                     d\  }}t           j                            d ||f          }t           j                            d ||f          }t          |||z
  d           t          ||dz  z
  ||z   d           d S )N)r  r$  c                     | S rv   r;   r   s    r3   rP   z)TestExpect.test_skellam.<locals>.<lambda>H      A r5   r[  c                     | dz  S r  r;   r   s    r3   rP   z)TestExpect.test_skellam.<locals>.<lambda>I  r  r5   r  r   rg   )r9   rD  r   r   )rs   r  p2r  r  s        r3   test_skellamzTestExpect.test_skellamD  s     B]!!++RH!==]!!..Bx!@@BG%0000RU
BG%888888r5   c           
          d\  }}t           j                            d ||f          }t          |t	          d t          ||          D                       ||z
  z  d           d S )N)r   q   c                     | S rv   r;   r   s    r3   rP   z)TestExpect.test_randint.<locals>.<lambda>Q  r)	  r5   c              3      K   | ]}|V  d S rv   r;   rA   r   s     r3   r  z*TestExpect.test_randint.<locals>.<genexpr>S  s"      55!A555555r5   r   r   )r9   rA  r   r   rm  r  )rs   r0  r1  r   s       r3   test_randintzTestExpect.test_randintM  s     Bm"";;R9955uR}}55555bA	O 	O 	O 	O 	O 	Or5   c                 T    t          t          t          j        j        d d           d S )Nc                     | dz  S r  r;   r   s    r3   rP   z&TestExpect.test_zipf.<locals>.<lambda>X  s
    q!t r5   r	  )r
   rz  r9   rr  r   rx   s    r3   	test_zipfzTestExpect.test_zipfU  s.    ^UZ%6#^T	+ 	+ 	+ 	+ 	+r5   c                     t           j                            d d          }t           j                            d dddd          }t          ||d	
           d S )Nc                     dS r   r;   r   s    r3   rP   z/TestExpect.test_discrete_kwds.<locals>.<lambda>\  rd	  r5   r	  r[  c                     dS r   r;   r   s    r3   rP   z/TestExpect.test_discrete_kwds.<locals>.<lambda>]  rd	  r5   r      r-  )r0  maxcount	chunksize	tolerancer.  rW  )r9   r  r   r   )rs   n0n1s      r3   test_discrete_kwdszTestExpect.test_discrete_kwdsZ  sd    ]!!++D!99]!!++D+/2 " O OBB//////r5   c                     d }dD ]=}t           j                            d|          }t          | ||          d           >d S )Nc                 H    | dz  d| dz  z  z   d| dz  z  z   d| dz  z  z   | z   S )Nr   r`   r   r  r   r?  rg   r;   r  s    r3   poiss_moment5z-TestExpect.test_moment.<locals>.poiss_moment5d  s8    q52b!e8#bQh.BE9B>>r5   )r   r  r   r  r   )r9   r  r  r   )rs   r}	  r^  m5s       r3   test_momentzTestExpect.test_momenta  se    	? 	? 	?  	? 	?B%%a,,BBb 1 1>>>>>	? 	?r5   c                    t          t          j                            dd          d           t          t          j                            dd          d           t          t          j                            dd          d           t          t          j                            d          d	           t          t          j                            d
          d
           d S )NrL  rT  rd   r}  r`   r]   )   r[  r	  U   r  )r   r9   r  r   r  rb  rx   s    r3   test_challenging_cases_gh8928z(TestExpect.test_challenging_cases_gh8928k  s     	
))b)<<bAAA
))b)<<bAAA
))b)<<bAAA***77===--"-55r:::::r5   c           	      t   t           j        }|                    dd          }t          |                    dd          |           t          |                    dddd          |           t          |                    dddd          |d	z             t          |                    ddddd
          |           t          |                    dddd          d           t          |                    dddd          d           t          |                    ddddd
          d           d S )Nr`   r   rd   r  ru  r*	  r  r.  r  Tr-	  r  g433333@g433333rV  )r9   r  r   r   r   )rs   rB   r  s      r3   test_lb_ub_gh15855zTestExpect.test_lb_ub_gh15855t  sH   }iiBai((!44c:::!bAA3GGG!rBBCGLLL!r04 $ 6 67:	< 	< 	< 	!rBBFKKK!rBBGLLL!r04 $ 6 679	; 	; 	; 	; 	;r5   N)r   r   r   r  r7	  rG	  rM	  rQ	  rZ	  ra	  rg	  rm	  rp	  rz	  r	  r	  r	  r;   r5   r3   r$	  r$	    s        
5 5 5 5 5 5& 3  3  3D	4 	4 	44 4 4K K K
8 
8 
89 9 9O O O+ + +
0 0 0? ? ?; ; ;; ; ; ; ;r5   r$	  c                   ~    e Zd Zd Zd Zd Zd Zd Zd Ze	j
                            ddd	g          d
             ZdS )TestNctc                     t          j        dd          }t          |                    d          d           t          j        dd          }t	          |                    d          dd           d S )Nr   r   r|   r   g K?r`   rW  )r9   nctr   rq   r   r  s     r3   test_nc_parameterzTestNct.test_nc_parameter  sc     Yq!__RVVAYY$$$Yq"BFF1II~rBBBBBBr5   c           	          t           j                            dt          j        dd          d d d f         t          j        ddd                    }t          g dg dg dg          }t          ||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;  r   )r9   r	  ri   rj   r  rY  r   r   )rs   r   rX   s      r3   test_broadcastingzTestNct.test_broadcasting  s    immAryAqqq$w7KQ224 4JJJJJJJJJL M M 	XD111111r5   c                 t    t          j        dd          }t          |                                d           d S )Nr   r   r   )r9   r	  r   rA  r  s     r3   test_variance_gh_issue_2401z#TestNct.test_variance_gh_issue_2401  s1    
 Yq!__RVVXXs#####r5   c                    t           j                             ddd          \  }}}}t          ||||gt          j        t          j        t          j        t          j        g           t           j                             ddd          \  }}}}t          t          j        |                     t          |||gt          j        t          j        t          j        g           t           j                             ddd          \  }}}}t          t          j        |||g                                                     t          |t          j                   d S )Nr   r~  r  )r  ncr  r&  r  )r9   r	  r   rj   r  r	   rG  r   r  s        r3   test_nct_inf_momentszTestNct.test_nct_inf_moments  s	   Y__V_DD
1aaAq\BFBFBFBF#CDDDY__V_DD
1aAaAY 8999Y__V_DD
1aQ1I&&**,,---Qr5   c                    t           j                            dd          }t           j                             dd          }ddg}t          ||d         d           t          ||d           t           j                            dd          }t           j                             dd          }d	d
g}t          ||d         d           t          ||d           d S )Nr   rg   g@ @g*dn?r   r  r   r   gqI  @gL) ?r  )r9   r	  r   r   )rs   nct_mean_df_1000nct_stats_df_1000expected_stats_df_1000nct_mean	nct_statsr  s          r3   test_nct_stats_large_df_valuesz&TestNct.test_nct_stats_large_df_values  s     !9>>$22!IOOD!44
 #56H!I(*@*C"	$ 	$ 	$ 	$)+A"	$ 	$ 	$ 	$ 9>>&!,,IOOFA..	,.@A."3%@@@@	>======r5   c           	      ~    t          t          j                            ddt	          d                    d           d S )Nrg   i   r   )r   r9   r	  rq   r   rx   s    r3   test_cdf_large_nczTestNct.test_cdf_large_nc  s0    	aE%LL991=====r5   zx, df, nc, expected)r  r`   ru  gAs69)rm  r  ru  gV1c                 h    t          t          j                            |||          |d           d S r  )r   r9   r	  ri   )rs   r_   r  r	  rX   s        r3   test_pdf_large_nczTestNct.test_pdf_large_nc  s1     		aR00(GGGGGGr5   N)r   r   r   r	  r	  r	  r	  r	  r	  r   r   r   r	  r;   r5   r3   r	  r	    s        C C C2 2 2$ $ $     > > >,> > >6 [2.+5
  H H	 H H Hr5   r	  c                   &    e Zd Zd Zd Zd Zd ZdS )TestRecipInvGaussc                 V    t           j                            dd          }|dk    sJ d S )Nr   r  r   )r9   recipinvgaussri   r  s     r3   test_pdf_endpointz#TestRecipInvGauss.test_pdf_endpoint  s+    ##As++Cxxxxxxr5   c                 l    t           j                            dd          }|t          j         k    sJ d S )Nr   r  )r9   r	  r   rj   r  rk  s     r3   test_logpdf_endpointz&TestRecipInvGauss.test_logpdf_endpoint  s/    "))!S11wr5   c                 n    t           j                            dd          }d}t          ||d           d S )NrF  r|   g9V9s;r  r   )r9   r	  rq   r   rs   rv  rX   s      r3   test_cdf_small_xz"TestRecipInvGauss.test_cdf_small_x  s<      ##D#..(8%000000r5   c                 l    t           j                            dd          }d}t          ||d           d S )Nr  r|   glH<r  )r9   r	  r   r   r	  s      r3   test_sf_large_xz!TestRecipInvGauss.test_sf_large_x  s7    ""2s++(8U+++++r5   N)r   r   r   r	  r	  r	  r	  r;   r5   r3   r	  r	    sP            1 1 1(, , , , ,r5   r	  c                        e Zd Zd Zd Zd ZdS )TestRicec                    g d}t          t          j        t          j                            |d                                                               t          t          j        t          j                            |d                                                               t          t          j        t          j                            |d                                                               t          t          j        t          j        	                    |d                                                               g d}t          t          j        t          j        
                    |d                                                               t          j                            dd          }t          t          j        |                                                     d}t          t          j                            |d          t          j                            ||          |d	           d S )
N)rQ  rT  rU  r   r2   )r]   r]   r|   r   r   r  r  r-  rn  )r	   rj   rG  r9   rY	  ri   r   r   rq   r  r   r   )rs   r_   r  r  r2   s        r3   test_rice_zero_bzTestRice.test_rice_zero_b  s   MMEJNN1N334488::;;;EJ--a2-6677;;==>>>EJNN1N334488::;;;EJ--a2-6677;;==>>>   EJNN1N334488::;;;z622D!!%%''(((
 
q!,,ejnnQ.B.BQ	( 	( 	( 	( 	( 	(r5   c                     t           j        j        }t           |d          j        d           t           |dd          j        d           d S )Nr  r	  r    )r   r   )r2   r   )r9   rY	  r   r   r   rB  r  s     r3   test_rice_rvszTestRice.test_rice_rvs%  sO    jnSS2YYY^Q'''SS2F+++16:::::r5   c                    t           j                            t          j        ddd          t          j        ddd                    }g d}t          ||           t          j        ddd          }t           j                            |dd          }g d	}t          ||           t          j         j                            d
t          j        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@r|      )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@)r9   rY	  rq   rj   r  r   r   r8   )rs   rq   cdf_expprobabilitiesr   ppf_exps         r3   test_rice_gh9836zTestRice.test_rice_gh9836*  s     jnnRYr333RYr35K5KLLO O O
 	W%%%	#q#..jnn]En;;L L L 	W%%%k""3	"c2(>(>??; ; ;
 	W%%%%%r5   N)r   r   r   r	  r	  r	  r;   r5   r3   r	  r	    sA        ( ( (*; ; ;
'& '& '& '& '&r5   r	  c                        e Zd Zd Zd Zd ZdS )
TestErlangc                 D    t           j                            d           d S r5  r6  rx   s    r3   r7  zTestErlang.setup_methodU  r8  r5   c           	         t          j                    5  t          j        dt                     t	          t          t
          j        j        dddd           g d}t
          j                            |d          }t
          j	                            |d          }t          ||d	
           d d d            d S # 1 swxY w Y   d S )Nrs  r  r   r    r   r  )r|   rT  r   r  r  r  r   )rw  rx  ry  rz  r`  r9   erlangr   r   r  r   )rs   r   result_erlangresult_gammas       r3   test_erlang_runtimewarningz%TestErlang.test_erlang_runtimewarningX  s     $&& 	D 	D!'>::: .,*CQaaI I I I (''D!L,,Ta,88M ;??4A?66LM<dCCCC	D 	D 	D 	D 	D 	D 	D 	D 	D 	D 	D 	D 	D 	D 	D 	D 	D 	Ds   BB==CCc                     t          t          j                            dddg          t          j                            dddg                     d S )Nr|   r    r   ru  )r   r9   r	  ri   r  rx   s    r3   test_gh_pr_10949_argcheckz$TestErlang.test_gh_pr_10949_argcheckk  sR    U\%%caW%55[__SQG_44	6 	6 	6 	6 	6r5   N)r   r   r   r7  r	  r	  r;   r5   r3   r	  r	  T  sD          D D D&6 6 6 6 6r5   r	  c                       e Zd Zd Zd Zd Zej                            dddg          d             Z	ej                            ddd	gd
dgg          d             Z
d Zd ZdS )TestRayleighc                 D    t           j                            d           d S )Nr  r6  rx   s    r3   r7  zTestRayleigh.setup_methodq  s    
	y!!!!!r5   c                 d    t           j                            d          }t          |d           d S )Nr=  g)Zx)r9   r+  r   r   rs   r2  s     r3   r  zTestRayleigh.test_logpdfu  s.    N!!"%%./////r5   c                 d    t           j                            d          }t          |d           d S )Nr=  i)r9   r+  r  r   r	  s     r3   r  zTestRayleigh.test_logsfy  s-    N  $$5!!!!!r5   r  )gB)Q?got|?)gS2P?g3?c                    t           j                            d||          }d } |||          }t           j                            ||          \  }}t	          ||           t	          ||           t           j                            |d          \  }}t	          |d           t           j                            |          \  }}t	          | |||                     d S )Nr"  rn  c                 b    t          j        | |z
  dz            dt          |           z  z  dz  S )Nrg   r|   )rj   rm  r  )r   r   s     r3   r  z(TestRayleigh.test_fit.<locals>.scale_mle  s/    FD4KA-..!c$ii-@RGGr5   r   r  r{  )r9   r+  r   r   r   )rs   r   r<  r   r  scale_expectre   rf   s           r3   ry  zTestRayleigh.test_fit}  s     ~!!sy!II	H 	H 	H !yw//^''7';;
US'"""UL))) ^''R'88
UUB
 ^''--
UUIIdC0011111r5   rS	  r~   g̡=E?g@1d?c                 ~    t           j                            d||          }t          t           j        |           d S )Nr"  rn  )r9   r+  r   r   )rs   r   r<  r   s       r3    test_fit_comparison_super_methodz-TestRayleigh.test_fit_comparison_super_method  s8    
 ~!!sy!II%end;;;;;r5   c                 8    t          t          j                   d S rv   )r  r9   r+  rx   s    r3   r  zTestRayleigh.test_fit_warnings  s    EN+++++r5   c                    t           j                            d          }d\  }}}t          j                            ||||          }t          j                            |          \  }}|t          j        |          k     sJ t          j                            ||          \  }}|t          j        |          k     sJ ||k    sJ d S )Ni  )r=  r  r  r   r{  )rj   r   r   r9   r+  r   r   r  )	rs   r   re   rf   r   r   r   r   r   s	            r3   test_fit_gh17088zTestRayleigh.test_fit_gh17088  s     i##C(('UDn  e$S II^'',,
$$$$"^//E/BB$$$$E!!!!!!r5   N)r   r   r   r7  r  r  r   r   r   ry  r	  r  r	  r;   r5   r3   r	  r	  p  s        " " "0 0 0" " " [03K3K3M N N2 2N N2. [0D$<4>
3K3M N N< <N N<, , ,
" 
" 
" 
" 
"r5   r	  c                   P   e Zd Zd Zd Zd Zej                            dg d          d             Z	ej                            dg d          d	             Z
ej                            dd
dg          d             Zej                            dddg          d             ZdS )TestExponWeibc                     d}d}d}t           j                            |||          }t           j                            |||          }t	          ||gddg           d S )Nr]   rT  r  g:23W+g~`9k)r9   	exponweibri   r   r   )rs   r_   r1   r  rv  r  s         r3   test_pdf_logpdfzTestExponWeib.test_pdf_logpdf  ss    O1a((%%aA..D	/1DE	G 	G 	G 	G 	Gr5   c                 ~   t          j        ddd          }d}d}t          j                            |||          }t          j                            ||          }t          ||           t          j                            |||          }t          j                            ||          }t          ||           d S )Nr  r   r   r    ra   )rj   r  r9   r	  ri   weibull_minr   r   rs   r_   r1   r  rv  rX   r  s          r3   test_a_is_1zTestExponWeib.test_a_is_1  s     KB""O1a(($((A..8$$$%%aA..$++Aq11h'''''r5   c                 z   t          j        ddd          }d}d}t          j                            |||          }t          j                            |          }t          ||           t          j                            |||          }t          j                            |          }t          ||           d S )Nr  r    r`   )rj   r  r9   r	  ri   rg  r   r   r	  s          r3   test_a_is_1_c_is_1z TestExponWeib.test_a_is_1_c_is_1  s    KAr""O1a((;??1%%8$$$%%aA..;%%a((h'''''r5   zx, a, c, ref))r    r  r   g=J ?)r=  r  r   g$+PR>)r!  r  r   g><)r"  r  r   ggX.X:)r"        ?r   gRy9)r   r	  r  g'vR0=)g_QPr  r  gQ:<c                 l    t           j                            |||          }t          ||d           d S r#  )r9   r	  r   r   )rs   r_   r1   r  r  r   s         r3   r  zTestExponWeib.test_sf  s8     _1a((Ce,,,,,,r5   zp, a, c, ref))rl  r  r   g;y$(@)gMg<r  r   gdbq^@)r  r    rg   gE@)gvIh%L=r	  r   g/7@)rR  r  r  gTWTOc                 l    t           j                            |||          }t          ||d           d S rT  )r9   r	  r  r   )rs   rv  r1   r  r  r  s         r3   r  zTestExponWeib.test_isf	  s8     o!!!Q**Su------r5   )rg   r   r  g)r   r|   r   gl̇կc                 l    t           j                            |||          }t          ||d           d S r  )r9   r	  r  r   )rs   r_   r1   r  r  r  s         r3   r  zTestExponWeib.test_logcdf  s8     ''1a00%000000r5   )gR]jp2r  r  gr)rY  rg   r`   gN~c                 l    t           j                            |||          }t          ||d           d S r  )r9   r	  r  r   )rs   r_   r1   r  r  r  s         r3   r  zTestExponWeib.test_logsf  s8     %%aA..s//////r5   N)r   r   r   r	  r	  r	  r   r   r   r  r  r  r  r;   r5   r3   r	  r	    sL       	G 	G 	G( ( ( ( ( (> [	5 	5 	5	 	- -	 	-* [	4 	4 	4 . . .
 [^AGIJ J1 1J J1
 [^IFHI I0 0I I0 0 0r5   r	  c                       e Zd Zd Zd ZdS )TestFatigueLifec                 j    t           j                            dd          }t          |dd           d S )Nr{  r  X9r   r   )r9   r!  r   r   r  s     r3   test_sf_tailzTestFatigueLife.test_sf_tail(  s8       ,,0u======r5   c                 n    d}t           j                            |d          }t          |dd           d S )Nr	  r  r{  r   r   )r9   r!  r  r   )rs   rv  r  s      r3   test_isf_tailzTestFatigueLife.test_isf_tail6  s:    !!!!S))5u------r5   N)r   r   r   r	  r	  r;   r5   r3   r	  r	  &  s2        > > >. . . . .r5   r	  c                       e Zd Zd Zd Zej                            dddg          d             Zd Z	ej                            dd	d
g          d             Z
dS )TestWeibullc                 f    t           j                            dd          }t          |d           d S r  )r9   r	  r   r   r	  s     r3   r  zTestWeibull.test_logpdf?  s0    $$Q**Qr5   c                    d}d}d}t           j                            |||          }t          |t	          j        d          dz             t           j                            |||          }t          |dt	          j        d          z
             t           j                            |||          }t          |t          j
        d                      t           j                            |||          }t          |t	          j        t          j
        d                                t           j                            |||          }t          |t	          j        d                     t           j                            |||          }	t          |	d           t           j                            ddd          }t          |t	          j        d	                     t           j                            ddd          }	t          |	d	           d
}t           j                            |||          }t          |t	          j        d          dz             t           j                            |||          }t          |dt	          j        d          z
             t           j                            |||          }t          |t	          j        d                     t           j                            |||          }t          |d           t           j                            |||          }t          |t          j
        d                      t           j                            |||          }	t          |	t	          j        t          j
        d                                t           j                            ddd          }t          |t          j
        d                      t           j                            ddd          }	t          |	t	          j        t          j
        d                                d S )Nr  r   r  r  r  r   r:  rg   r  rX  g&.gCe )r9   r	  ri   r   rj   r   r   r  rq   r   r  r  r   r  weibull_max)
rs   r_   r1   r2   rv  r7  r  lcrO   r  s
             r3   test_with_maxima_distribz$TestWeibull.test_with_maxima_distribD  s   .  !!!Qa!0026%==?+++%%a!%44EBF1II-...!!!Qa!00GM%000111%%a!%44BFGM%$8$8#899:::  AQ //26%==)))$$Q$33E"""   Qa 0026$<<((($$R!$44D!!! !!!Qa!0026%==?+++%%a!%44EBF1II-...!!!Qa!0026%==)))%%a!%44E"""  AQ //GM%000111$$Q$33BFGM%$8$8#899:::    33GM*@AAABBB$$UAQ$77BFGM2H$I$I#IJJKKKKKr5   rf   rT  r]   c                 x    t           j                            |dz  |dz  d|          }t          |d           d S )Nr  r  r   r  g>)r9   r	  r  r   r  s      r3   r  zTestWeibull.test_delta_cdf  sI    " !,,U3Ya38 - : :566666r5   c                    t           j                            d          }d\  }}}t          j        |||          }|                    d|          }t          j                            |dd          \  }}}	t          j                            |dd          \  }
}}||cxk    rdk    sn J ||
k    sJ t          j                            |ddd	
          \  }}}|dk    sJ t          j        |||          }|                    d          }t          j        |          t          j        |          f}t          ||           d S )Nl   > )rg   r;  r|   ra   r   r  r   r   r  r  r  r  r  )
rj   r   r   r9   r	  r   r   r   rB  r   )rs   r   r  re   rf   rB   r   c2r  r  c3r  r  c4r  r  r  r   r  s                      r3   test_fit_minzTestWeibull.test_fit_min  sT   i##$788#3 C//hhCch22 !,00c0BBD& ,00c0BBD&t    q      Rxxxx !,00a$0OOD&{{{{!"dF33kk$k''gcllEJsOO+S!!!!!r5   r,  )r=  r    r  )r   r   gq뮕qH)c                     t          t          j                            ||          |d           t          t          j                            ||          |d           d S rT  )r   r9   r	  r   r  r0  s       r3   r  zTestWeibull.test_sf_isf  sZ     	),,Q22CeDDDD)--c155quEEEEEEr5   N)r   r   r   r  r	  r   r   r   r  r	  r  r;   r5   r3   r	  r	  =  s          
RL RL RLh [WsCj117 7 217(" " "< [[+J,D+E F FF FF FF F Fr5   r	  c                       e Zd Zd Zd ZdS )TestDweibullc                 B   t           j                            d          }d|                    dd          z  }t          j                            |          }t          j                            |          t          j        d          z
  }t          ||d           d S )N   q>~). r`   ra   r	  r|   r   r   )
rj   r   r   r  r9   dweibullrH   r	  r  r   )rs   r   r  r   r  s        r3   rq  zTestDweibull.test_entropy  s     i##$788

2
...n$$Q''''**RVC[[8Su------r5   c                 P   t           j                            d          }d|                    dd          z  }d|                                z  }t
          j                            ||          }dt
          j                            ||          z  }t          ||d           d S )Nr	  r`   r    r	  r|   r   r   )
rj   r   r   r  r  r9   r	  r   r	  r   )rs   r   r  r_   r   r  s         r3   r  zTestDweibull.test_sf  s     i##$788


,,,n1%%E%((A...Su------r5   N)r   r   r   rq  r  r;   r5   r3   r	  r	    s2        . . .. . . . .r5   r	  c                   D    e Zd Zd Zd Zd Zd Zd Zd Zd Z	d Z
d	 Zd
S )TestTruncWeibullc                 r    t           j                            ddgddd          }t          |ddg           d S )Nr]   r   g)\(?gףp=
?r   )r9   truncweibull_minri   r   r	  s     r3   test_pdf_boundsz TestTruncWeibull.test_pdf_bounds  s<    "&&Sz3dCCQc
#####r5   c                     t           j                            dddt          j                  }t          |d           t           j                            dddd          }t          |d           d S )Nr   rT  r   r  g~?)r9   r	  r   rj   r  r   r   r	  s     r3   r  zTestTruncWeibull.test_logpdf  se    "))#sC@@Q "))#sC==./////r5   c                 r    t           j                            ddgddd          }t          |ddg           d S )Nr   rT  r   r]   )r9   r	  r   r   r	  s     r3   test_ppf_boundsz TestTruncWeibull.test_ppf_bounds  s<    "&&Sz3SAAQc
#####r5   c                     g d}t           j                            |ddd          }t           j                            |ddd          }t	          ||           d S N)r   r]   rl  r|   r   r   rT  r   r   r  )r9   r	  r   rq   r   rs   r  r_   q_outs       r3   test_cdf_to_ppfz TestTruncWeibull.test_cdf_to_ppf   s[    000"&&q"b"55&**1b"b995!!!!!r5   c                     g d}t           j                            |ddd          }t           j                            |ddd          }t	          ||           d S r
  )r9   r	  r  r   r   r
  s       r3   test_sf_to_isfzTestTruncWeibull.test_sf_to_isf  s[    000"&&q"b"55&))!RR885!!!!!r5   c                    dddfdt           j                            d          }t          |d           t           j                            d          }t	          fd          \  }}t          ||           t           j                            d          }t	          fd	          \  }}t          ||           t           j                            d
          }t	          fd          \  }}t          ||           t           j                            d          }	t	          fd          \  }
}t          |	|
           d S )Nr   rT  r  c                 T    | |z  t           j                            |           z  S rv   )r9   r	  ri   )r_   ru  r1   r2   r  s     r3   xnpdfz)TestTruncWeibull.test_munp.<locals>.xnpdf  s)    a4.221aA>>>>r5   r   r    c                      | d          S r   r;   r_   r
  s    r3   rP   z,TestTruncWeibull.test_munp.<locals>.<lambda>      a r5   rg   c                      | d          S r  r;   r
  s    r3   rP   z,TestTruncWeibull.test_munp.<locals>.<lambda>  r
  r5   r   c                      | d          S rX	  r;   r
  s    r3   rP   z,TestTruncWeibull.test_munp.<locals>.<lambda>   r
  r5   r   c                      | d          S )Nr   r;   r
  s    r3   rP   z,TestTruncWeibull.test_munp.<locals>.<lambda>$  r
  r5   )r9   r	  r  r   r   r   )rs   r  r  m1_expectedr   r  m2_expectedm3m3_expectedr  m4_expectedr1   r2   r  r
  s              @@@@r3   	test_munpzTestTruncWeibull.test_munp  s   	? 	? 	? 	? 	? 	? 	? #**1aA66R#**1aA663333Q::QK(((#**1aA663333Q::QK(((#**1aA663333Q::QK(((#**1aA663333Q::QK(((((r5   c                 ^   d}d}d}t          j        dt          j        dt          j        dt          j        d          z              z             z
            }t          j                            ||||          }t          |d           t          j                            ||||          }t          |t          j        d                      t          j        	                    d|||          }t          ||           t          j        
                    ||||          }t          |d           t          j                            ||||          }	t          |	t          j        d                      t          j                            d|||          }
t          |
|           d S )NrT  r  r   r    r|   r%  )rj   r  r  r   r9   r	  rq   r   r  r   r   r  r  )rs   r1   r2   r  x_medrq   r	  r   r   r  r  s              r3   test_reference_valuesz&TestTruncWeibull.test_reference_values'  so   BF3"rvbzz/0B)C)C#CDDDEE$((1a88S!!!#**5!Q::RVBZZK((($((aA66U####&&uaA66C   #))%Aq99RVBZZK((($((aA66U#####r5   c                 *   d}d}d}t           j        }d}t          j                            |||          }t          j                            |||||          }t          ||           t          j                            |||          }t          j                            |||||          }	t          ||	           t          j                            |||          }
t          j                            |||||          }t          |
|           t          j        	                    |||          }t          j        	                    |||||          }t          ||           t          j        
                    |||          }t          j        
                    |||||          }t          ||           t          j                            |||          }t          j                            |||||          }t          ||           t          j        
                    dd||d          }t          |t          j        d	                     t          j                            dd||d          }t          |d	           d S )
Nr  r   r   r  r  r:  rg   r   r  )rj   r  r9   r	  ri   r	  r   r   rq   r  r   r  r   )rs   r_   r  r1   r2   rf   rv  p_truncr7  lp_truncrq   	cdf_truncr	  lc_truncrO   s_truncr  ls_truncs                     r3   test_compare_weibull_minz)TestTruncWeibull.test_compare_weibull_min?  se    F!!!Qe!44(,,Q1au,EE7###%%a%%88)00Aq!50IIH%%%##Aq#66*..q!Q.GG	Y'''%%a%%88)00Aq!50IIH%%%  AU 33(++Aq!Qe+DD7###$$Q$77)//1a%/HHH%%% "%%b!Q%;;26$<<(((#))"aAQ)??D!!!!!r5   c                    d\  }}}t          j        ||d          }t          j                            ||||          }t          j                            ||||          }t          j                            ||          t          j                            ||          z
  }t          j                            ||          |z  }t          j                            ||          t          j                            ||          z
  |z  }	t           j                            ||           t           j                            ||	           d S )N)r  rl  r  ra   )	rj   rY  r9   r	  ri   rq   r	  r  r   )
rs   r  r1   r2   r_   r  rx  r  r  rz  s
             r3   test_compare_weibull_min2z*TestTruncWeibull.test_compare_weibull_min2h  s    "1aK1c""%))!Q155%))!Q155 $$Q**U->-B-B1a-H-HH $$Q**T1!%%a++e.?.C.CAq.I.II4O

""4...

""4.....r5   N)r   r   r   r	  r  r
  r
  r
  r
  r
  r"
  r$
  r;   r5   r3   r	  r	    s        $ $ $
0 0 0$ $ $
" " "" " ") ) )6$ $ $0'" '" '"R/ / / / /r5   r	  c                   f    e Zd Zd Zd Zej                            dg d          d             ZdS )	TestRdistc                     t           j        }g d}t          |                    |                    |d          d          |d           d S )Nr  g     @r   rW  )r9   rdistr   rq   r   )rs   r	  rX  s      r3   test_rdist_cdf_gh1285zTestRdist.test_rdist_cdf_gh1285z  sX    $$$FJJvzz&%'@'@%HH"A	/ 	/ 	/ 	/ 	/ 	/r5   c                 
   t          j        ddd          }d}t          dt          j        |dz  |dz                                |dz   dz            z  t          j        |                              |                     d S )NgGzr  r`   g@r|   rg   r    )rj   rY  r   r9   r  ri   r(
  )rs   r_   r  s      r3   test_rdist_betazTestRdist.test_rdist_beta  s    KtR((C
1Q3! 4 4 8 8!a% C CC!KNN..q11	3 	3 	3 	3 	3r5   r,  ))r  i  g?)r]      g;|bZ?)r|   i  g79)r   iU  g޻S/c                 f    t          t          j                            ||          |d           d S rT  )r   r9   r(
  r   r0  s       r3   test_rdist_sfzTestRdist.test_rdist_sf  s/     	q!,,c>>>>>>r5   N)	r   r   r   r)
  r+
  r   r   r   r.
  r;   r5   r3   r&
  r&
  y  st        / / /3 3 3 [	
 	
 	
 ? ? ? ? ?r5   r&
  c                   ~    e Zd Zd Zd Zd Zd Zd Zd Ze	j
                            dg d          d	             Zd
S )TestTrapezoidc                 V   g d}|D ]}d|dg}t          t          j                            |||          t          j                            ||                     t          t          j                            |||          t          j                            ||                     d S )N)r   r~  r|   r    r   r    )r   r9   r   ri   triangrq   )rs   modesmoder_   s       r3   test_reduces_to_triangz$TestTrapezoid.test_reduces_to_triang  s        	; 	;DD!A 3 3AtT B B % 0 0D 9 9; ; ; 3 3AtT B B % 0 0D 9 9; ; ; ;		; 	;r5   c                 b   t          j        ddd          }t          t          j                            |dd          t          j                            |                     t          t          j                            |dd          t          j                            |                     d S )Nr   r    r`   )rj   rY  r   r9   r   ri   r  rq   r  s     r3   test_reduces_to_uniformz%TestTrapezoid.test_reduces_to_uniform  s    K1b!!EO//1a88%-:K:KA:N:NOOOEO//1a88%-:K:KA:N:NOOOOOr5   c                 n   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   rg   r    r|   r   r  rQ  rT  r]   g      ?r  r   r	  ry  g      ?)r   r9   r   ri   rq   rx   s    r3   
test_caseszTestTrapezoid.test_cases  s   EO//1a88!<<<EO//1a88!<<<EO//Q<</	1 	1 	1EO//S#>>/	1 	1 	1 	EO//S#>>FFFEO//S#>>EEEEO//S#>>FFFEO//S#>>HHHEO//S#>>FFFEO//S#>>DDDEO//S#>>HHHEO//S#>>DDDDDr5   c                   	
 d\  	

	z
  	z
  z  	z
  	z
  z  		z
  f\  }}}}dz   
z
  	z
  z  	
fd} |d          } |d          |dz  z
  }dz
  
z   	z
  z  z   
z
  	z
  z  t          j        dz   
z
  	z
  z            z   }t          t          j                            ||||          |d           t          t          j                            ||||          |d           t          t          j                            ||||          |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 )N)r  r   rg   r   rg   c                     | dz   z  | dz   z  z
  z
  z  | dz   z  | dz   z  z
  z
  z  z
  z  | dz   z  | dz   z  S )Nrg   r    r;   )ru  r1   r2   r  drp  s    r3   r  z6TestTrapezoid.test_moments_and_entropy.<locals>.moment  sn    !ac(Q1X-!A#6!Hq1Q3x/AaC89 :qSqS" #r5   r    r|   r  rW  r   r  r3  r   r   )rj   r  r   r9   r   r   rA  rH   )rs   r  rf	  re   rf   r  r   rA  rH   r1   r2   r  r<
  rp  s            @@@@@r3   test_moments_and_entropyz&TestTrapezoid.test_moments_and_entropy  s    "
1acac]QqSQqSM1acABU1QqM	# 	# 	# 	# 	# 	# 	# 	# 	#
 vayyfQii$'!1Qq/QqSU1W-sac!eAg0G0GGEO00ReDD "	. 	. 	. 	.EO//BUCC	- 	- 	- 	-EO33BCGG#R	1 	1 	1 	1 	EO00Ar1==r2NNNNEO00Ar1==q"MMMMEO//1b!<<aLLLLLLr5   c                    t          j        g d          }t          j        ddg          d d d f         }t          j        g d          }t          j                            |||          }t          j        |||          \  }}}t          j        |j        |j                  }t          j	        |j                  }	t          |	|                                |                                |                                          D ]+\  }
}}}t          j                            |||          ||
<   ,t          ||                    |j                  d           t          j        t          j                            ||d	                    }t          j        ||          \  }}t          j        |j        d
f          }t          j	        |j                  }	t          |	|                                |                                          D ]+\  }
}}t          j                            ||d	          ||
<   ,t          ||j                            |j                  d           d S )Nru  r|   r  )r\  rl  r   r;  r   r   r  r  r   )rj   r   r9   r   ri   broadcast_arraysemptyr   r   r  r  ravelr   r?  rB  r   r  )rs   r  r<
  r_   r  ccddr  r   indr  rr  c1d1s                 r3   test_trapezoid_vectz!TestTrapezoid.test_trapezoid_vect  s
   H___%%Hc3Z  D)H&&&''O1a(((Aq11
Bhrwbh///i   bhhjj"((**bhhjjII 	5 	5MAr2r_((R44CFF3;;qw//e<<<< Ju,,Q6,BBCC$Q**Bh|$$i  S"((**bhhjj99 	C 	CIAr2_**2r6*BBCFF35==11>>>>>>r5   c                 (   t          j        ddd          }t          j        d          5  t          j                            |dd          }d d d            n# 1 swxY w Y   t          |t          j                            |                     d S )Nr   r    r`   z`trapz.pdf` is deprecatedr  )	rj   rY  r   deprecated_callr9   rI   ri   r   r  )rs   r_   r  s      r3   
test_trapzzTestTrapezoid.test_trapz  s    K1b!!#*EFFF 	. 	.[__Q1--F	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	.FEM$5$5a$8$899999s   "AA!Ar  )ri   r   rq   r  r   r  r   r  c                    d\  }} t          t          j        |          d||          }t          j        d| d          5   t          t          j        |          d||          }d d d            n# 1 swxY w Y   ||k    sJ d S )N)rQ  r   r    z`trapz.z` is deprecatedr  )r.  r9   r   r   rI
  rI   )rs   r  r  r<
  rX   r  s         r3   test_trapz_deprecationz$TestTrapezoid.test_trapz_deprecation  s     1375?F33Aq!<<#3F333
 
 
 	; 	; 2WU[&11!Q::F	; 	; 	; 	; 	; 	; 	; 	; 	; 	; 	; 	; 	; 	; 	; !!!!!!s   &A66A:=A:N)r   r   r   r5
  r7
  r9
  r=
  rG
  rJ
  r   r   r   rL
  r;   r5   r3   r0
  r0
    s        ; ; ;P P P
E E E(M M M6? ? ?4: : : [X (E (E (E F F" "F F" " "r5   r0
  c                       e Zd Zd ZdS )
TestTriangc                    t          j        d          5  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           d d d            d S # 1 swxY w Y   d S )Nraiser  r   r   r|   rT  r    r   rg   r   rl  )rj   r  r   r9   r2
  ri   rq   rx   s    r3   test_edge_caseszTestTriang.test_edge_cases  s   [W%%% 	6 	6))!Q//444))#q112666))!Q//444))!Q//333))#q112666))!Q//333))"b112666))#r22D999))#r22C888))"b112666))#r22D999))"b111555	6 	6 	6 	6 	6 	6 	6 	6 	6 	6 	6 	6 	6 	6 	6 	6 	6 	6s   H)IIIN)r   r   r   rQ
  r;   r5   r3   rN
  rN
    s#        6 6 6 6 6r5   rN
  c                       e Zd Zej                            dddg          d             Zej                            dg d          d             Zd Zd	 Z	d
S )TestMaxwellr  )r  gCR5&.)r~   g7q?c                 d    t          t          j                            |          |d           d S r#  )r   r9   maxwellr   r:  s      r3   r  zTestMaxwell.test_sf   s/     	((++Su======r5   rV  ))r  g1!@)g    ?g?)g      <gCuJ!@c                 d    t          t          j                            |          |d           d S r   )r   r9   rU
  r  rY  s      r3   r  zTestMaxwell.test_isf)  s/    
 	))!,,c>>>>>>r5   c                 l    d}t           j                            d          }t          ||d           d S )Ngi6ur  r  r   )r9   rU
  r  r   )rs   r  r  s      r3   r  zTestMaxwell.test_logcdf0  s7    %%%a((%000000r5   c                 l    d}t           j                            d          }t          ||d           d S )Ng_OԺr-  r  r   )r9   rU
  r  r   )rs   r  r  s      r3   r  zTestMaxwell.test_logsf6  s7    %##D))s//////r5   N)
r   r   r   r   r   r   r  r  r  r  r;   r5   r3   rS
  rS
    s        
 [X::<= => >= => [X: : :; ;? ?	; ;?1 1 10 0 0 0 0r5   rS
  c                       e Zd Zd Zd ZdS )
TestMielkec                    d\  }}t          t          j        ||                              d          t          j                   t          t          j        |d                              d          t          j                   t          t	          j        t          j        |d                              d                               d S )N)g rh@gv?r    rT  r0  )r   r9   mielker  rj   r  r	   rG  )rs   r\   rO   s      r3   r*  zTestMielke.test_moments>  s    1U\!Q''..q1126:::U\!S))0033RV<<<ELD1188;;<<=====r5   c                     t          j        ddd          }d\  }}t          t          j                            ||||z            t          j                            |||                     d S )Nr~   ra   r=  )g@gHzG@)rj   rY  r   r9   burrri   r\
  )rs   r_   r\   rO   s       r3   test_burr_equivalencez TestMielke.test_burr_equivalenceE  s^    Kc2&&1
q!QqS115<3C3CAq!3L3LMMMMMr5   N)r   r   r   r*  r_
  r;   r5   r3   rZ
  rZ
  =  s7        > > >N N N N Nr5   rZ
  c                   &    e Zd Zd Zd Zd Zd ZdS )TestBurrc           	      n   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             }t	          ||           d |D             }d |D             }t	          ||           d S )Nr    r    )r|   rg   )r    r    )rg   r|   r  r|   rT  )r    rg   r   c                 <    g | ]\  }}} |j         |j        g|R  S r;   r  r  s       r3   rC   z0TestBurr.test_endpoints_7491.<locals>.<listcomp>X  r  r5   c                     g | ]\  }}}|	S r;   r;   r  s       r3   rC   z0TestBurr.test_endpoints_7491.<locals>.<listcomp>Y  r  r5   c                 <    g | ]\  }}} |j         |j        g|R  S r;   )r   r1   r  s       r3   rC   z0TestBurr.test_endpoints_7491.<locals>.<listcomp>\  s4    @@@<2uayry&&&&@@@r5   c                 >    g | ]\  }}}t          j        |          S r;   )rj   r  r  s       r3   rC   z0TestBurr.test_endpoints_7491.<locals>.<listcomp>]  s)    FFF)=UI26)$$FFFr5   )r9   fiskr^
  burr12r   )rs   r   r  r  s       r3   test_endpoints_7491zTestBurr.test_endpoints_7491L  s     Zq!Z1%Z#Z1%\8S)\63'\63') >====>>>>>!#w///@@4@@@FFFFF!#w/////r5   c                     d\  }}t          j        ||                                           \  }}d\  }}t          ||           t          ||           d S )N)rU  r   )g4U?g&MI?)r9   r^
  r   )rs   r  r<
  r   variancemean_hcvariance_hcs          r3   test_burr_stats_9544zTestBurr.test_burr_stats_9544`  s_    1Aq))//11h  Gg&&&+.....r5   c                    d\  }}t          j        ||                                           \  }}t          t          j        |                     t          t          j        |                     d\  }}t          j        ||                                           \  }}t          t          j        |                     t          t          j        |                     d\  }}t           j                            t          j        g d          ||          \  }}}}t          t          j        |                     t          t          j        |                     t          t          j        |                     t          t          j        |                     d\  }}t           j                            g d||          \  }}}}t          t          j        |                     t          t          j        |                     t          t          j        |                     t          t          j        |                     d\  }}t           j                            g d||          \  }}}}t          t          j        |                     t          t          j        |                     t          t          j        |                     t          t          j        |                     d\  }}t           j                            g d||          \  }}}}t          t          j        |                     t          t          j        |                     t          t          j        |                     t          t          j        |                     d\  }}t           j                            g d||          \  }}}}t          t          j        |                     t          t          j        |                     t          t          j        |                     t          t          j        |                     d S )N)r|   r   )r  r   r  )r  r   )r;  r   )r9  r   )r9   r^
  r	   rj   rf  rG  r  r   )	rs   r  r<
  r   rl
  e1e2e3e4s	            r3   test_burr_nan_mean_var_9544z$TestBurr.test_burr_nan_mean_var_9544k  sb   1Aq))//11h""###1Aq))//11hD!!"""""###1))"(<<<*@*@!QGGBB1)),,,1==BBB   1)),,,1==BBB   B   1)),,,1==BBB   B   B   1)),,,1==BBB   B   B   B     r5   c                     d\  }}g d}g d}t          t          j                            |||          |d           d S )N)rU  r  )r]   r  r  r7  )g+&?gI$_@gKT@gؗVAr  r   )r   r9   r^
  r  )rs   r  r<
  r  r  s        r3   test_burr_isfzTestBurr.test_burr_isf  sV     1&&&# # #
q!Q//5AAAAAAr5   N)r   r   r   rj
  ro
  ru
  rw
  r;   r5   r3   ra
  ra
  K  sV        0 0 0(	/ 	/ 	/(! (! (!TB B B B Br5   ra
  c                       e Zd Zej                            dddg          d             Zd Zej                            dg d          d             Zd	S )

TestBurr12zscale, expected)rT  gd~)r;  gR톇T c                 r    t           j                            dddd|          }t          ||d           d S )Ng     jAg     jAr   r  r  r   r   )r9   ri
  r  r   )rs   rf   rX   rr  s       r3   r  zTestBurr12.test_delta_cdf  s?    ( ''S!Qe'DDxe444444r5   c                     d\  }}t           j        dz  }dt           j        dz  dz  z
  }t           j        dz  d|dz  z  z  }t           j        }||||g}t          j        ||                              d	          }t          ||d
           d S )Nrt  r   r    rg   ru  r   rt	  r  r  r  r   )rj   rk   r  r9   ri
  r   )	rs   r  r<
  r   rA  rB  r  r  r   s	            r3   test_moments_edgezTestBurr12.test_moments_edge  s    
 1uQw"%(2+ouaxCH%6S$)l1a  &&v..Su------r5   zp, c, d, ref))r  r  r|   ggx4/@)gҶOɃ;r  r|   gI8aS@)r  rl  #   gj\/ @)g#aMR/rl  r}
  g!GAc                 l    t           j                            |||          }t          ||d           d S r#  )r9   ri
  r  r   )rs   rv  r  r<
  r  r_   s         r3   test_isf_near_zerozTestBurr12.test_isf_near_zero  s8     LQ1%%3U++++++r5   N)	r   r   r   r   r   r   r  r|
  r
  r;   r5   r3   ry
  ry
    s        [.<;=> >5 5> >5(. . .  [	0 	0 	0 , , , , ,r5   ry
  c            	          e Zd Zg dZg dZg dZ ej        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          e                    Zddddej
        dfddej
        dfddej
        dfgZej        j        d             Zej        j        d             Zej                            e          ZdZ eej                            ee                    5 Z  e!j"        e           Z#ddd           n# 1 swxY w Y   ej        $                    de#d                   d             Z%ej        $                    de#d                   d             Z&ej        j        ej        '                    d          ej        $                    de#d                   d                                     Z(ej        j        d             Z)ej        j        d             Z*ej        $                    de          d              Z+ej        j        ej        '                    d          d!                         Z,ej        j        d"             Z-d# Z.d$ Z/d% Z0d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`"@rt  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@  r  r  r    r   r`   r  x   )rg   r  r.  r  )r]   r   i)#  gupm1f?)r    r`   r   g?<-=A?gv"?r   gd暕?ga-A?c                     | j         D ]>\  }}|\  }}}t          j                            |||          }t	          ||d           ?d S )Nr  r   )r   r9   studentized_rangerq   r   )rs   pvkr  
p_expectedr  r\   res_ps          r3   test_cdf_against_tablesz,TestStudentizedRange.test_cdf_against_tables    s_    i 	: 	:FC"J1+//1a88EE:D99999	: 	:r5   c                     | j         D ]>\  }}|\  }}}t          j                            |||          }t	          ||d           ?d S )Nr  r   )r   r9   r
  r   r   )rs   r
  
q_expectedrv  r  r\   res_qs          r3   test_ppf_against_tablesz,TestStudentizedRange.test_ppf_against_tables   s^    #y 	: 	:OCGAq!+//1a88EE:D99999	: 	:r5   z&data/studentized_range_mpmath_ref.jsonNcase_resultcdf_datac                     |d         }|d         }|d         |d         |d         f}t          j        j        | }t          |||d         |d                    d S 	Nsrc_case	mp_resultr  r\   r  expected_atolexpected_rtolrn  )r9   r
  rq   r   rs   r
  r
  r
  qkvr   s         r3   test_cdf_against_mpz(TestStudentizedRange.test_cdf_against_mp   s    z*,	smXc]HSM9%)3/Y%o6%o6	8 	8 	8 	8 	8 	8r5   pdf_datac                     |d         }|d         }|d         |d         |d         f}t          j        j        | }t          |||d         |d                    d S r
  )r9   r
  ri   r   r
  s         r3   test_pdf_against_mpz(TestStudentizedRange.test_pdf_against_mp   r
  r5   z+intermittent RuntimeWarning: invalid value.moment_datac                    |d         }|d         }|d         |d         |d         f}t          j        d          5  t          j        j        | }d d d            n# 1 swxY w Y   t          |||d         |d	         
           d S )Nr
  r
  r  r\   r  r  r  r
  r
  rn  )rj   r  r9   r
  r  r   )rs   r
  r
  r
  mkvr   s         r3   test_moment_against_mpz+TestStudentizedRange.test_moment_against_mp)   s     z*,	smXc]HSM9 [*** 	7 	7)0#6C	7 	7 	7 	7 	7 	7 	7 	7 	7 	7 	7 	7 	7 	7 	7 	Y%o6%o6	8 	8 	8 	8 	8 	8s   AA!$A!c                     d\  }}t          t          j        j        dt          j        ||f          }t          |d         d           d S )Nr   r`   r   r[  r    )r   r9   r
  ri   rj   r  r   )rs   r\   r  r   s       r3   test_pdf_integrationz)TestStudentizedRange.test_pdf_integration:   sG    15*.26AGGGA"""""r5   c                    d\  }}t          j        ddd          }t          j                            |||          dd          }t          j                            |||          }t          ||          }t          ||d           d S )	Nr
  r   r`   r~   )stepr    r  r   )rj   r  r9   r
  rq   ri   r   r   )rs   r\   r  r_   y_cdf	y_pdf_rawy_pdf_cumulatives          r3   test_pdf_against_cdfz)TestStudentizedRange.test_pdf_against_cdfA   s    1
 Ia$''''++Aq!44QRR8+//1a88	/	1== 	(%d;;;;;;r5   r_case_resultc                     |\  }}}}t          j        d          5  t          j                            |||          }d d d            n# 1 swxY w Y   t          ||           d S )Nr  r  )rj   r  r9   r
  rq   r   )rs   r
  r  r\   r  r_resr   s          r3   test_cdf_against_rz'TestStudentizedRange.test_cdf_against_rQ   s     '1a[*** 	7 	7)--aA66C	7 	7 	7 	7 	7 	7 	7 	7 	7 	7 	7 	7 	7 	7 	7U#####s   "AAAc                    t          j        d          5  t          j                            ddgddgddg          }d d d            n# 1 swxY w Y   t          |j        d	           t          j        t          d
          5  t          j                            dddgg d           d d d            d S # 1 swxY w Y   d S )Nr  r  r    rg   r   r   r`   r  r	  z...could not be broadcast...r  )r`   r  rV  )
rj   r  r9   r
  r  r   rB  r   r   r  r  s     r3   test_moment_vectorizationz.TestStudentizedRange.test_moment_vectorizationY   sh    [*** 	H 	H'--q!fq!fr2hGGA	H 	H 	H 	H 	H 	H 	H 	H 	H 	H 	H 	H 	H 	H 	H 	&&&]:-KLLL 	C 	C#))!aV\\\BBB	C 	C 	C 	C 	C 	C 	C 	C 	C 	C 	C 	C 	C 	C 	C 	C 	C 	Cs#   (A

AA&B88B<?B<c                    t                      5 }t          j        d          5  |                    t                     t
          j                            g d          \  }}}}d d d            n# 1 swxY w Y   d d d            n# 1 swxY w Y   t          t
          j        	                    ||                     d S )Nr  r  r  )
r   rj   r  rS   r   r9   r
  rt  r	   	_argcheck)rs   rm  r\   r  r   s        r3   test_fitstart_validz(TestStudentizedRange.test_fitstart_validi   sF      	GCX)F)F)F 	G 	GJJ)***1;;IIIFFKAr1a	G 	G 	G 	G 	G 	G 	G 	G 	G 	G 	G 	G 	G 	G 	G 	G 	G 	G 	G 	G 	G 	G 	G 	G 	G 	G 	G 	G 	G 	G 	'11!R8899999s5   B	AA2&B	2A6	6B	9A6	:B		BBc                    t           j                            ddt          j                  }t           j                            ddd          }t          ||dd           t           j                            ddt          j                  }t           j                            ddd          }t          ||dd           d S )Nr   r`   韆 r  rn  )r9   r
  ri   rj   r  r   rq   )rs   r   
res_finites      r3   test_infinite_dfz%TestStudentizedRange.test_infinite_dfq   s     %))!R88,00B>>
Zd>>>>%))!R88,00B>>
Zd>>>>>>r5   c                 Z   t           j                            ddd          }t           j                            ddd          }t           j                            ddd          }t          t          t
          ||dd           t          ||dd           t           j                            ddd          }t           j                            ddd          }t           j                            ddd          }t          t          t
          ||dd           t          ||dd           d S )Nr   r`   r   r
  i rv  rn  )r9   r
  ri   r`  AssertionErrorr   rq   )rs   r   r
  
res_sanitys       r3   test_df_cutoffz#TestStudentizedRange.test_df_cutoff|   s    %))!R88,00B>>
,00B>>
nosJd	, 	, 	, 	,
JTEEEE%))!R88,00B>>
,00B>>
nosJd	, 	, 	, 	,
JTEEEEEEr5   c                     d\  }}}t           j                            |||          }t          |dd           |dk    sJ d S )N)gaRA@r   iS  r   r  r   )r9   r
  r   r   )rs   r  r\   r  rv  s        r3   test_clippingz"TestStudentizedRange.test_clipping   sO     .1a#&&q!Q//15))))Avvvvvvr5   )1r   r   r   q05q01q001rj   rw  qspsr  vskslistr  r(   r   r_datar   r   r   r
  r   r
  ospathdirnamer  path_prefixrelative_pathopenrc  filejsonr  pregenerated_datar   r
  r
  xfail_on_32bitr
  r
  r
  r
  r
  r
  r
  r
  r
  r;   r5   r3   r
  r
    s       ' ' 'C' ' 'C( ( (D 
c4(	)	)B			B
QBRV	$B	B4GGBB'',,--D
 	*(	
Arv()	
Arv()	
B)*F [: : : [: : : '//(++K<M	bgll;66	7	7 ,4%DIdOO, , , , , , , , , , , , , , , [],=j,IJJ8 8 KJ8 [],=j,IJJ8 8 KJ8 [[ MNN[],=m,LMM8 8 NM ON 8 [# # # [< < < [_f55$ $ 65$ [[ MNNC C ON C [: : :	? 	? 	?F F F*    s   %DD	Dr
  c                       e Zd Zej                            dddd ej        dgdgdgg          g          d             Zd Z	d Z
d Zd	 Zd
S )TestTukeyLambdar  r   r  r  c                     t          j        ddd          }t          j                            ||          }t          j        |                                          sJ |dk                                    sJ d S )Nr  rU  r^   r   )rj   rY  r9   r+   ri   rG  r   )rs   r  r_   rv  s       r3   test_pdf_nonpositive_lambdaz+TestTukeyLambda.test_pdf_nonpositive_lambda   sm     Kc3''!!!S)){1~~!!#####C}}r5   c                    t          j        ddd          }t          j        dgdgdgg          }t          j                            ||          }t          j        |                                          sJ |d d         dk                                    sJ |d         dk                                    sJ |d         dk                                    sJ d S )Nr  rU  r^   r  r   r   rg   )	rj   rY  r   r9   r+   ri   rG  r   r   )rs   r_   r  rv  s       r3   test_pdf_mixed_lambdaz%TestTukeyLambda.test_pdf_mixed_lambda   s    Kc3''hu-..!!!S)){1~~!!##### "1"  """"" !s
!!!!!!  """""""r5   c                 (   t          j        g d          }t          j                            |          \  }}t          j        t           j        t           j        t           j        dddg          }t          ||           t          ||            d S )N)g      r   r   rl  r|   r   r   rg   r|   )rj   r   r9   r+   supportr  r   )rs   r  r1   r2   
expected_bs        r3   r1  zTestTukeyLambda.test_support   s|    h999:: ((--1Xrvrvrvq!SABB
Q
###Q$$$$$r5   c                 n    t           j                            ddgd          }t          |ddg           d S )Nr  r   r|   r   )r9   r+   ri   r   r  s     r3   test_pdf_support_boundaryz)TestTukeyLambda.test_pdf_support_boundary   s:     !!4+s33Qc
#####r5   c                 t   t           j                             dd          }dt          j        dz  dz  ddg}t	          ||d           t           j                             d	d          }g d
}t	          ||d           t           j                             dd          }g d}t	          ||d           d S )Nr   r  r  rg   r   ro  r`   rW  g
ףp=
	@)r   gz5v?r   gRTgQ?)r   gNg @r   gJFٻ)r9   r+   rj   rk   r   )rs   r  rX   s      r3   "test_tukeylambda_stats_ticket_1545z2TestTukeyLambda.test_tukeylambda_stats_ticket_1545   s     $$Q$77ruaxz1c*B"5555$$T6$::GGGB"5555$$T6$::GGGB"555555r5   N)r   r   r   r   r   r   rj   r   r
  r
  r1  r
  r
  r;   r5   r3   r
  r
     s        [	dD("(TFSED6#:;;<  	 # # #% % %$ $ $6 6 6 6 6r5   r
  c                   r    e Zd Zd Zd Zej                            dg d          d             Zd Z	d Z
dS )	TestLevyc                 "   t          j        g d          }t          j        g d          }t          j                            |          }t          ||d           t          j                            |          }t          ||d           d S )N)r   rT  r|   r]   r~   r  )g~V1?r3  gba"?g܄4ťY?r4  g$%+Ir  r   r   )rj   r   r9   r&  rq   r   r   )rs   r_   rX   r2  r  s        r3   test_levy_cdf_ppfzTestLevy.test_levy_cdf_ppf   s    H77788 8 5 5 5 6 6 JNN18%0000 Z^^H%%AE******r5   c                     t          j        g d          }t          j        g d          }t          j                            |          }t          ||d           d S )N)r  r  gMraB3GrL  )gep[>g}XEQ=gѭEG<g?_%~:r  r   )rj   r   r9   r&  r   r   )rs   r_   rX   r2  s       r3   test_levy_sfzTestLevy.test_levy_sf   sh    H---..8 4 4 4 5 5 JMM!8%000000r5   zp, expected_isf))r  gj1e2H)r-  g6C)r  gnE@)r  g!^x1?)r  gZ?)g   ?g:}!Nu?c                 h    t           j                            |          }t          ||d           d S )Nr  r   )r9   r&  r  r   )rs   rv  expected_isfr_   s       r3   test_levy_isfzTestLevy.test_levy_isf!  s2     JNN1<e444444r5   c                 p    d}d}t           j                            |          }t          ||d           d S )NrL  g?_%~r  r   )r9   r&  r  r   r[  s       r3   test_levy_logcdfzTestLevy.test_levy_logcdf!  s<    $""1%%%000000r5   c                 p    d}d}t           j                            |          }t          ||d           d S )Nr%  gޢاr  r   )r9   r&  r  r   r]  s       r3   test_levy_logsfzTestLevy.test_levy_logsf!  s<    %
  ##s//////r5   N)r   r   r   r
  r
  r   r   r   r
  r
  r
  r;   r5   r3   r
  r
     s        + + +.	1 	1 	1, [.J J JK K5 5K K51 1 10 0 0 0 0r5   r
  c                  ,   t          t          j                            d          ddd           t          t          j                            d          ddd           t          t          j                            dd	d
          ddd           d S )Ng2g?r`   test_540_567)rX  r]  g2gk?g;B.?gͣh?gHŎ*?rd   g.y?)r   r9   r  rq   r;   r5   r3   r
  r
  #!  s    
}557J "N< < < <
}557K "N< < < <
}.-; ' = =+ "N< < < < < <r5   zdocstrings strippedrj  c                      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	   r9   r  r  r;   r5   r3   test_regression_ticket_1421r
  /!  s9    (0EEFFFH--.....r5   c            	      	   t          j        d          5  t          t          j        t          j                            dt           j                                       t          t          j        t          j                            dt           j                                       t          t          j        t          j        	                    dt           j                                       t          t          j        t          j        
                    dt           j                                       t          t          j        t          j                            dt           j                                       t          t          j        t          j                            dt           j                                       t          t          j        t          j                            dt           j                                       t          t          j        t          j                            dt           j                                       t          t          j        t          j                            t           j        d                               t          t          j        t          j                            t           j        d                               t          t          j        t          j        	                    t           j        d                               t          t          j        t          j        
                    t           j        d                               t          t          j        t          j                            t           j        d                               t          t          j        t          j                            t           j        d                               t          t          j        t          j                            t           j        d                               t          t          j        t          j                            t           j        d                               d d d            d S # 1 swxY w Y   d S )Nr  r  r    r|   )rj   r  r	   rf  r9   r  r  r  rq   r  r   ri   r   r   r  r  rP  r:  r;   r5   r3    test_nan_arguments_gh_issue_1362r
  5!  s)   	X	&	&	& < <262233444Q//00111q"&1122333Arv..//000Q//00111262233444Q//00111Q//00111//<<==>>>,,RVS99::;;;..rvs;;<<===++BFC8899:::,,RVS99::;;;//<<==>>>,,RVS99::;;;,,RVS99::;;;#< < < < < < < < < < < < < < < < < <s   RR44R8;R8c                     t           j                            d           t          j        g d          } t          j                            | d         | d         | d         d          }t          j        d	          5  t          j        t          j                            |d
                    }d d d            n# 1 swxY w Y   t          || d           t          j        t          j                            |dd                    }t          || d           t          j        t          j                            |dd                    }t          || d           t          j        t          j                            |dd                    }t          || d           t           j                            d           d}d}t          j
                            |dd          }t          j        t          j
                            ||                    }t          j        |t          j        ||z
  dz                                            g          }t          ||d           d S )Ni.  )rl  r   r|   r   r    rg   ra   r;  r  )r	  r   r   rW  r|   )r   re   rl  )r   re   r  r   r   r   )rj   r   r   r   r9   r(  r   r  r   r   r  r  r   )truer_   r  re   r   rX   s         r3   test_frozen_fit_ticket_1536r
  J!  sL   INN48OOO$$D$q'47DG#>>A	H	%	%	% 9 9%-++AB+77889 9 9 9 9 9 9 9 9 9 9 9 9 9 9 a0000Xem''#1'==>>Fa0000Xem''d'::;;Fa0000Xem''d';;<<Fa0000INN4
CD
sBS))AXejnnQTn2233Fxrw4!'9'9';';<<=>>H!444444s    4C  CCc                      t           j                            d           t          j                            d          } t          j                            |           }d}t          ||d           d S )Ni	 ra   r;  )g
ףp=
?gʡE?r    rW  )rj   r   r   r9   r  r   r   r   )r   r  rX   s      r3   test_regression_ticket_1530r
  f!  sc    INN6
,



$
$C\c""FH!444444r5   c                     t           j                            d           t           j                            d          } dD ]K}t          j                            | |z             \  }}t          ||d           t          |dd           Ld S )Nr   r	  )r
  g   6ArT  r   r  )rj   r   r   r>  r9   r  r   r   )r_   offsetre   rf   s       r3   test_gh_pr_4806r
  o!  s    INN4
	A' . .\%%a&j11
UV#....s-----. .r5   c                      t          t          j        t          j                            dd                               d S )NrE  rS  )r	   rj   rG  r9   r  r:  r;   r5   r3   test_poisson_logpmf_ticket_1436r
  y!  s2    BK,,T3778899999r5   c                      ddg} | D ]6\  }}t           j                             |d          }t          ||           7d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))
    )rT  )r|   r  r   r  )r   )gUUUUUU?r3	  g^cQg333333r  r  N)r9   r  r   )rZ  r1   
exact_mvskr  s       r3   test_powerlaw_statsr
  }!  s[    > ,=?E 4 4:~##Av#66!$
33334 4r5   c                  f    t           j                            dd          } t          | d           d S )Nr   r    r   )r9   r  r   r   rv  s    r3   test_powerlaw_edger
  !  s/    a##ACr5   c                  x   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    r   )rl  rT  r  rT  )r9   r@  r   r   ri   rj   r  r
  s    r3   test_exponpow_edger   !  s    a##AC 	1...//ARVS#&'''a!1!1!122ARVS26'*+++++r5   c                  h    t           j                            ddd          } t          | d           d S )Nr   r    rT  )r9   gengammari   r   r
  s    r3   test_gengamma_edger  !  s1    1a##ACr5   za, c, ref, tol))g    `6Ar    g2j!@r   )r  r    ggA@r   )r  r    r  r   )r  r    g #@r   )g    `FAr  gI~mr   )gl$W}Lr  g.I#Ur   c                 f    t          t          j                            | |          ||           d S r  )r   r9   r  rH   )r1   r  r  rW  s       r3   test_gengamma_extreme_entropyr  !  s1    " EN**1a00#C@@@@@@r5   c                      t           j                            ddd          } | dk    sJ t           j                            ddd          }|t          j         k    sJ d S )Nr   r    r   r   )r9   r  ri   r   rj   r  )rv  r  s     r3   !test_gengamma_endpoint_with_neg_cr  !  sV    1a$$A8888>  Ar**DBF7??????r5   c                      t           j                            ddd          } t          | d           t           j                            ddd          } t          | d           d S )Nr   rS  rT  g>r`   gqq?)r9   r  r  r   r
  s    r3   test_gengamma_munpr	  !  sZ    Rb))A:&&&RR((A6"""""r5   c                     t          j        g d          } t          j        d          5  t                      5 }|                    t
          d           |                    t          d           t          j        	                    |            d d d            n# 1 swxY w Y   d d d            d S # 1 swxY w 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)
rj   r   r  r   rS   r   rz  r9   rL  r   )r<
  rm  s     r3   test_ksone_fit_freezer  !  sK   
		/ 		/ 		/
	0 
	0A 
X	&	&	&     	CJJ)#$ $ $ JJ~FH H HKOOA	 	 	 	 	 	 	 	 	 	 	 	 	 	 	                 s5   B5ABB5B!	!B5$B!	%B55B9<B9c            	      r   t          j        t          t          ddd                               } g d}t	          t          j                                        |           |d           t	          t          j                                        | dz             j        |d           t
          j                            | dz             d	z  j	        }t          j
        t
          j                            |           t
          j                            |           z
            }t	          ||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/Mr-  r   y        +=y        |=r  )rj   r   r
  r  r   r9   r  r  realimagr   r   )r_   rX   derivderiv_expecteds       r3   test_norm_logcdfr  !  s
    
Dq#q))**	+	++A0 0 0H EJLL''**H4@@@@ EJLL''F
338(NNNN Zq6z**506EVEJ--a005:3D3DQ3G3GGHHNE>666666r5   c                      t          j        g d          } t          j        g d          }t          j                            |           }t          ||d           d S )N)gMb{Gzg{Gztg~jtX)g#]<gOul2;gޢا6g̅7=!r   r   )rj   r   r9   r'  r   r   )r_   rX   r2  s      r3   test_levy_l_sfr  "  sh    
11122Ax 1 1 1 2 2H 	AAxe,,,,,,r5   c                      t          j        g d          } t          j                            |           }t          j                            |          }t          || d           d S )N)g [n<rl  r  rR  r   )rj   r   r9   r'  r  r   r   )rv  r_   r  s      r3   test_levy_l_isfr  "  s[    
&&&''AAAAqu%%%%%%r5   c                     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@r  ra   r  r    )r   r9   r  rB  r   r;   r5   r3   test_hypergeom_interval_1802r  $"  s    ))#vucBB! ! !))$sCC! ! ! ))#vucBB! ! ! $$T3Q77;;;$$QS!44a88888r5   c                     t           j                            d           t          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
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           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           t	          t
          t          j        j        | 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        	                    dd          d           t          j                            | dd	           t	          t
          t          j        j        | ddd
           t	          t
          t          j        j        | ddd
	           t	          t
          t          j        j        | dd	           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d	  	         t          j                            | dd
ddd           d S )Nr   r]   r  r   numrg   r   rT  r  r   r|   rd   r   r  r  r;  r3  r  )rj   r   r   rY  r`  r  r9   r  ri   r   rq   r   rH   r   r  rP  rg  r	  r  r   s    r3   test_distribution_too_many_argsr  3"  sX   INN4 	C!$$$A)U[_aA3????)U[_aAqcBBBB)U[_aAq!<<<)U[_aA3cJJJJ)U[_b!CHHHH)U[_aQCsKKKK)U[_aQCsKKKK)U[.A3cJJJJ)U[0"aSLLLL)U[_aQCsKKKK 
KOOAq!	KOOAq!Q	Kb!	Kb!Q	Kb!Q%%%	KOOB1a   	KOOEKOOBQO//444 
JNN1aQN)UZ^Q1a888)UZ^Q1!<<<< )U[_a<<<<)U_0!Q1#FFFF)U_0!Q1c3GGG)UY]Aq!QsCCCC)UY]Aq!Q3cJJJJ	IMM!Q1a%%%%%r5   c                      t           j                            t          j        ddd          dd          } t           j                            t          j        ddd          dd          }t          | |dd           d S )	Nr  r  rQ  rg   gYZ@r  r   rf  )r9   ncx2rq   rj   r  _cdfvecr   r  s     r3   test_ncx2_tails_ticket_955r!  ["  sm     	
ryR--q.AAA
29RS111nEEAAqt!,,,,,,r5   c            
      *   t          j                    5  t          j        dt                     t	          t
          j                            dt          j	        dd          d          d           t
          j        
                    dt          j	        dd          d          } d d d            n# 1 swxY w Y   t          t          j        |                                                      t          j                    5  t          j        dt                     t	          t
          j                            ddd	          d           t          t
          j        
                    ddd	          d
           d d d            d S # 1 swxY w Y   d S )Nrs  r    iT  i^  rg   r   r  r   rV  gsq6)rw  rx  ry  rz  r   r9   r  ri   rj   r  r   r	   r  r   r   )logvals    r3   test_ncx2_tails_pdfr$  c"  s    
	 	"	" > >g~666UZ^^Aryc':':A>>BBB""1biS&9&91==> > > > > > > > > > > > > > >
 BK##%%&&& 
	 	"	" M Mg~666UZ^^E1b111555
))%B779KLLLM M M M M M M M M M M M M M M M M Ms%   BB11B58B5A9FFFzmethod, expectedrq   gu%>g	;Q=ri   g<x>gRz\R>r   gʎK$/gԆX1r   gfHu@ge:@c                 z     t          t          j        |           dddgd          }t          ||d           d S )Nr]   r   r   r`   )r	  r  r   r   )r.  r9   r  r   )r  rX   r  s      r3   test_ncx2_zero_ncr&  t"  sE      )WUZ((!QB???FFH5111111r5   c                      t           j                            ddd          } t           j                            dd          }t	          | |d           d S )Nr`   r   r    )r  r	  r   )r  r   r   r   )r9   r  r   r  r   )r  rX   s     r3   test_ncx2_zero_nc_rvsr(  "  sO     Z^^raa^88Fz~~!~44HFH5111111r5   c                      dt          j        dd          z  } t          t          j                            dd|           d           d S )Nr`   r   r
  r    r  r	  r   )rj   r  r   r9   r  rq   )r	  s    r3   test_ncx2_gh12731r+  "  sB    	RYq"	Bb11155555r5   c                      t          j        g d          } d\  }}t          j                            | ||          }g d}t          ||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*  )rT  rT  rT  rT  rT  g?gy[qD?r   r   r   r   r   r   r  r   )rj   r   r9   r  r   r   )r_   nur  r   sf_expecteds        r3   test_ncx2_gh8665r/  "  st    
 " " " 	# 	#A
 %GB	qRC	(	(B' ' 'K
 B%000000r5   c            	         d} d}t          j        t          j                            d| |          t          j                            d| |          d          }t          j                            || |          }t          j                            || |z   t          j        d| z  d|z  z                       }t          ||d	
           d S )Ni,  i  r  r  r  r  rg   r   r  r   )	rj   rY  r9   r  r   ri   r  r  r   )r  r	  r_   ncx2_pdfgauss_approxs        r3   test_ncx2_gh11777r3  "  s    
 
B	B
EJNN5"b11JNN5"b11u	> 	> 	>Az~~aR((H:>>!R"Wbga"fq2vo.F.FGGLHl666666r5   zx, c, expected))rg   r    g|1q?)rg   rg   g~?)r|  r    V<3T1=) 7yQCr    u	lY<)r|  g   vHGBg^b!*1=)ry  rS  gS?c                 h    t           j                            | |          }t          ||d           d S Nr  )r9   
foldcauchyr   r   r_   r  rX   r   s       r3   test_foldcauchy_sfr;  "  s4     
			Q	"	"BB%(((((r5   r  ))rg   g?)r|  r4  )r5  r6  )g6ްPg8R#/c                 f    t           j                            |           }t          ||d           d S r8  )r9   rN  r   r   )r_   rX   r   s      r3   test_halfcauchy_sfr=  "  s2     
			Q		BB%(((((r5   zp, expected))g?g*wӄZ>)333333?gN9?)r|   rT  )r~   g8<,O@)r  g:B)gl ׶w/gMb}[Pc                 d    t           j                            |           }t          ||           d S rv   )r9   rN  r  r   )rv  rX   r_   s      r3   test_halfcauchy_isfr@  "  s0     	QAAx     r5   c                  x    t          j        dd          } t          |                     d          d           d S )Nr   r    r  )r9   r/  r   rq   )r  s    r3   test_foldnorm_zerorB  "  s6    		#	#	#BAr5   ))rg   r    gwo{?)r  r    gGd-P/)r`   r?  gof?)r  r?  gOul";c                 h    t           j                            | |          }t          ||d           d S )Nr  )r9   r/  r   r   r:  s       r3   test_foldnorm_sfrD  #  s3     
		1a	 	 BB%(((((r5   c                     t           j                             g ddd          } t           j                             ddgdd          }t          d |D                       }t          ||            t           j                             g d          } t           j                             ddg          }t          d	 |D                       }t          ||            d S )
N)r   r|   rT  r    r|   rT  c              3   R   K   | ]"}t           j        t           j        |f         V  #d S rv   rj   rN  r  rl	  s     r3   r  z-test_stats_shapes_argcheck.<locals>.<genexpr>#  s/      88q"%	*888888r5   )rg   r=  r   rg   r=  c              3   R   K   | ]"}t           j        |t           j        f         V  #d S rv   rG  rl	  s     r3   r  z-test_stats_shapes_argcheck.<locals>.<genexpr>#  s/      88q"%26	*888888r5   )r9   r"  tupler   r(  )mv3mv2mv2_augmenteds      r3   test_stats_shapes_argcheckrM  #  s     .

3
7
7C
.

Sz1c
2
2C88C88888M$$$ -

lll
+
+C
-

q#h
'
'C88C88888M$$$$$r5   c                       e Zd Zd ZdS )
_distr_genc                     dS Nr	  r;   rs   r_   r1   s      r3   _pdfz_distr_gen._pdf(#  s    rr5   Nr   r   r   rS  r;   r5   r3   rO  rO  '#  s#            r5   rO  c                       e Zd Zd ZdS )_distr2_genc                     d|z  |z   S rQ  r;   rR  s      r3   _cdfz_distr2_gen._cdf-#      Avzr5   N)r   r   r   rX  r;   r5   r3   rV  rV  ,#  s#            r5   rV  c                       e Zd Zd Zd ZdS )_distr3_genc                     ||z   S rv   r;   rs   r_   r1   r2   s       r3   rS  z_distr3_gen._pdf2#  s    1ur5   c                     d|z  |z   S rQ  r;   rR  s      r3   rX  z_distr3_gen._cdf5#  s     Avzr5   Nr   r   r   rS  rX  r;   r5   r3   r[  r[  1#  s2              r5   r[  c                       e Zd Zd Zd ZdS )_distr6_genc                     ||z  |z   S rv   r;   r]  s       r3   rS  z_distr6_gen._pdf=#  s    sQwr5   c                     d|z  |z   S rQ  r;   r]  s       r3   rX  z_distr6_gen._cdf@#  rY  r5   Nr_  r;   r5   r3   ra  ra  ;#  s2              r5   ra  c                   t    e Zd Zd Zd Zd Zd Zd Zd Zd Z	d Z
d	 Zd
 Zd Zd Zd Zd Zd Zd Zd ZdS )TestSubclassingExplicitShapesc                 r    t          dd          }t          |                    dd          d           d S )Ndummyr1   r  r  r    ru  r	  rO  r   ri   rs   dummy_distrs     r3   test_correct_shapesz1TestSubclassingExplicitShapes.test_correct_shapesG#  s:     gc:::[__Q!_,,b11111r5   c                 v    t          dd          }t          t          |j        dfi t	          d           d S )Nrg  Arh  r    ru  )rO  r`  r  ri   r  rj  s     r3   test_wrong_shapes_1z1TestSubclassingExplicitShapes.test_wrong_shapes_1K#  s>     gc:::i!AAtayyyAAAAAr5   c                 ~    t          dd          }t          ddd          }t          t          |j        dfi | d S )Nrg  za, b, crh  r    rg   r   r  )rO  r  r`  r  ri   )rs   rk  r/  s      r3   test_wrong_shapes_2z1TestSubclassingExplicitShapes.test_wrong_shapes_2O#  sJ     gi@@@Q!q!!!i!;;s;;;;;r5   c                 X    t          dd          }t          t          t          fi | d S )Nrg  r	  rh  )r  r`  r  rO  rs   r/  s     r3   test_shapes_stringz0TestSubclassingExplicitShapes.test_shapes_stringT#  s2    +++i33s33333r5   c                 X    t          dd          }t          t          t          fi | d S )Nrg  z(!)rh  r  r`  SyntaxErrorrO  rs  s     r3   test_shapes_identifiers_1z7TestSubclassingExplicitShapes.test_shapes_identifiers_1Y#  s2    ...k:5555555r5   c                 X    t          dd          }t          t          t          fi | d S )Nrg  4chanrh  rv  rs  s     r3   test_shapes_identifiers_2z7TestSubclassingExplicitShapes.test_shapes_identifiers_2^#  2    000k:5555555r5   c                 X    t          dd          }t          t          t          fi | d S )Nrg  zm(fti)rh  rv  rs  s     r3   test_shapes_identifiers_3z7TestSubclassingExplicitShapes.test_shapes_identifiers_3b#  s2    111k:5555555r5   c                 X    t          dd          }t          t          t          fi | d S )Nrg  za=2rh  rv  rs  s     r3   "test_shapes_identifiers_nodefaultsz@TestSubclassingExplicitShapes.test_shapes_identifiers_nodefaultsf#  s2    ...k:5555555r5   c                 X    t          dd          }t          t          t          fi | d S )Nrg  z*argsrh  rv  rs  s     r3   test_shapes_argsz.TestSubclassingExplicitShapes.test_shapes_argsj#  r|  r5   c                 X    t          dd          }t          t          t          fi | d S )Nrg  z**kwargsrh  rv  rs  s     r3   test_shapes_kwargsz0TestSubclassingExplicitShapes.test_shapes_kwargsn#  s2    
333k:5555555r5   c                 X    t          dd          }t          t          t          fi | d S )Nrg  za, b, c, lambdarh  rv  rs  s     r3   test_shapes_keywordsz2TestSubclassingExplicitShapes.test_shapes_keywordsr#  s3    (9:::k:5555555r5   c                      G d dt           j                  } |d          }t          |                    dd          t           j                            d          dz             d S )Nc                       e Zd Zd ZdS )FTestSubclassingExplicitShapes.test_shapes_signature.<locals>._dist_genc                 F    t           j                            |          |z  S rv   r9   r  rS  rR  s      r3   rS  zKTestSubclassingExplicitShapes.test_shapes_signature.<locals>._dist_gen._pdfz#      zq))A--r5   NrT  r;   r5   r3   	_dist_genr  y#  #        . . . . .r5   r  r1   r  r|   rg   ru  r9   rF   r   ri   r  rs   r  rB   s      r3   test_shapes_signaturez3TestSubclassingExplicitShapes.test_shapes_signaturew#  s|    	. 	. 	. 	. 	.+ 	. 	. 	. y$$$TXXcQX'')<)<Q)>?????r5   c                      G d dt           j                  } |d          }t          t          |j        dfi t          dd           d S )	Nc                       e Zd Zd ZdS )STestSubclassingExplicitShapes.test_shapes_signature_inconsistent.<locals>._dist_genc                 F    t           j                            |          |z  S rv   r  rR  s      r3   rS  zXTestSubclassingExplicitShapes.test_shapes_signature_inconsistent.<locals>._dist_gen._pdf#  r  r5   NrT  r;   r5   r3   r  r  #  r  r5   r  r  r  r|   r    rg   r  )r9   rF   r`  r  ri   r  r  s      r3   "test_shapes_signature_inconsistentz@TestSubclassingExplicitShapes.test_shapes_signature_inconsistent#  so    	. 	. 	. 	. 	.+ 	. 	. 	. y'''i3AA$a...AAAAAr5   c                     G d dt           j                  } |d          }t          |                    dd          t           j                            d          dz             t          |                    dd          t           j                            d          dz             t          t          |j        dfi t          d           d S )	Nc                       e Zd Zd ZdS )?TestSubclassingExplicitShapes.test_star_args.<locals>._dist_genc                 V    |d         }t           j                            |          |z  S Nr   r  )rs   r_   r0  extra_kwargs       r3   rS  zDTestSubclassingExplicitShapes.test_star_args.<locals>._dist_gen._pdf#  s$    "1gzq))K77r5   NrT  r;   r5   r3   r  r  #  s#        8 8 8 8 8r5   r  r  r  r|   !   r  )xxx)r9   rF   r   ri   r  r`  r  r  r  s      r3   test_star_argsz,TestSubclassingExplicitShapes.test_star_args#  s    	8 	8 	8 	8 	8+ 	8 	8 	8
 y...TXXcrX22EJNN34G4G4JKKKTXXc2&&
s(;(;B(>???i3??$2,,,?????r5   c                 p    G d dt           j                  } |d          }t          |                    ddd          t           j                            d          dz  dz              t          |                    ddd          t           j                            d          dz  dz              d S )	Nc                       e Zd Zd ZdS )ATestSubclassingExplicitShapes.test_star_args_2.<locals>._dist_genc                 \    |d         }t           j                            |          |z  |z   S r  r  )rs   r_   r
  r0  r  s        r3   rS  zFTestSubclassingExplicitShapes.test_star_args_2.<locals>._dist_gen._pdf#  s)    "1gzq))K7&@@r5   NrT  r;   r5   r3   r  r  #  s(        A A A A Ar5   r  zoffset, extra_kwargr  r|   o   r  )r
  r  r  r  s      r3   test_star_args_2z.TestSubclassingExplicitShapes.test_star_args_2#  s    	A 	A 	A 	A 	A+ 	A 	A 	A
 y 5666TXXc#2X>>Z^^C((+c1	3 	3 	3TXXc3++Z^^C((+c1	3 	3 	3 	3 	3r5   c                      G d dt           j                  } |d          }t          |                    dd          t           j                            d                     d S )Nc                       e Zd Zd ZdS )BTestSubclassingExplicitShapes.test_extra_kwarg.<locals>._distr_genc                 r    |                     dd          }t          j                            |          |z  S )Nr  r    )popr9   r  rS  )rs   r_   r0  kwargsr  s        r3   rS  zGTestSubclassingExplicitShapes.test_extra_kwarg.<locals>._distr_gen._pdf#  s0     %jj::zq))K77r5   NrT  r;   r5   r3   rO  r  #  s#        8 8 8 8 8r5   rO  r  r  r    r   r  r  )rs   rO  rB   s      r3   test_extra_kwargz.TestSubclassingExplicitShapes.test_extra_kwarg#  sw    	8 	8 	8 	8 	8, 	8 	8 	8 z///TXXaQX//1B1BCCCCCr5   c                      G d dt           j                  } |d          }t          |                    d          t           j                            d                     d S )Nc                       e Zd Zd ZdS )ITestSubclassingExplicitShapes.test_shapes_empty_string.<locals>._dist_genc                 @    t           j                            |          S rv   )r9   r  ri   r  s     r3   rS  zNTestSubclassingExplicitShapes.test_shapes_empty_string.<locals>._dist_gen._pdf#  s    z~~a(((r5   NrT  r;   r5   r3   r  r  #  s#        ) ) ) ) )r5   r   r  r|   r  r  s      r3   test_shapes_empty_stringz6TestSubclassingExplicitShapes.test_shapes_empty_string#  sq    	) 	) 	) 	) 	)+ 	) 	) 	) y###TXXc]]EJNN3$7$788888r5   N)r   r   r   rl  ro  rq  rt  rx  r{  r~  r  r  r  r  r  r  r  r  r  r  r;   r5   r3   re  re  D#  s"       2 2 2B B B< < <
4 4 4
6 6 6
6 6 66 6 66 6 66 6 66 6 66 6 6
@ @ @B B B@ @ @3 3 3D D D9 9 9 9 9r5   re  c                       e Zd Zd Zd Zej                            ed          d             Z	ej                            ed          d             Z
d Zd Zd	 Zd
 ZdS )TestSubclassingNoShapesc                 p    t          d          }t          |                    dd          d           d S )Nrg  r  r    ru  r	  ri  rj  s     r3   test_only__pdfz&TestSubclassingNoShapes.test_only__pdf#  s8     g...[__Q!_,,b11111r5   c                 p    t          d          }t          |                    dd          d           d S )Nrg  r  r    ru  )rV  r   ri   rj  s     r3   test_only__cdfz&TestSubclassingNoShapes.test_only__cdf#  s8    !w///KOOAO33Q77777r5   docstring strippedrj  c                     t          d          }t          |j        d           t          |j        d           t	          j        d|j                  }t          t          |          dk               d S )Nrg  r  r    r1   zlogpdf\(x, a, loc=0, scale=1\))	rO  r   numargsr  refindallr  r	   r  rs   rk  r   s      r3   test_signature_inspectionz1TestSubclassingNoShapes.test_signature_inspection#  sr     !g...[(!,,,['---j:$,. .CAr5   c                     t          d          }t          |j        d           t          |j        d           t	          j        d|j                  }t          t          |          dk               d S )Nrg  r  rg   r  z!logpdf\(x, a, b, loc=0, scale=1\)r    )	ra  r   r  r  r  r  r  r	   r  r  s      r3   test_signature_inspection_2argsz7TestSubclassingNoShapes.test_signature_inspection_2args#  sr     "w///[(!,,,['000j=$,. .CAr5   c                 >    t          t          t          d           d S )Nrg  r  )r`  r  r[  rx   s    r3   0test_signature_inspection_2args_incorrect_shapeszHTestSubclassingNoShapes.test_signature_inspection_2args_incorrect_shapes#  s    i7;;;;;;r5   c                 x     G d dt           j                  }t          t          |fi t	          d           d S )Nc                       e Zd ZddZdS )>TestSubclassingNoShapes.test_defaults_raise.<locals>._dist_genr	  c                     dS rQ  r;   rR  s      r3   rS  zCTestSubclassingNoShapes.test_defaults_raise.<locals>._dist_gen._pdf#      rr5   N)r	  rT  r;   r5   r3   r  r  #  s(             r5   r  rg  r  r9   rF   r`  r  r  rs   r  s     r3   test_defaults_raisez+TestSubclassingNoShapes.test_defaults_raise#  Z    	 	 	 	 	+ 	 	 	 	iAAd.@.@.@AAAAAr5   c                 x     G d dt           j                  }t          t          |fi t	          d           d S )Nc                       e Zd Zd ZdS )>TestSubclassingNoShapes.test_starargs_raise.<locals>._dist_genc                     dS rQ  r;   )rs   r_   r1   r0  s       r3   rS  zCTestSubclassingNoShapes.test_starargs_raise.<locals>._dist_gen._pdf#  r  r5   NrT  r;   r5   r3   r  r  #  #            r5   r  rg  r  r  r  s     r3   test_starargs_raisez+TestSubclassingNoShapes.test_starargs_raise#  r  r5   c                 x     G d dt           j                  }t          t          |fi t	          d           d S )Nc                       e Zd Zd ZdS )<TestSubclassingNoShapes.test_kwargs_raise.<locals>._dist_genc                     dS rQ  r;   )rs   r_   r1   r  s       r3   rS  zATestSubclassingNoShapes.test_kwargs_raise.<locals>._dist_gen._pdf#  r  r5   NrT  r;   r5   r3   r  r  #  r  r5   r  rg  r  r  r  s     r3   test_kwargs_raisez)TestSubclassingNoShapes.test_kwargs_raise#  r  r5   N)r   r   r   r  r  r   r   r  DOCSTRINGS_STRIPPEDr  r  r  r  r  r  r;   r5   r3   r  r  #  s        2 2 28 8 8
 [+4HII  JI [+4HII  JI< < <B B BB B BB B B B Br5   r  r  c                      g d} t           j        D ]l}t          t           |          }t          |t           j        t           j        z            r.| D ]+}t          t          j        ||j	                  d u            ,md S )N)z,\s*,z\(\s*,z^\s*:)
r9   rR   r.  rD  rE   rF   r	   r  searchr  )badonesdistnamerB   regexs       r3   r  r  #  s    ---GM @ @uh''dU.1DDFF 	@  @ @	%66$>????	@ @r5   c                      t          t          j                            t          j        dd          d           t          t          j                            t          j        dd          d           d S )Nr`   r  r   r  r]   r    )r   r9   rD  r   rj   r  r  rX  r;   r5   r3   test_infinite_inputr  $  sR    ((R88!<<<
377;;;;;r5   c                      t           j                            t           j                            dd          d          } t	          | d           d S r  )r9   lomaxr   rq   r   r
  s    r3   test_lomax_accuracyr  $  s=    22A66AAvr5   c                      t           j                            t           j                            dd          d          } t	          | d           d S r  )r9   r<  r   rq   r   r
  s    r3   test_truncexpon_accuracyr  $  sC    U-11&!<<a@@AAvr5   c                      t           j                            t           j                            dd          d          } t	          | dd           d S )Nr  r    r  r?  rW  )r9   r+  r  r   r   r
  s    r3   test_rayleigh_accuracyr  $  sF    5>,,Q22A66A3++++++r5   c                     t          j        d          5 } t          j        d           t          j                            dd           t          j                            dd           t          j                            dd           t          j                            t          j
         d           t          |           }t          |d           ddd           dS # 1 swxY w Y   dS )zregression test for gh-6219T)re  alwaysr|   r   r   N)rw  rx  ry  r9   r,  rq   ri   r   r   rj   r  r  r   )r  number_of_warnings_throwns     r3    test_genextreme_give_no_warningsr  $  s    
	 	-	-	- 3h'''R###R###R###---$'FF!.2223 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3s   B?C""C&)C&c                  P   t          j        t          j                            dt           j        ddgd                    } t           j        }t           j        dz  dz  }dt          j        d          z  t          j	        d	          z  t           j        d	z  z  }d
}||||g}t           j        gdz  x}}g d}t          | d d df         |d           t          | d d df         |           t          | d d df         |d           t          | d d d	f         |           d S )Nr   r    rX  r  r  rg   r3  rV  r   r=  r   )r   r    r   r3  r   r  r   )rj   r   r9   r,  r  euler_gammark   r  r   zetar   r   )	r   r   rA  rB  rC  ref_0ref_1ref_3ref_2s	            r3   test_moments_gh22400r  &$  s#   
 *U%++S"&!T,BF+SS
T
TC >D
%(Q,C

?W\!__,ruax7DD3d#EVHQJEEMMEC1Iu51111QQQTE"""C1Iu51111QQQTE"""""r5   c                     d} t           j                            d          }t          |d| z  dz   d           t           j                            d          }t          || dz   d           t           j                            d          }t	          |d           t           j                            d	d
          }t          || dz  t          j        d
          z   dz   d           t           j                            d
          }t          |d| z  dz   d           t           j                            d          }t          |d| z  dz   d           d S )Ngox?r  rg   r    r  r   r   rT  r  r`   r  r   r  rm  r  )r9   r,  rH   r   r   rj   r  )r  rp  s     r3   test_genextreme_entropyr  <$  sD   $K  &&AAq}q(u5555  ##AA{QU3333  %%AA  R 00AA{1}rvbzz1A5EBBBB  $$AAr+~)6666  %%AAr+~)666666r5   c                  R   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 )	Nr   r{  g'b%4gQ@ry  gؗҜ;r   g.l6?)r9   r,  r   r   r  )r_   rO   rs  s      r3   test_genextreme_sf_isfr  S$  s   2 	AAv&&AA-...				a	(	(BBAAu%%AA()))				a	'	'BBAAq!!AA-...				a	#	#BBr5   c                  l    d} t           j                            | dd          }t          |d           d S )Nr  rg   r   gH&8>)r9   ri
  r   r   )probr  s     r3   test_burr12_ppf_small_argr  $  s9    D|a++H H455555r5   c                      dd} t          j        g d          }t          j                            |d|           \  }}}t          |dd           |dk    sJ t          |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                 ,    t          | |||dd          S )Nr  )r0  dispxtolftol)r&   )r  x0r0  r  s       r3   r  z&test_invweibull_fit.<locals>.optimizer$  s    D"4dUKKKKr5   )r    r  rg   r  rt  r   r  r   r   r  r`   rV  @   c   )r   r  g`?rt  r   g [@N)r;   r   )rj   r   r9   r$  r   r   )r  r_   r  re   rf   s        r3   test_invweibull_fitr  $  s    .L L L L 	HHHIIA$((i(HHMAsEAxd++++!8888E8$//////r5   ))r   r  g7Ըh?)r   r  gz]r>)r   g     "@goBڱ9)r  r  g$=cC;c                 j    t           j                            | |          }t          ||d           d S r   )r9   r$  r   r   )r_   r  rX   rD  s       r3   test_invweibull_sfr  $  s7     ""1a((HHhU333333r5   zp, c, expected)r|   r  g9?)geK<r   g@c                 j    t           j                            | |          }t          ||d           d S r   )r9   r$  r  r   )rv  r  rX   rD  s       r3   test_invweibull_isfr	  $  s7     ##Aq))HHhU333333r5   z	df1,df2,x)r   rQ  rT  rD  r   r  r  r  rp  c                 ^   d}t           j                            || |          }t           j                            || ||          }t	          ||d           t           j                            || |          }t           j                            || ||          }t	          ||d           d S )Nr   r  r   rv  )r9   r  rq   r  r   ri   )df1df2r_   r	  expected_cdfcalculated_cdfr   calculated_pdfs           r3   test_ncf_edge_caser  $  s     
B7;;q#s++LY]]1c333NL.u==== 7;;q#s++LY]]1c333NL.t<<<<<<r5   c                  l    t           j                            ddd          } t          | dd           d S )Nrg   r3  r   g     `E@r  r   )r9   r  rA  r   )r  s    r3   test_ncf_variancer  $  s6     		aAAAu5))))))r5   c                      t           j                            dddd          } d}t          |t	          j        | d                     d S )Nr  r3  r  gffffff>@g&4I,)?)decimals)r9   r  rq   r   rj   round)	scipy_val	check_vals     r3   test_ncf_cdf_spotcheckr  $  sG    
 	b!R..IIIrx	A>>>?????r5   c                      t          j        ddd          } d| d<   dt          j        j        | gR  }fd| D             }t          ||           d S )Nr   r    r  r  )r]   rg   r   r   r    c                 >    g | ]}t          j        j        |gR  S r;   )r9   r  r   )rA   xipars     r3   rC   z,test_ncf_ppf_issue_17026.<locals>.<listcomp>$  s-    	.	.	.b%)-
!S
!
!
!	.	.	.r5   )rj   rY  r9   r  r   r   )r_   r  q0r  s      @r3   test_ncf_ppf_issue_17026r  $  sl    
Aq#AAaD
C	a#A	.	.	.	.A	.	.	.BArr5   c                   2    e Zd Zd Zd Zd Zd Zd Zd ZdS )TestHistogramc                 J   t           j                            d           t          j        g dd          }t	          j        |          | _        t          j                            dddd	          }t          j        |d
          }t	          j        |          | _	        d S )Nr   )r    rg   rg   r   r   r   r   r   r   r   r   r   r   r   r   r3  r3  r3  r3  r  r  r  r  r  r  r  binsrT  r  r  {   rA  r=  )
rj   r   r   	histogramr9   rG   templater  r   norm_template)rs   r%  r   norm_histograms       r3   r7  zTestHistogram.setup_method$  s    
	t L "@ "@ "@FGI I I	*955z~~#Su3~OOd444"/??r5   c                    t          j        g d          }t          j        g d          }t          | j                            |          |           t          | j                            d          d           t          | j                            d          d           t          | j                            d          d           t          | j                            d          d           t          j        d	d
d          }t          | j                            |          t          j
                            |dd          d           d S )Nr   r|   rT  r  r   r  r  r;  r  r9  rU        @r  r  r  r  r%        !@r  r6  )r   r   rM  rM  {Gz?r-  Q?r.  {Gz?r/  rQ  rQ  r/  r/  r.  r.  r.  r.  r   r   r%  r.  r,  r  r   r  r   rg   r`   rT  r  rd   r]   r   )rj   r   r   r   r&  ri   r   rY  r'  r9   r  )rs   rX  
pdf_valuesr_   s       r3   rR  zTestHistogram.test_pdf%  sX    M M M N NZ !I !I !I J J

 	))&11:>>> 	DM--c22H===DM--c22H=== 	DM--c22H===DM--d33X>>>KAr""*..q11
qc==C	I 	I 	I 	I 	I 	Ir5   c                 6   t          j        g d          }t          j        g d          }t          | j                            |          |           t          | j                            |dd                   |dd                    t          j        ddd          }t          | j                            | j                            |                    |           t          j        ddd          }t          | j                            | j                            |                    |           t          j        d	dd
          }t          | j                            |          t          j
                            |dd          d           d S )Nr*  )r   r   r   r  rM  r-  r.  
ףp=
?Q?{Gz?r  r|   r  (\?RQ?=
ףp=?)\(?r  rT  rT  rg   r   rT  r  ra   r   r   r`   r  rd   r]   r   )rj   r   r   r   r&  rq   r   rY  r'  r9   r  )rs   rX  
cdf_valuesr_   s       r3   r  zTestHistogram.test_cdf_ppf"%  s    M M M N NZ !M !M !M N N

 	))&11:>>>))*QrT*:;;VAbD\JJJ KS#&&))$-*;*;A*>*>??CCCKS#&&))$-*;*;A*>*>??CCCKAr""*..q11
qc==C	I 	I 	I 	I 	I 	Ir5   c                 :   d}| j                             |d          }t          t          j        |dk               d           t          t          j        |dk              d|z  d	           t          t          j        |d
k              d|z  d	           t          t          j        |dk              d|z  d	           t          t          j        |dk              d|z  d	           t          t          j        |dk              d|z  d	           t          t          j        |dk              d|z  d	           t          t          j        |dk              d|z  d	           t          t          j        |dk              d|z  d	           t          t          j        |dk              d|z  d	           t          t          j        |dk              d|z  d	           t          t          j        |dk              d|z  d	           t          t          j        |d k              d!|z  d	           t          t          j        |d"k              d#|z  d	           t          t          j        |d$k              d%|z  d	           t          t          j        |d&k              d|z  d	           t          t          j        |d&k              d|z  d	           t          t          j        |d&k              d           d S )'Nr  r$  r   rT  r   r   rM  rQ  r   r  r-  r  r.  r]   r;  r2  r  r3  r9  r4  rU  r  rF  r+  r|   r  r  r  r5  r  r6  r  r7  r%  r8  r,  r  r  )r&  r   r   rj   rm  r   )rs   r2  rW  s      r3   rJ  zTestHistogram.test_rvs8%  s   """<<RVFSL))3///v}--x!|#FFFFv}--x!|#FFFFv}--x!|#FFFFv}--x!|#FFFFv}--x!|#FFFFv}--x!|#FFFFv}--y1}4HHHHv}--y1}4HHHHv}--y1}4HHHHv}--y1}4HHHHv}--y1}4HHHHv}--y1}4HHHHv}--y1}4HHHHv}--y1}4HHHHv}--y1}4HHHHv}--y1}4HHHHRVFSL))3/////r5   c                     t          d          D ]R}t          | j                            |          t	          j        dd                              |          d           Sd S )Nr   rT  r  rF  r   )r  r   r'  r  r9   r  r  r  s     r3   r
  zTestHistogram.test_munpN%  sx    q 	G 	GAD.44Q77!JsC0077::G G G G G	G 	Gr5   c                     t          | j                                        t          j                            dd          d           d S )NrT  r  rd   rF  r   )r   r'  rH   r9   r  rx   s    r3   rq  zTestHistogram.test_entropyS%  sT    *2244
**s#*>>T	K 	K 	K 	K 	K 	Kr5   N)	r   r   r   r7  rR  r  rJ  r
  rq  r;   r5   r3   r   r   $  s        @ @ @ I I I4I I I,0 0 0,G G G
K K K K Kr5   r   c                     ddgg d}} t          j        | |fd          }t          j                            |                    ddg          ddg           |                                dk    sJ t          j        | |fd          }t          j                            |                    ddg          d	           |                                d
k    sJ d}t          j        t          |          5  t          j        | |f          }|                                d
k    sJ 	 d d d            n# 1 swxY w Y   t          j        | g df          }|                                dk    sJ d S )Nr    )r   r    r  F)r  r|   rS  r  Tg'^P?g     H@z(Bin widths are not constant. Assuming...r  r<  )
r9   rG   rj   r  r   ri   rH  r   r}  rz  )countsr#  rB   rp  s       r3   test_histogram_non_uniformr?  X%  s   FLLLDFvtne<<<DJtxxc
33c6]CCC;;==Avtnd;;;DJtxxc
33V<<<;;==F"""" 9G	nG	4	4	4 ' '!64.11{{}}&&&&&' ' ' ' ' ' ' ' ' ' ' ' ' ' '
 vyyy122D;;==As   61D44D8;D8c                   f    e Zd Zd Zej                            dddg          d             Zd ZdS )TestLogUniformc                 ~   t           j                            d          }t          j        dd          }|                    d|          }t           j                            d          }t          j        dd          }|                    d|          }t          ||           t          j        t          j	        |          d          \  }}d|
                                cxk    r|                                cxk    rd	k    sn J t          j        t          j        |          d
z
            dk    sJ d S )N   0o[ r  r    r  r   r`   r"  r!  iL  r   )rj   r   r   r9   
loguniformr   
reciprocalr   r%  log10r  r  r   rH  )rs   r   r  r   rv2r   rH  r   s           r3   
test_aliaszTestLogUniform.test_aliasp%  s    i##K00h00ff%cf22i##K00x11wwEw44c""",rx}}2666adhhjj6666DHHJJ6666$666666vbioo,--333333r5   r  r  r  c                 D   t           j                            d          }t          j                            ddd|          }t          j                            ||          \  }}}}|dk    sJ t          j                            |d|          \  }}}}|dk    sJ d S )	NrC  r]   r    r   r   r  rg   r  )rj   r   r   r9   rD  r   r   )rs   r  r   r   r1   r2   re   rf   s           r3   test_fit_overridez TestLogUniform.test_fit_override%  s     i##K00""33"GG +//F/CC1c5zzzz +//Af/MM1c5zzzzzzr5   c                    t           j                            d          }d\  }}t          j        ||          }|                    ddd          }t          |                    |                    |                    |           |	                    d          }t          |                    |                    |                    |           dt          j
        dd	          z  }|                    |          }t          |d d
         |dd          z  d           ||z
  t          j        |          t          j        |          z
  z  }	t          |                                |	           d S )Nl   eVi
P )gN~hr  r   r    r   r;  r  i8rS  r   r`   )rj   r   r   r9   rD  r  r   rq   r   r   r  ri   r  r   )
rs   r   r1   r2   rB   rq   r   r_   ri   r   s
             r3   test_overflowzTestLogUniform.test_overflow%  s<    i##$78811%% kk!QTk**#//555hhDh!!#//555 4%%%hhqkkCRCQRR("--- Aq		BF1II-.		T*****r5   N)	r   r   r   rH  r   r   r   rJ  rL  r;   r5   r3   rA  rA  o%  sd        4 4 4" [Xt}55
 
 65
+ + + + +r5   rA  c                      e Zd Zd Zej                            dddgddgddgg          d	             Zej                            d
ddg          d             Zej                            dg d          d             Z	ej                            dg d          d             Z
ej                            dg d          d             Zej                            dg d          d             Zej                            dg d          d             Zej                            dg d          d             Zej                            dg d          d              Zej                            dg d!          d"             Zd#S )$	TestArgusc                     t           j                            ddd          }t          t          j        d                                          |                                d           d S )Nr=  r  E  r   r   rW  )r9   argusr   r   r   r  s     r3   test_argus_rvs_large_chiz"TestArgus.test_argus_rvs_large_chi%  sR    KOOBSsO;;EKOO0022AFFHHaHHHHHHr5   zchi, random_stater]   rP  r     r;     c                     t           j                            |d|          }t          j        |d|f          \  }}t	          |dk               d S )Nr  r   rQ  rF  r9   rQ  r   rI  r	   )rs   r  r   r_   r   rv  s         r3   rJ  zTestArgus.test_rvs%  sN     KOOCcOEE|Aw001Dr5   r  r  rv  c                     t           j                            |dd          }t          j        |d           \  }}t	          |dk               d S )Nr  ie r   c                     dd| dz  z
  dz  z
  S )Nr    rg   r  r;   r   s    r3   rP   z.TestArgus.test_rvs_small_chi.<locals>.<lambda>%  s    a!Q$h#->)> r5   rF  rV  )rs   r  r  r   rv  s        r3   test_rvs_small_chizTestArgus.test_rvs_small_chi%  sN    
 KOOCcO??|A>>??1Dr5   zchi, expected_mean))r    g щi?)r`   g.憃?)r}  g\paP?)r~  g	?)r  g,6?c                 l    t           j                            |d          }t          ||d           d S )Nr    r  r   r   )r9   rQ  r   r   )rs   r  expected_meanr  s       r3   r  zTestArgus.test_mean%  s9     KS**=u555555r5   zchi, expected_var, rtol))r    gLH'B?r   )r`   go$?r  )r}  g
=>r-  )r~  g2W>r-  )r  gb~P>r-  c                 l    t           j                            |d          }t          |||           d S )Nr    r  r   )r9   rQ  rA  r   )rs   r  expected_varr   r  s        r3   r  zTestArgus.test_var%  s6     KOOCqO))<d333333r5   zchi, expected, rtol))r   gwM?r  )r|   gj?r  )r]   gHEA`!?r  )r~   gcف>r   )r  gF=r  )r  g߹Q#B=r  )rv  g0ޟ<r  )r  g*W%:r   c                 D    t          t          |          ||           d S r  )r   r$   )rs   r  rX   r   s       r3   test_argus_phi_small_chiz"TestArgus.test_argus_phi_small_chi%  s%     	
3======r5   zchi, expected))r|   )g-\/?g+p7c?g5Ϧ\?)rQ  )g
f?gi{?ٸ?gZ@7?)r]   )gB	?g]?gR{?)r~   )glͅw?g,?g[Z=?)r  )gz,?g]?gѪ1?)r  )g8<?gE-?gI?)rv  )g~7?g7B.?g9?)r  )g;?g9B.?g}5?c                     t          j        g d          }t          t          j                            ||          |d           d S )Nr]   r|   r   r   r   )rj   r   r   r9   rQ  ri   rs   r  rX   r_   s       r3   test_pdf_small_chizTestArgus.test_pdf_small_chi%  B     H___%%3//FFFFFFr5   ))r|   )gQ߹?e?grJ,?g򺗄?)rQ  )gΎd?g?gh?)r]   )g|
o?gvb?g+@?)r~   )g̈́On?ge?gC|[3?)r  )g!?m?g{jiH?gu53?)r  )g~m?g0-.?g3?)rv  )g~m?g7;B.?gŗ3?)r  )g~m?g9B.?gŗ3?c                     t          j        g d          }t          t          j                            ||          |d           d S )Nra  r  r   )rj   r   r   r9   rQ  r   rb  s       r3   test_sf_small_chizTestArgus.test_sf_small_chi%  sB     H___%%q#..uEEEEEEr5   zx, chi, expected))P?rl  g;=)rg  r  g[$7>)rW  r  g=~L]=)rW  r  gD,G
'=c                 j    t           j                            ||          }t          ||d           d S r  )r9   rQ  r   r   )rs   r_   r  rX   r   s        r3   test_sf_near_1zTestArgus.test_sf_near_1 &  s4     [^^As##H5111111r5   ))r|   )g+&?gk?gμm/?)rQ  )g?CL\f?g^N?g]AͭR?)r]   )ga>$?g;Lf?gaW?)r~   )gx?g 43n?gwY?)r  )go7🤎?g
+-oGn?gIQ9Y?)r  )gT?gGn?g@̋Y?)rv  )gU?g{Gn?giG͋Y?)r  )gDU?g {Gn?gH͋Y?c                     t          j        g d          }t          t          j                            ||          |d           d S )Nra  r  r   )rj   r   r   r9   rQ  rq   rb  s       r3   test_cdf_small_chizTestArgus.test_cdf_small_chi&  rd  r5   ))r|   )gQ)?g6{?r  )gB`"۹?)g?g8%?r  )r]   )g388?gV%?r   )r~   )g*j?gRS%?r   )r  )g!|?gZ%?r   )r  )gy[|?gG%?r   )rv  )g"3|?gW%?r   )r  )g!3|?gW%?r   c                 l    t           j                             |d          }t          |||           d S )Nr  r  r   )r9   rQ  r   )rs   r  rX   r   rI  s        r3   test_stats_small_chizTestArgus.test_stats_small_chi&  s9     kT22XD111111r5   N)r   r   r   rR  r   r   r   rJ  rY  r  r  r_  rc  rf  ri  rk  rm  r;   r5   r3   rN  rN  %  s       I I I
 [0#J#J#J3
  
  

 [UT4L11  21 [18 8 89 96 69 96
 [6B B BC C4 4C C4
 [2D D DE E> >E E> [	M 	M 	M	N 	NG G	N 	NG
 [	N 	N 	N	O 	OF F	O 	OF
 [	6 	6 	67 72 27 72
 [	P 	P 	P	Q 	QG G	Q 	QG
 [	C 	C 	C	D 	D2 2	D 	D2 2 2r5   rN  c                      e Zd Zd Zd Zd Zd Zej        	                    dg d          d             Z
ej        	                    dg d          d	             Zd
 Zej        	                    dg d          d             Zej                            d          ej        	                    dg d          ej        	                    dg d          ej        	                    dg d          d                                                 Zej        	                    dg d          ej        	                    dg d          d                         ZdS )TestNakagamic                 n    d}d}t           j                            ||          }t          |d           d S )Nr  r  gM+)r9   nakagamir   r   )rs   r-  r_   r  s       r3   r  zTestNakagami.test_logpdf+&  s<     ~$$Q++122222r5   c                     d}d}t           j                            ||          }t          |dd           t           j                            ||          }t          ||d           d S )Nr  rU  g"S+:r   r   )r9   rq  r   r   r  )rs   r-  r  r   rr  s        r3   r  zTestNakagami.test_sf_isf<&  sm     ^r2&&1>>>>^B''BU++++++r5   c                 v    d}d}d}t           j                            ||          }t          ||d           d S )Nr  r|   gUiּr  r   )r9   rq  r  r   )rs   r_   r-  r  r  s        r3   r  zTestNakagami.test_logcdfO&  sC    %&&q"--%000000r5   c                 v    d}d}d}t           j                            ||          }t          ||d           d S )NrF  rV  gD&`Ur  r   )r9   rq  r  r   )rs   r_   r-  r  r  s        r3   r  zTestNakagami.test_logsfW&  sC    %$$Q++s//////r5   zm, ref))r   g|Bd븿)r|   r  )r`   g/ۿc                 d    t          t          j                            |          |d           d S )Ng^~z=r   r   r9   rq  rH   rs   r  r  s      r3   rq  zTestNakagami.test_entropy_&  s/      	..q113WEEEEEEr5   ))r  g}Ô%I)r  gq_)g    ScAgB4U)g    cAgu1U)r  gb>%)r  gWҙ\c                 `    t          t          j                            |          |           d S rv   rv  rw  s      r3   test_extreme_nuzTestNakagami.test_extreme_nuq&  s*    
 	..q11377777r5   c                     t          j        t          j                            d                    sJ t          j        t          j                            d                    sJ d S )Nr  r  )rj   rG  r9   rq  _entropyrx   s    r3   test_entropy_overflowz"TestNakagami.test_entropy_overflowx&  sT    {5>22599:::::{5>226::;;;;;;;r5   znu, ref))r  g2H?)r  g:U?)r  g>c                 d    t          t          j                            |          |d           d S r  )r   r9   rq  r   )rs   r-  r  s      r3   r  zTestNakagami.test_mean|&  s/     	++B//5AAAAAAr5   z+Fit of nakagami not reliable, see gh-10908.rj  r-  )r  r  r  re   )r   r`   r}
  rf   )r  r   r  c                   
 d
t           j                            
|||d          t           j                                      \  }}}t	          ||d           t	          ||d           t	          ||d           fd}
fd}
fd}	t	           ||||          d	d
           t	           ||||          d	d
           t	           |	|||          d	d
           d S )Nra   9  r   r-  re   rf   r   rQ  r   c                     d| z  dz   t          j        d|z
  z            z  d| z  |dz  z  t          j        |z
            z  z   S )Nr   r    rg   rj   rm  )r-  re   rf   r\  s      r3   	dlogl_dnuz(TestNakagami.test_fit.<locals>.dlogl_dnu&  sR    UQY"&GcM):";";;d5!8mbfWs]&;&;;< =r5   c                     dt          j        |           z   t          d|           z
  z  dt          j        t          j        |z
  |z                      z  z   t          j        |z
  |z  dz            z
  S )Nr    r   rg   )rj   r  r   rm  r-  re   rf   r2  r\  s      r3   
dlogl_dlocz)TestNakagami.test_fit.<locals>.dlogl_dloc&  sv    RVBZZ)Ar*:*::;rvw}&=>>???@fw}59::; <r5   c                 h    dz  | z  |z  d| z  |dz  z  t          j        |z
  dz            z  z   S )Nr   rg   r   r  r  s      r3   dlogl_dscalez+TestNakagami.test_fit.<locals>.dlogl_dscale&  sG    !GbL5("fuz)BFGcMa3G,H,HHI Jr5   r   r  r   )r9   rq  r   r   r   )rs   r-  re   rf   nu_estloc_est	scale_estr  r  r  r2  r\  s             @@r3   ry  zTestNakagami.test_fit&  s[    .$$!+0t % E E%*^%7%7%@%@"----3////	5s3333	= 	= 	= 	= 	=	< 	< 	< 	< 	< 	<
	J 	J 	J 	J 	J 	J 			&'9==qtLLLL

67I>>MMMMVWi@@!$OOOOOOr5   c                    d}d}t           j                            ||||d          }t           j                            ||          \  }}}t	          j        |          }	t	          j        t	          j        ||z
  dz                      }
t          ||d           t          ||	d           t          ||
d           d S )	Nr|   ra   r  r  r  rg   r-  r   )	r9   rq  r   r   rj   r  r  r   r   )rs   re   rf   r-  ru  r\  r  r  r  loc_theo
scale_theos              r3   test_fit_nuzTestNakagami.test_fit_nu&  s    
 .$$!+0t % E E%*^%7%7B%7%G%G" 6'??WRWg&7A%=>>??
....5555	:D999999r5   N)r   r   r   r  r  r  r  r   r   r   rq  ry  r|  r  r  ry  r  r;   r5   r3   ro  ro  )&  s       3 3 3", , ,&1 1 10 0 0 [X	% 	% 	%& &F F	& &F [X	D 	D 	DE E8 8	E E8< < < [Y> > >? ?B B	? ?B [KLL[T???33[UNNN33[Wkkk22P P 32 43 43 MLP8 [UNNN33[Wkkk22: : 32 43: : :r5   ro  c                        e Zd Zd Zd Zd ZdS )TestWrapCauchyc                 L   t          j        ddgddgg          }t          j        dgdgg          }t          j                            ||          }|j        dk    sJ d t          j        ||f          D             }t          |                                |d	
           d S )NQ?rl  r|   r   rT  r  rt  c                 T    g | ]%\  }}t           j                            ||          &S r;   )r9   
wrapcauchyrq   )rA   rr  rE
  s      r3   rC   z>TestWrapCauchy.test_cdf_shape_broadcasting.<locals>.<listcomp>&  s?     < < <%b" )--b"55 < < <r5   r   r   )	rj   r   r9   r  rq   rB  nditerr   rA
  )rs   r  r_   rv  scalar_valuess        r3   test_cdf_shape_broadcastingz*TestWrapCauchy.test_cdf_shape_broadcasting&  s     HtTlS$K011HsecU^$$  A&&w&    < <)+Aq6):):< < <		=u======r5   c                 ~    t           j                            t          j        d          }t          |dd           d S )Nr  r|   r  r   )r9   r  rq   rj   rk   r   r  s     r3   test_cdf_centerzTestWrapCauchy.test_cdf_center&  s7      --3U++++++r5   c                    d}d}d}t           j                            ||g|          }d|z   d|z
  z  }t          |d         t	          j        |t	          j        |dz            z            t          j        z             t          |d         dt	          j        |t	          j        t          j        |dz  z
            z            t          j        z  z
             d S )NrT  r  r   r    r   rg   )r9   r  rq   r   rj   arctantanrk   )rs   rr  rs  r  rv  crs         r3   r\  zTestWrapCauchy.test_cdf&  s      "b1--!ea!e_!bi26"Q$<<88>???!a")Brvbebdl/C/C,C"D"DRU"JJKKKKKr5   N)r   r   r   r  r  r\  r;   r5   r3   r  r  &  sF        
> 
> 
>, , ,L L L L Lr5   r  c                       G d dt           j                  }  | d          }t          t          d          5  |                                 d d d            d S # 1 swxY w Y   d S )Nc                       e Zd Zd ZdS )/test_rvs_no_size_error.<locals>.rvs_no_size_genc                     dS r   r;   rx   s    r3   _rvsz4test_rvs_no_size_error.<locals>.rvs_no_size_gen._rvs&  s    1r5   N)r   r   r   r  r;   r5   r3   rvs_no_size_genr  &  s#        	 	 	 	 	r5   r  rvs_no_sizer  z_rvs\(\) got (an|\d) unexpectedr  )r9   rF   r`  r  r   )r  r  s     r3   test_rvs_no_size_errorr  &  s        %-    "/}555K	y(J	K	K	K                   s   AA!$A!zdistname, argsc                 n   | t           v rt          j        d|  d           t          t          |           }t          |t          j                  rt          |          dk    rA |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.  r9   rD  rF   r  r
  r   rj   r  )r  r0  rB   a0b0r(  r)  a1b1r1   r2   s              r3   test_support_gh13294_regressionr  &  s8   777 0$,0 0 0 	1 	1 	15(##D$+,,  t99>>!T\4(FBRV$$$RV$$$
 f2t2T26222BR   R     t|T"1QQr5   c                  \   t           j                            g dg d          \  } }t          j        t          j         t          j         t          j         t          j        g          }t          j        t          j        t          j        t          j        t          j        g          }t          | |           t          ||           | j        |j        k    sJ |j        |j        k    sJ t           j                            g g           \  }}t          j        g           t          j        g           }}t          ||           t          ||           |j        |j        k    sJ |j        |j        k    sJ t           j                            g ddg          \  }}	t          j        dt          j        gz            }
t          j        dt          j        gz            }t          ||
           t          |	|           |j        |
j        k    sJ |	j        |j        k    sJ d S )N)r   r   r   r    )r    r    r    r   r   r   )	r9   r  r
  rj   r   r  r  r   rB  )r  r  ex_a0ex_b0r  r  ex_a1ex_b1r  r  ex_a2ex_b2s               r3   ,test_support_broadcasting_gh13294_regressionr  &  s   Zmmm<<FBHrvgw899EHbfbfbfbf566EUU8u{""""8u{""""ZB''FB8B<<"5EUU8u{""""8u{""""Zrd33FBHQxZ  EHQxZ  EUU8u{""""8u{""""""r5   c                  2   ddg} dgdgdgg}t          t          j                            | |          ddgddgddgg           t	          j        d          } t	          j        d          }t          j                            | |          j        dk    sJ d S )	Nr   rT  r   r  r  r  r?   r  )r   r9   r  rA  rj   r@
  rB  rd   s     r3   *test_stats_broadcasting_gh14953_regressionr  '  s    r(CTB4"EU++r2hR2r(-KLLL
(5//CHVE:>>#u%%+v555555r5   )gn!	g:I"<)gQ	@gE?c                     t          t          j                            |           |           t          t          j                            |            |           d S rv   )r   r9   cosinerq   r   )r_   rX   s     r3   test_cosine_cdf_sfr  &'  sJ     EL$$Q''222ELOOQB''22222r5   ))rv  gkM6O)r  g0ӭ!	)r>  g;'u(@c                     t          t          j                            |           |           t          t          j                            |           |            d S rv   )r   r9   r  r   r  )rv  rX   s     r3   test_cosine_ppf_isfr  .'  sL    
 EL$$Q''222EL$$Q''(33333r5   c                      t           j                            t          j         t          j        g          } t          | d           d S )NgrB)r9   r  r   rj   rk   r   )r  s    r3   test_cosine_logpdf_endpointsr  7'  s:    <//D d./////r5   c                      d t           D             } d t          D             }| |k    sJ d t          D             }d t          D             }||k    sJ d S )Nc                 B    h | ]\  }}t          |t                    |S r;   )rD  rL   rA   r  r   s      r3   	<setcomp>z*test_distr_params_lists.<locals>.<setcomp>B'  s:     4 4 474'c224$ 4 4 4r5   c                     h | ]\  }}|S r;   r;   r  s      r3   r  z*test_distr_params_lists.<locals>.<setcomp>D'  s    AAAgdATAAAr5   c                     h | ]\  }}|S r;   r;   r  s      r3   r  z*test_distr_params_lists.<locals>.<setcomp>G'  s    333wtQd333r5   c                     h | ]\  }}|S r;   r;   r  s      r3   r  z*test_distr_params_lists.<locals>.<setcomp>H'  s    999'$999r5   )r!   r"   r   r   )discrete_distnamesinvdiscrete_distnamescont_distnamesinvcont_distnamess       r3   test_distr_params_listsr  >'  s    4 4l 4 4 4AAAAA!6666633(333N99[999.......r5   c                      t           j                            dd           t           j                            dd          dk    sJ t           j                            dd          dk    sJ d S )Nr   r\   rw  r   )r  r1   r  )r9   r  _statsr  r  r;   r5   r3   test_moment_order_4r  L'  sr     
NAs+++ >  qA ..#5555 >1%%,,,,,,r5   c                      e Zd Zej        d             Zej                            dg d          d             Zej                            dg d          d             Z	ej        j
        ej                            d ej        d	d
ej        j                   ej        ddej        j
                   ej        ddej        j
                  g          d                         ZdS )TestRelativisticBWc                     t          j        t          t                    j        dz            }t           j                            |j        d          }|S )a&  Sample data points for pdf computed with CERN's ROOT

        See - https://root.cern/

        Uses ROOT.TMath.BreitWignerRelativistic, available in ROOT
        versions 6.27+

        pdf calculated for Z0 Boson, W Boson, and Higgs Boson for
        x in `np.linspace(0, 200, 401)`.
        z-data/rel_breitwigner_pdf_sample_data_ROOT.npyzx,pdf,rho,gammar  r  r|  s     r3   ROOT_pdf_sample_dataz'TestRelativisticBW.ROOT_pdf_sample_dataa'  sM     wNN!;<
 
 v  /@ AAr5   zrho,gamma,rtol))qVEB@gj+@rR  )FC@Gz @r  )ND@_LU?r*  c                     ||d         |k    |d         |k    z           }|d         |d         }}t          |t          j                            |||          |           d S )Nrhor  r_   ri   r  r   )r   r9   rel_breitwignerri   )rs   r  r  r  r   r   r_   ri   s           r3   test_pdf_against_ROOTz(TestRelativisticBW.test_pdf_against_ROOTt'  s     $!%(C/#G,57
 cDK3&**1c*??d	
 	
 	
 	
 	
 	
r5   zrho, Gamma, rtol))r  r  r*  )r  r  r*  )r  r  g&.>c                    d }t          j        ddd          }t          j                            |||          }t          j                            |||          } ||||z  |          }t          |||           d S )Nc                    t          j        |dz  |dz  |dz  z   z            }dt          j        d          z  |z  |z  |z  t           j        t          j        |dz  |z             z  z  }|| dz  |dz  z
  dz  |dz  |dz  z  z   z  S r  )rj   r  rk   )Er  Gammar  r\   s        r3   ri   zFTestRelativisticBW.test_pdf_against_simple_implementation.<locals>.pdf'  s    GAqDAqD5!8O455ERWQZZ!#e+e3EBGAqD5L1113AA1q(1a4q=899r5   rF  r@  r`   r  r   )rj   rY  r9   r  r   ri   r   )	rs   r  r  r   ri   rv  r_   r   r  s	            r3   &test_pdf_against_simple_implementationz9TestRelativisticBW.test_pdf_against_simple_implementation'  s    	: 	: 	: KdB''!%%aE%::#''3e'<<c!SY&&St,,,,,,r5   z	rho,gammar  r  r  r  r  r  r  c                    d}t           j                            |          }t          j                            ||d|          }t          j                            |d          }t          |d         |d         f||fd           |d	         dk    sJ t          j                            |d|
          }t          |d         |d           |d	         |d         fd|fk    sJ dS )zpTests fit for cases where floc is set.

        `rel_breitwigner` has special handling for these cases.
        l   s;
rl# r   )rf   r   r   r   r   rg   rQ  r   r    rB  r~   N)rj   r   r   r9   r  r   r   r   )rs   r  r  r   r   r   r   s          r3   test_fit_flocz TestRelativisticBW.test_fit_floc'  s    & #i##D))$((u4c ) 
 
 #''1'55QQ(3,TBBBB1v{{{{#''1U'CCA$////AAAu:------r5   N)r   r   r   r   r  r  r   r   r  r  r   r  r   r  r;   r5   r3   r  r  `'  se       ^  ^$ [ 
 
 
 
 
 
 [/ 2 2 2  - - - [[FL"F&+2B   FL!50A   FL &0A  

 . .  . . .r5   r  c                   Z    e Zd Zej                            dddg          d             ZdS )TestJohnsonSUr  )r  r  r  r  g={?g?5mV>gn?g\}\1@)g`fo@g$ں_@r   r    gvUgN<fj?gwCg~l
@c                 v    t          j        j         |d d         ddi}t          ||dd          d           d S )Nr   r  r  r  r   )r9   rU  r   )rs   r  r   s      r3   test_moment_gh18071z!TestJohnsonSU.test_moment_gh18071'  sF     o#T"1"X>v>>T!""XE222222r5   N)r   r   r   r   r   r   r  r;   r5   r3   r  r  '  sZ        [VOJ&K L L
3 3L L
3 3 3r5   r  c                   2   e Zd Zd Zej                            dddg          ej                            dddg          ej                            dddg          ej                            dddg          d                                                 Zd	S )
TestTruncParetoc                 H   d\  }}t          j        dd          }t          j        ||                              |          }t          j        |                              |          t          j        |                              |          z  }t          ||           d S )N)?333333@r  r  )rj   rY  r9   truncparetori   r  rq   r   )rs   r2   r  r_   r   r  s         r3   rR  zTestTruncPareto.test_pdf'  s    1KS!!1%%))!,,l1oo!!!$$u|A':':1'='==S!!!!!r5   r   TFr>  r  r  c                    t           j                            d          }d\  }}}}	t          j        ||||	          }
|
                    d|          }i }|r||d<   |r|	|d<   |r||d<   |r||d	<   |rU|rS|rQ|rOd
}t          j        t          |          5  t          j        j	        |fi | d d d            d S # 1 swxY w Y   d S t          t          j        |fi | d S )Nl   Z#Lm )r  r  r    r  rd   r  r   r   r   r   r	  r  r  )rj   r   r   r9   r  r   r   r   r  r   r   )rs   r   r>  r  r  r   r2   r  re   rf   rB   r   r   rp  s                 r3   ry  zTestTruncPareto.test_fit'  s}    i##$788+1c5 A3e<<<xxSsx33 	DL 	#"DN 	DJ 	DJ 	Ky 	KU 	Ku 	KKG|7;;; 4 4!%d33d3334 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 *%*;TJJTJJJJJs   B>>CCN)r   r   r   rR  r   r   r   ry  r;   r5   r3   r  r  '  s        " " " [Yu66[[4-88[WtUm44[WtUm44K K 54 54 98 76K K Kr5   r  c                       e Zd Zd ZdS )
TestKappa3c                     dt           j                            dd          z
  }t           j                            dd          }t	          ||           d S )Nr    r|   g     j@)r9   kappa3rq   r   r   )rs   sf0ry  s      r3   r  zTestKappa3.test_sf'  sK     %,""3,,,looc3''S!!!!!r5   N)r   r   r   r  r;   r5   r3   r  r  '  s#        " " " " "r5   r  c                       e Zd Z ej        dd          Z ej        d          Z ej        d          Zd Z	d Z
d Zd Zd Zd	 Zd
 Zd Zd ZdS )TestIrwinHallr   r    r`   c                 V    t          | j                            d          d           d S )Nr  )r   g?r   gQ)r   ih10r9   rx   s    r3   test_stats_ih10zTestIrwinHall.test_stats_ih10'  s)     	TY__V446JKKKKKr5   c                      g d} fdt          t          |                    D             }t          ||            j                            d          }d}t          ||           d S )N)	r   gUUUUU9@g     0a@i  gx@gn۶mˠ@g    :PAg8)Ag   SAc                 J    g | ]}j                             |d z              S rc
  )r  r  )rA   ru  rs   s     r3   rC   z3TestIrwinHall.test_moments_ih10.<locals>.<listcomp>(  s-    CCCQ49##AaC((CCCr5   r=  g2⍚H)r  r  r   r  r  )rs   rH  r  m50	m50_exacts   `    r3   test_moments_ih10zTestIrwinHall.test_moments_ih10(  s~    - - - DCCC%D		2B2BCCCWd+++ir""H	S),,,,,r5   c                     t          j        ddd          }| j                            |          }| j                            |          }t          ||d           d S Nr   r    ra   r`   maxulp)rj   rY  unifri   ih1r   )rs   ptspdf_unifpdf_ih1s       r3   test_pdf_ih1_unifzTestIrwinHall.test_pdf_ih1_unif(  sX     k!Q$$9==%%(,,s##Whr::::::r5   c                    t          j        d          }d}t          j        dd|          }t          j        dd|          }d||dz   dz  d          z
  ||dz   dz  d <   |                    |          }t          ||d           d S )Nrg   r^   r   r    r`   r   )r9   r  rj   rY  ri   r   )rs   ih2nptsr  rX   pdf_ih2s         r3   test_pdf_ih2_trianglez#TestIrwinHall.test_pdf_ih2_triangle(  s    oa  k!Q%%;q!T**%&4!8/2B2B)C%C$(q!!"''#,,Whr::::::r5   c                     t          j        ddd          }| j                            |          }| j                            |          }t          ||d           d S r  )rj   rY  r  rq   r  r   )rs   r  cdf_unifcdf_ih1s       r3   test_cdf_ih1_unifzTestIrwinHall.test_cdf_ih1_unif)(  sV    k!Q$$9==%%(,,s##Whr::::::r5   c                     t          j        dd          }t          j        |          }|                    |dz            }t          j        dt          |                    }t          ||d           d S )Nr    r`   rg   r|   r   )rj   r  r9   r  rq   r  r  r   )rs   ru  ihih_cdfexacts        r3   r\  zTestIrwinHall.test_cdf1(  sg    Ia_QA	#s1vv&&VU2666666r5   c                 2   g d}t          | j                            t          j        d                    |d           t          | j                            d          dd           d}t          | j                            d          |d           d S )	N)r   g\xO~>g-P2?g8ʝi?gg{Ȇ?r|   g&M?g׈Y?gB?gDl?r    r  r`   r   r]   g+|<rT  g#@)r   r  rq   rj   r  r>  s      r3   test_cdf_ih10_exactz!TestIrwinHall.test_cdf_ih10_exact:(  s    ( ( ( 	TY]]29R==994KKKKTY]]4002EbQQQQ1TY]]5113rBBBBBBr5   c                     g d}|dg|d d d         z   z  }t          | j                            t          j        d                    |d           d S )N)r   g4V>gP3NV?g*8f?gNgX?gjS?r   r  r`   r   )r   r  ri   rj   r  r  s     r3   test_pdf_ih10_exactz!TestIrwinHall.test_pdf_ih10_exactJ(  s\     JII$ttt*,,TY]]29R==994KKKKKKr5   c           	      ,   t          | j                            t          j        d                    d| j                            t          j        d                    z
             d}t          | j                            d          |d           d S )Nr  r    rT  r]   r`   r   )r   r  r   rj   r  rq   r   )rs   r  s     r3   test_sf_ih10_exactz TestIrwinHall.test_sf_ih10_exactQ(  st    	RYr]]33QryQS}}9U9U5UVVV 2TY\\$//R@@@@@@r5   N)r   r   r   r9   r  r  r  r  r  r  r  r  r  r  r\  r  r  r  r;   r5   r3   r  r  '  s        5=AD
%/!

C5?2DL L L- - -
; 
; 
;; ; ;; ; ;7 7 7C C C L L LA A A A Ar5   r  c                       e Zd Zd ZdS )TestDParetoLognormc                    d\  }}}}}t          j        ||||          }t          j                            |                    |          d           t          j                            |                    |          d           d S )N)r  r   ro  r  rg   g6Z?g@#
?)r9   dpareto_lognormrj   r  r   ri   rq   )rs   r_   r  rO   r1   r2   rB   s          r3   test_against_Rz!TestDParetoLognorm.test_against_R[(  su     ,1aA$Q1a00

""488A;;0CDDD

""488A;;0CDDDDDr5   N)r   r   r   r  r;   r5   r3   r  r  Z(  s(        E E E E Er5   r  r  ))r  NNNN)r  NNNN)r(  NNNN)r  NNNN)r  NNNNc                    | \  }}}}}t          j        d          }|pd}|pd}|pd}|pd}t          t          |          }t	          t
                    |         } || }	t          j        ||          }
|	                    |	                    |
                    }t          ||
||           dt          j        ||d          z
  }
|	                    |	                    |
                    }t          ||
||           d S )	Nr|   iir   r  rn  r    r  )
rj   rF  r.  r9   r  r   r  r   r  r   )r  r  lp1lp2r   r   lpmrB   r  dist_frozenr  r   s               r3   test_sf_isf_overridesr%  o(  s
    &*"Hc3d
(3--C
+C
*C91D=5D5(##D(^^H%F$-K +c3

C
..--
.
.CC4d3333 bk#sB''
'C
..--
.
.CC4d333333r5   rv   )NF(  r  rw  r  r   r	  pathlibr   r
  r
  r_  numpy.testingr   r   r   r   r   r	   r
   r   r   r   r   r   r   r`  numpyrj   r   r   numpy.lib.recfunctionsr   r8   r   scipy._lib._utilr   r1  r   r   r   r   scipy.statsr9   !scipy.stats._distn_infrastructurer   scipy.stats._constantsr   scipy.stats.distributionsscipy.specialr   r   r   scipy.stats._distr_paramsr   r   test_discrete_basicr!   r"   scipy.stats._continuous_distnsr#   r$   scipy.optimizer%   r&   r'   	itertoolsr(   flagsoptimizer  rb  r  r  r4   r<   rY   r[   r   r  r   r   r1  r3  r^  r  r  r  r  r  r  r5  rA  rh  r  r  r  r  r  r  r  r+  r2  rL  rQ  rd  rn  r  r  r  r-  rQ  r\  r_  r  r  r  r  r
  r#  r9  rG  ro  ry  r  r  r  r  r  r  r  r  r  r,  rT  r  r  r  rk   r  r  r  r  r  r  r-  r:  r>  rB  rM  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&
  r0
  rN
  rS
  rZ
  ra
  ry
  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/  r3  r;  r=  r@  rB  rD  rM  rF   rO  rV  r[  ra  re  r  r  r  r  r  r  r  r  r  r  r  r  r  r	  r  r  r  r  r   r?  rA  rN  ro  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r%  r;   r5   r3   <module>r7     s!     				 



        				  : : : : : : : : : : : : : : : : : : : : : : : : : :
  * * * * * *     " " " " " " " " 4 4 4 4 4 4       / / / / / /3 3 3 3 3 3 3 3 3 3 3 3       8 8 8 8 8 8 ( ( ( ( ( (         0 0 0 0 0 0 0 0 0 0 ; ; ; ; ; ; ; ; > > > > > > > > C C C C C C C C = = = = = = = = = =       i(1,  |x'M.>h.>.@.@H.L )6z'B $$ $ $ $8 8 8
  G@ G@ G@ G@ G@ G@ G@ G@T4 4 4 4D. . .. > > >? ?

6 
6? ?

69 9 9 9 9 9 9 9>G, G, G, G, G, G, G, G,T* * * * * * * *       >       O, O, O, O, O, O, O, O,d<B <B <B <B <B <B <B <B~D/ D/ D/ D/ D/ D/ D/ D/N# # # # # # # #<E; E; E; E; E; E; E; E;PG G G G G G G GT0 0 0 0 0 0 0 0@CD CD CD CD CD CD CD CDLHU HU HU HU HU HU HU HUT2 2 2 2 2 2 2 2"'K 'K 'K 'K 'K 'K 'K 'KT> > > > > > > >*7D 7D 7D 7D 7D 7D 7D 7Dt
C 
C 
C 
C 
C 
C 
C 
CA2 A2 A2 A2 A2 A2 A2 A2H4 4 4 4 4 4 4 4DI6 I6 I6 I6 I6 I6 I6 I6X* * * * * * * *. G  G  G  G  G  G  G  GFK" K" K" K" K" K" K" K"\=I =I =I =I =I =I =I =I@`& `& `& `& `& `& `& `&FhD hD hD hD hD hD hD hDVD D D D D D D D,D D D D D D D D,U< U< U< U< U< U< U< U<p%( %( %( %( %( %( %( %(P"B "B "B "B "B "B "B "BJT# T# T# T# T# T# T# T#nt: t: t: t: t: t: t: t:nUO UO UO UO UO UO UO UOp< < < < < < < <~'* '* '* '* '* '* '* '*TO@ O@ O@ O@ O@ O@ O@ O@d/ / / / / / / /2!, !, !, !, !, !, !, !,HME ME ME ME ME ME ME ME`5 5 5 5 5 5 5 5.v0 v0 v0 v0 v0 v0 v0 v0r7F 7F 7F 7F 7F 7F 7F 7FtT6 T6 T6 T6 T6 T6 T6 T6n9J 9J 9J 9J 9J 9J 9J 9Jx0D 0D 0D 0D 0D 0D 0D 0DfDD DD DD DD DD DD DD DDN( ( ( ( ( ( ( (Vp0 p0 p0 p0 p0 p0 p0 p0faJ aJ aJ aJ aJ aJ aJ aJH*9 *9 *9 *9 *9 *9 *9 *9Z+G +G +G +G +G +G +G +Gl 7271ag;		
	gbga25k""	"	
	1RU7$$	$		AbeG%%	%		GBGAqwK((	(	* + 	7271c"%i=!!7271c"%i=)))7271c"%i=)))50gbgaRUm,,,g53ru9...	9; < "WRWQruW


r'"'!BE'"""
#

WRWQruW%%%
&

#gbgag&&&
'

D25)))
*

eGBGAbeG,,,
-

wrwqw///
0

 '!  H^V ^V ^V ^V ^V ^V ^V ^VB6 6 6 6 6 6 6 6&5 5 5 5 5 5 5 5:
8 
8 
8 
8 
8 
8 
8 
8s5 s5 s5 s5 s5 s5 s5 s5l/, /, /, /, /, /, /, /,d	E 	E 	E 	E 	E 	E 	E 	E       /G /G /G /G /G /G /G /Gd7B 7B 7B 7B 7B 7B 7B 7BtC C C C C C C CDf" f" f" f" f" f" f" f"Rw> w> w> w> w> w> w> w>t	3 	3 	3P6 P6 P6 P6 P6 P6 P6 P6fB= B= B= B= B= B= B= B=J& & & & & & & &>$ $ $ $ $ $ $ $*W+ W+ W+ W+ W+ W+ W+ W+t) ) ) ) ) ) ) )       , , , mG mG mG mG mG mG mG mG`e, e, e, e, e, e, e, e,Pw; w; w; w; w; w; w; w;t]H ]H ]H ]H ]H ]H ]H ]H@", ", ", ", ", ", ", ",JB& B& B& B& B& B& B& B&J6 6 6 6 6 6 6 68;" ;" ;" ;" ;" ;" ;" ;"|u0 u0 u0 u0 u0 u0 u0 u0p. . . . . . . ..SF SF SF SF SF SF SF SFl. . . . . . . .2J/ J/ J/ J/ J/ J/ J/ J/Z ?  ?  ?  ?  ?  ?  ?  ?Fh" h" h" h" h" h" h" h"V6 6 6 6 6 6 6 6(0 0 0 0 0 0 0 0DN N N N N N N NRB RB RB RB RB RB RB RBj2, 2, 2, 2, 2, 2, 2, 2,jC C C C C C C CL>6 >6 >6 >6 >6 >6 >6 >6BD0 D0 D0 D0 D0 D0 D0 D0N	< 	< 	< '0EFF/ / GF/
< < <*5 5 585 5 5. . .: : :#4 #4 #4L  	, 	, 	,   )J J JK K
A 
AK K
A  # # #  07 7 72	- 	- 	-& & &9 9 9%& %& %&P- - -M M M" +
HBHo7889
HBHo7889xrx|4556
HBHk;/001	.  2 2 22 2 26 6 61 1 167 7 7: ); ; ;< <) )< <) : : :; ;
) ); ;
) ; ; ;< <! !< <!
  ( ); ; ;< <
) )< <
)
% % %*    $   
    %%   
    %%       %%   u9 u9 u9 u9 u9 u9 u9 u9p8B 8B 8B 8B 8B 8B 8B 8Bv '0DEE@ @ FE@< < <
    , , ,3 3 3# # #,7 7 7.) ) )X6 6 60 0 0D )F F FG G
4 4G G
4 )<<>? ?4 4? ?4
 !!!""""#___ = = = * * *@ @ @  ^K ^K ^K ^K ^K ^K ^K ^KB  .3+ 3+ 3+ 3+ 3+ 3+ 3+ 3+lA2 A2 A2 A2 A2 A2 A2 A2HP: P: P: P: P: P: P: P:fL L L L L L L L8	 	 	 )?[+HII    JI 4# # #26 6 6 ;467 73 37 73
 7 7 78 84 4	8 84
0 0 0/ / /- - -(W. W. W. W. W. W. W. W.t3 3 3 3 3 3 3 3 #K #K #K #K #K #K #K #KL" " " " " " " "]A ]A ]A ]A ]A ]A ]A ]A@E E E E E E E E*  "G "G "G H H
4 4H H
4 4 4r5   