
    S_fA                        d dl Z d dlZd dlmZ d dlZd dlmZ d dl	m
Z
 d dlZd dlm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mZmZmZmZmZmZm Z m!Z! d dl"m#Z# d dl$m%Z% 	 d	Z&e jN                  d
k  Z(g dZ)g dZ*g dZ+g dZ,g dZ-g dg dz   dgz   Z.e,e-e.z   dZ/g dZ0g dg dz   dgz   Z1e0e-e1z   dZ2h dZ3g Z4g dddZ5ddgg ddZ6 ejn                  e5e6gddg      D ]G  \  Z8Z9 ejt                  dmi e8de9iZ; ejx                  e;e9      Z=e4j}                  e= e?       f       I d  Z@ej                  j                  d! e@             ej                  j                  d"d#g      d$               ZCej                  j                  d! e@             d%        ZDd& ZEd' ZFej                  j                  ej                  j                  d( eF             d)               ZHej                  j                  d*e#      d+        ZIej                  j                  d,g d-      d.        ZJd/ ZKd0 ZLd1 ZMd2 ZNd3 ZOd4 ZPd5 ZQd6 ZRd7 ZSd8 ZTd9 ZUd: ZVd; ZWd< ZXd= ZYd> ZZd? Z[d@ Z\dA Z]dB Z^dC Z_dD Z`dE ZadF ZbdG ZcdH ZddI Zeej                  j                  dJg dK      ej                  j                  dLe#      dM               ZfdN ZgdO ZhdP ZidQ Zjej                  j                  dR        ZldS ZmdT ZndUg gdVd ggdWd ggdXd ggdYd ggdZd ggd[d ggd\d ggd]d ggd^dggd_g gd`g gdag gdbg gdcg gddg ggZodedfggdgg gdhdigggZpej                  j                  djeoepz         dk        Zqdl Zry)n    N)raises)IntegrationWarning)stats   )check_normalizationcheck_momentcheck_mean_expectcheck_var_expectcheck_skew_expectcheck_kurt_expectcheck_entropycheck_private_entropycheck_entropy_vect_scalecheck_edge_supportcheck_named_argscheck_random_state_propertycheck_meth_dtypecheck_ppf_dtypecheck_cmplx_derivcheck_picklingcheck_rvs_broadcastcheck_freezingcheck_munp_expect)distcont)rv_continuous_frozen   l        )
recipinvgaussvonmiseskappa4vonmises_line
gausshypernorminvgaussgeninvgaussgenhyperbolic	truncnormtruncweibull_min)studentized_rangekstwoksone
wrapcauchygenexpon)r'   r   r    r)   r(   r   r+   )exponpow	exponweibr!   r+   halfgennormgompertz	johnsonsb	johnsonsur   r)   r(   	kstwobignmielkencfnctpowerlognorm	powernormr   	trapezoidr   r    levy_stablerv_histogram_instancer'   )argusr,   r-   r!   r+   genhalflogisticr.   r/   r0   r   r2   r   r8   
truncexponr   r    r'   )alpha	betaprimebradfordburrburr12cauchycrystalballffisk
foldcauchy
genextreme	genpareto
halfcauchyinvgamma	jf_skew_tkappa3levylevy_l
loglaplacelomaxr3   nakagamir4   
skewcauchyttukeylambda
invweibullrel_breitwigner)r$   r1   r)   r(   r5   paretor7   r6   pearson3MLEMM)rA   r,   r-   r!   r+   r.   r/   r0   r1   r   r)   r(   r2   r9   r3   r4   ncx2r6   r7   rdistr   r8   truncparetor   r    r'   )r>   r?   rA   rB   rC   rD   rE   rF   rG   rH   rI   rJ   rK   rL   rM   rN   rO   rP   rQ   r3   rR   r4   r5   rS   rT   r_   rV   )r$   r1   r)   r(   rX   r7   r6   >4   rE   rT   chir4   r5   betachi2r]   ricer;   gammar)   r(   r^   cosinedgammaerlangrO   gennormmaxwelldweibullgengammarK   loggammalogisticrR   rY   powerlawskewnormr   r?   rI   rL   r0   r2   r%   rG   r!   rJ   
loguniform
reciprocalr#   r.   rU   halflogisticr"   r$   r    rW   r&   r'   r:   )r      rs      rt   rt      ru   ru   ru   r   r   r   r   r      rv   rv   rv      rw   rw      rx   	   rx   )abins)r   r   
   TFdensity)r}   c               #   @  K   t         d d  t        z   D ]  \  } }| dk(  r| t        v r3t        j                  | |t        j
                  j                         G| t        v r3t        j                  | |t        j
                  j                         | |f  y w)Nr9   marks)	r   histogram_test_instancesdistslowpytestparammarkslow	distxslowxslow)distnameargs     Glib/python3.12/site-packages/scipy/stats/tests/test_continuous_basic.pycases_test_cont_basicr      s     !!'??  #}$!,,xFKK4D4DEE",,xFKK4E4EFFC- s   BBzdistname,argzsn, n_fit_samples)i     c                    	 t        t        |       }t        j                  j                  d      } |j                  |||d} |j                  | \  }}| dvrt        |||       t        |||        t        |||        t        |||        t        |||        t        |||        t        |||        t        |||        t        |||        t!        |||        t#        |||        d}	| dk(  rt%        |j&                  ||	|       n| dk7  rt%        | ||	|       d}
