
    Cpjs                     $   S SK rS SKrS SKJr  S SKJrJrJr  S SKJ	r	J
r
  S SKJrJr  S SKJrJr  \R                   " S5      r\R$                  R'                  \(       + SS	9/r\R$                  R*                  rS
\SSSS.S jr\" SSS	9\R$                  R1                  S\	" \R2                  0 5      \	" \R4                  0 5      \	" \R6                  SS05      \	" \R8                  SS05      \	" \0 5      /5      \R$                  R1                  SSS/5      \R$                  R1                  S/ SQ5      S 5       5       5       5       r\" SSS	9\" SSS	9\R$                  R1                  S/ \	" \R<                  SS05      P\	" \R<                  SS 05      P\	" \R>                  0 5      P\	" \R>                  S!S05      P\	" \R@                  0 5      P\	" \R@                  SSS".5      P\	" \RB                  0 5      P\	" \RB                  S#S05      P\	" \RD                  S$S%05      P\	" \RD                  S$S05      P\	" \RD                  S$S&05      P\	" \RD                  S$S'05      P\	" \RF                  S$S%05      P\	" \RF                  S$S05      P\	" \RH                  0 5      P\	" \RH                  S S%S(.5      P\	" \RJ                  0 5      P\	" \RJ                  S S%S(.5      P\	" \RL                  0 5      P\	" \RL                  S S%SS).5      P\	" \RN                  0 5      P\	" \RN                  S#S05      P\	" \0 5      P\	" \S%SS*.5      P\	" \RP                  0 5      P\	" \RP                  S#S%05      P\	" \RR                  \*" S+S,95      P\	" \RV                  \*" S+S-S.95      P\	" \RX                  \*" S+S/S.95      P\	" \RZ                  \*" S+S0S.95      P\	" \R\                  \*" S SS195      P\	" \R^                  \*" S SS295      P\	" \R`                  S3S405      P\	" \R`                  S5S6S7.5      P\	" \Rb                  0 5      P\	" \Rb                  S8S605      P\	" \Rd                  S9S:05      P5      \R$                  R1                  S/ SQ5      S; 5       5       5       5       r3\
" \Rh                  5      \R$                  R1                  S<S=S
/5      \R$                  R1                  S>S?S/5      \R$                  R1                  S/ SQ5      S@ 5       5       5       5       r5\
" \Rl                  5      \" SSS	9\" SSS	9\R$                  R1                  S/ SQ5      \R$                  R1                  SA0 S!S0/5      SB 5       5       5       5       5       r7\" SSCS	9\" SSS	9\R$                  R1                  SD\	" \Rp                  5      \	" \Rr                  5      \	" \Rt                  5      /5      \R$                  R1                  S#S S%/5      \R$                  R1                  S/ SQ5      SE 5       5       5       5       5       r;\" SSS	9\" SSS	9\R$                  R1                  SF\	" \Rx                  5      \	" \Rz                  5      \	" \R|                  5      /5      \R$                  R1                  SG/ SHQ5      \R$                  R1                  S/ SQ5      SI 5       5       5       5       5       r?\" SSS	9\" SSS	9\R$                  R                  SJ5      \R$                  R1                  SK\	" \R                  \B" 5       5      \	" \R                  \B" 5       5      \	" \R                  \B" 5       5      \	" \R                  \B" 5       5      \	" \R                  \R                  R                  45      /5      \R$                  R1                  SG/ SHQ5      \R$                  R1                  S/ SQ5      SL 5       5       5       5       5       5       rI\" SSS	9\" SSS	9\R$                  R1                  SM\	" \R                  0 5      \	" \R                  SNSO05      \	" \R                  SNSP05      \	" \R                  SNSQ05      \	" \R                  SNSR05      \	" \R                  SNSS05      \	" \R                  SNST05      /5      \R$                  R1                  S#S S%/5      \R$                  R1                  S/ SQ5      SU 5       5       5       5       5       rL\
" \R                  5      \" SSS	9\R$                  R1                  SV/ SWQ5      \R$                  R1                  S/ SQ5      SX 5       5       5       5       rN\
" \R                  5      \" SSS	9\R$                  R1                  SG/ SHQ5      \R$                  R1                  SYSS/5      SZ 5       5       5       5       rPS[ rQ\
" \R                  5      \" SSS	9\R$                  R                  S\5      \R$                  R1                  S]SS/5      \R$                  R1                  SS S%/5      S^ 5       5       5       5       5       rS\
" \R                  5      \" SSS	9\R$                  R1                  S_S%S/5      \R$                  R1                  S`Sa/5      \R$                  R1                  SbSS/5      \R$                  R1                  SG/ SHQ5      \R$                  R1                  SVSc/5      \R$                  R1                  S/ SQ5      Sd 5       5       5       5       5       5       5       5       rU\
" \R                  5      \" SSS	9\R$                  R1                  SeSS/5      \R$                  R1                  SG/ SHQ5      \R$                  R1                  SVSc/5      \R$                  R1                  S/ SQ5      Sf 5       5       5       5       5       5       rW\" SSS	9\R$                  R1                  SD\	" \R                  5      \	" \R                  5      /5      \R$                  R1                  SV/ SgQ5      \R$                  R1                  SG/ SHQ5      \R$                  R1                  S/ SQ5      Sh 5       5       5       5       5       rZ\" SSS	9\R$                  R1                  S\	" \R                  0 5      \	" \R                  SiS605      \	" \R                  SVSj05      /5      \R$                  R1                  S/ SQ5      Sk 5       5       5       r]\
" \R                  5      \" SSS	9\R$                  R1                  SD\	" \R                  5      \	" \R                  5      /5      \R$                  R1                  SV/ SlQ5      \R$                  R1                  SG/ SHQ5      \R$                  R1                  S/ SQ5      Sm 5       5       5       5       5       5       r_\
" \R                  5       " Sn So5      5       ra\" SSS	9\" SSS	9\R$                  R1                  S\	" \R                  0 5      \	" \R                  0 5      \	" \R                  SSp05      \	" \R                  SSq05      \	" \R                  SSr05      \	" \R                  SYS05      \	" \R                  SYS05      \	" \R                  0 5      \	" \R                  SSp05      \	" \R                  SSq05      \	" \R                  SSr05      /5      \R$                  R1                  S/ SQ5      Ss 5       5       5       5       rh\" SStS	9\R$                  R1                  S\	" \R                  0 5      /5      \R$                  R1                  S/ SQ5      Su 5       5       5       rj\" SSS	9\" SSS	9\R$                  R1                  SF\	" \R                  5      \	" \R                  5      \	" \R                  5      /5      \R$                  R1                  SG/ SHQ5      \R$                  R1                  S/ SQ5      Sv 5       5       5       5       5       rn\" SSS	9\" SwSxS	9\R$                  R1                  Sy\	" \R                  SVSz05      \	" \R                  SVS{05      \	" \R                  SVS|05      \	" \R                  SVS{05      \	" \R                  SGS}05      \	" \R                  SGS~05      \	" \R                  SGS05      /5      \R$                  R1                  S/ SQ5      S 5       5       5       5       rr\
" \R                  5      \" SSS	9\" SSS	9\R$                  R1                  SSS/5      \R$                  R1                  SSS/5      \R$                  R1                  S/ SQ5      S 5       5       5       5       5       5       rt\
" \R                  5      \" SStS	9\R$                  R1                  SV/ SQ5      \R$                  R1                  S/ SQ5      S 5       5       5       5       rv\
" \R                  5      \" SStS	9\R$                  R1                  S<SS
/5      \R$                  R1                  SS%S&/5      S 5       5       5       5       rxg)    N)stats)xp_assert_closexp_assert_equal_count_nonmasked)make_xp_pytest_parammake_xp_test_case)SCIPY_ARRAY_APIis_torch)_xp_mean_xp_varmarrayzcspecial function dispatch to marray required for these tests is hidden behind SCIPY_ARRAY_API flag.)reasonfloat64)      Tl   !wN4 )dtypexpshape
all_uniqueseedc                   [         R                  U5      n[        R                  R	                  U5      n/ / p[        U 5       H  n
U(       a  UR                  US9O&UR                  [        R                  " U5      S-  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[        X5       HS  u  pUR                  UR                  XS95        UR                  5       nUR                  UU'   UR                  U5        MU     XmU4$ )N)size   complex
   y              $@g      ?mask)r   _get_namespacenprandomdefault_rngrangeintegersmin
startswithstandard_normalastypeappendzipasarraycopynan)n_arraysr   r   r   r   r   mxprngdatasmasksidatar   marrays
nan_arraysarray	nan_arrays                    Y/var/www/html/pdf-tiff/venv/lib/python3.13/site-packages/scipy/stats/tests/test_marray.py
get_arraysr9      s@   



#C
))


%Cr58_*4


&\\"&&-1"45\A 	I&&d7S=!4!4%!4!@@D{{5!Tzzuz%,T  GJ5(s{{5{45JJL	&&	$)$	 ) ##    z	jax.numpyz"JAX doesn't allow item assignment.zfun, kwargspr   alphag?weightedFaxisr      Nc                     [        SUS9u  pVnU(       a  US   OS nU(       a  US   OS n	U " US   4XS.UD6n
U " US   4U	SUS.UD6n[        U
R                  UR                  U5      5        g )Nr   r   r@   r   )weightsr>   omit)rC   
nan_policyr>   )r9   r   r3   r*   )funkwargsr=   r>   r   r.   r4   narraysmweightsnweightsresrefs               r8   !test_one_sample_weighted_reducingrM   3   sy     'qR0C'%wqz4H%wqz4H
gaj
@(
@
@C
gaj
S(vD
SF
SCCHHbjjo.r:   z
dask.arrayz/Arrays need `device` attribute: dask/dask#11711ordercenterg      ?bias)fisherrP   ddofnr@         )lowhigh)rV   rW   	normalize)
correctionkeepdims)皙?g?)limits)FT)r\   	inclusive)TF)FF)
lowerlimitr]   )
upperlimitr]   interpolationnearest)r   Z   normal)r/   scalerd   proportiontocutr[   c                     [        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 )Nr@   rB   r>   r   rE   rD   )r9   dictr   r3   r*   	rF   rG   r>   r   r.   r4   rH   rK   rL   s	            r8   test_one_sample_reducingrj   F   se    V 'qR0C't_v%F
gaj
#F
#C
gaj
6V
6v
6CCHHbjjo.r:   r   int32r   r   c           	         [        SUSUS9u  pEn[        R                  " UR                  US   [	        X@5      5      5      n[        R                  " USS06n[        UR                  R                  UR                  UR                  R                  U 5      5      5        [        UR                  R                  UR                  UR                  5      5        g )Nr@   F)r   r   r   r   rE   rD   )	r9   r   moder'   getattrr   r3   r*   count)	r   r   r>   r   r.   r4   rH   rK   rL   s	            r8   	test_moderp   x   s    
 'q%BOC'
**SZZ
GC,?@
AC
**g
1&
1CCHHMM2::chhooe.D#EFCIINNBJJsyy$9:r:   rG   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                  [        R                  " UR                  R                  SS95      5        [	        UR                  R                  UR                  [        R                  " UR                  R                  SS95      5        [	        UR                  R                  UR                  [        R                  " UR                  R                  SS95      5        g )	Nr@   rB   rg   r   rE   rD   T)r+   )r9   rh   r   describer   nobsr3   r*   minmaxvariancer   skewnesskurtosis)r>   rG   r   r.   r4   rH   rK   rL   s           r8   test_describerx      su    '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%%JJrzz#,,*;*;$GHJCLL%%JJrzz#,,*;*;$GHJCLL%%JJrzz#,,*;*;$GHJr:   z#shape (nan,) in _xp_var when ddof=1rF   c           	      L   U [         R                  :X  a
  [        SUS9O	[        SUS9u  pEnU " XQUS.6nUR                  U " XaSUS.65      n[	        UR
                  UR                  )    XR                  U5      )    5        [        UR                  US   R                  5        g )Nr   rB   r@   )rR   r>   rD   )rR   rE   r>   r   )	r   zmapr9   r*   r   r3   r   isnanr   )	rF   rR   r>   r   r.   r4   rH   rK   rL   s	            r8   test_one_sample_non_reducingr|      s     69EJJ5FZb1",Q2"6 C'
