B 0d@sddlZddlZddlmZddlmZmZmZm Z m Z m Z m Z m Z ddlZddlmZddlmZddlmZddlmZdd lmZd d lmZdd lmZy&ddlZd ejd<ddlmZ dZ!Wne"k rdZ!YnXddddddddddg Z#ddddddddddg Z$ddddddddddg Z%ddddddddddg Z&ddd d!dd"ddd#dg Z'd$d%d$dddddd&dg Z(d!d'dd(ddd)d*ddg Z)ddddddddddg Z*dddddd'd'dddg Z+ddd"dddddd'dg Z,Gd+d,d,Z-Gd-d.d.Z.Gd/d0d0Z/Gd1d2d2Z0Gd3d4d4Z1Gd5d6d6Z2Gd7d8d8Z3Gd9d:d:Z4Gd;d<d<Z5Gd=d>d>e5Z6Gd?d@d@Z7GdAdBdBZ8GdCdDdDZ9GdEdFdFZ:GdGdHdHZ;GdIdJdJZGdOdPdPZ?GdQdRdRZ@dSdTdUdVdWdXdYdZd[d\d]d^d_d`dadbdcdddedfdgdhdidjdkdldmdndodpdqdrdsdtdudvdwdxdydzd{d|d}d~ddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddd d d d d ddddddddddddddddddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;d<d=d>d?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^d_d`dadbdcdddedfdgdhdidjdkdldmdndodpdqdrdsdtdudvdwdxdydzd{d|d}d~dddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddÐdĐdŐdƐdǐdȐdɐdʐdːd̐d͐dΐdϐdАdѐdҐdӐdԐdՐd֐dאdؐdِdڐdېdܐdݐdސdߐdddddddddddddddddgZAGdddZBGdddZCGdddZDGdddZEGdddZFGdddZGGdddZHGdddZIdS(N) RandomState)assert_array_equalassert_almost_equalassert_array_lessassert_array_almost_equalassert_assert_allclose assert_equalsuppress_warnings)raises)optimize)stats) _abw_state)check_named_results)_get_wilcoxon_distr)_binary_search_for_binom_tstZAggbackendTFgjt?gZd;?gV-?g?gX9v?g-?gx&1?g+?g+?gCl?gV-?gʡE?g/$?gףp= ?gGz?gCl?gGz?grh|?g\(\?g/$?gh|?5?g1Zd?gMb?gjt?g}?5^I?g)\(?c@s$eZdZddZddZddZdS) TestBayes_mvscCs~dddddddg}t|\}}}t|jdt|jdd d t|jd t|jd d d t|jdddt|jdd d dS)N  g"@)g4`H#j@gO[%@g+=)rtolg$@)gai @gӇu8@g& .>g _紫@)decimal)gYs?g1O@)r bayes_mvsr statisticrminmax)selfdatameanvarstdr'R/var/www/html/venv/lib/python3.7/site-packages/scipy/stats/tests/test_morestats.py test_basic/s  zTestBayes_mvs.test_basiccCstttjgdS)N) assert_raises ValueErrorr r)r"r'r'r(test_empty_inputAszTestBayes_mvs.test_empty_inputcCs4td}d}t|}x|D]}t||qWdS)N)r r!)nparanger rr)r"x attributesresir'r'r(test_result_attributesDs    z$TestBayes_mvs.test_result_attributesN)__name__ __module__ __qualname__r)r,r4r'r'r'r(r.src@s,eZdZddZddZddZddZd S) TestMvsdistcCsdddddddg}t|\}}}t|dt|dd d d t|d t|dd dd t|dddt|ddd d dS)Nrrrrrrg"@g?)g4`H#j@gO[%@g+=)rg$@)gai @gӇu8@g& .>g _紫@r)r)gYs?g1O@)r mvsdistrr$rinterval)r"r#r$r%r&r'r'r(r)Ns   zTestMvsdist.test_basiccCstttjgdS)N)r*r+r r9)r"r'r'r(r,]szTestMvsdist.test_empty_inputcCsdg}tttj|dS)Nr)r*r+r r9)r"r#r'r'r( test_bad_arg`szTestMvsdist.test_bad_argc Cs\tJtdtddtdddgDddtddddd gDWdQRXdS) NerrorcSsg|] }|qSr')r$).0r0r'r'r( jsz*TestMvsdist.test_warns..rrcSsg|] }|qSr')r$)r=r0r'r'r(r>ks)warningscatch_warnings simplefilterRuntimeWarningr r9)r"r'r'r( test_warnses  zTestMvsdist.test_warnsN)r5r6r7r)r,r;rFr'r'r'r(r8Msr8c@s<eZdZddZddZddZddZd d Zd d Zd S) TestShapiroc Csddddddddd d d d d dddddddg}t|\}}t|}t|dddt|jdddt|dddt|jddddddddddd d!d"d#d$dd%d&d'd(d)d*d+g}t|\}}t|}t|d,ddt|jd,ddt|d-d.dt|jd-d.dtjjd/d.d0d1d2}t|\}}t|}t|d3ddt|jd3ddt|d4d.dt|jd4d.dd5d6d7d8d9d:d;dd?d@dAdBd'dCdDdEdFdGdHdIdJdKdLg}dM}dN} t|\}}t|}t||dOdt|j|dOdt|| d/dt|j| d/ddS)PNg)\(?g{Gz@gq= ףp@gHzG$@g@gQ @gq= ףp?gQ@gzG?g@gQ?gRQ?gQ @g(\?gzG?gzG@g(\#@g333333@gGz?gp= ף@g?r)rgɌ?g(\?g= ףp=?g\(\@gffffff@g\(\?g(\?gGz@g(\gףp= @g?g)\(?gRQgp= ף?g@g ףp= ?g{Gz?g\(\ @g{Gz@gQ @gIddY?gDio?r?rAdiNa)locscalesize random_stategE?gOٴ?gn?gjt?gffffff?gMb?gjt?g;On?gjt?gX9v?gS?g(\?gI +?gʡE?gx?g|?5^?gCl?g1Zd@gS@g(\ @gGz @g rh @g"~j@gQ@gtV@gʡE @ghyܝ?g 0M?r@)r shapirorr pvaluenormrvs) r"x1wpw shapiro_testx2Zx3Zx4Z W_expectedZ p_expectedr'r'r(r)osH      zTestShapiro.test_basicc Cs ddddddddd d g d d d dddddddg g}t|\}}t|}t|dddt|jdddt|dddt|jddddddddddd d!d"g d#d$dd%d&d'd(d)d*d+g g}t|\}}t|}t|d,ddt|jd,ddt|d-d.dt|jd-d.ddS)/Ng)\(?g{Gz@gq= ףp@gHzG$@g@gQ @gq= ףp?gQ@gzG?g@gQ?gRQ?gQ @g(\?gzG?gzG@g(\#@g333333@gGz?gp= ף@g?r)rgɌ?g(\?g= ףp=?g\(\@gffffff@g\(\?g(\?gGz@g(\gףp= @g?g)\(?gRQgp= ף?g@g ףp= ?g{Gz?g\(\ @g{Gz@gQ @gIddY?gDio?r?)r rMrr rN)r"rQrRrSrTrUr'r'r(test_2ds$  zTestShapiro.test_2dcCs&tttjgtttjggggdS)N)r*r+r rM)r"r'r'r(r,szTestShapiro.test_empty_inputcCs4tttjddgtttjtjgdggtddS)Nrr)dtype)r*r+r rMr.arrayobject)r"r'r'r(test_not_enough_valuessz"TestShapiro.test_not_enough_valuescCsdg}tttj|dS)Nr)r*r+r rM)r"r0r'r'r(r;szTestShapiro.test_bad_argcCs`td}tj|d<t|\}}t|}t|tjt|jtjt|dt|jddS)Ng$@rg?) r.r/nanr rMr r rrN)r"r0rRrSrTr'r'r(test_nan_inputs     zTestShapiro.test_nan_inputN) r5r6r7r)rVr,rZr;r\r'r'r'r(rGns +rGc@sDeZdZddZddZddZddZd d Zd d Zd dZ dS) TestAndersoncCstd}|jdd}|jdd}t|\}}}t|dd|t|\}}}t||ddtd}d|d<t|\}}}t|ddS) NiI2)rK rg @) rstandard_exponentialstandard_normalr andersonrr.onesr)r"rsrQrUAcritsigvr'r'r( test_normals    zTestAnderson.test_normalc Cstd}|jdd}|jdd}t|d\}}}t||ddtjddt|d\}}}WdQRXt||dkdS) NiIr^)rKZexponr`ignore)allr_) rrbrcr rdrr.errstater)r"rfrQrUrgrhrir'r'r( test_expons  zTestAnderson.test_exponc Cstd}d|d<t|d\}}}t|}tj|\}}tj|||}tj|||} t d|d} | t d| d|| ddd} t || dS)NrHgrgumbelrrr_) r.rer rdlengumbel_lfitlogcdflogsfr/r$r) r"rjZa2rhrinZxbarsrtrur3Z expected_a2r'r'r( test_gumbels *zTestAnderson.test_gumbelcCstttjdgdddS)Nrplate_of_shrimp)dist)r*r+r rd)r"r'r'r(r;szTestAnderson.test_bad_argcCs0td}|jdd}t|}d}t||dS)NiIr^)rK)r critical_valuessignificance_level)rrbr rdr)r"rfr0r2r1r'r'r(r4s   z#TestAnderson.test_result_attributesc CsFtd}|jdd}t|d\}}}t|d\}}}t||dS)NiIrH)rKrprr)rrpr rdr) r"rfr0A1crit1sig1A2crit2sig2r'r'r( test_gumbel_ls  zTestAnderson.test_gumbel_lc Csptd}|jdd}td}d|d<t|d\}}}t|d\}}} t||ddt||dkdS) NiIrH)rKgZd;?rZgumbel_rr`r_)rrpr.rer rdrr) r"rfrQrUr}r~rrrrr'r'r( test_gumbel_rs  zTestAnderson.test_gumbel_rN) r5r6r7rkrorxr;r4rrr'r'r'r(r]s  r]c@sTeZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ dS)TestAndersonKSampc Csddddddddg}td d d d d dddg}tddddddddg}tddddddddg}tj||||fdd\}}}t|d d!td"d#d$d%d&g|d'd(d)t|d*d+d,dS)-NgYC@gD@gfffffE@g@F@gF@gG@gG@gM@gC@gfffffC@gC@g33333D@gfffffD@g33333sE@gfffffE@gfffffF@gA@gA@gC@gD@gE@gF@gF@gffffffA@g33333A@gB@gfffffB@gD@gffffffE@F)midrankg@r?gCl?gvq -?g,Ԛ?gl@gPk @rrAr@gJ4a?gMb0?)atol)r.rXr anderson_ksamprrr)r"t1t2t3t4Tktmpr'r'r(test_example1as z TestAndersonKSamp.test_example1ac Cstddddddddg}td d d d d dddg}tddddddddg}tddddddddg}tj||||fdd\}}}t|d d!td"d#d$d%d&g|d'd(d)t|d*d+d,dS)-NgYC@gD@gfffffE@g@F@gF@gG@gG@gM@gC@gfffffC@gC@g33333D@gfffffD@g33333sE@gfffffE@gfffffF@gA@gA@gC@gD@gE@gF@gF@gffffffA@g33333A@gB@gfffffB@gD@gffffffE@T)rgQ@r?gCl?gvq -?g,Ԛ?gl@gPk @rrAr@gMb`?gMb0?)r)r.rXr rrrr)r"rrrrrrrr'r'r(test_example1b0s z TestAndersonKSamp.test_example1bcCs>ddddddg}ddd d d d d ddddddg }ddddddddddg }ddddd d!dd"d#d$d%d&d'd(ddd)d*d+d,d-d'd.dg}d/d0d1d2d0g}d3dd4dd5d6d1d7dd8d(d9g }d:d;d#dd?d@dAdBdg }d.dCdDddEddd?dFdGd@d7dHd$dId6dEdJdKddGdJddJdLddMdLdNdOg}dPdQdJdRdSddTdUdVdWdMdLdXdYdZd[dd\d]d^d_d`ddaddbd"g} dcd'dddedd_dKddfdgd%dhd]dIdIdddidjdPdkd.dldg} dmd d6dndodKdrZra<=18OT,;i6LFeJ0iri7i@h/iWxG*rA r?r4_a3r@DMPjR6.o'?r^fHXrigii[rii#ibU[+ BF)rgM @g^I +?g~:?g+e?gH.!@gcZB>h@rg ^)p?gMb0?)r)r rrrr)r"rrrrt5t6t7t8t9t10t11t12t13t14rrrr'r'r(test_example2a@s4"  z TestAndersonKSamp.test_example2acCs>ddddddg}ddd d d d d ddddddg }ddddddddddg }ddddd d!dd"d#d$d%d&d'd(ddd)d*d+d,d-d'd.dg}d/d0d1d2d0g}d3dd4dd5d6d1d7dd8d(d9g }d:d;d#dd?d@dAdBdg }d.dCdDddEddd?dFdGd@d7dHd$dId6dEdJdKddGdJddJdLddMdLdNdOg}dPdQdJdRdSddTdUdVdWdMdLdXdYdZd[dd\d]d^d_d`ddaddbd"g} dcd'dddedd_dKddfdgd%dhd]dIdIdddidjdPdkd.dldg} dmd d6dndodKdh@rg ^)p?gMb0?)r)r rrrr)r"rrrrrrrrrrrrrrrrrr'r'r(test_example2b`s4"  z TestAndersonKSamp.test_example2bc Cs.tddd}t0}|jtddtj||dgdd\}}}WdQRXt|dd t|d t,}|jtddt||dg\}}}WdQRXt|d d t|d t0}|jtd dtj||d gdd\}}}WdQRXt|ddt|dt,}|jtd dt||d g\}}}WdQRXt|ddt|dtj||dgdd\}}}t|ddt |ddddt||dg\}}}t|ddt |ddddtj||dgdd\}}}t|ddt |ddddtj||dgdd\}}}t|d dt |d!d"dddS)#NrrHzp-value floored)messageg@D@F)rg= ףpD@r?gMbP?gGzD@zp-value cappedg?g?ܵr@g?g?ܵg@g!u?gףp= ׳?g{Gzt?r)rrrgBiq?g鷯?g'@g:M@g46W[1@g1ZGUS?g-C6?) r.linspacer filter UserWarningr rrr r)r"rQsuprw_rr'r'r(test_R_kSamplessB!&  "  &  "      z!TestAndersonKSamp.test_R_kSamplescCstttjtddS)NrA)r*r+r rr.re)r"r'r'r(test_not_enough_samplessz)TestAndersonKSamp.test_not_enough_samplescCs"tttjtdtdfdS)NrA)r*r+r rr.re)r"r'r'r(test_no_distinct_observationssz/TestAndersonKSamp.test_no_distinct_observationscCstttjtdgfdS)NrA)r*r+r rr.re)r"r'r'r(test_empty_samplesz#TestAndersonKSamp.test_empty_samplec CsRddddddddg}td d d d d dddg}tj||fdd}d}t||dS)NgYC@gD@gfffffE@g@F@gF@gG@gG@gM@gC@gfffffC@gC@g33333D@gfffffD@g33333sE@gfffffE@gfffffF@F)r)r r{r|)r.rXr rr)r"rrr2r1r'r'r(r4s z(TestAndersonKSamp.test_result_attributesN) r5r6r7rrrrrr r r r4r'r'r'r(rs  Prc @seZdZddZddZddZddZd d Zd d Zd dZ e j dddddgddddgddfddddgddddgddfdddgdddddgddfdddgdddddgddfdddddgdddgddfdddddgdddgddfgddZ dd Zd!S)" TestAnsaric Csldddddg}ddddddddg}t"}|tdt||\}}WdQRXt|ddt|d ddS) Nrrr?r@rz%Ties preclude use of exact statistic.g7@rg8#oG?)r rrr ansarir)r"r0yrWpvalr'r'r( test_smalls  zTestAnsari.test_smallc Cs^td}td}t"}|tdt||\}}WdQRXt|ddt|dddS)N)rkrHcrrmlrrr`rrrqtrnr)rrrrigrrrrHrrrrrrrrrrz%Ties preclude use of exact statistic.g0g@rg|:?)r.rXr rrr r r)r"ZramsayZparekhrrrr'r'r( test_approxs    zTestAnsari.test_approxc Cs@tddddgddddd d g\}}t|d d t|d ddS)Nrrr?r@r-rArrrarg$@rg?r)r r r)r"rrr'r'r( test_exacts$ zTestAnsari.test_exactcCs(tttjgdgtttjdggdS)Nr)r*r+r r )r"r'r'r(r;szTestAnsari.test_bad_argc Cs^dddddg}ddddddddg}t}|tdt||}WdQRXd}t||dS)Nrrr?r@rz%Ties preclude use of exact statistic.)r rN)r rrr r r)r"r0rrr2r1r'r'r(r4s z!TestAnsari.test_result_attributesc CsHddddg}ddddg}d }tt|d tj||d d WdQRXdS) Nrrr?r@rArrrz!'alternative' must be 'two-sided')matchfoo) alternative)r*r+r r )r"rQrUrr'r'r(test_bad_alternatives   zTestAnsari.test_bad_alternativec Csdddddddg}dd d d d d g}t||\}}tj||ddj}tj||ddj}|dksbt|dksntt|t|t|}t||d|ddt||dddt|d||dddtj||ddj}tj||ddj} |dkst| dkstdS)NrrArar-rrg@g!@g#@g%@g'@g)@less)r!greatergffffff?g?g-q=)rr)r r rNAssertionErrorrpmfrqr) r"rQrUr rpval_lpval_gZprobZpval_l_reverseZpval_g_reverser'r'r(test_alternative_exacts   z!TestAnsari.test_alternative_exactzx, y, alternative, expectedrrr?r@rArrrr$g@A?r%gG$I?g%I$I?cCs$tj|||dj}t||dddS)N)r!g-q=)r)r r rNr)r"r0rr!expectedrr'r'r(test_alternative_exact_with_R2sz(TestAnsari.test_alternative_exact_with_RcCstjjddddd}tjjddddd}tj||ddj}tj||d dj}t|d d d t|d d d tjjddddd}tjjddddd}t||j}tj||ddj}tj||d dj}t||dd d t|d|dd d dS)NrrArH{)rKrLrr$)r!r%g?g-q=)rgrg?r)r rOrPr rNr)r"rQrUr(r)rr'r'r(test_alternative_approxQsz"TestAnsari.test_alternative_approxN)r5r6r7rrrr;r4r"r*pytestmark parametrizer,r.r'r'r'r(r s   (r c@s4eZdZddZddZddZddZd d Zd S) TestBartlettc CsBtttttttttt g }t j |\}}t |ddt |dddS)NgzU@/4@rgG?) g1g2g3g4g5g6g7g8g9g10r bartlettr)r"argsTrr'r'r( test_datais zTestBartlett.test_datacCstttjdgdS)Nr)r*r+r r=)r"r'r'r(r;pszTestBartlett.test_bad_argc Cs4tttttttttt g }t j |}d}t ||dS)N)r rN) r3r4r5r6r7r8r9r:r;r<r r=r)r"r>r2r1r'r'r(r4ts z#TestBartlett.test_result_attributesc Cs6tttttttttt gf }t t j t j ft j|dS)N)r3r4r5r6r7r8r9r:r;r<r r.r[r r=)r"r>r'r'r(test_empty_argzszTestBartlett.test_empty_argcCs*tddgddgg}tttjt|dS)Nrrr?r@)r.rXr*r+r r=r3)r"r0r'r'r( test_1d_inputszTestBartlett.test_1d_inputN)r5r6r7r@r;r4rArBr'r'r'r(r2gs r2c@sTeZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ dS) TestLevenec CsBtttttttttt g }t j |\}}t |ddt |dddS)NgYpK?rgI{]?) r3r4r5r6r7r8r9r:r;r<r levener)r"r>rrr'r'r(r@s zTestLevene.test_datacCsFtjtttdd\}}tjtttddd\}}t||t||dS)Nr$)centertrimmedg)rEproportiontocut)r rDr3r4r5r)r"W1pval1W2pval2r'r'r( test_trimmed1s    zTestLevene.test_trimmed1c Csddddddddg}d dd dd dd d g}tjdtj|}tj||ddd\}}tj||ddd\}}tj|dd|dddd\}} t||t||t|| dS)Ng333333?g@g@g@g@g@g@gY@gg @g@g@gi@irFg?)rErGrr_r$)rE)r.randomseed permutationr rDr) r"r0rrUZW0Zpval0rHrIrJrKr'r'r( test_trimmed2s      $  zTestLevene.test_trimmed2cCsntddd}tjdtj|}|d}tj||dd\}}tj||dd\}}t||t||dS) Nr_rrir?r$)rEZmedian)r.rrMrNrOr rDr)r"r0rUrrHrIrJrKr'r'r(test_equal_mean_medians   z!TestLevene.test_equal_mean_mediancCs&tddd}tttj||dddS)Nr_rrg?) portiontocut)r.rr* TypeErrorr rD)r"r0r'r'r(test_bad_keywordszTestLevene.test_bad_keywordcCs&tddd}tttj||dddS)Nr_rrtrim)rE)r.rr*r+r rD)r"r0r'r'r(test_bad_center_valuesz TestLevene.test_bad_center_valuecCstttjdgdS)Nr)r*r+r rD)r"r'r'r(test_too_few_argsszTestLevene.test_too_few_argsc Cs4tttttttttt g }t j |}d}t ||dS)N)r rN) r3r4r5r6r7r8r9r:r;r<r rDr)r"r>r2r1r'r'r(r4s z!TestLevene.test_result_attributescCs*tddgddgg}tttjt|dS)Nrrr?r@)r.rXr*r+r rDr3)r"r0r'r'r(rBszTestLevene.test_1d_inputN) r5r6r7r@rLrPrQrTrVrWr4rBr'r'r'r(rCs  rCc@sBeZdZdZeejZddZddZ ddZ dd Z d d Z d S) TestBinomPzTests for stats.binom_test.cCsR|dd}t|dd|dd}t|dd|jdd gd d }t|d ddS) NrHg-"^?rrig`w?ig?)rgz?)binom_test_funcr)r"rr'r'r(r@s     zTestBinomP.test_datacCstt|jdddgdS)Nrrr?)r*r+r[)r"r'r'r(test_bad_len_xszTestBinomP.test_bad_len_xcCs(tt|jdgtt|jdgdddS)NrHr^)rv)r*r+r[)r"r'r'r( test_bad_nszTestBinomP.test_bad_ncCstt|jddgdddS)Nr^g@)r)r*r+r[)r"r'r'r( test_bad_pszTestBinomP.test_bad_pcCsX|jddddd}t|d|jddddd}t|d|jdddd d}t|d dS) NrgUUUUUU?r$)rr!g-$Ǻl?r%gs0,j.?z two-sidedgJfJ"f?)r[r)r"r2r'r'r(test_alternativess   zTestBinomP.test_alternativesN) r5r6r7__doc__ staticmethodr Z binom_testr[r@r\r]r^r`r'r'r'r(rXs rXc@seZdZdZedddZdS)TestBinomTestPzJ Tests for stats.binomtest as a replacement for stats.binom_test. N? two-sidedcCst|tj}t|dkr8|d|d}|d}n@t|dkrp|d}|dks\||krdtdt|}ntdtj||||d}|jS)Nrrrzn must be >= xzIncorrect length for x.)rr!) r.Z atleast_1dZastypeint_rqr+r binomtestrN)r0rvrr!resultr'r'r(r[s    zTestBinomTestP.binom_test_func)Nrdre)r5r6r7rarbr[r'r'r'r(rcsrcc @s@eZdZdZddZddZddZdd Zej d d d d gddZ ej d dddgddZ ej ddddgddZ ej ddddgdd Zej d!d"d#d$d%d&d'gd"d#d$d(d)d*gd"d#d+d%d,d-gd"d#d+d(d.d/gd"d0d$d%d1d2gd"d0d$d(d1d3gd"d0d+d%d1d4gd"d0d+d(d1d5gd"d6d$d%d7d8gd"d6d$d(d9d8gd"d6d+d%d:d8gd"d6d+d(d;d8gdgd)r rgrrN)r"rr2r'r'r(test_two_sided_pvalues1sz%TestBinomTest.test_two_sided_pvalues1cCsd}tjdddd}t|jd|dtddd}t|jd |dtd dd}t|jd |dtd dd}t|jd |dtddd}t|jd|dtddd}t|jd|dtddd}t|jd|ddS)Ng|=rrgQ?)rvrgX-h?)rr@g㤶y?rg y}?rg+,?r?rag?g?rg?g{Gz?g333333?g<,?)r rgrrN)r"rr2r'r'r(test_two_sided_pvalues2sz%TestBinomTest.test_two_sided_pvalues2cCsd}tddd}t|jd|dtddd }t|jd|dtd d d }t|jd|dtd dd}t|jd|dtdd d}t|jd|dtddd}t|jd|ddS)Ng|=iig?r)rr?rgAL &W?rrg! YB?r-rgE]t?rrr)r rgrrN)r"rr2r'r'r(test_edge_cases-szTestBinomTest.test_edge_casescs,ddd}ttd}tj|}tjtj||kdd}tfdd| t}|t|tj|k}t ||d d d }tt d}tj|}tjtj||kdd}tfd d|dt }|d}t ||d d dS) Nrag?r?rr)axiscstj| S)N)r binomr')rQ)rvrr'r(Isz>TestBinomTest.test_binary_srch_for_binom_tst..g& .>)rrcstj|S)N)r rnr')rQ)rvrr'r(roUs) r.r/ceilr rnr'sumrintrfloor)r"kr3dy1Zixy2r')rvrr(test_binary_srch_for_binom_tst<s$ z,TestBinomTest.test_binary_srch_for_binom_tstz"alternative, pval, ci_low, ci_high)r$g$W ?gg?)r%g}?gGI~?g?)z two-sidedg}G?g>?6?g4M?cCsXtjddd|d}t|j|ddt|jd|jdd }t|j|jf||fdddS) NrrHg?)rvrr!g-q=)rg?gffffff?)confidence_level) r rgrrNr proportion_estimate proportion_cilowhigh)r"r!rci_lowci_highr2cir'r'r(test_confidence_intervals1\s   z(TestBinomTest.test_confidence_intervals1)r$ga&+w?ggny?)r%g5!Rx?g5%?g?)z two-sidedg~th?gSuR&|?g|KnG?cCsXtjddd|d}t|j|ddt|jd|jdd }t|j|jf||fdddS) Nr?r^g?)rvrr!gư>)rgQ?gGz?)ry) r rgrrNr rzr{r|r})r"r!rr~rr2rr'r'r(test_confidence_intervals2ks   z(TestBinomTest.test_confidence_intervals2zalternative, pval, ci_high)r$g‹լ?g̓A?)r%g?g?)z two-sidedgE ow?g;Kڞj?cCsNtjddd|d}t|j|dd|jdd}t|jd t|j|dddS) Nrrag?)rr!gư>)rgffffff?)ryg)r rgrrNr{r r|r})r"r!rrr2rr'r'r(!test_confidence_interval_exact_k0zs   z/TestBinomTest.test_confidence_interval_exact_k0zalternative, pval, ci_low)r$g?g)r%g6>g@>}_?)z two-sidedg6>gcڒ ?cCsNtjddd|d}t|j|dd|jdd}t|jdt|j|dddS) Nrag?)rr!gư>)rgffffff?)ryg?)r rgrrNr{r r}r|)r"r!rr~r2rr'r'r(%test_confidence_interval_exact_k_is_ns   z3TestBinomTest.test_confidence_interval_exact_k_is_nz+k, alternative, corr, conf, ci_low, ci_highr?z two-sidedTgffffff?g`?g?gGz?grt!?g/?FgVK5?g5dM?gXCVu^?g*r[?r$ggP-V?g^'?gS5a?g ?r%g!Շ`ⰸ?g?g$x" ?gW~=?g L?rgX ?gN?g/<7#9?gAA?rag7?g؍s?ghadn?gAB/?c CsLtj|dd|d}|rd}nd}|j||d} t| j| jf||fdddS) Nrag?)rvrr!ZwilsonccZwilson)rymethodgư>)r)r rgr{rr|r}) r"rtr!Zcorrconfr~rr2rrr'r'r(test_ci_wilson_methods z#TestBinomTest.test_ci_wilson_methodcCs*tddd}t|jdt|jddS)Nr@rg?g?)r rgr rzrN)r"r2r'r'r(&test_estimate_equals_hypothesized_props z4TestBinomTest.test_estimate_equals_hypothesized_propzk, n)rr)r_rc Cs*tjtddt||WdQRXdS)Nz must be an integer not less than)r)r/r r+r rg)r"rtrvr'r'r(test_invalid_k_ns zTestBinomTest.test_invalid_k_nc Cs,tjtddtdddWdQRXdS)Nzk must not be greater than n)rrrag?)r/r r+r rg)r"r'r'r(test_invalid_k_too_bigs z$TestBinomTest.test_invalid_k_too_bigc Cs:tjdddd}tjtdd|jddWdQRXdS) Nr?rag?)rvrzmust be in the interval)rr_)ry)r rgr/r r+r{)r"r2r'r'r(test_invalid_confidence_levelsz+TestBinomTest.test_invalid_confidence_levelc Cs:tjdddd}tjtdd|jddWdQRXdS) Nr?rag?)rvrzmethod must be)rzplate of shrimp)r)r rgr/r r+r{)r"r2r'r'r(test_invalid_ci_methodsz$TestBinomTest.test_invalid_ci_methodN)r5r6r7rarjrkrlrxr/r0r1rrrrrrrrrrr'r'r'r(rish      ric@sDeZdZddZddZddZddZd d Zd d Zd dZ dS) TestFlignercCs&td}tt||ddddS)NrAr)gnf @gDq?r)r.r/rr fligner)r"rQr'r'r(r@s zTestFligner.test_datac svtjdfdd}|t}|t}|t}tj|||dd\}}tj|||ddd\}}t||t||dS) Nr-cs t|dt|S)Ng|=)r.Zasarrayrandnrqtolist)g)rfr'r(roz+TestFligner.test_trimmed1..r$)rErFg)rErG) r.rMrr3r4r5r rr) r"Z_perturbZg1_Zg2_Zg3_Xsq1rIXsq2rKr')rfr(rLs     zTestFligner.test_trimmed1cCszddddddddg}d dd dd dd d g}tj||ddd\}}tj|dd|dddd\}}t||t||dS)Ng333333?g@g@g@g@g@g@gY@gg @g@g@gi@rFg?)rErGrr_r$)rE)r rr)r"r0rrrIrrKr'r'r(rPs  $ zTestFligner.test_trimmed2cCs&tddd}tttj||dddS)Nr_rrg?)rR)r.rr*rSr r)r"r0r'r'r(rT szTestFligner.test_bad_keywordcCs&tddd}tttj||dddS)Nr_rrrU)rE)r.rr*r+r r)r"r0r'r'r(rVsz!TestFligner.test_bad_center_valuecCstttjdgdS)Nr)r*r+r r)r"r'r'r(test_bad_num_argsszTestFligner.test_bad_num_argscCs.td}ttjtjft||dgdS)NrAr)r.r/r r[r r)r"r0r'r'r(rAs zTestFligner.test_empty_argN) r5r6r7r@rLrPrTrVrrAr'r'r'r(rsrc@sneZdZddZddZddZddZd d Zd d Zd dZ e j j dde j ddddgddZdS)TestMoodcCs&td}tt||ddddS)NrAr)g8!giT?r)r.r/rr mood)r"rQr'r'r( test_moods zTestMood.test_moodcCs`tjdtjdd}tjdd}t||\}}t||\}}t||g| |gdS)Nirarr-)r.rMrNrr rr)r"rQrUZz1p1Zz2p2r'r'r(test_mood_order_of_args%s  z TestMood.test_mood_order_of_argscCsddddddddd d d d d dddddddg}dddddddddddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2g}t|}t|}d3|_d4|_ttj||dd5d6d7gdS)8Ng g۟?gy<7xg1)E?g9cç?g FAg<=ӟ 2?g{![?g_l?gڻ{ӿgGSsA0?g 1?g 3ggL?g_lg,*.gː53?gvNqG?gW=?gd"g`Y#?g^g?gNgB agp5_%?g(?gv]lοgg?ggr6ʺ?ḡj?gK!ٿg{g&?g*h}gZ?gyBU?g .ۋ4?g|w;?g&ck@gRj2g}+w(o?g@F?g}_9t?g =g S?gx (gg7Y?g{ih?)rar)r-r)rmg4g^}E ?)r.rXshaperr r)r"rQrUr'r'r(test_mood_with_axis_none/s.   z!TestMood.test_mood_with_axis_nonec Csd}tjdtjd|}tjd|}t||\}}xFt|D]:}t||||gt|dd|f|dd|fqFW|}|}tj||dd\}}xFt|D]:}t||||gt||ddf||ddfqWdS)NrAirar-r)rm) r.rMrNrr rrangerZ transpose)r"nyrQrU z_vectest pval_vectestjr3r'r'r( test_mood_2dKs *zTestMood.test_mood_2dc CsNd}tjdtjj|}tjj|}xtdD]}tj|||d\}}ddgddgddgf}xt|||dD]}xt|||dD]} |dkr|dd|| f} |dd|| f} nR|dkr||dd| f} ||dd| f} n$||| ddf} ||| ddf} t||| f||| fgt| | qWqzWq4WdS)N)rarArir?)rmrrr)r.rMrNrrr rr) r"rrQrUrmrrZaxes_idxr3rZslice1Zslice2r'r'r( test_mood_3dbs(     zTestMood.test_mood_3dcCstttjdggdS)Nr)r*r+r r)r"r'r'r(test_mood_bad_arg~szTestMood.test_mood_bad_argc Cstjdtjjddd}tjjddd}tj||dd\}}tj||dd\}}tj||d d\}}||kr||ksntt|dd d t||d t|d |d t j t ddtj||ddWdQRXdS)Nrg?rH)rJrKg?z two-sided)r!r$r%gHz>)rrrzalternative must be...)rz ekki-ekki) r.rMrNr rOrPrr&rr/r r+) r"r0rZstat1rZstat2rZstat3Zp3r'r'r(test_mood_alternatives zTestMood.test_mood_alternativez,SciPy needs tie correction like R (gh-13730))reasonzalternative, expected)z two-sided)g!}?g-?)r$)g~?g4?)r%)g!}?g-?cCsvddddddddd dd d d ddd dd ddg}ddddddddd dd ddd ddd dddg}tj|||d}t||dS)NrrrHrrrrrrrrrrrrrrrr)r!)r rr)r"r!r+r0rr2r'r'r(test_mood_alternative_against_Rs z(TestMood.test_mood_alternative_against_RN)r5r6r7rrrrrrrr/r0Zxfailr1rr'r'r'r(rs  rc@sVeZdZddZddZddZejje ddd d Z d d Z d dZ ddZ dS) TestProbplotcCstjjddd}tj|dd\}}dddd d d d d ddddddddddddg}t|t|t||tj|dd\}}dddg}t||dS)Nri90)rKrLF)rsg+/gY3gtg5Z0gFjg!2gsR{ܿg7Կg>hǿgcz隇gcz隇?g>h?g7?gsR{?g!2?gFj?g5Z0?gt?gY3?g+/?TgƝV?g?gà?)r rOrPprobplotrr.sort)r"r0osmosrZ osm_expectedr2Zres_fitZres_fit_expectedr'r'r(r)s   zTestProbplot.test_basicc Cstjjddd}tj|ddd\}}tj|ddd\}}tj|ddd\}}t||t||t||t||tj|ddd\}} dS)NrHi@)rKrLF)sparamsrsrr')r rOrPrr) r"r0osm1osr1osm2osr2Zosm3Zosr3rrr'r'r(test_sparams_keywords    z!TestProbplot.test_sparams_keywordcCstjjddd}tj|dddd\}}tj|dtjdd\}}t||t||tttj|dd tttj|gd Gd d d }tj|d dd \}}tj||dd\}}t||t||dS)Nri90)rKrLFt)r?)rsrzrzwrong-dist-name)rzc@seZdZdZddZdS)z3TestProbplot.test_dist_keyword..custom_distz6Some class that looks just enough like a distribution.cSstjj|ddS)Nr)rI)r rOppf)r"qr'r'r(rsz7TestProbplot.test_dist_keyword..custom_dist.ppfN)r5r6r7rarr'r'r'r( custom_distsr)r)rrs)rzrs) r rOrPrrrr*r+AttributeError)r"r0rrrrrr'r'r(test_dist_keywords   zTestProbplot.test_dist_keywordz no matplotlib)rc Cst}|dtjjdddd}tj|td\}}ttj|dd\}}tj|dtd}ttj|ddd}tt |t |kot |kot |kod knt ||t ||t ||t ||t}|d} tj|d| dtdS) Nrr?rHit)rKrL)plotF)rsrr) pltfigure add_subplotr rrPrcloserrqr) r"figr0Zres1Zfitres1Zres2Zfitres2Zres3Zres4axr'r'r(test_plot_kwargs$ @     zTestProbplot.test_plot_kwargcCstttjdgdddS)Nrry)rz)r*r+r r)r"r'r'r(test_probplot_bad_argssz#TestProbplot.test_probplot_bad_argscCsZttjgddtgtgfttjgddtgtgftjtjdffdS)NF)rsTg)r r rr.rXr[)r"r'r'r( test_emptys zTestProbplot.test_emptyc CsTtjdd>ttjdgddtdgtdgftjtjdffWdQRXdS)Nrl)invalidrT)rsg)r.rnr r rrXr[)r"r'r'r(test_array_of_size_onesz#TestProbplot.test_array_of_size_oneN)r5r6r7r)rrr/r0skipifhave_matplotlibrrrrr'r'r'r(rsrc @seZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ e jddddgdddddgdddddddd ggd!d"Zd#d$Zd%S)& TestWilcoxoncCshtttjdgddgtttjddgddgdtttjddgddgddtttjdgddddS)Nrrdummy)r!raZxyz)mode)r*r+r wilcoxon)r"r'r'r(test_wilcoxon_bad_arg s z"TestWilcoxon.test_wilcoxon_bad_argcCsRtd}tttj||dddtttj||dddttj||dddddS)Nrwilcoxapprox)rprattzsplit)g@Z@g?)r.r/r*r+r rr )r"r0r'r'r(test_zero_diffs zTestWilcoxon.test_zero_diffc CsXddddg}ddddg}t&}|jtddtj||dd d }WdQRXt|d dS) Nrrr?r@rAzSample size too small)rrr) zero_methodr)gg |N?)r rrr rr)r"r0rrr2r'r'r( test_pratts   zTestWilcoxon.test_prattc CsNdddddddddddg }tj|ddd}tj|d dd}tj|d dd}dS) Nrrr?rr_rr)rrrr)r r)r"Zarrrr'r'r(test_wilcoxon_arg_type(sz#TestWilcoxon.test_wilcoxon_arg_typecCshdddddddddg }tdd}td d t||D}t|j}tj||d d d \}}t|dt|dtj||dd d \}}t|dt|dtj||dd d \}}t|dt|dt ddddddddddd d!g }t d"d#d$d%d!d&d'dd(dd)d*g }tj||d+d d,\}}t |d-t|d.d/d0tj||d1d d,\}}t |d-t|d2d/d0dS)3Nrr@rr-rrArcSsg|]\}}|g|qSr'r')r=urjr'r'r(r>4sz7TestWilcoxon.test_accuracy_wilcoxon..rr)rig.i?rigFYv]Uj?riGg%^Ez?rrrryrrpwsrF) correctionrrg^҂)rTg:b+?) rr.Z concatenatezipzerosrKr rrrXr )r"freqnumsr0rr?rr'r'r(test_accuracy_wilcoxon1s*        ""  z#TestWilcoxon.test_accuracy_wilcoxoncCshtddddddddd d d d g }td dddd ddd ddddg }tj||ddd}d}t||dS)NrrrrrrrrrrrrrrrrrrrrrFr)rr)r rN)r.rXr rr)r"r0rr2r1r'r'r(test_wilcoxon_result_attributesNs ""z,TestWilcoxon.test_wilcoxon_result_attributescCsntjdgddd\}}d}t|dt||ddtjdgdd dd \}}d }t|dt||dddS) Ng?rar)rg4IťY?rgư>)rT)rrg c2_?)r rr r)r"statrZ expected_pr'r'r(test_wilcoxon_tieUs   zTestWilcoxon.test_wilcoxon_tiec Cspddddddddddg }dddddd d d dd g }t*}|jtd dtj||ddd\}}WdQRXt|dt|dddt,}|jtd dtj||dddd\}}WdQRXt|dt|dddt*}|jtd dtj||ddd\}}WdQRXt|dt|dddt,}|jtd dtj||dddd\}}WdQRXt|dt|ddddS)N}rrrzr|r-zSample size too small)rr$r)r!rrg13}?r)rT)r!rrg7a%?r%g?gڏoJT?)r rrr rr r)r"r0rrrRrr'r'r( test_onesidedhs0         zTestWilcoxon.test_onesidedcCsPxJtddD]<}t|}t||dddt|tt|d|q WdS)Nrrr)rrr rqrq)r"rvZcntr'r'r(test_exact_basicszTestWilcoxon.test_exact_basicc CsVtddddddddd d g }td d d dddddddg }tj||ddd\}}t|dddtj||ddd\}}t|dddtj||ddd\}}t|dddtdd d!}td dd"}tj||ddd\}}t|d#ddtj||ddd\}}t|d$ddtj||ddd\}}t|d%ddtd&d'}tttj|dd(dS))Ng(\?g= ףp=?g(\?gQ޿gQ?g{Gz?gp= ףgq= ףp?ggQ?gQ?g?gɿg333333?ggܿgzGgQοg(\gRQz two-sidedexact)r!rg֔?r)rr$gYJ?r%g*?rrg?r_g(?g(?gS?rr)r)r.rXr rrr/r*r+)r"r0rrrrur'r'r(test_exact_pvals(  zTestWilcoxon.test_exact_pvalr0r_r`r?rrrAr#irrcCs@t|\}}t|}||dk}t||t|ddS)Nrr)r rr.rXrqr )r"r0rRrZwtruer'r'r(test_exact_p_1s   zTestWilcoxon.test_exact_p_1c Cstddd}tddd}tt||tj||ddtdd}t"}|jtdd t|\}}WdQRXttj|d d||ftd d }tt|tj|d ddS) Nrrg?r_r)rrzExact p-value calculation)rrrr)r.r/r r rr rr)r"r0rrurrRrr'r'r( test_autos   zTestWilcoxon.test_autoN)r5r6r7rrrrrrrrrrr/r0r1rrr'r'r'r(r s    &  rc@s,eZdZddZddZddZddZd S) TestKstatcstjdtjdfdddD}dddd g}t||d d tjd d }tjdd }tjdd }t|||f|ddddddS)Ni}i90csg|]}t|qSr')r kstat)r=rv)r#r'r(r>sz>TestKstat.test_moments_normal_distribution..)rrr?r@gHQ,?glZ)rI?g88}?gcJN?g-C6?)rr)momentrr?r_g{Gz?g{Gz?)rr)r.rMrNrrr r)r"Zmomentsr+m1m2Zm3r')r#r( test_moments_normal_distributions   z*TestKstat.test_moments_normal_distributioncCstttjgdS)N)r*r+r r)r"r'r'r(r,szTestKstat.test_empty_inputcCs*td}tj|d<tt|tjdS)Ng$@r)r.r/r[r r r)r"r#r'r'r(r\s  zTestKstat.test_nan_inputcCs.td}xdD]}tttj||dqWdS)Nra)rg/$@)rv)r.r/r*r+r r)r"r#rvr'r'r(test_kstat_bad_args  zTestKstat.test_kstat_bad_argN)r5r6r7rr,r\rr'r'r'r(rsrc@s$eZdZddZddZddZdS) TestKstatVarcCstttjgdS)N)r*r+r kstatvar)r"r'r'r(r,szTestKstatVar.test_empty_inputcCs*td}tj|d<tt|tjdS)Ng$@r)r.r/r[r r r)r"r#r'r'r(r\s  zTestKstatVar.test_nan_inputcCs dg}d}tttj||ddS)Nrra)rv)r*r+r r)r"r#rvr'r'r(r;szTestKstatVar.test_bad_argN)r5r6r7r,r\r;r'r'r'r(rsrc@sNeZdZddZddZddZejje ddd d Z d d Z d dZ dS) TestPpccPlotcCstjjddddd|_dS)NrAiit)rKrL)r loggammarPr0)r"r'r'r( setup_methodszTestPpccPlot.setup_methodcCsNd}tj|jdd|d\}}ddddd g}t|tjdd|d t||dS) NrAira)Ng9% ?gxM _?g"; 9?g®XZ?g?)num)r ppcc_plotr0rr.r)r"rsvalsppcc ppcc_expectedr'r'r(r)s zTestPpccPlot.test_basiccCstj|jdddd\}}tj|jddtjd\}}t||ddt||ddt|jdd\}}t||ddt||dddS)Nira tukeylambda)rzg#B ;)r)r r r0rr)r"Zsvals1Zppcc1Zsvals2Zppcc2Zsvals3Zppcc3r'r'r( test_dist s zTestPpccPlot.test_distz no matplotlib)rcCsZt}|d}tj|jddtd|||d}tj|jdd|dtdS)Nrir)r)rrrr r r0delaxesr)r"rrr'r'r(rs   zTestPpccPlot.test_plot_kwargcCs4tttj|jddtttjdddgdddddS)Nrrrr?ry)rz)r*r+r r r0)r"r'r'r(test_invalid_inputs$sz TestPpccPlot.test_invalid_inputscCs@tgdd\}}t|tjddddt|tjdtddS)Nrrr)r )rW)r r rr.rrfloat)r"r r r'r'r(r,szTestPpccPlot.test_emptyN) r5r6r7rr)rr/r0rrrrrr'r'r'r(rs   rc@s,eZdZddZddZddZddZd S) TestPpccMaxcCsdg}tttj|dddS)Nrry)rz)r*r+r ppcc_max)r"r#r'r'r(test_ppcc_max_bad_arg6sz!TestPpccMax.test_ppcc_max_bad_argcCs2tjjddddddd}tt|dd d dS) Ngffffffrg?i'i)rIrJrKrLg@g{r)r)r rrPrr)r"r0r'r'r(test_ppcc_max_basic;s zTestPpccMax.test_ppcc_max_basiccCsptjjddddddd}tj|dd }tj|tjd }t|d d d t|d d d t|}t|d d d dS) Ngffffffrg?i'i)rIrJrKrLg@r)rzg{rA)r)r rrPrr)r"r0Zmax1Zmax2Zmax3r'r'r(r@s  zTestPpccMax.test_distcCs`tjjddddddd}tttj|dd ttj|d d d d d ttj|dd d d d dS)Ngffffffrg?i'i)rIrJrKrLg@)gg?g?)brack)rrg{r)r)r`r)r rrPr*r+rr)r"r0r'r'r( test_brackNs  zTestPpccMax.test_brackN)r5r6r7rrrrr'r'r'r(r5src@s4eZdZddZddZddZddZd d Zd S) TestBoxcox_llfcCsTtjjdddd}d}t||}|j dtt|d}t ||dS)Ni'rai1)rKrIrLrg@r) r rOrP boxcox_llfrKr.logrqr&r)r"r0lmbdallfZ llf_expectedr'r'r(r)\s  $zTestBoxcox_llf.test_basiccCsDtjjdddd}d}t||}t|t|}t||dddS)NrHrai1)rKrIrLrg-q=)r)r rOrPrlistr)r"r0rrllf2r'r'r(test_array_likecs  zTestBoxcox_llf.test_array_likecCsPtjjdddd}d}t||}t|t||gj}t||g|dddS)NrHrai1)rKrIrLrg-q=)r)r rOrPrr.vstackr?r)r"r0rrrr'r'r( test_2d_inputjs  zTestBoxcox_llf.test_2d_inputcCstttdgdS)Nr)rr.isnanr r)r"r'r'r(rtszTestBoxcox_llf.test_emptycCs&ddddg}td|}t|ddS)Ngh@g m@gx@ig#:x1)r rr)r"r#rr'r'r( test_gh_6873ws  zTestBoxcox_llf.test_gh_6873N)r5r6r7r)r r"rr$r'r'r'r(rZs  riU>iϵii_ i7iisivi+ i*i-qir:iiii i_i4i iDii,iEi\ii|igiM iӢi[i!i if6iPiiiIi/iF&i ii`#iii&i iiQiai iXZiwi_ imiDi-iBi:Ci i'ii %ixiiz_iQi!#i=i ixi iU7i_i1kiW9i&ixiIiiTi iiirvimi͟iiOivi)iP i4i=iiiUiiig>ix i`i|i0i`iM i'i[i8ii3[ii) i`Ni ii[i:+ipi %i iVi[$iXi_ ibieZii`? itiiiEiiTiyiimvi<iZi!ii(ir ii iɪi|ii >iiii* i#ixi i&ii#i1giP i)iۆi{%i?i۽i|i i&i-iư)i+ci7iQi oi&i|_iy i6iiii:Ji#ji8ii%i i%Wi(i`i:inaii i%i$i4iA(iqii"iyiXZi izrii$i;isigYii7ip ipiL6"iDHii!3iiI(i|Ciш iimiSii{8iiiiPi2Li(i@isiui2i_iji)migZi_iu i9iki';iiiiKi܃ii =ivi=i ia iriii}Wi*iiJi ii2iUFiNj iV,"ipiAi%iiv9iI5iiT-i i8~i?e`i1 iVi/Xi = i4iiحi^i i/igi(ii|B;ipiiciE'iEi$ ii{iki:iIiA_iiPioPi6_iiܲ iO iA7iLiAiiViwmiiZh i|ViԇCiw i&i~ii'i!ii i}iei3 iii_ii@)ižiViiiFi?i& i"Y iwiiHiSirdibi@iKiA;iz|id+ i i 8i"i iOi_i$li!9 iiiTiuL iQ'%iqpiCi]K@i!iiiIp|ii? Biiio0ii ii5%iii*iiiv 0i*i*iwiivei $ii-iD~i-ii,4ibii iP0i)imi iRiQi iic@seZdZddZddZddZddZd d Zd d Ze j d dddgddZ ddZ e j ddddgddZddZdS) TestBoxcoxcCstjjddddd}tj|dd}t||dtj|dd}t|dd|tj|dd}t|t|tjt|dd}t|t|dS) NrAr^i90)rKrLr)rr_r)r rrPboxcoxrr.rr)r"r0xtr'r'r(test_fixed_lmbdaszTestBoxcox.test_fixed_lmbdacCsLd}tjjdddd}||d| }t|\}}t|d|dd dS) Ng@raiPi)rIrKrLrr_r)r)r rOrPr&r)r"rr0x_invr'maxlogr'r'r(test_lmbda_Nones zTestBoxcox.test_lmbda_NonecCstjd}tjjdd|dd}tj|dd\}}}t|ddgtj|d d\}}}t|d d gtjjd d |dd}tj|dd\}}}t|ddgtj|dd\}}}t|ddgdS)NirAr^)rKrLg?)alphagx@gv[%@g?gxˉEk?g8Bhk @rir-gMbP?g\?gx̡&@g+?g7(U@gp: sY@)r.rMrr rrPr&r)r"rngr0rr:r'r'r( test_alphas zTestBoxcox.test_alphacCsRtddg}tttj|tttjtdgtttjtdgdggdS)Nr_rr)r.rXr*r+r r&)r"r0r'r'r(test_boxcox_bad_argszTestBoxcox.test_boxcox_bad_argcCsttgjdkdS)N)r)rr r&r)r"r'r'r(rszTestBoxcox.test_emptycCs tt\}}t|ddddS)NgsHjdrgh㈵>)r)r r& _boxcox_datar)r"rZlamr'r'r(r$szTestBoxcox.test_gh_6873bounds)r_r)g?r)r`gcsFfdd}tjtd|d\}}d|kr<dksBntdS)Ncstj|ddS)Nbounded)r1r)r minimize_scalar)fun)r1r'r( optimizerszBTestBoxcox.test_bounded_optimizer_within_bounds..optimizer)rr5rr)r r&r0r&)r"r1r5rrr')r1r($test_bounded_optimizer_within_boundss z/TestBoxcox.test_bounded_optimizer_within_boundscsjtjtdd\}}|d|dfddifdd}tjtd|d\}}||ksXtt|d dS) N)rg?rZxatolg-q=cstj|ddS)Nr2)r1roptions)r r3)r4)r1r7r'r(r5szPTestBoxcox.test_bounded_optimizer_against_unbounded_optimizer..optimizer)rr5r)r r&r0r&r)r"rrr5Z lmbda_boundedr')r1r7r(2test_bounded_optimizer_against_unbounded_optimizers  z=TestBoxcox.test_bounded_optimizer_against_unbounded_optimizerr5str)rrg?c Cs.tjtddtjtd|dWdQRXdS)Nz`optimizer` must be a callable)r)rr5)r/r r+r r&r0)r"r5r'r'r($test_bad_optimizer_type_raises_errorsz/TestBoxcox.test_bad_optimizer_type_raises_errorc Cs:dd}d}tjt|dtjtd|dWdQRXdS)NcSsdS)Nrr')r4r'r'r(r5szCTestBoxcox.test_bad_optimizer_value_raises_error..optimizerz;`optimizer` must return an object containing the optimal...)r)rr5)r/r r+r r&r0)r"r5rr'r'r(%test_bad_optimizer_value_raises_errorsz0TestBoxcox.test_bad_optimizer_value_raises_errorN)r5r6r7r(r+r.r/rr$r/r0r1r6r8r:r;r'r'r'r(r%s   r%c@sleZdZddZddZddZddZej d d d d gej d dddgddZ ddZ ddZ dS)TestBoxcoxNormmaxcCstjjddddd|_dS)NrAr^i90)rKrL)r rrPr0)r"r'r'r(r&szTestBoxcoxNormmax.setup_methodcCst|j}t|ddddS)Ngm?gư>)r)r boxcox_normmaxr0r)r"r*r'r'r( test_pearsonr)s zTestBoxcoxNormmax.test_pearsonrcCs<tj|jdd}t|dddt|j\}}t||dS)Nmle)rg .!?gư>)r)r r=r0rr&)r"r*rZ maxlog_boxcoxr'r'r(test_mle-szTestBoxcoxNormmax.test_mlecCs&tj|jdd}t|ddgdddS)Nrm)rgm?g .!?gư>)r)r r=r0r)r"Z maxlog_allr'r'r(test_all5szTestBoxcoxNormmax.test_allrr?Zpearsonrrmr1)r_r)g?r)r`gcsNfdd}tj|j||d}td|ks4tt|dksJtdS)Ncstj|ddS)Nr2)r1r)r r3)r4)r1r'r(r5=szITestBoxcoxNormmax.test_bounded_optimizer_within_bounds..optimizer)rr5rr)r r=r0r.rmr&)r"rr1r5r*r')r1r(r69s   z6TestBoxcoxNormmax.test_bounded_optimizer_within_boundscsvt|j}t|d}t|d|ddGdddfdd}tj|j|d}||ksftt||d dS) NrAg{Gz?ic@s eZdZdS)z?TestBoxcoxNormmax.test_user_defined_optimizer..MyResultN)r5r6r7r'r'r'r(MyResultLsrBcs:g}xD]}|||q W}t||_|S)N)appendr.Zargminr0)r4Zobjsrr2)rB lmbda_ranger'r(r5Os  z@TestBoxcoxNormmax.test_user_defined_optimizer..optimizer)r5gh㈵>)r r=r0r.roundrr&r)r"rZ lmbda_roundedr5Zlmbda2r')rBrDr(test_user_defined_optimizerFs    z-TestBoxcoxNormmax.test_user_defined_optimizerc CsHtj}tj|jd|dtjtddtj|jd|dWdQRXdS)N)rr5z,`brack` must be None if `optimizer` is given)r)gg@)r r3r r=r0r/r r+)r"r5r'r'r(2test_user_defined_optimizer_and_brack_raises_error\s  zDTestBoxcoxNormmax.test_user_defined_optimizer_and_brack_raises_errorN) r5r6r7rr>r@rAr/r0r1r6rFrGr'r'r'r(r<%s r<c@sFeZdZddZddZejje ddddZ d d Z d d Z d S)TestBoxcoxNormplotcCstjjddddd|_dS)NrAiit)rKrL)r rrPr0)r"r'r'r(rnszTestBoxcoxNormplot.setup_methodcCsNd}tj|jdd|d\}}ddddd g}t|tjdd|d t||dS) NrAira)rg!X4}?g΁i?g- 15?gW$?g\ܱ{?)r )r boxcox_normplotr0rr.r)r"rZlmbdasr r r'r'r(r)qs zTestBoxcoxNormplot.test_basicz no matplotlib)rcCsZt}|d}tj|jddtd|||d}tj|jdd|dtdS)Nrir)r)rrrr rIr0rr)r"rrr'r'r(rys   z"TestBoxcoxNormplot.test_plot_kwargcCs.tttj|jddtttjddgdddS)Nrrr_)r*r+r rIr0)r"r'r'r(rsz&TestBoxcoxNormplot.test_invalid_inputscCsttgddjdkdS)Nrr)rr rIrK)r"r'r'r(rszTestBoxcoxNormplot.test_emptyN) r5r6r7rr)r/r0rrrrrr'r'r'r(rHms  rHc@s$eZdZddZddZddZdS)TestYeojohnson_llfcCsDtjjdddd}d}t||}t|t|}t||dddS)NrHri1)rKrIrLrg-q=)r)r rOrPyeojohnson_llfrr)r"r0rrrr'r'r(r s  z"TestYeojohnson_llf.test_array_likecCsPtjjdddd}d}t||}t|t||gj}t||g|dddS)NrHrai1)rKrIrLrg-q=)r)r rOrPrKr.r!r?r)r"r0rrrr'r'r(r"s  z TestYeojohnson_llf.test_2d_inputcCstttdgdS)Nr)rr.r#r rK)r"r'r'r(rszTestYeojohnson_llf.test_emptyN)r5r6r7r r"rr'r'r'r(rJsrJc@seZdZddZejdddddgdd Zd d Zd d Z ejde j e j gddZ ejde je je je jgddZdS)TestYeojohnsoncCstjd}tjjdd|dd}t|dks4ttj|dd}t ||tj|dd}t |dd|dtj|dd}t |t |dtj|dd}t ||tjjdd|dd}t|dksttj|d d}t |t | d tj|dd}t ||tj|d d}t |d| ddtjjdd|dd }t|dkr`tt|dkrtt|dk}tj|dd}t ||||tj|dd}t ||dd||dtj|dd}t ||t ||dtj|dd}t |||||}tj|d d}t ||t || d tj|dd}t ||||tj|d d}t ||d|| dddS) Ni90rAr^)rKrLrr)rr_rr?) r.rMrr rrPrmr& yeojohnsonrr)r"r-r0r'posnegr'r'r(r(sL     zTestYeojohnson.test_fixed_lmbdarrg?g?rcCsdd}d}tjdtjjdd|d}|||}t|\}}t||dd tdtj |||d d td| dd td| dd dS) NcSstj|j|jd}|dk}t|tdkrDt||d||<n$t|||dd|d||<t|dtdkrdtd| ||ddd|||<ndt|| ||<|S)N)rWrg?rr)r.rrrWabsspacingexppower)r0rr)rNr'r'r(_inverse_transforms$z:TestYeojohnson.test_lmbda_None.._inverse_transformi Nirr)rIrJrKg{Gz?)rr)r) r.rMrNnormalr rMrrZlinalgrOr$r&)r"rrTZ n_samplesr0r)r'r*r'r'r(r+s  zTestYeojohnson.test_lmbda_NonecCsttgjdkdS)N)r)rr rMr)r"r'r'r(rszTestYeojohnson.test_emptycCsDtjjdddd}t|\}}tt|\}}t||dddS)NrHri1)rKrIrLg-q=)r)r rOrPrMrr)r"r0Zxt1rZxt2r'r'r(r szTestYeojohnson.test_array_likerWc Cs:tjd|d}d}tjt|dt|WdQRXdS)Nr)rWz>Yeo-Johnson transformation is not defined for complex numbers.)r)r.r/r/r r+r rM)r"rWr0err_msgr'r'r(test_input_dtype_complexsz'TestYeojohnson.test_input_dtype_complexcCsZtjd|d}tjdtjd}t|\}}t|\}}t||ddt||dddS)Nr)rWgHz>)r)r.r/Zfloat64r rMr)r"rWZx_intZx_floatZxt_intZ lmbda_intZxt_floatZ lmbda_floatr'r'r(test_input_dtype_integer s z'TestYeojohnson.test_input_dtype_integerN)r5r6r7r(r/r0r1r+rr r.Z complex64Z complex128rWZint8uint8Zint16Zint32rXr'r'r'r(rLs /%rLc@s$eZdZddZddZddZdS)TestYeojohnsonNormmaxcCstjjddddd|_dS)NrAr^i90)rKrL)r rrPr0)r"r'r'r(rsz"TestYeojohnsonNormmax.setup_methodcCst|j}t|ddddS)Ng?gư>)r)r yeojohnson_normmaxr0r)r"r*r'r'r(r@s zTestYeojohnsonNormmax.test_mlecCsDddddddddd d d d d ddg}t|}tj|ddds@tdS)Ngffffff@g g?g@gffffff?g333333@g @gffffff@g?g @g@g@g"@g@ggzG?gMbP?)r)r r[r.Zallcloser&)r"r0rr'r'r(test_darwin_examples z)TestYeojohnsonNormmax.test_darwin_exampleN)r5r6r7rr@r\r'r'r'r(rZsrZc@sLeZdZejdejdfejdfej dfgddZ ddZ ejd eje j feje jfej e jfgd d Zd d ZddZddZejdejdfejdfej dfgddZejdejejej gddZejdejejej gddZejdeje jdddfeje jdddfej e jdddfgd d!Zejdejd"e d#d$d%d&d'd(ge d)d*gdfejd+e d,d-d-d.d/d/ge dd0gdfej d1e d2d3d3d4d5d5ge d6d7gdfgd8d9Zejdejdfejdfej dfgd:d;Zejdejejej gdd?Zejd@dAdBdCdDdEdFe jge dAdBdCdDdEdFe jgdGdHdIdJe jdKdLgggejdejejej gdMdNZejd@dAdBdCdDdEdFe jge dAdBdCdDdEdFe jgdGdHdIdJe jdKdLgggejdejejej gdOdPZdQdRZ dSdTZ!dUdVZ"dWS)X TestCircFuncsztest_func,expectedgA:wv?gk~_ЀBE@gE|2@cCs0tddddddg}t||dd|d d dS) NicrArigrai^ih)r}gHz>)r)r.rXr)r" test_funcr+r0r'r'r(test_circfuncs'szTestCircFuncs.test_circfuncsc Cstdddddddg}|}tj|dd }t||d d |}tj|dd }t||d d |}tj |dd }t||d d dS) Nrrrrg4@g3333333@ih)r}gh㈵>)rg-C6?) r.rXr$r circmeanrr%circvarr&circstd)r"r0M1M2V1V2S1S2r'r'r(test_circfuncs_small/sz"TestCircFuncs.test_circfuncs_smallztest_func, numpy_funccCs<tdgddgd}||}||}t||dddS)Ng˛ɦv9?raguv9?rHg:0yU>)r)r.rXr)r"r^Z numpy_funcr0ZcircstatrUr'r'r(test_circfuncs_close=sz"TestCircFuncs.test_circfuncs_closec stddddddgddd d d d gd d ddd dggtjdd}tjdd}t||ddtjddd}fddtjdD}t||ddtjddd}fddtjdD}t||dddS)NicrArigrai^i_rr@i`ri]ierifidih)r}g+=)rr)r}rmcsg|]}tj|ddqS)ih)r})r ra)r=r3)r0r'r(r>Qsz4TestCircFuncs.test_circmean_axis..rcs&g|]}tjdd|fddqS)Nih)r})r ra)r=r3)r0r'r(r>Us)r.rXr raravelrrr)r"rdrer')r0r(test_circmean_axisHsz TestCircFuncs.test_circmean_axisc stddddddgddd d d d gd d ddd dggtjdd}tjdd}t||ddtjddd}fddtjdD}t||ddtjddd}fddtjdD}t||dddS)NicrArigrai^i_rr@i`ri]ierifidih)r}gdy=)rr)r}rmcsg|]}tj|ddqS)ih)r})r rb)r=r3)r0r'r(r>bsz3TestCircFuncs.test_circvar_axis..rcs&g|]}tjdd|fddqS)Nih)r})r rb)r=r3)r0r'r(r>fs)r.rXr rbrlrrr)r"rfrgr')r0r(test_circvar_axisXszTestCircFuncs.test_circvar_axisc stddddddgddd d d d gd d ddd dggtjdd}tjdd}t||ddtjddd}fddtjdD}t||ddtjddd}fddtjdD}t||dddS)NicrArigrai^i_rr@i`ri]ierifidih)r}gdy=)rr)r}rmcsg|]}tj|ddqS)ih)r})r rc)r=r3)r0r'r(r>ssz3TestCircFuncs.test_circstd_axis..rcs&g|]}tjdd|fddqS)Nih)r})r rc)r=r3)r0r'r(r>ws)r.rXr rcrlrrr)r"rhrir')r0r(test_circstd_axisiszTestCircFuncs.test_circstd_axiscCs*ddddddg}t||dd|d d dS) NicrArigrai^ih)r}gHz>)r)r)r"r^r+r0r'r'r(test_circfuncs_array_likezsz'TestCircFuncs.test_circfuncs_array_liker^cCstt|gdS)N)rr.r#)r"r^r'r'r(rszTestCircFuncs.test_emptycCs.ddddddtjg}tt||dddS) NicrArigrai^ih)r})r.r[rr#)r"r^r0r'r'r(test_nan_propagatesz TestCircFuncs.test_nan_propagateg5:v@gf Pb?)NrrgڴV0@gLԿAB@g#q@gauy+@c Cstdddddddgdd d d d d tjgdtjtjtjtjtjtjgg}xf|D]Z}||d|d}|dkr|tt|qRt|d||ddtt|ddqRWdS)NicrArigrai^ri_rr@i`ri]ih)r}rmrgHz>)r)r.rXr[keysrr#rrm)r"r^r+r0rmoutr'r'r(test_nan_propagate_arrays"z&TestCircFuncs.test_nan_propagate_arraygrQvv@gv@g@g@g8v@g#@gu@g5@v?g2>Bjv@gtUuYK@gR@gzw<5?g~/ϔ(@g K ?gsQP@gJPm$@gj@g?g)h; @g{?gun2@g+fa @c Cstddddddtjgddd d d d tjgtjtjtjtjtjtjtjgg}xh|D]\}||d d|d}|dkrt|||ddqVt|dd||ddtt|dqVWdS)NicrArigrai^i_rr@i`ri]ihomit)r} nan_policyrmgHz>)rr_)r.rXr[rrrrr#)r"r^r+r0rmrsr'r'r(test_nan_omit_arrays$z!TestCircFuncs.test_nan_omit_arraycCs0ddddddtjg}t||ddd |d d dS) NicrArigrai^ihru)r}rvgHz>)r)r.r[r)r"r^r+r0r'r'r( test_nan_omitszTestCircFuncs.test_nan_omitcCs2tjtjtjtjtjg}tt||dddS)Nru)rv)r.r[rr#)r"r^r0r'r'r(test_nan_omit_allszTestCircFuncs.test_nan_omit_allcCsjttjtjtjtjtjgtjtjtjtjtjgg}||ddd}tt|tt|dkdS)Nrur)rvrmr)r.rXr[rr#rmrq)r"r^r0rsr'r'r(test_nan_omit_all_axiss z$TestCircFuncs.test_nan_omit_all_axisr0icrArigrai^i_rr@i`ri]cCstt||ddddS)Nihraise)r}rv)r*r+)r"r^r0r'r'r(test_nan_raiseszTestCircFuncs.test_nan_raisecCstt||ddddS)Nihfoobar)r}rv)r*r+)r"r^r0r'r'r(test_bad_nan_policysz!TestCircFuncs.test_bad_nan_policycCs$d}|}t|}t||dddS)Ng?gh㈵>)r)r rar)r"r0rdrer'r'r(test_circmean_scalars z"TestCircFuncs.test_circmean_scalarcCs@ttdddtjtj }t|tjkt|tj kdS)Nrrg?)r rar.r/pir)r"mr'r'r(test_circmean_rangesz!TestCircFuncs.test_circmean_rangecCsZtjddgdd}ttj|dddttj|dddd d ttj|ddd d d dS) NrarY)rWr)r}g@e@g.}}VW{@gHz>)rg)#p_4@)r.rXr r rarrbrc)r"r0r'r'r(test_circfuncs_unit8sz"TestCircFuncs.test_circfuncs_unit8N)#r5r6r7r/r0r1r rarbrcr_rjr.r$r%r&rkrmrnrorprrqr[rtrXrwrxryrzr|r~rrrr'r'r'r(r]&s      r]c@sdeZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ dS)TestMedianTestcCstttjdddgdS)Nrrr?)r*r+r median_test)r"r'r'r(test_bad_n_samples sz!TestMedianTest.test_bad_n_samplescCstttjgdddgdS)Nrrr?)r*r+r r)r"r'r'r(r  sz TestMedianTest.test_empty_samplecCs,tttjddddgdddgddgdddS)Nrrrrl)ties)r*r+r r)r"r'r'r(test_empty_when_ties_ignored sz+TestMedianTest.test_empty_when_ties_ignoredcCs@tttjdddgdddgtttjdddgdddgdddS)Nrabove)r)r*r+r r)r"r'r'r(test_empty_contingency_row sz)TestMedianTest.test_empty_contingency_rowcCs"tttjdddgddgdddS)Nrrr?r@rAr )r)r*r+r r)r"r'r'r( test_bad_ties szTestMedianTest.test_bad_tiescCs"tttjdddgddgdddS)Nrrr?r@rAr})rv)r*r+r r)r"r'r'r(r~ sz"TestMedianTest.test_bad_nan_policycCs"tttjdddgddgdddS)Nrrr?r@rAr )r )r*rSr r)r"r'r'r(rT# szTestMedianTest.test_bad_keywordcCs`dddg}dddg}t||\}}}}t|dt|ddgddggt|dt|ddS)Nrrr?g@r)r rr r)r"r0rrrZmedtblr'r'r( test_simple' s    zTestMedianTest.test_simplecCsddddg}ddg}ddd g}t|||\}}}}t|dt|d ddgddd ggtj|||d d \}}}}t|dt|d ddgdd d ggtj|||d d \}}}}t|dt|d ddgdd d ggdS)Nrrr?r@rArrrrrrl)rr)r rr )r"r0rzrrrrr'r'r(test_ties_options6 s     z TestMedianTest.test_ties_optionscCsddtjg}dddg}tj||dd}tj||dd\}}}}t|tjtjtjdft|d t|d t|d t|td dgddggtttj||d ddS)Nrrr@rAr propagate)rvrug?gإvao?g@rr{) r.r[r rr rrXr*r+)r"r0rZmt1rwrrrr'r'r(test_nan_policy_optionsJ s     z&TestMedianTest.test_nan_policy_optionsc Cs<dddddg}ddddg}t||\}}}}t|dt|ddgddggt|\}}} } t||t||tj||dd \}}}}t|dt|ddgddggtj|dd \}}} } t||t||tj||d d \}}}}t|dt|ddgddggtj|d d \}}} } t||t||dS) Nrrr?r@rArrr)Zlambda_F)r)r rr Zchi2_contingencyr) r"r0rrrrrZexp_statZexp_pZdofer'r'r(r)W s(         zTestMedianTest.test_basicN)r5r6r7rr rrrr~rTrrrr)r'r'r'r(rs  r)JrBnumpyr.Z numpy.randomrZ numpy.testingrrrrrrr r r/r r*Zscipyr r Zscipy.stats.morestatsrZ common_testsrZ _hypotestsrZscipy.stats._binomtestrZ matplotlibZrcParamsZmatplotlib.pyplotZpyplotrr Exceptionr3r4r5r6r7r8r9r:r;r<rr8rGr]rr r2rCrXrcrirrrrrrrrrr0r%r<rHrJrLrZr]rr'r'r'r(s (          !ZVGE'SD^G6%) $$$$$$$$$$$$$$$$$$$$$ $$$$ qH#qY