|j(                  |j*                  |j&                  |j,                  |j.                  g}ddd	d
ddd}|j1                  | d      }| dk(  rd}n| dk(  rd}t3        ||||
|       t5        ||       | dv rt6        rt9        j:                  d       nt=        ||       t?        ||       | dvrtA        |||        |jB                  dk(  rtE        ||       |jF                  jH                  t        jJ                  jH                  k7  r | dk7  rtM        ||t        jJ                         tO        jP                         5 }|jS                  tT        d       |jS                  tT        d       |jS                  tV        d       tY        ||       d d d        t[        ||       t]        ||       t_        |||       ta        ||       | tb        vrte        ||       | dk7  rtg        |||        dD ];  }| th        |   vrtk        |||d | |       | tl        |   vs+to        |||d | |       = y # t        $ r | }d} Y Rw xY w# 1 sw Y   xY w)Nr:   i )sizerandom_state>   laplace_asymmetric{Gz?r#   r   r               ?g	@333333?      @)weibull_maxrO   rX   r_   rU   r:         ?rV   r   r)   )rt   rW   fails on Linux 32-bit)r2   r(   r4   r   r   z The occurrence of roundoff errorzExtremely bad integrandzinvalid valuer%   rZ   )8getattrr   	TypeErrornprandomRandomStatervscheck_sample_meanvar_check_cdf_ppfcheck_sf_isfcheck_cdf_sfcheck_ppf_isf	check_pdfcheck_pdf_logpdfcheck_pdf_logpdf_at_endpointscheck_cdf_logcdfcheck_sf_logsfcheck_ppf_broadcastcheck_distribution_rvscdfpdflogpdflogcdflogsfgetr   r   	_IS_32BITr   skipr   r   r   numargscheck_vecentropy	__class___entropyrv_continuousr   nptsuppress_warningsfilterr   RuntimeWarningr   check_retrieving_supportr   r   r   fails_cmplxr   check_ppf_privateskip_fit_testcheck_fit_argsskip_fit_fix_testcheck_fit_args_fix)r   r   snn_fit_samplesdistfnrngr   mvr>   locscale_defaultsmethsspec_xxsupmethods                   r   test_cont_basicr      sU   
+)
 ))


'C
&**s"
5C6<<DAq--aC(&#x(h'h'&#x(fc8$VS(+!&#x8VS(+63)X.E**vzz3s;	]	" 	xeS9ZZ

FMM\\E "TC'*,F 	

8S!A<	W	VQ%6>,&&9+,vs#63 44fc8,~~%!!U%8%8%A%AAJ&fc5+>+>?				  .C

%'IJ

%'@A

>?3 -	. VS)vs#VS%(FC {"&#&;&#x0 I=0063N](;VD,V44vsC,?HIc  +*+~. .s   M %AM(M%$M%(M1c                 0   	 t        t        |       }t        j                   |j
                  |       sJ t        j                   |j
                  |ddi      sJ t        j                   |j
                  |dd i      sJ y # t        $ r | }d} Y w xY w)Nr:   r    )r   r   r   r   isscalarr   )r   r   r   s      r   test_rvs_scalarr      s    +)
 ;;zvzz3'(((;;zvzz30R0111;;zvzz32T2333  +*+s   B BBc                  8    t        t        j                  d       y )N)r   皙?)r   r   r9   r       r   &test_levy_stable_random_state_propertyr     s       1 1:>r   c               #     K   t               } dh}dh}t        d d  t        z   D ]  \  }}|dk(  r|t        v r=t	        j
                  ||ddddt        j                  j                  d             Q|| v}||v}||v}t               }t	        j
                  |||||d|       |r|r|rt	        j
                  ||ddddt        j                  j                  g|z           y w)	Nr4   r1   r9   Ttoo slow)reasonr   F)
