B `<@sddlZddlZddlmZddlmZmZmZm Z m Z m Z m Z m Z ddlZddlmZddlmZddlmZdd lmZy&ddlZd ejd <ddlmZd ZWnek rd ZYnXddddddddddg Zddddddddddg Zddddddddddg Z ddddddddddg Z!ddddddddd dg Z"d!d"d!dddddd#dg Z#dd$dd%ddd&d'ddg Z$ddddddddddg Z%dddddd$d$dddg Z&ddddddddd$dg Z'Gd(d)d)e(Z)Gd*d+d+e(Z*Gd,d-d-e(Z+Gd.d/d/e(Z,Gd0d1d1e(Z-Gd2d3d3e(Z.Gd4d5d5e(Z/Gd6d7d7e(Z0Gd8d9d9e(Z1Gd:d;d;e(Z2Gdd?d?e(Z4Gd@dAdAe(Z5GdBdCdCe(Z6GdDdEdEe(Z7GdFdGdGe(Z8GdHdIdIe(Z9GdJdKdKe(Z:dLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[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ߐddddddddddgZ;Gddde(ZGddde(Z?Gddde(Z@Gddde(ZAGddde(ZBGddde(ZCdS(N) RandomState)assert_array_equalassert_almost_equalassert_array_lessassert_array_almost_equalassert_assert_allclose assert_equalsuppress_warnings)raises)stats)check_named_results)_get_wilcoxon_distrZAggbackendTFgjt?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$O/home/dcms/DCMS/lib/python3.7/site-packages/scipy/stats/tests/test_morestats.py test_basic-s  zTestBayes_mvs.test_basiccCstttjgdS)N) assert_raises ValueErrorr r)rr$r$r%test_empty_input?szTestBayes_mvs.test_empty_inputcCs4td}d}t|}x|D]}t||qWdS)N)rr)nparanger rr)rx attributesresir$r$r%test_result_attributesBs    z$TestBayes_mvs.test_result_attributesN)__name__ __module__ __qualname__r&r)r1r$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)rr r!r"r#r$r$r%r&Ls   zTestMvsdist.test_basiccCstttjgdS)N)r'r(r r6)rr$r$r%r)[szTestMvsdist.test_empty_inputcCsdg}tttj|dS)Nr )r'r(r r6)rr r$r$r% test_bad_arg^szTestMvsdist.test_bad_argc Cs\tJtdtddtdddgDddtddddd gDWdQRXdS) NerrorcSsg|] }|qSr$)r!).0r-r$r$r% hsz*TestMvsdist.test_warns..r rcSsg|] }|qSr$)r!)r:r-r$r$r%r;is)warningscatch_warnings simplefilterRuntimeWarningr r6)rr$r$r% test_warnscs  zTestMvsdist.test_warnsN)r2r3r4r&r)r8rCr$r$r$r%r5Ksr5c@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.dtjd/tjj d0d.d1d2}t|\}}t|}t|d3ddt|jd3ddt|d4d.dt|jd4d.dd5d6d7d8d9d:d;dd?d@dAdBd'dCdDdEdFdGdHdIdJdKdLg}dM}dN} t|\}}t|}t||dOdt|j|dOdt|| d0dt|j| d0ddS)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<iNar>d)locscalesizegE?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 shapirorrpvaluer+randomseednormrvs) rx1wpw shapiro_testx2Zx3Zx4Z W_expectedZ p_expectedr$r$r%r&msJ       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 rIrrrJ)rrOrPrQrRrSr$r$r%test_2ds$  zTestShapiro.test_2dcCs&tttjgtttjggggdS)N)r'r(r rI)rr$r$r%r)szTestShapiro.test_empty_inputcCs4tttjddgtttjtjgdggtddS)Nr r)dtype)r'r(r rIr+arrayobject)rr$r$r%test_not_enough_valuessz"TestShapiro.test_not_enough_valuescCsdg}tttj|dS)Nr )r'r(r rI)rr-r$r$r%r8szTestShapiro.test_bad_argcCs`td}tj|d<t|\}}t|}t|tjt|jtjt|dt|jddS)Ng$@rg?) r+r,nanr rIr rrrJ)rr-rPrQrRr$r$r%test_nan_inputs     zTestShapiro.test_nan_inputN) r2r3r4r&rTr)rXr8rZr$r$r$r%rDls ,rDc@sDeZdZddZddZddZddZd d Zd d Zd dZ dS) TestAndersoncCstd}|jdd}|jdd}t|\}}}t|dd|t|\}}}t||ddtd}d|d<t|\}}}t|ddS) NiI2)rH rg @) rstandard_exponentialstandard_normalr andersonrr+onesr)rrsrOrSAcritsigvr$r$r% test_normals    zTestAnderson.test_normalc Cstd}|jdd}|jdd}t|d\}}}t||ddtjddt|d\}}}WdQRXt||dkdS) NiIr\)rHZexponr^ignore)allr]) rr`rar rbrr+errstater)rrdrOrSrerfrgr$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)NrEgrgumbelr rr]) r+rcr rblengumbel_lfitlogcdflogsfr,r!r) rrhZa2rfrgnZxbarsrrrsr0Z expected_a2r$r$r% test_gumbels *zTestAnderson.test_gumbelcCstttjdgdddS)Nr plate_of_shrimp)dist)r'r(r rb)rr$r$r%r8szTestAnderson.test_bad_argcCs0td}|jdd}t|}d}t||dS)NiIr\)rH)rcritical_valuessignificance_level)rr`r rbr)rrdr-r/r.r$r$r%r1s   z#TestAnderson.test_result_attributesc CsFtd}|jdd}t|d\}}}t|d\}}}t||dS)NiIrE)rHrnrp)rrnr rbr) rrdr-A1crit1sig1A2crit2sig2r$r$r% test_gumbel_ls  zTestAnderson.test_gumbel_lc Cshtd}|jdd}td}t|d\}}}t|d\}}} t||ddt||dkdS)NiIrE)rHZgumbel_rr^r])rrnr+rcr rbrr) rrdrOrSr{r|r}r~rrr$r$r% test_gumbel_r s  zTestAnderson.test_gumbel_rN) r2r3r4rirmrvr8r1rrr$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 @rr>r=gJ4a?gMb0?)atol)r+rVr anderson_ksamprrr)rt1t2t3t4Tktmpr$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 @rr>r=gMb`?gMb0?)r)r+rVr rrrr)rrrrrrrrr$r$r%test_example1b,s 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 d6dndodKdrZr_<=18OT,;i6LFeJ0iri7i@h/iWxG*r> r<r 4_a3r=DMPjR6.o'?r\fHXrigii[rii#ibU[+ BF)rgM @g^I +?g~:?g+e?gH.!@gcZB>h@rg ^)p?gMb0?)r)r rrrr)rrrrrt5t6t7t8t9t10t11t12t13t14rrrr$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 d6dndodKdrr<rr rrrrr=rrrrrrrrrrrrrrrrr\rrrrrrrrrigii[riririrrrrrrrrrrT)rg'1Z @g^I +?g~:?g+e?gH.!@gcZB>h@rg ^)p?gMb0?)r)r rrrr)rrrrrrrrrrrrrrrrrrr$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)#Nr rEzp-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)rrOsupru_rr$r$r%test_R_kSamples|sB!&  "  &  "      z!TestAndersonKSamp.test_R_kSamplescCstttjtddS)Nr>)r'r(r rr+rc)rr$r$r%test_not_enough_samplessz)TestAndersonKSamp.test_not_enough_samplescCs"tttjtdtdfdS)Nr>)r'r(r rr+rc)rr$r$r%test_no_distinct_observationssz/TestAndersonKSamp.test_no_distinct_observationscCstttjtdgfdS)Nr>)r'r(r rr+rc)rr$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)rryrz)r+rVr rr)rrrr/r.r$r$r%r1s z(TestAndersonKSamp.test_result_attributesN) r2r3r4rrrrrrrr r1r$r$r$r%rs  Prc@s4eZdZddZddZddZddZd d Zd S) TestAnsaric Csldddddg}ddddddddg}t"}|tdt||\}}WdQRXt|ddt|d ddS) Nr rr<r=rz%Ties preclude use of exact statistic.g7@rg8#oG?)r rrr ansarir)rr-yrWpvalr$r$r% test_smalls  zTestAnsari.test_smallc Cs^td}td}t"}|tdt||\}}WdQRXt|ddt|dddS)N)rkrEcrrmlrrr`rrrqtrnr)rrrrigrrrrErrrrrrrrrrz%Ties preclude use of exact statistic.g0g@rg|:?)r+rVr rrr r r)rZramsayZparekhrr rr$r$r% test_approxs    zTestAnsari.test_approxc Cs@tddddgddddd d g\}}t|d d t|d ddS)Nr rr<r=r*r>rrr_rg$@rg?r)r r r)rr rr$r$r% test_exacts$ zTestAnsari.test_exactcCs(tttjgdgtttjdggdS)Nr )r'r(r r )rr$r$r%r8szTestAnsari.test_bad_argc Cs^dddddg}ddddddddg}t}|tdt||}WdQRXd}t||dS)Nr rr<r=rz%Ties preclude use of exact statistic.)rrJ)r rrr r r)rr-r rr/r.r$r$r%r1s z!TestAnsari.test_result_attributesN)r2r3r4rrrr8r1r$r$r$r%r s  r c@s4eZdZddZddZddZddZd d Zd S) TestBartlettc CsBtttttttttt g }t j |\}}t |ddt |dddS)NgzU@/4@rgG?) g1g2g3g4g5g6g7g8g9g10r bartlettr)rargsTrr$r$r% test_datas zTestBartlett.test_datacCstttjdgdS)Nr )r'r(r r()rr$r$r%r8szTestBartlett.test_bad_argc Cs4tttttttttt g }t j |}d}t ||dS)N)rrJ) rrr r!r"r#r$r%r&r'r r(r)rr)r/r.r$r$r%r1s z#TestBartlett.test_result_attributesc Cs6tttttttttt gf }t t j t j ft j|dS)N)rrr r!r"r#r$r%r&r'r r+rYr r()rr)r$r$r%test_empty_argszTestBartlett.test_empty_argcCs*tddgddgg}tttjt|dS)Nr rr<r=)r+rVr'r(r r(r)rr-r$r$r% test_1d_input$szTestBartlett.test_1d_inputN)r2r3r4r+r8r1r,r-r$r$r$r%r s rc@sTeZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ dS) TestLevenec CsBtttttttttt g }t j |\}}t |ddt |dddS)NgYpK?rgI{]?) rrr r!r"r#r$r%r&r'r levener)rr)r rr$r$r%r++s zTestLevene.test_datacCsFtjtttdd\}}tjtttddd\}}t||t||dS)Nr!)centertrimmedg)r0proportiontocut)r r/rrr r)rW1pval1W2pval2r$r$r% test_trimmed12s    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@ir1g?)r0r2r r]r!)r0)r+rKrL permutationr r/r) rr-r rSZW0Zpval0r3r4r5r6r$r$r% test_trimmed2;s      $  zTestLevene.test_trimmed2cCsntddd}tjdtj|}|d}tj||dd\}}tj||dd\}}t||t||dS) Nr]r rir<r!)r0Zmedian)r+rrKrLr8r r/r)rr-rSr r3r4r5r6r$r$r%test_equal_mean_medianMs   z!TestLevene.test_equal_mean_mediancCs&tddd}tttj||dddS)Nr]r rg?) portiontocut)r+rr' TypeErrorr r/)rr-r$r$r%test_bad_keywordWszTestLevene.test_bad_keywordcCs&tddd}tttj||dddS)Nr]r rtrim)r0)r+rr'r(r r/)rr-r$r$r%test_bad_center_value[sz TestLevene.test_bad_center_valuecCstttjdgdS)Nr )r'r(r r/)rr$r$r%test_too_few_args_szTestLevene.test_too_few_argsc Cs4tttttttttt g }t j |}d}t ||dS)N)rrJ) rrr r!r"r#r$r%r&r'r r/r)rr)r/r.r$r$r%r1bs z!TestLevene.test_result_attributescCs*tddgddgg}tttjt|dS)Nr rr<r=)r+rVr'r(r r/r)rr-r$r$r%r-iszTestLevene.test_1d_inputN) r2r3r4r+r7r9r:r=r?r@r1r-r$r$r$r%r.)s  r.c@s4eZdZddZddZddZddZd d Zd S) TestBinomPcCsRtdd}t|ddtdd}t|ddtjdd gd d }t|d ddS) NrEg-"^?rrig`w?ig?)rgz?)r binom_testr)rrr$r$r%r+ps     zTestBinomP.test_datacCstttjdddgdS)Nr rr<)r'r(r rD)rr$r$r%test_bad_len_xxszTestBinomP.test_bad_len_xcCs(tttjdgtttjdgdddS)NrEr\)rt)r'r(r rD)rr$r$r% test_bad_n|szTestBinomP.test_bad_ncCstttjddgdddS)Nr\g@)r)r'r(r rD)rr$r$r% test_bad_pszTestBinomP.test_bad_pcCsXtjddddd}t|dtjddddd}t|dtjdddd d}t|d dS) NrgUUUUUU?less)r alternativeg-$Ǻl?greatergs0,j.?z two-sidedgJfJ"f?)r rDr)rr/r$r$r%test_alternativess   zTestBinomP.test_alternativesN)r2r3r4r+rErFrGrLr$r$r$r%rAns rAc@sDeZdZddZddZddZddZd d Zd d Zd dZ dS) TestFlignercCs&td}tt||ddddS)Nr>r)gnf @gDq?r)r+r,rr fligner)rrOr$r$r%r+s zTestFligner.test_datac svtjdfdd}|t}|t}|t}tj|||dd\}}tj|||ddd\}}t||t||dS) N{cs t|dt|S)Ng|=)r+Zasarrayrandnrotolist)g)rdr$r%z+TestFligner.test_trimmed1..r!)r0r1g)r0r2) r+rKrrrr r rNr) rZ_perturbZg1_Zg2_Zg3_Xsq1r4Xsq2r6r$)rdr%r7s     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@r1g?)r0r2r r]r!)r0)r rNr)rr-r rUr4rVr6r$r$r%r9s  $ zTestFligner.test_trimmed2cCs&tddd}tttj||dddS)Nr]r rg?)r;)r+rr'r<r rN)rr-r$r$r%r=szTestFligner.test_bad_keywordcCs&tddd}tttj||dddS)Nr]r rr>)r0)r+rr'r(r rN)rr-r$r$r%r?sz!TestFligner.test_bad_center_valuecCstttjdgdS)Nr )r'r(r rN)rr$r$r%test_bad_num_argsszTestFligner.test_bad_num_argscCs.td}ttjtjft||dgdS)Nr>r)r+r,r rYr rN)rr-r$r$r%r,s zTestFligner.test_empty_argN) r2r3r4r+r7r9r=r?rWr,r$r$r$r%rMsrMc@s<eZdZddZddZddZddZd d Zd d Zd S)TestMoodcCs&td}tt||ddddS)Nr>r)g8!giT?r)r+r,rr mood)rrOr$r$r% test_moods zTestMood.test_moodcCs`tjdtjdd}tjdd}t||\}}t||\}}t||g| |gdS)Nir_r r*)r+rKrLrPr rYr)rrOrSZz1p1Zz2p2r$r$r%test_mood_order_of_argss  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?)r_r)r*r)axisg4g^}E ?)r+rVshaperr rY)rrOrSr$r$r%test_mood_with_axis_nones.   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)Nr>ir_r*r )r^) r+rKrLrPr rYrangerZ transpose)rnyrOrS z_vectest pval_vectestjr0r$r$r% test_mood_2ds *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)r_r>rir<)r^r rr)r+rKrLrPrar rYr) rr_rOrSr^rcrdZaxes_idxr0reZslice1Zslice2r$r$r% test_mood_3ds(     zTestMood.test_mood_3dcCstttjdggdS)Nr )r'r(r rY)rr$r$r%test_mood_bad_arg6szTestMood.test_mood_bad_argN) r2r3r4rZr]r`rfrgrhr$r$r$r%rXs  rXc@sVeZdZddZddZddZejje ddd d Z d d Z d dZ ddZ dS) TestProbplotcCstjdtjjdd}tj|dd\}}dddd d d d d ddddddddddddg}t|t|t||tj|dd\}}dddg}t||dS)Ni90r)rHF)rqg+/gY3gtg5Z0gFjg!2gsR{ܿg7Կg>hǿgcz隇gcz隇?g>h?g7?gsR{?g!2?gFj?g5Z0?gt?gY3?g+/?TgƝV?g?gà?) r+rKrLr rMrNprobplotrsort)rr-osmosrZ osm_expectedr/Zres_fitZres_fit_expectedr$r$r%r&>s    zTestProbplot.test_basicc Cstjdtjjdd}tj|ddd\}}tj|ddd\}}tj|ddd\}}t||t||t||t||tj|ddd\}} dS)Ni@rE)rHF)sparamsrqrr$)r+rKrLr rMrNrjr) rr-osm1osr1osm2osr2Zosm3Zosr3rlrmr$r$r%test_sparams_keywordNs     z!TestProbplot.test_sparams_keywordcCstjdtjjdd}tj|dddd\}}tj|dtjdd\}}t||t||t t tj|dd t t tj|gd Gd d d t }tj|d dd \}}tj||dd\}}t||t||dS)Ni90r)rHFt)r<)rqrxrnzwrong-dist-name)rxc@seZdZdZddZdS)z3TestProbplot.test_dist_keyword..custom_distz6Some class that looks just enough like a distribution.cSstjj|ddS)Nr)rF)r rMppf)rqr$r$r%rujsz7TestProbplot.test_dist_keyword..custom_dist.ppfN)r2r3r4__doc__rur$r$r$r% custom_disthsrx)r)rnrq)rxrq) r+rKrLr rMrNrjrtrr'r(AttributeErrorrW)rr-rorprqrrrxr$r$r%test_dist_keyword]s    zTestProbplot.test_dist_keywordz no matplotlib)reasonc Cstjdt}|dtjjddd}tj |td\}}t tj |dd\}}tj |dtd}t tj |ddd}t t |t |kot |kot |kod knt ||t ||t ||t ||t}|d} tj |d| dt dS) Nitrr<rE)rH)plotF)rqr|r)r+rKrLpltfigure add_subplotr rtrNrjcloserror) rfigr-Zres1Zfitres1Zres2Zfitres2Zres3Zres4axr$r$r%test_plot_kwargrs&  @     zTestProbplot.test_plot_kwargcCstttjdgdddS)Nr rw)rx)r'r(r rj)rr$r$r%test_probplot_bad_argssz#TestProbplot.test_probplot_bad_argscCsZttjgddtgtgfttjgddtgtgftjtjdffdS)NF)rqTg)r r rjr+rVrY)rr$r$r% test_emptys zTestProbplot.test_emptyc CsTtjdd>ttjdgddtdgtdgftjtjdffWdQRXdS)Nrj)invalidr T)rqg)r+rlr r rjrVrY)rr$r$r%test_array_of_size_onesz#TestProbplot.test_array_of_size_oneN)r2r3r4r&rsrzpytestmarkskipifhave_matplotlibrrrrr$r$r$r%ri<sric @seZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ e jddddgdddddgdddddddd ggd!d"Zd#d$Zd%S)& TestWilcoxoncCshtttjdgddgtttjddgddgdtttjddgddgddtttjdgddddS)Nr rdummy)rJr_Zxyz)mode)r'r(r wilcoxon)rr$r$r%test_wilcoxon_bad_args z"TestWilcoxon.test_wilcoxon_bad_argcCsRtd}tttj||dddtttj||dddttj||dddddS)Nrwilcoxapprox)rprattzsplit)g@Z@g?)r+r,r'r(r rr )rr-r$r$r%test_zero_diffs zTestWilcoxon.test_zero_diffc CsXddddg}ddddg}t&}|jtddtj||dd d }WdQRXt|d dS) Nr rr<r=r>zSample size too small)rrr) zero_methodr)gg |N?)r rrr rr)rr-r rr/r$r$r% test_pratts   zTestWilcoxon.test_prattc CsNdddddddddddg }tj|ddd}tj|d dd}tj|d dd}dS) Nr rr<rr]rr)rrrr)r r)rZarrrr$r$r%test_wilcoxon_arg_typesz#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)3Nr r=rr*rr>rcSsg|]\}}|g|qSr$r$)r:urhr$r$r%r;sz7TestWilcoxon.test_accuracy_wilcoxon..rr)rig.i?rigFYv]Uj?riGg%^Ez?rrrryrrpwsrF) correctionrrg^҂)rTg:b+?) rar+Z concatenatezipzerosrHr rrrVr )rfreqnumsr-r r*rr$r$r%test_accuracy_wilcoxons*        ""  z#TestWilcoxon.test_accuracy_wilcoxoncCshtddddddddd d d d g }td dddd ddd ddddg }tj||ddd}d}t||dS)NrrrrrrrrrrrrrrrrrrrrrFr)rr)rrJ)r+rVr rr)rr-r r/r.r$r$r%test_wilcoxon_result_attributess ""z,TestWilcoxon.test_wilcoxon_result_attributescCsntjdgddd\}}d}t|dt||ddtjdgdd dd \}}d }t|dt||dddS) Ng?r_r)rg4IťY?rgư>)rT)rrg c2_?)r rr r)rstatrZ expected_pr$r$r%test_wilcoxon_ties   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|rOzSample size too small)rrIr)rJrrg13}?r)rT)rJrrg7a%?rKg?gڏoJT?)r rrr rr r)rr-r rrPrr$r$r% test_onesideds0         zTestWilcoxon.test_onesidedcCsPxJtddD]<}t|}t||dddt|tt|d|q WdS)Nr rr)rarr rosum)rrtZcntr$r$r%test_exact_basic#szTestWilcoxon.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)rJrg֔?r)rrIgYJ?rKg*?rrg?r]g(?g(?gS?rr )r)r+rVr rrr,r'r()rr-r rrdr$r$r%test_exact_pval)s(  zTestWilcoxon.test_exact_pvalr-r]r^r<rrr>irrcCs@t|\}}t|}||dk}t||t|ddS)Nrr )r rr+rVrr )rr-rPrZwtruer$r$r%test_exact_p_1Gs   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)rrr r)r+r,r r rr rr)rr-r rrrPrr$r$r% test_autoQs   zTestWilcoxon.test_autoN)r2r3r4rrrrrrrrrrrr parametrizerrr$r$r$r%rs    &  rc@s,eZdZddZddZddZddZd S) TestKstatcstjdtjdfdddD}dddd g}t||d d tjd d }tjdd }tjdd }t|||f|ddddddS)Ni}i90csg|]}t|qSr$)r kstat)r:rt)r r$r%r;hsz>TestKstat.test_moments_normal_distribution..)r rr<r=gHQ,?glZ)rI?g88}?gcJN?g-C6?)rr )momentrr<r]g{Gz?g{Gz?)rr)r+rKrLrPrr r)rZmomentsexpectedm1m2Zm3r$)r r% test_moments_normal_distributiones   z*TestKstat.test_moments_normal_distributioncCstttjgdS)N)r'r(r r)rr$r$r%r)sszTestKstat.test_empty_inputcCs*td}tj|d<tt|tjdS)Ng$@r)r+r,rYr r r)rr r$r$r%rZvs  zTestKstat.test_nan_inputcCs.td}xdD]}tttj||dqWdS)Nr_)rg/$@)rt)r+r,r'r(r r)rr rtr$r$r%test_kstat_bad_arg|s  zTestKstat.test_kstat_bad_argN)r2r3r4rr)rZrr$r$r$r%rdsrc@s$eZdZddZddZddZdS) TestKstatVarcCstttjgdS)N)r'r(r kstatvar)rr$r$r%r)szTestKstatVar.test_empty_inputcCs*td}tj|d<tt|tjdS)Ng$@r)r+r,rYr r r)rr r$r$r%rZs  zTestKstatVar.test_nan_inputcCs dg}d}tttj||ddS)Nr r_)rt)r'r(r r)rr rtr$r$r%r8szTestKstatVar.test_bad_argN)r2r3r4r)rZr8r$r$r$r%rsrc@sNeZdZddZddZddZejje ddd d Z d d Z d dZ dS) TestPpccPlotcCs&tjdtjjdddd|_dS)Nitr>i)rH)r+rKrLr loggammarNr-)rr$r$r% setup_methods zTestPpccPlot.setup_methodcCsNd}tj|jdd|d\}}ddddd g}t|tjdd|d t||dS) Nr>ir_)Ng9% ?gxM _?g"; 9?g®XZ?g?)num)r ppcc_plotr-rr+r)rrsvalsppcc ppcc_expectedr$r$r%r&s zTestPpccPlot.test_basiccCstj|jdddd\}}tj|jddtjd\}}t||ddt||ddt|jdd\}}t||ddt||dddS)Nir_ tukeylambda)rxg#B ;)r)r rr-rr)rZsvals1Zppcc1Zsvals2Zppcc2Zsvals3Zppcc3r$r$r% test_dists zTestPpccPlot.test_distz no matplotlib)r{cCsZt}|d}tj|jddtd|||d}tj|jdd|dtdS)Nrir)r|)r}r~rr rr-delaxesr)rrrr$r$r%rs   zTestPpccPlot.test_plot_kwargcCs4tttj|jddtttjdddgdddddS)Nr rrr<rw)rx)r'r(r rr-)rr$r$r%test_invalid_inputssz TestPpccPlot.test_invalid_inputscCs@tgdd\}}t|tjddddt|tjdtddS)Nrr r)r)rU)r rrr+rrfloat)rrrr$r$r%rszTestPpccPlot.test_emptyN) r2r3r4rr&rrrrrrrrr$r$r$r%rs   rc@s,eZdZddZddZddZddZd S) TestPpccMaxcCsdg}tttj|dddS)Nr rw)rx)r'r(r ppcc_max)rr r$r$r%test_ppcc_max_bad_argsz!TestPpccMax.test_ppcc_max_bad_argcCs<tjdtjjdddddd}tt|dd d dS) Nigffffffrg?i')rFrGrHg@g{r)r)r+rKrLr rrNrr)rr-r$r$r%test_ppcc_max_basics zTestPpccMax.test_ppcc_max_basiccCsztjdtjjdddddd}tj|dd }tj|tjd }t|d d d t|d d d t|}t|d d d dS) Nigffffffrg?i')rFrGrHg@r)rxg{r>)r)r+rKrLr rrNrr)rr-Zmax1Zmax2Zmax3r$r$r%rs  zTestPpccMax.test_distcCsjtjdtjjdddddd}tttj|dd t tj|d d d d d t tj|dd d d d dS)Nigffffffrg?i')rFrGrHg@)gg?g?)Zbrack)rr g{r)r)r^r) r+rKrLr rrNr'r(rr)rr-r$r$r% test_bracks  zTestPpccMax.test_brackN)r2r3r4rrrrr$r$r$r%rsrc@s4eZdZddZddZddZddZd d Zd S) TestBoxcox_llfcCs^tjdtjjddd}d}t||}|j dtt | d}t ||dS)Ni1i'r_)rHrFr g@r) r+rKrLr rMrN boxcox_llfrHlogrr#r)rr-lmbdallfZ llf_expectedr$r$r%r&s   $zTestBoxcox_llf.test_basiccCsNtjdtjjddd}d}t||}t|t|}t||dddS)Ni1rEr_)rHrFr g-q=)r) r+rKrLr rMrNrlistr)rr-rrllf2r$r$r%test_array_likes   zTestBoxcox_llf.test_array_likecCsZtjdtjjddd}d}t||}t|t||gj}t ||g|dddS)Ni1rEr_)rHrFr g-q=)r) r+rKrLr rMrNrvstackr*r)rr-rrrr$r$r% test_2d_inputs   zTestBoxcox_llf.test_2d_inputcCstttdgdS)Nr )rr+isnanr r)rr$r$r%r szTestBoxcox_llf.test_emptycCs&ddddg}td|}t|ddS)Ngh@g m@gx@ig#:x1)r rr)rr rr$r$r% test_gh_6873s  zTestBoxcox_llf.test_gh_6873N)r2r3r4r&rrrrr$r$r$r%rs  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@s<eZdZddZddZddZddZd d Zd d Zd S) TestBoxcoxcCstjdtjjdddd}tj|dd}t||dtj|dd}t|dd|tj|dd}t|t|tjt |dd}t|t|dS) Ni90r>r\)rHr )rr]r) r+rKrLr rrNboxcoxrrr)rr-xtr$r$r%test_fixed_lmbdaOs zTestBoxcox.test_fixed_lmbdacCsbtjdtjdd}tjjddd}||d| }t|\}}t|d|d d dS) Niig@r_iP)rFrHr r]r)r)r+rKrLr rMrNrr)rrr-x_invrmaxlogr$r$r%test_lmbda_None^s  zTestBoxcox.test_lmbda_NonecCstjdtjjdddd}tj|dd\}}}t|ddgtj|d d\}}}t|d d gtjjd d dd}tj|dd\}}}t|ddgtj|dd\}}}t|ddgdS)Nir>r\)rHg?)alphagx@gv[%@g?gxˉEk?g8Bhk @rir*gMbP?g\?gx̡&@g+?g7(U@gp: sY@)r+rKrLr rrNrr)rr-rr7r$r$r% test_alphajs zTestBoxcox.test_alphacCsRtddg}tttj|tttjtdgtttjtdgdggdS)Nr]rr )r+rVr'r(r r)rr-r$r$r%test_boxcox_bad_arg{szTestBoxcox.test_boxcox_bad_argcCsttgjdkdS)N)r)rr rr_)rr$r$r%rszTestBoxcox.test_emptycCs tt\}}t|ddddS)NgsHjdrgh㈵>)r)r r _boxcox_datar)rr Zlamr$r$r%rszTestBoxcox.test_gh_6873N) r2r3r4rrrrrrr$r$r$r%rMs   rc@s,eZdZddZddZddZddZd S) TestBoxcoxNormmaxcCs&tjdtjjdddd|_dS)Ni90r>r\)rH)r+rKrLr rrNr-)rr$r$r%rs zTestBoxcoxNormmax.setup_methodcCst|j}t|ddddS)Ngm?gư>)r)r boxcox_normmaxr-r)rrr$r$r% test_pearsonrs zTestBoxcoxNormmax.test_pearsonrcCs<tj|jdd}t|dddt|j\}}t||dS)NZmle)methodg .!?gư>)r)r rr-rr)rrrZ maxlog_boxcoxr$r$r%test_mleszTestBoxcoxNormmax.test_mlecCs&tj|jdd}t|ddgdddS)Nrk)rgm?g .!?gư>)r)r rr-r)rZ maxlog_allr$r$r%test_allszTestBoxcoxNormmax.test_allN)r2r3r4rrrrr$r$r$r%rsrc@sFeZdZddZddZejje ddddZ d d Z d d Z d S)TestBoxcoxNormplotcCs&tjdtjjdddd|_dS)Nitr>i)rH)r+rKrLr rrNr-)rr$r$r%rs zTestBoxcoxNormplot.setup_methodcCsNd}tj|jdd|d\}}ddddd g}t|tjdd|d t||dS) Nr>ir_)rg!X4}?g΁i?g- 15?gW$?g\ܱ{?)r)r boxcox_normplotr-rr+r)rrZlmbdasrrr$r$r%r&s zTestBoxcoxNormplot.test_basicz no matplotlib)r{cCsZt}|d}tj|jddtd|||d}tj|jdd|dtdS)Nrir)r|)r}r~rr rr-rr)rrrr$r$r%rs   z"TestBoxcoxNormplot.test_plot_kwargcCs.tttj|jddtttjddgdddS)Nr rr])r'r(r rr-)rr$r$r%rsz&TestBoxcoxNormplot.test_invalid_inputscCsttgddjdkdS)Nrr )rr rrH)rr$r$r%rszTestBoxcoxNormplot.test_emptyN) r2r3r4rr&rrrrrrrr$r$r$r%rs  rc@s$eZdZddZddZddZdS)TestYeojohnson_llfcCsNtjdtjjddd}d}t||}t|t|}t||dddS)Ni1rEr)rHrFr g-q=)r) r+rKrLr rMrNyeojohnson_llfrr)rr-rrrr$r$r%rs   z"TestYeojohnson_llf.test_array_likecCsZtjdtjjddd}d}t||}t|t||gj}t ||g|dddS)Ni1rEr_)rHrFr g-q=)r) r+rKrLr rMrNrrr*r)rr-rrrr$r$r%rs   z TestYeojohnson_llf.test_2d_inputcCstttdgdS)Nr )rr+rr r)rr$r$r%rszTestYeojohnson_llf.test_emptyN)r2r3r4rrrr$r$r$r%rsrc@seZdZddZejdddddgdd Zd d Zd d Z ejde j e j gddZ ejde je je je jgddZdS)TestYeojohnsoncCstjdtjjdddd}t|dks2ttj|dd}t ||tj|dd}t |dd|dtj|dd}t |t |dtj|dd}t ||tjjdddd}t|dksttj|d d}t |t | d tj|dd}t ||tj|d d}t |d| ddtjjdddd }t|dkrZtt|dkrnt|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) Ni90r>r\)rHrr )rr]rr<) r+rKrLr rrNrkAssertionError yeojohnsonrr)rr-rposnegr$r$r%rsL     zTestYeojohnson.test_fixed_lmbdarrg?g?rcCsdd}tjdd}tjjdd|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)rUrg?r r)r+rr_rUabsspacingexppower)r-rrrr$r$r%_inverse_transforms$z:TestYeojohnson.test_lmbda_None.._inverse_transformii Nrr )rFrGrHg{Gz?)rr)r) r+rKrLnormalr rrrZlinalgrMr!r#)rrr Z n_samplesr-rrrr$r$r%rs  zTestYeojohnson.test_lmbda_NonecCsttgjdkdS)N)r)rr rr_)rr$r$r%r6szTestYeojohnson.test_emptycCsNtjdtjjddd}t|\}}tt|\}}t||dddS)Ni1rEr)rHrFg-q=)r) r+rKrLr rMrNrrr)rr-Zxt1rZxt2r$r$r%r9s  zTestYeojohnson.test_array_likerUc Cs:tjd|d}d}tjt|dt|WdQRXdS)Nr)rUz>Yeo-Johnson transformation is not defined for complex numbers.)match)r+r,rr r(r r)rrUr-err_msgr$r$r%test_input_dtype_complex@sz'TestYeojohnson.test_input_dtype_complexcCsZtjd|d}tjdtjd}t|\}}t|\}}t||ddt||dddS)Nr)rUgHz>)r)r+r,float64r rr)rrUZx_intZx_floatZxt_intZ lmbda_intZxt_floatZ lmbda_floatr$r$r%test_input_dtype_integerHs z'TestYeojohnson.test_input_dtype_integerN)r2r3r4rrrrrrrr+Z complex64Z complex128rZint8uint8int16int32rr$r$r$r%rs /%rc@s$eZdZddZddZddZdS)TestYeojohnsonNormmaxcCs&tjdtjjdddd|_dS)Ni90r>r\)rH)r+rKrLr rrNr-)rr$r$r%rSs z"TestYeojohnsonNormmax.setup_methodcCst|j}t|ddddS)Ng?gư>)r)r yeojohnson_normmaxr-r)rrr$r$r%rWs 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 rr+Zallcloser)rr-rr$r$r%test_darwin_example[s z)TestYeojohnsonNormmax.test_darwin_exampleN)r2r3r4rrrr$r$r$r%rRsrc@seZdZejdejdfejdfej dfgddZ ddZ d d Z d d Z d dZejdejdfejdfej dfgddZejdejejej gddZejdejejej gddZejdejejdddfejejdddfej ejdddfgddZejdejded d!d"d#d$d%ged&d'gdfejd(ed)d*d*d+d,d,gedd-gdfej d.ed/d0d0d1d2d2ged3d4gdfgd5d6Zejdejdfejdfej dfgd7d8Zejdejejej gd9d:Zejdejejej gd;d<Zejd=d>d?d@dAdBdCejged>d?d@dAdBdCejgdDdEdFdGejdHdIgggejdejejej gdJdKZejd=d>d?d@dAdBdCejged>d?d@dAdBdCejgdDdEdFdGejdHdIgggejdejejej gdLdMZdNdOZdPdQZdRdSZdTS)U TestCircFuncsztest_func,expectedgA:wv?gk~_ЀBE@gE|2@cCs0tddddddg}t||dd|d d dS) Nicr>rigr_i^ih)highgHz>)r)r+rVr)r test_funcrr-r$r$r%test_circfuncseszTestCircFuncs.test_circfuncsc Cstdddddddg}|}tj|dd }t||d d |}tj|dd }t||d d |}tj |dd }t||d d dS) Nrrrrg4@g3333333@ih)rgh㈵>)rg-C6?) r+rVr!r circmeanrr"circvarr#circstd)rr-M1M2V1V2S1S2r$r$r%test_circfuncs_smallmsz"TestCircFuncs.test_circfuncs_smallc stddddddgddd d d d gd d ddd dggtjdd}tjdd}t||ddtjddd}fddtjdD}t||ddtjddd}fddtjdD}t||dddS)Nicr>rigr_i^i_rr=i`ri]ierifidih)rg+=)rr )rr^csg|]}tj|ddqS)ih)r)r r)r:r0)r-r$r%r;sz4TestCircFuncs.test_circmean_axis..rcs&g|]}tjdd|fddqS)Nih)r)r r)r:r0)r-r$r%r;s)r+rVr rravelrrar_)rr r!r$)r-r%test_circmean_axis{sz TestCircFuncs.test_circmean_axisc stddddddgddd d d d gd d ddd dggtjdd}tjdd}t||ddtjddd}fddtjdD}t||ddtjddd}fddtjdD}t||dddS)Nicr>rigr_i^i_rr=i`ri]ierifidih)rgdy=)rr )rr^csg|]}tj|ddqS)ih)r)r r)r:r0)r-r$r%r;sz3TestCircFuncs.test_circvar_axis..rcs&g|]}tjdd|fddqS)Nih)r)r r)r:r0)r-r$r%r;s)r+rVr rr'rrar_)rr"r#r$)r-r%test_circvar_axisszTestCircFuncs.test_circvar_axisc stddddddgddd d d d gd d ddd dggtjdd}tjdd}t||ddtjddd}fddtjdD}t||ddtjddd}fddtjdD}t||dddS)Nicr>rigr_i^i_rr=i`ri]ierifidih)rgdy=)rr )rr^csg|]}tj|ddqS)ih)r)r r)r:r0)r-r$r%r;sz3TestCircFuncs.test_circstd_axis..rcs&g|]}tjdd|fddqS)Nih)r)r r)r:r0)r-r$r%r;s)r+rVr rr'rrar_)rr$r%r$)r-r%test_circstd_axisszTestCircFuncs.test_circstd_axiscCs*ddddddg}t||dd|d d dS) Nicr>rigr_i^ih)rgHz>)r)r)rrrr-r$r$r%test_circfuncs_array_likesz'TestCircFuncs.test_circfuncs_array_likercCstt|gdS)N)rr+r)rrr$r$r%rszTestCircFuncs.test_emptycCs.ddddddtjg}tt||dddS) Nicr>rigr_i^ih)r)r+rYrr)rrr-r$r$r%test_nan_propagatesz TestCircFuncs.test_nan_propagateg5:v@gf Pb?)Nrr gڴV0@gLԿAB@g#q@gauy+@c Cstdddddddgdd d d d d tjgdtjtjtjtjtjtjgg}xf|D]Z}||d|d}|dkr|tt|qRt|d||ddtt|ddqRWdS)Nicr>rigr_i^r i_rr=i`ri]ih)rr^rgHz>)r)r+rVrYkeysrrrrk)rrrr-r^outr$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)Nicr>rigr_i^i_rr=i`ri]ihomit)r nan_policyr^gHz>)rr])r+rVrYr-rrr)rrrr-r^r.r$r$r%test_nan_omit_arrays$z!TestCircFuncs.test_nan_omit_arraycCs0ddddddtjg}t||ddd |d d dS) Nicr>rigr_i^ihr0)rr1gHz>)r)r+rYr)rrrr-r$r$r% test_nan_omitszTestCircFuncs.test_nan_omitcCs2tjtjtjtjtjg}tt||dddS)Nr0)r1)r+rYrr)rrr-r$r$r%test_nan_omit_allszTestCircFuncs.test_nan_omit_allcCsjttjtjtjtjtjgtjtjtjtjtjgg}||ddd}tt|tt|dkdS)Nr0r )r1r^r)r+rVrYrrrkro)rrr-r.r$r$r%test_nan_omit_all_axiss z$TestCircFuncs.test_nan_omit_all_axisr-icr>rigr_i^i_rr=i`ri]cCstt||ddddS)Nihraise)rr1)r'r()rrr-r$r$r%test_nan_raise szTestCircFuncs.test_nan_raisecCstt||ddddS)Nihfoobar)rr1)r'r()rrr-r$r$r%test_bad_nan_policysz!TestCircFuncs.test_bad_nan_policycCs$d}|}t|}t||dddS)Ng?gh㈵>)r)r rr)rr-r r!r$r$r%test_circmean_scalars z"TestCircFuncs.test_circmean_scalarcCs@ttdddtjtj }t|tjkt|tj kdS)Nrrg?)r rr+r,pir)rmr$r$r%test_circmean_range!sz!TestCircFuncs.test_circmean_rangecCsZtjddgdd}ttj|dddttj|dddd d ttj|ddd d d dS) Nr_r)rUr)rg@e@g.}}VW{@gHz>)rg)#p_4@)r+rVr r rrrr)rr-r$r$r%test_circfuncs_unit8(sz"TestCircFuncs.test_circfuncs_unit8N)r2r3r4rrrr rrrrr&r(r)r*r+rr,r+rYr/rVr2r3r4r5r7r9r:r=r?r$r$r$r%rds    rc@sdeZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ dS)TestMedianTestcCstttjdddgdS)Nr rr<)r'r(r median_test)rr$r$r%test_bad_n_samples3sz!TestMedianTest.test_bad_n_samplescCstttjgdddgdS)Nr rr<)r'r(r rA)rr$r$r%r 7sz TestMedianTest.test_empty_samplecCs,tttjddddgdddgddgdddS)Nr rrrj)ties)r'r(r rA)rr$r$r%test_empty_when_ties_ignored;sz+TestMedianTest.test_empty_when_ties_ignoredcCs@tttjdddgdddgtttjdddgdddgdddS)Nr above)rC)r'r(r rA)rr$r$r%test_empty_contingency_rowCsz)TestMedianTest.test_empty_contingency_rowcCs"tttjdddgddgdddS)Nr rr<r=r>foo)rC)r'r(r rA)rr$r$r% test_bad_tiesOszTestMedianTest.test_bad_tiescCs"tttjdddgddgdddS)Nr rr<r=r>r8)r1)r'r(r rA)rr$r$r%r9Ssz"TestMedianTest.test_bad_nan_policycCs"tttjdddgddgdddS)Nr rr<r=r>rG)rG)r'r<r rA)rr$r$r%r=VszTestMedianTest.test_bad_keywordcCs`dddg}dddg}t||\}}}}t|dt|ddgddggt|dt|ddS)Nr rr<g@r)r rAr r)rr-r rrZmedtblr$r$r% test_simpleZs    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)Nr rr<r=r>rrrrrrj)rCrE)r rAr )rr-r zrrr<rIr$r$r%test_ties_optionsis     z TestMedianTest.test_ties_optionscCsddtjg}dddg}tj||dd}tj||dd\}}}}t|tjtjtjdft|d t|d t|d t|td dgddggtttj||d ddS)Nr rr=r>r propagate)r1r0g?gإvao?g@rr6) r+rYr rAr rrVr'r()rr-r Zmt1rurr<rtr$r$r%test_nan_policy_options}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) Nr rr<r=r>rrr)Zlambda_F)r)r rAr Zchi2_contingencyr) rr-r rrr<rIZexp_statZexp_pZdofer$r$r%r&s(         zTestMedianTest.test_basicN)r2r3r4rBr rDrFrHr9r=rJrLrNr&r$r$r$r%r@1s  r@)Dr?numpyr+Z numpy.randomrZ numpy.testingrrrrrrr r rr r'Zscipyr Z common_testsrZ _hypotestsrZ matplotlibZrcParamsZmatplotlib.pyplotZpyplotr}r Exceptionrrr r!r"r#r$r%r&r'rWrr5rDr[rr rr.rArMrXrirrrrrrrrrrrrrrr@r$r$r$r%s (       ![SG,E#DgbG7%,$$$$$$$$$$$$$$$$$$$$$ $$$$ C$rN