o ̋iN@s&ddlZddlZddlmmZddlm Z dZ e e de Z e e de ZedejZedejZdZedZejdZejdZejd Zgd Zd d Zd dZd&ddZd&ddZddZd&ddZd&ddZ d&ddZ!ddZ"ddZ#d&d d!Z$d"d#Z%d&d$d%Z&dS)'N) _derivativei<)gSˆBgAAz?g}<ٰj_g#+K?g88CgJ?gllfgUUUUUU?cCs6d|}t|d|td|tt||S)N?r)nplog_LOG_2PIpolyval_STIRLING_COEFFS)nrnrX/home/air/goalskill_sales/back/venv/lib/python3.10/site-packages/scipy/stats/_ksstats.py_log_nfactorial_div_n_pow_n]s.rcCst|ddS)z%clips a probability to range 0<=p<=1.r )r clip)prrr _clip_probgsrTcCst|||}t|S)z>Selects either the CDF or SF, and then clips to range 0<=p<=1.)r wherer)cdfprobsfprobcdfrrrr_select_and_clip_problsrcCs^|dkr tdd|S||}|dkrtdd|Stt|}||}d|d}t||g}td|d}d||} t|} d} |D]} | | | d<| | } | | d| 9<qGtd|dd|d||} d| | | d<td|D]}| d||d||dd|f<q|| |dddf<tj | dd |dddf<t t |d}|}d}d}|dkr|drt ||}||7}t ||}|d9}t ||d|dftkr|t}|t7}|d}|dks||d|df}td|dD]}|||}t |tkr|t9}|t8}q|dkr't||}t|d||S) zComputes the Kolmogorov CDF: Pr(D_n <= d) using the MTW approach to the Durbin matrix algorithm. Durbin (1968); Marsaglia, Tsang, Wang (2003). [1], [3]. r r?rrrN)axis)rintr ceilzerosarangeemptymaxrangeflipeyeshapematmulabs_EP128_E128_EM128ldexp)rdrndkhmHintmvwfacjttiHpwrnnexpntHexpntrrrr _kolmogn_DMTWrs`      "&      rAc Cs|dkr| |d||d}}nLt|dd\}}|dkrN||dkr8|||d|||d}}n'|d||d||d|d}}n|d|d|||d}}t|ddt||fS)z0Compute the endpoints of the interval for row i.rrr)divmodr%min) r<rllceilfroundfj1j2ip1div2ip1mod2rrr_pomeranz_compute_j1j2s $,"rKc Cs||}tt|}d||}t|d|}|dkrdnd}|dkr&dnd}d|d} t| } t| } t| } d| d<d| d<d| d<d} ||d||dd||}}}td| D]&}| |d||| |<| |d||| |<| |d||| |<qdt| g}t| g}d|d<d\}}td||||\}}tdd|dD]}|}||}}||}}|dt|||||\}}|dks|d|dkr| }n|dr| n| }||d}|dkr:t |||||||d|}||}||d}|||||d|<dt |kr*t kr4nn|t 9}| t 8} |||}q|||}td|dD]}t|t krZ|t 9}| t 7} ||9}qH| dkrkt|| }t|d||}|S) z[Computes Pr(D_n <= d) using the Pomeranz recursion algorithm. Pomeranz (1974) [2] r rrrr)rrrN)r r floorrCr$r&r"rKfillconvolver%r.r,r-r+r/r) rxrtrDfgrErFnpwrsgpower twogpower onem2gpowerr?g_over_n two_g_over_none_minus_two_g_over_nr4V0V1V0sV1srGrHr<k1pwrsln2conv conv_startconv_lenansrrr_kolmogn_Pomeranzsl     (       ( "     rec% Cs.|dkr tdd|dS|dkrtdd|dSt||}|d|d|d|df\}}}}t d|}|tkrAtdd|dSt|} | } td} d|d|} d|d |td} td d|d }td d |d }td|d|d }td|d|d}d|d|d}td}t t d |tj }t |ddD]G}d|d }|d|d|d}}}t | d|}td| | || | ||||||||||g}||9}||7}q|| 9}|t9}|t|d|d|dd|dg}tt d|} t|dd}|d}t|}tj |}| |} t|| }!|!ttd|9}!|d|!7<t|||||| }"|"ttd|9}"|d|"7<t |dtt|d}#||#}|s|d9}|dd 7<t|}$|$S)aPComputes the Pelz-Good approximation to Prob(Dn <= x) with 0<=x<=1. Start with Li-Chien, Korolyuk approximation: Prob(Dn <= x) ~ K0(z) + K1(z)/sqrt(n) + K2(z)/n + K3(z)/n**1.5 where z = x*sqrt(n). Transform each K_(z) using Jacobi theta functions into a form suitable for small z. Pelz-Good (1976). [6] rr rrrrrr@i`iZrrHiP i@)rr sqrt _PI_SQUARED_MIN_LOGexp_PI_FOUR_PI_SIXr"r r!pir&powerarray_SQRT2PIr#_SQRT3sumlen)%rrOrzzsquaredzthreezfourzsixqlogqk1ak1bk2ak2bk2ck3dk3ck3bk3aK0to3maxkr2r4msquaredmfourmsixqpowercoeffsksksquaredsqrt3zkspiqpwersk2extrak3extra powers_of_nKsumrrr_kolmogn_PelzGood#sl $     * rcCsPt|r|St||ks|dkrtjS|dkrtdd|dS|dkr*tdd|dS||}|dkrr|dkr=tdd|dS|dkrWttd|dd|d|d}ntt||t d|d}t|d||dS||dkrdd||}td|||dS|dkrdt j ||}td|||dS||}|dkr|d krt ||d d}t|d||dS|d krt||d d}t|d||dSdt j ||}td|||dS|s|d krdS|d krdt j ||}t|S|dkrd}n|dkr||ddkrt ||d d}nt||d d}t|d||dS)zComputes the CDF(or SF) for the two-sided Kolmogorov-Smirnov statistic. x must be of type float, n of type integer. Simard & L'Ecuyer (2011) [7]. rr rrfrrrg0q&?Trg w@g@g2@ig?gffffff?)r isnanr nanrprodr#rxrr scipyspecialsmirnovrArerr)rrOrrPprob nxsquaredrrrr_kolmognvsX ,$  rcsDtrStksdkrtjS|dks|dkrdS|}|dkr`|dkr,dSdkrDttddd|d}nttdtd|d}|ddS|dkrrdd|dS|dkrdt j j |S|d}t ||d}t |d|}fd d }t|||d d S) zvComputes the PDF for the two-sided Kolmogorov-Smirnov statistic. x must be of type float, n of type integer. rr rrrrrg@cs t|SN)kolmogn)_xrrr_kks z_kolmogn_p.._kkrh)dxorder)r rr rrr#rxrr rstatsksonepdfrCr)rrOrPprddeltarrrr _kolmogn_ps. ((  rcstrStksdkrtjSdkrdS|dkr"dStttjd}|dkrB|ddSt t|d }|ddkrY|St t }t |dd}fdd}tjj|d|dd S) zeComputes the PPF/ISF of kolmogn. n of type integer, n>= 1 p is the CDF, q the SF, p+q=1 rr rrrtcst|Sr)r)rOrrrr_fsz_kolmogni.._fg+=)xtol)r rr rrxr rrloggammaexpm1scu _kolmogcirurCoptimizebrentq)rrrrrOx1rrrr _kolmognis$ $ rc Cstj|||dgdgdtjtjtjgd}|D](\}}}}t|r&||d<qt||kr3td|tt|||d|d<q|jd}|S)aComputes the CDF for the two-sided Kolmogorov-Smirnov distribution. The two-sided Kolmogorov-Smirnov distribution has as its CDF Pr(D_n <= x), for a sample of size n drawn from a distribution with CDF F(t), where :math:`D_n &= sup_t |F_n(t) - F(t)|`, and :math:`F_n(t)` is the Empirical Cumulative Distribution Function of the sample. Parameters ---------- n : integer, array_like the number of samples x : float, array_like The K-S statistic, float between 0 and 1 cdf : bool, optional whether to compute the CDF(default=true) or the SF. Returns ------- cdf : ndarray CDF (or SF it cdf is False) at the specified locations. The return value has shape the result of numpy broadcasting n and x. N zerosize_ok)flags op_dtypes.n is not integral: rfr) r nditerfloat64bool_rr ValueErrorroperands) rrOrit_nr_cdfrresultrrrrs   rcCsnt||dg}|D]%\}}}t|r||d<q t||kr&td|tt|||d<q |jd}|S)aComputes the PDF for the two-sided Kolmogorov-Smirnov distribution. Parameters ---------- n : integer, array_like the number of samples x : float, array_like The K-S statistic, float between 0 and 1 Returns ------- pdf : ndarray The PDF at the specified locations The return value has shape the result of numpy broadcasting n and x. N.rr)r rrr rrr)rrOrrrrrrrrkolmognps   rc Cst|||dg}|D]7\}}}}t|r||d<q t||kr(td||r0|d|fnd||f\}} tt||| |d<q |jd} | S)aComputes the PPF(or ISF) for the two-sided Kolmogorov-Smirnov distribution. Parameters ---------- n : integer, array_like the number of samples q : float, array_like Probabilities, float between 0 and 1 cdf : bool, optional whether to compute the PPF(default=true) or the ISF. Returns ------- ppf : ndarray PPF (or ISF if cdf is False) at the specified locations The return value has shape the result of numpy broadcasting n and x. N.rrr)r rrr rrr) rrrrr_qrr_pcdf_psfrrrrkolmogni;s    r)T)'numpyr scipy.specialrscipy.special._ufuncsr_ufuncsrscipy._lib._finite_differencesrr-r/ longdoubler,r.rur{r~r r rwrrvryrzrrrrrArKrerrrrrrrrrrrs8C       K  T S<* %