setr   r   distxslow_test_momentsr   r   r   r   listxfail)	fail_normalizationfail_higherfail_momentr   r   cond1cond2cond3r   s	            r   cases_test_momentsr     s    'K-K!!'?? B#}$--,,xdD$%+[[%6%6j%6%IK K 22++ ll8S% / 	/ E ,,xdD$&,kk&7&7%85%@B B9Bs   B,C,/=C,z=distname,arg,normalization_ok,higher_ok,moment_ok,is_xfailingc           
         	 t        t        |       }t        j                         5 }|j                  t        d       |j                  t        d       |j                  t        d       |r|j                  t                |j                  |ddi\  }}	}
}t        j                  d      5  |rt        |||        |rJt        ||||        t        ||||	|
|        t        ||||	|        t        ||||	||        t        |||        d d d        t        ||||	|        |rt!        ||||	|        d d d        y # t        $ r | }d} Y 8w xY w# 1 sw Y   FxY w# 1 sw Y   y xY w)	Nr:   z9The integral is probably divergent, or slowly convergent.z#The maximum number of subdivisions.z The algorithm does not converge.momentsmvskignoreall)r   r   r   r   r   r   r   r   errstater   r	   r   r
   r   r   check_loc_scaler   )r   r   normalization_ok	higher_ok	moment_okis_xfailingr   r   r   r   sks               r   test_momentsr   2  sd   +)
 
			  6C

%N	P

%8	:

%5	7 JJ)*!V\\377
1a[[X& 		9#FC:!&#q(;!&#q!QA aH=!&#q!QA!&#x8		9 	Q84aH576 6	  +*+"		9 		96 6s6   D: BE-AE	(E:E
EE	EE#zdist,shape_argsc                    | dv rt        j                  d       | dv rt        rt        j                  d       | dv }t        t        |       }t        j                  d      }t        j                  d      }|j                  }g }ddg}t        |      D ]P  }	|	d	z   fd
|	dz   z  z   }
|j                  ||	   t        j                  |
      z         |j                  d|	d	z          R |j                  ||g       t        || |||d       y )N)r!   r'   r   r   r   )r;   r?   rf   rj   	exponnormr$   r#   r9   r5   r"   rc   ro   semicircularrh   rl   rs   )rt   r   rt   ru   r   r   d)r   r   r   r   r   r   zerosonesr   rangeappendinsertextendr   )dist
shape_args
shape_onlydistfunclocscalenargsallargsbshaper   shps              r   test_rvs_broadcastr  \  s   22J""y+,  M MJ
 ud#H
((1+CGGFOEEGVF5\  1uhq1u%z!}RWWS\12aQ  NNC<  $SIr   zx,n,sf,cdf,pdf,rtol))gh㈵> ʚ;g$?gm%?g0eD@g+<)g&.!>r  gי?gscl)D>g ^RUU!@vIh%,=)gMb@?r  g_H'9$      ?gL.'r  c                    t         j                  }t        j                  |j	                  | |      |j                  | |      |j                  | |      g      }t        j                  |||g      }t        j                  |||       t        j                  |d   |d   z   d       t        j                  |j                  ||      g| gd       y )Nrtolr   r   r  :0yE>)r   r)   r   arraysfr   r   r   assert_allcloseassert_equalisf)	r   nr  r   r   r  ksvalsexpecteds	            r   test_gh17775_regressionr    s    $ 
B88RUU1a["&&A,q!=>DxxS#'HhT2T!WtAw&,A!48r   c                     t         j                  j                  d      } t        j                  j                  t        j                  d      d|       }t        j                  |      }t        j                  t        j                  |dk7        d       t        j                  j                  dt        j                  d      |       }t        j                  |      }t        j                  t        j                  |dk7        d       t        j                  j                  t        j                  d      t        j                  d      |       }t        j                  |      }t        j                  t        j                  |dk7        d       t        j                  j                  t        j                  dgdgg      t        j                  d      |       }t        j                  |j                               }t        j                  t        j                  |dk7        d       t        t        t        j                  j
                  ddgddggddgddggd       t        t        t        j                   j
                  g dddd       t        t        t        j                   j
                  g d	g d
dgdggd       y )N{   r   r   )r  r  r   r   z0All the values are equal, but they shouldn't be!)rs   rt   ru   r   )rs   rs   )r   r   r   r   )r   r   r   r   rs   )ru   )r   r   r   r   normr   r   diffr   assert_r   r   r  ravelassert_raises
ValueErrorrd   )r   r  r   s      r   test_rvs_gh2069_regressionr'    s    ))


$C::>>bhhqk>ED
AKKqAv RS::>>arwwqz>DD
AKKqAv RS::>>bhhqk#>ND
AKKqAv RS::>>bhhaSz2"''!*'*  ,D


