
    9i0                        S SK r S SKrS SKJr  S SKJrJr  S SKJ	r	J
r
Jr  S SKJr  \ R                  " S5      r\ R                   R"                  rS\SS	S
.S jr\" SSS9\" SSS9\" SSS9\ R                   R)                  S\R*                  0 4\R,                  0 4\R.                  SS04/5      \ R                   R)                  SS S/5      S 5       5       5       5       5       r\" SSS9\" SSS9\" SSS9\ R                   R)                  S/ SQ5      \ R                   R)                  SSS/5      S 5       5       5       5       5       r\" SSS9\" SSS9\" SSS9\ R                   R)                  S\R4                  SS04\R6                  0 4\R6                  S S04\R8                  0 4\R8                  S S04\R:                  0 4\R<                  S!S04\R<                  S!S04\R<                  S!S"04\R<                  S!S#04\R>                  S!S04\R>                  S!S04\R@                  0 4\RB                  0 4\RD                  0 4\
0 4\RF                  S$S%04\RH                  S$S%04\RJ                  S&S'04\RL                  S(S'04\RN                  S$S%04\RP                  S$S%04/5      \ R                   R)                  S/ SQ5      S) 5       5       5       5       5       r)\" SSS9\" SSS9\" SSS9\ R                   R)                  SS S/5      \ R                   R)                  S*0 /5      S+ 5       5       5       5       5       r*\" SSS9\" SSS9\" SSS9\ R                   R)                  S,\RV                  \RX                  \RZ                  /5      \ R                   R)                  S/ SQ5      S- 5       5       5       5       5       r.\" SSS9\" SSS9\" SSS9\" S.S/S9\ R                   R)                  S0/ S1Q5      \ R                   R)                  S/ SQ5      S2 5       5       5       5       5       5       r/\" SSS9\" SSS9\" SSS9\" S.S/S9\ R                   Ra                  S35      \ R                   R)                  S0/ S4Q5      \ R                   R)                  S/ SQ5      S5 5       5       5       5       5       5       5       r1S6 r2\" S7 S\2S89SDS S9.S: jj5       r3\" SSS9\" SSS9\" SSS9\" S.S/S9\ R                   R)                  S;/ S<Q5      \ R                   R)                  S=S S/5      \ R                   R)                  S/ SQ5      S> 5       5       5       5       5       5       5       r4\" SSS9\" SSS9\" SSS9\" S.S/S9\ R                   R)                  S?/ S@Q5      \ R                   R)                  S/ SQ5      SA 5       5       5       5       5       5       r5\" SSS9\" SSS9\" SSS9\" S.S/S9SB 5       5       5       5       r6SC r7g)E    N)stats)xp_assert_closexp_assert_equal)_xp_mean_xp_var_length_nonmasked)_axis_nan_policy_factorymarrayfloat64)      l   !wN4 )dtypexpshapeseedc                j   [         R                  U5      n[        R                  R	                  U5      n/ / p[        U 5       H  n	UR                  US9n
UR                  S5      (       a  SU
-  S-  UR                  US9-  n
U
R                  U5      n
UR                  U
5        UR                  US9S:  nUR                  U5        M     / n/ n[        Xx5       HR  u  pUR                  UR                  XS95        UR                  5       nUR                  X'   UR                  U5        MT     X\U4$ )N)sizecomplex
   y              $@g      ?mask)r
   _get_namespacenprandomdefault_rngrange
startswithstandard_normalastypeappendzipasarraycopynan)n_arraysr   r   r   r   mxprngdatasmasksidatar   marrays
nan_arraysarray	nan_arrays                   ]/var/www/html/land-doc-ocr/venv/lib/python3.13/site-packages/scipy/stats/tests/test_marray.py
get_arraysr1      s   



#C
))