w
-C
**S'dK
LCCHHchhY'hhsm^)<=CHHgajoo.r:   falternativelessgreater	two-sidedc                    U R                   n[        SUS9u  pVn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 " XaUS.6nU " XqS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 )	Nr   rB   ttest_1sampr@   r>   rZ   )r~   r>   rD   )r~   rE   r>   )__name__r9   meanr   nanmeanr   	statisticr3   r*   pvalueconfidence_intervalrV   rW   )r}   r~   r>   r   f_namer.   r4   rH   rK   rL   res_ciref_cis               r8   test_ttestsr      s    ZZF&qR0C'XXgajt$>NXO
ZZ
D@PQ

WD
9C
W&t
LCCMM&&

3==(ABCJJOORZZ

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

%;<FKK$$bjj&=>r:   z7ignore::scipy.stats._axis_nan_policy.SmallSampleWarningzf, argsc                    [        SUSS9u  pVnU [        R                  [        R                  1;   a  SU0nO,US:w  a$  [        R
                  " SU R                   S35        0 nU " / UQUQ70 UDSU0D6n	U " / UQUQ70 UDS	US
.D6n
[        U	R                  R                  UR                  U
R                  5      5        [        U	R                  R                  UR                  U
R                  5      SS9  g )Nr@   r      r   r   r~   r   zstr(z( does not support multiple alternatives.r>   rD   rE   r>   gV瞯<atol)r9   r   skewtestkurtosistestpytestskipr   r   r   r3   r*   r   )r}   argsr~   r>   r   r.   r4   rH   kwdsrK   rL   s              r8   test_goodness_of_fitr      s     'qRx@C'U^^U//00{+)#KK$qzzl*RST

/W
/t
/t
/$
/C

BW
Bt
Bt
BT
BCCMM&&

3==(ABCJJOORZZ

%;%Hr:   z
f, lambda_lambda_pearsonzlog-likelihoodzfreeman-tukeyzmod-log-likelihoodneymanzcressie-readc                    [        SUSS9u  pVn[        X2S9nU " US   40 UDUD6n	[        R                  " US   4SS0UD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UD6n	[        R                  " USS0UDUD6n
[	        U	R
                  R                  UR                  U
S   5      5        [	        U	R                  R                  UR                  U
S   5      5        g )Nr   )      r   )r>   rR   r   rE   rD   r@   Tr   )r9   rh   r   power_divergencer   r   r3   r*   r   r   r{   nansum)r}   r   rR   r>   r   r.   r4   rH   rG   rK   rL   common_maskrX   s                r8   test_power_divergence_chisquarer      s    'qRv>C't'F GAJ
,'
,V
,C

 
 
T
T'
TV
TCCMM&&

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

3q6(:;CJJOORZZA%78r:   method)rQ   r   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 )Nr   r   r   r   r>   r   rE   rD   r   rC   r@   )rC   rE   )	r9   rh   r   combine_pvaluesr   r   r3   r*   r   )	r   r>   r   r.   r4   rH   rG   rK   rL   s	            r8   test_combine_pvaluesr     s5    '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

%;<r:   	equal_varc           
      L   Sn[        SX#S9u  pEn[        R                  " [        R                  " U Vs/ s H  n[        R                  " U5      PM     sn5      SS9R                  [        5      nU Vs/ s H  owU)    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'   [        XS	9n	[        R                  " U0 U	D6n
[        R                  " U0 U	D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   rg   r   d   r   )r~   r   )r9   r   sumstackr{   r'   boolrh   r   ttest_ind_from_statsr*   anyr   r   r3   r   r   r   )r~   r   r   r   r.   r4   rH   argr   rG   rK   rL   s               r8   test_ttest_ind_from_statsr   !  s   
 E&qR=C'66"((W=WcBHHSMW=>QGNNtTD%,-WcD5zWG-$QZ#-wqzC/?GAJ
$QZ#-wqzC/?GAJ
k?F

$
$g
8
8C

$
$g
8
8C::dD66$<<BFFD5MM))CMM&&u-rzz#--/HICJJOOTE*BJJszz,BCCMM&&-CJJOOT*==%'''::u$$$! >-s    H5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   r@   )r>   r   )r   r   r   r*   r   raisesNotImplementedErrorr   )r   r3   r   marrs       r8   1test_length_nonmasked_marray_iterable_axis_raisesr   ;  sz    			r	"B#Jc
#DENT5M*D::d:&D 
*G
IFr2
I 
I 
Is   A''
A5zignore::RuntimeWarningrX   c                 
   [        SSUS9u  p4n[        R                  " X@US.6nUu  nUS:X  a  [        R                  " USS5      n[        UR                  S   5       H  nXx   n	U	[        R                  " [        R                  " U	5      SS9) S S 24   n	[        R                  " XS9n
[        UR                  R                  US	4   UR                  U
R                  5      5        [        UR                  R                  U   UR                  U
R                  5      S
   5        M     UR                  UR                  R                  5      (       a   eUR                  UR                  R                  5      (       a   eg )Nr@   )      rT   r   r   )rX   r>   r   rg   )rX   . )r9   r   directional_statsr   swapaxesr"   r   r   r{   r   mean_directionr3   r*   mean_resultant_lengthr   )rX   r>   r   r.   r4   rH   rK   xr2   xirL   s              r8   test_directional_statsr   I  sG    'qCC'

!
!7d
KC	BAqyKK1a 1771:T2..12%%b>**//37

3#5#56	81166q9

3#<#<=bA	C  vvc((--....vvc//4455555r:   	n_sampleszero_methodzsplitrY   
asymptoticc                 x   [        XSS9u  pxn	[        XX4S9n
[        R                  " USU0U
D6n[        R                  " U	SUS.U
D6n[	        UR
                  R                  UR                  UR
                  5      5        [	        UR                  R                  UR                  UR                  5      5        g )N   "wN4 r   r   )r   rY   r~   r   r>   rD   r   )	r9   rh   r   wilcoxonr   r   r3   r*   r   )r   r   rY   r~   r   r>   r   r.   r4   rH   rG   rK   rL   s                r8   test_wilcoxonr   b  s     'ynMC'k):F
..'
7
7
7C
..'f4
J6
JCCMM&&

3==(ABCJJOORZZ

%;<r:   use_continuityc                 z   [        SUSS9u  pVn[        XUS9n[        R                  " USU0UD6n	[        R                  " USUS.UD6n
[	        U	R
                  R                  UR                  U
R
                  5      5        [	        U	R                  R                  UR                  U
R                  5      5        g )Nr   r   r   )r   r~   r   r>   rD   r   )	r9   rh   r   mannwhitneyur   r   r3   r*   r   )r   r~   r   r>   r   r.   r4   rH   rG   rK   rL   s              r8   test_mannwhitneyur   t  s     'qRnEC'QWXF


g
;D
;F
;C


g&t
Nv
NCCMM&&

3==(ABCJJOORZZ

%;<r:   )exactapproxasympc                    [        SUSS9u  pVn[        XUS9nU " / UQ[        R                  R                  P70 UD6n	[        R
                  " / UQ[        R                  R                  P7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	R                  R                  UR                  U
R                  5      5        [        U	R                  R                  UR                  U
R                  UR                  S95        g )Nr@   r   r   r   r~   r>   rE   rD   r   )r9   rh   r   normcdfks_1sampr   r   r3   r*   r   r   statistic_locationstatistic_signint8rF   r   r~   r>   r   r.   r4   rH   rG   rK   rL   s              r8   test_ks_1sampr     s     'qRnEC'tDF

1w
1


1&
1C
..
O'
O5::>>
Of
O
OCCMM&&

3==(ABCJJOORZZ

%;<C**//C<R<R1STC&&++JJs11JACr:   distributionr   c                    U [         R                  :X  a  Uc  [        R                  " S5        [	        SUSS9u  pEnU " USU0UD6nU " USUS.UD6n[        UR                  R                  UR                  UR                  5      5        [        UR                  R                  UR                  UR                  5      5        g )Nz"Sample too large for exact method.r   r   r   r>   rD   r   )
r   cramervonmises_2sampr   r   r9   r   r   r3   r*   r   ri   s	            r8   test_two_sample_testsr     s     e(((T\89&qRnEC'
w
,T
,V
,C
w6
?
?CCMM&&

3==(ABCJJOORZZ

%;<r:   )r   r   autoc                    [        SUSS9u  pVn[        XUS9nU " U0 UD6n	[        R                  " U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	R                  R                  UR                  U
R                  UR                  S95        g )Nr   r   r   r   rE   rD   r   )r9   rh   r   ks_2sampr   r   r3   r*   r   r   r   r   r   s              r8   test_ks_2sampr     s     'qRnEC'tDF
w
!&
!C
..'
?f
?
?CCMM&&

3==(ABCJJOORZZ

%;< C&&++JJs11JACr:   c                   j   \ rS rSr\R
                  R                  SSS/5      \R
                  R                  SSS/5      \R
                  R                  S/ S	Q5      \R
                  R                  S
/ SQ5      S 5       5       5       5       r\R
                  R                  SSS/5      S 5       rSr	g)TestKendallTaui  r   r   r   variantbcr~   r   r>   r?   c                    [        SSUSS9u  pgn[        XUS9n	[        R                  " U0 U	DSU0D6n
[        R                  " U0 U	DSUS.D6n[	        U
R
                  R                  UR                  UR
                  5      5        [	        U
R                  R                  UR                  UR                  5      5        g )	Nr   )      r   )r   r   r   )r   r   r~   r>   rD   r   )	r9   rh   r   
kendalltaur   r   r3   r*   r   )selfr   r   r~   r>   r   r.   r4   rH   rG   rK   rL   s               r8   test_omit_masked_elements(TestKendallTau.test_omit_masked_elements  s    
 !+1H.0~!GgV+N=6==P6Pf4P**BJJs}},EF

CJJ)?@r:   znan_policy, message)	propagatez;`nan_policy='propagate'` is incompatible with MArray input.)raisezThe input contains nan values.c                     [         R                  U5      nUR                  SSSUR                  /5      n[        R
                  " [        US9   [        R                  " XUUS9  S S S 5        g ! , (       d  f       g = f)Nr@   r   rT   r   )rE   )	r   r   r*   r,   r   r   
ValueErrorr   r   )r   rE   messager   r.   r   s         r8   test_input_validation$TestKendallTau.test_input_validation  sY    
 ##B'KKAq"&&)*]]:W5Qj9 655s   A--
A;r   N)
r   
__module____qualname____firstlineno__r   markparametrizer   r  __static_attributes__r   r:   r8   r   r     s    [[Xg'>?[[Yc
3[[],LM[[V\2A 3 N 4 @A [[2T35 :	:r:   r   medianr   trimmedc                 ,   [        SUS9u  pEnU " USU0UD6nU " USUS.UD6n[        UR                  R                  UR	                  UR                  5      5        [        UR
                  R                  UR	                  UR
                  5      5        g )NrT   rB   r>   rD   r   r9   r   r   r3   r*   r   ri   s	            r8   test_k_sample_testsr    s|    " 'qR0C'
w
,T
,V
,C
w6
?
?CCMM&&

3==(ABCJJOORZZ

%;<r:   z&JAX currently incompatible with marrayc                 .   [        SSUS9u  pEnU " USU0UD6nU " USUS.UD6n[        UR                  R                  UR	                  UR                  5      5        [        UR
                  R                  UR	                  UR
                  5      5        g )NrT   )r   	   r   r>   rD   r   r  ri   s	            r8   test_k_sample_paired_testsr    s~     'q2>C'
w
,T
,V
,C
w6
?
?CCMM&&

3==(ABCJJOORZZ

%;<r:   c                    [        SUS9u  pEnU [        R                  :X  a  0 OSU0nU " U0 UDSU0D6nUu  pUS:X  a  U	R                  U
R                  pOLUcI  U	R	                  5       [
        R                  S S 24   U
R	                  5       [
        R                  S S 24   p[        U	R                  S   5       GH  nXS4   XS4   pUc  SOUn[
        R                  " U5      [
        R                  " U5      -  nU " X)    X)    40 UD6n[        U5      (       a  U [        R                  :X  a  SOS	n[        UR                  R                  U   UR                  UR                  5      S   US
9  [        UR                   R                  U   UR                  UR                   5      S   US
9  U [        R"                  :X  d  GM  UR%                  5       u  nnUR%                  5       u  nn[        UR                  U   UR                  U5      S   5        [        UR                  U   UR                  U5      S   5        GM     g )Nr   rB   r~   r>   r   .r   gHz>g        r   )r9   r   pointbiserialrTravelr   newaxisr"   r   r{   r
   spearmanrhor   r   r3   r*   r   pearsonrr   )r}   r~   r>   r   r.   r4   rH   rG   rK   r   yr2   r   yir   rL   r   
res_ci_lowres_ci_high
ref_ci_lowref_ci_highs                        r8   test_correlationr    s    'qR0C',,,R=+2NF
W
*
*T
*C DAqyssACC1	wwyQ'2::q=)A11771:c6AfIB,BAxx|bhhrl*5	2e9// e.?.?)?tb**1-rzz#--/H/LSWX

*BJJszz,B2,FTR&)&=&=&?#J&)&=&=&?#JJOOA.

:0Fr0JKK,,Q/K1H1LM r:   array_api_strictzissue with clip.z	f, methodhierarchicalseparatejointr   r   r   c                 0   [        SSSUS9u  pEnU " USU0UD6nU " USUS.UD6n[        UR                  R                  UR	                  UR                  5      5        [        UR
                  R                  UR	                  UR
                  5      5        U [        R                  :X  as  [        UR                  R                  UR	                  UR                  5      5        [        UR                  R                  UR	                  UR                  5      5        g U [        R                  :X  a  [        UR                  R                  UR	                  UR                  5      5        [        UR                  R                  UR	                  UR                  5      5        [        UR                  R                  UR	                  UR                  5      5        [        UR                  R                  UR	                  UR                  5      5        g g )Nr   )r   r   l    wN4 )r   r   r   r>   rD   r   )r9   r   sloper3   r*   	interceptr   theilslopes	low_slope
high_slope
linregressrvaluer   stderrintercept_stderr)	r}   r   r>   r   r.   r4   rH   rK   rL   s	            r8   test_slopes_interceptsr.  #  sf    'q~RTUC'
W
*4
*6
*C
Wd
=f
=CCIINNBJJsyy$9:CMM&&

3==(ABE**BJJs}},EF++RZZ-GH	
e	

CJJ)?@

CJJ)?@

CJJ)?@,,112::c>R>R3ST	 
r:   z/Dask doesn't like the /= when base is not None.qkbasec                     [        U (       a  SOSUS9u  pEn[        R                  " XQUS.6n[        R                  " XaSUS.6n[        UR                  UR                  U5      5        g )Nr   r@   rB   )r0  r>   rD   )r0  rE   r>   )r9   r   entropyr   r3   r*   )	r/  r0  r>   r   r.   r4   rH   rK   rL   s	            r8   test_entropyr3  A  sQ     'BqA"=C'
--$
7C
--T
JCCHHbjjo.r:   )averager$   maxdenseordinalc           	         [        SUS9u  p4n[        R                  " X@US.6n[        R                  " XPSUS.6n[        UR                  UR
                  )    UR                  U[        R                  " U5      )    5      5        [        UR
                  UR                  [        R                  " U5      5      5        g )Nr@   rB   r   rD   )r   rE   r>   )	r9   r   rankdatar   r3   r   r*   r   r{   )r   r>   r   r.   r4   rH   rK   rL   s           r8   test_rankdatar:  N  s    
 'qR0C'
..'t
<C
..'V$
OCCHHchhY'C#4G)HICHHbjj#78r:   float32r-   c                 @   [        XSUS9u  p4n[        R                  " U6 n[        R                  " USS06n[        Xg5       HX  u  p[	        UR
                  UR                  )    UR                  U	[        R                  " U	5      )    [        X 5      S95        MZ     g )N)   )r   r   r   rE   rD   r   )r9   r   obrientransformr)   r   r3   r   r*   r   r{   rn   )
r   r-   r   r.   r4   rH   rK   rL   res_iref_is
             r8   test_obrientransformrA  Z  s     'xEbQC'



)C



<V
<CC

EJJ;/

5"((5/)9#:'"BT
U	W &r:   )ynumpyr   r   scipyr   scipy._lib._array_apir   r   r   r   r   r	   r
   scipy.stats._stats_pyr   r   importorskipr   r  skipif
pytestmarkskip_xp_backendsskip_backendr9   r  gmeanhmeanpmean	expectilerM   momentskewrw   semkstatkstatvarcircmeancircvarcircstdgstd	variationtmeanrh   tvartstdtsemtmintmaxiqrmedian_abs_deviation	trim_meanrj   rm   rp   rr   rx   zscoregzscorerz   r|   r   	ttest_rel	ttest_indr   filterwarningsr   tupler   
normaltestjarque_beracramervonmisesr   r   r   	chisquarer   r   r   r   r   r   r   r   r   r   r   r   r   r   kstestr   brunnermunzelr   r   r   r   r   r   bartlettalexandergovernlevenef_onewaykruskalflignerr  friedmanchisquarer  r  r  r  r  siegelslopesr'  r*  r.  r2  r3  r9  r:  r>  rA  r   r:   r8   <module>rv     s      T T I ; 3 
		X	&
KK6  
 {{++"+&T"$6 k"FGb)b)sAh/7C.92&)  eT]3./ / 4 H/ l#TUk"FG%%,,!5 %%,,38%%**b)% %**vuo6% %.."-	%
 %..UE*JK% %))R(% %))fa[1% %++Qx0% %++Qx0% %++Qx0% %++Qx0% %..3(3% %..3(3% %.."-% %..!Q*?@%  %--,!%" %--A)>?#%$ %--,%%& %--A4)PQ'%( %**b))%* %**vqk2+%, '2&-%. '!#FG/%0 %//2.1%2 %//FA;73%4 %++t:'>?5%6 %**d*&VW7%8 %**d*&VW9%: %**d*
&ST;%< %**dcT&JK=%> %**dcU&KL?%@ %))oy%ABA%B %))X%IJC%D %44b9E%F %44w6IJG%H %//,=s+CDI%&N ./ /O& H VT/ 5::7I"672v,/.; / 0 8 ; 5>>"l#TUk"FG.B#89J : / H V #
J" l#HIk"FG!5ell!C!5emm!D!5ejj!A!C D !Q(./ / )D H J/ l#TUk"FG3E4E4EF3EOOD3EOODF G (HI.? / JG H V?  l#TUk"FGUV%..%'2%,,eg6%**EG4%++UW5%..0AB%  (HI.I / J W H VI" l#TUk"FG"-//)Y1GH//)=M1NO//)_1MN//#%9:<//)X1FG//)^1LM	( 	 !Q(.9 / )	 H V94 5(()k"FG $; <.= /< H *
=, 5--.k"FG(HIudm4% 5 J H /%,3 5**+k"FG45udm4!Q(6 ) 5 6 H ,
6( 5>>"k"FGq!f-
3t}5(HIL>2.= / 3 J 6 4 . H #= 5%%&k"FG)E4=9(HIL>2.= / 3 J : H '= k"FG!5enn!E!5ell!C!E F#?@(HI.	C / J AF H	C k"FG,,b1,,~x.HI33h5HI)  .= / H= 5>>"k"FG!5enn!E!5ell!C!E F#=>(HI.C / J ?F H #C 5##$: : %:2 l#TUk"FG,..3(';<&'9:)'<=+t)<=+u)=>+8(<=6(:;9(=>)  .= / H V = k"JK00"5)  .= / L
= l#TUk"FG(--.**+ 
 (HI.N / J H VN> k"FG );<++h~-FG++hz-BC**Xw,?@**Xz,BC))M6+BC))M9+EF))M;+GH'  .U / = HU$ 5==!l#TUk"FGt}-$+./ / , . H V "/ 5>>"k"JK#PQ.9 / R L #9 5(()k"JK9i"89aV,W - : L *Wr:   