AKKqAv RS*ejjnn1v1v.>q6Aq6"A'*ekkoo|Q6J*ekkoo|\3*d$r   c                      t         j                  } t        j                  | j	                  ddt
        j                        d       t        j                  | j                  dgdgt
        j                  g      d       t        j                  | j	                  dt
        j                   d      d       t        j                  | j                  dgt
        j                   gdg      d       t        j                  | j                  dgdgt
        j                  g      d       t        j                  | j	                  ddt
        j                        d       t        j                  | j                  dgt
        j                   gdg      d       t        j                  | j	                  dt
        j                   d      d       t        j                  | j	                  dt
        j                   d      d       y )Nr   r   gA?g|N?)r   r%   r   assert_almost_equalr   r   inf_cdf)tns    r   test_nomodify_gh9900_regressionr.    s{    
B BFF1a0.0BGGQC!rvvh7.0 BFF2w2/1BGGRDBFF7)aS9/1 BGGQC!rvvh7.0BFF1a0.0 BGGRDBFF7)aS9/1BFF1rvvgq1BFF2w2/1r   c            	          t        j                  g d      } t        j                  g d      }t        | |      D cg c]&  \  }}t        j                  j                  d||      ( }}}t        j                  t        j                  j                  d| |      |       t        | |      D cg c]&  \  }}t        j                  j                  d||      ( }}}t        j                  t        j                  j                  d| |      |       t        | |      D cg c]&  \  }}t        j                  j                  |||      ( }}}t        j                  t        j                  j                  | | |      |       t        | |      D cg c]&  \  }}t        j                  j                  |||      ( }}}t        j                  t        j                  j                  || |      |       y c c}}w c c}}w c c}}w c c}}w )N)r   rs   rt   ru   r   rv   )rx      r       r   0   rw   r   )r   r  zipr   rq   r   r   assert_array_almost_equal)rz   b_a_banss        r    test_broadcast_gh9990_regressionr9    s    	#$A
&'A9<Qq
Bvr252r*
BC
B!!%"2"2"6"6q!Q"?E9<Qq
Bvr252r*
BC
B!!%"2"2"6"6q!Q"?E:=a(
CB5B+
CC
C!!%"2"2"6"6q!Q"?E:=a(
CB5B+
CC
C!!%"2"2"6"6q!Q"?E C C D Ds   +G80+G>"+H+H
c            	          t         j                  j                  t        j                  g d      dt        j                  g d      z
  dz  t        j
                  t        j                  g d      d       y )N)      @       @r  r   )g      @r   g      @r;  )rz   r5  r  r  )r   r%   r   r   r  r+  r   r   r    test_broadcast_gh7933_regressionr=    sQ    	OO
