o tBh@sddlZddlZddlZddlmZddlmZmZm Z m Z m Z m Z m Z mZddlZddlmZddlmZddlmZddlmZdd lmZd d lmZdd lmZzddlZd ejd<ddlm Z!dZ"Wn e#yrdZ"YnwgdZ$gdZ%gdZ&gdZ'gdZ(gdZ)gdZ*gdZ+gdZ,gdZ-GdddZ.GdddZ/Gdd d Z0Gd!d"d"Z1Gd#d$d$Z2Gd%d&d&Z3Gd'd(d(Z4Gd)d*d*Z5Gd+d,d,Z6Gd-d.d.e6Z7Gd/d0d0Z8Gd1d2d2Z9Gd3d4d4Z:Gd5d6d6Z;Gd7d8d8ZGd=d>d>Z?Gd?d@d@Z@GdAdBdBZAgdCZBGdDdEdEZCGdFdGdGZDGdHdIdIZEGdJdKdKZFGdLdMdMZGGdNdOdOZHGdPdQdQZIGdRdSdSZJdS)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_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$rrrr!rg1Zd?r) r"Mb?rgjt?rrg}?5^I?g)\(?rr) rrrrrrrrrr) rrrr!rr%r%rrr ) r r!r#rrrr rr%rc@$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@& .> _紫@decimalgYs?g1O@)r bayes_mvsr statisticrminmaxselfdatameanvarstdrEw/var/www/html/riverr-enterprise-integrations-main/venv/lib/python3.10/site-packages/scipy/stats/tests/test_morestats.py test_basic0s   zTestBayes_mvs.test_basiccCtttjgdSN) assert_raises ValueErrorr r<r@rErErFtest_empty_inputBzTestBayes_mvs.test_empty_inputcCs0td}d}t|}|D]}t||qdS)N)r=r>)nparanger r<r)r@x attributesresirErErFtest_result_attributesEs   z$TestBayes_mvs.test_result_attributesN)__name__ __module__ __qualname__rGrMrVrErErErFr'/s r'c@,eZdZddZddZddZddZd S) TestMvsdistcCsgd}t|\}}}t|dt|ddddt|dt|ddd dt|d d d t|dd dddS)Nr(r/g?r0r1r2r4r5r6r7r8r9r;)r mvsdistrrBrintervalr?rErErFrGOs    zTestMvsdist.test_basiccCrHrIrJrKr r\rLrErErFrM^rNzTestMvsdist.test_empty_inputcCdg}tttj|dSNrr^r@rArErErF test_bad_argazTestMvsdist.test_bad_argcCsjt'tdtddtgdDddtgdDWddS1s.wYdS)NerrorcSg|]}|qSrErB.0rRrErErF kz*TestMvsdist.test_warns..rrcSrerErfrgrErErFrilrjrrrl)warningscatch_warnings simplefilterRuntimeWarningr r\rLrErErF test_warnsfs  "zTestMvsdist.test_warnsN)rWrXrYrGrMrbrtrErErErFr[Ns  r[c@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)r9Ɍ?)(\?= ףp=?\(\@ffffff@\(\?(\?Gz@(\ףp= @皙?)\(?RQrp= ף?@ ףp= ?{Gz?\(\ @{Gz@Q @IddY?Dio?rlrodiNalocscalesize 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?rn)r shapirorr=pvaluenormrvs) r@x1wpw shapiro_testx2x3x4 W_expected p_expectedrErErFrGps<    zTestShapiro.test_basiccCsgdgdg}t|\}}t|}t|dddt|jdddt|dddt|jdddgdgdg}t|\}}t|}t|d ddt|jd ddt|d d dt|jd d ddS) N) rvrwrxryrzr{r|r}r~r) rrrrrrrrrrrr)r9r) rrrrrrrrrr) rrrrrrrrrrrrrl)r rrr=r)r@rrrrrrErErFtest_2ds$  zTestShapiro.test_2dcCs&tttjgtttjggggdSrIrJrKr rrLrErErFrMzTestShapiro.test_empty_inputcCs4tttjddgtttjtjgdggtddS)Nrrdtype)rJrKr rrParrayobjectrLrErErFtest_not_enough_valuess"z"TestShapiro.test_not_enough_valuescCr_r`rr@rRrErErFrbrczTestShapiro.test_bad_argcCs`td}tj|d<t|\}}t|}t|tjt|jtjt|dt|jddS)Nr4r*r) rPrQnanr rr r=rr)r@rRrrrrErErFtest_nan_inputs     zTestShapiro.test_nan_inputN) rWrXrYrGrrMrrbrrErErErFruos+ ruc@DeZdZddZddZddZddZd 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 andersonrrPonesr)r@rsrrAcritsigvrErErF test_normals    zTestAnderson.test_normalcCstd}|jdd}|jdd}t|d\}}}t||ddtjddt|d\}}}Wdn1sr,r!Zr<=1r88OT,;rri6Lr3r&Fer< Jr%0rir+r=rir4 7i@r/h/rr9iWr,xr8r&rIGrJrB*r0ror+rO rlr8rQrSr8rSr)rrT4_a3rSrnr#DMPrrTjR6r".o'?r#r#r.rr3fHr'rfrlrOr1Xrdrororr'rXr9r.r8 gr*r+ii[rlrFrirroi#r+ ir#rr,broU[+rlr;rkr8r%rb r$r4rCr8r}Br,r!FrgM @rlg^I +?g~:?g+e?gH.!@gcZB>h@rrorn ^)p?rrr rrrrr@rrr r t5t6t7t8t9t10t11t12t13t14r r r rErErFtest_example2aAs0   z TestAndersonKSamp.test_example2acCr)Nrrr r(r:r?rDrMrWrjrsrurxr~Trg'1Z @rlrrrornrrrrrrErErFtest_example2bas0   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@Frg= ףpD@rlMbP?gGzD@zp-value capped?g?ܵrn?g?ܵ@g!u?gףp= ׳?g{Gzt?rrr3r)gBiq?g鷯?'@g:M@g46W[1@g1ZGUS?-C6?) rPlinspacer filter UserWarningr rrr r)r@rsupr_r rErErFtest_R_kSamplessJ!            z!TestAndersonKSamp.test_R_kSamplescCstttjtddSNrorJrKr rrPrrLrErErFtest_not_enough_samplesrz)TestAndersonKSamp.test_not_enough_samplescCs"tttjtdtdfdSrrrLrErErFtest_no_distinct_observationssz/TestAndersonKSamp.test_no_distinct_observationscCstttjtdgfdSrrrLrErErFtest_empty_samplez#TestAndersonKSamp.test_empty_samplecCs:gd}tgd}tj||fdd}d}t||dS)NrrFrr)rPrr rr)r@rrrTrSrErErFrVs z(TestAndersonKSamp.test_result_attributesN) rWrXrYrrrrrrrrrVrErErErFrs  P rc @seZdZddZddZddZddZd d Zd d Zd dZ e j dgdgdddfgdgdddfgdgdddfgdgdddfgdgdddfgdgdddfgddZ ddZdS) TestAnsaricCsngd}gd}t}|tdt||\}}Wdn1s$wYt|ddt|dddS)Nrrrlrlrnrlrr)rr)rrnr%Ties preclude use of exact statistic.g7@rSg8#oG?)r rrr ansarir)r@rRyrWpvalrErErF test_smalls  zTestAnsari.test_smallcCsrtd}td}t}|tdt||\}}Wdn1s&wYt|ddt|dddS)Nrekrcrkr_mlrFrr>`rXrkrqtrnryrrr_ryigrrrFrrrrrFrrrrr_rrg0g@rSg|:?)rPrr rrr rr)r@ramsayparekhrrrrErErF test_approxs    zTestAnsari.test_approxcCs4tgdgd\}}t|ddt|dddS)Nrrrlrn)rOror0r-rr+r4rSg?r,)r rr)r@rrrErErF test_exacts zTestAnsari.test_exactcCs(tttjgdgtttjdggdSr`rJrKr rrLrErErFrbszTestAnsari.test_bad_argcCs`gd}gd}t}|tdt||}Wdn1s"wYd}t||dS)Nrrrr=r)r rrr rr)r@rRrrrTrSrErErFrVs z!TestAnsari.test_result_attributescCsVgd}gd}d}tt|dtj||ddWddS1s$wYdS)Nrror)r,r-z!'alternative' must be 'two-sided'matchfoo alternativer)r@rrrrErErFtest_bad_alternatives "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)rrorrOr0r5)rg!@#@g%@rg)@lessrgreaterffffff?皙?r-q=rr)r rrrpmfrr) r@rrr=rpval_lpval_gprobpval_l_reversepval_g_reverserErErFtest_alternative_exacts   z!TestAnsari.test_alternative_exactzx, y, alternative, expectedrrrg@A?rrk)rnror)r,r-gG$I?g%I$I?rm)r)r,r-cCs$tj|||dj}t||dddS)Nrrr)r rrr)r@rRrrexpectedrrErErFtest_alternative_exact_with_R3sz(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)Nrror{rrrrrrrrrrr*g?r)r rrrrr)r@rrrrrrErErFtest_alternative_approxRsz"TestAnsari.test_alternative_approxN)rWrXrYrrrrbrVrrpytestmark parametrizerrrErErErFrs(   rc@4eZdZddZddZddZddZd d Zd S) TestBartlettc CBtttttttttt g }t j |\}}t |ddt |dddS)NgzU@/4@r,gG?) g1g2g3g4g5g6g7g8g9g10r bartlettr)r@argsTrrErErF test_dataj zTestBartlett.test_datacCtttjdgdSr`)rJrKr rrLrErErFrbqzTestBartlett.test_bad_argc C4tttttttttt g }t j |}d}t ||dSNr) rrrrrrrrrrr rrr@rrTrSrErErFrVu z#TestBartlett.test_result_attributesc Cs6tttttttttt gf }t t j t j ft j|dSrI)rrrrrrrrrrr rPrr r)r@rrErErFtest_empty_arg{szTestBartlett.test_empty_argcC*tddgddgg}tttjt|dSNrrrlrn)rPrrJrKr rrrrErErF test_1d_inputzTestBartlett.test_1d_inputN)rWrXrYrrbrVrrrErErErFrhs  rc@r) TestLevenec Cr)NgYpK?r,gI{]?) rrrrrrrrrrr levener)r@rrrrErErFrrzTestLevene.test_datacCsFtjtttdd\}}tjtttddd\}}t||t||dS)NrBcentertrimmedrrproportiontocut)r rrrrr)r@W1pval1W2pval2rErErF 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?rrrrBr)rPrandomseed permutationr rr) r@rRrrW0pval0rrr r rErErF test_trimmed2s     $  zTestLevene.test_trimmed2cCsntddd}tjdtj|}|d}tj||dd\}}tj||dd\}}t||t||dS) NrrrBrrlrBrmedian)rPrrrrr rr)r@rRrrrrr r rErErFtest_equal_mean_medians   z!TestLevene.test_equal_mean_mediancC&tddd}tttj||dddSNrrrB皙?) portiontocut)rPrrJ TypeErrorr rrrErErFtest_bad_keywordrzTestLevene.test_bad_keywordcCrNrrrBtrimr)rPrrJrKr rrrErErFtest_bad_center_valuerz TestLevene.test_bad_center_valuecCrr`)rJrKr rrLrErErFtest_too_few_argsszTestLevene.test_too_few_argsc Crr) rrrrrrrrrrr rrrrErErFrVrz!TestLevene.test_result_attributescCrr)rPrrJrKr rrrrErErFrrzTestLevene.test_1d_inputN) rWrXrYrr rrr$r'r(rVrrErErErFrs   rc@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-"^?rSrig`w?i?r gz?binom_test_funcr)r@rrErErFrs    zTestBinomP.test_datacCstt|jgddSNrkrJrKr/rLrErErFtest_bad_len_xzTestBinomP.test_bad_len_xcCs(tt|jdgtt|jdgdddS)Nrrrr1rLrErErF test_bad_nszTestBinomP.test_bad_ncCstt|jddgdddS)Nrr r-r1rLrErErF test_bad_ps  zTestBinomP.test_bad_pcCsX|jddddd}t|d|jddddd}t|d|jdddd d}t|d dS) NrYgUUUUUU?rr rg-$Ǻl?rgs0,j.? two-sidedgJfJ"f?r.r@rTrErErFtest_alternativess   zTestBinomP.test_alternativeslz32-bit does not overflowreasoncCstt|jddtjjddS)Nrr)r-)rJ OverflowErrorr/sys float_infominrLrErErFtest_boost_overflow_raisessz%TestBinomP.test_boost_overflow_raisesN)rWrXrY__doc__ staticmethodr binom_testr/rr2r5r6r;rrskipifr?maxsizerBrErErErFr)s  r)c@seZdZdZedddZdS)TestBinomTestPzJ Tests for stats.binomtest as a replacement for stats.binom_test. Nrr9cCst|tj}t|dkr|d|d}|d}n t|dkr8|d}|dus.||kr2tdt|}ntdtj||||d}|jS)Nrrrzn must be >= xzIncorrect length for x.r8) rP atleast_1dastypeint_rrKr binomtestr)rRrr rresultrErErFr/s    zTestBinomTestP.binom_test_func)Nrr9)rWrXrYrCrDr/rErErErFrHsrHc@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?rr2iΙgX7?i ϙg=?iϙgybw?r6r rLrrr@r3rTrErErFtest_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)NrOr*rBrPrr gX-h?r2rng㤶y?rSg y}?r,g+,?rlrrg?rg?g{Gz?g333333?g<,?rQrRrErErFtest_two_sided_pvalues2"sz%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)NrOiirrr2rlrIgAL &W?r.rdg! YB?rOr3gE]t?r,r)rSrQrRrErErFtest_edge_cases3szTestBinomTest.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) Nrrrlrraxiscstj| SrIr binomrrrTrErFOrNz>TestBinomTest.test_binary_srch_for_binom_tst..r6r2r,cstj|SrIrYr[rTrErFr\[) rPrQceilr rZrsumrintrfloor)r@krUdy1ixy2rErTrFtest_binary_srch_for_binom_tstBs( z,TestBinomTest.test_binary_srch_for_binom_tstz"alternative, pval, ci_low, ci_high))rg$W ?rg?)rg}?gGI~?r)r9g}G?g>?6?g4M?cCXtjddd|d}t|j|ddt|jd|jdd }t|j|jf||fdddS) Nr0rrrr rrr2皙?rconfidence_level r rLrrr proportion_estimate proportion_cilowhighr@rrci_lowci_highrTcirErErFtest_confidence_intervals1b   z(TestBinomTest.test_confidence_intervals1))rga&+w?rgny?)rg5!Rx?g5%?r)r9g~th?gSuR&|?g|KnG?cCrh) Nrlrrjriư>r2gQ?r"rkrmrrrErErFtest_confidence_intervals2qrwz(TestBinomTest.test_confidence_intervals2zalternative, pval, ci_high))rg‹լ?g̓A?)rrr)r9gE ow?g;Kڞj?cCsNtjddd|d}t|j|dd|jdd}t|jd t|j|dddS) Nrrrr8rxr2rrkr)r rLrrror rprq)r@rrrtrTrurErErF!test_confidence_interval_exact_k0   z/TestBinomTest.test_confidence_interval_exact_k0zalternative, pval, ci_low))rrr)r6>g@>}_?)r9r|gcڒ ?cCsNtjddd|d}t|j|dd|jdd}t|jdt|j|dddS) Nrrr8rxr2rrkr)r rLrrror rqrp)r@rrrsrTrurErErF%test_confidence_interval_exact_k_is_nr{z3TestBinomTest.test_confidence_interval_exact_k_is_nz+k, alternative, corr, conf, ci_low, ci_high)rlr9Trg`?g?)rlr9Tr"grt!?g/?)rlr9FrgVK5?g5dM?)rlr9Fr"gXCVu^?g*r[?)rlrTrrgP-V?)rlrTr"rg^'?)rlrFrrgS5a?)rlrFr"rg ?)rlrTrg!Շ`ⰸ?r)rlrTr"g$x" ?r)rlrFrgW~=?r)rlrFr"g L?r)rr9TrrgX ?)rr9FrrgN?)rrTrrg/<7#9?)rrFrrgAA?)rrTrrr)rrFrrr)rr9Trg7?r)rr9Frg؍s?r)rrTrrr)rrFrrr)rrTrghadn?r)rrFrgAB/?rc CsLtj|dd|d}|rd}nd}|j||d} t| j| jf||fdddS) Nrr!riwilsonccwilson)rlmethodrxr2)r rLrorrprq) r@rbrcorrconfrsrtrTrrurErErFtest_ci_wilson_methods z#TestBinomTest.test_ci_wilson_methodcCs*tddd}t|jdt|jddS)NrnrTrr)r rLr rnrr:rErErF&test_estimate_equals_hypothesized_props z4TestBinomTest.test_estimate_equals_hypothesized_propzk, n)rr)rrcCs@tjtddt||WddS1swYdS)Nz must be an integer not less thanrrr rKr rL)r@rbrrErErFtest_invalid_k_ns "zTestBinomTest.test_invalid_k_ncCsBtjtddtdddWddS1swYdS)Nzk must not be greater than nrrSrrrrLrErErFtest_invalid_k_too_bigs "z$TestBinomTest.test_invalid_k_too_bigcCPtjdddd}tjtdd|jddWddS1s!wYdS) Nrlrr!rTzmust be in the intervalrrrkr rLrr rKror:rErErFtest_invalid_confidence_level"z+TestBinomTest.test_invalid_confidence_levelcCr) Nrlrr!rTzmethod must berzplate of shrimprrr:rErErFtest_invalid_ci_methodrz$TestBinomTest.test_invalid_ci_methodN)rWrXrYrCrSrUrVrgrrrrvryrzr}rrrrrrrErErErFrN sr           rNc@r) TestFlignercC&td}tt||ddddS)Nror)gnf @gDq?rS)rPrQrr flignerr@rrErErFr zTestFligner.test_datac svtjdfdd}|t}|t}|t}tj|||dd\}}tj|||ddd\}}t||t||dS) Nrcs t|dt|S)NrO)rPasarrayrandnrtolist)grrErFr\s z+TestFligner.test_trimmed1..rBrrrr) rPrrrrrr rr) r@_perturbg1_g2_g3_Xsq1rXsq2r rErrFr s    zTestFligner.test_trimmed1cCsbgd}gd}tj||ddd\}}tj|dd|dddd \}}t||t||dS) Nr rrrrrrrBr)r rr)r@rRrrrrr rErErFrs  $ zTestFligner.test_trimmed2cCrr )rPrrJr#r rrrErErFr$rzTestFligner.test_bad_keywordcCrr%)rPrrJrKr rrrErErFr'rz!TestFligner.test_bad_center_valuecCrr`)rJrKr rrLrErErFtest_bad_num_argsrzTestFligner.test_bad_num_argscCs.td}ttjtjft||dgdS)Nror)rPrQr rr rrrErErFrs $zTestFligner.test_empty_argN) rWrXrYrr rr$r'rrrErErErFrs rc@sleZdZddZddZddZddZd d Zd d Zd dZ e j j dde j dgdddZdS)TestMoodcCr)Nror)g8!giT?rS)rPrQrr moodrrErErF test_mood$rzTestMood.test_moodcCs`tjdtjdd}tjdd}t||\}}t||\}}t||g| |gdS)NrrrrO)rPrrrr rr)r@rrz1p1z2p2rErErFtest_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)rOrrWg4g^}E ?)rPrshaperr r)r@rrrErErFtest_mood_with_axis_none5s z!TestMood.test_mood_with_axis_nonec Csd}tjdtjd|}tjd|}t||\}}t|D]}t||||gt|dd|f|dd|fq"|}|}tj||dd\}}t|D]}t||||gt||ddf||ddfqVdS)NrorrrOrrW) rPrrrr rranger transpose)r@nyrr z_vectest pval_vectestjrUrErErF test_mood_2dQs"  " "zTestMood.test_mood_2dc Cs>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)rror)rrlrWrrr)rPrrrrr rr) r@rrrrXrraxes_idxrUrslice1slice2rErErF test_mood_3dhs2       zTestMood.test_mood_3dcCstttjdggdSr`)rJrKr rrLrErErFtest_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)Nrr,r)rrg?r9rrrHz>rrrzalternative must be...rz ekki-ekki) rPrrr rrrrrr rK) r@rRrstat1rstat2rstat3p3rErErFtest_mood_alternatives "zTestMood.test_mood_alternativez,SciPy needs tie correction like R (gh-13730)r<zalternative, expected))r9)!}?g-?)r)g~?g4?)r)rg-?cCs.gd}gd}tj|||d}t||dS)Nrrr)r rr)r@rrrRrrTrErErFtest_mood_alternative_against_Rs z(TestMood.test_mood_alternative_against_RN)rWrXrYrrrrrrrrrxfailrrrErErErFr#s  rc@sVeZdZddZddZddZejje ddd d Z d d Z d dZ ddZ dS) TestProbplotcCsltjjddd}tj|dd\}}gd}t|t|t||tj|dd\}}gd}t||dS) Nr090rFr)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 rrprobplotrrPsort)r@rRosmosr osm_expectedrTres_fitres_fit_expectedrErErFrGs zTestProbplot.test_basicc Cstjjddd}tj|ddd\}}tj|ddd\}}tj|ddd\}}t||t||t||t||tj|ddd\}} dS)Nri@rFsparamsrrrE)r rrrr) r@rRosm1osr1osm2osr2osm3osr3rrrErErFtest_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)Nr0rrFt)rl)rrrzwrong-dist-namerc@seZdZdZddZdS)z3TestProbplot.test_dist_keyword..custom_distz6Some class that looks just enough like a distribution.cSstjj|ddS)Nr)r)r rppf)r@qrErErFrr]z7TestProbplot.test_dist_keyword..custom_dist.ppfN)rWrXrYrCrrErErErF custom_dists r)rr)rr) r rrrrrrJrKAttributeError)r@rRrrrrrrErErFtest_dist_keywords   zTestProbplot.test_dist_keyword no matplotlibr<c Cst}|dtjjdddd}tj|td\}}ttj|dd\}}tj|dtd}ttj|ddd}tt |t |koXt |koXt |koXd knt ||t ||t ||t ||t}|d} tj|d| dtdS) NrerlrtrplotF)rrr) pltfigure add_subplotr rrrcloserrr) r@figrRres1fitres1res2fitres2res3res4axrErErFtest_plot_kwargs$ @      zTestProbplot.test_plot_kwargcCrr)rJrKr rrLrErErFtest_probplot_bad_argsz#TestProbplot.test_probplot_bad_argscCsZttjgddtgtgfttjgddtgtgftjtjdffdS)NFrTr)r r rrPrrrLrErErF test_emptys zTestProbplot.test_emptycCsjtjdd%ttjdgddtdgtdgftjtjdffWddS1s.wYdS)Nr)invalidrTrr)rPrr r rrrrLrErErFtest_array_of_size_ones "z#TestProbplot.test_array_of_size_oneN)rWrXrYrGrrrrrFhave_matplotlibrrrrrErErErFrs  rc@seZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ e jdgdgdgdgddZddZdS) TestWilcoxoncCshtttjdgddgtttjddgddgdtttjddgddgddtttjdgddddS)Nrrdummyrrxyzmode)rJrKr wilcoxonrLrErErFtest_wilcoxon_bad_args z"TestWilcoxon.test_wilcoxon_bad_argcCsRtd}tttj||dddtttj||dddttj||dddddS)Nr0wilcoxapproxrprattzsplit)g@Z@r)rPrQrJrKr rr rrErErFtest_zero_diffs zTestWilcoxon.test_zero_diffcCsdgd}gd}t}|jtddtj||ddd}Wdn1s&wYt|ddS) Nr)rrrlroSample size too smallrrr zero_methodr)rg |N?)r rrr rr)r@rRrrrTrErErF test_pratt$szTestWilcoxon.test_prattcCs<gd}tj|ddd}tj|ddd}tj|ddd}dS)N) rrrlrrrlrrrrrrrrrr)r r)r@arrrrErErFtest_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) rrnrTrOr-rnrorrrocSsg|] \}}|g|qSrErE)rhurrErErFri:sz7TestWilcoxon.test_accuracy_wilcoxon..rrrig.i?rigFYv]Uj?riGg%^Ez? rOrrrnyrhrp wr rsrr[F correctionrr!g^҂TestKstat.test_moments_normal_distribution..r)gHQ,?glZ)rI?g88}?gcJN?rr2r)momentrrlrg{Gz?{Gz?r)rPrrrrr r5)r@momentsrm1m2m3rEr4rF test_moments_normal_distributions  "z*TestKstat.test_moments_normal_distributioncCrHrI)rJrKr r3rLrErErFrMrNzTestKstat.test_empty_inputcC*td}tj|d<tt|tjdSNr4r)rPrQrr r r3rarErErFr  zTestKstat.test_nan_inputcCs*td}dD] }tttj||dqdS)Nr)rg/$@r4)rPrQrJrKr r3r@rArrErErFtest_kstat_bad_args zTestKstat.test_kstat_bad_argN)rWrXrYr;rMrrArErErErFr2s  r2c@r&) TestKstatVarcCrHrIrJrKr kstatvarrLrErErFrMrNzTestKstatVar.test_empty_inputcCr<r=r>rarErErFrr?zTestKstatVar.test_nan_inputcCs dg}d}tttj||ddS)Nrrr4rCr@rErErFrbszTestKstatVar.test_bad_argN)rWrXrYrMrrbrErErErFrBs rBc@sNeZdZddZddZddZejje ddd d Z d d Z d dZ dS) TestPpccPlotcCtjjddddd|_dSNrorrr loggammarrRrLrErErF setup_methodrzTestPpccPlot.setup_methodcCHd}tj|jdd|d\}}gd}t|tjdd|dt||dS)NrorN)g9% ?gxM _?g"; 9?g®XZ?g?num)r ppcc_plotrRrrPr)r@rOsvalsppcc ppcc_expectedrErErFrG  zTestPpccPlot.test_basiccCstj|jdddd\}}tj|jddtjd\}}t||ddt||ddt|jdd\}}t||ddt||dddS)NrMr tukeylambdarg#B ;r2)r rRrRrWr)r@svals1ppcc1svals2ppcc2svals3ppcc3rErErF test_dists  zTestPpccPlot.test_distrr<cCZt}|d}tj|jddtd|||d}tj|jdd|dtdSNreir0r)rrrr rRrRdelaxesrr@rrrErErFr    zTestPpccPlot.test_plot_kwargcCs2tttj|jddtttjgddddddS)Nrrrkrr)rJrKr rRrRrLrErErFtest_invalid_inputs*s z TestPpccPlot.test_invalid_inputscCs@tgdd\}}t|tjddddt|tjdtddS)Nrrr^rPr)r rRrrPrrfloat)r@rSrTrErErFr2szTestPpccPlot.test_emptyN) rWrXrYrKrGr^rrrFrrrdrrErErErFrEs  rEc@rZ) TestPpccMaxcCsdg}tttj|dddSr)rJrKr ppcc_maxrarErErFtest_ppcc_max_bad_arg<sz!TestPpccMax.test_ppcc_max_bad_argcCs2tjjddddddd}tt|dd d dS) Nffffffrr'r@{r,r9)r rWrrrgrrErErFtest_ppcc_max_basicAs zTestPpccMax.test_ppcc_max_basiccCsptjjddddddd}tj|dd }tj|tjd }t|d d d t|d d d t|}t|d d d dS) NrirrrjrkrrlrWrrmror9)r rWrrgr)r@rRmax1max2max3rErErFr^Fs zTestPpccMax.test_distcCs`tjjddddddd}tttj|dd ttj|d d d d d ttj|dd d d d dS)Nrirrrjrkrrl)rrr)brack)rrrmr,r9)rr)r rWrrJrKrgrrrErErF test_brackTs zTestPpccMax.test_brackN)rWrXrYrhrnr^rsrErErErFrf;s  rfc@r) TestBoxcox_llfcCsTtjjdddd}d}t||}|j dtt|d}t ||dS)Nrjr1rrrrr r) r rr boxcox_llfrrPlogr_rDr)r@rRlmbdallf llf_expectedrErErFrGbs  $zTestBoxcox_llf.test_basiccCDtjjdddd}d}t||}t|t|}t||dddSNrrrurvrrr2)r rrrwlistrr@rRryrzllf2rErErFtest_array_likei  zTestBoxcox_llf.test_array_likecCPtjjdddd}d}t||}t|t||gj}t||g|dddSr})r rrrwrPvstackrrrrErErF test_2d_inputps  zTestBoxcox_llf.test_2d_inputcCtttdgdSr`)rrPisnanr rwrLrErErFrzzTestBoxcox_llf.test_emptycCs"gd}td|}t|ddS)N)gh@ m@rgx@ig#:x1)r rwr)r@rArzrErErF test_gh_6873}s zTestBoxcox_llf.test_gh_6873N)rWrXrYrGrrrrrErErErFrt`s  rt(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) Nrorrrrryrr)r rJrboxcoxrrPrxr~)r@rRxtrErErFtest_fixed_lmbdaszTestBoxcox.test_fixed_lmbdacCsLd}tjjdddd}||d| }t|\}}t|d|dd dS) Ng@riPi)rrrrrrr9)r rrrr)r@ryrRx_invrmaxlogrErErFtest_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)Nrrorrr,)alphagx@gv[%@rgxˉEk?g8Bhk @r,rHrOrg\?gx̡&@rg7(U@gp: sY@)rPrrr rJrrr)r@rngrRrr]rErErF test_alphas zTestBoxcox.test_alphacCsRtddg}tttj|tttjtdgtttjtdgdggdS)Nrrr)rPrrJrKr rrrErErFtest_boxcox_bad_args zTestBoxcox.test_boxcox_bad_argcCttgjdkdSN)r)rr rrrLrErErFrrzTestBoxcox.test_emptycCs tt\}}t|ddddS)NgsHjdrh㈵>r2r r _boxcox_datar)r@rlamrErErFrszTestBoxcox.test_gh_6873bounds)rr)rr)rr*csHfdd}tjtd|d\}}d|krdks"JJdS)Nctj|ddSNbounded)rrr minimize_scalarfunrrErF optimizerzBTestBoxcox.test_bounded_optimizer_within_bounds..optimizerryrrr)r rr)r@rrrryrErrF$test_bounded_optimizer_within_boundss (z/TestBoxcox.test_bounded_optimizer_within_boundscsjtjtdd\}}|d|dfddifdd}tjtd|d\}}||ks,Jt|d dS) Nrr!rxatolrcstj|ddS)Nr)rroptionsrrrrrErFrszPTestBoxcox.test_bounded_optimizer_against_unbounded_optimizer..optimizerrrr)r@rryr lmbda_boundedrErrF2test_bounded_optimizer_against_unbounded_optimizers z=TestBoxcox.test_bounded_optimizer_against_unbounded_optimizerr)str)rrr!cCsDtjtddtjtd|dWddS1swYdS)Nz`optimizer` must be a callablerrrr rKr rrr@rrErErF$test_bad_optimizer_type_raises_errors"z/TestBoxcox.test_bad_optimizer_type_raises_errorcCsPdd}d}tjt|dtjtd|dWddS1s!wYdS)NcSsdSr`rErrErErFr#szCTestBoxcox.test_bad_optimizer_value_raises_error..optimizerz;`optimizer` must return an object containing the optimal...rrr)r@rrrErErF%test_bad_optimizer_value_raises_errors "z0TestBoxcox.test_bad_optimizer_value_raises_errorN)rWrXrYrrrrrrrrrrrrrrErErErFrs      rc@sheZdZddZddZddZddZej d gd ej d gd d dZ ddZ ddZ dS)TestBoxcoxNormmaxcCrFNrorrrrIrLrErErFrK,rzTestBoxcoxNormmax.setup_methodcCt|j}t|ddddS)Nm?rxr2r boxcox_normmaxrRrr@rrErErF test_pearsonr/ zTestBoxcoxNormmax.test_pearsonrcCs<tj|jdd}t|dddt|j\}}t||dS)Nmler .!?rxr2)r rrRrr)r@rr maxlog_boxcoxrErErFtest_mle3szTestBoxcoxNormmax.test_mlecCs&tj|jdd}t|ddgdddS)Nrrrrrxr2r)r@ maxlog_allrErErFtest_all;szTestBoxcoxNormmax.test_allr)rpearsonrrrrcsNfdd}tj|j||d}td|ksJt|dks%JdS)NcrrrrrrErFrCrzITestBoxcoxNormmax.test_bounded_optimizer_within_bounds..optimizer)rrrr)r rrRrPr)r@rrrrrErrFr?s  z6TestBoxcoxNormmax.test_bounded_optimizer_within_boundscsvt|j}t|d}t|d|ddGdddfdd}tj|j|d}||ks3Jt||d dS) Nror6ic@s eZdZdS)z?TestBoxcoxNormmax.test_user_defined_optimizer..MyResultN)rWrXrYrErErErFMyResultRsrcs6g}D] }|||q}t||_|SrI)appendrPargminrR)robjsryrTr lmbda_rangerErFrUs z@TestBoxcoxNormmax.test_user_defined_optimizer..optimizer)rr)r rrRrProundrr)r@ry lmbda_roundedrlmbda2rErrFtest_user_defined_optimizerLs   z-TestBoxcoxNormmax.test_user_defined_optimizercCs^tj}tj|jd|dtjtddtj|jd|dWddS1s(wYdS)N)rrrz,`brack` must be None if `optimizer` is givenr)gr )r rr rrRrr rKrrErErF2test_user_defined_optimizer_and_brack_raises_errorbs "zDTestBoxcoxNormmax.test_user_defined_optimizer_and_brack_raises_errorN) rWrXrYrKrrrrrrrrrrErErErFr+s  rc@sFeZdZddZddZejje ddddZ d d Z d d Z d S)TestBoxcoxNormplotcCrFrGrIrLrErErFrKtrzTestBoxcoxNormplot.setup_methodcCrL)NrorMrrN)g!X4}?g΁i?g- 15?gW$?g\ܱ{?rP)r boxcox_normplotrRrrPr)r@rOlmbdasrTrUrErErFrGwrVzTestBoxcoxNormplot.test_basicrr<cCr_r`)rrrr rrRrarrbrErErFrrcz"TestBoxcoxNormplot.test_plot_kwargcCs.tttj|jddtttjddgdddS)Nrrr)rJrKr rrRrLrErErFrdsz&TestBoxcoxNormplot.test_invalid_inputscCsttgddjdkdSr.)rr rrrLrErErFrrzTestBoxcoxNormplot.test_emptyN) rWrXrYrKrGrrrFrrrdrrErErErFrss  rc@r&)TestYeojohnson_llfcCr|)Nrrrurvrrr2)r rryeojohnson_llfr~rrrErErFrrz"TestYeojohnson_llf.test_array_likecCrr})r rrrrPrrrrrErErFrs  z TestYeojohnson_llf.test_2d_inputcCrr`)rrPrr rrLrErErFrrzTestYeojohnson_llf.test_emptyN)rWrXrYrrrrErErErFrs rc@seZdZddZejdgdddZddZd d Z ejd e j e j gd d Z ejd e je je je jgddZdS)TestYeojohnsoncCstjd}tjjdd|dd}t|dksJtj|dd}t||tj|dd}t|dd|dtj|dd}t|t |dtj|dd}t||tjjdd|dd}t|dksiJtj|d d}t|t | d tj|dd}t||tj|d d}t|d| ddtjjdd|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) Nrrorrrrrrrrl) rPrrr rJrr yeojohnsonrrx)r@rrRrposnegrErErFrsL     $zTestYeojohnson.test_fixed_lmbdary)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)rPrrrabsspacingexppower)rRryrrrErErF_inverse_transforms$  z:TestYeojohnson.test_lmbda_None.._inverse_transformi Nrkrr)rrrr6rrr9) rPrrnormalr rrrlinalgrrBrD)r@ryr n_samplesrRrrrrErErFrs  zTestYeojohnson.test_lmbda_NonecCrr)rr rrrLrErErFrrzTestYeojohnson.test_emptycCsDtjjdddd}t|\}}tt|\}}t||dddS)Nrrrurvrr2)r rrrr~r)r@rRxt1rxt2rErErFrszTestYeojohnson.test_array_likercCsPtjd|d}d}tjt|dt|WddS1s!wYdS)Nr)rz>Yeo-Johnson transformation is not defined for complex numbers.r)rPrQrr rKr r)r@rrRerr_msgrErErFtest_input_dtype_complex s  "z'TestYeojohnson.test_input_dtype_complexcCsZtjd|d}tjdtjd}t|\}}t|\}}t||ddt||dddS)Nr-rrr2)rPrQfloat64r rr)r@rx_intx_floatxt_int lmbda_intxt_float lmbda_floatrErErFtest_input_dtype_integers z'TestYeojohnson.test_input_dtype_integerN)rWrXrYrrrrrrrrP complex64 complex128rint8uint8int16int32rrErErErFrs/ $ rc@r&)TestYeojohnsonNormmaxcCrFrrIrLrErErFrKrz"TestYeojohnsonNormmax.setup_methodcCr)Ng?rxr2)r yeojohnson_normmaxrRrrrErErFrrzTestYeojohnsonNormmax.test_mlecCs*gd}t|}tj|dddsJdS)N)gffffff@g rr gffffff?g333333@rgffffff@g?g @rrg"@rggzG?rr)r rrPallclose)r@rRryrErErFtest_darwin_example#s z)TestYeojohnsonNormmax.test_darwin_exampleN)rWrXrYrKrrrErErErFrs rc@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|dddSNcrorrtr^hrqrr2rPrrr@ test_funcrrRrErErFtest_circfuncs-szTestCircFuncs.test_circfuncscCs~tgd}|}tj|dd}t||dd|}tj|dd}t||dd|}tj |dd}t||dddS)N)r0rBr'r#g4@g3333333@rrrr2r) rPrrBr circmeanrrCcircvarrDcircstd)r@rRM1M2V1V2S1S2rErErFtest_circfuncs_small5sz"TestCircFuncs.test_circfuncs_smallztest_func, numpy_funccCs<tdgddgd}||}||}t||dddS)Ng˛ɦv9?rguv9?rg:0yU>rr)r@r numpy_funcrRcircstatrrErErFtest_circfuncs_closeCsz"TestCircFuncs.test_circfuncs_closectgdgdgdgtjdd}tjdd}t||ddtjddd }fd d tjd D}t||ddtjdd d }fd d tjdD}t||dddS)Nr_r,rn`r*]ier*r-ifrnidrrr1r2rrqrXcg|] }tj|ddqSrrr rrhrUrRrErFriWz4TestCircFuncs.test_circmean_axis..rc&g|]}tjdd|fddqSNrrrrrrErFri[&)rPrr rravelrrr)r@rrrErrFtest_circmean_axisNs z TestCircFuncs.test_circmean_axiscr)Nrrrrrdy=r2rrcrrr rrrrErFrihrz3TestCircFuncs.test_circvar_axis..rcrrr#rrrErFrilr)rPrr rr rrr)r@rr rErrFtest_circvar_axis^ zTestCircFuncs.test_circvar_axiscr)Nrrrrrr"r2rrcrrr rrrrErFriyrz3TestCircFuncs.test_circstd_axis..rcrrr&rrrErFri}r)rPrr rr rrr)r@r r rErrFtest_circstd_axisor%zTestCircFuncs.test_circstd_axiscCs"gd}t||dd|dddSr)rrrErErFtest_circfuncs_array_likesz'TestCircFuncs.test_circfuncs_array_likercCstt|gdSrI)rrPr)r@rrErErFrrzTestCircFuncs.test_emptycCs.ddddddtjg}tt||dddS) NrrorrtrrrrrPrrrr@rrRrErErFtest_nan_propagatesz TestCircFuncs.test_nan_propagateg5:v@gf Pb?r.gڴV0@gLԿAB@g#q@gauy+@c Cstgdddddddtjgdtjtjtjtjtjtjgg}|D]-}||d |d }|dur8tt|q#t|d ||d d tt|ddq#dS)N)rrorrtrrrrr,rnrr*rrrrrrr2)rPrrkeysrrrrr@rrrRrXoutrErErFtest_nan_propagate_arrays  z&TestCircFuncs.test_nan_propagate_arraygrQvv@)gv@rrg8v@rgu@g5@v?g2>Bjv@gtUuYK@)gR@zw<5?r0g~/ϔ(@ K ?r1gsQP@gJPm$@)gj@竃?r2g)h; @{?r3gun2@g+fa @c Cstddddddtjgddd d d d tjgtjtjtjtjtjtjtjgg}|D].}||d d|d}|durBt|||ddq*t|dd||ddtt|dq*dS)Nrrorrtrrrr,rnrr*rromit)rq nan_policyrXrr2r)rPrrr,rrrr-rErErFtest_nan_omit_arrays z!TestCircFuncs.test_nan_omit_arraycCs0ddddddtjg}t||ddd |d d dS) Nrrorrtrrrr4rqr5rr2)rPrrrrErErF test_nan_omits zTestCircFuncs.test_nan_omitcCs2tjtjtjtjtjg}tt||dddS)Nr4r5r)r*rErErFtest_nan_omit_allszTestCircFuncs.test_nan_omit_allcCsjttjtjtjtjtjgtjtjtjtjtjgg}||ddd}tt|tt|dkdS)Nr4r)r5rXr)rPrrrrrr)r@rrRr.rErErFtest_nan_omit_all_axiss z$TestCircFuncs.test_nan_omit_all_axisrRrrorrtrrrr,rnrr*rcCtt||ddddS)Nrraiser7rJrKr*rErErFtest_nan_raisezTestCircFuncs.test_nan_raisecCr<)Nrfoobarr7r>r*rErErFtest_bad_nan_policyr@z!TestCircFuncs.test_bad_nan_policycCs$d}|}t|}t||dddS)Nrrr2)r rr)r@rRrrrErErFtest_circmean_scalars z"TestCircFuncs.test_circmean_scalarcCs@ttdddtjtj }t|tjkt|tj kdS)Nrrr!)r rrPrQpir)r@mrErErFtest_circmean_rangesz!TestCircFuncs.test_circmean_rangecCsZtjddgdd}ttj|dddttj|dddd d ttj|ddd d d dS) Nrrrrrg@e@g.}}VW{@rr2g)#p_4@)rPrr r rrrrrrErErFtest_circfuncs_unit8sz"TestCircFuncs.test_circfuncs_unit8N)#rWrXrYrrrr rrrrr rPrBrCrDrr!r$r'r(rr+rr/rr6r8r:r;r?rBrCrFrHrErErErFr,s                           rc@sdeZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ dS)TestMedianTestcCstttjgddSr0rJrKr median_testrLrErErFtest_bad_n_samples r3z!TestMedianTest.test_bad_n_samplescCstttjggddSr0rJrLrErErFr rz TestMedianTest.test_empty_samplecCs&tttjgdgdddgdddS)N)rrrr)rrrrrrtiesrJrLrErErFtest_empty_when_ties_ignored s z+TestMedianTest.test_empty_when_ties_ignoredcCs8tttjgdgdtttjgdgddddS)N)rrraboverMrJrLrErErFtest_empty_contingency_row s z)TestMedianTest.test_empty_contingency_rowcC tttjgdddgdddS)NrkrnrorrMrJrLrErErF test_bad_ties"  zTestMedianTest.test_bad_tiescCrR)NrkrnrorAr9rJrLrErErFrB& s z"TestMedianTest.test_bad_nan_policycCrR)Nrkrnror)r)rJr#r rKrLrErErFr$) rTzTestMedianTest.test_bad_keywordcCs\gd}gd}t||\}}}}t|dt|ddgddggt|dt|ddS)Nrkr rrr)r rKr r)r@rRrrr medtblrErErF test_simple- s  zTestMedianTest.test_simplecCsgd}ddg}gd}t|||\}}}}t|dt|gdgdgtj|||dd\}}}}t|dt|gdgd gtj|||d d\}}}}t|dt|gd gd gdS) Nrror))r,r-r*)rrrl)rnrrrrM)rnrrrP)rrrl)r rKr )r@rRrzrr rErVrErErFtest_ties_options< s   z TestMedianTest.test_ties_optionscCsddtjg}gd}tj||dd}tj||dd\}}}}t|tjtjtjdft|dt|dt|d t|td dgddggtttj||d ddS) Nrr)rnror) propagater9r4g?gإvao?rrr=) rPrr rKr rrrJrK)r@rRrmt1rr rErrErErFtest_nan_policy_optionsP 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) Nrm)rrnr)r-rnrrr)lambda_F)r)r rKr chi2_contingencyr) r@rRrrr rErVexp_statexp_pdoferErErFrG] s(        zTestMedianTest.test_basicN)rWrXrYrLrrOrQrSrBr$rWrYr\rGrErErErFrI s  rI)Krpr?numpyrP numpy.randomr numpy.testingrrrrrrr r rr rJscipyr r scipy.stats._morestatsr common_testsr _hypotestsrscipy.stats._binomtestr matplotlibrcParamsmatplotlib.pyplotpyplotrr Exceptionrrrrrrrrrrr'r[rurrrrrr)rHrNrrrrr2rBrErfrtrrrrrrrrrIrErErErFs (          !ZVGE,SD^G6%(2qH#qY