L i/<dZddlZddlZddlZddlmZddlmZddlm cm Z ddl Z ddl Z ddlmZmZmZmZmZmZmZmZmZddlZddlmZddl mZmZmZmZddl Zddl m!Z!ddl"m!cm#Z#ddl$m!cm%Z&ddl'm(Z(dd l)m*Z*dd l+m,Z,dd l-m.Z.d d l/m0Z0ddl1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7ddl8m9Z9m:Z:m;Z;mZ>m?Z?ddl@mAZAddlBmCZCddlDmEZEmFZFmGZGmHZHmIZImJZJmKZKmLZLmMZMddlNmOZOmPZPddlQmRcmSZTe!gZUejjZW egdeXZYegdeXZZegdeXZ[egdeXZ\egdeXZ]egdeXZ^egdeXZ_GddZ`GddZaeMe!jGdd ZcGd!d"ZdGd#d$ZeGd%d&Zfd'Zgd(Zhd)Ziejjd*ZkGd+d,Zld-Zmd.Znd/Zod0ZpGd1d2ZqGd3d4Zrd5Zsd6Ztd7ZuGd8d9ZvGd:d;ZweMe!jGd<d=ZyeMe!jGd>d?Z{eMe!jGd@dAZ}eMe!jGdBdCZGdDdEZdFZGdGdHZeMe!jGdIdJZGdKdLZeMe!j GdMdNeZeMe!jGdOdPeZe jjdQZeMe!jGdRdSZGdTdUZedVgdWZegdXddddYdZdYej"d[zd\ej"d]zzzdZd^ej"d_zd^ej"d`zzzdaWegdXgdbdddcdZdYej"d_zd^ej"d[zzd^ej"ddzzzdZdZej"d[zdeej"d_zzdZej"dfzzzdgWegdXd^dddYdZdYej"d[zd\ej"d]zzzdZd^ej"d_zd^ej"d`zzzdaWegdhgdhddddddWgZegdddddddWeej6ggggj&dddgdigdigdigdiWeej6ggggdddggggWgZeMe!j*GdjdkZeMe!j.GdldmZdnZGdodpZGdqdrZGdsdtZduZdvZdwZejjAdxejBdydzfejBd{d|fgd}Zd~ZejjAdgddZejjAde!jej8ddffe!jJej8dej8dffgdZddddZeMe!jPe!jRdZdZdZejj[dGddZejj[dGddZGddZeMe!jPGddZdZeMe!jPGddZeMe!jRGddZejjddejj[ddZdZejjnejj[dejj[deMe!jdZeWdddZeMe!jejjdddZeMe!jvGddZGddZeMe!j|GddeZeMe!jGddeZeMe!jGddeZGddZeMe!jGddZGddZdZdZddddddZdZdZdZeMe!jGddZeMe!jGd„dëZeMe!jGdĄdūZeMe!jGdƄdǫZdȄZdɄZdʄZGd˄d̫ZGd̈́dΫZGdτdЫZGdфdҫZGdӄdԫZeMe!jGdՄd֫ZGdׄdثZGdلdګZGdۄdܫZGd݄dޫZGd߄dZGddZGddZejjdZejdZejdZejjAde!jeffe!jee!jjffe!jeeffe!jeeffgdZGddZGddZGddZGddZdZeWdd򬦫dZy)z Test functions for stats module WRITTEN BY LOUIS LUANGKESORN FOR THE STATS MODULE BASED ON WILKINSON'S STATISTICS QUIZ https://www.stanford.edu/~clint/bench/wilk.txt Additional tests by a host of SciPy developers. N) namedtuple)product) assert_ assert_equalassert_almost_equalassert_array_almost_equalassert_array_equalassert_approx_equalassert_allclosesuppress_warningsassert_array_less)raises)arrayarangefloat32power)kolmogn)FuncData)binom)optimize)check_named_results)_broadcast_concatenateSmallSampleWarningtoo_small_nd_omittoo_small_nd_not_omittoo_small_1d_omittoo_small_1d_not_omit)_permutation_distribution_t _chk_asarray_momentLinregressResult_xp_mean_xp_var _SimpleChi2) AxisError)skip_xp_invalid_arg) array_namespace eager_warns is_lazy_arrayis_numpyis_torchxp_default_dtypexp_sizeSCIPY_ARRAY_APImake_xp_test_case)xp_assert_closexp_assert_equal) r ) rrrrrrrrr) iiiiiiiii) g~?g< ?gEk?g?ge (?gq?g?g8C?gG?) gmBg}BgyӅBgBg@0BgyӕBg BwBgBg ^B) -q=g-=gAfc=g-=gdy=gAfc=gYLɞ=g-=gSSʣ=) ??@ @@@@@!@ceZdZejej j Zee jdZ ee je jjddZee j"dZee j&dZeddd Zee j.d Zeddd Zee j&e j.d Zedd ee j&e j.dZee j8dZy)TestTrimmedStatsc t|}|jt|}tj|dd}t ||jdtj|dd}tj|d}t |||j |jdd }tj|d }t ||j|tj|d }t ||j|d tj|d }t ||j|d tj|d d}t ||jdtj|dd }gd}t ||j|tj|ddd }gd}t ||j|tj|dddfj|jd} tj| dd }dddd|j|j|jg}t ||j|tj|dd }ddd d!|j|j|j|j|jg }t ||j|tj|dd"d }ddd d!|j|j|j|j|jg }t ||j|y)#Ndtyper3r9TT@)FFlimits inclusiverNO@r:r8axisrr)r3=)rNrTg?@)r3)'@r: TF)rNrOrT)%@rXr:rYrZr[r\T)copy)rr\r8r@rArBr5rYrVFT) r-asarrayXstatstmeanr1reshapermeanxpxatsetnan) selfxp default_dtypexyy1y2x_2dy_true x_2d_with_nans b/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/scipy/stats/tests/test_stats.py test_tmeanzTestTrimmedStats.test_tmeanRs(, JJq J . KK6< 02::c?+ [[6^ D [[4 (Bzz"))C.&1 KK4 (2774=) KK1 %2774a701 KK1 %2774a701 KKW4 82::d+, KKW1 5.2::f-. KKW A N02::f-.t RW-11"&&t1D KK gA >S#sBFFBFFBFF;2::f-. KKW1 5RRH2::f-. KKW"/a 9r2r266266266266266J2::f-.z>ignore:invalid value encountered in divide:RuntimeWarning:daskc|jtj}tj|dd}t ||jdtj|d}t ||j |d|j|jdd }tj|d }t ||j |dtj|d }t ||jd d tj|d }t ||jddtj|ddd}t |d |jdtj|ddd}t |d |jdt|d|j|jy)NrJrKrMg@rPr correctionrQrRrSrr8gv@)r:g㪪@rr6)rNrTrOr>rr7g@) rdretolistrftvarr1varrhrfullr2rm)rnrorqrrrus rx test_tvarzTestTrimmedStats.test_tvarsc JJqxxz " JJq< @2::&89: JJq &266!623zz"))C.&1 JJt$ '266$1656 JJt! $2774/0 JJt! $277445 JJtFl K!bjjo. JJtFl K!bjj):;<!bjj01rzc|jtj}tj|dd}t ||jdtj|d}t ||j |dy)NrJrKgB Ć/H@rPrr|)rdrerrftstdr1stdrnrorqrrs rx test_tstdzTestTrimmedStats.test_tstdsc JJqxxz " JJq&, /2::&89: JJq &266!623rzc J|jd}ttj||j dttj|d|j dttj|dd|j d|j |d}ttj|dd|j d dgttj|d |j gd ttj|d |j dt j|jdd j|j}ttj||j |j|j |jdd}tj|dd }t||j tjdddgy)N$@r lowerlimitF)rrO?r6r3@rrS)rr@@ @r:r5r5r5)rrTr9r[) rr2rftminrdrhrjrkrlrmnprnrorqress rx test_tminzTestTrimmedStats.test_tminsk IIcN 1 rzz"~6 13RZZ^D 1eDbjjQSnU JJq& ! 1eD B8, . 11-rzz:N/OP 140"**R.A FF299S>1 % ) )"&& 1 1 rzz"&&'9: JJryy}f -jjqq1RZZAr(:;1Cz5 2 11 20Cz5 3 12 3 3 3 2 2 3 3 3 3=A DC95DD(D9D >DD DDc |jd}ttj||j dttj|d|j dttj|dd|j d|j |d}ttj|dd|j dd gttj|d |j gd ttj|d |j dt j|jddj|j}ttj||j |jt5}|jtd |j |jdd}tj|dd }t||j dddtjgdddy#1swYyxYw)Nr"@r: upperlimitF)rrOrr@rrS)r@rLrrzAll-NaN slice encounteredrrrZ)rrTr4r8)rr2rftmaxrdrhrjrkrlrmr filterrr)rnrorqrrs rx test_tmaxzTestTrimmedStats.test_tmaxs IIcN 1 rzz"~6 13RZZ^D 1eDbjjQSnU JJq& ! 1eD B8, . 11-rzz:N/OP 140"**R.A FF299S>1 % ) )"&& 1 1 rzz"&&'9:  AC JJ~'B C 299R=&1A**Q2A6C CQ2rvv,>!? @  A A As A=IIc:ttjddtjd}tj |d<t 5}|jtdttj|ddd}tt| 5tj|d dddd }tt| 5tj|d ddddddy#1swYExYw#1swYxYw#1swYyxYw) Nr5rr7rrrrrrrrr) rrfrrrrmr rrrrrs rxtest_tmax_scalar_and_nanpolicyz/TestTrimmedStats.test_tmax_scalar_and_nanpolicyrrcn|j|jd|jdj}t |r t |n |j }ttj||jddg|ttj||jddg|ttj|d |j|jd gttj|d |jd |jgy)NrYrHr3r6rr6r5r:r7rrr4rr) rhrint16Tr*r-rIr2rfrrdrrm)rnrorq expect_dtypes rxtest_tmin_tmax_int_dtypez)TestTrimmedStats.test_tmin_tmax_int_dtypes JJryy288y4f = ? ? 0=Q/?'+QWW  1 rzz1a& z'MN 1 rzz1a& z'MN  13RZZ 5MN 13RZZRVV 5MNrzz%Only with data-dependent output dtype eager_onlyrc|jdd}ttj|dddf|ttj|dddf|y)Nll )rr2rfrrrnrorqs rx test_gh_22626zTestTrimmedStats.test_gh_22626 sJ IIeX & 1T1W:.2 1T1W:.2rzch|jtj}tj|dd}|jgd}t ||j |dt|dzz t tj|dd g tj|d y) N)r4r9rcrM)rrLrrrrr|r<r_rYrP)rdrerrftsemr1rr.)rnrorqrry_refs rx test_tsemzTestTrimmedStats.test_tsems JJqxxz " JJq= A /0266%A669LLM 1b"X6 1T8RSrzN)__name__ __module__ __qualname__rfinfofloat64 precisiondprecr0rfrgryrpytestmarkfilterwarningsrrrrrskip_xp_backendsrrrrrrrrrzrxrFrFNs BHHRZZ * *Eu{{#,/$,/\uzz" [[H2#28uzz"4#4uzz"=#=*dSU 3U 3uzz"A#A.dSU 3U 3uzz5::. O/ O.UVuzz5::.3/W3 uzz"T#TrzrFceZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZdZdZdZdZy)TestPearsonrWilkinsonE W.II.D. Compute a correlation matrix on all the variables. All the correlations, except for ZERO and MISS, should be exactly 1. ZERO and MISS should have undefined or missing correlations with the other variables. The same should go for SPEARMAN correlations, if your program has them. cbtjtt}|d}t|dyNrr)rfpearsonrrer rnrrrs rxtest_pXXzTestPearsonrWilkinson.test_pXX%s% NN1Q  aDAc"rzcbtjtt}|d}t |dyr)rfrreBIGr rs rx test_pXBIGz TestPearsonrWilkinson.test_pXBIG*s% NN1S ! aDAc"rzcbtjtt}|d}t |dyr)rfrreLITTLEr rs rx test_pXLITTLEz#TestPearsonrWilkinson.test_pXLITTLE/s% NN1V $ aDAc"rzcbtjtt}|d}t |dyr)rfrreHUGEr rs rx test_pXHUGEz!TestPearsonrWilkinson.test_pXHUGE4% NN1T " aDAc"rzcbtjtt}|d}t |dyr)rfrreTINYr rs rx test_pXTINYz!TestPearsonrWilkinson.test_pXTINY9rrzcbtjtt}|d}t |dyr)rfrreROUNDr rs rx test_pXROUNDz"TestPearsonrWilkinson.test_pXROUND>s% NN1U # aDAc"rzcbtjtt}|d}t|dyr)rfrrr rs rx test_pBIGBIGz"TestPearsonrWilkinson.test_pBIGBIGCs% NN3s # aDAc"rzcbtjtt}|d}t |dyr)rfrrrr rs rxtest_pBIGLITTLEz%TestPearsonrWilkinson.test_pBIGLITTLEHs% NN3v & aDAc"rzcbtjtt}|d}t |dyr)rfrrrr rs rx test_pBIGHUGEz#TestPearsonrWilkinson.test_pBIGHUGEM% NN3t $ aDAc"rzcbtjtt}|d}t |dyr)rfrrrr rs rx test_pBIGTINYz#TestPearsonrWilkinson.test_pBIGTINYRrrzcbtjtt}|d}t |dyr)rfrrrr rs rxtest_pBIGROUNDz$TestPearsonrWilkinson.test_pBIGROUNDWs% NN3u % aDAc"rzcbtjtt}|d}t|dyr)rfrrr rs rxtest_pLITTLELITTLEz(TestPearsonrWilkinson.test_pLITTLELITTLE\s% NN6& ) aDAc"rzcbtjtt}|d}t |dyr)rfrrrr rs rxtest_pLITTLEHUGEz&TestPearsonrWilkinson.test_pLITTLEHUGEa% NN6$ ' aDAc"rzcbtjtt}|d}t |dyr)rfrrrr rs rxtest_pLITTLETINYz&TestPearsonrWilkinson.test_pLITTLETINYfrrzcbtjtt}|d}t |dyr)rfrrrr rs rxtest_pLITTLEROUNDz'TestPearsonrWilkinson.test_pLITTLEROUNDks% NN6% ( aDAc"rzcbtjtt}|d}t|dyr)rfrrr rs rxtest_pHUGEHUGEz$TestPearsonrWilkinson.test_pHUGEHUGEp% NN4 % aDAc"rzcbtjtt}|d}t |dyr)rfrrrr rs rxtest_pHUGETINYz$TestPearsonrWilkinson.test_pHUGETINYur rzcbtjtt}|d}t |dyr)rfrrrr rs rxtest_pHUGEROUNDz%TestPearsonrWilkinson.test_pHUGEROUNDz% NN4 & aDAc"rzcbtjtt}|d}t|dyr)rfrrr rs rxtest_pTINYTINYz$TestPearsonrWilkinson.test_pTINYTINYr rzcbtjtt}|d}t |dyr)rfrrrr rs rxtest_pTINYROUNDz%TestPearsonrWilkinson.test_pTINYROUNDrrzcbtjtt}|d}t|dyr)rfrrr rs rxtest_pROUNDROUNDz&TestPearsonrWilkinson.test_pROUNDROUNDs% NN5 ' aDAc"rzN)rrr__doc__rrrrrrrrrrrrrrrr r rrrrrrzrxrrsp# # # # # # # # # # # # # # # # # # # # #rzrceZdZdZdZdZdZdZejjdddgd Z d Z d Z d Zd ZdZejj#ddZedejjdgddZdZdZdZejj1dejj3dejjddejjdddZejj6ejjddd Zejjd!d"d#gd$Zd%Zd&Zejj#dd'Z d(Z!ejjd!gd)ejjdgdd*Z"y+), TestPearsonrctjtt}d}t||t |j |j yN correlationpvalue)rfrrerrr statisticrnr attributess rxtest_pearsonr_result_attributesz,TestPearsonr.test_pearsonr_result_attributess3nnQ". C,S__cmm4rzc |jd}tj||\}}t||j ddt||j dt j dt jdzy)NrrV瞯<atolrr3rrfrr1rdrsqrtspacingrnroarprobs rxtest_r_almost_exactly_pos1z'TestPearsonr.test_r_almost_exactly_pos1sc IIcN..A&42::c?7 bjjoBGGAbjjo>!Q'DAq Arzz"&&1 2 Arzz"&&1 2  3 3 3 B B;;CrIrrc tt|}t||}|jdddtjd|zg|}|jddddtjd|zzg|}d}t j t j|5t j||dddy#1swYyxYw)Nr3rHr4r7z/An input array is nearly constant; the computedr) getattrrrdr(rr>rfNearConstantInputWarningr)rnrorInpdtyperqrrrs rxtest_near_constant_inputz%TestPearsonr.test_near_constant_inputs"e$E" JJ1a"**Qg">>?uJ M JJ1a!BJJq$@"@@AJ O? \\%88 D ! NN1a  ! ! !s #CC cN|jgd|j}|jgd|j}tj||\}}t ||jd|jt ||jd|jy)N)g y)r?gxd{s?gnp?g$9"o?ge =:k?rH)goȟg` ,gzg2l?g>??' ge;E?g3R?rdrrfrr1rnrorqrrrrAs rxtest_very_small_input_valuesz)TestPearsonr.test_very_small_input_valuess JJVZZ  ) JJNZZ  )~~a#1 2::&8 :KL2::&8 :KLrzcVd|jgd|jz}d|jd|jz}tj||\}}t ||jd|jt ||jd|jy)NgY) kR)rrrrrrrrHr8r2g͗~-?)rdrrrfrr1rKs rxtest_very_large_input_valuesz)TestPearsonr.test_very_large_input_valuess 1D D 1BJJ/ /~~a#1 2::&8 :KL2::&:"**:MNrzcN|jgd|j}|jgd|j}tj||\}}t ||jd|jt ||jd|jy)N)g7 igNig_b4igZbirH)g Dig<'(½igmQiZbtig*D{?gݣ ?rJrKs rx!test_extremely_large_input_valuesz.TestPearsonr.test_extremely_large_input_valuess JJ9J L JJ;2::J N~~a#1 2::&7rzz:JK2::&7rzz:JKrzc0|jddg}|jddg}tj||}|\}}|jd}t||t|||j \}} t|| t| |y)NrrrrLrdrfrr2confidence_interval rnrorqrrrrrAonelowhighs rxtest_length_two_pos1z!TestPearsonr.test_length_two_pos1s JJBx  JJBx nnQ"1jjn33++- TcT"c"rzc2|jddg}|jddg}tj||}|\}}|jd}t|| t|||j \}} t|| t| |y)NrrrrLrSrUs rxtest_length_two_neg1z!TestPearsonr.test_length_two_neg1s JJBx  JJBx nnQ"1jjnC4 3++- TcT"c"rz*ignore:invalid value encountered in dividecd}tjtj|5|j ddg}|j ddg}tj ||\}}t ||j |jt ||j |jdddy#1swYyxYw)Nr7rr9r;r<r=r@s rxtest_length_two_constant_inputz+TestPearsonr.test_length_two_constant_inputs + \\%44C @ 3 E5>*A E5>*A>>!Q'DAq Arzz"&&1 2 Arzz"&&1 2  3 3 3rCTrz$alternative, pval, rlow, rhigh, sign)) two-sided?gmgƎ?r)lessK˅?r_g? ?r)greater.?gդζrr)r`ragfǎgl?r_)rbrer.gդζ?r_)rdrcg? rr_cgd}tjgd|z}tj|||} t | j d|zdt | j |d| j} t | ||fdy)Nrr3r4r5)rrr<r alternativeg ?r;rtolư>)rrrfrr rrrT) rnripvalrlowrhighsignrorqrrresultcis rxtest_basic_examplezTestPearsonr.test_basic_example$st  HH^ $t +1+>((*r$)rdrfrr1rTr2rUs rx#test_length3_r_exactly_negative_onez0TestPearsonr.test_length3_r_exactly_negative_one=s JJ| $ JJ 'nnQ"1jjoC4 1S5t,++- TcT"c"rzcgd}dg}d}tjt|5tj||dddgd}ddg}d}tjt|5tj||ddddg}dg}d }tjt|5tj||dddgd }gd }d }tjt|5tj||dddd }tjt|5tjddgd dgddddtjddgd dg}tjt|5|j ddddy#1swYgxYw#1swY0xYw#1swYxYw#1swYxYw#1swYxYw#1swYyxYw)Nrr3r4r5z3`x` and `y` must have the same length along `axis`.rr6"`x` and `y` must be broadcastable.rr3(`x` and `y` must have length at least 2.)yyz+This function does not support complex dataz"`method` must be an instance of...r4 asymptoticmethodexact)rrrrfrrT)rnrqrrmessagers rxtest_input_validationz"TestPearsonr.test_input_validationKs  CG ]]:W 5 ! NN1a  !  F6 ]]:W 5 ! NN1a  !C C< ]]:W 5 ! NN1a  !  ? ]]:W 5 ! NN1a  !7 ]]:W 5 @ NNAq6Aq6, ? @nnaVaV, ]]:W 5 4  # #7 # 3 4 45 ! !  ! !  ! !  ! ! @ @ 4 4sGF!)F.+F;/G+GG!F+.F8;GGGG(rYz-Monte Carlo method needs > a few kB of memoryrirbrdr` method_name permutation monte_carlo monte_carlo2c"tjjd}|dk(rdnd}|j|}|j|}t j |t j |jfdzt j d d }||}t j||||d } t j|||d } t| j| jdt| j| jdd|dk(rqt j d }t j||||d } t| j| jt| j| jyy)Nlo#[mrr3d)r3sizerngr3rvsirr_)rirrTrirTr#rj{Gz?MbP?rkr%r) rrandom default_rngnormalrfPermutationMethodMonteCarloMethodrr rrr) rnrrirrrqrrmethodsrrrefres2s rxtest_resampling_pvaluez#TestPearsonr.test_resampling_pvaluems< ii##$56&-7xY JJDJ ! JJDJ !"'"9"9c"B"'"8"8cjj]1_"M#(#9#9d#CE%nnQ{6PRSnnQ{D s}}5A CJJTE . (++5F>>!QKUWXD  7 cjj 1 )rzc,tjjd}|jd}|jd}t j |||d}tjjd}t j |}|j|}|j}t||d tjjd}t j | }|j|} t| |y) Nl_YFrrr_rl|#(AkP random_statergQ?r$r) rrrrrfrBootstrapMethodrTr ) rnrirrqrrrrres_ciref_cires_ci2s rxtest_bootstrap_cizTestPearsonr.test_bootstrap_cisii##$45 JJHJ % JJHJ %nnQ{Dii##$67&&C8(((7((*V4ii##$67&&3/)))8(rzrTrrctjjd}d}|jd|z\}}t j |||}|j }|dk(r|j|j}}t|jdD]}t j ||||} | j } t|j|| jt|j|| jt|j|| jt|j|| jy)NI.v#r:rYr3rrSr)rrrrrfrrTrrangeshaper rrrWrX) rnrTrrrqrrrrrires_ici_is rx test_axis01zTestPearsonr.test_axis01sii##K0zzte|z,1nnQ-  $ $ & 1933qAqwwqz" 3ANN1Q41.E,,.D CMM!,eoo > CJJqM5<< 8 BFF1Itxx 0 BGGAJ 2  3rzctjjd}d}|jd|z\}}t j ||d}|j }t j |j|j}|j }t|j|jt|j|jt||y)NrrrrrS) rrrrrfrrTravelr rr) rnrrrqrrrrrrci_refs rxtest_axis_NonezTestPearsonr.test_axis_Nonesii##K0zzte|z,1nnQ-  $ $ &nnQWWY 2((* s}}5 CJJ/F#rzc|jdx}}d}tjt|5t j ||ddddd}tjt|5t j ||jdddddd }tjt|5t j |jd|jdddddd }tjt|5t j ||jd ddddd }t |s`|jd }tjt|5t j ||t jdddyy#1swYnxYw#1swY.xYw#1swYxYw#1swYxYw#1swYyxYw)Nrz`axis` must be an integer.rr=rSz2`x` and `y` must have the same length along `axis`r3rrrr~r4r6z:`method` must be `None` if arguments are not NumPy arrays.rYr) onesrrrrfrr+rr)rnrorqrrrs rxtest_nd_input_validationz%TestPearsonr.test_nd_input_validationsA. ]]:W 5 + NN1ac * +G ]]:W 5 7 NN1bggfoA 6 7= ]]:W 5 E NN2776?BGGFO! D E7 ]]:W 5 7 NN1bggfoA 6 7O| " Az9 Gq!E,C,C,EF G G! + + 7 7 E E 7 7  G Gs;F6/(G<7G(G+G(6GG GG%(G1ctjjd}|jd|jd}}d}tjt j |5d\|d<|d<|j||j|}}t j||d }|j} |j|j|jg|j } t|jd d | t|jd d | t| jd d | t| j d d | |j#|j%|jd dsJ|j#|j%|jd dsJ|j#|j%| jd dsJ|j#|j%| j d dsJ dddd }tjt j&|5d\|d<|d<|j||j|}}t j||d ddd|jdd gdd gd dgd dgg}|jdd gd dgdd gd dgg}|j)d} t j||d }|j} t+|j|jgdt+|j| t+| j| t+| j | y#1swY]xYw#1swYxYw)NlU!Jr5r6r7rrr3)r.)r.rrSrHrr3z!An input array is nearly constant)g?g@rrrrrr5r_)rr_r_r)rrrrr>rfr?rdrrTrmrr2rrrWrXallisfiniterFrr1) rnrorx0y0rrqrrrrrnansrs rxtest_nd_special_casesz"TestPearsonr.test_nd_special_casessii##N3F#SZZ%7B. \\%44G D 4%) "BvJ6 ::b>2::b>qA..AA.C((*B::rvvrvv.bjj:AD CMM!A. 5 CJJqOT 2 BFF1QK . BGGAaL$ /66"++cmmAB&789 9966"++cjjn56 6666"++bffQRj12 2266"++bggabk23 33 46 \\%88 H )!5 BtHbh::b>2::b>qA NN1aa ( ) JJAAAB8 9 JJAAAB8 9wwqznnQ+  $ $ & rzz/'BC D)&&= 4 4  ) )s)F)M.AM;.M8;Nc&tjjd}|jd}|jd}tj||d}tj||d}t |j |j y)Nl[LY)r4rYrYrrSr_)rrrrfrrrrnrorrqrrrrs rxtest_different_dimensionalityz*TestPearsonr.test_different_dimensionalitysjii##$56 JJw  JJrNnnQ*nnQ+S]]CMM2rz)rrNctjd\}}tj|j ||j |||}tj||||}t |j |j |j t |j|j |j|j}|j} t |j|j | jt |j|j | jy)N)r3rYrZrrTri) rrrfrrdr1rrrTrWrX) rnrorTrirqrrrrrrs rxtest_array_apizTestPearsonr.test_array_apiszz{z+1nnRZZ]BJJqM"&KAnnQ+F rzz#--'@A BJJszz$:;((*((* BJJvzz$:; RZZ %<=rzN)#rrrr!r,r/r5rBrr parametrizerHrLrNrQrYr[rr^rrsrwr{r fail_slowxfail_on_32bitrslowrrrrrrrrrzrxrrs!5 PP/ 3 [[Wy)&<= !> ! M O L # # [[ LM 3N 3&d# [[C HI6I$6F # 4D [[2 [[ OP [[],LM [[]KM2MNQ 2( [[ [[],LM)N)& [[VaV,3-3 $G0 [[ LM#'N#'J3$ [[V\2 [[],LM >N3 >rzrceZdZdZdZdZdZdZejjdZ dZ dZ d Zd Zejj!d gd d ZdZejj'dejjdZejj'dejjdZejj-dZy)TestFisherExactaSome tests to show that fisher_exact() works correctly. Note that in SciPy 0.9.0 this was not working well for large numbers due to inaccuracy of the hypergeom distribution (see #1218). Fixed now. Also note that R and SciPy have different argument formats for their hypergeometric distribution functions. R: > phyper(18999, 99000, 110000, 39000, lower.tail = FALSE) [1] 1.701815e-09 ctj}|ddgddggd}t|dd|d d gd d ggd}t|d d|d dgdd ggd}t|dd|d dgddggd}t|dd|d dgddggd}t|dd|d dgddggd}t|dd|dd gddggd}t|dd|d dgddggd}t|dd|ddgdd ggd}t|d|dd gddggd}t|d|d dgdd gg}t|dddt|ddy) N8 N0u@rg؞Y?r5 significantrr3rr6g,Ԛ?r8r9g2c?r7rYgY A?g_?rgO^M?rgօa?r4rggE?g$I$I?)rf fisher_exactr )rnrrs rxr5zTestFisherExact.test_basic&s)) UENUEN;? @!fq!f O P!fq!f (>? @!fq!f O P $ 9LE5$$RZZ%67C JJ * *3q658R37 + 9 9rzc ddgddgg}tj|}t|ddddgdd gg}tj|}t|dd d d gd dgg}tj|}t|ddy)Nr7%lrgmt?rfg,:^0i i\BgNQ8rfrr rnrqrs rx test_gh4130zTestFisherExact.test_gh4130YsWsCj !  #A 12!Wq#h   #A 56"Xe} %  #A 56rzc`ddgddgg}tj|}t|dddy)NiiXiViWrrgѧRr$rrs rx test_gh9231zTestFisherExact.test_gh9231os8w '7!3 4  #A/rzcgd}t|gdD]1\}}tjddgd|ggd}t||d3tjd d gd d ggd}t|d dy)N)g?=gNuP=gs=)KLMi(Eii)rr5riPFi8ri_g^K=?)ziprfrr )rnpvalsrmnumrs rxtest_large_numbersz"TestFisherExact.test_large_numbersvs1UL1 :ID#$$uclT3K%@A!DC Tq 9 :  5%.5%.!AB1ECQ7rzcd}tjt|5tjt j ddddy#1swYyxYw)Nz+The input `table` must have two dimensions.rr7)rrrrfrrrrnrs rx test_raiseszTestFisherExact.test_raisess@? ]]:W 5 -   ryy| , - - -s )AAcddgddggddgddggddgddggddgddggf}|D]@}tj|\}}t|dt|tjBy)Nrr6rYr)rfrrrrm)rntablesr oddsratiorms rxtest_row_or_col_zeroz$TestFisherExact.test_row_or_col_zerosq6Ar7#r7QF#q6Ar7#q6B7#% ,E#007OIt s # BFF + ,rzc ddgddggddgddggddgdd ggd d gdd ggd dgdd ggddgddggdd gddggd dgddggdd gddggf }ddgddgddgddgddgddgddgddgddgf }t||D]h\}}g}|jtj|dd|jtj|d dt ||d d!"jy)#Nr3r8r9ri,rVr7ii irr4rr5gI[?gq}!?rg2\&gQp6g:O_?ga?皙?ffffff??333333?UUUUUU?r3rbrhrdrz)r%rk)rappendrfrr )rnrrrrmrs rxtest_less_greaterz!TestFisherExact.test_less_greatersWVaV 1X3x "X4y !3Z#s $VaV VaV VaV VaV VaV   "#5 6 ) * ( )  " #J #J #J 3K $K vu- :KE4C JJu))%VDQG H JJu))%YGJ K CAD 9  :rzc@tjddgddgg\}}y)Nrr3r:i")rfr)rnoddsrs rx test_gh3014zTestFisherExact.test_gh3014s%))Aq6Ax=*AB frzrir`rbrdctjddgddgg}tj||}t |j |j f|y)Nrrrrrh)rrrfrrrr)rnrirrs rx test_resultzTestFisherExact.test_resultsF5%.5%.9:  K@cmmSZZ0#6rzctjjd}tjddgddgg}d}t j t |5tj|}tj||d dddd }t j t |5tj|}tj|| dddd }t j t |5tjtjj }tj|| dddd}t j t |5tjtjddddtjj!tjgdgdtjj!tjdgdgdggdtjj!tjtjddy#1swYxYw#1swYxYw#1swY(xYw#1swYxYw)N9o8or3r8r9z4`alternative` must be the default \(None\) unless...rrrbrriz6...not recognized; if provided, `method` must be an...rz3If the `method` argument of `fisher_exact` is an...rz2`table` must have at least one row and one column.rrr}rrr4r3r4)rrrrdrrrrfrrrrnormrzerosrr)rnrrrrs rx$test_input_validation_edge_cases_rxcz4TestFisherExact.test_input_validation_edge_cases_rxcsii##$78 QFQF+,I ]]:W 5 I,,5F   uV H IK ]]:W 5 5**s3F   uV 4 5H ]]:W 5 5++ ?F   uV 4 5G ]]:W 5 1   rxx/ 0 1  2 2I; ?H  2 2QC!qc? CVL  2 2288F3C DfM- I I  5 5  5 5  1 1s1/I,.I?AI+*)I8II(+I58JrYctjjd}tjddgddgg}t j |}|j }t j|jd|jdj|}t j|}t j || }t|j |d t|j|t j|}t j || }t|j |j d t|j|y) Nrr3r8r9rrSrrrg{Gzd?r$)rrrrdrfrr random_tablesumpmfrr rrr)rnrrr ref_pvalueref_statrrs rxtest_resampling_2x2z#TestFisherExact.test_resampling_2x2sii##$78 QFQF+,  'ZZ %%eiiQi&79JKOOPUV''C0  v6 JV<S]]H-((S1  v6 CJJV<S]]H-rzcgdgdgdgdgdg}tj|}d}tjjd}t j |}t j || }t|j|d t j|d }t j || }t|j|d y)N)rr3r3rrrr)r3rrr3r4rr)rrrrr3r8r4)rrr3rrrr)rrrrrrrgk$?l| KrrgMb@?r$i)r n_resamples) rrdrrrfrrr rr)rnrr$rrrs rxtest_resampling_rxcz#TestFisherExact.test_resampling_rxcs'&&&& (  5!% ii##M2((S1  v6 JT:''CUC  v6 JT:rzc tjjd}tj|}t ddD]}t ddD]}t ddD]|}t ddD]k}tj ||g||gg}tj|}tj||} t| j|jdm~y) Nrrrr4r5r+=r$) rrrrfrrrdrr r) rnrrr*r4cdrrrs rxtest_resampling_exact_2x2z)TestFisherExact.test_resampling_exact_2x2sii##$78((S1q! LA1a[ Lq!LA"1a[L " QFQF+; <#007#00vF' CJJUK LL L LrzN)rrrrr5rrrrrrrrrrrrrrrr&r)xslowr.rrzrxrrs .6907,0 [[88- , :DC  [[],LM7N7 N> [[2 [[.." [[2 [[;;: [[ L LrzrceZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZ dZ!d Z"d!Z#d"Z$d#Z%y$)%TestCorrSpearmanrrctjdd}ttj|j y)Nrr)rf spearmanrrrisnanr)rnrrs rx test_scalarzTestCorrSpearmanr.test_scalar(s) OOB # !"rzctttjgdddgtttjgddy)Nrr3rr9r:)rrrfr3rns rxtest_uneven_lengthsz%TestCorrSpearmanr.test_uneven_lengths,s*j%//9q!fEj%//9a@rzcVtjjdtjjdd}tjjdd}t j ||j jdk(sJt j |j|jdjjdk(sJtttj ||dtttj |j|jy)N鄋r5r4r3r6r6rrS) rrseedrandnrfr3rrrrrrrnrqrrs rxtest_uneven_2d_shapesz'TestCorrSpearmanr.test_uneven_2d_shapes0s v IIOOAq ! IIOOAq !q!$..44>>>qssACCa077==GGGj%//1aa@j%//133rrrfr3r flattenrnrqs rxtest_ndim_too_highz$TestCorrSpearmanr.test_ndim_too_high;s v IIOOAq! $j%//15j%//1a8j%//1dDA148 QYY[qI Krzctjd}tj|d<tt j ||tjtjftt j ||ddt ttj ||dt ttj ||dy)Nrr:rr)rrrr)rrrmr rfr3rrrCs rxtest_nan_policyz!TestCorrSpearmanr.test_nan_policyEs IIcNvv!5??1a02662662BC5??1aFC% 'j%//1aGLj%//1aHMrzc8tjjdtjjdd}d}tj|dd|f<tj ||d}t j|d\}}t j|\}}tj tj ||d|d}tj tj ||d|d}t||d t||d y) Nr6rYr7rrSrrrr+r$) rrr=randrmdeleterfr3r )rnrqkrrcorxpxcorypys rxtest_nan_policy_bug_12458z+TestCorrSpearmanr.test_nan_policy_bug_12458Ns q IINN1b ! &&!Q$ IIa #??18b??1%byy43QQ? YYryyQQ/ ;d/BU+rzctjjdd}d}tjj||}tj|d<tj|d<t j |dd\}}t|Dcgc]G}t|Dcgc]0}t j ||ddf||ddfj2c}I}}}t||ycc}wcc}}w)Nr6rYrr)r4r_r propagaterTr) rrr=r>rmrfr3rrr ) rnmnrqcorrrjrrs rxtest_nan_policy_bug_12411z+TestCorrSpearmanr.test_nan_policy_bug_12411[s q   IIOOAq !&&$66%qq[I fa"FK1XN!Q$1a41;;N""c"O"sC4%5C/ C4/C4cbtjtt}|d}t|dyr)rfr3rer rs rxtest_sXXzTestCorrSpearmanr.test_sXXgs% OOAa  aDAc"rzcbtjtt}|d}t |dyr)rfr3rerr rs rx test_sXBIGzTestCorrSpearmanr.test_sXBIGls% OOAc " aDAc"rzcbtjtt}|d}t |dyr)rfr3rerr rs rx test_sXLITTLEzTestCorrSpearmanr.test_sXLITTLEqs% OOAf % aDAc"rzcbtjtt}|d}t |dyr)rfr3rerr rs rx test_sXHUGEzTestCorrSpearmanr.test_sXHUGEv% OOAd # aDAc"rzcbtjtt}|d}t |dyr)rfr3rerr rs rx test_sXTINYzTestCorrSpearmanr.test_sXTINY{rarzcbtjtt}|d}t |dyr)rfr3rerr rs rx test_sXROUNDzTestCorrSpearmanr.test_sXROUNDs% OOAe $ aDAc"rzcbtjtt}|d}t|dyr)rfr3rr rs rx test_sBIGBIGzTestCorrSpearmanr.test_sBIGBIGs% OOC $ aDAc"rzcbtjtt}|d}t |dyr)rfr3rrr rs rxtest_sBIGLITTLEz!TestCorrSpearmanr.test_sBIGLITTLEs% OOC ' aDAc"rzcbtjtt}|d}t |dyr)rfr3rrr rs rx test_sBIGHUGEzTestCorrSpearmanr.test_sBIGHUGE% OOC % aDAc"rzcbtjtt}|d}t |dyr)rfr3rrr rs rx test_sBIGTINYzTestCorrSpearmanr.test_sBIGTINYrlrzcbtjtt}|d}t |dyr)rfr3rrr rs rxtest_sBIGROUNDz TestCorrSpearmanr.test_sBIGROUNDs% OOC & aDAc"rzcbtjtt}|d}t|dyr)rfr3rr rs rxtest_sLITTLELITTLEz$TestCorrSpearmanr.test_sLITTLELITTLEs% OOF6 * aDAc"rzcbtjtt}|d}t |dyr)rfr3rrr rs rxtest_sLITTLEHUGEz"TestCorrSpearmanr.test_sLITTLEHUGE% OOF4 ( aDAc"rzcbtjtt}|d}t |dyr)rfr3rrr rs rxtest_sLITTLETINYz"TestCorrSpearmanr.test_sLITTLETINYrurzcbtjtt}|d}t |dyr)rfr3rrr rs rxtest_sLITTLEROUNDz#TestCorrSpearmanr.test_sLITTLEROUNDs% OOF5 ) aDAc"rzcbtjtt}|d}t|dyr)rfr3rr rs rxtest_sHUGEHUGEz TestCorrSpearmanr.test_sHUGEHUGE% OOD & aDAc"rzcbtjtt}|d}t |dyr)rfr3rrr rs rxtest_sHUGETINYz TestCorrSpearmanr.test_sHUGETINYr|rzcbtjtt}|d}t |dyr)rfr3rrr rs rxtest_sHUGEROUNDz!TestCorrSpearmanr.test_sHUGEROUND% OOD ' aDAc"rzcbtjtt}|d}t|dyr)rfr3rr rs rxtest_sTINYTINYz TestCorrSpearmanr.test_sTINYTINYr|rzcbtjtt}|d}t |dyr)rfr3rrr rs rxtest_sTINYROUNDz!TestCorrSpearmanr.test_sTINYROUNDrrzcbtjtt}|d}t|dyr)rfr3rr rs rxtest_sROUNDROUNDz"TestCorrSpearmanr.test_sROUNDROUNDs% OOE% ( aDAc"rzctjtt}d}t||t |j |j yr)rfr3rerrrrrs rx test_spearmanr_result_attributesz2TestCorrSpearmanr.test_spearmanr_result_attributess3ooa#. C,S__cmm4rzcgd}gd}tj||}tjtj||gj}t ||y)Nrr3r4r5r6r7)rr3r4r5r7r6)rfr3rrdrr rnx1x2res1rs rx test_1d_vs_2dzTestCorrSpearmanr.test_1d_vs_2dsG  r2&rzz2r(3556d#rzcdD]}dtjddddg}dddddtjg}tj|||}tjtj||gj |}t ||y) N)rRrrr4r5r6r7r3r)rrmrfr3rdrr )rnrrrrrs rxtest_1d_vs_2d_nansz$TestCorrSpearmanr.test_1d_vs_2d_nanss|/ (JRVVQ1a(BQ1a(B??2rjAD??2::r2h#7#9#9jQD D$ '  (rzctjd}| }tjgd}tj|||gj}t j |}tjgdgdgdg}tjdt}d|d d d f<d|d d d f<t|j|t|j|y) Nr7)rrr3r4r6r5)rr_3+?)r_r3+)rrr)r4r4rHg+Hs]s?r3r) rrrrdrrfr3rfloatr rr)rnrrx3rqactual expected_correxpected_pvalues rx test_3colszTestCorrSpearmanr.test_3colss YYq\S XX( ) JJB| $ & &#"5"6">"@A ((67"/1Q3"/!Q((-8 7rzc tjtjddddddgdtjddd ddggd gj}tjtjtjtjgtjtjtjgtjtjd gg}t t j |d j|t j |d j}t |dd|dd|ddfddy)NrrrLffffff@rffffff"@ffffff@g333333@皙@)r<rffffff@ffffff@rrffffff@rrRrrrrr3)gacJC?g%ln?g_0nYt!޿rlrj)rrrmrr rfr3r)rnrqrVrs rx test_gh_9103zTestCorrSpearmanr.test_gh_9103s HHrvvsCc3<BFFCc3<9; <<=A xx"&&"&&"&&1&&"&&"&&1&&"&&"-/0 kBLL ooaF3==QCF1Is1vay9: Grzcd}tjjd}|j|}|j|dkD}|dkD}tj|}t j ||dj}tj||<t j ||dj}|jtj}t j ||dj} gd} t||| g| y)NriHr r<rr)gonj?)Wn%?r) rr RandomStaterHrrfr3rrmastypeint32r ) rnrUrrqrTr*r4rrres3expecteds rx test_gh_8111zTestCorrSpearmanr.test_gh_8111s ii##F+ HHQK HHQK# V HHQKq!7AAvv!q!7AA HHRXX q!7AA:tT*H5rzN)&rrrrr5r9r@rDrFrOrXrZr\r^r`rcrergrirkrnrprrrtrwryr{r~rrrrrrrrrrrrzrxr1r1s#A =KN , ## # # # # # # # # # # # # # # # # # # # # 5 $(8 G6rzr1ceZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZej(j+dddZy)TestCorrSpearmanr2z-Some further tests of the spearmanr function.cgd}gd}d}tj||}t|d|dt|d|dy)Nrr3r4r5r6r6r7r8r9r8gh C?gR?rr)rfr3r rnrrrrs rxtest_spearmanr_vs_rz&TestCorrSpearmanr2.test_spearmanr_vs_r sG >oob"%CFHQK0CFHQK0rzcttjggtjtjfyN)rrfr3rrmr8s rxtest_empty_arraysz$TestCorrSpearmanr2.test_empty_arrays*s$U__R,rvvrvv.>?rzctjjd}tj|j ddd|j dddg}ddgddgg}tj tj j||}d}tj|d|d}t|d|dt|d|dy) Nzrlocscalerrr)gCzYW?ge*=r) rrrrrdotlinalgcholeskyrfr3r )rnrrqrVrrs rxtest_normal_drawsz$TestCorrSpearmanr2.test_normal_draws-sii##D) HHcjjQacj:jjQacj:< =c c  FF299%%d+Q /?ooadAaD)CFHQK0CFHQK0rzcRttjgdgdddy)Nrrr3rr)r rfr3r8s rx test_corr_1zTestCorrSpearmanr2.test_corr_19sEOOIyA!DcJrzctjd}tj|d<tt j ||tjtjft t j ||ddtttj ||dtttj ||dy)Nrr:rr)rrrr) rrrmr rfr3r rrrCs rxtest_nan_policiesz$TestCorrSpearmanr2.test_nan_policies<s IIcNvv!5??1a02662662BC1@  "j%//1aGLj%//1aHMrzctjd}tjd}tttj ||y)Nr4@)rrrrrfr3r?s rxtest_unequal_lengthsz'TestCorrSpearmanr2.test_unequal_lengthsEs. IIcN IIcNj%//1a8rzcgd}dddtjg}tj||d}tj|dd|ddd}t ||y)Nrgr9r8r7rrr4)rrmrfr3rrs rxtest_omit_paired_valuez)TestCorrSpearmanr2.test_omit_paired_valueJsS Arvv r2&9r"1vr"1v&AT4 rzcdttd}ttd}|d|dc|d<|d<|d|dc|d<|d<|d|dc|d<|d<|jtj|jdt t j||d dd y) Nr:rirYiirrrgV-?)listrrrrmrrfr3r?s rx#test_gh_issue_6061_windows_overflowz6TestCorrSpearmanr2.test_gh_issue_6061_windows_overflowQs t  t qT1Q4 !ad#" "qvD'1S6#$   EOOAqVDQGOrzcbd}tjtj|5tjgdgd\}}t |t jt |t jtjgdgd\}}t |t jt |t jtjgdgd\}}t |t jt |t jdddy#1swYyxYw)Nr7rr3r3r3r3rr3)rr>rfr?r3rrrm)rnwarn_msgrrAs rx test_tie0zTestCorrSpearmanr2.test_tie0^s/ \\%44H E $??9i8DAq BFF # BFF #??9i8DAq BFF # BFF #??9i8DAq BFF # BFF # $ $ $s C4D%%D.cgd}gd}gd}gd}tj||}tj||}t||y)Nrrrr)rrrr)rr>r>r)rfr3rr)rnrqrrxryrsrprs rx test_tie1zTestCorrSpearmanr2.test_tie1lsB   ! !__Q " ^^B #B#rzcgd}gd}ddddtjg}ddddtjg}tj||}tj||d }t ||y) N)rr3r>r3)rr4r>r5rr3r>r4r5rr)rrmrfr3r)rnrrsrrtsr1sr2s rx test_tie2zTestCorrSpearmanr2.test_tie2ysd CBFF #CBFF #oob"%oob"8C%rzctjgdgdg}tjgdgdg}tjgdgdg}d}tjtj |5t j |d\}}t|tjt|tjt j |d\}}t|tjt|tjt j |d\}}t|tjt|tjdddy#1swYyxYw)Nrrrrrgr7rrrS rrrr>rfr?r3rrm)rnz1z2z3rrrAs rxtest_ties_axis_1z#TestCorrSpearmanr2.test_ties_axis_1s XX|\2 3 XX|\2 3 XX|\2 3/ \\%44H E $??2A.DAq BFF # BFF #??2A.DAq BFF # BFF #??2A.DAq BFF # BFF # $ $ $s 9C+E--E6crtjgd}tjgd}d}tjtj |5t j ||\}}t|tjt|tjdddy#1swYyxYw)N) rrrrrrrrrr rgN} ?rrg /`?gY%uX-H?gC1tq1?rrg *I?r7rr)rnrqrrrrrAs rx test_gh_11111z TestCorrSpearmanr2.test_gh_11111s HHG H HH- .0 \\%44H E $??1a(DAq BFF # BFF # $ $ $s AB--B6ctjgd}tjgd}tttj ||dy)N) rrrrrrrrrrrr3rS)rrrrrfr3r?s rxtest_index_errorz#TestCorrSpearmanr2.test_index_errors: HHG H HH- . j%//1aa@rzcgd}gd}d}tj||d}t|d|dt|dd|ddz z tj||d }t|d|dt|d|ddz tjt d 5tj||d dddy#1swYyxYw) Nrrrrbrhrrr3rd`alternative` must be 'less'...r ekki-ekki)rfr3r rrrrs rxtest_alternativez#TestCorrSpearmanr2.test_alternatives ?oob"&9CFHQK0CFA!q$9:oob")<CFHQK0CFHQK!O4 ]]:-N O = OOB < = = =s 'C  Crircgd}gd}|tjgz}|tjgz}ttj||tjtjftj||d|}tj|||}t ||d}t jt|5tj||d|dddd }t jt|5tj||d |dddy#1swYIxYw#1swYyxYw) Nrrrrrirhrrrrr) rrmr rfr3r rrr) rnrirrx1nanx2nan res_actual res_expectedrs rxtest_alternative_nan_policyz.TestCorrSpearmanr2.test_alternative_nan_policys bffX bffX  5??5%8266266:JK__UEf1<> r2;G  L12 ]]:W 5 5 OOE5W(3 5 5 1 ]]:W 5 5 OOE5[(3 5 5 5  5 5  5 5s D+D7+D47EN)rrrrrrrrrrrrrrrrrrrrrrrrrzrxrrs|71@ 1KN9 ! P $ $ &$ $A=. [[],LM5N5rzrcpd}gd}gd}d}|D]>}tj|||}t|d|dt|d|d@gd}gd }d}|D]>}tj|||}t|d|dt|d|d@gd }gd }d }|D]>}tj|||}t|d|dt|d|d@gd }gd}d}|D]>}tj|||}t|d|dt|d|d@tjd}tjd}d}|D]>}tj|||}t|d|dt|d|d@|d}|d|d<||d<d}|D]>}tj|||}t|d|dt|d|d@|d}|d|d<||d<d}|D]>}tj|||}t|d|dt|d|d@tjd}tjdddd}d}|D]>}tj|||}t|d|dt|d|d@|d}|d|d<||d<d}|D]>}tj|||}t|d|dt|d|d@|d}|d|d<||d<d}|D]>}tj|||}t|d|dt|d|d@t gd}t gd}d}ttj||dd|d}ttj||d d||d|d<t ttj||d!"t ttj||d#"t ttj||d$gd%}gd&}d'}tj||}t|d|dt|d|dd(} |D]F}tj|||}t|| t|j|jH|D]}ttjgd)gd)|tjtjfttjgd*gd)|tjtjfttjgd)gd*|tjtjftjtd+,5ttjggtjtjfdddtj j#d-} tj | j%ddd./| j%ddd./g}d0d1gd1d0gg} tj&tj(j+| |}d2}tj|d|d}t|d|dt|d|dttjgd3gd3ddd0ttjgd3gd3d dd4tjd5}tj|d6<t-tj||tjtjft/tj||d78d9d:;t/tj||d7d<=d>d:;t ttj||d?8t ttj||d@8tjd5}tjdA}t ttj||tjtd+,5tjdgdg\} } dddttj ttj tjdBt0C}tj2j5|dD}tjdBt0C}tj6|dEd|ddEf}t9tj:tj<j||dy#1swYxYw#1swYxYw)FN)r4r,)r6r3rr4r7r5r8r9)r6r3r7r4rr9r8r5r)variantrr) rr6r3rr4r7r5r8r9) r6r3rr7r4rr9r8r5)r6r3rr4r7r5r8)r6r3r7r4rr8r5)gJ$I¿gmK?)r3rr4r7r5r8r9)r3r7r4rr9r8r5)gJkaa?rrY)r紞xO~>r3)g?>?aV>r6r7)g}'}'?o&5>r_)r.r)g?>r)g}'}'r) rr3r3r5r5r7r7r9r:r:) rr3r5r5r5r5r9r9r9rYgok|?r4gffffff?r,rrbananarmsr[r3rr[r3rr5r8rr)g}+޿g`=i?rrrOne or more sample...rrrrrr)g\zf?g^n)=rg#q?rr:rr)rg/>rlrjr)rr)rg&?rrrrrHir)rf kendalltaur rrrrrrrrrrmrr>rrrrrrrr r rmamasked_greater concatenaterrmstats)variantsrqrrrtauxrr4rrr rrVtaup_values rxtest_kendalltaur sH !A AH1q!T2CFHQK0CFHQK01 $A#AH1q!T2CFHQK0CFHQK01 AA/H1q!T2CFHQK0CFHQK01 AA%H1q!T2CFHQK0CFHQK01 " A " A)H1q!T2CFHQK0CFHQK01 !A Q4AaD AaD8H1q!T2CFHQK0CFHQK01 !A Q4AaD AaD8H1q!T2CFHQK0CFHQK01 " A " ddA*H1q!T2CFHQK0CFHQK01 !A Q4AaD AaD9H1q!T2CFHQK0CFHQK01 !A Q4AaD AaD9H1q!T2CFHQK0CFHQK01 ,-A -.AH((As;A>IH((As;A>I Q4AaD*e..1WE*e..1XF*e..1eD B B:H   2r "CA ,A ,+J5r2t4C,S__cmm45 'U%%iDIffbff% 'U%%iDIffbff% 'U%%iDIffbff% ' ' (0G HAU%%b"-/?@A ))   %C #**!#*6**!#*68 9A #J #J D ryy!!$'+A   R = > > >s :FF%)rbr`rd)rrr<)rRUUUU?ZUUUU?)g~?r$r%)?rr&)ˮ[?rr')g?gSnj?gSnj?)g: ?g3 ?grr,rnrir+r)rqrrr*s rxtest_against_R_n1z+TestKendallTauAlternative.test_against_R_n19sWsQC1 \\,4K L O OOAq+sM: N O O Os AAcDddgddg}}d}|j||||||y)Nrr3r4r5g?r,r.s rxtest_against_R_n2z+TestKendallTauAlternative.test_against_R_n2Cs.1v1v1*  1k3 zJrzcDgdgd}}d}|j||||||y)Nr}rr1r.s rxtest_against_R_c0z+TestKendallTauAlternative.test_against_R_c0Ls&)1  1k3 zJrzcDgdgd}}d}|j||||||y)Nrg)rr3r5r4gVUUUUU?r1r.s rxtest_against_R_c1z+TestKendallTauAlternative.test_against_R_c1Us&\1*  1k3 zJrzcDgdgd}}d}|j||||||y)Nr)rr6r5r3r4rr1r.s rxtest_against_R_no_correlationz7TestKendallTauAlternative.test_against_R_no_correlation_s&1  1k3 zJrzcDgdgd}}d}|j||||||y)Nrr3r4r5r6r7r8r9)r9r7rr4r3r6r5r8rr1r.s rxtest_against_R_no_correlationbz8TestKendallTauAlternative.test_against_R_no_correlationbis'')A1  1k3 zJrzcDgd}gd}d}|j||||||y)N) g333333F@g33333F@g33333D@gfffffJ@gYF@g F@gYI@gF@g N@) g@@r>rL @rg@ffffff@rgqq?r1r.s rxtest_against_R_lt_171z/TestKendallTauAlternative.test_against_R_lt_171rs* C 9*  1k3 zJrzctjjd}|jd}|jd}d}|j ||||||y)NrrgUm*,rrrrHr,rnrir+r)rrqrrr*s rxtest_against_R_lt_171bz0TestKendallTauAlternative.test_against_R_lt_171bsKii##A& HHSM HHSM,  1k3 zJrzctjjd}|jd}|jd}d}|j ||||||y)NrgH4?rBrCs rxtest_against_R_lt_171cz0TestKendallTauAlternative.test_against_R_lt_171csKii##A& HHSM HHSM*  1k3 zJrzzalternative, revc6tjjd}|jd}|jd}t j ||d|}t j ||d|}t |d|dt|d|ddy) Nrirrrrrrj)rrrrHrfrrr )rnrir)rrqrrres0rs rx test_gt_171z%TestKendallTauAlternative.test_gt_171sii##A& HHSM HHSM1W,791\,79T!Wd1g&Qat4rzr)rrrircgd}gd}|tjgz}|tjgz}tj||||}tjtjf}t ||tj||d||}tj||||}t ||d} t j t| 5tj||d||dddd } t j t| 5tj||d ||dddy#1swYJxYw#1swYyxYw) Nrr6r7r8r9r:rr)rrrirrrrr)rrmrfrr rrr) rnrrirrrrrrrs rxrFz)TestKendallTauAlternative.test_nan_policys< bffX bffX %%eU-3N '  L1%%eUv-3N ''Bv4?A  L12 ]]:W 5 E   UEg$*  E E 1 ]]:W 5 E   UEk$*  E E E  E E  E EsD6E6D?E N),rrrr# alternativesrrmp_n1p_n2p_c0p_c1p_no_correlationp_no_correlationb p_n_lt_171 p_n_lt_171b p_n_lt_171cr,rrreversed case_R_n1rrrr/ case_R_n2r2 case_R_c0r4 case_R_c1r6case_R_no_corrr8 case_no_cor_br; case_R_lt_171r@case_R_lt_171brDcase_R_lt_171crG case_gt_171rJrFrrzrxr r s'>h4L FFBFFBFF #D D 0D >D<=DJCKGK+c,ugai89L(4.4&(CDEI [[;YGOHO c,ugai89L(4.4&(CDEI [[;YGKHK c,ugai89L(4.4&(CDEI [[;YGKHK c,ugai89L(4.4&(CDEI [[;YGKHK 3|-=wqyIJSx8H/I"&+,,N [[;^LKMK #l,=wqyIJC h7H.I!%q*++M [[;]KKLK #lJ BCC hz.BTF1HMNOM [[;]KKLK3|[5'!)DESx /D"&+,,N [[;^LKMK3|[5'!)DESx /D"&+,,N [[;^LKMKL5'!)45L4&(345K [[/= 5> 5 [[X'>? [[],LMEN@Erzr c gd}gd}tj||\}}t|dttj |tj||d\}}t|dttj |tj||d\}}t|d ttj |tj||}d }t ||t|j|jtj||d \}}t|d ttj |tj||d \}}t|d ttj |tj||dd\}}t|dttj |tj||dd\}}t|dttj |tj||d \}}t|dttj |tj||dd\}}t|d ttj |tj||dd\}}t|d ttj |tjt j|tj|\}}t|dtjt j|tj|\}}t|dtjt j|tjt j|tj\}}t|dtjtd5tjgg\}}dddttj |ttj |tjtd5tjdgdg\}}dddttj |ttj |tt tjddggdtt tjddgddggdgd}ddddtj g}tj||\}}t|dddtj ddg}tj||\}}t|dgd }d!d"d#d!tj g}tj||\}}t|dd$d%tj d$d%g}tj||\}}t|dgd&}gd'}tj||\}}t|d(d$d%tj d$tj g}tj||\}}t|d(tj d"d#tj tj g}tj||\}}t|d(y#1swYxYw#1swYRxYw))Nrrgs$F)additivegROocyNrrrqs rxz"test_weightedtau..rz)weigherrr)rankgڿgn)rjrcg @ڿg[đ(gIکnTcyrerrfs rxrgz"test_weightedtau..rhrz)rjricyrerrfs rxrgz"test_weightedtau..rhrzrHrrrrrrr3r5r8r[r3)(@rrrnrrrrrnr)rnrrrnr)rrrrrgt34+)rf weightedtaur rrrmrrrrdrrrr>rrr)rqrrr r rr s rxtest_weightedtaurps>AA$$Q*LC12!$$QE:LC12!$$Q;?LC12!   Aq !C*JZ(#--0$$Q5LC01!$$Q5LC01!$$QuELC12!$$QuELC12!$$Q6LC12!$$QkJLC12!$$QkJLC12!$$RZZ%DaHLC12$$RZZ%BAFLC12$$RZZ%D%'ZZ%DFLC12 (0G H1((R0 W1! (0G H3((!qc2 W3!*e//!QC*e//!Q!QKA Aq!RVVA$$Q*LC12 QAA$$Q*LC12#A c3RVV$A$$Q*LC12 sBFFD#&A$$Q*LC12#A!A$$Q*LC01 sBFFD"&&)A$$Q*LC01 c266266*A$$Q*LC01K1133s2[#[[ [cd}tjt|5tjdgdgtjdgdgtjt j gdgdddy#1swYyxYw)Nz)One or more sample arguments is too smallrrr4)rr>rrfrorrm)rs rxtest_segfault_issue_9710rs!sm:G ( 8* 1#u% 1#u% 266(RD) ***s AA>>Bc*d}tj|dzjt}tj|dzjt}tj|d<t j ||dd\}}t|dy)Nrr_rr)rrr)rrrrrmrfrr)rUrqrr_rms rxtest_kendall_tau_largerw-sr A !a%&A !a%&A FFAbEq!GGGAtsrzc "d}d}tjjd}tddD]}g}t|D] }||g|zz } t |}|j ||j |tj t|tj}tdD]U}dD]=} |||||| } tj||||| j} t| | ?|j |Wy) Nchdx}x}x}x}} ttt|tt|D]\} } |r||| ||| zn||| ||| z} || z }|| || k(r|| z }|| || k(r| | z } || || kr || || ks|| || kDr|| || kDr|| z }|| || kr || || kDs|| || kDs|| || ks|| z }||z tj||z z tj|| z z S)Nr)rrlenrr') rqrrrjriaddtotconcdiscuvrrWws rxwkqz*test_weightedtau_vs_quadratic..wkq9sm$%%%d%T%AeCFmU3q6]; FQ7:Q 747#33T!W%Q(88  1HCtqt|Qtqt|Qtad{qtad{adQqTkadQqTk 1!1!!qt !qt   t rwwsQw//"''#'2BBBrzcd|dzz S)Nrrrrfs rxriz.test_weightedtau_vs_quadratic..weigherIsQU|rzr r4rYrHr3r]) rrrrrrrrzintprfrorr ) rrirrr*rr4rjrvr{rrs rxtest_weightedtau_vs_quadraticr7sC  ))   #C 1R[ q A !QJA  G A AyyQrww/q A$ 6q!T7C8**1awDNN#Hf5 6 KK   rzceZdZdZdZy)TestFindRepeatscgd}d}tj|5tj|\}}dddt gdt gdy#1swY&xYw)N) rr3r4r5rr3r4r5rr3r6+`scipy.stats.find_repeats` is deprecated...rrg)r4r4r3r3rdeprecated_callrf find_repeatsr )rnr*rrnumss rxr5zTestFindRepeats.test_basicbsW -?  # #' 2 .**1-IC .3 -4. . .s AA$cgdgfD]S}d}tj|5tj|\}}dddt gt gUy#1swY$xYw)N)rYr2(rrr)rnr*rrepeatedcountss rxtest_empty_resultz!TestFindRepeats.test_empty_resultjsg&+ +ACG''g6 9#(#5#5a#8 & 9 x , vr *  + 9 9s AA' N)rrrr5rrrzrxrr`s /+rzrcleZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZy)TestRegressionctjtt}t |j dt |j dt |jdt |jdy)Nirr) rf linregressrerr interceptrvaluestderrintercept_stderrrnrqs rxtest_linregressBIGXz"TestRegression.test_linregressBIGXusR!!!S)F,,h7FMM3/ FMM3/F33S9rzctjtt}t|jdt|j dt|j dt|jdy)Nrr)rfrrerrrrrrs rxtest_regressXXzTestRegression.test_regressXXsR !!!Q'F,,c2FMM3/FMM3/F33S9rzc\tjtt}t |j dt jtjd5tjttj}dddt |jy#1swY xYw)NrzAn input array...r) rfrreZEROrrrr>r?rrr)rnrq ref_rvalues rxtest_regressZEROXz TestRegression.test_regressZEROXst !!!T*F,,c2 \\%441&9 Q$++/%:;1)< T[[1_5{{dkk8T[[88888 < rr`rhgؗh?gJK@:Cgag?g->g5ՒO?g]؊E @) rfrr sloperrrr'rrr)rnrqrrrs rxtest_regress_against_Rz%TestRegression.test_regress_against_Rs ? 4q!= ?3 '89 BGGM$:; $56 O4,,o>rzctjd}tjdd}|ddgxxdzcc<|ddgxxdz cc<tj||}d}||jd ||j d ||j d ||jd ||jd ||jdy)NrZr6rrrr_ct||dS)NrWr)rrqrrs rx assert_aez1TestRegression.test_linregress..assert_aes&q!R8 8rzrrLgeP Un?g.bt>gڽE?gʺL7?) rrrfrrrrrrr)rnrqrrrqrs rxtest_linregresszTestRegression.test_linregresss IIbM IIa  A+! A+!!!!Q'  9&,,$&""C(&--!12&--1&--!56&))+>?rzcd\}}tj|d|z|}tjd|z||}tj||}t |j dk\t |j dt tj|j t tj|j y)N)gJr 11順r3r_) rrrfrrrrr4rr)rnr*rUrqrrrqs rx test_regress_simple_negative_corz/TestRegression.test_regress_simple_negative_cors1 KK1q5! $ KKAq! $!!!Q'  #$FMM2. BHHV]]++,BHHV44556rzcbtjddd}dtjdddzdz}|tjtjdddz }tj||}t }t t||d}t||dt|vsJy)NrrrrYr)rrrrrr) rrrrfrr"rrrdir)rnrqrrrqrr s rx!test_linregress_result_attributesz0TestRegression.test_linregress_result_attributess KK3 $ "++ac* *R / RVVBKK2s+ ,,!!!Q' 62&'J FJ/!S[000rzc tjd}tjdd}tj||}t |j dt |j dt |jdy)Nr3r4r6r)rrrfrrrrrrnrqrrrqs rxtest_regress_two_inputsz&TestRegression.test_regress_two_inputs s` IIaL IIaO!!!Q' FMM3/ FMM3/F33S9rzctjd}tjd}tj||}t |j dt |jdt |jdy)Nr3rr) rrrrfrrrrrrs rx'test_regress_two_inputs_horizontal_linez6TestRegression.test_regress_two_inputs_horizontal_line s^ IIaL GGAJ!!!Q' FMM3/ FMM3/F33S9rzcNgd}gd}tj||}t|jdt|jdt|j dzdt|j dt|jdt|jd y) N)$rgfffffu@ǧ]@g̤@333333$@gPl@gfffff҄@gfffff"@g |@gH@gq@皙?333333?g<@g33333ׄ@g u@g{@g333333'@g`@33333l@gfffff@g̼@g ^@rrgffffff@g1u@g@g8@gX@333333&@g33333]@gfffffl@g@fffff|@r<)$r g,u@gfffff]@@rrg@4@g|@33333W@gy@rr gP@gfffff@4u@rg%@gm@gl@g0@gfffffƋ@gfffff]@rrgl@rrrrgffffff$@gffffff]@g̜l@g33333@g33333|@rg?g!пr3gpX?rgh-h*rrfrrrr4rnrs rxtest_empty_inputzTestRegression.test_empty_inputK sS \\,4K L )""2r*C66"((3-( (( ) ) )s AA''A0ctjd}tj|d<tjd5t j ||}dddt }tt|t|tjfdzt|jtjy#1swYgxYw)Nrr:ignoreinvalidr6) rrrmerrstaterfrr"rrr rr)rnrqrqrs rxtest_nan_inputzTestRegression.test_nan_inputP s IIcNvv! [[ * ,%%a+F , 62&'6BFF9Q;/V,,bff5 , ,s B<<Cctjd}tjjd}d}tt|5t j ||dddy#1swYyxYw)NrYz$Cannot calculate a linear regressionr)rrrrrrfrrnrqrrrs rxtest_identical_xzTestRegression.test_identical_x] sW HHRL II  R 4 :S 1 #   Q " # # #s A((A1N)rrrrrrrrrrrrrrrrrrrrrzrxrrtsU: :* 7 B9*?&@,7$1 : :G6 7) 6#rzrctjgd\}}}}t|dt|dd}tjt |5tjgdddddtjgdd\}}}}t|dt|dgd }gd }tj||d d \}}}}t|d t|dt|ddt|ddtj||d d\}}}}t|d t|dt|ddt|ddy#1swYxYw)Nrrrr<zHmethod must be either 'joint' or 'separate'.'joint_separate' is invalid.rjoint_separaterjointr)rr3r4r5rYr[)r:rr-7NgQ?separater5rgQ@r3rgGz @r)rf theilslopesrrrr)rrlowerupperrrqrrs rxtest_theilslopesre sI%*%6%6w%?"E9eUs# 3' *C z -> ),<=>&+%6%6y>E&G"E9eUs# 3' !A#A%*%6%6q!T>H&J"E9eUq! 3'tQ/tQ/%*%6%6q!T>E&G"E9eUq! 3'tQ/tQ//>>s EE c(gd}tj|d\}}}}t|tjgdtj|dd\}}}}t |dk(d}tj|dd}t ||y) Nrr5r3rr4rr5numbins)rrrLr)r=r6)rdefaultreallimitsr4)cumcountrbinsize extrapoints)rfcumfreqrrrrr)rqcumfreqslowlimrrr rs rx test_cumfreqr  sA-2]]1a-H*Hfg{h1A(BC-2]] 1.2*Hfg{ K1 FJ --1 ACZ(rzc6tjgd}tj|d\}}}}t |tgdd}tj|d}t ||tjgdd\}}}}t ||y)Nrr5r)r<K}\UU?r r ) frequencyrrr)rrrfrelfreqrr)r*relfreqsr rrr r relfreqs2s rx test_relfreqr s #$A-2]]1a-H*Hfg{h#$MNPGJ --1 %CZ(/4mm@AB D[eBi"e6=?@A C[eBi"E6>@AB D[eCj!12u6=?@A C[eCj!12u6>@AB D[,!7Y6=?@B D[,!796>@AD F[,!7W6=?@A C[,!776>@AC Erzc *tddz}tjgd}tj|gd}t ||t t|tjt tj|tjgd|tjtjdjdtjgdd }tgd gd gd gd g}t ||y) Nr9r<)rr?r)rrrr[r4r5)rrrrrrS)rr5r9)Q?gQ@g(\ @r4r8rZ) rrrrfrr rrndarrayrh)rnrqrrr expected2s rxtest_sequence_perz'TestScoreatpercentile.test_sequence_per s 1IO88N+%%a6X& 3 +,//288L3IJ  "&&ryy}'<'[M:B? 9   ''2.U[["%UC ''2A6U[[$'UI&rzctttjddgddtttjdgdtttjdgdy)Nrr3rrr&er_)rrrfrr8s rxtest_exceptionz$TestScoreatpercentile.test_exception sKj%"9"9Aq62+3 5j%"9"9A3Dj%"9"9A3Crzcfttjgdtjttjtj gggdtjttjgddgtjtjgy)Nrc)rrfrrrmrr8s rx test_emptyz TestScoreatpercentile.test_empty smU,,R4bff=U,,RXXr2h-?DbffMU,,R"b:RVVRVVrrrfmoderrrrnvalsrs rxrFzTestMode.test_empty s\ \\,4I J * ::b>LD& *T288B<(VRXXb\* * *s BB ctjd\}}t|tjdgt|tjdgy)Nrr)rfrKrrrrLs rxr5zTestMode.test_scalar# s;zz"~ fT288RD>*VRXXqc]+rzcrgd}tj|}t|ddt|ddy)N r4r6rrYr4r3r7r9r7rYr7rr7rr4rfrKr)rndata1rMs rxr5zTestMode.test_basic( s17zz% T!Wa T!Wa rzc gd}gd}gd}gd}gd}tj|||||g}tj|dd}t |dtjd ggt |d tjd ggtj|dd}t |dtjgd gt |d tjgd gtj|d d}t |dtjdgdgdgd gd ggt |d tjdgdgdgdgdggy)N)rYrYrr)rYrYrYrY)rrYrr)rrrr)rrrrTrTkeepdimsrrrr9)rYrYrr)r3r4r4r3rYrr3r5r4)rrrfrKr)rnrSdata2data3data4data5arrrMs rx test_axeszTestMode.test_axes. s+     hhueUE:;zz#D48T!Wbhhv./T!Wbhhuo.zz#A5T!Wbhh(8'9:;T!Wbhh ~67zz#A5T!WbhhrdRD2$'EFGT!WbhhaS1#sQC'@ABrzrTrc<tjjdtjjdddd}t j ||j |z}t j ||}tjj||y)N:rYrZr[r\rS) rrr=rHrfrKndimrr )rnrTr*rIrs rxtest_negative_axes_gh_15375z$TestMode.test_negative_axes_gh_15375B sf y! IINN2r2r *zz!!&&+.zz!$' %%dD1rzcgd}g}tj|}d}t||tjt t 5tj|}dddt|y#1swYxYw)NrP)rKcountr)rfrKrrr>rr)rnrSrWrr actual2s rxtest_mode_result_attributesz$TestMode.test_mode_result_attributesJ sf7E"& FJ/ \\,4I J (jj'G (GZ0 ( (s A44A=c Jdtjdddddddddddg }tj|}t |d tj|d }t |d t t tj|d t t tj|d y)Nr4r6rrYrQr3r7r9)r7r4rrrr)rrmrfrKrrr)rnrSrs rx test_mode_nanzTestMode.test_mode_nanT s|BFFAq"b!Q1aQ?E"VV$Ef5VV$j%**eHj%**eIrzdata)r4r6rrr4r4r6r)r4r6rrVFTc|tj|d|}|rt|dddyt|ddy)Nr)rrVrrrR)rnrhrVrqs rxtest_smallest_equalzTestMode.test_smallest_equal^ s9DVhG  1q ) A &rzr`ctjjd}|jdj |}t j ||d}t|j}|j|tjj|j j|tjj|jj|y)Nr_r4r5r6rFrU) rrruniformrrfrKrrpoprr rc)rnrTrIrr*rreference_shapes rxtest_mode_shape_gh_9955z TestMode.test_mode_shape_gh_9955l sii##I. KKYK ' . .u 5jj6qww-D! %%chhnnoF %%ciiooGrzcdtjdtjg}tj|}tj|jr|j dk(sJyNr3r)rrmrfrKr4rc)rnr*rs rx!test_nan_policy_propagate_gh_9815z*TestMode.test_nan_policy_propagate_gh_9815v sI266 "jjmxx!cii1n44nrzc. tjd}tj|dd}|jj|j jcxk(rdk(sJJtj|dd}|jj|j jcxk(rdk(sJJdddtj gddtj dgg}tj|dd}t|jddgt|j d dgtj|dd}t|jdgdggt|j d gdggtj|}tj|dd}tj|jd }t|||jj|jjcxk(rd k(sJJtj|dd}tj|jd }t|jj|jj|jjd k(sJt|j j|j j|j jd k(sJdtj tj tj dgtj tj tj tj d gdd tj d d gg}tj|ddd}t|jgdt|j gdtj|ddd}t|jdgd gd ggt|j d gdgd ggtj|}tj|ddd}tj|jdd}t|||jj|jjcxk(rd k(sJJtj|ddd}tj|jdd}t|jj|jj|jjd k(sJt|j j|j j|j jd k(sJy)N)rr3r4rrFrU)rr4rT)rrr4rr4r3rVrrr6r)rTrVr)rr3r6)r3rr3)rVr) rrrfrKrrcrmr rrr)rnr*rrs rx test_keepdimszTestMode.test_keepdims} s HH\ "jjU3xx~~=I=====jjT2xx~~@L@@@@@Arvv Arvvq 1 2jjU3388aV,399q!f-jjT2388qcA3Z0399sQCj1 HHQKjj6jjU33$xx~~5255555jj5jjT2SXX^^%sxx~~'78xx~~'''SYY__& (9:yy&((( +ffbffbffbffa 0BFFAq ! #jjUvF388Y/399i0jjTfE388qcA3_5399sQC!o6 HHQKjj&IjjUvF3$xx~~5255555jjHjjTfESXX^^%sxx~~'78xx~~'''SYY__& (9:yy&(((rzrrRrcd}tj|}tj|d<tj|dd|}t |jgdt |j gdtjdtjfd tjd fg}tjd | }d }tjt|5tj||dddy#1swYyxYw)N)r5r4rrF)r*rTrVrrr3r4r4r4asdfqwerr=rYrHz=Argument `a` is not....|An argument has dtype...|The DType...rr)rrrmrfrKr rcrIuint8rrrr TypeError)rnrrrhrmy_dtypetestrs rx test_gh16955zTestMode.test_gh16955 swwu~VVT jj4a%JO388\2399l388fbhh/&"**d1KLMxx(+Q ]]9G 4 4 JJt 3 4 4 4s DD c~tjtt5t j g}dddt jdf}t|tjtt5t j t jgd}dddt||gdt jt jt jgg}tjtt5t j |dd}ddddt jgd dgf}t||t j |dd }dt jgd d gf}t||t jggg}tjtt5t j |d }dddt jt jgddgf}t||y#1swYxYw#1swYyxYw#1swYxYw#1swYaxYw) Nrrrr)rrrrrSrr3rRr4rS) rr>rrrfrKrrmrrrrr)rnrrr*zs rx test_gh9955zTestMode.test_gh9955 s\\,4I J !**R.C !vvqkS# \\,4E F :**bffX&9C :S# rvvrvvrvv6 7 \\,4E F ;**QQ6:C ;BFF|aV$S#jj{;BFF|aV$S# HHb"X  \\,4I J (**QQ'C (!Q(S#/ ! !  : :  ; ; ( (s/H ;'HH&;H3 HH#&H03H<zignore::RuntimeWarningrrmrc|jdk(rAtjtt5t j |dd}dddnt j |dd}tj|dd}dj|djcxk(r|jcxk(rdk(sJJy#1swY\xYw)NrrTrUrr<) rrr>rrrfrKrrir)rnrrrs rx test_gh17214zTestMode.test_gh17214 s 66Q;08MN >jj= > >**QTD9CggadT21v||s1v||EsyyEIEEEEE  > >s B<<Cctrdnd}Gdd}tjt|5t j |t jddddtjt|5t j t jdtdddy#1swY\xYw#1swYyxYw)Nz'...only boolean and numerical dtypes...z*Argument `a` is not recognized as numeric.ceZdZdZddZy)8TestMode.test_raise_non_numeric_gh18254..ArrLikec||_yr)_xrCs rx__init__zATestMode.test_raise_non_numeric_gh18254..ArrLike.__init__ s rzNc@|jjtSr)rrobject)rnrIras rx __array__zBTestMode.test_raise_non_numeric_gh18254..ArrLike.__array__ sww~~f--rz)NN)rrrrrrrzrxArrLiker s   .rzrrr4rH) r/rrr|rfrKrrr)rnrrs rxtest_raise_non_numeric_gh18254z'TestMode.test_raise_non_numeric_gh18254 s@O<D  . .]]9G 4 . JJwryy|, - . ]]9G 4 3 JJryy&1 2 3 3 . . 3 3s/B:/C:CCN)rrrrFr5r5r\rrrrrrarergrmrjrrprsrvrrremptyrrrrzrxrHrH s+ , ! C( [[VYRYYr1%562721J [[V BFFAq!Q BFFAq &  [[Z%7'8 ' [[VYRYYr1%5624**H7H5:)x [[\K+@A4B4": [[ 89 [[S8288I#68K"LMFN:F3rzrHceZdZgdZdZej jddZdZ e ddd Z y ) TestSEMrrr\c|j|jd}t|r>tjt t 5tj|}dddnftjj5}|jt|jttj|}ddd|jsJy#1swYxYw#1swY)xYw)Nrr)rdscalar_testcaser+rr>rrrfsemrrr r UserWarningrr4)rnrorrrrs rxtest_sem_scalarzTestSEM.test_sem_scalar s**T%9%9:2> B<08MN /IIo. / /--/ /3 ;' >*IIo. /xx{{ / / / /s C"AC."C+.C7c|j|j}tj|}t ||jdt |j}t tj|d||dz z dzztj|d|j d}|j|dk(|j|}ttj||j|jy)Ng㝎?rddofr3r<rr:) rdtestcaserfrr1rzrwherermr2)rnrorrrrUrqs rxtest_semzTestSEM.test_sem s::dmm, IIh 2::l34   (3q!A#wnD (3 5 IIcN HHQ!VRVVQ ' ! bjj&89rzTz(`nan_policy` only supports NumPy backendrctjd}tj|d<tt j |ddt ttj |dt ttj |dy)Nrr:rrgH=6?rr)rrrmrrfrrrrs rxtest_sem_nan_policyzTestSEM.test_sem_nan_policy) sX IIcNvv!UYYqV46HIj%))Q7Cj%))Q8DrzN) rrrrrrrrrrrrrrzrxrr sX HO [[ LMN$ :dGIEIErzrceZdZejj dgdgdfgdgdfgdZdZdZejj dd d gd Z ejj dd d gd Z e d ddZ ejjdejjddZejjddZy)TestZmapzx, yrry)rrrrrc|j||j|}}||j|z |j|dz }tj||}t ||y)Nrr|)rdrirrfzmapr1)rnrqrrrorrs rx test_zmapzTestZmap.test_zmap6 sZzz!}bjjm1 NbffQ1f&== JJq! 8$rzc~|jgdgdgdg}d}d}d}tj||d}tj||d }| | d z | d z d gd || d z |g|| d z || gg}gd | | | d ggdg} |j|}|j| } t||t|| yNrrrr)rrrr)rrrr.! ?3Ey?;f?rrSrr3rr.r.rrgLXz?)rr.rr.rdrfrr1 rnrorqt1t2t3z0r z0_expected z1_expecteds rxtest_zmap_axiszTestZmap.test_zmap_axisC s JJ,,,. /   ZZ11 % ZZ11 %bSURCE3/R"Q+RCE2s+- .bS2#w/-/ jj- jj- K(K(rzc|jgdgdg}tj||dd}|jgddz }|jgddz }t|d ddf|t|dddf|y Nr)rrrrr)rTr)rr<r<g3Ey?)rr<r=g4?rrrnrorqrrrs rxtest_zmap_ddofzTestZmap.test_zmap_ddof\ s} JJ,,. / JJq!!! ,jj!78*E jj!78D !Q$-!Q$-rzrrr3c|jdddtjg}|jdddddtjg}tj|||d}tj||dd | }t ||y) Nr`r_r3r8r[rrrr6r)rdrrmrfrr1)rnrroscorescomparerrs rxtest_zmap_nan_policy_omitz"TestZmap.test_zmap_nan_policy_omith svRQ/0**b"aB78 JJvwTf Ejj!483rzc |j|jddd}tjtjdddd}tj|d<tj|d<tj|d <|j |d ddftj |d ddf}|j |d ddftj |d ddf}|j |}tj||d d | }tj|d ddf||}tj|d ddf||} |j|| f} t|| y)Nrr3r_rr7)rr5rrrrr)rrTrr) rhrrrrmrdr4rfrstackr1) rnrrorrcompare_0_notnacompare_1_notnarrIrrs rx#test_zmap_nan_policy_omit_with_axisz,TestZmap.test_zmap_nan_policy_omit_with_axist s;BIIdC0':**R[[Q3W=   **WQT]BHHWQT]4K3K%LM**WQT]BHHWQT]4K3K%LM**W% JJvw6 Mzz&A,#%zz&A,#%88T4L)8$rzTlazy arrays don't do 'raise'.rc|jgd}|jddddd|jg}tjtd5t j ||d dddy#1swYyxYw) Nr}rr`r3r8r[zinput contains nanrrr)rdrmrrrrfr)rnrorrs rxtest_zmap_nan_policy_raisez#TestZmap.test_zmap_nan_policy_raise scI&**b"aB78 ]]:-A B < JJvw7 ; < < jjL12::>N3OPjj9:S!rzN)rrrrrrrrrrrrrrrrrrzrxrr3 s [[ , - , - / %  %)2 . [[VaV,  -   [[VaV,%-%(.MN<O<  [[ WX [[ VW"XY" [[ WX"Y"rzrceZdZdZdZdZdZdZdZe dd d Z d Z e jjd d Ze jjd dZe jjd dZe jjd dZe jj'dej*gej,dgdZedZy) TestZscorectj|jgd}gd}t||j|y)Nrg)gOT\wg%ܿg%?gOT\w?)rfzscorerdr1)rnrorrdesireds rx test_zscorezTestZscore.test_zscore s5 LLL1 262::g./rzcr|jgdgdgdg}d}d}d}tj|d}tj|d }| | d z | d z d gd || d z |g|| d z || gg}gd | | | d ggdg} t||j|t||j| yrrdrfrr1rs rxtest_zscore_axiszTestZscore.test_zscore_axis s JJ,,,. /   \\!! $ \\!! $bSURCE3/R"Q+RCE2s+- .bS2#v.-/  BJJ{34BJJ{34rzc|jgdgdg}tj|dd}|jgddz }|jgddz }t|d ddf|t|dddf|yrrrs rxtest_zscore_ddofzTestZscore.test_zscore_ddof s{ JJ,,. / LL +jj!78*E jj!78+F !Q$-!Q$-rzc|jddtjddg}tj|d}t ||j |j|jy)Nrr3r5r6rRr)rdrrmrfrr2rrrnrorqrs rxtest_zscore_nan_propagatez$TestZscore.test_zscore_nan_propagate sL JJ1bffa+ , LL{ 3277177BFF34rzc|jdd|jddg}tj|d}|jdd|jd d g}t ||y) Nrr3r5r6rrgIHb=gIHb=gIHb=?gIHb=?)rdrmrfrr1rnrorqrrs rxtest_zscore_nan_omitzTestZscore.test_zscore_nan_omit sa JJ1bffa+ , LLv .::22!vv11 !" 8$rzc|j|jdddddg}tj|dd}|j |j|jgtj|ddd g}t ||y) NrrrLrrrrrr)rdrmrfrconcatr1rs rxtest_zscore_nan_omit_with_ddofz)TestZscore.test_zscore_nan_omit_with_ddof sp JJS#sC8 9 LLv 699bjj"&&2ELL12Q4OPQ8$rzTrrc|jdd|jddg}tjtd5t j |ddddy#1swYyxYw) Nrr3r5r6zThe input contains nan...rrr)rdrmrrrrfrrs rxtest_zscore_nan_raisez TestZscore.test_zscore_nan_raise sQ JJ1bffa+ , ]]:-H I 0 LLw / 0 0 0s AA&c|jdgdz}t|td5tj|}dddt |j |j|jy#1swY:xYw)NgʡEr4rr) rdr)rrfrr2rrrmrs rxtest_zscore_constant_input_1dz(TestZscore.test_zscore_constant_input_1d sb JJx!| $ N2N O  QA 277177BFF34  s A77Brc B|jgdgdg}t|td5tj|d}dddt |j|j dddg|j dddggt|td5tj|d }dddt|j|j|j |j |j |j gtj|d ddfgtj|d}t||jtj|j|d |j|jd }t|td5tj|d}dddt||j|j|j y#1swYxYw#1swY]xYw#1swYTxYw) Nrrrrrr8rn*@rrrrSr.rrr_)r4r7) rdr)rrfrr1rmr2rrhrrr)rnrorqrrrrrs rxtest_zscore_constant_input_2dz(TestZscore.test_zscore_constant_input_2d s JJ002 3 N2N O )aa(B )BJJtT(B)+c3(?(AB CN2N O )aa(B )BHHbjj"&&"&&"&&"&&1Q&R&+ll1QT7&;&=> ? LL &2::ell2::a3G&H!''RS GGFO N2N O + QT*A +277177BFF34! ) )  ) ) + +s#G;H*H;HHHc .|jgdddd|jgdd|jdgg}d}d}t|td5t j |d d }dddt |j|j| d |jg|jd d |jg|j||j|jggt|td5t j |d d }dddt |j|j|j|j|jg| d ||jg| dz ||j| dz ggy#1swYxYw#1swYxYw)Nrrr8rnrrrrrrrrTr.rrr3)rdrmr)rrfrr1)rnrorqrs2rrs rx-test_zscore_constant_input_2d_nan_policy_omitz8TestZscore.test_zscore_constant_input_2d_nan_policy_omit sr JJ0tT2662tRVVT24 5   N2N O <aF;B <BJJ!T266(B)+C(@)+BFFBFF(C(EF GN2N O <aF;B <BJJ(H*+Q266(:*,QBFFRCE(B(DE F < <  < -F >F Fc |jtjtjtjtjggdg}tj|dd}t ||jtjtjtjtjggdgy)N)rrrnrnrrrr)rdrrmrfrr1rs rxtest_zscore_2d_all_nan_rowz%TestZscore.test_zscore_2d_all_nan_row s JJ802 3 LLvA 62::'G'='?@ Arzc|jd|j}tj|dd}t ||y)Nrrr)rrmrfrr2)rnrorrrs rxtest_zscore_2d_all_nanz!TestZscore.test_zscore_2d_all_nan% s4 GGFBFF # LLvD 91rzrq)r4rr6ch|j|}tj|}t||yr)rdrfrr2)rnrqrors rxtest_zscore_empty_inputz"TestZscore.test_zscore_empty_input, s& JJqM LLO1rzctjjd}|jd}tj|}tj j ||}d|jd<tj|dd}tjtj|rJtj|}t|dd|tj|d}t|dd||d|ddtjtd5tj|}dddt!|ddtj"tjtd5tj|d}dddt!|ddtj"y#1swYxYw#1swY3xYw) Ni_rYTrrrSrr)rrrstandard_normal zeros_liker masked_arraymaskrfranyr4r rr>rrrm)rnrorrqrrrrrs rx$test_zscore_masked_element_0_gh19039z/TestZscore.test_zscore_masked_element_0_gh190392 s`ii##G,    #}}Q EE  q$ 'q ll1QR5!66"((3-(((ll1oAB%ll14(AB%!!" \\.0L M ",,q/C "SWbff% \\.0L M -,,qt,C -SWbff%  " " - -s$F8:G8GG N)rrrrrrrrrrrrrrrrrrrrrrrrr'rrrzrxrr s.05. .5 %% .MN0O0 5  [[ VW5X5* [[ VWFXF& [[ VWAXA [[ VWX  [[S8288B<)1D"EFG &&rzrc"eZdZdZedZy) TestGZscorec tjgd}tj|j|}tj|tj |z tjtj |dz }t||j||jdjy)NrgrrrrH) rrdrfgzscoreloggmeangstdr1rI)rnrorqrrs rxtest_gzscore_normal_arrayz%TestGZscore.test_gzscore_normal_arrayM s JJ| $ MM"**Q- (&&U[[^+,rvvejj6K/LL2::gRZZ^5I5I:JKrzctjgd}gd}tjj||}t j |}ddtj ddg}tjj||}t|j|jt|j|jt|tjjsJy)N)rr3r_r4r5)rrrrrrg0jg`ȿg/A?g(/?) rrrrrfrrr compressedrr MaskedArray)rnrqrmxrrs rxtest_gzscore_masked_arrayz%TestGZscore.test_gzscore_masked_arrayS s HH% & UU    - MM" #_bffn"$%%$$W4$8 (:(:(<= -!RUU..///rzN)rrrr r'rrrzrxrrK sL  0 0rzrc eZdZdZdZdZdZdZejjdgddZ ejjd d e je jd d gfd e je je jd gfgd ZejjddgdfdgdZdZy)TestMedianAbsDeviationctjdddddddddddd d d d d d d d d d ddtjg|_tjgd|_y)N皙@rr>皙@r?333333@= ףp=@r=(\ @rr?r>皙 @)\(@Q@)rrrrr>rr?rrrr=rrrrr?r>rrrrrrg33333<@)rrrmdat_nandatr8s rx setup_classz"TestMedianAbsDeviation.setup_classb sgxxtS#sCc!%tT4c3!$c3S$bff!NO 88?@rzcttj|jdd|jj dd}tj|d}t j gd}t||y)NrSgQ?r7r5r)gףp= ?r<?r)rrfmedian_abs_deviationrrhrrdr)rnrmad mad_expecteds rxtest_median_abs_deviationz0TestMedianAbsDeviation.test_median_abs_deviationj saE66txxdK! #hhq!$((15zz"9: !#|4rzc^tj|jd}t|dy)Nrrg(\?)rfr!rr)rnr"s rxtest_mad_nan_omitz(TestMedianAbsDeviation.test_mad_nan_omitr s"((&IC&rzctjddddtjggdg}tj|d}t |tjtjdgy)Nrrrr)rrrLrrrrSrrrmrfr!r)rnrqr"s rxtest_axis_and_nanz(TestMedianAbsDeviation.test_axis_and_nanv sV HHsCc2662/1 2((3S"((BFFC=12rzc tjdddddtjtjg}t j |d}t |dy) Nrr4r5r7rErrr)rrrmrrfr!r)rnrr"s rxtest_nan_policy_omit_with_infz4TestMedianAbsDeviation.test_nan_policy_omit_with_inf| sC HHaAq"bffbff5 6((v>S#rzrT)rrr3Nctjd}tj||}t |tj |j |tjy)N)r4rr5rS fill_value)rrrfr!r full_liker"rm)rnrTrqr"s rxtest_size_zero_with_axisz/TestMedianAbsDeviation.test_size_zero_with_axis sF HHY ((6S",,quu$u'7BFFKLrzznan_policy, expectedrr=rRc xtjtjtjtjtjtjtjgddddtjtjggdg}tj||d}t ||y)Nrr6r4r7)r6r7r8r:r:rYrr()rnrrrqr"s rxtest_nan_policy_with_axisz0TestMedianAbsDeviation.test_nan_policy_with_axis sy HHrvvrvvrvvrvvrvvrvvF!Q2662662)+ ,((zJS(#rzzaxis, expectedr)r>rrn)Nr@ctjddddtjggdgdg}tj|tj d|}t ||d d y) Nrr3r5r:)rrrrr[)r4r4rrr)centerrrTr#r)rrrmrfr!rir )rnrTrrqr"s rxtest_center_mean_with_nanz0TestMedianAbsDeviation.test_center_mean_with_nan s[ HHq!Q266*&-/ 0((277v.24XE>rzctjtd5tjgdddddy#1swYyxYw)Ncallabler)rr3r4r6rEr5)rrr|rfr!r8s rxtest_center_not_callablez/TestMedianAbsDeviation.test_center_not_callable s9 ]]9J 7 @  & &|B ? @ @ @s ?AN)rrrrr$r&r)r+rrrr0rrrmr2r6r:rrzrxrra s@5'3   [[V_5M6M  [[3%xrxxc0B'CD*HBHHbffbffc5J,KLNO$O$ [[- "23[AC?C?@rzrcltt||d|D]}t|j|uy)z Checks that all of the warnings from a list returned by `warnings.catch_all(record=True)` are of the required type and that the list contains expected number of warnings. znumber of warningsN)rrzrcategory) warn_list expected_type expected_lenwarn_s rx_check_warningsrA s5 Y/CD1-/01rzceZdZdZdZej jdgejdgdZ dZ dZ dZ d Zd Zd Zd Zd ZdZy)TestIQRctjddz}tjj|t t j |dy)Nr9r<r)rrrrrrfiqrrCs rxr5zTestIQR.test_basic s8 IIaL3  !UYYq\4(rzc tjd}tj|tj|dtj|dtj|dtj|ddtj|dddtj|dddd tj|dd d d d tj|ddddddy)Nr<rr)rYZ)rrr)rrr=rR)rrrrlineargٿrrT)rrrfrE)rnr-s rxtest_apizTestIQR.test_api s GGFO !  !T !Q !V !T8$ !T8S) !T8S+6 !T8XwA !T8T67DArzrqrctjtt5t t j |tjdddy#1swYyxYwrJ) rr>rrrrfrErrmrCs rxrFzTestIQR.test_empty s= \\,4I J / 1rvv . / / /s .AA cRtjd}ttj|dt tj|dtj dt tj|dtj dttj|d dttj|d dttj|d dttj|d dttj|d dtjdtjdz}t tj|dtj dt tj|dtj dt tj|dtjddt tj|dtj dt tj|dtjddt tj|dtjddy)N)r8r5rrrSr5rr8rH interpolationmidpointnearestrr.r5r6r7r7)r6r7)r5r7r3rr>rrr3r6rr) rrrrfrEr rrrr?s rx test_constantzTestIQR.test_constant s GGFOUYYq\3'599QQ/!=599QQ/!=UYYq93?UYYq ;SAUYYq :C@UYYq8#>UYYq93? GGI 1 -599QQ/&1AB599QQ/&1AB599QQ/1EF599QV4bhhqkB599QV4bgganE599QV4bgganErzctjddz}ttj|ddttj|dt tj|ddgy)NrrrrTru)rrrrfrEr rCs rxtest_scalarlikezTestIQR.test_scalarlike sS IIaL3 UYYqt_c*UYYq\3'599Q6>rzctjdjd}tt j |dt t j |dtjddt t j |dtjd d t t j |d dt t j |d dy) NrrrrrSr6rLrr4rrrQ)rrrhrrfrEr rrCs rxtest_2DzTestIQR.test_2D s IIbM ! !& )UYYq\3'599QQ/B@599QQ/B@599QV4c:599QV4c:rzc 2tjjd}tj|gdz}t j |}t t j |d|tj|dd}t t j |d||jdd }t t j |d ||jdd }t t j |d t j |dt t j |d t j |dtjd }tjj||jd}t t j |d dt j |dddddddfjt t j |dd t j |ddddd ddfjt t j |ddt j |dddddddfjt t j |ddt j |dddddddfjt t j |ddt j |dd ddddfjt t j |ddt j |dddddd fjt t j |ddt j |ddddddfjtttj |dtttj |dy)N)GrQrrYrrSr_rrrrQrmrir4r6r8rZrrr4)r4rr]r3)r4rr3r4r3)rr)rr4r3r3r5r)rrrdstackrfrErmoveaxisswapaxesrrrhrrr&r)rnorqqr-s rxr@zTestIQR.test_axis s II  (  + IIqcBh  IIaLUYYqv.2 KK2q !UYYqv.2 JJq! UYYqv.2 JJq! UYYqy1YYqt, .UYYqt,YYqq) + IIn % ! IIm $UYYqy1!4YYq1Q{0023 5UYYqy1!4YYq1a{0023 5UYYqz215YYq1a{0023 5UYYqy1!4YYq1Qqz//12 4UYYqv.t4YYqAa{0023 5UYYqw/5YYqAq!}2245 7UYYqv.t4YYqAq|1134 6 iAA6j%))QV)g)@r)rYrg?)rrBr)rr<) rrrrfrErrrrmr|rCs rxtest_rngzTestIQR.test_rng s IIaLUYYq\1%UYYqj137UYYqj137EIIaX6<j%))QH=j%))QRVVRLAiA;?rzctjd}tjd}ttj|dttj|dttj|ddttj|ddttj|ddttj|dd d ttj|ddttj|d dttj|dd dttj|d dttj|d dttj|d d ttj|ddttj|dd dttj|dddD]}tj||t t tj|dy)Nr6r5r3r=rHrLr.)rP)rrMr4rrOrrNr>) inverted_cdfaveraged_inverted_cdfclosest_observationinterpolated_inverted_cdfhazenweibullmedian_unbiasednormal_unbiasedr)rrrrfrErr)rnrqrrrs rxtest_interpolationzTestIQR.test_interpolation s IIaL IIaLUYYq\1%UYYq\3'UYYq91=UYYq93?UYYq91=UYYqhhGKUYYq91=UYYq8!<UYYqhgFJUYYq8!<UYYq :A>UYYq :A>UYYq ;Q?UYYqhjI3OUYYq ;Q?* /F IIav .  / j%))QhGrzctjd}ttj|ddj dttj|ddj dttj|ddj dttj|d dj d ttj|d dj d ttj|d dj dttj|ddj dttj|ddj dttj|ddj dttj|ddj dttj|d dj dttj|d dj dttj|d dj dttj|ddj dy)NrZFrUrr3)r4r6rZr)r8rZ)rr4)r6r8)rr3rrr3r4rur[r~Tr)r4r6rrZ)rrr8rZ)rr6r8r)r4rr8rZ)rrr8r)rrrrfrErrCs rxrvzTestIQR.test_keepdims= s GGM "UYYqte<BBBGUYYqq59??LUYYqv>DDgNUYYqv>DDfMUYYqte<BBJOUYYq,?EErJUYYqy5AGGNUYYqtd;AA<PUYYqq48>> NUYYqv=CC]SUYYqv=CC\RUYYqtd;AA=QUYYq,>DDlSUYYqy4@FF Urzc jtjdjd}tt j |ddtt j |ddtt j |ddtj |d<tjd 5tjd tt j |dtj tt j |d d ddtj ddgtt j |dd dtj dgdddtjd 5tjd tt j |ddtt j |d d tjddtt j |dd gddddtttj |dtttj |d d tttj |dd tttj |dy#1swY@xYw#1swYxYw)N.@rrRrr8rrrTralwaysrrSr6rr3rC)r3r>r3barfood) rrrhrrfrErmwarningscatch_warnings simplefilterrrrrCs rxtest_nanpolicyzTestIQR.test_nanpolicyP s IIdO # #F + UYYq[91=UYYqV4a8UYYqW5q9&&$  $ $D 1 )  ! !( + 1= ! 11EQ1- / 11ERVVQ )  ) $ $D 1 O  ! !( + 18# > 11@"''!Q- P 11@+ N  O j%))Q7Cj%))QQ7Kj%))QQ7K j%))Q9E) ) ) O Os1B-JN L  !!85@!B "!RVVQ 89 D F 11CKPRVVQ ) )" UYYq?EEIIaxFK+ -UYYq?F j%))Qh?/ ) )s 1EJJ"N)rrrr5rIrrrrrrFrRrTrVr@rerprvr{r~rrzrxrCrC sw) B [[S2yryy|"45/6/F,? ;'=R @ HDV&F@!@rzrCceZdZdZgdZdZejjdejjdZ e jjdddge jjd ed d e jjd d Ze jjdddge jjdd e jjd dZdZdZe jjdgde jjdddgdZdZdZdZdZe jjdgd e jjdgde jjdd d!ge jjd d"Zy )# TestMomentsa Comparison numbers are found using R v.1.5.1 note that length(testcase) = 4 testmathworks comes from documentation for the Statistics Toolbox for Matlab and can be found at both https://www.mathworks.com/help/stats/kurtosis.html https://www.mathworks.com/help/stats/skewness.html Note that both test cases came from here. rrr rrY)rYr3zm, crr)Nrrz?ignore:divide by zero encountered in divide:RuntimeWarning:daskcPtjjd}|j|j|}t j |||}||j |dn|}|j||z |zd|jdz }t||dy)NlrHZrr9rrS缉ؗҜ.test_cases s RZZ^,A Arzz"&&1 2 RZZ"**Z=>A Arzz"&& zC D RXXf-A6A Arxx~ . RZZ-A6A Arzz266(3 4 RZZ-aV!DA Arxx/ 0rzzSee documentation for...zMean of empty slice.z invalid value)rdrrfrrr1rrrr+r>rrrr rr)rnrorrrrrrs ` rx test_momentzTestMoments.test_moment s"::dmm, LLD$8$89 :2::c?+ LL1 %2::c?+ LL1 %2::c?+ LL1 %2::d+, LL1 %2::c?+ LL1 %2::f-. LL< 02::&:;< LL3 '2::c?+= ]]:W 5 ( LL3 ' ( LL#3 42::&:;< 1 B<08RS    --/ 3 >+AB >?;   - ( (&    s$K K 4K! KK!K*cvtjd}tj|d<tt j |dtjt t j |ddtttj |dtttj |dy) Nrr:r3rrrrr) rrrmrrfrrrrrCs rxrFzTestMoments.test_nan_policy sn IIcNvv!U\\!Q'0ELLv>Dj%,,gFj%,,hGrzrI)rr complex128z expect, orderrrQc|dk(rt|rtjt||}|j t j jd|}tj||}t||j ||tj|j|dd|}t||jd||tj|j|d d |}t||jd ||tj|j|d d|}t||jd ||y) Nrr6rHr)r7r6r)rTr)r6rr3)rr3r5r6r) r,rskiprErdrrrHrfrr2 broadcast_tor)rnrIexpectrrorqrrs rxtest_constant_momentsz!TestMoments.test_constant_moments s , 8B< KKME" JJryy~~a(J 6 LL% (2::fE::; LLF3!5 I2774u7=> LLO<1$ &277<u7EF LLO<4$ &2772vU7;? R0557 T%B%BBG IrzrrTr1r5Nrc tjjd}|jd}tj|j ||||}|j t ||||}t||y)NlhU~)r6r7r8r)rTr5)ri)rrrrfrrdr!r1) rnrorrTr5rrqrrs rxtest_moment_array_apiz!TestMoments.test_moment_array_api/sgii##N3 JJIJ &ll2::a=%d6JjjE4f=>S!rz)rrrrrrrrr=rHrrrrrrrrrrFrrrrrrrrzrxrr s HOIINN4!yy~~b1 [[Vb']3 [[VW\<%HI [[I.J4 . [[Vb']3 [[S,/ [[I"04 "3jH [[W&JK [[_vv.>?=@L=&D= "I [[Wl3 [[VZ0 [[Xay1 [[I"214 "rzrc&eZdZdZgdZgdZdZy)SkewKurtosisTestrr)gp= ף?g?߾?gD9?gQI?g}?5^Ic|jg}t|r:tjtt 5|j |}dddndtjj5}|jtd|jtd|j |}dddt|j|jy#1swY/xYw#1swY;xYw)NrMean of empty slicezinvalid value encountered)rdr+rr>rrstat_funrrr rrr2rm)rnrorqrrs rx test_empty_1dzSkewKurtosisTest.test_empty_1dBs JJrN B<08MN 'mmA& ' '--/ '3 >+@A >+FGmmA&  ' RZZ/0 ' ' ' 'sC!5>C-!C*-C6N)rrrrr testmathworksrrrzrxrr=sOH2::&89F JJrzz$--0 12::c?7rzctjd}tj|d<tjd5t t j |tjdddt t j |ddttt j |dttt j |d y#1swYmxYw) Nrr:rrrrrrr) rrrmrrrfrrrrCs rxrFzTestSkew.test_nan_policyes IIcNvv! [[ * 0 A / 0UZZf5r:j%**aGDj%**aHE  0 0s .CC"cTttjtddy)NrYr)rrfrrr8s rxtest_skewness_scalarzTestSkew.test_skewness_scalarosUZZr +S1rzc|jd}|j|d}tj|dj |j }t jd5tj|j|d}dddt|jd|j gy#1swY1xYw) NrrrQrrrrSr) rrhrjrkrlrmrrrfrrdr2)rnror*rs rxtest_skew_propagate_nanz TestSkew.test_skew_propagate_nanss IIbM JJq' " FF1IdO   ' [[ * 2 2::a=q1A 22::q"&&k23 2 2s +'CC c|j|jdgd}t|td5t t j ||j|jt t j |dz|j|jt t j |dz |j|jt t j |d|j|j|jdgd z}t t j ||j|jd |jd d d zz}t t j ||j|jdddy#1swYyxYw)N9ѿrYPrecision loss occurredrgCFrg,@r8rr5r) repeatrdr)rr2rfrrmrrnror*s rxtest_skew_constant_valuez!TestSkew.test_skew_constant_value}s6 IIbjj+/ 4 N2K L ? EJJqM2::bff+= > EJJqx0"**RVV2D E EJJqx0"**RVV2D E EJJqu5rzz"&&7I J D6!8$A EJJqM2::bff+= >RYYsA&u,,A EJJqM2::bff+= > ? ? ?s FGG Trc tjtd5tjj d}|j d}d|dddf<t j|j|dddy#1swYyxYw)Nrrl:"z`)rrYr)\(?r) rr>rrrrrfrrd)rnrorr*s rxtest_precision_loss_gh15554z$TestSkew.test_precision_loss_gh15554sm \\.0I J &))'' 4C *AAadG JJrzz!} %  & & &s ABB rTr_rr3NrFctjjd}j|jd}fd}t j |||}||||}t ||y)Nf#"rlc|j|d}d}j||d}j||z dz|}j||}||dzz }|s#|j|}||dz |zdz|d z z z}|S) NrrTrUr4rSrr<r)rhrirr) r*rTrrimu3rrrUros rxskewnessz-TestSkew.test_vectorization..skewnesss|JJq%(7714$77D''1t8a-d'3C&&&&Cq.CGGDMSA #-S99Jrz)rTr)rrrrdrfrr1) rnrorTrrrqrrrs ` rxtest_vectorizationzTestSkew.test_vectorizationsa ii##M2 JJszz), - jjD1qt$/S!rzN)rrrrrrrrrFrrrrrrrrrzrxrrPs [[O 8 8F24? &&'& [[V%56 [[VeT]3"47"rzrc8eZdZdZej j ddZdZdZ dZ dZ ej jdgd ej jd d d gej jd d d gdZ y) TestKurtosisc,tj|Sr)rfkurtosisrCs rxrzTestKurtosis.stat_funs~~a  rzz1ignore:invalid value encountered in scalar dividec tj|j|j}|j |sJtj|j|j ddd}t ||jdtj|j|j dd}t ||jdtj|j|jdd}t ||jd|jd}|j|dk(|j|}ttj||j|jy) NrrfisherrgO߻S@g x|N @g= ףp=?rr9) rfrrdrr4rr1rrrrmr2)rnrorrrqs rx test_kurtosiszTestKurtosis.test_kurtosiss NN2::d&:&:; <xx{{ NN2::d&8&891QQ O2::o67 NN2::d&8&89!! L2::&789 NN2::dmm4a ;2::d+, IIcN HHQ!VRVVQ 'q)2::bff+=>rzctjd}tj|d<tt j |ddt ttj |dt ttj |dy)Nrr:rrGzrr)rrrmrrfrrrrCs rxtest_kurtosis_nan_policyz%TestKurtosis.test_kurtosis_nan_policysU IIcNvv!ENN1@)Lj%..!Hj%..!Irzcttttjgdtj y)Nr})rtyperfrrrr8s rxtest_kurtosis_array_scalarz'TestKurtosis.test_kurtosis_array_scalarsT%..34bjjArzc.tjdjddjt}tj |d<t j|dd}tjj|dtj gd y) Nr9r3r_rQrrRrSg(\r#r$) rrrhrrrmrfrrr )rnr*rJs rxtest_kurtosis_propagate_nanz(TestKurtosis.test_kurtosis_propagate_nansi IIaL B ' . .u 5&&$ NN11 = ""1ubffoE"Brzc |jdgdz}t|td5|jt j |dsJ|jt j |t dzdsJ|jt j |t dz dsJ|jt j |ddsJ dddy#1swYyxYw) NrrYrrF)rl r)rdr)rr4rfrrrs rxtest_kurtosis_constant_valuez)TestKurtosis.test_kurtosis_constant_values JJ }R' ( N2K L I88ENN1U;< <<88ENN1uU|+;EJK KK88ENN1uU|+;EJK KK88ENN1UGH HH  I I Is B:C,,C5rTrrFTrctjjd}j|jd}fd}t j ||||}|||||} t || y)NrrPcP| j|d}d} j||d} j||z dz|} j||d}|r ||dzz d z }n8|j|}|d z |dz |d z zz |d z|z|dzz d |d z zz z}|r|S|d zS) NrrTrUr5rSrTr}r3r4r)rhrirr) r*rTrrrimu4mu2rrUros rxrz1TestKurtosis.test_vectorization..kurtosiss|JJq%(7714$77D''1t8a-d'3C&&!&4CCFlQ&GGDMs!!}-!a%3sAv1E1Q31OP!3 -cAg -rz)rTrr)rrrrdrfrr1) rnrorTrrrrqrrrs ` rxrzTestKurtosis.test_vectorizationse ii##M2 JJszz), - .&nnQTVDqt$v>S!rzN)rrrrrrrrrrrrrrrrzrxrrs! [[ ST?U?>JBCI [[V%56 [[VeT]3 [[Xt}5"647"rzrctdd}tjdd}|tjtj ||}g}t j5t jdtt|jD]}tjt5tj |d|}t j"|d kDr9t j"t j$|r|j'|ddd dddt)j*||t(j,j/|}||fS#1swYxYw#1swYSxYw) NF) allow_nanallow_infinityrr3)min_dimsmin_side)rIelementsrerrorrr)dictnpst array_shapesarraysrrrxryrzrrzr contextlibsuppress Exceptionrfrrrr hypothesisassume strategies sampled_from)drawrrrhok_axesrTrs rxttest_data_axis_strategyrs3eE:H   q1 5E  "**xuM NDG  ")g&#djj/* )D$$Y/ )ll4qt<66#'?rvvbkk#.>'?NN4( ) ) ))g  %%227; 7A*E2! E>2E; 7E>>FceZdZgdZgdZdZdZdZdZdZ dZ d Z dZ ed z Z d ed z z Zej j#d ej j#d dZdZej j#ddZeddej j-dgddZdZeddej j2ej6ej8j;ddeej j-ddd gd!Zy")#TestStudentTestr.rr)rrrrrgMXzg4/?gMXz gFX5t?gMXz?r3rrrc*t5}tjdd5|jtdt |s|j dnd}tj|d\}}ddddddt|j |jt|j |jtj|j |jd\}}t||j |jt||j |jtj|j |jd}d}t|||tj|j |j d\}}t||j |j"t||j |j$tj|j |jd \}}t||j |j&t||j |j(tj|j |jd \}}t||j |j*t||j |j,y#1swYxYw#1swYxYw) Nrrdivide!Degrees of freedom <= 0 for slicerrrrrrorr)r rrrrr+rdrf ttest_1sampr2rmX1r1T1_0P1_0rX2T2_0P2_0T1_1P1_1T1_2P1_2)rnrorr*trArr s rxtest_onesamplezTestStudentTest.test_onesampleIs"  ,C HX> , JJ~'J K&.rl 2A$$Q+DAq  , , 2::bff-.2::bff-.  DGG!4b912::dii012::dii01 477 3R8, C3  DGG!4b912::dii012::dii01  DGG!4b912::dii012::dii01  DGG!4b912::dii012::dii01; , , , ,s#LAK;1L;L LLct|s^|jddd|jg}d}tjt |5t j|dddddyt jjdd d d }tj|d <tjd5tt j|dtjtjftt j|dddttt j|ddttt j|dddddy#1swYyxYw#1swYyxYw)Nrrr%Use of `nan_policy` and `keepdims`...rrrr6rY3itrrrrrrrrL)g̈́^Bg/kCm?rr)r+rdrmrrNotImplementedErrorrfrrrrrr rrr)rnrorqrs rxtest_onesample_nan_policyz)TestStudentTest.test_onesample_nan_policyks| BB/0A=G2'B <!!!RF; <  JJNNq'N J" [[ * / u00C8266266:J K %e&7&736&R&Q S *e&7&7CG T *e&7&7C%- /  / /  <  / /sE$6B%E0$E-0E9z+ignore:divide by zero encountered in dividecd}tjt|5tj|j |j dddddtj|j |j dd\}}t||j |jt||j |jtj|j |j dd\}}t||j |jt||j |jy#1swYxYw) N8`alternative` must be 'less', 'greater', or 'two-sided'.rrrrhrrbrd) rrrrfrrdrr1P1_1_lr P1_1_g)rnrorrrAs rxtest_1samp_alternativez&TestStudentTest.test_1samp_alternativesL ]]:W 5 L   bjj127 K L  DGG!4bfM12::dkk232::dii01  DGG!4biP12::dkk232::dii01 L Ls 2EE jax.numpyGeneric stdtrit mutates array.rrirctjjd}d}|j|dd}|j}|j dj }|j ||}|j ||}dd gd tj gtj d gd }tj||| } | jd} t| j|j ||dt| j|j ||dt| j|j |dz y)Nl ?@C ,1"i/!Q..t.4g;TRTZZ  $oobjj5 jCUK C(rzN) rrrrrrrr r r rr r rrrrrrrrrrr'r*r/rgivenrfloatsrr0rrzrxrr6sl B B D D D D D D D D AXF $(^F [[ WX [[ VW2XY2@/( [[ MN 2O 2k*JK [[],LM1NL109d;7 [[ZJ1188H 8 :< [[]VY,?@)A<8 )rzrc eZdZdZej j dgddZej j dgddZej j dgddZ ej j dgd d Z ej j dgdd Z ej j dgdd Z ej j dgd d Z ej j ddgdfdgdfdgdfdgdfgdZej j ddgdfdgdfdgdfdgdfgdZdgdej"fdej"gdej"fdej"ggdej"ej"ej"gfdddgddej"gddej"gfd ddej"ggdgd!fd ddgd"dej"gd"dej"gfd ej"ej"ggdej"ej"ej"gfgZej j d#ed$Zd%ddd&ej"ggd'd(fd%gd'ddd&ej"gd(fgZej j d)ed*Zej j d+gd,d-Zy.)/TestPercentileOfScorec,tj|i|Sr)rfpercentileofscore)rnargskwargss rxfzTestPercentileOfScore.fs&&777rzz kind, result))rjr)ri#strictr)weakrcHgd}t|j|d||y)N rr3r4r5r6r7r8r9r:rYr5kindrr9rnrArqr*s rx test_uniquez!TestPercentileOfScore.test_uniques" ,TVVAqtV,f5rz))rjr)rirr;)r=rcHgd}t|j|d||y)N) rr3r4r5r5r6r7r8r9r:r5r@rBrCs rxtest_multiple2z$TestPercentileOfScore.test_multiple2" +TVVAqtV,f5rz))rjr)rirr;)r=rdcHgd}t|j|d||y)N) rr3r4r5r5r5r6r7r8r9r5r@rBrCs rxtest_multiple3z$TestPercentileOfScore.test_multiple3rGrz))rjr)rirr;)r=rcHgd}t|j|d||y)N) rr3r4r6r7r8r9r:rYrZr5r@rBrCs rx test_missingz"TestPercentileOfScore.test_missings" -TVVAqtV,f5rzcHgd}t|j|d||y)N rYrrrrrdFrgrGrrr@rBrCs rxrz(TestPercentileOfScore.test_large_numberss" 6TVVArV-v6rzcHgd}t|j|d||y)N) rYrrrrrrrdrNrgrr@rBrCs rxtest_large_numbers_multiple3z2TestPercentileOfScore.test_large_numbers_multiple3s" 5TVVArV-v6rzcHgd}t|j|d||y)N rYrrrrdrNrgrGrnrr@rBrCs rxtest_large_numbers_missingz0TestPercentileOfScore.test_large_numbers_missings" 7TVVArV-v6rzrj)rrYrrri)rr6_rr<)rrrGrr=cLgd}t|j|gd||y)NrR)rrYrSrr@rBrCs rxtest_boundariesz%TestPercentileOfScore.test_boundariess# 7TVVA0tVBRVV3D Eq!RVVni >q!fq!RVVnq"bffo >rvvrvv& BFFBFFBFF3K L  NE [[7?B@B 1aBFF#Y ( * )aArvv. ( * E [[8%@0A0 [[W' >  >rzr4Case)f_obsf_exprrTchi2rmod_logcr)r5r9r[r9r5r3r<r[r=r9rrgfx(@)r3rr[r3rrr?g{T4@)r4r6r8r:rrrc<eZdZdZej j deej j dgddZdZ dZ ej j de ej j dgdd Z d Z d Zd Zy )TestPowerDivergencec$|jdj}|j||}||j||n|}| t|} n)||j||n|f} | dj|} t 5} | j tdtj|||||\} } t| |j|||dk(s|dk(r9tj||||\} } t| |j||dddtj|}tjjj|| dz |z }t |j||y#1swYlxYw) NrrHrrrfrrTlambda_rpearson)rfrrT)rdrIr.broadcast_arraysrr rrrfpower_divergencer1 chisquarer distributionsrgsf)rnrerfrrTrp expected_statrorInum_obsrrstatrA expected_ps rxcheck_power_divergencez*TestPowerDivergence.check_power_divergenceysz 2$$ 5 .272C 5 . <enG;@;Lb))%7!8 Qiood+G  NC JJ~'< =,, %U%)7?! N Ns BFFcase lambda_stat))Nrgrqrg)rrglog-likelihoodrmod-log-likelihoodrh cressie-readrirric |\}}t||}|j|j|j|j|j |||yr)rEr{rerfrrTrnr|r}rorpattrrws rxr5zTestPowerDivergence.test_basicsG $ d+  ##DJJ DIItyy$+]B @rzc td}td}tj|j|jf}tjtj|jtj |jz|j f}|j|}||j|n|}|j||ddd|j|jg||j||ddd|j|jg||j||ddd|j|jg||j||ddd|j|jg||j|j|jd}|j|dddd|j|y) Nrrrqrrrrr])power_div_1d_casesrvstackre ones_likerirfrdr{rgrrhrirh)rnrocase0case1rerf f_obs_reshapes rxr@zTestPowerDivergence.test_axiss"1%"1% 5;; 45 2<< 4RWWU[[5II ;;() 5!%*%6 5!E ##eQ5::uzz2r $ ; ##eQ%))UYY!7B $ @ ##eQ#emmU]]%C $ L ##eQ%((3 $ <  2::ekk#:FC  ##dAt%** $ -rzctd}td}tj|j|jfj}tjtj |jtj |jz|jfj}|j|jg}|jdj}|j||}|j||}|j||}|jdgdgg}tj|||\} } t| |tj|||d\} } tj|||d\} } |j| |jddf| |jddffd}t| |y) NrrrrHrrrQrS)rrrrerrrirfrgrdrIrfrsr1rr)rnrorrrerf expected_chi2rIrryrArvp0p1rzs rxtest_ddof_broadcastingz*TestPowerDivergence.test_ddof_broadcastings #1%"1% 5;; 4577 2<< 4RWWU[[5II ;;())* UZZ0  2$$ 5 . 5 . = > zzA3*%((DAam,&&ue$t*E2&&ue$t*E2YY2::q= 12bjj!m3DEAYN :&rz)r~rrrrc |\}}t||}tj5|j|j|j |j |j|||dddy#1swYyxYwr)rErxryr{rerfrrTrs rxtest_empty_casesz$TestPowerDivergence.test_empty_casessh $ d+  $ $ & ,  ' ' DJJ 499 , , , ,s AA//A8cLtdj}tdj}tdj}tdj}|j dj }|j ||}tj||||d}d}t|||y)NrrrHrqrorr) rrerfrrTrdrIrfrsr) rnrorerfrrTrIrr s rx'test_power_divergence_result_attributesz;TestPowerDivergence.test_power_divergence_result_attributess"1%++"1%++!!$))!!$)) 2$$ 5 .$$U%d*. C, C3rzcL|jddgddgg}|jddgddgg}d}tjt| 5t j ||jdd g dddtjt| 5t j ||d dddt j || \}}t ||jddgt ||jddgy#1swYxYw#1swYjxYw)Nrr>@rLrtgA@g9@For each axis slice...rN@rfr)rfrTg̑m@gUUU@gj^4;?g4;?)rdrrrrfrsr1)rnrorerfrryrms rxtest_power_divergence_gh_12282z2TestPowerDivergence.test_power_divergence_gh_12282s S#Jc 34 RISz23* ]]:W 5 H  " "5 C:0F G H ]]:W 5 ?  " "5A > ?++E? dbjj*j)ABCbjj*j)ABC  H H ? ?s )DDDD#c |jgd}d}|jd|jddz}|j|j ||j |j ||zz }|j |||zz}|j ||jddf||jddffj}|jgd}|j|d}t|jdD]G}||df||df} } tj|dddf|dddf| \} } t| | d Iy) N)rtr8,@g1@rLr8rrrrrrr8rrrrrg*kqrr) g$g@rgr@rgffffffP@ggLD@rgA@r.g=@rg:@rg8@r<gffffff7@gq= ףp?g7@rg333336@r=g6@rgfffff6@rg8@rLgA@rgj@)r_r3r)rpg{Gzt?rj)rdrrrr"exprrrrhrrfrsr1) rnroobsbetarr+expected_countstable4table5rprwryrAs rx/test_power_divergence_against_cressie_read_datazCTestPowerDivergence.test_power_divergence_against_cressie_read_data sPjj@A IIb#))A,+ ,rvvc{RVVBFF46N%;;<&&a0C A .+BJJM:<==>Q &FG,v||A' GD! D-d ;  AA"c6tjjd}d}|jdd|}|j |}|j |}|j ||j }tj||dd }||z |j|z }|j|d z}t|j|t|j ||j } t|j| j|y) Nl*m{mrYrrrrHr_F)rfr sum_checkr3)rrrrmpoissonrdrIrfrtr'r"r1rr%rrv) rnrorrUlamrqrrrrs rxtest_chisquare_12282bz#TestChisquare.test_chisquare_12282bFsii##$45 kk$1k- KK jjo JJq J *ooasuE W $FF1a4L  y1 AY__= > BEE)$45rzzn, dtype)rr{)i@Brc zt||}|j|dg|}|j|dz|dzg|}tj||}|\}}t ||j||jdj dt |j|t |j|y)NrrHr3rvIh%<=rj) rErdrfrtr1rIr2rr) rnrUrIrorrrryrAs rx#test_chiquare_data_types_attributesz1TestChisquare.test_chiquare_data_types_attributesYsE"jj!Quj-jj!q&!q&)j7ooc3'abjj"**R.2F2FjGeT t, A&rzN) rrrrrrrrrrrzrxrr;s806& [[Z.:L)MN 'O 'rzrc tgdtgdtgdtgdg}tgdtgdtgdtgdg}tgd tgd tgd tgd g}ttj|d |d|d|ddttj|d |d|d|ddttj|d |d|d|ddt t tj|d |dd}tj|}t ||ttj|d |d|d|ddttj|d |d|d|ddt t tj|d |dy)N)g"~j?g^I +?gI +?gjt?gMbX9?gʡE?x&?-?g?rGz?g+?gv?g9v?)g~jt?gx?gOn?rg"~j?n?gK7A`?rg7A`?rbX9?gZd;O?g1Zd?gK?)gy&1?gzG?g`"?g!rh?g'1Z?gZd;O?grh|?gjt?gʡE?rgzG?gS?gJ4?gʡE?)gK7?gh|?5?gl?g/$?g#~j?rgQ??g?rrg rh?g333333? ףp= ?) r5r4r6r4r6r4r3r6r5r5r5r4) r3r3rr3r4rr3r4r3rrr4) r3r5r4r4r5r4r4r5r5rr3r) r4r6r5r4r5r5r4r4r4r5r5r5)rg#@rDrg$@)g333333@g@g@r?g@)rrrAr? @)g!@g!@333333 @ffffff @g333333"@rrr3r4)gNt$@gBv?)g __2@g5"]i2?)g\(\%@g+ԋ?r)rrrffriedmanchisquarerrrr)rrrr rs rxtest_friedmanchisquarergs 8 9 9 : : ; 9 :  ;B ) * ) * ) * ) * ,B & ' % & % & % & (B e55beBqE"Q%1NDFe55beBqE"Q%1NFHe55beBqE"Q%1N9;*e55beBqEB)J  ! !2 &CZ(f66r!ube79!ubeEDF f66r!ube79!ubeE9;*f66r!uRUCrzc4eZdZdZ ddZddZdZdZdZy) TestKSTestzLTests kstest and ks_1samp agree with K-S various sizes, alternatives, modes.ctj|d||}tj||g}t tj|||yNrrirKr)rfkstestrrr rnrqriexpected_statistic expected_probrKrrqrs rx_testOnezTestKSTest._testOnes?a[tL88/?@!"((6"2HgNrzctj|d||}tj|tjj||}t t j|||yr)rfrks_1samprcdfrrr)rnrqrirKrrq result_1samps rx_test_kstest_and_ks1sampz#TestKSTest._test_kstest_and_ks1sampsKa[tL~~a2=DJ !"((6"2L'Rrzcztjddd}d}tj|d}t ||y)Nr_rr:rr)rrrfrrrnrqr rs rxtest_namedtuple_attributesz%TestKSTest.test_namedtuple_attributess3 KKAq !, ll1f%C,rzc$tjddd}|j|dtjddd}|j|dgd}|j|d|j|dd |j|d d y) Nr_rr:r`r rQ?g333333g(\?Q?g(\ſg{GzgHzG?g\(\gGzrdrrKrb)rrrrCs rxtest_agree_with_ks_1sampz#TestKSTest.test_agree_with_ks_1samps KKAq ! %%a5 KKR # %%a5 N %%a5 %%a%A %%ag%>rzctj ddtjg}tj|tjj }tj|tjj d}tj tj|sJt||tj|jrJtj|jrJy)NrrT)_no_deco) rrrfrcauchyrrrrr4rr)rnrMrrs rxtest_pm_inf_gh20386zTestKSTest.test_pm_inf_gh20386sArvv&ll4!1!12ll4!1!1DAvvbkk#&'''S#88CMM***88CJJ''''rzNautorW) rrrrrrrrrrrzrxrrs%V')O S - ? (rzrceZdZdZ d dZdZdZdZdZe jjde je jge jjdgd d Zy ) TestKSOneSamplezY Tests kstest and ks_samp 1-samples with K-S various sizes, alternatives, modes. ctj|tjj||}t j ||g}t t j |||y)Nrr)rfrrrrrrrs rxrzTestKSOneSample._testOnesH5::>>{QUV88/?@!"((6"2HgNrzctjddd}d}tj|tjj }t ||y)Nr_rr:r)rrrfrrrrrs rxrz*TestKSOneSample.test_namedtuple_attributess; KKAq !, nnQ /C,rzc8tjddd}|j|dddtjddd}|j|dd d gd }|j|dd d |j|dd dd|j|ddddy)Nr_rr:r`g |N?g7.s?rrg{CTp?g M)r?Tgv(?)rrjTgT ?)@g\(\?Fr)r?Fg (>)rgQ?Fg@c?)r?FgX+ʳP?rmr4)dtypes) rrr4rrmrrrdrcheckintrbool)rndatasets rxtest_ks1samp_allpathsz%TestKSOneSample.test_ks1samp_allpathssD123 :-E F " CD ! "Q-./**  4 '9a066sE4>P6Q= " "s CCksfunc*alternative, x6val, ref_location, ref_sign))rdr7r7r)rbr8r8r_)r`r7r7r)r`r8r8r_ctjddz}||d<tjdj}||||}t |j dd|j|k(sJ|j|k(sJy) NrYr<r7r}rhr r#rj) rrrfrmrr rstatistic_locationstatistic_sign) rnrrix6val ref_locationref_signrqrrs rxtest_location_signz"TestKSOneSample.test_location_signsy IIbMC !mm"%))Q5 s7%%555!!X---rzNr)rrrrrrrrrrrrrfrrrrrzrxrrs| ')O - UP!RF [[X enn'EF [[I78 . 8G .rzrceZdZdZ ddZdZdZdZdZdZ dZ d Z e jjd Ze jjd Zd Zd Ze jj&dZe jj&dZe jjdZdZe jjdZe jj3dgdgfdggfggffdZe jj&dZdZe jj3dej<ej>ge jj3dgddZ y)TestKSTwoSampleszs#s$) fe['3? feY= feVWc: fe['3? feY= feVQ4rzcptjddg}|dz}|dz }tjgd}|j||ddd|j||ddd |j||d d d |j||dd d|j||dd d|j||d ddy)Nrrrrr`r<g?rdg?rbrj皙?r:rrrr r s rx testTwoVsFourzTestKSTwoSamples.testTwoVsFourJs#s$-. fe['7C feY@ feVWg> fe['6B feY@ feVQ4rzcNtjddd}|dzdz}|dzdz }|j||ddd|j||ddd |j||d d d |j||dd d |j||dd d|j||d d d y)Nrrr3r r`r2g~z?rdgz;.B?rbrr{Gz?gn2IU?r)rnx100 x100_2_p1 x100_2_m1s rx test100_100zTestKSTwoSamples.test100_100Ws{{1c3'1HsN 1HsN  dI{I?QR dIy)=OP dIvq#6 dI{IsC dIy)=NO dIvq#6rzc|tjddd}tjddd}|dzdz}|dzdz }|j||ddd|j||d dd |j||d d d|j||dd d|j||d d d|j||d ddy)NrrrSrr r`g?g@߿?rdg}n?rbrg:[?g]O );?gѫ);?rrr)rnrx110 x110_20_p1 x110_20_m1s rx test100_110zTestKSTwoSamples.test100_110bs{{1c3'{{1c3'BY_ BY_  dJ \CWX dJ cJ !qA37*aS2X5a?sK!rQC!G+qcBh6!q@L eUK9KL eUIv7JK eUFFC8 eUK>PQ eUI{ eU1Wi= eU1WfeR8 eU3Y UC@ eU3Y 5$? eU3Yr: eU3Y UC@ eU3Y 5#> eU3YtT"  $ aK)<>T"  $rzcHtjjd}|jd}|jddz}|j ||dddd |j ||ddd d |j ||d ddd |j ||dddd y)Nr6r2ri'r=r`g _r!?gAD5rrr7rrdgלN#y7rbgvqw?r8r9s rxtest_gh11184_biggerz$TestKSTwoSamples.test_gh11184_biggersii##F+ JJEJ " JJEJ "S ( aK)<>T"  $ aK)<>T"  $ aI':EEENN5%g>EEE eQY / eQY / 0rzc hd\}}|dz}d|z |z dz dz }tjdd||z }tjdd|}|j||dd|z d d |j||dd|z d d |j||dd|z d d |j||dd|z d|j||dd|z dt5}d}|j t ||j||dd|z dd |j||dd|z dd dddy#1swYyxYw)N)r2i*r8rr3rrr`g@gkHY?rrgLɔ.?rrrdg.LbG2?rbrgr?֎?r&)rrrr rr) rnr)r*r3r+rqrrrrs rx testLargeBothzTestKSTwoSamples.testLargeBothseBgBr ! A  KK3 #e + KK3 # aK6H"  $ aK6H"  $ aK6H!  # aIus{4FG aFD3J0BC   (CAG JJ~w / MM!Q 53;8J&  ( MM!Qs 4F&  (  ( ( (s A D((D1cRd}tjddgdg}t||y)Nrrr3r4)rfrrrnr rs rxtestNamedAttributesz$TestKSTwoSamples.testNamedAttributess', nnaVaS)C,rzc ddlm}m}|dddd|ddddtjd5t t |dd ddt t |d ddddddy#1swYyxYw) Nr)_count_paths_outside_method!_compute_outer_prob_inside_methodrrirrr/iKr)scipy.stats._stats_pyrHrIrrrFloatingPointError)rnrHrIs rxtest_some_code_pathsz%TestKSTwoSamples.test_some_code_pathsst *!Q15#D!Q5 [[ ) , ,.Ia , ,.Ia , , , ,s )A''A0r|rc tjtt5t j |}t |jtjt |jtjdddy#1swYyxYwrJ) rr>rrrfrrrrrmrrnr|rs rxtest_argument_checkingz'TestKSTwoSamples.test_argument_checkings[\\,4I J -..$'C  / RVV , - - - ABB c*tjjdd}tjj |dd}|dz}tj ||ddtj ||d dtj ||d dy ) zEnsure gh-12218 is fixed.Nai rrr rdrrrbr`N)rrr=rfrmrr)rnr)rvs1rvs2s rx test_gh12218zTestKSTwoSamples.test_gh12218'ss x }}  bb :ax tTywG tTvGD tT{IrzcVtjjd}|jddz}|jd}d}tjt |5t j||d }t|jd d dddy#1swYyxYw)Ni{fr=iqrr<iqz(ks_2samp: Exact calculation unsuccessfulrrbrhrr+r$) rrrrr>rrfrr r)rnrrSrWrrs rxtest_warnings_gh_14019z'TestKSTwoSamples.test_warnings_gh_140194sii###2  $s*  $< \\. 8 7..6BC CJJ 6 7 7 7s %1BB(rr))rd皙@rYr)rbffffff@rr_)r`rYrYr)r`rZrr_ctjdtj}|j}||d<t j |||}|j dk(sJ|j|k(sJ|j|k(sJy)NrYrHr7rhr ) rrrrarfrrrr) rnrrirrrrqrrrs rxrz#TestKSTwoSamples.test_location_sign@sx IIb + FFH!nnQ{;}}###%%555!!X---rzN)r)!rrrrrrrrrrr"r$rrrr,r0r4r:r/r<rArCrFrLrrOrUrXrfrrrrrzrxrr-sF> 4 5 5 7 : L = [[226 [[228 K$ [[ $ $ [[ 0 0 [[((.-  [[,,  [[VrA3i1#rRH%EF-G- [[ J J 7 [[X enn'EF [[I;< . <G .rzrc $d\}}|| g||gf}tjddd}tjddd}tjtjdddtjdddg}tjtjdddtjdddg}tj||d\}}t ||g||ftj|j |j d\}}t ||g|tj||d\}}t ||g|t5} tjdd 5| jtd tjd d \}}ddddddttj|ttj|d } tj||d} t| | tj|||g} tj|||g} tj| | d\}}t tj||t tj||t!|j"dtjtj$| ddtj$| ddd\}}t tj||t tj||t!|j"dt't(tj||dtj||dd\}}t+|d|dz z t+||tj||dd\}}t+||dz t+||tj,j/d}tj0j3ddd|}tj4|d<tj0j3ddd|tj0j3dd|z}tj4|d<tjd5t7tj||tj4tj4fdddt tj||d d!t't(tj||d" t't(tj||d# t9j:td$%5tjgd&gd'\}}dddt!tj||ftj<dftjd5t!tjgd&gd&tj4tj4ftjdtj4gd(dgg}t!tj|tj>d)dtj4gdtj4gfdddtj@d*}t't(tj|jCd+|jCd,d-}tjD|}tj4|ddd.d/f<tj4|ddd0d1f<t9j:tFtH%5tj||dd \}}dddt9j:tFtH%5tj||ddd2\}}dddt+||d34tjd5t+||||dd34dddt9j:tFtH%5tj||ddd2\}}dddt+||d34tjd5t+||||dd34dddy#1swYxYw#1swYxYw#1swYxYw#1swY3xYw#1swY0xYw#1swY\xYw#1swY$xYw#1swYxYw#1swYxYw#1swYyxYw)5N)gu?gȵI?rrrgKX@rrSrrrrrrrr3r\rrhrbrrdrRr6rYrrr)rrrrrr)ghm#1?gry̧?rrrrrkr<r_r]r)r9r4r3r4r5cD|dkr|dk(s |dkDr |dk(r|dz Sd|dz z SNrrbrdr3rrrrAalts rxconvertztest_ttest_rel..convert3 EcVmQ3)3Cq5LAE{rzrrrrrr+rj)%rrrrf ttest_relrrr rrrrr4rr^absrrr_rrr rrrrrmr rr>rrrrh vectorizerr)trrtprrSrTrvs1_2Drvs2_2DrrArr rrvs1_3Drvs2_3Drrqrrananrc converters rxtest_ttest_relrpSs 4EBr s8RG C ;;qS !D ;;tF3 'Dhh Ac#. D0LMNGhh D4bkk!C6LMNG //$1 -CAaqeRG, //'))WYYQ 7CAaqeC( //'7 3CAaqeC(  ' KK :' >#FGr2&1'' BHHQK BHHQK)J //$1 -CZ(ii12Gii12G //'7 3CAabffQi,bffQi,&! ??2;;w15;;w15 ! #DAqbffQi,bffQi,&!*eootTwO ??4A6 BDAqAq2a4x Ar ??4A9 EDAqAr!tAr ))   )C 1BSsCA VVAcF ARc D c# > ?A VVAcF X &D5??1a02662662BCDeooavFGI*eooq!H*eooq!I n,E F5y)415"&&)Q"&&!- X &1U__Y :RVVRVVrrrfrer _stats_py TtestResultrrrm)rqs rx#test_ttest_rel_empty_1d_returns_nanrysj (0E F)R() feoo99 :: :"&&"&&)*))s BBb, expected_shaperr6rrrrrr4rctjd}tjj5}|j t t tj||d}dddttjjsJtj|tj}t|j|t|j |y#1swYxYw)Nr4rrr_rSr-)rrrr rrrrfrerrwrxrrmrrr)r4expected_shaper*rrqexpected_values rxtest_ttest_rel_axis_size_zerors A  % % '03 %'<=AB/0 feoo99 :: :WW^?N!!>2/00s 3C''C0cTtjd}tjd}tj||d}t |tj j sJt|jjdt|jjdyN)rr9rr6r9rrrSr6r) rrrfrerrwrxrrrr)r*r4rqs rx test_ttest_rel_nonaxis_size_zerorsv A A __Q *F feoo99 :: :!!''0$$f-rzrirctjjd}d}|j|dd}|j|dd}ddgdtjgtj d gd }t j ||| }|jd }t|||t|j|dz y)NlC67l: rYr=r3r gZ?Ygi{a?g8gC?r!rhr#r$r) rrrrrrfrerTr rr&)rirrUrqrrrrrrs rxtest_ttest_rel_ci_1drs ))   3 4C A s! ,A q *A,->?)2662VVG01 3C //!QK 8C $ 7BBK()1rzztest_fun, argsrYc||}d}tjt|5|jddddy#1swYyxYw)Nr)rrYr$)rrrrT)test_funr7rrs rxtest_ttest_ci_ivrsF D/CDG z 15 4555s ?ArcV t||ndfd }||||||zS)Ncj|}j||}j||d}|j|}|||fS)NrSrr)rdrirr)rqrTmurnobsros rx_statsz_desc_stats.._stats+sO JJqM WWQTW "ffQTaf0wwt}3}rzrY)r()rrrTrors ` rx _desc_statsr(s5$&JR BB "d fR. ..rzc |jd}|jd}|j|| g}|j||g}|jddd}|jddd}|j||g}|j||g}tj||d} | \} } t | |t | |tj t||} | \} } t | |t | |tj|j|jd} t | j|t | j|tj t|j|j} t | j|t | j|tj||d} t | j|t | j|tj t||d} t | j|t | j|d } tjt| 5tj||d dddt|j|j} tjt| 5tj | d d idddtj||d \} } t | d|dz z t | |tj||d \} } t | |dz t | |tj|j|jdd}t|j|j} tj | d di}t |j|jt |j|jtj|j|jdd}t|j|j} tj | d di}t |j|jt |j|j|j|j}|j|j!|j"ddk(||}tj||d} t%| j|t%| j|tj t||} t%| j|t%| j|y#1swYxYw#1swYxYw)N ru?g8Mѱ?r6irrrrSrrrrhrirbr3rdr)rdrrrf ttest_indr1ttest_ind_from_statsrrrrrrrrmrrrr2)rorhrtr_2Dpr_2DrSrTrjrkrrrArr7rrNaNs rxtest_ttest_indr5s& & 'B ' (B HHb2#Y E HHb"X E ;;q#s #D ;;q#s #Dhhd|$Ghhd|$G //$1 -C DAqArAr  $ $k$&= >C DAqArAr //'))WYYQ 7CCMM5)CJJ&  $ $k'))WYY&G HCCMM5)CJJ& //'7 3CCMM5)CJJ&  $ $k'7&K LCCMM5)CJJ& IG z 1; d':; wyy')) ,D z 1A ""D@@A ??46 :DAqAqBqDz"Ar ??49 =DAqAr!tAr ??799giiaV LD wyy')) ,D  % %t @ @DDNNDNN3DKK- ??799giiaV LD wyy')) ,D  % %t @ @DDNNDNN3DKK- **RVV C 88BIIdjjm,13 =D //$1 -CCMM3'CJJ$  $ $k$&= >CCMM3'CJJ$M;;AAs#U*?U7*U47Vc ^tjddd}tjddd}tj||g}tj||g}tj|||g}tj|||g}tjj d}t jjddd|}tj|d <t jjddd |}tjd 5tt j||tjtjfdddtt j||d dttt j||d ttt j||d t!j"t$d5t jgdgd\} } dddt'tj(  ftj*dftjd 5t't jgdgdtjtjftjdtjgddgg} t't j| tj,ddtjgdtjgfdddtj|ddddddf<tj|ddddddf<d} tj.| } t j||dd \}}t j||dd d\} } t1| |d t1| | ||dd t j||dd d!\} } t1| |d t1| | ||d!d y#1swYxYw#1swYxYw#1swYxYw)"Nr6rrrrRrYr]rrrrrr)guͷ?g*,?rrrrrkr<rr_r]rr7r[cD|dkr|dk(s |dkDr |dk(r|dz Sd|dz z Sr`rras rxrcz*test_ttest_ind_nan_policy..convertrdrzrSrb)rTrrir+rjrd)rrrr^rrrfrrrmrr rrrrrr>rrrfrrrgr )rSrTrjrkrlrmrrqrrrrArnrcrorhrs rxtest_ttest_ind_nan_policyrs% ;;q#s #D ;;q#s #Dhhd|$Ghhd|$Gii'734Gii'734G ))   )C 1BSsCA VVAcF 1BSsCA X &D5??1a02662662BCDeooavFHJ*eooq!H*eooq!I n,E F5y)415"&&)Q"&&!- X &1U__Y :RVVRVV#FGr2&1'' 88A;; 88A;; ''''s"B0BBB BB&z&ignore:Arguments...:DeprecationWarningcreZdZdZej j dejejdezdzej j dezdzfZ ejejedzdzej j edzfZ ejdZ ejddzZ ddgZ ddgZej j dej j#d dd j%dd j&Zej j#d dd Zd dgZd dgZgdZe e ddiefe j&e j&ddiefe dddfe dddfddiedfe dddfj3e dddfj3ddiedfe e dddefe e ej j5dddefe e ddid feedddefe eidfe e ej j7dddefg Zej<j>ej<jAdedZ!dZ"dZ#dZ$dZ%ej<jMdZ'd Z(d!Z)d"Z*ej<jAd#gd$ej<jAd%d&d'gd(Z+ej<jAd#gd$ej<jAd%d&d'gd)Z,d*Z-e.dd+,d-Z/y).Test_ttest_ind_permutationsrrr4r5rrYrr3r6rrr9g'^P?gCq?g$R?)g=B?g?goŜ?gp?g'^?rTN)rrT equal_varTr_)rTrr3za,b,update,p_dcddd}ddddd}|j||j|tj||fi|\}}tj||fi|\} } t|| dt| |y)NFrTrrr)rTr permutationsrr6)updaterfrr) rnr*r4rp_d options_a options_pstat_arvstat_prs rxtest_ttest_ind_permutationsz7Test_ttest_ind_permutations.test_ttest_ind_permutationss"6 !%)1>   OOAq6I6 A;;!&&!4!&#.rzctjjd}d}|jd|d}|jd|d}ddd}|j dt j ||fi|}t j ||fi|}|j d t j ||fi|}t j ||fi|} |j d t j ||fi|} t j ||fi|} t|j|jt|j| jt|j|j t|j| j t| j| j t| j| jt|j| jt|j|j|jd k} t|j| | j| z| j| t|j| |j| z| j| y) Nrr4r3rr)rTrrdrhrbr`r< rrrrHrrfrrrr) rnrNr*r4rres_g_abres_g_bares_l_abres_l_bares_2_abres_2_bars rx test_ttest_ind_exact_alternativez Y/??1a595V,??1a595[1??1a595 X__x6In59:: < #%HOOD11 -D :Qxu556 .T ;HOOTE22 .T ;Qxt445 -D :rzctjjd}d}|j|d}|j|d}tj|d<tj|d<tj|d<tj|d<ddd}|j d t td 5tj||fi|}dddt5}|jtd |j d tj||fi|}tj|jdtj|jdz}tj|dd|f|dd|ffi|} t|j |tjt|j"|tjt%|j || j t%|j"|| j"tj|j'|j'fi|}tj|j sJtj|j"sJ dddy#1swYxYw#1swYyxYw)Nrrr6r6r)r9r3)r:r4rrrrrrrrRrS)rrrrHrmrrrrfrr rrr4rrrrr r) rnrrr*r4rrrrrs rx$test_ttest_ind_permutation_nanpolicyz@Test_ttest_ind_permutations.test_ttest_ind_permutation_nanpolicys ii##A&  HHQN HHQN&&$&&$&&$&&$%)1=  G, :-L M 5//!Q4)4C 5  +C JJ~'7 8     4//!Q4)4C88A;???*RXXa[__!_-DDD??1QX;!dU( IyID D)266 2 t,bff 5 CJJu-t{{ ; CMM4%0$.. A//!'')QWWYD)DC88CJJ' ''88CMM* **# + +  5 5 + +s:J+$F=J8+J58Kcttd5tj|j|j ddddttd5tj|j|j ddddttd5tj|j |jddd dddy#1swYxYw#1swY^xYw#1swYyxYw) NzPermutations must berr`rr=z'hello' cannot be usedrhello)rrrT)rrrfrrb2r*r4r8s rx'test_ttest_ind_permutation_check_inputszCTest_ttest_ind_permutations.test_ttest_ind_permutation_check_inputss :-C D ? OODGGTWW2 > ? :-C D @ OODGGTWW3 ? @ :-E F : OODFFDFF)0q : : :  ? ? @ @ : :s#-C-C"/C.C"C+.C7cd}tjj|d}tjj|d}tj||dj }t d|vd|vsJy)NrYrrrr)rrrHrfrrprint)rnrr*r4p_valuess rx)test_ttest_ind_permutation_check_p_valueszETest_ttest_ind_permutations.test_ttest_ind_permutation_check_p_valuessb  IINN1b ! IINN1b !??1aa8?? c!"("""rzrirr)r[)r3r[ctjjd}|j|}|jd}td}tjjd}t j dd|i|}t j ||d|| }d } tjjd}t j||f| fd||d |} t|j| jt|j| jt j ||d| } t|j| jt|j| jd y)NNvf_aBrr\r(N\Sju rr_rTrircFtj|||jSNrSrfrrrqrrrTs rxrzFTest_ttest_ind_permutations.test_permutation_method..statistic%//!QT*J*T*T#Trz)rTrrirr2rjr) rrrrrfrrpermutation_testrrrr ) rnrirrrqrrr8rrrrs rxtest_permutation_methodz3Test_ttest_ind_permutations.test_permutation_methods2ii##$45 JJEJ " JJBJ #&ii##O4((;S;F;ooaVT Uyy$$_5$$aVYHRS1<H@FH S]]CMM2SZZ,ooaES]]CMM2 CJJT:rzc8tjjd}|j|}|jd}td}d}tjjd}|j|jg}t j ||ff||d|d |} tjjd}|j|jg}t jdd |i|} t j||d|| } t| j| jt| j| jt jdd di|} t j||d|| } t| j| jt| j| jt j||d| } t| j| jt| j| jdy)Nrrr\rrcFtj|||jSrrrs rxrzFTest_ttest_ind_permutations.test_monte_carlo_method..statisticrrzrr_)rrrTrirrrrrrjr) rrrrrrfmonte_carlo_testrrrrrr ) rnrirrrqrrr8rrrrrs rxtest_monte_carlo_methodz3Test_ttest_ind_permutations.test_monte_carlo_methodsii##$45 JJEJ " JJBJ #& Uii##O4""C$7$78$$aVH PR1<H@FHii##O4""C$7$78'':C:6:ooaVTS]]CMM2SZZ,''FOFvFooaVTS]]CMM2SZZ,ooaES]]CMM2 CJJT:rzcd}tjt|5tjgdgdddddy#1swYyxYw)NzC`method` must be an instance of `PermutationMethod`, an instance...rr}rP migratoryr)rrrrfrrs rx test_resampling_input_validationzW ]]:W 5 F OOIy E F F Fs AA Uses NumPy for pvalue, CIcpu_onlyrc|j|j|j|j}}d}t|rt j ||dn?@A 9299QT?S("))*:*:1a4*@ABA 2B 2 B QB QBIINN1 ::>>ar"  $$+GCOAA ::>>ar> 4D < C~&G+G A{C accFA;$ 1a4!AqD'FD>3q62 1a4 1QT7>>+fd^SVD A1-s3 A 5 5a 8!DcJ R+t$f- tb!4g> RS A 5 5a 8!DgN  F [[ [[-v6 /7 /(-T *'&<: [[::B"+H:# [[],LM [[Wug&67;8N;4 [[],LM [[Wug&67;8N;BF t,GH7I7rzrceZdZejj ejj ddddddiiggdejj d d d gd d gd Zejj ddddddiiggdejj dddgdZy)Test_ttest_ind_commonkwdsrrrtrimr)rrbasic)idsrTF unequal_varc Rtjjdtjjdddddd}tjjddddd}t j ||fd d i|}d \}}}||dd|dddddf} |ddddd|ddf} t j | | fd d i|} t |j|dd||ddf| jt |j|dd||ddf| jtjtj|d d d} tjtj|dd d} | jdd}tj|}tj|}td|DD]C}| |}| |}t j ||fd di|}|j||<|j||<Et||jt||jy)Nrr6r5r8rr7r9r3rTr`)r3r4rrrrrrr3rr_r6rr5rrrc32K|]}t|ywrr.0rs rx z=Test_ttest_ind_common.test_ttest_many_dims..As 9aq 9)rrr=rHrfrrrrr_tilerrrr )rnrrr*r4rrrWrJrrrrqrrr statisticspvaluesindicesxiyirs rxtest_ttest_many_dimsz*Test_ttest_ind_common.test_ttest_many_dims"s q IINN1aAq! , IINN1aAq )ooa44t41a q!Q1a  q!Q1} r27B7$7S]]1aAq=1^^ %SZZ1aA .[[ " KK#56B ? KK#56B ? XXe_ ((5/ 95 9: +G7B7B??2r;;d;D"&..Jw #{{GG   +  CMM2,rz)rrrrTr_cRtjjd}|jddj d}|jddj d}tj |ddd<tj |ddd <tj tj||z| }t5}tjd 5|jtd |jtdtj||fd|i|}ddddddtj j} t| |tj |j } t| |y#1swYhxYw#1swYlxYw)NlAoI;rY)r6r4rYrrrr3r4r7rSrrz'invalid value encountered in less_equalrrT)rrrintegersrrmr4r"r rrrrfrrr r) rnrrTrr*r4rrrp_nansstatistic_nanss rxtest_nans_on_axisz'Test_ttest_ind_common.test_nans_on_axisKsPii##$9: LL*L - 4 4W = LL*L - 4 4W =VV!Q VV!Q 88BFF1q5t45  ;CX)F ; JJ~@ B JJ~'@ A//!Q:T:T:C  ; ; #**%68,#--0>84 ; ; ; ;s% F$AF*FF FF&N) rrrrrr/rrr rrzrxrrs [[ [[VsA&N'-rlB&8!BD [[[4-"-}!=?!-?D !-F [[VsA&N'-rlB&8!BD [[Vb!W-5.D5rzrc veZdZgdgddddggdgddddggdgdd d d ggd gd dddggdgddddggZej j dedZdZdZ ej j dddZ dZ e dddZ ej j d gd!d"Zy#)$Test_ttest_trimr}rrg@g9(?ger)rg33333`@r[g33333^@g(\P@rgfffffڇ@gBM)u?g ]@gyd?g s5v@g{Gz? rrrrffffff?rrrrrr rBg@rr?r<rg333333@rr;g@rgn4aƂg?^)g , gm?giI @gE|x!gvwQ?glfPg_)gXG~?g=n@g8$mg({g]:m?gYv gb?gb gSTß}g9yg.Ͽv?g޺i?g 'X@)g>_\?g hO?gau?@g׌g+?g:؜^gz 54g7f?gDO%@g#xG @g\F?g4E?gnfg>P"ۿguX@gvR?g%?g2Ydzgo?gih?gn?g!nu?gJ+bza,b,pr,tr,trimcttj|||d\}}t||dt||dy)a Using PairedData's yuen.t.test method. Something to note is that there are at least 3 R packages that come with a trimmed t-test method, and comparisons were made between them. It was found that PairedData's method's results match this method, SAS, and one of the other R methods. A notable discrepancy was the DescTools implementation of the function, which only sometimes agreed with SAS, WRS2, PairedData and this implementation. For this reason, most comparisons in R are made against PairedData's method. Rather than providing the input and output for all evaluations, here is a representative example: > library(PairedData) > a <- c(1, 2, 3) > b <- c(1.1, 2.9, 4.2) > options(digits=16) > yuen.t.test(a, b, tr=.2) Two-sample Yuen test, trim=0.2 data: x and y t = -0.68649512735573, df = 3.4104431643464, p-value = 0.5361949075313 alternative hypothesis: true difference in trimmed means is not equal to 0 95 percent confidence interval: -3.912777195645217 2.446110528978550 sample estimates: trimmed mean of x trimmed mean of y 2.000000000000000 2.73333333333333 Frrr#r$Nrfrr )rnr*r4rrhrrrs rxtest_ttest_compare_rz$Test_ttest_trim.test_ttest_compare_r}s6@"OOAqtuM 6 2E2/rzcgd}gd}tj||dd\}}t|ddt|d dy) N) r[rWrrr,r[rWrrrr)rbrrWr[rrrbrrWr[rrrg ףp= ?FrgRbv?rlr$gOi?rrnr*r4rrs rxtest_compare_SASz Test_ttest_trim.test_compare_SASsA = D"OOAqseL 6t4 8$7rzcgd}gd}tj||d\}}t|ddt|ddy ) a The PairedData library only supports unequal variances. To compare samples with equal variances, the multicon library is used. > library(multicon) > a <- c(2.7, 2.7, 1.1, 3.0, 1.9, 3.0, 3.8, 3.8, 0.3, 1.9, 1.9) > b <- c(6.5, 5.4, 8.1, 3.5, 0.5, 3.8, 6.8, 4.9, 9.5, 6.2, 4.1) > dv = c(a,b) > iv = c(rep('a', length(a)), rep('b', length(b))) > yuenContrast(dv~ iv, EQVAR = TRUE) $Ms N M wgt a 11 2.442857142857143 1 b 11 5.385714285714286 -1 $test stat df crit p results -4.246116897032513 12 2.178812829667228 0.00113508833897713 rrrrgR?rr$g^Nrrs rxtest_equal_varzTest_ttest_trim.test_equal_vars@& D C!OOAqr: 6 3%@ #5EBrzz alt,pr,tr))rdgjΜ>5?r)rbgn4aƂW?rcgd}gd}tj||dd|\}}t||dt||dy) z > library(PairedData) > a <- c(2.7,2.7,1.1,3.0,1.9,3.0,3.8,3.8,0.3,1.9,1.9) > b <- c(6.5,5.4,8.1,3.5,0.5,3.8,6.8,4.9,9.5,6.2,4.1) > options(digits=16) > yuen.t.test(a, b, alternative = 'greater') rrrF)rrrirr$Nr)rnrbrrhr*r4rrs rxtest_alternativesz!Test_ttest_trim.test_alternativessE D C!OOAqrU8;= 6/ 2E2rzc d}tt|5d}tjt|5t j ddgddgddddddddtt|5d }tjt|5t j ddgddgddd ddddddy#1swYpxYw#1swYtxYw#1swY*xYw#1swYyxYw) Nz>Use of `permutations` is incompatible with with use of `trim`.rz5Arguments {'permutations'} are deprecated, whether...rr3r4r)rrz9Arguments {.*'random_state'.*} are deprecated, whether...)rrr)rrrr>DeprecationWarningrfr)rnrrs rxtest_errors_unsupportedz'Test_ttest_trim.test_errors_unsupportedsP .e < IMG0@ IAARaH I I .e < 0QG0@ 0AARa-.0 0 0 0 I I I I 0 0 0 0sFC B?C 1C#C.C#?C C  CC C##C,Trrc*d}|jd|jddz}}t|rtj||dyt j t |5tj||ddddy#1swYyxYw)Nz3Use of `trim` is compatible only with NumPy arrays.rYrr rr)rr+rfrrrr)rnrorr*r4s rxrz5Test_ttest_trim.test_permutation_not_implement_for_xpsoGyy}biimAo1 B< OOAqs +2'B 013/ 0 0 0s 'B  Br)gɿr<rcd}tt|5tjddgddg|dddy#1swYyxYw)Nz/Trimming percentage should be 0 <= `trim` < .5.rrr3r)rrrfr)rnrrs rxtest_trim_bounds_errorz&Test_ttest_trim.test_trim_bounds_errors>A :U 3 7 OOQFQF 6 7 7 7s :AN)rrrrrrrrrrrr"rrr%rrzrxr r is O%8:M  1? / 5 1? / 6 @ @ ,b 2 H "34Db JKF& [[-v6!07!0F 8C4 [[[N3 3 0t,GH0I0 [[V\2737rzr c@eZdZejdZgded<dddej dged <ddd d ej ged <gd ed<dddej dged<ddddej ged<dddejejged<dddejejged<dddejejged<dd d!ejejged"<dd d#ejejged$<dd d%ejejged&<ejjd'gd(ejjd)d*d+gejjd,d-d.ge d/d01d2Z y3)4 Test_ttest_CI)r4r3r3r6)r)gͿe13@gv7?gę"Ͽg8{M(?rkr(r)gv7?gx_?r|g /?gD.ɿ)r3rr)reϿ-&'a@ ff\+.IBGGDM@RRS+.( 2vsD rzz)'<4H 2T: BJJv$6TB((-B BFFBJJsO$ ? BGGRZZ%5D ArzN) rrrrrrrrmrrrrr0rrzrxr'r's< |$ANAgJibffWlKAgJiM266JAgJLAgJibffWlKAgJiL"&&IAgJ%r+=rvvrvvNAgJ%r+=rvvrvvNAgJ%r+=rvvrvvNAgJ%r+=rvvrvvNAgJ%r+=rvvrvvNAgJ%r+=rvvrvvNAgJ [[],LM [[[5$-8 [[VaX.k*JKBL/9NBrzr'c Dtjjdtjjdddddd}tjjddddd}t ||fd }tj |d }tj |d d }t d|jDD]`}|\}}}}}} ||jd kr||||||| f||||||| fk(r6J||||||jd z || f||||||| fk(r`Jy)Nrr6r5r4rr7r9r3r`rSr)N.rc32K|]}t|ywrrrs rxrz.test__broadcast_concatenate..Ts5585r)rrr=rHrrrr) r*r4r,indexrrWrJr/rTrUs rxtest__broadcast_concatenater4Js1IINN1 q!Q1a(A q!Q1%A1vB/A %&A ) 01A5QWW56L 1aAq qwwr{?Q1aA%&!Aq!Q1,<*== ==Q1a!''"+oq!34!Q1a:J8KK KK LrzcbeZdZeej dZejjdejjddZ dZ dZ ejjdejdd fejd d fgd Zd ZdZy) TestTTestIndc|jgd}|jgd}|jd}|jd}tj||d\}}t||t||tjt ||ddi\}}t||t|||jgd}|jd }|jd }tj||d\}}t||t||tjt ||ddi\}}t||t|||jd }|jd }|jd }|jd} |j || g} |j ||g} |jddd} |jddd} |jddd}|j || g}|j | |g}tj|| dd\}}t||t||tjt || ddi\}}t||t||tj|| dd\}}t||t|| tjt || ddi\}}t||t|| tj|j|jdd}t|j| t|j| t |j|j}tj|ddi}t|j| t|j| tj||dd}t|j| t|j| t ||d}tj|ddi}t|j| t|j| y)Nryr g9(?geFrrrgJ?g9ʿrg@[?gωұ?ghG?rrrr6rrrrS) rdrfrr1rrrrrrr)rnror*r4rrhrrA tr_uneq_n pr_uneq_nrrrvs3rTrSrjrkrr7s rxtest_ttest_ind_with_uneq_varz)TestTTestInd.test_ttest_ind_with_uneq_var^s JJ| $ JJ ' ZZ+ , ZZ- .q!u5122));q!+<NN122 JJ' ( ZZ+ , ZZ- .q!u5122));q!+<NN122ZZ* +JJ23 ZZ+ ,JJ23 "rc#"b"{{1c2&{{1c3'{{1c3'((D$<(((D$<(tTUC122));tT+BTeT122tTUC19%9%));tT+BTeT19%9%oogiieL u- E*799gii0(($@%@ u- E*oogwQ%H u- E*7G!4(($@%@ u- E*rz0ignore:divide by zero encountered:RuntimeWarning/ignore:invalid value encountered:RuntimeWarningc|jd}|jd}t|td5t j ||d\}}dddt |j|j t |jdtjd5t j ||d\}}t ||j|jt ||j|j|jd |jgd d gg}t j ||jd d\}}t ||jdtjgt ||jd tjgdddy#1swYqxYw#1swYyxYw) Nr4rrFr8rrrrr_r]r) rrr)rrfrr2rdrrrrm)rnrorqrrrrArns rxtest_ttest_ind_zero_divisionz)TestTTestInd.test_ttest_ind_zero_divisionsc HHQK GGAJ N2K L :??1a59DAq : 2::rvvg./2::b>* [[X & 9??1a59DAq Arzz"&&1 2 Arzz"&&1 2::266{RG45D??4&)9UKDAq Arzz2rvv,7 8 Arzz2rvv,7 8 9 9  : : 9 9sG0DG G  Gctjdddg}gd}tj||d}tj||d}t |j |j dt |j |j dtj||d d}t ||dt |d dy) Nrrrrrrrr#r$r)g8HgW@?)rrmrfrr rr)rnrqrrrrrts rxtest_ttest_ind_nan_2nd_argz'TestTTestInd.test_ttest_ind_nan_2nd_args VVS#s #  __Qf 5 __Qf 5 r||m%@ 29959__Q!" &BU+ G" $rzc4t|r]tjtt5t j |jg|jg}dddn4t j |jg|jg}tt jjsJ|j|jd}t|j|t|j|y#1swYzxYw)Nrr)r+rr>rrrfrrdrrwrxrmr2rrrnrorrs rx#test_ttest_ind_empty_1d_returns_nanz0TestTTestInd.test_ttest_ind_empty_1d_returns_nans B<08MN FoobjjnbjjnE F F//"**R."**R.AC#u::;;;jj $ s+ C( F Fs 5DDrzr{rr|r}c|jd}|j||j}tjj 5}|j tttj||d}dddttjjsJ|j||j}t!|j"|t!|j$|y#1swYyxYw)NrrHr_rSr-)rrdrIrrr rrrrfrrrwrxrrmr2rr)rnr4rror*rrrs rxtest_ttest_ind_axis_size_zeroz*TestTTestInd.test_ttest_ind_axis_size_zeros HHY  JJqJ ( ZZ ) ) + 1s JJ)+@ A//!QR0C 1#u::;;;BFFC ~6 N3 1 1s 3C88Dc0|jd}|jd}tj||d}t|tjj sJ|j jdk(sJ|jjdk(sJyr rrfrrrwrxrrrrnror*r4rs rx test_ttest_ind_nonaxis_size_zeroz-TestTTestInd.test_ttest_ind_nonaxis_size_zeros| HHY  HHY ooa+#u::;;;}}""V+++zz6)))rzc0|jd}|jd}tj||d}t|tjj sJ|j jdk(sJ|jjdk(sJy)N)rr8rrrrSrrJrKs rx2test_ttest_ind_nonaxis_size_zero_different_lengthsz?TestTTestInd.test_ttest_ind_nonaxis_size_zero_different_lengths s| HHY  HHY ooa+#u::;;;}}""V+++zz6)))rzN)rrrr0rfrr<rrrrArCrFrrrrHrLrNrrzrxr6r6\su112N+3N+` [[: [[99 9,$* ) [[0&bhhy16:%RXXi0&9;< 4< 4 * *rzr6ceZdZejj dddZejjddZy)TestTTestIndFromStatsTz"Other backends don't like integersrc|jddg|jddg}}|jddg|jddg}}|jddg|jdd g}}tj||||||y) Nrr3r4r5r6r)rdrfr)rnromean1mean2std1std2nobs1nobs2s rx test_gh5686z!TestTTestIndFromStats.test_gh5686szz1a&)2::q!f+=uZZA'QF);dzz3*-rzz3*/Eu ""5$udEJrzr>c |jd}|jd}|j|j}tj||||||d}t |j |t |j |y)NrrFr8)rdrmrfrr2rr)rnrozerosixrrs rx%test_ttest_ind_from_stats_inputs_zeroz;TestTTestIndFromStats.test_ttest_ind_from_stats_inputs_zero$skzz"~jjnjj ((tS$c38: s+ C(rzN) rrrrrrr[rr_rrzrxrPrPsV [[!!$'K"MKMK [[ QR)S)rzrPTzTest uses ks_1samprr>ctjjd}|j|j d}|j|j d}|j d}t j||dd}tj|j}t j|t j jjd kDsJttj|||d t j||dd}tj|j}t j|t j jjd kDsJttj|||d |jgd |jd g}}t j||d}t||jk(rdnd}t|j |jd|t|j|jd|y)NlbQ)r2r3r)r2rrTr_)rrTr rr$)r3r4r6r=r8rlrgzNd?rjgJ?)rrrrdrrmrfrrrrr quantiler-rr1r)rorrqrrrbrrrks rxtest_ttest_uniform_pvaluesrb0s ))   0C 3::::./A 3::::./A  A //!Q$R 8C ZZ #F >>&%--/"5"5 6 = = CC CBKK*AD9 //!Q$R 8C ZZ #F >>&%--/"5"5 6 = = CC CBKK*AD9 ::i "**cU"3qA //!Q$ /C#B'2::545DCMM2::o#>TJCJJ > :Frzc|j|dk(}|j|dk(}|dk|z|dkD|zz}|j||dz d|dz z S)Nrbrdrr3r)rdr)rrArbrorbrdrs rx_convert_pvalue_alternativerdNsc ::cVm $Djj )*G a%4QUg-.A 88AqsA!G $$rzrrcR d\}}}tjjdd|||f}|j|}|j d||f}tj ||d\}}tj |dd\}} tj |ddddfd\} } t ||d t |d | d t|j||f|j |d|f}tj ||d\}}tj |dd\}} tj |ddddfd\} } t ||d t |d | d t|j||f|j ||df}tj ||d \}}tj |dd \}} tj |ddddfd\} } t ||d t |d | d t|j||ftj |jgd d\} } t|j| |j|jt| |jdtj |ddddddfd\}}tj |ddddddfdd\} } t||d|}t | |t | |tj |ddddddfdd\} } t||d|}t | |t | |tjd5tj |jgd d}t|j|j|j t|j"|j|j |jdtj gddgg}tj |d}t|j|jd|j gt|j"|jd|j gdddy#1swYyxYw)NrYrrr6rYrrrrSrr+rjrr3)rrrrrdrhrbrr@r.)rfrrrdrrr1rrr2rfrrdrrrrmr)ror)r*n3rvn1r"rrrp2rp3rrArhrpcrrns rxtest_ttest_1samp_newrlWs JBB ::>>arR > =D ::d Dggq"bk"G   tW1 5FB   tRa 0FB   tAq!G}b 1FBB'BtHbu-B8$ggr1bk"G   tW1 5FB   tRa 0FB   tAq!G}b 1FBB'BtHbu-B8$ggr2qk"G   tW1 5FB   tRa 0FB   tAq!G}b 1FBB'BtHbu-B8$   RZZ 5r :DAqBFF1Irzz"&&12Arzz"~&   tAq!G}b 1FB   T!Q']BI FDAq $RY ;BArAr   T!Q']BF CDAq $RVR 8BArAr  " > < 8"= rzz"&&'9: BJJrvv$67zzB<#r34b) rzz2rvv,'?@ BJJBFF|$<= > > >s 1D#RR&z(Only NumPy has nan_policy='omit' for nowrc2d\}}}tjjdd|||f}|j|}|j|ddddd d f<tj |ddddddfd d \}}tj |ddddddfd d d\}}t ||d|} t|| t||tj |ddddddfd d d\}}t ||d|} t|| t||y)N)r6rYrr6rYrrr3rr4r5r9rrrrdrrb)rfrrrdrmrrdr1) ror)r*rgrhrhrrrArks rxtest_ttest_1samp_new_omitrnsJBB ::>>arR > =D ::d D&&D1ac1Q3   tAq!G}bV DFB   T!Q']B6)2 4DAq $RY ;BArAr   T!Q']B6)/ 1DAq $RVR 8BArArrzrrrctjjd}|jd}|j|}d}|j|jd}t j t |5tj||dddd|j|jd }tj||d}|jjd k(sJ|j\}}|j|d }|j|d }tj||d}|j|jd |j |jj}t!|j|tj||d}t!|j|y#1swY5xYw)NliI}pC)rrrrz%`popmean.shape\[axis\]` must equal 1.)r6r3rrr)r"rT)r6rr)r6rrSg?rH)rrrrdrrrrfrrrrTr.rrrr1) rorrqrr"rr/rrs rxtest_ttest_1samp_popmean_arrayrpsu ))   3 4C $A 1 A6Gjj45G z 17 !W267jj45G   AwR 8C ==  ' )) )  " " $DAq qr"A qr"A   Aqr 2C //"**T*N>N OCCJJ$   Aqr 2CCJJ$!77s G  GcxeZdZejj ddZdZdZdZ dZ dZ dZ d Z y ) TestDescriberct5}tjdd5|jtdt j |jdd\}}}}}}dddddddk(sJtd|jdt|d|jdt|jdt|j|jt|j|jt|j|jy#1swYxYw#1swYxYw)Nrrrrrrr) r rrrrrfdescriberdr2rm) rnrorrUrrTrskkurts rxtest_describe_scalarz!TestDescribe.test_describe_scalars   GCkk(8< G JJ~'J K$)NN2::b>"3E$F !Ar1aT G GAv v1rzz#/1rzz#/2::c?+2::bff-.BJJrvv./bjj01 G G G Gs#EAE(EE EEc|j|jd|jddf}d}|jgd|jgdf}|jgd}|jgd}|jd gd z}|jd gd z}t j |\} } } } } }| |k(sJt | d |d t | d |d t| |d |j| jjzt| |d |j| jjzt| |t||t j |jd \} } } } } }| |k(sJt | d |d t | d |d t| |d |j| jjzt| |d |j| jjzt| |t||y)Nr1r3r5rr6rrrrrrrrffffff?r}r}r})rrrrH, p ?r5TUUUUUrrrjrS) rrrrdrfrtr2r1rrIepsrrnrorqncmmcmcvcskckurtcrUrrTrrurvs rxtest_describe_numbersz"TestDescribe.test_describe_numberss IIrwwv(;< = zz*+RZZ8H-IJ ZZ, - ZZ, -jj-.23 ./!34 %q 12q!RBww1s1v&1s1v&2A(9(=(=$=>2A(9(=(=$=>C e$ %qss ;2q!RBww1s1v&1s1v&2A(9(=(=$=>2A(9(=(=$=>C e$rzc>tjd}tj|d<d\}}d}d}d}d}tj|d \}} } } } } t ||t | |t | |t | |t | |t | |d y) Nrr:)r:)rrrrCrgGzrrr\r)rrrmrfrtrr)rnrqrrrrrrrUrrTrrurvs rxdescribe_nan_policy_omit_testz*TestDescribe.describe_nan_policy_omit_tests IIcNvv!!C  # %qV D2q!RQRQQ!"c*!$r:rzc|jd}|j|dk(|j|}t|r0 ]]:W 5 + NN2::b> * + + +s %A  AN)rrrrrrrwrrrrrrrrrzrxrrrrsI [[ VW 2X 2%4;"!3F0%"%,+rzrrcZeZdZejj dgddZdZy)NormalityTestsrirc|j}tt|}|j|jd}|j|jd}i}|dvr=|dvr$|dk(r|dz nd|dz z }|dk(rd|z n|}||d<nt j d |jd dz}||fi|} | \} } t| |t| |t| d | y) Nrr>rbrd>skewtest kurtosistestrbr3rriz,`alternative` not available for `normaltest`)rr_rrr3rrr_rrr3rrr_rrr3rrr_rrr3rrr) test_namerErfrdcase_refrrr1r) rnrirorr ref_statisticr$r8rqr res_statistic res_pvalues rxtest_against_RzNormalityTests.test_against_RasNN 5),zz$--"23 ZZ a 01  - -88-8F-BZ\*UV, -6*-DQz\* (3}% JK JJ. / 2q#F#$'! z }5 J/C!8R@rzctt|j}|jd}|j |j |j }|j|dk(||}tjd5||}t|j|t|j|dddy#1swYyxYw)NrrHrrr) rErfrrrdrmrIrrrr2rr)rnrorrqrrs rxtest_nanzNormalityTests.test_nans5$..1 IIcNjjqwwj/ HHQ"Wc1 % [[ * -1+C CMM3 / CJJ , - - -s ?5B==CN)rrrrrrrrrrzrxrr_s. [[],LMANA< -rzrc eZdZdZdZdZdZy) TestSkewTestr)gvO?gbH`?ctjjddd}|j|}tj|dj }t ||jd|jd y) Nrr2{)r*rrrdrhrrHg?̔>r$)rfskewnormrrdrrr1rI)rnrora1_xprms rxtest_intuitivezTestSkewTest.test_intuitives] ^^  !%c  B 2~~e;BBbjjEKKj@tLrzctj|jd|jd}t|rtnd}t j t|5tj|}|j|j}t|j|t|j|dddy#1swYyxYw)Nrrz1`skewtest` requires at least 8 valid observationsr) rfrrr+rrr>rrdrmr2rr)rnrorqrrrs rx"test_skewtest_too_few_observationsz/TestSkewTest.test_skewtest_too_few_observationss ryy~& IIcN,4RL(K  \\,G < -..#C**RVV$C CMM3 / CJJ ,  - - -s $AC  CN)rrrrrrrrrzrxrrsI9HM -rzrc&eZdZdZdZdZdZdZy)TestKurtosisTestr)gglg2J@?ctjjdd}|j|}tj|dj }t ||jd|jd y) Nr2r)rrrdrhrrHr#r$)rflaplacerrdrrr1rI)rnrora2_xprms rxrzTestKurtosisTest.test_intuitives] ]]  E  < 2!!%Y?FFbjjEKKj@uMrzc gd}tjt|Dcgc]\}}tj||c}}}|j ||j }t j|ddksJycc}}w)N) rr:r8r)rrrrHrr)rhstack enumeraterrdrrfr)rnrorrr,rqs rxtest_gh9033_regressionz'TestKurtosisTest.test_gh9033_regressionso7 II61BCArwwq!}C D JJq J +!!!$Q'$...Ds B ctj|jdt|rtnd}t j t|5tj|jd}|j|j}t|j|t|j|dddy#1swYyxYw)NrLz(`kurtosistest` requires at least 5 validrr) rfrrr+rrr>rrdrmr2rr)rnrorrrs rx&test_kurtosistest_too_few_observationsz7TestKurtosisTest.test_kurtosistest_too_few_observationss 299S>*,4RL(B  \\,G < -$$RYYr]3C**RVV$C CMM3 / CJJ ,  - - -s A,CCN)rrrrrrrrrrzrxrrsI:HN/ -rzrceZdZdZdZdZy)TestNormalTest normaltest)gاc@g?c~tj|jdtjt 5tj|jd}|j |j}t|j|t|j|dddy#1swYyxYw)Nrr) rfrrrr>rrdrmr2rrrEs rxtest_too_few_observationsz(TestNormalTest.test_too_few_observationss 2'\\, - -""299R=1C**RVV$C CMM3 / CJJ ,  - - -s A,B33B<N)rrrrrrrrzrxrrsI9H -rzrceZdZejj dejj dd\ZZe jjdgddZ dZ dZy ) TestRankSumsrr3rYrirctj|j|j|j}tj |j|jd|j}t ||y)NrhF)use_continuityri)rfranksumsrqrrr mannwhitneyur )rnrirrs rxtest_ranksums_result_attributesz,TestRankSums.test_ranksums_result_attributess\~~dffdff*577=v !!$&&$&&.9;;A6 d#rzcptj|j|j}t |dy)Nr)rfrrqrrrrs rxtest_ranksums_named_resultsz(TestRankSums.test_ranksums_named_resultss%nnTVVTVV,C!89rzcttd5tj|j|j ddddy#1swYyxYw)Nz`alternative` must be 'less'rrrh)rrrfrrqrrr8s rxrz"TestRankSums.test_input_validations> :-K L A NN466466x @ A A As -AAN)rrrrrr=rHrqrrrrrrrrrrzrxrrsZIINN1 99>>!R DAq [[],LM$N$:ArzrcVeZdZdZeddZedddZd Zy ) TestJarqueBeracgd}|j|}|jddg}tj|}t|j|dt|j |dy)N)g8MK~Ŀg/?g6r=TgGg!J_ lgq"?gPHK?rr)rdrf jarque_berar1rr)rnrorqrrs rxtest_jarque_bera_against_Rz)TestJarqueBera.test_jarque_bera_against_Rs^  ; JJqMjj*O<=" s1v. CF+rzTr_cdtjjdtjjddd}t j t |x}\}}t j t|x}\}}t j |jddx} \} } ||cxk(r6| cxk(r0|jcxk(r |jcxk(r| jk(sJJ||cxk(r6| cxk(r0|jcxk(r |jcxk(r| jk(sJJy)Nrrrrr3P) rrr=rrfrrtuplerhrr) rnrorqjb_test1JB1rjb_test2JB2rijb_test3JB3rjs rxtest_jarque_bera_array_likez*TestJarqueBera.test_jarque_bera_array_likes y! II  Q6 *"..tAw7773"..uQx8873"..qyyE/BCC73cbSbH$6$6b(:L:LbPXPbPbbbbbbRV2VVHOOVxVVVVVrzarray_api_strictzNoisy; see TestSkewrcF|jg}tjtt5t j |}|j|j}t|j|t|j|dddy#1swYyxYwrJ) rdrr>rrrfrrmr2rr)rnrorqrrs rx%test_jarque_bera_too_few_observationsz4TestJarqueBera.test_jarque_bera_too_few_observationssr JJrN \\,4I J -##A&C**RVV$C CMM3 / CJJ ,  - - -s ABB ctjjd}|j|jd}t j |d}t j |j |d}t|j|jt|j|jt j |d}t j |dddf\}}t j |dddf\}} t|j|j||gt|j|j|| gt j |jd} t|j| jt|j| jy) NiKrW)r3rrrSrrr) rrrrdrfrrhr2rrr1rr) rnrorrqrrs0rs1rresTs rxr@zTestJarqueBera.test_axiss=ii###3 JJszzwz/ 0- 1e 45 s}}5 CJJ/*""1QT7+B""1QT7+B rxxR'9: BHHb"X$67  1- t~~6 DKK0rzN)rrrrrrrr@rrzrxrrsE ,d# W$ W(1FG-H-1rzrcDeZdZgdZgdZdZdZdZdZdZ dZ d Z y ) TestMannWhitneyU)g U3@g^–3@gt3@g]o5@gzS4@gt<4@gM X3@gXmCk4@gLo&3@gfɘ2@g{m;^3@gaA$|3@gMѢ_4@g#3@g19T_3@ggC2@gqR4@gKk/4@g܁3@gZ2@g_ 3@g$`3@g x4@g3#л5@g;VND1@g>H2@gr4@gyܤ2@g?4@g=bɇ3@)gpG3@gL:j0@g2@gJNnC1@gR;`(3@g+Pi,2@g6:Z2@gZ"3=2@gF uSa 3@gXҀ2@gX3^83@g21@gJ2@gZ)E2@gU83@g50@gԆ3@gFz33@gz 3@gڴ3@rWctj|j|jd\}}tj|j|jd\}}tj|j|jd\}}tj|j|jd\}}t ||t ||t ||k7t |dt |dt |dt |dt |d|jt |d|jy) Nrbrhrdrg=_A?rglPz.?rfrreYrrr r rnu1ru2riu3rju4p4s rxtest_mannwhitneyu_one_sidedz,TestMannWhitneyU.test_mannwhitneyu_one_sided>s##DFFDFFGB##DFFDFF JB##DFFDFF JB##DFFDFFGBRRbRRRRB 1t?O?OPB 3AQAQRrzc8tj|j|jd\}}tj|j|jd\}}t ||t |dt |dt |d|j y)Nr`rhrrglPz.?rrfrrerrr rrnrrrris rxtest_mannwhitneyu_two_sidedz,TestMannWhitneyU.test_mannwhitneyu_two_sidedNsv##DFFDFF LB##DFFDFF LBRRRB 2(,(8(8 :rzctj|j|jdd\}}tj|j|jdd\}}tj|j|jdd\}}tj|j|jdd\}}t ||t ||t ||k7t |dt |dt |dt |dt |d|jt |d |jy) NFrbrhrdrrgii5?rg22?rrs rx&test_mannwhitneyu_no_correct_one_sidedz7TestMannWhitneyU.test_mannwhitneyu_no_correct_one_sidedXs##DFFDFFE068B##DFFDFFE09;B##DFFDFFE09;B##DFFDFFE068B RRbRRRRB 1t?O?OPB 4$BRBRSrzc<tj|j|jdd\}}tj|j|jdd\}}t ||t |dt |dt |d|j y)NFr`rhrrg22?rrrs rx&test_mannwhitneyu_no_correct_two_sidedz7TestMannWhitneyU.test_mannwhitneyu_no_correct_two_sidedls##DFFDFFE0;=B##DFFDFFE0;=B RRRB 4(,(8(8 :rzc,tjgd}tjgd}ttj||ddttj||ddttj||dd y) N)rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr)rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrbrh) @g?չ>rd)rg&)E?r`)rg?չ ?)rrr rfrr?s rxtest_mannwhitneyu_onesz'TestMannWhitneyU.test_mannwhitneyu_onesxs HH. /& HH & ' **1aVD: < **1aYG2 4**1a[I5 7rzcxd}tj|j|jd}t ||y)Nrrbrh)rfrrerrrEs rx#test_mannwhitneyu_result_attributesz4TestMannWhitneyU.test_mannwhitneyu_result_attributess-,   VDC,rzN) rrrrerrrrrrrrrrzrxrr,s9 -A NA KS :T( :)7V-rzrcgd}gd}ttj||dddd}tj||}t||t |j |j y)N)"rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr)"g-@g+@g(@rrrZg333333@ffffff@g333333@r?r皙 @rr?r?r>rffffff@@333333?r r=?r rrrrr rr<rrr rg/"?r6r)rrfpointbiserialrrrrr)rqrrr rs rxtest_pointbiserialr se A &A,,Q215wB+J   q! $CZ(#--0rzctjgd}tj|}gd}t |d|tjgd}tj|}tjgd}t |d|tj||\}}t ||dt ||dtj|||\}}}t ||dt ||dt ||dtj d}tjgdgdg}ttj|d |z|d tdd } tjgd } tj| | } tjgd} tj| | }tj| }t|d|d y)N)rr3r5)r8rr8r)rr4r7r:)rrrrr6)㪪@I?g&>ڿr r )5@㪪@gIr r r3r9rrZ)r6rZr:r4r3r3)g_v @g1%?g"u?gZӼ?g!u @g q&@r5) rrrfobrientransformr rrrrr) rrrrrr*r4r,rqvaluesrepsrhtransformed_valuess rxtest_obrientransformr s ) B   r "BHBqE8$ , B   r "Bxx'HBqE8$  R (DAqBqEBqE##BB/GAq!BqEBqEBqE 1B XXDFHIFe33B"=vqQ1b\F 88' (D 99VT "D#<=yy+T2H  " "4 (FfQi1=rzrz)rTrIrkweightsc,|xs |j}t|dk(r|\}}n|\}} }|j||}|j||}||j||n|}t|dk(r|fn| f}|||||d} t| ||y)Nr3rH)rTrIr rj)rrzrdr1) romean_funrTrIrkr r7 array_likerrArqs rxcheck_equal_xmeanr s  RZZE 4yA~" G!% AwJe4Jjjj.G292Ebjjj.7GINJ=QD$T@AAwT*rzc8t|dtji|SNr )r rfrr7r8s rxcheck_equal_gmeanr  d CU[[ CF CCrzc8t|dtji|Sr )r rfhmeanr s rxcheck_equal_hmeanr r rzc8t|dtji|Sr )r rfpmeanr s rxcheck_equal_pmeanr r rzceZdZejj ddZdZejj dejj ddZejj ddZ dZ d Z ejj dd Z d Z ejj dd Zed ddZdZdZdZedZy) TestHMeanrc,gd}d}t|||yN)rrr3rrr rnror*rs rxtest_0zTestHMean.test_0s !W,rzcTgd}d}t|||gd}d}t|||y)NrMg=b#A@rrggQ?r% r& s rxtest_1dzTestHMean.test_1d s. 5!W, :!W,rzrcTtjddg}d}t|||dy)Nrrrrorkrrr r& s rxtest_1d_with_zerozTestHMean.test_1d_with_zeros) HHaV !W#6rzr=ctjgd}d}tjt|5t ||j |ddddy#1swYyxYw)Nrrr_z$The harmonic mean is only defined...rrr+ )rrrr>rr rm)rnror*rs rxtest_1d_with_negative_valuez%TestHMean.test_1d_with_negative_valuesL HHZ 8 \\. 8 : aBS 9 : : :s AA!c`gdgdgdg}d}ttj|||y)NrYrrrrrdrNrgrGrrSxgmWUC@r)r rrr& s rxtest_2dzTestHMean.test_2d%s) /1D E"((1+w26rzcfgdgdgdg}tjgd}t||d|y)Nr2 r3 r4 )U|ϊ6@g4.=C@JQ6LsJ@E]P@rrTror, r& s rx test_2d_axis0zTestHMean.test_2d_axis0+s- /1D E((OP!W14rzcfgdgdgdg}tjgd}t||d|y)NrYrrrr3 r4 )r8 rr9 r: rr; r, r& s rxtest_2d_axis0_with_zeroz!TestHMean.test_2d_axis0_with_zero1s- .0C D((GH!W14rzcfgdgdgdg}tjgd}t||d|y)Nr2 r3 r4 )g3333333@ O@y@Y@rr; r, r& s rx test_2d_axis1zTestHMean.test_2d_axis17s- /1D E((<=!W14rzcfgdgdgdg}tjgd}t||d|y)Nr> r3 r4 )rrA rB rr; r, r& s rxtest_2d_axis1_with_zeroz!TestHMean.test_2d_axis1_with_zero=s- .0C D((;<!W14rzT,array-likes only supported for NumPy backendrcgd}gd}d}tj||}t|tj|dy)Nr3rYr7rYr6r4rr h㈵>rj)rfr r1rrdrnror*r rrs rxtest_weights_1d_listzTestHMean.test_weights_1d_listCs: kk!W-RZZ0trzc:gdgdgdg}d}t|||y)Nr2 r3 r4 g/,$qJ@rra r& s rxr6 zTestGMean.test_2ds! /1D E!W,rzcgdgdgdg}tjgd}t||d|tgdgdgdg}tgd}t||dd| y) Nr2 r3 r4 )g4@A@g5H@g`8|wЭN@gm~&+R@rr; rgr+rTrkrorrr r& s rxr< zTestGMean.test_2d_axis0sX /1D E((NO!W14 <|< = %!W15R@rzcgdgdgdg}tjgd}t||d|tgdgdgdg}tdd }t|||g}t||dd | y) Nr2 r3 r4 )gD" 6"6@gcP@g ҧZ@rr; rgrrjr+rf )rrr r)rnror*rrs rxrC zTestGMean.test_2d_axis1sj /1D E((CD!W14 <|< = - )Aq "!W15R@rzc@tgd}d}t||d|y)N)g}Ô%ITrPgu<7~rPrrc )rr r& s rxtest_large_valueszTestGMean.test_large_valuess ' (!W5R8rzcgd}d}tjd5t|||dddy#1swYyxYw)N) rYrrrrrdrNrgrGrrrr@r)rrr r& s rxtest_1d_with_0zTestGMean.test_1d_with_0s; 4 [[X & 1 aR 0 1 1 1s5>z;ignore:invalid value encountered in log:RuntimeWarning:daskcgd}tj}tjd5t|||dddy#1swYyxYw)N) rYrrrrrdrNrgrGr_rrr)rrmrr r& s rx test_1d_negzTestGMean.test_1d_negs? 5&& [[ * 1 aR 0 1 1 1s AA TrF rcgd}gd}d}tj||}t|tj|dy)Nrr3r6r7r5r4GZ*oG8@rJ rK rj)rfrr1rrdrL s rxrM zTestGMean.test_weights_1d_lists: ! kk!W-RZZ0trr rm)rnror*rArs rxr0 z%TestPMean.test_1d_with_negative_value sNxx #T15 \\. 8 3 aBFFr 2 3 3 3s AA#rw rYrrrdrGrrr<cttjtj||}t ||||y)Nrru rx rrr )rnr*rArors rxtest_2d_axisnonezTestPMean.test_2d_axisnones, ++BHHQK;!QB/rzr2 r3 r4 r> c  tt|dDcgc]Q}tjt j tt|Dcgc] }||| c}|S}}}t |||d|ycc}wcc}}w)Nrr; )rrzru rx rrr )rnr*rArorWrrs rxr< zTestPMean.test_2d_axis0s3qt9%    % %5Q=9a!A$q'9:A   !QaB7 : s9B B#B B c|Dcgc]+}tjtj||-}}t |||d|ycc}w)Nrr; r )rnr*rAroa_rs rxrC zTestPMean.test_2d_axis1)sD IJJ"9,,RXXb\1=JJ!QaB7Ks0Acgdd}}gd}tjtj|||}t ||||d|y)NrH ރBrI rK rO )ru rz rrr rnror*rAr rs rxrQ zTestPMean.test_weights_1d2s>;1,,RXXa[!WE!QtKrzTrF rcgdd}}gd}tjtj|||}t j |||}t |tj|dy)NrH r rI rJ rK rj)ru rz rrrfr r1rd)rnror*rAr rrs rxrM zTestPMean.test_weights_1d_list8sW ;1,,RXXa[!WEkk!Q0RZZ0t.funWs 11!Q@@rzr3r6rYr7rr4)rTr rK rS )rErfrrr ) rnrTr rAror r*r rs ` rxtest_weights_2dzTestPMean.test_weights_2dOs ) ) A%*C HHq!fr1g1v. /((RGaVaV45adG4!QdG$SUVrzN)rrrrx rz r| r) rrrr- r0 rrrr r< rC rQ rrM r'r] r rrzrxru ru s.A@ 0 [[ VW [[ WX:YX: 3  [[ 8b"XCy )4 0 "((RHr2hS 2 3S 9 ; 0  0 [[ -/B CT J ,.A BC H J 8  8 [[ -/B CT J ,.A BC H J 8  8L == =33 [[!  W  Wrzru c eZdZejddzj ZdZejedj Z dZ e ddd Z e d Zej j#d ej j#d ej j%d ddej&ej(fdZdZdZdZdZdZdZdZy)TestGSTDrrgvaZ@r^ctj|j|j}t ||j|j yr)rfr rdarray_1dr1 gstd_array_1drnro gstd_actuals rx test_1d_arrayzTestGSTD.test_1d_arrayhs6jjDMM!:;  RZZ0B0B%CDrzTz$Only NumPy supports array-like inputrctjt|j}t ||j yr)rfr rr r r r s rx test_1d_numeric_array_like_inputz)TestGSTD.test_1d_numeric_array_like_inputls)jjt}}!56  T%7%78rzcd}tjttf|5t j ddddy#1swYyxYw)Nz.could not convert string to float|The DType...rz*You cannot take the logarithm of a string.)rrrr|rfr rs rx#test_raises_error_non_numeric_inputz,TestGSTD.test_raises_error_non_numeric_inputqs@B ]]J 2' B E JJC D E E Es AA rr bad_valuerr_c|j|j|gz}tj|rGt |srrfr r2rm)rnr rorqrrs rxtest_returns_nan_invalid_valuez'TestGSTD.test_returns_nan_invalid_valuews JJt}} {2 3 ;;y !-*:KGnG< $jjm $ $**Q-CRZZ/0  $ $s B;;Cc|jgd|jdddgg}tj|d}t ||jdt jgy)N)rrrrrr3r4rSr5)rdrmrfr r1r)rnror*r s rxtest_propagates_nan_valuesz#TestGSTD.test_propagates_nan_valuessM JJ 1a'89 :jj+  RZZBFF %<=rzc|j|j}tj||jd}t ||j|j y)Nrr)rdr rfr rr2rmrs rx)test_ddof_equal_to_number_of_observationsz2TestGSTD.test_ddof_equal_to_number_of_observationssC JJt}} %jj,RZZ/0rzc|j|j}tj|d}|j|j}t ||yr)rdarray_3drfr r r1rnrorqr rs rx test_3d_arrayzTestGSTD.test_3d_arraysB JJt}} %jj. jj++, S)rzc|j|j}tj|d}|jddg}t ||y)NrrSg~d@g< ?rdr rfr r1r s rxtest_3d_array_axis_type_tuplez&TestGSTD.test_3d_array_axis_type_tuplesA JJt}} %jj0 jj*j12 S)rzc|j|j}tj|d}|jgdgdgdg}t ||y)NrrS)gĶ?@gQӫ@g֌@gן7R@)g%Ȁ@g–e@gv6B5@g_?)gE ?gީ?gk̊Y?g%: ?r rnrorqr gstd_desireds rxtest_3d_array_axis_0zTestGSTD.test_3d_array_axis_0sL JJt}} %jj+ zz O O O#   \2rzc|j|j}tj|d}|jgdgdg}t ||y)NrrS)gN@g85@g'ĥ?gwth?)gHnaX?gG?gBu?gE?r r s rxtest_3d_array_axis_1zTestGSTD.test_3d_array_axis_1sI JJt}} %jj+ zz L L#    \2rzc|j|j}tj|d}|jgdgdg}t ||y)Nr3rS)gt80?gі?g?)gۘ]~?gmT6(?g G?r r s rxtest_3d_array_axis_2zTestGSTD.test_3d_array_axis_2sI JJt}} %jj+ zz ? ?#    \2rzN)rrrrrrr r rhr r rr r'r rrrrrrmr r r r r r r r rrzrxr r as )$q(002H"Mrzz(I.557HEd+QR9S9EE  [[ WX [[ VW [[[1b"&&"&&*AB1CXY1> 1 * * 333rzr c tjtjdddtjdddtjdddf}d}d }gd }t||D]6\}}t t j |||j|d d | 8t t j dddjdd y)Nr rr6r g?r#ffffff?r]i)rrgs gG|G{g ɺM>6 g> ztt+g#ӌR M/gF׸fz2gU8u5g8M8gФ)Ba?g%վ?g*+a5?gś?gڒJ|>r[z fail forp=)rerr_msgrrgVݶ3;r)rrrrr rf binomtestr)pprUrqrarArs rxtest_binomtestr s S#q1T43T435 6B A A(Gb'"F3EOOAq!4;;S(*j4D FFC5<</$&(rzc  ddggdgdgdgdgdgdgdgd gd g }td d D]P}t|d zDcgc]#}tj||d j%}}t |||d z dRycc}w)Nr)r<rr<)rjrrrj)rrrrr)r?rrr r)r??rr r r)rr?rrr rr) ???@?rr r r r ) p???@?rrr r r r ) `???? ?rr r r r r rrZr<rYr)rrfr rr)rrJrrs rxtest_binomtest2r s c  +8G  * E  D"1b\9;@Q<Ha1c*11HHD$qs)R89Hs(Bc tddDcgc]:}tddD])}tj|||zd|z j+<}}}t |t j t|tt jgd}t jgd}tddDcgc]=}tddD],}tj|dz||zd|z j.?}}}tddDcgc]=}tddD],}tj|dz ||zd|z j.?}}}t||dt||dycc}}wcc}}wcc}}w) NrrZr3r)Zr<grq?g?glq?gx0f#?g%d)MO?g o?g6?g:X@?rgs w?g?gR'jMV?gEf?gF5a?gAQU?g'W?gk?r g ņB?g?g$?g{dU'?gD:B?gwV?gR0Nf?gOq?r g[k^?gT+?gO?gggh(?gP-@?gOQ?g}^?gV7_i?r ge$?gр?g#>?g5z @?g:?gf?gzP ?g??gg*),?g}A_?g}aA?gH]@_?g y7s?gPG?g>)\R?gjc6?K?go;,?g!O0?g?g}e?g5?g?!A^?gN/#?g*?@?g ?g?g3?g.1?gd T\1-?gt0W59?goQC?gX4J??g ?gf?C"?g\6I?g4b?g|$or?gWNS~?gT?gXnc?\?gY\?gp ^a?gX-K?g#c?g 3?g75Q?gMJ[ ?gP8R?)Zr<gw %?g?gSg?gv^?g*uCi?ga?g]?gE9Q?rg/?g?g x&~?g#?g5B ?g A'9?gpZ?gWV$u?r gqIb?g@N?g1Y?g5?gO?gc-?gk?g)?r glMV?g?$?gT*r?gCqRv?grx?gωZ?gExY?gceZdZdZdZdZy)TestTrimc `tjd}ttjt j |dtjdttjt j |dtjdttjt j |ddtjddttjt j |d dtjd dtt j |d gtt j |d dgtt j gdgtt j gd dgtt j gd gtjd j dd}tjdd j dd}d}t j |dd|}ttj|||d}t j |jdd|}ttj|||jy)NrZr rYrr:left)tailr3tE]t?r4rrrr7r5r6r)r rTrSr)rrrsortrftrim1rhr)rnr*rrTtrimmeds rx test_trim1zTestTrim.test_trim1Fs IIbMRWWU[[C01299R=ARWWU[[C01299Q<@RWWU[[Cf=>YYq"% 'RWWU[[E?@YYq"% 'U[[C("-U[[Cf5r: U[[S)2.U[[U8"=U[[T*B/ IIbM ! !!Q 'ii2&&q!,++a6=RWWW40#6++acc3V$?RWWW40#%%8rzc tjd}ttjt j |dtjddttjt j |dtj gdttjt j tjdjdd dtjd d jd d ttjt j tjdjd djd tj gd gd gtttj tjdjd djdtt j gdgtt j gdgtt j gdgy)NrZr r4r9r)r3r4r5r6r7r8r9rr7r5rr3)r3r9rWr)r4r:rrVrr ) rrrr rftrimbothrrhrrr)rnr*s rx test_trimbothzTestTrim.test_trimbothbss IIbMRWWU^^Au56 !QHRWWU^^As34XX34 6RWWU^^BIIbM,A,A!Q,GMNYYq"%--a3 5RWWU^^BIIbM,A,A!Q,G,I,I,023XX~~>? A j%..iim++Aq133T ; U^^B,b1U^^B.3U^^B-r2rzctjgd}tjgd}tjdjdd|ddf}tjdjddd|ddf}t t j |dtjgd t t j |dtjgd tjgd }tjdjdd|ddf}t t j |dtjgd gd }t t j |ddt t j gdddtjjdtjjdd}dD]Q}t j |d|}t j tj||dd} t || St j |dd}t j |jd} t || ttt j |dt t j gdtjt t j gdtjy)N) r5r9r3rr:r6rYrr8r4r7)r4r6rrr3r5rr7r5Frr3)r>rDg-@g4@r)rrr4r3)rrr8rnrr)r8rZr[rVrr7rrr6rrrYrbr:rrrQrr3rWr5r\r9r4rX)r6r5r4rr3rr>rr)r6r7r5r8r)rrr3r4r_rSrrr)rrrrhrrf trim_meanrr=randintr_rrrrm) rnr*idxrridx4a4rTrrs rxtest_trim_meanzTestTrim.test_trim_meants  HH7 8hh)* YYr] " "1a (a 0 YYr] " "1as " 3CF ;U__R.XX45 7U__R.XX23 5xx % YYr] " "1a (q 1U__R.XX;< > %U__Q-t4U__]D93? t II  b|  4$ %D??1d6D??2;;q$#:DAD t $ % q$T2qwwy$/T4 j%//1c: U__R-rvv6U__R-rvv6rzN)rrrr r r rrzrxr r Ds983$$7rzr c*eZdZdZdZdZdZdZy) TestSigmaClipc$tjtjdddtjdddf}d}tj|\}}}t |j |kDt |j|kt||j||jzz t||j||jzzt|j|jy)Nr;r^rrr6r5 rrrrf sigmacliprminmaxrrirrrnr*factr,rWupps rxtest_sigmaclip1zTestSigmaClip.test_sigmaclip1s NNBKKT26 Ar18MN Oooa( 3# # S!&&(T!%%'\12S!&&(T!%%'\12QVVQVV$rzc@tjtjdddtjdddf}d}tj|||\}}}t |j |kDt |j|kt||j||jzz t||j||jzzt|jdt|jd y) Nr;r^r rrr6r=r5$r r s rxtest_sigmaclip2zTestSigmaClip.test_sigmaclip2s NNBKKT26 Ar18MN Oooat4 3# # S!&&(T!%%'\12S!&&(T!%%'\12QVVQQVVR rzc*tjtjdddtjdddf}d}tj|||\}}}t |j |kDt |j|kt||j||jzz t||j||jzzt|tjdddy)Nr;r^rZr4?) rrrrfr rr r rrirr s rxtest_sigmaclip3zTestSigmaClip.test_sigmaclip3s NNBKKT26KKc157 8ooat4 3# # S!&&(T!%%'\12S!&&(T!%%'\12Q Cr23rzctjtjdddtjdddf}d}tj|||}d}t ||y) Nr;r^rZr r r4r )clippedrr)rrrrfr r)rnr*r rr s rx test_sigmaclip_result_attributesz.TestSigmaClip.test_sigmaclip_result_attributessZ NNBKKT26KKc157 8ooat,2 C,rzcrtjd}ttj|d|y)NrYr)rrrrfr rCs rx test_std_zerozTestSigmaClip.test_std_zeros' GGBKU__Q'*A.rzN)rrrr r r r r rrzrxr r s% ! 4-/rzr ceZdZdZej j dddggfddgdfddgdgfgdZdZdZ dZ d Z d Z d Z d Zd Zy)TestAlexanderGovernc`gdgdgdgdg}|Dcgc]'}tj|tj)}}|Dcgc]'}tj|tj)}}|Dcgc]'}tj|tj)}}|Dcgc]'}tj|tj )}}t j|}t j|}t j|} t j|} |j|jcxk(r | jcxk(r| jk(sJJ|j|jcxk(r | jcxk(r| jk(sJJycc}wcc}wcc}wcc}w)N) r\r\r\r\r\r\r\r[r[) rWr\r[r[r[r[r[rZrZ) rWrWr\r\r\r\r\r[r[) rrWr\r\r\r[r[r[rZrH) rrdrrr{rrfalexandergovernrr) rnr7arg args_int16 args_int32 args_uint8 args_float64 res_int16 res_int32 res_uint8 res_float64s rxtest_compare_dtypesz'TestAlexanderGovern.test_compare_dtypess|44446BFF#bjjBHH5F FAEF#bjjBHH5F FAEF#bjjBHH5F FEIJc 3bjj9J J)):6 )):6 )):6 ++\:   I$4$47  7$/$6$67 87 87##y':':=##='2'<'<= >= >=GFFJs,F,F!7,F&),F+r|rr3c tjtt5t j |}t |jtjt |jtjdddy#1swYyxYwrJ) rr>rrrfr rrrrmrrNs rxtest_too_small_inputsz)TestAlexanderGovern.test_too_small_inputss]\\,4I J -''.C  / RVV , - - -rPcVtjd5tjddgtjtjg}dddt j tjt |jtjy#1swYRxYw)Nrrrr3) rrrfr rrrrmrrs rxtest_bad_inputsz#TestAlexanderGovern.test_bad_inputssn [[ * B''A0@AC BS]]BFF+SZZ( B Bs 7BB(c gd}gd}gd}tj|tj|tj|}}}tj|||}t |j dt |j dy)a Data generated in R with > set.seed(1) > library("onewaytests") > library("tibble") > y <- c(rnorm(40, sd=10), + rnorm(30, sd=15), + rnorm(20, sd=20)) > x <- c(rep("one", times=40), + rep("two", times=30), + rep("eight", times=20)) > x <- factor(x) > ag.test(y ~ x, tibble(y,x)) Alexander-Govern Test (alpha = 0.05) ------------------------------------------------------------- data : y and x statistic : 1.359941 parameter : 2 p.value : 0.5066321 Result : Difference is not statistically significant. ------------------------------------------------------------- Example adapted from: https://eval-serv2.metpsy.uni-jena.de/wiki-metheval-hp/index.php/R_FUN_Alexander-Govern )(g5gb?gl"k g^s/@gQ\ @gRh gIF@gpZ>q@g7D@g-ѪZng(Q<.@gON/@g Ogҷ%6gs&@gw)ܿg789Ŀgzp"@gz l @g57@g>m@*a"@g,ZbI@gX9 ]?g3g1@gg"Jbgy:@j-g2A g;@gL97vp,+@gp/rg"@gfT|7g^v+g٘Yg0/gglA!&@go@)g!_5*gq TEgg1$@g a, @gˤ$g9%gZ@gT6U'@gS g6aGn*@gO@g\\"g^4w@g0gr~5@g DP=@gٿԙg:4R/gOx]r!@gU4gta)B@g,Gg2$@gD%?gI,VL&g@gч ;g 5@gWJc@gYjK@@)gj7#@ggIK'@g?Ң@g7C5@g;C>ϯUg7@g)n}.^@gFQ?gsdlT6?Nrrdrfr r rr)rnrVtwoeightsolns rxtest_compare_rz"TestAlexanderGovern.test_compare_rsr: " I7**S/2::c?BJJu %;rrfr rrrmr)rnrrrs rxrBz'TestAlexanderGovern.test_constant_inputsz ZZ- . ZZ- . \\.0L M 0''B/C 0S]]BFF+SZZ( 0 0s B22B;N)rrrr rrrr r r r# r% r( r* r0 rBrrzrxr r sz>( [[Vq!fb\QFA;!Q! $NO-P-)B=H11f10f,= >)rzr c eZdZdZdZdZdZdZdZdZ dZ e jjd ejgd ejgd ej d ffejgd ejgd ej"ej"ffgd Ze jjdgddZdZdZdZdZe jjddgdfgdZdZdZy) TestFOneWaycntjddgddg\}}t|dt|dy)Nrr3rr)rff_onewayrrnr rAs rx test_trivialzTestFOneWay.test_trivials2~~q!fq!f-1QQrzctjddgddg\}}t|dt|dt j dz dy) Nrr3r5rrr<r+rj)rfr5 rr rr'r6 s rxr5zTestFOneWay.test_basicsC~~q!fq!f-1Q1rwws|+%8rzcgdgdgdg}tj|ddi\}}t|ddt|d dy) Ng(\5IgHz'D@gףp= 2g ףp=A@g333333g)\(?gq= ףp7@r@g333333.@gRQ#@gzG'@rg= ףp#@gGz."@gp= ף @rFgM;yQ?r+rjgWe?rfr5 r )rnsamplesr rAs rxtest_unequal_varzTestFOneWay.test_unequal_varsG=+35~~w8%81& ,59,59rzcgdgdgdg}d}tjt|5tj|ddidddy#1swYyxYw)Nr: r; r< z(Expected a boolean value for 'equal_var'rrFalse)rrr|rfr5 )rnr> rs rxtest_equal_var_input_validationz+TestFOneWay.test_equal_var_input_validationsM=+35= ]]9G 4 8 NNG 7w 7 8 8 8s AAcxtjdgdggd\}}t|ddt|ddy)Nr3r^rr#rjrr= r6 s rxtest_known_exactzTestFOneWay.test_known_exacts6~~qcA3 213U+3U+rzctjddgtj}tjddgtj}tj||\}}t |ddy) NrHgŸ?r+rj)rruint16rfr5 r )rnr*r4r rAs rxtest_large_integer_arrayz$TestFOneWay.test_large_integer_arraysU HHc3Zryy 1 HHc3Zryy 1~~a#1 .U;rzctjddgtj}tjddgtj}tj||}d}t ||y)NrF rG rHrH rI r)rrrJ rfr5 r)rnr*r4rr s rxtest_result_attributesz"TestFOneWay.test_result_attributessQ HHc3Zryy 1 HHc3Zryy 1nnQ", C,rzcgd}|D]l}d}tjjtjjtjj t d|}t |5}|jjd}dddddDcgc]"}|jr|j$}}tj|d} | j\} } | jt} tj| } t!|d d }| D cgc] } | | | k( }} t#j$|}d }||vrd }t'|d ||d |oy#1swYxYwcc}wcc} w)N) z SiRstv.datz SmLs01.datz SmLs02.datz SmLs03.datz AtmWtAg.datz SmLs04.datz SmLs05.datz SmLs06.dat SmLs07.dat SmLs08.dat SmLs09.datrzzdata/nist_anova rrrd)skiprowsrr_)rO rP rQ rszFailing testcase: )rkr )ospathabspathjoindirname__file__openreadsplitstriprloadtxtrrruniquerrfr5 r )rn filenames test_caserkfnamer9contentline certifieddatafrrrqcatyrxlistrhard_tcs rx test_nistzTestFOneWay.test_nistsi? # FIDGGOOBGGLL1J1BI%OPEe /&&(... /29"R.*$ JJL*I*JJur2E77DAq A99Qrfr?r5 )rnr*r4rrr9rAs rxrBzTestFOneWay.test_constant_input sW 6 \\%44C @ $>>!Q'DAq #a8m #1 $ $ $s &AA!rT)rr_rrc htjgdgdgdgdgdg}tjgdgdgdgdgd gd gd gd g}tjgd gdgdgdg}|dvr'|j}|j}|j}d}nd}d}tjt j |5t j||||\}}ddddD]}} t jtj|| |tj|| |tj|| |\} } t| | dt| | ddD]} tjt j |5t jtj|| |tj|| |tj|| |\} } t| | t| | dddy#1swY9xYw#1swYxYw)N)rr5r4r4)r3r6r4r4)r4r7r4r4rx)r4rr6r4)r5r7r6r4)r5r4r6r4)rr6r6r4)r6r6r6r4)r3r4r6r4)r9r3r6r4)r3r3r6r4)r5r4r5r4)r5r3r5r4)r6r5r5r4r_rrrrl rrSrr+rjr) rrrrr>rfr?r5 taker r) rnrTr*r4r, take_axisrr9rArWfjpjs rxtest_2d_inputszTestFOneWay.test_2d_inputs+s HHl"""" $ % HHl"""""""$ % HHl"""$ % 7?AAAII: \\%44H E 6>>!Q5DAq 6  2A^^BGGAq)$<$&GGAq)$<$&GGAq)$<>FB AaD"5 1 AaD"5 1  2  'Ae88I '1i(@(*1i(@(*1i(@BBQqT2&QqT2&  ' ' ' 6 6 ' 'sHA8H(H%(H1 c dtjddjdddz }dtjddjdd dz }tjdtjdd jdddz }t j |||d \}}|j d k(sJ|j d k(sJt|j d D]s}t|j dD]V}t j ||dd|f||dd|f||dd|f\}} t||||ft| |||fXuy)Nrrr5r6r8r3r9qrS)r5r8r) rrrhcosrfr5 rrr ) rnr*r4r,r9rArrWfijpijs rxtest_3d_inputszTestFOneWay.test_3d_inputs[sC biiY'//1a8 8 biiY'//1a8 8 FF1RYYsI.66q!Q?? @~~aAA.1ww&   ww&   qwwqz" .A1771:& . >>!Aq!G*a1aj!Aq!G*MSQq!tW-Qq!tW- . .rzctjtt5t j gdggd}t |tjtjfdddy#1swYyxYw)Nrr}r9) rr>rrrfr5 rrrmrs rxtest_length0_1d_errorz!TestFOneWay.test_length0_1d_errorlsP \\,4I J 3^^Ir<@F "&&"&&!1 2 3 3 3s AA//A8ctjtt5d}t j d|f}t j d|f}t j d|f}t j|||\}}t j|ftj}t||t||dddy#1swYyxYw)Nrr4r5rr6r-) rr>rrrrrfr5 rrmr)rnncolsr*r4r,r9rArs rxtest_length0_2d_errorz!TestFOneWay.test_length0_2d_errorrs \\,4I J "EE #AE #AE #A>>!Q*DAq77E87D D ! D ! " " "s B C  Cctjt5tjdgdgdgdg}t |t jt jfdddy#1swYyxYw)NrYrZr[r\)rr>rrfr5 rrrmrs rxtest_all_length_onezTestFOneWay.test_all_length_one}sW \\, - 3^^RD2$rd;F "&&"&&!1 2 3 3 3s AA**A3r7rr}cxd}tt|5tj|dddy#1swYyxYw)Nz$At least two samples are required...r)rr|rfr5 )rnr7rs rxtest_too_few_inputszTestFOneWay.test_too_few_inputss28 9G 4 " NND ! " " "s09ctjd}tjd}tt5t j ||ddddy#1swYyxYw)Nr1r6r5r3rS)rrrr&rfr5 rnr*r4s rxtest_axis_errorzTestFOneWay.test_axis_errorsI GGFO GGFO 9 % ) NN1aa ( ) ) ) AA%ctjd}tjd}tt5t j ||ddddy#1swYyxYw)Nr1r rrS)rrrrrfr5 r s rxtest_bad_shapeszTestFOneWay.test_bad_shapessI GGFO GGFO : & ) NN1aa ( ) ) )r N)rrrr7 r5r? rB rD rK rM rj rrrrrrrmrBrs r{ r} r r r r r rrzrxr3 r3 s 9:88,<-F@ [[- , )!4rvvqkB , ,!7"&&"&&9IJ0  $  $ [[V^4-'5-'^."3 "3  [[Vb9,%78"9" ) )rzr3 cHeZdZdZdZdZdZdZdZdZ dZ d Z d Z y ) TestKruskalcdg}dg}tj||\}}t|dt|tjj j |dtjtj|tj|\}}t|dt|tjj j |dy)Nrr3r) rfkruskalrr rurgrvrrrnrqrrhrAs rx test_simplezTestKruskal.test_simples C C}}Q"1QAu2277::1a@A}}RXXa["((1+61QAu2277::1a@Arzcgd}gd}tj||\}}t|ddt|tjjj ddtjt j|t j|\}}t|ddt|tjjj ddy)Nrr4r6r8r:r3r5r7r9rYr rYrr)rfr r rurgrvrrr s rxr5zTestKruskal.test_basics  }}Q"1Au"5Au2277::5!DE}}RXXa["((1+61Au"5Au2277::5!DErzcndg}ddg}d}d}||z }tj||\}}t||y)Nrr3r r:)rfr rrnrqrrh_uncorrrVrr rAs rxtest_simple_tiezTestKruskal.test_simple_tiesE C F*d?}}Q"1 Q!rzcgd}gd}d}dtddz z }||z }tj||\}}t||y)N)rrrr3)r3r3r3r3rr)rrfr r r s rxtest_another_tiezTestKruskal.test_another_tiesM  >5,-::d?}}Q"1Ax(rzcgd}gd}ddg}d}dtddz z }||z }tj|||\}}t||t|tjj j |dy)Nr<rr3rLrr r )rrfr r rurgrv) rnrqrrrr rVrr rAs rxtest_three_groupszTestKruskal.test_three_groupss{   FG5,-::d?}}Q1%1Ax(Au2277::1a@Arzcgd}gd}g}tjtt5t t j |||tjtjfdddy#1swYyxYw)Nr<rr) rr>rrrrfr rrm)rnrqrrrs rxrFzTestKruskal.test_emptys^    \\,4I J C q!Q/"&&"&&1A B C C Cs AA33A<c\gd}gd}tj||}d}t||y)Nr r r)rfr r)rnrqrrrr s rxtest_kruskal_result_attributesz*TestKruskal.test_kruskal_result_attributess+  mmAq!, C,rzctjd}tj|d<tt j ||tjtjft t j ||ddtttj ||dtttj ||dy)Nrr:rrrrr) rrrmrrfr rrrrCs rxrFzTestKruskal.test_nan_policys} IIcNvv!U]]1a(266266*:;EMM!Q6BJOj%--A'Jj%--A(Krzcd}tjj|}tjj|dz}tj||\}}d}t ||y)Nrrr)rrr>rfr r )rnrUrqrrr rArs rxtest_large_no_samplesz!TestKruskal.test_large_no_samplessS  IIOOA  IIOOA  #}}Q"1Ax(rzcd}tjt|5tjdddy#1swYyxYw)Nz-Need at least two groups in stats.kruskal\(\)r)rrrrfr rs rxtest_no_args_gh20661z TestKruskal.test_no_args_gh20661s3B ]]:W 5  MMO   s <AN) rrrr r5r r r rFr rFr r rrzrxr r s9BF ") BC-L)rzr ceZdZejj dgddZejj dgdddfgdd d fgd Zgd Zejj d gdejj dedZ ejj dedZ ejj deejj dddgdZ y)TestCombinePvaluesz+method, expected_statistic, expected_pvalue))rg'"x-@g({#H?)stoufferv!: @/?)rqgQ4gF?)tippettrgL^i?)mudholkar_georgegzP@guKqN?cgd}tj|j||}t|j|j|t|j |j|y)Nrrrr)rfcombine_pvaluesrdr1rr)rnrorrrrqrs rxtest_reference_valuesz(TestCombinePvalues.test_reference_valuessR ##BJJqM&A rzz2D'EF BJJ$?@rzz,weights, expected_statistic, expected_pvaluerrrr r )rrrga;?grmT?c |jgd}tj|d|j|}t|j|j|t|j |j|y)Nr r )rr )rdrfr r1rr)rnror rrrqrs rxtest_weighted_stoufferz)TestCombinePvalues.test_weighted_stouffer s^ JJ} %##Aj"**WBUV rzz2D'EF BJJ$?@rz)rrqr r r r)singlerrrc d\}}tjjd}|dk(rr|j|j |j|||f}|j |j |jdd|d|ddddfgd}np|d k(r0|j|jdd|||fj}n;|d k(r6|j|j |jd d||f d }|j tjd Dcgc],}ttj||ddf| d.c}} |j!| dd| ddz d k\sJycc}w)N)rYr8l TrU( "b-=r r r rn rrSrrrrrr_)rrrrrdrrhrrr rmrrrrfr r) rnrrrorTrUrpvaluessrcombined_pvaluess rxtest_monotonicityz$TestCombinePvalues.test_monotonicity s~1ii##$KL h rzz#**Q-'@1a&IHyy"**R[[c1-Ew"O"*1ab5/"39:!>!,-'  %''AvFqI J'  vv&qr*-=cr-BBaGHHH ' s11F ctj|jgd|}t|j|dt|j |dy)Nr rrr)rfr rdr2rr)rnrrors rxrzTestCombinePvalues.test_result5 sA##BJJ}$=fM s1v. CF+rzrTrrcptjjd}|j|jd}|dk(r |jn|}t j |||}|j|j|d}t j ||}t|j|jt|j|jy|dk(r |jn|}|dddf|dddf} }t j ||} t j | |} t|jd| jt|jd| jt|jd| jt|jd| jy) Nl2'$FVr3rYrr)rTrrrr) rrrrdrrfr rhr1rr) rnrrTrorrqrrrrref0ref1s rxr@zTestCombinePvalues.test_axis; sL ii##O4 JJszzwz/ 0AIACCA##AD@ < 1e$A''&9C CMM3== 9 CJJ 3 AIACCA1a4!AqD'B$$R7$$R7 a($..9 a($..9 1 t{{3 1 t{{3rzN) rrrrrrr r rr rr@rrzrxr r s- [[5 GHAHA  [[ 7 )+> ? )+= > @ AA AA OG [[Y(CD [[Xw/I0EI0 [[Xw/,0,  [[Xw/ [[VaV,4-04rzr c.eZdZdZdZdZdZdZdZy)TestCdfDistanceValidationzg Test that _cdf_distance() (via wasserstein_distance()) raises ValueErrors for bad inputs. c tttjdgdgdgddgtttjdgdgddgy)Nrr3r5r4rrrrfwasserstein_distancer8s rx&test_distinct_value_and_weight_lengthsz@TestCdfDistanceValidation.test_distinct_value_and_weight_lengths] sG j%"<"<cA3aV -j%"<"? ]]:-O P Bzz{z3Hzzyz1H--hAA B]]:-O P Bzzuz-Hzzwz/H--hAA B]]:<> Bzzwz/Hzzvz.H--hAA  B B B B B B B Bs#;D;;E7;E;EEEu_size)rrYrv_sizectjjd}|j|df}|j|df}|j}|j}t j ||}t j ||} t j ||} t| |t| ||j|} |j|} t j ||| | }t j ||| | } t j ||| | } t| |t| |y)NlKu@*rr)rrrrrfr r r ) rnr r rr r u_values_flat v_values_flatd1d2d3r r s rxtest_optimization_vs_analyticalz9TestWassersteinDistanceND.test_optimization_vs_analytical!s#ii##K0::FA;:/::FA;:/ (  (  ' ' } E  * *8X >  * *=- HBBJJFJ+ JJFJ+  ' ' }(19>  * *8X(19>  * *=-(19>BBrzN)rrrrr rrrr r r r r r r r rrzrxr r } s"& [[Y 4 [[VY/ [[]G4,505," [[T:. [[T;/ [[VZ0 "10/ " [[T;/ [[T;/ [[VY/"000" + [[T<0 [[T<0 [[VY/"011"" [[T;/ [[T;/ [[VY/"000"$B  [[X{3 [[X{3 44 rzr c:eZdZdZdZdZdZdZdZdZ dZ y ) TestWassersteinDistancez5 Tests for wasserstein_distance() output values. c<ttjddgdgddgdgdttjddgdgddgdgdttjddgdgddgdgdttjgdgddy) Nrrr<r4rjr3rmr}r rfr r8s rxr z#TestWassersteinDistance.test_simple0!s   & &1vsQFQC @   22 FQC!Q!&   22 FQC!Q!&   22 y"  rzc ttjgdgddttjgdddggdddgdy Nr}r_r)rrrr5r5rrr4)rrfr r8s rxtest_same_distributionz.TestWassersteinDistance.test_same_distribution@!sE U// 9EqI  & &|aV'3aV =  rzc (ttjdgdgdttjdgdgdttjgdgddttjgdgd gd gd d y) Nrrrr6rYr)rZr[r\rWr)r@g@r )rr8r)r4rr)rr4rr>r r8s rx test_shiftz"TestWassersteinDistance.test_shiftI!sz 22A3  & &y)'0) =  >rzctjddd}tj|}tt j ||tj tj|tjt|}|ddd}tt j ||||tjtj||y)Nr4rrr_rJ ) rrrr rfr rirfrzr )rnrrr r s rx test_collapsez%TestWassersteinDistance.test_collapse`!s IIc2s # MM!   & &q! , GGBFF1I  IIc!f% ddO   & &q!Y B JJrvvay) 4 6rzc ttjgdddggdddgtjddgddgddgddgyN)rr3rrr,r3r r8s rxr z(TestWassersteinDistance.test_zero_weightp!sN  & &~1v'01a& :  & &1v1v1v1v F Hrzcttjddtjgddgtjttjddtjgtj dgtjttjdtj tjgddgtjt 5}|j tdttjddtjgtjdgtjdddy#1swYyxYwNrr3r) rrfr rrr rrrmrnrs rxr z'TestWassersteinDistance.test_inf_valuesw!s   & &1bff~1v > FF    & &1bff~| D FF    & &BFF7BFF';aV D FF   C JJ~'7 8 **Aq"&&>BFFA;G     ,A&EE$N) rrrrr r r r r r r rrzrxr r ,!s+  >6 Hrzr c4eZdZdZdZdZdZdZdZdZ y) TestEnergyDistancez0 Tests for energy_distance() output values. cttjddgdgddgdgtjddzttjddgdgddgdgtjddzttjddgdgddgdgdttjgdgd tjdd zy) Nrrr3r<r4rjrrmr}rrrfenergy_distancerr'r8s rxr zTestEnergyDistance.test_simple!s   ! !1a&1#1vs ; GGAJO  E11 FQC!Q!& GGAJ   E11 FQC!Q!&     ! !)Y 7 GGAJ* * ,rzc ttjgdgddttjgdddggdddgdyr )rrfr r8s rxr z)TestEnergyDistance.test_same_distribution!sB U**9i@!D  ! !,A q!f M  rzcttjdgdgtjdttjdgdgtjddzy)Nrrr3rr6gS[:XL @r r8s rxr zTestEnergyDistance.test_shift!sS E111#s;RWWQZH  ! !2$ , GGAJ  !rzc ttjgdgdgdgdtjgdgdgdgdyr rrfr r8s rxr z'TestEnergyDistance.test_combine_weights!s<   ! !"79N"79N P  ! !)Y 9 M Orzc ttjgdddggdddgtjddgddgddgddgyr r# r8s rxr z#TestEnergyDistance.test_zero_weight!sK  ! !.1a&)aV L  ! !1a&1a&1a&1a& A Crzcttjddtjgddgtjttjddtjgtj dgtjttjdtj tjgddgtjt 5}|j tdttjddtjgtjdgtjdddy#1swYyxYwr ) rrfr rrr rrrmr s rxr z"TestEnergyDistance.test_inf_values!s U**Aq"&&>Aq6BBFFK  ! !1a.BFF7A, ? FF    ! !1rvvgrvv"6A ? FF   C JJ~'7 8 %%q!RVVnrvvqkB    r N) rrrrr r r r r r rrzrxr r !s', !OC rzr ceZdZgdZgdZdZdZdZdZdZ dZ d Z e jjd d gid giggd gdZdZdZdZdZdZy)TestBrunnerMunzel)rr3rrrrrrrrr3r5rr r4r4r5r4rr3r4rrr6r5r\ctj|j|jd\}}tj|j|jd\}}tj|j|jd\}}tj|j|jd\}}t |||j t |||j t ||k7t |d|j t |d|j t |d|j t |d|j t |d|j t |d|j y) Nrbrhrdr| & @| & gҏGg?g@-pL?)rf brunnermunzelrerr rrrs rxtest_brunnermunzel_one_sidedz.TestBrunnerMunzel.test_brunnermunzel_one_sided!s6$$TVVTVVHB$$TVVTVVKB$$TVVTVVKB$$TVVTVVHBB0@0@AB0@0@AbB 2(,(8(8 :B 3(,(8(8 :B 2(,(8(8 :B 3(,(8(8 :B 5(,(8(8 :B 3(,(8(8 :rzctj|j|jd\}}tj|j|jd\}}t |||j t |d|j t |d|j t |d|j y)Nr`rhrr* r+ ҏGw?rfr, rerr rrs rxtest_brunnermunzel_two_sidedz.TestBrunnerMunzel.test_brunnermunzel_two_sided!s$$TVVTVVMB$$TVVTVVMBB0@0@AB 2(,(8(8 :B 3(,(8(8 :B 5(,(8(8 :rzcxtj|j|j\}}tj|j|j\}}t |||j t |d|j t |d|j t |d|j y)Nrr* r+ r/ r0 rs rxtest_brunnermunzel_defaultz,TestBrunnerMunzel.test_brunnermunzel_default!s$$TVVTVV4B$$TVVTVV4BB0@0@AB 2(,(8(8 :B 3(,(8(8 :B 5(,(8(8 :rzc d}d}d}t|dvtttj|j |j |||y)NrrrRr!rrrrfr, rerrnri distributionrs rx$test_brunnermunzel_alternative_errorz6TestBrunnerMunzel.test_brunnermunzel_alternative_error"sI    #CCDj))ffff!"  "rzctj|j|jd\}}tj|j|jd\}}t |||j t |d|j t |d|j t |d|j y)Nrr7 rr* r+ g"H[?r0 rs rx$test_brunnermunzel_distribution_normz6TestBrunnerMunzel.test_brunnermunzel_distribution_norm"s$$TVVTVV(KB$$TVVTVV(KBB0@0@AB 2(,(8(8 :B 3(,(8(8 :B 5(,(8(8 :rzc d}d}d}t|dvtttj|j |j |||y)Nr`rrR)rrr5 r6 s rx%test_brunnermunzel_distribution_errorz7TestBrunnerMunzel.test_brunnermunzel_distribution_error"sH!    ?23j))ffff!"  "rz kwarg_updaterrrqrcX|j|jd}|j|tjt t 5tjdi|\}}dddttjttjy#1swY>xYw)Nrrr) rerrrr>rrrfr, rrrm)rnr> r8rrs rxtest_brunnermunzel_empty_imputz0TestBrunnerMunzel.test_brunnermunzel_empty_imput)"s{vvDFF+ l# \\,4I J > % 3 3 =f = Iv >Y'VRVV$ > >s B  B)cddddddddddddddtjg}gd}tj||d\}}tj||d\}}t |tjt |tjt |tjt |tjy)Nrr3r5r( rRr)rrmrfr, rrnrerrrrris rx&test_brunnermunzel_nan_input_propagatez8TestBrunnerMunzel.test_brunnermunzel_nan_input_propagate3"s 1aAq!Q1aArvv > -$$QkBB$$QkBBR R R R rzcddddddddddddddtjg}gd}d}d}d}tttj |||||tttj |||||y)Nrr3r5r( r`rr)rrmrrrfr, )rnrerrir7 rs rx"test_brunnermunzel_nan_input_raisez4TestBrunnerMunzel.test_brunnermunzel_nan_input_raise>"s 1aAq!Q1aArvv > -!   j))!"  " j))!"  "rzcvddddddddddddddtjg}gd}tj||d\}}tj||d\}}t |||j t |d|j t |d |j t |d |j y) Nrr3r5r( rrrr* r+ r/ )rrmrfr, r rrB s rx!test_brunnermunzel_nan_input_omitz3TestBrunnerMunzel.test_brunnermunzel_nan_input_omitT"s 1aAq!Q1aArvv > -$$Qf=B$$Qf=BB0@0@AB 2(,(8(8 :B 3(,(8(8 :B 5(,(8(8 :rzcgd}gd}d}tjt|5tj||ddddy#1swYyxYw)z| tests that a warning is emitted when p is nan p-value with t-distributions can be nan (0/0) (see gh-15843) r}rLzDp-value cannot be estimated|divide by zero|invalid value encounteredrrr: N)rr>rrfr, rs rxtest_brunnermunzel_return_nanz/TestBrunnerMunzel.test_brunnermunzel_return_nanb"sH  T \\. 4 8   13 7 8 8 8s AAcgd}gd}tjtd5tj||d\}}dddt dy#1swYxYw) zo tests that a p is 0 for datasets that cause p->nan when t-distribution is used (see gh-15843) r}rLzdivide by zerorrr: Nr)rr>rrfr, r)rnrqrrrvrAs rxtest_brunnermunzel_normal_distz0TestBrunnerMunzel.test_brunnermunzel_normal_distm"sX   \\.0@ A D&&q!(CDAq DQ D Ds AAN)rrrrerrr- r1 r3 r8 r; r= rrrr@ rC rE rG rI rK rrzrxr' r' !s2A)AK:. : : " : " [[^sBi#r46R.@.BC%C% !", : 8 rzr' c eZdZdZdZej jdgdgdddej dd gd d d ejd ggdZ ej jdgdgdgdZ ej jdddgej jdd d gej jdd dgdZ dZ y)TestQuantileTestzb Test the non-parametric quantile test, including the computation of confidence intervals c(gd}d}tjt|5tj|gdddd}tjt|5tj|ddgdddd}tjt|5tj|d d g dddtjt|5tj|d dddtjt|5tj|d dddd }tjt|5tj|ddddd}tjt|5tj|j ddddy#1swYxYw#1swYTxYw#1swY"xYw#1swYxYw#1swYxYw#1swYxYw#1swYyxYw)Nr}z/`x` must be a one-dimensional array of numbers.rz`q` must be a scalar.rr3)rbz-`p` must be a float strictly between 0 and 1.r<r:)rArz`alternative` must be one of...z one-sidedrhr))rrrrf quantile_testrT)rnrqrs rxtest_quantile_test_ivz&TestQuantileTest.test_quantile_test_iv~"s C ]]:W 5 %    $ %* ]]:W 5 -   aV , -B ]]:W 5 2   c4[ 1 2 ]]:W 5 (   Q ' ( ]]:W 5 +   T * +4 ]]:W 5 <   { ; <I ]]:W 5 :    " 6 6q 9 : :) % % - - 2 2 ( ( + + < < : :sSF=G GG$G0G<%H=G GG!$G-0G9<HHzp, alpha, lb, ub, alternative)rr gX݊?g蜶?r`)r<r g-tA?gP"?r`rjr g{gA?rbrr gg@rdctjtjddd}tj|||}t |j |||gdy)Nrrr)rArir#rj)rrrrfrO r rT)rnrAr+lbubrirqrs rxtest_R_ci_quantilez#TestQuantileTest.test_R_ci_quantile"sN FF299Qd+ ,!!!qkB//6RuMrzzq, p, alternative, ref)rrr`gͯI ?)r r<r`gɍQ%e?ctjtjddd}tj||||}t |j |dy)NrrrrbrArir;rj)rrrrfrO r r)rnrbrArirrqrs rx test_R_pvaluezTestQuantileTest.test_R_pvalue"sE FF299Qd+ ,!!!qA;G Ce4rzr| continuousdiscreterir+ctdt|zt|z|z}tjj |}|dk(r(|jd\}}|jd}n6|j ddd}|j}|j dd}t j||||} | j| } |d k(r|| jk} n|| jk\} || D]/} t j|| ||} | jd|z kDr/J|| D]/} t j|| ||} | jd|z kr/Jy) Nr8rX r3rrrrZrV r$rb) rrzrrrrrfrO rTrXrWr) rnr|rir+r=rrArbrrrri_insiderqs rxtest_pval_ci_matchz#TestQuantileTest.test_pval_ci_match"scAs4yL3{#33U:;ii##D) < ::1:%DAq**#*&C,,q"3,/C A Q#A!!#a[I  $ $e $ < & bgg~Hbff}HX *A%%cQ!MC::E ) )) *hY *A%%cQ!MC::E ) )) *rzc<gd}d}tj|ddd}t|j|dd gd zd gd zz}tjd dj d }tj|dd d}t|j|dy)N)rgeX?ri r:r`rV rK rj;r9rUhr<p)rArUrrdrdrr$)rfrO r rrr#)rnrqpvalue_expectedrs rxtest_match_conover_examplesz,TestQuantileTest.test_match_conover_examples"s  !!!sd L O$?DFbT5\ !++s377!7<!!!rSiH O%@rzN)rrrrrP rrrrrrT rW r\ rp rrzrxrM rM y"s:6 [[' G F rvvg/ 8 s%rvvy 9 ; N N [[ 2 1 3 5  5 [[VlJ%?@ [[]VY,?@ [[WsDk2$*3AA$*LArzrM c eZdZejj dejj ddZejj ddZdddd efd d dd efd d dd gdgdgdgfddddgdgdgdgfddddgdgdgdgdgdgdgdgdgdgdgdgdgdgdgfddddgdgdgdgdgdgdgdgdgdgdgdgdgdgd gd!gd!gd!gd!gd!gd!gd!gfd"d#ddgd$gd$gd%gfd&d'ddgd(gd(gd(gd(gd(gd(gd)gd*gd*gd*g fd+d,ddgdgdgdgdgdgdgdgdgdgdgdgdgd gd!gd!gd!gd!gd!gd!gfd-d.ddgd/gd/gd0gd1gfd2d3ddgdgdgdgdgdgdgdgdgdgd gd!g fg Z e jjd4e d5Z d6d7ddgd8gd9gfd:d;ddgd8gd8gd8gd<gd=gfd>d?ddgd@gd@gd@gd@gd@gd@gd@gd@gd@gdAgdBgdBgdBgdBgdBgfgZe jjd4e e jjdCZdDZdEZdFZyG)HTestPageTrendTestrr4rrYi2g:hў?FrigsH?i,0gY@?)rrIr:r`rrl rr r1r+rNrr r rrrUrrgrr)Drrrd"rrAXrVrrr r CErrrrWrNrw rru rw )rr\Ur:Or[\VrV@rr rx rbrYrXrw rw r 6r\Brri gXY?r)r=rg @r6rrZ)r6r5r?rYrrV)rr rYr[rWriLgs˜?T)r5r4r3r)r4r5rr3rggĞ?)r4r3rr_r}g>K=?)r7r6r5r4r3r)rr4r5r6r3r7igqူ?)r6r5r4r3r)r5rr4r3r6rga?ig;?)r8r7r6r5r4r3r)r7r6r8r5r4r3rrug[^6?zL, p, ranked, method, datactjjdtj|||}t ||j t||jt ||jyNr )rankedr rrr=rfpage_trend_testrrr rrrnLrAr rrhrs rx test_accuracyzTestPageTrendTest.test_accuracyC#sP r##DGQ &3::&VSZZ(rzig-lp W?) rYr:r9r8r7r6r5r4r3r) rr9r5r8r7r6r:r4r3rYi*gh[?) r:r3r9r8r7r6r5r4rYrr?ig&]2?)r9r8r7r6r5r4r3r)rr4r6r7r5r8r3r9r:ctjjdtj|||}t ||j t||jt ||jyr r r s rxtest_accuracy2z TestPageTrendTest.test_accuracy2`#sR r##DGQ &3::&VSZZ(rzcFtjjdd\}}tjd|dz}tjj tj|}tjj ||}t j|d}t j|}t j|d}t j|d} t j||} t j|dd|f||} t|j|jt|j| jt|j| jt|j| jy) Nr )rYrrrSTr F)predicted_ranks) rrr=rrrHrfrankdatar rr) rnrTrUr permrhranksrrrres4res5s rx test_optionszTestPageTrendTest.test_optionsi#s  r1))Aqs+yy$$RYYq\2yy~~a#t!,$$U+$$U48$$T%8$$UOL$$U1d7^5DT5JLT^^T^^4T^^T^^4T^^T^^4T^^T^^4rzctjjdgdgdgdgdgdgdg}tj|j}tj dd }t j|d |d }t|jd t|jddt j|d |d }t|jd t|jddy)Nr )rBr o)rrGk)gy)rrSr )r )r5 trr8Fr)r r rigy&1l?r5rrgHPsb?) rrr=rrrrfr rrrr)rnrhr rs rxtest_Ames_assayz!TestPageTrendTest.test_Ames_assay{#s r/Cxx~))Aq/##D4C+79 S]]C(CJJ:##D4C+24 S]]C(CJJ:rzcDttd5tjddddttd5tjgdddttd5tjddgdddttd5tjdgggdddttd5tjtj j dddddttd5tjtj j dddddd}tt|5tjgdgdggd  dddtt|5tjgdgdggd  dddtt|5tjgdgdggd  dddtt|5tjgdgdgd dddttd5tjgdgdgddddttd5tjgdgdgddddttd5tjgdddtjggddddttd5tjgdgdgddddttd5tjgdgdgddddy#1swY;xYw#1swYxYw#1swYxYw#1swYxYw#1swYxYw#1swYPxYw#1swY"xYw#1swYxYw#1swYxYw#1swYxYw#1swYwxYw#1swYNxYw#1swYxYw#1swYxYw#1swYyxYw)Nz`data` must be a 2d array.rrr3zPage's L is only appropriater4z+`predicted_ranks` must include each integerr}rm)rhr )rr3r4)rr3r4r4rz`data` is not properly ranked)rr3r4Tr z`data` contains NaNsFr z`method` must be inekki)rhrz`ranked` must be boolean.)rhr ) rrrfr rrrHrmr|rs rxrz'TestPageTrendTest.test_input_validation#sa :-I J (  ! !$ ' ( :-I J &  ! !" % & :-I J *  ! !1a& ) * :-I J +  ! !QC5' * +:-K L 8  ! !"))..A"6 7 8 :-K L 8  ! !"))..A"6 7 8@ :W 5 =  ! ! 9'=2; = =:W 5 ?  ! ! 9'=2= ? ?:W 5 @  ! ! 9'=2> @ @:W 5 =  ! ! 9'=2; = = :-L M @  ! !9i"8$ ? @ :-L M @  ! !9i"8$ ? @:-C D 0  ! !9q!RVVn"=). 0 0:-B C 1  ! ! 9'=)/ 1 19,G H 1  ! ! 9'=)/ 1 1 1W ( ( & & * * + + 8 8 8 8  = = ? ? @ @ = =  @ @ @ @ 0 0 1 1 1 1sM!M.0M;!N4N 4N"/ N/( N<! O OO#O0=-O=P :P!M+.M8;NNN"N,/N9<O OO #O-0O:=P PPN)rrrrrr=rH data_3_25 data_10_26tsrrrr ts2rr r r rrrzrxrr rr #sIINN1 q"%IB'J $e\9 E %ulJ G $e\ = = =  > ? %ug % # & ( ) "D' , l , l , l , ( ) "D' y)Y 9 y)Y 9 y)Y 9 y) - . "D'  24F G I "D' _o _o _o    "D' y)Y 9 y)Y 9 y)Y 9    !4 !#8 !#8 : ; "D' y)Y 9 y)Y  C DO* BX [[92>)?) #T7 , , . / $dG ,.M ,.M , . / $dG %'? %'? %'? %'? %'? %'? %'? % ' ( C* [[92> [[)?)5$;*/1rzrr iv5z fun, argsc||ddi}||ddi}t|||jd}tjtt j |5||ddddddy#1swYyxYw)NrrrKz#() got multiple values for argumentr)rrK)rrrrr|reescape)r r7rrerrs rxtest_rename_mode_methodr #st t $G $C  #7 #Ddll^> ?C y # 71 T'0111s  A--A6ceZdZdZdZdZdZejjdgdejjddd gd Z ejjddd gd Z y ) TestExpectilectjjd}|jd}tt j |dtj |y)Nr rrr<r+)rrrr rf expectilerirnrrqs rxtest_same_as_meanzTestExpectile.test_same_as_mean#sCii##B' JJBJ 5rwwqzBrzctjjd}|jd}tt j |dtj |y)Nr rrrr )rrrr rfr aminr s rx test_minimumzTestExpectile.test_minimum#Cii##B' JJBJ 3RWWQZ@rzctjjd}|jd}tt j |dtj |y)Nr rrrr )rrrr rfr amaxr s rx test_maximumzTestExpectile.test_maximum#r rzc tjjd}d fd}d}|j|}|j}|j|}tj|||}||||}t ||y)Nlb@lNctj||}d|z |||k<tj||z||z dzzS)Nr-rr3)rr/r")rr*r+r rs rxr z'TestExpectile.test_weights..fun#sC Q51AE Aa1fI66!g+Q 23 3rzctj|tj|f}tj||||fj S)N)bracketr7)rr r rminimize_scalarrq)r*r+r r r s rx expectile2z.TestExpectile.test_weights..expectile2#sEffQi*G++C23UG1DFFGa HrzrYrJ )rrrrfr r ) rnrr rUr*r+r rrr s @rx test_weightszTestExpectile.test_weights#s{ ii##$78 4  H  JJqM **Q-ooa8E7+S!rzr+)rg?r<g/#?rrUrrc htjjd}|j|}dD]8}t t j tj||||:|j}t t j ||z|t j |||zt t j ||z |t j |||z t t j ||z||t j ||z|j|d}|dk(rd }n |dkDrd }nd }|t j tj||z|t j ||t j ||z|j|d }t j ||t j ||ksJ|j|d}d D]b}|t j d|z |z||zz|d|z t j ||z|t j ||zzdt t j | |t j |d|z  y)u See Section 6 of I. Steinwart, C. Pasin, R.C. Williamson & S. Zhang (2014). "Elicitation and Identification of Properties". COLT. http://proceedings.mlr.press/v35/steinwart14.html and Propositions 5, 6, 7 of F. Bellini, B. Klar, and A. Müller and E. Rosazza Gianin (2013). "Generalized Quantiles as Risk Measures" http://doi.org/10.2139/ssrn.2225751 r r)rrr<)rr.r rYr>r<ct||yr)r r*r4s rx assert_opz:TestExpectile.test_expectile_properties..assert_op;$s 1%rzc||ksJyrrr s rxr z:TestExpectile.test_expectile_properties..assert_op?$ 1u urzc||kDsJyrrr s rxr z:TestExpectile.test_expectile_properties..assert_opC$r rzr6)r r<rrN) rrrrr rfr r exponentiallogisticr_)rnr+rUrrqr,rrr s rxtest_expectile_propertiesz'TestExpectile.test_expectile_properties$sO$ii##B' JJAJ   A aA >eL   OO  OOAE / OOAU +a /   OOAE / OOAU +a /   OOAE / / /  LLaRL ( C< &S[    OOBEE!a%L 6 OOAU +ooau- .  JJA1J % OOAU +uq/N N N LLaRL (  A 1a!A# U;177EOOAU334    OOQBe , __Qag . . rzctjjd}|jd|}g}tjdtj dd}tj d|d|dd d z dfD](}|jtj|| *tjtj|dkDsJy) Nr r3)r*rrr<rrrr_r ) rrrparetologspacelog10r rrfr rdiff)rnrUrrqe_list alpha_seqr+s rxtest_monotonicity_in_alphaz(TestExpectile.test_monotonicity_in_alphag$sii##B' JJJ #KKRXXc]C8 UU1iYv2v->)>AB ;E MM%//!59 : ;vvbggfo)***rzN) rrrr r r r rrrr r rrzrxr r #sC A A "4 [[: [[S2t*-_ ._ B [[S2t*-+.+rzr c `eZdZgdZdgdej ej dgZdZe dZ e jjdedd ggd ggzd Ze jjd ed Ze jjdedZe jjdedZe jjdddgdge jjdddge jjdddgdZdZy) TestLMoment)yףp= ?r p= ף?r=r rr=r<rrr {Gz?g@(\?gףp= ? ףp= ?g)\(?gRQ@r{Gz?ffffff@rg{Gz@gRQ?gGz@gQ?(\?r gzG?g ףp= ?gQ?ffffff?gGz @g(\?g(\?gQ?q= ףp?g(\@Q?gGz @r Q?r p= ף@r gp= ף@gq= ףp@gRQ?Gz?g ףp= ?r gHzG?g@r gQ?rr RQ?r gQ?p= ף?gQ@gGz?gQ?gp= ף?q= ףp?r gp= ף?gQ?r r g@r g)\(@gzG@r rgGz?g(\?r333333?)\(?rr gGz@rg(\?g= ףp=?r r gHzG?{Gz?r g ףp= ?gףp= ?r r r gQ?r}g?r r g(\?r gQ?gffffff@gQ?g?r gGz?r g{Gz?gHzG@r r r g\(\?g(\?r g(\?r=)rr3r?za duckcd}tjt|5tjt j |jt jdddy#1swYyxYwNz*`sample` must be an array of real numbers.rrH) rrrrflmomentrrrhrrs rx test_dtype_ivzTestLMoment.test_dtype_iv$sM> ]]:W 5 D MM"((499BMMB C D D Ds AA**A3cd}tjt|5tjt j |jtdddy#1swYyxYwr ) rrrrfr rrrhrrs rxtest_dtype_iv_non_numericz%TestLMoment.test_dtype_iv_non_numeric$sE> ]]:W 5 = MM"((499F; < = = =s 9A  A)rrr_r}cd}tjt|5tj|j |dddy#1swYyxYw)Nz*`order` must be a scalar or a non-empty...rrrrrrfr rh)rnrrs rx test_order_ivzTestLMoment.test_order_iv$s<> ]]:W 5 2 MM$))5 1 2 2 2 "A  ArTcd}tjt|5tj|j |dddy#1swYyxYw)Nz"`axis` must be an integer, a tuplerrSr )rnrTrs rx test_axis_ivzTestLMoment.test_axis_iv$s<6 ]]:W 5 0 MM$))$ / 0 0 0r sortedcd}tjt|5tj|j |dddy#1swYyxYw)Nz`sorted` must be True or False.r)r r )rnr rs rxtest_sorted_ivzTestLMoment.test_sorted_iv$s<3 ]]:W 5 4 MM$))F 3 4 4 4r standardizecd}tjt|5tj|j |dddy#1swYyxYw)Nz$`standardize` must be True or False.r)r r )rnr rs rxtest_standardize_ivzTestLMoment.test_standardize_iv$s<8 ]]:W 5 > MM$)) = > > >r rr5rgFTc.tjgd}|s|ddxxx|dzccc|rtj|jn |j}t j ||||}t ||tj|dz y)N)g0ʻEH?gk?gz[?g+3?r3r)r r )rrdr rhrfr r )rnrr r rrhrs rx test_lmomentzTestLMoment.test_lmoment$s}jjBC Gs1v G%+rwwtyy!mmD%[PSE!21!456rzctj}tj|j}t j |j |}t j |}|jj|k(sJt||dtj}tjgd}t j |j |}t j |j tj}|jjtjk(sJt||dy)Nrsrjrr#) rrrdrhrfr rrIrr int64r)rnrIsamplerrs rx test_dtypezTestLMoment.test_dtype$s DII&mmFMM%01mmF#yy~~&&&St,O,mmFMM%01mmFMM"**56yy~~+++Su-rzN)rrrrhrrmr not_integersr r'r rrrr r r r r r rrzrxr r s$sN NDbffbffh?LD ==  [[WlaR)5M&MN2O2  [[V\2030  [[X|4454  [[]L9>:>  [[Wq!\&:; [[]UDM: [[Xt}576;<7" .rzr ceZdZejj dgdejj dddgejj dddgdZd Zejjd ejjd d Z ejjd dZ e ddZ dZ ejjddZdZdZy) TestXP_MeanrTNrr_)rr3r NTrVFctjjd}|jd}|j|}dx}} |r;|jd}|j|} tj||\}}t || ||} tj ||||} t| |j| y)N-fHrlr)r rTrV)rrrrdrrr#r r1) rnrorTr rVrrqx_xprw_xprrs rxtest_xp_mean_basiczTestXP_Mean.test_xp_mean_basic$sii##N3 JJy !zz!}D  6"A::a=D&&q!,DAqtTxHjjAD8DRZZ_-rzc|jd|jd}}d}tjt|5t ||dddy#1swYyxYw)Nrr6z/Array shapes are incompatible for broadcasting.rrJ )rrrrrr#)rnrorqrrs rxtest_non_broadcastablez"TestXP_Mean.test_non_broadcastable$sMyy~rxx{1C ]]:W 5 # Q " # # #s AA rrc|jgd}t|jgd|}t||j|jt|jgd|}t||jtj t|jgd|}t||jtj y)Nrr rJ )rrr)rrr)rdr#r1rmrr)rnror rs rxtest_special_caseszTestXP_Mean.test_special_cases$s**]+rzz,/ARZZ/0rzz,/ARZZ01rzz,/ARZZ/0rzr>c@|jd}|dk(}|j||j|}t|r2t j t d5t|ddddn1t j td5t|ddddt|}t|d}|j|j}t||t||t|d }|j||}t|||jd }|j||j|}t|jd|d }|j||}t||y#1swYxYw#1swYxYw) Nrr4rrrrrrRrrY)r r)rrrmr*rrr|r#rrdr2rir1r) rnrorqrrrrrr s rxrFzTestXP_Mean.test_nan_policy$s_ IIcNQ HHT2661 %  y0OP 0w/ 0 0z1PQ 0w/ 0{k2jj c"c"qV,ggahS!''"+((41ryy~w6JggahS!/ 0 0 0 0sF FFFrcT|jd}|j|dk(|j|}d}tjt |5t |tjzd}|j|j}t||dddy#1swYyxYwNrr4zAfter omitting NaNs...rrr) rrrmrr>rr#rrdr2rnrorqrrrs rxtest_nan_policy_warnsz!TestXP_Mean.test_nan_policy_warns%s IIcN HHQ!VRVVQ '+ \\. 8 &1rvv:&9C**RVV$C C % & & & ABB'cLd}tjt|5t|j g}|j |j }t ||dddd}tjt|5t|j ggd}|j |j g}t ||dddt|j ggd}|j g}t ||y#1swYxYw#1swYPxYwNz,One or more sample arguments is too small...rz2All axis-slices of one or more sample arguments...rrSr)rr>rr#rdrmr2rnrorrrs rxrFzTestXP_Mean.test_empty#%s@ \\,G < &2::b>*C**RVV$C C % & G \\,G < &2::rd+!4C**bffX&C C % & rzz2$'a0jjnS! & &  & &ADADDD#4ignore:overflow encountered in reduce:RuntimeWarningc|j|jj}tj||gtj}|j |}tj d5t |}tj|}tjj|tjt||j |dddt ||j}|j tjtj|tj}t||y#1swYtxYw)NrHr)over) rrr rrdrr#rirrrr1rrnror x_npr rrs rxr zTestXP_Mean.test_dtype4%shhrzz"&&zz3*BJJ7zz$[[h ' 24.C''$-C JJ # #C 0 CC 1  2t2::.jjD !CDES! 2 2s 4A*EEc|jd}|jd}tt|t|tt||t||y)NrYrrJ )rr2r#rs rx test_integerzTestXP_Mean.test_integerH%sF IIbM IIcN Xa[1A.A0FGrzcXtjjd}|jd\}}}}t|j ||dzz|j ||dzz}tj ||dzz||dzz}t ||j |y)Nr )r5rrrJ )rrrr#rdr r1) rnrorrqrrwxwyrrs rxrz TestXP_Mean.test_complex_gh22404O%sii##N3zz'* 1b"rzz!ad(+RZZRU 5KLjjQrT22:6RZZ_-rz)rrrrrrr r rr rFrr rFr r' rrrzrxr r $s [[V%;< [[Yt 5 [[Z%7.86=. # [[ WX [[ VW 1XY 1 [[9""@& &' &"" [[>"""H.rzr ceZdZejj dgdejj dddgejj dddgejj d d d gd Zd ZdZe ddZ e ddZ dZ ejjddZdZdZy) TestXP_VarrTr rVFTr}rrrrRrc tjjd}|jd}tj}|dk(rB|j|jdkD} tj || <tj }|j|} t| ||||} t5} | jtd|||||} dddt| |j y#1swY%xYw) Nr rlrrr<)rTrVr}rr)rTrVr) rrrrrrmnanvarrdr$r rrr1)rnrorTrVr}rrrqvar_refnan_maskr rrrs rxtest_xp_var_basiczTestXP_Var.test_xp_var_basicX%s ii##N3 JJy !&&  zzqwwz/#5H&&AhKiiGzz!}dZ!+-  LC JJ~'J K!$ KC L RZZ_-  L Ls 3#C::Dct|jddgd}t||j|jy)Nrrr4r|)r$rdr1rm)rnrors rxr zTestXP_Var.test_special_casesq%s2bjj"b*q9RZZ/0rzc\|jd}|dk(}|j||j|}t|}t|d}|j |j}t ||t ||t|d}|j ||}t||y)Nrr4rRrr)rrrmr$rdr2rr1)rnrorqrrrrrs rxrFzTestXP_Var.test_nan_policyv%s IIcNQ HHT2661 %qzq[1jj c"c"aF+ffQuXS!rzrcT|jd}|j|dk(|j|}d}tjt |5t |tjzd}|j|j}t||dddy#1swYyxYwr ) rrrmrr>rr$rrdr2r s rxr z TestXP_Var.test_nan_policy_warns%s IIcN HHQ!VRVVQ '+ \\. 8 &!bff*8C**RVV$C C % & & &r cd}tjt|5t|j dd|j gddddy#1swYyxYw)Nrrrr3rr)rrrr$rdrmrs rxr* z TestXP_Var.test_nan_policy_raise%sM2 ]]:W 5 D BJJ1bff~.7 C D D Ds *AAcLd}tjt|5t|j g}|j |j }t ||dddd}tjt|5t|j ggd}|j |j g}t ||dddt|j ggd}|j g}t ||y#1swYxYw#1swYPxYwr )rr>rr$rdrmr2r s rxrFzTestXP_Var.test_empty%s@ \\,G < &"**R.)C**RVV$C C % & G \\,G < &"**bT*3C**bffX&C C % & bjj"&Q/jjnS! & &  & &r r! cj|j|jj}tj||dz gtj}|j |}t |}tj |}tjj|tjt||j |t ||j}|j tj tj|tj}t||y)Nr3rH) rrr rrdr$rrrrr1rr$ s rxr zTestXP_Var.test_dtype%shhrzz"&&zz3A,bjj9zz$dmffTl RVV,RZZ_-d"**-jj 4rzz BCDS!rzc|jd}|jd}tt|t|y)NrYr)rr2r$rs rxr' zTestXP_Var.test_integer%s. IIbM IIcN GAJ/rzctjjd}|jd\}}t|j ||dzz}tj ||dzz}t ||j |dy)Nr )r3rrF) check_dtype)rrrr$rdrr1rs rxrzTestXP_Var.test_complex_gh22404%smii##N3zz'"1bjjQrT*+ffQ2XRZZ_%@rzN)rrrrrrr1 r rFrr r* rFrr r' rrrzrxr, r, W%s [[V%;< [[Z%7 [[\Aq62 [[\K+@A.B38=.*1 ""& &' &&D'D "" [[> " "0 Arzr, c 2tjjd}|jd}|j|}d}t |||\}}t ||j|t ||d}t |||\}}t ||j|jt |dd}t |d||\}}t ||jtj|dt ||y) Nl~Mr^rrr; rr3rk) rrrrdr r2rr atleast_1d)rorrrqrTx_outaxis_outs rxtest_chk_asarrayr? %s ))   .C  #B 2A D"14B7OE8E2::b>*4 D"14B7OE8E2::bhhj121 D"1W:DR@OE8E2::bmmBwK&@AB4 rznumpyz,These parameters *are* compatible with NumPycJ|jgd}d}tjt|5t j |ddddtjt|5t j |ddddy#1swYExYw#1swYyxYw)NrgrrrrTru)rdrrrrfr)rorqrs rx'test_axis_nan_policy_keepdims_nanpolicyrB %s < A5G *' :) 1() *' :% 1t$%%))%%sB ,B BB"rY)rrT r rx collectionsr itertoolsrhypothesis.extra.numpyextrar@ rrr numpy.testingrrrrr r r r r rrrrrrrr scipy.statsrfscipy.stats.mstatsrscipy.stats._mstats_basic _mstats_basicrscipy.stats._ksstatsrscipy.special._testutilsr scipy.specialrscipyr common_testsrscipy.stats._axis_nan_policyrrrrrrrJrr r!r"r#r$r%scipy._lib._utilr&scipy.conftestr'scipy._lib._array_apir(r)r*r+r,r-r.r/r0scipy._lib._array_api_no_0dr1r2scipy._lib.array_api_extra_libarray_api_extrarjlazy_xp_modulesrrrrerrrrrrrFrrrrr1rr rrrrr rprsrwrrrrr rrrHrrrrrrrrrrArCrrrrrrrr compositerrrr4 PowerDivCaserrrrrsrmrtrrrrrrpruryrrrrrrerrrrrrrrrrr r'r4r6rPrbrdrrlrnrprtrrrrrrrrrrrrrr r r r r r r r" rr_ r ru r r r r r r r r r3 r r r r r r r r' rM rr rrrrqrrwilcoxonrrrrrr r r r r, r? rB rrzrxr] s  "%%... *//##00(--TTUUU&.GGGI((';;// u% %(  !&( 238 : DeL DeL 3U;KTKT\p#p#f5>>"E>E>#E>P DLDLN{6{6|~5~5VV:r 4 <33$TETEnY2x *&R++(n#n#b0D )3"rNrNjf3f3R599+E+E+E\5::j"j"j"Z5<< f&f&!f&R5==!00"0*<@<@~1f@f@R5<< e"e"!e"P11&5::_"_"_"D5>>"`"#`"#`"F   !<5$$%J)J)&J)Zp>p>f& }Dqt&"&&+ 6266$<78AfbffSkMAfbffTlN:;I K }N&"&&+ &"&&,66266#;FGAfbffSkMBvrvvd|O;as mKL1 2}AAD&"&&+ 6266$<78AfbffSkMAfbffTlN:;I K |$>?Q%@Q%h7AtDEM7M7FM7` DEG5G5FG5TQ7Q7h5??#IBIB$IBXL$5??#y*y*$y*x5--.))/),t4HIMNGOJG8%STRS5$$%9>&TU9>x$'QRS,5$$%k2RS%T&%:5>>"N+N+#N+b*-*-Z5>>"->-#-65%%&-~-'-B5##$ -^ -% - AA.5$$%6161&61r{-{-|1"(>V15D + DDD5;;q3q3 q3h5;;n3n3 n3b5;;tWtW tWn5::R3R3R3j(.90[;|T7T7n,/,/^i)i)X\)\)~XXx5(()b4b4*b4J#A#ALl l ^[[|@@FiiXHAHAV{1{1| iiI&JJrNJJrN >>A40 >>Auzz~~+>? >>Aq62 <