L i0 ddlZddlZddlmZddlmZmZddlm Z m Z m Z ddl m Z ejdZej j"Zdedd d d Zed d eddeddej j)dej*ifej,ifej.ddifgej j)dddgdZed d eddeddej j)dgdej j)dddgdZed d eddeddej j)dej4ddifej6ifej6d difej8ifej8d difej:ifej<d!difej<d!difej<d!d"ifej<d!d#ifej>d!difej>d!difej@ifejBifejDife ifejFd$d%ifejHd$d%ifejJd&d'ifejLd(d'ifejNd$d%ifejPd$d%ifgej j)dgdd)Z)ed d eddeddej j)dddgej j)d*igd+Z*ed d eddeddej j)d,ejVejXejZgej j)dgdd-Z.ed d eddedded.d/ej j)d0gd1ej j)dgdd2Z/ed d eddedded.d/ej jad3ej j)d0gd4ej j)dgdd5Z1d6Z2e d7de28dDdd9d:Z3ed d eddedded.d/ej j)d;gd<ej j)d=ddgej j)dgdd>Z4ed d eddedded.d/ej j)d?gd@ej j)dgddAZ5ed d eddedded.d/dBZ6dCZ7y)EN)stats)xp_assert_closexp_assert_equal)_xp_mean_xp_var_length_nonmasked)_axis_nan_policy_factorymarrayfloat64)l!wN4)dtypexpshapeseedc|tj|}tjj |}gg}}t |D]} |j|} |j drd| zdz|j|z} | j|} |j| |j|dkD} |j| g} g} t||D]W\}} | j|j|| |j}|j|| <| j|Y|| | fS)N)sizecomplex y$@g?mask)r _get_namespacenprandom default_rngrange startswithstandard_normalastypeappendzipasarraycopynan)n_arraysrrrrmxprngdatasmasksidatarmarrays nan_arraysarray nan_arrays c/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/scipy/stats/tests/test_marray.py get_arraysr1s/    #C ))   %Cr5E 8_zzuz%   I &d7S=!4!4%!4!@@D{{5! Tzzuz%, TGJ5%(% ts{{5t{45JJL && $)$ %  ##z dask.arrayz/Arrays need `device` attribute: dask/dask#11711)reasonz jax.numpyz"JAX doesn't allow item assignment.torchz marray#99z fun, kwargspaxisctd|\}}}||df|d|d|}||df|dd|d|}t|j|j|y)Nr6rrr8)weightsr7omit)r; nan_policyr7)r1rr+r" funkwargsr7rr&r,narraysresrefs r0 test_xmeanrD'sk'qR0C' gaj B'!*4 B6 BC gaj U'!*d Uf UCCHHbjjo.r2)rr8NkeepdimsFTctd|\}}}t||}t|dfd|di|}t|df|ddd|}t|j|j |y) Nr6r:r7rErr;r8r<r;r=)r1dictrrr+r") r7rErr&r,rAr@rBrCs r0 test_xp_meanrJ5ss 'qR0C' th /F 71: s r0 test_severalrVBsf:'qR0C' t_v %F gaj #F #C gaj 6V 6v 6CCHHbjjo.r2r@ctd|\}}}t||z}tj|dfi|}tj|dfddi|}t |j j |j|j t |jdj |j|jdj t |jdj |j|jdj t |jj |j|jj t |jj |j|jj t |jj |j|jj yrT) r1rIrdescribernobsr+r"minmaxvarianceskewnesskurtosis)r7r@rr&r,rArBrCs r0 test_describer^fsJ 'qR0C' t_v %F .. .v .C .. A A& ACCHHMM2::chh#78CJJqM&& 3::a=3E3E(FGCJJqM&& 3::a=3E3E(FGCLL%%rzz#,,2C2C'DECLL%%rzz#,,2C2C'DECLL%%rzz#,,2C2C'DEr2r?cZ|tjk(r td|n td|\}}}||d|i}|j||d|d}t |j |j ||j|t|j |dj y)Nr6r:r8r7r<r=r7r) rzmapr1r"rr+risnanr)r?r7rr&r,rArBrCs r0 test_zscorercxs 69EJJ5FZb1",Q2"6C' w "T "C **S'f4@ ACCHHchhY'bhhsm^)<=CHHgajoo.r2cupyzspecial functions won't workf_name) ttest_1samp ttest_rel ttest_indctt|}td|\}}}|dk(r<|j|d||du|d<t j |d||du|d<||d|i}||d|d}t |jj|j|jt |jj|j|j|j} |j} t | jj|j| jt | jj|j| jy) Nr6r:rfr8rGr7r<r`)getattrrr1meanrnanmeanr statisticr+r"pvalueconfidence_intervallowhigh) rer7rfr&r,rArBrCres_ciref_cis r0 test_ttestrus$ vA&qR0C' XXgajtd$>NXO ZZ D@PQ W 4 C Wd 3CCMM&& 3==(ABCJJOORZZ %;<  $ $ &F  $ $ &FFJJOORZZ %;<FKK$$bjj&=>r2z7ignore::scipy.stats._axis_nan_policy.SmallSampleWarning)skewtest kurtosistest normaltest jarque_beracNtt|}td|d\}}}||d|i}||d|d}t|jj |j |jt|jj |j |jy)Nr8r rrr7r<r`)rjrr1rrmr+r"rn) rer7rrrr&r,rArBrCs r0test_normality_testsr~s vA&qRx@C' W 4 C Wd 3CCMM&& 3==(ABCJJOORZZ %;rs d r2)paired n_samplesrUc6tj|||||S)N)r7ddoflambda_rpower_divergence)f_obsrrrr7s r0power_divergence_refrs  ! !%Tg VVr2r)pearsonzlog-likelihoodz freeman-tukeyzmod-log-likelihoodneymanz cressie-read chisquarerctd|d\}}}t||}dk(rdd}nfd}||d fi|} t|d fd d |} t| jj |j | d t| jj |j | d tj|d tj|d z} tj|d | z|d tj|d | z|d z } |d xx|j | zcc<|d xx| zcc<||i|} t|d d |} t| jj |j | d t| jj |j | d y)Nr6)r})r7rrrc,tj|i|Sr)rr)rr@s r0rrz*test_power_divergence_chisquare..fs??D3F3 3r2c2tj|di|S)Nrr)rr@rs r0rrz*test_power_divergence_chisquare..fs))4KKFK Kr2rr<)r=rr8TrG) r1rIrrrmr+r"rnrrbnansum) rrr7rr&r,rAr@rrrBrC common_mask normalizes ` r0test_power_divergence_chisquarers'qRv>C' t$ 'F+ 4 L GAJ !& !C wqz Xfg XQW XCCMM&& 3q6(:;CJJOORZZA%78((71:&'!*)==K71: 44$O99WQZ;,6TDQRI AJ"**Y''J AJ)J W  C FG Vv VCCMM&& 3q6(:;CJJOORZZA%78r2method)fisherrmudholkar_georgetippettstoufferctd|d\}}}t||}tj|dfi|}tj|dfddi|}t |j j |j|j t |jj |j|j|dk7rytj|dfd |d i|}tj|df|d dd |}t |j j |j|j t |jj |j|jy) Nr6r{r})rr7rr=r<rr;r8rH) r1rIrcombine_pvaluesrrmr+r"rn) rr7rr&r,rAr@rBrCs r0test_combine_pvaluesrs7'qRx@C' d +F    5f 5C    Hv H HCCMM&& 3==(ABCJJOORZZ %;<     IGAJ I& IC     =GAJ+1 =5; =CCMM&& 3==(ABCJJOORZZ %;rs BFFA   X &{{++ #,&~$2l#TU k"FG gk*%++r):*/++r):*/++Qx)@)BC!Q(/)C+HV/l#TU k"FG gk*.eT]3/4/+HV /l#TU k"FG g45 llWaL! jj" jj65/" nnb nnvuo& ii_ kkC8 kkC8 kkC8 kkC8 nnsAh nnsAh nnb mmR mmR r] kkHj)* jj8Z() jj<%& jj<%& jj8Z() jj8Z()+0.//16HV8/l#TU k"FG g45!Q(B4( F))6HV Fl#TU k"FG g45u}}ejj IJ.//K6HV /l#TU k"FG g45 f;<#LM. ?/N=6HV ? l#TU k"FG g45 f;<UV$AB.=/BW=6HV==3DKXWAWYWl#TU k"FG g45 f;<%23!Q(.9/)3=6HV9Bl#TU k"FG g45 f;<$;<.=/<=6HV=*l#TU k"FG g45 f;<%=6HV%( 4r2