!/**c
1
&&HH_%  r   c            	         t         j                  ddd   } t         j                  t        j                  d       t        j                  d      f   }t        | |      D cg c]4  \  }}t        j
                  j                  ||t         j                        6 }}}t        j
                  j                  | |t         j                        }t        j                  ||       y c c}}w )Nrs   y             @Y@2   3   )
r   r_r   r3  r   r%   r   r+  r   r4  )r   rz   _xr6  r  r8  s         r   test_gh2002_regressionrD    s     	b4iA
rwwr{lBGGBK'(ABEa)LB##BBFF3LHL
//

aBFF
+C!!#x0 Ms   "9C$c                  |    d} t         j                  j                  dt        j                  | g| dz   gg             y )Ng(\@r   )r   rH   ppfr   r  )cs    r   test_gh1320_regressionrH    s3    A	bhha#gY'789r   c                     t         j                  j                  d       g d} ddt        j                  d      z  dz  z
  }ddt        j                  d      z  dz  z   }t        t        t        j                        t        j                  #  | d      \  }}t        j                  ||d	
       t        j                  ||z   |d	
       y )Ni  )r   r   r   r   r   皙?rs   rt   r   r\   r   ru   )decimal)r   r   seedsqrtsupertyper   uniformfitr   r*  )r   rz   r5  r  r  s        r   test_method_of_momentsrS  &  s    IINN4Aa
l1nAa
l1nAtEMM*EMM>q>NJCCA.CIq!4r   c                     t        j                  |       rt        ||        t        j                  |      rt        ||       y y N)r   isfinitecheck_sample_meancheck_sample_var)popmeanpopvarsamples      r   r   r   2  s3    	{{7&'*	{{6( r   c                 R    t        j                  | |      j                  }|dkD  sJ y )Nr   )r   ttest_1samppvalue)r[  rY  probs      r   rW  rW  9  s&    VW-44D$;;r   c                     t        j                  | fd d      }|j                  }|j                  |j                  }}||cxk  r|k  sJ  J y )Nc                 (    | j                  d|      S )Nr   )ddofaxis)var)r   rc  s     r   <lambda>z"check_sample_var.<locals>.<lambda>E  s    140 r   gףp=
?)confidence_level)r   	bootstrapconfidence_intervallowhigh)r[  rZ  resconfri  rj  s         r   rX  rX  ?  sT     //		0C
 ""D$))C& D     r   c                     g d}t        j                   | j                   | j                  |g| g| |t        |dz          y )N)gMbP?r   g+?z - cdf-ppf roundtriprL  err_msg)r   r*  r   rF  DECIMAL)r   r   msgvaluess       r   r   r   M  sF     FJFJJzvzz&'?3'?F#F"GS2>34r   c                     t        j                   | j                   | j                  g dg| g| g dt        |dz          y )Nr   r   ?z - sf-isf roundtriprn  )r   r*  r  r  rp  r   r   rq  s      r   r   r   T  sB    IFIIjfjj&G3&GN#N+Wc1G23r   c                     t        j                   | j                  ddgg| d | j                  ddgg| z
  t        |dz          y )Nr   ru  r  z - cdf-sf relationshiprn  )r   r*  r   r  rp  rv  s      r   r   r   Z  sO    JFJJSz8C8)&))S#J"="==$+S4656r   c                     t        j                  ddg      }t        j                   | j                  |g|  | j
                  d|z
  g| t        |dz          y )Nr   ru  r   z - ppf-isf relationshiprn  )r   r  r   r*  r  rF  rp  )r   r   rq  ps       r   r   r   a  sW    
#sAJFJJq/3/AaC1F#1F$+S5667r   c                 B    | j                   dg| }d} | j                  |g| }|dk  s|dkD  r|dz   } | j                  |g| } | j                  ||z   g|  | j                  ||z
  g| z
  |z  dz  }|dz  }t        j                  ||t
        |       y )	Nr   gư>g-C6?     @r   r<  z - cdf-pdf relationshiprn  )rF  r   r   r   r*  rp  )r   r   rq  medianepspdfvcdfdiffs          r   r   r   h  s    VZZ"c"F
C6::f#s#Dt #vzz&'3'vzz&3,--vzz&3,--./23367G $$CD'7CHr   c                    t        j                  g d      } | j                  |g| }|t        j                  |         } | j                  |g| } | j
                  |g| }||dk7  t        j                  |      z     }|t        j                  |         }|dz  }t        j                  t        j                  |      |d|       y )N)rJ  r   皙?r   333333?ffffff?皙?r    - logpdf-log(pdf) relationshiprw   rn  	r   r  rF  rV  r   r   r   r*  logr   argsrq  pointsr  r   r   s          r   r   r   z  s    XX9:F6::f$t$DD!"D
&**T
!D
!CV]]4'$'F
sax2;;s++
,CBKK'(F,,CBFF3KCHr   c                    t        j                  ddg      } | j                  |g| }|t        j                  |         } | j                  |g| } | j
                  |g| }||dk7  t        j                  |      z     }|t        j                  |         }|dz  }t        j                  t        j                  |      |d|       y )Nr   r   r  rw   rn  r  r  s          r   r   r     s    XXq!fF6::f$t$DD!"D
&**T
!D
!CV]]4'$'F
sax2;;s++
,CBKK'(F,,CBFF3KCHr   c                 v   t        j                  g d      } | j                  |g| }|t        j                  |         } | j                  |g| } | j
                  |g| }||dk7     }|t        j                  |         }|dz  }t        j                  t        j                  |      |d|       y )N)	        rJ  r   r  r   r  r  r  r  r   z - logsf-log(sf) relationshiprw   rn  )	r   r  rF  rV  r  r   r   r*  r  )r   r  rq  r  r  r  r   s          r   r   r     s    XXCDF6::f$t$DD!"D	4	$	BFLL%%E	B!GB"++e$%E**CBFF2Jq#Fr   c                 v   t        j                  g d      } | j                  |g| }|t        j                  |         } | j                  |g| } | j
                  |g| }||dk7     }|t        j                  |         }|dz  }t        j                  t        j                  |      |d|       y )N)	r   rJ  r   r  r   r  r  r  r  r   z - logcdf-log(cdf) relationshiprw   rn  )	r   r  rF  rV  r   r   r   r*  r  )r   r  rq  r  r  r   r   s          r   r   r     s    XXABF6::f$t$DD!"D
&**T
!D
!CV]]4'$'F
cQh-CBKK'(F,,CBFF3KCHr   c                     d}g |z  }|r&|D cg c]  }t        j                  |g|z         }} | j                  dg| } | j                  dg| }|dz  }t        j                  ||g|z  d|       y c c}w )Nr   r   z - ppf multiplerw   rn  )r   r  rF  r   r*  )r   r   rq  num_repeatsr  _r|  medianss           r   r   r     s    KD
589!{*+99VZZ"c"Ffjj$t$GCGfX%;QPST :s    A9c                 $   t        j                  || |d      \  }}||k  rot        j                  | | |d      \  }}t        j                  ||kD  dt	        |      z   dz   t	        |      z   dz   t	        |      z   dz   t	        |      z          y y )Ni  )r  NzD = z	; pval = z
; alpha = z
args = )r   kstestr   r#  str)r  r  r>   r   Dpvals         r   r   r     s     ll3448GAtu ,,tT=4D5L&3q6/K"?#d)"K #!#&u:#.0;#<>A$i#H 	I 	r   c                 d    t        j                   | j                  |  | j                  |        y rU  )r   r  
vecentropyr   )r   r  s     r   r   r     s+    &V&&-v/EFr   c                     t        j                  ddg      t        j                  ddg      }} | j                  |||d\  }}t        j                  ||z  |z   |       t        j                  ||z  |z  |       y )N      $@g      4@r  r  )r   r  r   r   r  )	r   r   r   r   rq  r  r  mtvts	            r   r   r     so     4,'4,)?CV\\3Cu5FB%#r*%r*r   c                      | j                   t        j                  g d      g| }t        j                  t        j
                  t        j                  |             |dz          y )Nrt  zppf private is nan)_ppfr   r  r   r#  anyisnan)r   r   rq  ppfss       r   r   r     sH    6;;rxx0737DKKBFF288D>**C2F,FGr   c                     d\  }} | j                   | } | j                   |||d}t        j                  t        j                  |      |z  |z   t        j                  |             y )N)r   rs   r  )supportr   r*  r   r  )r   r  r  r  suppsupp_loc_scales         r   r   r     s]    JC6>>4 D#V^^Ts%@NBHHTN5036HH^46r   c                 
   t        j                  d      5  t        j                         5 }|j	                  t
        d       |j	                  t
        d       | j                  ||      }| j                  |d|      }d d d        d d d        t        j                  t              d	t        |      z   k(         t        j                  t              d	t        |      z   k(         y # 1 sw Y   lxY w# 1 sw Y   pxY w)
Nr   r   !The shape parameter of the erlangcategorymessagezfloating point number truncatedrK  powell)	optimizerr   rs   	r   r   r   r   r   r   rR  r#  len)r   r   r   r   r   r  vals2s          r   r   r     s    		" CC$9$9$; Cs

N> 	 	@

N< 	 	>zz#fz-

3(6
BC C KKD	Qs3xZ'(KKE
aCj()C C C Cs#   C9AC-C9-C6	2C99Dc           	         t        j                  d      5  t        j                         5 }|j	                  t
        d       | j                  |d|      }| j                  |d|      }t        j                  t        |      d	t        |      z   k(         t        j                  |d
   dk(         t        j                  |d   dk(         t        j                  t        |      d	t        |      z   k(         t        |      dkD  rb| j                  ||d   |      }t        j                  t        |      d	t        |      z   k(         t        j                  |d   |d   k(         t        |      dkD  rb| j                  ||d   |      }t        j                  t        |      d	t        |      z   k(         t        j                  |d   |d   k(         t        |      d	kD  rb| j                  ||d	   |      }	t        j                  t        |	      d	t        |      z   k(         t        j                  |	d	   |d	   k(         d d d        d d d        y # 1 sw Y   xY w# 1 sw Y   y xY w)Nr   r   r  r  r   )flocr   r   )fscaler   rs   r?  r)  )f0r   )f1r   )f2r   r  )
r   r   r   r   r   r  r  vals3vals4vals5s
             r   r   r     s   		" ,C$9$9$; ,s

N> 	 	@ zz#Afz5

3q
8CI3s8+,DHM"E"IN#CJ!CH*,-s8a<JJss1vfJ=EKKE
aCj01KKaCF*+s8a<JJss1vfJ=EKKE
aCj01KKaCF*+s8a<JJss1vfJ=EKKE
aCj01KKaCF*++, , , , , ,s#   I)H II)I&	"I))I2r   )r   r   r   r   r  r   rF  r  zdistname, argsc           	      N   t        t        |      }t        ||       }|dk(  r| j                  d      rddg}nddg}|D cg c]  }|gdz  
 }}ddg}dd	g}	 ||g|||	d
}
t        j                  |
t        |g|||	 D cg c]  } || 	 c}dd       y c c}w c c}w )NrV   r  r   rs   r   rJ  r   r   g)\(?r  g+=r  )r  atol)r   r   
startswithr   r  r3  )r   r   r  r  rE   r   rz   shape2r  r  resultr   s               r   test_methods_with_listsr    s     5(#DfA<F$5$5e$<!H#J!"qc!e"F"c(CIEq060s%0F(+A(C(C(CU(CD1AD"0	 #
 Es    BB"
c                      t         j                  j                  ddd      } t        | t              sJ t         j
                  j                  dd      }t        |t              sJ y )Nr   r   ru   rx   )r   rA   moment
isinstancefloatrF   )vals0vals1s     r   (test_burr_fisk_moment_gh13234_regressionr  %  sP    JJaA&EeU###JJa#EeU###r   c            
         t         j                  j                  dt        j                  g d      d      } t        j                  g d      }t        j                  | |       t         j                  j                  dt        j                  g d      d      }t        j                  t        j                  t        j                  t        j                  g      }t        j                  ||       t         j                  j                  dt        j                  g d      g d      }t        j                  t        j                  dt        j                  g      }t        j                  ||       t         j                  j                  dd	d
      }t        j                  }t        j                  ||       t        ||j                        sJ t         j                  j                  dg dg d      }t        j                  dt        j                  t        j                  g      }	t        j                  ||	       t         j                  j                  dg dg d      }
t        j                  dt        j                  dg      }t        j                  |
|       t         j                  j                  dg dg d      }t        j                  t        j                  t        j                  t        j                  g      }t        j                  ||       t         j                  j                  ddd	d	      }t        j                  }t        j                  ||       t        ||j                        sJ t         j                  j                  dg dd	d	      }t        j                  t        j                  t        j                  t        j                  g      }t        j                  ||       t         j                  j                  dg dg dg d      }t        j                  dt        j                  t        j                  g      }t        j                  ||d       t         j                  j                  dddgddg      }t        j                  ddg      }t        j                  ||d       dd	ddt        j                  g}dd	ddt        j                  g}dd	t        j                  d      g}d d	ddt        j                  g}t        j                  |      j                  d!      }t        j                  |      j                  d"      }t        j                  |      j                  d#      }t        j                  |      }t         j                  j                  d||||$      }t        j                   ||||      \  }}}}t        j"                  |      D ]z  }t        j$                  d%d%&      5  |d	   }t         j                  j                  d||   ||   ||   ||   $      }t        j&                  j                  ||   |       d d d        | y # 1 sw Y   xY w)'Nr   r   rs   rt   orderr  r  r  r<  r;  r)  )r   r   r<  rs   r   )r   r   rs   )rt   r  r  )r   r   r   )r   r?  g      Y@g      9@r{  )g      r?  g      Yr  dfr  r  )r  r  r<  )r  g      r  g ﲙ"@r  r  r   r  g      :@g      @gvIh%<=gg@g )r)  r   r   r   )r)  r   r   )r)  r   )r  rz   r5  r  r  r   )invaliddivide)r   r!  r  r   r  r   r  nanr  r   r`   r  pirN  reshapera   broadcast_arraysndenumerater   testing)r  	expected0r  	expected1r  	expected2r  	expected3r  	expected4r  	expected5vals6	expected6vals7	expected7vals8	expected8vals9	expected9vals10
expected10rz   r5  r  r  vals11ir  s                                r   *test_moments_with_array_gh12192_regressionr  -  s   JJA288I+>aHE&IUI& JJA288I+>bIE"&&"&&"&&12IUI& JJA288I+>$.  0E"&&"bff-.IUI& JJA1B7EIUI&eY00111 JJA9KHE#rvvrvv./IUI& JJA9NKE#rvvv./IUI& JJA9<LME"&&"&&"&&12IUI& II1;EIUI&eY00111 II1CE"&&"&&"&&12IUI& II1#0  2E*bffbff56Iyt4 ZZq2r(RH5F3+&J
7 
q!S"%% A	q!S"%% ABGGAJ
C1abee$E
M*A
J'A
((3-


(CHHUOEZZQ!qcGF**1ae<Aq#u^^A 9[[(; 	9!Azz((qAaDAaD-0V58 ) EHJJ##F1Ix8	9 	99	9 	9s   AY::Z	c                     t         j                  j                  dt        j                  g d      dgg      } t        j                  g dg      }t        j                  | |       | j                  |j                  k(  sJ t         j                  j                  dt        j                  dgdgdgg      g d      }t        j                  g dg dg d	g      }t        j                  ||       |j                  |j                  k(  sJ t         j                  j                  dg dd
d      }t        j                  g d      }t        j                  ||d       |j                  |j                  k(  sJ t         j                  j                  ddgdgdggg dg d      }t        j                  t        j                  t        j                  dgt        j                  t        j                  dgt        j                  t        j                  dgg      }t        j                  ||d       |j                  |j                  k(  sJ y )Nr   r  r  r  rs   rt   )r  r  r  )r<  r<  r<  )r;  r;  r;  r  r  r  )g e3E?g> ?g e3E?r  r  r<  r;  )r  r  r<  )g      r  r;  g@3F@gw,W.
@gf3&@)r   r!  r  r   r  r   r  shaper`   r  r  )r  r  r  r  r  r  r  r  s           r   /test_broadcasting_in_moments_gh12192_regressionr    s   JJA288I+>seLE,(IUI&;;)//)))JJA288aS1#sO+D$-  /E,lCDIUI&;;)//)))II12RHE=>Iyt4;;)//)))II12$rd);#0  2E266266:666266:666266:68 9I yt4;;)//)))r   c                  ^   g d} d}t        j                  |D cg c]1  }| D cg c]#  }t        j                  j                  ||      % c}3 c}}      }t        j                  t        j                  j                  | |            }t	        j
                  ||       y c c}w c c}}w )N)r   r         @      @      @r   r   )r   r  r   rM   r   r  )shapesr   r  r  rk  res2s         r   test_kappa3_array_gh13582r    s    &FG
((6=?,2( \\''v'> ( ? @C88ELL&&vw&?@DT"( ?s   	B)
(B$B)
$B)
c                     t        j                  g d      } t        j                  g d      }d}t        j                  |D cg c]A  }t        d      D cg c]*  }t        j                  j                  | |   ||   |      , c}C c}}      }t        j                  t        j                  j                  | ||            }t        j                  ||       t        j                  g d      } t        j                  g d      }t        j                  |D cg c]A  }t        d      D cg c]*  }t        j                  j                  | |   ||   |      , c}C c}}      }t        j                  t        j                  j                  | ||            }t        j                  ||       t        j                  g d	      } t        j                  g d
      d d d f   }t        j                  t        j                  j                  | ||            }|j                  dk(  sJ y c c}w c c}}w c c}w c c}}w )N)r   r  r  r  r  )r   r   g      r   r  r   r   r  )r)        пr  r   r)  r   )r   r   r   gUUUUUUտr)  r   rv   )r)  r   r   )r)  r   r   r   )ru   ru   rt   )r   r  r   r   r   r   r  r  )hr   r   r  r  rk  r  s          r   test_kappa4_array_gh13582r    s   
*+A
()AG
((4;=*0E!H& \\''!adF'C & = >C88ELL&&q!W&=>DT" 	+,A
)*A
((4;=*0E!H& \\''!adF'C & = >C88ELL&&q!W&=>DT" 	A
!"1d7+A88ELL&&q!W&=>D::"""#& =& =s0   H:
/H5H:
 I
2/I !I
5H:
 I
c                     d} t        j                  t        |       5  t        j                         j
                   d d d        t        j                  t        |       5  t        j                         j                   d d d        dt        j                  _        t        j                         }t        |t              sJ t        t        j                  d       y # 1 sw Y   xY w# 1 sw Y   kxY w)Nz.'rv_continuous_frozen' object has no attribute)matchherringpmf)
r   r   AttributeErrorr   r!  r   logpmfr  r   delattr)r  frozen_norms     r   test_frozen_attributesr    s     ?G	~W	5 

	~W	5 

EJJN**,Kk#7888EJJ  s   C C)C&)C2c                     t         j                  j                  d      } t         j                   d}}t	        j
                  t        j                  j                  ||      t        j                  j                  |             | j                  d      | j                  d      }}| j                  d      dk  }d||<   ||   }t        j                  j                  ||      }t	        j
                  ||   t        j                  j                  |             t	        j
                  ||    t        j                  j                  ||    ||                 y )Nr   )rt   rt   )r   r   )
r   r   default_rngr+  r   r  r   ro   r   r!  )r   r   rz   maskx_normrk  s         r   test_skewnorm_pdf_gh16038r
    s   
))


"CFF7AqAU^^''1-uzz~~a/@A::6:"CJJFJ$;qA::6:"S(DAdGtWF
..

Q
"CSY

v 67S$Z!3!3AteHah!GHr   r   r   r   r   r   r  r   rF  r  r  entropyexpectr|  meanstdrd  intervalgffffff?r  r   mvcasec                    | \  }}t        t        j                         |      } || }| t        v rt	        |t
        j                        sJ y t	        |d   t
        j                        sJ t	        |d   t
        j                        sJ y )Nr   r   )r   r   r!  
scalar_outr  r   number)r  method_namer  r   rk  s        r   test_scalar_for_scalarr    su     KUZZ\;/F
$-Cz#ryy)))#a&")),,,#a&")),,,r   c                     t         j                  j                  g d      } t        | d   t        j
                        sJ t        | d   t        j
                        sJ t         j                  j                  g d      } t        | d   t        j
                        sJ t        | d   t        j
                        sJ t         j                  j                  dg d      } t        | t        j
                        sJ y )Nr  r   r   r   )r   r!  rR  r  r   r  fit_loc_scalennlf)rk  s    r   test_scalar_for_scalar2r    s    
**..
#Cc!fbii(((c!fbii(((
**
"
"9
-Cc!fbii(((c!fbii(((
**//&)
,Cc299%%%r   r   )ssysnumpyr   numpy.testingr  r   r   r   r%  scipy.integrater   	itertoolsscipyr   common_testsr   r   r	   r
   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   scipy.stats._distr_paramsr   !scipy.stats._distn_infrastructurer   rp  maxsizer   r   r   r   skip_fit_test_mleslow_fit_test_mmfail_fit_test_mmr   skip_fit_fix_test_mlefail_fit_fix_test_mmr   r   r   case1case2productr  r}   	histogram_histrv_histogram_rv_histr   tupler   r   parametrizer   r   r   r   r   r   r  r  r'  r.  r9  r=  rD  rH  rS  r   rW  rX  r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r  r  r  r  r  r   r  r  r
  r  scalars_outr  r  r   r   r   <module>r4     s   
    * .  
? 
? 
? 
? 
? 
? / B	 [[5 	-
 N	I C ) EFF #|$  *'*::<. IBB &,'  2+.BBD ?$  *34	6!fj)&Y&&u~e}E 9MD'BLL1411E!u!!%9H##Xuw$789	  )>)@A,zl;[I < B[I| )>)@A
4 B
4?"BJ  '+-/#6/ #6L *H5&J 6&Jj .?	@9	@9$$:!1HF,1:	5)!4367I$
I
I
G
I
UI&G+H6
*,2  $A B)840 5B0&$Q9h*4# # #0
I bkEA3<(QC51#,!oqc{WqcNUQCLqclXsOi_xnnvrlUBK%F
 TF#i_w6GH k!9:
- ;
-	&r   