U md@sDdZddlZddlmZddlmZmZmZd dd Z d d d Z dS)zb Created on Sun Sep 25 21:23:38 2011 Author: Josef Perktold and Scipy developers License : BSD-3 N)stats) array_like bool_likeint_likenormTcCs|t|ddd}t|d}t|d}tj||d}|j|}|r|dkrttj||d|}ttj|d|d |}|||} t j | } qt |r| |}|j |f|} qtd n"t |r|j |f|} ntd td|d} dg|j} td| |<t| } tdg|j} tddd | |<t| } tjd | | d|t| t| |  |d}| |}|S)a Calculate the Anderson-Darling a2 statistic. Parameters ---------- x : array_like The data to test. dist : {'norm', callable} The assumed distribution under the null of test statistic. fit : bool If True, then the distribution parameters are estimated. Currently only for 1d data x, except in case dist='norm'. params : tuple The optional distribution parameters if fit is False. axis : int If dist is 'norm' or fit is False, then data can be an n-dimensional and axis specifies the axis of a variable. Returns ------- {float, ndarray} The Anderson-Darling statistic. xN)ndimfitaxis)r r)Zddofr z!dist must be 'norm' or a Callablez+if fit is false, then dist must be callableg?)rrrnpsortshapeZ expand_dimsZmeanZstdrrZcdfcallabler ValueErrorZaranger slicetuplesumloglog1p)rdistr paramsr yZnobsZxbarswziZsl1Zsl2Za2rrR/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/statsmodels/stats/_adnorm.pyanderson_statistics<        0 r!cCst|dd|d}|j|}|dd|d|d}t|dkr|dkrx|d krxdtd d |d |d}n|d krdtdd|d|d}nT|dkrtdd|d|d}n,|dkrtdd|d|d}nd}ntdd d dg}dd}dd}dd} dd} dd} ||| | | g} tj||dd } tjt|}t d!D]"}| |k}| |||||<qj||fS)"a Anderson-Darling test for normal distribution unknown mean and variance. Parameters ---------- x : array_like The data array. axis : int The axis to perform the test along. Returns ------- ad2 : float Anderson Darling test statistic. pval : float The pvalue for hypothesis that the data comes from a normal distribution with unknown mean and variance. See Also -------- statsmodels.stats.diagnostic.anderson_statistic The Anderson-Darling a2 statistic. statsmodels.stats.diagnostic.kstest_fit Kolmogorov-Smirnov test with estimated parameters for Normal or Exponential distributions. rT)rr r r g?g@rgg?Zd;*)\HY@(\k@g(\?A`Т SeE@MbM@g333333? Ac]?V-@Gz? \m?t@Έ ?cSstjt|S)N)rnan ones_likead2arrr xznormal_ad..cSs"dtdd|d|dS)Nr r"r#r$rrexpr1rrr r3yscSs"dtdd|d|dS)Nr r%r&r'rr5r1rrr r3{scSstdd|d|dS)Nr(r)r*rr5r1rrr r3}r4cSstdd|d|dS)Nr,r-r.rr5r1rrr r3~r4right)Zside) r!rrsizer6arrayZ searchsortedr/r0range)rr Zad2nr2ZpvalZboundsZpval0Zpval1Zpval2Zpval3Zpval4Zpvalliidxrmaskrrr normal_adIs4 $$   r?)rTrr)r) __doc__numpyrZscipyrZstatsmodels.tools.validationrrrr!r?rrrr s   ;