
    xfĠ                        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                  dki 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 Zcej                  j                  dHg dI      ej                  j                  dJe#      dK               ZddL ZedM ZfdN ZgdO Zhej                  j                  dP        ZjdQ ZkdR ZldSg gdTd ggdUd g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]g gd^g gd_g gd`g gdag gdbg ggZmdcddggdeg gdfdggggZnej                  j                  dhemenz         di        Zodj Zpy)l    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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   r3   rQ   r4   r5   rR   rS   r^   rU   )r$   r1   r)   r(   rW   r7   r6   >3   rE   rS   chir4   r5   betachi2r\   ricer;   gammar)   r(   r]   cosinedgammaerlangrN   gennormmaxwelldweibullgengammarK   loggammalogisticrQ   rX   powerlawskewnormr   r?   rI   r0   r2   r%   rG   r!   rJ   
loguniform
reciprocalr#   r.   rT   halflogisticr"   r$   r    rV   r&   r'   r:   )r      rr      rs   rs      rt   rt   rt   r   r   r   r   r      ru   ru   ru      rv   rv      rw   	   rw   )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     I/usr/lib/python3/dist-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        |||        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}|j-                  | d      }| dk(  rd}n| dk(  rd}t/        ||||
|       t1        ||       | dv rt2        rt5        j6                  d       nt9        ||       t;        ||       | dvrt=        |||        |j>                  dk(  rtA        ||       |jB                  jD                  t        jF                  jD                  k7  r | dk7  rtI        ||t        jF                         tK        jL                         5 }|jO                  tP        d       |jO                  tP        d       |jO                  tR        d       tU        ||       d d d        tW        ||       tY        ||       t[        |||       t]        ||       | t^        vrta        ||       | dk7  rtc        |||        dD ];  }| td        |   vrtg        |||d | |       | th        |   vs+tk        |||d | |       = y # t        $ r | }d} Y 8w xY w# 1 sw Y   xY w)Nr:   i )sizerandom_state>   laplace_asymmetric{Gz?r#   r   r               ?g	@333333?      @)weibull_maxrN   rW   r^   rT   r:         ?rU   r   r)   )rs   rV   fails on Linux 32-bit)r2   r(   r4   r   r   z The occurrence of roundoff errorzExtremely bad integrandzinvalid valuer%   rY   )6getattrr   	TypeErrornprandomRandomStatervscheck_sample_meanvar_check_cdf_ppfcheck_sf_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      s=   
+)
 ))


'C
&**s"
5C6<<DAq--aC(&#x(h'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_  +*+z. .s   L: AM:M
MM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   0  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?   re   ri   	exponnormr$   r#   r9   r5   r"   rb   rn   semicircularrg   rk   rr   )rs   r   rs   rt   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
  Z  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!)rr   rs   rt   r   )rr   rr   )r   r   r   r   )r   r   r   r   rr   )rt   )r   r   r   r   normr   r   diffr   assert_r   r   r  ravelassert_raises
