Q(hIfdZddlZddlZddlZddlZddlmZddlZddlZddl Z ddl m Z m Z m Z mZmZmZmZmZmZmZmZddlZddlmZddlZddlmZmZddlmZddlm Z dd l!m"Z"dd l#m$Z$m%Z%m&Z&m'Z'ddl(m)Z)dd l*m+Z+dd l,m-Z-ddl.Zdd l/m0Z0m1Z1m2Z2ddl3m4Z4m5Z5ddl6m7Z7m8Z8ddl9m:Z:m;Z;ddlZ>m?Z?ddl@mAZAejjdkDZDejdk(xre jdk(ZFddgZGddZHdZIdZJGddZKddZLdZMejjdgd d!ZPGd"d#ZQGd$d%ZRGd&d'ZSGd(d)ZTGd*d+ZUGd,d-ZVGd.d/ZWGd0d1ZXGd2d3ZYGd4d5ZZGd6d7Z[Gd8d9Z\Gd:d;Z]Gd<d=Z^Gd>d?Z_Gd@dAZ`GdBdCZaGdDdEZbGdFdGZcGdHdIZdGdJdKZeGdLdMZfGdNdOZgGdPdQZhGdRdSZiGdTdUZjGdVdWZkGdXdYZlGdZd[ZmGd\d]ZnGd^d_ZoGd`daZpGdbdcZqGdddeZrGdfdgZsGdhdiZtGdjdkZuGdldmZvGdndoZwGdpdqZxGdrdsZyGdtduZzGdvdwZ{GdxdyZ|Gdzd{Z}Gd|d}Z~Gd~dZGddZGddZGddZGddZGddZGddZGddZddejddejzz zddejddejzz zddejddejzz zddejddejzz zddejddejzz zg fdejddejzz ddejddejzz zddejddejzz zdz ddejddejzz zdz ddejddejzz zdz g fddejdejzz dddejdejzzz dddejdejzzz dddejdejzzz dddejdejzzz dddejdejzzz dddejdejzzz dgfgZGddZGddZGddZGddZGddZGddZGddZGddZGddZGd„dëZGdĄdūZGdƄdǫZGdȄdɫZdʄZGd˄d̫ZGd̈́dΫZGdτdЫZGdфdҫZGdӄdԫZGdՄd֫ZGdׄdثZdلZGdڄd۫ZGd܄dݫZGdބd߫ZGddZGddZGddZGddZGddZGddZGddZGddZGddZGddZGddZGddZGddZGddZGddZGddZGddZGddZGddZGddZdZejjqeDd  d Zd Zd ZdZdZdZdZdZdZdZejjdgddZÐdZĐdZŐdZƐdZǐdZȐdZɐdZʐdZːd Z̐d!Zejjd"d#ej8d$d%gfd&ej8d'd(gfd)ej8d*d+gfd,ej8d-d.gfgd/Zΐd0Zϐd1ZАd2Zѐd3Zejjd4gd5d6Zejjd7gd8d9Zejjd:gd;d<ZՐd=Zejjd4gd>d?Zאd@ZGdAdBe)jZGdCdDe)jZGdEdFe)jZGdGdHe)jZGdIdJZGdKdLZejjqeDdM dNZdOZdPZdQZdRZdSZdTZdUZdVZdWZdXZejjd4gdYdZZejjd[d\d]gd^Zejjd_ddgd`fdadbgd`fdcddgdefgdfZdgZdhZdiZGdjdkZdlZGdmdnZGdodpZGdqdrZGdsdtZduZejjdve8e5zdwZdxZdyZejjd7dzd{gd|Zejjd:gd}d~ZdZdZdZGddZGddZGddZGddZGddZGddZejjdgddZy(z! Test functions for stats module N)Path) assert_equalassert_array_equalassert_almost_equalassert_array_almost_equalassert_allcloseassert_ assert_warnsassert_array_lesssuppress_warningsassert_array_max_ulpIS_PYPY)raises) typecodesarray)rec_append_fields)special)check_random_state)IntegrationWarningquad trapezoidcumulative_trapezoid) argsreduce)_XMAX)xlogy polygammaentr)distcont invdistcont) distdiscreteinvdistdiscrete) FitDataError _argus_phi)rootfmindifferential_evolution)productdarwinx86_64 tukeylambdapearson3cD||d|}tt|||y)Nz does not have attribute msg)r hasattr)abr/s b/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/scipy/stats/tests/test_distributions.py_assert_hasattrr45s( {,QC0 GAqMs#cLttjjdy)Nf_gen)r4scipystats distributionsr5r3test_api_regressionr<;sEKK--w7r5cttjjj}t Dcgc]}|d }}t Dcgc]}|d }}gd}||z|z}ttd|}||k(sJycc}wcc}w)Nr) rv_discrete rv_continuous rv_histogramentropytrapzc8t|jd S)N<)str startswithss r3z.test_distributions_submodule..JsA(9(9#(>$>r5)setr8r9r:__all__rr!filter)actualdist continuousdiscreteotherexpecteds r3test_distributions_submodulerS@s **22 3F&./d$q'/J/$01DQ1H1 !EH$u,H6>IJH X  01s B B c reZdZejj dgdejj dddejddgdZd Z d Z ejj d gd d Z ejj dgddZ dZ ejj dgddZdZejj ejj dddgejj dgdejj dddgejj dddgdZejj$dZejj d d!dgd"Zd#Zd$Zd%Zy&)' TestVonMisesk)皙?r exrr dcd}d}||dd|||dd|||dd|||dd|||dd|||dd|y)Nctj|||}t|j||j|dtj z|zzy)Nlocscale)r9vonmisesrpdfnppirVLrHrYvms r3check_vonmises_pdf_periodiczHTestVonMises.test_vonmises_periodic..check_vonmises_pdf_periodicSs@q2B q 266!q255y1}2E+F Gr5ctj|||}t|j|dz|j|dtj z|zzdzy)Nr^r ra)r9rbrcdfrdrerfs r3check_vonmises_cdf_periodiczHTestVonMises.test_vonmises_periodic..check_vonmises_cdf_periodicWsMq2B q A "qAIM': ;a ? Ar5rr rZr;)selfrVrYrirls r3test_vonmises_periodicz#TestVonMises.test_vonmises_periodicPsd H A $Aq!Q/#Aq!Q/#Aq"a0#Aq!Q/#Aq!Q/#Aq"a0r5cttjjtj ttjj tj yN)rr9 vonmises_liner1rdrer2rms r3test_vonmises_line_supportz'TestVonMises.test_vonmises_line_supportds8U((**RUUF3U((**BEE2r5cdtjd}t|jddy)N r?)r9rbrrk)rmrhs r3test_vonmises_numericalz$TestVonMises.test_vonmises_numericalhs" ^^C BFF1Is+r5zx, kappa, expected_pdf))rW{Gz?g|65?)rW9@g?U?)rWrug"?)@rxgDfI?)rzryg1<)rzruc`tjj||}t||dyNV瞯<rtol)r9rbrcr)rmrYkappa expected_pdfrcs r3test_vonmises_pdfzTestVonMises.test_vonmises_pdfvs&nn  E*\6r5zkappa, expected_entropy))r gA ?)g,eޞ?)r[gBb9d22)g G)g c^tjj|}t||dyNvIh%<=r)r9rbrAr)rmrexpected_entropyrAs r3test_vonmises_entropyz"TestVonMises.test_vonmises_entropys%..((/!1>r5cLd}tjj|}tjj|}tjj|}tjdddj |}tjddtj zdj |}tjdddtj zt|z dzj |}t||dt||dy) Ni@}r rr^ random_staterar~atol) rdrandom default_rngr9rbrvsreabsr)rmseedrng1rng2rng3rvs1rvs2rvs3s r3test_vonmises_rvs_gh4598z%TestVonMises.test_vonmises_rvs_gh4598syy$$T*yy$$T*yy$$T*~~aQa044$4G~~aQruuWA6:::M~~aQ%&ruuWSY%6q%8;;>3D3;Q d/d/r5zx, kappa, expected_logpdf))rWrxg EJ?)rWrygoι?)rWrug0,)rzrxgdty)rzrygʹw[A)rzrugc`tjj||}t||dyr})r9rblogpdfr)rmrYrexpected_logpdfrs r3test_vonmises_logpdfz!TestVonMises.test_vonmises_logpdfs&&&q%0e.r5r racyrr;rs r3rIz3TestVonMises.test_vonmises_expect..rr5c2tjd|zS)Ny?)rdexprs r3rIz3TestVonMises.test_vonmises_expect..sBFF2a4Lr5 complex_funcN) rdrrr9rbexpectr issubdtypedtypefloatingreanglecomplexfloating)rmrngr_rlbresboundss r3test_vonmises_expectz!TestVonMises.test_vonmises_expectsUii##$78A+UBnnE299+FQ}}SYY 444R!bee)^#9ennE299+OOQ}}SYY 444R!bee)^#9ennE299:PS;ASPQS sag7}}SYY(:(:;;;r5rvs_locra rvs_shape)r r[חAfix_locTF fix_shapecF|r|rtjdtjj d}t j j|d||}ddi}|r||d<|r||d<tt j |t j jfi|y) NzNothing to fit.rr)sizer_rfscaler flocf0) pytestskiprdrrr9rbr_assert_less_or_close_loglikennlf)rmrrrrrdatakwdss r3test_fit_MLE_comp_optimizerz(TestVonMises.test_fit_MLE_comp_optimizers  KK) *ii##$78~~!!)$G/2"4!} "DL "DJ%ennd&+nn&9&9 C=A Cr5cbgd}tj|}dtjz}tjj ||\}}}|tj tjk(sJttj|tjjd|y)N)g!*Kg4,%OԿgZ۵?g ge_@g-8gPghn?grrag d?)r8r9rbrrdrerrs r3test_vonmises_fit_boundsz%TestVonMises.test_vonmises_fit_bounds sT    !WA 6    "%%'9"5!6Q ?r5N)__name__ __module__ __qualname__rmark parametrizerdrernrsrwrrrrrxslowrslowrrrrrr;r5r3rUrUOs [[S-0 [[S1aC"891:11$3, [[5/0707 [[6;< ? < ? 0  [[8BC=C=<4 [[ [[YA/ [[[-8 [[Yu 6 [[[4-8C9790 C$ [[ O O [[Vb!W- (. ( M9 @r5rUc X| |j}|j|fi|}tt|||fi|}|st j ||k7sJ|||}|||}||kst j ||dsJd|vr |d|dk(sJd|vr|d|dk(sJyy)a^ This utility function checks that the negative log-likelihood function (or `func`) of the result computed using dist.fit() is less than or equal to the result computed using the generic fit method. Because of normal numerical imprecision, the "equality" check is made using `np.allclose` with a relative tolerance of 1e-15. Nr~rrrr)rrsupertyperdanyallclose) rNrfuncmaybe_identicalrmle_analytical numerical_optll_mle_analyticalll_numerical_opts r3rrs |yyTXXd+d+N$t*d/==M vvn 5666^T2M40 !1 1 KK)+;% HJ I ~b!T&\1114b!T(^333r5c ddg}|jr.t|jjd}|gdd|z }tt |t j t|}gd}tjtd5|j|fi|dddtjtd5|jt jgdddtjtd5|jt jgdddtjtd 5|j|d dddtjtd 5|j|gd gt|d z zdddy#1swYxYw#1swYxYw#1swYxYw#1swYvxYw#1swYyxYw)Nrr,)rf1f2r rar3All parameters fixed. There is nothing to optimize.matchz#The data contains non-finite valueszUnknown keyword arguments:ra) extra_keywordzToo many positional arguments.r )shapeslensplitdictziprdarangerr RuntimeErrorr ValueErrornaninf TypeError)rNparamnshapes all_fixedrs r3assert_fit_warningsr;s X E {{dkk'',- #HW--S #e* 567I D |& '$ ##$ zB D "&& zB D "&& y(D E( Q'( y(H I.-SZ!^,-..$$((..s<F2!F? !G G$G#2F<?G GG #G,rN)alpha betaprime fatiguelifeinvgammainvgauss invweibull johnsonsblevylevy_llognormgibrat powerlognormrayleighwaldctt}||}tt|}t |j |j g|dt|j|j g|tj t |j |jg|dt|j|jg|tj y)zgh-6235rN) r rgetattrr9rrcr1rrrdrr2)rNdctargss r3 test_supportr+Rs x.C t9D 5$ D/$/3TVV+d+bffW5/$/3TVV+d+bffW5r5c$eZdZdZdZdZdZy) TestRandIntcBtjjdyNrrdrrrrs r3 setup_methodzTestRandInt.setup_methode tr5c0tjjddd}tt j |dkt j |dk\ztt |dk(tjjddd}tt j|dk(t|jjtdvtjjdd}t|dk\|dkztt|tjtt| tjddjd }t|jjtdvy) Nrr[rra2 AllInteger.r.r)r9randintrr rdrr shapercharr isinstance ScalarTypereprrrmvalsvals r3test_rvszTestRandInt.test_rvshs}}  BS 1tby!BFF419$556D S !}}  BW 5')* 9\#::;mmB'sRx() 3 .DcODmmB#''* )L"99:r5ctjdd}tj|dk\|dkzdd}tjj |dd}t ||y)Nr$rr4{Gz?)rdr_wherer9r;pmfr)rmrVoutrBs r3test_pdfzTestRandInt.test_pdfusR EE!BKhhQ1r6*J:}}  Ar*!$,r5ctjddd}tj|}tj|dk\|dk\gd|dz dzd z gd}tj j |dd}t||d y) NrrFr[r4r?@r ry decimal)rdlinspacefloorselectr9r;rkr)rmrYrVrKrBs r3test_cdfzTestRandInt.test_cdf{sv KK2s # HHQKiib!q&)C!C%'F1C+DaH}}  Ar*!$R8r5N)rrrr1rDrLrVr;r5r3r-r-ds ;- 9r5r-cBeZdZdZdZdZdZdZdZdZ dZ d Z y ) TestBinomcBtjjdyr/r0rrs r3r1zTestBinom.setup_methodr2r5ctjjddd}tt j |dk\t j |dkztt j |dk(t|jjtdvtjjdd}tt|ttjddjd}tt|tjt|jjtdvyNrZrr6r5rr8r) r9binomrr rdrr<rr=rr>intndarrayrAs r3rDzTestBinom.test_rvss{{r4g6tqy!BFF42:$667')* 9\#::;kkoob$' 3$%kk"d#''* 3 +, )L"99:r5ctjjddd}tjjddd}t|dddt|dddy)Nr[r rrNr~rrr9r\rJr)rmvals1vals2s r3test_pmfzTestBinom.test_pmfsJ S!, 3*sQ7sQ7r5ctjdd}tjgd}t t || }|j }t||tjdd}|j }t|dtjdd}|j }t|dy)Nrarv)?rvrfr{rN) r9r\rdrsumrrArr)rmr2 expected_p expected_hhs r3 test_entropyzTestBinom.test_entropys KK3 XX/0 % J788 IIK:& KK3  IIKQ KK3  IIKQr5cFtj5tjdtt t j ddjdt t j ddjddddy#1swYyxYw)Nerrorrarnp) warningscatch_warnings simplefilterRuntimeWarningrr9r\rstdrrs r3 test_warns_p0zTestBinom.test_warns_p0so  $ $ & 9  ! !'> : qA.335q 9 qA.224a 8 9 9 9s A9BB cVd}tjjd|d|k(sJy)Nr333333?rN)qrorp)r9r\rrmros r3 test_ppf_p1zTestBinom.test_ppf_p1s( {{S1Q666r5cd}d}tjd}tjj |||}tj j |||z}t ||dy)NglyCgq]\ T3r]r^rAs r3rDzTestBernoulli.test_rvss""4g"6tqy!BFF419$556')* 9\#::;oo!!$' 3$%ood#''* 3 +, )L"99:r5ctjd}dtjdzdtjdzz }|j }t ||tjd}|j }t |dtjd}|j }t |dy)Nrfпrr{rN)r9rrdlogrArr)rmr2rirjs r3rkzTestBernoulli.test_entropys OOD !266$<'$rvvd|*;; IIK:& OOC  IIKQ OOC  IIKQr5N)rrrr1rDrkr;r5r3rrs ; r5rceZdZdZy) TestBradfordcd}tjdd}tjj ||}tjj ||}t ||y)NrW)rdlogspacer9bradfordrkrr)rmcrYryxxs r3 test_cdf_ppfzTestBradford.test_cdf_ppfsM  KKR  NN  q! $ ^^  1 %2r5N)rrrrr;r5r3rrsr5rc|eZdZdZej j dddddddd d d d ed fg dZej j dgdej j de jjdfe jjdfgdZ ej j dgddZej j dgdej j de jjdfe jj dfgdZy) TestCauchycPtjjd}|dk(sJy)NZbtir{)r9cauchyrcrs r3test_pdf_no_overflow_warningz'TestCauchy.test_pdf_no_overflow_warnings# LL  U #Cxxr5x, ref)r{罡HP)gr)gIL8r)gOd@gHP)Mb@?gP)rWg*6y)?g8@(J)gNgmCg'K@$[U)rgonьgs 32chtjj|| g}t|||gdyr})r9rrr)rmrYrlogps r3 test_logpdfzTestCauchy.test_logpdf s.||""Ar7+sCju5r5))g7y1gu lY<)g^?)rrf)rrv)r r)rg)nE?)g7y1C?z method, sgnr rc6|||z}t||dyr}r)rmrYrmethodsgnrps r3 test_cdf_sfzTestCauchy.test_cdf_sfs 3q5M3U+r5))gRyXsg!;v+͋)(,* EgjLҳ)$@g )r{9B.)g$gCį )g :@sgg>Z)g 5g*\Bctjj|}t||dtjj | }t||dyN+g /)rf)rvr{)rrN) !?gqmA)g?gqJRBc6|||z}t||dyr}r)rmrprrrrYs r3 test_ppf_isfzTestCauchy.test_ppf_isf>s q M3U+r5N)rrrrrrrrrr9rrkrrrrisfrr;r5r3rrsa [[ # & % ' $ # # # # $ $ % ' (6 (6  [[ %  [[ ,,  A  ,,//2   ,  ,  [[X;<0<0 [[ /  [[ ,,  A  ,,  B  !", " ,r5rceZdZdZdZdZdZdZdZe jjddefd gd Z e jjdgd d Z y )TestChignT9;g2 >h?@cttjjdd}t||jdy)NrZrr~r)r9chirr CHI_SF_10_4rmrHs r3test_sfzTestChi.test_sfZs( IILLQ 4++%8r5cttjj|jd}t |ddy)NrrZr~r)r9rrrrrmrYs r3test_isfzTestChi.test_isf^s( IIMM$**A .2E*r5chd}d}tjj||}t|ddy)Nrr9ga rr)r9rrr)rmrYdfrs r3 test_logcdfzTestChi.test_logcdfbs/  !!!R( 6UCr5chd}d}tjj||}t|ddy)Nrxr9@y`rr)r9rrr)rmrYrrs r3 test_logsfzTestChi.test_logsfis-   2&5EBr5df, ref@@)gļBgcAcZttjj||dyN-q=r)rr9rrrmrrs r3 test_meanzTestChi.test_meanvs  r*Ce?gG6ƴ?g9b?|^?߈܁r?gT C?rrgm?gFZۤ?g HB?gTXek?g?bȓE?gLZ;x?gJ>+=r@)g2c}`?g>t\f?g˂ q?gg'T){?g mw?g>t\?h'T)?7-h J?u8n?A#?g:Q?rrrrgZrԢ?g߄q?g)aL?gr'Orvrrr_r`)g3c}?g>t\?g˂ ?f'T)?#n?5-h J??u8n?vL1??#?||P?g:Q?rrrrrrrr)rdrSr9 crystalballrcrrrmX calculatedrRs r3rLzTestCrystalBall.test_pdfs KKc2 &s +&&**13#*> 88 K L *59&&**13#*> 88 J K *59&&**13#3c*R 88 G H *59r5ctjddddd}tjj |dd}tj gd }t ||d tjj |dd }tj gd }t ||d tjj |dd dd}tj gd}t ||d y)rrrOrNrrNrzr)ghQ| )?gj,,?g d}9?gH6?g|^?gEv?glQ| )?g d}9?g|^?go[?gsX?g& Ռ?g~M+?gZR c?gtE_s?gi?gB?gEg?gvP ?gx훟?rrr)gpr?gt\v?ge}?gݠn^?gCȘU?at\?ݠn^?~[n|D?gS4s/?2s?`qfhG?IgL?荩s?rA?IH?guk?ghi"?gP ?g.q?gs?rvr)go?rge?rgPU,?rgjݗ ?rg?rg_!Y?rga?rgR?r g( ?r g܉x?r )rdrSr9rrkrrrs r3rVzTestCrystalBall.test_cdfs KKc2 &s +&&**13#*> 88 E F *59&&**13#*> 88 E F *59&&**13#3c*R 88 E F *59r5zx, beta, m, rootref))(@rNrzgI9)"@@?gy <)rWgjt?g +,)grzrgGA*?)g>rvrOg^v'?)gd~Qr皙?g眭?cdtjj|||}t||dy)Nrrr)r9rrr)rmrYrrrootrefrs r3rzTestCrystalBall.test_sfs-   ! !!$! ! 4G%0r5ctjgd}tjgd}tjgd}tjj d||}t ||dtjgd}tjdd tj d d g}||z }tjj d ||}t ||dtjtj tj tj d dg}||z } tjj d||} t | | dtjtj tj tj tj dg}||z } tjj d||} t | | dtjtj tj tj tj dg}||z } tjj d||}t | |dtjtj tj tj tj dg}||z }tjj d||}t ||dy)zs All values are calculated using the pdf formula and the integrate function of Mathematica )rzrNrrzr)rrrzrr )rNrNrNrNrNrMbP?r)gԚh@g [&@gͪ@g2dQ@gn6-@g?ޫV&̿gBgQCRgQF\jr g @gn(@rag?Җ)rgMg'#@rg('UHrN)rdrr9r_munprr)rmrrexpected_0th_momentcalculated_0th_momentnormr1expected_1th_momentcalculated_1th_momentexpected_2th_momentcalculated_2th_momentexpected_3th_momentcalculated_3th_momentexpected_4th_momentcalculated_4th_momentexpected_5th_momentcalculated_5th_moments r3 test_momentszTestCrystalBall.test_momentss* xx12 HH. /!hh'@A % 1 1 7 74 C+-BO xxHI HHh"&&)YG H$h % 1 1 7 74 C+-BO HHbffbffbfffh? @$h % 1 1 7 74 C+-BO HHbffbffbffbffjA B$h % 1 1 7 74 C+-BO HHbffbffbffbffg> ?$h % 1 1 7 74 C+-BO HHbffbffbffbffh? @$h % 1 1 7 74 C+-BOr5ctjdd}|j}d\}}}tj|||}t t |j||}t||dy)Nrar)ir4@ Hz>r) r9rrArdrSrrrcr)rmcbres1lohiNrYres2s r3rkzTestCrystalBall.test_entropyLs`   q! $zz|& B KKB "bffQi!,d.r5N) rrrrLrVrrrrr$rkr;r5r3rrsE<:|<:D [[ 011+PZ/r5rc$eZdZdZdZdZdZy) TestNBinomcBtjjdyr/r0rrs r3r1zTestNBinom.setup_methodXr2r5cntjjddd}tt j |dk\tt j |dk(t|jjtdvtjjdd}tt|ttjddjd}tt|tjt|jjtdvyr[) r9nbinomrr rdrr<rr=rr>r]r^rAs r3rDzTestNBinom.test_rvs[s||Dw7tqy!"')* 9\#::;llr4( 3$%ll2t$((+ 3 +, )L"99:r5c ,ttjtjj dddtjj dddtjjj ddd}t|dy)Niigp= ף?rr ) rrdrr9r2logpmfrJr8r)rmrCs r3rdzTestNBinom.test_pmffshu||223TBC ((c48 :kk  ''1a0S!r5ctjjgddd}tjtjj gddd}t ||y)N)rrrr333333@?rn)r9r2rrdrrkrrmrBrs r3test_logcdf_gh16159zTestNBinom.test_logcdf_gh16159nsK||""<3$"?ffU\\%%lcT%BCc"r5N)rrrr1rDrdr9r;r5r3r/r/Ws ;#r5r/c eZdZdZej j dZej j dZej j dZ ej j dZ dZ dZ dZ d Zy ) TestGenInvGausscBtjjdyr/r0rrs r3r1zTestGenInvGauss.setup_methodvr2r5ctjdd}tj|jdd|j\}}t |dkDdy)Nffffff@rrr皙?Tr9 geninvgausskstestrrkrrmgigrrps r3test_rvs_with_mode_shiftz(TestGenInvGauss.test_rvs_with_mode_shiftyIS)||CGGDGA377K1QXt$r5ctjdd}tj|jdd|j\}}t |dkDdy)Nrrr?rrr@TrArDs r3test_rvs_without_mode_shiftz+TestGenInvGauss.test_rvs_without_mode_shiftsIT*||CGGDGA377K1QXt$r5ctjdd}tj|jdd|j\}}t |dkDdy)NrW皙?r?rrr@TrArDs r3test_rvs_new_methodz#TestGenInvGauss.test_rvs_new_methodrGr5c|d}t|dddt|dddt|dddy)Nctj||}|jdd}tj||jddkDS)Nr?rrr r@)r9rBrrCrk)rpr2rErs r3 my_ks_checkz4TestGenInvGauss.test_rvs_p_zero..my_ks_checksE##Aq)C''t$'7C<<SWW-a047 7r5rrKTrr)r)rmrOs r3test_rvs_p_zerozTestGenInvGauss.test_rvs_p_zeros< 8 [C($/[C($/[C($/r5cttjddjdddtjddjddz y)NrarZrrr r)rr9rBrrrs r3test_rvs_negative_pz#TestGenInvGauss.test_rvs_negative_psP!!$*..BT.JE%%c1-11r1MM Or5ctjjdddd}ttj|ddgddkDd d t j d dd }}tjj|dd|z | }t|tj|j|tjj|dd|z | }t|tj|j|y)Nr?rr r)rrpr2rrr*333333?Tr[rxrZ)rpr2r`) r9rBrrrCrdrSrcrrrk)rmigmurYpdf_igcdf_igs r3 test_invgausszTestGenInvGauss.test_invgausss    " " " MU\\"js;A>EtLR[[q"-A""&&qDAF"&Er 2 6 6q 9:""&&qDAF"&Er 2 6 6q 9:r5ctjgd}tjddd}t|tj j |ddy)N) g;g8y8?g\}Z?ghܲ0|?gRq?rxrrZrvr )rdrrSrr9rBrc)rmvals_RrYs r3 test_pdf_RzTestGenInvGauss.test_pdf_RsJ,- KKa $ 1 1 5 5aa @Ar5cttjjddddttjjddddy)Nrrvg>Ar7ra)rr9rBrcrrs r3 test_pdf_zerozTestGenInvGauss.test_pdf_zerosBU&&**1c37; U&&**3A6:r5N)rrrr1rrrrFrIrLrPrSr[r^r`r;r5r3r;r;us [[%%  [[%%  [[%%  [[00O ; B;r5r;ceZdZdZdZdZejjdgddZ ejjdgddZ d Z d Z d Z d Zd ZdZy)TestGenHyperboliccBtjjdyr/r0rrs r3r1zTestGenHyperbolic.setup_methodr2r5ctjgd}d\}}}d\}}|||z||zf}tj|||d}tjddd} t |j | |dd y) N) gF^T=g{X'=gNunƢsy>g7!t?gɫzI?g̸?g?gq 6.?gmPZz?g:PgN?rarar rvrr^rZrrrr)rdrr9 genhyperbolicrSrrc rmr]lmbdarrrXdeltar*ghrYs r3 test_pdf_rzTestGenHyperbolic.test_pdf_rs~ %ud EuU{DJ/  $Be < KKR $q 6>r5ctjgd}d\}}}d\}}|||z||zf}tj|||d}tjddd} t |j | |dd y) N) g=R0W<=gEB=g#Nb>g_PG>g! 1Or?gUM?g u./?gfy?gl5\?gkV?rerfr^rgrZrư>rh)rdrr9rirSrrkrjs r3 test_cdf_rzTestGenHyperbolic.test_cdf_rs~ %ud EuU{DJ/  $Be < KKR $q 6=r5zx, p, a, b, loc, scale, ref) )rarrrvrg8ɹvF(;)rrrZrrfr rg[ Ϡ?)rrrZrg?rr g✝`D;)rr?r):?rr g dL;)rrsrrtrr gzK^7?)rrrtrr gj?)rrurrr ga=vG:)rrurrvrr g?)(rurrvrr g1bV?)<rurrvrr g?cjtjj||||||}t||dyNr^dy=r)r9rirkr) rmrYrpr1r2r_r`rrks r3test_cdf_mpmathz!TestGenHyperbolic.test_cdf_mpmaths3!!%%aAqc%GSu-r5))rrprPrrr gC!,?)rr@@r rg,Ћf?)rrr}r~r rrN)rarrrvrgm"چ >),rZrrfr rg:;)rxrurrvrr gw=)Krurrvrr g3Z0&K4 [[% < . . [[% @ - -=. %$"*r5rbceZdZejj dddgdZejj dddgdZd Zy ) TestHypSecantz x, reference)r4gA0=)r7g!b;c^tjj|}t||dyr)r9 hypsecantrr)rmrY referencers r3rzTestHypSecant.test_sfs$__   "IE2r5z p, reference)rpg9bT*@)rgTW-;@c^tjj|}t||dyr)r9rrr)rmrprrYs r3rzTestHypSecant.test_isfs$ OO   "951r5cd}d}tjj|}t||dtjj | }t||dy)NI@g!brr)r9rrrrrs r3rzTestHypSecant.test_logcdf_logsfsN %''*%0%%qb)s/r5N) rrrrrrrrrr;r5r3rrsg  [[^9:<=3=3 [[^89;<2<20r5rc~eZdZdZdZdZejjdgddZ dZ dZ d Z d Z y ) TestNormInvGausscBtjjdyr/r0rrs r3r1zTestNormInvGauss.setup_methodr2r5ctjgd}tjgd}tjj |dd}t ||dy)N)gp~٧>g @X>g}e?g{)߳?gwH?irrr9r rvr1r2rr)rdrr9rrkr)rmr_cdfx_testvals_cdfs r3 test_cdf_RzTestNormInvGauss.test_cdf_RsO <=,-%%))&A)=%d3r5ctjgd}tjgd}tjj |dd}t ||dy)N)gp϶>g\H55$?geСt&?go NiH?g]u>rr rvrrr)rdrr9rrcr)rmr_pdfrvals_pdfs r3r^zTestNormInvGauss.test_pdf_RsM<=,-%%))&A)=%d3r5zx, a, b, sf, rtol))rr r!7?r)rr r?=r)r rrRgy5`?r)rZrrRgMLC9rctjj|||}t|||tjj |||}t|||yNrr9rrrr)rmrYr1r2rrrHrs r3test_sf_isf_mpmathz#TestNormInvGauss.test_sf_isf_mpmathsP    ! !!Q *2D)    " "2q! ,14(r5cddg}ddg}d}ddg}tjj|||}t||dd tjj |||}t||d y) Nrrr rrrrr}r`rprr)rmrYr1r2rrHrs r3test_sf_isf_mpmath_vectorizedz.TestNormInvGauss.test_sf_isf_mpmath_vectorizedsr H F  "8 9    ! !!Q *2E6    " "2q! ,14(r5ctjdd}tjddd}|j |}|j |}t ||y)Nr rrra)r9rrdrrrr)rmdstrYrrs r3 test_gh8718zTestNormInvGauss.test_gh8718sJ  A& IIaQ  VVAYggbkQr5c(d\}}tj|dz|dzz }||z |dz|dzz d|z|tj|zz ddd|dzz|dzz zz|z f}t|tjj||dy) Nr rvrarrr rmvskmoments)rdrrr9r)rmr1r2gammav_statss r3 test_statszTestNormInvGauss.test_statss11q!t $u9adUAXosQw!bggen:L/M!a!Q$hAo-.68We0066q!V6LMr5cd\}}tjgd}tjj |||}t |tjj |||y)Nrrrvr)rdrr9rrrrk)rmr1r2rrBs r3test_ppfzTestNormInvGauss.test_ppfsS1-.!!%%fa3 2 2 6 6tQ BCr5N)rrrr1rr^rrrrrrrrr;r5r3rrsU 44 [[0JK ) K ) ) NDr5rcNeZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z y ) TestGeomcBtjjdyr/r0rrs r3r1zTestGeom.setup_methodr2r5chtjjdd}tt j |dk\tt j |dk(t|jjtdvtjjd}tt|ttjdjd}tt|tjt|jjtdvy)Nrr6r5rr8r) r9geomrr rdrr<rr=rr>r]r^rAs r3rDzTestGeom.test_rvsszz~~d~1tqy!"')* 9\#::;jjnnT" 3$%jj""1% 3 +, )L"99:r5cttjjd}tjj tj dd|}|jtjk(sJtj|tjtjjkDsJy)NlA:irr) rdrrr9rrrrint64riinfoint32max)rmrrs r3 test_rvs_9313zTestGeom.test_rvs_9313sxii##O4jjnnRVVC[qsnCyyBHH$$$vvcBHHRXX.222333r5cdtjjgdd}t|gdy)Nrrv)rvrfrs)r9rrJrrmrBs r3rdzTestGeom.test_pmfs!zz~~i-!$(:;r5c(tjtjj gdd}tjj gdd}t ||ddtjj dd}t|dy)Nrrvr~rr`r r{)rdrr9rrJr4rr)rmrbrcrCs r3 test_logpmfzTestGeom.test_logpmf seuzz~~i56 !!)S1u5q9jj1%S#r5ctjjgdd}tjjgdd}t gd}t ||t |d|z y)Nrrvrvr?r )r9rrkrrrrmrBvals_sfrRs r3rzTestGeom.test_cdf_sfsMzz~~i-**-- 3/+,!$1!'1X:6r5c$tjjgdd}tjjgdd}t gd}t |t j|t |t j| y)Nrrvr) r9rrrrrrdrlog1prs r3rzTestGeom.test_logcdf_logsfsazz  C0**""9c2+,!$x(89!'288XI+>?r5cztjjgdd}tgd}t ||y)Nrrv)rNrzr)r9rrrrrmrBrRs r3rzTestGeom.test_ppf!s+zz~~0#6)!$1r5c\ttjjddddy)NrrNrr)rr9rrrrs r3test_ppf_underflowzTestGeom.test_ppf_underflow&s ue4cFr5cftjdj}t|ddy)Ng<,Ԛ?g۩@r~r)r9rrAr)rmrjs r3test_entropy_gh18226zTestGeom.test_entropy_gh18226*s( JJv  & & (,59r5ctjjd}tjj dd|dkDj sJy)NiKH9rZrr)rdr RandomStater9rrr)rmrs r3test_rvs_gh18372zTestGeom.test_rvs_gh183720sByy,,Y7  u2LIAMRRTTTr5N)rrrr1rDrrdrrrrrrrr;r5r3rrs< ;4<7@2 G: Ur5rceZdZdZdZdZy) TestPlanckcBtjjdyr/r0rrs r3r1zTestPlanck.setup_method8r2r5cztjjgdd}tgd}t ||y)NrrO)g|dy?g8'\>ggrU!>)r9planckrrrrs r3rzTestPlanck.test_sf;s1||y"-23 "$1r5cztjjgdd}tgd}t ||y)N)r@@p@r)gP.gh>gTF)r9rrrrrs r3rzTestPlanck.test_logsfBs.||!!"7?:;!$1r5N)rrrr1rrr;r5r3rr7s22r5rc$eZdZdZdZdZdZy) TestGennormcgd}tjj|d}tjj|}t ||yNrr )r9gennormrcrrrmpointspdf1pdf2s r3 test_laplacezTestGennorm.test_laplaceIs;}}  +}}  (D$'r5cgd}tjj|d}tjj|d}t ||yNrrag;f?r`)r9rrcrrrs r3 test_normzTestGennorm.test_normPs=}}  +zz~~fF~3D$'r5ctjjdtjd}|j d}tj ||jjdkDsJtjd}|j d}tjj d}tj||jdkDsJtjd}|j d}tjj dd }tj||jdkDsJy) Nrrvrr5rWr rag;f?r`r) rdrrr9rrrCrkpvaluerks_2sampr)rmrNr rvs_laplacervs_norms r3rDzTestGennorm.test_rvsWs q}}S!hhDh!||C*11C777}}QhhDh!mm''T'2 ~~c;/66<<<}}QhhDh!::>>t><~~c8,33c999r5ctjjdtjddgddgg}|j gd}tj |ddddftjdjdd kDsJtj |ddddftjdjdd kDsJtj |ddddftjdjdd kDsJtj |ddddftjdjdd kDsJy) NrrvrNrzrO)rrarar5r rW)rdrrr9rrrCrk)rmrNrs r3test_rvs_broadcastingz!TestGennorm.test_rvs_broadcastinghs q}}sBi"b23hhLh)||C1aL%--*<*@*@A!DsJJJ||C1aL%--*<*@*@A!DsJJJ||C1aL%--*<*@*@A!DsJJJ||C1aL%--*<*@*@A!DsJJJr5N)rrrrrrDrr;r5r3rrHs((:"Kr5rcTeZdZejj dddgdZy) TestGibratx, sfx)rgD.?)gJ]bH>r5r ceZdZdZej j dgddZdZdZ ej j dgdd Z y ) TestGompertzctjjtjjddd}t |dyN0.++r )r9gompertzrrkrrs r3test_gompertz_accuracyz#TestGompertz.test_gompertz_accuracys3 NN  u~~11&! ))!Q/5A**32AEBr5cld}d}d}tjj||}t||dy)N @rWg _rr)r9rrr)rmrYrrrs r3rzTestGompertz.test_logcdfs3  %&&q!,%0r5cld}d}d}tjj||}t||dy)Ng"ul/rPgH 7Wrr)r9rrr)rmrYrrrs r3rzTestGompertz.test_logsfs3  $$Q*s/r5c, ref))rgeUT8?)r gtqq?)rgs{)rgN*6cZttjj||dyr)rr9rrArmrrs r3rkzTestGompertz.test_entropys ..q13UCr5N) rrrrrrrrrrrkr;r5r3rrso#" [[[+MNC NC10 [[X(EFD FDr5rcTeZdZejj dddgdZy) TestFoldNorm x, c, ref)r:0yE>g:0N<?)rrg:?c\ttjj|||dyr})rr9foldnormrkrmrYrrs r3rVzTestFoldNorm.test_cdfs! **1a0#EBr5N)rrrrrrrVr;r5r3r$r$s7 [[[+N+M+OPCPCr5r$ceZdZejj dddgdZejj dgddZejj dd d gejj d gd ejj d ddgejj dddgdZdZ y) TestHalfNormr )r |N?)rZ Oul2;cttjj||dttjj ||dyr)rr9halfnormrrrs r3rzTestHalfNorm.test_sf_isfs: ))!,c>**3/?r5r))Ww'&l7g;G<^͛7)C]r2r rvs_scalerxr[rrTF fix_scalectjjd}tjj ||d|}|rK|rId}t jt|5tjj|||dddyi}|r||d<|r||d<ttj|fi|d d iy#1swYyxYw Nrrr_r`rrrrrrrrrT) rdrrr9r/rrrr halflogisticrr rmrr5rr7rr error_msgrs r3rz(TestHalfNorm.test_fit_MLE_comp_optimizersii##$78~~!!gYT/2"4 y%I|9= M""&&t')&L M  "DL &DN &ennd )g?g@guVJrr)r9r<rrrmrYrrs r3rzTestHalfLogistic.test_logcdfOs. $##**1-%0r5cfd}d}tjj|}t||dy)Nrg+rr)r9r<rrrmrYrrs r3rzTestHalfLogistic.test_logsfVs. $""((+s/r5rr4rr5r6rTFr7ctjjd}tjj ||d|}i}|rK|rId}t jt|5tjj|||dddy|r||d<|r||d<ttj|fi|d d iy#1swYyxYwr9) rdrrr9r<rrrrrr) rmrr5rr7rrrr>s r3rz,TestHalfLogistic.test_fit_MLE_comp_optimizer]sii##$78!!%%'36&8 y%I|9= M""&&t')&L M  "DL &DN &e&8&8$ <$ <6: < M s &#CC cd}tt|5tjj gdddddy#1swYyxYw)Nz; Maximum likelihood estimation with 'halflogistic' requiresr)rrarr r) assert_raisesr#r9r<rrmr/s r3test_fit_bad_flocz"TestHalfLogistic.test_fit_bad_floczs?L g+c\ttjj|||dyN绽|=rrr9 truncnormrArmr1r2rs r3rkzTestTruncnorm.test_entropys!$ //15sGr5))rrghiɮ9?)rrr)g0.+rr)}Ô%IrZ_2?c\ttjj|||dyrrrs r3test_extreme_entropyz"TestTruncnorm.test_extreme_entropys! //15sGr5c tjjgddddgdzd}tjtj dd dd d tj g}t ||y) NrrrrvH.?r rarrNrrar^r fO?l@r)r9rrrdrrrrs r3test_ppf_ticket1131z!TestTruncnorm.test_ppf_ticket1131\""#EsB()s1uA#788RVVQ Az1bffMN!$1r5c tjjgddddgdzd}tjtj dd dd d tj g}t ||y) NrrrNrrrar^rrrr )r9rrrdrrrrs r3test_isf_ticket1131z!TestTruncnorm.test_isf_ticket1131rr5cd\}}tjj||ddd}t||j cxkxr|j cxkxr|kncd\}}tjj||ddd}t||j cxkxr |j cxkxr |kycy)N)irgrr rZr5)rZ r9rrr minrrmlowhighrYs r3test_gh_2477_small_valuesz'TestTruncnorm.test_gh_2477_small_valuess T OO  T1ab  9aeeg..$./ T OO  T1ab  9aeeg..$././r5cZd\}}tjj||ddd}t||j cxkxr|j cxkxr|knct |||gfd\}}tjj||ddd}t||j cxkxr|j cxkxr|kncd\}}tjj||ddd}t||j cxkxr|j cxkxr|kncd\}}tjj||ddd}t||j cxkxr |j cxkxr |kycy) N)r[rXrr rZr5)r)'i')i)r9rrr rrrErs r3test_gh_2477_large_valuesz'TestTruncnorm.test_gh_2477_large_valuessF T OO  T1ab  9quuw1!%%'1T12CdA4GG T OO  T1ab  9aeeg..$./ T OO  T1ab  9aeeg..$./" T OO  T1ab  9aeeg..$././r5cddgddgfD]\}}tjtj ||tjg}||zdz }tjj |||}tjj |||}tjj|||}tjgd}tjgd} tjgd} |d krtjgd } t||t|| t|| ttj| d | d z |d ztjgd} tjj| ||} tj|tj|dz|g} t| | |d krWttjj |||dttjj |||dnVttjj |||dttjj |||dtjj|||}ttj|| d z |dzd z y)Nrrrrzrrr r rNrNr{r{)rd._MT @BKg?rr)rrrrr rarvrrvrNgĖy @gdv*?g,mj%V?rf) rdrrr9rrkrrcrrrr)rmrrxvalsxmidcdfssfspdfs expected_cdfs expected_sfs expected_pdfspvalsppfs expected_ppfsrcs r3test_gh_9403_nontail_valuesz)TestTruncnorm.test_gh_9403_nontail_values s<a&2r(+ MICHHrvvgsD"&&9:EHc>D??&&uc48D//$$UC6C??&&uc48DHH\2M88$89LHH%ABMQw ")E F m 4 \ 2 m 4 }Q'7 a8H'H I #c  +HH]+E??&&uc48DHHc2773< +A4%HIM m 4Qw#EOO$6$6tS$$G$68#EOO$7$7c4$H$68$EOO$7$7c4$H$68#EOO$6$6tS$$G$68//%%dC6C s=+;'; rfrr) rdrrr9rrkrrcrrrrrS)rmrrrrrrrrrrrrrrcxvals2s r3test_gh_9403_medium_tail_valuesz-TestTruncnorm.test_gh_9403_medium_tail_values0sFr(S#J/, PICHHrvvgsD"&&9:EHc>D??&&uc48D//$$UC6C??&&uc48DHH\2M88$89LHH%KLMQw "*=!> m 4 \ 2 m 4 }Q'7 a8H'H I #c  +HH]+E??&&uc48DHHc2773< +CT%JKM m 4??&&tS$7D e ,Qw#EOO$6$6tS$$G$68#EOO$7$7c4$H$9;$EOO$7$7c4$H$68#EOO$6$6tS$$G$9;//%%dC6C s=+;'; K@casec|\}}}}}}tjj||d\}} } } t|| | | g||||gdy)NrrFFgxx{q#c(m+++saeeem+,-quu!u},-.r5cttjdr@tjj dddtjj yy)Nrrgrrr)r0rdrr9rrrrrs r3test_rvs_Generatorz TestTruncnorm.test_rvs_GeneratorsC 299m , OO  Ra-/YY-B-B-D  F -r5ctjtj tj dtj dg}tjtjtjddtjg}tjgd}gd}tt j ||j ||tt j | | j| |y)NrZr)rZ@r r)g Oul"gځ=!gL!g>EG fg^K[X)rdrrrr9rrr)rmr1r2rYrRs r3test_logcdf_gh17064z!TestTruncnorm.test_logcdf_gh17064s HHrvvgwRVVGR8 9 HHbffbffaRVV4 5 HH* ++ 1-44Q7BQB/55qb98Dr5cjtjddjd}d}t||y)Nrrrgl).0S?)r9rrr)rmrrs r3test_moments_gh18634z"TestTruncnorm.test_moments_gh18634s0oob!$++A. S!r5N)rrrr1rrrrkrrrrrrrrrdr_truncnorm_stats_datarr$rrrrrrr;r5r3rzrzsJ [[[BCH CH  [[[DE H E H2 2 00&!MF-P^,  1 = < BFF       &&!            ;"F%BHH%:; [[V%:;D<D + 0 /F E"r5rzceZdZejj dgddZejj dgddZejj ddd gd Zejj d d d gdZ ejj d ddgdZ ejj dddgdZ y)TestGenLogistic x, expected))ig4`n)ig4 ncg)rgX =)r[gh@ X)rg hr]r^rAs r3rDzTestHypergeom.test_rvs s""2r17";vvdai 266$!)#4444xx~(((zz)L"9999oo!!"a,#s###oob!R(,,Q/#rzz***yy~~rh)g@rg@g@g@)r rg) 9g"1)rdrr9rrr) rmorangespears fruits_eatenquantileeatenrrR quantilesr- expected2s r3test_precision2zTestHypergeom.test_precision2.sxx =>D (*!!(GeOWeL**889:XAD90 !!)Wu_guM> iad;*s,B6c>tjddd}|j}tjddg}tj t || }t||tjddd}|j}t|dy)Nrr rrfr{) r9rrArdrrgrrr)rmhgrjrhris r3rkzTestHypergeom.test_entropyAsz __Q1 % JJLXXtTl+ ffU:z:;; :& __Q1 % JJLQr5cd}d}d}d}tjj||||}d}t||dd}d }d }d }tjj||||}d }t||d y)NrcA.Aj@gorrQr @Xrg_7 j$r9)r9rrrrmrVrror,resultrRs r3rzTestHypergeom.test_logsfMs    &&q!Q2FHa8    &&q!Q2 FHb9r5c&d}d}d}d}tjj||||}d}t||dd}d }d }d }tjj||||}d }t||d d}d }d}d}tjj||||}d}t||d t j gd}d }d }d }tjj||||}t j dd }t||d y)Nr rrrg)\UrrQrwrr7rg*@Vr9}rgר)rwrwrw)r9rrrrdrfullrs r3rzTestHypergeom.test_logcdfes    ''1a3FHa8    ''1a3&FHb9    ''1a3 FHb9 HH\ "   ''1a377112FHb9r5czd}d}d}tjj|||}||z |z}t||y)NipiPi.)r9rrr)rmrror,hmrms r3test_mean_gh18511zTestHypergeom.test_mean_gh18511sB    __ ! !!Q * UQYBr5cd}d}tjdd}d|z}tjj |dz |||}tj |dkDsJtj tj |dksJy)NrZrrr9rr r)rdrr9rrrdiff)rmror,rpopulation_sizerps r3test_sf_gh18506zTestHypergeom.test_sf_gh18506sw   IIa q& OO  q1uoq! <vva!e}}vvbggaj1n%%%r5N)rrrr1rDrrrr rkrrrrrrr"r;r5r3rrsU 9>FL<& :0,:\   [[ & &r5rcDeZdZejj dgddZejj dgddZdZdZ d Z d Z ejj d d d gdZ ejj dgddZ y) TestLoggammaz x, c, cdf))r rag;Y%?)rgRٲa6_<)gHrgi|e?)rg6l ?)i+rWg֬CCf9)irgֺl ctjj||}t||dtjj ||}t||dyr)r9loggammarkrr)rmrYrrkrpys r3rzTestLoggamma.test_cdf_ppfsH NN  q! $3U+ NN  sA &15)r5zx, c, sf))rrgQOu3;)r[g\Z{0)r&rgeɐ?)ig{Gzd?gr0T?ctjj||}t||dtjj ||}t||dyr)r9r(rrr)rmrYrrrHr)s r3rzTestLoggamma.test_sf_isfsH NN  a #2E* NN  r1 %15)r5c`tjjdd}t|ddy)N rag@rr)r9r(rr)rmlps r3rzTestLoggamma.test_logpdfs& ^^ " "4 +G%0r5cld}d}tjj||}d}t||dy)Nr@gI.rr)r9r(rr)rmrYrrrs r3rzTestLoggamma.test_logcdfs3  &&q!,%%0r5cld}d}tjj||}d}t||dy)Ng9 @g!{1+ҷrr)r9r(rr)rmrYrrrs r3rzTestLoggamma.test_logsfs3  $$Q*#s/r5ctjgdjdd}|D];\}}}}}tjj|d}t |||||gd=y)N)rvg"~jg46<@goŏrrNg h"lxg Q?gZd;333333@r g{/L@g??gгYҿgh|?5?rrmsvkrrrQ)rdrreshaper9r(r)rmtablerrvarskewkurtcomputeds r3rzTestLoggamma.test_statsst wr1~  ). 1 $AtS$~~++Av+>H %hsD$0G./ 1 1r5rrWrctjj|d}tj|j sJtjj |}tjtj||kt|}|jd}|jdcxkr|jksJJy)Nrr5r)confidence_levelrv) r9r(rrdisfinitermedian binomtest count_nonzeror  proportion_cirr)rmrrYmedbtestcis r3rDzTestLoggamma.test_rvss NN  qv  .{{1~!!###nn##A& 0 0S 93q6B  % 8vv%bgg%%%%%r5r ))r&g3H໱k3@)r r)rg8,[H^} )rg90$)rgtum\cZttjj||dyr)rr9r(rAr"s r3rkzTestLoggamma.test_entropy s ..q13UCr5N)rrrrrrrrrrrrrDrkr;r5r3r$r$s [[[EF*F* [[ZBC * C * 110 1 [[S3,/ &0 & [[X<= D = Dr5r$cXeZdZgdZej j dedZy) TestJohnsonsu))r-r r g?r&)rr r gH';vIh%,=)rr r g?x7rIrc|\}}}}}ttjj||||dttjj |||||yNrIr)rr9 johnsonsurrrmrrYr1r2rtols r3rzTestJohnsonsu.test_sf_isf% P1aS**1a3ReD++B15qsCr5Nrrrcasesrrrrr;r5r3rHrH s0 ;E [[VU+D,Dr5rHcXeZdZgdZej j dedZy) TestJohnsonb))rr r rr')rr r g0jA:rI)rr r gQbF5rIrc|\}}}}}ttjj||||dttjj |||||y)NrIrr)rr9rrrrMs r3rzTestJohnsonb.test_sf_isf; rOr5NrPr;r5r3rSrS, s0 =E [[VU+D,Dr5rSceZdZdZdZdZdZdZejjdddgd Z d Z ejjd d d gdZ dZy) TestLogisticctjdd}tjj |}tjj |}t ||yNrr)rdrSr9logisticrkrrrmrYr)rs r3rzTestLogistic.test_cdf_ppfD sD KKR  NN  q ! ^^   "2r5ctjdd}tjj |}tjj |}t ||yrX)rdrSr9rYrrrrZs r3rzTestLogistic.test_sf_isfJ sD KKR  NN  a  ^^   "2r5cd}d}ttjjd|z |ttjj ||y)Ng.funck sDAqD AqA~.RVVTAXN335689A>B$(a&&$(a014&&$(a014678:;$(#&(TAX./ &ennd;r5N)rrrrrr^r`rcrrrrprtrzrr;r5r3rVrVB s  84)  [[03J3K3MNCNC,F  [[\D%=9 1: 1r]r^rAs r3rDzTestLogser.test_rvs s||73vvdai   xx~(((zz)L"9999llt$#s###ll4 $$Q'#rzz***yy~~rpr)rdrrr)rmrNrzr_r`s r3rpzTestGumbel_r_l.test_fit sH :; ;XXa[ U S!3345DAr5N) rrrrfixturerrrr9gumbel_rgumbel_lrtrpr;r5r3rr sV^^*%+&+ [[Vennenn%EF [[Y 3 [[[*5 [[1#UmeT];=:=64G :  [[[ENNA+>,1NNB+?+ABBBBr5rc eZdZdZdZej ddZejjdejjddd gejjd d d gejjd dd gdZ ejjddd gejjd d d gejjd dd gejjde ddgdDcgc]}d|vr| c}}e jddZe jddZdZdZycc}}w) TestParetocJ tj5tjdttj j dd\}}}}t |tjt |tjt |tjt |tjtj j dd\}}}}t |tjt |tjt |tjt |tjtj j dd\}}}}t |dt |tjt |tjt |tjtj j dd\}}}}t |dt |tjt |tjt |tjtj j d d\}}}}t|d t|d t |tjt |tjtj j dd\}}}}t|dt|d t |tjt |tjtj j d d\}}}}t|dt|dt|dtjdzt |tjtj j dd\}}}}t|dt|dt|dtjdzt |tjtj j dd\}}}}t|dt|dt|dtjdzt|ddddy#1swYyxYw)NrmrvrrrNrrrzr}g?grq@rr2gffffff?gK~?g2@g۶m۶m?rUUUUUU?gqq?rr0g%I$I?gQ?gUUUUUU@grq?g88Nb@) rqrrrsrtr9paretorrdrrrrrmrrrHrVs r3rzTestPareto.test_stats sJ $ $ &7 N  ! !'> :++C+@JAq!Q BFF # BFF # BFF # BFF #++C+@JAq!Q BFF # BFF # BFF # BFF #++C+@JAq!Q C BFF # BFF # BFF #++C+@JAq!Q C BFF # BFF # BFF #++C+@JAq!Q Ay ) A2 3 BFF # BFF #++C+@JAq!Q As # At $ BFF # BFF #++C+@JAq!Q Ay ) A2 3 A 2777+;; < BFF #++C+@JAq!Q Ay ) Az * A0277;3GG H BFF #++C+@JAq!Q Ay ) A2 3 A RWWW-== > AL Mo7 N7 N7 Ns Q;RR"c~d}d}d}tjj||d|}||z |z}t||y)NgeArarrr^)r9rrr)rmrYr2r`rprRs r3rzTestPareto.test_sf$ sB   LLOOAqauO 5!Ga<8$r5rrc@tjjdSr/rrrs r3rzTestPareto.rng, rr5z2ignore:invalid value encountered in double_scalarsrr rarrr5rc tjjd||||}tjj|ddd}tjj|ddd}tjj|ddd}||cxk(r |cxk(rdk(sJJtjjd|||dz|}tjj|d \} } } t | dz|j |dz } | j d} t | | tjtj| | j z z t | dy) Nr[rr2r`r_rrp= ף?)rr)rfix_b)rfbrar) r9rrrrrr<rdrgr)rmrrr5rrshape_mle_analytical1shape_mle_analytical2shape_mle_analytical3 shape_mle_a loc_mle_a scale_mle_a data_shiftndatas r3rpzTestPareto.test_fit0 so ||SIY$+# ?!& 0 0A$ 0 G J % 0 0AT 0 J1 M % 0 0A$ 0 G J%)>.%.)-. /. /.||SIY%,q[ E.3ll.>.>t!.>.L+ Y [1_dhhj1AX   #[RVVBFF:jnn6F+F$GHH JY"r5rWfix_shape, fix_loc, fix_scaleTFrrepeatignoreinvalidctjjd||||}i} |r|| d<|r|| d<|r|| d<ttj|fi| y)Nr[rrrr)r9rrr rmrrr5rrr7rrrs r3rz&TestPareto.test_fit_MLE_comp_optimizerL sf||SIY$+# ? "DJ "DL &DN%ellDADAr5c d\}}}tjj|||dtjj d}t tj|y)N)r rr r[ið&r)r9rrrdrrr)rmr<locationr`rs r3test_fit_known_bad_seedz"TestPareto.test_fit_known_bad_seedb sR ")x||xS-/YY-B-B7-K M%ellD9r5cttjtttjj gddtttjj gdddy)Nrrar)rrarr rr;)rr9rrXr#rrrs r3test_fit_warningszTestPareto.test_fit_warningsl sCELL)lELL$4$4iaHlELL$4$4ia r5ctjjddd|}t|dtjj |}y)Ni~r r[)r_r2rrr)r9rrr r)rmrrrs r3test_negative_datazTestPareto.test_negative_datat s?||DACcJ$" LL  T "r5N)rrrrrrrrrfilterwarningsrrpr(rderrstaterrrr.0rps00r3rr s:Nx%V^^*%+&+ [[!12 [[[1a&1 [[YA/ [[[1a&1#2022 #. [[[2q'2 [[YA/ [[[1a&1 [[<)0$q)I-A!&! -.R[[" B#.203 BR[[":#: #I-s$ Frc eZdZdZdZdZdZdZdZdZ dZ d Z e jjd d gd fd dddej dz ej"gfddddej dzdgfdgdfgdZdZy) TestGenparetocbdD]c}tj|}tjj |\}}t |dt tj|etjd}tjj |\}}t||gddgy)NrNr{r{rv) rdrr9 genpareto _get_supportrr isposinfr)rmrr1r2s r3test_abzTestGenpareto.test_ab s $A 1 A??//2DAq B  BKKN #  $ JJsO++A.1AS *r5cLtjd}tjddd}t |j |tj j |t |j|tj j|t |j|tj j|tjddd}t |j|tj j|y)Nr{rrrr4rNrZ) r9rrdrSrrcr_rkrrrmrvrYrys r3test_c0zTestGenpareto.test_c0 s__r " KK3 #q 5;;??1#56q 5;;??1#56a%++.."34 KKB #q 5;;??1#56r5ctjd}tjddd}t |j |tj j |t |j|tj j|t |j|tj j|tjddd}t |j|tj j|t |jd dy) Nrrrrr4r{rNrZr ) r9rrdrSrrcuniformrkrrrrs r3test_cm1zTestGenpareto.test_cm1 s __s # KK3 #q 5==#4#4Q#78q 5==#4#4Q#78a%--"2"21"56 KKB #q 5==#4#4Q#78  ! a(r5ctjd}t|jtj |j tj gddgtt j|jtj tjd}t|jtj |j tj gddgtt j|jtj tjd}t|jtj |j tj gddgtt j|jtj y)NrWrr{rNr) r9rrrcrdrrkr isneginfrrmrs r3 test_x_infzTestGenpareto.test_x_inf s __s #82r(C BIIbff-./ __r "82r(C BIIbff-./ __s #82r(C BIIbff-./r5ctjddd}dD]}tjj ||}dD]3}tjj |||z}t ||d5tjj ||}dD]3}tjj |||z}t ||d5y) NrrZr4rrrg+rr)rr)rdrSr9rrcrrk)rmrYrpdf0dcpdfccdf0cdfcs r3test_c_continuityzTestGenpareto.test_c_continuity s KK2r " 8A??&&q!,D% 8**1a"f5d7 8??&&q!,D$ 8**1a"f5d7 8 8r5c ztjtjdddtjddddd tjdddz f}d D]Z}tj j ||}d D]3}tj j |||z}t||d 5\y NrrxrWbaser r4FendpointrN)r{rrr)rdrHrrSr9rrr)rmryrppf0rppfcs r3test_c_continuity_ppfz#TestGenpareto.test_c_continuity_ppf EE"++eT4++dArE:r{{5$S99: ; 8A??&&q!,D% 8**1a"f5d7 8 8r5c ztjtjdddtjddddd tjdddz f}d D]Z}tj j ||}d D]3}tj j |||z}t||d 5\yr)rdrHrrSr9rrr)rmryrisf0risfcs r3test_c_continuity_isfz#TestGenpareto.test_c_continuity_isf rr5c ^tjtjdddtjddddd tjdddz f}d D]L}t t j jt j j||||d Ny) NrrxrWrr r4FrrN)r&gC]r2r~gV瞯Ҽr~r) rdrHrrSrr9rrkr)rmryrs r3test_cdf_ppf_roundtripz$TestGenpareto.test_cdf_ppf_roundtrip s EE"++eT4++dArE:r{{5$S99: ;/ +A EOO//0C0CAq0I1ME + +r5c`tjjdddd}t|dy)Nrrxrr gpEȜ)r9rrrrbs r3rzTestGenpareto.test_logsf s'$$T31512r5zc, expected_statsr)r r rar*rfrgqq @rZragqq?g?g$I$ ?rq?rgll0@r)rvUUUUUU?r333333cdtjj|d}t||ddy)Nrrrr~r`)r9rr)rmrexpected_statsrs r3rzTestGenpareto.test_stats s+&&q&&9UGr5c^tjjd}t|ddy)Nr&gvǼ ?rr)r9rr8r)rmrs r3test_varzTestGenpareto.test_var s# OO   %,59r5N)rrrrrrrrrrrrrrrrdrrrrr;r5r3rr~ s + 7 ) 0 888+3  [[ \  T2gbggaj="&&1 2 ]T72771:$5v> ? " # %& H & H:r5rc<eZdZdZdZdZdZdZdZdZ dZ y ) TestPearson3cBtjjdyr/r0rrs r3r1zTestPearson3.setup_method r2r5ctjjdd}tj|dk(sJ|j j tdvsJtjjd}t|tsJtjdjd}t|tjsJ|j j tdvsJt|dk(sJy)NrWr6r5AllFloatrvr) r9r,rrdr<rr=rr>rr^r rAs r3rDzTestPearson3.test_rvs s~~!!#G!4xx~(((zz)J"7777nn  %#u%%%nnS!%%a(#rzz***yy~~:!66663x1}}r5ctjjdgd}t|t j gddtjjdd}t|t j dgdtjjgd d}t|t j gd dy) Nrar{rWrK)gtT?g+q?g?rprrrWND}i?rrrrr )rgj?gdC?gs?g'M?)r9r,rcrrdrrs r3rLzTestPearson3.test_pdf s~~!!!_5bhh'KL! #~~!!"c*bhh |44@~~!!"4c:bhh(@AGK Mr5cPtjjdgd}t|t j gddtjjdd}t|dgdtjjgd d}t|gd dy) Nrar)grE?gTp]0?g~?rprrrWgE&"J?r)giZ*J?gh+8w?gcK?g|rdnumber)rmrs r3test_return_array_bug_11746z(TestPearson3.test_return_array_bug_11746# sl&&q!,VQ&")),,,&&q(3VQ&")),,,r5cgd}d}tjjtjj|||}t ||t j dgdgg}t jdd}t tjj||tjj| | t tjj||tjj| | t tjj||tjj||  y)Nr rvrrrra) r9r,rrkrrdrrSrcrr)rmr r rr9rYs r3test_ppf_bug_17050zTestPearson3.test_ppf_bug_17050. s!nn  !3!3FE!BEJV$xx$#( KKA **1d3**A2u5 7**1d3))1"te4 6**1d3++Au55 7r5cgd}gd}gd}ttjj|||dttjj|dtjj|dy)N)rWrvrN皙)rOrrr)g7>g~׼=gp'+7g;rrr)rr9r,rr)rmr9rYrs r3rzTestPearson3.test_sfA s\% ">))!T2CeD))!Q/q1ANr5N) rrrr1rDrLrVrrrrr;r5r3rr s- MK ( -7& Or5rc6eZdZdZdZdZdZdZdZdZ y) TestKappa4cgd}d}dD]P}tjj|||}tjj|| }t ||Ry)N)r{rWrKrvrN) gffffffrr皙ɿrrWrKrvrNffffff?)r9kappa4rkrrrmrYrjrVrB vals_comps r3test_cdf_genparetozTestKappa4.test_cdf_genparetoN sW   -A<<##Aq!,D++Ar2I D) ,  -r5ctjddd}d}tjddd}tjj |||}tj j ||}t ||y)NrrrZr{rr)rdrSr9rrk genextremerrs r3test_cdf_genextremezTestKappa4.test_cdf_genextremeY se KKAr "  KKAr "||1a($$((A. i(r5ctjddd}d}d}tjj |||}tj j |}t ||y)NrrZrNr{)rdrSr9rrkr_rrs r3test_cdf_exponzTestKappa4.test_cdf_exponb sR KK2r "  ||1a(KKOOA& i(r5ctjddd}d}d}tjj |||}tj j |}t ||y)NrrrZr{)rdrSr9rrkrrrs r3test_cdf_gumbel_rzTestKappa4.test_cdf_gumbel_rk sT KKAr "  ||1a(NN&&q) i(r5ctjddd}d}d}tjj |||}tj j |}t ||y)NrrrZrr{)rdrSr9rrkrYrrs r3test_cdf_logisticzTestKappa4.test_cdf_logistict sT KKAr "  ||1a(NN&&q) i(r5ctjddd}d}d}tjj |||}tj j |}t ||y)NrrrZrN)rdrSr9rrkrrrs r3test_cdf_uniformzTestKappa4.test_cdf_uniform} sT KKAr "  ||1a(MM%%a( i(r5c0tjddyrg)r9rrrs r3test_integers_ctorzTestKappa4.test_integers_ctor s  Qr5N) rrrrr!r#r%r'r)r+r;r5r3rrM s% -)))))r5rc*eZdZdZdZdZdZdZy) TestPoissoncBtjjdyr/r0rrs r3r1zTestPoisson.setup_method r2r5ctjd}tjj gd|}d|dz |dzdz g}t ||y)Nrarr rarvr)rdrr9r~rJr)rmln2rBrRs r3test_pmf_basiczTestPoisson.test_pmf_basic sGffQi}}  C0QQq)h'r5ctjjgdd}gd}t||tjj dd}t |dy)Nr0r)r rrffffff?rr)r9r~rJrintervalr)rmrBrRr6s r3test_mu0zTestPoisson.test_mu0 sG}}  A.4*==))$2Xv&r5ctjjdd}tj|dk\sJtj |dk(sJ|j jtdvsJtjjd}t|tsJtjdjd}t|tjsJ|j jtdvsJy)Nrvr6r5rr8r) r9r~rrdrr<rr=rr>r]r^rAs r3rDzTestPoisson.test_rvs s}}  7 3vvdai   xx~(((zz)L"9999mm$#s###mmC $$Q'#rzz***yy~~rvrNra) rdrSrr8rr?rr9r@rrrAr)rmrYrorBrCrDrRrs r3rzTestKSTwo.test_sf s KK1b !) /A!SUAaCaAgq9:Aa%!B&&qs+B!#q266":aCxxc!"R#X!"5;;>>#q#9!9!$QrT3!45HkknnQ*G GX . /r5ctjddddd}gd}|D][}|tj|z }tjj ||}tj |}t|d]y)Nrrarr )r7r[rKirrr&)rdrSrr9rArkr r )rmrYns_xxnprobsdiffss r3test_cdf_sqrtnzTestKSTwo.test_cdf_sqrtn sn KK1b !!" % , +Bbggbk!BKKOOB+EGGENE eT *  +r5ctjddd}dD]Q}tjj ||}tjj ||}t |d|z SyNrr rr>)rdrSr9rArkrr)rmrYrorrs r3rzTestKSTwo.test_cdf_sf sY KK1b !) =A{{q!,HkknnQ*G %hG < =r5ctjddd}dD]i}|tj|z }tjj ||}tjj ||}t|d|z kyrN)rdrSrr9rArkrr)rmrYrorIrrs r3test_cdf_sf_sqrtnzTestKSTwo.test_cdf_sf_sqrtn sj KK1b !) =ARWWQZB{{r1-HkknnR+G %hG <  =r5ctjddd}dD]l}||d|z kD}tjj ||}d|k|dkz}tjj ||}t ||||dny) Nrr rr>rvrrr)rdrSr9rArkrrrmrYrorIrcondrBs r3test_ppf_of_cdfzTestKSTwo.test_ppf_of_cdf s KK1b !) =A1s1u9B{{r1-HLX_5D;;??8Q/D DJ4t <  =r5ctjddd}dD]l}||d|z kD}tjj ||}d|k|dkz}tjj ||}t ||||dny) Nrr rr>rvrNrr)rdrSr9rArrr)rmrYrorIvals_isfrSrBs r3test_isf_of_sfzTestKSTwo.test_isf_of_sf s KK1b !) =A1s1u9B{{r1-HLX^4D;;>>(A.D DJ4t <  =r5c<tjddd}dD]}|tj|z |d|z kD}tjj ||}d|k|dkz}tjj ||}t||||y)Nrr rr>rvrN)rdrSrr9rArkrrrRs r3test_ppf_of_cdf_sqrtnzTestKSTwo.test_ppf_of_cdf_sqrtn s KK1b !) 2Abggaj.!c!e),B{{r1-HLX^4D;;??8Q/D DJ4 1  2r5c<tjddd}dD]}|tj|z |d|z kD}tjj ||}d|k|dkz}tjj ||}t||||y)Nrr rr>rvr4)rdrSrr9rArrr)rmrYrorIrrSrBs r3test_isf_of_sf_sqrtnzTestKSTwo.test_isf_of_sf_sqrtn s KK1b !) 2Abggaj.!c!e),BkknnR+GKGdN3D;;??7A.D DJ4 1  2r5ctjddddd}dD]N}tjj ||}tjj ||}t ||PyrN)rdrSr9rArrkr)rmrJrorIrs r3rzTestKSTwo.test_ppf s\ Aq"%ab)) -A*B{{r1-H He , -r5c gd}tjgd}tjgdgdgdgdgdgdg}t|D]x\}}|tjd ztjtj d z |z z}t jj||}t|||d zy) N)rZr7r[rKrr)rfgUUUUUU?rvr rar)gRT>gr ?gK5 ?gn "5?gU ?gk?)g&6#>gb E>gI5?g\ ?g;?g%1?)g6>gL>g<?g䓀^?gO|6$?g qO?)g}+:>gwQ9>gH{?g]qv (?g)~/v?gmJ?)gV27J=gGk(>gWX?g Y?g{0`?gZ H?)g_R=g4>g̪?gT_@?g<e?g2?rar4r) rdr enumeraterrrer9rArkr)rmrGratiosrRidxrorYrs r3test_simard_lecuyer_table1z$TestKSTwo.test_simard_lecuyer_table1& s+8988 - - - - - -    m @FC"RWWRUU1WQY%77A{{q!,H Hhsm$ ? @r5N)rrrr1rVrrLrrPrTrWrYr[rrar;r5r3r;r; s?0( / +====22-@r5r;ceZdZdZdZdZy)TestZipfcBtjjdyr/r0rrs r3r1zTestZipf.setup_methodI r2r5ctjjdd}tj|dk\sJtj |dk(sJ|j jtdvsJtjjd}t|tsJtjdjd}t|tjsJ|j jtdvsJy)Nrr6r5r r8r) r9zipfrrdrr<rr=rr>r]r^rAs r3rDzTestZipf.test_rvsL szz~~c~0vvdai   xx~(((zz)L"9999jjnnS!#s###jjo!!!$#rzz***yy~~rrrrs r3r$zTestZipf.test_momentsW syzz#&1 AQzz#t41BKKA'++--.r5N)rrrr1rDr$r;r5r3rcrcH s 9/r5rcc$eZdZdZdZdZdZy) TestDLaplacecBtjjdyr/r0rrs r3r1zTestDLaplace.setup_methodb r2r5c"tjjdd}tj|dk(sJ|j j tdvsJtjjd}t|tsJtjdjd}t|tjsJ|j j tdvsJtjjdJy)Nrr6r5r8rr) r9dlaplacerrdr<rr=rr>r]r^rAs r3rDzTestDLaplace.test_rvse s~~!!#G!4xx~(((zz)L"9999nn  %#s###nnS!%%a(#rzz***yy~~@g @gGz@c*tjjd}tjj d||||}t ttjtj"}||}tjj|}t||||dd}tjj|dd}t||ttj||tj||dz z dkDsJttj||dz ttj|dttj||tjjdd y) Nrr[)rrXr_r`rrra)rr}rr r;) rdrrr9rrrrrrrrrand) rmrrr5rr super_fit super_fitted invgauss_fit __class__s r3rz(TestInvgauss.test_fit_MLE_comp_optimizer s9ii##D)~~!!sv&-YS"R$u~~.C   ~~))$/ \<0!A15 ~~))$QA)> \<0 &enndIvvtw{+q0111%ennd1M &enndC &ennd-/YY^^A->q-A Cr5cttjtjt 5tjj gdddddy#1swYyxYwr@)rr9rrrr#rrrs r3test_fit_raise_errorsz"TestInvgauss.test_fit_raise_errors sFENN+ ]]< ( 2 NN  yq  1 2 2 2s $A  A)cgd}gd}tjjd|}t||tjjdd}t |dtjj dd}t |d tjjd d }t |d tjjd d }t |dy)N)g4wT;?gjdV&}?g{i\0>gDV_h?gqac X?)r r r r r 皙?rXr?gfFngr^:g9̗?rgx_;:g!?gٷ ??)r9rrkrrr)rmrXrRrM cdf_actual sf_actuals r3rzTestInvgauss.test_cdf_sf s."##CB#/Xv&^^''$'7  $9:NN%%cd%3  #78 ##GV4 34##Hf5 12r5chtjjdd}t|dtjjdd}t|dtjj dd}t|dtjj dd}t|d y) NrrrgiJ.rgr^ߺrgpgFgpQ^L)r9rrrr)rmrrs r3rzTestInvgauss.test_logcdf_logsf s&&v$&7 12&&sD1 56$$Ut$456$$S$/01r5zmu, ref))rg,9)rg% !)rxg40)rgK' @)rg`p @cZttjj||dyrK)rr9rrA)rmrXrs r3rkzTestInvgauss.test_entropy s ..r2CeDr5)rrrr1rrrrprrrrrk __classcell__rs@r3rzrz s [[7')>?A>A>: [[7')>?A CA CD2 !3F2. [[Y)DE E E Er5rzcZeZdZejj dgddZdZy) TestLandauname)rcrkrrrc|dvrtjdddf}ntjdddf}ttj|}ttj |}||}|g|dd}t ||d y) N>rrrWrrrrZr rr)rdrSr(r9landau levy_stabler)rmrrY landau_method levy_methodrrs r3test_landau_levy_agreementz%TestLandau.test_landau_levy_agreement s > ! Ca()A B2&'A d3 e//6 Q#1#a##Su-r5cttjjdtjfdzttjj dtjy)Nrrrr)rr9rrdrrrrs r3r$zTestLandau.test_moments& sEU\\'''7"&&1EU\\((+RVV4r5N)rrrrrrrr$r;r5r3rr s, [[V%GH .I .5r5rceZdZejj dgdejj dgddZejj dgdfdZd Zd Z d Z d Z d Z xZ S) TestLaplacer)rrr rar5)r rarrZctjjd}tjj d|||}tj |}tjtj||z t|z }tjj|\}}t||ddt||ddtjj||\}}t||ddtjj||\}}t|||dz}tjtj||z t|z }tjj||\}}t||tjj||\}}t||tttjj||| tttjjtj gtttjjtj"gy) Nrr[rr~rhrrrrar;)rdrrr9rrr?rgrr rrrrXrrrr) rmrr5rrloc_mlerr_r`s r3rpzTestLaplace.test_fit- s ii##D)}}  cwi.1!3))D/FF266$.12SY> ]]&&t, UW5u=yu5A]]&&t'&: Uyu5A]]&&tI&> UW%kFF266$*-.T: ]]&&t#&6 UY&]]&&tI&> UWc" lEMM$5$5t'& ( j%--"3"3bffX>j%--"3"3bffX>r5rvs_loc,rvs_scale))rrZrZr)rvrKctjjd}tjj d|||}d}tjj |\}}tttjtj|\}} ||||} ||| |} | | kstj| | ddsJyy)Nrrrc dt| tjd|zzd|z tjtj||z zz zS)Nrrar )r rdrrgr)r_r`rs r3llz3TestLaplace.test_fit_MLE_comp_optimizer..lle sOCI%8E'266"&&*<#==>? ?r5r~rh) rdrrr9rrrrrr) rmrr5rrrr_r`loc_opt scale_optll_mlell_optrs r3rz'TestLaplace.test_fit_MLE_comp_optimizer\ sii##D)}}  dy.1!3 ? ]]&&t, U"4 #6#(==66:) C%GY-"++ff38u#F FF#Fr5ctjgd}tjj |d\}}t |dddtjj |d\}}t |dddy)N)rNrNrrOrrr*rrr~rhrr)rdrr9rrr)rmrr_r`s r3test_fit_simple_non_random_dataz+TestLaplace.test_fit_simple_non_random_datas sgxx78]]&&t!&4 Uqu59]]&&tA&6 UQU7r5c<d}tjj| }|dk(sJtjj|}|dk(sJtjj|}|dk(sJtjj| }|dk(sJy)Nrr{rN)r9rrkr)rmrYp0p1s r3test_sf_cdf_extremesz TestLaplace.test_sf_cdf_extremes| s  ]]  r "Syy]]  q !Syy ]]  a Syy]]  qb !Syyr5cd}tjj|}t|t j | dz dy)NrKrarr)r9rrrrdr)rmrYrps r3rzTestLaplace.test_sf s5  MM  Q 2661":a)rmrmomrs r3test_moments_statsz!TestLogLaplace.test_moments_stats s~QUm IIc1s7C ( ((//15rvv>66"++e&6&6&<&s r3test_fit_analytic_mlez$TestLogLaplace.test_fit_analytic_mle sii##$78##A3e#14$6} DJ "DNCI  A:MI j9K 3  $$T2T2 3 %e&6&6EE  3 s !CCN) rrrrrrrrrrr;r5r3rr sDE [[S,/O0O [[S/2 [[\L>: [[WtUm4 [[[4-8F95;3Fr5rc eZdZejj dddgdZejddZejj dgd ejj d gd ejj d gd ejj de ddgdDcgc]}d|vr| c}}dZ dZ dZ dZ ycc}}w) TestPowerlawzx, a, sf)rfrzg?)g?gp?gSQ>c\ttjj|||dyr})rr9powerlawr)rmrYr1rs r3rzTestPowerlaw.test_sf s! ))!Q/%@r5rrc@tjjdSr/rrrs r3rzTestPowerlaw.rng rr5r)rWrvrr rarrr5rrTFrrc*tjjd||||}t} |r|| d<|r6t j |j tj | d<|r|| d<ttj|fi| ddiy)Nr)rr1r_r`rrrrrT) r9rrr rd nextafterrrrrs r3rz(TestPowerlaw.test_fit_MLE_comp_optimizer s~~!!siW(1"Ev "DJ << RVVG [[Y 3 [[[*5 [[<)0$q)I-A!&! -.<.64? <$ D466{-s> C/rc,eZdZejj dgddZejj dddgdZejj dgdd Zejj dgd d Z y ) TestPowerLogNormz x, c, s, ref))r[rr ga!R()rrr g;?)rrr gE?)MDrr g=c^ttjj||||dyr)rr9r$rrmrYrrHrs r3rzTestPowerLogNorm.test_sfGs% **--aA6%Hr5z q, c, s, ref)g?rr rx)gR4 -rr rc^ttjj||||dy)Ng|=r)rr9r$r)rmryrrHrs r3rzTestPowerLogNorm.test_isfQs% **..q!Q75Ir5))rrr g?)rprr gv6)rprKr g+ 94|7)rxrKr g?c^ttjj||||dy)Ngt =r)rr9r$rkrs r3rVzTestPowerLogNorm.test_cdfWs% **..q!Q75Ir5))rrr gS^Ռe 9)@xDrr g 0g;)g\)c=Hrr g7H7c^ttjj||||dy)NgAfc=r)rr9r$rcrs r3rLzTestPowerLogNorm.test_pdfjs% **..q!Q75Ir5N) rrrrrrrrrVrLr;r5r3rr6s" [[^EF I F I  [[^ACEFJFJ [[^AB J B J [[^GHJ HJr5rceZdZejj dgddZejj dgddZejj dgddZy ) TestPowerNormr%))rr g|<)rrage()r[rg%!@6)rKrxgT;-c\ttjj|||dyr)rr9 powernormrr)s r3rzTestPowerNorm.test_sfzs! **1a0#EBr5r))r4rgĿ)gwJ?r[g!X)rrg缂U)r@rgͫ1@)r1ragD@)Jz5rgU!zYr@c\ttjj|||dyNr{r)rr9rrrs r3rzTestPowerNorm.test_isfs! ++Aq13UCr5))irg.u T9)rar?)rrgsK-)rrgk8>)rrg^u1-c\ttjj|||dyrK)rr9rrkr)s r3rVzTestPowerNorm.test_cdfs! ++Aq13UCr5N) rrrrrrrrrVr;r5r3rrrs [[[CD C D C [[[=>D>D [[[BC D C Dr5rcreZdZdZdZdZdZdZejjdddgd Z y ) TestInvGammac tj5tjdttj j dd}gd}t ||gd}tj j |d}gdtjdd gtjd d gtjtjd gf}t||D]\}}t|| dddy#1swYyxYw) Nrmg(\O3@rrk)g~?gCX&?g*7gS?gBqFq@)r@gffffff@)rg Ey?gў3oz?g^Yb?g,r?gOD@g2*Z@gFN8@) rqrrrsrtr9rrrdrrrr)rmrrRr1rYr)s r3test_invgamma_inf_gh_1866z&TestInvGamma.test_invgamma_inf_gh_1866s $ $ & *  ! !'> :>>''%'@D%H D( +A>>''!V'>!5UI9>>!#w/>>s (B5 BcDtjjddd\}}}}ttj|ttj|ttj|ttj| y)Nr@rr)r9rr rdr>rs r3test_f_momentszTestF.test_f_momentsseWW]]2sF]; 1a A A ABKKN"#r5ctj5tjdttj j dgdzgdddddy#1swYyxYw)Nrmrr)rarr*rrdfndfdr)rqrrrsrtr9rrrs r3test_moments_warningszTestF.test_moments_warningssS  $ $ & H  ! !'> : GGMMrd1f,M G H H Hs AA!!A*ctjdgdgg}tjddg}tjj||d\}}}}||dz z gdz}t ||d|dzz||zdz z|z |dz dzz |dz z }t ||d|z|zdz tj d|dz zz|d z tj |||zdz zzz } t || d|d |zd z z||zdz z|dz |dz dzzzz} ||d z z|dz z||zdz z} | | z } t || y) NrrrPrr rarrr*r)rdrr9rrr) rmr r rrrHrVrtv2s2k2numk2denk2s r3test_stats_broadcastzTestF.test_stats_broadcastsuhhbT{#hhBx WW]]sV]D 1aS1Wo q 2 aZ39q= )C /37Q, >#' J2us{Q"''!S1W+"66a2773c A #67792cQsURZ(C#IM:Qw37Q,./0sQw37+sSy1}= U]2r5N)rrrrr rrr;r5r3rrs0$H r5rc eZdZdZdZdZejjdddgdZ ejjdgd ejjd d d gd d gd d ggd gdgd d gd d ggd gd gggdZ ejjd d d gd d gd d ggd gdgd d gd d ggd gd gggdZ dZ ejjdgddZ y) TestStudentTc^ttjjddgddgy)Nrr*g?g?)rr9rrurrs r3test_rvgeneric_stdzTestStudentT.test_rvgeneric_stds"!%''++q!f"5 J7OPr5cttjjddtjtj tj tj fttjjdddtjtj tj fttjjdddtjtj tj fttjjddddtj tjfttjjd d tj tjfttjjd d dtjfttjjd d dtjfttjjd d dy)Nr r)rr)\(?r{ragGz@g i@rrjgGz@rg ףp= @)r{gq@)rr9rrdrrrrs r3test_moments_tzTestStudentT.test_moments_tsXUWW]]a]8VVRVVRVVRVV4 6UWW]]dF];"&&"&&"&&1 3UWW]]a]8"&&"&&"&&1 3UWW]]dF];/266266: <UWW]]a]68HIUWW]]dD]9C=IUWW]]a]6bff FUWW]]dD]9;RSr5cjgd}gd}ttjj||dy)N)r rarr[)g?@gTM]?ggY?U?rrrr9rrA)rmrrRs r3test_t_entropyzTestStudentT.test_t_entropy+s' ;+XEBr5zv, ref)r[r)rrcZttjj||dyrr )rmrrs r3test_t_extreme_entropyz#TestStudentT.test_t_extreme_entropy2s *Ce?S*%z!ZK.'ABr5cTtjjdtj|t}tjj dd|j }tj||<tjj|ddd}tjjddd }tjj||ddd}td D]-}t|||||t|||||/tjj|dd }tjjdd }tjj||dd }t|||t|||y) Nrr'rZr5rr r)rr_r`rr_r`rrr(r^)rdrrrr)rr<rr9rrrrrA)rmr%rr res_ex_inf res_ex_noinfrs r3test_t_inf_df_stats_entropyz(TestStudentT.test_t_inf_df_stats_entropyYs` qZZ $7 YY  q":+;+;  <:ggmmrq6mBZZ%%!1f%E ww}}J;Qa,2%4 q ?A Q +Z] ; Q ,l1o > ?ggoo!o4ZZ''AQ'7 ww"j[/qJ S_j1S*%|4r5cgd}gd}gd}gd}ttjj|||dttjj |||dy)N)r rrZr )rd~QJrr )gZ_2g 9g!ǟuIgdg)g73?rgմFhAW;gm0_?rr)rr9rrrc)rmrYr logpdf_refpdf_refs r3test_logpdf_pdfzTestStudentT.test_logpdf_pdfpsS  #@ >q"-zF Ar*G%@r5 x, df, ref))gRr9guSaG)rr9r)R@r9gGn`?ctjj||}t||dtjj | |}t||dyr)r9rrrr)rmrYrrrrs r3rzTestStudentT.test_logcdf_logsf~sF 2&%0 qb"%s/r5N)rrrrrr!rrrr#r1r6r;rr;r5r3rrsUQ TC [[X68:; =; = [[Z*<= [[\QFQFQF-6 ,B./VaV,<./S1#J,89C 9= C  [[\QFQFQF-6 ,B./VaV,<./S1#J,895 95& A [[\CD0 D0r5rcTeZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zy )TestRvDiscretecBtjjdyr/r0rrs r3r1zTestRvDiscrete.setup_methodr2r5cgd}gd}d}tjd||f}|j|}t|tj sJt ||D]0\}}tt||k(t|z |z dkr0J|j}t jt|tjsJy)N)rrr rarr)r{rxrr{rxr{rsample)rvaluesr5r@) r9r>rr>rdr^rrrgrrrinteger)rmstates probabilitysamplesrrYrHrps r3rDzTestRvDiscrete.test_rvss$4    8V[4I J EEwE !RZZ((( , >DAqs16{5>1A56= == > EEG}}T!Wbjj111r5c:tjgd}tjgd|f}t t || }|j }t||tjgdgdf}|j }t|dy)N)rfr7rxr0rD)rNrrr{) rdrr9r>rgrrArr)rmrrprirjs r3rkzTestRvDiscrete.test_entropysy*+   i%7 8%u-.. IIK:&   i%= > IIKQr5cgd}gd}tj||f}ddgddgg}t|j|dd gd d ggd y)NrrvrxrKrJrNrrrarvrKr{rxrr)r9r>rrJ)rmxkpkrrYs r3rdzTestRvDiscrete.test_pmfs]     r2h /"X!W q sc$*/ 1r5cgd}gd}tj||f}gd}gd}t|j||dt|Dcgc]}|j|c}|dycc}w)NrrLrJ)rrNrrrzrrr)rrvrvrvrrr r rr)r9r>rrk)rmrMrNrx_valuesrRrs r3rVzTestRvDiscrete.test_cdfsf     r2h /55x((? h77 u .7A2cgd}gd}tj||f}gd}gd}t|j||dt|Dcgc]}|j|c}|dycc}w)NrrLrJ)rWrvr}rrrN)r r rararrrr)r9r>rr)rmrMrNrq_valuesrRrys r3rzTestRvDiscrete.test_ppfsf     r2h /0%x((? H5q5 u .5rQc gdgdf}tj|}t|j|j |j dddz|j ddy)N)r rarrr)rWrKrxrxrWrJrr&r )r9r>rrrkrM)rmrBrs r3test_cdf_ppf_nextz TestRvDiscrete.test_cdf_ppf_nextsU!:;   d +266"&&s"4t";<559 &r5cDtjdjd}tjgdgdgdg}t j ||f}t |jtj|j|jzdy)NrP)rr)rWrWrVr@)rWrWr@r@rJrr) rdrr6rr9r>rrrgrMrNrmrMrNrs r3test_multidimensionz"TestRvDiscrete.test_multidimensionss YYr] " "6 * XX---/0  r2h / RVVBEEBEEM%:Gr5cgd}ddg}tttjfit ||fgd}tttjfit ||fgd}gd}tttjfit ||fgd}gd}tttjfit ||fddg}ddg}tttjfit ||fy)NrrvrJ)rv333333?gffffff濩r rarrr)rxrxrxrxrr )rXrr9r>r rmrMrNs r3test_bad_inputzTestRvDiscrete.test_bad_inputs 3Zj%"3"3MtB87LM j%"3"3MtB87LM  j%"3"3MtB87LM  'j%"3"3MtB87LMV3Zj%"3"3MtB87LMr5cntjdjdtjdd}}t t t jfit||ftjdjdtjdd}}t t t jfit||ftjdjdtjdd}}tt j||fjddy) NrrararargUUUUUU?rJr*rrar) rdrr6rrXrr9r>r rrJr\s r3test_shape_rv_samplez#TestRvDiscrete.test_shape_rv_samples1%%f-rwwvs/CBj%"3"3MtB87LM1%%f-rwwvs/CBj%"3"3MtB87LM1%%f-rwwvs/CBU&&r2h7;;A>Dr5cgd}gd}tj||f}t|jt j |j |jzdy)N)r rarr*rr)rWrKrKrKrKrWrJrr)r9r>rrrdrgrMrNrWs r3 test_expect1zTestRvDiscrete.test_expect1sG  +   r2h / RVVBEEBEEM%:Gr5c tgd}gd}tj||f}t|j|j dt|jt dt ||Ddt|jdt dt ||Ddy) N)/gi@gr@gy@g@@g@g@@g @rg0@g@gP@g@gp@g@g@g @g@rgh@g0@g@g@g@gP@g@g@g@rg8@g@gȩ@g@gX@g @g@g@gx@g@@g@gh@g̰@g0@g@g@g\@g@)/g-C6:?r{gF%uk?g:vz?r{r{g.nr?g|гY?r{r{r{g@6*?4U0*C?g@ +?gC6z?gnr?g ?gOjM?g n?g1w-!?g:pΈ?gףp= ?g_Le?r{g^)p?g~jtx?gJ4q?gGzt?gJ4q?gMb@?g@Psׂ?g@^)ˀ?r{r{gN@s?g1w-!_?rggൄ|г?r{gH}}?r{g F%u?r{r{g^)ˀ?gv?r{rJrrc3,K|] \}}||zywrpr;rrws r3 z.TestRvDiscrete.test_expect2..'s9daAE9sc |dzSNrar;rs r3rIz-TestRvDiscrete.test_expect2..* AqDr5c32K|]\}}|dz|zyw)raNr;ris r3rkz.TestRvDiscrete.test_expect2..+s<AAqD1Hrrrrgr)rmr)pyrs r3 test_expect2zTestRvDiscrete.test_expect2s =?   q"g .  RWWYU; 9c!Rj99 G  .1<Q<<5 Jr5N)rrrr1rDrkrdrVrrUrXr]rbrdrqr;r5r3r@r@sC 2  1 . .&HN( EH#Jr5r@ceZdZdZdZy)TestSkewCauchyctjddd}ttjj |dtj j |ttjj|dtj j|ttjj|dtj j|yNrrr[rri) rdrSrr9 skewcauchyrcrrkrrs r3 test_cauchyzTestSkewCauchy.test_cauchy/s KKAs #!%"2"2"6"6qA"6">"',,"2"21"5 7!%"2"2"6"6qA"6">"',,"2"21"5 7!%"2"2"6"6qA"6">"',,"2"21"5 7r5ctjjdtjjddzdz }tjjddzdz }gd}gd}t t j j|||t t j j|||t t j j|||y)NrrZrar r) g&45?gQʶ?g-4[u?g a2?g Rn?gѦ}cي?) gN@Q?gPPz@ ?g?g; 7?glg?g!?%?gQҙ?gC"?go?g@Sg?) rdrrrrr9rvrcrkr)rmr1rYrcrks r3test_skewcauchy_Rz TestSkewCauchy.test_skewcauchy_R8s& q IINN2  "Q & IINN2  #a '%$ ((,,Q2C8((,,Q2C8((,,S!4a8r5N)rrrrwryr;r5r3rsrs.s 7 9r5rsc~eZdZdZej dZejjdgddZ y) TestJFSkewTcdx}}|dz}gd}gd}tj||}tj|}t|j ||j |t|j ||j |t|j ||j |t|jd|jdy)Nrra)rr{rNrz)r{rWrfrrrNr)r9 jf_skew_trrrcrkr)rmr1r2rrYryjfrs r3test_compare_tzTestJFSkewT.test_compare_t\s A U ! - __Q " GGBKq 1558,q 1558,q 1558,(!''&/:r5ctjttjdz }tj j |dS)aSample data points computed using the `ST5` distribution from the GAMLSS package in R. The pdf has been calculated for (a,b)=(2,3), (a,b)=(8,4), and (a,b)=(12,13) for x in `np.linspace(-10, 10, 41)`. N.B. the `ST5` distribution in R uses an alternative parameterization in terms of nu and tau, where: - nu = (a - b) / (a * b * (a + b)) ** 0.5 - tau = 2 / (a + b) z"data/jf_skew_t_gamlss_pdf_data.npyz x,pdf,a,bnames)rdloadr__file__parentrec fromarraysrss r3gamlss_pdf_datazTestJFSkewT.gamlss_pdf_datalsAww N ! !$H H vv  [ 99r5za,b)r`)rr)rP c||d|k(|d|k(z}|d|d}}t|tj||j|dy)zCompare the pdf with a table of reference values. The table of reference values was produced using R, where the Jones and Faddy skew t distribution is available in the GAMLSS package as `ST5`. r1r2rYrcrrN)rr9r}rc)rmrr1r2rrYrcs r3test_compare_with_gamlss_rz&TestJFSkewT.test_compare_with_gamlss_r|s_  S !Q &?3+?1+D E cDK3U__Q266q9Fr5N) rrrrrrrrrrr;r5r3r{r{[sH;  ^^ : : [[U$>? G@ Gr5r{rarg@rgףp= 1@r9gRZ@igS㥛Ԋ@rWrXg&@iQi'i0ybilHO1Zi3rirZir[ikrKi1rii4i Ni(lEI.r&iceZdZdZdZdZdZdZdZdZ e jjde d Zd Zd Zd Zy ) TestSkewNormc$td|_yr/)rrrrs r3r1zTestSkewNorm.setup_methods%d+r5ctjddd}ttjj |dtj j |yru)rdrSrr9skewnormrcrrs r3 test_normalzTestSkewNorm.test_normalsB KKAs #!%.."4"4Q!"4"<"'**.."3 5r5cd}tjjd||j}t ||j tjjd||j}t ||j y)N)rrrr)r1rrr)r9rrrrr<)rmr<rYs r3rDzTestSkewNorm.test_rvssa NN  E  IUAGG$ NN  %dhh  GUAGG$r5ctjjdtddd|j}t j |t j|tj|tj|g}tjjdddd}t||dtjjd tddd|j}t j |t j|tj|tj|g}tjjd ddd}t||dy) Nrrrra)r1rr_r`rr)r1r_r`rrQr) r9rrr]rrdrr8r9kurtosisr)rmrrRr;s r3r$zTestSkewNorm.test_momentss  NN  Sq,0HH  6GGAJq 5::a=%..:KL>>''!!V'L!(Ha@ NN  #c(,0HH  6GGAJq 5::a=%..:KL>>''"!1f'M!(Ha@r5cgdgdgdgdgdgdgdg}|D]4\}}}tjj||}t||d 6y) N)rwrgEAV)rwrgV|@$a)rwrg0rY)rwrvZk)rrr)rrļ)ragcrr&r)r9rrr)rm logpdfvalsrYr1 logpdfvalrs r3test_pdf_large_xzTestSkewNorm.test_pdf_large_xsY 2 3 0 2 4 5 5  * 8OAq)>>((A.D D)$ 7 8r5ctjjgdd}t|t j ddtjjdd}t|ddy) N)rZrr4rrrrrr}rN)r9rrkrrdonesrs r3test_cdf_large_xzTestSkewNorm.test_cdf_large_xsP NN  |R 02771:E2 NN  r3 '3U+r5cgdgdgdgdgdg}|D]d\}}}tjj||}t||dtjj | | }t||dfy)N)rr gzbLe9)rragn'/2;)rrg::)rrg|<)rgrr-r&r)r9rrkrr)rmcdfvalsrYr1cdfvalrps r3test_cdf_sf_small_valuesz%TestSkewNorm.test_cdf_sf_small_valuessy . + , . /  $ 2LAq&""1a(A AvD 1!!1"qb)A AvD 1  2r5z a, momentsct|dD]3\}}tjj||}t ||d5y)Nr )startrr)r^r9rrr)rmr1rorderrRrs r3test_noncentral_momentsz$TestSkewNorm.test_noncentral_momentss>(: 7OE8..''q1C C 6 7r5c Xtjjd}d\}}}tj|||}|j d|}tjj |dd\}}} tjj |dd\} } } || cxk(rdk(sJJ|| k7sJtjj |ddd \} }}|dk(sJtj| ||}|jd }tj|tj|f}t||tjj d d|}tjj |}tjtj|sJtjj |d \}}}tj|sJtj|tj|}}t|||tjdtj z zzt||dzd dtj z z ztjj |d\}}}tjj | d\}}}t|||g| | |gtjj |d \}} }!tjj | d \}"}#}$t|"|#|$g| | |!gy)Nl# ke)rr2rvr[rrRrrgmmrrmsrr rramle)rdrrr9rrrrr9rrrr>isinfr8rre)%rmrr1r_r`rNra2loc2scale2a3loc3scale3a4loc4scale4dist4rrra5loc5scale5rra6ploc6pscale6pa6mloc6mscale6ma7ploc7pscale7pa7mloc7mscale7ms% r3rpzTestSkewNorm.test_fitsii##$78$ 3~~ae,hhCch2!>>--c4a-@D& >>--c4a-@D&t q     Rxx!>>--c1Qt-LD&{{r40kk$k'ggclEJJsO+S!||#>nn  &vvbkk#&'''!>>--d4-@D&xx||wwt}bffTl14&2771RUU7+;";;<619AI 67$nn00U0CUG#nn00#e0DUGeW-ufg/FG#nn00T0BUG#nn00#d0CUGeW-ufg/FGr5cX tjgdddgzzdgz}tjj |}tjj ||}tjj |d}tjj ||}||dz ksJtj jd d d d d } fd }tjtj|||}tjj||jdy)N)rrr4rPr rT)superfitrvly ND)rr)rgrZ)r}rZr1r_r`c t||S)N)r)r')funrrs r3 optimizerz0TestSkewNorm.test_fit_gh19332..optimizerHs)#v3? ?r5)rrr) rdrr9rrrrrtestingrparams) rmrYrr params_superrrr fit_resultrs @r3test_fit_gh19332zTestSkewNorm.test_fit_gh193323s HH*R1#X5; <##A&nn!!&!, ~~))!d); nn!!,2S3Yii##$78y;G @YYu~~q&IN  ""6:+<+<4"Hr5c\ttjjddddy)Nrxrgi}#%?rr)rr9rrrrs r3rzTestSkewNorm.test_ppfNs" **457HuUr5N)rrrr1rrDr$rrrrrr_skewnorm_noncentral_momentsrrprrr;r5r3rrs`,5 % A8 ,2" [[\+GH7I7 .H`I6Vr5rc$eZdZdZdZdZdZy) TestExponcVttjjddyr)rr9r_rcrrs r3 test_zerozTestExpon.test_zeroVsU[[__Q'+r5cttjjddttjj tjj ddy)Nr1rw)rr9r_rkrrrrs r3 test_tailzTestExpon.test_tailYs=U[[__U+U3U[[__U[[^^B%78"=r5ctjdddddtjg}ttt j j|yNo_?46@yX5ͻ@j+?UX @)rdrrrXrr9r_rrs r3test_nan_raises_errorzTestExpon.test_nan_raises_error]5 HHfffffbffE Fj%++//15r5ctjdddddtjg}ttt j j|yr)rdrrrXrr9r_rrs r3test_inf_raises_errorzTestExpon.test_inf_raises_errorbrr5N)rrrrrrrr;r5r3rrUs,>6 6r5rcfeZdZdZdZdZejjdddgdZ y) TestNormctjdddddtjg}ttt j j|yr)rdrrrXrr9rrrs r3rzTestNorm.test_nan_raises_errori5 HHfffffbffE Fj%**..!4r5ctjdddddtjg}ttt j j|yr)rdrrrXrr9rrrs r3rzTestNorm.test_inf_raises_errornrr5cbgd}tttjj|dy)Nrshrimp)plate)rXrr9rrrs r3test_bad_keyword_argzTestNorm.test_bad_keyword_argss i(Cr5r_rr cd}tjjd|zd|z|}t||dtjjd|z d|z | }t||dy)Ngg {F.:rrPr_rr)r9r _delta_cdfr)rmr_rRrls r3test_delta_cdfzTestNorm.test_delta_cdfwsm) %%bfbf#%>xe4 %%3i2c6%Exe4r5N) rrrrrrrrrrr;r5r3rrhs;5 5 D [[UQF+ 5, 5r5rceZdZdZdZdZy) TestUniformgh-10300ctjdddddtjg}ttt j j|yr)rdrrrXrr9rrrs r3rz!TestUniform.test_nan_raises_error7 HHfffffbffE Fj%--"3"3Q7r5ctjdddddtjg}ttt j j|yr)rdrrrXrr9rrrs r3rz!TestUniform.test_inf_raises_errorrr5N)rrr__doc__rrr;r5r3rrs8 8r5rc 6eZdZdZdZdZdZejjdgddZ ejjdgd gd gd gd gd gdZ ejjdgdgdgdgdgdgdgdZ y) TestExponNormc2d}d\}}}d||zz }tjj|||d}t|||||d\}}}d||zz }tjj|||d}t|||||d\}}}d||zz }tjj|||d}t|||||d\}}}d||zz }tjj|||d}t|||||y) Ncdd||zdzz z}d||zdzz |dzz}dd||zdzzdzz}|d|z z||zd||zz z||gS)NrNr rarrR@rr;)lamsigrXopK2exp_skewexp_kurts r3get_momsz,TestExponNorm.test_moments..get_momsszc#g\))DC#I>)D4L8Ha39q.0B77H3JC#s3w- 78L Lr5)rr r rNrr3)rrarW)rrr )rrr2)r9 exponnormr)rmrrXrrKstss r3r$zTestExponNorm.test_momentss. M C 39 oo##A2S&#IC#sB!78! C 39 oo##A2S&#IC#sB!78 C 39 oo##A2S&#IC#sB!78" C 39 oo##A2S&#IC#sB!78r5ctjdddddtjg}ttt j j|ddy Nrrrrrrr r;)rdrrrXrr9rrrs r3rz#TestExponNorm.test_nan_raises_error; HHfffffbffE Fj%//"5"5qqKr5ctjdddddtjg}ttt j j|ddyr)rdrrrXrr9rrrs r3rz#TestExponNorm.test_inf_raises_errorr r5cTttjjdddttjjdddttjjdddttjjdddy)Ni|r r{rx)rr9rrcrrs r3test_extremes_xzTestExponNorm.test_extremes_xsnEOO//a8#>EOO//a8#>EOO//d;SAEOO//d;SAr5zx, K, expected))rrxg6 N-)r rxg՜n+H?)rrxgi?)rrxgbJI-)rZr gI8?)rZrgQ3|-0?c\ttjj|||dyr)rr9rrc)rmrYrrRs r3 test_std_pdfzTestExponNorm.test_std_pdfs! ++Aq18%Hr5zx, K, scale, expected)rrxr gVAҤ?)r{Gzt?r g'^ >)rxrxr[r{)rxrxrgbr ;)rrr gVMe?c~tjj|||}|dk(r|dk(sJyt||dy)Nrr{rr)r9rrkrrmrYrr`rRrps r3test_cdf_small_KzTestExponNorm.test_cdf_small_Ks= OO  1E  2 s?8O8 Axe 4r5)rZrxr gh G}$;)rarr g ]fJ?)rrrvg31"g#;)rZrrvgf +-)rrrvr{)rrr ga9S?c~tjj|||}|dk(r|dk(sJyt||dy)Nrr{-a=r)r9rrrrs r3test_sf_small_KzTestExponNorm.test_sf_small_Ks= OO  q!5  1 s?8O8 Axe 4r5N) rrrr$rrr rrrrrrr;r5r3rrs96L L B$ [[-BCICI [[4>D4G? AB 5 B 5& [[4B@DF3@ BC5C5r5rceZdZdZdZej jdgddZej jdgddZ y) TestGenExponcddlm}tjj t j dddddd}t||dddy) Nr)simpsonrZrxrvrz)dxr )scipy.integraterr9genexponrcrdrr)rmrrps r3test_pdf_unity_areaz TestGenExpon.test_pdf_unity_area s?+ NN  ryyB5sC EGA$/A6r5ctjjtjdddddd}tj d|k|dkzsJy)NrrZrxrvrzr )r9rrkrdrr)rmrks r3test_cdf_boundszTestGenExpon.test_cdf_boundssHnn  1b$!7c3GvvqCxC1H-...r5z x, p, a, b, c))r9gM((++Aq13WE((,,S!4aeDr5N)rrrrr;r5r3r%r%:sEr5r%ceZdZdZy) TestExponpowcttjjdddttjj tjj ddddy)Nrrzrrr)rr9exponpowrkrrrrs r3rzTestExponpow.test_tailGsLENN..ub95AENN..u~~/@/@B/GL r5N)rrrrr;r5r3r)r)Fsr5r)ceZdZdZdZdZy) TestSkellamctjdd}d\}}tjgd}ttj j ||||dy)Nrgr9r)gYjP'?g$S?gQ`s2?gC/qF?gD<]Y?g/Xj?g86y?g]a¨?g:?g?gQ>?g?g:?g]a¨?g96?g0X?gA<]?gB/q?gQ`s?g$S?gYjP'?gX_?g+rx?gfSr?gJXx~?rQ)rdrrrr9skellamrJ)rmrVmu1mu2skpmfRs r3rdzTestSkellam.test_pmfNsO IIc2 S - . EMM--ac:FBOr5ctjdd}d\}}tjgd}ttj j ||||dy)Nrgr9r)gƃQ?gS `X'?gVy)>?g: S?gei.f?gȼJ]x?g)?g{^'+i?gLIg>?gnZ ?g?rvͪ?gA?g0?g#J?gqSm . ?g~B?gd?gFBD?gQ9?gmF^?gcMy?gGJq?gC^c?g.){?gñݸ?rrQ)rdrrrr9r/rk)rmrVr0r1skcdfRs r3rVzTestSkellam.test_cdfcsO IIc2 S - . EMM--ac:FANr5cd\}}}ttjj|||ddttjj |||ddy)N)rr gs1Crr}rr )rr9r/rJrk)rmrYr0r1s r3test_extreme_mu2zTestSkellam.test_extreme_mu2xsK. 3 ))!S#6F ))!S#6Fr5N)rrrrdrVr6r;r5r3r-r-MsP*O*Gr5r-c eZdZdZdZej ddZejjdddgejjd gd ejjd gd ejjd e ddDcgc]}d|vr| c}}e jddZ dZycc}}w) TestLognormctj5tjdttj j gdd}t|gddddy#1swYyxYw)Nrmrrvr r )r{g~r?ge3E?)rqrrrsrtr9r"rcrrmrcs r3rLzTestLognorm.test_pdfsZ $ $ & J  ! !'> :--##K3C %c+H I J J Js A A))A2c d\}}}ttjj||z |tjjt j ||z |z ttjj||z |tjjt j ||z |z y)N)g(\5i@gOn?rG)rr9r"rrrdrr)rmrjrXsigmas r3rzTestLognorm.test_logcdfs* B ((B%(8 bffRUmE&9: < ++BrEU+; ((2u)<= ?r5rrc@tjjdSr/rrrs r3rzTestLognorm.rngrr5rrWrar)rrrar5)rKr rr)FTrrFrrctjjd||||}i} |r|| d<|r|| d<|r|| d<ttj|fi| ddiy)Nr[)rrHr`r_rrrrrT)r9r"rrrs r3rz'TestLognorm.test_fit_MLE_comp_optimizersu}}  cYi%,3!@ "DJ "DL &DN &emmT a"&&!"56r5cfgd}tttjj|dddy)NrWrvr}rr r)rrr)rXrr9rrrs r3test_fit_bad_keyword_argsz"TestBeta.test_fit_bad_keyword_argss$ i1$ &r5cdgd}tttjj|ddy)NrNrv)fafix_a)rXrr9rrrs r3#test_fit_duplicated_fixed_parameterz,TestBeta.test_fit_duplicated_fixed_parameters! j%**..!3Gr5zOverflow, see gh-14901reasoncfd\}}}ttjj|||dy)N)g?r=g0xAgx)>)rr9rr)rmrpr1r2s r3test_issue_12635zTestBeta.test_issue_12635s+71a q!Q/1FGr5c0tjgd}tjgd}d}tjj ||dzd|z }t ||tjj ||dzd|z }t ||y)N)g@3@?g^?gNC?)rZr[rrr r)rdrr9rrrr)rminv_R count_listrpinvrs r3test_issue_12794zTestBeta.test_issue_12794s~12XXo. jjnnQ Q0CDU#jjmmCa*1DEQr5cd}tjdd}d}d|z |dz||z }}}tjj |||}tjj |||}t |d|z y)Nh㈵>r rr)rdrr9rrrkr) rmalpha_2count_nobsryr1r2r[rs r3test_issue_12796zTestBeta.test_issue_12796sq1b!g+vz4&=a1jjnnQ1%jjnnS!Q'Q[)r5chd\}}ttjjd||tj d\}}ttjjd||tj d\}}ttjjd||dttjjd||dd\}}ttjjd||dttjjd ||dy) Nrr )rKrr)r rr+ph)rr rN)rr9rrcrdr)rmr1r2s r3rzTestBeta.test_endpointss1UZZ^^Aq!,bff51UZZ^^Aq!,bff51UZZ^^Aq!,a0UZZ^^FAq1151UZZ^^Aq!,a0UZZ^^Ha3Q7r5zDoes not convert boost warningcd\}}}tjt5tjj |||dddy#1swYyxYw)N)gףp= ?gvH7B@0B)rwarnsrtr9rr)rmryr1r2s r3test_boost_eval_issue_14606z$TestBeta.test_boost_eval_issue_14606sB'1a \\. ) $ JJNN1a # $ $ $s "A  Ara, b)rd)@)rjrdc<d} ||||y#t$rYywxYw)Nr) OverflowError)rmrr1r2rps r3 test_beta_ppf_with_subnormal_a_bz)TestBeta.test_beta_ppf_with_subnormal_a_bs-   1aO   s  z x, a, b, ref)rrr}g>׵NlIq)& .!>rwr}g'rOcbtjj|||}t||dyr)r9rrrrmrYr1r2rrs r3rzTestBeta.test_logcdf3s(""1a+%0r5)rnrr}gp}^U)rnrwr}g 1*c`tjj|||}t||dy)Nr)r9rrrrmrYr1r2rrs r3rzTestBeta.test_logsf;s(   Aq)sE*r5r|))rvrvgο)rr g)r rg&׾k )rrg*)Mc`ttj||j|yrprr9rrArs r3rkzTestBeta.test_entropyLs"  1a(002C8r5za, b, ref, tol))r rZgCpr)rZrg8EOr)gNAgNAg(4Fr)gSAgSAg8TAUr&)rg _Bg[VVuD&r)r8r8gR#Lr~)rarg?r5r)rar\r)rZrgnQc}4r)rZrg{CEr)rZr8%\r)rrarur)rrrvr)r8rZrwrcdttj||j||yrrt)rmr1r2rrNs r3rzTestBeta.test_extreme_entropyTs%H  1a(002CcBr5N)rrrrrLrOrSrrskipif MACOS_INTELrWr\rbrxfailrrhrr9rrrrmrrrkrr;r5r3rErEs,7& H  [[ ,DE HF H [[ ,DE F " [[ ,DE *F *8. [[w'GH$I$  [[X  'GH [[Vnn%EFGI. [[^GCEF1F1  [[^GEGH+H+ [[[DE 9 E 9 [[ . C/. Cr5rEcfeZdZgdZdZdZejjdgddZ ejjdedZ d Z ejjd egd zd Z ejjd d dgdZ dZdZdZdZdZy) TestBetaPrime) )rY@r@D>?)rr~r@m嗽?)rr@rWgPK?)rr~r@g%p?)rr@rWg#|/?)rrrgsM <)rr@r~$5Є*?)^ 9^;r@rWgompP?)rrrg7j"!9)rr@r~z ^J?)rr@rWgܷ"+>)rrrgrNs: )rr@r~gK>cXd\}}tjgd}tj||}t tj |j |jt|j|tj|j |y)NrIrJ) rdrr9rr r>rrrrcrrKs r3rzTestBetaPrime.test_logpdfsn t HH_ % OOE4 ( AHHQK(,,./a"&&!"56r5c tjjddd}t|dd\}}t j gd}tjj|||}t t j|jtjj}|Dcgc]}|tj|||}}t||ddycc}w) NrrKrxr{rIrJ-=rh) r9rrkrrdrr r>rr? _cdf_singler)rmrYrrrgen_cdfrCcdfs_gs r3rVzTestBetaPrime.test_cdfs OO  3 ,Q t HH_ %""1eT2 D!%%'(%%11HIJ'%//3t<JJf159Ks5 C'zp, a, b, expected))rxrr}gV)?)rrr}gWJB9*=)r1rr}gOGJ<)rrf@gΣ01)?rfrg8ǘ`?)>q?r@rWgMDcbtjj|||}t||dyr)r9rrr)rmrpr1r2rRrYs r3rzTestBetaPrime.test_ppfs( OO  1a (8%0r5z x, a, b, pc^ttjj||||dyr)rr9rr)rmrYr1r2rps r3test_ppf_gh_17631zTestBetaPrime.test_ppf_gh_17631s!++Aq!4aeDr5ctjd}tjd}tjd}ttjj |||ddy)NrNrvrr)rdrrr9r_ppf)rmr1r2rps r3 test__ppfzTestBetaPrime.test__ppfsH HHSM HHSM HHSM,,Q15sGr5zx, a, b, expected))rrrr)rr@rWgV[?)rr@rWrc^ttjj||||dyr)rr9rrk)rmrYr1r2rRs r3test_cdf_gh_17631zTestBetaPrime.test_cdf_gh_17631s# ++Aq!4hUKr5)r8r@rWg%?)r8r~r@goXr?cptjj|||}|dksJt||dy)NrNgh㈵>r)r9rrkr)rmrYr1r2rRr)s r3test_cdf_extreme_tailsz$TestBetaPrime.test_cdf_extreme_tailss3 OO  1a (3ww8$/r5cgd}gd}gd}gd}tjj|||}t||dy)N)rrrar@r@r@r@r~r~r@r@r@rr)rrar rWrWrWrWr@r@r~r~r~rr)rrꌠ9Y>)Frrrrrrrrrrr)g/:g U/7gKH9g<![a?gw 4?g)J?g5?rrrrg%?gsM  [[ B11 [[\84E5EH [[   L L [[ . / 120 204*100 "r5r}c ReZdZdZdZdZdZdZdZe jjddd gd Z e jjd gd d Z e jjdgde jjdgde jjdgde jjdddge jjdddge jjdddgdZy) TestGammactjjddd}t|dtjjddd}t|dy) NZirKrg CTb?rrZgBΖ?)r9rrcrr;s r3rLzTestGamma.test_pdf%sFkkoob#To2C-kkooa4o0C+r5c\tjjdd}t|dyr)r9rrrrGs r3rzTestGamma.test_logpdf-s$##Aq)FA&r5cdgd}tttjj|ddy)NrNrr)rr)rXrr9rrrs r3rOz#TestGamma.test_fit_bad_keyword_args3s i!!8Lr5ctjtjj ddddsJtjtjj dddd sJy) Nrr gplC@rrrr[g1}t@r)rdiscloser9rrrrs r3rzTestGamma.test_isf7s\"zz%++//%3+%9 99zz%++//%5+%9 99r5cld}d}d}tjj||}t||dy)NPrgޟ򑁺rr)r9rrrrs r3rzTestGamma.test_logcdfMs3  $##Aq)%0r5cld}d}d}tjj||}t||dy)NrrgDmrr)r9rrrrs r3rzTestGamma.test_logsfTs3  % !!!Q's/r5r`rNrOcrtjj|dz|dzd|}t|ddy)Nrrrg>S*rr)r9rrrrmr`rls r3rzTestGamma.test_delta_cdf[s6 &&uSy%)Qe&L6UCr5z a, ref, rtol))rgt.r~)rarr~)r[gX>k-־ @r)rgwM7@r~)gNgmCg%og$6@r~)roܘ#]@r~cZttjj|||yr)rr9rrA)rmr1rrs r3rkzTestGamma.test_entropyis  ++A.$?r5r1)rxr r~r_)rxrr~rRTFrr7c.tjjd}tjj |||d|}i} |r|| d<|r|| d<|r|| d<dt | z } | dk(rId } tjt| 5tjj|fd d i| dddytjj|fd d i| } tj| } | d k\r-t| jtj|| dk\r1t| jdtj|dz| dk\r2t| jdtj|dzyy#1swYyxYw)Nrr[r:rQrrrrrrrrr ra)rdrrr9rrr rrrrrrr)rmr1r_r`rRrr7rrrrr>thetarNs r3 test_fit_mmzTestGamma.test_fit_mmzsWii##$78{{qcS,/1 DJ DL "DNCI  A:MIz; ; :T:T: ;  :T:T:{{E" A: DIIK 7 A: DKKNBGGD!G,< = A: DKKNBGGD!G,< =  ; s #F  FN)rrrrLrrOrrrrrrrrkrr;r5r3rr$s),' M9,10 [[WsCj1 D2 D [[^BC@C@ [[S.1 [[UN3 [[Wn5 [[WtUm4 [[Yu 6 [[[4-8>975642 >r5rcd}tjd|dzz d|dzz dd}t|jddy) Nrg;f?rar rr^rNgV{q@)r9rrrc)jitterZs r3test_pdf_overflow_gh19616rsCF 6FAI-q619}!1MAAEE#J 12r5ceZdZdZej j dgddZej j dgddZej j dgdd Z d Z y ) TestDgammactjjd}d}|jd|}|j d|}t j j||}t jjtj||dz }t||t j |}t|j||dy)NlOP($rZr)rrrarr) rdrrnormalrr9dgammarcrrr)rmrrrYr1rrrNs r3rLzTestDgamma.test_pdfsii##$78 JJRdJ + KKRdK +llq!$kkoobffQi+a/S!||A Su5r5zx, a, expected))rr ggrU>)rr g$IC<)irg2Wj<)rsg~)A.=)rr gf?cxtjj||}t||dtjj ||}t||dtjj | |}t||dtjj ||}t|| dyr)r9rrkrrrr)rmrYr1rRrkrrrs r3test_cdf_ppf_sf_isf_tailz#TestDgamma.test_cdf_ppf_sf_isf_tailsllq!$XE2llx+QU+ \\__aR #H51llx+aRe,r5r))rgmqn@)?g?)rgx?cZttjj||dyrrr9rrArs r3rkzTestDgamma.test_entropys  ,,Q/5Ar5))rr)rg>_)r4g5i)rgo@)rg4Fa3@)rgrO]@cZttjj||dyrrrs r3test_entropy_entreme_valuesz&TestDgamma.test_entropy_entreme_valuess"  ,,Q/5Ar5ctjgd}tjj |}t t |D],}||tjj ||k(r,Jy)N)r rrr4)rdrr9rrArr )rmrYr)rs r3test_entropy_array_inputz#TestDgamma.test_entropy_array_inputsa HH' ( LL  #s1v 6AQ45<<//!55 55 6r5N) rrrrLrrrrrkrrr;r5r3rrs 6, [[-:; - ; - [[X89 B 9 B [[X:; B; B6r5rceZdZdZej j dgddZej j dgddZdZ ej j dgd d Z d Z y ) TestChi2cttjjddddttjjddddy)Nrgޞw1D?r%rQr[g6:֜?rr9chi2rcrrs r3rzTestChi2.test_precisions@EJJNN468N$& (EJJNN346J$& (r5r<))gp@rg~#e)g^@r9g tA)rrgzĀ׿c`tjj||}t||dyr)r9rrr)rmrYrrrs r3rzTestChi2.test_logcdfs&""1b)%0r5))rr9r)rrwgc)rrZg]3c`tjj||}t||dyr)r9rrr)rmrYrrrs r3rzTestChi2.test_logsfs&   B's/r5c|d}tjjd|}t|ddtjjd|}t|ddd}tjjd |}t|d dtjjd |}t|d dy) Nr6gλm:=6g'bd5)r gƯʉ?)r[gn>@)g͔v0@)rg9:3@cbttj|j|dyr)rr9rrArs r3rkzTestChi2.test_entropy4s!  2..0#EBr5cZttjjddddy)Nr{rarvr%rrrs r3test_regression_ticket_1326z$TestChi2.test_regression_ticket_1326=sEJJNN32Cd3ej8ej:ej:fgd?d@ggdAZejjdBgdCejjdDej@jBdEej.dFd3dGdHfej@jDdEej.dFd3dGdHfej@jBdIej.d3dJdGdEfej@jDdIej.d3dJdGdHfgdKZ#ejjdLgdMdNZ$ejj"ejjdLgdOdPZ%dQZ&dRZ'yS)TTestLevyStableT)autousecdtj_dtj_dtj_tj j tj_y)z2Setup default parameters for levy_stable generatorS1 piecewiseN)r9rparameterizationcdf_default_methodpdf_default_method _levy_stable _QUAD_EPSquad_epsrrs r3reset_levy_stable_paramsz'TestLevyStable.reset_levy_stable_paramswsH.2*/:,/:,%*%7%7%A%A"r5ctjttjdz }tj j |jd}|S)a Sample data points for pdf computed with Nolan's stablec See - http://fs2.american.edu/jpnolan/www/stable/stable.html There's a known limitation of Nolan's executable for alpha < 0.2. The data table loaded below is generated from Nolan's stablec with the following parameter space: alpha = 0.1, 0.2, ..., 2.0 beta = -1.0, -0.9, ..., 1.0 p = 0.01, 0.05, 0.1, 0.25, 0.35, 0.5, and the equivalent for the right tail Typically inputs for stablec: stablec.exe << 1 # pdf 1 # Nolan S equivalent to S0 in scipy .25,2,.25 # alpha -1,-1,0 # beta -10,10,1 # x 1,0 # gamma, delta 2 # output file z.data/levy_stable/stable-Z1-pdf-sample-data.npyx,p,alpha,beta,pctrrdrrrrrrTrss r3nolan_pdf_sample_dataz$TestLevyStable.nolan_pdf_sample_datasL6ww N ! ! < = vv  /C D r5ctjttjdz }tj j |jd}|S)a#Sample data points for cdf computed with Nolan's stablec See - http://fs2.american.edu/jpnolan/www/stable/stable.html There's a known limitation of Nolan's executable for alpha < 0.2. The data table loaded below is generated from Nolan's stablec with the following parameter space: alpha = 0.1, 0.2, ..., 2.0 beta = -1.0, -0.9, ..., 1.0 p = 0.01, 0.05, 0.1, 0.25, 0.35, 0.5, and the equivalent for the right tail Ideally, Nolan's output for CDF values should match the percentile from where they have been sampled from. Even more so as we extract percentile x positions from stablec too. However, we note at places Nolan's stablec will produce absolute errors in order of 1e-5. We compare against his calculations here. In future, once we less reliant on Nolan's paper we might switch to comparing directly at percentiles (those x values being produced from some alternative means). Typically inputs for stablec: stablec.exe << 2 # cdf 1 # Nolan S equivalent to S0 in scipy .25,2,.25 # alpha -1,-1,0 # beta -10,10,1 # x 1,0 # gamma, delta 2 # output file z.data/levy_stable/stable-Z1-cdf-sample-data.npyrrrrss r3nolan_cdf_sample_dataz$TestLevyStable.nolan_cdf_sample_datasMJww N ! ! < = vv  /C D r5cdtjttjdz }|S)a&Sample data where loc, scale are different from 0, 1 Data extracted in similar way to pdf/cdf above using Nolan's stablec but set to an arbitrary location scale of (2, 3) for various important parameters alpha, beta and for parameterisations S0 and S1. z1data/levy_stable/stable-loc-scale-sample-data.npy)rdrrrrrss r3nolan_loc_scale_sample_dataz*TestLevyStable.nolan_loc_scale_sample_datas/ww N ! ! ? @  r5 sample_sizer7r?marksrS0rz alpha,beta))rNr)rNr)rr)rrvz gamma,deltar rrac|tj_tj||||}tj|j |d|j \}} | dkDsJy)N)rrr`r_rrr@)r9rrrCrrk) rmrrrrrlrlsrrps r3rDzTestLevyStable.test_rvssb(.>*   d%U || FF $F 7 14xxr5rrvr ctjjdd}d}d}tjj ||||d}tj |d||||f\}}|d kDsJy ) z3Additional test cases for rvs for alpha equal to 1.h:rNrvrr r_r`rrrUrxN)rdrrr9rrrC)rmrrr_r`rYstatrps r3test_rvs_alpha1zTestLevyStable.test_rvs_alpha1sv y!    ! !%3e'+ " -,,q-%*D#u$=?a4xxr5czgd}tjj|\}}}}t|dddt |ddt |dd t |d d|gd z}tjj|\}}} } t |dt |d t | d d t | dd y)N)wgrr{r{r{r{羼u?r r r r 67',?r r r r tA}˜.?r r r r Gz?rrxr`)\(̿ragۧ1?rg9]c?)r r r r r rgv?r )r9rrkrrr rmrYalpha1beta1loc1scale1rjalpha2beta2rrs r3rpzTestLevyStable.test_fits  ',&7&7&A&A!&D#tV148E4+FGQ/ '1 9 9&+&7&7&A&A"&E#tVVQUBFFA.D&!,r5zUnknown problem with fitstart.rTzalpha,beta,delta,gamma)rrrar)rNrrarparametrizationc|tj_tjj||||dd}tjj |}|\}} } } t ||||g|| | | gdy)z7Test that fit agrees with rvs for each parametrization.rrr:rxrN)r9rrrrkr) rmrrrlrrrr alpha_obsbeta_obs delta_obs gamma_obss r3 test_fit_rvszTestLevyStable.test_fit_rvss-<)  $$ 4U%e$% ))$/471 8Y  D% ' )Y 7 r5cNtjgd}tjj |\}}}}tjj | \}}}} t |d|dk7sJt ||t || t || t | |y)N r r rrrZrZrZr4r4r[r[r r)rdrr9rrkrr) rmrYrrrrrrrrs r3test_fit_beta_flipz!TestLevyStable.test_fit_beta_flip8s HH? @&+&7&7&A&A!&D#tV&+&7&7&A&A1"&E#tVUAqyyFF+EE6*D4%(FF+r5c6d}tjgd}tjj | \}}}}tjj | |z\}}} } t ||t ||t | ||zt | |y)Nr r)rdrr9rrkr) rmSHIFTrYrrrrrrrrs r3test_fit_delta_shiftz#TestLevyStable.test_fit_delta_shiftDs HH? @&+&7&7&A&A1"&E#tV&+&7&7&A&A1"u*&M#tVFF+E5)D$,/FF+r5cpgd}tjj|\}}}}|dks Jd||t|ksJdt|d|gd}tjj|\}}} } |dkDs Jd|| t |kDsJdt |d| y) N) r r rrrZrZrZr4r4r#r zExpected alpha < 1, got zExpected loc < z, got ) r r rrrZrZrZr4r4r$zExpected alpha > 1, got zExpected loc > )r9rrkrrrs r3test_fit_loc_extrapz"TestLevyStable.test_fit_loc_extrapOs 6&+&7&7&A&A!&D#tVz>5fX>>zc!f}DAxvdVDD} 7&+&7&7&A&A"&E#tVz>5fX>>zc"g~FR vFF~r5z pct_range,alpha_range,beta_range)rxrvr)rWr ra)rrr)rxr@rvr4r)rWrvr rra)grrrxr}r ) rxr@rWrfffffff?rv?rrr4rrWrarrrcP|}tj}|jdk(xr|jdk(dj |j|j|j g}ddfdgddfd gddfd gdd fd gd dfdgd dfdgg}t |D]Y\} \} } } | tj_ | || |n|} t5}|jtdtjj| d| d| ddd}tjd5t!| gd|tj"|| dz tj"|| dz tj"| dz g}ddd|d| k\tj$|z}d| d | d!|d"|j&j(d#| }t+|| d| |d$%ddd\y#1swYnxYw#1swYtxYw)&z2Test pdf values against Nolan's stablec.exe outputLinuxi686/dnir'c vtj|dtj|dztj|dz|ddk(|ddk(z|ddk\|ddk\z|ddk(zz|ddktj|dd d gzz|dd ktj|dd d gzz|ddktj|dddgzz|ddk(tj|dddgztjtj|dgdzz|ddk(tj|ddgztjtj|ddgzz|ddk(tj|dddgztjtj|dgdzz|ddk(|ddk(z|ddk(zz|ddk(|ddk(z|ddk(zz|ddk(|dd k(z|ddk(zz|ddk(tj|ddgztjtj|dgdzz|ddk(tj|dddgztjtj|dgdzz|ddk(tj|dd d gztjtj|dddgzz|ddk\zzS)Npctrrrrvrrrrxrrxr@r4rKrWrfr)rvr}r~r&r')gٿ333333ӿrxrrvr/rNrWrKrxr)rrrNrr)rdisinrr alpha_range beta_range pct_ranges r3rIz7TestLevyStable.test_pdf_nolan_samples..sGGAeHi0GGAgJ 45GGAfIz23 vY!^uX_.vY#-wZ3.0uX_. wZ3.GGAeHsCj9: wZ3.GGAeHsCj9:$wZ3.GGAeHr2h78%,wZ3.GGAeHsCj9:GGBFF1V9$5|DE-6wZ3.GGAeHrd34GGBFF1V9$5t<=7@wZ3.GGAeHsCj9:GGBFF1V9$57MNOA#JwZ3.vY#-/uX-/K(TwZ3.vY$.0uX-/U-^wZ3.vY#-/uX-/_2hwZ2-GGAeHrd34GGBFF1V9$57GHIi7rwZ2-GGAeHsCj9:GGBFF1V9$5|DEs<|wZ2-GGAeHsCj9:GGBFF1V9$5RyAB}AF7s*GC,EHr5rrctj|dtj|dztj|dz|ddkDz|ddk7zS)Nr.rrrKrNrdr1r2s r3rIz7TestLevyStable.test_pdf_nolan_samples..snGGAeHi0GGAgJ 45GGAfIz23wZ#%'wZ2% 'r5c|ddk( ztj|dzdvztj|dzS)NrrNr.rr7)rr3r4 is_linux_32r5s r3rIz7TestLevyStable.test_pdf_nolan_samples..s]wZ2%$_&GGAeHi01;&(GGAfIz2 3r5g& .=ctj|dtj|dztj|dz|ddkzS)Nr.rrrKr7r2s r3rIz7TestLevyStable.test_pdf_nolan_samples..s[GGAeHi0GGAgJ 45GGAfIz23wZ3&(r5 fft-simpsonr4c|ddk\tj|dztj|dztj|dzS)Nrrr.rr7r2s r3rIz7TestLevyStable.test_pdf_nolan_samples..s[wZ3&GGAeHi01GGAgJ 45GGAfIz23r5rpctj|dtj|dztj|dz|ddkDz|ddkzS)Nr.rrr rr7r2s r3rIz7TestLevyStable.test_pdf_nolan_samples..smGGAeHi0GGAgJ 45GGAfIz23wZ!^%wZ#% 'r5Nz2Density calculations experimental for FFT method.*rYrrr rr`r_rrcalcabserrrelerrrprBz pdf test  failed with method 'z ' [platform: z]  Ferr_msgverbose)platformunamesystemmachinejoin processorr^r9rrr recordrtrcrdrrrisnanrrr)rmrr5r3r4rrI platform_desctestsixdefault_methodr filter_funcsubdatasuprpsubdata2failuresmessager9s ``` @r3test_pdf_nolan_samplesz%TestLevyStable.test_pdf_nolan_samples[s4% llg-I%--62I  \\5==%// :< tJL ^U% U% W' t& t& kC L"+5!1*  B3AE   0 + 7;t,=A "$%  "H%%))CLG$FO *[[h/ 04FF1ws|#34FF1ws|#34rvvgcl7KK H $H%-((1+  t#88HI""/HNN4H4H3IH:W CL#! ?% %  * $  % % s'*AHAH"A!HH HH% cz|}ddfdgddfdgddfdgdd fd gdd fd gdd fdgg}t|D]V\}\}} } |tj_| || |n|} t 5} | j t dtjj| d| d| ddd} tjd5t| gd| tj| | dz tj| | dz tj| dz g}ddd|d| k\tj| z}d|d|d|jjd|}t| | d| |d !dddYy#1swYkxYw#1swYqxYw)"z4 Test cdf values against Nolan's stablec.exe output.rrcXtj|dtj|dztj|dz|ddk(tj|dgdz|ddk(z|ddk(tj|dgdz|ddk(zzzS Nr.rrrN)r/rrrxrWrKrxrr7r2s r3rIz7TestLevyStable.test_cdf_nolan_samples..PsGGAeHi0GGAgJ 45GGAfIz23 wZ2-GGAfI/ABCuX-/ wZ2-GGAfI?@uX-/  r5r@cVtj|dtj|dztj|dz|ddk(tj|dgdz|ddk(zz|ddk(tj|dgdz|ddk(zzSr]r7r2s r3rIz7TestLevyStable.test_cdf_nolan_samples..esGGAeHi0GGAgJ 45GGAfIz237r)& +=>?5T)+ 7r)& ?;<5T)+ r5r;r4ctj|dtj|dztj|dz|ddkDzS)Nr.rr333333?r7r2s r3rIz7TestLevyStable.test_cdf_nolan_samples..ws[GGAeHi0GGAgJ 45GGAfIz23wZ#%'r5rctj|dtj|dztj|dz|ddkDz|ddkzS)Nr.rrrrar7r2s r3rIz7TestLevyStable.test_cdf_nolan_samples..nGGAeHi0GGAgJ 45GGAfIz23wZ#%'wZ3& (r5rctj|dtj|dztj|dz|ddkDz|ddkzS)Nr.rrrrr7r2s r3rIz7TestLevyStable.test_cdf_nolan_samples..rcr5rxctj|dtj|dztj|dz|ddkDz|ddkzS)Nr.rrrNrr7r2s r3rIz7TestLevyStable.test_cdf_nolan_samples..rcr5Nz[Cumulative density calculations experimental for FFT method. Use piecewise method instead.*rYrrr rr>rrr?rprBz cdf test rCz' rDFrE)r^r9rrr rNrtrkrdrrrrOrrr)rmrr5r3r4rrQrRrSrrTrUrVrprWrXrYs ``` r3test_cdf_nolan_samplesz%TestLevyStable.test_cdf_nolan_samples3s*2%U% ,T$ &t& t& t& t& GL \"+5!1(  B3AE   0 + 7;t,=A "$#  "@ %%))CLG$FO *[[h/ 04FF1ws|#34FF1ws|#34rvvgcl7KK H $H%-((1+'rd*??Os&nn2232hZACL#! ;# #  ( $  # # s'AF0AF$:AF0$F- )F00F: rrrrcrkc tj}|jdk(xrdtjdv}|r|dk(rt j d|}dt j_dt j_ ||d|k(}d|t j_ |d vsJ|dk(rt jjnt jj}||d |d |d d d} t| ||dy)zGTests for pdf and cdf where loc, scale are different from 0, 1 r)32bitrrcz4Test unstable on some platforms; see gh-17839, 17859rrS)rcrkrYrrrarr>r4N)rHrIrJ architecturerrr9rrrrrcrkr) rmrrrrIr9rrUrrBs r3test_location_scalez"TestLevyStable.test_location_scales llg-W'X=R=R=TUV=W2W  45= KKN O*0;,/:,tG}-./0[*~%%%%)U]E   ! !8I8I8M8M  CL''*GFO1!  GDM40r5zmethod,decimal_placesr,rrc tjgd}tjgd}tjgd}tjd5t5}|j t d|t j_t jj|d|dd }t||||d d d d d d y #1swYxYw#1swYy xYw) z sample points extracted from Tables and Graphs of Stable Probability Density Functions - Donald R Holt - 1973 - p 187. )rrrrr r r r rararararrrrrrrr)gtF_?gܵ|?gQ?g@?gec]?gHP?gPkw?g!rh?gTN?g ?gDioɴ?g r?gH?g+?gHPs?gX?g&?g 0*?gv/?ga2U0*?)rrfrvr rrfrvr rrfrvr rrfrvr rrfrvr rrzDensity calculation unstable.*)categoryrYr rr>N) rdrrr rLrtr9rrrcr)rmrdecimal_placesxsdensitybetasrVrcs r3'test_pdf_alpha_equals_one_beta_non_zeroz6TestLevyStable.test_pdf_alpha_equals_one_beta_non_zerosXX H ((     [[X & (9(; s JJ'8  4:E   0##''AuA1'EC Wnf       s% C#'ACC#C C##C,zparams,expected)r r rr )rarrZr)rZr0rrc|tjj|d|d|d|dd}t||y)Nrr rarrr3)r9rr)rmrrRobserveds r3rzTestLevyStable.test_stats sG$$** 1IvayfQivay+  Hh/r5r)rfrvrzfunction,beta,points,expectedrNrrZr{rrcd|cxkrdksJJt||||tjt||y)a\Ensure the pdf/cdf routines do not return nan outside support. This distribution's support becomes truncated in a few special cases: support is [mu, infty) if alpha < 1 and beta = 1 support is (-infty, mu] if alpha < 1 and beta = -1 Otherwise, the support is all reals. Here, mu is zero by default. rr rrN)rrdrr )rmrrrrrRs r3!test_distribution_outside_supportz0TestLevyStable.test_distribution_outside_supports@P5}1}}} V5t 4 GGCK * r5zx,alpha,beta,expected))r/iZ?s0?gz5F?)r~H?|/mg>m?)r.d5 ?L ,ٿg깃\?)r>?ЀjoؿgA]4?)rpa2?Յ пgqQ?cdtj_ttjj ||||y)aTest pdf for x equal to zeta. With S1 parametrization: x0 = x + zeta if alpha != 1 So, for x = 0, x0 will be close to zeta. When case "x equal zeta" is not handled properly and quad_eps is not low enough: - pdf may be less than 0 - logpdf is nan The points from the parametrize block are found randomly so that PDF is less than 0. Reference values taken from MATLAB https://www.mathworks.com/help/stats/stable-distribution.html Afc=rwNr9rrrrcrmrYrrrRs r3test_x_equal_zetaz TestLevyStable.test_x_equal_zetaIs9Z&-"    ! !!5t ! <  r5) )rryrzgz¶?)rr{r|gV?)rr}r~gRE)3\?)rrrgB?)rrrg/^?)-C6ryrzgH?)rr{r|gn?)rr}r~gx\?)rrrgVM%?)rrrgm?cdtj_ttjj ||||y)aTest pdf for x near zeta. With S1 parametrization: x0 = x + zeta if alpha != 1 So, for x = 0, x0 will be close to zeta. When case "x near zeta" is not handled properly and quad_eps is not low enough: - pdf may be less than 0 - logpdf is nan The points from the parametrize block are found randomly so that PDF is less than 0. Reference values taken from MATLAB https://www.mathworks.com/help/stats/stable-distribution.html rrwNrrs r3test_x_near_zetazTestLevyStable.test_x_near_zeta}s9D&-"    ! !!5t ! <  r5ctjj}tdddd}tj }t j d i|}d|_d|_|jd i|d|d d }|jd|d  }tj||k(rJd|_d|_|jd i|d|d d }|jd|d  }t||t||y) NrrWr{rNrrr_r`rrrZ rr;) rdrrr r9rrrrr rmrr unfrozenfrozen unfrozen_afrozen_a unfrozen_bfrozen_bs r3$test_frozen_parameterization_gh20821z3TestLevyStable.test_frozen_parameterization_gh20821sii##Ccs#>$$"",V,$(!"&!X\\QFQ#i.Q ::2C N:C66(j0111$(!"&!X\\QFQ#i.Q ::2C N:CXz*Z*r5ctjj}tdddd}tj }d|_t j d i|}|jd i|d|dd }|jd|d }t||d |_t j d i|}|jd i|d|dd }|jd|d }t||y) NrrWr{rNrrrZrrrr;) rdrrr r9rrrrrs r3%test_frozen_parameterization_gh20821bz4TestLevyStable.test_frozen_parameterization_gh20821bsii##Ccs#>$$$(!"",V,!X\\QFQ#i.Q ::2C N:CXz*$(!"",V,!X\\QFQ#i.Q ::2C N:CXz*r5N)(rrrrrrrrrrrrrrDrrrpr{rrr!r%rdrSrZrfrkrsrrrr9rrkrcrxrrrrr;r5r3rrvsV^^D!B"B ^^B ^^))V ^^   [[ [[ FLL lfll4v{{7G7GH   [[/$> [[C [[]VV,<=>? $ [[ [[Vc1X. / -0 [[>? [[    [[D$<  @   , , G [[ [[* FLL  FLL(#(kk&&   FLLD CB' B2&kk''  - *@+,@D [[* FLL  FLL(#(kk&&   FLLD CB' B2&kk''  - *z+*zx [[Wq!f- [[VeU^415.1@ [[ AJ !  @ [[ !RVVRVVRVV!< =  / 00 [[W&78 [['!!%% CB'  !!%% CB'  !!%% Ar2&  !!%% Ar2&  ' : ;9<  [[& )8 98 0 [[ [[   !" 0+,+r5rceZdZdZdZy)TestArrayArgumentcBtjjdyr/r0rrs r3r1zTestArrayArgument.setup_methodr2r5ctjjtjdtj dd}t |jdy)Nrrr)r9rrrdrrrr<rmrs r3test_noexceptionz"TestArrayArgument.test_noexceptions>jjnn"))A,rwwqz")+SYY(r5N)rrrr1rr;r5r3rrs )r5rceZdZdZdZy) TestDocstringc:tjj3tdtjjj vtj j4tdtj jj vyy)Nr%r)r9r%rr lowerrrrs r3test_docstringszTestDocstring.test_docstringssg >> ! ! - J%.."8"8">">"@@ A ?? " " . K5??#:#:#@#@#BB C /r5cTtjtjyrp)r9r?r>rrs r3test_no_name_argzTestDocstring.test_no_name_args  r5N)rrrrrr;r5r3rrs Dr5rctgd}t|dkD|d\}}t|gdt|dgtddkD|d\}}t||t|dgtj|zt|dkD|d\}}t||t|dgtj|zy)N)r rrar rarrr ra)rrararrr)rrrrdrr1r2rs r3test_args_reducers #$A a!eQ "DAqq/*q1# a!eQ "DAqq!q1# *+ a!eQ "DAqq!q1# *+r5cReZdZgdZdZgdZejjde dZ dZ dZ dZ d Zd Zd Zd Zejj$ejjd ddgdZejjd ddgdZy) TestFitMethod)ncfr@rA irwinhallcBtjjdyr/r0rrs r3r1zTestFitMethod.setup_methodr2r5)r_rrrz dist,argsc||jvrtj|dtjdddddtj g}tjdddddtj g}tt|}tt|j|dtt|j|dy ) rz fit known to fail or deprecatedrrrrrr rrN) fitSkipNonFiniterrrdrrrr(r9rXrr)rmrNr*rYr)distfuncs r3!test_fit_w_non_finite_data_valuesz/TestFitMethod.test_fit_w_non_finite_data_valuess 4(( ( KK4& @A B HHfffffbffE F HHfffffbffE F5$'j(,,!<j(,,!.mlefuncNsAAB!"BKKA&Eufw{{1~566ufw{{1~5668DKr5)rsrfrvrr r;rprra)rrrr4)rrrrvr:)rrr)rrrr)rrrr) rdrr9rrrrrrXr)rmrrYr1r2r_r`dadbrjrb2rrr)s r3test_fix_fit_betazTestFitMethod.test_fix_fit_betaKs  HH' ( ::>>!!A>>1c5S!UA1a(1a&t< HH' ( ::>>!!>D1c5QS!UAAq!BAD)U$zz~~bQQq~IBfRT1VQR$BAD)B" j%**..!#aH HHZ j%**..!!AFj%**..!!A!Lj%**..!!A!L j%**..!aQ (r5ctjgd}tjj |\}}t |dt |dtjj |d\}}t |dt |dtjj |d\}}t |dt |dy)N)rararrrrrrrarrrrrr)rdrr9r_rrrs r3test_expon_fitzTestFitMethod.test_expon_fits HH- .[[__Q' US!UA[[__Qq_1 US!UA[[__QQ_/ US!UAr5c @tjgd}tj|dz }tjj |d\}}}t ||jdt|dt |tj|jdtjj |dd\}}}t |tj|tjdz dzjdt|dt|dtjj |dd \}}}t|d t|dt |tj|jdy) N)rrrZr9rr rrrr*r;rar)rfix_s) rdrrr9r"rrrurrrr)rmrYlnxm1r<r_r`s r3test_lognorm_fitzTestFitMethod.test_lognorm_fits3 HH- .q1u !MM--aa-8sEuyy{7S!rvvejjl3%@!MM--aa-BsErww):Q(>'D'D'FG" $S!UA!MM--aat-DsEUD!S!rvvejjl3%@r5ctjgd}tjj |\}}t ||j t |tj|tjj |d\}}t |dt ||jtjj |d\}}t |dt |dtttjj|dtttjj|dy)N)rNrrZr rrrZrrrzrO) rdrr9rrrrrrrXrrs r3test_uniform_fitzTestFitMethod.test_uniform_fits HH) *]]&&q) US!%%'"UBFF1I&]]&&qq&1 US!UAEEG$]]&&q&4 US!UBj%--"3"3QSAj%--"3"3QsCr5rMLEMMc >d\}}tjj||dd}tjj|d|}tjj|d|}t ||dd tjj|d| }t ||dd tjj|d | }tjj|d | }t ||dd tjj|d |}t ||dd t t tjj|dd|t t tjj|dddd|tjj|ddd|} | \} } } } t| | | ggdd}tjj|d}tjj|||\} } } t| |y)N)rrr[rrr)rr)rQrrrh)rRrr)rr)rr)rrr ra)rQrrrr)rQrrrr)rQrrr)rrr r5) r9rrrrrXrrr)rmrr1r2rYres_1res_2res_3res_4res_5aabbrssrs r3 test_fshapeszTestFitMethod.test_fshapess 1 JJNN1acN = qR7 qR7u5u= q6:u5u= qR7 qR7u5u= q6:u5u= j%**..!aO j%**..!aQv 7  qRa&IBBb"b\:. {{qs+[[__Ta_? BRr5ctj}|jdd}td}t t |j |fi|d|iy)Nrar[)rri) enikibenikir)r9rrr rXrr)rmrrNrr)s r3test_extra_paramszTestFitMethod.test_extra_paramssCxx!#x&t$i4F3FvFr5N)rrrrr1rrrrrrrrrrrrrrrrr;r5r3rrs 1DA [[[(3=4=@ $3L5(n A(D$ [[ [[Xt}5$6$L [[Xt}5G6Gr5rcBeZdZdZdZdZdZdZdZdZ dZ d Z y ) TestFrozencBtjjdyr/r0rrs r3r1zTestFrozen.setup_methodr2r5ctj}tjdd}|jd}|jddd}t|||j d}|j ddd}t|||j d}|j ddd}t|||j d}|j ddd}t|||jd}|jddd}t|||j}|jdd}t|||j}|jdd}t|||j}|jdd}t|||j}|jdd}t|||j}|jdd}t|||jd}|jddd}t||t|j|jt|j|jy)Nrrr^rrfra)r9rrcrrkrrrr?rr8rurArr1r2)rmrNrresult_frs r3rzTestFrozen.test_norms zzC0::d#$D4Xv&::d#$D4Xv&::d#$D4Xv&::d#$D4Xv&99T?4s3Xv&==?S1Xv&;;=t3/Xv&::<d#.Xv&::<d#.Xv&>>#$c2Xv&==#QD4Xv&VXXtvv&VXXtvv&r5cd}tj}tj|}|jd}|jd|}t|||j d}|j d|}t|||j d}|j d|}t|||j d}|j d|}t|||jd}|jd|}t|||j}|j|}t|||j}|j|}t|||j}|j|}t|||j}|j|}t|||j}|j|}t|||jd}|jd|}t||t|j|jjt|j |jj y)Nrzrrfrra)r9rrcrrkrrrr?rr8rurArr1rNr2)rmr1rNrrrs r3 test_gammazTestFrozen.test_gamma$s {{Q::d#$"Xv&::d#$"Xv&::d#$"Xv&::d#$"Xv&99T?q!Xv&==?QXv&;;=1Xv&::<!Xv&::<!Xv&>>#aXv&==#Q"Xv&VXXv{{}}-VXXv{{}}-r5ctjd}|jd}|jd|jd}t||y)Nr rarr)r9r"rr)rmrm1rts r3test_regression_ticket_1293z&TestFrozen.test_regression_ticket_1293XsHq! ]]1  V $]]1  Rr5cd}tj|}|jj|\}}t ||gddgd}tjj d|t |jj|dt jgd}tj|}|jj|\}}t ||gddgd}tjj d|t |jj|jjftjj|tjd}t|j|judD]}t j|}tj|}|j|j}}t |dtt j|t jd}tjj|\}}t||gdd gy) Nrrr{rrWrrrrv)r9rrNrrrcrdrr1r2r rrr)rmrrr1r2rv1s r3rzTestFrozen.test_abgs  __q !ww##A&1aVb#Y'  A#RWW))!,q"&&k:  __q !ww##A&1aVb#Y'  Aq!bggii+U__-I-I!-LMoo$'( /A 1 A1%B44qA B  BKKN # 3A??//2DAq QFRI . /r5cJtttjdy)N rv_frozen)r r0r9r:rrs r3test_rv_frozen_in_namespacez&TestFrozen.test_rv_frozen_in_namespaces++[9:r5cftj}tt|dd|_t |jj tjjdj tjjd}|jd|y)Nr*rrr) r9rr r0rr get_staterdrrr)rmrrndms r3test_random_statezTestFrozen.test_random_states/0!V((224YY**2.88: <yy$$T*  -r5cTtjdd}tjd}tjgdgdf}|||fD]}d|_|j d t j|}|j d }t j|}|j d }t|||jd |jd g} t| d | d t|j| d |j| d y) Ngdsz@g@e?r)rr rarr0rJrrr5rvrr ) r9rr~r>rrpickledumpsloadsrrrk) rmrpoissrCdistfnrHr0 unpickledr1medianss r3 test_picklingzTestFrozen.test_picklings zz,.AB b!""<+?+ABUF+ 4F"&F  JJAJ  V$A#B QIA&B R zz# c(:;G WQZ 0 GAJ/"wqz2 4 4r5c d}tjddd}tjdd5|j |ddd }tjj |d ddddd }dddt tj dd }|j |}tj j |dd}t ||y#1swYlxYw)Nc|Srpr;rs r3rz$TestFrozen.test_expect..funcsHr5rarrrr)rdivider T)rub conditionalrar*r_r`rr r r)rr*r_)r9rrdrrrr~)rmrgmgm_val gamma_valrpp_val poisson_vals r3 test_expectzTestFrozen.test_expects [[1!1 - [[( ; IYYtaTYBF **4d./A4+II I  * MM! #mm**4d*B {+ I Is .s1qs r5rrar^rr%rQc|Srpr;rs r3rIz&TestExpect.test_norm..sr5r@r4cyrr;rs r3rIz&TestExpect.test_norm..rr5r_r`rr rcyrr;rs r3rIz&TestExpect.test_norm..rr5Tr_r`rr r rN)r9rrrr)rmrrrr prob90prob90cs r3rzTestExpect.test_norms JJ  3!  DAq"- JJ  m!  <Aq"- ZZ^^Daq^ 1 ZZ^^Daq^ 1"";AQ2""MFC4**##KQaB204$6GR4r5c 0tjjdddd}t|ddtjjd dd d }t|d dtjj d dddd}tjj ddddd}tjjddd d ||d}t|ddtjjdddd||d}t|ddy)Nc|dz |dz zS)NUUUUUU@r;rs r3rIz&TestExpect.test_beta..s51U7(;r5rrra)r*r_r`qq?rrQc|Srpr;rs r3rIz&TestExpect.test_beta..sr5rOrzr r4rZr^r@cy)NrNr;rs r3rIz&TestExpect.test_beta..rr5)rZrZFr rcyrr;rs r3rIz&TestExpect.test_beta..rr5TrN)r9rrrr)rmrrr rr r s r3 test_betazTestExpect.test_betas JJ  ;'"#1  .Avr2 JJ  kRr  JAub1 ZZ^^D"baq^ 9 ZZ^^D"baq^ 9"".Qr5)rrZrr rrQc|dz dzSNr rar;rs r3rIz+TestExpect.test_hypergeom..sadQYr5r%c|dz dzSr# r;rs r3rIz+TestExpect.test_hypergeom..sQrTAIr5r)r*r_rr r cyrr;rs r3rIz+TestExpect.test_hypergeom.. rr5r*rPcyrr;rs r3rIz+TestExpect.test_hypergeom..rr5T)r*r_rr r cyrr;rs r3rIz+TestExpect.test_hypergeom..rr5r)r*rr )r9rrrrJrg) rmm_truev_truerrv_bounds prob_true prob_boundsprob_bcprob_bs r3test_hypergeomzTestExpect.test_hypergeomsx..r2qb.A OO " ";[b " IAvr2 OO " "#6[') # +Avr2??))*=/:.0Q2*? Hfb9eoo))1b'2r1!)DHHJJ oo,,[{13b-B KB?//((;B,-"$)HGQ3'' ++,(4FAr2r5ctjjdddd}dtjjddz }t ||d tjjd ddd }t |dd y) Ncyrr;rs r3rIz)TestExpect.test_poisson..rr5r rF)r*rr r rar%rQcyrr;rs r3rIz)TestExpect.test_poisson..!rr5T)r9r~rrkr)rmr, prob_b_trueprob_lbs r3 test_poissonzTestExpect.test_poissonsymm**;Ta7<+>  ))!Q// KbA--&&{!37'9GQ3r5ctj}|jd}|jd|jd}t||dy)N)rrU)rvr%rQ)r9genhalflogisticrr)rmhalflogr)r-s r3test_genhalflogisticzTestExpect.test_genhalflogistic%sI''~~6~*F#~~6~*D$3r5cttjtjj ddttjtjj ddttjtjj ddttjtjj ddy)NrvGz?cyrr;rs r3rIz/TestExpect.test_rice_overflow..4rr5)r; rUcyrmr;rs r3rIz/TestExpect.test_rice_overflow..5rr5cyNrr;rs r3rIz/TestExpect.test_rice_overflow..6rr5)r rdr>r9ricercrrrs r3test_rice_overflowzTestExpect.test_rice_overflow/s  EJJNN3567 EJJ--k-HIJ EJJ--k-HIJ EJJ--k-HIJr5cd\}}tjjd|f}t|||dz z t j d|z z dtjjd|f|}t|||zdy) N)rxrc|Srpr;rVs r3rIz(TestExpect.test_logser..;ar5rUrNr~rc|Srpr;rD s r3rIz(TestExpect.test_logser..ArE r5r )r9rrrrdr)rmrpr_res_0res_ls r3 test_logserzTestExpect.test_logser8s~3 ##Kqd#;QV rvvb1f~5E C ##Kqd#Dus{7r5cd\}}tjjd||f}tjjd||f}t|||z dt||dzz ||zdy)N)rrc|Srpr;rs r3rIz)TestExpect.test_skellam..HsAr5rUc |dzSrmr;rs r3rIz)TestExpect.test_skellam..Irnr5rrra)r9r/rr)rmrp2rrts r3 test_skellamzTestExpect.test_skellamDsoB ]] ! !+RH ! = ]] ! !.Bx ! @BG%0RU BG%8r5c d\}}tjjd||f}t|t dt ||D||z z dy)N)rqc|Srpr;rs r3rIz)TestExpect.test_randint..Qr! r5c3 K|]}|ywrpr;rrs r3rkz*TestExpect.test_randint..Ss5!A5s r~r)r9r;rrrgr)rmr*r+rs r3 test_randintzTestExpect.test_randintMsQBmm"";R95uR}55bA Or5cZtttjjddy)Nc |dzSrmr;rs r3rIz&TestExpect.test_zipf..Xs q!tr5r )r rtr9rfrrrs r3 test_zipfzTestExpect.test_zipfUs^UZZ%6%6#T +r5ctjjdd}tjjddddd}t||d y) Ncyrr;rs r3rIz/TestExpect.test_discrete_kwds..\rr5r rUcyrr;rs r3rIz/TestExpect.test_discrete_kwds..]rr5r r&)r*maxcount chunksize tolerancer%rQ)r9r~rr)rmn0n1s r3test_discrete_kwdszTestExpect.test_discrete_kwdsZsL ]] ! !+D ! 9 ]] ! !+D+/2"OBB/r5cd}dD]6}tjjd|}t|||d8y)NcH|dzd|dzzzd|dzzzd|dzzz|zS)NrrZrrrr9rar;rs r3 poiss_moment5z-TestExpect.test_moment..poiss_moment5ds8q52b!e8#bQh.BE9B> >r5)rrrrr)r9r~rr)rmrd rXm5s r3 test_momentzTestExpect.test_momentas? ? ?B%%a,B B b 1 > ?r5cttjjdddttjjdddttjjdddttjjdd ttj jd d y) NrFrNr^rwrZrW)rUrh Ur)rr9rrrrYrrs r3test_challenging_cases_gh8928z(TestExpect.test_challenging_cases_gh8928ks  ))b)rrs r3test_nct_inf_momentszTestNct.test_nct_inf_momentssYY__V_D 1aaAq\BFFBFFBFFBFF#CDYY__V_D 1a AaAY 89YY__V_D 1a Q1I&**,-Qr5ctjjdd}tjjdd}ddg}t||ddt||dtjjdd}tjjdd}d d g}t||ddt||d y) Nrrag@@g*dn?rrrrgqI@gL)?r)r9rp rr)rmnct_mean_df_1000nct_stats_df_1000expected_stats_df_1000nct_mean nct_statsrs r3test_nct_stats_large_df_valuesz&TestNct.test_nct_stats_large_df_valuess!99>>$2!IIOOD!4 #56H!I(*@*C" $)+A" $99>>&!,IIOOFA. ,.@A."3%@ >=r5c lttjjddt ddy)Nrair)rr9rp rkrrrs r3test_cdf_large_nczTestNct.test_cdf_large_ncs! aE%L91=r5zx, df, nc, expected)rrZrmgAs69)rgrrmgV1c^ttjj||||dyr)rr9rp rc)rmrYrrw rRs r3test_pdf_large_nczTestNct.test_pdf_large_ncs!  aR0(Gr5N) rrrrq rs ru rx r r rrrr r;r5r3rn rn sTC2$  >,>6 [[2.+5  H  Hr5rn c$eZdZdZdZdZdZy)TestRecipInvGausscRtjjdd}|dk(sJy)Nrr}r{)r9 recipinvgaussrcrs r3test_pdf_endpointz#TestRecipInvGauss.test_pdf_endpoints%    # #As +Cxxr5cptjjdd}|tj k(sJy)Nrr})r9r rrdrrbs r3test_logpdf_endpointz&TestRecipInvGauss.test_logpdf_endpoints,""))!S1wr5cdtjjdd}d}t||dy)Nr@rvg9V9s;rr)r9r rkrrmrprRs r3test_cdf_small_xz"TestRecipInvGauss.test_cdf_small_xs-    # #D# .(8%0r5cbtjjdd}d}t||dy)NrrvglHr9r@ rcrrrkrrr)rmrYryrr2s r3test_rice_zero_bzTestRice.test_rice_zero_bsR  EJJNN1N3488:; EJJ--a2-67;;=> EJJNN1N3488:; EJJ--a2-67;;=>  EJJNN1N3488:;zz62 D!%%'(  q!,ejjnnQ.BQ (r5ctjj}t|djdt|ddj dy)Nrr r )rr)r2r)r9r@ rrrr<rs r3 test_rice_rvszTestRice.test_rice_rvs%s8jjnnS2Y^^Q'S2F+116:r5ctjjtjdddtjddd}gd}t ||tjddd}tjj |dd}gd }t ||tjjj d tjdd d}gd }t ||y) NrZ)g/ Ƹ?g[jNJ\?g ?g9LJ?g#O?gXx?glO?g٥a%?gVzӯ?gTQ+xQ?g?`J?go?gbu=?gIo|?g"Z6?rWr g@_@rr) g<~@gwd @gC?@gb 0@g` A@@g'NxP@g33-a@g4pv@g$F@rv)gJ-$@g+f4@gce*D>@g$D@gnHGI@grN@g< uQ@giffT@g9r[V@g3bXQY@gq xJ[@g`*DD^@g@`@gdAa@)r9r@ rkrdrrrr8)rmrkcdf_exp probabilitiesrppf_exps r3test_rice_gh9836zTestRice.test_rice_gh9836*sjjnnRYYr33RYYr35KL O W% #q#. jjnn]En; L W%kk""3 "c2(>? ; W%r5N)rrrr r r r;r5r3r r s(*; '&r5r ceZdZdZdZdZy) TestErlangcBtjjdyr/r0rrs r3r1zTestErlang.setup_methodUr2r5c tj5tjdtt tt j jddddgd}t j j|d}t jj|d}t||d dddy#1swYyxYw) Nrmrrr rr)rvrNrzrrrr) rqrrrsrtrXr9erlangrrrr)rmr result_erlang result_gammas r3test_erlang_runtimewarningz%TestErlang.test_erlang_runtimewarningXs $ $ & D  ! !'> : .,,**CQaa I (D!LL,,Ta,8M ;;??4A?6L M.scale_mles,FFD4KA-.!c$i-@RG Gr5rr}rr)r9r%rrr)rmrr5rr scale_expectr_r`s r3rpzTestRayleigh.test_fit}s~~!!sy!I H!w/ ^^''7'; US'"UL)^^''R'8 UUB ^^''- UUIdC01r5r; rxg̡=E?g@1d?c|tjjd||}ttj|y)Nrre)r9r%rr)rmrr5rs r3 test_fit_comparison_super_methodz-TestRayleigh.test_fit_comparison_super_methods- ~~!!sy!I%ennd;r5c6ttjyrp)rr9r%rrs r3rzTestRayleigh.test_fit_warningssENN+r5ctjjd}d\}}}tjj ||||}tjj |\}}|tj|ksJtjj ||\}}|tj|ksJ||k(sJy)Ni)r7rrrrr)rdrrr9r%rrr) rmrr_r`rrrrrs r3test_fit_gh17088zTestRayleigh.test_fit_gh17088sii##C('UDnn  e$S I^^'', $$$"^^//E/B$$$E!!!r5N) rrrr1rrrrrrpr rr r;r5r3r r ps"0" [[03K3K3MN2N2. [[0D$<4> 3K3MN<N< , "r5r c>eZdZdZdZdZejjdgddZ ejjdgdd Z ejjdd d gd Z ejjdd dgdZ y) TestExponWeibcd}d}d}tjj|||}tjj|||}t ||gddgy)NrWrNr~g:23W+g~`9k)r9 exponweibrcrr)rmrYr1rrprs r3test_pdf_logpdfzTestExponWeib.test_pdf_logpdfsY    OO  1a (%%aA.D /1DE Gr5cntjddd}d}d}tjj |||}tj j ||}t ||tjj|||}tj j||}t ||y)Nrrrr r[)rdrr9r rc weibull_minrrrmrYr1rrprRrs r3 test_a_is_1zTestExponWeib.test_a_is_1s KKB "   OO  1a ($$((A.8$%%aA.$$++Aq1h'r5cjtjddd}d}d}tjj |||}tj j |}t ||tjj|||}tj j|}t ||y)Nrr rZ)rdrr9r rcr_rrr s r3test_a_is_1_c_is_1z TestExponWeib.test_a_is_1_c_is_1s KKAr "   OO  1a (;;??1%8$%%aA.;;%%a(h'r5z x, a, c, ref))r r}rg=J ?)r7r}rg$+PR>)rr}rg ><)rr}rggX.X:)r?rgRy9)rr rg'vR0=)g_QPrrgQ: [[ 5 - -* [[ 4..  [[^AGIJ1J1  [[^IFHI0I0r5r ceZdZdZdZy)TestFatigueLifec`tjjdd}t|ddy)Nrfr}X9rr)r9rrrrs r3 test_sf_tailzTestFatigueLife.test_sf_tail(s)     ,0u=r5cdd}tjj|d}t|ddy)Nr r}rfrr)r9rrr)rmrprys r3 test_isf_tailzTestFatigueLife.test_isf_tail6s+ !    ! !!S )5u-r5N)rrrr r r;r5r3r r &s  >.r5r ceZdZdZdZej jdddgdZdZ ej jdd d gd Z y ) TestWeibullc\tjjdd}t|dyr)r9r rrr s r3rzTestWeibull.test_logpdf?s$    $ $Q *Qr5cXd}d}d}tjj|||}t|t j ddz tjj |||}t|dt jdz tjj|||}t|tjd tjj|||}t|t jtjd tjj|||}t|t j dtjj|||} t| dtjjddd}t|t j d tjjddd} t| d d }tjj|||}t|t j ddz tjj |||}t|dt jdz tjj|||}t|t j dtjj|||}t|dtjj|||}t|tjd tjj|||} t| t jtjd tjjd dd}t|tjd  tjjd dd} t| t jtjd  y) Nrrzrrrrr4rarrRg& .gCe)r9r rcrrdrrrrkrrrrr weibull_max) rmrYr1r2rpr.rlcrHrs r3test_with_maxima_distribz$TestWeibull.test_with_maxima_distribDs?.       ! !!Qa ! 0266%=?+    % %a! % 4EBFF1I-.    ! !!Qa ! 0GMM%001    % %a! % 4BFFGMM%$8#89:    AQ /266%=)    $ $Q $ 3E"    Qa 0266$<(    $ $R! $ 4D!     ! !!Qa ! 0266%=?+    % %a! % 4EBFF1I-.    ! !!Qa ! 0266%=)    % %a! % 4E"    AQ /GMM%001    $ $Q $ 3BFFGMM%$8#89:     3GMM*@AAB    $ $UAQ $ 7BFFGMM2H$I#IJKr5r`rNrWcntjj|dz|dzd|}t|dy)Nrrrrg>)r9r rrrs r3rzTestWeibull.test_delta_cdfs;"!!,,U3Ya38-:56r5ctjjd}d\}}}tj|||}|j d|}tjj |dd\}}} tjj |dd\} } } || cxk(rdk(sJJ|| k7sJtjj |ddd \} }}|dk(sJtj| ||}|jd }tj|tj|f}t||y) Nl>)rar2rvr[rrrrrrrrr) rdrrr9r rrrr9r)rmrrr_r`rNrc2rrc3rrc4rrrrrs r3 test_fit_minzTestWeibull.test_fit_mins2ii##$78# 3  C/hhCch2!,,00c0BD& ,,00c0BD&t q     Rxx!,,00a$0OD&{{!!"dF3kk$k'ggclEJJsO+S!r5r%)r7r r)rrgq뮕qH)cttjj|||dttjj |||dyrK)rr9r rrr)s r3rzTestWeibull.test_sf_isfsB )),,Q2CeD))--c15quEr5N) rrrrr rrrrr rr;r5r3r r =su RLh [[WsCj1727("< [[[+J,D+EFFFFr5r ceZdZdZdZy) TestDweibullc2tjjd}d|jddz}tj j |}tjj |tjdz }t||dy)Nq>~).rZr[rrvr~r) rdrrrr9dweibullrAr rr)rmrrrrs r3rkzTestDweibull.test_entropyssii##$78  2 . .nn$$Q'''*RVVC[8Su-r5c6tjjd}d|jddz}d|j z}t j j||}dt jj||z}t||dy)Nr rZr rrvr~r) rdrrrrr9r rr r)rmrrrYrrs r3rzTestDweibull.test_sfsii##$78   , ,  nn1%E%%((A..Su-r5N)rrrrkrr;r5r3r r s  ..r5r cBeZdZdZdZdZdZdZdZdZ dZ d Z y ) TestTruncWeibullchtjjddgddd}t|ddgy)NrWrzg)\(?gףp= ?r{)r9truncweibull_minrcrr s r3test_pdf_boundsz TestTruncWeibull.test_pdf_boundss0  " " & &Sz3d CQc #r5ctjjdddtj}t |dtjjdddd}t |dy)NrzrNr{rg~?)r9r rrdrrrr s r3rzTestTruncWeibull.test_logpdfsU  " " ) )#sC @Q  " " ) )#sC =./r5chtjjddgddd}t|ddgy)Nr{rNrzrW)r9r rrr s r3test_ppf_boundsz TestTruncWeibull.test_ppf_boundss0  " " & &Sz3S AQc #r5cgd}tjj|ddd}tjj|ddd}t ||yN)r{rWrfrvrrrNrzr{r)r9r rrkrrmryrYq_outs r3test_cdf_to_ppfz TestTruncWeibull.test_cdf_to_ppfsI 0  " " & &q"b" 5&&**1b"b95!r5cgd}tjj|ddd}tjj|ddd}t ||yr )r9r rrrr s r3test_sf_to_isfzTestTruncWeibull.test_sf_to_isfsI 0  " " & &q"b" 5&&))!RR85!r5c d d d fdtjjd }t|dtjjd }t fd \}}t ||tjjd }t fd  \}}t ||tjjd }t fd  \}}t ||tjjd } t fd  \} }t | | y)NrzrNrcT||ztjj|zSrp)r9r rc)rYror1r2rs r3xnpdfz)TestTruncWeibull.test_munp..xnpdfs)a4..221aA>> >r5rr c|dSrr;rYr s r3rIz,TestTruncWeibull.test_munp.. a r5rac|dSrmr;r s r3rIz,TestTruncWeibull.test_munp..r r5rc|dSr? r;r s r3rIz,TestTruncWeibull.test_munp.. r r5rc|dS)Nrr;r s r3rIz,TestTruncWeibull.test_munp..$r r5)r9r rrrr)rmrr m1_expectedrrt m2_expectedm3 m3_expectedru m4_expectedr1r2rr s @@@@r3 test_munpzTestTruncWeibull.test_munp s#    ? # # * *1aA 6R  # # * *1aA 63Q: QK(  # # * *1aA 63Q: QK(  # # * *1aA 63Q: QK(  # # * *1aA 63Q: QK(r5c 8d}d}d}tjdtjdtjdtjdz zz }tj j ||||}t|dtj j||||}t|tjd tj jd|||}t||tj j||||}t|dtj j||||} t| tjd tj jd|||} t| |y)NrNrrzr rvr) rdrrrr9r rkrrrrrr) rmr1r2rx_medrkr rrrrs r3test_reference_valuesz&TestTruncWeibull.test_reference_values'sB   BFF3"rvvbz/0B)C#CDDE$$((1a8S!  # # * *5!Q :RVVBZK($$((aA6U#  # # & &uaA 6C  # # ) )%Aq 9RVVBZK($$((aA6U#r5cd}d}d}tj}d}tjj |||}tj j |||||}t ||tjj|||}tj j|||||} t || tjj|||} tj j|||||} t | | tjj|||} tj j|||||} t | | tjj|||}tj j|||||}t ||tjj|||}tj j|||||}t ||tj jdd||d}t |tjd tj jdd||d}t |d y) Nrrzr{rrr4rarr) rdrr9r rcr rrrkrrrr)rmrYrr1r2r`rpp_truncr.lp_truncrk cdf_truncr lc_truncrHs_truncrls_truncs r3test_compare_weibull_minz)TestTruncWeibull.test_compare_weibull_min?s1    FF    ! !!Qe ! 4((,,Q1au,E7#    % %a% % 8))00Aq!50IH%##Aq#6**..q!Q.G Y'    % %a% % 8))00Aq!50IH%    AU 3((++Aq!Qe+D7#    $ $Q $ 7))//1a%/HH%  " " % %b!Q % ;266$<(  # # ) )"aAQ ) ?D!r5cd\}}}tj||d}tjj ||||}tjj ||||}tj j ||tj j ||z }tj j |||z }tj j ||tj j ||z |z } tjj||tjj|| y)N)r}rfrr[) rdrSr9r rcrkr rr) rmrr1r2rYrrprrrrs r3test_compare_weibull_min2z*TestTruncWeibull.test_compare_weibull_min2hs"1a KK1c "%%))!Q15%%))!Q15  $$Q*U->->-B-B1a-HH  $$Q*T1!!%%a+e.?.?.C.CAq.II4O ""4. ""4.r5N) rrrr rr r r r r r r r;r5r3r r s0$ 0$ " " )6$0'"R/r5r c`eZdZdZdZej jdgddZy) TestRdistctj}gd}t|j|j |dd|dy)Nrg@rrQ)r9rdistrrkr)rmrrDs r3test_rdist_cdf_gh1285zTestRdist.test_rdist_cdf_gh1285zs5$FJJvzz&%'@%H"A /r5ctjddd}d}tdtj|dz |dz j |dzdz ztj |j |y)NgGzrrZg@rvrar )rdrSrr9rrcr )rmrYrs r3test_rdist_betazTestRdist.test_rdist_betasd KKtR ( C 1Q3! 4 8 8!a% CC!KKN..q1 3r5r%))rig?)rWg;|bZ?)rvig79)riUg޻S/c\ttjj|||dyrK)rr9r rr)s r3 test_rdist_sfzTestRdist.test_rdist_sfs  q!,c>r5N) rrrr r rrrr r;r5r3r r ys7/3 [[ ??r5r cxeZdZdZdZdZdZdZdZe jjdgdd Z y ) TestTrapezoidcHgd}|D]}d|dg}ttjj|||tjj||ttjj |||tjj ||y)N)rrxrvr rr )rr9rrctriangrk)rmmodesmoderYs r3test_reduces_to_triangz$TestTrapezoid.test_reduces_to_triangs  ;DD! A  3 3AtT B % 0 0D 9 ;  3 3AtT B % 0 0D 9 ;  ;r5cRtjddd}ttjj |ddtj j |ttjj|ddtj j|y)Nrr rZ)rdrSrr9rrcrrkrs r3test_reduces_to_uniformz%TestTrapezoid.test_reduces_to_uniformsl KK1b !EOO//1a8%--:K:KA:NOEOO//1a8%--:K:KA:NOr5c ttjjddddttjjddddttjjddddttjjddddttjjd ddd ttjjdddd ttjjd ddd ttjj d ddd ttjj ddddttjj ddddttjj d dddttjj ddddy)Nrrar rvrrrKrNrWg?rrr rsg?)rr9rrcrkrrs r3 test_caseszTestTrapezoid.test_casessfEOO//1a8!<EOO//1a8!<EOO//Q</ 1EOO//S#>/ 1 EOO//S#>FEOO//S#>EEOO//S#>FEOO//S#>HEOO//S#>FEOO//S#>DEOO//S#>HEOO//S#>Dr5cV d\ z z z z z z z f\}}}}d z z z z fd}|d}|d|dzz }d z z z z z z z z tjd z z z zz}ttjj |||||dttjj |||||dttjj|||||dttjj ddd d d dttjj ddd d ddttjj ddd d d dy) N)rrrarrac|dzz|dzzz z z |dzz|dzzz z z z z|dzz |dzz S)Nrar r;)ror1r2rdrjs r3rz6TestTrapezoid.test_moments_and_entropy..momentsn!ac(Q1X-!A#6!Hq1Q3x/AaC89:qSqS" #r5r rvrrQrrr*rr)rdrrr9rrr8rA)rmrrM r_r`rrr8rAr1r2rr# rjs @@@@@r3test_moments_and_entropyz&TestTrapezoid.test_moments_and_entropys" 1acac]QqSQqSM1acABU 1QqM # # ayQi$'!1Qq/QqSU1W-sac!eAg0GGEOO00ReD " .EOO//BUC -EOO33BCG#R 1 EOO00Ar1=r2NEOO00Ar1=q"MEOO//1b!r5ctjddd}tjd5tj j |dd}dddttjj |y#1swY3xYw)Nrr rZz`trapz.pdf` is deprecatedr) rdrSrdeprecated_callr9rBrcrr)rmrYrs r3 test_trapzzTestTrapezoid.test_trapzsg KK1b !  # #*E F .[[__Q1-F .FEMM$5$5a$89 . .s "BB r)rcrrkrrrrrcd\}}ttj|d||}tjd|d5ttj |d||}ddd|k(sJy#1swYxYw)N)rKrr z`trapz.z` is deprecatedr)r(r9rrr0 rB)rmrrr# rRrs r3test_trapz_deprecationz$TestTrapezoid.test_trapz_deprecations1375??F3Aq!<  # #F8?3  ;2WU[[&1!Q:F ;!!!  ; ;s #A55A>N) rrrr r r r$ r. r1 rrrr3 r;r5r3r r sP;P E(M6?4: [[X(EF"F"r5r ceZdZdZy) TestTriangcJtjd5ttjj dddttjj dddttjj dddttjj dddttjj dddttjj ddd ttjj dddttjj ddd ttjj dddttjj dddttjj ddd ttjj ddddddy#1swYyxYw) NraisermrrzrvrNr r{rarrf)rdrrr9r rcrkrrs r3test_edge_caseszTestTriang.test_edge_casessg [[W % 6 ))!Q/ 4 ))#q12 6 ))!Q/ 4 ))!Q/ 3 ))#q12 6 ))!Q/ 3 ))"b12 6 ))#r2D 9 ))#r2C 8 ))"b12 6 ))#r2D 9 ))"b11 5 6 6 6s G9HH"N)rrrr8 r;r5r3r5 r5 s6r5r5 ceZdZejj dddgdZejj dgddZdZd Z y ) TestMaxwellr)rgCR5&.)rxg7q?cZttjj||dyr)rr9maxwellrr3s r3rzTestMaxwell.test_sf s  ((+Su=r5rN))rg1!@)g?g?)gr5cbd}tjjd}t||dy)Ngi6urrr)r9r< rr)rmrrs r3rzTestMaxwell.test_logcdf0s'%%%a(%0r5cbd}tjjd}t||dy)Ng_OԺr&rr)r9r< rr)rmrrs r3rzTestMaxwell.test_logsf6s'% ##D)s/r5N) rrrrrrrrrrr;r5r3r: r: sf  [[X::<=>=>  [[X:;? ;?1 0r5r: ceZdZdZdZy) TestMielkecd\}}ttj||jdtj ttj|djdtj t t jtj|djdy)N)g rh@gv?r rNr)rr9mielkerrdrr r>)rmrVrHs r3r$zTestMielke.test_moments>sz1U\\!Q'..q1266:U\\!S)003RVV< ELLD188;<=r5ctjddd}d\}}ttjj ||||z tj j |||y)Nrxr[r7)g@gHzG@)rdrSrr9burrrcrC )rmrYrVrHs r3test_burr_equivalencez TestMielke.test_burr_equivalenceEsO KKc2 &1 q!QqS15<<3C3CAq!3LMr5N)rrrr$rF r;r5r3rA rA =s >Nr5rA c$eZdZdZdZdZdZy)TestBurrc tjddgtjddgtjddgtjddgtjddgtjddgtjd d gg}|Dcgc]#\}}}|j|j g|%}}}}|Dcgc]\}}}| }}}}t |||Dcgc]#\}}}|j|j g|%}}}}|Dcgc]\}}}tj|}}}}t ||ycc}}}wcc}}}wcc}}}wcc}}}w) N)r r )rvra)r r )rarvrrvrN)r rarz) r9fiskrE burr12rcr1rrrdr)rmrrrrrrrs r3test_endpoints_7491zTestBurr.test_endpoints_7491Ls7ZZq ! ZZ1 % ZZ # ZZ1 % \\8S ) \\63 ' \\63 ')9=== E1vrvvbdd#U#==9=>>!5UI9>>!#w/;?@@<2uayryy&&@@AEFF)=UI266)$FF!#w/ >>AFs(D?9 E(E  Ecd\}}tj||j\}}d\}}t||t||y)N)rOr)g4U?g&MI?)r9rE r)rmrr# rvariancemean_hc variance_hcs r3test_burr_stats_9544zTestBurr.test_burr_stats_9544`sI1Aq)//1h Gg&+.r5c<d\}}tj||j\}}ttj|ttj|d\}}tj||j\}}ttj |ttj|d\}}tjj tjgd||\}}}}ttj|ttj|ttj|ttj|d\}}tjj gd||\}}}}ttj |ttj|ttj|ttj|d\}}tjj gd||\}}}}ttj |ttj |ttj|ttj|d\}}tjj gd||\}}}}ttj |ttj |ttj |ttj|d\}}tjj gd||\}}}}ttj |ttj |ttj |ttj |y)N)rvr)rrr)r}r)r2r)r0r)r9rE r rdrOr>rr) rmrr# rrN e1e2e3e4s r3test_burr_nan_mean_var_9544z$TestBurr.test_burr_nan_mean_var_9544ks1Aq)//1h"#1Aq)//1h D!""#1))"((<*@!QGBB    1)),1=BB B    1)),1=BB B  B   1)),1=BB B  B  B  1)),1=BB B  B  B  B r5cxd\}}gd}gd}ttjj||||dy)N)rOr)rWrrr0)g+&?gI$_@gKT@gؗVArr)rr9rE r)rmrr# ryrs r3 test_burr_isfzTestBurr.test_burr_isfs61 &# q!Q/5Ar5N)rrrrL rQ rW rY r;r5r3rH rH Ks0( /(!TBr5rH ceZdZejj dddgdZdZejj dgddZy ) TestBurr12zscale, expected)rNgd~)r2gR톇T chtjjdddd|}t||dy)NgjAgjArrrrr)r9rK rr)rmr`rRrls r3rzTestBurr12.test_delta_cdfs/( ''S!Qe'Dxe4r5c0d\}}tjdz }dtjdzdz z }tjdzd|dzzz }tj}||||g}tj||jd }t ||d y) Nr_rr rarmrr[ rrrr)rdrerr9rK r) rmrr# rr8r9rrrs r3test_moments_edgezTestBurr12.test_moments_edges 1uuQw"%%(2+ouuaxCH%66S$)ll1a &&v.Su-r5z p, c, d, ref))rrrvggx4/@)gҶOɃ;rrvgI8aS@)rrf#gj\/@)g#aMR/rfr_ g!GAcbtjj|||}t||dyr)r9rK rr)rmrprr# rrYs r3test_isf_near_zerozTestBurr12.test_isf_near_zeros( LL  Q1 %3U+r5N) rrrrrrrr^ ra r;r5r3r[ r[ sa [[.<;=>5>5( .  [[ 0,,r5r[ c eZdZgdZgdZgdZejeeefZgdZ ddddd ejgZ gd Z e eee e e eZd d ddejd fddejdfddejdfgZej&j(dZej&j,dZej2j5eZdZeej2j?ee5Z e!jDe Z#dddej&jIde#ddZ%ej&jIde#ddZ&ej&j,ej&jOdej&jIde#ddZ(ej&j(dZ)ej&j,dZ*ej&jIded Z+ej&j,ej&jOdd!Z,ej&j,d"Z-d#Z.d$Z/d%Z0y#1swYjxYw)&TestStudentizedRange)gQ1@g33333F@g ףp=*K@gHzM@g/$@g~j!@g33333$@g{Gz&@gh|?5 @gQ8@g&1@g|?5@g@gOn@g!rhm@gB`"@rhgʡEs@g|?5^@g/$@gV-@g/$@gF@g?5^I @)gRV@gffffffl@gp@gr@gʡ @gHzG/@gQ82@gQ3@g!rh@g@g +@gʡEs @gjt@gB`"[@g@gnJ@gV- @gFx@gsh|?@g+N@g/$ @g @g@gGz@)gfffff"@ii i gHzG2@g(\A@gQC@gfffffE@g @gNbX9"@gHz$@g(\&@g-@gn@@gw/@g= ףp @gJ +@g-'@gn@gHz@gV-@gnJ@g$@g/ݤ@)r4rrr rrZrx)rarr%r)rWri)#gupm1f?)r rZrg?<-=A?gv"?rgd暕?ga-A?c|jD]:\}}|\}}}tjj|||}t ||d<y)Nrr)rr9studentized_rangerkr)rmpvkry p_expectedrrVres_ps r3test_cdf_against_tablesz,TestStudentizedRange.test_cdf_against_tables sKii :FC" J1++//1a8E E:D 9 :r5c|jD]:\}}|\}}}tjj|||}t ||d<y)Nrr)rr9rf rr)rmrg q_expectedrprrVres_qs r3test_ppf_against_tablesz,TestStudentizedRange.test_ppf_against_tables sJ#yy :OCGAq!++//1a8E E:D 9 :r5z&data/studentized_range_mpmath_ref.jsonN case_resultcdf_datac|d}|d}|d|d|df}tjj|}t|||d|dy Nsrc_case mp_resultryrVr expected_atol expected_rtolrh)r9rf rkrrmro rs rt qkvrs r3test_cdf_against_mpz(TestStudentizedRange.test_cdf_against_mp `z* , smXc]HSM9%%))3/Y%o6%o6 8r5pdf_datac|d}|d}|d|d|df}tjj|}t|||d|dyrr )r9rf rcrrw s r3test_pdf_against_mpz(TestStudentizedRange.test_pdf_against_mp rz r5z+intermittent RuntimeWarning: invalid value. moment_datac|d}|d}|d|d|df}tjd5tjj|}dddt ||d|d  y#1swYxYw) Nrs rt rrVrrrru rv rh)rdrr9rf rr)rmro rs rt mkvrs r3test_moment_against_mpz+TestStudentizedRange.test_moment_against_mp) sz* , smXc]HSM9[[ * 7))00#6C 7 Y%o6%o6 8 7 7s A**A3cd\}}ttjjdtj ||f}t |ddy)NrrZrrUr )rr9rf rcrdrr)rmrVrrs r3test_pdf_integrationz)TestStudentizedRange.test_pdf_integration: s=15**..266AGA"r5cd\}}tjddd}tjj |||dd}tjj |||}t ||}t||dy) Nr rrZrx)stepr rr)rdrr9rf rkrcrr)rmrVrrYy_cdf y_pdf_rawy_pdf_cumulatives r3test_pdf_against_cdfz)TestStudentizedRange.test_pdf_against_cdfA sv1 IIa$ '''++Aq!4QR8++//1a8 / 1= (%d;r5 r_case_resultc|\}}}}tjd5tjj |||}dddt |y#1swYxYw)Nrr)rdrr9rf rkr)rmr ryrVrr_resrs r3test_cdf_against_rz'TestStudentizedRange.test_cdf_against_rQ sV'1a [[ * 7))--aA6C 7U# 7 7s "AAc~tjd5tjj ddgddgddg}dddt j d tjtd 5tjj dddggd dddy#1swYixYw#1swYyxYw) Nrrr rarrrZrr z...could not be broadcast...r)rZrrP) rdrr9rf rrr<rrrrs r3test_moment_vectorizationz.TestStudentizedRange.test_moment_vectorizationY s[[ * H''--q!fq!fr2hGA H & ]]:-K L C  # # ) )!aV\ B C C  H H  C Cs(B'8&B3'B03B<c`t5}tjd5|jtt j jgd\}}}}ddddddtt j jy#1swY;xYw#1swY?xYw)Nrrr) r rdrrLrr9rf rkr _argcheck)rmrVrVrrs r3test_fitstart_validz(TestStudentizedRange.test_fitstart_validi s   GCX)F G JJ) *11;;IFKAr1a G G ''11!R89  G G G Gs"B$ Zd>%%))!R8,,00B> Zd>r5c0tjjddd}tjjddd}tjjddd}ttt ||ddt ||ddtjj ddd}tjj ddd}tjj ddd}ttt ||ddt ||ddy)NrrZrr irprh)r9rf rcrXAssertionErrorrrk)rmrr res_sanitys r3test_df_cutoffz#TestStudentizedRange.test_df_cutoff| s%%))!R8,,00B> ,,00B> nosJd , JTE%%))!R8,,00B> ,,00B> nosJd , JTEr5c|d\}}}tjj|||}t|dd|dk\sJy)N)gaRA@riSrrr)r9rf rr)rmryrVrrps r3 test_clippingz"TestStudentizedRange.test_clipping s@.1a  # # & &q!Q /15)Av vr5)1rrrq05q01q001rdroqspsrvskslistrr(rr_datarrrrj rrn ospathdirnamer path_prefix relative_pathopenrLfilejsonrpregenerated_datarry r} xfail_on_32bitr r r r r r r r r r;r5r3rc rc s 'C  'C  (D c4( )B B QBRVV $B B GBB', -D *( Arvv() Arvv() B)* F [[::  [[:: ''//(+Kr)rmrrYrps r3test_pdf_nonpositive_lambdaz+TestTukeyLambda.test_pdf_nonpositive_lambda sZ KKc3 '    ! !!S ){{1~!!###C}}r5ctjddd}tjdgdgdgg}tjj ||}tj |jsJ|dddkDjsJ|ddkDjsJ|ddk(jsJy)NrrOrXrr{rzra) rdrSrr9r+rcr>rr)rmrYrrps r3test_pdf_mixed_lambdaz%TestTukeyLambda.test_pdf_mixed_lambda s KKc3 'hhu-.    ! !!S ){{1~!!###"1   """!s !!!!   """r5c2tjgd}tjj |\}}tjtj tj tj dddg}t ||t || y)N)grr{rfrvrzrrarv)rdrr9r+supportrr)rmrr1r2 expected_bs r3r+zTestTukeyLambda.test_support shhh9:  ((-1XXrvvrvvrvvq!SAB Q #Q $r5cdtjjddgd}t|ddgy)Nrrzrvr{)r9r+rcrrs r3test_pdf_support_boundaryz)TestTukeyLambda.test_pdf_support_boundary s.    ! !4+s 3Qc #r5cbtjjdd}dtjdzdz ddg}t ||dtjjd d}gd }t ||dtjjd d}gd }t ||dy) NrrrrarrZrZrQg ףp= @)rgz5v?rgRTgQ?)rgNg@rgJFٻ)r9r+rdrer)rmrrRs r3"test_tukeylambda_stats_ticket_1545z2TestTukeyLambda.test_tukeylambda_stats_ticket_1545 s   $ $Q $ 7ruuaxz1c*B"5    $ $T6 $ :GB"5    $ $T6 $ :GB"5r5N) rrrrrrrdrr r r+r r r;r5r3r r s^ [[ dD("((TFSED6#:;<  #%$6r5r cleZdZdZdZej jdgddZdZ dZ y) TestLevyctjgd}tjgd}tjj |}t ||dtjj |}t ||dy)N)rrNrvrWrxr)g~V1?r,gba"?g܄4ťY?r-g$%+Irrr)rdrr9r rkrr)rmrYrRr)rs r3test_levy_cdf_ppfzTestLevy.test_levy_cdf_ppf sf HH7 88856 JJNN1 8%0ZZ^^H %AE*r5ctjgd}tjgd}tjj |}t ||dy)N)rrgMraB3Gr8)gep[>g}XEQ=gѭEG,,RVVS9:;..rvvs;<=++BFFC89:,,RVVS9:;//<=>,,RVVS9:;,,RVVS9:;#<<A H %9%--++AB+789a0 XXemm''#1'= >Fa0 XXemm''d': ;Fa0 XXemm''d'; H!4)99s 95IIctjjdtjj d}tjj |}d}t||dy)Ni r[r5)g ףp= ?gʡE?r rQ)rdrrr9rrrr)rrrRs r3test_regression_ticket_1530r f!sNIINN6 ,,    $C \\  c "FH!4r5ctjjdtjjd}dD]C}tj j ||z\}}t||dt|ddEy)Nrr)rg6ArNrr})rdrrr*r9rrr)rYoffsetr_r`s r3test_gh_pr_4806r o!sgIINN4 A'.\\%%a&j1 UV#.s-.r5c|ttjtjj ddy)Nr?rK)r rdr>r9r~r4r;r5r3test_poisson_logpmf_ticket_1436r y!s$ BKK ,,T37 89r5czddg}|D]2\}}tjj|d}t||4y)aTest the powerlaw stats function. This unit test is also a regression test for ticket 1548. The exact values are: mean: mu = a / (a + 1) variance: sigma**2 = a / ((a + 2) * (a + 1) ** 2) skewness: One formula (see https://en.wikipedia.org/wiki/Skewness) is gamma_1 = (E[X**3] - 3*mu*E[X**2] + 2*mu**3) / sigma**3 A short calculation shows that E[X**k] is a / (a + k), so gamma_1 can be implemented as n = a/(a+3) - 3*(a/(a+1))*a/(a+2) + 2*(a/(a+1))**3 d = sqrt(a/((a+2)*(a+1)**2)) ** 3 gamma_1 = n/d Either by simplifying, or by a direct calculation of mu_3 / sigma**3, one gets the more concise formula: gamma_1 = -2.0 * ((a - 1) / (a + 3)) * sqrt((a + 2) / a) kurtosis: (See https://en.wikipedia.org/wiki/Kurtosis) The excess kurtosis is gamma_2 = mu_4 / sigma**4 - 3 A bit of calculus and algebra (sympy helps) shows that mu_4 = 3*a*(3*a**2 - a + 2) / ((a+1)**4 * (a+2) * (a+3) * (a+4)) so gamma_2 = 3*(3*a**2 - a + 2) * (a+2) / (a*(a+3)*(a+4)) - 3 which can be rearranged to gamma_2 = 6 * (a**3 - a**2 - 6*a + 2) / (a*(a+3)*(a+4)) )rN)rvrr{r)rz)gUUUUUU?r g^cQg333333rrN)r9rr)rQr1 exact_mvskrs r3test_powerlaw_statsr }!sG>, = ?E4 :~~##Av#6!$ 34r5c\tjjdd}t|dy)Nrr r{)r9rrrrps r3test_powerlaw_edger !s" a#ACr5cvtjjdd}t|dtjj dgd}t|t j ddgtjjdgd}t|t j dt j gy)Nrr r{)rfrNrrN)r9r+rrrcrdrr s r3test_exponpow_edger !s a#AC 1./ARVVS#&' a!12ARVVS266'*+r5c^tjjddd}t|dy)Nrr rN)r9gengammarcrr s r3test_gengamma_edger !s$ 1a#ACr5za, c, ref, tol))g`6Ar g2j!@r~)rr ggA@r~)rr rr~)rr g#@r)g`FArgI~mr~)gl$W}Lrg.I#Ur~c\ttjj||||yr)rr9r rA)r1rrrNs r3test_gengamma_extreme_entropyr !s!"ENN**1a0#C@r5ctjjddd}|dk(sJtjjddd}|tj k(sJy)Nrr rr{)r9r rcrrdr)rprs r3!test_gengamma_endpoint_with_neg_cr !sO 1a$A 8O8 >> Ar *D BFF7??r5ctjjddd}t|dtjjddd}t|dy)NrrKrNg>rZgqq?)r9r rrr s r3test_gengamma_munpr !sF Rb)A:& RR(A6"r5c\tjgd}tjd5t5}|j t d|j t dtjj|ddddddy#1swYxYw#1swYyxYw)N)0g>X*ȿgPix#?g=?g&^?goϿg뵏˿gO_ZQ?g\?g~?g)/?gRܶ?g핚οgH߃[g?g S"gm_ пg7k?g#BF?g KU= ?gT?gZg[Qs?g2?g6Hwƭ8?g*- ?ga]?g'?g޻?g?J_?g~sw?g0FfͿgg"@ӿg?%:˿g&zM?g?Gg F?g~,g HgW2ҿgTzy?gwm ǿg@gW=ֿg 't?g".v1glg.Hꮿg,?g*˿rrz:The maximum number of subdivisions .50. has been achieved.z-floating point number truncated to an integer) rdrrr rLrrtr9r@r)r# rVs r3test_ksone_fit_freezer !s  / 0A X &   C JJ)# $ JJ~F H KKOOA     s# B"A BB"B B""B+c Ttjttddd }gd}t t j j||dt t j j|dzj|dt j j|dzd z j}tjt j j|t j j|z }t ||d y) Nrrd r)gDB.g8_$g%`OAgjhwHRgWMB@v`g7VY}igۦx\Arg xg#g/dg3$gegd)gNq{3g يGǓgr 4g37' gfFg?dJgضg2, gm%gӅJgz;pg ɿ{gk2gij<%giVSͶgBR4g Q/Mr&ry+=y|=r) rdrr rrr9rrrealimagrr)rYrRderivderiv_expecteds r3test_norm_logcdfr !s Dq#q)* ++A0HEJJL''*H4@EJJL''F 388(NZZ  q6z *5 0 6 6EVVEJJ--a05::3D3DQ3GGHNE>6r5ctjgd}tjgd}tjj |}t ||dy)N)gMb{Gzg{Gztg~jtX)g#]g~6UZZ^^Aryyc':A>B""1biiS&91=>  BKK  # # %&  "Mg~6UZZ^^E1b115 ))%B79KLMM>>MMsBE&,A1E2&E/2E;zmethod, expectedrkg u%>g ;Q=rcggRz\R>rgʎK$/gԆX1rgfHu@ge:@cjttj|dddgd}t||dy)NrWrrrZ)rw rr~r)r(r9r r)rrRrs r3test_ncx2_zero_ncr t"s/ )WUZZ (!QB ?FFH51r5ctjjddd}tjjdd}t ||dy)NrZrr )rrw r)rrr~r)r9r rrr)rrRs r3test_ncx2_zero_nc_rvsr "s>ZZ^^raa^ 8Fzz~~!~4HFH51r5cdtjddz}ttjj dd|dy)NrZrrr rrw r)rdrrr9r rk)rw s r3test_ncx2_gh12731r "s4 RYYq"  Bb115r5ctjgd}d\}}tjj |||}gd}t ||dy)N) gю] @gj%@gb y/7@gH@gIZ@g Xl@g'/2?8@gݰmА@gU @gy]`'@g@g*\@g48c@)rgh?8@r ) rNrNrNrNrNg?gy[qD?r{r{r{r{r{r{rr)rdrr9r rr)rYnurr sf_expecteds r3test_ncx2_gh8665r "sM " #A %GB qRC (B'K B %0r5c d}d}tjtjj d||tjj d||d}tjj |||}tj j |||ztjd|zd|zz}t||d y) Ni,irrrr rarrr) rdrSr9r rrcrrr)rrw rYncx2_pdf gauss_approxs r3test_ncx2_gh11777r "s B B EJJNN5"b1JJNN5"b1u >Azz~~aR(H::>>!R"Wbgga"fq2vo.FGLHl6r5zx, c, expected))rar g|1q?)rarag~?)rfr V<3T1=)7yQCr u lY<)rfgvHGBg^b!*1=)rsrKgS?c^tjj||}t||dyNry)r9 foldcauchyrrrYrrRrs r3test_foldcauchy_sfr "s(     Q "BB%(r5r))rag?)rfr )r r )g6ްPg8R#/c\tjj|}t||dyr )r9rFrr)rYrRrs r3test_halfcauchy_sfr "s&     Q BB%(r5z p, expected))g?g*wӄZ>)333333?gN9?)rvrN)rxg8<,O@)rg:B)gl ׶w/gMb}[PcZtjj|}t||yrp)r9rFrr)rprRrYs r3test_halfcauchy_isfr" "s$ QAAx r5chtjdd}t|jddy)Nrr r)r9r(rrk)rs r3test_foldnorm_zeror$ "s$  #BAr5))rar gwo {?)rr gGd-P/)rZr9gof?)rr9g Oul";c^tjj||}t||dy)Nr)r9r(rrr s r3test_foldnorm_sfr& #s&   1a BB%(r5ctjjgddd}tjjddgdd}td|D}t||tjjgd}tjjddg}td |D}t||y) N)r{rvrNr rvrNc3bK|]'}tjtj|f)ywrprdrHrrS s r3rkz-test_stats_shapes_argcheck..#s!8q"%% *8-/)rar4rrar4c3bK|]'}tj|tjf)ywrpr) rS s r3rkz-test_stats_shapes_argcheck..#s!8q"%%266 *8r* )r9rtuplerr")mv3mv2 mv2_augmenteds r3test_stats_shapes_argcheckr0 #s ..  3 7C ..  Sz1c 2C8C88M$ --  l +C --  q#h 'C8C88M$r5ceZdZdZy) _distr_gencyNrr;rmrYr1s r3_pdfz_distr_gen._pdf(#sr5Nrrrr6 r;r5r3r2 r2 '#sr5r2 ceZdZdZy) _distr2_gencd|z|zSr4 r;r5 s r3_cdfz_distr2_gen._cdf-#Avzr5N)rrrr; r;r5r3r9 r9 ,#sr5r9 ceZdZdZdZy) _distr3_genc ||zSrpr;rmrYr1r2s r3r6 z_distr3_gen._pdf2#s 1u r5cd|z|zSr4 r;r5 s r3r; z_distr3_gen._cdf5#sAvzr5Nrrrr6 r; r;r5r3r> r> 1#s r5r> ceZdZdZdZy) _distr6_genc||z|zSrpr;r@ s r3r6 z_distr6_gen._pdf=#ssQwr5cd|z|zSr4 r;r@ s r3r; z_distr6_gen._cdf@#r< r5NrB r;r5r3rD rD ;#s r5rD creZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZy)TestSubclassingExplicitShapescXtdd}t|jdddy)Ndummyr1rr r rirr2 rrcrm dummy_distrs r3test_correct_shapesz1TestSubclassingExplicitShapes.test_correct_shapesG#s% gc: [__Q!_,b1r5cjtdd}tt|jdfit dy)NrJ ArK r ri)r2 rXrrcr rM s r3test_wrong_shapes_1z1TestSubclassingExplicitShapes.test_wrong_shapes_1K#s' gc: i!AtayAr5crtdd}tddd}tt|jdfi|y)NrJ za, b, crK r rarr)r2 r rXrrc)rmrN r)s r3test_wrong_shapes_2z1TestSubclassingExplicitShapes.test_wrong_shapes_2O#s1 gi@ Q!q!i!;s;r5cHtdd}tttfi|y)NrJ rrK )r rXrr2 rmr)s r3test_shapes_stringz0TestSubclassingExplicitShapes.test_shapes_stringT#s+i3s3r5cHtdd}tttfi|y)NrJ z(!)rK r rX SyntaxErrorr2 rV s r3test_shapes_identifiers_1z7TestSubclassingExplicitShapes.test_shapes_identifiers_1Y#s.k:55r5cHtdd}tttfi|y)NrJ 4chanrK rY rV s r3test_shapes_identifiers_2z7TestSubclassingExplicitShapes.test_shapes_identifiers_2^#0k:55r5cHtdd}tttfi|y)NrJ zm(fti)rK rY rV s r3test_shapes_identifiers_3z7TestSubclassingExplicitShapes.test_shapes_identifiers_3b#s1k:55r5cHtdd}tttfi|y)NrJ za=2rK rY rV s r3"test_shapes_identifiers_nodefaultsz@TestSubclassingExplicitShapes.test_shapes_identifiers_nodefaultsf#s.k:55r5cHtdd}tttfi|y)NrJ z*argsrK rY rV s r3test_shapes_argsz.TestSubclassingExplicitShapes.test_shapes_argsj#r_ r5cHtdd}tttfi|y)NrJ z**kwargsrK rY rV s r3test_shapes_kwargsz0TestSubclassingExplicitShapes.test_shapes_kwargsn#s 3k:55r5cHtdd}tttfi|y)NrJ za, b, c, lambdarK rY rV s r3test_shapes_keywordsz2TestSubclassingExplicitShapes.test_shapes_keywordsr#s(9:k:55r5cGddtj}|d}t|jddtjjddzy)NceZdZdZy)FTestSubclassingExplicitShapes.test_shapes_signature.._dist_gencFtjj||zSrpr9rr6 r5 s r3r6 zKTestSubclassingExplicitShapes.test_shapes_signature.._dist_gen._pdfz#zzq)A--r5Nr7 r;r5r3 _dist_genrl y# .r5rp r1r rvrarir9r?rrcrrmrp rNs r3test_shapes_signaturez3TestSubclassingExplicitShapes.test_shapes_signaturew#sI .++ .$TXXcQX')?r5cGddtj}|d}tt|jdfit ddy) NceZdZdZy)STestSubclassingExplicitShapes.test_shapes_signature_inconsistent.._dist_gencFtjj||zSrprn r5 s r3r6 zXTestSubclassingExplicitShapes.test_shapes_signature_inconsistent.._dist_gen._pdf#ro r5Nr7 r;r5r3rp rx #rq r5rp rirr rvr rar)r9r?rXrrcr rt s r3"test_shapes_signature_inconsistentz@TestSubclassingExplicitShapes.test_shapes_signature_inconsistent#s; .++ .'i3A$a.Ar5cGddtj}|d}t|jddtjjddzt|jddtjjddzt t |jdfitdy) NceZdZdZy)?TestSubclassingExplicitShapes.test_star_args.._dist_gencP|d}tjj||zSNrrn )rmrYr* extra_kwargs r3r6 zDTestSubclassingExplicitShapes.test_star_args.._dist_gen._pdf#s#"1g zzq)K77r5Nr7 r;r5r3rp r} #s 8r5rp r rr rv!r )xxx)r9r?rrcrrXrr rt s r3test_star_argsz,TestSubclassingExplicitShapes.test_star_args#s 8++ 8  .TXXcrX2EJJNN34G4JKTXXc2& s(;B(>?i3?$2,?r5cJGddtj}|d}t|jdddtjjddzdzt|jdddtjjddzdzy) NceZdZdZy)ATestSubclassingExplicitShapes.test_star_args_2.._dist_gencV|d}tjj||z|zSr rn )rmrYr r*r s r3r6 zFTestSubclassingExplicitShapes.test_star_args_2.._dist_gen._pdf#s("1g zzq)K7&@@r5Nr7 r;r5r3rp r #s Ar5rp zoffset, extra_kwargrr rvor )r r rs rt s r3test_star_args_2z.TestSubclassingExplicitShapes.test_star_args_2#s A++ A  56TXXc#2X>ZZ^^C(+c1 3TXXc3+ZZ^^C(+c1 3r5cGddtj}|d}t|jddtjjdy)NceZdZdZy)BTestSubclassingExplicitShapes.test_extra_kwarg.._distr_gencj|jdd}tjj||zS)Nr r )popr9rr6 )rmrYr*kwargsr s r3r6 zGTestSubclassingExplicitShapes.test_extra_kwarg.._distr_gen._pdf#s,%jj: zzq)K77r5Nr7 r;r5r3r2 r #s 8r5r2 r rr r rr rs )rmr2 rNs r3test_extra_kwargz.TestSubclassingExplicitShapes.test_extra_kwarg#sD 8,, 8/TXXaQX/1BCr5cGddtj}|d}t|jdtjjdy)NceZdZdZy)ITestSubclassingExplicitShapes.test_shapes_empty_string.._dist_genc@tjj|Srp)r9rrcrs r3r6 zNTestSubclassingExplicitShapes.test_shapes_empty_string.._dist_gen._pdf#szz~~a((r5Nr7 r;r5r3rp r #s )r5rp rr rvrs rt s r3test_shapes_empty_stringz6TestSubclassingExplicitShapes.test_shapes_empty_string#s? )++ )#TXXc]EJJNN3$78r5N)rrrrO rR rT rW r[ r^ ra rc re rg ri ru rz r r r r r;r5r3rH rH D#s\2B< 4 6 666666 @B @ 3 D9r5rH ceZdZdZdZej jeddZ ej jeddZ dZ dZ d Z d Zy ) TestSubclassingNoShapescVtd}t|jdddy)NrJ rr rirrL rM s r3test_only__pdfz&TestSubclassingNoShapes.test_only__pdf#s# g. [__Q!_,b1r5cVtd}t|jdddy)NrJ r r ri)r9 rrcrM s r3test_only__cdfz&TestSubclassingNoShapes.test_only__cdf#s#!w/ KOOAO3Q7r5docstring strippedrTctd}t|jdt|jdt j d|j }tt|dk(y)NrJ r r r1zlogpdf\(x, a, loc=0, scale=1\)) r2 rnumargsr refindallrr r rmrN rs r3test_signature_inspectionz1TestSubclassingNoShapes.test_signature_inspection#sX!g. [((!,[''-jj:$,,.CA r5ctd}t|jdt|jdt j d|j }tt|dk(y)NrJ r rariz!logpdf\(x, a, b, loc=0, scale=1\)r ) rD rr r r r rr r r s r3test_signature_inspection_2argsz7TestSubclassingNoShapes.test_signature_inspection_2args#sX"w/ [((!,[''0jj=$,,.CA r5c0tttdy)NrJ r )rXrr> rrs r30test_signature_inspection_2args_incorrect_shapeszHTestSubclassingNoShapes.test_signature_inspection_2args_incorrect_shapes#si7;r5clGddtj}tt|fit dy)NceZdZddZy)>TestSubclassingNoShapes.test_defaults_raise.._dist_gencyr4 r;r5 s r3r6 zCTestSubclassingNoShapes.test_defaults_raise.._dist_gen._pdf#r5N)rr7 r;r5r3rp r #s r5rp rJ r r9r?rXrr rmrp s r3test_defaults_raisez+TestSubclassingNoShapes.test_defaults_raise#* ++  iAd.@Ar5clGddtj}tt|fit dy)NceZdZdZy)>TestSubclassingNoShapes.test_starargs_raise.._dist_gencyr4 r;)rmrYr1r*s r3r6 zCTestSubclassingNoShapes.test_starargs_raise.._dist_gen._pdf#r r5Nr7 r;r5r3rp r # r5rp rJ r r r s r3test_starargs_raisez+TestSubclassingNoShapes.test_starargs_raise#r r5clGddtj}tt|fit dy)NceZdZdZy)._dist_gencyr4 r;)rmrYr1r s r3r6 zATestSubclassingNoShapes.test_kwargs_raise.._dist_gen._pdf#r r5Nr7 r;r5r3rp r #r r5rp rJ r r r s r3test_kwargs_raisez)TestSubclassingNoShapes.test_kwargs_raise#r r5N)rrrr r rrryDOCSTRINGS_STRIPPEDr r r r r r r;r5r3r r #sy28  [[+4HIJ [[+4HIJ<BBBr5r r cgd}tjD]p}tt|}t|tjtj zs?|D]-}t tj||jdu/ry)N)z,\s*,z\(\s*,z^\s*:) r9rKr(r>r>r?r r searchr)badonesdistnamerNregexs r3rr#sp-GMM@uh' dU..1D1DD F  @ %6$>? @@r5cttjjtj dddttj jtj dddy)NrZrrrrWr )rr9r/rrdrr r; r;r5r3test_infinite_inputr $sB ((R8!< 37;r5ctjjtjjddd}t |dyr)r9lomaxrrkrr s r3test_lomax_accuracyr $s/  2A6AAvr5ctjjtjjddd}t |dyr)r9r'rrkrr s r3test_truncexpon_accuracyr $s7 U--11&!>,,Q2A6A3+r5ctjd5}tjdtjj ddtjj ddtjjddtjjtj dt|}t|ddddy#1swYyxYw)zregression test for gh-6219T)rNalwaysrvrr{N) rqrrrsr9r rkrcrrrdrr r)rjnumber_of_warnings_throwns r3 test_genextreme_give_no_warningsr $s   -3h' R# R# R# -$'F!.2333s B>D %%(Q,C  ?W\\!_ ,ruuax 7D D 3d #EVVHQJEE EC1Iu51QTE"C1Iu51QTE"r5cd}tjjd}t|d|zdzdtjjd}t||dzdtjjd}t |dtjjd d }t||d zt j d zdzdtjjd }t|d |zdzdtjjd}t|d|zdzdy)Ngox?rrar rrrrNrrZrrrrgr)r9r rArrrdr)r rjs r3test_genextreme_entropyr <$s $K   &AAq}q(u5   #AA{QU3   %AA   R 0AA{1}rvvbz1A5EB   $AAr+~)6   %AAr+~)6r5c d}tjj|d}t|dtjj |d}t||d}tjj|d}t|dtjj |d}t||d}tjj|d}t|dtjj |d}t||y) Nrrug'b%4gQ@rsgؗҜ;rg.l6?)r9r rrr)rYrHrjs r3test_genextreme_sf_isfr S$s2 A Av&AA-.     a (BB A Au%AA()     a 'BB A Aq!AA-.     a #BBr5cbd}tjj|dd}t|dy)Nr}rargH&8>)r9rK rr)probrs r3test_burr12_ppf_small_argr $s, D||a+HH45r5cd d}tjgd}tjj |d|\}}}t |dd|dk(sJt |ddy ) a? Test fitting invweibull to data. Here is a the same calculation in R: > library(evd) > library(fitdistrplus) > x = c(1, 1.25, 2, 2.5, 2.8, 3, 3.8, 4, 5, 8, 10, 12, 64, 99) > result = fitdist(x, 'frechet', control=list(reltol=1e-13), + fix.arg=list(loc=0), start=list(shape=2, scale=3)) > result Fitting of the distribution ' frechet ' by maximum likelihood Parameters: estimate Std. Error shape 1.048482 0.2261815 scale 3.099456 0.8292887 Fixed parameters: value loc 0 rc$t||||ddS)Nr)r*dispxtolftol)r&)rx0r*r s r3rz&test_invweibull_fit..optimizer$sD"4dUKKr5)r rrar}rhrrrrrrZrP@c)rrg`?r^rg [@N)r;r)rdrr9rrr)rrYrr_r`s r3test_invweibull_fitr $s_.L HIA$$((i(HMAsEAxd+ !8O8E8$/r5))rrg7Ըh?)rrgz]r>)rg"@goBڱ9)rrg$=cC;c`tjj||}t||dyr})r9rrr)rYrrRr;s r3test_invweibull_sfr $s( ""1a(HHhU3r5zp, c, expected)rvr}g9?)geKrrrrwr[cLd}tjj|||}tjj||||}t ||dtjj |||}tjj ||||}t ||dy)Nrrrrp)r9rrkrrrc)df1df2rYrw expected_cdfcalculated_cdfrcalculated_pdfs r3test_ncf_edge_caser $s B77;;q#s+LYY]]1c33NL.u=77;;q#s+LYY]]1c33NL.t@g&4I,)?)decimals)r9rrkrrdround) scipy_val check_vals r3test_ncf_cdf_spotcheckr $s6  b!R.IIIrxx A>?r5ctjddd}d|d<d}tjj|g|}|Dcgc]"}tjj|g|$}}t ||ycc}w)Nrr rr})rWrarrr )rdrSr9rrr)rYparryxiq0s r3test_ncf_ppf_issue_17026r $sm Aq#A AaD C a#A,- .b%))-- !S ! .B .Ar /s'A9c0eZdZdZdZdZdZdZdZy) TestHistogramcPtjjdtjgdd}t j ||_tjjdddd }tj|d }t j ||_ y) Nr)r rararrrrrrrrrrrrr*r*r*r*rrrrrrrbinsrNr}r{r:r7) rdrr histogramr9r@templaterr norm_template)rmr rnorm_histograms r3r1zTestHistogram.setup_method$s{ tLL"@FGI **95 zz~~#Su3~Od4"//?r5ctjgd}tjgd}t|jj ||t |jj ddt |jj ddt |jj ddt |jj ddtjd d d }t|jj |tjj |d d dy)Nr{rvrNrrzr}rr2rr0rO@rrrrr!@r r!)r{r{rGrG{Gz?r Q?r {Gz?r rKrKr r r r r r r{r{rr r r r{rrrarZrNr}r^rWr) rdrrrr rcrrSr r9r)rmrD pdf_valuesrYs r3rLzTestHistogram.test_pdf%sMNZZ!IJ  ))&1:> DMM--c2H=DMM--c2H= DMM--c2H=DMM--d3X> KKAr "**..q1 qc=C Ir5ctjgd}tjgd}t|jj ||t|jj |dd|ddtjddd}t|jj |jj ||tjddd}t|jj |jj ||tjd dd }t|jj |tjj |dd d y)Nr )r{r{r{rrGr r ףp= ?Q?{Gz?rrvr}(\?RQ?= ףp=?)\(?r rNrNrarrNr r[r{rrZr}r^rWr) rdrrrr rkrrSr r9r)rmrD cdf_valuesrYs r3rzTestHistogram.test_cdf_ppf"%s)MNZZ!MN  ))&1:> ))*Qr*:;VAb\J KKS# & ))$--*;*;A*>?C KKS# & ))$--*;*;A*>?C KKAr "**..q1 qc=C Ir5cd}|jj|d}ttj|dkdt tj|dkd|zd t tj|d kd |zd t tj|d kd |zd t tj|dkd|zd t tj|dkd|zd t tj|dkd|zd t tj|dkd|zd t tj|dkd|zd t tj|dkd|zd t tj|dkd|zd t tj|dkd|zd t tj|d kd!|zd t tj|d"kd#|zd t tj|d$kd%|zd t tj|d&kd|zd t tj|d&kd|zd ttj|d&kDdy)'Nrr rrNr{rzrGrKrr}r rr rWr2r rr r0r rOrr@r rvrr}rr rr rr rr r r r )r rrrdrgr)rmr,rCs r3rDzTestHistogram.test_rvs8%s """<RVVFSL)3/v}-x!|#Fv}-x!|#Fv}-x!|#Fv}-x!|#Fv}-x!|#Fv}-x!|#Fv}-y1}4Hv}-y1}4Hv}-y1}4Hv}-y1}4Hv}-y1}4Hv}-y1}4Hv}-y1}4Hv}-y1}4Hv}-y1}4Hv}-y1}4HRVVFSL)3/r5ctdD]L}t|jj|t j ddj |dNy)NrrNr}r@r)rrr rr9rrrzs r3r zTestHistogram.test_munpN%sNq GA D..44Q7!JJsC077: G Gr5ct|jjtjjdddy)NrNr}r^r@r)rr rAr9rrrs r3rkzTestHistogram.test_entropyS%s5**224 **s#*>T Kr5N) rrrr1rLrrDr rkr;r5r3r r $s%@ I4I,0,G Kr5r cddggd}}tj||fd}tjj |j ddgddg|j dk(sJtj||fd}tjj |j ddgd |j d k(sJd }tjt| 5tj||f}|j d k(sJ dddtj|gd f}|j dk(sJy#1swY8xYw)Nr )rr rF)rqrvrKrTg'^P?gH@z(Bin widths are not constant. Assuming...rr0) r9r@rdrrrcr?rrgrt)countsr rNrYs r3test_histogram_non_uniformr! X%s&FLDF   vtne tjjd}tjj ddd|}tjj ||\}}}}|dk(sJtjj |d|\}}}}|dk(sJy) Nr% rWr rrrrar)rdrrr9r& rr)rmrrrr1r2r_r`s r3test_fit_overridez TestLogUniform.test_fit_override%sii##K0""33"G ++//F/C1c5zz ++//Af/M1c5zzr5ctjjd}d\}}tj||}|j ddd}t |j|j|||jd}t |j|j||dtjdd z}|j|}t |dd |ddz d ||z tj|tj|z z } t |j| y) NleVi P)gN~hrrr rr5ri8rKrrZ)rdrrr9r& rrrkrrrrcrr) rmrr1r2rNrkrrYrcrs r3 test_overflowzTestLogUniform.test_overflow%sii##$7811%kk!QTk*#/5hhDh!#/5 4% %hhqkCRQR("-Aq BFF1I-. T*r5N) rrrr* rrrr, r. r;r5r3r# r# o%s54" [[Xt}5 6 +r5r# ceZdZdZej j dddgddgddggd Zej j d d d gd Zej j dgddZ ej j dgddZ ej j dgddZ ej j dgddZ ej j dgddZ ej j dgddZej j dgdd Zej j dgd!d"Zy#)$ TestArgusctjjddd}ttjdj |j dy)Nr7rErrrQ)r9argusrrrrs r3test_argus_rvs_large_chiz"TestArgus.test_argus_rvs_large_chi%s= KKOOBSsO ;EKKO002AFFHaHr5zchi, random_staterWr2 rr2ctjj|d|}tj|d|f\}}t |dkDy)Nrrr3 r@r9r3 rrCr )rmrrrYrrps r3rDzTestArgus.test_rvs%s@ KKOOCc O E||Aw01Dr5rrrpctjj|dd}tj|d\}}t |dkDy)Nrie rcdd|dzz dzz S)Nr rarr;rs r3rIz.TestArgus.test_rvs_small_chi..%sa!Q$h#->)>r5r@r8 )rmrrrrps r3test_rvs_small_chizTestArgus.test_rvs_small_chi%s= KKOOCcO ?||A>?1Dr5zchi, expected_mean))r gщi?)rZg.憃?)rwg\paP?)rxg ?)r g,6?cbtjj|d}t||dy)Nr rrr)r9r3 rr)rmr expected_meanrs r3rzTestArgus.test_mean%s) KK  S  *=u5r5zchi, expected_var, rtol))r gLH'B?r)rZgo$?r)rwg =>r&)rxg2W>r&)r gb~P>r&cbtjj|d}t|||y)Nr rr)r9r3 r8r)rmr expected_varrrs r3rzTestArgus.test_var%s& KKOOCqO )r~)rgF=r)rg߹Q#B=r)rpg0ޟ)rOr}g=~L]=)rOrgD,G '=c`tjj||}t||dyr)r9r3 rr)rmrYrrRrs r3test_sf_near_1zTestArgus.test_sf_near_1&s$[[^^As #H51r5))rv)g+&?gk?gμm/?)rK)g?CL\f?g^N?g]AͭR?)rW)ga>$?g;Lf?gaW?)rx)gx?g43n?gwY?)r)go7🤎?g +-oGn?gIQ9‹Y?)r)g T?gGn?g@̋Y?)rp)gU?g{Gn?giG͋Y?)r)gDU?g{Gn?gH͋Y?ctjgd}ttjj |||dy)NrC rr)rdrrr9r3 rkrD s r3test_cdf_small_chizTestArgus.test_cdf_small_chi &rF r5))rv)g Q)?g6{?r)gB`"۹?)g?g8%?r)rW)g388?gV%?r)rx)g*j?gRS%?r)r)g! |?gZ%?r)r)gy[|?gG%?r)rp)g"3|?gW%?r)r)g!3|?gW%?rcbtjj|d}t|||y)Nrrr)r9r3 r)rmrrRrrCs r3test_stats_small_chizTestArgus.test_stats_small_chi&s)kkT2XD1r5N)rrrr4 rrrrDr; rrrA rE rH rK rM rO r;r5r3r0 r0 %sI  [[0 #J #J #J3      [[UT4L12 [[189 6 9 6  [[6BC 4 C 4  [[2DE>E> [[ M NG NG  [[ N OF OF  [[ 67 2 7 2  [[ P QG QG  [[ C D2 D2r5r0 ceZdZdZdZdZdZejjdgddZ ejjdgdd Z d Z ejjd gd d Z ejjdejjdgdejjdgdejjdgddZejjdgdejjdgddZy) TestNakagamicdd}d}tjj||}t|dy)Nr}rgM+)r9nakagamirr)rmr rYrs r3rzTestNakagami.test_logpdf+&s/ ~~$$Q+12r5cd}d}tjj||}t|ddtjj ||}t||dy)Nr}rOg"S+:rr)r9rS rrr)rmr r rris r3rzTestNakagami.test_sf_isf<&sS  ^^  r2 &1> ^^  B 'BU+r5cld}d}d}tjj||}t||dy)NrrvgU iּrr)r9rS rr)rmrYr rrs r3rzTestNakagami.test_logcdfO&s3  %&&q"-%0r5cld}d}d}tjj||}t||dy)Nr@rPgD&`Urr)r9rS rr)rmrYr rrs r3rzTestNakagami.test_logsfW&s3  %$$Q+s/r5zm, ref))rg|Bd븿)rvr)rZg/ۿcZttjj||dy)Ng^~z=rrr9rS rArmrrs r3rkzTestNakagami.test_entropy_&s ..q13WEr5))rg}Ô%I)rgq_)gScAgB4U)gcAgu1U)rgb >%)rgWҙ\cVttjj||yrprX rY s r3test_extreme_nuzTestNakagami.test_extreme_nuq&s ..q137r5ctjtjj dsJtjtjj dsJy)Nrr)rdr>r9rS _entropyrrs r3test_entropy_overflowz"TestNakagami.test_entropy_overflowx&sD{{5>>2259:::{{5>>226:;;;r5znu, ref))rg2H?)rg:U?)rg>cZttjj||dyr)rr9rS r)rmr rs r3rzTestNakagami.test_mean|&s ++B/5Ar5z+Fit of nakagami not reliable, see gh-10908.rTr )rr}rr_)ryrZr_ r`)rrrc d tjj |||d tjj \}}}t ||dt ||dt ||d fd} fd} fd} t ||||d d t ||||d d t | |||d d y) Nr[9rr r_r`rrKrcd|zdztjd|z z zd|z|dzz tj|z zzS)Nrr rardrg)r r_r`rHs r3 dlogl_dnuz(TestNakagami.test_fit..dlogl_dnu&sPUQY"&&GcM):";;d5!8mbffWs]&;;< =r5cdtj|ztd|z zdtjtj|z |z zztj|z |z dzz S)Nr rra)rdrrrgr r_r`r,rHs r3 dlogl_dlocz)TestNakagami.test_fit..dlogl_dloc&spRVVBZ)Ar*::;rvvw}&=>??@ffw}59:; .dlogl_dscale&sF!GbL5("fuz)BFFGcMa3G,HHI Jr5rrr)r9rS rrr) rmr r_r`nu_estloc_est scale_estre rh rj r,rHs @@r3rpzTestNakagami.test_fit&s ..$$!+0t%E%*^^%7%7%@"-3/ 5s3 = <  J  &'9=qtL 67I>M VWi@!$Or5cxd}d}tjj||||d}tjj||\}}}t j |} t j t j||z dz} t||dt|| dt|| dy) Nrvr[ra rb rrar'r) r9rS rrrdrrrr) rmr_r`r rorHrk rl rm loc_theo scale_theos r3 test_fit_nuzTestNakagami.test_fit_nu&s  ..$$!+0t%E%*^^%7%7B%7%G"66'?WWRWWg&7A%=>? .5 :D9r5N)rrrrrrrrrrrkr[ r^ rr{rprq r;r5r3rQ rQ )&sS3",&10 [[X %& F & F [[X DE8 E8< [[Y>?B ?B [[KL [[T?3 [[UN3 [[Wk2P344MP8 [[UN3 [[Wk2:34:r5rQ ceZdZdZdZdZy)TestWrapCauchyctjddgddgg}tjdgdgg}tjj ||}|j dk(sJtj ||fDcgc]%\}}tjj ||'}}}t|j|d ycc}}w) NQ?rfrvrrNrr_rr) rdrr9 wrapcauchyrkr<nditerrr( )rmrrYrprir, scalar_valuess r3test_cdf_shape_broadcastingz*TestWrapCauchy.test_cdf_shape_broadcasting&s HHtTlS$K0 1 HHsecU^ $    A &ww&   )+Aq6):<%b"))--b"5< < =u=?!a"))Brvvbeebdl/C,C"DRUU"JJKr5N)rrrry r{ rVr;r5r3rs rs &s >,Lr5rs cGddtj}|d}ttd5|j dddy#1swYyxYw)NceZdZdZy)/test_rvs_no_size_error..rvs_no_size_gencyrr;rrs r3_rvsz4test_rvs_no_size_error..rvs_no_size_gen._rvs&sr5N)rrrr r;r5r3rvs_no_size_genr &s r5r rvs_no_sizer z_rvs\(\) got (an|\d) unexpectedr)r9r?rXrr)r r s r3test_rvs_no_size_errorr &sJ%--"}5K y(J Ks AAzdistname, argsct|tvrtjd|dtt|}t |tj rt|dk7rF|j|\}}t|tjt|tjd\}}|jg|||\}}t|tjt|tjy|j|\} } t| tjt| tjy)Nz6skipping test for the support method for distribution .rr) $skip_test_support_gh13294_regressionrrr(r9r>r?r r rrdr) r r*rNa0b0rra1b1r1r2s r3test_support_gh13294_regressionr &s77 $$,:Q0 1 5( #D$++, t9>!T\\4(FB RVV $ RVV $  f2t2T262BR R t||T"1QQr5ctjjgdgd\}}tjtj tj tj tj g}tjtj tj tj tj g}t||t|||j|jk(sJ|j|jk(sJtjjgg\}}tjgtjg}}t||t|||j|jk(sJ|j|jk(sJtjjgddg\}} tjdtj gz} tjdtj gz} t|| t| | |j| jk(sJ| j| jk(sJy)N)rrrr )r r r rrr) r9rr rdrrrrr<) r r ex_a0ex_b0r r ex_a1ex_b1rrex_a2ex_b2s r3,test_support_broadcasting_gh13294_regressionr &s ZZ   m >#u % + +v 55 5r5)gn! g:I"<)gQ @gE?cttjj||ttjj | |yrp)rr9cosinerkr)rYrRs r3test_cosine_cdf_sfr &'s6ELL$$Q'2ELLOOQB'2r5))rpgkM6O)rg0ӭ! )r g;'u(@cttjj||ttjj || yrp)rr9r rr)rprRs r3test_cosine_ppf_isfr .'s8 ELL$$Q'2ELL$$Q'(3r5ctjjtj tjg}t |dy)NgrB)r9r rrdrer )rs r3test_cosine_logpdf_endpointsr 7's1 <<   /Dd./r5c0tDchc]\}}t|tr|}}}tDchc]\}}| }}}||k(sJtDchc]\}}| }}}t Dchc]\}}| }}}||k(sJycc}}wcc}}wcc}}wcc}}wrp)r!r>rEr"rr)rrdiscrete_distnamesinvdiscrete_distnamescont_distnamesinvcont_distnamess r3test_distr_params_listsr >'s/;474'c2441@AgdATAA !6 66 6*23wtQd3N3-89'$99 . .. .4A49sB B B ) Bctjjddtjjdddk(sJtjj dddk(sJy)NrrVrkr)rr1r)r9r_statsrrr;r5r3test_moment_order_4r L's` NNAs+ >> qA .# 55 5 >>  1 % ,, ,r5c &eZdZejdZej jdgddZej jdgddZ ej jej jdejd d ej j ejd d ej j ejddej j gdZ y)TestRelativisticBWctjttjdz }tj j |jd}|S)a&Sample data points for pdf computed with CERN's ROOT See - https://root.cern/ Uses ROOT.TMath.BreitWignerRelativistic, available in ROOT versions 6.27+ pdf calculated for Z0 Boson, W Boson, and Higgs Boson for x in `np.linspace(0, 200, 401)`. z-data/rel_breitwigner_pdf_sample_data_ROOT.npyzx,pdf,rho,gammarrrss r3ROOT_pdf_sample_dataz'TestRelativisticBW.ROOT_pdf_sample_dataa'sLww N ! ! ; < vv  /@ A r5zrho,gamma,rtol))qVEB@gj+@rI)‰FC@Gz@r)ND@_LU?rc||d|k(|d|k(z}|d|d}}t|tjj||||y)NrhorrYrcrr)rr9rel_breitwignerrc)rmr r rrrrYrcs r3test_pdf_against_ROOTz(TestRelativisticBW.test_pdf_against_ROOTt'sg$ !% (C /#G,5 7 cDK3 &&**1c*?d r5zrho, Gamma, rtol))r r r)r r r)r r g& .>cd}tjddd}tjj |||}tjj |||}||||z|}t |||y)Nc tj|dz|dz|dzzz}dtjdz|z|z|ztjtj|dz|zzz }||dz|dzz dz|dz|dzzzz Srm)rdrre)ErGammarrVs r3rczFTestRelativisticBW.test_pdf_against_simple_implementation..pdf'sGGAqDAqD5!8O45ERWWQZ!#e+e3EEBGGAqD5L113AA1q(1a4q=89 9r5r@r4rZrr)rdrSr9r rrcr) rmr r rrcrprYrrs r3&test_pdf_against_simple_implementationz9TestRelativisticBW.test_pdf_against_simple_implementation'ss : KKdB '  ! ! % %aE % :##''3e'<!SY&St,r5z rho,gammar r rr r r r cd}tjj|}tjj ||d|}tjj |d}t|d|df||fd|d dk(sJtjj |d| }t|d|d |d |dfd|fk(sJy ) zpTests fit for cases where floc is set. `rel_breitwigner` has special handling for these cases. ls ; rl#r)r`rrrrrarKrr r;rxN)rdrrr9r rrr)rmr rrrrrs r3 test_fit_flocz TestRelativisticBW.test_fit_floc's&#ii##D)$$(( u4c) ##''1'5QQ(3,TB1v{{##''1U'CA$/AAAu:---r5N)rrrrrr rrr r rrrr r;r5r3r r `'s  ^^$ [[    [[/2  -  - [[ [[ FLL"F&++2B2B  FLL!5 0A0A  FLL & 0A0A    . .r5r cTeZdZejj dddgdZy) TestJohnsonSUr)r rrrg={?g?5mV>gn?g\}\1@)g`fo@g$ں _@rr gvUg Nv>T!"XE2r5N)rrrrrrr r;r5r3r r 's9 [[V O J&KL 3 L 3r5r c eZdZdZej j dddgej j dddgej j dddgej j dddgdZy ) TestTruncParetoc.d\}}tjdd}tj||j |}tj |j |tj |j |z }t||y)N)?333333@r r )rdrSr9 truncparetorcrrkr)rmr2rrYrrs r3rLzTestTruncPareto.test_pdf'sr1 KKS !1%))!,ll1o!!!$u||A':':1'==S!r5rTFr7rrctjjd}d\}}}} tj|||| } | j d|} i} |r|| d<|r| | d<|r|| d<|r|| d <|rM|rK|rI|rGd } t jt| 5tjj| fi| dddyttj| fi| y#1swYyxYw) NlZ #Lm)r r r r}r^rrrrrrrr) rdrrr9r rrrrrr)rmrr7rrrr2rr_r`rNrrrYs r3rpzTestTruncPareto.test_fit's ii##$78+1c5  A3e<xxSsx3 DL "DN DJ DJ yUuKG|7; 4!!%%d3d3 4 4 *%*;*;T JT J 4 4s !CC%N)rrrrLrrrrpr;r5r3r r 's" [[Yu 6 [[[4-8 [[WtUm4 [[WtUm4K5597Kr5r ceZdZdZy) TestKappa3cdtjjddz }tjjdd}t ||y)Nr rvgj@)r9kappa3rkrr)rmsf0rqs r3rzTestKappa3.test_sf's=%,,""3,,llooc3'S!r5N)rrrrr;r5r3r r 's"r5r ceZdZejddZej dZej dZdZ dZ dZ dZ dZ d Zd Zd Zd Zy ) TestIrwinHallrr rZcNt|jjddy)Nr)rg?rgQ)r ih10r9rrs r3test_stats_ih10zTestIrwinHall.test_stats_ih10's TYY__V46JKr5cgd}tt|Dcgc] }|jj|dz"}}t |||jjd}d}t ||ycc}w)N) rgUUUUU9@g0a@igx@gn۶mˠ@g:PAg8)AgSAr r7g2⍚H)rr r rr )rmrBrorm50 m50_exacts r3test_moments_ih10zTestIrwinHall.test_moments_ih10(sm-38D 2BCQ499##AaC(CCWd+iir"H S), Ds%A8ctjddd}|jj|}|jj|}t ||dyNrr r[rZmaxulp)rdrSunifrcih1r )rmptspdf_unifpdf_ih1s r3test_pdf_ih1_unifzTestIrwinHall.test_pdf_ih1_unif(sFkk!Q$99==%((,,s#Whr:r5ctjd}d}tjdd|}tjdd|}d||dzdzdz ||dzdzd|j |}t ||dy)NrarXrr rZr )r9rrdrSrcr )rmih2nptsr rRpdf_ih2s r3test_pdf_ih2_trianglez#TestIrwinHall.test_pdf_ih2_triangle(s|ooa kk!Q%;;q!T*%&4!8/2B)C%C$(q!"''#,Whr:r5ctjddd}|jj|}|jj|}t ||dyr )rdrSr rkr r )rmr cdf_unifcdf_ih1s r3test_cdf_ih1_unifzTestIrwinHall.test_cdf_ih1_unif)(sDkk!Q$99==%((,,s#Whr:r5ctjdd}tj|}|j |dz }tj dt |}t||dy)Nr rZrarvr )rdrr9rrkrr r )rmroihih_cdfexacts r3rVzTestIrwinHall.test_cdf1(sQ IIa  __Q A #s1v&VU26r5c gd}t|jjtjd|dt|jjdddd}t|jjd|dy) N) rg\xO~>g-P2?g8ʝi?gg{Ȇ?rvg&M?g׈Y?gB?gD l?r rrZr rWg+|gP3NV?g*8f?gNgX?gjS?rrrZr )r r rcrdrrs r3test_pdf_ih10_exactz!TestIrwinHall.test_pdf_ih10_exactJ(sDJ $tt*,,TYY]]299R=94Kr5c $t|jjtjdd|jj tjdz d}t |jjd|dy)Nrr rNrWrZr )rr rrdrrkr )rmrs r3test_sf_ih10_exactz TestIrwinHall.test_sf_ih10_exactQ(s\ RYYr]3QryyQS}9U5UV2TYY\\$/R@r5N)rrrr9rr rr r r r r r r rVr r r r;r5r3r r 'se 5==A D %//! C 5??2 DL - ;;;7C LAr5r ceZdZdZy)TestDParetoLognormcd\}}}}}tj||||}tjj |j |dtjj |j |dy)N)rrrZrrag6Z?g@# ?)r9dpareto_lognormrdrrrcrk)rmrYrrHr1r2rNs r3test_against_Rz!TestDParetoLognorm.test_against_R[(sc, 1aA$$Q1a0 ""488A;0CD ""488A;0CDr5N)rrrr r;r5r3r r Z(sEr5r r))r NNNN)rNNNN)r"NNNN)r NNNN)rNNNNc|\}}}}}tjd}|xsd}|xsd}|xsd}|xsd}tt|}t t |}||} tj ||} | j| j| } t| | ||dtj ||dz } | j| j| } t| | ||y) Nrviirrrhr r) rdr( r(r9r rrrrr) rr lp1lp2rrlpmrNr dist_frozenrrs r3test_sf_isf_overridesr o(s&*"Hc3d ((3-C +C *C 91D =5D 5( #D (^H %F-K ++c3 C ..- .CC4d3 bkk#sB' 'C ..- .CC4d3r5rp)NF(rrqr rrpathlibrr r rH numpy.testingrrrrrr r r r r rrrrXnumpyrdrrnumpy.lib.recfunctionsrr8rscipy._lib._utilrrrrrr scipy.statsr9!scipy.stats._distn_infrastructurerscipy.stats._constantsrscipy.stats.distributions scipy.specialrrrscipy.stats._distr_paramsrrtest_discrete_basicr!r"scipy.stats._continuous_distnsr#r$scipy.optimizer%r&r' itertoolsr(flagsoptimizer rKrzr r4r<rSrUrrrrr+r-rXrrrrrrr/r;rbrrrrrr rr$r+rDrIr\rfrzrrr$rHrSrVrrrrrrr-r;rcrmrzrrrrrrrrrr@rsr{rrerrrrrrrr%r)r-r8rEr}rrrrrrrrrrrrr rn r r r r r r r r r r r r5 r: rA rH r[ rc r r r ryr r r r r r r r r r r r r r r r r r r r r r r r r r r r r" r$ r& r0 r?r2 r9 r> rD rH r rr r r r r r r r r r r r r r r r r r! r# r0 rQ rs r r r r r r r r r r r r r r r r r;r5r3r s4   :::: *"4/338( 00;>C==ii((1,||x'M.>h.>.>.@H.L )6z'B$$ 8 G@G@T4D..>? 6 ? 699>G,G,T**>O,O,d<B<B~D/D/N##>*7D7Dt C CA2A2H44DI6I6X**. G GFK"K"\=I=I@`&`&FhDhDVDD,DD,U<Uw>t 3P6P6fB=B=J&&>$$*W +W +t))   , mGmG`e,e,Pw;w;t]H]H@",",JB&B&J668;";"|u0u0p...SFSFl..2J/J/Z ? ?Fh"h"V66(00D N NRBRBj2,2,jCCL>6>6BD0D0N <'0EF/G/ <*585.:#4L , )JK AK A#072 -& 9%&P-M"+ HBHHo7 89 HBHHo7 89 xrxx|456 HBHHk;/ 01 . 2  226 16 7:);<)<):; ) ; );<!<! ();< ) < ) %*$$ %%% %%%%%%u9u9p8B8Bv'0DE@F@<   , 3#,7.)X 60D)FG 4 G 4 )<<>?4?4  !" "#_ =  = *@^K^KB.3+3+lA2A2HP:P:fLL8 )?[+HI J 4#26;467373 784 84 0 /-(W.W.t 3 3 #K#KL""]A]A@EE*"GH 4 H 4r5