o 5ήcR@sddlZddlZddlZddlmZddlmZmZm Z m Z m Z m Z m Z mZddlZddlmZddlmZddlmZddlmZdd lmZd d lmZmZdd lmZzddlZd ejd<ddl m!Z"dZ#Wn e$ytdZ#YnwgdZ%gdZ&gdZ'gdZ(gdZ)gdZ*gdZ+gdZ,gdZ-gdZ.ddZ/GdddZ0Gdd d Z1Gd!d"d"Z2Gd#d$d$Z3Gd%d&d&Z4Gd'd(d(Z5Gd)d*d*Z6Gd+d,d,Z7Gd-d.d.Z8Gd/d0d0e8Z9Gd1d2d2Z:Gd3d4d4Z;d5d6ZGd;d<d<Z?Gd=d>d>Z@Gd?d@d@ZAGdAdBdBZBGdCdDdDZCGdEdFdFZDgdGZEGdHdIdIZFGdJdKdKZGGdLdMdMZHGdNdOdOZIGdPdQdQZJGdRdSdSZKGdTdUdUZLGdVdWdWZMdS)XN) 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_get_wilcoxon_distr2)_binary_search_for_binom_tstAggbackendTF) jt?Zd;?V-??gX9v?-?x&1?+?+?r) rrrr灕Cl?rrrrV-?) ʡE?g/$?rrףp= ?rrrrr) gGz?rrrr"rrrrr) rrgCl?Gz?rrh|?rrg\(\?r) 羟/$?gh|?5?r%rr rr"rg1Zd?r) r#Mb?rgjt?rrg}?5^I?g)\(?rr) rrrrrrrrrr) rrrr"rr&r&rrr!) r!r"r$rrrr!rr&rcOsttjj|i|SN)nplogr gammarvs)argskwargsr.G/tmp/pip-target-vg8gfxp4/lib/python/scipy/stats/tests/test_morestats.py_old_loggamma_rvs1r0c@$eZdZddZddZddZdS) TestBayes_mvscCstgd}t|\}}}t|jdt|jdddt|jdt|jdddt|jd d d t|jd dddS) N r9 "@g4`H#j@gO[%@+=rtol$@gai @gӇu8@& .> _紫@decimalgYs?g1O@)r bayes_mvsr statisticrminmaxselfdatameanvarstdr.r.r/ test_basic6s   zTestBayes_mvs.test_basiccCtttjgdSr') assert_raises ValueErrorr rHrLr.r.r/test_empty_inputHzTestBayes_mvs.test_empty_inputcCs0td}d}t|}|D]}t||qdS)N)rIrJ)r(aranger rHr)rLx attributesresir.r.r/test_result_attributesKs   z$TestBayes_mvs.test_result_attributesN)__name__ __module__ __qualname__rQrVr^r.r.r.r/r35s r3c@,eZdZddZddZddZddZd S) TestMvsdistcCsgd}t|\}}}t|dt|ddddt|dt|ddd dt|d d d t|dd dddS)Nr4r;g?r<r=r>r@rArBrCrDrErG)r mvsdistrrNrintervalrKr.r.r/rQUs    zTestMvsdist.test_basiccCrRr'rSrTr rdrUr.r.r/rVdrWzTestMvsdist.test_empty_inputcCdg}tttj|dSNrrfrLrMr.r.r/ test_bad_arggzTestMvsdist.test_bad_argcCsjt'tdtddtgdDddtgdDWddS1s.wYdS)NerrorcSg|]}|qSr.rN.0rZr.r.r/ qz*TestMvsdist.test_warns..rrcSrmr.rnror.r.r/rqrrrrrrt)warningscatch_warnings simplefilterRuntimeWarningr rdrUr.r.r/ test_warnsls  "zTestMvsdist.test_warnsN)r_r`rarQrVrjr|r.r.r.r/rcTs  rcc@DeZdZddZddZddZddZd d Zd d Zd dZ dS) 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= ף@?r5rEɌ?)(\?= ףp=?\(\@ffffff@\(\?(\?Gz@(\ףp= @皙?)\(?RQrp= ף?@ ףp= ?{Gz?\(\ @{Gz@Q @IddY?Dio?rtrwdiNalocscalesize 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?rv)r shapirorrIpvaluenormr+) rLx1wpw shapiro_testx2x3x4 W_expected p_expectedr.r.r/rQvs<    zTestShapiro.test_basiccCsgdgdg}t|\}}t|}t|dddt|jdddt|dddt|jdddgdgdg}t|\}}t|}t|d ddt|jd ddt|d d dt|jd d ddS) N) rrrrrrrrrr) rrrrrrrrrrrr5rEr) rrrrrrrrrr) rrrrrrrrrrrrrt)r rrrIr)rLrrrrrr.r.r/test_2ds$  zTestShapiro.test_2dcCs&tttjgtttjggggdSr'rSrTr rrUr.r.r/rVzTestShapiro.test_empty_inputcCs4tttjddgtttjtjgdggtddS)Nrrdtype)rSrTr rr(arrayobjectrUr.r.r/test_not_enough_valuess"z"TestShapiro.test_not_enough_valuescCrgrhrrLrZr.r.r/rjrkzTestShapiro.test_bad_argcCs`td}tj|d<t|\}}t|}t|tjt|jtjt|dt|jddS)Nr@r6r) r(rYnanr rr rIrr)rLrZrrrr.r.r/test_nan_inputs     zTestShapiro.test_nan_inputcCs2tgd\}}t|}d}t||dddS)N)ii =i@)gPq~?gF?h㈵>r>)r boxcoxrr)rL trans_valmaxlogr\refr.r.r/ test_gh14462s zTestShapiro.test_gh14462N) r_r`rarQrrVrrjrrr.r.r.r/r~us+ r~c@r}) TestAndersoncCstd}|jdd}|jdd}t|\}}}t|dd|t|\}}}t||ddtd}d|d<t|\}}}t|ddS) NI2r rg @) rstandard_exponentialstandard_normalr andersonrr(onesr)rLrsrrAcritsigvr.r.r/ test_normals    zTestAnderson.test_normalcCstd}|jdd}|jdd}t|d\}}}t||ddtjddt|d\}}}Wdn1sr8r-Zr<=1rD8OT,;r r+i6Lr?r2FerH Jr10r ir7rIr ir@ 7i@r;h/r!rEiWr8xrDr2rUGrVrN*r<rwr7r[ rtrDr]r_rDr_r5rr`4_a3r_rvr/DMPrr`jR6r..o'?r/r/r:rr?fHr3rrrtrXr=Xrprwrwr)r3rdrEr:rD gr6r7ii[rtrRrirrwi#r7 ir/rr8brwU[+rtrGrwrDr1rn r0r@rOrDrBr8r-Fr gM @rtg^I +?g~:?g+e?gH.!@gcZB>h@rrwrv ^)p?r rr rrrrrLrrrrt5t6t7t8t9t10t11t12t13t14rrrr.r.r/test_example2aXs0   z TestAndersonKSamp.test_example2acCr)Nrr#r,r4rFrKrPrYrcrvrrrrTr g'1Z @rtrrrwrvrr rrrr.r.r/test_example2bxs0   z TestAndersonKSamp.test_example2bcCs~tddd}t}|jtddtj||dgdd\}}}Wdn1s*wYt|dd t|d t}|jtddt||dg\}}}Wdn1s[wYt|d d t|d t}|jtd dtj||d gdd\}}}Wdn1swYt|ddt|dt}|jtd dt||d g\}}}Wdn1swYt|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@Fr g= ףpD@rtMbP?gGzD@zp-value capped?g?ܵrv?g?ܵ@g!u?gףp= ׳?g{Gzt?rrr?r5gBiq?g鷯?'@g:M@g46W[1@g1ZGUS?-C6?) r(linspacer filter UserWarningr rrr r)rLrsupr_rr.r.r/test_R_kSamplessJ!            z!TestAndersonKSamp.test_R_kSamplescCstttjtddSNrwrSrTr rr(rrUr.r.r/test_not_enough_samplesr1z)TestAndersonKSamp.test_not_enough_samplescCs"tttjtdtdfdSrrrUr.r.r/test_no_distinct_observationssz/TestAndersonKSamp.test_no_distinct_observationscCstttjtdgfdSrrrUr.r.r/test_empty_samplez#TestAndersonKSamp.test_empty_samplecCs:gd}tgd}tj||fdd}d}t||dS)NrrFr r)r(rr rr)rLrrr\r[r.r.r/r^s z(TestAndersonKSamp.test_result_attributesN) r_r`rarrrrrrrrr^r.r.r.r/r6s  P rc @seZdZddZddZddZddZd d Zd d Zd dZ e j dgdgdddfgdgdddfgdgdddfgdgdddfgdgdddfgdgdddfgddZ ddZdS) TestAnsaricCsngd}gd}t}|tdt||\}}Wdn1s$wYt|ddt|dddS)Nrrrtrtrvrtrr5rr5rrvr%Ties preclude use of exact statistic.g7@r_g8#oG?)r rrr ansarir)rLrZyrWpvalr.r.r/ test_smalls  zTestAnsari.test_smallcCsrtd}td}t}|tdt||\}}Wdn1s&wYt|ddt|dddS)NrqkrcrwrkmlrRrrJ`rdrwrqtrnrrrrkrigrrrRrrrrrRrrrrrkrrg0g@r_g|:?)r(rr rrr rr)rLramsayparekhrrrr.r.r/ test_approxs    zTestAnsari.test_approxcCs4tgdgd\}}t|ddt|dddS)Nrrrtrv)rXrwr<r9rr7r@r_g?r8)r rr)rLrrr.r.r/ test_exacts zTestAnsari.test_exactcCs(tttjgdgtttjdggdSrhrSrTr rrUr.r.r/rjszTestAnsari.test_bad_argcCs`gd}gd}t}|tdt||}Wdn1s"wYd}t||dS)NrrrrIr)r rrr rr)rLrZrrr\r[r.r.r/r^s z!TestAnsari.test_result_attributescCsVgd}gd}d}tt|dtj||ddWddS1s$wYdS)Nrrwr5r8r9z!'alternative' must be 'two-sided'matchfoo alternativer)rLrrrr.r.r/test_bad_alternative's "zTestAnsari.test_bad_alternativec Csgd}gd}t||\}}tj||ddj}tj||ddj}|dks(J|dks.Jt|t|t|}t||d|d d t||d d d t|d||d d d tj||ddj}tj||ddj} |dksrJ| dksxJdS) N)rrwrrXr<rA)rg!@#@g%@rg)@lessrgreaterffffff?皙?r-q=rr)r rrrpmfrr) rLrrrIrpval_lpval_gprobpval_l_reversepval_g_reverser.r.r/test_alternative_exact/s   z!TestAnsari.test_alternative_exactzx, y, alternative, expectedrrrg@A?rrs)rvrwr5r8r9gG$I?g%I$I?ru)r5r8r9cCs$tj|||dj}t||dddS)Nrrr)r rrr)rLrZrrexpectedrr.r.r/test_alternative_exact_with_RJsz(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)Nrrwr{rrrrrrrrrrr6g?r)r rr+rrr)rLrrrrrr.r.r/test_alternative_approxisz"TestAnsari.test_alternative_approxN)r_r`rarrrrjr^rrpytestmark parametrizerrr.r.r.r/rs(   rc@4eZdZddZddZddZddZd d Zd S) TestBartlettc CBtttttttttt g }t j |\}}t |ddt |dddS)NgzU@/4@r8gG?) g1g2g3g4g5g6g7g8g9g10r bartlettr)rLr,Trr.r.r/ test_data zTestBartlett.test_datacCtttjdgdSrh)rSrTr rrUr.r.r/rjzTestBartlett.test_bad_argc C4tttttttttt g }t j |}d}t ||dSNr) rrrrrrrrrrr rrrLr,r\r[r.r.r/r^ z#TestBartlett.test_result_attributesc Cs6tttttttttt gf }t t j t j ft j|dSr')rrrrrrrrrrr r(rr r)rLr,r.r.r/test_empty_argszTestBartlett.test_empty_argcC*tddgddgg}tttjt|dSNrrrtrv)r(rrSrTr rrrr.r.r/ test_1d_inputzTestBartlett.test_1d_inputN)r_r`rarrjr^rr r.r.r.r/rs  rc@r) TestLevenec Cr)NgYpK?r8gI{]?) rrrrrrrrrrr levener)rLr,rrr.r.r/rrzTestLevene.test_datacCsFtjtttdd\}}tjtttddd\}}t||t||dS)NrNcentertrimmedrrproportiontocut)r r rrrr)rLW1pval1W2pval2r.r.r/ 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?rrrrNr )r(randomseed permutationr r r) rLrZrrW0pval0rrrrr.r.r/ test_trimmed2s     $  zTestLevene.test_trimmed2cCsntddd}tjdtj|}|d}tj||dd\}}tj||dd\}}t||t||dS) NrrrNr rtrNr median)r(rr"r#r$r r r)rLrZrrrrrrr.r.r/test_equal_mean_medians   z!TestLevene.test_equal_mean_mediancC&tddd}tttj||dddSNrrrN皙?) portiontocut)r(rrS TypeErrorr r rr.r.r/test_bad_keywordrzTestLevene.test_bad_keywordcCr*NrrrNtrimr )r(rrSrTr r rr.r.r/test_bad_center_valuerz TestLevene.test_bad_center_valuecCrrh)rSrTr r rUr.r.r/test_too_few_argsszTestLevene.test_too_few_argsc Crr) rrrrrrrrrrr r rrr.r.r/r^rz!TestLevene.test_result_attributescCrr)r(rrSrTr r rrr.r.r/r r zTestLevene.test_1d_inputN) r_r`rarrr'r)r/r2r3r^r r.r.r.r/r s   r c@s`eZdZdZeejZddZddZ ddZ dd Z d d Z e jjejd kd dddZdS) TestBinomPzTests for stats.binom_test.cCsR|dd}t|dd|dd}t|dd|jdd gd d }t|d ddS) Nrg-"^?r_r*ig`w?i?rgz?binom_test_funcr)rLrr.r.r/rs    zTestBinomP.test_datacCstt|jgddSNrsrSrTr:rUr.r.r/test_bad_len_xzTestBinomP.test_bad_len_xcCs(tt|jdgtt|jdgdddS)Nrrrr<rUr.r.r/ test_bad_nszTestBinomP.test_bad_ncCstt|jddgdddS)Nrrr8r<rUr.r.r/ test_bad_ps  zTestBinomP.test_bad_pcCsX|jddddd}t|d|jddddd}t|d|jdddd d}t|d dS) NregUUUUUU?rrrg-$Ǻl?rgs0,j.? two-sidedgJfJ"f?r9rLr\r.r.r/test_alternativess   zTestBinomP.test_alternativeslz32-bit does not overflowreasoncCstt|jddtjjddS)Nrr5r8)rS OverflowErrorr:sys float_infominrUr.r.r/test_boost_overflow_raisessz%TestBinomP.test_boost_overflow_raisesN)r_r`ra__doc__ staticmethodr binom_testr:rr=r@rArFrrskipifrJmaxsizerMr.r.r.r/r4s  r4c@seZdZdZedddZdS)TestBinomTestPzJ Tests for stats.binomtest as a replacement for stats.binom_test. NrrDcCst|tj}t|dkr|d|d}|d}n t|dkr8|d}|dus.||kr2tdt|}ntdtj||||d}|jS)Nrrrzn must be >= xzIncorrect length for x.rC) r( atleast_1dastypeint_rrTr binomtestr)rZrrrresultr.r.r/r:s    zTestBinomTestP.binom_test_func)NrrD)r_r`rarNrOr:r.r.r.r/rS srSc@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?rBr rWrrrLr?r\r.r.r/test_two_sided_pvalues1+sz%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)NrZr6rNr[rrgX-h?r>rvg㤶y?r_g y}?r8g+,?rtrrg?rg?g{Gz?g333333?g<,?r\r]r.r.r/test_two_sided_pvalues29sz%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)NrZiirrr>rtrUgAL &W?r:rpg! YB?rXr?gE]t?r8r5r_r\r]r.r.r/test_edge_casesJszTestBinomTest.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) Nrrrtrraxiscstj| Sr'r binomrrr_r.r/frWz>TestBinomTest.test_binary_srch_for_binom_tst..rBr>r8cstj|Sr'rdrfr_r.r/rgr) r(rYceilr rersumrintrfloor)rLkr]dy1ixy2r.r_r/test_binary_srch_for_binom_tstYs( z,TestBinomTest.test_binary_srch_for_binom_tstz"alternative, pval, ci_low, ci_high))rg$W ?rg?)rg}?gGI~?r)rDg}G?g>?6?g4M?cCXtjddd|d}t|j|ddt|jd|jdd }t|j|jf||fdddS) Nr<rrrrrrr>皙?rconfidence_level r rWrrr proportion_estimate proportion_cilowhighrLrrci_lowci_highr\cir.r.r/test_confidence_intervals1y   z(TestBinomTest.test_confidence_intervals1))rga&+w?rgny?)rg5!Rx?g5%?r)rDg~th?gSuR&|?g|KnG?cCrs) Nrtrrurtư>r>gQ?r#rvrxr}r.r.r/test_confidence_intervals2rz(TestBinomTest.test_confidence_intervals2zalternative, pval, ci_high))rg‹լ?g̓A?)rrr)rDgE ow?g;Kڞj?cCsNtjddd|d}t|j|dd|jdd}t|jd t|j|dddS) NrrrrCrr>rrvr)r rWrrrzr r{r|)rLrrrr\rr.r.r/!test_confidence_interval_exact_k0   z/TestBinomTest.test_confidence_interval_exact_k0zalternative, pval, ci_low))rrr)r6>g@>}_?)rDrgcڒ ?cCsNtjddd|d}t|j|dd|jdd}t|jdt|j|dddS) NrrrCrr>rrvr)r rWrrrzr r|r{)rLrrr~r\rr.r.r/%test_confidence_interval_exact_k_is_nrz3TestBinomTest.test_confidence_interval_exact_k_is_nz+k, alternative, corr, conf, ci_low, ci_high)rtrDTrg`?g?)rtrDTr#grt!?g/?)rtrDFrgVK5?g5dM?)rtrDFr#gXCVu^?g*r[?)rtrTrrgP-V?)rtrTr#rg^'?)rtrFrrgS5a?)rtrFr#rg ?)rtrTrg!Շ`ⰸ?r)rtrTr#g$x" ?r)rtrFrgW~=?r)rtrFr#g L?r)rrDTrrgX ?)rrDFrrgN?)rrTrrg/<7#9?)rrFrrgAA?)rrTrrr)rrFrrr)rrDTrg7?r)rrDFrg؍s?r)rrTrrr)rrFrrr)rrTrghadn?r)rrFrgAB/?rc CsLtj|dd|d}|rd}nd}|j||d} t| j| jf||fdddS) Nrr,rtwilsonccwilson)rwmethodrr>)r rWrzrr{r|) rLrmrcorrconfr~rr\rrr.r.r/test_ci_wilson_methods z#TestBinomTest.test_ci_wilson_methodcCs*tddd}t|jdt|jddS)Nrvr`rr)r rWr ryrrEr.r.r/&test_estimate_equals_hypothesized_props z4TestBinomTest.test_estimate_equals_hypothesized_propzk, n)rr)rrcCs@tjtddt||WddS1swYdS)Nz must be an integer not less thanrrr rTr rW)rLrmrr.r.r/test_invalid_k_ns "zTestBinomTest.test_invalid_k_ncCsBtjtddtdddWddS1swYdS)Nzk must not be greater than nrr_rrrrUr.r.r/test_invalid_k_too_bigs "z$TestBinomTest.test_invalid_k_too_bigcCPtjdddd}tjtdd|jddWddS1s!wYdS) Nrtrr,r_zmust be in the intervalrrrvr rWrr rTrzrEr.r.r/test_invalid_confidence_level"z+TestBinomTest.test_invalid_confidence_levelcCr) Nrtrr,r_zmethod must berzplate of shrimprrrEr.r.r/test_invalid_ci_methodrz$TestBinomTest.test_invalid_ci_methodN)r_r`rarNr^r`rarrrrrrrrrrrrrrrr.r.r.r/rY$sr           rYc@r}) TestFlignercCs&td}tt||ddddS)Nrwr)gnf @gDq?r_)r(rYrr fligner)rLrr.r.r/rs zTestFligner.test_datac svtjdfdd}|t}|t}|t}tj|||dd\}}tj|||ddd\}}t||t||dS) Nrcs t|dt|S)NrZ)r(asarrayrandnrtolist)grr.r/rgs z+TestFligner.test_trimmed1..rNr rrr) r(r"rrrrr rr) rL_perturbg1_g2_g3_Xsq1rXsq2rr.rr/rs    zTestFligner.test_trimmed1cCsbgd}gd}tj||ddd\}}tj|dd|dddd \}}t||t||dS) Nrrrr!rrrrNr )r rr)rLrZrrrrrr.r.r/r's  $ zTestFligner.test_trimmed2cCr*r+)r(rrSr.r rrr.r.r/r/)rzTestFligner.test_bad_keywordcCr*r0)r(rrSrTr rrr.r.r/r2-rz!TestFligner.test_bad_center_valuecCrrh)rSrTr rrUr.r.r/test_bad_num_args1rzTestFligner.test_bad_num_argscCs.td}ttjtjft||dgdS)Nrwr)r(rYr rr rrr.r.r/r5s $zTestFligner.test_empty_argN) r_r`rarrr'r/r2rrr.r.r.r/rs rc csgd}gd}t|D]S\}}tj|}|d}|jdddd}|jdddd}t||D]\}} t|d|| D]} |||| <q>q1||t|d \} } | | d g||RVq dS) N))g]YCgQɣ?)gme5gϵ5?)g;jgH?)iFeiHcMi%:rrrrw)r{r|rrrr) enumerater(r" default_rngintegersziprangeshufflesplit) expected_resultsseedssir#rngxytie_indnum_ties_per_indr]rjrZrr.r.r/mood_cases_with_ties:s    rc@sneZdZejdeddZejdgdddZdd Z d d Z d d Z ddZ ddZ ddZdS)TestMoodz$x,y,alternative,stat_expect,p_expectcCs4tj|||d\}}t||ddt||dddS)am Example code used to generate SAS output: DATA myData; INPUT X Y; CARDS; 1 0 1 1 1 2 1 3 1 4 2 0 2 1 2 4 2 9 2 16 ods graphics on; proc npar1way mood data=myData ; class X; ods output MoodTest=mt; proc contents data=mt; proc print data=mt; format Prob1 17.16 Prob2 17.16 Statistic 17.16 Z 17.16 ; title "Mood Two-Sample Test"; proc print data=myData; title "Data for above results"; run; rgؗҜd}tjdtjj|}tjj|}tdD]}tj|||d\}}ddgddgddgf}t|||dD]b}t|||dD]U} |dkr^|dd|| f} |dd|| f} n)|dkru||dd| f} ||dd| f} n||| ddf} ||| ddf} t||| f||| fgt| | qEq9qdS)N)rrwr5r rtrbrrr)r(r"r#rrr rr) rLrrrrcrraxes_idxr]rslice1slice2r.r.r/ test_mood_3ds2       zTestMood.test_mood_3dcCstttjdggdSrh)rSrTr rrUr.r.r/test_mood_bad_argzTestMood.test_mood_bad_argc Cstjdtjjddd}tjjddd}tj||dd\}}tj||dd\}}tj||d d\}}||kr?|ksBJJt|dd d t||d t|d |d tj t ddtj||ddWddS1stwYdS)Nrr7r)rrg?rDrrrHz>rrrzalternative must be...rz ekki-ekki) r(r"r#r rr+rrrr rT) rLrZrstat1rstat2rstat3p3r.r.r/test_mood_alternatives "zTestMood.test_mood_alternativeN)r_r`rarrrrrrrrrrrrr.r.r.r/rQs      rc@sVeZdZddZddZddZejje ddd d Z d d Z d dZ ddZ dS) TestProbplotcCsltjjddd}tj|dd\}}gd}t|t|t||tj|dd\}}gd}t||dS) Nr<90rFr)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 rr+probplotrr(sort)rLrZosmosr osm_expectedr\res_fitres_fit_expectedr.r.r/rQs zTestProbplot.test_basicc Cstjjddd}tj|ddd\}}tj|ddd\}}tj|ddd\}}t||t||t||t||tj|ddd\}} dS)Nri@rFsparamsrrr.)r rr+rr) rLrZosm1osr1osm2osr2osm3osr3rrr.r.r/test_sparams_keywords    z!TestProbplot.test_sparams_keywordcCstjjddd}tj|dddd\}}tj|dtjdd\}}t||t||tttj|dd tttj|gd Gd d d }tj|d dd \}}tj||dd\}}t||t||dS)Nr<rrFt)rt)rrrzwrong-dist-namerc@seZdZdZddZdS)z3TestProbplot.test_dist_keyword..custom_distz6Some class that looks just enough like a distribution.cSstjj|ddS)Nr)r)r rppf)rLqr.r.r/r rhz7TestProbplot.test_dist_keyword..custom_dist.ppfN)r_r`rarNrr.r.r.r/ custom_dists r)rr)rr) r rr+rrrrSrTAttributeError)rLrZrrrrrr.r.r/test_dist_keywords   zTestProbplot.test_dist_keyword no matplotlibrGc Cst}|dtjjdddd}tj|td\}}ttj|dd\}}tj|dtd}ttj|ddd}tt |t |koXt |koXt |koXd knt ||t ||t ||t ||t}|d} tj|d| dtdS) NrqrtrtrplotF)rrr) pltfigure add_subplotr rr+rcloserrr) rLfigrZres1fitres1res2fitres2res3res4axr.r.r/test_plot_kwarg(s$ @      zTestProbplot.test_plot_kwargcCrr)rSrTr rrUr.r.r/test_probplot_bad_argsAz#TestProbplot.test_probplot_bad_argscCsZttjgddtgtgfttjgddtgtgftjtjdffdS)NFrTr)r r rr(rrrUr.r.r/ test_emptyEs zTestProbplot.test_emptycCsjtjdd%ttjdgddtdgtdgftjtjdffWddS1s.wYdS)Nr)invalidrTrr)r(rr r rrrrUr.r.r/test_array_of_size_oneLs "z#TestProbplot.test_array_of_size_oneN)r_r`rarQrrrrrQhave_matplotlibrrrrr.r.r.r/rs  rc@seZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ejdgdgdgdgddZddZdS) TestWilcoxoncCshtttjdgddgtttjddgddgdtttjddgddgddtttjdgddddS)Nrrdummyrrxyzmode)rSrTr wilcoxonrUr.r.r/test_wilcoxon_bad_argTs z"TestWilcoxon.test_wilcoxon_bad_argcCsRtd}tttj||dddtttj||dddttj||dddddS)Nr<wilcoxapproxr prattzsplit)g@Z@r)r(rYrSrTr r r rr.r.r/test_zero_diff]s zTestWilcoxon.test_zero_diffcCsdgd}gd}t}|jtddtj||ddd}Wdn1s&wYt|ddS) Nr)rrrtrwSample size too smallrrr zero_methodr )rg |N?)r rrr r r)rLrZrrr\r.r.r/ test_pratthszTestWilcoxon.test_prattcCs<gd}tj|ddd}tj|ddd}tj|ddd}dS)N) rrrtrrrtrrrrrrrrrr)r r )rLarrrr.r.r/test_wilcoxon_arg_typersz#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) rrvr`rXr9rvrwrrrwcSsg|] \}}|g|qSr.r.)rpurr.r.r/rq~sz7TestWilcoxon.test_accuracy_wilcoxon..rrr ig.i?rigFYv]Uj?riGg%^Ez? r[rr"rzyrtrp wr!rsrrgF correctionr r-g^҂Tg:b+?) rr( concatenaterzerosrr r rrr )rLfreqnumsrZrrrr.r.r/test_accuracy_wilcoxon{s*          z#TestWilcoxon.test_accuracy_wilcoxoncCs@tgd}tgd}tj||ddd}d}t||dS)Nrr"Frr+r)r(rr r r)rLrZrr\r[r.r.r/test_wilcoxon_result_attributess z,TestWilcoxon.test_wilcoxon_result_attributescCstjd}|d|d}}tj||dd}tj|jd}t|j |tj||dd}t |dr7Jt||}t |drDJdS)Nl m$SrXrr rexact zstatistic) r(r"rr r rrrrr4hasattr)rLrrZrr\rr.r.r/test_wilcoxon_has_zstatistics   z)TestWilcoxon.test_wilcoxon_has_zstatisticcCsntjdgddd\}}d}t|dt||ddtjdgdd dd \}}d }t|dt||dddS) Nr,rrr g4IťY?rrr>Tr+g c2_?)r r r r)rLstatr expected_pr.r.r/test_wilcoxon_ties   zTestWilcoxon.test_wilcoxon_tiecCsgd}gd}t}|jtddtj||ddd\}}Wdn1s(wYt|dt|d d d t}|jtddtj||dd dd \}}Wdn1sZwYt|dt|dd d t}|jtddtj||ddd\}}Wdn1swYt|dt|dd d t}|jtddtj||dd dd \}}Wdn1swYt|dt|dd d dS)N) }r)rGr;r)r;r:r;) rzr:r[r;|rr<rrrrrr rOg13}?r5rET)rr,r g7a%?rg?gڏoJT?)r rrr r r r)rLrZrrrrr.r.r/ test_onesideds<         zTestWilcoxon.test_onesidedcCsZtddD]%}t|}t|}t||dddt|tt|dt||qdS)Nrrer)rrrr rrjr)rLrpmf1pmf2r.r.r/test_exact_basics zTestWilcoxon.test_exact_basiccCstgd}tgd}tj||ddd\}}t|dddtj||d dd\}}t|d ddtj||d dd\}}t|d ddtd dd}tdd d}tj||ddd\}}t|dddtj||d dd\}}t|dddtj||d dd\}}t|ddddS)N) g(\?g= ףp=?g(\?gQ޿gQ?g{Gz?gp= ףgq= ףp?ggQ?) rg?gɿg333333?皙gܿgzGgQοg(\gRQrDr3rAg֔?r5rErgYJ?rg*?rr<rrg(?g(?gS?)r(rr r rrY)rLrZrrrr.r.r/test_exact_pvals zTestWilcoxon.test_exact_pvalrZ)rrrt)rrrrw)rrrtrrir8r9cCs@t|\}}t|}||dk}t||t|ddSNrr)r r r(rrjr )rLrZrrwtruer.r.r/test_exact_p_1 s   zTestWilcoxon.test_exact_p_1cCstddd}tddd}tt||tj||ddtdd}t}|jtdd t|\}}Wdn1s@wYttj|d d||ftd d }tt|tj|d ddS) NrrArrr3r r:zExact p-value calculationrrrra)r(rYr r r r rr)rLrZrrnrrrr.r.r/ test_autos   zTestWilcoxon.test_autoN)r_r`rar rrrr1r2r6r9rBrErGrrrrKrLr.r.r.r/rSs$    &  rc@rb) TestKstatcstjdtjdfdddD}gd}t||ddtjd d }tjd d }tjd d }t|||f|dd ddddS)Ni}rcsg|]}t|qSr.)r kstat)rprrMr.r/rq+sz>TestKstat.test_moments_normal_distribution..r)gHQ,?glZ)rI?g88}?gcJN?rr>r)momentrrtrg{Gz?{Gz?r)r(r"r#rrr rP)rLmomentsrm1m2m3r.rOr/ test_moments_normal_distribution(s  "z*TestKstat.test_moments_normal_distributioncCrRr')rSrTr rNrUr.r.r/rV6rWzTestKstat.test_empty_inputcC*td}tj|d<tt|tjdSNr@r5r(rYrr r rNrir.r.r/r9  zTestKstat.test_nan_inputcCs*td}dD] }tttj||dqdS)Nr)rg/$@r?)r(rYrSrTr rNrLrMrr.r.r/test_kstat_bad_arg?s zTestKstat.test_kstat_bad_argN)r_r`rarVrVrr\r.r.r.r/rM's  rMc@r2) TestKstatVarcCrRr'rSrTr kstatvarrUr.r.r/rVGrWzTestKstatVar.test_empty_inputcCrWrXrYrir.r.r/rJrZzTestKstatVar.test_nan_inputcCs dg}d}tttj||ddS)Nrrr?r^r[r.r.r/rjPszTestKstatVar.test_bad_argN)r_r`rarVrrjr.r.r.r/r]Fs r]c@sNeZdZddZddZddZejje ddd d Z d d Z d dZ dS) TestPpccPlotcCtddddd|_dSNrwrrr0rZrUr.r.r/ setup_methodXr1zTestPpccPlot.setup_methodcCHd}tj|jdd|d\}}gd}t|tjdd|dt||dS)NrwrN)g9% ?gxM _?g"; 9?g®XZ?g?num)r ppcc_plotrZrr(r)rLrisvalsppcc ppcc_expectedr.r.r/rQ[ zTestPpccPlot.test_basiccCstj|jdddd\}}tj|jddtjd\}}t||ddt||ddt|jdd\}}t||ddt||dddS)Nrgr tukeylambdarg#B ;r>)r rlrZrqr)rLsvals1ppcc1svals2ppcc2svals3ppcc3r.r.r/ test_distcs  zTestPpccPlot.test_distrrGcCZt}|d}tj|jddtd|||d}tj|jdd|dtdSNrqir<r)rrrr rlrZdelaxesrrLrrr.r.r/ro    zTestPpccPlot.test_plot_kwargcCs2tttj|jddtttjgddddddS)Nrrrsrr)rSrTr rlrZrUr.r.r/test_invalid_inputs|s z TestPpccPlot.test_invalid_inputscCs@tgdd\}}t|tjddddt|tjdtddS)Nrrrjrjr)r rlrr(rr.float)rLrmrnr.r.r/rszTestPpccPlot.test_emptyN) r_r`rarerQrxrrrQrrr~rr.r.r.r/r`Ws  r`c@rb) TestPpccMaxcCsdg}tttj|dddSr)rSrTr ppcc_maxrir.r.r/test_ppcc_max_bad_argsz!TestPpccMax.test_ppcc_max_bad_argcCs2tjjddddddd}tt|dd d dS) Nffffffrr'r@{r8rE)r rqr+rrrr.r.r/test_ppcc_max_basics zTestPpccMax.test_ppcc_max_basiccCsptjjddddddd}tj|dd }tj|tjd }t|d d d t|d d d t|}t|d d d dS) NrrrrrrrrqrrrwrE)r rqr+rr)rLrZmax1max2max3r.r.r/rxs zTestPpccMax.test_distcCs`tjjddddddd}tttj|dd ttj|d d d d d ttj|dd d d d dS)Nrrrrrrr)rrr)brack)rrrr8rE)rr)r rqr+rSrTrrrr.r.r/ test_bracks zTestPpccMax.test_brackN)r_r`rarrrxrr.r.r.r/rs  rc@r) TestBoxcox_llfcCsTtjjdddd}d}t||}|j dtt|d}t ||dS)Nrr1rrrrrr) r rr+ boxcox_llfrr(r)rjrPr)rLrZlmbdallf llf_expectedr.r.r/rQs  $zTestBoxcox_llf.test_basiccCDtjjdddd}d}t||}t|t|}t||dddSNrrrrrrr>)r rr+rlistrrLrZrrllf2r.r.r/test_array_like  zTestBoxcox_llf.test_array_likecCPtjjdddd}d}t||}t|t||gj}t||g|dddSr)r rr+rr(vstackrrrr.r.r/ test_2d_inputs  zTestBoxcox_llf.test_2d_inputcCtttdgdSrh)rr(isnanr rrUr.r.r/rzTestBoxcox_llf.test_emptycCs"gd}td|}t|ddS)N)gh@ m@rgx@ig#:x1)r rr)rLrMrr.r.r/ test_gh_6873s zTestBoxcox_llf.test_gh_6873N)r_r`rarQrrrrr.r.r.r/rs  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) TestBoxcoxcCstddddd}tj|dd}t||dtj|dd}t|dd|tj|dd}t|t|tjt|dd}t|t|ttd d }t|t d dS) Nrwrrrrrrrrr) r0r rrr(r)rrr r.)rLrZxtr.r.r/test_fixed_lmbdaszTestBoxcox.test_fixed_lmbdacCsLd}tjjdddd}||d| }t|\}}t|d|dd dS) Ng@riPi)rrrrrrrE)r rr+rr)rLrrZx_invrrr.r.r/test_lmbda_None s zTestBoxcox.test_lmbda_NonecCstjd}tdd|dd}tj|dd\}}}t|ddgtj|d d\}}}t|d d gtd d |dd}tj|dd\}}}t|ddgtj|dd\}}}t|ddgdS)Nr rwrrr7)alphagx@gv[%@rgxˉEk?g8Bhk @r8rcrXrg\?gx̡&@rg7(U@gp: sY@)r(r"rr0r rr)rLrrZrrer.r.r/ test_alpha*s zTestBoxcox.test_alphacCsRtddg}tttj|tttjtdgtttjtdgdggdS)Nrrr)r(rrSrTr rrr.r.r/test_boxcox_bad_arg;s zTestBoxcox.test_boxcox_bad_argcCttgjdkdSN)r)rr rrrUr.r.r/rDr1zTestBoxcox.test_emptycCs tt\}}t|ddddS)NgsHjdrrr>r r _boxcox_datar)rLrlamr.r.r/rGszTestBoxcox.test_gh_6873bounds)rr)rr)rrFcsHfdd}tjtd|d\}}d|krdks"JJdS)Nctj|ddSNbounded)rrr minimize_scalarfunrr.r/ optimizerRzBTestBoxcox.test_bounded_optimizer_within_bounds..optimizerrrrr)r rr)rLrrrrr.rr/$test_bounded_optimizer_within_boundsOs (z/TestBoxcox.test_bounded_optimizer_within_boundscsjtjtdd\}}|d|dfddifdd}tjtd|d\}}||ks,Jt|d dS) Nrr,rxatolrcstj|ddS)Nr)rroptionsrrrrr.r/rdszPTestBoxcox.test_bounded_optimizer_against_unbounded_optimizer..optimizerrrr)rLrrr lmbda_boundedr.rr/2test_bounded_optimizer_against_unbounded_optimizerYs z=TestBoxcox.test_bounded_optimizer_against_unbounded_optimizerr)str)rrr,cCsDtjtddtjtd|dWddS1swYdS)Nz`optimizer` must be a callablerrrr rTr rrrLrr.r.r/$test_bad_optimizer_type_raises_errorns"z/TestBoxcox.test_bad_optimizer_type_raises_errorcCsPdd}d}tjt|dtjtd|dWddS1s!wYdS)NcSsdSrhr.rr.r.r/ryszCTestBoxcox.test_bad_optimizer_value_raises_error..optimizerz;`optimizer` must return an object containing the optimal...rrr)rLrrr.r.r/%test_bad_optimizer_value_raises_errorts "z0TestBoxcox.test_bad_optimizer_value_raises_errorN)r_r`rarrrrrrrrrrrrrr.r.r.r/r s      rc@sheZdZddZddZddZddZej d gd ej d gd d dZ ddZ ddZ dS)TestBoxcoxNormmaxcCraNrwrrrrdrUr.r.r/rer1zTestBoxcoxNormmax.setup_methodcCt|j}t|ddddS)Nm?rr>r boxcox_normmaxrZrrLrr.r.r/ test_pearsonr zTestBoxcoxNormmax.test_pearsonrcCs<tj|jdd}t|dddt|j\}}t||dS)Nmler .!?rr>)r rrZrr)rLrr maxlog_boxcoxr.r.r/test_mleszTestBoxcoxNormmax.test_mlecCs&tj|jdd}t|ddgdddS)Nrrrrrr>r)rL maxlog_allr.r.r/test_allszTestBoxcoxNormmax.test_allr)rpearsonrrrrcsNfdd}tj|j||d}td|ksJt|dks%JdS)Ncrrrrrr.r/rrzITestBoxcoxNormmax.test_bounded_optimizer_within_bounds..optimizer)rrrr)r rrZr(r)rLrrrrr.rr/rs  z6TestBoxcoxNormmax.test_bounded_optimizer_within_boundscsvt|j}t|d}t|d|ddGdddfdd}tj|j|d}||ks3Jt||d dS) NrwrQic@s eZdZdS)z?TestBoxcoxNormmax.test_user_defined_optimizer..MyResultN)r_r`rar.r.r.r/MyResultsrcs6g}D] }|||q}t||_|Sr')appendr(argminrZ)robjsrr\r lmbda_ranger.r/rs z@TestBoxcoxNormmax.test_user_defined_optimizer..optimizer)rr)r rrZr(roundrr)rLr lmbda_roundedrlmbda2r.rr/test_user_defined_optimizers   z-TestBoxcoxNormmax.test_user_defined_optimizercCs^tj}tj|jd|dtjtddtj|jd|dWddS1s(wYdS)N)rrz,`brack` must be None if `optimizer` is givenr)gr)r rr rrZrr rTrr.r.r/2test_user_defined_optimizer_and_brack_raises_errors "zDTestBoxcoxNormmax.test_user_defined_optimizer_and_brack_raises_errorN) r_r`rarerrrrrrrrrr.r.r.r/rs  rc@sFeZdZddZddZejje ddddZ d d Z d d Z d S)TestBoxcoxNormplotcCrarbrdrUr.r.r/rer1zTestBoxcoxNormplot.setup_methodcCrf)Nrwrgrrh)g!X4}?g΁i?g- 15?gW$?g\ܱ{?rj)r boxcox_normplotrZrr(r)rLrilmbdasrnror.r.r/rQrpzTestBoxcoxNormplot.test_basicrrGcCryrz)rrrr rrZr{rr|r.r.r/rr}z"TestBoxcoxNormplot.test_plot_kwargcCs.tttj|jddtttjddgdddS)Nrrr)rSrTr rrZrUr.r.r/r~sz&TestBoxcoxNormplot.test_invalid_inputscCsttgddjdkdSrI)rr rrrUr.r.r/rrzTestBoxcoxNormplot.test_emptyN) r_r`rarerQrrrQrrr~rr.r.r.r/rs  rc@r2)TestYeojohnson_llfcCr)Nrrrrrrr>)r rr+yeojohnson_llfrrrr.r.r/rrz"TestYeojohnson_llf.test_array_likecCrr)r rr+rr(rrrrr.r.r/rs  z TestYeojohnson_llf.test_2d_inputcCrrh)rr(rr rrUr.r.r/rrzTestYeojohnson_llf.test_emptyN)r_r`rarrrr.r.r.r/rs rc@seZdZddZejdgdddZddZd d Z ejd e j e j gd d Z ejd e je je je jgddZddZdS)TestYeojohnsoncCstjd}tdd|dd}t|dksJtj|dd}t||tj|dd}t|dd|dtj|dd}t|t|dtj|dd}t||tdd|dd}t|dkseJtj|d d}t|t| d tj|dd}t||tj|d d}t|d| ddtdd|dd }t|dkrJt|dkrJ|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) Nrrwrrrrrrrrt) r(r"rr0rr yeojohnsonrr))rLrrZrposnegr.r.r/rsL     $zTestYeojohnson.test_fixed_lmbdar)rr,rrcCsdd}d}tjdtjjdd|d}|||}t|\}}t||dd tdtj |||d d td| dd td| dd dS) NcSstj|j|jd}|dk}t|tdkr"t||d||<nt|||dd|d||<t|dtdkrZdtd| ||ddd|||<|Sdt|| ||<|S)Nrrrrr)r(r.rrabsspacingexppower)rZrrrr.r.r/_inverse_transform6s$  z:TestYeojohnson.test_lmbda_None.._inverse_transformi Nrrr)rrrrQrrrE) r(r"r#normalr rrrlinalgrrNrP)rLrr n_samplesrZrrrr.r.r/r1s  zTestYeojohnson.test_lmbda_NonecCrr)rr rrrUr.r.r/rVr1zTestYeojohnson.test_emptycCsDtjjdddd}t|\}}tt|\}}t||dddS)Nrrrrrr>)r rr+rrr)rLrZxt1rxt2r.r.r/rYszTestYeojohnson.test_array_likercCsPtjd|d}d}tjt|dt|WddS1s!wYdS)Nr5rz>Yeo-Johnson transformation is not defined for complex numbers.r)r(rYrr rTr r)rLrrZerr_msgr.r.r/test_input_dtype_complex_s  "z'TestYeojohnson.test_input_dtype_complexcCsZtjd|d}tjdtjd}t|\}}t|\}}t||ddt||dddS)Nr9rrr>)r(rYfloat64r rr)rLrx_intx_floatxt_int lmbda_intxt_float lmbda_floatr.r.r/test_input_dtype_integergs z'TestYeojohnson.test_input_dtype_integercCsNtgd}t|\}}t|d\}}t||ddt||dddS)N)(HAG."A 5fAg(@AgVAgHzuoAg)\¿+Arg(\^@AgRArg#Ag\µ=Ag ףNArgp=oHAg 6bAg{csAg)\tmhAg/BAg'|Ag(\wipAg ݃hAgQlBAgףp@Arrr>)r(rr rrr)rLrZxt_yeolam_yeoxt_boxlam_boxr.r.r/test_input_high_varianceps z'TestYeojohnson.test_input_high_varianceN)r_r`rarrrrrrrr( complex64 complex128rint8uint8int16int32rr r.r.r.r/rs/ $   rc@r2)TestYeojohnsonNormmaxcCrarrdrUr.r.r/rer1z"TestYeojohnsonNormmax.setup_methodcCr)Ng?rr>)r yeojohnson_normmaxrZrrr.r.r/rrzTestYeojohnsonNormmax.test_mlecCs*gd}t|}tj|dddsJdS)N)gffffff@g rrgffffff?g333333@rgffffff@g?g @rrg"@rggzG?rr)r rr(allclose)rLrZrr.r.r/test_darwin_examples z)TestYeojohnsonNormmax.test_darwin_exampleN)r_r`rarerrr.r.r.r/rs rc@sBeZdZejdejdfejdfej 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 jd(d)e dd*gdfej d+e gd,e d-d.gdfgd/d0Zejdejdfejdfej dfgd1d2Zejdejejej gd3d4Zejdejejej gd5d6Zejd7d8d9d)d:d;dd?d@e jdAdBgggejdejejej gdCdDZejd7d8d9d)d:d;dd?d@e jdAdBgggejdejejej gdEdFZ dGdHZ!dIdJZ"dKdLZ#dMS)N TestCircFuncsztest_func,expectedgA:wv?g@pz?gE|2@cCs(tgd}t||dd|dddSNcrwrrr^hr|rr>r(rrrL test_funcrrZr.r.r/test_circfuncsszTestCircFuncs.test_circfuncscCstgd}|}tj|dd}t||dd|tjd}|d}tj|dd}t||dd| }tj |dd}t||dddS) N)r<rNr3r/g4@g3333333@rrrr>rrr) r(rrNr circmeanrpirOcircvarrPcircstd)rLrZM1M2V1V2S1S2r.r.r/test_circfuncs_smallsz"TestCircFuncs.test_circfuncs_smallztest_func, numpy_funccCs<tdgddgd}||}||}t||dddS)Ng˛ɦv9?rguv9?rg:0yU>rr)rLr numpy_funcrZcircstatrr.r.r/test_circfuncs_closesz"TestCircFuncs.test_circfuncs_closectgdgdgdgtjdd}tjdd}t||ddtjddd }fd d tjd D}t||ddtjdd d }fd d tjdD}t||dddS)Nr_r8rv`r6]ier6r9ifrvidrrr=r>rr|rccg|] }tj|ddqSrrr r rpr]rZr.r/rqz4TestCircFuncs.test_circmean_axis..rc&g|]}tjdd|fddqSNrrr7r8r9r.r/rq&)r(rr r ravelrrr)rLr$r%r.r9r/test_circmean_axiss z TestCircFuncs.test_circmean_axiscr.)Nrr/r3rrdy=r>rr4cr5r6r r"r8r9r.r/rqr:z3TestCircFuncs.test_circvar_axis..rcr;r<rAr8r9r.r/rqr=)r(rr r"r>rrr)rLr&r'r.r9r/test_circvar_axis zTestCircFuncs.test_circvar_axiscr.)Nrr/r3rrr@r>rr4cr5r6r r#r8r9r.r/rqr:z3TestCircFuncs.test_circstd_axis..rcr;r<rDr8r9r.r/rqr=)r(rr r#r>rrr)rLr(r)r.r9r/test_circstd_axisrCzTestCircFuncs.test_circstd_axiscCs"gd}t||dd|dddSr)rrr.r.r/test_circfuncs_array_likesz'TestCircFuncs.test_circfuncs_array_likercCstt|gdSr')rr(r)rLrr.r.r/rrzTestCircFuncs.test_emptycCs.ddddddtjg}tt||dddS) Nrrwrrrrrrr(rrrrLrrZr.r.r/test_nan_propagatesz TestCircFuncs.test_nan_propagateg5:v@gf Pb?rIgee?gOQOv?g#q@gauy+@c Cstgdddddddtjgdtjtjtjtjtjtjgg}|D]-}||d |d }|dur8tt|q#t|d ||d d tt|ddq#dS)N)rrwrrrrrr0r8rvr1r6r2rrr4rrr>)r(rrkeysrrrrrLrrrZrcoutr.r.r/test_nan_propagate_array s  z&TestCircFuncs.test_nan_propagate_arraygrQvv@)gv@rrg8v@rgu@g5@v?g2>Bjv@gG2?)gaF+?z?rNgA ~@碲?rOrrgX҄?gJPm$@)gj@竃?rPg)h; @{?rQgun2@g+fa @c Cstddddddtjgddd d d d tjgtjtjtjtjtjtjtjgg}|D].}||d d|d}|durBt|||ddq*t|dd||ddtt|dq*dS)Nrrwrrrrr0r8rvr1r6r2romit)r| nan_policyrcrr>r)r(rrrJrrrrKr.r.r/test_nan_omit_array s z!TestCircFuncs.test_nan_omit_arraycCs0ddddddtjg}t||ddd |d d dS) NrrwrrrrrrRr|rSrr>)r(rrrr.r.r/ test_nan_omit4 s zTestCircFuncs.test_nan_omitcCs2tjtjtjtjtjg}tt||dddS)NrRrSrGrHr.r.r/test_nan_omit_all= szTestCircFuncs.test_nan_omit_allcCsjttjtjtjtjtjgtjtjtjtjtjgg}||ddd}tt|tt|dkdS)NrRr)rSrcr)r(rrrrrr)rLrrZrLr.r.r/test_nan_omit_all_axisC s z$TestCircFuncs.test_nan_omit_all_axisrZrrwrrrr0r8rvr1r6r2cCtt||ddddS)NrraiserUrSrTrHr.r.r/test_nan_raiseL zTestCircFuncs.test_nan_raisecCrZ)NrfoobarrUr\rHr.r.r/test_bad_nan_policyU r^z!TestCircFuncs.test_bad_nan_policycCs$d}|}t|}t||dddS)Nrrr>)r r r)rLrZr$r%r.r.r/test_circmean_scalar^ s z"TestCircFuncs.test_circmean_scalarcCs@ttdddtjtj }t|tjkt|tj kdS)Nrrr,)r r r(rYr!r)rLmr.r.r/test_circmean_ranged sz!TestCircFuncs.test_circmean_rangecCsZtjddgdd}ttj|dddttj|dddd d ttj|ddd d d dS) Nrr rrrg@e@gLwqA?rr>g)#p_4@)r(rr r r rr"r#rr.r.r/test_circfuncs_uint8k sz"TestCircFuncs.test_circfuncs_uint8N)$r_r`rarrrr r r"r#rr*r(rNrOrPr-r?rBrErFrrIrrMrr!rTrVrXrYr]r`rarcrer.r.r.r/rs                          rc@sdeZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ dS)TestMedianTestcCstttjgddSr;rSrTr median_testrUr.r.r/test_bad_n_samplesv r>z!TestMedianTest.test_bad_n_samplescCstttjggddSr;rgrUr.r.r/rz rz TestMedianTest.test_empty_samplecCs&tttjgdgdddgdddS)N)rrrr)rrrrrrtiesrgrUr.r.r/test_empty_when_ties_ignored~ s z+TestMedianTest.test_empty_when_ties_ignoredcCs8tttjgdgdtttjgdgddddS)N)rrraboverjrgrUr.r.r/test_empty_contingency_row s z)TestMedianTest.test_empty_contingency_rowcC tttjgdddgdddS)NrsrvrwrrjrgrUr.r.r/ test_bad_ties  zTestMedianTest.test_bad_tiescCro)Nrsrvrwr_rWrgrUr.r.r/r` s z"TestMedianTest.test_bad_nan_policycCro)Nrsrvrwr)r)rSr.r rhrUr.r.r/r/ rqzTestMedianTest.test_bad_keywordcCs\gd}gd}t||\}}}}t|dt|ddgddggt|dt|ddS)Nrsrrrr)r rhr r)rLrZrr7rmedtblr.r.r/ test_simple s  zTestMedianTest.test_simplecCsgd}ddg}gd}t|||\}}}}t|dt|gdgdgtj|||dd\}}}}t|dt|gdgd gtj|||d d\}}}}t|dt|gd gd gdS) Nrrwr5)r8r9r6)rrrt)rvrrrrj)rvrrrm)rrrt)r rhr )rLrZrzr7rrbrsr.r.r/test_ties_options s   z TestMedianTest.test_ties_optionscCsddtjg}gd}tj||dd}tj||dd\}}}}t|tjtjtjdft|dt|dt|d t|td dgddggtttj||d ddS) Nrr)rvrwr5 propagaterWrRg?gإvao?rrr[) r(rr rhr rrrSrT)rLrZrmt1rrrbrr.r.r/test_nan_policy_options 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) Nru)rrvr5r9rvrrr)lambda_F)r,)r rhr chi2_contingencyr) rLrZrr7rrbrsexp_statexp_pdofer.r.r/rQ s(        zTestMedianTest.test_basicN)r_r`rarirrlrnrpr`r/rtrvryrQr.r.r.r/rft s  rf)NrxrJnumpyr( numpy.randomr numpy.testingrrrrrrr r rr rSscipyr r scipy.stats._morestatsr common_testsr _hypotestsrrscipy.stats._binomtestr matplotlibrcParamsmatplotlib.pyplotpyplotrr Exceptionrrrrrrrrrrr0r3rcr~rrrrr r4rSrYrrrrrrMr]r`rrrrrrrrrrrfr.r.r.r/s (         !kVGE,SD%^U6%(2uH#e