ValueErrorrc   )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+    sh    
B BFF1a02DEBGGQC!rvvh79KL BFF2w24GHBGGRDBFF7)aS9;NO BGGQC!rvvh79KLBFF1a02DE BGGRDBFF7)aS9;NOBFF1rvvgq115BFF2w24GHr   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   rr   rs   rt   r   ru   )rw      r       r   0   rv   r   )r   r  zipr   rp   r   r   assert_array_almost_equal)ry   b_a_banss        r    test_broadcast_gh9990_regressionr6    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      @r8  )ry   r2  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 )Nrr   y             @Y@2   3   )
r   r_r   r0  r   r%   r   r(  r   r1  )r   ry   _xr3  r  r5  s         r   test_gh2002_regressionrA    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_regressionrE    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   皙?rr   rs   r   r[   r   rt   )decimal)r   r   seedsqrtsupertyper   uniformfitr   r'  )r   ry   r2  r  r  s        r   test_method_of_momentsrP    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   '  s3    	{{7&'*	{{6( r   c                 R    t        j                  | |      j                  }|dkD  sJ y )Nr   )r   ttest_1samppvalue)rX  rV  probs      r   rT  rT  .  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   r`  s     r   <lambda>z"check_sample_var.<locals>.<lambda>:  s    140 r   gףp=
?)confidence_level)r   	bootstrapconfidence_intervallowhigh)rX  rW  resconfrf  rg  s         r   rU  rU  4  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rI  err_msg)r   r'  r   rC  DECIMAL)r   r   msgvaluess       r   r   r   B  sF     FJFJJzvzz&'?3'?F#F"GS2>34r   c                 "   t        j                   | j                   | j                  g dg| g| g dt        |dz          t        j                   | j
                  ddgg| d | j                  ddgg| z
  t        |dz          y )Nr   r   ?z - sf-isf roundtriprk  r   rr  r  z - cdf-sf relationship)r   r'  r  r  rm  r   )r   r   rn  s      r   r   r   I  s    IFIIjfjj&G3&GN#N+Wc1G23 JFJJSz8C8)&))S#J"="==$+S4656r   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   r9  z - cdf-pdf relationshiprk  )rC  r   r   r   r'  rm  )r   r   rn  medianepspdfvcdfdiffs          r   r   r   S  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)rG  r   皙?r   333333?ffffff?皙?r    - logpdf-log(pdf) relationshiprv   rk  	r   r  rC  rS  r   r   r   r'  logr   argsrn  pointsr  r   r   s          r   r   r   e  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~  rv   rk  r  r  s          r   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)	        rG  r   rz  r   r{  r|  r}  r  r   z - logsf-log(sf) relationshiprv   rk  )	r   r  rC  rS  r  r   r   r'  r  )r   r  rn  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   rG  r   rz  r   r{  r|  r}  r  r   z - logcdf-log(cdf) relationshiprv   rk  )	r   r  rC  rS  r   r   r   r'  r  )r   r  rn  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 multiplerv   rk  )r   r  rC  r   r'  )r   r   rn  num_repeatsr  _ru  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 rR  )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   rn  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 )Nrq  zppf private is nan)_ppfr   r  r   r   anyisnan)r   r   rn  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   rr   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 truncatedrH  powell)	optimizerr   rr   	r   r   r   r   r   r   rO  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   rr   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   rC  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 )NrU   r  r   rr   r   rG  r   r   g)\(?r  g+=r  )r  atol)r   r   
startswithr   r  r0  )r   r   r  r   rE   r   ry   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   rt   rw   )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   rr   rs   orderr  r  r  r9  r8  r&  )r   r   r9  rr   r   )r   r   rr   )rs   r  r  )r   r   r   )r   r<  g      Y@g      9@rt  )g      r<  g      Yr  dfr  r  )r  r  r9  )r  g      r  g ﲙ"@r  r  r   r  g      :@g      @gvIh%<=gg@g )r&  r   r   r   )r&  r   r   )r&  r   )r  ry   r2  r  r  r   )invaliddivide)r   r  r  r   r  r   r  nanr  r   r_   r  pirK  reshaper`   broadcast_arraysndenumerater   testing)r  	expected0r  	expected1r  	expected2r  	expected3r  	expected4r  	expected5vals6	expected6vals7	expected7vals8	expected8vals9	expected9vals10
expected10ry   r2  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  rr   rs   )r  r  r  )r9  r9  r9  )r8  r8  r8  r  r  r  )g e3E?g> ?g e3E?r  r  r9  r8  )r  r  r9  )g      r  r8  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  l  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   rL   r   r  )shapesr   r  r  rh  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   ru   )r&  r   r   )r&  r   r   r   )rt   rt   rs   )r   r  r   r   r   r   r  r  )hr   r   r  r  rh  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   )rs   rs   )r   r   )
r   r   default_rngr(  r   r  r   rn   r   r  )r   r   ry   maskx_normrh  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   rC  r  r  entropyexpectru  meanstdra  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   rh  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  rO  r  r   r  fit_loc_scalennlf)rh  s    r   test_scalar_for_scalar2r    s    
**..
#Cc!fbii(((c!fbii(((
**
"
"9
-Cc!fbii(((c!fbii(((
**//&)
,Cc299%%%r   r   )q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   rm  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+  r6  r:  rA  rE  rP  r   rT  rU  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>r-     s   
    * .  
? 
? 
? 
? 
? 
? / B	 [[5 	-
 N	I C ) EFF #|$  *'*::<. 6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;YI < BYIx )>)@A
4 B
4?"BJ  '+-/#6/ #6L *H5&J 6&Jj .?	@9	@9$$:I6F,1:	5)!46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   