%Cr58_zzuz%I&&d7S=!4!4%!4!@@D{{5!Tzzuz%,T  GJ5(s{{5{45JJL	&&	)$	 ) ##    z
dask.arrayz/Arrays need `device` attribute: dask/dask#11711)reasonz	jax.numpyz"JAX doesn't allow item assignment.torchz	marray#99zfun, kwargsp   axis   c                     [        SUS9u  pEnU " US   4US   US.UD6nU " US   4US   SUS.UD6n[        UR                  UR                  U5      5        g )Nr6   r   r   r8   )weightsr7   omit)r;   
nan_policyr7   )r1   r   r+   r"   	funkwargsr7   r   r&   r,   narraysresrefs	            r0   
test_xmeanrD   '   si     'qR0C'
gaj
B'!*4
B6
BC
gaj
U'!*d
Uf
UCCHHbjjo.r2   )r   r8   NkeepdimsFTc                     [        SUS9u  p4n[        XS9n[        US   4SUS   0UD6n[        US   4US   SS.UD6n[        UR                  UR                  U5      5        g )	Nr6   r:   r7   rE   r   r;   r8   r<   r;   r=   )r1   dictr   r   r+   r"   )	r7   rE   r   r&   r,   rA   r@   rB   rC   s	            r0   test_xp_meanrJ   5   so     'qR0C't/F
71:
<wqz
<V
<C
71:
Owqzf
O
OCCHHbjjo.r2   zarray-api-compat#242orderbiasn      limits)g?g?
lowerlimitg      ?
upperlimitc                     [        SUS9u  pEn[        US9U-  nU " US   40 UD6nU " US   4SS0UD6n[        UR                  UR	                  U5      5        g Nr8   r:   r7   r   r=   r<   )r1   rI   r   r+   r"   r>   s	            r0   test_severalrV   B   sd    : 'qR0C't_v%F
gaj
#F
#C
gaj
6V
6v
6CCHHbjjo.r2   r@   c                    [        SUS9u  p4n[        U S9U-  n[        R                  " US   40 UD6n[        R                  " US   4SS0UD6n[	        UR
                  R                  UR                  UR
                  5      5        [	        UR                  S   R                  UR                  UR                  S   R                  5      5        [	        UR                  S   R                  UR                  UR                  S   R                  5      5        [	        UR                  R                  UR                  UR                  R                  5      5        [	        UR                  R                  UR                  UR                  R                  5      5        [	        UR                  R                  UR                  UR                  R                  5      5        g rT   )r1   rI   r   describer   nobsr+   r"   minmaxvarianceskewnesskurtosis)r7   r@   r   r&   r,   rA   rB   rC   s           r0   test_describer^   f   sH    'qR0C't_v%F
..
.v
.C
..
A
A&
ACCHHMM2::chh#78CJJqM&&

3::a=3E3E(FGCJJqM&&

3::a=3E3E(FGCLL%%rzz#,,2C2C'DECLL%%rzz#,,2C2C'DECLL%%rzz#,,2C2C'DEr2   r?   c                 L   U [         R                  :X  a
  [        SUS9O	[        SUS9u  p4nU " USU06nUR                  U " USUS.65      n[	        UR
                  UR                  )    XrR                  U5      )    5        [        UR                  US   R                  5        g )Nr6   r:   r8   r7   r<   r=   r7   r   )	r   zmapr1   r"   r   r+   r   isnanr   )r?   r7   r   r&   r,   rA   rB   rC   s           r0   test_zscorerc   x   s     69EJJ5FZb1",Q2"6 C'
w
"T
"C
**S'f4@
ACCHHchhY'hhsm^)<=CHHgajoo.r2   cupyzspecial functions won't workf_name)ttest_1samp	ttest_rel	ttest_indc                    [        [        U 5      n[        SUS9u  pEnU S:X  a5  UR                  US   XS LS9US'   [        R
                  " US   XS LS9US'   U" USU06nU" USUS.6n[        UR                  R                  UR                  UR                  5      5        [        UR                  R                  UR                  UR                  5      5        UR                  5       n	UR                  5       n
[        U	R                  R                  UR                  U
R                  5      5        [        U	R                  R                  UR                  U
R                  5      5        g )	Nr6   r:   rf   r8   rG   r7   r<   r`   )getattrr   r1   meanr   nanmeanr   	statisticr+   r"   pvalueconfidence_intervallowhigh)re   r7   r   fr&   r,   rA   rB   rC   res_ciref_cis              r0   
test_ttestru      s    	vA&qR0C'XXgajt$>NXO
ZZ
D@PQ

W
 4
 C
Wd
3CCMM&&

3==(ABCJJOORZZ

%;<$$&F$$&FFJJOORZZ

%;<FKK$$bjj&=>r2   z7ignore::scipy.stats._axis_nan_policy.SmallSampleWarning)skewtestkurtosistest
normaltestjarque_berac                 F   [        [        U 5      n[        SUSS9u  pEnU" USU06nU" USUS.6n[        UR                  R
                  UR                  UR                  5      5        [        UR                  R
                  UR                  UR                  5      5        g )Nr8   r      r   r   r7   r<   r`   )rj   r   r1   r   rm   r+   r"   rn   )	re   r7   r   rr   r&   r,   rA   rB   rC   s	            r0   test_normality_testsr~      s     	vA&qRx@C'
W
 4
 C
Wd
3CCMM&&

3==(ABCJJOORZZ

%;<r2   c                 0    U R                  SS 5      b  S$ S$ )Nf_expr6   r8   )get)r@   s    r0   pd_nsamplesr      s    

7D)51<1<r2   c                      [        U 5      $ N)tuple)argss    r0   <lambda>r      s    dr2   )paired	n_samplesrU   c                .    [         R                  " XXBUS9$ )N)r7   ddoflambda_r   power_divergence)f_obsr   r   r   r7   s        r0   power_divergence_refr      s    !!%TgVVr2   r   )pearsonzlog-likelihoodzfreeman-tukeyzmod-log-likelihoodneymanzcressie-read	chisquarer   c                 ~  ^  [        SUSS9u  pEn[        X!S9nT S:X  a  Sm S nOU 4S jnU" US	   40 UD6n	[        US	   4S
T S.UD6n
[        U	R                  R
                  UR                  U
S	   5      5        [        U	R                  R
                  UR                  U
S   5      5        [        R                  " US	   5      [        R                  " US   5      -  n[        R                  " US   U) -  USS9[        R                  " US	   U) -  USS9-  nUS	==   UR                  U5      -  ss'   US	==   U-  ss'   U" U0 UD6n	[        US
T S.UD6n
[        U	R                  R
                  UR                  U
S	   5      5        [        U	R                  R
                  UR                  U
S   5      5        g )Nr6   )      r}   )r7   r   r   r   c                  .    [         R                  " U 0 UD6$ r   )r   r   )r   r@   s     r0   rr   *test_power_divergence_chisquare.<locals>.f   s    ??D3F33r2   c                  4   > [         R                  " U ST0UD6$ )Nr   r   )r   r@   r   s     r0   rr   r      s    ))4KKFKKr2   r   r<   )r=   r   r8   TrG   )r1   rI   r   r   rm   r+   r"   rn   r   rb   nansum)r   r   r7   r   r&   r,   rA   r@   rr   rB   rC   common_mask	normalizes   `            r0   test_power_divergence_chisquarer      s    'qRv>C't'F+	4	L GAJ
!&
!C
wqz
Xfg
XQW
XCCMM&&

3q6(:;CJJOORZZA%78 ((71:&'!*)==K71:44$O99WQZ;,6TDQRIAJ"**Y''JAJ)J
W

C
FG
Vv
VCCMM&&

3q6(:;CJJOORZZA%78r2   method)fisherr   mudholkar_georgetippettstoufferc                    [        SUSS9u  p4n[        XS9n[        R                  " US   40 UD6n[        R                  " US   4SS0UD6n[	        UR
                  R                  UR                  UR
                  5      5        [	        UR                  R                  UR                  UR                  5      5        U S:w  a  g [        R                  " US   4S	US
   0UD6n[        R                  " US   4US
   SS.UD6n[	        UR
                  R                  UR                  UR
                  5      5        [	        UR                  R                  UR                  UR                  5      5        g )Nr6   r{   r}   )r   r7   r   r=   r<   r   r;   r8   rH   )	r1   rI   r   combine_pvaluesr   rm   r+   r"   rn   )	r   r7   r   r&   r,   rA   r@   rB   rC   s	            r0   test_combine_pvaluesr      s3    'qRx@C'+F




5f
5C




Hv
H
HCCMM&&

3==(ABCJJOORZZ

%;<




IGAJ
I&
IC



 =GAJ+1=5;=C CMM&&

3==(ABCJJOORZZ

%;<r2   c                 :   Sn[        SXS9u  p#n[        R                  " [        R                  " [        R                  " U Vs/ s H  n[        R
                  " U5      PM     sn5      SS9[        5      nU Vs/ s H  oUU)    PM
     nnUS   S-  US   S-  sUS'   US'   US   S-  US   S-  sUS'   US'   [        R                  " U6 n[        R                  " U6 nU R                  U5      nU R                  U5      (       a  U R                  U) 5      (       d   e[        UR                  R                  U)    U R                  UR                  5      5        [        UR                  R                  U)    U R                  UR                  5      5        [        UR                  R                  U5        [        UR                  R                  U5        UR                  R                   U:X  d   eUR                  R                   U:X  d   eg s  snf s  snf )	Nr{   r   r}   r   rU   r6   d   r   )r1   r   r   sumstackrb   boolr   ttest_ind_from_statsr"   anyr   rm   r+   rn   r   r   )	r   r   r&   r,   rA   argr   rB   rC   s	            r0   test_ttest_ind_from_statsr      s   
 E&qR=C'99RVVBHHw%Gwbhhsmw%GHqQSWXD%,-WcD5zWG-$QZ#-wqzC/?GAJ
$QZ#-wqzC/?GAJ


$
$g
.C

$
$g
.C::dD66$<<BFFD5MM))CMM&&u-rzz#--/HICJJOOTE*BJJszz,BCCMM&&-CJJOOT*==%'''::u$$$ &H-s    H;Hc                      [         R                  [        5      n SS/SS//nSS/SS//nU R                  XS9n[        R
                  " [        SS	9   [        US
U S9  S S S 5        g ! , (       d  f       g = f)Ng      ?g       @g      @g      @FTr   z7`axis` must be an integer or None for use with `MArray`)match)r   r8   )r7   r   )r
   r   r   r"   pytestraisesNotImplementedErrorr   )r   r+   r   marrs       r0   1test_length_nonmasked_marray_iterable_axis_raisesr     sz    			r	"B#Jc
#DENT5M*D::d:&D 
*G
I$V3
I 
I 
Is   A''
A5r   )8r   numpyr   scipyr   scipy._lib._array_apir   r   scipy.stats._stats_pyr   r   r   scipy.stats._axis_nan_policyr	   importorskipr
   markskip_xp_backendsskip_backendr1   parametrizegmeanhmeanpmeanrD   rJ   momentskewr]   semkstatkstatvarcircmeancircvarcircstdtmeantvartmintmaxtstdtsemrV   r^   zscoregzscorera   rc   ru   filterwarningsr~   r   r   r   r   r   r    r2   r0   <module>r      s      B F F A 
		X	&{{++ #,&~ $2 l#TUk"FGgk*%++r):*/++r):*/++Qx)@)B C !Q(/ )C + H V/ l#TUk"FGgk*.eT]3/ 4 / + H V
/ l#TUk"FGg45llWaL!jj"jj65/"nnbnnvuo&ii_kkC8kkC8kkC8kkC8nnsAhnnsAhnnbmmRmmRr]kkHj)*jj8Z()jj<%&jj<%&jj8Z()jj8Z()+0 ./ /1 6 H V8/ l#TUk"FGg45!Q(B4(
F ) ) 6 H V

F l#TUk"FGg45u}}ejj IJ./ / K 6 H V
/ l#TUk"FGg45f;<#LM.? / N = 6 H V?  l#TUk"FGg45f;<UV $A B.= /B W = 6 H V== 3DKXWA W YW l#TUk"FGg45f;< %2 3 !Q(.9 / )3 = 6 H V9B l#TUk"FGg45f;< $; <.= /< = 6 H V=* l#TUk"FGg45f;<% = 6 H V%(4r2   