3 Od<@sddlZddlZddlmZddlmZmZmZm Z m Z m Z m Z m Z ddlZddlmZddlmZddlmZdd lmZy$ddlZd ejd <ddljZd 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}tj|\}}}t|jdt|jdd d t|jd t|jddd t|jdddt|jdd d dS)N  g"@4`H#j@O[%@g+=)rtolg$@ai @Ӈu8@g& .>g _紫@)decimalYs?疡1O@)rr)rr)r r!)r bayes_mvsr statisticrminmax)selfdatameanvarstdr*T/var/www/html/virt/lib64/python3.6/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_input?szTestBayes_mvs.test_empty_inputcCs4tjd}d}tj|}x|D]}t||qWdS)Nr#r$)r#r$)nparanger r"r)r%x attributesresir*r*r+test_result_attributesBs    z$TestBayes_mvs.test_result_attributesN)__name__ __module__ __qualname__r,r/r7r*r*r*r+r,src@s,eZdZddZddZddZddZd S) TestMvsdistcCsdddddddg}tj|\}}}t|jdt|jddd d t|jd t|jdddd t|jdddt|jddd d dS)Nrrrrrrg"@g?4`H#j@O[%@g+=)rg$@ai @Ӈu8@g& .>g _紫@r)rYs?疡1O@)r<r=)r>r?)r@rA)r mvsdistrr'rinterval)r%r&r'r(r)r*r*r+r,Ls     zTestMvsdist.test_basiccCstttjgdS)N)r-r.r rB)r%r*r*r+r/[szTestMvsdist.test_empty_inputcCsdg}tttj|dS)Nr )r-r.r rB)r%r&r*r*r+ test_bad_arg^szTestMvsdist.test_bad_argcCs\tjJtjdtddtjdddgDddtjddddd gDWdQRXdS) NerrorcSsg|] }|jqSr*)r').0r3r*r*r+ hsz*TestMvsdist.test_warns..r rcSsg|] }|jqSr*)r')rFr3r*r*r+rGis)warningscatch_warnings simplefilterRuntimeWarningr rB)r%r*r*r+ test_warnscs  zTestMvsdist.test_warnsN)r8r9r:r,r/rDrOr*r*r*r+r;Ksr;c@s<eZdZddZddZddZddZd d Zd d Zd S) TestShapiroc Csddddddddd d d d d dddddddg}tj|\}}tj|}t|dddt|jdddt|dddt|jdddddddddddPd!d"d#dQdd%d&d'd(d)d*d+g}tj|\}}tj|}t|d,ddt|jd,ddt|d-d.dt|jd-d.dtjjd/tjj d0d.d1d2}tj|\}}tj|}t|d3ddt|jd3ddt|d4d.dt|jd4d.dd5d6d7d8d9d:d;dd?d@dAdBd'dCdDdEdFdGdHdIdJdKdLg}dM}dN} tj|\}}tj|}t||dOdt|j|dOdt|| d0dt|j| d0ddS)RNg)\(?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)\(?gRQ?gp= ף?g@g ףp= ?g{Gz?g\(\ @g{Gz@gQ @gIddY?gDio?rHiNarJd)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?rIg(\gRQ) r shapirorr#pvaluer1randomseednormrvs) r%x1wpw shapiro_testx2Zx3Zx4Z W_expectedZ p_expectedr*r*r+r,msJ       zTestShapiro.test_basicc Cs ddddddddd d g d d d dddddddg g}tj|\}}tj|}t|dddt|jdddt|dddt|jddddddddddd/d!d"g d#d0dd%d&d'd(d)d*d+g g}tj|\}}tj|}t|d,ddt|jd,ddt|d-d.dt|jd-d.ddS)1Ng)\(?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)\(?gRQ?gp= ף?g@g ףp= ?g{Gz?g\(\ @g{Gz@gQ @gIddY?gDio?rHg(\gRQ)r rUrr#rV)r%r[r\r]r^r_r*r*r+test_2ds$  zTestShapiro.test_2dcCs&tttjgtttjggggdS)N)r-r.r rU)r%r*r*r+r/szTestShapiro.test_empty_inputcCs4tttjddgtttjtjgdggtddS)Nr r)dtype)r-r.r rUr1arrayobject)r%r*r*r+test_not_enough_valuessz"TestShapiro.test_not_enough_valuescCsdg}tttj|dS)Nr )r-r.r rU)r%r3r*r*r+rDszTestShapiro.test_bad_argcCs`tjd}tj|d<tj|\}}tj|}t|tjt|jtjt|dt|jddS)Ng$@rg?) r1r2nanr rUr r#rrV)r%r3r\r]r^r*r*r+test_nan_inputs     zTestShapiro.test_nan_inputN) r8r9r:r,r`r/rdrDrfr*r*r*r+rPls ,rPc@sDeZdZddZddZddZddZd d Zd d Zd dZ dS) TestAndersoncCstd}|jdd}|jdd}tj|\}}}t|dd |tj|\}}}t||d dtjd}d|d<tj|\}}}t|ddS) NiI2)rTr r rg @) rstandard_exponentialstandard_normalr andersonrr1onesr)r%rsr[r_Acritsigvr*r*r+ test_normals    zTestAnderson.test_normalc Cstd}|jdd}|jdd}tj|d\}}}t||d dtjddtj|d\}}}WdQRXt||d kdS) NiIrh)rTZexponrignore)allr rkrj) rrlrmr rnrr1errstater)r%rpr[r_rqrrrsr*r*r+ test_expons  zTestAnderson.test_exponc Cstjd}d|d<tj|d\}}}t|}tjj|\}}tjj|||}tjj|||} tj d|d} | tj d| d|| ddd} t || dS)NrQgrgumbelr rrj) r1ror rnlengumbel_lfitlogcdflogsfr2r'r) r%rtZa2rrrsnZxbarsr~rr6Z expected_a2r*r*r+ test_gumbels *zTestAnderson.test_gumbelcCstttjdgdddS)Nr plate_of_shrimp)dist)r-r.r rn)r%r*r*r+rDszTestAnderson.test_bad_argcCs0td}|jdd}tj|}d}t||dS)NiIrh)rTr#critical_valuessignificance_level)r#rr)rrlr rnr)r%rpr3r5r4r*r*r+r7s   z#TestAnderson.test_result_attributesc CsFtd}|jdd}tj|d\}}}tj|d\}}}t||dS)NiIrQ)rTrzr|)rrzr rnr) r%rpr3A1crit1sig1A2crit2sig2r*r*r+ test_gumbel_ls  zTestAnderson.test_gumbel_lc Cshtd}|jdd}tjd}tj|d\}}}tj|d\}}} t||ddt||dkdS) NiIrQ)rTZgumbel_rrr rkrj)rrzr1ror rnrr) r%rpr[r_rrrrrrr*r*r+ test_gumbel_r s  zTestAnderson.test_gumbel_rN) r8r9r:ruryrrDr7rrr*r*r*r+rgs  rgc@sTeZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ dS)TestAndersonKSampc Csddddddddg}tjd d d d d dddg}tjddddddddg}tjddddddddg}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@rHgCl?gvq -?g,Ԛ?gl@gPk @rrJrIgJ4a?gMb0?)atol)r1rbr anderson_ksamprrr)r%t1t2t3t4Tktmpr*r*r+test_example1as z TestAndersonKSamp.test_example1ac Cstjddddddddg}tjd d d d d dddg}tjddddddddg}tjddddddddg}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@rHgCl?gvq -?g,Ԛ?gl@gPk @rrJrIgMb`?gMb0?)r)r1rbr rrrr)r%rrrrrrrr*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 d6dndodKdrZri<=18OT,;i6LFeJ0iri7i@h/iWxG*rJ rHr 4_a3rIDMPjR6.o'?rhfHXrigii[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%rrrrrrrrrrrrrr rrrr*r*r+test_example2b\s4"  z TestAndersonKSamp.test_example2bc"Cs.tjddd}t0}|jtddtj||dgdd\}}}WdQRXt|dd t|d t,}|jtddtj||dg\}}}WdQRXt|d d t|d t0}|jtd dtj||d gdd\}}}WdQRXt|d#dt|dt,}|jtd dtj||d g\}}}WdQRXt|d$dt|dtj||dgdd\}}}t|ddt |ddddtj||dg\}}}t|ddt |ddddtj||dgdd\}}}t|ddt |ddddtj||dgdd\}}}t|d dt |d!d"dddS)%Nr rQzp-value floored)messageg@D@F)rg= ףpD@rHgMbP?gGzD@zp-value cappedg?g?ܵ?rIg?g?ܵ?g@g!u?gףp= ׳?g{Gzt?r)rrrgBiq?g鷯?g'@g:M@g46W[1@g1ZGUS?g-C6?g?ܵg?ܵ) r1linspacer filter UserWarningr rrr r)r%r[supr_rr*r*r+test_R_kSamples|sB!&  "  &  "      z!TestAndersonKSamp.test_R_kSamplescCstttjtjddS)NrJ)r-r.r rr1ro)r%r*r*r+test_not_enough_samplessz)TestAndersonKSamp.test_not_enough_samplescCs"tttjtjdtjdfdS)NrJ)r-r.r rr1ro)r%r*r*r+test_no_distinct_observationssz/TestAndersonKSamp.test_no_distinct_observationscCstttjtjdgfdS)NrJ)r-r.r rr1ro)r%r*r*r+test_empty_samplesz#TestAndersonKSamp.test_empty_samplec CsRddddddddg}tjd 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)rr#rr)r#rr)r1rbr rr)r%rrr5r4r*r*r+r7s z(TestAndersonKSamp.test_result_attributesN) r8r9r:rrr r rrrrr7r*r*r*r+rs  Prc@s4eZdZddZddZddZddZd d Zd S) TestAnsaric Csldddddg}ddddddddg}t"}|jtdtj||\}}WdQRXt|ddt|d ddS) Nr rrHrIrz%Ties preclude use of exact statistic.g7@rg8#oG?)r rrr ansarir)r%r3yrWpvalr*r*r+ test_smalls  zTestAnsari.test_smallcCs^tjd}tjd}t"}|jtdtj||\}}WdQRXt|ddt|dddS)NrkrQcrrmlrr`rqtnrigrz%Ties preclude use of exact statistic.g0g@rg|:?)rrrQrrrrrrrrr rrrr!r"r!r#r)rrrrr$r%r#r$rrQr rr%rr&r&r!rrr)r1rbr rrr rr)r%ZramsayZparekhrrrr*r*r+ test_approxs  zTestAnsari.test_approxcCs@tjddddgddddd d g\}}t|d d t|d ddS)Nr rrHrIr0rJrrrirg$@rg?r)r rr)r%rrr*r*r+ test_exacts$ zTestAnsari.test_exactcCs(tttjgdgtttjdggdS)Nr )r-r.r r)r%r*r*r+rDszTestAnsari.test_bad_argc Cs^dddddg}ddddddddg}t}|jtdtj||}WdQRXd }t||dS) Nr rrHrIrz%Ties preclude use of exact statistic.r#rV)r#rV)r rrr rr)r%r3rrr5r4r*r*r+r7s z!TestAnsari.test_result_attributesN)r8r9r:rr'r(rDr7r*r*r*r+rs  rc@s4eZdZddZddZddZddZd d Zd S) TestBartlettc CsBtttttttttt g }t j |\}}t |ddt |dddS)NgzU@/4@rgG?) g1g2g3g4g5g6g7g8g9g10r bartlettr)r%argsTrr*r*r+ test_datas zTestBartlett.test_datacCstttjdgdS)Nr )r-r.r r4)r%r*r*r+rDszTestBartlett.test_bad_argc Cs4tttttttttt g }t j |}d}t ||dS)Nr#rV)r#rV) r*r+r,r-r.r/r0r1r2r3r r4r)r%r5r5r4r*r*r+r7s z#TestBartlett.test_result_attributesc Cs6tttttttttt gf }t t j t j ft j|dS)N)r*r+r,r-r.r/r0r1r2r3r r1rer r4)r%r5r*r*r+test_empty_argszTestBartlett.test_empty_argcCs*tjddgddgg}tttjt|dS)Nr rrHrI)r1rbr-r.r r4r*)r%r3r*r*r+ test_1d_input$szTestBartlett.test_1d_inputN)r8r9r:r7rDr7r8r9r*r*r*r+r) s r)c@sTeZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ dS) TestLevenec CsBtttttttttt g }t j |\}}t |ddt |dddS)NgYpK?rgI{]?) r*r+r,r-r.r/r0r1r2r3r levener)r%r5rrr*r*r+r7+s zTestLevene.test_datacCsFtjtttdd\}}tjtttddd\}}t||t||dS)Nr')centertrimmedg)r<proportiontocut)r r;r*r+r,r)r%W1pval1W2pval2r*r*r+ test_trimmed12s    zTestLevene.test_trimmed1c Csddddddddg}d dd dd dd d g}tjjdtjj|}tj||ddd\}}tj||ddd\}}tj|dd|dddd\}} t||t||t|| dS)Ng333333?g@g@g@g@g@g@gY@gg @g@g@gi@ir=g?)r<r>r r')r<rjrj)r1rWrX permutationr r;r) r%r3rr_ZW0Zpval0r?r@rArBr*r*r+ test_trimmed2;s      $  zTestLevene.test_trimmed2cCsntjddd}tjjdtjj|}|d}tj||dd\}}tj||dd\}}t||t||dS) Nr rirHr')r<Zmedianrj)r1r rWrXrDr r;r)r%r3r_rr?r@rArBr*r*r+test_equal_mean_medianMs   z!TestLevene.test_equal_mean_mediancCs&tjddd}tttj||dddS)Nr rg?) portiontocutrj)r1r r- TypeErrorr r;)r%r3r*r*r+test_bad_keywordWszTestLevene.test_bad_keywordcCs&tjddd}tttj||dddS)Nr rtrim)r<rj)r1r r-r.r r;)r%r3r*r*r+test_bad_center_value[sz TestLevene.test_bad_center_valuecCstttjdgdS)Nr )r-r.r r;)r%r*r*r+test_too_few_args_szTestLevene.test_too_few_argsc Cs4tttttttttt g }t j |}d}t ||dS)Nr#rV)r#rV) r*r+r,r-r.r/r0r1r2r3r r;r)r%r5r5r4r*r*r+r7bs z!TestLevene.test_result_attributescCs*tjddgddgg}tttjt|dS)Nr rrHrI)r1rbr-r.r r;r*)r%r3r*r*r+r9iszTestLevene.test_1d_inputN) r8r9r:r7rCrErFrIrKrLr7r9r*r*r*r+r:)s  r:c@s4eZdZddZddZddZddZd d Zd S) TestBinomPcCsRtjdd}t|ddtjdd}t|ddtjdd gdd }t|d ddS)NrQg-"^?rrig`w?ig@rI)rgz?g?)r binom_testr)r%rr*r*r+r7ps     zTestBinomP.test_datacCstttjdddgdS)Nr rrH)r-r.r rP)r%r*r*r+test_bad_len_xxszTestBinomP.test_bad_len_xcCs(tttjdgtttjdgdddS)NrQrh)r)r-r.r rP)r%r*r*r+ test_bad_n|szTestBinomP.test_bad_ncCstttjddgdddS)Nrhg@)r)r-r.r rP)r%r*r*r+ test_bad_pszTestBinomP.test_bad_pcCsXtjddd dd}t|dtjddd dd}t|d tjdddd d}t|d dS)Nrg?rless)r alternativeg-$Ǻl?greatergs0,j.?z two-sidedgJfJ"f?gUUUUUU?gUUUUUU?gUUUUUU?)r rPr)r%r5r*r*r+test_alternativess   zTestBinomP.test_alternativesN)r8r9r:r7rQrRrSrXr*r*r*r+rMns rMc@sDeZdZddZddZddZddZd d Zd d Zd dZ dS) TestFlignercCs&tjd}ttj||ddddS)NrJrnf @Dq?r)rZr[)r1r2rr fligner)r%r[r*r*r+r7s zTestFligner.test_datac svtjjdfdd}|t}|t}|t}tj|||dd\}}tj|||ddd\}}t||t||dS) N{cs tj|djt|jS)Ng|=)r1Zasarrayrandnr{tolist)g)rpr*r+sz+TestFligner.test_trimmed1..r')r<r=g)r<r>) r1rWrr*r+r,r r\r) r%Z_perturbZg1_Zg2_Zg3_Xsq1r@Xsq2rBr*)rpr+rCs     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@r=g?)r<r>r r')r<rjrj)r r\r)r%r3rrbr@rcrBr*r*r+rEs  $ zTestFligner.test_trimmed2cCs&tjddd}tttj||dddS)Nr rg?)rGrj)r1r r-rHr r\)r%r3r*r*r+rIszTestFligner.test_bad_keywordcCs&tjddd}tttj||dddS)Nr rrJ)r<rj)r1r r-r.r r\)r%r3r*r*r+rKsz!TestFligner.test_bad_center_valuecCstttjdgdS)Nr )r-r.r r\)r%r*r*r+test_bad_num_argsszTestFligner.test_bad_num_argscCs.tjd}ttjtjftj||dgdS)NrJr)r1r2r rer r\)r%r3r*r*r+r8s zTestFligner.test_empty_argN) r8r9r:r7rCrErIrKrdr8r*r*r*r+rYsrYc@s<eZdZddZddZddZddZd d Zd d Zd S)TestMoodcCs&tjd}ttj||ddddS)NrJrg8!?iT?r8!)rgrf)r1r2rr mood)r%r[r*r*r+ test_moods zTestMood.test_moodcCs`tjjdtjjdd}tjjdd}tj||\}}tj||\}}t||g| |gdS)Nirir r0)r1rWrXr^r rhr)r%r[r_Zz1p1Zz2p2r*r*r+test_mood_order_of_argss  z TestMood.test_mood_order_of_argscCsd9dd:ddd;ddd ddd?d@dddg}dAdddBdCdddDddEdd dFdGd#dHd%d&d'd(d)dId+d,d-dJd/dKd1d2g}tj|}tj|}dL|_dM|_ttj||dd6dNd8gdS)ONg ?g۟?gy<7x?g1)E?g9cç?g FA?g<=ӟ 2?g{![?g_l?gڻ{?gGSsA0?g 1?g 3?g@gL?g_l?g,*.?gː53?gvNqG?gW=?gd"?g`Y#?g^g?gN?gB a?gp5_%?g(?gv]l?gg?g?gr6ʺ?ḡj?gK!?g{?g&?g*h}@gZ?gyBU?g .ۋ4?g|w;?g&ck@gRj2?g}+w(o?g@F?g}_9t?g =@g S?gx (?gg7Y?g{ih?rirr0)axisg4?g^}E ?g gy<7xg FAgڻ{ӿg 3gg_lg,*.gd"gNgB agv]lοggK!ٿg{g*h}gRj2g =gx ()rir)r0rg4)r1rbshaperr rh)r%r[r_r*r*r+test_mood_with_axis_nones.   z!TestMood.test_mood_with_axis_nonec Csd}tjjdtjjd|}tjjd|}tj||\}}xFt|D]:}t||||gtj|dd|f|dd|fqFW|j}|j}tj||dd\}}xFt|D]:}t||||gtj||ddf||ddfqWdS)NrJirir0r )rm) r1rWrXr^r rhrangerZ transpose)r%nyr[r_ z_vectest pval_vectestjr6r*r*r+ test_mood_2ds *zTestMood.test_mood_2dc CsNd }tjjdtjj|}tjj|}xtdD]}tj|||d\}}ddgd dgd dgf}xt|||d D]}xt|||dD]} |d kr|dd|| f} |dd|| f} nR|dkr||dd| f} ||dd| f} n$||| ddf} ||| ddf} t||| f||| fgtj| | qWqzWq4WdS) NrirJrirH)rmr rr)rirJr)r1rWrXr^rpr rhr) r%rnr[r_rmrrrsZaxes_idxr6rtZslice1Zslice2r*r*r+ test_mood_3ds(     zTestMood.test_mood_3dcCstttjdggdS)Nr )r-r.r rh)r%r*r*r+test_mood_bad_arg6szTestMood.test_mood_bad_argN) r8r9r:rirlrorurvrwr*r*r*r+res  rec@sVeZdZddZddZddZejje ddd d Z d d Z d dZ ddZ dS) TestProbplotcCstjjdtjjdd}tj|dd\}}ddddddddddddd d d d d dddg}t|tj|t||tj|dd\}}dddg}t||dS)Ni90r)rTF)r}g+/?gY3?gt?g5Z0?gFj?g!2?gsR{?g7?g>h?gcz隇?TgƝV?g?gà?g+/gY3gtg5Z0gFjg!2gsR{ܿg7Կg>hǿgcz隇) r1rWrXr rYrZprobplotrsort)r%r3osmosrZ osm_expectedr5Zres_fitZres_fit_expectedr*r*r+r,>s    zTestProbplot.test_basicc Cstjjdtjjdd}tj|ddd\}}tj|ddd\}}tj|fdd\}}t||t||t||t||tj|fdd\}} dS)Ni@rQ)rTF)sparamsr}r)r1rWrXr rYrZryr) r%r3osm1osr1osm2osr2Zosm3Zosr3r{r|r*r*r+test_sparams_keywordNs     z!TestProbplot.test_sparams_keywordcCstjjdtjjdd}tj|dddd\}}tj|dtjdd\}}t||t||t t tj|dd t t tj|gd Gd d d t }tj|ddd \}}tj||dd\}}t||t||dS)Ni90r)rTFtrH)r}rr}zwrong-dist-name)rc@seZdZdZddZdS)z3TestProbplot.test_dist_keyword..custom_distz6Some class that looks just enough like a distribution.cSstjj|ddS)Nr)rR)r rYppf)r%qr*r*r+rjsz7TestProbplot.test_dist_keyword..custom_dist.ppfN)r8r9r:__doc__rr*r*r*r+ custom_disthsrr)r}r})rr})rH)rH)r) r1rWrXr rYrZryrrr-r.AttributeErrorrc)r%r3r~rrrrr*r*r+test_dist_keyword]s    zTestProbplot.test_dist_keywordz no matplotlib)reasonc Cstjjdtj}|jdtjjddd}tj |td\}}tj tj |dd\}}tj |dtd}tj tj |ddd}t t |t |kot |kot |kod knt ||t ||t ||t ||tj}|jd} tj |d| dtj dS) NitrrHrQ)rT)plotF)r}rr)r1rWrXpltfigure add_subplotr rrZrycloserr{r) r%figr3Zres1Zfitres1Zres2Zfitres2Zres3Zres4axr*r*r+test_plot_kwargrs&  @     zTestProbplot.test_plot_kwargcCstttjdgdddS)Nr r)r)r-r.r ry)r%r*r*r+test_probplot_bad_argssz#TestProbplot.test_probplot_bad_argscCsZttjgddtjgtjgfttjgddtjgtjgftjtjdffdS)NF)r}Tg)r r ryr1rbre)r%r*r*r+ test_emptys zTestProbplot.test_emptyc CsTtjdd>ttjdgddtjdgtjdgftjtjdffWdQRXdS)Nrv)invalidr T)r}g)r1rxr r ryrbre)r%r*r*r+test_array_of_size_onesz#TestProbplot.test_array_of_size_oneN)r8r9r:r,rrpytestmarkskipifhave_matplotlibrrrrr*r*r*r+rx<srxc @seZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ e jjdd#d$dgd%dd&d'dgd(d)dd*d+d,ddggddZd d!Zd"S)- TestWilcoxoncCshtttjdgddgtttjddgddgdtttjddgddgddtttjdgddddS)Nr rdummy)rVriZxyz)mode)r-r.r wilcoxon)r%r*r*r+test_wilcoxon_bad_args z"TestWilcoxon.test_wilcoxon_bad_argcCsRtjd}tttj||dddtttj||dddttj||dddd dS) Nrwilcoxapprox)rprattzsplitrrI?i@Z@)rr)r1r2r-r.r rr )r%r3r*r*r+test_zero_diffs zTestWilcoxon.test_zero_diffc CsXddddg}ddddg}t&}|jtddtj||dd d }WdQRXt|d dS)Nr rrHrIrJzSample size too small)r rr) zero_methodr |N?)rr)r rrr rr)r%r3rrr5r*r*r+ test_pratts   zTestWilcoxon.test_prattc CsNddddd ddddddg }tj|ddd}tj|ddd}tj|d dd}dS) Nr rrHrrr)rrrrrj)r r)r%Zarrrr*r*r+test_wilcoxon_arg_typesz#TestWilcoxon.test_wilcoxon_arg_typec Cshdddddddddg }td2d}tjdd t||D}tj|j}tj||d d d \}}t|d t|dtj||dd d \}}t|dt|dtj||dd d \}}t|dt|dtj dddddddddddd g }tj d!d"d#d$d d%d&dd'dd(d)g }tj||d*d d+\}}t |d,t|d-d.d/tj||d0d d+\}}t |d,t|d1d.d/dS)3Nr rIrr0rrJrcSsg|]\}}|g|qSr*r*)rFurtr*r*r+rGsz7TestWilcoxon.test_accuracy_wilcoxon..rr)rig.i?rigFYv]Uj?riGg%^Ez?rr&rryrr!pwsrF) correctionrrg^҂)rTg:b+?) rpr1Z concatenatezipzerosrTr rrrbr )r%freqnumsr3rr6rr*r*r+test_accuracy_wilcoxons*        ""  z#TestWilcoxon.test_accuracy_wilcoxonc Cshtjddddddddd d d d g }tjd dddd ddd ddddg }tj||ddd}d}t||dS)Nrr&rrrrrrrr!rrrrrrrrrrrFr)rrr#rV)r#rV)r1rbr rr)r%r3rr5r4r*r*r+test_wilcoxon_result_attributess ""z,TestWilcoxon.test_wilcoxon_result_attributescCsntjdgddd\}}d}t|dt||ddtjdgdd dd \}}d }t|dt||dddS) Ng?rir)rg4IťY?rgư>)rT)rrg c2_?)r rr r)r%statrZ 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}rrr#zr|r]zSample size too small)r rUr)rVrrg13}?r)rT)rVrrg7a%?rWg?gڏoJT?)r rrr rr r)r%r3rrr\rr*r*r+ test_onesideds0         zTestWilcoxon.test_onesidedcCsPxJtddD]<}t|}t||dddt|tt|d|q WdS)Nr rr)rprr r{sum)r%rZcntr*r*r+test_exact_basic#szTestWilcoxon.test_exact_basicc CsVtjdddd(ddd)dd*d g }tjd d d+dd,d-d.d/d0d1g }tj||ddd\}}t|dddtj||ddd\}}t|dddtj||ddd\}}t|dddtjdd d!}tjd dd2}tj||ddd\}}t|d#ddtj||ddd\}}t|d$ddtj||ddd\}}t|d%ddtjd&d"}tttj|dd'dS)3Ng(\?g= ףp=?g(\?gQ?gQ?g{Gz?gp= ף?gq= ףp?g?gQ?gQ?g?g?g333333?g?g?gzG?gQ?g(\?gRQ?z two-sidedexact)rVrg֔?r)rrUgYJ?rWg*?rrg?r g(?g(?gS?r)rgQ޿gp= ףggɿggܿgzGgQοg(\gRQrj)r1rbr rrr2r-r.)r%r3rrrdr*r*r+test_exact_pval)s(  zTestWilcoxon.test_exact_pvalr3r rrHrIrJrrrcCs@tj|\}}tj|}||dkj}t||t|ddS)Nrr )r rr1rbrr )r%r3r\rZwtruer*r*r+test_exact_p_1Gs   zTestWilcoxon.test_exact_p_1c Cstjddd}tjddd }ttj||tj||ddtjdd}t"}|jtdd tj|\}}WdQRXttj|d d||ftjdd }ttj|tj|d ddS) Nrrg?r r)rrzExact p-value calculation)r rrrj)r1r2r r rr rr)r%r3rrrr\rr*r*r+ test_autoQs   zTestWilcoxon.test_autoNrjrkrjrrjrkri)r8r9r:rrrrrrrrrrrr parametrizerrr*r*r*r+rs    &  rc@s,eZdZddZddZddZddZd S) TestKstatcstjjdtjjdfdddD}d d d d g}t||d dtjdd}tjdd}tjdd}t|||f|ddddddS)Ni}i90csg|]}tj|qSr*)r kstat)rFr)r&r*r+rGhsz>TestKstat.test_moments_normal_distribution..r rrHrIgHQ,?glZ)rI?g88}?gcJN?g-C6?)r)momentg{Gz?g{Gz?)rr)r rrHrIrj)r1rWrXr^rr r)r%Zmomentsexpectedm1m2Zm3r*)r&r+ test_moments_normal_distributiones   z*TestKstat.test_moments_normal_distributioncCstttjgdS)N)r-r.r r)r%r*r*r+r/sszTestKstat.test_empty_inputcCs*tjd}tj|d<ttj|tjdS)Ng$@r)r1r2rer r r)r%r&r*r*r+rfvs  zTestKstat.test_nan_inputcCs.tjd}xdD]}tttj||dqWdS)Nrir/$@)r)rr)r1r2r-r.r r)r%r&rr*r*r+test_kstat_bad_arg|s  zTestKstat.test_kstat_bad_argN)r8r9r:rr/rfrr*r*r*r+rdsrc@s$eZdZddZddZddZdS) TestKstatVarcCstttjgdS)N)r-r.r kstatvar)r%r*r*r+r/szTestKstatVar.test_empty_inputcCs*tjd}tj|d<ttj|tjdS)Ng$@r)r1r2rer r r)r%r&r*r*r+rfs  zTestKstatVar.test_nan_inputcCs dg}d}tttj||ddS)Nr ri)r)r-r.r r)r%r&rr*r*r+rDszTestKstatVar.test_bad_argN)r8r9r:r/rfrDr*r*r*r+rsrc@sNeZdZddZddZddZejje ddd d Z d d Z d dZ dS) TestPpccPlotcCs&tjjdtjjdddd|_dS)NitrJi)rT)r1rWrXr loggammarZr3)r%r*r*r+ setup_methods zTestPpccPlot.setup_methodcCsNd}tj|jd d|d\}}dddddg}t|tjd d|d t||dS) NrJri)Ng9% ?gxM _?g"; 9?g®XZ?g?)numii)r ppcc_plotr3rr1r )r%rsvalsppcc ppcc_expectedr*r*r+r,s zTestPpccPlot.test_basiccCstj|jdddd\}}tj|jddtjd\}}t||ddt||ddtj|jdd\}}t||ddt||dddS) Nri tukeylambda)rg#B ;)riii)r rr3rr)r%Zsvals1Zppcc1Zsvals2Zppcc2Zsvals3Zppcc3r*r*r+ test_dists zTestPpccPlot.test_distz no matplotlib)rcCsZtj}|jd}tj|jddtd|j||jd}tj|jdd|dtjdS)Nrr)rii)rrrr rr3delaxesr)r%rrr*r*r+rs   zTestPpccPlot.test_plot_kwargcCs4tttj|jddtttjdddgdddddS)Nr rrrHr)r)r-r.r rr3)r%r*r*r+test_invalid_inputssz TestPpccPlot.test_invalid_inputscCs@tjgdd\}}t|tjddddt|tjdtddS)Nrr r)r)ra)r rrr1r rfloat)r%rrr*r*r+rszTestPpccPlot.test_emptyN) r8r9r:rr,rrrrrrrrr*r*r*r+rs   rc@s,eZdZddZddZddZddZd S) TestPpccMaxcCsdg}tttj|dddS)Nr r)r)r-r.r ppcc_max)r%r&r*r*r+test_ppcc_max_bad_argsz!TestPpccMax.test_ppcc_max_bad_argcCs<tjjdtjjd ddddd}ttj|d d d dS) Nigffffff?rg?i')rRrSrTg@g{?r)rgffffffg{)r1rWrXr rrZrr)r%r3r*r*r+test_ppcc_max_basics zTestPpccMax.test_ppcc_max_basiccCsztjjdtjjd ddddd}tj|dd }tj|tjd }t|dd d t|dd d tj|}t|dd d dS)Nigffffff?rg?i')rRrSrTg@r)rg{?rJ)rgffffffg{g{g{)r1rWrXr rrZrr)r%r3Zmax1Zmax2Zmax3r*r*r+rs  zTestPpccMax.test_distcCsjtjjdtjjdddddd}tttj|dd t tj|dd dddt tj|dd ddddS)Nigffffff?r?i')rRrSrTg@?)Zbrackrr g{?r)rgffffff)rrr)rr g{rk)rkrg{) r1rWrXr rrZr-r.rr)r%r3r*r*r+ test_bracks  zTestPpccMax.test_brackN)r8r9r:rrrrr*r*r*r+rsrc@s4eZdZddZddZddZddZd d Zd S) TestBoxcox_llfcCs^tjjdtjjddd}d}tj||}|j dtjtj |j d}t ||dS)Ni1i'ri)rTrRr g@r) r1rWrXr rYrZ boxcox_llfrTlogrr)r)r%r3lmbdallfZ llf_expectedr*r*r+r,s   $zTestBoxcox_llf.test_basiccCsNtjjdtjjddd}d}tj||}tj|t|}t||dddS)Ni1rQri)rTrRr g-q=)r) r1rWrXr rYrZrlistr)r%r3rrllf2r*r*r+test_array_likes   zTestBoxcox_llf.test_array_likecCsZtjjdtjjddd}d}tj||}tj|tj||gj}t ||g|dddS)Ni1rQri)rTrRr g-q=)r) r1rWrXr rYrZrvstackr6r)r%r3rrrr*r*r+ test_2d_inputs   zTestBoxcox_llf.test_2d_inputcCsttjtjdgdS)Nr )rr1isnanr r)r%r*r*r+r szTestBoxcox_llf.test_emptycCs&ddddg}tjd|}t|ddS)Ngh@g m@gx@rg#:x1@ig#:x1)r rr)r%r&rr*r*r+ test_gh_6873s  zTestBoxcox_llf.test_gh_6873N)r8r9r:r,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) TestBoxcoxcCstjjdtjjdddd}tj|dd}t||dtj|dd}t|dd|tj|dd}t|tj|tjt |dd}t|tj|dS) Ni90rJrh)rTr )rrrj) r1rWrXr rrZboxcoxrrr)r%r3xtr*r*r+test_fixed_lmbdaOs zTestBoxcox.test_fixed_lmbdacCsbtjjdtjjdd}tjjddd}||d| }tj|\}}t|d |dd dS) Niig@riiP)rRrTr r)rrj)r1rWrXr rYrZrr)r%rr3x_invrmaxlogr*r*r+test_lmbda_None^s  zTestBoxcox.test_lmbda_NonecCstjjdtjjdddd}tj|dd\}}}t|ddgtj|d d\}}}t|d d gtjjd d dd}tj|dd\}}}t|ddgtj|dd\}}}t|ddgdS)NirJrh)rTg?)alphagx@gv[%@g?gxˉEk?g8Bhk @rir0gMbP?g\?gx̡&@g+?g7(U@gp: sY@)r1rWrXr rrZrr)r%r3rrCr*r*r+ test_alphajs zTestBoxcox.test_alphacCsRtjddg}tttj|tttjtjdgtttjtjdgdggdS)Nr rrj)r1rbr-r.r r)r%r3r*r*r+test_boxcox_bad_arg{szTestBoxcox.test_boxcox_bad_argcCsttjgjdkdS)Nr)r)rr rrn)r%r*r*r+rszTestBoxcox.test_emptycCs tjt\}}t|ddddS)NgsHjdr?gh㈵>)rgsHjdr)r r _boxcox_datar)r%rZlamr*r*r+rszTestBoxcox.test_gh_6873N) r8r9r:rr r r rrr*r*r*r+rMs   rc@s,eZdZddZddZddZddZd S) TestBoxcoxNormmaxcCs&tjjdtjjdddd|_dS)Ni90rJrh)rT)r1rWrXr rrZr3)r%r*r*r+rs zTestBoxcoxNormmax.setup_methodcCstj|j}t|ddddS)Ngm?gư>)r)r boxcox_normmaxr3r)r%r r*r*r+ test_pearsonrs zTestBoxcoxNormmax.test_pearsonrcCs<tj|jdd}t|dddtj|j\}}t||dS)NZmle)methodg .!?gư>)r)r rr3rr)r%r rZ maxlog_boxcoxr*r*r+test_mleszTestBoxcoxNormmax.test_mlecCs&tj|jdd}t|ddgdddS)Nrw)rgm?g .!?gư>)r)r rr3r)r%Z maxlog_allr*r*r+test_allszTestBoxcoxNormmax.test_allN)r8r9r:rrrrr*r*r*r+rsrc@sFeZdZddZddZejje ddddZ d d Z d d Z d S)TestBoxcoxNormplotcCs&tjjdtjjdddd|_dS)NitrJi)rT)r1rWrXr rrZr3)r%r*r*r+rs zTestBoxcoxNormplot.setup_methodcCsNd}tj|jd d|d\}}dddddg}t|tjd d|d t||dS) NrJri)rg!X4}?g΁i?g- 15?gW$?g\ܱ{?)rii)r boxcox_normplotr3rr1r )r%rZlmbdasrrr*r*r+r,s zTestBoxcoxNormplot.test_basicz no matplotlib)rcCsZtj}|jd}tj|jddtd|j||jd}tj|jdd|dtjdS)Nrr)rii)rrrr rr3rr)r%rrr*r*r+rs   z"TestBoxcoxNormplot.test_plot_kwargcCs.tttj|jddtttjddgdddS)Nr rrj)r-r.r rr3)r%r*r*r+rsz&TestBoxcoxNormplot.test_invalid_inputscCsttjgddjdkdS)Nrr )rr rrT)r%r*r*r+rszTestBoxcoxNormplot.test_emptyN) r8r9r:rr,rrrrrrrr*r*r*r+rs  rc@s$eZdZddZddZddZdS)TestYeojohnson_llfcCsNtjjdtjjddd}d}tj||}tj|t|}t||dddS)Ni1rQr)rTrRr g-q=)r) r1rWrXr rYrZyeojohnson_llfrr)r%r3rrrr*r*r+rs   z"TestYeojohnson_llf.test_array_likecCsZtjjdtjjddd}d}tj||}tj|tj||gj}t ||g|dddS)Ni1rQri)rTrRr g-q=)r) r1rWrXr rYrZrrr6r)r%r3rrrr*r*r+rs   z TestYeojohnson_llf.test_2d_inputcCsttjtjdgdS)Nr )rr1rr r)r%r*r*r+rszTestYeojohnson_llf.test_emptyN)r8r9r:rrrr*r*r*r+rsrc@seZdZddZejjdddddgdd Zd d Zd d Z ejjde j e j gddZ ejjde je je je jgddZdS)TestYeojohnsoncCstjjdtjjdddd}tj|dks2ttj|dd}t ||tj|d d}t |dd|dtj|dd}t |tj |dtj|dd}t ||tjjdddd}tj|dksttj|dd}t |tj | d tj|dd}t ||tj|d d}t |d| ddtjjdddd}tj|dk s\ttj|dk srt|dk}tj|dd}t ||||tj|d d}t ||dd||dtj|dd}t ||tj ||dtj|dd}t |||||}tj|dd}t ||tj || d tj|dd}t ||||tj|d d}t ||d|| dddS) Ni90rJrh)rTrr )rrrHrjrj) r1rWrXr rrZrwAssertionError yeojohnsonrr)r%r3rposnegr*r*r+rsL     zTestYeojohnson.test_fixed_lmbdarrg?g?rcCsdd}tjjdd}tjjdd|d}|||}tj|\}}t||dd tdtjj |||d d td|j dd td|j dd dS) NcSstj|j|jd}|dk}t|tjdkrDtj||d||<n$tj|||dd|d||<t|dtjdkrdtjd| ||ddd|||<ndtj|| ||<|S)N)rarg?r r)r1rrnraabsspacingexppower)r3rrrr*r*r+_inverse_transforms$z:TestYeojohnson.test_lmbda_None.._inverse_transformii Nrr )rRrSrTg{Gz?)rr)r) r1rWrXnormalr rrrZlinalgrYr'r))r%rr"Z n_samplesr3rrr r*r*r+r s  zTestYeojohnson.test_lmbda_NonecCsttjgjdkdS)Nr)r)rr rrn)r%r*r*r+r6szTestYeojohnson.test_emptycCsNtjjdtjjddd}tj|\}}tjt|\}}t||dddS)Ni1rQr)rTrRg-q=)r) r1rWrXr rYrZrrr)r%r3Zxt1rZxt2r*r*r+r9s  zTestYeojohnson.test_array_likerac Cs:tjd|d}d}tjt|dtj|WdQRXdS)Nr)raz>Yeo-Johnson transformation is not defined for complex numbers.)match)r1r2rr r.r r)r%rar3err_msgr*r*r+test_input_dtype_complex@sz'TestYeojohnson.test_input_dtype_complexcCsZtjd|d}tjdtjd}tj|\}}tj|\}}t||ddt||dddS)Nr)ragHz>)r)r1r2Zfloat64r rr)r%raZx_intZx_floatZxt_intZ lmbda_intZxt_floatZ lmbda_floatr*r*r+test_input_dtype_integerHs z'TestYeojohnson.test_input_dtype_integerN)r8r9r:rrrrr rrr1Z complex64Z complex128r&Zint8uint8Zint16Zint32r'r*r*r*r+rs /%rc@s$eZdZddZddZddZdS)TestYeojohnsonNormmaxcCs&tjjdtjjdddd|_dS)Ni90rJrh)rT)r1rWrXr rrZr3)r%r*r*r+rSs z"TestYeojohnsonNormmax.setup_methodcCstj|j}t|ddddS)Ng?gư>)r)r yeojohnson_normmaxr3r)r%r r*r*r+rWs zTestYeojohnsonNormmax.test_mlecCsDddddddddd d d d d ddg}tj|}tj|ddds@tdS)Ngffffff@g @g?g@gffffff?g333333@g @gffffff@g?g @g@g@g"@g@g@gzG?gMbP?)rg g)r r*r1Zallcloser)r%r3rr*r*r+test_darwin_example[s z)TestYeojohnsonNormmax.test_darwin_exampleN)r8r9r:rrr+r*r*r*r+r)Rsr)c @seZdZejjdejdfejdfej dfgddZ ddZ d d Z d d Z d dZejjdejdfejdfej dfgddZejjdejejej gddZejjdejejej gddZejjdejejdddfejejdddfej ejdddfgddZejjdejdejd d!d"d#d$d%gejd&d'gdfejd(ejd)d*d*d+d,d,gejdd-gdfej d.ejd/d0d0d1d2d2gejd3d4gdfgd5d6Zejjdejdfejdfej dfgd7d8Zejjdejejej gd9d:Zejjdejejej gd;d<Zejjd=d>d?d@dAdBdCejgejd>d?d@dAdBdCejgdDdEdFdGejdHdIgggejjdejejej gdJdKZejjd=d>d?d@dAdBdCejgejd>d?d@dAdBdCejgdDdEdFdGejdHdIgggejjdejejej gdLdMZdNdOZdPdQZdRdSZdTS)U TestCircFuncsztest_func,expectedgA:wv?gk~_ЀBE@gE|2@cCs0tjddddddg}t||dd|d d dS) NicrJrigrii^ih)highgHz>)r)r1rbr)r% test_funcrr3r*r*r+test_circfuncseszTestCircFuncs.test_circfuncscCstjdddddddg}|j}tj|dd }t||d d |j}tj|dd }t||d d |j}tj |dd }t||d d dS) Nrrrrg4@g3333333@ih)r-gh㈵>)rg-C6?) r1rbr'r circmeanrr(circvarr)circstd)r%r3M1M2V1V2S1S2r*r*r+test_circfuncs_smallmsz"TestCircFuncs.test_circfuncs_smallc stjddddddgddd d d d gd d ddd dggtjdd}tjjdd}t||ddtjddd}fddtjdD}t||ddtjddd}fddtjdD}t||dddS)NicrJrigrii^i_rrIi`ri]ierifidih)r-g+=)rr )r-rmcsg|]}tj|ddqS)ih)r-)r r1)rFr6)r3r*r+rGsz4TestCircFuncs.test_circmean_axis..rcs&g|]}tjdd|fddqS)Nih)r-)r r1)rFr6)r3r*r+rGs)r1rbr r1ravelrrprn)r%r4r5r*)r3r+test_circmean_axis{sz TestCircFuncs.test_circmean_axisc stjddddddgddd d d d gd d ddd dggtjdd}tjjdd}t||ddtjddd}fddtjdD}t||ddtjddd}fddtjdD}t||dddS)NicrJrigrii^i_rrIi`ri]ierifidih)r-gdy=)rr )r-rmcsg|]}tj|ddqS)ih)r-)r r2)rFr6)r3r*r+rGsz3TestCircFuncs.test_circvar_axis..rcs&g|]}tjdd|fddqS)Nih)r-)r r2)rFr6)r3r*r+rGs)r1rbr r2r;rrprn)r%r6r7r*)r3r+test_circvar_axisszTestCircFuncs.test_circvar_axisc stjddddddgddd d d d gd d ddd dggtjdd}tjjdd}t||ddtjddd}fddtjdD}t||ddtjddd}fddtjdD}t||dddS)NicrJrigrii^i_rrIi`ri]ierifidih)r-gdy=)rr )r-rmcsg|]}tj|ddqS)ih)r-)r r3)rFr6)r3r*r+rGsz3TestCircFuncs.test_circstd_axis..rcs&g|]}tjdd|fddqS)Nih)r-)r r3)rFr6)r3r*r+rGs)r1rbr r3r;rrprn)r%r8r9r*)r3r+test_circstd_axisszTestCircFuncs.test_circstd_axiscCs*ddddddg}t||dd|d d dS) NicrJrigrii^ih)r-gHz>)r)r)r%r.rr3r*r*r+test_circfuncs_array_likesz'TestCircFuncs.test_circfuncs_array_liker.cCsttj|gdS)N)rr1r)r%r.r*r*r+rszTestCircFuncs.test_emptycCs.ddddddtjg}ttj||dddS) NicrJrigrii^ih)r-)r1rerr)r%r.r3r*r*r+test_nan_propagatesz TestCircFuncs.test_nan_propagateg5:v@gf Pb?)Nrr gڴV0@gLԿAB@g#q@gauy+@c Cstjdddddddgdd d d d d tjgdtjtjtjtjtjtjgg}xf|jD]Z}||d|d}|dkr|ttj|qRt|d||ddttj|ddjqRWdS)NicrJrigrii^r i_rrIi`ri]ih)r-rmrgHz>)r)r1rbrekeysrrrrw)r%r.rr3rmoutr*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 Cstjddddddtjgddd d d d tjgtjtjtjtjtjtjtjgg}xh|jD]\}||d d|d}|dkrt|||ddqVt|dd||ddttj|dqVWdS)NicrJrigrii^i_rrIi`ri]ihomit)r- nan_policyrmgHz>)rr rjrj)r1rbrerArrr)r%r.rr3rmrBr*r*r+test_nan_omit_arrays$z!TestCircFuncs.test_nan_omit_arraycCs0ddddddtjg}t||ddd |d d dS) NicrJrigrii^ihrD)r-rEgHz>)r)r1rer)r%r.rr3r*r*r+ test_nan_omitszTestCircFuncs.test_nan_omitcCs2tjtjtjtjtjg}ttj||dddS)NrD)rE)r1rerr)r%r.r3r*r*r+test_nan_omit_allszTestCircFuncs.test_nan_omit_allcCsjtjtjtjtjtjtjgtjtjtjtjtjgg}||ddd}ttj|jtt|dkdS)NrDr )rErmr)r1rbrerrrwr{)r%r.r3rBr*r*r+test_nan_omit_all_axiss z$TestCircFuncs.test_nan_omit_all_axisr3icrJrigrii^i_rrIi`ri]cCstt||ddddS)Nihraise)r-rE)r-r.)r%r.r3r*r*r+test_nan_raise szTestCircFuncs.test_nan_raisecCstt||ddddS)Nihfoobar)r-rE)r-r.)r%r.r3r*r*r+test_bad_nan_policysz!TestCircFuncs.test_bad_nan_policycCs$d}|}tj|}t||dddS)Ng?gh㈵>)r)r r1r)r%r3r4r5r*r*r+test_circmean_scalars z"TestCircFuncs.test_circmean_scalarcCs@tjtjdddtjtj }t|tjkt|tj kdS)Nrrg?)r r1r1r2pir)r%mr*r*r+test_circmean_range!sz!TestCircFuncs.test_circmean_rangecCsZtjddgdd}ttj|dddttj|dddd d ttj|ddd d d dS) Nrir()rar)r-g@e@g.}}VW{@gHz>)rg)#p_4@)r1rbr r r1rr2r3)r%r3r*r*r+test_circfuncs_unit8(sz"TestCircFuncs.test_circfuncs_unit8N)r8r9r:rrrr r1r2r3r/r:r<r=r>r?rr@r1rerCrbrFrGrHrIrKrMrNrQrSr*r*r*r+r,ds    r,c@sdeZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ dS)TestMedianTestcCstttjdddgdS)Nr rrH)r-r.r median_test)r%r*r*r+test_bad_n_samples3sz!TestMedianTest.test_bad_n_samplescCstttjgdddgdS)Nr rrH)r-r.r rU)r%r*r*r+r7sz TestMedianTest.test_empty_samplecCs,tttjddddgdddgddgdddS)Nr rrrv)ties)r-r.r rU)r%r*r*r+test_empty_when_ties_ignored;sz+TestMedianTest.test_empty_when_ties_ignoredcCs@tttjdddgdddgtttjdddgdddgdddS)Nr above)rW)r-r.r rU)r%r*r*r+test_empty_contingency_rowCsz)TestMedianTest.test_empty_contingency_rowcCs"tttjdddgddgdddS)Nr rrHrIrJfoo)rW)r-r.r rU)r%r*r*r+ test_bad_tiesOszTestMedianTest.test_bad_tiescCs"tttjdddgddgdddS)Nr rrHrIrJrL)rE)r-r.r rU)r%r*r*r+rMSsz"TestMedianTest.test_bad_nan_policycCs"tttjdddgddgdddS)Nr rrHrIrJr[)r[)r-rHr rU)r%r*r*r+rIVszTestMedianTest.test_bad_keywordcCs`dddg}dddg}tj||\}}}}t|dt|ddgddggt|dt|ddS)Nr rrHg@r)r rUr r)r%r3rrrZmedtblr*r*r+ test_simpleZs    zTestMedianTest.test_simplecCsddddg}ddg}ddd g}tj|||\}}}}t|dt|d ddgddd ggtj|||d d \}}}}t|dt|d ddgdd d ggtj|||d d \}}}}t|dt|d ddgdd d ggdS)Nr rrHrIrJrrrrrrv)rWrY)r rUr )r%r3rzrrrPr]r*r*r+test_ties_optionsis     z TestMedianTest.test_ties_optionscCsddtjg}dddg}tj||dd}tj||dd\}}}}t|tjtjtjdft|d t|d t|d t|tjd dgddggtttj||d ddS)Nr rrIrJr propagate)rErDg?gإvao?g@rrJ) r1rer rUr rrbr-r.)r%r3rZmt1rrrPrr*r*r+test_nan_policy_options}s     z&TestMedianTest.test_nan_policy_optionsc Cs<dddddg}ddddg}tj||\}}}}t|dt|ddgddggtj|\}}} } t||t||tj||dd \}}}}t|dt|ddgddggtj|dd \}}} } t||t||tj||d d \}}}}t|dt|ddgddggtj|d d \}}} } t||t||dS) Nr rrHrIrJrrr)Zlambda_F)r)r rUr Zchi2_contingencyr) r%r3rrrrPr]Zexp_statZexp_pZdofer*r*r+r,s(         zTestMedianTest.test_basicN)r8r9r:rVrrXrZr\rMrIr^r`rbr,r*r*r*r+rT1s  rT)DrKZnumpyr1Z numpy.randomrZ numpy.testingrrrrrrr r rr r-Zscipyr Z common_testsrZ _hypotestsrZ matplotlibZrcParamsZmatplotlib.pyplotZpyplotrr Exceptionr*r+r,r-r.r/r0r1r2r3rcrr;rPrgrrr)r:rMrYrerxrrrrrrrrrrrrr)r,rTr*r*r*r+s (       ![SG,E#DgbG7%,$$$$$$$$$$$$$$$$$$$$$ $$$$ C$rN