a /'a@sddlZddlZddlmZddlmZmZmZm Z m Z m Z m Z m Z ddlZddlmZddlmZddlmZddlmZdd lmZd d lmZdd lmZz&ddlZd ejd<ddlmZ dZ!Wne"ydZ!Yn0gdZ#gdZ$gdZ%gdZ&gdZ'gdZ(gdZ)gdZ*gdZ+gdZ,GdddZ-GdddZ.Gdd d Z/Gd!d"d"Z0Gd#d$d$Z1Gd%d&d&Z2Gd'd(d(Z3Gd)d*d*Z4Gd+d,d,Z5Gd-d.d.e5Z6Gd/d0d0Z7Gd1d2d2Z8Gd3d4d4Z9Gd5d6d6Z:Gd7d8d8Z;Gd9d:d:Zd>Z>Gd?d@d@Z?GdAdBdBZ@gdCZAGdDdEdEZBGdFdGdGZCGdHdIdIZDGdJdKdKZEGdLdMdMZFGdNdOdOZGGdPdQdQZHGdRdSdSZIdS)TN) 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_tstZAggZbackendTF) jt?Zd;?V-??gX9v?-?x&1?+?+?r) rrrr灕Cl?rrrrV-?) ʡE?g/$?rrףp= ?rrrrr) gGz?rrrrrrrrr) rrgCl?Gz?rrh|?rrg\(\?r) 羟/$?gh|?5?r"rrrrrg1Zd?r) r Mb?rgjt?rrg}?5^I?g)\(?rr) rrrrrrrrrr) rrrrrr#r#rrr) rrr!rrrrrr#rc@s$eZdZddZddZddZdS) TestBayes_mvscCstgd}t|\}}}t|jdt|jdddt|jdt|jdddt|jd d d t|jd dddS) N r* "@g4`H#j@gO[%@+=rtol$@gai @gӇu8@& .> _紫@ZdecimalgYs?g1O@)r bayes_mvsr statisticrminmaxselfdatameanvarstdrA?lib/python3.9/site-packages/scipy/stats/tests/test_morestats.py test_basic/s  zTestBayes_mvs.test_basiccCstttjgdSN) assert_raises ValueErrorr r8r<rArArBtest_empty_inputAszTestBayes_mvs.test_empty_inputcCs0td}d}t|}|D]}t||qdS)N)r9r:)nparanger r8r)r<x attributesresirArArBtest_result_attributesDs   z$TestBayes_mvs.test_result_attributesN)__name__ __module__ __qualname__rCrHrPrArArArBr$.sr$c@s,eZdZddZddZddZddZd S) TestMvsdistcCsgd}t|\}}}t|dt|ddddt|dt|ddd dt|d d d t|dd dddS)Nr%r,g?r-r.r/r1r2r3r4r5r6r7)r mvsdistrr>rintervalr;rArArBrCNs   zTestMvsdist.test_basiccCstttjgdSrDrErFr rUrGrArArBrH]szTestMvsdist.test_empty_inputcCsdg}tttj|dSNrrWr<r=rArArB test_bad_arg`szTestMvsdist.test_bad_argcCshtLtdtddtgdDddtgdDWdn1sZ0YdS)NerrorcSsg|] }|qSrAr>.0rLrArArB jz*TestMvsdist.test_warns..rrcSsg|] }|qSrAr\r]rArArBr_kr`rrrb)warningscatch_warnings simplefilterRuntimeWarningr rUrGrArArB test_warnses  zTestMvsdist.test_warnsN)rQrRrSrCrHrZrjrArArArBrTMsrTc@s<eZdZddZddZddZddZd d Zd d Zd S) TestShapiroc Csgd}t|\}}t|}t|dddt|jdddt|dddt|jdddgd}t|\}}t|}t|dddt|jdddt|dd dt|jdd dtjjd d d d d }t|\}}t|}t|dddt|jdddt|dd dt|jdd dgd}d}d} t|\}}t|}t||ddt|j|ddt|| d dt|j| d ddS)N))\(?{Gz@q= ףp@HzG$@@Q @q= ףp?Q@zG?@Q?RQ?Q @(\?zG?zG@(\#@333333@Gz?p= ף@?r&r6Ɍ?)(\?= ףp=?\(\@ffffff@\(\?(\?Gz@(\ףp= @皙?)\(?RQrp= ף?@ ףp= ?{Gz?\(\ @{Gz@Q @IddY?Dio?rbrediNalocscalesize random_stategE?gOٴ?)gn?gjt?gffffff?gMb?gjt?g;On?gjt?gX9v?gS?g(\?gI +?gʡE?gx?g|?5^?rgCl?g1Zd@gS@g(\ @gGz @g rh @g"~j@gQ@gtV@gʡE @ghyܝ?g 0M?rd)r shapirorr9pvaluenormrvs) r<x1wpw shapiro_testx2Zx3Zx4Z W_expectedZ p_expectedrArArBrCos<    zTestShapiro.test_basiccCsgdgdg}t|\}}t|}t|dddt|jdddt|dddt|jdddgdgdg}t|\}}t|}t|d ddt|jd ddt|d d dt|jd d ddS) N) rlrmrnrorprqrrrsrtru) rvrwrxryrzr{r|r}r~rrr&r6r) rrrrrrrrrr) rrrrrrrrrrrrrb)r rrr9r)r<rrrrrrArArBtest_2ds$  zTestShapiro.test_2dcCs&tttjgtttjggggdSrDrErFr rrGrArArBrHszTestShapiro.test_empty_inputcCs4tttjddgtttjtjgdggtddS)Nrrdtype)rErFr rrJarrayobjectrGrArArBtest_not_enough_valuessz"TestShapiro.test_not_enough_valuescCsdg}tttj|dSrXrr<rLrArArBrZszTestShapiro.test_bad_argcCs`td}tj|d<t|\}}t|}t|tjt|jtjt|dt|jddS)Nr1r'r) rJrKnanr rr r9rr)r<rLrrrrArArBtest_nan_inputs     zTestShapiro.test_nan_inputN) rQrRrSrCrrHrrZrrArArArBrkns +rkc@sDeZdZddZddZddZddZd d Zd d Zd dZ dS) TestAndersoncCstd}|jdd}|jdd}t|\}}}t|dd|t|\}}}t||ddtd}d|d<t|\}}}t|ddS) NI2r rg @) rstandard_exponentialstandard_normalr andersonrrJonesr)r<rsrrAcritsigvrArArB test_normals    zTestAnderson.test_normalcCstd}|jdd}|jdd}t|d\}}}t||ddtjdd"t|d\}}}Wdn1sx0Yt||dkdS) NrrrZexponrignore)allr) rrrr rrrJerrstater)r<rrrrrrrArArB test_expons  0zTestAnderson.test_exponc Cstd}d|d<t|d\}}}t|}tj|\}}tj|||}tj|||} t d|d} | t d| d|| ddd} t || dS)Nrrgumbelrrr) rJrr rlengumbel_lfitlogcdflogsfrKr>r) r<rZa2rrnZxbarsrrrOZ expected_a2rArArB test_gumbels *zTestAnderson.test_gumbelcCstttjdgdddSNrplate_of_shrimpdist)rErFr rrGrArArBrZszTestAnderson.test_bad_argcCs0td}|jdd}t|}d}t||dS)Nrrrr9Zcritical_valuesZsignificance_level)rrr rr)r<rrLrNrMrArArBrPs   z#TestAnderson.test_result_attributesc CsFtd}|jdd}t|d\}}}t|d\}}}t||dS)Nrrrrr)rrr rr) r<rrLA1crit1sig1A2crit2sig2rArArB test_gumbel_ls  zTestAnderson.test_gumbel_lc Csptd}|jdd}td}d|d<t|d\}}}t|d\}}} t||ddt||dkdS) NrrrrrZgumbel_rrr)rrrJrr rrr) r<rrrrrrrrrrArArB test_gumbel_rs  zTestAnderson.test_gumbel_rN) rQrRrSrrrrZrPrrrArArArBrs  rc@sTeZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ dS)TestAndersonKSampcCsgd}tgd}tgd}tgd}tj||||fdd\}}}t|ddtgd |d d d t|d dddS)NgYC@gD@gfffffE@g@F@gF@gG@gG@gM@gC@gfffffC@gC@g33333D@gfffffD@g33333sE@gfffffE@gfffffF@A@gA@gC@gD@E@rgF@gF@rffffffA@rg33333A@gB@gfffffB@gD@gffffffE@FZmidrankg@rbgCl?gvq -?g,Ԛ?gl@gPk @rrerdgJ4a?Mb0?atolrJrr anderson_ksamprrrr<t1t2t3t4TktmprArArBtest_example1as  z TestAndersonKSamp.test_example1acCstgd}tgd}tgd}tgd}tj||||fdd\}}}t|ddtgd |d d d t|d dddS)NrrrrTrgQ@rbrrrerdgMb`?rrrrrArArBtest_example1b0s  z TestAndersonKSamp.test_example1bcCsgd}gd}gd}gd}gd}gd}gd}gd}gd } gd } gd } gd } gd } gd}tj||||||||| | | | | |fdd\}}}t|ddtgd|dddt|ddddS)NrI)! ir5:%rAr'i$v "rC9>r)rZr<=1r58OT,;rri6Lrr Fer  Jr 0rir(r!rir 7i@rh/rriWr)xr5r r-Gr.r&*rrer(r3 rbr5r5r7r5r7r rr84_a3r7rdrDMPrr8jR6r.o'?rrrrrfHr rJrbrIrXrHrererr r<rr+r5 gr'r(ii[rbr*rirrei#r( irrr)breU[+rbrrOr5r rF rrr'r5raBrrFrgM @rbg^I +?g~:?g+e?gH.!@gcZB>h@rrerd ^)p?rrr rrrrr<rrrrZt5Zt6Zt7Zt8Zt9Zt10Zt11Zt12Zt13Zt14rrrrArArBtest_example2a@s0    z TestAndersonKSamp.test_example2acCsgd}gd}gd}gd}gd}gd}gd}gd}gd } gd } gd } gd } gd } gd}tj||||||||| | | | | |fdd\}}}t|ddtgd|dddt|ddddS)Nrrrr rr#r(r1r;rNrWrYr\rbTrg'1Z @rbrhrrerdrirrrjrkrArArBtest_example2b`s0    z TestAndersonKSamp.test_example2bcCstddd}t:}|jtddtj||dgdd\}}}Wdn1sT0Yt|dd t|d t6}|jtddt||dg\}}}Wdn1s0Yt|d d t|d t:}|jtd dtj||d gdd\}}}Wdn1s0Yt|ddt|dt6}|jtd dt||d g\}}}Wdn1s0Yt|ddt|dtj||dgdd\}}}t|ddt |ddddt||dg\}}}t|ddt |ddddtj||dgdd\}}}t|ddt |ddddtj||dgdd\}}}t|d dt |d!d"dddS)#Nrrzp-value flooredmessageg@D@Frg= ףpD@rbMbP?gGzD@zp-value capped?g?ܵrd?g?ܵ@g!u?gףp= ׳?g{Gzt?rrr0r&gBiq?g鷯?'@g:M@g46W[1@g1ZGUS?-C6?) rJlinspacer filter UserWarningr rrr r)r<rsupr_rrArArBtest_R_kSamplessB!:  6  <  8      z!TestAndersonKSamp.test_R_kSamplescCstttjtddSNrerErFr rrJrrGrArArBtest_not_enough_samplessz)TestAndersonKSamp.test_not_enough_samplescCs"tttjtdtdfdSr}r~rGrArArBtest_no_distinct_observationssz/TestAndersonKSamp.test_no_distinct_observationscCstttjtdgfdSr}r~rGrArArBtest_empty_samplesz#TestAndersonKSamp.test_empty_samplecCs:gd}tgd}tj||fdd}d}t||dS)NrrFrr)rJrr rr)r<rrrNrMrArArBrPs z(TestAndersonKSamp.test_result_attributesN) rQrRrSrrrlrmr|rrrrPrArArArBrs  Prc @seZdZddZddZddZddZd d Zd d Zd dZ e j dgdgdddfgdgdddfgdgdddfgdgdddfgdgdddfgdgdddfgddZ ddZdS) TestAnsaricCsngd}gd}t,}|tdt||\}}Wdn1sH0Yt|ddt|dddS)Nrrrbrbrdrbrr&rr&rrdr%Ties preclude use of exact statistic.g7@r7g8#oG?)r rxryr ansarir)r<rLyrzWpvalrArArB test_smalls . zTestAnsari.test_smallcCsrtd}td}t,}|tdt||\}}Wdn1sL0Yt|ddt|dddS)NrIkrcrOrCmlr*rr"`r<rOrqtrnr]rrrCr]igrrr*rrrrr*rrrrrCrrg0g@r7g|:?)rJrr rxryr rr)r<ZramsayZparekhrzrrrArArB test_approxs   . zTestAnsari.test_approxcCs4tgdgd\}}t|ddt|dddS)Nrrrbrd)rIrerr*rr(r1r7g?r))r rr)r<rrrArArB test_exacts zTestAnsari.test_exactcCs(tttjgdgtttjdggdSrXrErFr rrGrArArBrZszTestAnsari.test_bad_argcCs`gd}gd}t(}|tdt||}Wdn1sD0Yd}t||dS)Nrrrr9r)r rxryr rr)r<rLrrzrNrMrArArBrPs *z!TestAnsari.test_result_attributescCsTgd}gd}d}tt|d tj||ddWdn1sF0YdS)Nrrer&r)r*z!'alternative' must be 'two-sided'matchfoo alternativer)r<rrrrArArBtest_bad_alternatives zTestAnsari.test_bad_alternativec Csgd}gd}t||\}}tj||ddj}tj||ddj}|dksPJ|dks\Jt|t|t|}t||d|d d t||d d d t|d||d d d tj||ddj}tj||ddj} |dksJ| dksJdS) N)rrerrIrr)rsg!@#@g%@rug)@lessrgreaterffffff?皙?r-q=rr)r rrrpmfrr) r<rrr9rpval_lpval_gZprobZpval_l_reverseZpval_g_reverserArArBtest_alternative_exacts   z!TestAnsari.test_alternative_exactzx, y, alternative, expectedrrrg@A?rra)rdrer&r)r*gG$I?g%I$I?rc)r&r)r*cCs$tj|||dj}t||dddS)Nrrr)r rrr)r<rLrrexpectedrrArArBtest_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)Nrrer{rrrrrrrrrrrg?r)r rrrrr)r<rrrrrrArArBtest_alternative_approxQsz"TestAnsari.test_alternative_approxN)rQrRrSrrrrZrPrrpytestmark parametrizerrrArArArBrs&   rc@s4eZdZddZddZddZddZd d Zd S) TestBartlettc CsBtttttttttt g }t j |\}}t |ddt |dddS)NgzU@/4@r)gG?) g1g2g3g4g5g6g7g8g9g10r bartlettr)r<argsTrrArArB test_datais zTestBartlett.test_datacCstttjdgdSrX)rErFr rrGrArArBrZpszTestBartlett.test_bad_argc Cs4tttttttttt g }t j |}d}t ||dSNr) rrrrrrrrrrr rrr<rrNrMrArArBrPts z#TestBartlett.test_result_attributesc Cs6tttttttttt gf }t t j t j ft j|dSrD)rrrrrrrrrrr rJrr r)r<rrArArBtest_empty_argzszTestBartlett.test_empty_argcCs*tddgddgg}tttjt|dSNrrrbrd)rJrrErFr rrrrArArB test_1d_inputszTestBartlett.test_1d_inputN)rQrRrSrrZrPrrrArArArBrgs rc@sTeZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ dS) TestLevenec CsBtttttttttt g }t j |\}}t |ddt |dddS)NgYpK?r)gI{]?) rrrrrrrrrrr levener)r<rrrrArArBrs zTestLevene.test_datacCsFtjtttdd\}}tjtttddd\}}t||t||dS)Nr>centertrimmedrrZproportiontocut)r rrrrr)r<W1pval1W2pval2rArArB test_trimmed1s    zTestLevene.test_trimmed1c Csgd}gd}tjdtj|}tj||ddd\}}tj||ddd\}}tj|dd|ddd d \}} t||t||t|| dS) Ng333333?@@@@@@gY@rr @rg@rg@gi@r?rrrr>r)rJrandomseed permutationr rr) r<rLrrZW0Zpval0rrrrrArArB test_trimmed2s      $  zTestLevene.test_trimmed2cCsntddd}tjdtj|}|d}tj||dd\}}tj||dd\}}t||t||dS) Nrrr&rrbr>rZmedian)rJrwrrrr rr)r<rLrrrrrrrArArBtest_equal_mean_medians   z!TestLevene.test_equal_mean_mediancCs&tddd}tttj||dddSNrrr&皙?)Z portiontocut)rJrwrE TypeErrorr rrrArArBtest_bad_keywordszTestLevene.test_bad_keywordcCs&tddd}tttj||dddSNrrr&Ztrimr)rJrwrErFr rrrArArBtest_bad_center_valuesz TestLevene.test_bad_center_valuecCstttjdgdSrX)rErFr rrGrArArBtest_too_few_argsszTestLevene.test_too_few_argsc Cs4tttttttttt g }t j |}d}t ||dSr) rrrrrrrrrrr rrrrArArBrPs z!TestLevene.test_result_attributescCs*tddgddgg}tttjt|dSr)rJrrErFr rrrrArArBrszTestLevene.test_1d_inputN) rQrRrSrrrrrrrrPrrArArArBrs  rc@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) Nrg-"^?r7rig`w?i?rgz?binom_test_funcr)r<rrArArBrs     zTestBinomP.test_datacCstt|jgddSNrarErFrrGrArArBtest_bad_len_xszTestBinomP.test_bad_len_xcCs(tt|jdgtt|jdgdddS)NrrrrrGrArArB test_bad_nszTestBinomP.test_bad_ncCstt|jddgdddS)NrrrrrGrArArB test_bad_ps zTestBinomP.test_bad_pcCsX|jddddd}t|d|jddddd}t|d|jdddd d}t|d dS) Nr=gUUUUUU?rrrg-$Ǻl?rgs0,j.? two-sidedgJfJ"f?rr<rNrArArBtest_alternativess   zTestBinomP.test_alternativesN) rQrRrS__doc__ staticmethodr Z binom_testrrrrrrrArArArBrs rc@seZdZdZedddZdS)TestBinomTestPzJ Tests for stats.binomtest as a replacement for stats.binom_test. NrqrcCst|tj}t|dkr8|d|d}|d}n@t|dkrp|d}|dus\||krdtdt|}ntdtj||||d}|jS)Nrrrzn must be >= xzIncorrect length for x.r) rJZ atleast_1dZastypeZint_rrFr binomtestr)rLrrrresultrArArBrs    zTestBinomTestP.binom_test_func)Nrqr)rQrRrSrrrrArArArBrsrc@sxeZdZdZddZddZddZdd Zej d gd d d Z ej d gdddZ ej dgdddZ ej dgdddZej dgdgdgdgdgdgdgd gd!gd"gd#gd$gd%gd&gd'gd(gd)gd*gd+gd,gd-gd.gd/gd0gd1gd2d3Zd4d5Zej d6d7d8gd9d:Zd;d<Zd=d>Zd?d@ZdAS)B TestBinomTestzTests for stats.binomtest.cCsd}tddd}t|jd|dtddd}t|jd|dtd dd}t|jd |dtd dd}t|jd d ddS)N绽|=iΙi@o@Q?rr/iΙgX7?i ϙg=?iϙgybw?r3r rrrr<r0rNrArArBtest_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)Nrr'r&rrrgX-h?r/rdg㤶y?r7g y}?r)g+,?rbrrqg?rg?g{Gz?g333333?g<,?rrrArArBtest_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)Nriirqrr/rbr-gAL &W?r+rHg! YB?rIrgE]t?r)r&r7rrrArArBtest_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) Nrrqrbrraxiscstj| SrDr binomrrr rArBIsz>TestBinomTest.test_binary_srch_for_binom_tst..r3r/r)cstj|SrDrrr rArBrUs) rJrKZceilr rrsumrintrZfloor)r<krOdZy1ZixZy2rAr rBtest_binary_srch_for_binom_tst<s( z,TestBinomTest.test_binary_srch_for_binom_tstz"alternative, pval, ci_low, ci_high))rg$W ?rg?)rg}?gGI~?r)rg}G?g>?6?g4M?cCsXtjddd|d}t|j|ddt|jd|jdd }t|j|jf||fdddS) Nrrrrrrrrr/皙?rconfidence_level r rrrr proportion_estimate proportion_cilowhighr<rrci_lowci_highrNcirArArBtest_confidence_intervals1\s   z(TestBinomTest.test_confidence_intervals1))rga&+w?rgny?)rg5!Rx?g5%?r)rg~th?gSuR&|?g|KnG?cCsXtjddd|d}t|j|ddt|jd|jdd }t|j|jf||fdddS) Nrbrrrư>r/gQ?r rrr!rArArBtest_confidence_intervals2ks   z(TestBinomTest.test_confidence_intervals2zalternative, pval, ci_high))rg‹լ?g̓A?)rrr)rgE ow?g;Kڞj?cCsNtjddd|d}t|j|dd|jdd}t|jd t|j|dddS) Nrrrrrr&r/rrr)r rrrrr rr )r<rrr#rNr$rArArB!test_confidence_interval_exact_k0zs   z/TestBinomTest.test_confidence_interval_exact_k0zalternative, pval, ci_low))rrr)r6>g@>}_?)rr)gcڒ ?cCsNtjddd|d}t|j|dd|jdd}t|jdt|j|dddS) Nrrrrr&r/rrr)r rrrrr r r)r<rrr"rNr$rArArB%test_confidence_interval_exact_k_is_ns   z3TestBinomTest.test_confidence_interval_exact_k_is_nz+k, alternative, corr, conf, ci_low, ci_high)rbrTrg`?g?)rbrTr grt!?g/?)rbrFrgVK5?g5dM?)rbrFr gXCVu^?g*r[?)rbrTrrgP-V?)rbrTr rg^'?)rbrFrrgS5a?)rbrFr rg ?)rbrTrg!Շ`ⰸ?r)rbrTr g$x" ?r)rbrFrgW~=?r)rbrFr g L?r)rrTrrgX ?)rrFrrgN?)rrTrrg/<7#9?)rrFrrgAA?)rrTrrr)rrFrrr)rrTrg7?r)rrFrg؍s?r)rrTrrr)rrFrrr)rrTrghadn?r)rrFrgAB/?rc CsLtj|dd|d}|rd}nd}|j||d} t| j| jf||fdddS) NrrrZwilsonccZwilson)rmethodr&r/)r rrrrr ) r<rrZcorrZconfr"r#rNr+r$rArArBtest_ci_wilson_methods z#TestBinomTest.test_ci_wilson_methodcCs*tddd}t|jdt|jddS)Nrdr8rrr)r rr rrrrArArB&test_estimate_equals_hypothesized_props z4TestBinomTest.test_estimate_equals_hypothesized_propzk, n)rr)rrcCs>tjtddt||Wdn1s00YdS)Nz must be an integer not less thanrrr rFr r)r<rrrArArBtest_invalid_k_nszTestBinomTest.test_invalid_k_ncCs@tjtddtdddWdn1s20YdS)Nzk must not be greater than nrr7rrrr.rGrArArBtest_invalid_k_too_bigsz$TestBinomTest.test_invalid_k_too_bigcCsNtjdddd}tjtdd|jddWdn1s@0YdS) Nrbrrr zmust be in the intervalrrrr rrr rFrrrArArBtest_invalid_confidence_levelsz+TestBinomTest.test_invalid_confidence_levelcCsNtjdddd}tjtdd|jddWdn1s@0YdS) Nrbrrr zmethod must berzplate of shrimpr+r1rrArArBtest_invalid_ci_methodsz$TestBinomTest.test_invalid_ci_methodN)rQrRrSrr r r rrrrr%r'r(r*r,r-r/r0r2r4rArArArBrsp       rc@sDeZdZddZddZddZddZd d Zd d Zd dZ dS) TestFlignercCs&td}tt||ddddS)Nrer)gnf @gDq?r7)rJrKrr flignerr<rrArArBrs  zTestFligner.test_datac svtjdfdd}|t}|t}|t}tj|||dd\}}tj|||ddd\}}t||t||dS) Nrcs t|dt|S)Nr)rJZasarrayrandnrtolist)grrArBrr`z+TestFligner.test_trimmed1..r>rrrr) rJrrrrrr r6r) r<Z_perturbZg1_Zg2_Zg3_Xsq1rXsq2rrAr;rBrs     zTestFligner.test_trimmed1cCsbgd}gd}tj||ddd\}}tj|dd|dddd \}}t||t||dS) Nrrrrrrrr>r)r r6r)r<rLrr<rr=rrArArBrs  $ zTestFligner.test_trimmed2cCs&tddd}tttj||dddSr)rJrwrErr r6rrArArBr szTestFligner.test_bad_keywordcCs&tddd}tttj||dddSr)rJrwrErFr r6rrArArBrsz!TestFligner.test_bad_center_valuecCstttjdgdSrX)rErFr r6rGrArArBtest_bad_num_argsszTestFligner.test_bad_num_argscCs.td}ttjtjft||dgdS)Nrer)rJrKr rr r6rrArArBrs zTestFligner.test_empty_argN) rQrRrSrrrrrr>rrArArArBr5sr5c@sleZdZddZddZddZddZd d Zd d Zd dZ e j j dde j dgdddZdS)TestMoodcCs&td}tt||ddddS)Nrer)g8!giT?r7)rJrKrr moodr7rArArB test_moods  zTestMood.test_moodcCs`tjdtjdd}tjdd}t||\}}t||\}}t||g| |gdS)NrrrrI)rJrrr8r r@r)r<rrZz1p1Zz2p2rArArBtest_mood_order_of_args%s  z TestMood.test_mood_order_of_argscCsNgd}gd}t|}t|}d|_d|_ttj||ddddgdS)N)g 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?)rr)rIrr g4g^}E ?)rJrshaperr r@)r<rrrArArBtest_mood_with_axis_none/s  z!TestMood.test_mood_with_axis_nonec Csd}tjdtjd|}tjd|}t||\}}t|D]:}t||||gt|dd|f|dd|fqD|}|}tj||dd\}}t|D]:}t||||gt||ddf||ddfqdS)NrerrrIrr ) rJrrr8r r@rangerZ transpose)r<Znyrr z_vectest pval_vectestjrOrArArB test_mood_2dKs   " "zTestMood.test_mood_2dc Cs@d}tjdtjj|}tjj|}tdD]}tj|||d\}}ddgddgddgf}t|||dD]}t|||dD]} |dkr|dd|| f} |dd|| f} nR|dkr||dd| f} ||dd| f} n$||| ddf} ||| ddf} t||| f||| fgt| | qqtq0dS)N)rrer&rrbr rrr)rJrrr8rGr r@r) r<rErrrrHrIZaxes_idxrOrJZslice1Zslice2rArArB test_mood_3dbs,      zTestMood.test_mood_3dcCstttjdggdSrX)rErFr r@rGrArArBtest_mood_bad_arg~szTestMood.test_mood_bad_argc Cstjdtjjddd}tjjddd}tj||dd\}}tj||dd\}}tj||d d\}}||kr||ksnJt|dd d t||d t|d |d tj t dd tj||ddWdn1s0YdS)Nrrr)rrg?rrrrHz>rrrzalternative must be...rz ekki-ekki) rJrrr rrr@rrr rF) r<rLrZstat1rBZstat2rCZstat3Zp3rArArBtest_mood_alternatives zTestMood.test_mood_alternativez,SciPy needs tie correction like R (gh-13730)reasonzalternative, expected))r)!}?g-?)r)g~?g4?)r)rRg-?cCs.gd}gd}tj|||d}t||dS)Nrrr)r r@r)r<rrrLrrNrArArBtest_mood_alternative_against_Rs z(TestMood.test_mood_alternative_against_RN)rQrRrSrArDrFrKrLrMrOrrZxfailrrSrArArArBr?s  r?c@sVeZdZddZddZddZejje ddd d Z d d Z d dZ ddZ dS) TestProbplotcCsltjjddd}tj|dd\}}gd}t|t|t||tj|dd\}}gd}t||dS) Nr90rFr)g+/gY3gtg5Z0gFjg!2gsR{ܿg7Կg>hǿgcz隇gcz隇?g>h?g7?gsR{?g!2?gFj?g5Z0?gt?gY3?g+/?T)gƝV?g?gà?)r rrprobplotrrJsort)r<rLosmosrZ osm_expectedrNZres_fitZres_fit_expectedrArArBrCs zTestProbplot.test_basicc Cstjjddd}tj|ddd\}}tj|ddd\}}tj|ddd\}}t||t||t||t||tj|ddd\}} dS)Nri@rFsparamsrrrA)r rrrWr) r<rLosm1osr1osm2osr2Zosm3Zosr3rYrZrArArBtest_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)NrrUrFt)rb)rrr\zwrong-dist-namerc@seZdZdZddZdS)z3TestProbplot.test_dist_keyword..custom_distz6Some class that looks just enough like a distribution.cSstjj|ddS)Nr)r)r rppf)r<qrArArBrcsz7TestProbplot.test_dist_keyword..custom_dist.ppfN)rQrRrSrrcrArArArB custom_distsre)rr[)rr) r rrrWrbrrErFAttributeError)r<rLr]r^r_r`rerArArBtest_dist_keywords   zTestProbplot.test_dist_keyword no matplotlibrPc Cst}|dtjjdddd}tj|td\}}ttj|dd\}}tj|dtd}ttj|ddd}tt |t |kot |kot |kod knt ||t ||t ||t ||t}|d} tj|d| dtdS) NrIrbrtrplotF)rrkr) pltfigure add_subplotr rbrrWcloserrr) r<figrLZres1Zfitres1Zres2Zfitres2Zres3Zres4axrArArBtest_plot_kwargs$ @     zTestProbplot.test_plot_kwargcCstttjdgdddSr)rErFr rWrGrArArBtest_probplot_bad_argssz#TestProbplot.test_probplot_bad_argscCsZttjgddtgtgfttjgddtgtgftjtjdffdS)NFrVTr)r r rWrJrrrGrArArB test_emptys zTestProbplot.test_emptycCshtjddHttjdgddtdgtdgftjtjdffWdn1sZ0YdS)Nr)ZinvalidrTrVr)rJrr r rWrrrGrArArBtest_array_of_size_ones  z#TestProbplot.test_array_of_size_oneN)rQrRrSrCrargrrskipifhave_matplotlibrrrsrtrurArArArBrTs rTc@seZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ e jdgdgdgdgddZddZdS) TestWilcoxoncCshtttjdgddgtttjddgddgdtttjddgddgddtttjdgddddS)NrrZdummyrrZxyzmode)rErFr wilcoxonrGrArArBtest_wilcoxon_bad_arg s z"TestWilcoxon.test_wilcoxon_bad_argcCsRtd}tttj||dddtttj||dddttj||dddddS)Nrwilcoxapproxryprattzsplit)g@Z@r)rJrKrErFr r{r rrArArBtest_zero_diffs zTestWilcoxon.test_zero_diffcCsdgd}gd}t0}|jtddtj||ddd}Wdn1sL0Yt|ddS) Nr)rrrbreSample size too smallrnrr~Z zero_methodrz)rg |N?)r rxryr r{r)r<rLrrzrNrArArB test_pratts 0zTestWilcoxon.test_prattcCs<gd}tj|ddd}tj|ddd}tj|ddd}dS)N) rrrbrrrbrrrrrrr~rrr})r r{)r<Zarrr{rArArBtest_wilcoxon_arg_type(sz#TestWilcoxon.test_wilcoxon_arg_typecCs2gd}tdd}tddt||D}t|j}tj||ddd\}}t|d t|d tj||d dd\}}t|d t|d tj||ddd\}}t|dt|dt gd}t gd}tj||ddd\}}t |dt|dddtj||ddd\}}t |dt|ddddS)N) rrdr8rIr*rdrerrrecSsg|]\}}|g|qSrArA)r^urrArArBr_4r`z7TestWilcoxon.test_accuracy_wilcoxon..rr~ryig.i?rigFYv]Uj?r}iGg%^Ez? r3rrrRyrLrp wrrsrr?F correctionrzrg^҂TestKstat.test_moments_normal_distribution..r)gHQ,?glZ)rI?g88}?gcJN?rvr/r)momentrrbrg{Gz?{Gz?rt)rJrrr8rr r)r<ZmomentsrZm1Zm2Zm3rArrB test_moments_normal_distributions  z*TestKstat.test_moments_normal_distributioncCstttjgdSrD)rErFr rrGrArArBrHszTestKstat.test_empty_inputcCs*td}tj|d<tt|tjdSNr1r&rJrKrr r rrYrArArBrs  zTestKstat.test_nan_inputcCs*td}dD]}tttj||dqdS)Nr)rg/$@r)rJrKrErFr rr<r=rrArArBtest_kstat_bad_args zTestKstat.test_kstat_bad_argN)rQrRrSrrHrrrArArArBrsrc@s$eZdZddZddZddZdS) TestKstatVarcCstttjgdSrDrErFr ZkstatvarrGrArArBrHszTestKstatVar.test_empty_inputcCs*td}tj|d<tt|tjdSrrrYrArArBrs  zTestKstatVar.test_nan_inputcCs dg}d}tttj||ddS)NrrrrrrArArBrZszTestKstatVar.test_bad_argN)rQrRrSrHrrZrArArArBrsrc@sNeZdZddZddZddZejje ddd d Z d d Z d dZ dS) TestPpccPlotcCstjjddddd|_dSNrerirr loggammarrLrGrArArB setup_methodszTestPpccPlot.setup_methodcCsHd}tj|jdd|d\}}gd}t|tjdd|dt||dS)NrerN)g9% ?gxM _?g"; 9?g®XZ?g?Znum)r ppcc_plotrLrrJrw)r<rsvalsppcc ppcc_expectedrArArBrCs zTestPpccPlot.test_basiccCstj|jdddd\}}tj|jddtjd\}}t||ddt||ddt|jdd\}}t||ddt||dddS)Nrr tukeylambdarg#B ;r/)r rrLrr)r<Zsvals1Zppcc1Zsvals2Zppcc2Zsvals3Zppcc3rArArB test_dist s  zTestPpccPlot.test_distrhrPcCsZt}|d}tj|jddtd|||d}tj|jdd|dtdSNrIirrj)rlrmrnr rrLdelaxesror<rprqrArArBrrs   zTestPpccPlot.test_plot_kwargcCs2tttj|jddtttjgddddddS)Nrrrarr)rErFr rrLrGrArArBtest_invalid_inputs$sz TestPpccPlot.test_invalid_inputscCs@tgdd\}}t|tjddddt|tjdtddS)NrrrBrr)r rrrJrwrfloat)r<rrrArArBrt,szTestPpccPlot.test_emptyN) rQrRrSrrCrrrrvrwrrrrtrArArArBrs  rc@s,eZdZddZddZddZddZd S) TestPpccMaxcCsdg}tttj|dddSr)rErFr ppcc_maxrYrArArBtest_ppcc_max_bad_arg6sz!TestPpccMax.test_ppcc_max_bad_argcCs2tjjddddddd}tt|dd d dS) Nffffffrrq'r@{r)r6)r rrrrrrArArBtest_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) Nrrrqrrrrrrrrer6)r rrrr)r<rLZmax1Zmax2Zmax3rArArBr@s zTestPpccMax.test_distcCs`tjjddddddd}tttj|dd ttj|d d d d d ttj|dd d d d dS)Nrrrqrrrr)rrrq)brack)rrrr)r6)rr)r rrrErFrrrrArArB test_brackNszTestPpccMax.test_brackN)rQrRrSrrrrrArArArBr5src@s4eZdZddZddZddZddZd d Zd S) TestBoxcox_llfcCsTtjjdddd}d}t||}|j dtt|d}t ||dS)Nrr1rrrrrr) r rr boxcox_llfrrJlogrr@r)r<rLlmbdallfZ llf_expectedrArArBrC\s  $zTestBoxcox_llf.test_basiccCsDtjjdddd}d}t||}t|t|}t||dddSNrrrrrrr/)r rrrlistrr<rLrrZllf2rArArBtest_array_likecs  zTestBoxcox_llf.test_array_likecCsPtjjdddd}d}t||}t|t||gj}t||g|dddSr)r rrrrJvstackrrrrArArB test_2d_inputjs  zTestBoxcox_llf.test_2d_inputcCstttdgdSrX)rrJisnanr rrGrArArBrttszTestBoxcox_llf.test_emptycCs"gd}td|}t|ddS)N)gh@ m@rgx@ig#:x1)r rr)r<r=rrArArB test_gh_6873ws zTestBoxcox_llf.test_gh_6873N)rQrRrSrCrrrtrrArArArBrZs  r(iU>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 gdddZ ddZ e j dgdddZddZdS) TestBoxcoxcCstjjddddd}tj|dd}t||dtj|dd}t|dd|tj|dd}t|t|tjt|dd}t|t|dS) NrerrUrrrrr)r rrboxcoxrrJrr)r<rLxtrArArBtest_fixed_lmbdaszTestBoxcox.test_fixed_lmbdacCsLd}tjjdddd}||d| }t|\}}t|d|dd dS) Ng@riPi)rrrrrrr6)r rrrr)r<rrLx_invrmaxlogrArArBtest_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)Nrrerrr)Zalphagx@gv[%@rgxˉEk?g8Bhk @r)rrIrpg\?gx̡&@rg7(U@gp: sY@)rJrrr rrrr)r<rngrLr{rVrArArB test_alphas zTestBoxcox.test_alphacCsRtddg}tttj|tttjtdgtttjtdgdggdS)Nrrr)rJrrErFr rrrArArBtest_boxcox_bad_argszTestBoxcox.test_boxcox_bad_argcCsttgjdkdSN)r)rr rrErGrArArBrtszTestBoxcox.test_emptycCs tt\}}t|ddddS)NgsHjdrh㈵>r/r r _boxcox_datar)r<rZlamrArArBrszTestBoxcox.test_gh_6873bounds)rr)rr)rrcsFfdd}tjtd|d\}}d|kr<dksBnJdS)Ncstj|ddSNbounded)rr+r minimize_scalarfunrrArB optimizerszBTestBoxcox.test_bounded_optimizer_within_bounds..optimizerrrrr)r rr)r<rrr{rrArrB$test_bounded_optimizer_within_boundss z/TestBoxcox.test_bounded_optimizer_within_boundscsjtjtdd\}}|d|dfddifdd}tjtd|d\}}||ksXJt|d dS) NrrrZxatolrcstj|ddS)Nr)rr+optionsrrrrrArBrszPTestBoxcox.test_bounded_optimizer_against_unbounded_optimizer..optimizerrrr)r<r{rrZ lmbda_boundedrAr rB2test_bounded_optimizer_against_unbounded_optimizers  z=TestBoxcox.test_bounded_optimizer_against_unbounded_optimizerr)str)rrrcCsBtjtdd tjtd|dWdn1s40YdS)Nz`optimizer` must be a callablerrrr rFr rrr<rrArArB$test_bad_optimizer_type_raises_errorsz/TestBoxcox.test_bad_optimizer_type_raises_errorcCsNdd}d}tjt|d tjtd|dWdn1s@0YdS)NcSsdSrXrArrArArBrszCTestBoxcox.test_bad_optimizer_value_raises_error..optimizerz;`optimizer` must return an object containing the optimal...rrr )r<rrorArArB%test_bad_optimizer_value_raises_errorsz0TestBoxcox.test_bad_optimizer_value_raises_errorN)rQrRrSrrrrrtrrrrrr rrrArArArBrs    rc@sheZdZddZddZddZddZej d gd ej d gd d dZ ddZ ddZ dS)TestBoxcoxNormmaxcCstjjddddd|_dSNrerrUrrrGrArArBr&szTestBoxcoxNormmax.setup_methodcCst|j}t|ddddS)Nm?r&r/r boxcox_normmaxrLrr<rrArArB test_pearsonr)s zTestBoxcoxNormmax.test_pearsonrcCs<tj|jdd}t|dddt|j\}}t||dS)Nmler3 .!?r&r/)r rrLrr)r<rr{Z maxlog_boxcoxrArArBtest_mle-szTestBoxcoxNormmax.test_mlecCs&tj|jdd}t|ddgdddS)Nrr3rrr&r/r)r<Z maxlog_allrArArBtest_all5szTestBoxcoxNormmax.test_allr+)rZpearsonrrrrcsNfdd}tj|j||d}td|ks4Jt|dksJJdS)Ncstj|ddSrrrrrArBr=szITestBoxcoxNormmax.test_bounded_optimizer_within_bounds..optimizer)r+rrr)r rrLrJr)r<r+rrrrArrBr9s   z6TestBoxcoxNormmax.test_bounded_optimizer_within_boundscsvt|j}t|d}t|d|ddGdddfdd}tj|j|d}||ksfJt||d dS) Nreric@s eZdZdS)z?TestBoxcoxNormmax.test_user_defined_optimizer..MyResultN)rQrRrSrArArArBMyResultLsrcs6g}D]}|||q}t||_|SrD)appendrJZargminrL)rZobjsrrNrZ lmbda_rangerArBrOs z@TestBoxcoxNormmax.test_user_defined_optimizer..optimizer)rr)r rrLrJroundrwr)r<rZ lmbda_roundedrZlmbda2rArrBtest_user_defined_optimizerFs    z-TestBoxcoxNormmax.test_user_defined_optimizercCs\tj}tj|jd|dtjtdd"tj|jd|dWdn1sN0YdS)N)rrz,`brack` must be None if `optimizer` is givenr)gr)r rr rrLrr rFr rArArB2test_user_defined_optimizer_and_brack_raises_error\s  zDTestBoxcoxNormmax.test_user_defined_optimizer_and_brack_raises_errorN) rQrRrSrrrrrrrrrr rArArArBr%s rc@sFeZdZddZddZejje ddddZ d d Z d d Z d S)TestBoxcoxNormplotcCstjjddddd|_dSrrrGrArArBrnszTestBoxcoxNormplot.setup_methodcCsHd}tj|jdd|d\}}gd}t|tjdd|dt||dS)Nrerrr)g!X4}?g΁i?g- 15?gW$?g\ܱ{?r)r boxcox_normplotrLrrJrw)r<rZlmbdasrrrArArBrCqs zTestBoxcoxNormplot.test_basicrhrPcCsZt}|d}tj|jddtd|||d}tj|jdd|dtdSr)rlrmrnr r"rLrrorrArArBrrys   z"TestBoxcoxNormplot.test_plot_kwargcCs.tttj|jddtttjddgdddS)Nrrr)rErFr r"rLrGrArArBrsz&TestBoxcoxNormplot.test_invalid_inputscCsttgddjdkdSr)rr r"rrGrArArBrtszTestBoxcoxNormplot.test_emptyN) rQrRrSrrCrrrvrwrrrrtrArArArBr!ms  r!c@s$eZdZddZddZddZdS)TestYeojohnson_llfcCsDtjjdddd}d}t||}t|t|}t||dddS)Nrrrrrrr/)r rryeojohnson_llfrrrrArArBrs  z"TestYeojohnson_llf.test_array_likecCsPtjjdddd}d}t||}t|t||gj}t||g|dddSr)r rrr$rJrrrrrArArBrs  z TestYeojohnson_llf.test_2d_inputcCstttdgdSrX)rrJrr r$rGrArArBrtszTestYeojohnson_llf.test_emptyN)rQrRrSrrrtrArArArBr#sr#c@seZdZddZejdgdddZddZd d Z ejd e j e j gd d Z ejd e je je je jgddZdS)TestYeojohnsoncCstjd}tjjdd|dd}t|dks4Jtj|dd}t||tj|dd}t|dd|dtj|dd}t|t |dtj|dd}t||tjjdd|dd}t|dksJtj|d d}t|t | d tj|dd}t||tj|d d}t|d| ddtjjdd|dd }t|dkr`Jt|dkrtJ|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) NrUrerrrrrrrrb) rJrrr rrr yeojohnsonrr)r<rrLrposnegrArArBrsL     zTestYeojohnson.test_fixed_lmbdar)rrrqrcCsdd}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)Nrrrrr)rJrrErabsZspacingZexpZpower)rLrrr'rArArB_inverse_transforms$ z:TestYeojohnson.test_lmbda_None.._inverse_transformi Nrrr)rrrrrrr6) rJrrnormalr r&rrZlinalgrr>r@)r<rr*Z n_samplesrLrrrrArArBrs  zTestYeojohnson.test_lmbda_NonecCsttgjdkdSr)rr r&rErGrArArBrtszTestYeojohnson.test_emptycCsDtjjdddd}t|\}}tt|\}}t||dddS)Nrrrrrr/)r rrr&rr)r<rLZxt1r{Zxt2rArArBrszTestYeojohnson.test_array_likercCsNtjd|d}d}tjt|dt|Wdn1s@0YdS)Nr&rz>Yeo-Johnson transformation is not defined for complex numbers.r)rJrKrr rFr r&)r<rrLerr_msgrArArBtest_input_dtype_complexsz'TestYeojohnson.test_input_dtype_complexcCsZtjd|d}tjdtjd}t|\}}t|\}}t||ddt||dddS)Nr*rrNr/)rJrKZfloat64r r&r)r<rZx_intZx_floatZxt_intZ lmbda_intZxt_floatZ lmbda_floatrArArBtest_input_dtype_integer s z'TestYeojohnson.test_input_dtype_integerN)rQrRrSrrrrrrtrrJZ complex64Z complex128r-Zint8uint8Zint16Zint32r.rArArArBr%s/ $ r%c@s$eZdZddZddZddZdS)TestYeojohnsonNormmaxcCstjjddddd|_dSrrrGrArArBrsz"TestYeojohnsonNormmax.setup_methodcCst|j}t|ddddS)Ng?r&r/)r yeojohnson_normmaxrLrrrArArBrs zTestYeojohnsonNormmax.test_mlecCs*gd}t|}tj|ddds&JdS)N)gffffff@g rrgffffff?g333333@rgffffff@g?g @rrg"@rsggzG?rpr)r r1rJZallclose)r<rLrrArArBtest_darwin_examples z)TestYeojohnsonNormmax.test_darwin_exampleN)rQrRrSrrr2rArArArBr0sr0c@s4eZdZejdejdfejdfej dfgddZ ddZ ejd eje j feje jfej e jfgd d Zd d ZddZddZejdejdfejdfej dfgddZejdejejej gddZejdejejej gddZejdeje jdddfeje jdddfej e jdddfgd d!Zejdejd"e gd#e d$d%gdfejd&e gd'e dd(gdfej d)e gd*e d+d,gdfgd-d.Zejdejdfejdfej dfgd/d0Zejdejejej gd1d2Zejdejejej gd3d4Zejd5d6d7d8d9d:d;e jge d6d7d8d9d:d;e jgdd?e jd@dAgggejdejejej gdBdCZejd5d6d7d8d9d:d;e jge d6d7d8d9d:d;e jgdd?e jd@dAgggejdejejej gdDdEZdFdGZ dHdIZ!dJdKZ"dLS)M TestCircFuncsztest_func,expectedgA:wv?gk~_ЀBE@gE|2@cCs(tgd}t||dd|dddSNcrerrXr^hr rNr/rJrrr< test_funcrrLrArArBtest_circfuncs'szTestCircFuncs.test_circfuncscCs~tgd}|}tj|dd}t||dd|}tj|dd}t||dd|}tj |dd}t||dddS)N)rr&r rg4@g3333333@r8r9rr/rv) rJrr>r circmeanrr?circvarr@circstd)r<rLM1M2V1V2S1S2rArArBtest_circfuncs_small/sz"TestCircFuncs.test_circfuncs_smallztest_func, numpy_funccCs<tdgddgd}||}||}t||dddS)Ng˛ɦv9?rguv9?rg:0yU>rr:)r<r<Z numpy_funcrLZcircstatr+rArArBtest_circfuncs_close=sz"TestCircFuncs.test_circfuncs_closecstgdgdgdgtjdd}tjdd}t||ddtjddd }fd d tjd D}t||ddtjdd d }fd d tjdD}t||dddS)Nr5_r)rd`r']ier'r*ifrdidr8r9r.r/rr rcsg|]}tj|ddqSr8r9r r?r^rOrLrArBr_Qr`z4TestCircFuncs.test_circmean_axis..rcs&g|]}tjdd|fddqSNr8r9rQrRrSrArBr_Ur`)rJrr r?ravelrrGrE)r<rBrCrArSrBtest_circmean_axisHs z TestCircFuncs.test_circmean_axiscstgdgdgdgtjdd}tjdd}t||ddtjddd }fd d tjd D}t||ddtjdd d }fd d tjdD}t||dddS)Nr5rJrNr8r9dy=r/rrOcsg|]}tj|ddqSrPr r@rRrSrArBr_br`z3TestCircFuncs.test_circvar_axis..rcs&g|]}tjdd|fddqSrTrXrRrSrArBr_fr`)rJrr r@rUrrGrE)r<rDrErArSrBtest_circvar_axisXs zTestCircFuncs.test_circvar_axiscstgdgdgdgtjdd}tjdd}t||ddtjddd }fd d tjd D}t||ddtjdd d }fd d tjdD}t||dddS)Nr5rJrNr8r9rWr/rrOcsg|]}tj|ddqSrPr rArRrSrArBr_sr`z3TestCircFuncs.test_circstd_axis..rcs&g|]}tjdd|fddqSrTrZrRrSrArBr_wr`)rJrr rArUrrGrE)r<rFrGrArSrBtest_circstd_axisis zTestCircFuncs.test_circstd_axiscCs"gd}t||dd|dddSr4)rr;rArArBtest_circfuncs_array_likezsz'TestCircFuncs.test_circfuncs_array_liker<cCstt|gdSrD)rrJr)r<r<rArArBrtszTestCircFuncs.test_emptycCs.ddddddtjg}tt||dddS) Nr6rerrXrr7r8r9rJrrrr<r<rLrArArBtest_nan_propagatesz TestCircFuncs.test_nan_propagateg5:v@gf Pb?rgڴV0@gLԿAB@g#q@gauy+@c Cstgdddddddtjgdtjtjtjtjtjtjgg}|D]Z}||d |d }|durptt|qFt|d ||d d tt|ddqFdS)N)r6rerrXrr7rrKr)rdrLr'rMrr8rOrrNr/)rJrrkeysrrrrr<r<rrLroutrArArBtest_nan_propagate_arrays  z&TestCircFuncs.test_nan_propagate_arraygrQvv@)gv@rrg8v@rgu@g5@v?g2>Bjv@gtUuYK@)gR@zw<5?rdg~/ϔ(@ K ?regsQP@gJPm$@)gj@竃?rfg)h; @{?rggun2@g+fa @c Cstddddddtjgddd d d d tjgtjtjtjtjtjtjtjgg}|D]\}||d d|d}|durt|||ddqTt|dd||ddtt|dqTdS)Nr6rerrXrr7rKr)rdrLr'rMr8omit)r  nan_policyrrNr/r)rJrrr`rrrrarArArBtest_nan_omit_arrays z!TestCircFuncs.test_nan_omit_arraycCs0ddddddtjg}t||ddd |d d dS) Nr6rerrXrr7r8rhr rirNr/)rJrrr;rArArB test_nan_omitszTestCircFuncs.test_nan_omitcCs2tjtjtjtjtjg}tt||dddS)Nrhrir]r^rArArBtest_nan_omit_allszTestCircFuncs.test_nan_omit_allcCsjttjtjtjtjtjgtjtjtjtjtjgg}||ddd}tt|tt|dkdS)Nrhr)rirr)rJrrrrrr)r<r<rLrbrArArBtest_nan_omit_all_axiss z$TestCircFuncs.test_nan_omit_all_axisrLr6rerrXrr7rKr)rdrLr'rMcCstt||ddddS)Nr8raiserkrErFr^rArArBtest_nan_raiseszTestCircFuncs.test_nan_raisecCstt||ddddS)Nr8foobarrkrqr^rArArBtest_bad_nan_policysz!TestCircFuncs.test_bad_nan_policycCs$d}|}t|}t||dddS)Nrrr/)r r?r)r<rLrBrCrArArBtest_circmean_scalars z"TestCircFuncs.test_circmean_scalarcCs@ttdddtjtj }t|tjkt|tj kdS)Nrrr)r r?rJrKZpir)r<mrArArBtest_circmean_rangesz!TestCircFuncs.test_circmean_rangecCsZtjddgdd}ttj|dddttj|dddd d ttj|ddd d d dS) Nrr/rrr9g@e@g.}}VW{@rNr/g)#p_4@)rJrr r r?rr@rArrArArBtest_circfuncs_unit8sz"TestCircFuncs.test_circfuncs_unit8N)#rQrRrSrrrr r?r@rAr=rHrJr>r?r@rIrVrYr[r\rtr_rrcrrjrlrnrorrrtrurwryrArArArBr3&s                        r3c@sdeZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ dS)TestMedianTestcCstttjgddSrrErFr median_testrGrArArBtest_bad_n_samples sz!TestMedianTest.test_bad_n_samplescCstttjggddSrr{rGrArArBr sz TestMedianTest.test_empty_samplecCs&tttjgdgdddgdddS)N)rrrr)rrrrrrZtiesr{rGrArArBtest_empty_when_ties_ignored sz+TestMedianTest.test_empty_when_ties_ignoredcCs8tttjgdgdtttjgdgddddS)N)rrrabover~r{rGrArArBtest_empty_contingency_row sz)TestMedianTest.test_empty_contingency_rowcCs tttjgdddgdddS)Nrardrerr~r{rGrArArB test_bad_ties szTestMedianTest.test_bad_tiescCs tttjgdddgdddS)Nrardrersrmr{rGrArArBrt sz"TestMedianTest.test_bad_nan_policycCs tttjgdddgdddS)Nrardrer)r)rErr r|rGrArArBr# szTestMedianTest.test_bad_keywordcCs\gd}gd}t||\}}}}t|dt|ddgddggt|dt|ddS)Nrarrrr)r r|r r)r<rLrrrZmedtblrArArB test_simple' s  zTestMedianTest.test_simplecCsgd}ddg}gd}t|||\}}}}t|dt|gdgdgtj|||dd\}}}}t|dt|gdgd gtj|||d d\}}}}t|dt|gd gd gdS) Nrrer&)r)r*r')rrrb)rdrrrr~)rdrrr)rrrb)r r|r )r<rLrzrrrvrrArArBtest_ties_options6 s   z TestMedianTest.test_ties_optionscCsddtjg}gd}tj||dd}tj||dd\}}}}t|tjtjtjdft|dt|dt|d t|td dgddggtttj||d ddS) Nrr)rdrer&Z propagatermrhg?gإvao?rrrp) rJrr r|r rrrErF)r<rLrZmt1rrrvrbrArArBtest_nan_policy_optionsJ s    z&TestMedianTest.test_nan_policy_optionsc Cs2gd}gd}t||\}}}}t|dt|ddgddggt|\}}} } t||t||tj||dd\}}}}t|dt|ddgddggtj|dd\}}} } t||t||tj||dd \}}}}t|dt|ddgddggtj|dd \}}} } t||t||dS) Nrc)rrdr&r*rdrrr)Zlambda_F)r)r r|r Zchi2_contingencyr) r<rLrrrrvrZexp_statZexp_pZdoferArArBrCW s(        zTestMedianTest.test_basicN)rQrRrSr}rrrrrtrrrrrCrArArArBrzs  rz)JrfZnumpyrJZ numpy.randomrZ numpy.testingrrrrrrr r rr rEZscipyr r Zscipy.stats.morestatsrZ common_testsrZ _hypotestsrZscipy.stats._binomtestrZ matplotlibZrcParamsZmatplotlib.pyplotZpyplotrlrw Exceptionrrrrrrrrrrr$rTrkrrrrrrrrr5r?rTrxrrrrrrrrr!r#r%r0r3rzrArArArBs~ (           !ZVGE'SD^G6%(2qH#qY