+
    nDj                    B   R t ^ RIt^ RIt^ RIt^ RIHt ^ RIHt ^ RIH	u H
t ^ RIt^ RIt^ RIHtHtHtHtHtHtHt ^ RIt^ RIHt ^ RI
HtHtHt ^ RI
t^ RIHt ^ RIH t  ^ RI!H"t" ^ R	I#H$t$H%t% ^R
I&H't' ^ RI(H)t)H*t*H+t+H,t,H-t- ^ RI.H/t/H0t0H1t1H2t2H3t3H4t4 ^ RI5H6t6 ^ RI7H8t8 ^ RI9H:t:H;t;H<t<H=t=H>t>H?t?H@t@HAtAHBtBHCtCHDtDHEtEHFtFHGtGHHtHHItIHJtJ ^ RIKHLtLHMtMHNtN ^ RIOHPu HQtR ^ RI5HStS ].tT]P                  P                  tV]P                  P                  tW ]! . RO]X4      tY]! . RO]X4      tZ]! . RO]X4      t[]! . RO]X4      t\]! . RO]X4      t]]! . RO]X4      t^]! . RO]X4      t_ ! R R4      t` ! R R4      ta]B! ]P                  4       ! R R4      4       tc ! R R4      td ! R R4      te ! R R4      tf]B! ]P                  4       ! R  R!4      4       th]B! ]P                  4       ! R" R#4      4       ti ! R$ R%4      tj]B! ]P                  4       ! R& R'4      4       tl ! R( R)4      tm]B! ]P                  4       ! R* R+]m4      4       to]B! ]P                  4       ! R, R-]m4      4       tq ! R. R/4      tr]B! ]P                  4       ! R0 R14      4       tt]B! ]P                  4       ! R2 R34      4       tv]B! ]P                  4       ! R4 R54      4       tx]B! ]P                  4       ! R6 R74      4       tz]B! ]P                  4       ! R8 R94      4       t|]B! ]P                  4       ! R: R;4      4       t~]B! ]P                  4       ! R< R=4      4       t]B! ]EP                  4       ! R> R?4      4       t ! R@ RA4      t]B! ]EP                  4       ! RB RC]4      4       t]B! ]EP                  4       ! RD RE]4      4       t]EP                  EP                  RF 4       t]B! ]EP                  4       ! RG RH4      4       t ! RI RJ4      t]! RK. RMO4      t]! . ROR^ R^^^]EP                  ! R4      ,          ^]EP                  ! R4      ,          ,           ,          ^^]EP                  ! R4      ,          ^]EP                  ! R4      ,          ,           ,          RRM7      ]! . RO. RO^ R^^^]EP                  ! R4      ,          ^]EP                  ! R4      ,          ,           ^]EP                  ! R4      ,          ,           ,          ^^]EP                  ! R4      ,          ^]EP                  ! R4      ,          ,           ^]EP                  ! R4      ,          ,           ,          RRM7      ]! . RO^^ R^^^]EP                  ! R4      ,          ^]EP                  ! R4      ,          ,           ,          ^^]EP                  ! R4      ,          ^]EP                  ! R4      ,          ,           ,          RRM7      ]! . RO. RO^ ^ ^ ^ ^ ^ RM7      .t]! . R^ ^ ^ ^ ^ ^ RM7      ]! ]P2                  ! . . . .4      EP"                  R^ ^ . RO. RO. RO. RORM7      ]! ]P2                  ! . . . .4      R^ ^ . . . . RM7      .t]B! ]EP&                  4       ! RN RO4      4       t]B! ]EP*                  4       ! RP RQ4      4       t]B! ]EP.                  4       ! RR RS4      4       t]B! ]EP2                  4       ! RT RU4      4       t]B! ]EP6                  4       ! RV RW4      4       t]B! ]EP:                  4       ! RX RY4      4       t]P                  EP?                  RZ4      ]P                  EP?                  R[4      ]B! ]EP@                  4       ! R\ R]4      4       4       4       t]P                  EPE                  R^]J! ]EP                  ]P4                  ! ^
4      ^ 34      ]J! ]EP@                  ]P4                  ! ^
4      ]P4                  ! ^
4      34      .4      R_ 4       tRR`R/Ra llt]B! ]EPJ                  ]EPL                  4       ! Rb Rc4      4       t]B! ]EPJ                  4      ]V! RdReRf7      ]P                  EP?                  Rg4       ! Rh Ri4      4       4       4       t]B! ]EPJ                  4       ! Rj Rk4      4       t]B! ]EPJ                  4       ! Rl Rm4      4       t]B! ]EPJ                  4       ! Rn Ro4      4       t]B! ]EPJ                  4       ! Rp Rq4      4       t]B! ]EPL                  4       ! Rr Rs4      4       t]B! ]EP                  4       ! Rt Ru4      4       t]B! ]EP^                  4       ! Rv Rw4      4       t ! Rx Ry4      t]B! ]EPd                  4       ! Rz R{]4      4       t]B! ]EPh                  4       ! R| R}]4      4       t]B! ]EPl                  4       ! R~ R]4      4       t ! R R4      t]B! ]EPr                  4       ! R R4      4       t]B! ]EPv                  4       ! R R4      4       t]B! ]EPz                  4       ! R R4      4       tRLRRRRRRR/R ltR tR tR t]B! ]EP                  4       ! R R4      4       t]B! ]EP                  4       ! R R4      4       t]B! ]EP                  4       ! R R4      4       t]B! ]EP                  4       ! R R4      4       t ! R R4      t]B! ]EP                  4       ! R R4      4       t]B! ]EP                  4       ! R R4      4       t]B! ]EP                  4       ! R R4      4       t]B! ]EP                  4       ! R R4      4       t]B! ]EP                  4       ! R R4      4       t]B! ]EP                  4       ! R R4      4       t ! R R4      t ! R R4      t ! R R4      t ! R R4      t]B! ]EP                  4       ! R R4      4       t]S! RRR4      R 4       t]B! ]EP                  4       ! R R4      4       t ! R R4      t]B! ]EP                  4       ! R R4      4       t]B! ]EP                  4       ! R R4      4       t]B! ]24       ! R R4      4       t]B! ]34       ! R R4      4       t]P                  EP                  RRRf7      R 4       t]V! RRRf7      ]P                  EPE                  R]J! ]EP                  4      ]J! ]EP6                  4      ]J! ]EP:                  4      ]J! ]EP2                  4      .4      R 4       4       tR# )zTest functions for stats module

WRITTEN BY LOUIS LUANGKESORN <lluang@yahoo.com> 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_equalassert_almost_equalassert_array_almost_equalassert_array_equalassert_approx_equalassert_allclose)raises)arrayarangepower)kolmogn)FuncData)optimizespecial)check_named_results)SmallSampleWarningtoo_small_nd_omittoo_small_nd_not_omittoo_small_1d_omittoo_small_1d_not_omit)_chk_asarray_momentLinregressResult_xp_mean_xp_var_SimpleChi2)	AxisError)skip_xp_invalid_arg)array_namespaceeager_warnsis_lazy_arrayis_numpyis_torchxp_default_dtypexp_sizeSCIPY_ARRAY_APImake_xp_test_casexp_ravelxp_swapaxesxp_result_typeis_jaxis_cupyxp_copy
xp_promotemake_xp_pytest_param)xp_assert_closexp_assert_equalxp_assert_less)_apply_over_batch      ?      ?c            	         a  ] tR t^Nt o ]P
                  ! ]P                  4      P                  t]	! ]
P                  4      R 4       t]	! ]
P                  4      ]P                  P!                  R4      R 4       4       t]	! ]
P$                  4      R 4       t]	! ]
P(                  4      R 4       t]P                  P-                  R]
P(                  ]
P.                  .4      R 4       t]! RR	R
7      ]P                  P-                  R]! ]
P(                  R4      ]! ]
P.                  R4      .4      R 4       4       t]	! ]
P.                  4      R 4       t]	! ]
P(                  ]
P.                  4      R 4       t]! RRR
7      ]	! ]
P(                  ]
P.                  4      R 4       4       t]	! ]
P>                  4      R 4       t Rt!V t"R# )TestTrimmedStatsc           	        \        V4      pVP                  \        VR 7      p\        P                  ! VRR4      p\        WAP                  R4      4       \        P                  ! VRRR7      p\        P                  ! VRR7      p\        WV4       VP                  VP                  R4      R4      p\        P                  ! VRR7      p\        WAP                  V4      4       \        P                  ! V^ R7      p\        WAP                  V^ R7      4       \        P                  ! V^R7      p\        WAP                  V^R7      4       \        P                  ! VRRR7      p\        WAP                  R	4      4       \        P                  ! VR^ R7      p. ROp\        WAP                  V4      4       \        P                  ! VRR^ R
7      p. ROp\        WAP                  V4      4       \        P                  ! V4      RRR13,          P                  VP                  RR7      p	\        P                  ! V	R^ R7      p^RRRVP                  VP                  VP                  .p\        WAP                  V4      4       \        P                  ! VR^R7      p^^
^^VP                  VP                  VP                  VP                  VP                  .	p\        WAP                  V4      4       \        P                  ! VRR^R
7      pR^
^^VP                  VP                  VP                  VP                  VP                  .	p\        WAP                  V4      4       R# )dtypeT      @limits	inclusiveNr?        O@axis)r?   rD   g     ?@)r?   r@   rD   )copy      @      @      @      TT)FF	      )rJ   =   )rJ      )         '@rN   
            TF)      %@rS   rN   rT   rU   rV   rW   )   rW   FT)r&   asarrayXstatstmeanr2   reshaper   meanxpxatsetnan)
selfxpdefault_dtypexyy1y2x_2dy_truex_2d_with_nans
   &&        V/data/cameron/venvs/s3viz/lib/python3.14/site-packages/scipy/stats/tests/test_stats.py
test_tmeanTestTrimmedStats.test_tmeanR   ss   (,JJqJ.KK6<0::c?+[[6^D[[4(zz"))C.&1KK4(774=)KK1%774a701KK1%774a701KKW48::d+,KKW15.::f-.KKWAN0::f-.tRW-11"&&t1DKKgA>S#sBFFBFFBFF;::f-.KKW15RRH::f-.KKW"/a9r2r266266266266266J::f-.    z>ignore:invalid value encountered in divide:RuntimeWarning:daskc                   VP                  \        P                  4       4      p\        P                  ! VRRR7      p\        W1P                  R4      4       \        P                  ! VRR7      p\        W1P                  V^R7      4       VP                  VP                  R4      R4      p\        P                  ! VRR7      p\        W1P                  V^R7      4       \        P                  ! V^ R7      p\        W1P                  RR4      4       \        P                  ! V^R7      p\        W1P                  RR	4      4       \        P                  ! VR^RR
7      p\        V^ ,          VP                  R4      4       \        P                  ! VR^RR
7      p\        V^ ,          VP                  R4      4       \        V^,          VP                  VP                  4      4       R# )rJ   r>   g@NrA   
correctionrB   rC   g     v@g㪪@)r?   rD   r@         @g@rI   rL   rM   rO   )rN   r\             )r^   r_   tolistr`   tvarr2   varrb   r   fullr3   rg   )rh   ri   rk   rl   ro   s   &&   rr   	test_tvarTestTrimmedStats.test_tvar   sY   
 JJqxxz"JJq<@::&89:JJq&66!623zz"))C.&1JJt$'66$1656JJt!$774/0JJt!$77445 JJtFlK!bjjo. JJtFlK!bjj):;<!bjj01ru   c                   VP                  \        P                  4       4      p\        P                  ! VRR4      p\        W1P                  R4      4       \        P                  ! VRR7      p\        W1P                  V^R7      4       R# )rJ   gBĆ/H@NrA   rw   rI   rL   )r^   r_   r   r`   tstdr2   stdrh   ri   rk   rl   s   &&  rr   	test_tstdTestTrimmedStats.test_tstd   s_    JJqxxz"JJq&,/::&89:JJq&66!623ru   c           	     \   VP                  R 4      p\        \        P                  ! V4      VP	                  R4      4       \        \        P                  ! V^ R7      VP	                  R4      4       \        \        P                  ! V^ RR7      VP	                  R4      4       VP                  VR
4      p\        \        P                  ! V^ RR7      VP	                  RR.4      4       \        \        P                  ! V^R7      VP	                  . RO4      4       \        \        P                  ! VRR7      VP	                  R4      4       \        P                  ! VP                  R 4      ^	4      P                  VP                  4      p\        \        P                  ! V4      VP	                  VP                  4      4       VP                  VP                  ^4      R4      p\        P                  ! V^^R	7      p\        W1P	                  \        P                  ^^^.4      4       R# )      $@        
lowerlimitF)r   r@         ?       @rC   N)r   rD   r|   rJ   )r   r         @      @       @   r   )r   r3   r`   tminr^   rb   rd   re   rf   rg   nprh   ri   rk   ress   &&  rr   	test_tminTestTrimmedStats.test_tmin   si   IIcN

1rzz"~6

13RZZ^D

1eDbjjQSnUJJq&!

1eD

B8,	.

11-rzz:N/OP

140"**R.AFF299S>1%))"&&1

1rzz"&&'9: JJryy}f-jjqq1ZZAr(:;<ru   tfunc                *    \        V! ^4      ^4       R# )r   N)r   )rh   r   s   &&rr   test_tmin_tmax_scalar_input,TestTrimmedStats.test_tmin_tmax_scalar_input   s    T!Wa ru   T'lazy -> reduced nan_policy capabilities
eager_onlyreasonz	tfun, refr         "@c                   VP                  R 4      pVP                  V^&   \        V! VRR7      VP                  V4      4       Rp\	        \
        VR7      ;_uu_ 4        V! VRR7       RRR4       Rp\	        \
        VR7      ;_uu_ 4        V! VRR7       RRR4       R#   + '       g   i     L@; i  + '       g   i     R# ; i)	r   omit
nan_policyThe input contains nan valuesmatchraiseNnan_policy must be one of...foobar)r   rg   r3   r^   assert_raises
ValueError)rh   r   refri   rk   msgs   &&&&  rr   test_tmin_tmax_nanpolicy)TestTrimmedStats.test_tmin_tmax_nanpolicy   s     IIcNvv!Q62BJJsOD-:S11w' 2,:S11x( 21 21 211s   B"B5"B2	5C	c                   VP                  R 4      p\        \        P                  ! V4      VP	                  R4      4       \        \        P                  ! V^	R7      VP	                  R4      4       \        \        P                  ! V^	RR7      VP	                  R4      4       VP                  VR4      p\        \        P                  ! V^	RR7      VP	                  RR.4      4       \        \        P                  ! V^R7      VP	                  . RO4      4       \        \        P                  ! VRR7      VP	                  R4      4       \        P                  ! VP                  R 4      ^	4      P                  VP                  4      p\        \        P                  ! V4      VP	                  VP                  4      4       \        P                  ! 4       ;_uu_ 4        \        P                  ! R	R
\        4       VP                  VP                  ^4      R4      p\        P                  ! V^^R7      p\        W1P	                  ^^^\        P                  .4      4       RRR4       R#   + '       g   i     R# ; i)r   r   
upperlimitF)r   r@   r         @rC   NignorezAll-NaN slice encountered)r   rD   r   )r         @r=   r   r   r   )r   r3   r`   tmaxr^   rb   rd   re   rf   rg   warningscatch_warningsfilterwarningsRuntimeWarningr   r   s   &&  rr   	test_tmaxTestTrimmedStats.test_tmax   s   IIcN

1rzz"~6

13RZZ^D

1eDbjjQSnUJJq&!

1eD

B8,	.

11-rzz:N/OP

140"**R.AFF299S>1%))"&&1

1rzz"&&'9: $$&&##5~G

299R=&1A**Q2A6CCQ2rvv,>!?@ '&&&s   #BI00J	c                   VP                  VP                  ^
VP                  R7      R4      P                  p\	        V4      '       d   \        V4      MVP                  p\        \        P                  ! V4      VP                  ^ ^.VR7      4       \        \        P                  ! V4      VP                  ^^	.VR7      4       \        \        P                  ! V^R7      VP                  VP                  R.4      4       \        \        P                  ! V^R7      VP                  RVP                  .4      4       R# )rT   r;   r   r   r   r   NrJ   r|   )rb   r   int16Tr#   r&   r<   r3   r`   r   r^   r   rg   )rh   ri   rk   expect_dtypes   &&  rr   test_tmin_tmax_int_dtype)TestTrimmedStats.test_tmin_tmax_int_dtype   s    JJryy288y4f=??
 0=Q/?/?'+QWW

1rzz1a&z'MN

1rzz1a&z'MN 	

13RZZ5MN

13RZZRVV5MNru   z%Only with data-dependent output dtypec                    VP                  RR4      p\        \        P                  ! VR,          4      V4       \        \        P                  ! VR,          4      V4       R# )rJ   Nl            l   
        )NNNN)r   r3   r`   r   r   rh   ri   rk   s   && rr   test_gh_22626TestTrimmedStats.test_gh_22626   sB     IIeX&

1W:.2

1W:.2ru   c                ~   VP                  \        P                  4       4      p\        P                  ! VRRR7      pVP                  . RO4      p\        W1P                  V^R7      \        V4      R,          ,          4       \        \        P                  ! VR	^
.R7      \        P                  ! VRR7      4       R# )
   r>   rw   r6   rA   N)r   rK   r]   )r   r=   r   r   r   rZ   )r^   r_   r   r`   tsemr2   r   r'   )rh   ri   rk   rl   y_refs   &&   rr   	test_tsemTestTrimmedStats.test_tsem  s~    JJqxxz"JJq=A

/066%A669LLM

1b"X6

1T8RSru    N)#__name__
__module____qualname____firstlineno__r   finfofloat64	precisiondprecr)   r`   ra   rs   r   pytestmarkr   r   r   r   r   r   parametrizer   r   skip_xp_backendsr1   r   r   r   r   r   r   __static_attributes____classdictcell____classdict__s   @rr   r9   r9   N   s    HHRZZ **Eu{{#,/ $,/\ uzz"[[H2 #28 uzz"4 #4 uzz"= #=* [[Vejj%**%=>! ?! .WX[[[+?

B+O+?

B+O+Q R	)R Y	) uzz"A #A0 uzz5::.O /O .UVuzz5::.3 / W3 uzz"T #Tru   r9   c                      a  ] tR tRt o RtR tR tR tR tR t	R t
R	 tR
 tR tR tR tR tR tR tR tR tR tR tR tR tR tRtV tR# )TestPearsonrWilkinsoni     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.
c                n    \         P                  ! \        \        4      pV^ ,          p\        VR4       R# r~   r   N)r`   pearsonrr_   r	   rh   rl   rs   &  rr   test_pXXTestPearsonrWilkinson.test_pXX  s%    NN1QaDAc"ru   c                n    \         P                  ! \        \        4      pV^ ,          p\	        VR4       R# r   )r`   r   r_   BIGr	   r   s   &  rr   
test_pXBIG TestPearsonrWilkinson.test_pXBIG  s%    NN1S!aDAc"ru   c                n    \         P                  ! \        \        4      pV^ ,          p\	        VR4       R# r   )r`   r   r_   LITTLEr	   r   s   &  rr   test_pXLITTLE#TestPearsonrWilkinson.test_pXLITTLE   s%    NN1V$aDAc"ru   c                n    \         P                  ! \        \        4      pV^ ,          p\	        VR4       R# r   )r`   r   r_   HUGEr	   r   s   &  rr   test_pXHUGE!TestPearsonrWilkinson.test_pXHUGE%  %    NN1T"aDAc"ru   c                n    \         P                  ! \        \        4      pV^ ,          p\	        VR4       R# r   )r`   r   r_   TINYr	   r   s   &  rr   test_pXTINY!TestPearsonrWilkinson.test_pXTINY*  r   ru   c                n    \         P                  ! \        \        4      pV^ ,          p\	        VR4       R# r   )r`   r   r_   ROUNDr	   r   s   &  rr   test_pXROUND"TestPearsonrWilkinson.test_pXROUND/  s%    NN1U#aDAc"ru   c                n    \         P                  ! \        \        4      pV^ ,          p\        VR4       R# r   )r`   r   r   r	   r   s   &  rr   test_pBIGBIG"TestPearsonrWilkinson.test_pBIGBIG4  s%    NN3s#aDAc"ru   c                n    \         P                  ! \        \        4      pV^ ,          p\	        VR4       R# r   )r`   r   r   r   r	   r   s   &  rr   test_pBIGLITTLE%TestPearsonrWilkinson.test_pBIGLITTLE9  s%    NN3v&aDAc"ru   c                n    \         P                  ! \        \        4      pV^ ,          p\	        VR4       R# r   )r`   r   r   r   r	   r   s   &  rr   test_pBIGHUGE#TestPearsonrWilkinson.test_pBIGHUGE>  %    NN3t$aDAc"ru   c                n    \         P                  ! \        \        4      pV^ ,          p\	        VR4       R# r   )r`   r   r   r   r	   r   s   &  rr   test_pBIGTINY#TestPearsonrWilkinson.test_pBIGTINYC  r  ru   c                n    \         P                  ! \        \        4      pV^ ,          p\	        VR4       R# r   )r`   r   r   r  r	   r   s   &  rr   test_pBIGROUND$TestPearsonrWilkinson.test_pBIGROUNDH  s%    NN3u%aDAc"ru   c                n    \         P                  ! \        \        4      pV^ ,          p\        VR4       R# r   )r`   r   r   r	   r   s   &  rr   test_pLITTLELITTLE(TestPearsonrWilkinson.test_pLITTLELITTLEM  s%    NN6&)aDAc"ru   c                n    \         P                  ! \        \        4      pV^ ,          p\	        VR4       R# r   )r`   r   r   r   r	   r   s   &  rr   test_pLITTLEHUGE&TestPearsonrWilkinson.test_pLITTLEHUGER  %    NN6$'aDAc"ru   c                n    \         P                  ! \        \        4      pV^ ,          p\	        VR4       R# r   )r`   r   r   r   r	   r   s   &  rr   test_pLITTLETINY&TestPearsonrWilkinson.test_pLITTLETINYW  r  ru   c                n    \         P                  ! \        \        4      pV^ ,          p\	        VR4       R# r   )r`   r   r   r  r	   r   s   &  rr   test_pLITTLEROUND'TestPearsonrWilkinson.test_pLITTLEROUND\  s%    NN6%(aDAc"ru   c                n    \         P                  ! \        \        4      pV^ ,          p\        VR4       R# r   )r`   r   r   r	   r   s   &  rr   test_pHUGEHUGE$TestPearsonrWilkinson.test_pHUGEHUGEa  %    NN4%aDAc"ru   c                n    \         P                  ! \        \        4      pV^ ,          p\	        VR4       R# r   )r`   r   r   r   r	   r   s   &  rr   test_pHUGETINY$TestPearsonrWilkinson.test_pHUGETINYf  r'  ru   c                n    \         P                  ! \        \        4      pV^ ,          p\	        VR4       R# r   )r`   r   r   r  r	   r   s   &  rr   test_pHUGEROUND%TestPearsonrWilkinson.test_pHUGEROUNDk  %    NN4&aDAc"ru   c                n    \         P                  ! \        \        4      pV^ ,          p\        VR4       R# r   )r`   r   r   r	   r   s   &  rr   test_pTINYTINY$TestPearsonrWilkinson.test_pTINYTINYp  r'  ru   c                n    \         P                  ! \        \        4      pV^ ,          p\	        VR4       R# r   )r`   r   r   r  r	   r   s   &  rr   test_pTINYROUND%TestPearsonrWilkinson.test_pTINYROUNDu  r.  ru   c                n    \         P                  ! \        \        4      pV^ ,          p\        VR4       R# r   )r`   r   r  r	   r   s   &  rr   test_pROUNDROUND&TestPearsonrWilkinson.test_pROUNDROUNDz  s%    NN5'aDAc"ru   r   N)r   r   r   r   __doc__r   r   r   r   r  r  r  r  r  r  r  r  r  r  r"  r%  r)  r,  r0  r3  r6  r   r   r   s   @rr   r   r     sx     #
#
#
#
#
#
#
#
#
#
#
#
#
#
#
#
#
#
#
#
# #ru   r   c                   >  a  ] tR tRt o R tR tR tR tR t]	P                  P                  RRR	.4      R
 4       tR tR tR t]	P                  P!                  R4      ]	P                  P!                  R4      R 4       4       t]	P                  P!                  R4      ]	P                  P!                  R4      R 4       4       t]	P                  P!                  R4      R 4       t]	P                  P                  R. R'O4      R 4       tR t]	P                  P!                  R4      R 4       tR t]	P                  P1                  ^
4      ]	P                  P3                  R4      ]	P                  P                  RR(4      ]	P                  P                  RR)4      R 4       4       4       4       t]	P                  P6                  ]	P                  P                  RR(4      R 4       4       t]	P                  P                  R^ ^.4      R 4       tR  tR! t]	P                  P!                  R4      ]	P                  P!                  R4      R" 4       4       t R# t!]	P                  P                  R. R*O4      ]	P                  P                  R. R(O4      R% 4       4       t"R&t#V t$R$# )+TestPearsonri  c                    \         P                  ! VP                  \        4      VP                  \        4      4      pRp\	        W#VR7       \        VP                  VP                  4       R# )correlationri   Nr<  pvalue)r`   r   r^   r_   r   r3   r<  	statistic)rh   ri   r   
attributess   &&  rr   test_pearsonr_result_attributes,TestPearsonr.test_pearsonr_result_attributes  sC    nnRZZ]BJJqM:.
C37ru   c           
     &   VP                  R 4      p\        P                  ! W"4      w  r4\        W1P	                  R4      RR7       \        WAP	                  R4      \
        P                  ! ^\
        P                  ! R4      ,          4      R7       R# )r   r   V瞯<atolr   Nr   r`   r   r2   r^   r   sqrtspacingrh   ri   ar   probs   &&   rr   test_r_almost_exactly_pos1'TestPearsonr.test_r_almost_exactly_pos1  s[    IIcN..&::c?7 	jjoBGGAbjjo<M4NOru   c           
     (   VP                  R 4      p\        P                  ! W") 4      w  r4\        W1P	                  R4      RR7       \        WAP	                  R4      \
        P                  ! ^\
        P                  ! R4      ,          4      R7       R# )r   r   rE  rF  r   N      rH  rK  s   &&   rr   test_r_almost_exactly_neg1'TestPearsonr.test_r_almost_exactly_neg1  s^    IIcN..B'::d+%8 	jjoBGGAbjjo<M4NOru   c                    VP                  . RO4      pVP                  . RO4      p\        P                  ! W#4      w  rE\        WAP                  R4      4       \        WQP                  R4      4       R# )r\   NrZ   r~   r\   )r~   r~   r   LXz?UUUUUU?)r^   r`   r   r2   )rh   ri   rL  br   rM  s   &&    rr   
test_basicTestPearsonr.test_basic  sQ     JJz"JJy!..&::h/0jjo.ru   c                   VP                  . RO4      pVP                  . RO4      pRp\        \        P                  WAR7      ;_uu_ 4        \        P                  ! W#4      w  rV\        WQP                  VP                  4      4       \        WaP                  VP                  4      4       RRR4       R#   + '       g   i     R# ; i)MbX?An input array is constantr   ri   Nr\  r\  r\  rh|?v/?gsh|??r^   r"   r`   ConstantInputWarningr   r2   rg   rh   ri   rk   rl   r   r   ps   &&     rr   test_constant_input TestPearsonr.test_constant_input  s     JJ,-JJ,-*333FF>>!'DAAzz"&&12Azz"&&12 GFFF   A!B77C	r<   float32r   c                   \        \        V4      p\        W!4      pVP                  ^^^\        P                  ! ^VR7      ,           .VR7      pVP                  ^^^^\        P                  ! ^VR7      ,          ,           .VR7      pRp\	        \
        P                  WbR7      ;_uu_ 4        \
        P                  ! WE4       RRR4       R#   + '       g   i     R# ; i)rJ   r;   z/An input array is nearly constant; the computedr^  N)getattrr   r^   rJ  r"   r`   NearConstantInputWarningr   )rh   r<   ri   npdtyperk   rl   r   s   &&&    rr   test_near_constant_input%TestPearsonr.test_near_constant_input  s    "e$"JJ1a"**Qg">>?uJMJJ1a!BJJq$@"@@AJO?77sJJ NN1  KJJJs   .CC 	c                J   VP                  . ROVP                  R7      pVP                  . ROVP                  R7      p\        P                  ! W#4      w  rE\	        WAP                  RVP                  R7      4       \	        WQP                  RVP                  R7      4       R# )y)r?r;   ge;E?g3R?N)rr  gxd{s?gnp?g$9"o?ge =:k?)goȟg`
,gzg2l?g>??'r^   r   r`   r   r2   rh   ri   rk   rl   r   rf  s   &&    rr   test_very_small_input_values)TestPearsonr.test_very_small_input_values  s     JJVZZ  )JJNZZ  )~~a# 	::&8

:KL::&8

:KLru   c                b   R VP                  . ROVP                  R7      ,          pR VP                  ^VP                  R7      ,          p\        P                  ! W#4      w  rE\        WAP                  RVP                  R7      4       \        WQP                  RVP                  R7      4       R# )gY)	kRr;   rV  g͗~-?N)r~   r~   r~   r\   r\   r\   r\   )r^   r   r   r`   r   r2   rt  s   &&    rr   test_very_large_input_values)TestPearsonr.test_very_large_input_values  s     1DD1BJJ//~~a# 	::&8

:KL::&:"**:MNru   c                J   VP                  . ROVP                  R7      pVP                  . ROVP                  R7      p\        P                  ! W#4      w  rE\	        WAP                  RVP                  R7      4       \	        WQP                  RVP                  R7      4       R# )7	ir;   g*D{?gݣ?N)r{  gNig_b4igZbi)gDig<'(½igmQiZbtirs  rt  s   &&    rr   !test_extremely_large_input_values.TestPearsonr.test_extremely_large_input_values  sx     JJ9JLJJ;2::JN~~a# 	::&7rzz:JK::&7rzz:JKru   4ignore:invalid value encountered:RuntimeWarning:dask5ignore:divide by zero encountered:RuntimeWarning:daskc                &   VP                  R R.4      pVP                  RR.4      p\        P                  ! W#4      pVw  rVVP                  R 4      p\        WW4       \        Wg4       VP	                  4       w  r\        W) 4       \        W4       R# )r   r   r   r=   Nr^   r`   r   r3   confidence_interval
rh   ri   rk   rl   r   r   rf  onelowhighs
   &&        rr   test_length_two_pos1!TestPearsonr.test_length_two_pos1  s|    
 JJBx JJBx nnQ"jjn++-	T""ru   c                (   VP                  R R.4      pVP                  RR.4      p\        P                  ! W#4      pVw  rVVP                  R4      p\        WW) 4       \        Wg4       VP	                  4       w  r\        W) 4       \        W4       R# )r   r   r   r=   Nr  r  s
   &&        rr   test_length_two_neg1!TestPearsonr.test_length_two_neg1  s~    
 JJBx JJBx nnQ"jjn4 ++-	T""ru   *ignore:invalid value encountered in dividec                   VP                  R R .4      pVP                  RR.4      pRp\        \        P                  WAR7      ;_uu_ 4        \        P                  ! W#4      w  rV\        WQP                  VP                  4      4       \        WaP                  VP                  4      4       RRR4       R#   + '       g   i     R# ; i)r\  ra  rb  r]  r^  Nrc  re  s   &&     rr   test_length_two_constant_input+TestPearsonr.test_length_two_constant_input  s    
 JJu~&JJu~&*333FF>>!'DAAzz"&&12Azz"&&12 GFFFri  z$alternative, pval, rlow, rhigh, signc                   VP                  . RO4      pVP                  . RO4      V,          p\        P                  ! WxVR7      p	\        V	P                  VP                  RV,          4      4       \        V	P
                  VP                  V4      4       V	P                  4       p
\        V
P                  VP                  V4      4       \        V
P                  VP                  V4      4       R# )r   alternativeg?Nr   r   r   r   )r   r   r6   r   )	r^   r`   r   r2   r@  r?  r  r  r  )rh   r  pvalrlowrhighsignri   rk   rl   resultcis   &&&&&&&    rr   test_basic_exampleTestPearsonr.test_basic_example  s     JJ'(JJ()D0+>(("**5G5L*MNrzz$'78'')

4 01E!23ru   c                &   VP                  R 4      pV) p\        P                  ! W#RR7      p\        P                  ! W#RR7      p\        VP                  VP                  R4      4       \        VP                  VP                  R4      RR7       R# )	r   greaterr  lessr   r   g#B;rF  N)r   r`   r   r2   r?  r^   )rh   ri   rk   rl   test_greater	test_lesss   &&    rr   (test_negative_correlation_pvalue_gh177955TestPearsonr.test_negative_correlation_pvalue_gh17795*  sh    IIcNB~~a	BNN1V<	++RZZ^<	(("**R.uEru   c                <   VP                  . RO4      pVP                  . RO4      p\        P                  ! W#4      pVw  rVVP                  R 4      p\        WW) 4       \        V^ V,          RR7       VP	                  4       w  r\        W) 4       \        W4       R# )r   Hz>rF  Nr   r   r   )r=   g      g      *)r^   r`   r   r2   r  r3   r  s
   &&        rr   #test_length3_r_exactly_negative_one0TestPearsonr.test_length3_r_exactly_negative_one2  s~    JJ|$JJ'nnQ" jjo4 1S5t,++-	T""ru   c                   . ROp^.pRp\         P                  ! \        VR7      ;_uu_ 4        \        P                  ! W4       RRR4       . ROp^^.pRp\         P                  ! \        VR7      ;_uu_ 4        \        P                  ! W4       RRR4       ^.p^.pRp\         P                  ! \        VR7      ;_uu_ 4        \        P                  ! W4       RRR4       . ROp. ROpRp\         P                  ! \        VR7      ;_uu_ 4        \        P                  ! W4       RRR4       Rp\         P                  ! \        VR7      ;_uu_ 4        \        P                  ! ^^.^^.RR	7       RRR4       \        P                  ! ^^.^^.4      p\         P                  ! \        VR7      ;_uu_ 4        VP                  R
R	7       RRR4       R#   + '       g   i     EL; i  + '       g   i     ELe; i  + '       g   i     EL.; i  + '       g   i     L; i  + '       g   i     L; i  + '       g   i     R# ; i)r\   z3`x` and `y` must have the same length along `axis`.r   N"`x` and `y` must be broadcastable.(`x` and `y` must have length at least 2.z+This function does not support complex dataz"`method` must be an instance of...
asymptoticmethodexactr\   rJ   r   )             y              y             )r   r   r   r`   r   r  )rh   rk   rl   messager   s   &    rr   test_input_validation"TestPearsonr.test_input_validationA  s   CG]]:W55NN1  6 F6]]:W55NN1  6 CC<]]:W55NN1  6 ?]]:W55NN1  6 7]]:W55NNAq6Aq6,? 6 nnaVaV,]]:W55##7#3 655 655 655 655 65 65 655sG   G;G.HHH)=H<G+	.G?	H	H&	)H9	<I	z-Monte Carlo method needs > a few kB of memoryr  method_namec           	     D   \         P                  P                  R 4      pVR8X  d   RMRpVP                  VR7      pVP                  VR7      pR\        P
                  ! VR7      R\        P                  ! VP                  3^,          R7      R\        P                  ! RR7      /pWq,          p\        P                  ! WVW(RR7      p	\        P                  ! WVVRR	7      p
\        V	P                  V
P                  R
R7       \        V	P                  V
P                  RRR7       VR8X  ds   \        P                  ! RR7      p\        P                  ! WVW(RR7      p\        VP                  V	P                  4       \        VP                  V	P                  4       R# R# )l   o#[mpermutationsizerngmonte_carlorvsmonte_carlo2i  )r  r  rD   r  rD   rE  rtol{Gz?MbP?r  rG  NrJ   d   )rJ     rZ   )r   randomdefault_rngnormalr`   PermutationMethodMonteCarloMethodr   r
   r@  r?  r   )rh   r  r  r  r  rk   rl   methodsr  r   r   res2s   &&&         rr   test_resampling_pvalue#TestPearsonr.test_resampling_pvaluec  s6    ii##$56&-7xYJJDJ!JJDJ! %"9"9c"B %"8"8cjj]1_"M!5#9#9d#CE %nnQ{PRSnnQ{Ds}}5A

CJJTE.(++5F>>!KUWXD7cjj1	 )ru   c                .   \         P                  P                  R 4      pVP                  R
R7      pVP                  R
R7      p\        P
                  ! W4VRR7      p\         P                  P                  R4      p\        P                  ! VR7      pVP                  VR7      pVP                  4       p\        WxRR7       \         P                  P                  R4      p\        P                  ! VR7      pVP                  VR7      p	\        W4       R	# )l   _Y F r  r  l   |#(AkPrandom_stater  gQ?rF  r  Nr  rZ   )	r   r  r  r  r`   r   BootstrapMethodr  r
   )
rh   r  r  rk   rl   r   r  res_ciref_cires_ci2s
   &&        rr   test_bootstrap_ciTestPearsonr.test_bootstrap_ci|  s     ii##$45JJHJ%JJHJ%nnQ{D ii##$67&&C8(((7((*V4 ii##$67&&3/)))8(ru   rD   c                   \         P                  P                  R 4      pRpVP                  RV,           R7      w  rE\        P
                  ! WEVR7      pVP                  4       pV^ 8X  d   VP                  VP                  rT\        VP                  ^ ,          4       F  p\        P
                  ! WH,          WX,          4      p	V	P                  4       p
\        VP                  V,          V	P                  4       \        VP                  V,          V	P                  4       \        VP                  V,          V
P                  4       \        VP                  V,          V
P                  4       K  	  R# )   I.v# r  rC   NrN   rT   rJ   )r   r  r  r  r`   r   r  r   rangeshaper
   r@  r?  r  r  )rh   rD   r  r  rk   rl   r   r  ires_ici_is   &&         rr   test_axis01TestPearsonr.test_axis01  s    ii##K0zzte|z,nnQ-$$&1933qqwwqz"ANN14.E,,.DCMM!,eoo>CJJqM5<<8BFF1Itxx0BGGAJ		2 #ru   c                   \         P                  P                  R 4      pRpVP                  RV,           R7      w  r4\        P
                  ! W4RR7      pVP                  4       p\        P
                  ! VP                  4       VP                  4       4      pVP                  4       p\        VP                  VP                  4       \        VP                  VP                  4       \        Wh4       R# )r  r  NrC   r  r  )r   r  r  r  r`   r   r  ravelr
   r@  r?  )	rh   r  r  rk   rl   r   r  r   ci_refs	   &        rr   test_axis_NoneTestPearsonr.test_axis_None  s    ii##K0zzte|z,nnQ-$$&nnQWWY	2((*s}}5

CJJ/#ru   c                   VP                  R4      ;r#Rp\        P                  ! \        VR7      ;_uu_ 4        \        P
                  ! W#RR7       RRR4       Rp\        P                  ! \        VR7      ;_uu_ 4        \        P
                  ! W!P                  R4      ^R7       RRR4       Rp\        P                  ! \        VR7      ;_uu_ 4        \        P
                  ! VP                  R4      VP                  R4      ^R7       RRR4       Rp\        P                  ! \        VR7      ;_uu_ 4        \        P
                  ! W!P                  R4      ^R7       RRR4       R	p\        V4      '       gl   VP                  ^
4      p\        P                  ! \        VR7      ;_uu_ 4        \        P
                  ! W"\        P                  ! 4       R
7       RRR4       R# R#   + '       g   i     EL; i  + '       g   i     ELa; i  + '       g   i     EL; i  + '       g   i     L; i  + '       g   i     R# ; i)rJ   z`axis` must be an integer.r   r7   rC   Nz2`x` and `y` must have the same length along `axis`r  r  z:`method` must be `None` if arguments are not NumPy arrays.r  r   rJ   r\   r   r|   )	onesr   r   r   r`   r   r$   r   r  )rh   ri   rk   rl   r  s   &&   rr   test_nd_input_validation%TestPearsonr.test_nd_input_validation  sk   .]]:W55NN1c* 6 G]]:W55NN1ggfoA6 6 =]]:W55NN2776?BGGFO!D 6 7]]:W55NN1ggfoA6 6 O||		"Az99qE,C,C,EF :9 ! 655 655 655 65 :99s;   G, (H 8H<(H(4,H;,G=	 H	H%	(H8	;I	c                   \         P                  P                  R 4      pVP                  R
4      VP                  R
4      rC^VR&   ^VR&   VP                  V4      VP                  V4      reRp\	        \
        P                  WqR7      ;_uu_ 4        \
        P                  ! WV^R7      pVP                  4       p	VP                  VP                  VP                  .VP                  R7      p
\        VP                  R,          V
4       \        VP                  R,          V
4       \        V	P                  R,          V
4       \        V	P                  R,          V
4       VP!                  VP#                  VP                  R,          4      4      '       g   Q hVP!                  VP#                  VP                  R,          4      4      '       g   Q hVP!                  VP#                  V	P                  R,          4      4      '       g   Q hVP!                  VP#                  V	P                  R,          4      4      '       g   Q h RRR4       RRuVR&   VR&   VP                  V4      VP                  V4      reRp\	        \
        P$                  WqR7      ;_uu_ 4        \
        P                  ! WV^R7       RRR4       VP                  ^^.^^.^^.^R	..4      pVP                  ^^.^^.^^.^R	..4      pVP'                  ^4      p\
        P                  ! WVRR7      pVP                  4       p	\)        VP                  VP                  . RO4      4       \)        VP                  V4       \)        V	P                  V) 4       \)        V	P                  V4       R#   + '       g   i     ELb; i  + '       g   i     EL; i)l   U!Jr]  r^  rC   r;   r~   rJ   NrJ   NNNz!An input array is nearly constantr   r   r|   r~   .r\   .g     ?g      @r~   r~   r\   r\   rZ   )r\   rZ   rZ   r   )r   r  r  r^   r"   r`   rd  r   r  rg   r   r3   r@  r?  r  r  allisfiniterm  r  r2   )rh   ri   r  x0y0rk   rl   r  r   r  nansr  s   &&          rr   test_nd_special_cases"TestPearsonr.test_nd_special_cases  s    ii##N3F#SZZ%7B6
6
zz"~rzz"~1.337JJ..A.C((*B::rvvrvv.bjj:ADCMM#.5CJJsOT2BFF3K.BGGCL$/66"++cmmB&78999966"++cjjn5666666"++bffRj1222266"++bggbk23333 K '	4"T(zz"~rzz"~1577wNNNN1a( O JJAAAB89JJAAAB89wwqznnQ+$$&rzz/'BC

D)&&9 KJJ  ONNs+   D?N8N
8NNN3N0	3O	c                (   \         P                  P                  R 4      pVP                  R4      pVP                  ^
4      p\        P                  ! W4^R7      p\        P                  ! W4RR7      p\        VP                  VP                  4       R# )l   [LYrC   N)r   rT   rZ   )r   r  r  r`   r   r   r@  rh   ri   r  rk   rl   r   r   s   &&     rr   test_different_dimensionality*TestPearsonr.test_different_dimensionality  sf     ii##$56JJwJJrNnnQ*nnQ+S]]CMM2ru   Nc                   \         P                  P                  R 4      pVP                  RR7      w  rV\        P
                  ! VP                  V4      VP                  V4      W#R7      p\        P
                  ! WVW#R7      p\        VP                  VP                  VP                  4      4       \        VP                  VP                  VP                  4      4       VP                  4       p	VP                  4       p
\        V	P                  VP                  V
P                  4      4       \        V	P                  VP                  V
P                  4      4       R# )i8r  rD   r  N)rJ   rT   rU   )r   r  r  r  r`   r   r^   r2   r@  r?  r  r  r  )rh   ri   rD   r  r  rk   rl   r   r   r  r  s   &&&&       rr   test_array_apiTestPearsonr.test_array_api   s     ii##J/zz{z+nnRZZ]BJJqM"&AnnQFrzz#--'@A

BJJszz$:;((*((*

BJJvzz$:;RZZ%<=ru   r   ))	two-sided?gmgƎ?r\   )r  K˅?rQ  g?
?r\   )r  .?gդζr   r\   )r  r	  gfǎgl?rZ   )r  r  rQ  gդζ?rZ   )r  r
  g?
r   rZ   r  r  r  )r  r  r  r~   r\   N)%r   r   r   r   rB  rN  rR  rY  rg  r   r   r   ro  ru  rx  r}  r   r  r  r  r  r  r  r  	fail_slowxfail_on_32bitr  slowr  r  r  r  r  r  r  r   r   r   s   @rr   r:  r:    s    8PP/	3 [[Wy)&<=
! >
!MOL [[ VW[[ WX# Y X# [[ VW[[ WX# Y X# [[ LM
3 N
3& [[CHI4I4F [[ WX# Y# 4D [[2[[ OP[[],LM[[]KM2M N Q 
2( [[[[],LM) N )& [[VaV,3 -3 
$G0 [[ VW[[ WX$' Y X$'L3$ [[V\2[[],LM> N 3>ru   r:  c                     a  ] tR tRt o RtR tR tR tR t]	P                  P                  R 4       tR tR	 tR
 tR t]	P                  P#                  R. RO4      R 4       tR t]	P                  P)                  ^
4      ]	P                  P                  4       R 4       4       t]	P                  P)                  ^
4      ]	P                  P                  4       R 4       4       t]	P                  P/                  4       R 4       tRtV tR# )TestFisherExacti  a  Some 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
c                F   \         P                  pV! R R.RR..4      ^,          p\        VR^R7       V! ^d^.R^..4      ^,          p\        VR^R7       V! ^^.^^..4      ^,          p\        VR^R7       V! ^^.^
^
..4      ^,          p\        VR	^R7       V! ^^.^^..4      ^,          p\        VR
^R7       V! ^^.^^..4      ^,          p\        VR^R7       V! ^
^.^
^..4      ^,          p\        VR	^R7       V! ^^ .^^..4      ^,          p\        VR^R7       V! ^ ^.^^..4      ^,          p\        VR4       V! ^ ^.^^..4      ^,          p\        VR4       V! ^^.^^..4      p\        V^,          R^R7       \        V^ ,          R4       R# )8   N  0u  @  g؞Y?significantr  g,Ԛ?g2c?gY
A?g_?gO^M?gօa?r   ggE?Ng$I$I?)r`   fisher_exactr	   )rh   r  r   s   &  rr   rY  TestFisherExact.test_basic  s   ))UENUEN;<Q?Ca8S!HtQi01!4CQ7QFQF+,Q/C:QFRH-.q1C:QGb"X./2C:QGb"X./2C:RGb!W-.q1C:QFQF+,Q/C;QFQF+,Q/C%QFQF+,Q/C.QFQF+,CFI1=CFH-ru   c                   ^d^.R^..R3^^.^^..R3^^.^
^
..R	3^^.^^..R
3^^.^^..R3^
^.^
^..R3^
^.^
^ ..R3^^ .^^..\         P                  R33^ ^.^^..R3^^.^ ^..\         P                  R33^ ^.^^..R3.pV F`  w  r#\        P                  ! \         P                  ! V4      4      p\         P
                  P                  V^,          V^,          ^RR7       Kb  	  R# )r  r  gaa?gaa?T)decimalverboseN)gAn&	?g}K
T?)g	7?g?)g!@j)A?)gU!a?gKY?)g7X?g~ltP?)gFu|?r  )r   gQN^?)r   r   )r   infr`   r  r^   testingr   )rh   tablisttableres_rr   s   &    rr   test_preciseTestFisherExact.test_precise:  s^    Ahq	"$TU!fq!fOP!fr2h!QR"gBx "RS"gBx "RS1gAw!QR1gAw!QR!fq!f(>?@!fq!fOP!fq!f(>?@!fq!fOP $LE$$RZZ%67CJJ**3q658R37 + 9 $ru   c                ,   ^^%.^l^..p\         P                  ! V4      p\        V^,          R4       ^^ .^ ^f..p\         P                  ! V4      p\        V^,          R4       ^^^0.RR..p\         P                  ! V4      p\        V^,          R4       R# )r   gmt?g,:i  i\B  gNQ8Nr`   r  r
   rh   rk   r   s   &  rr   test_gh4130TestFisherExact.test_gh4130R  s     WsCj!  #A 12 !Wq#h  #A 56 "Xe}%  #A 56ru   c                l    R R.RR..p\         P                  ! V4      p\        V^,          ^ RR7       R# )iiX iV iW gѧRrF  Nr(  r)  s   &  rr   test_gh9231TestFisherExact.test_gh9231h  s8     w'7!34  #A/ru   c                    . ROp\        V. RO4       F5  w  r#\        P                  ! RR.RV..4      ^,          p\        WB^R7       K7  	  \        P                  ! RR.RR..4      ^,          p\        VR	^R7       R
# )?=i(E  i  i)  r  iPF  i8 r  i_ g^K=?N)r0  gNuP=gs=)K   L   M   )zipr`   r  r	   )rh   pvalsr  numr   s   &    rr   test_large_numbers"TestFisherExact.test_large_numberso  sx     1UL1ID$$uclT3K%@A!DCq9 2   5%.5%.!AB1ECQ7ru   c                    R p\         P                  ! \        VR7      ;_uu_ 4        \        P                  ! \
        P                  ! ^4      4       RRR4       R#   + '       g   i     R# ; i)z+The input `table` must have two dimensions.r   N)r   r   r   r`   r  r   r   rh   r  s   & rr   test_raisesTestFisherExact.test_raisesz  s=    ?]]:W55ryy|, 6555s   +AA-	c                    ^ ^ .^^
..^^
.^ ^ ..^ ^.^ ^
..^^ .^
^ ..3pV FA  p\         P                  ! V4      w  r4\        VR4       \        V\        P                  4       KC  	  R# r   )r`   r  r   r   rg   )rh   tablesr#  	oddsratior  s   &    rr   test_row_or_col_zero$TestFisherExact.test_row_or_col_zero  s    q6Ar7#r7QF#q6Ar7#q6B7#% E#007OIs#BFF+ ru   c                   ^^.^^..^^.^R..^^.^R..^R.^R..^ ^.^^ ..^^.^^..^^ .^^..^ ^.^^..^^ .^^..3	pRR.RR.RR	.R
R.RR.RR.RR.RR.RR.3	p\        W4       Fq  w  r4. pVP                  \        P                  ! VRR7      ^,          4       VP                  \        P                  ! VRR7      ^,          4       \	        WT^ RR7       Ks  	  R# )rJ   i,  i  i   i  gI[?gq}!?r   g2\&gQp6g:O_?g a?皙?ffffff??333333?r  r  r  r  )rG  r  NUUUUUU?rW  )r4  appendr`   r  r
   )rh   r>  r5  r#  r  r   s   &     rr   test_less_greater!TestFisherExact.test_less_greater  sL    VaV1X3x "X4y!3Z#s$ VaVVaVVaVVaVVaV  "#56)*()	"#J#J#J3K$K v-KECJJu))%VDQGHJJu))%YGJKCAD9	 .ru   c                B    \         P                  ! ^^.^	R..4      w  rR# )r\   i"N)r`   r  )rh   oddsr?  s   &  rr   test_gh3014TestFisherExact.test_gh3014  s%     ))Aq6Ax=*ABfru   r  c                    \         P                  ! R R.RR..4      p\        P                  ! W!R7      p\	        VP
                  VP                  3V4       R# )r  r  r  r  r  N)r   r   r`   r  r   r@  r?  )rh   r  r#  r   s   &&  rr   test_resultTestFisherExact.test_result  sD    5%.5%.9:  @cmmSZZ0#6ru   c                   \         P                  P                  R 4      p\         P                  ! ^^.^^..4      pRp\        P
                  ! \        VR7      ;_uu_ 4        \        P                  ! VR7      p\        P                  ! W$RR7       RRR4       Rp\        P
                  ! \        VR7      ;_uu_ 4        \        P                  ! VR7      p\        P                  ! W$R7       RRR4       R	p\        P
                  ! \        VR7      ;_uu_ 4        \        P                  ! \        P                  P                  R
7      p\        P                  ! W$R7       RRR4       Rp\        P
                  ! \        VR7      ;_uu_ 4        \        P                  ! \         P                  ! R4      4       RRR4       \         P                  P!                  \        P                  ! . RO.4      R4       \         P                  P!                  \        P                  ! ^.^.^..4      R4       \         P                  P!                  \        P                  ! \         P                  ! R4      4      R4       R#   + '       g   i     EL; i  + '       g   i     EL; i  + '       g   i     ELM; i  + '       g   i     EL; i)   9o8o z4`alternative` must be the default \(None\) unless...r   r  r  r  r  Nz6...not recognized; if provided, `method` must be an...r  z3If the `method` argument of `fisher_exact` is an...r  z2`table` must have at least one row and one column.r~   r\   r  r  rJ   r   )r   r  r  r^   r   r   r   r`   r  r  r  r  normr  zerosr!  r   )rh   r  r#  r  r  s   &    rr   $test_input_validation_edge_cases_rxc4TestFisherExact.test_input_validation_edge_cases_rxc  s   ii##$78

QFQF+,I]]:W55,,5FuH 6 K]]:W55**s3Fu4 6 H]]:W55++

?Fu4 6 G]]:W55rxx/0 6
 	

 2 2I; ?H


 2 2QC!qc? CVL


 2 2288F3C DfM- 655
 655
 655
 655s1   "0J  /JAJ(+J< J	J%	(J9	<K	c                   \         P                  P                  R 4      p\         P                  ! ^^.^^..4      p\        P
                  ! V4      pVP                  p\        P                  ! VP                  ^R7      VP                  ^ R7      4      P                  V4      p\        P                  ! VR7      p\        P
                  ! W&R7      p\        VP                  VRR7       \        VP                  V4       \        P                  ! VR7      p\        P
                  ! W&R7      p\        VP                  VP                  RR7       \        VP                  V4       R# )rS  rC   r  r  g{Gzd?rF  N)r   r  r  r^   r`   r  r?  random_tablesumpmfr  r
   r   r@  r  )rh   r  r#  r   
ref_pvalueref_statr  r   s   &       rr   test_resampling_2x2#TestFisherExact.test_resampling_2x2  s    ii##$78

QFQF+,  'ZZ
%%eiiQi&79JKOOPUV''C0  6

JV<S]]H-((S1  6

CJJV<S]]H-ru   c                   . R
O. RO. RO. RO. RO.p\         P                  ! V4      pRp\         P                  P                  R4      p\        P
                  ! VR7      p\        P                  ! WR7      p\        VP                  VRR7       \        P                  ! VRR7      p\        P                  ! WR7      p\        VP                  VRR7       R	# )r\   gk$?l   |Kr  r  gMb@?rF  i )r  n_resamplesN)r\   rJ   rJ   r\   r\   r~   r\   )rJ   r~   r~   rJ   r   r~   r~   )r~   r\   r\   r\   rJ   rO   r   )r\   r\   rJ   r~   r~   r~   r\   )r~   r\   r\   r\   r\   r~   r~   )
r   r^   r  r  r`   r  r  r
   r?  r  )rh   r#  r_  r  r  r   s   &     rr   test_resampling_rxc#TestFisherExact.test_resampling_rxc  s     '&&&&	(
 

5!%
ii##M2((S1  6

JT:''CUC  6

JT:ru   c           
        \         P                  P                  R 4      p\        P                  ! VR7      p\        ^^4       F  p\        ^^4       F  p\        ^^4       F  p\        ^^4       Fl  p\         P                  ! W4.WV..4      p\        P                  ! V4      p\        P                  ! WrR7      p	\        V	P                  VP                  RR7       Kn  	  K  	  K  	  K  	  R# )rS  r  r  +=rF  N)
r   r  r  r`   r  r  r^   r  r
   r?  )
rh   r  r  rL  rX  cdr#  r   r   s
   &         rr   test_resampling_exact_2x2)TestFisherExact.test_resampling_exact_2x2  s     ii##$78((S1q!A1a[q!A"1a[ "

QFQF+; <#007#00F'

CJJUK	 ) % ! ru   r   Nr  r  r  )r   r   r   r   r8  rY  r%  r*  r-  r   r   r  r7  r;  r@  rI  rM  r   rP  rY  r  ra  re  xslowrk  r   r   r   s   @rr   r  r    s    .6907,0 [[8 8-, :DC
 [[],LM7 N7
N> [[2[[.  ." [[2[[;  ;: [[L Lru   r  c                   >  a  ] tR tRt o RtR tR tR tR tR t	R t
R	 tR
 t]P                  P                  R^ ^.4      R 4       tR tR tR tR tR tR tR tR tR tR tR tR tR tR tR tR t R t!R t"R t#R  t$R! t%R" t&R# t'R$ t(R% t)R& t*R' t+R(t,V t-R)# )*TestCorrSpearmanri  r   c                N    \         P                  P                  R 4      V n        R# )iGNr   r  r  r  rh   s   &rr   setup_methodTestCorrSpearmanr.setup_method!  s    99((3ru   c                    \         P                  ! R R4      p\        \        P                  ! V4      P                  4       4       R# )r   r   N)r`   	spearmanrr   r   isnanr  )rh   rl   s   & rr   test_scalarTestCorrSpearmanr.test_scalar$  s)    OOB#!"ru   c                    \        \        \        P                  . RO^^	.4       \        \        \        P                  . RO^4       R# r\   N)r\   rJ   r\   )r   r   r`   rw  rs  s   &rr   test_uneven_lengths%TestCorrSpearmanr.test_uneven_lengths(  s*    j%//9q!fEj%//9a@ru   c                   V P                   P                  R4      pV P                   P                  R4      p\        P                  ! W4      P                  P
                  R8X  g   Q h\        P                  ! VP                  VP                  ^R7      P                  P
                  R8X  g   Q h\        \        \        P                  W^R7       \        \        \        P                  VP                  VP                  4       R# )r   rC   Nr   r   r   rJ   r|   r|   )
r  standard_normalr`   rw  r@  r  r   r?  r   r   rh   rk   rl   s   &  rr   test_uneven_2d_shapes'TestCorrSpearmanr.test_uneven_2d_shapes,  s    HH$$V,HH$$V,q$..44>>>qssACCa077==GGGj%//1a@j%//133<ru   c           	        V P                   P                  R4      p\        \        \        P
                  V4       \        \        \        P
                  W4       \        \        \        P
                  VRR4       \        \        P
                  ! WRR7      \        P
                  ! VP                  4       VP                  4       ^ R7      4       R# )r   NrC   )r   r   rJ   )r  r  r   r   r`   rw  r
   flattenrh   rk   s   & rr   test_ndim_too_high$TestCorrSpearmanr.test_ndim_too_high6  s|    HH$$Y/j%//15j%//18j%//1dDA48		QYY[qI	Kru   c                   \         P                  ! R 4      p\         P                  V^	&   \        \        P
                  ! W4      \         P                  \         P                  34       \        \        P
                  ! WRR7      R4       \        \        \        P
                  WRR7       \        \        \        P
                  WRR7       R# )r   r   r   r   r   N)r   r   )r   r   rg   r   r`   rw  r   r   r  s   & rr   test_nan_policy!TestCorrSpearmanr.test_nan_policy?  sx    IIcNvv!5??102662662BC5??1FC%	'j%//1GLj%//1HMru   c                   \         P                  P                  R 4      pVP                  R4      p^p\         P                  VRV3&   \         P                  ! W#^R7      p\
        P                  ! VRR7      w  rV\
        P                  ! V4      w  rx\         P                  ! \         P                  ! WS^R7      V^ R7      p\         P                  ! \         P                  ! Wc^R7      V^ R7      p\        WWRR7       \        WhRR7       R# )	l   LG r   rC   r   r   rh  rF  Nr|   rT   )r   r  r  rg   deleter`   rw  r
   )	rh   r  rk   krl   corxpxcorypys	   &        rr   test_nan_policy_bug_12458+TestCorrSpearmanr.test_nan_policy_bug_12458H  s    ii##J/JJw&&!Q$IIa#??18??1%yy43QQ?YYryyQ/;/U+ru   c                   ^p^
pV P                   P                  W34      p\        P                  VR&   \        P                  VR&   \        P
                  ! V^RR7      w  rE\        V4       UUu. uFM  p\        V4       Uu. uF4  p\        P
                  ! W7R3,          W6R3,          4      P                  NK6  	  upNKO  	  ppp\        WH4       R# u upi u uppi )r|   	propagaterD   r   r   Nr\   r~   )r   rZ   )	r  r  r   rg   r`   rw  r  r@  r
   )	rh   mnrk   corrr?  jr  r   s	   &        rr   test_nan_policy_bug_12411+TestCorrSpearmanr.test_nan_policy_bug_12411U  s    HH$$aV,&&$66%qq[Ia" 1 FK1XNXQ$a41;;XN  	 "" O "s   0C:C<CCrD   c           	        \         P                  ! R4      p\         P                  ! ^ ^
4      VR&   \         P                  ! ^ ^
4      ) VR&   \         P                  ! \         P                  \         P                  \         P                  \         P                  .\         P                  \         P                  \         P                  \         P                  .\         P                  \         P                  RR	.\         P                  \         P                  R	R..4      pV^8X  d   VP
                  p\        P                  ! \        P                  P                  4      ;_uu_ 4        \        P                  ! W!R7      pRRR4       \        XP                  V4       \        VP                  ^ V,          RR7       R#   + '       g   i     LG; i)
rT   r   rC   NgJz5rF  )rT   r   )r   rJ   )r   r   rQ  )r   rX  r   r   rg   r   r   warnsr`   _warnings_errorsrd  rw  r
   r@  r?  )rh   rD   r_   r  r   s   &&   rr   test_constant_input_gh22816-TestCorrSpearmanr.test_constant_input_gh22816`  s   HHW))Ar"$99Q##$xx"&&"&&"&&"&&9&&"&&"&&"&&9&&"&&"c2&&"&&#r24 5 19A\\%00EEFF//!/C Gt,

AdF7 GFs   'F>>G	c                n    \         P                  ! \        \        4      pV^ ,          p\        VR4       R# r   )r`   rw  r_   r	   r   s   &  rr   test_sXXTestCorrSpearmanr.test_sXXq  s%    OOAa aDAc"ru   c                n    \         P                  ! \        \        4      pV^ ,          p\	        VR4       R# r   )r`   rw  r_   r   r	   r   s   &  rr   
test_sXBIGTestCorrSpearmanr.test_sXBIGv  s%    OOAc"aDAc"ru   c                n    \         P                  ! \        \        4      pV^ ,          p\	        VR4       R# r   )r`   rw  r_   r   r	   r   s   &  rr   test_sXLITTLETestCorrSpearmanr.test_sXLITTLE{  s%    OOAf%aDAc"ru   c                n    \         P                  ! \        \        4      pV^ ,          p\	        VR4       R# r   )r`   rw  r_   r   r	   r   s   &  rr   test_sXHUGETestCorrSpearmanr.test_sXHUGE  %    OOAd#aDAc"ru   c                n    \         P                  ! \        \        4      pV^ ,          p\	        VR4       R# r   )r`   rw  r_   r   r	   r   s   &  rr   test_sXTINYTestCorrSpearmanr.test_sXTINY  r  ru   c                n    \         P                  ! \        \        4      pV^ ,          p\	        VR4       R# r   )r`   rw  r_   r  r	   r   s   &  rr   test_sXROUNDTestCorrSpearmanr.test_sXROUND  s%    OOAe$aDAc"ru   c                n    \         P                  ! \        \        4      pV^ ,          p\        VR4       R# r   )r`   rw  r   r	   r   s   &  rr   test_sBIGBIGTestCorrSpearmanr.test_sBIGBIG  s%    OOC$aDAc"ru   c                n    \         P                  ! \        \        4      pV^ ,          p\	        VR4       R# r   )r`   rw  r   r   r	   r   s   &  rr   test_sBIGLITTLE!TestCorrSpearmanr.test_sBIGLITTLE  s%    OOC'aDAc"ru   c                n    \         P                  ! \        \        4      pV^ ,          p\	        VR4       R# r   )r`   rw  r   r   r	   r   s   &  rr   test_sBIGHUGETestCorrSpearmanr.test_sBIGHUGE  %    OOC%aDAc"ru   c                n    \         P                  ! \        \        4      pV^ ,          p\	        VR4       R# r   )r`   rw  r   r   r	   r   s   &  rr   test_sBIGTINYTestCorrSpearmanr.test_sBIGTINY  r  ru   c                n    \         P                  ! \        \        4      pV^ ,          p\	        VR4       R# r   )r`   rw  r   r  r	   r   s   &  rr   test_sBIGROUND TestCorrSpearmanr.test_sBIGROUND  s%    OOC&aDAc"ru   c                n    \         P                  ! \        \        4      pV^ ,          p\        VR4       R# r   )r`   rw  r   r	   r   s   &  rr   test_sLITTLELITTLE$TestCorrSpearmanr.test_sLITTLELITTLE  s%    OOF6*aDAc"ru   c                n    \         P                  ! \        \        4      pV^ ,          p\	        VR4       R# r   )r`   rw  r   r   r	   r   s   &  rr   test_sLITTLEHUGE"TestCorrSpearmanr.test_sLITTLEHUGE  %    OOF4(aDAc"ru   c                n    \         P                  ! \        \        4      pV^ ,          p\	        VR4       R# r   )r`   rw  r   r   r	   r   s   &  rr   test_sLITTLETINY"TestCorrSpearmanr.test_sLITTLETINY  r  ru   c                n    \         P                  ! \        \        4      pV^ ,          p\	        VR4       R# r   )r`   rw  r   r  r	   r   s   &  rr   test_sLITTLEROUND#TestCorrSpearmanr.test_sLITTLEROUND  s%    OOF5)aDAc"ru   c                n    \         P                  ! \        \        4      pV^ ,          p\        VR4       R# r   )r`   rw  r   r	   r   s   &  rr   test_sHUGEHUGE TestCorrSpearmanr.test_sHUGEHUGE  %    OOD&aDAc"ru   c                n    \         P                  ! \        \        4      pV^ ,          p\	        VR4       R# r   )r`   rw  r   r   r	   r   s   &  rr   test_sHUGETINY TestCorrSpearmanr.test_sHUGETINY  r  ru   c                n    \         P                  ! \        \        4      pV^ ,          p\	        VR4       R# r   )r`   rw  r   r  r	   r   s   &  rr   test_sHUGEROUND!TestCorrSpearmanr.test_sHUGEROUND  %    OOD'aDAc"ru   c                n    \         P                  ! \        \        4      pV^ ,          p\        VR4       R# r   )r`   rw  r   r	   r   s   &  rr   test_sTINYTINY TestCorrSpearmanr.test_sTINYTINY  r  ru   c                n    \         P                  ! \        \        4      pV^ ,          p\	        VR4       R# r   )r`   rw  r   r  r	   r   s   &  rr   test_sTINYROUND!TestCorrSpearmanr.test_sTINYROUND  r  ru   c                n    \         P                  ! \        \        4      pV^ ,          p\        VR4       R# r   )r`   rw  r  r	   r   s   &  rr   test_sROUNDROUND"TestCorrSpearmanr.test_sROUNDROUND  s%    OOE%(aDAc"ru   c                    \         P                  ! \        \        4      pRp\        W4       \	        VP
                  VP                  4       R# )r<  Nr>  )r`   rw  r_   r   r   r<  r@  )rh   r   rA  s   &  rr    test_spearmanr_result_attributes2TestCorrSpearmanr.test_spearmanr_result_attributes  s1    ooa#.
C,S__cmm4ru   c                    . ROp. ROp\         P                  ! W4      p\         P                  ! \        P                  ! W.4      P                  4      p\        W44       R# )r\   N)r\   rJ   r   r   r|   r   )r\   rJ   r   r   r   r|   )r`   rw  r   r^   r   r
   rh   x1x2res1r  s   &    rr   test_1d_vs_2dTestCorrSpearmanr.test_1d_vs_2d  sA    r&rzz2(3556#ru   c                $   R F  p^\         P                  ^^^^.p^^^^^\         P                  .p\        P                  ! W#VR7      p\        P                  ! \         P                  ! W#.4      P
                  VR7      p\        WE4       K  	  R# )r  r   N)r  r   )r   rg   r`   rw  r^   r   r
   )rh   r   r  r  r  r  s   &     rr   test_1d_vs_2d_nans$TestCorrSpearmanr.test_1d_vs_2d_nans  sq    /JRVVQ1a(BQ1a(B??2jAD??2::rh#7#9#9jQDD' 0ru   c                   \         P                  ! ^4      pV) p\         P                  ! . RO4      p\         P                  ! WV.4      P                  p\
        P                  ! V4      p\         P                  ! . RO. RO. RO.4      p\         P                  ! R\        R7      pRVR	&   RVR
&   \        VP                  V4       \        VP                  V4       R# )r   r;   g+Hs]s?N)r~   r\   rJ   r   r|   r   )r\   rZ   3+?)rZ   r\   3+)r  r  r\   r   r   )rJ   r  )r  rJ   )r   r   r   r^   r   r`   rw  rX  floatr
   r@  r?  )rh   r  r  x3rk   actualexpected_correxpected_pvalues   &       rr   
test_3colsTestCorrSpearmanr.test_3cols  s    YYq\SXX()JJ|$&&#"5"6">"@ A ((67"/"/((-87ru   c           
        \         P                  ! \         P                  R RRRRR.R\         P                  RRRRR.. RO.4      P                  p\         P                  ! \         P                  \         P                  \         P                  .\         P                  \         P                  \         P                  .\         P                  \         P                  R	..4      p\	        \
        P                  ! VR
R7      P                  V4       \
        P                  ! VRR7      P                  p\	        V^ ,          ^,          V^ ,          ^,          V^,          ^,          3RRR7       R# )r   r   r=   ffffff@r   ffffff"@ffffff@g333333@皙@r   r  r   r   ư>r  N)r6   r   ffffff@ffffff@r   r  ffffff@)gacJC?g%ln?g_0nYt!޿)r   r   rg   r   r
   r`   rw  r@  )rh   rk   r  r   s   &   rr   test_gh_9103TestCorrSpearmanr.test_gh_9103   s   HHrvvsCc3<BFFCc3<9; <<=A 	
 xx"&&"&&"&&1&&"&&"&&1&&"&&"-/ 0 	kBLL	 ooaF3==QCF1Is1vay9:	Gru   c                   ^dp\         P                  P                  R4      pVP                  V4      pVP                  V4      R8  pVR8  p\         P                  ! V4      p\
        P                  ! WVRR7      P                  p\         P                  Wd&   \
        P                  ! WVRR7      P                  pVP                  \         P                  4      p\
        P                  ! WVRR7      P                  p	. ROp
\        WxV	.V
4       R# )r  iH rD  r6   r   r   N)gonj?)Wn%?r  )r   r  RandomStaterandr   r`   rw  r@  rg   astypeint32r
   )rh   r  r  rk   r  rL  rX  r  r  res3expecteds   &          rr   test_gh_8111TestCorrSpearmanr.test_gh_8111  s    ii##F+HHQKHHQK# VHHQKq7AA vvq7AA HHRXXq7AA:T*H5ru   r  N).r   r   r   r   r8  rt  ry  r}  r  r  r  r  r  r   r   r   r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r   r   r   s   @rr   rp  rp    s     4#A=KN,	# [[VaV,8 -8 #
#
#
#
#
#
#
#
#
#
#
#
#
#
#
#
#
#
#
#
#
5$(8 G6 6ru   rp  c                      a  ] tR tRt o RtR tR tR tR tR t	R t
R	 tR
 tR tR tR tR tR tR tR t]P*                  P-                  RR4      R 4       tRtV tR# )TestCorrSpearmanr2i'  z-Some further tests of the spearmanr function.c                    . ROp. ROpRp\         P                  ! W4      p\        V^ ,          V^ ,          4       \        V^,          V^,          4       R# )r\   Nr\   rJ   r   r   r|   r|   r   rO   rK   rO   gh
C?gR?)r`   rw  r	   rh   r  r  r  r   s   &    rr   test_spearmanr_vs_r&TestCorrSpearmanr2.test_spearmanr_vs_r*  sE     >oob%CFHQK0CFHQK0ru   c                    \        \        P                  ! . . 4      \        P                  \        P                  34       R # N)r   r`   rw  r   rg   rs  s   &rr   test_empty_arrays$TestCorrSpearmanr2.test_empty_arrays4  s$    U__R,rvvrvv.>?ru   c           	        \         P                  P                  R 4      p\         P                  ! VP	                  ^^RR7      VP	                  ^^RR7      .4      pRR.RR..p\         P
                  ! \         P                  P                  V4      V4      pRp\        P                  ! V^ ,          V^,          4      p\        V^ ,          V^ ,          4       \        V^,          V^,          4       R# )z    locscaler  r   rF  N)gCzYW?ge*=)r   r  r  r   r  dotlinalgcholeskyr`   rw  r	   )rh   r  rk   r  r  r   s   &     rr   test_normal_draws$TestCorrSpearmanr2.test_normal_draws7  s    ii##D)HHcjjQacj:jjQacj:< =c
c
FF299%%d+Q/?ooadAaD)CFHQK0CFHQK0ru   c                ^    \        \        P                  ! . RO. RO4      ^ ,          R4       R# )r\   r   Nr\   r\   rJ   )r	   r`   rw  rs  s   &rr   test_corr_1TestCorrSpearmanr2.test_corr_1C  s    EOOIyA!DcJru   c                   \         P                  ! R 4      p\         P                  V^	&   \        \        P
                  ! W4      \         P                  \         P                  34       \        \        P
                  ! WRR7      R4       \        \        \        P
                  WRR7       \        \        \        P
                  WRR7       R# )r   r   r   r   r   N)r   r~   )	r   r   rg   r   r`   rw  r
   r   r   r  s   & rr   test_nan_policies$TestCorrSpearmanr2.test_nan_policiesF  sx    IIcNvv!5??102662662BC@ 	"j%//1GLj%//1HMru   c                    \         P                  ! R 4      p\         P                  ! R4      p\        \        \        P
                  W4       R# )r         4@N)r   r   r   r   r`   rw  r  s   &  rr   test_unequal_lengths'TestCorrSpearmanr2.test_unequal_lengthsO  s,    IIcNIIcNj%//18ru   c                    . ROp^^^\         P                  .p\        P                  ! WRR7      p\        P                  ! VR,          VR,          RR7      p\	        W44       R# )r\   r   r   :Nr   NNr\   rJ   r   r   )r   rg   r`   rw  r   r  s   &    rr   test_omit_paired_value)TestCorrSpearmanr2.test_omit_paired_valueT  sK    Arvvr&9r"vr"v&AT ru   c                   \        \        R 4      4      p\        \        R 4      4      pV^	,          V^ ,          uV^ &   V^	&   VR,          V^
,          uV^
&   VR&   VR,          VR,          uVR&   VR&   VP                  \        P                  4       VP                  R4       \        \        P                  ! WRR7      ^ ,          R4       R# )	  i  i  i  r   r   r   gV-?N)listr  rH  r   rg   r   r`   rw  r  s   &  rr   #test_gh_issue_6061_windows_overflow6TestCorrSpearmanr2.test_gh_issue_6061_windows_overflow[  s    ttqT1Q4
!ad#""qvD'1S6#$ 	
	EOOAVDQGOru   c                   R p\         P                  ! \        P                  VR7      ;_uu_ 4        \        P                  ! . RO. RO4      w  r#\        V\        P                  4       \        V\        P                  4       \        P                  ! . RO. RO4      w  r#\        V\        P                  4       \        V\        P                  4       \        P                  ! . RO. RO4      w  r#\        V\        P                  4       \        V\        P                  4       RRR4       R#   + '       g   i     R# ; i)r]  r   NrJ   rJ   rJ   rJ   r~   rJ   )r   r  r`   rd  rw  r   r   rg   )rh   warn_msgr   rf  s   &   rr   	test_tie0TestCorrSpearmanr2.test_tie0h  s    /\\%44HEE??9i8DABFF#BFF#??9i8DABFF#BFF#??9i8DABFF#BFF# FEEEs   C4D//E 	c                    . ROp. ROp. ROp. ROp\         P                  ! W4      p\         P                  ! W44      p\        WV4       R# )r   Nr  )r   r   r   r   )r   ry   ry   r   )r`   rw  r   r   )rh   rk   rl   xryrsrprs   &      rr   	test_tie1TestCorrSpearmanr2.test_tie1v  s<      !! __Q"^^B#B#ru   c                    . ROp. ROp^^R^\         P                  .p^^R^\         P                  .p\        P                  ! W4      p\        P                  ! W4RR7      p\	        WV4       R# )r\   ry   r   r   N)r\   rJ   ry   rJ   )r\   r   ry   r   )r   rg   r`   rw  r   )rh   r  rm   r  rn   sr1sr2s   &      rr   	test_tie2TestCorrSpearmanr2.test_tie2  s^     CBFF#CBFF# oob%oob8C%ru   c                   \         P                  ! . RO. RO.4      p\         P                  ! . RO. RO.4      p\         P                  ! . RO. RO.4      pRp\        P                  ! \        P
                  VR7      ;_uu_ 4        \        P                  ! V^R7      w  rV\        V\         P                  4       \        V\         P                  4       \        P                  ! V^R7      w  rV\        V\         P                  4       \        V\         P                  4       \        P                  ! V^R7      w  rV\        V\         P                  4       \        V\         P                  4       RRR4       R#   + '       g   i     R# ; i)r\   r]  r   rC   Nr\   r\   r\   r\   r=  	r   r   r   r  r`   rd  rw  r   rg   )rh   z1z2z3rH  r   rf  s   &      rr   test_ties_axis_1#TestCorrSpearmanr2.test_ties_axis_1  s    XX|\23XX|\23XX|\23/\\%44HEE??2A.DABFF#BFF#??2A.DABFF#BFF#??2A.DABFF#BFF# FEEEs   C+E::F	c                   \         P                  ! . RO4      p\         P                  ! . RO4      pRp\        P                  ! \        P
                  VR7      ;_uu_ 4        \        P                  ! W4      w  rE\        V\         P                  4       \        V\         P                  4       RRR4       R#   + '       g   i     R# ; i)r   r]  r   N)
r   r   r   r   r   r   r   r   r   r   
r~   gN}	?r~   r~   g	/`?gY%uX-H?gC1tq1?r~   r~   g*I?rY  )rh   rk   rl   rH  r   rf  s   &     rr   test_gh_11111 TestCorrSpearmanr2.test_gh_11111  s~    HHGHHH - . 0\\%44HEE??1(DABFF#BFF# FEEEs   !AB88C		c                    \         P                  ! . RO4      p\         P                  ! . RO4      p\        \        \        P
                  W^R7       R# )r   rC   N)
r   r   r   r   r   r   r   r   r   r   r`  )r   r   r   r   r`   rw  r  s   &  rr   test_index_error#TestCorrSpearmanr2.test_index_error  s8    HHGHHH - . 	j%//1a@ru   c                &   . ROp. R	OpR
p\         P                  ! WRR7      p\        V^ ,          V^ ,          4       \        V^,          ^V^,          ^,          ,
          4       \         P                  ! WRR7      p\        V^ ,          V^ ,          4       \        V^,          V^,          ^,          4       \        P                  ! \
        RR7      ;_uu_ 4        \         P                  ! WRR7       RRR4       R#   + '       g   i     R# ; i)r\   r  r  r  `alternative` must be 'less'...r   	ekki-ekkiNr  r  r  )r`   rw  r	   r   r   r   r  s   &    rr   test_alternative#TestCorrSpearmanr2.test_alternative  s      ? oob&9CFHQK0CFA!q$9: oob)<CFHQK0CFHQK!O4]]:-NOOOOB< POOOs   C??D	r  c           	        . R
Op. ROpV\         P                  .,           pV\         P                  .,           p\        \        P                  ! WE4      \         P                  \         P                  34       \        P                  ! WERVR7      p\        P                  ! W#VR7      p\        Wg4       Rp\        P                  ! \        VR7      ;_uu_ 4        \        P                  ! WERVR7       RRR4       Rp\        P                  ! \        VR7      ;_uu_ 4        \        P                  ! WER	VR7       RRR4       R#   + '       g   i     LZ; i  + '       g   i     R# ; i)r\   r   r   r  r  r   r   r   Nr   rh  r  r  )	r   rg   r   r`   rw  r
   r   r   r   )	rh   r  r  r  x1nanx2nan
res_actualres_expectedr  s	   &&       rr   test_alternative_nan_policy.TestCorrSpearmanr2.test_alternative_nan_policy  s     bffXbffX 	5??58266266:JK __Uf1<>
r;G
1 2]]:W55OOEW(35 6
 1]]:W55OOE[(35 65 65 655s   E!EE	E)	r   Nrm  )r   r   r   r   r8  r   r$  r/  r3  r6  r:  r>  rC  rI  rP  rU  r]  ra  rd  ri  r   r   r   rq  r   r   r   s   @rr   r  r  '  s     71@
1KN9
!P$$&$ 	$A=. [[],LM5 N5ru   r  c                     a  ] tR tRt o ]P
                  P                  R. RO. R ORR3. R!O. R"ORR3. R#O. R$OR%R3. R&O. R'ORR3. R(O. R(ORR3. R(O. R)ORR	3. R(O. R*OR
R3. R(O. R+OR,R3. R(O. R-OR.R	3. R(O. R/OR0R3.
4      ]P
                  P                  RRR.4      R 4       4       tR tR t	]P
                  P                  RRR.4      R 4       t
]P
                  P                  R. R1O. R1O3. R2O. R1O3. R1O. R2O3.4      ]P
                  P                  RRR.4      R 4       4       t]P
                  P                  R. R3O4      R 4       tR tR t]P
                  P                  RR7      R 4       tR tRtV tR# )4TestKendallTaui  zx, y, statistic, pvaluer   r   gmK?gJkaa?gxO~>g?>?gaV>g}'}'?go&5>variantrX  ri  c                    \         P                  ! VP                  V4      VP                  V4      VR 7      p\        V^ ,          VP                  V4      4       \        V^,          VP                  V4      4       R# ru  N)r`   
kendalltaur^   r2   )rh   rk   rl   r@  r?  ru  ri   r   s   &&&&&&& rr   test_kendalltau_no_ties&TestKendallTau.test_kendalltau_no_ties  sV    . rzz!}bjjmWMA

9 56A

6 23ru   c                `   VP                  . R	O4      pVP                  . R
O4      pRp\        \        P                  ! W#RR7      ^ ,          VP                  V4      4       Rp\        \        P                  ! W#RR7      ^ ,          VP                  V4      4       VP                  . RO4      pVP                  . RO4      pRp\        P                  ! WV4      p\	        V^ ,          VP                  V^ ,          4      4       \	        V^,          VP                  V^,          4      4       VP                  . RO4      p\        \        P                  ! W"RR7      ^ ,          VP                  R4      4       \        \        P                  ! W"RR7      ^ ,          VP                  R4      4       R# )r\   gok|?rX  rx  gffffff?ri  r   g#q?N
r\   rJ   rJ   r   r   r   r   rK   rN   rN   )
r\   rJ   r   r   r   r   rK   rK   rK   rT   rV   rJ   r\   rV   rJ   r\   r   rO   r\   r~   )g}+޿g`=i?r2  )r^   r2   r`   ry  r	   )rh   ri   rk   rl   r  r  r  r   s   &&      rr   test_kendalltau_ties#TestKendallTau.test_kendalltau_ties  s9    JJ56JJ67((s;A>

8@TU((s;A>

8@TU
 ZZ)*ZZ(>r&CFBJJx{$;<CFBJJx{$;< JJy!((s;A>

3P((s;A>

:@VWru   c                   VP                  . RO4      pVP                  . RO4      pRp\        P                  ! \        VR7      ;_uu_ 4        \        P
                  ! W#RR7       RRR4       Rp\        P                  ! \        VR7      ;_uu_ 4        \        P
                  ! W#RR7       RRR4       Rp\        P                  ! \        VR7      ;_uu_ 4        \        P
                  ! W#R	R
7       RRR4       \        \        RVR7      ;_uu_ 4        \        P
                  ! VP                  . 4      VP                  . 4      4      pRRR4       \        XP                  VP                  VP                  4      4       \        VP                  VP                  VP                  4      4       \        P                  ! \        RR7      ;_uu_ 4        \        P
                  ! VP                  ^ .4      VP                  ^ .4      4      w  rgRRR4       \        XVP                  \        P                  4      4       \        XVP                  \        P                  4      4       \        P                  ! R4      p\        P                  ! R4      pRp\        P                  ! \        VR7      ;_uu_ 4        \        P
                  ! W#4       RRR4       R#   + '       g   i     EL; i  + '       g   i     ELM; i  + '       g   i     EL; i  + '       g   i     EL; i  + '       g   i     EL&; i  + '       g   i     R# ; i)r\   z(Ties found; exact method cannot be used.r   r  r  Nz$Unknown `method` 'banana' specified.bananaz"Unknown `variant` 'rms' specified.rmsrx  One or more sample...r^  r   r9  /Array shapes are incompatible for broadcasting.r}  
r\   rJ   rJ   r   r   r   rK   rK   rK   rT   )r^   r   r   r   r`   ry  r"   r   r3   r@  rg   r?  r  r   r   )rh   ri   rk   rl   r  r   taup_values   &&      rr    test_input_validation_edge_cases/TestKendallTau.test_input_validation_edge_cases  s   JJ56JJ67 =]]:W55Q'2 6 9]]:W55Q(3 6 7]]:W55Q51 6 +3JrRR""2::b>2::b>BC Srzz"&&'9:

BJJrvv$67 \\,4KLL ++BJJsORZZ_MLC MRZZ/0BFF!34 IIcNIIcNC]]:W55Q" 65; 655
 655
 655 SRR MLL 655sH   KK1L6L:L-<MK.	1L	L	L*	-L>	M	c                    VP                  . RO4      pVP                  . RO4      pRp\        P                  ! W4VR7      p\        WeVR7       \	        VP
                  VP                  4       R# )r\   rx  r=  Nr}  r  r>  )r^   r`   ry  r   r3   r<  r@  )rh   ru  ri   r  r  rA  r   s   &&&    rr   test_attributesTestKendallTau.test_attributesE  sS    ZZ67ZZ78.
rw7C37ru   x, yc                4   VP                  V4      VP                  V4      r!\        P                  ! WVR 7      p\        VP                  VP                  VP
                  4      4       \        VP                  VP                  VP
                  4      4       R# rw  )r^   r`   ry  r3   r@  rg   r?  )rh   rk   rl   ru  ri   r   s   &&&&& rr   test_degenerate_input$TestKendallTau.test_degenerate_inputO  sc     zz!}bjjm1qW5rzz"&&'9:

BJJrvv$67ru   r<   Nc           	        Vf   \        V4      M
\        W!4      p\        P                  P	                  R4      p\        P
                  ! VP                  ^^RR7      VP                  ^^RR7      .4      pRR.RR..p\        P                  ! \        P                  P                  V4      V4      pVP                  V^ ,          VR7      pVP                  V^,          VR7      pRp\        P                  ! Wg4      p	\        V	^ ,          VP                  V^ ,          VR7      4       \        V	^,          VP                  V^,          VR7      4       R # )Nr'  r(  r)  r   rF  r;   )g\zf?g^n)=)r&   rl  r   r  r  r   r  r,  r-  r.  r^   r`   ry  r2   )
rh   r<   ri   r  rk   r  r  r  r  r   s
   &&&       rr   test_large_arrays TestKendallTau.test_large_arrays\  s    ). $72;Mii##D)HHcjjQacj:jjQacj:< =c
c
FF299%%d+Q/ZZ!EZ*ZZ!EZ*@r&A

8A;e
 DEA

8A;e
 DEru   c                z   \         P                  ! R 4      p\         P                  V^	&   VP                  VP	                  4       4      p\
        P                  ! W"4      p\        VP                  VP                  VP                  4      4       \        VP                  VP                  VP                  4      4       \
        P                  ! W"RR7      p\        VP                  VP                  R4      4       \        VP                  VP                  R4      RR7       \
        P                  ! W"RRR7      p\        VP                  VP                  R4      4       \        VP                  VP                  R	4      RR7       R
p\        P                  ! \        VR7      ;_uu_ 4        \
        P                  ! W"RR7       RRR4       Rp\        P                  ! \        VR7      ;_uu_ 4        \
        P                  ! W"RR7       RRR4       R#   + '       g   i     LY; i  + '       g   i     R# ; i)r   r   r   r   g/>r	  r  r  )r   r  g&?r   r   r   Nr   r   )r   r   rg   r^   r   r`   ry  r3   r@  r?  r2   r   r   r   )rh   ri   rk   r   r  s   &&   rr   r  TestKendallTau.test_nan_policym  sk   IIcNvv!JJqxxz"q$rzz"&&'9:

BJJrvv$67q7rzz#7

BJJ|$<5Iq|Lrzz#7

BJJ/E$FUS1]]:W55Qg6 6 1]]:W55Qh7 65	 65 655s   ,H3H)H&	)H:	c                0   VP                  . R	O4      pVP                  \        P                  RRR.4      p\        P                  ! W#RR7      p\        P                  ! VR,          VR,          4      p\        VP                  VP                  RR7       R# )
r   333333@333333@r   r   r\   NNrE  rF  Nr  )r^   r   rg   r`   ry  r2   r@  )rh   ri   rk   rl   r1r2s   &&    rr   "test_kendalltau_nan_2nd_arg_gh61341TestKendallTau.test_kendalltau_nan_2nd_arg_gh6134  sl    JJ'(JJS#./av6aeQrU+bll?ru   zfails in parallelr   c                   ^ RI pVP                  R4       . ROpR	pVP                  VP                  W4R7      4      pVP                  VP                  W4R7      4      p\        P
                  ! WV4      p\        VP                  VP                  R4      4       \        VP                  VP                  R4      RR7       R# )
r~   Ni_ r  g*8_2\S?gd`TR?gMb`?rF  r\   rJ   r   r   r|   r   rO   i@ )	r  seedr^   choicesr`   ry  r2   r@  r?  )rh   ri   r  classes	n_samplesrk   rl   r   s   &&      rr    test_kendalltau_gh18139_overflow/TestKendallTau.test_kendalltau_gh18139_overflow  s    
 	G'	JJv~~g~;<JJv~~g~;<q$rzz2G'HI 	

BJJv$6TBru   c                   ^p\         P                  ! V^,           4      P                  \        4      p\         P                  ! V^,           4      P                  \        4      p\         P                  VR&   VP                  VP                  4       4      VP                  VP                  4       4      rC\        P                  ! W4RRR7      w  rV\        WaP                  R4      4       R# )   r  r   )r  r   r   NrZ   )
r   r   r  r  rg   r^   r   r`   ry  r3   )rh   ri   r  rk   rl   _r  s   &&     rr   !test_kendall_tau_large_nan_policy0TestKendallTau.test_kendall_tau_large_nan_policy  s    IIa!e##E*IIa!e##E*"zz!((*%rzz!((*'=1""1FKjjo.ru   r   )r|   rJ   r\   r   r   r   rO   rK   )r|   rJ   r   r   r\   rK   rO   r   )	r~   r|   rJ   r\   r   r   r   rO   rK   )	r|   rJ   r~   r   r   r\   rK   rO   r   )r|   rJ   r\   r   r   r   rO   )r|   rJ   r   r   r\   rO   r   gJ$I¿)rJ   r\   r   r   r   rO   rK   )rJ   r   r   r\   rK   rO   r   )
r~   r\   rJ   r   r   r|   r   rO   rK   rN   )
r~   rJ   r\   r   r   r|   r   rO   rK   rN   )
r~   rJ   r\   r   r   r   r|   rO   rK   rN   )
rN   rK   rO   r   r|   r   r   rJ   r\   r~   rQ  )
rN   rO   rK   r   r|   r   r   rJ   r\   r~   g?>)
rN   rO   rK   r   r|   r   r   rJ   r\   r~   g}'}'rF  rG  Nrj  r   )r   r   r   r   r   r   r   rz  r  r  r  r  r  r  r  thread_unsafer  r  r   r   r   s   @rr   rt  rt    s    [[6 
"#;S#F	$&A3L	 5~}U	 5~sK	')G	 	"	')G	/	1	')G	/	1	')G	!	#	')G	0	2	')G	0	2%9 * [[Yc
34 4+,4
X4$#L [[Yc
38 48 [[V	I	I	I& 
 [[Yc
38 48 [[W&BCF DF 84@ [[&9:C ;C"/ /ru   rt  c            	       l  a  ] tR tRt o R tRt]P                  ]P                  ]P                  .t. ROt	. ROt
. ROt. ROt. ROt. ROt. ROt. ROtR t]! ]! ]]R.^,          4      4      ]! ]! ]]! ]4      R.^,          4      4      ,           t]P.                  P1                  R]4      R 4       t]! ]! ]]	R.^,          4      4      ]! ]! ]]! ]	4      R.^,          4      4      ,           t]P.                  P1                  R]4      R 4       t]! ]! ]]
R.^,          4      4      ]! ]! ]]! ]
4      R.^,          4      4      ,           t]P.                  P1                  R]4      R	 4       t]! ]! ]]R.^,          4      4      ]! ]! ]]! ]4      R.^,          4      4      ,           t]P.                  P1                  R]4      R
 4       t]! ]! ]]R.^,          4      4      ]! ]! ]]! ]4      R.^,          4      4      ,           t ]P.                  P1                  R] 4      R 4       t!]! ]! ]]R.^,          4      4      ]! ]! ]]! ]4      R.^,          4      4      ,           t"]P.                  P1                  R]"4      R 4       t#]! ]! ]]R.^,          4      4      ]! ]! ]]! ]4      R.^,          4      4      ,           t$]P.                  P1                  R]$4      R 4       t%]! ]! ]]R.^,          4      4      ]! ]! ]]! ]4      R.^,          4      4      ,           t&]P.                  P1                  R]&4      R 4       t']! ]! ]]R.^,          4      4      ]! ]! ]]! ]4      R.^,          4      4      ,           t(]P.                  P1                  R](4      R 4       t)]! ]! ]R.^,          4      4      ]! ]! ]R.^,          4      4      ,           t*]P.                  P1                  R]*4      R 4       t+]P.                  P1                  RR 4      ]P.                  P1                  RR!4      R 4       4       t,Rt-V t.R# )"TestKendallTauAlternativei  c                   . R	Op. R
Op\         P                  ! VP                  V4      VP                  V4      RR7      pV^ ,          ^ 8  g   Q h\         P                  ! VP                  V4      VP                  V4      RR7      p\        V^ ,          VP                  V^ ,          4      4       \	        V^,          VP                  ^V^,          ^,          ,
          4      4       \         P                  ! VP                  V4      VP                  V4      RR7      p\        V^ ,          VP                  V^ ,          4      4       \	        V^,          VP                  V^,          ^,          4      4       VP                  4        \         P                  ! VP                  V4      VP                  V4      RR7      pV^ ,          ^ 8  g   Q h\         P                  ! VP                  V4      VP                  V4      RR7      p\        V^ ,          VP                  V^ ,          4      4       \	        V^,          VP                  ^V^,          ^,          ,
          4      4       \         P                  ! VP                  V4      VP                  V4      RR7      p\        V^ ,          VP                  V^ ,          4      4       \	        V^,          VP                  V^,          ^,          4      4       \        P                  ! \        RR7      ;_uu_ 4        \         P                  ! VP                  V4      VP                  V4      RR7       RRR4       R#   + '       g   i     R# ; i)r\   r  r  r  r  rg  r   rh  Nr  r  )	r`   ry  r^   r3   r2   reverser   r   r   )rh   ri   r  r  r  r   s   &&    rr   &test_kendalltau_alternative_asymptotic@TestKendallTauAlternative.test_kendalltau_alternative_asymptotic  sY     ##BJJrNBJJrN0;={Q rzz"~rzz"~6RA

8A; 78A

1a+@ AB rzz"~rzz"~9UA

8A; 78A

8A;? ;< 	

 ##BJJrNBJJrN0;={Q rzz"~rzz"~9UA

8A; 78A

1a+@ AB rzz"~rzz"~6RA

8A; 78A

8A;? ;<]]:-NOORZZ^RZZ^U POOOs   8MM,	c                6   VP                  V4      VP                  V4      r!V'       d   V) pVR,          p\        P                  ! WRVR7      p\        VP                  VP                  V4      4       \        VP
                  VP                  V4      4       R# )r\   r  rT  NrZ   )r^   r`   ry  r2   r@  r?  )	rh   rk   rl   r  revstat_expected
p_expectedri   r   s	   &&&&&&&& rr   
exact_test$TestKendallTauAlternative.exact_test  so    zz!}bjjm1ARMqGMrzz-'@A

BJJz$:;ru   FTzalternative, p_expected, revc                    ^.^.re\         P                  p\        P                  ! \        RR7      ;_uu_ 4        V P                  WVWWrVR7       RRR4       R#   + '       g   i     R# ; i)r\   r  r   r=  N)r   rg   r   r  r   r  rh   r  r  r  ri   rk   rl   r  s   &&&&&   rr   test_against_R_n1+TestKendallTauAlternative.test_against_R_n1  sN    sQC1\\,4KLLOOA+MRTOU MLLLs   AA+	c           
     B    ^^.^^.reRpV P                  WVWWrVR7       R# )r\   g?r=  Nr  r  s   &&&&&   rr   test_against_R_n2+TestKendallTauAlternative.test_against_R_n2  s*    1v1v1*kbQru   c           
     B    . RO. ROreRpV P                  WVWWrVR7       R# )r\   r   r=  Nr  r  r  s   &&&&&   rr   test_against_R_c0+TestKendallTauAlternative.test_against_R_c0  s"    )1kbQru   c           
     B    . RO. ROreRpV P                  WVWWrVR7       R# )r\   gVUUUUU?r=  Nr=  )r\   rJ   r   r   r  r  s   &&&&&   rr   test_against_R_c1+TestKendallTauAlternative.test_against_R_c1  s"    \1*kbQru   c           
     B    . RO. ROreRpV P                  WVWWrVR7       R# )r\   r   r=  Nr  )r\   r|   r   rJ   r   r  r  s   &&&&&   rr   test_against_R_no_correlation7TestKendallTauAlternative.test_against_R_no_correlation"  s"    1kbQru   c           
     B    . RO. ROreRpV P                  WVWWrVR7       R# )r\   r   r=  Nr\   rJ   r   r   r|   r   rO   rK   )rK   r   r\   r   rJ   r|   r   rO   r  r  s   &&&&&   rr   test_against_R_no_correlationb8TestKendallTauAlternative.test_against_R_no_correlationb,  s#    ')A1kbQru   c           
     D    . ROp. ROpRpV P                  WVWWrVR7       R# )333333F@gqq?r=  N)	r  g33333F@g33333D@gfffffJ@gYF@gF@gYI@gF@gN@)	g@@ry   r=   @r   g@ffffff@r  r  r  s   &&&&&   rr   test_against_R_lt_171/TestKendallTauAlternative.test_against_R_lt_1715  s(     C9*kbQru   c           
         \         P                  P                  ^ 4      pVP                  ^d4      P	                  4       pVP                  ^d4      P	                  4       pRpV P                  WgWWVR7       R# )r~   r=  NgUm*,r   r  r  r  r   r  	rh   r  r  r  ri   r  rk   rl   r  s	   &&&&&    rr   test_against_R_lt_171b0TestKendallTauAlternative.test_against_R_lt_171bB  s[    ii##A&HHSM  "HHSM  ",kbQru   c           
         \         P                  P                  ^ 4      pVP                  ^4      P	                  4       pVP                  ^4      P	                  4       pRpV P                  WgWWVR7       R# )r~   gH4?r=  Nr  r  s	   &&&&&    rr   test_against_R_lt_171c0TestKendallTauAlternative.test_against_R_lt_171cN  s[    ii##A&HHSM  "HHSM  "*kbQru   zalternative, revc                   \         P                  P                  ^ 4      pVP                  VP	                  R4      P                  4       4      pVP                  VP	                  R4      P                  4       4      p\        P                  ! WVRVR7      p\        P                  ! WVRVR7      p\        V^ ,          V^ ,          4       \        V^,          V^,          RR7       R# )r~   i  r  rT  r  r  r  N)
r   r  r  r^   r  r   r`   ry  r3   r2   )	rh   r  r  ri   r  rk   rl   res0r  s	   &&&&     rr   test_gt_171%TestKendallTauAlternative.test_gt_171Y  s    ii##A&JJsxx}++-.JJsxx}++-.W+N\{SQa)Qat4ru   r  r  c           
     8   . R
Op. ROpV\         P                  .,           pV\         P                  .,           pVP                  V4      VP                  V4      rTVP                  V4      VP                  V4      rv\        P                  ! WgWR7      p\        VP                  VP                  VP                  4      4       \        VP                  VP                  VP                  4      4       \        P                  ! WgRWR7      p\        P                  ! WEVVR7      p	\        VP                  V	P                  4       \        VP                  V	P                  4       Rp
\        P                  ! \        V
R7      ;_uu_ 4        \        P                  ! WgRWR7       RRR4       Rp
\        P                  ! \        V
R7      ;_uu_ 4        \        P                  ! WgR	WR7       RRR4       R#   + '       g   i     LZ; i  + '       g   i     R# ; i)r\   rT  r   )r   r  r  r   r   r   Nr   rh  r  )r|   r   rO   rK   rN   )r   rg   r^   r`   ry  r3   r@  r?  r   r   r   )rh   r  r  ri   r  r  rm  rn  ro  rp  r  s   &&&&       rr   r  )TestKendallTauAlternative.test_nan_policyc  s    bffXbffXBBBzz%("**U*;u %%e-3N

,,bjj.@A
))2::bff+=> %%ev-3N
''v4?A
,,l.D.DE
))<+>+>? 2]]:W55Ug$*E 6
 1]]:W55Uk$*E 65 65 655s   	G5H5H	H	r   N)r  r  r  )r   r   r6   )r   RUUUU?ZUUUU?)g~?r  r  )?r   r  )ˮ[?r   r  )g?gSǌ?gSǌ?)g:
?g3
?g<t'.?)gN  A?gss?gYss?)r  r  rm  )/r   r   r   r   r  alternativesr   rg   p_n1p_n2p_c0p_c1p_no_correlationp_no_correlationb
p_n_lt_171p_n_lt_171bp_n_lt_171cr  rB  r4  reversed	case_R_n1r   r   r   r  	case_R_n2r  	case_R_c0r  	case_R_c1r  case_R_no_corrr  case_no_cor_br  case_R_lt_171r  case_R_lt_171br  case_R_lt_171cr  case_gt_171r  r  r   r   r   s   @rr   r  r    s    )Vl 4LFFBFFBFF#DD2D>D>?DJCKGK< c,ugai89L(4.4&(CDEI [[;YGV HV c,ugai89L(4.4&(CDEI [[;YGR HR
 c,ugai89L(4.4&(CDEI [[;YGR HR
 c,ugai89L(4.4&(CDEI [[;YGR HR
 3|-=wqyIJSx8H/I"&+ ,,N [[;^LR MR
 #l,=wqyIJCh7H.I!%q* ++M [[;]KR LR
 #lJ	BCChz.BTF1HMNOM [[;]KR LR 3|[5'!)DESx/D"&+ ,,N [[;^LR MR 3|[5'!)DESx/D"&+ ,,N [[;^LR MR L5'!)45L4&(345K [[/=5 >5 [[X'>?[[],LM!E N @!Eru   r  c                   2   a  ] tR tRt o R tR tR tRtV tR# )TestWeightedTaui  c                   . ROp. ROp\         P                  ! W4      w  r4\        VR4       \        \        P
                  V4       \         P                  ! WRR7      w  r4\        VR4       \        \        P
                  V4       \         P                  ! WR R7      w  r4\        VR4       \        \        P
                  V4       \         P                  ! W4      pRp\        WV4       \        VP                  VP                  4       \         P                  ! WRR7      w  r4\        VR4       \        \        P
                  V4       \         P                  ! W!RR7      w  r4\        VR4       \        \        P
                  V4       \         P                  ! WRRR7      w  r4\        VR4       \        \        P
                  V4       \         P                  ! W!RRR7      w  r4\        VR4       \        \        P
                  V4       \         P                  ! WRR7      w  r4\        VR4       \        \        P
                  V4       \         P                  ! WRR	 R
7      w  r4\        VR4       \        \        P
                  V4       \         P                  ! W!RR R
7      w  r4\        VR4       \        \        P
                  V4       \         P                  ! \        P                  ! V\        P                  R7      V4      w  r4\        VR4       \         P                  ! \        P                  ! V\        P                  R7      V4      w  r4\        VR4       \         P                  ! \        P                  ! V\        P                  R7      \        P                  ! V\        P                  R7      4      w  r4\        VR4       \        P                  ! \        RR7      ;_uu_ 4        \         P                  ! . . 4      w  r4RRR4       \        \        P
                  V4       \        \        P
                  V4       \        P                  ! \        RR7      ;_uu_ 4        \         P                  ! ^ .^ .4      w  r4RRR4       \        \        P
                  V4       \        \        P
                  V4       \        \         \         P                  ^ ^.. RO4       \        \         \         P                  ^ ^.^ ^.. RO4       . ROp^^^^\        P
                  .p\         P                  ! W4      w  r4\        VR4       ^^\        P
                  ^^.p\         P                  ! W4      w  r4\        VR4       . R OpRRRR\        P
                  .p\         P                  ! W4      w  r4\        VR4       RR\        P
                  RR.p\         P                  ! W4      w  r4\        VR4       . R!Op. R"Op\         P                  ! W4      w  r4\        VR#4       RR\        P
                  R\        P
                  .p\         P                  ! W4      w  r4\        VR#4       \        P
                  RR\        P
                  \        P
                  .p\         P                  ! W4      w  r4\        VR#4       R#   + '       g   i     EL; i  + '       g   i     ELZ; i)$rV   F)additivec                     ^# r\   r   rk   s   &rr   <lambda>2TestWeightedTau.test_weightedtau.<locals>.<lambda>  s    ru   )weigherN)rank)r  r  Tc                     ^# r  r   r  s   &rr   r  r	        Aru   )r  r
  c                     ^# r  r   r  s   &rr   r  r	    r  ru   r;   r  r         (@r   r   r   r   r~  r  gs$gROor  r>  gڿgng @ڿg[đ(gIکn࿩r~   r\   rJ   )r  r   r   r  r   )r  r   r   r  r   )r   r   r   r   r   gt34+)r`   weightedtaur	   r   r   rg   r   r<  r@  r^   r   r   r   r  r   r   r   )rh   rk   rl   r  r  r   rA  s   &      rr   test_weightedtau TestWeightedTau.test_weightedtau  s   ((.C!56RVVW%((>C!56RVVW%(({CC!56RVVW% %.
C,S__cmm4 ((D9C!45RVVW%((D9C!45RVVW%((D5IC!56RVVW%((D5IC!56RVVW%((E:C!56RVVW%((D+NC!56RVVW%((D+NC!56RVVW%((ARZZ)H!LC!56((ARXX)FJC!56((ARZZ)H)+ARZZ)HJC!56\\,4KLL ,,R4LC MRVVS!RVVW%\\,4KLL ,,aS1#6LC MRVVS!RVVW%j%"3"3aVYGj%"3"3aVaVYO1a ((.C!56BFFB"((.C!56'#sC(((.C!563c*((.C!56'%((.C!453bff-((.C!45VVS#rvvrvv.((.C!45K MLL MLLs   4[.[[	[,	c                >   R p\         P                  ! \        VR7      ;_uu_ 4        \        P                  ! ^.R.4       \        P                  ! ^.R.4       \        P                  ! \
        P                  .^4.4       RRR4       R#   + '       g   i     R# ; i)z)One or more sample arguments is too smallr   r   N)r   r  r   r`   r  r   rg   r:  s   & rr   test_segfault_issue_9710(TestWeightedTau.test_segfault_issue_9710  sj    
 >\\,G<<qcC5)qcC5)rvvh-	 =<<<s   ABB	c           
     <   R  pR p\         P                  P                  ^*4      p\        ^^
4       F  p. p\        V4       F  pWV.V,          ,          pK  	  \	        V4      pVP                  V4       VP                  V4       \         P                  ! \        V4      \         P                  R7      p\        ^4       FT  p	R F:  p
V! WWWV
4      p\        P                  ! WWWV
4      P                  p\        W4       K<  	  VP                  V4       KV  	  K  	  R# )c                    ^ ;p;p;p;r\        \        \        V 4      4      \        \        V 4      4      4       EF>  w  rV'       d"   V! W*,          4      V! W+,          4      ,           M V! W*,          4      V! W+,          4      ,          pW\,          pW
,          W,          8X  d	   W,          pW,          W,          8X  d	   W,          p	W
,          W,          8  d   W,          W,          8  g'   W
,          W,          8  d   W,          W,          8  d   Wl,          pK  W
,          W,          8  d   W,          W,          8  g-   W
,          W,          8  g   EK   W,          W,          8  g   EK6  W|,          pEKA  	  Wg,
          \        P                  ! WX,
          4      ,          \        P                  ! WY,
          4      ,          # r~   )r   r  lenr   rI  )rk   rl   r  r
  addtotconcdiscuvr  r  ws   &&&&&        rr   wkq:TestWeightedTau.test_weightedtau_vs_quadratic.<locals>.wkq  s%   ())C)$))!%A-s1v?;>GDG$wtw'77 )GDG,<< 414<FA414<FA4!$;14!$;!$+!$+IDTAD[QTAD[AD14KAD14KID @ K27737#33bggcg6FFFru   c                 "    R V ^,           ,          # )r   r   r  s   &rr   r
  >TestWeightedTau.test_weightedtau_vs_quadratic.<locals>.weigher  s    Q<ru   r;   NrX   )r   r  r  r  rB  shuffler   r  intpr`   r  r@  r	   )rh   r"  r
  r  srL  r  rX  r  r  r  r  r  s   &            rr   test_weightedtau_vs_quadratic-TestWeightedTau.test_weightedtau_vs_quadratic  s    	G 	  ii##B'qAA1XSU
 QAKKNKKN99SV2773D1X(C"1<H"..qTCHRRF'9 )
 D!  ru   r   N)	r   r   r   r   r  r  r)  r   r   r   s   @rr   r  r    s     Y6x
.&" &"ru   r  c                      a  ] tR tRt o R tR tR tR tR t]	P                  P                  R. RO4      R	 4       tR
 tR tR tR tR tR tR tR tR tR tR tRtV tR# )TestRegressioni  c                   VP                   p\        P                  ! VP                  \        4      VP                  \
        4      4      p\        VP                  VP                  R VR7      4       \        VP                  VP                  RVR7      4       VP                  RVR7      p\        VP                  VRR7       \        VP                  VRR7       R# )g   ؃חAr;   r   r   r  rF  N)r   r`   
linregressr^   r_   r   r2   	interceptrvaluestderrintercept_stderr)rh   ri   r<   r  zeros   &&   rr   test_linregressBIGX"TestRegression.test_linregressBIGX  s    

!!"**Q-CA(("**Ye**LMrzz#Uz'CD zz#Uz+t$7//DAru   c                   \         P                  ! VP                  \        P	                  4       4      VP                  \        P	                  4       4      4      p\        VP                  VP                  R 4      RR7       \        VP                  VP                  R4      4       \        VP                  VP                  R 4      RR7       \        VP                  VP                  R 4      RR7       R# )r   rE  rF  r   r  N)
r`   r.  r^   r_   r   r2   r/  r0  r1  r2  rh   ri   r  s   && rr   test_regressXXTestRegression.test_regressXX*  s    
 !!"**QXXZ"8"**QXXZ:PQ(("**S/Frzz#7rzz#TB//CtLru   c           	     t   \         P                  ! VP                  \        P	                  4       4      VP                  \
        P	                  4       4      4      p\        VP                  VP                  R 4      4       \        \         P                  RVR7      ;_uu_ 4        \         P                  ! VP                  \        P	                  4       4      VP                  \
        P	                  4       4      4      P                  pRRR4       \        VP                  X4       R#   + '       g   i     L(; i)r   zAn input array...r^  N)r`   r.  r^   r_   r   ZEROr2   r/  r"   rd  r   r@  r0  )rh   ri   r  
ref_rvalues   &&  rr   test_regressZEROX TestRegression.test_regressZEROX?  s    
 !!"**QXXZ"8"**T[[]:ST(("**S/:33;NSUVV

188:(>(*

4;;=(ACCL9  W 	z2 WVs   #A$D''D7	c                f   VP                  ^ ^d^d4      pRVP                  ^ ^d^d4      ,          ^
,           pW1P                  VP                  ^ ^^d4      4      ,           p\        P                  ! W#4      p\        p\        WE4      '       g   Q h\        VP                  VP                  R4      4       R# )r~   皙?g4_Qc?N)	linspacesinr`   r.  r   
isinstancer2   r1  r^   )rh   ri   rk   rl   r  lrs   &&    rr   test_regress_simple"TestRegression.test_regress_simpleK  s    KK3$"++ac**R/r{{1b#.//!!!'&%%%%rzz2G'HIru   c                   VP                  ^ ^d^d4      pRVP                  ^ ^d^d4      ,          ^
,           pW1P                  VP                  ^ ^^d4      4      ,           p\        P                  ! \        RR7      ;_uu_ 4        \
        P                  ! W#RR7       RRR4       \
        P                  ! W#RR7      p\
        P                  ! W#RR7      p\        VP                  ^VP                  ^,          ,
          4       \
        P                  ! W#R	R7      p\        VP                  VP                  ^,          4       VP                  VP                  u;8X  d   VP                  8X  g   Q h Q hR#   + '       g   i     L; i)
r~   r@  rg  r   rh  r  Nr  r  r  )
rA  rB  r   r   r   r`   r.  r2   r?  r0  )rh   ri   rk   rl   r  r  r  s   &&     rr   test_regress_alternative'TestRegression.test_regress_alternativeV  s   KK3$"++ac**R/r{{1b#.//]]:-NOOQ{; P +> &9Q$++/%:; )<T[[1_5{{dkk8T[[88888 POs   E33F	r<   Nc           
        Vf   TM
\        W!4      pVP                  . R	OVR7      pVP                  . R
OVR7      p\        P                  ! W4RR7      p\	        VP
                  VP                  RVR7      4       \	        VP                  VP                  RVR7      4       \	        VP                  VP                  VP                  RVR7      4      4       \	        VP                  VP                  RVR7      4       \	        VP                  VP                  RVR7      4       \	        VP                  VP                  RVR7      4       R # )Nr;   r  r  gؗh?gag?g->g5ՒO?g]؊E @)
                              )
?   Q   8   [   /   9   r2  H   >   0   gJK@:C)rl  r^   r`   r.  r2   sloper/  r0  rI  r?  r1  r2  )rh   r<   ri   rk   rl   r   s   &&&   rr   test_regress_against_R%TestRegression.test_regress_against_Rk  s     GB,>JJIQVJWJJ?uJMq=		2::oU:#KLrzz2C5z'QR

BGGBJJ}EJ,R$ST

BJJ/@J$NO

BJJeJ$LM,,bjjPUj.VWru   c                  a \         P                  ! ^4      p\         P                  ! ^^4      pV^R	.;;,          ^,          uu&   V^ R
.;;,          ^,          uu&   \        P                  ! SP	                  V4      SP	                  V4      4      pV3R lpV! VP
                  R4       V! VP                  R4       V! VP                  R4       V! VP                  R4       V! VP                  R4       V! VP                  R4       R# )rU   c                 >   < \        V SP                  V4      R R7      # )rh  rF  )r2   r^   )rk   rl   ri   s   &&rr   	assert_ae1TestRegression.test_linregress.<locals>.assert_ae  s    "1bjjm%@@ru   r   r=   gePUn?g.bt>gڽE?gʺL7?NrZ   )r   r   r`   r.  r^   r^  r/  r0  r?  r1  r2  )rh   ri   rk   rl   r  rc  s   &f    rr   test_linregressTestRegression.test_linregress  s    IIbMIIa	A+!	A+!!!"**Q-A?	A&,,$&""C(&--!12&--1&--!56&))+>?ru   c                   R Rr2VP                   pVP                  V^V,          W4R7      pVP                  ^V,          W#VR7      p\        P                  ! WV4      pVP                  R8  g   Q h\        VP                  VP                  RVR7      4       VP                  VP                  4      '       d   Q hVP                  VP                  4      '       d   Q hR# )gJr 11順 r;   NrZ   rQ  )
r   rA  r`   r.  r0  r2   r^   rx  r1  r2  )rh   ri   rL  r  r<   rk   rl   r  s   &&      rr    test_regress_simple_negative_cor/TestRegression.test_regress_simple_negative_cor  s     f1

KK1q5!K1KKAq5K1!!!' }}"""rzz#Uz'CD 88FMM****88F3344444ru   c                ^   VP                  ^ ^d^d4      pRVP                  ^ ^d^d4      ,          ^
,           pW1P                  VP                  ^ ^^d4      4      ,           p\        P                  ! W#4      p\        p\        WE4      '       g   Q hRp\        WFVR7       R\        V4      9   g   Q hR# )r~   r@  r=  r2  N)r^  r/  r0  r?  r1  )rA  rB  r`   r.  r   rC  r   dir)rh   ri   rk   rl   r  rD  rA  s   &&     rr   !test_linregress_result_attributes0TestRegression.test_linregress_result_attributes  s    KK3$"++ac**R/r{{1b#.//!!!' &%%%% J
F26!S[000ru   c                   VP                  ^4      pVP                  ^^4      p\        P                  ! W#4      p\        VP                  VP                  R4      4       \        VP                  VP                  R4      4       \        VP                  VP                  R4      4       VP                  VP                  4      p\        VP                  V4       \        VP                  V4       \        VP                  V4       R# )rJ   r   r   N)r   r`   r.  r2   r^  r^   r/  r0  rg   r3   r?  r1  r2  rh   ri   rk   rl   r  NaNs   &&    rr   test_regress_two_inputs&TestRegression.test_regress_two_inputs  s    IIaLIIaO!!!'bjjo6(("**S/:rzz#7jj s+s+//5ru   c                   VP                  R 4      pVP                  ^4      p\        P                  ! W#4      p\	        VP
                  VP                  R4      4       \	        VP                  VP                  R4      4       VP                  VP                  4      p\        VP                  V4       \        VP                  V4       \        VP                  V4       \        VP                  V4       R# )r   r   r   N)r   r  r`   r.  r2   r^  r^   r/  rg   r3   r0  r?  r1  r2  rq  s   &&    rr   'test_regress_two_inputs_horizontal_line6TestRegression.test_regress_two_inputs_horizontal_line  s    IIbMGGAJ!!!' 	bjjo6(("**S/:jj s+s+s+//5ru   c                   . R NRNRNRNRNRNRNRNRNR	NR
NRNRNRNRNRNRNRNRNRNRNRNRNRNRNRNRNRNRNRNRNRNRNR NR!NR"Np. R#NR$NR%NR&NR'NRNR(NR)NR*NR+NR,NRNR#NR-NR.NR/NR!NR0NR1NR2NR3NR4NR5NRNRNR6NR/NR&NR)NR+NR7NR8NR9NR:NR;NR Np\         P                  ! VP                  V4      VP                  V4      4      p\        VP                  VP                  R<4      4       \        VP
                  VP                  RD4      4       \        VP                  ^,          VP                  R=4      4       \        VP                  VP                  R>4      R?R@7       VP                  P                  VP                  8X  dM   \        VP                  VP                  RA4      4       \        VP                  VP                  RB4      4       RC# RC# )Er@  gfffffu@ǧ]@g̤@333333$@g     Pl@gfffff҄@gfffff"@g	|@g     H@gq@皙?333333?g     <@g33333ׄ@g      u@g     {@g333333'@g     `@g33333l@gfffff@g̼@g^@rF  gffffff@g1u@g@g     8@g     X@333333&@g33333]@gfffffl@g@gfffff|@r6   rC  g,u@gfffff]@g     @r  g     @g     4@g|@g33333W@gy@gP@gfffff@g4u@g%@gm@gl@g     0@gfffffƋ@gfffff]@gl@gffffff$@gffffff]@g̜l@g33333@g33333|@g?gpX?r   g#aMR/rF  gh-h*<?gg?Ng!п)r`   r.  r^   r2   r^  r/  r0  r?  r<   r   r1  r2  )rh   ri   rk   rl   r  s   &&   rr   test_nist_norrisTestRegression.test_nist_norris  sd   7s 7E 75 7% 7 7 777 7"'7).725777 7"'7).7157 7 7 !7 #(7 */7 367 	7 	7 !	7 #(	7 */	7 16	7
 7
 7
 !7
 #(7
 */7
 3677s 7E 75 7% 7 7 777 7"'7).725777 7"'7).7157 7 7 !7 #(7 */7 367 	7 	7 !	7 #(	7 */	7 16	7
 7
 7
 !7
 #(7
 */7
 367 !!"**Q-A?bjj1A&BC(("**5G*HIq("**5F*GHrzz#UC=="**,FMM2::6F+GHF33RZZ@P5QR -ru   c                0   \         P                  ! ^ ^d^d4      pR\         P                  ! ^ ^d^d4      ,          ^
,           pV\         P                  ! \         P                  ! ^ ^^d4      4      ,          p\        P                  ! VP                  V4      VP                  V4      4      p\         P                  ! W#^4      p\        VP                  VP                  V^ ,          4      4       \        VP                  VP                  V^,          4      4       R# )r~   r@  N)
r   rA  rB  r`   r.  r^   polyfitr2   r^  r/  )rh   ri   rk   rl   r  polys   &&    rr   test_compare_to_polyfit&TestRegression.test_compare_to_polyfit  s    KK3$"++ac**R/	RVVBKK2s+,,!!"**Q-A?zz!" 	bjja&9:(("**T!W*=>ru   c                   VP                  R R.4      pVP                  R R.4      p\        P                  ! W#4      pVP                  VP                  4      p\	        VP
                  V4       \	        VP                  V4       \	        VP                  V4       \        VP                  VP                  R4      4       \        VP                  VP                  R 4      4       R# )r   r   N)r^   r`   r.  rg   r3   r?  r1  r2  r2   r^  r/  )rh   ri   rk   rl   r   rr  s   &&    rr   (test_linregress_two_points_nan_inference7TestRegression.test_linregress_two_points_nan_inference  s    JJBx JJBx q$jj 

C(

C(,,c2 			2::c?3rzz#7ru   c                   \        \        R VR7      ;_uu_ 4        \        P                  ! VP	                  . 4      VP	                  . 4      4      pVP	                  VP
                  4      p\        VP                  V4       \        VP                  V4       \        VP                  V4       \        VP                  V4       \        VP                  V4       \        VP                  V4       RRR4       R#   + '       g   i     R# ; i)r  r^  N)r"   r   r`   r.  r^   rg   r3   r^  r/  r0  r?  r1  r2  rh   ri   r   rr  s   &&  rr   test_empty_inputTestRegression.test_empty_input	  s    +3JrRR""2::b>2::b>BC**RVV$CCIIs+CMM3/CJJ,CJJ,CJJ,C00#6 SRRRs   CC::D	c                   \         P                  ! R 4      p\         P                  V^	&   VP                  VP	                  4       4      p\         P
                  ! RR7      ;_uu_ 4        \        P                  ! W"4      pRRR4       \        p\        XV4      '       g   Q hVP                  VP                  4      p\        VP                  V4       \        VP                  V4       \        VP                  V4       \        VP                  V4       \        VP                  V4       \        VP                   V4       R#   + '       g   i     L; i)r   r   invalidN)r   r   rg   r^   r   errstater`   r.  r   rC  r3   r^  r/  r0  r?  r1  r2  )rh   ri   rk   r  rD  rr  s   &&    rr   test_nan_inputTestRegression.test_nan_input	  s    IIcNvv!JJqxxz"[[**%%a+F + &"%%%%jj c*((#.s+s+s+//5 +*s   (EE	c                   \         P                  P                  R 4      pVP                  ^
4      pVP	                  VP                  ^
4      P                  4       4      p\         P                  ! RR7      ;_uu_ 4        \        P                  ! W44      pRRR4       VP	                  VP                  4      p\        XP                  V4       \        VP                  V4       \        VP                  V4       \        VP                  V4       \        VP                  V4       \        VP                   V4       R#   + '       g   i     L; i)l   ,w* r   r  N)r   r  r  rX  r^   r   r  r`   r.  rg   r3   r^  r/  r0  r?  r1  r2  )rh   ri   r  rk   rl   r   rr  s   &&     rr   test_identical_xTestRegression.test_identical_x-	  s    ii##J/HHRLJJszz"~,,./[[**""1(C +jj 		3's+

C(

C(

C(,,c2 +*s   >D>>E	r   r  )r   r   r   r   r4  r8  r=  rE  rH  r   r   r   r_  rf  rj  rn  rs  rv  r}  r  r  r  r  r  r   r   r   s   @rr   r,  r,    s     
B
M*
3	J9* [[W&BCX DX&@,5&1 66 S8
?8 	76&3 3ru   r,  c                   4  a  ] tR tRt o R t]P                  P                  R. RO4      ]P                  P                  RRR.4      ]P                  P                  RRR.4      ]P                  P                  R	RR.4      R
 4       4       4       4       tRt	V t
R# )HistFunctionsTesti<	  c                   \        \        V P                  4      pVP                  ^4      p^
pRpVP	                  ^4      pRp\
        P                  ! \        VR7      ;_uu_ 4        V! W4^
4       RRR4       \
        P                  ! \        VR7      ;_uu_ 4        V! W4R4       RRR4       \
        P                  ! \        VR7      ;_uu_ 4        V! W4VRRR1,          4       RRR4       Rp\
        P                  ! \        VR7      ;_uu_ 4        V! VP                  W1P                  4      WE4       RRR4       \
        P                  ! \        VR7      ;_uu_ 4        V! W4WQP                  WaP                  4      4       RRR4       \        V4      '       g<   Rp\
        P                  ! \        VR7      ;_uu_ 4        V! W4WV) 4       RRR4       R# R#   + '       g   i     ELu; i  + '       g   i     ELS; i  + '       g   i     EL'; i  + '       g   i     L; i  + '       g   i     L; i  + '       g   i     R# ; i)	   z,If specified, `defaultreallimits` must be...r   Nz8`a` and \(if specified\) `weights` must have real dtype.z#All `weights` must be non-negative.)r~   r  r  rZ   )rl  r`   histfunr   r  r   r   r   r  
complex128r#   )rh   ri   r  rL  numbinsdefaultreallimitsweightsr  s   &&      rr   r  'HistFunctionsTest.test_input_validation=	  s   %.IIbM#''"+@]]:W55A# 6]]:W55A	* 6]]:W55A 1$B$ 78 6 N]]:W55BIIa/L 6]]:W55A 199Wmm3TU 6 W%%;Gz99$5x@ :9 & 655555555 6555
 :99sH   '
G
G1H#H$#H,H?G.	1H	H	H)	,H<	?I	r<   Nedge_pointsFTspecify_limitsspecify_weightsc           
        \         P                  P                  R 4      p\        VP	                  ^
^4      4      pV'       d   V'       g   \
        P                  ! R4       V'       dX   \        V4      '       g1   VR8w  d   \        V4      '       g   VR8w  d(   \        V4      '       d   \
        P                  ! R4       \         P                  ! VP                  ^d4      V4      pV'       d>   \         P                  ! \         P                  ! VP                  ^4      4      V4      p	T	p
MjRVP                  4       VP                  4       ,
          ,          V^,
          ,          pVP                  4       V,
          VP                  4       V,           3p
Rp	V'       dt   \         P                  ! WV
RR7      w  r\         P                  ! WP	                  ^VP                   R7      P#                  4       4      p\         P$                  ! W.VR7      pV'       d1   \         P                  ! VP                  VP                   4      V4      MRp\         P                  ! WV
RR7      w  pp\         P                  ! WV
VRR	7      w  ppVf   \'        V4      M
\)        WQ4      p\)        \*        V P,                  4      pVf   TMVP/                  WR7      pV! VP/                  WR7      WyV4      pVP/                  VVR7      pWP0                  8X  d   R
MRpV\*        P2                  8X  d+   VP5                  V4      p\7        VP8                  VVR7       MZVf   VP                   ^ ,          MVP;                  V4      pVP/                  VV,          VR7      p\7        VP<                  VVR7       \7        VP>                  VP/                  V^ ,          VR7      4       \7        VP@                  VP/                  V^,          V^ ,          ,
          VR7      4       \7        VPB                  VP/                  VPD                  \;        VP#                  4       4      ,
          4      4       R# )l   =k z2Limits must be specified to determine edge points.r   z<Some backends are having trouble with edge point arithmetic.NF)binsr  densityr  r;   )r  r  r  r  -q=gh㈵>r  r6   )#r   r  r  intintegersr   skipr-   r%   r.   r  sortmaxmin	histogramrepeatr  r   concatr&   rl  r`   r  r^   r   cumfreqcumulative_sumr2   cumcountr]  	frequencyr   binsizeextrapointsr  )rh   r<   r  r  r  ri   r  nbinsrL  r?   range_r(  r  edgesa2r  
bin_countsref_binsref_histr  r   r  ref_cumcountdenref_frequencys   &&&&&&                   rr   test_against_reference(HistFunctionsTest.test_against_referenceW	  s    ii##J/CLLR()~KKLMF2JJ % 2x|| % 2wr{{KKVWIIcjjou-YYrwwszz!}5u=FF 1557*+uqy9Aeegk1557Q;/FF||AOHA5,,qu{{,"C"J"J"LMB		1'/A;J"))CJJqww/7PT!||AQVW
Hll1+2EC! ). $72;M%.$_'"**W*2Rbjjj0%I::he:4+uemm#,,X6LCLL,TB '!''!*RVVG_CJJx|5JAMCMM=tD

8A;e
(LMRZZhqk0IQVZ%WXAFFSARARAT=U4U)VWru   r   r  )r   r   r   r   r  r   r   r   r  r   r   r   s   @rr   r  r  <	  s     A4 [[W&BC[[]UDM:[[-t}=[[.>2X ? > ; D2Xru   r  c                   *   a  ] tR tRt o RtR tRtV tR# )TestCumfreqi	  r  c                4   VP                  . RO4      p\        P                  ! V^R7      w  r4rV\        W1P                  . RO4      4       \        P                  ! V^RR7      w  r4rVV^8X  g   Q hRp\        P                  ! V^RR7      p\	        WVR7       R# )	r\   r  )r  r  r=  Nr\   r   rJ   r\   r   r\   )r   r   r=   r   )r7   r|   )r  r   r  r  )r^   r`   r  r3   r   )	rh   ri   rk   cumfreqslowlimr  r  rA  r   s	   &&       rr   test_cumfreqTestCumfreq.test_cumfreq	  s    JJ)*16q!1L.'**-=">?16qH26.'a J
mmAqHEC3ru   r   N)r   r   r   r   r  r  r   r   r   s   @rr   r  r  	  s     G4 4ru   r  c                   0   a  ] tR tRt o RtR tR tRtV tR# )TestRelfreqi	  relfreqc                    VP                  . RO4      p\        P                  ! V^R7      w  r4rV\        W1P                  . RO4      4       Rp\        P                  ! V^R7      p\	        WVR7       R# )r\   r  r=  Nr  )r6   K}\UU?r  r  )r  r   r  r  )r^   r`   r  r2   r   )	rh   ri   rL  relfreqsr  r  r  rA  r   s	   &&       rr   test_relfreqTestRelfreq.test_relfreq	  sZ    JJ)*16q!1L.'**-V"WX K
mmAq)C3ru   c                    . ROp\         P                  ! V^R7      w  p  p\         P                  ! \        P                  ! V4      ^R7      w  p  p\	        W$4       R# )r\   r  Nr  )r`   r  r   r^   r3   )rh   rL  r   r  r   s   &    rr   test_relfreq_array_like#TestRelfreq.test_relfreq_array_like	  sG    }}Q2Q1}}RZZ]A>Q1!ru   r   N)	r   r   r   r   r  r  r  r   r   r   s   @rr   r  r  	  s     G4" "ru   r  c                   P   a  ] tR tRt o R tR tR tR tR tR t	R t
R	 tR
tV tR# )TestScoreatpercentilei	  c                <    . ROV n         . ROV n        . ROV n        R# )r   N)r   r   r|   rT   r[   r   )r   re  rK   rO   r   rJ   r\   )r   r   r|   rT   r[   r  r  r   a1r  a3rs  s   &rr   rt  "TestScoreatpercentile.setup_method	  s    *,1ru   c                    \        ^4      R,          p\        \        P                  ! V^ 4      R4       \        \        P                  ! V^d4      R4       \        \        P                  ! V^24      R4       R# )rK   r6   r         @      ?N)r   r   r`   scoreatpercentiler  s   & rr   rY   TestScoreatpercentile.test_basic	  sT    1IOU,,Q2B7U,,Q4c:U,,Q3T:ru   c           	     P   \         P                  p\        V! \        \	        ^
4      4      ^24      R4       \        V! \        \	        ^
4      4      ^2R4      R4       \        V! \        \	        ^d4      4      ^2R	R7      R4       \        V! \
        P                  ! . R
O4      ^2R4      ^74       \        V! \
        P                  ! . R
O4      ^2R4      R4       \        V! \        \	        ^
4      4      ^2RR7      R4       \        V! \        \	        ^
4      4      ^2RRR7      R4       \        V! \        \	        ^d4      4      ^2R	RR7      R4       \        V! \
        P                  ! . R
O4      ^2RRR7      ^74       \        V! \
        P                  ! . R
O4      ^2RRR7      R4       R# )rT   rF   )limitrG   fractioninterpolation_methodr  r  NrJ   rO   r\   rK   r\   rT   r  rT   r  r\   rT   r`   r  r   rB  r  r   r   rh   scoreatpercs   & rr   test_fraction#TestScoreatpercentile.test_fraction	  sP   -- 	[eBi"5s;[eBi"e<cB[eCj!12VDcJ[+!6HErJ[+!6FCSI 	[eBi":V	[eBi"F6@B	 	[eCj!12V6@B	 	[+!6I6@B	 	[+!6F6@B	ru   c           	     b   \         P                  p\        V! \        \	        ^
4      4      ^2RR7      ^4       \        V! \        \	        ^
4      4      ^2RR7      ^4       \        V! \        \	        ^
4      4      ^2RRR7      ^4       \        V! \        \	        ^
4      4      ^2RRR7      ^4       \        V! \        \	        ^d4      4      ^2RRR7      ^4       \        V! \        \	        ^d4      4      ^2RRR7      ^4       \        V! \
        P                  ! . RO4      ^2R	RR7      ^
4       \        V! \
        P                  ! . RO4      ^2R	RR7      ^d4       \        V! \
        P                  ! . RO4      ^2R
RR7      ^4       \        V! \
        P                  ! . RO4      ^2R
RR7      ^
4       R# )rT   lowerr  higherr  Nr  r  r  r  r  r  r  s   & rr   test_lower_higher'TestScoreatpercentile.test_lower_higher	  s~   -- 	[eBi"6=?@A	C[eBi"6>@AB	D[eBi"e6=?@A	C[eBi"E6>@AB	D[eCj!12u6=?@A	C[eCj!12u6>@AB	D[,!7Y6=?@B	D[,!796>@AD	F[,!7W6=?@A	C[,!776>@AC	Eru   c           	     >   \        ^4      R,          p\        P                  ! . RO4      p\        P                  ! V. RO4      p\        W24       \        \        V\        P                  4      4       \        \        P                  ! V\        P                  ! . RO4      4      V4       \        P                  ! \        P                   ! ^4      P                  R4      \        P                  ! . RO4      ^R7      p\        . RO. R	O. R
O. R
O.4      p\        WE4       R# )rK   r6   rC   N)r~   r  r  )r~   r  2   r   r   )r~   r\   r  r  )r~   r   rK   )Q?gQ@g(\ @r   rO   rU   )
r   r   r   r`   r  r
   r   rC  ndarrayrb   )rh   rk   r  r   r  	expected2s   &     rr   test_sequence_per'TestScoreatpercentile.test_sequence_per	  s    1IO88N+%%a6&
3

+,//288L3IJ 	" &&ryy}'<'<U'C')xx0@'AK9-%%' (	 	(ru   c                &   \         P                  p\        ^4      P                  ^^4      p\	        V! VR4      . RO4       . RO. RO. RO.p\	        V! VR^ R7      V4       . R	O. R
O. RO.p\	        V! VR^R7      V4       \        . RO. RO. RO. RO. RO.4      p\         P                  ! V^24      p\	        VP                  R4       \	        VR4       \         P                  ! V^2^ R7      p\	        VP                  R4       \	        V. RO4       R# )rV   rC   r   N)   r  r  )g      @rG         &@)rJ   r   r   r|   )r   r|   r   rO   )rK   rN   rT   rU   )      ?g      @g     !@)r7   rG         #@r  r\   r\   r\   )r   r   r   r   r   )r`   r  r   rb   r   r   r  )rh   r  rk   r0r  scores   &     rr   	test_axisTestScoreatpercentile.test_axis	
  s    --2Jq!$[M24EFL.9[M:B? /:>[M:B?9	 
 ''2.U[["%UC ''2A6U[[$'UI&ru   c                    \        \        \        P                  ^^.^8RR7       \        \        \        P                  ^.^e4       \        \        \        P                  ^.R4       R# )r\   r   r  NrZ   )r   r   r`   r  rs  s   &rr   test_exception$TestScoreatpercentile.test_exception!
  sK    j%"9"9Aq62+3	5j%"9"9A3Dj%"9"9A3Cru   c                p   \        \        P                  ! . ^24      \        P                  4       \        \        P                  ! \        P
                  ! . . .4      ^24      \        P                  4       \        \        P                  ! . ^2^c.4      \        P                  \        P                  .4       R# )r  N)r   r`   r  r   rg   r   rs  s   &rr   
test_empty TestScoreatpercentile.test_empty'
  sm    U,,R4bff=U,,RXXr2h-?DbffMU,,R"b:RVVRVV<LMru   r  N)r   r   r   r   rt  rY  r  r  r  r	  r  r  r   r   r   s   @rr   r  r  	  s6     2
;4E2)$'0DN Nru   r  c                   *  a  ] tR tRt o R tR tR tR tR t]	P                  P                  R]! R$^ 4      4      R 4       tR	 tR
 t]! RRR7      R 4       t]! RRR7      ]	P                  P                  R. R%O^]P&                  ^^^^.. R&O^]P&                  ^^..4      ]	P                  P                  RRR.4      R 4       4       4       t]	P                  P                  R]! R'^4      4      R 4       tR tR tR tR t]	P                  P                  RRR.4      R 4       tR t]	P                  P9                  R4      ]	P                  P                  R]P:                  ! R(4      ]P:                  ! R)4      .4      R  4       4       tR! tR"t V t!R## )*TestModei-
  c                H   \        \        \        VR 7      ;_uu_ 4        \        P                  ! VP                  . 4      4      w  r#RRR4       \        XVP                  VP                  4      4       \        XVP                  R4      4       R#   + '       g   i     LR; i)r^  Nr   )r"   r   r   r`   moder^   r3   rg   )rh   ri   valscountss   &&  rr   r  TestMode.test_empty0
  sd    +3HRPP ::bjjn5LD Qbjj01

2/ QPs   (BB!	c                    \         P                  ! R 4      w  r\        V\        P                  ! R .4      4       \        V\        P                  ! ^.4      4       R# )r   N)r`   r  r   r   r   )rh   r  r  s   &  rr   ry  TestMode.test_scalar6
  s9    zz"~T288RD>*VRXXqc]+ru   c                    VP                  . RO4      p\        P                  ! V4      p\        V^ ,          VP                  ^4      4       \        V^,          VP                  ^4      4       R# )r   Nr   r|   r\   rT      r   rJ   r   rK   r   rT   r   r^   r`   r  r3   )rh   ri   data1r  s   &&  rr   rY  TestMode.test_basic;
  sK    

BCzz% QA/QA/ru   c           
        . ROp. ROp. ROp. ROp. ROpVP                  W#WEV.4      p\        P                  ! VRRR7      p\        V^ ,          VP                  ^..4      4       \        V^,          VP                  ^..4      4       \        P                  ! V^ RR7      p\        V^ ,          VP                  . R	O.4      4       \        V^,          VP                  . R
O.4      4       \        P                  ! V^RR7      p\        V^ ,          VP                  ^
.^
.^.^.^..4      4       \        V^,          VP                  ^.^.^.^.^..4      4       R# )rT   NTrD   keepdimsrT   rT      (   rT   rT   rT   rT   r  rT   r  r  r$  r$  r$  r$  r%  r$  r$  r$  rT   rT   r$  r$  rJ   r   r   rJ   r  	rh   ri   r  data2data3data4data5arrr  s	   &&       rr   test_axes_keepdimsTestMode.test_axes_keepdimsA
  s(        jj%e<=zz#D48QbTF!34QaSE!23zz#A5Q-=,>!?@Q\N!;<zz#A5QbTB4"tbT,J!KLQaS1#sQC!,E!FGru   c                   . ROp. ROp. ROp. ROp. ROpVP                  W#WEV.4      p\        P                  ! VRR7      p\        V^ ,          VP                  ^4      4       \        V^,          VP                  ^4      4       \        P                  ! V^ R7      p\        V^ ,          VP                  . RO4      4       \        V^,          VP                  . R	O4      4       \        P                  ! V^R7      p\        V^ ,          VP                  . R
O4      4       \        V^,          VP                  . RO4      4       R# )rT   NrC   r#  r&  r'  r(  r)  r*  r+  )rT   rT   r  r$  r$  )rJ   r   r   r   r   r  r,  s	   &&       rr   	test_axesTestMode.test_axesU
  s         jj%e<=zz#D)QB0QA/zz#A&Q,<!=>QL!9:zz#A&Q,@!ABQO!<=ru   rD   c                   \         P                  P                  R 4      pVP                  VP                  R4      4      p\        P
                  ! WDP                  V,           R7      p\        P
                  ! WAR7      p\        VP
                  VP
                  4       \        VP                  VP                  4       R# )l   q!<M rC   N)rT   rU   rV   rW   )	r   r  r  r^   r`   r  ndimr3   count)rh   rD   ri   r  rL  r  r  s   &&&    rr   test_negative_axes_gh_15375$TestMode.test_negative_axes_gh_15375i
  sv    ii##J/JJszz"234zz!&&+.zz!'		499-

DJJ/ru   c                R   VP                  . RO4      pVP                  . 4      p\        P                  ! V4      pRp\        WEVR7       \	        \
        \        VR7      ;_uu_ 4        \        P                  ! V4      pRRR4       \        XWQR7       R#   + '       g   i     L; i)r   r=  r^  Nr  )r  r9  )r^   r`   r  r   r"   r   r   )rh   ri   r  r-  r  rA  actual2s   &&     rr   test_mode_result_attributes$TestMode.test_mode_result_attributesr
  st    

BC

2E"&
F26+3HRPPjj'G QGZ7 QPs   (BB&	c                *   VP                  ^\        P                  ^^^
^^^^^^^
^.4      p\        P                  ! V4      p\        V^ ,          VP                  ^VP                  R7      4       \        V^,          VP                  ^4      4       R# )r   r;   N)r^   r   rg   r`   r  r3   r<   )rh   ri   r  r  s   &&  rr   test_nan_propagateTestMode.test_nan_propagate|
  sr    

Arvvq!RQ1aBJKE"q	2::au{{:#CDq	2::a=1ru   Tlazy arrays don't do 'raise'.r   c                   VP                  ^\        P                  ^^^
^^^^^^^
^.4      p\        P                  ! VRR7      p\        VP                  VP                  R4      4       \        VP                  VP                  ^4      4       \        \        \        P                  VRR7       \        \        \        P                  VRR7       R# )r   r   r   r   r   r   N)	r^   r   rg   r`   r  r3   r9  r   r   )rh   ri   r  r   s   &&  rr   test_nan_omitTestMode.test_nan_omit
  s    

Arvvq!RQ1aBJKjj62"**R.1		2::a=1j%**eHj%**eIru   zlazy arrays don't do 'omit'.datar"  Fc                    \         P                  ! VP                  V4      R VR7      pV'       d,   \        V^ ,          ^ ,          VP                  R4      4       R# \        V^ ,          VP                  R4      4       R# )r   )r   r"  r   N)r`   r  r^   r3   )rh   rG  r"  ri   r  s   &&&& rr   test_smallest_equalTestMode.test_smallest_equal
  sR     BJJt,(SF1IaL"**R.9F1Irzz"~6ru   c                   \         P                  P                  R 4      pVP                  VP	                  RR7      4      p\
        P                  ! WARR7      p\        VP                  4      pVP                  V4       \         P                  P                  VP                  P                  V4       \         P                  P                  VP                  P                  V4       R# )i:r  Fr!  Nr   r   r|   )r   r  r  r^   uniformr`   r  rB  r  popr!  r   r9  )rh   rD   ri   r  rL  r   reference_shapes   &&&    rr   test_mode_shape_gh_9955 TestMode.test_mode_shape_gh_9955
  s    ii##I.JJs{{	{23jj6qww-D!


%%chhnnoF


%%ciiooGru   c                    VP                  ^\        P                  ^\        P                  .4      p\        P                  ! V4      pVP                  VP                  4      '       d   VP                  ^8X  g   Q hR# rJ   N)r^   r   rg   r`   r  rx  r9  rh   ri   rL  r   s   &&  rr   !test_nan_policy_propagate_gh_9815*TestMode.test_nan_policy_propagate_gh_9815
  sU     JJ2661bff-.jjmxx!!cii1n44nru   c                |   VP                  R4      p\        P                  ! V^RR7      pVP                  P                  VP                  P                  u;8X  d	   R8X  g   Q h Q h\        P                  ! V^RR7      pVP                  P                  VP                  P                  u;8X  d	   R8X  g   Q h Q hR# )r\   Fr!  TN)r\   rJ   r   r~   )r\   r   r~   )r\   r\   r   r~   )rX  r`   r  r  r9  rT  s   &&  rr   test_keepdims_emptyTestMode.test_keepdims_empty
  s    HH\"jjU3xx~~=I=====jjT2xx~~@L@@@@@ru   c                V   VP                  ^^^\        P                  .^^\        P                  ^..4      p\        P                  ! V^RR7      p\        VP                  VP                  RR.4      4       \        VP                  VP                  ^^.4      4       \        P                  ! V^RR7      p\        VP                  VP                  R.R..4      4       \        VP                  VP                  ^.^..4      4       VP                  V4      p\        P                  ! VRRR7      p\        P                  ! \        V4      RR7      p\        VP                  VP                  4       \        VP                  VP                  4       VP                  P                  VP                  P                  u;8X  d	   R8X  g   Q h Q h\        P                  ! VRRR7      p\        P                  ! \        V4      RR7      p\        \        VP                  4      \        VP                  4      4       VP                  P                  R	8X  g   Q h\        \        VP                  4      \        VP                  4      4       VP                  P                  R	8X  g   Q hR# )
r\   Fr!  r   r   TNr"  r   r  )	r^   r   rg   r`   r  r3   r9  r*   r  )selfselfri   rL  r   r   s   &&   rr   test_keepdims_nonemptyTestMode.test_keepdims_nonempty
  s   JJAq"&&)Aq"&&!+<=>jjU3"**b"X"67		2::q!f#56jjT2"**rdRD\":;		2::sQCj#9:JJqMjj6jj!u5#((+		399-xx~~5255555jj5jj!t4*HSXX,>?xx~~'''+Xcii-@Ayy&(((ru   c                h   ^\         P                  \         P                  \         P                  ^.\         P                  \         P                  \         P                  \         P                  ^.^^\         P                  ^^..p\        P                  ! V^RRR7      p\	        VP                  . RO4       \	        VP
                  . RO4       \        P                  ! V^RRR7      p\	        VP                  ^.^.^..4       \	        VP
                  ^.^.^..4       \         P                  ! V4      p\        P                  ! VRRRR7      p\        P                  ! VP                  4       RRR7      p\	        W#4       VP                  P                  VP                  P                  u;8X  d	   R	8X  g   Q h Q h\        P                  ! VRRRR7      p\        P                  ! VP                  4       RRR7      p\        VP                  P                  4       VP                  P                  4       4       VP                  P                  R
8X  g   Q h\        VP
                  P                  4       VP
                  P                  4       4       VP
                  P                  R
8X  g   Q hR# )r\   Fr   )rD   r"  r   TN)r"  r   )r\   rJ   r|   )rJ   r\   rJ   r   r  )
r   rg   r`   r  r   r9  r   r  r  r   )rh   rL  r   r   s   &   rr   test_keepdims_nan_omitTestMode.test_keepdims_nan_omit
  s   +ffbffbffbffa0BFFAq!# jjUvF388Y/399i0jjTfE388qcA3_5399sQC!o6HHQKjj&IjjUvF3$xx~~5255555jjHjjTfESXX^^%sxx~~'78xx~~'''SYY__&		(9:yy&(((ru   r   r  r   c                F   R
p\         P                  ! V4      p\         P                  VR&   \        P                  ! V^RVR7      p\        VP                  . RO4       \        VP                  . RO4       \         P                  ! R\         P                  3R\         P                  R3.4      p\         P                  ! ^
VR7      pRp\        P                  ! \        VR7      ;_uu_ 4        \        P                  ! WaR7       R	R	R	4       R	#   + '       g   i     R	# ; i)r   F)rL  rD   r"  r   asdfqwerr;   z=Argument `a` is not....|An argument has dtype...|The DType...r   r   Nr  r  rX  rJ   r   r   r   r  )r   r  rg   r`   r  r   r9  r<   uint8r   rX  r   r   	TypeError)rh   r   r  rG  r   my_dtypetestr  s   &&      rr   test_gh16955TestMode.test_gh16955
  s     wwu~VVT
jj4a%JO388\2399l3 88fbhh/&"**d1KLMxx(+Q]]9G44JJt3 5444s   -DD 	c                   \         P                  ! \        \        R 7      ;_uu_ 4        \        P
                  ! . 4      pRRR4       \        P                  ^ 3p\        XV4       \         P                  ! \        \        R 7      ;_uu_ 4        \        P
                  ! \        P                  .RR7      pRRR4       \        W4       . RO\        P                  \        P                  \        P                  ..p\         P                  ! \        \        R 7      ;_uu_ 4        \        P
                  ! V^RR7      pRRR4       ^\        P                  .^^ .3p\        W4       \        P
                  ! V^RR7      p^\        P                  .^^.3p\        W4       \        P                  ! . . .4      p\         P                  ! \        \        R 7      ;_uu_ 4        \        P
                  ! V^R7      pRRR4       \        P                  \        P                  .^ ^ .3p\        W4       R#   + '       g   i     EL; i  + '       g   i     EL; i  + '       g   i     EL&; i  + '       g   i     L}; i)r   Nr   r   r  r  rC   )r   r9  r9  )r   r  r   r   r`   r  r   rg   r   r   r   r   r   )rh   r   r   rL  zs   &    rr   test_gh9955TestMode.test_gh9955
  s    \\,4IJJ**R.C KvvqkS#\\,4EFF**bffX&9C GSrvvrvvrvv67\\,4EFF**QQ6:C GBFF|aV$Sjj{;BFF|aV$SHHb"X\\,4IJJ**QQ'C K!Q(S/ KJJ
 GFF
 GFF KJs/   H3(I$I!I/3I	I	I,	/I?	zignore::RuntimeWarningrm  c                   VP                  V4      pVP                  ^ 8X  dL   \        P                  ! \        \
        R7      ;_uu_ 4        \        P                  ! VRRR7      pRRR4       M\        P                  ! VRRR7      pVP                  VRRR7      pX^ ,          P                  V^,          P                  u;8X  d   VP                  u;8X  d	   R8X  g   Q h Q hR#   + '       g   i     Ln; i)r~   r   NTr!  r  )
r^   r  r   r  r   r   r`   r  rc   r  )rh   rm  ri   r   r   s   &&&  rr   test_gh17214TestMode.test_gh17214  s     JJqM66Q;08MNNjj= ON **QTD9CggadTg21v||s1v||EsyyEIEEEEE ONs   C%%C5	c           	         ! R  R4      p\         '       d   RMRp\        P                  ! \        VR7      ;_uu_ 4        \        P
                  ! V! \        P                  ! ^4      4      4       RRR4       \         '       d   RMRp\        P                  ! \        VR7      ;_uu_ 4        \        P
                  ! \        P                  ! ^\        R7      4       RRR4       R#   + '       g   i     L; i  + '       g   i     R# ; i)c                   0   a  ] tR tRt o R tRR ltRtV tR# )8TestMode.test_raise_non_numeric_gh18254.<locals>.ArrLikei'  c                    Wn         R # r#  _xr  s   &&rr   __init__ATestMode.test_raise_non_numeric_gh18254.<locals>.ArrLike.__init__(  s    ru   Nc                @    V P                   P                  \        4      # r#  )rx  r  object)rh   r<   rE   s   &&&rr   	__array__BTestMode.test_raise_non_numeric_gh18254.<locals>.ArrLike.__array__+  s    ww~~f--ru   rw  )NN)r   r   r   r   ry  r}  r   r   r   s   @rr   ArrLikeru  '  s     . .ru   r  z'...only boolean and numerical dtypes...zCannot interpret...r   Nz*Argument `a` is not recognized as numeric.r;   )	r(   r   r   rg  r`   r  r   r   r|  )rh   r  r  s   &  rr   test_raise_non_numeric_gh18254'TestMode.test_raise_non_numeric_gh18254%  s    	. 	. AP<- 	 ]]9G44JJwryy|,- 5 AP<D 	]]9G44JJryy&12 54 54
 544s   1C',1C:'C7	:D	r   N)r   r|   r\   r\   r   )r   r|   r   r[   r  r2  )"r   r   r   r   r  ry  rY  r2  r5  r   r   r   r  r:  r>  rA  r   rE  r   rg   rI  rP  rU  rX  r]  r`  rj  rn  r   emptyrq  r  r   r   r   s   @rr   r  r  -
  s    0,
0H(>( [[VU2q\20 3082 .MNJ OJ .LM[[V	
BFFAq!Q	
BFFAq	&  [[Z%77 8 N7 [[VU2q\2H 3H5A)4)6 [[\K+@A4 B4": [[ 89[[S288I#68K"LMF N :F3 3ru   r  c                      a  ] tR tRt o . ROtRt]P                  P                  R4      R 4       t	R t
]! RRR7      R	 4       tR
tV tR# )TestSEMi:  r   r  c                ^   VP                  V P                  4      R,          p\        V4      '       dI   \        P                  ! \
        \        R 7      ;_uu_ 4        \        P                  ! V4      pRRR4       Mr\        P                  ! 4       ;_uu_ 4        \        P                  ! R\        4       \        P                  ! R\        4       \        P                  ! V4      pRRR4       VP                  X4      '       g   Q hR#   + '       g   i     L+; i  + '       g   i     L>; i)r   Nr   r   )r^   scalar_testcaser$   r   r  r   r   r`   semr   r   simplefilterUserWarningr   rx  )rh   ri   r  rl   s   &&  rr   test_sem_scalarTestSEM.test_sem_scalar@  s     **T%9%9:2>B<<08MNNIIo. ON ((**%%h<%%h?IIo. + xx{{{ ON +*s   D	AD	D	D,	c                @   VP                  V P                  4      p\        P                  ! V4      p\	        W1P                  R 4      4       \        V P                  4      p\	        \        P                  ! V^ R7      WD^,
          ,          R,          ,          \        P                  ! V^R7      4       VP                  R4      pVP                  V^	8H  VP                  V4      p\        \        P                  ! V4      VP                  VP                  4      4       R# )g㝎?ddofr6   r   N)
r^   testcaser`   r  r2   r  r   whererg   r3   )rh   ri   r  rl   r  rk   s   &&    rr   test_semTestSEM.test_semS  s    ::dmm,IIh::l34		(3qA#wnD		(3	5 IIcNHHQ!VRVVQ'		!bjj&89ru   Tr   r   c                   VP                  R 4      pVP                  V^	&   \        \        P                  ! VRR7      VP                  R4      4       Rp\        P                  ! \        VR7      ;_uu_ 4        \        P                  ! VRR7       RRR4       Rp\        P                  ! \        VR7      ;_uu_ 4        \        P                  ! VR	R7       RRR4       R#   + '       g   i     LY; i  + '       g   i     R# ; i)
r   r   r   gH=6?r   r   r   Nr   r   )	r   rg   r2   r`   r  r^   r   r   r   rh   ri   rk   r  s   &&  rr   test_sem_nan_policyTestSEM.test_sem_nan_policy_  s    IIcNvv!		!7DV9WX1]]:W55IIaG, 60]]:W55IIaH- 65 65 655s   8C"?C5"C2	5D	r   Nr  )r   r   r   r   r  r  r   r   r   r  r  r   r  r   r   r   s   @rr   r  r  :  sV       HO[[ LM N$
: .WX	. Y	.ru   r  c                     a  ] tR tRt o ]P
                  P                  R. RO. RO3. RO. RO3.4      R 4       tR tR t	]P
                  P                  R^ ^.4      R 4       t
]P
                  P                  R^ ^.4      R 4       t]! R	R
R7      R 4       t]P
                  P                  R4      ]P
                  P                  R4      R 4       4       t]P
                  P                  R4      R 4       tRtV tR# )TestZmapil  r  c                    VP                  V4      VP                  V4      r!WP                  V4      ,
          VP                  V^ R7      ,          p\        P                  ! W4      p\        WT4       R# )r~   rw   N)r^   rc   r   r`   zmapr2   )rh   rk   rl   ri   r  rm  s   &&&&  rr   	test_zmapTestZmap.test_zmapo  sR     zz!}bjjm1
NbffQ1f&==JJq$ru   c                   VP                  . RO. RO. RO.4      pRpRpR	p\        P                  ! W"^ R7      p\        P                  ! W"^R7      pV) V) ^,          V) ^,          R .R WU) ^,          V.W5) ^,          WS) ..p. R
OV) V) V) R.. RO.p	VP                  V4      pVP                  V	4      p	\        Wh4       \        Wy4       R# r   rC   Nr   r   r   r   )r   r   r   r   )r   r   r   r   .!	?gLXz?3Ey?;f?rQ  rQ  r   r   )r   rQ  r   rQ  r^   r`   r  r2   
rh   ri   rk   t1t2t3z0rZ  z0_expectedz1_expecteds
   &&        rr   test_zmap_axisTestZmap.test_zmap_axis|  s    JJ,,,. / ZZ1%ZZ1%bSURCE3/RQ+CE2s+- .bS2#w/-/ jj-jj-((ru   c                   VP                  . RO. RO.4      p\        P                  ! W"^^R7      pVP                  . RO4      R,          pVP                  . RO4      R,          p\        VR	,          V4       \        VR
,          V4       R# r   )rD   r  Nr  )r   r   r   r   )      r  r6   r6   g3Ey?)      r  r6   r7   g4?r~   r   r\   r   r  rh   ri   rk   rm  r  r  s   &&    rr   test_zmap_ddofTestZmap.test_zmap_ddof  ss    JJ,,. / JJq!!,jj!78*Ejj!78D$-$-ru   r  c                   VP                  RR^\        P                  .4      pVP                  RR^^^\        P                  .4      p\        P                  ! W4VRR7      p\        P                  ! W4R,          VR7      p\        WV4       R# )	r   r   r  r   :Nr|   Nr  Nr[   rZ   )r^   r   rg   r`   r  r2   )rh   r  ri   scorescomparerm  r   s   &&&    rr   test_zmap_nan_policy_omit"TestZmap.test_zmap_nan_policy_omit  sn     RQ/0**b"aB78JJvTfE jj48ru   c                   VP                  VP                  RR4      R4      p\        P                   ! \        P                  ! R^^4      R4      p\        P                  VR	&   \        P                  VR
&   \        P                  VR&   VP                  VR,          \        P                  ! VR,          4      ( ,          4      pVP                  VR,          \        P                  ! VR,          4      ( ,          4      pVP                  V4      p\        P                  ! W4R^VR7      p\        P                  ! VR,          VVR7      p\        P                  ! VR,          VVR7      p	VP                  W34      p
\        Wz4       R# )r=   r   r   )r   rD   r  r  N      rJ   rZ   r  )r~   r   r}   r  r  r  )rb   r   r   rA  rg   r^   rx  r`   r  stackr2   )rh   r  ri   r  r  compare_0_notnacompare_1_notnarm  r  r  r  s   &&&        rr   #test_zmap_nan_policy_omit_with_axis,TestZmap.test_zmap_nan_policy_omit_with_axis  s   BIIdC0':**R[[Q3W= **WT]BHHWT]4K3K%LM**WT]BHHWT]4K3K%LM**W%JJv6Mzz&,#%zz&,#%88TL)$ru   TrC  r   c                    VP                  . RO4      pVP                  RR^^^VP                  .4      p\        P                  ! \        RR7      ;_uu_ 4        \
        P                  ! W#RR7       RRR4       R#   + '       g   i     R# ; i)	r\   zinput contains nanr   r   r   Nr  r  r[   )r^   rg   r   r   r   r`   r  )rh   ri   r  r  s   &&  rr   test_zmap_nan_policy_raise#TestZmap.test_zmap_nan_policy_raise  s_    I&**b"aB78]]:-ABBJJv7; CBBBs   A<<B	r  r  c                `   VP                  ^4      pVP                  ^4      pVP                  VP                  ) VP                  VP                  .4      p\        \        RVR7      ;_uu_ 4        \        P                  ! W#4      pRRR4       \        XV4       R#   + '       g   i     L; i)r   Precision loss occurred...r^  N)
r   r  r^   r   rg   r"   r   r`   r  r3   )rh   ri   r  r  r   r   s   &&    rr   r  TestZmap.test_degenerate_input  sz     1''!*jj266'26626623/KPRSS**V-C TS! TSs   0BB-	c                    \         P                  ! VP                  . RO4      VP                  . RO4      4      pVP                  . RO4      p\        W#4       R# )r\   Nr=  )r\                 ?rZ   r  )y      ?        y       @        y      @        y      @        )r`   r  r^   r2   )rh   ri   r   r   s   &&  rr   test_complex_gh22404TestZmap.test_complex_gh22404  s<    jjL12::>N3OPjj9:!ru   r   Nr  r  )r   r   r   r   r   )r   r   r   r   r   r   r   r  r  r  r  r  r   r  r   r  r  r   r   r   s   @rr   r  r  l  s     [[
,	-
,	-	/
%
%)2
. [[VaV,	  -	  [[VaV,% -%( .MN< O< [[ WX[[ VW" X Y" [[ WX" Y"ru   r  c                   
  a  ] tR tRt o R tR tR tR tR tR t	]
! RR	R
7      R 4       tR t]P                  P                  R4      R 4       t]P                  P                  R4      R 4       t]P                  P                  R4      R 4       t]P                  P                  R4      R 4       t]P                  P)                  R]P,                  ! . 4      ]P.                  ! R4      .4      R 4       t]R 4       tRtV tR# )
TestZscorei  c                    \         P                  ! VP                  . RO4      4      p. ROp\        W!P                  V4      4       R# )r\   Nr=  )gOT\wg%ܿg%?gOT\w?)r`   zscorer^   r2   )rh   ri   rl   desireds   &&  rr   test_zscoreTestZscore.test_zscore  s3     LLL126::g./ru   c                   VP                  . RO. RO. RO.4      pRpRpR	p\        P                  ! V^ R7      p\        P                  ! V^R7      pV) V) ^,          V) ^,          R .R WU) ^,          V.W5) ^,          WS) ..p. R
OV) V) V) R.. RO.p	\        WaP                  V4      4       \        WqP                  V	4      4       R# r  r^   r`   r  r2   r  s
   &&        rr   test_zscore_axisTestZscore.test_zscore_axis  s    JJ,,,. / \\!!$\\!!$bSURCE3/RQ+CE2s+- .bS2#v.-/ 	JJ{34JJ{34ru   c                   VP                  . RO. RO.4      p\        P                  ! V^^R7      pVP                  . RO4      R,          pVP                  . RO4      R,          p\        VR	,          V4       \        VR
,          V4       R# r  r  r  s   &&    rr   test_zscore_ddofTestZscore.test_zscore_ddof  ss    JJ,,. / LL+jj!78*Ejj!78+F$-$-ru   c                    VP                  ^^\        P                  ^^.4      p\        P                  ! VRR7      p\        W1P                  VP                  VP                  4      4       R# )r\   r  r   N)r^   r   rg   r`   r  r3   r   r  rh   ri   rk   rm  s   &&  rr   test_zscore_nan_propagate$TestZscore.test_zscore_nan_propagate  sJ    JJ1bffa+,LL{377177BFF34ru   c                    VP                  ^^VP                  ^^.4      p\        P                  ! VRR7      pVP                  RRVP                  RR.4      p\	        W44       R# )r\   r   r   gIHb=?gIHb=?NgIHb=gIHb=)r^   rg   r`   r  r2   rh   ri   rk   rm  r  s   &&   rr   test_zscore_nan_omitTestZscore.test_zscore_nan_omit  s_    JJ1bffa+,LLv.::22!vv11	! " 	$ru   c                $   VP                  VP                  R RRRR.4      p\        P                  ! V^RR7      pVP	                  VP                  VP                  .4      \        P                  ! VR,          ^R7      .4      p\        W44       R	# )
r   r   r=   r   r   r   r  r  r  N)r^   rg   r`   r  r  r2   r  s   &&   rr   test_zscore_nan_omit_with_ddof)TestZscore.test_zscore_nan_omit_with_ddof  sl    JJS#sC89LLv699bjj"&&2ELL2Q4OPQ$ru   TrC  r   c                    VP                  ^^VP                  ^^.4      p\        P                  ! \        RR7      ;_uu_ 4        \
        P                  ! VRR7       RRR4       R#   + '       g   i     R# ; i)r\   zThe input contains nan...r   r   r   N)r^   rg   r   r   r   r`   r  r   s   && rr   test_zscore_nan_raise TestZscore.test_zscore_nan_raise  sO    JJ1bffa+,]]:-HIILLw/ JIII   A((A9	c                0   VP                  R.^,          4      p\        \        RVR7      ;_uu_ 4        \        P                  ! V4      pRRR4       \        XVP                  VP                  VP                  4      4       R#   + '       g   i     LB; i)gʡE?r  r^  NgʡE)	r^   r"   r   r`   r  r3   r   r  rg   r  s   &&  rr   test_zscore_constant_input_1d(TestZscore.test_zscore_constant_input_1d%  sa    JJx!|$/KPRSSQA T277177BFF34 TSs   BB	r  c                   VP                  . RO. RO.4      p\        \        RVR7      ;_uu_ 4        \        P                  ! V^ R7      pRRR4       \        XVP                  VP                  RRR.VP                  RRR..4      4       \        \        RVR7      ;_uu_ 4        \        P                  ! V^R7      pRRR4       \        XVP                  VP                  VP                  VP                  VP                  VP                  .4      \        P                  ! VR	,          4      .4      4       \        P                  ! VRR7      p\        WQP                  \        P                  ! VP                  VR
4      4      VP                  4      4       VP                  R4      p\        \        RVR7      ;_uu_ 4        \        P                  ! VRR7      pRRR4       \        WQP                  WaP                  4      4       R#   + '       g   i     EL; i  + '       g   i     ELk; i  + '       g   i     L^; i)r   r  r^  rC   Nr   r   r   r   r   r   r  r        *@rQ  r  rZ   )r   r   )r^   r"   r   r`   r  r2   rg   r3   r  rb   r  r  	full_like)rh   ri   rk   r  rZ  rm  rl   s   &&     rr   test_zscore_constant_input_2d(TestZscore.test_zscore_constant_input_2d+  s   JJ002 3/KPRSSaa(B TBJJtT(B)+c3(?(A B 	C /KPRSSaa(B TBHHbjj"&&"&&"&&"&&1Q&R&+ll1T7&;&= > 	? LL&::ell2::a3G&H!''RSGGFO/KPRSSQT*A T<<6623! TSS
 TSS TSs#   H&H!H5H	!H2	5I	c                z   VP                  . R	OR RRVP                  .R RVP                  R ..4      pR
pRp\        \        RVR7      ;_uu_ 4        \        P
                  ! VR^ R7      pRRR4       \        XVP                  VP                  V) RVP                  .VP                  ^ RVP                  .VP                  W1P                  VP                  ..4      4       \        \        RVR7      ;_uu_ 4        \        P
                  ! VR^R7      pRRR4       \        XVP                  VP                  VP                  VP                  VP                  .V) ^ W1P                  .V) ^,          WAP                  V) ^,          ..4      4       R#   + '       g   i     EL4; i  + '       g   i     L; i)r   r  r  r  r^  r   r   rD   Nr   r  r  r  rQ  )r^   rg   r"   r   r`   r  r2   )rh   ri   rk   r(  s2r  rZ  s   &&     rr   -test_zscore_constant_input_2d_nan_policy_omit8TestZscore.test_zscore_constant_input_2d_nan_policy_omitA  sm   JJ0tT2662tRVVT24 5 /KPRSSaF;B TBJJ!T266(B)+C(@)+FFBFF(C(E F 	G /KPRSSaF;B TBJJ(H*+Q66(:*,QFFRCE(B(D E 	F TSS TSs   F=F*F'	*F:	c           	        VP                  \        P                  \        P                  \        P                  \        P                  .. RO.4      p\        P                  ! VR^R7      p\        W1P                  \        P                  \        P                  \        P                  \        P                  .. RO.4      4       R# )r   r   r  N)r   r   r  r  r  )r^   r   rg   r`   r  r2   r  s   &&  rr   test_zscore_2d_all_nan_row%TestZscore.test_zscore_2d_all_nan_rowU  s     JJ802 3LLvA6::'G'='? @ 	Aru   c                    VP                  RVP                  4      p\        P                  ! VRRR7      p\	        W24       R# )rJ   r   Nr  rV  )r   rg   r`   r  r3   )rh   ri   rl   rm  s   &&  rr   test_zscore_2d_all_nan!TestZscore.test_zscore_2d_all_nan^  s2     GGFBFF#LLvD9ru   rk   c                j    VP                  V4      p\        P                  ! V4      p\        W14       R # r#  )r^   r`   r  r3   )rh   rk   ri   rm  s   &&& rr   test_zscore_empty_input"TestZscore.test_zscore_empty_inpute  s$    JJqMLLOru   c                ,   \         P                  P                  R 4      pVP                  ^
4      p\         P                  ! V4      p\         P
                  P                  W44      pRVP                  ^ &   \        P                  ! VR,          4      p\         P                  ! \         P                  ! V4      4      '       d   Q h\        P                  ! V4      p\        VR,          V4       \        P                  ! VRR7      p\        VR,          V4       V^,          VR&   \        P                  ! \        RR7      ;_uu_ 4        \        P                  ! V4      pRRR4       \!        VR,          \         P"                  4       \        P                  ! \        RR7      ;_uu_ 4        \        P                  ! VRR7      pRRR4       \!        VR,          \         P"                  4       R#   + '       g   i     L; i  + '       g   i     LF; i)i_ Tr  NrC   r  r   )r   r  r  r  
zeros_likemamasked_arraymaskr`   r  anyrx  r
   r   r  r   r   rg   )rh   ri   r  rk   r  rl   r   r   s   &&      rr   $test_zscore_masked_element_0_gh19039/TestZscore.test_zscore_masked_element_0_gh19039k  sQ    ii##G,#}}QEEq'q	ll1R5!66"((3-((((ll1oB%ll14(B%!"\\.0LMM,,q/C NSWbff%\\.0LMM,,qt,C NSWbff% NM NMs   G0,H0H 	H	r   N)r   r~   r|   )r   r   r   r   r  r  r  r  r  r  r   r  r  r   r   r   r  r  r  r  r   r   r   rX  r  r    r  r   r   r   s   @rr   r  r    s-    05.
.5
%% .MN0 O0
5 [[ VW4 X4* [[ VWF XF& [[ VWA XA [[ VW X [[S288B<)1D"EF G
 & &ru   r  c                   6   a  ] tR tRt o R t]R 4       tRtV tR# )TestGZscorei  c           	        \         P                  ! . RO4      p\        P                  ! VP                  V4      4      p\         P                  ! V\        P
                  ! V4      ,          4      \         P                  ! \        P                  ! V^ R7      4      ,          p\        W1P                  WAP                  R4      P                  R7      4       R# )r\   r  r   r;   Nr=  )	r   r^   r`   gzscoreloggmeangstdr2   r<   )rh   ri   rk   rm  r  s   &&   rr   test_gzscore_normal_array%TestGZscore.test_gzscore_normal_array  s{    JJ|$MM"**Q-(&&U[[^+,rvvejj6K/LL::gZZ^5I5I:JKru   c                   \         P                  ! . RO4      p. ROp\         P                  P                  WR7      p\        P
                  ! V4      pRR\         P                  RR.p\         P                  P                  WRR7      p\        VP                  4       VP                  4       4       \        VP                  VP                  4       \        V\         P                  P                  4      '       g   Q hR# )	r\   r  g/A?g(/?N)r\   rJ   rZ   r   r   )r~   r~   r\   r~   r~   g0jg`ȿ)r   r   r	  r
  r`   r  r   r
   
compressedr  rC  MaskedArray)rh   rk   r  mxrm  r  s   &     rr   test_gzscore_masked_array%TestGZscore.test_gzscore_masked_array  s    HH%&UU-MM"#_bffn"$%%$$W$8(:(:(<=-!RUU..////ru   r   N)	r   r   r   r   r  r    r  r   r   r   s   @rr   r  r    s     L 
0 
0ru   r  c                     a  ] tR tRt o R tR t]! RRR7      R 4       tR t]! RRR7      R	 4       t	]
P                  P                  R
. RO4      R 4       t]
P                  P                  RR]P                  RR.3R]P                  ]P                  R.3.4      R 4       t]
P                  P                  R^. RO3R.4      R 4       tR tRtV tR# )TestMedianAbsDeviationi  c                r    R R RRRRRRRRRRR	R	R	R
RRRRRRR\         P                  .V n        . ROV n        R# )皙@r  ry   皙@r  333333@=
ףp=@r  (\
@r  r  r  皙@)\(@Q@N)r"  r"  r  r  ry   r#  r  r$  r%  r%  r  r&  r  r  r  r  r  r'  r'  r'  r'  r(  r)  g33333<@)r   rg   dat_nandatrs  s   &rr   rt  #TestMedianAbsDeviation.setup_method  sL    dCc3SdD$S#sS#sCtRVVE6ru   c                \   \        \        P                  ! VP                  V P                  4      R R7      VP                  R4      4       VP                  VP                  V P                  4      R4      p\        P                  ! V^ R7      pVP                  . RO4      p\        W44       R # )NrC   gQ?)r   r   )gףp=
?r6   g?rz  )r2   r`   median_abs_deviationr^   r+  rb   )rh   ri   r+  madmad_expecteds   &&   rr   test_median_abs_deviation0TestMedianAbsDeviation.test_median_abs_deviation  sx    222::dhh3GdS

5)	+jjDHH-v6((15zz"9:*ru   	jax.numpyzlazy -> no nan_policy="omit"r  c                    \         P                  ! VP                  V P                  4      R R7      p\	        W!P                  R4      4       R# )r   r   g(\?N)r`   r.  r^   r*  r2   )rh   ri   r/  s   && rr   test_mad_nan_omit(TestMedianAbsDeviation.test_mad_nan_omit  s2    ((DLL)AfUZZ-.ru   c                    VP                  R RRR\        P                  .. RO.4      p\        P                  ! V^R7      p\        W1P                  \        P                  R.4      4       R# )r   r   r   r   rC   N)r   r   r=   r   r   )r^   r   rg   r`   r.  r2   )rh   ri   rk   r/  s   &&  rr   test_axis_and_nan(TestMedianAbsDeviation.test_axis_and_nan  sT    JJc3RVV413 4((3ZZ67ru   c           	         VP                  ^^^^^c\        P                  \        P                  .4      p\        P
                  ! VRR7      p\        W1P                  R4      4       R# )r\   r   r   r   N)r^   r   rg   r   r`   r.  r2   )rh   ri   rm  r/  s   &&  rr   test_nan_policy_omit_with_inf4TestMedianAbsDeviation.test_nan_policy_omit_with_inf  sI    JJ1aB78((v>ZZ_-ru   rD   Nc           
     t   VP                  R4      pVR9   d   \        \        R\        R7      M\        P
                  ! 4       pV;_uu_ 4        \        P                  ! W1R7      pRRR4       \        XVP                  VP                  W1R7      VP                  R7      4       R#   + '       g   i     LI; i)r   N	too smallr^  rC   
fill_value)r   r~   r   >   Nr\   )rX  r"   r   r   
contextlibnullcontextr`   r.  r2   r  r]  rg   )rh   rD   ri   rk   contextr/  s   &&&   rr   test_size_zero_with_axis/TestMedianAbsDeviation.test_size_zero_with_axis  s    HHYi' 1L-7-C-C-E 	W,,Q:C R\\"&&&*>266\RS Ws   B''B7	znan_policy, expectedr   r7   r  c           	        VR 8X  d(   \        V4      '       g   \        P                  ! R4       VP                  \        P
                  \        P
                  \        P
                  \        P
                  \        P
                  \        P
                  .^^^^\        P
                  \        P
                  .. RO.4      pVR 8X  d   \        \        R\        R7      M\        P                  ! 4       pV;_uu_ 4        \        P                  ! WA^R7      p\        WcP                  V4      4       RRR4       R#   + '       g   i     R# ; i)r   z8nan_policy='omit' with n-d input only supported by NumPyr>  r^  r  N)r|   r   rO   rN   rN   rT   )r$   r   r  r^   r   rg   r"   r   rA  rB  r`   r.  r2   )rh   r   r  ri   rk   rC  r/  s   &&&&   rr   test_nan_policy_with_axis0TestMedianAbsDeviation.test_nan_policy_with_axis  s     vhrllKKRSJJHAq!RVVRVV4+- .
 !F* 1L0:0F0F0H 	W,,QANCCH!56 WWWs   =3D::E	zaxis, expectedc                    \         P                  ! ^^^^	\         P                  .. RO. RO.4      p\        P                  ! V\         P
                  RVR7      p\        WBRRR7       R# )r\   r   )centerr   rD   rE  r  N)r~   r\   r\   r\   rV   )rK  rK  r  r  )r   r   rg   r`   r.  rc   r2   )rh   rD   r  rk   r/  s   &&&  rr   test_center_mean_with_nan0TestMedianAbsDeviation.test_center_mean_with_nan  sY     HHq!Q266*&-/ 0 ((277v.24E>ru   c                    \         P                  ! \        R R7      ;_uu_ 4        \        P                  ! VP                  . RO4      ^cR7       RRR4       R#   + '       g   i     R# ; i)callabler   rJ  N)r\   rJ   r   r|   )r   r   rg  r`   r.  r^   rh   ri   s   &&rr   test_center_not_callable/TestMedianAbsDeviation.test_center_not_callable  s;    ]]9J77&&rzz,'?K 8777   *AA*	)r+  r*  )r~   r\   rJ   N)ry   r   r  )NrF   )r   r   r   r   rt  r1  r   r5  r8  r;  r   r   r   rD  r   rg   rG  rL  rR  r   r   r   s   @rr   r   r     s    6+ k*HI/ J/8 k*HI. J.
 [[V_5T 6T [[3%S'9:*RVVRVVS,ABDE7E7 [[- "23[AC?C?L Lru   r   c                     a  ] tR tRt o ]P
                  P                  R. R O4      R 4       tR t]	! RRR7      R	 4       t
]P
                  P                  R
. ]P                  ! ^ 4      .4      R 4       tR tR tR tR tR tR tR t]	! RRR7      R 4       t]	! RRR7      R 4       tR tR tR t]	! RRR7      R 4       tR tR tRtV tR# )!TestIQRi  r<   Nc           	     4   \         P                  ! ^4      R,          p\         P                  P                  V4       Vf   \	        V4      M
\        W!4      p\        \        P                  ! VP                  W1R7      4      VP                  RVR7      4       R# )rK   r6   Nr;   r  )
r   r   r  r&  r&   rl  r3   r`   iqrr^   rh   r<   ri   rk   s   &&& rr   rY  TestIQR.test_basic  sj    IIaL3
		!(- $72;M		"**Q*"<=

4u
5	7ru   c                t   VP                  R4      p\        P                  ! V4       \        P                  ! VR4       \        P                  ! V^4       \        P                  ! VR4       \        P                  ! VRR4       \        P                  ! VRRR4       \        P                  ! VRR	RR4       R# )
r|   Nr   r7   r  r  rU  )rT   Z   )r$  r  r  r1  r  r`   rX  rh   ri   rj  s   && rr   test_apiTestIQR.test_api  sy    GGFO		!		!T		!Q		!V		!T8$		!T8S)		!T8S+6ru   r3  r   r  c           	         VP                  R4      p\        P                  ! VRR	RRR4       \        P                  ! VRR
RRRR4       R# )r|   Nr  r   linearr   r  Tr  )r  r  r]  gٿr^  r_  s   && rr   test_api_eagerTestIQR.test_api_eager  s=    GGFO		!T8XwA		!T8T67DAru   rk   c           	        \        \        \        VR 7      ;_uu_ 4        \        \        P
                  ! VP                  V4      4      VP                  VP                  4      4       RRR4       R#   + '       g   i     R# ; ir^  N)r"   r   r   r3   r`   rX  r^   rg   )rh   rk   ri   s   &&&rr   r  TestIQR.test_empty  sK    +3HRPPEIIbjjm4bjj6HI QPPPs   A	A22B	c                B   VP                  R4      pVP                  R4      p\        \        P                  ! V4      V4       \        \        P                  ! V^ R7      VP                  ^4      4       \        \        P                  ! V^R7      VP                  ^4      4       \        \        P                  ! VRR7      V4       \        \        P                  ! VRR7      V4       \        \        P                  ! VRR7      V4       \        \        P                  ! VRR7      V4       \        \        P                  ! VRR7      V4       VP                  R4      VP                  R	4      ,          p\        \        P                  ! V^ R7      VP                  R4      4       \        \        P                  ! V^R7      VP                  R4      4       \        \        P                  ! V^R7      VP                  RR
4      4       \        \        P                  ! VRR7      VP                  ^4      4       \        \        P                  ! VRR7      VP                  ^R4      4       \        \        P                  ! VRR7      VP                  ^R4      4       R# )rO   r   rC   rc  interpolationmidpointnearestr  r  r   ry   r   N)rO   r   r   r|   r   )r|   r   )r   r   r  rU  r~   rJ   r\   rJ   )r  r^   r3   r`   rX  rX  r   r   )rh   ri   rk   r3  rl   s   &&   rr   test_constantTestIQR.test_constant  s   GGFOzz#		!d+		!!,bhhqk:		!!,bhhqk:		!8<dC		!:>E		!9=tD		!7;TB		!8<dC GGI2.		!!,bhhv.>?		!!,bhhv.>?		!!,bggfc.BC		!&1288A;?		!&12771b>B		!&12771b>Bru   c                d   VP                  R 4      R,           p\        \        P                  ! V^ ,          4      VP	                  R4      4       \        \        P                  ! V4      VP	                  R4      4       \        \        P                  ! VRR7      VP	                  R.4      4       R# )r   r   r   Tr[  N)r   r3   r`   rX  r^   r   s   && rr   test_scalarlikeTestIQR.test_scalarlike  sl    IIbMC		!A$C9		!bjjo6		!d3RZZ5FGru   c                2   VP                  VP                  ^4      R4      p\        \        P                  ! V4      VP                  R4      4       \        \        P                  ! V^ R7      VP                  ^R4      4       \        \        P                  ! V^R7      VP                  ^R4      4       \        \        P                  ! VRR7      VP                  R4      4       \        \        P                  ! VRR7      VP                  R4      4       R# )	   r   rC   r=   r   Nr  rU  r  )rb   r   r3   r`   rX  r^   r   r   s   && rr   test_2DTestIQR.test_2D$  s    JJryy}f-		!bjjo6		!!,bggan=		!!,bggan=		!&12::c?C		!&12::c?Cru   c           	     d	   \         P                  P                  RR7      p\         P                  ! V.^
,          4      pVP	                  V4      VP	                  V4      r2VP                  \        P                  ! V4      R	4      p\        \        P                  ! VR
R7      V4       VP                  VR^ 4      p\        \        P                  ! VRR7      V4       \        V^ ^VR7      p\        \        P                  ! VRR7      V4       \        V^ ^VR7      p\        \        P                  ! VRR7      \        P                  ! VRR7      4       \        \        P                  ! VRR7      \        P                  ! V^ R7      4       \         P                  ! R4      p\         P                  P                  V4       VP                  R4      pVP	                  V4      p\        \        P                  ! VRR7      ^ ,          \        P                  ! \        VR,          4      4      4       \        \        P                  ! VRR7      ^,          \        P                  ! \        VR,          4      4      4       \        \        P                  ! VRR7      ^,          \        P                  ! \        VR,          4      4      4       \        \        P                  ! VRR7      ^,          \        P                  ! \        VR,          4      4      4       \        \        P                  ! VRR7      R,          \        P                  ! \        VR,          4      4      4       \        \        P                  ! VRR7      R,          \        P                  ! \        VR,          4      4      4       \        \        P                  ! VRR7      R,          \        P                  ! \        VR,          4      4      4       \        P                   ! \"        RR7      ;_uu_ 4        \        P                  ! V^R7       RRR4       \        P                   ! \$        RR7      ;_uu_ 4        \        P                  ! VR R7       RRR4       R#   + '       g   i     LW; i  + '       g   i     R# ; i)!G   r  rC   r=  Nz`axis` is out of bounds...r   z`axis` must contain only...)r{  r  )rT   rU  rZ   r  ro  r  r  i  r   r|   rO   rU   )r   r   r   r~   r~   r\   r   )r   r   r\   r   )r   r\   r  )r   r   rJ   r   )r   r\   rJ   )rJ   r   r   r   r   rJ   )rJ   r\   r   r   )r\   re  )rJ   r   r   r\   r\   r   rJ   rJ   )rJ   r   rJ   r   r  )r   r  r  dstackr^   broadcast_tor`   rX  r3   moveaxisr+   r   r&  rb   r*   r   r   r   r   )rh   ri   ork   qrj  s   &&    rr   r	  TestIQR.test_axis,  s    II(+IIqcBhzz!}bjjm1OOEIIaL%0		!&115KK2q!		!&115!Q2&		!&115!Q2&		!)4		!$/	1		!$/		!!,	. IIn% 			!IIm$JJqM		!)4Q7		(1X;"78	:		!)4Q7		(1X;"78	:		!*5a8		(1X;"78	:		!)4Q7		(1W:"67	9		!&1$7		(1X;"78	:		!'248		(1Z="9:	<		!&1$7		(1Y<"89	; ]]9,HIIIIaa  J]]:-JKKIIaf% LK JIKKKs   #R(RR	R/	c                   VP                  ^4      p\        \        P                  ! V4      VP	                  R4      4       \        \        P                  ! VR
R7      VP	                  R4      4       \        \        P                  ! VRR7      VP	                  R4      4       \        \        P                  ! VRR7      VP	                  R4      4       Rp\
        P                  ! \        VR7      ;_uu_ 4        \        P                  ! VRR7       RRR4       Rp\
        P                  ! \        VR7      ;_uu_ 4        \        P                  ! V\        P                  ^3R7       RRR4       R	p\
        P                  ! \        VR7      ;_uu_ 4        \        P                  ! VRR7       RRR4       R#   + '       g   i     L; i  + '       g   i     Ll; i  + '       g   i     R# ; i)r|   r   r  ry   g?z2Elements of `rng` must be in the range \[0, 100\].r   Nz`rng` must not contain NaNs.z%`rng` must be a two element sequence.)r  g     U@)g      )@r1  rT   r  )r~   e   )r~   r  <   )r   r2   r`   rX  r^   r   r   r   r   rg   rg  r  s   &&  rr   test_rngTestIQR.test_rngY  s(   IIaL		!bjjn5		!4bjjoF		!4bjjoF		!2BJJsODG]]:W55IIaX& 6 1]]:W55IIabffb\* 6 :]]9G44IIa[) 54 65 65 544s$   :F;)GG!;G	G	!G2	c                   VP                  ^4      pVP                  ^4      p\        \        P                  ! V4      VP	                  R4      4       \        \        P                  ! V4      VP	                  R4      4       \        \        P                  ! VRR7      VP	                  R4      4       \        \        P                  ! VRR7      VP	                  R4      4       \        \        P                  ! VRR7      VP	                  R4      4       \        \        P                  ! VRRR7      VP	                  R4      4       \        \        P                  ! VRR7      VP	                  R4      4       \        \        P                  ! VRR7      VP	                  R4      4       \        \        P                  ! VRRR7      VP	                  R4      4       \        \        P                  ! VRR7      VP	                  R4      4       \        \        P                  ! VR	R7      VP	                  R4      4       \        \        P                  ! VR	R7      VP	                  R
4      4       \        \        P                  ! VRR7      VP	                  R4      4       \        \        P                  ! VRRR7      VP	                  R4      4       \        \        P                  ! VRR7      VP	                  R4      4       R F  p\        P                  ! W4R7       K  	  \
        P                  ! \        RR7      ;_uu_ 4        \        P                  ! VRR7       RRR4       R#   + '       g   i     R# ; i)r|   r   r7   rc  rj  r  )r  rk  r   r  rm  r   rl  ry   z`method` must be one of...r   r   N)r  P   )inverted_cdfaveraged_inverted_cdfclosest_observationinterpolated_inverted_cdfhazenweibullmedian_unbiasednormal_unbiased)r   r3   r`   rX  r^   r   r   r   )rh   ri   rk   rl   r  s   &&   rr   test_interpolationTestIQR.test_interpolationl  s$   IIaLIIaL		!bjjn5		!bjjo6		!8<bjjnM		!8<bjjoN		!8<bjjnM		!J

2	(		!8<bjjnM		!7;RZZ^L		!I

2	(		!7;RZZ^L		!9=rzz"~N		!9=rzz"~N		!:>

2O		!L

3	)		!:>

2O*F IIa.	* ]]:-IJJIIax0 KJJJs   M((M9	c                   VP                  R4      p\        \        P                  ! VRRR7      P                  R4       \        \        P                  ! V^RR7      P                  R4       \        \        P                  ! VR	RR7      P                  R
4       \        \        P                  ! VRRR7      P                  R4       \        \        P                  ! VRRR7      P                  R4       \        \        P                  ! VRRR7      P                  R4       \        \        P                  ! VRRR7      P                  R4       \        \        P                  ! VRRR7      P                  R4       \        \        P                  ! V^RR7      P                  R4       \        \        P                  ! VR	RR7      P                  R4       \        \        P                  ! VRRR7      P                  R4       \        \        P                  ! VRRR7      P                  R4       \        \        P                  ! VRRR7      P                  R4       \        \        P                  ! VRRR7      P                  R4       R# )r   NFr!  r[  Tr|  r   )r   r|   rU   rU  )rO   rU   )r~   r   )r|   rO   r  r  r~   r\   rJ   r   r}  rz   rX  )r   r|   r\   rU   )r\   r\   rO   rU   )r\   r|   rO   r\   )r   r\   rO   rU   )r\   r\   rO   r\   )r  r   r`   rX  r  r   s   && rr   test_keepdimsTestIQR.test_keepdims  s   GGM"UYYqte<BBBGUYYqq59??LUYYqv>DDgNUYYqv>DDfMUYYqte<BBJOUYYq,?EErJUYYqy5AGGNUYYqtd;AA<PUYYqq48>>NUYYqv=CC]SUYYqv=CC\RUYYqtd;AA=QUYYq,>DDlSUYYqy4@FFUru   z(lazy -> reduced nan_policy capabilitiesyc           
        VP                  VP                  R 4      R4      p\        \        P                  ! VRR7      VP                  R4      4       \        \        P                  ! VRR7      VP                  R4      4       \        \        P                  ! VRR7      VP                  R4      4       \        P                  ! V4      R,          P                  VP                  4      p\        \        P                  ! VRR7      VP                  VP                  4      4       \        \        P                  ! V^ RR7      VP                  ^^VP                  ^^.4      4       \        \        P                  ! V^RR7      VP                  ^VP                  ^.4      4       \        \        P                  ! VRR7      VP                  R4      4       Rp\        P                  ! \        VR	7      ;_uu_ 4        \        P                  ! VRR7       R
R
R
4       Rp\        P                  ! \        VR	7      ;_uu_ 4        \        P                  ! VRR7       R
R
R
4       R
#   + '       g   i     LY; i  + '       g   i     R
# ; i)      .@r  r   r   r   r   r        @r   r   Nr   barfoodr  rp  )rb   r   r3   r`   rX  r^   rd   re   rf   rg   r   r   r   r  s   &&  rr   test_nanpolicyTestIQR.test_nanpolicy  s   JJryy/ 			!<bjjnM		!7BH		!8"**R.I FF1IdO'		!<

266*	,		!!D

Aq"&&!Q#78	:		!!D

Arvvq>2	4 			!7CI1]]:W55IIaG, 6 1]]:W55IIaI. 65 65
 655s   =I'I:'I7	:J	Tz=nan_policy w/ multidimensional arrays only available w/ NumPynp_onlyr   c                   VP                  VP                  R 4      R4      pVP                  VR	&   \        \        P
                  ! V^ RR7      VP                  ^R4      4       \        \        P
                  ! V^RR7      VP                  . R
O4      4       Rp\        P                  ! \        VR7      ;_uu_ 4        \        P
                  ! V^ RR7       RRR4       \        P                  ! \        VR7      ;_uu_ 4        \        P
                  ! V^RR7       RRR4       R#   + '       g   i     LX; i  + '       g   i     R# ; i)r  r   r  r=   r   r   r   Nr  rp  )rJ   ry   rJ   )rb   r   rg   r3   r`   rX  r   r^   r   r   r   r  s   &&  rr   test_nanpolicy_ndTestIQR.test_nanpolicy_nd  s     JJryy/&&$		!!?2	(		!!?

;/	1 2]]:W55IIaaG4 6]]:W55IIaaG4 65 65555s   >D(D;(D8	;E	c                n   VP                  VP                  R 4      R4      p\        \        P                  ! VRR7      VP                  R4      4       \        \        P                  ! VRR7      VP                  R	4      4       \        \        P                  ! VRR7      VP                  R4      4       R# )
r  r   r+  r   r  r   r  Nr  g=V^w@)rb   r   r3   r`   rX  r^   r2   r   s   && rr   test_scale_no_nansTestIQR.test_scale_no_nans  sr    JJryy/		!3/B@		!84bjj6OP		!3/CAru   c                   VP                  VP                  R 4      R4      p\        P                  ! V4      R,          P	                  VP
                  4      pVP                  VP
                  4      p\        \        P                  ! VRRR7      V4       \        \        P                  ! VRRR7      V4       \        \        P                  ! VRRR7      V4       R# )	r  r   r  r+  r   r  r   Nr  rp  )
rb   r   rd   re   rf   rg   r^   r3   r`   rX  )rh   ri   rk   rg   s   &&  rr   test_scale_with_nansTestIQR.test_scale_with_nans  s    JJryy/FF1IdO'jj 		!3;GM		!8LcR		!3;GMru   c                    VP                  R 4      pRp\        P                  ! \        VR7      ;_uu_ 4        \        P
                  ! VRR7       RRR4       R#   + '       g   i     R# ; i)r  z"foobar not a valid scale for `iqr`r   r   r  N)r   r   r   r   r`   rX  r  s   &&  rr   test_invalid_scaleTestIQR.test_invalid_scale  sB    IIdO6]]:W55IIax( 6555s   AA,	c                   VP                  VP                  R 4      R	4      p\        P                  ! V4      R
,          P	                  VP
                  4      p\        \        P                  ! VRRR7      VP                  R4      4       \        \        P                  ! VRRR7      VP                  R4      4       \        \        P                  ! VRRR7      VP                  R4      4       R# )r  r   r   r  r  r  r   g      @Nr  rp  g	I1=@)rb   r   rd   re   rf   rg   r3   r`   rX  r^   r2   r   s   && rr   test_scale_nan_policy_omit"TestIQR.test_scale_nan_policy_omit  s    JJryy/FF1IdO'		!36BBJJsOT		!8G

?3	5		!36BBJJtDTUru   c           	     Z   VP                  VP                  R 4      R4      p\        P                  ! V4      R	,          P	                  VP
                  4      p\        \        P                  ! V^RRR7      VP                  ^\        P
                  ^.4      4       \        \        P                  ! V^RRR7      VP                  ^\        P
                  ^.4      R,          4       \        \        P                  ! V^RRR7      VP                  ^\        P
                  ^.4      4       R# )
r  r   r  )rD   r+  r   r  gF7k?r   Nr  rp  )rb   r   rd   re   rf   rg   r3   r`   rX  r^   r   r2   r   s   && rr   test_scale_nanpolicy_ndTestIQR.test_scale_nanpolicy_nd  s    JJryy/FF1IdO'		!!3;O

Arvvq>2	4		!!8T

Arvvq>2Y>	@		!!3;O

Arvvq>2	4ru   c                    VP                  R 4      R,          p\        P                  ! VRR7      p\        P                  ! V4      p\        W44       R# )r   r6   r  N)r1  r  )r   r`   rX  r3   rh   ri   rk   r   r   s   &&   rr   test_rng_orderTestIQR.test_rng_order  s9    IIbMCiix(iil!ru   r   r  )r   r   r   r   r   r   r   rY  r`  r   rd  r   r   r  rq  rt  rx  r	  r  r  r  r  r  r  r  r  r  r  r  r   r   r   s   @rr   rV  rV    s)     [[W&BC7 D77 k*STB UB
 [[S2ryy|"45J 6JC.HD+&Z*&$1LV& k*TU/ V/8 dNP5P5BN) k*STV UV
4" "ru   rV  c            
       N  a  ] tR tRt o Rt. ROtRt]P                  P                  R4      t
]
P                  ^*4      t]P                  P                  R^
R .4      ]P                  P                  R]! ]! R!R"4      4      4      R 4       4       t]P                  P                  R^
R .4      ]P                  P                  R	R"4      R
 4       4       tR t]! RRR7      R 4       tR tR t]P                  P                  R. R#O4      ]P                  P                  RR$R%.4      R 4       4       tR tR tR tR t]P                  P                  R. R!O4      ]P                  P                  R. R&O4      ]P                  P                  RRR.4      R 4       4       4       tRtV tR# )'TestMomentsi  aC  
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.
r   l   Jf r  zm, cNc                r   \         P                  P                  R 4      pVP                  VP                  VR7      4      p\        P
                  ! WbVR7      pVf   VP                  V^ R7      MTpVP                  Wc,
          V,          ^ R7      VP                  ^ ,          ,          p\        WxRR7       R# )l   rHZ r  rP  NrC   缉ؗҜ<rF  )
r   r  r  r^   r`   momentrc   r]  r  r2   )	rh   r  r  ri  ri   r  rk   r   r   s	   &&&&&    rr    test_moment_center_scalar_moment,TestMoments.test_moment_center_scalar_moment  s     ii##$78JJszztz,-ll1*"#)BGGAAGffaeaZaf(3u-ru   ri  c                r   \         P                  P                  R 4      pVP                  VP                  VR7      4      p. ROp\        P
                  ! WVVR7      pTP                  V Uu. uF.  p\        P
                  ! WXVR7      VP                  R3,          NK0  	  up4      p	\        Wy4       R# u upi )l   %~.}~= r  rP  .Nr  )	r   r  r  r^   r`   r  r  newaxisr2   )
rh   r  ri  ri   r  rk   r  r   r  r   s
   &&&&      rr   test_moment_center_array_moment+TestMoments.test_moment_center_array_moment  s     ii##$78JJszztz,-ll1*iiPQRPQ1a15bjj#oFFPQRS! Ss   ,4B4c                   VP                  V P                  4      p\        P                  ! VP                  V P                  4      4      p\        W1P                  R 4      4       \        P                  ! V^ 4      p\        W1P                  R4      4       \        P                  ! V^4      p\        W1P                  R 4      4       \        P                  ! V^4      p\        W1P                  R4      4       \        P                  ! V^4      p\        W1P                  R 4      4       \        P                  ! V^4      p\        W1P                  R4      4       \        P                  ! V. RO4      p\        W1P                  . RO4      4       \        P                  ! VR 4      p\        W1P                  R4      4       \        P                  ! V. RO4      p\        W1P                  . RO4      4       R# )r   r         ?     @Nr=  )r~   r  r~   r  )r   rJ   r   r   )r^   r  r`   r  r  r2   )rh   ri   r  rl   s   &&  rr   test_momentTestMoments.test_moment'  sL   ::dmm,LLD$8$89:::c?+LL1%::c?+LL1%::c?+LL1%::d+,LL1%::c?+LL1%::f-. LL<0::&:;< LL3'::c?+LL#34::&:;<ru   r3  zarray value-dependent errorr  c                    VP                  V P                  4      pR p\        P                  ! \        VR7      ;_uu_ 4        \
        P                  ! VR4       RRR4       R#   + '       g   i     R# ; i)z)All elements of `order` must be integral.r   333333?N)r^   r  r   r   r   r`   r  )rh   ri   r  r  s   &&  rr   test_order_input_validation'TestMoments.test_order_input_validationD  sG    ::dmm,=]]:W55LL3' 6555s   A$$A5	c           
     J   R p\        \        W!R7      ;_uu_ 4        \        P                  ! VP	                  . 4      4      p\        W1P	                  VP                  4      4       RRR4       \        \        W!R7      ;_uu_ 4        \        P                  ! VP	                  . VP                  R7      4      p\        W1P	                  VP                  VP                  R7      4       RRR4       Rp\        \        W!R7      ;_uu_ 4        \        P                  ! VP	                  . .4      ^R7      p\        W1P	                  VP                  .4      4       RRR4       \        P                  ! VP                  R4      ^ R7      p\        W1P                  R4      4       \        P                  ! VP	                  . .4      ^ ^.^ R7      p\        W1P                  R	4      4       R#   + '       g   i     EL; i  + '       g   i     EL; i  + '       g   i     L; i)
zOne or more sample arguments...r^  Nr;   zAll axis-slices of one or morerC   orderrD   r  r  )rJ   r~   )
r"   r   r`   r  r^   r3   rg   rj  rX  r  )rh   ri   r  rl   s   &&  rr   test_zero_sizeTestMoments.test_zero_sizeK  sf   3+7BBRZZ^,AAzz"&&12 C +7BBRZZ"**Z=>AAzz"&&

zCD C 3+7BBRZZ-A6AAzz266(34 C LL&)288D>*LLRD)!Qa@88F+,! CBB CBB
 CBs&   A
G*A"G>AH*G;	>H	H"	c                ~   \         P                  ! R 4      p\         P                  V^	&   \        \        P
                  ! V^4      \         P                  4       \        \        P
                  ! VRR7      R4       \        \        \        P
                  VRR7       \        \        \        P
                  VRR7       R# )r   r   r   r   r   r   N)	r   r   rg   r   r`   r  r   r   r   r  s   & rr   r  TestMoments.test_nan_policy_  sn    IIcNvv!U\\!Q'0ELLv>Dj%,,gFj%,,hGru   r<   zexpect, orderc                   VR 8X  d*   \        V4      '       d   Rp\        P                  ! V4       \        WA4      p\        P
                  P                  R4      pVP                  VP                  ^4      VR7      p\        P                  ! WsR7      p\        WP                  W!R7      4       Rp\        \        WTR7      ;_uu_ 4        \        P                  ! VP                  VR	4      ^ VR7      p\        WP                  R
W!R7      4       RRR4       \        \        WTR7      ;_uu_ 4        \        P                  ! VP                  VR4      ^VR7      p\        WP                  RW!R7      4       RRR4       \        P                  ! VP                  VR4      RVR7      p\        WP                  RW!R7      4       R#   + '       g   i     L; i  + '       g   i     Lj; i)r  z?torch can't round complex input; data-apis/array-api-compat#407l   1G r;   r  z-Precision loss occurred in moment calculationr^  )rD   r  N)r   r|   )r|   r  )r\   rJ   r   r|   r   )r%   r   r  rl  r   r  r  r^   r`   r  r3   r"   r   r  r   )	rh   r<   expectr  ri   r  r  rk   rl   s	   &&&&&    rr   test_constant_moments!TestMoments.test_constant_momentsg  sY    ,8B<<WGKK "ii##J/JJszz!}EJ2LL(::f::;Aw>>R__Q7auMAAwwtVwAB ? w>>R__Q@q#(*AAww|VwIJ ?
 LLO<4$&772v7;< ?> ?>s   
AG1AGG	G)	c                T   VP                  VP                  R 4      R4      p\        P                  ! V4      R,          P	                  VP
                  4      p\        P                  ! VP                  V4      ^^R7      p\        W1P                  RVP
                  .4      RR7       R# )r   rC   r  rE  rF  Nr  r  )
rb   r   rd   re   rf   rg   r`   r  r^   r2   )rh   ri   rL  mms   &&  rr   test_moment_propagate_nan%TestMoments.test_moment_propagate_nan  sq     JJryy}g.FF1IdO'\\"**Q-3JJbff~6UCru   c                    \         P                  ! \        R R7      ;_uu_ 4        \        P                  ! VP                  . RO4      . R7       RRR4       R#   + '       g   i     R# ; i)z1`order` must be a scalar or a non-empty 1D array.r   r  Nr=  )r   r   r   r`   r  r^   rQ  s   &&rr   test_moment_empty_order#TestMoments.test_moment_empty_order  sM    ]]: .E F FLLL1<F F F FrT  c                    VP                  ^
4      p\        P                  ! V^R7      p\        P                  ! V^R7      p\        W44       R# )rT   )r  r  N)r   r`   r  r3   r  s   &&   rr   test_rename_moment_order$TestMoments.test_rename_moment_order  s9     IIbMll1Q'll1A&!ru   c                   V P                   \        P                  ! V P                   4      ,
          p\        \        P                  ! V^*4      P                  4       \
        P                  ! V P                   ^*4      4       R# )*   N)testcase_moment_accuracyr   rc   r
   r   r`   r  )rh   
tc_no_means   & rr   test_moment_accuracy TestMoments.test_moment_accuracy  sY     33 = =>?
R0557T%B%BBG	Iru   r  rD   rJ  r   c           	     L   \         P                  P                  R 4      pVP                  RR7      p\        P                  ! VP                  V4      W#VR7      pVf   TM\         P
                  ! V4      pVP                  \        WbW4R7      R,          4      p\        Wx4       R# )l   hU~r  )rD   rJ  NrP  )r|   r   rO   r   )r   r  r  r`   r  r^   r   r2   )	rh   ri   r  rD   rJ  r  rk   r   r   s	   &&&&&    rr   test_moment_array_api!TestMoments.test_moment_array_api  sx     ii##N3JJIJ&ll2::a=%6J!>rzz&/Ajj4?CD!ru   r   r  )rT   rJ   r  )Nr~   r\   )rj  r   r  rU  r  rU  ) r   r   r   r   r8  r  r  r   r  r  r  r  r   r   r   rB  r   r  r  r  r   r  r  r  r  r  r  r  r  r  r   r   r   s   @rr   r  r    s      HO
))



+C"zz"~[[Vb']3[[VT',*M%NO. P 4. [[Vb']3[[S,/" 0 4"=: k*GH( I(-(H [[W&JK[[_vv.>?= @ L=0D="I [[Wl3[[VZ0[[Xbz2" 3 1 4"ru   r  c                   :   a  ] tR tRt o Rt. ROt. ROtR tRtV t	R# )SkewKurtosisTesti  r   c                   VP                  . 4      p\        \        \        VR 7      ;_uu_ 4        V P	                  V4      pRRR4       \        XVP                  VP                  4      4       R#   + '       g   i     L7; irg  )r^   r"   r   r   stat_funr3   rg   r   s   &&  rr   test_empty_1dSkewKurtosisTest.test_empty_1d  sW    JJrN+3HRPP--"C Q 	RZZ/0 QPs   A11B	r   Nr  )gp=
ף?g?߾?gD9?gQI?g}?5^I)
r   r   r   r   r  r  testmathworksr  r   r   r   s   @rr   r  r    s     OH<M1 1ru   r  c                     a  ] tR tRt o R t]P                  P                  R4      R 4       tR t	R t
R tR tR	 t]P                  P                  R
. RO4      ]P                  P                  RRR.4      R 4       4       tRtV tR# )TestSkewi  c                .    \         P                  ! V4      # r#  )r`   skewr  s   &&rr   r  TestSkew.stat_fun  s    zz!}ru   r  c                r   \         P                  ! VP                  V P                  4      4      p\	        W!P                  VP
                  4      4       \         P                  ! VP                  V P                  4      4      p\	        W!P                  R4      RR7       \         P                  ! VP                  V P                  4      ^ R7      p\	        W!P                  R4      RR7       \         P                  ! VP                  V P                  4      4      p\	        W!P                  R4      RR7       R# )g7l*?绽|=rF  biasr   Ng7l*ҿg2۠ۿ)r`   r  r^   r  r2   rg   r  r  )rh   ri   rl   s   && rr   test_skewnessTestSkew.test_skewness  s    
 JJrzz$"6"678::bff-. JJrzz$"4"456::&78uEJJrzz$"4"45A>::&89FJJrzz$--01::c?7ru   c                   \         P                  ! R 4      p\         P                  V^	&   \         P                  ! RR7      ;_uu_ 4        \	        \
        P                  ! V4      \         P                  4       RRR4       \	        \
        P                  ! VRR7      R4       \        \        \
        P                  VRR7       \        \        \
        P                  VRR7       R#   + '       g   i     Lv; i)	r   r   r  Nr   r   r   r   r   )	r   r   rg   r  r   r`   r  r   r   r  s   & rr   r  TestSkew.test_nan_policy  s    IIcNvv![[**A/ +UZZf5r:j%**aGDj%**aHE	 +*s   	/C&&C6	c                X    \        \        P                  ! \        ^
4      4      R4       R# )rT   r   N)r   r`   r  r   rs  s   &rr   test_skewness_scalarTestSkew.test_skewness_scalar  s    UZZr
+S1ru   c                   VP                  R 4      pVP                  VR4      p\        P                  ! V4      R,          P	                  VP
                  4      p\        P                  ! RR7      ;_uu_ 4        \        P                  ! VP                  V4      ^R7      pRRR4       \        XVP                  ^ VP
                  .4      4       R#   + '       g   i     L9; i)r   r   r  rC   Nr  r  )r   rb   rd   re   rf   rg   r   r  r`   r  r^   r3   )rh   ri   rL  r(  s   &&  rr   test_skew_propagate_nan TestSkew.test_skew_propagate_nan  s     IIbMJJq'"FF1IdO'[[**

2::a=q1A +2::q"&&k23 +*s   9(CC"	c           	        VP                  VP                  R	.4      ^
4      p\        \        RVR7      ;_uu_ 4        \	        \
        P                  ! V4      VP                  VP                  4      4       \	        \
        P                  ! VR
,          4      VP                  VP                  4      4       \	        \
        P                  ! VR
,          4      VP                  VP                  4      4       \	        \
        P                  ! VRR7      VP                  VP                  4      4       VP                  R.^,          4      p\	        \
        P                  ! V4      VP                  VP                  4      4       RVP                  R^4      R,          ,           p\	        \
        P                  ! V4      VP                  VP                  4      4       RRR4       R#   + '       g   i     R# ; i)9?Precision loss occurredr^  Fr  g,@r   r  N9ѿg      C      )	r  r^   r"   r   r3   r`   r  rg   r   rh   ri   rL  s   && rr   test_skew_constant_value!TestSkew.test_skew_constant_value  s4   IIbjj+/4/HRPPEJJqM2::bff+=>EJJqx0"**RVV2DEEJJqx0"**RVV2DEEJJqu5rzz"&&7IJ 

D6!8$AEJJqM2::bff+=>RYYsA&u,,AEJJqM2::bff+=> QPPPs   F G''G8	c                *   \        \        R VR7      ;_uu_ 4        \        P                  P	                  R4      pVP                  RR7      pRVR&   \
        P                  ! VP                  V4      4       RRR4       R#   + '       g   i     R# ; i)r  r^  l   :"z` r  )\(?N)r  rT   r   r~   )r"   r   r   r  r  r`   r  r^   )rh   ri   r  rL  s   &&  rr   test_precision_loss_gh15554$TestSkew.test_precision_loss_gh15554  se     /HRPP))''4C

	
*AAdGJJrzz!}%	 QPPPs   ABB	rD   Nr  FTc                   a \         P                  P                  R 4      pSP                  VP                  R4      4      pV3R lp\        P
                  ! WRVR7      pV! WRVR7      p\        Wx4       R# )   f#"c                 l  < Vf   SP                  V R4      p ^ pSP                  WRR7      pSP                  W,
          ^,          VR7      pSP                  WR7      pWE^,          ,          pV'       g?   V P                  V,          pWgR,
          V,          R,          VR,
          ,          ,          pV# )NTr!  rC   r   r6   r   r  )rb   rc   r   r  )	rL  rD   r  rc   mu3r   r   r  ri   s	   &&&     rr   skewness-TestSkew.test_vectorization.<locals>.skewness  s    |JJq%(771$77D''18a-d'3C&&&&Cq.CGGDMSA#-S99Jru   )rD   r  NrL  )r   r  r  r^   r`   r  r2   )	rh   ri   rD   r  r  rk   r  r   r   s	   &f&&     rr   test_vectorizationTestSkew.test_vectorization   s[    
 ii##M2JJszz),-	 jjD1q$/!ru   r   rZ   r~   rJ   N)r   r   r   r   r  r   r   r   r   r  r  r  r  r  r   r  r   r   r   s   @rr   r  r    s      [[>88F24? & [[V%56[[VeT]3" 4 7"ru   r  c                   L  a  ] tR tRt o R t]P                  P                  R4      R 4       tR t	R t
R tR t]P                  P                  R	. RO4      ]P                  P                  RRR.4      ]P                  P                  RRR.4      R 4       4       4       tRtV tR
# )TestKurtosisi  c                .    \         P                  ! V4      # r#  )r`   kurtosisr  s   &&rr   r  TestKurtosis.stat_fun  s    ~~a  ru   r  c                2   \         P                  ! VP                  V P                  4      4      pVP	                  V4      '       g   Q h\         P                  ! VP                  V P
                  4      ^ ^ ^R7      p\        W!P                  R4      4       \         P                  ! VP                  V P
                  4      ^ ^ R7      p\        W!P                  R4      4       \         P                  ! VP                  V P                  4      ^ ^ 4      p\        W!P                  R4      4       VP                  R4      pVP                  V^8H  VP                  V4      p\        \         P                  ! V4      VP                  VP                  4      4       R# )r~   fisherr  gO߻S@gx|N@g=
ףp=?r   N)r`   r#  r^   r  rx  r  r2   r  r   r  rg   r3   )rh   ri   rl   rk   s   &&  rr   test_kurtosisTestKurtosis.test_kurtosis   s    NN2::d&:&:;<xx{{{ NN2::d&8&891QQO::o67 NN2::d&8&89!!L::&789NN2::dmm4a;::d+,IIcNHHQ!VRVVQ'q)2::bff+=>ru   c                    \         P                  ! R 4      p\         P                  V^	&   \        \        P
                  ! VRR7      R4       \        \        \        P
                  VRR7       \        \        \        P
                  VRR7       R# )r   r   r   r   r   NGz)r   r   rg   r   r`   r#  r   r   r  s   & rr   test_kurtosis_nan_policy%TestKurtosis.test_kurtosis_nan_policy@  sU    IIcNvv!ENN1@)Lj%..!Hj%..!Iru   c                x    \        \        \        P                  ! . RO4      4      \        P
                  4       R# )r\   Nr  )r   typer`   r#  r   r   rs  s   &rr   test_kurtosis_array_scalar'TestKurtosis.test_kurtosis_array_scalarH  s    T%..34bjjAru   c                4   \         P                  ! ^4      P                  ^R4      P                  \        4      p\         P
                  VR&   \        P                  ! V^RR7      p\         P                  P                  VR\         P
                  .RR7       R# )	rK   r  r  rE  rF  NrZ   r  g(\)
r   r   rb   r  r  rg   r`   r#  r!  r
   )rh   rL  r  s   &  rr   test_kurtosis_propagate_nan(TestKurtosis.test_kurtosis_propagate_nanL  si     IIaL  B'..u5&&$NN11=


""1ubffoE"Bru   c           
     b   VP                  R.^
,          4      p\        \        RVR7      ;_uu_ 4        VP                  \        P
                  ! VRR7      4      '       g   Q hVP                  \        P
                  ! V\        R4      ,          RR7      4      '       g   Q hVP                  \        P
                  ! V\        R4      ,          RR7      4      '       g   Q hVP                  \        P
                  ! VRRR7      4      '       g   Q h RRR4       R#   + '       g   i     R# ; i)	r  r  r^  F)r'  r&  Nr  l           )r^   r"   r   rx  r`   r#  r  r  s   && rr   test_kurtosis_constant_value)TestKurtosis.test_kurtosis_constant_valueU  s    JJ}R'(/HRPP88ENN1U;<<<<88ENN1uU|+;EJKKKK88ENN1uU|+;EJKKKK88ENN1UGHHHH	 QPPPs   CDDD.	rD   Nr  FTr'  c                   a \         P                  P                  R 4      pSP                  VP                  R4      4      pV3R lp\        P
                  ! WbW4R7      pV! WbW4R7      p	\        W4       R# )r  c                   < Vf   S	P                  V R4      p ^ pS	P                  WRR7      pS	P                  W,
          ^,          VR7      pS	P                  W^ R7      pV'       d   WV^,          ,          ^,
          pMpV P                  V,          pV^,
          V^,
          V^,
          ,          ,          V^,           V,          V^,          ,          ^V^,
          ,          ,
          ,          pV'       d   V# V^,           # )NTr!  rC   rD   rx   r  )rb   rc   r   r  )
rL  rD   r  r'  rc   mu4mu2r   r  ri   s
   &&&&     rr   r#  1TestKurtosis.test_vectorization.<locals>.kurtosisg  s    |JJq%(771$77D''18a-d'3C&&!&4CFlQ&GGDMs!!}-!a%3sAv1E1Q31OP !3-cAg-ru   )rD   r  r'  Nrn  )r   r  r  r^   r`   r#  r2   )
rh   ri   rD   r  r'  r  rk   r#  r   r   s
   &f&&&     rr   r  TestKurtosis.test_vectorization^  s[     ii##M2JJszz),-	.& nnQDq$>!ru   r   r  )r   r   r   r   r  r   r   r   r(  r,  r0  r3  r6  r   r  r   r   r   s   @rr   r!  r!    s     ! [[ VW? X?>JBCI [[V%56[[VeT]3[[Xt}5" 6 4 7"ru   r!  c                 ~   \        R R R7      p\        P                  ! ^^R7      pV ! \        P                  ! \        P
                  WR7      4      p. p\        P                  ! 4       ;_uu_ 4        \        P                  ! R4       \        \        VP                  4      4       F  p\        P                  ! \        4      ;_uu_ 4        \        P                   ! V^VR7      p\        P"                  ! V^ 8  4      '       dB   \        P"                  ! \        P$                  ! V4      4      '       d   VP'                  V4       RRR4       K  	  RRR4       \(        P*                  ! V4       V ! \(        P,                  P/                  V4      4      pW53#   + '       g   i     EK  ; i  + '       g   i     Lc; i)F)	allow_nanallow_infinity)min_dimsmin_side)r<   elementsr  errorr  N)dictnpstarray_shapesarraysr   r   r   r   r  r  r  r  rA  suppress	Exceptionr`   r  r  r  rH  
hypothesisassume
strategiessampled_from)drawrD  r  rG  ok_axesrD   r   s   &      rr   ttest_data_axis_strategyrR    s(    eE:Hq15E "**xMND G 
	 	 	"	"g&#djj/*D$$Y//ll4qt<66#'??rvvbkk#.>'?'?NN4( 0/ + 
# g 
%%227;<D: 0// 
#	"s,   1AF,A(F4FF,F)"
F,,F<	c                     a  ] tR tRt o . R Ot. R!OtRtRtR"tRt	R#t
RtRtRt]	^,          t^]	^,          ,
          t]P"                  P%                  R4      ]P"                  P%                  R4      R	 4       4       tR
 t]! RRR7      R 4       t]P"                  P%                  R4      R 4       t]! RRR7      ]P"                  P1                  R. R$O4      R 4       4       tR t]! RRR7      ]P"                  P6                  ]P:                  ! ]P<                  P?                  RR%4      ] ! 4       R7      ]P"                  P1                  RRR.4      R 4       4       4       4       t!Rt"V t#R# )&TestTTest_1sampi  r   r   gMXz?g4/?gFX5t?r  r  c           	        \         P                  ! 4       ;_uu_ 4        \        P                  ! R R R7      ;_uu_ 4        \         P                  ! R R\
        4       \        V4      '       g   VP                  R4      MRp\        P                  ! VR4      w  r4RRR4       RRR4       \        XVP                  VP                  4      4       \        XVP                  VP                  4      4       \        P                  ! VP                  V P                  4      R4      w  r4\        W1P                  V P                  4      4       \        WAP                  V P                  4      4       \        P                  ! VP                  V P                  4      R4      pR
p\!        WVVR7       \        P                  ! VP                  V P"                  4      R4      w  r4\        W1P                  V P$                  4      4       \        WAP                  V P&                  4      4       \        P                  ! VP                  V P                  4      R4      w  r4\        W1P                  V P(                  4      4       \        WAP                  V P*                  4      4       \        P                  ! VP                  V P                  4      R	4      w  r4\        W1P                  V P,                  4      4       \        WAP                  V P.                  4      4       R#   + '       g   i     EL; i  + '       g   i     EL; i)r   r  divide!Degrees of freedom <= 0 for slicer   r   Nr   r=  r   r   r@  r?  )r   r   r   r  r   r   r$   r^   r`   ttest_1sampr3   rg   X1r2   T1_0P1_0r   X2T2_0P2_0T1_1P1_1T1_2P1_2)rh   ri   rL  trf  r   rA  s   &&     rr   test_onesampleTestTTest_1samp.test_onesample  s    $$&&HX>>##=~O&.rll

2A$$Q+DA	 ? ' 	2::bff-.2::bff-.  DGG!4b9::dii01::dii01

477 3R8,
C3  DGG!4b9::dii01::dii01  DGG!4b9::dii01::dii01  DGG!4b9::dii01::dii01; ?>> '&&s#   !L0AL	L0L-'	L00M	c                   \         P                  P                  ^^
^3RR7      p\        P                  V^2&   VP                  V\        V4      R7      p\         P                  ! VR4      p\        VP                  VP                  VP                  4      4       \        VP                  VP                  VP                  4      4       R# )r|   t r*  r+  r  r  r;   r=   N)r`   rW  r  r   rg   r^   r&   rZ  r3   r@  r?  r   s   &&  rr   #test_onesample_nan_policy_propagate3TestTTest_1samp.test_onesample_nan_policy_propagate  s    JJNNq'NJ"JJq 0 4J53'rzz"&&'9:

BJJrvv$67ru   TrC  r   c                   \         P                  P                  ^^
^3RR7      p\        P                  V^2&   VP                  V\        V4      R7      p\         P                  ! VRRR7      p\        VP                  VP                  R4      4       \        VP                  VP                  R4      4       \        P                  ! \        RR	7      ;_uu_ 4        \         P                  ! VRR
R7       RRR4       \        P                  ! \        RR	7      ;_uu_ 4        \         P                  ! VRRR7       RRR4       R#   + '       g   i     LX; i  + '       g   i     R# ; i)r|   ri  rj  r;   r=   r   r   g/kCm?r   r   r   Nnan_policy must be one ofr   g̈́^B)r`   rW  r  r   rg   r^   r&   rZ  r2   r@  r?  r   r   r   r   s   &&  rr   $test_onesample_nan_policy_omit_raise4TestTTest_1samp.test_onesample_nan_policy_omit_raise  s    JJNNq'NJ"JJq 0 4J536:rzz2E'FG

BJJ/C$DE]]:-LMMa9 N ]]:-HIIa: JI NM JIIs   E EE	E(	z+ignore:divide by zero encountered in dividec                   R p\         P                  ! \        VR7      ;_uu_ 4        \        P                  ! VP                  V P                  4      RRR7       RRR4       \        P                  ! VP                  V P                  4      RRR7      w  r4\        WAP                  V P                  4      4       \        W1P                  V P                  4      4       \        P                  ! VP                  V P                  4      RRR7      w  r4\        WAP                  V P                  4      4       \        W1P                  V P                  4      4       R#   + '       g   i     EL; i)	8`alternative` must be 'less', 'greater', or 'two-sided'.r   r   rE  r  Nr   r  r  )r   r   r   r`   rZ  r^   r[  r2   P1_1_lra  P1_1_g)rh   ri   r  re  rf  s   &&   rr   test_1samp_alternative&TestTTest_1samp.test_1samp_alternative  s    L]]:W55bjj127K 6   DGG!4bfM::dkk23::dii01  DGG!4biP::dkk23::dii01 655s   3EE-	r3  Generic stdtrit mutates array.r  r  r  r  c                   \         P                  P                  R 4      p^
pVP                  VR^R7      pVP                  4       pVP	                  R4      P
                  pVP	                  WWR7      pVP	                  WgR7      pRRR.RR	\         P                  .R
\         P                  ) R./p\        P                  ! WVVR7      p	V	P                  RR7      p
\        V
P                  VP	                  W,          ^ ,          4      4       \        V
P                  VP	                  W,          ^,          4      4       \        V	P                  VP	                  V^,
          4      4       R# )l   <PleH r7   r  r*  r+  r   r;   r  gÍ]?g$(}w@r  gS1?r  g]@)popmeanr  333333?confidence_levelN)r   r  r  r  r^   r<   r   r`   rZ  r  r2   r  r  r3   df)rh   ri   r  r  r  rk   rz  r<   r   r   r  s   &&&        rr   test_1samp_ci_1d TestTTest_1samp.test_1samp_ci_1d  s$    ii##$78JJA3aJ0**, 

3%%JJqJ&**W*2/1CD-rvv6!235 L$$d$;

3+;A+> ?@C,<Q,?!@A

1Q30ru   c                    \         P                  ! VP                  R 4      R4      pRp\        P                  ! \
        VR7      ;_uu_ 4        VP                  ^
R7       RRR4       R#   + '       g   i     R# ; i)r   r   4`confidence_level` must be a number between 0 and 1.r   r|  N)r`   rZ  r   r   r   r   r  )rh   ri   r   r  s   &&  rr   test_1samp_ci_iv TestTTest_1samp.test_1samp_ci_iv  sR    		#3H]]:W55##R#8 6555s   A**A;	z	Too slow.r  rE  )alpha	data_axisc                   Vw  rVVP                  V4      p\        P                  ! VR W6R7      pVP                  VR7      w  rVR8X  d   TMT	p
VP	                  WR7      p
\        P                  ! WZW6R7      p\        VP                  4      pVP                  V4       VP                  VP                  ^V,
          VP                  R7      V4      p\        VP                  V4       R# )r   r  r|  r  rC   r;   N)r^   r`   rZ  r  expand_dimsrB  r  rN  r  r   r2   r?  )rh   r  r  r  ri   rG  rD   r   lr  rz  r  r   s   &&&&&        rr   test_pvalue_ciTestTTest_1samp.test_pvalue_ci  s     
zz$b,7D&&&>"i/!Q...4;RTZZ 		$oobjj5

jCUK

C(ru   r   NrQ  r   r   )r   r   r   gMXzgMXzrm  g?)$r   r   r   r   r[  r^  r\  r]  ra  rb  rc  rd  r_  r`  rs  rt  r   r   r   rf  rk  r   ro  ru  r   r  r  rn  rL  givenrN  floatsrR  r  r   r   r   s   @rr   rT  rT    s     
B	BDDDDDDDDAXF$(^F[[ WX[[ VW2 X Y2B8 .MN; O;  [[ MN2 O2 k*JK[[],LM1 N L109 d;7[[J1188H 8 :<[[]VY,?@) A<  8
)ru   rT  c                     a  ] tR tRt o R t]P                  P                  R. RO4      R 4       t]P                  P                  R. RO4      R	 4       t	]P                  P                  R. R O4      R
 4       t
]P                  P                  R. R!O4      R 4       t]P                  P                  R. RO4      R 4       t]P                  P                  R. R O4      R 4       t]P                  P                  R. R!O4      R 4       t]P                  P                  RR. R"O3R. R#O3R. R$O3R. R"O3.4      R 4       t]P                  P                  RR. R%O3R. R&O3R. R'O3R. R%O3.4      R 4       tR. ^]P$                  3R]P$                  .^]P$                  3R]P$                  .. R(O]P$                  ]P$                  ]P$                  .3R^^.^^]P$                  .^2^d]P$                  .3R^^]P$                  .. R(O. R)O3R^^.^ ^]P$                  .^ ^2]P$                  .3R]P$                  ]P$                  .. R(O]P$                  ]P$                  ]P$                  .3.t]P                  P                  R]4      R 4       tR^^^]P$                  .. R*OR3R. R*O^^^]P$                  .R3.t]P                  P                  R]4      R 4       t]P                  P                  R. R+O4      R 4       tR tRtV tR# ),TestPercentileOfScorei3  c                .    \         P                  ! V/ VB # r#  )r`   percentileofscore)rh   argskwargss   &*,rr   fTestPercentileOfScore.f5  s    &&777ru   zkind, resultr  rc   strictweakc                J    . ROp\        V P                  V^VR7      V4       R# )r\   kindN
r\   rJ   r   r   r|   r   rO   rK   rN   rT   r   r  rh   r  r  rL  s   &&& rr   test_unique!TestPercentileOfScore.test_unique8  s"    
 ,TVVAqtV,f5ru   c                J    . ROp\        V P                  V^VR7      V4       R# )r\   r  N)
r\   rJ   r   r   r   r|   r   rO   rK   rN   r  r  s   &&& rr   test_multiple2$TestPercentileOfScore.test_multiple2@  "    
 +TVVAqtV,f5ru   c                J    . ROp\        V P                  V^VR7      V4       R# )r\   r  N)
r\   rJ   r   r   r   r   r|   r   rO   rK   r  r  s   &&& rr   test_multiple3$TestPercentileOfScore.test_multiple3H  r  ru   c                J    . ROp\        V P                  V^VR7      V4       R# )r\   r  N)
r\   rJ   r   r|   r   rO   rK   rN   rT   rU   r  r  s   &&& rr   test_missing"TestPercentileOfScore.test_missingP  s"    
 -TVVAqtV,f5ru   c                J    . ROp\        V P                  V^(VR7      V4       R# )rT   r  N
rT   r  r$  r%  r  r  F   r  r\  r  r  r  s   &&& rr   r7  (TestPercentileOfScore.test_large_numbersX  s"    
 6TVVArV-v6ru   c                J    . ROp\        V P                  V^(VR7      V4       R# )rT   r  N)
rT   r  r$  r%  r%  r%  r  r  r  r  r  r  s   &&& rr   test_large_numbers_multiple32TestPercentileOfScore.test_large_numbers_multiple3`  s"    
 5TVVArV-v6ru   c                J    . ROp\        V P                  V^(VR7      V4       R# )rT   r  N
rT   r  r$  r  r  r  r  r\  r  n   r  r  s   &&& rr   test_large_numbers_missing0TestPercentileOfScore.test_large_numbers_missingh  s"    
 7TVVArV-v6ru   c                N    . ROp\        V P                  V. ROVR7      V4       R# )rT   r  Nr  )r~   rT   r     r  r  s   &&& rr   test_boundaries%TestPercentileOfScore.test_boundariesp  s#    
 7TVVA0tV<fEru   c           
         ^^^^^^^^^	\         P                  5.
p\        V P                  V\         P                  ) ^\         P                  5.VR7      V4       R# )r\   r  N)r   r   r   r  r  s   &&& rr   test_infTestPercentileOfScore.test_infx  sP    
 1aAq!Q0TVVARVVG44V@&Iru   r  r   zpolicy, a, score, resultc                @    \        V P                  W#VR 7      V4       R# )r   Nr  )rh   policyrL  r  r  s   &&&&&rr   test_nans_ok"TestPercentileOfScore.test_nans_ok  s    TVVAV8&Aru   r   r   zpolicy, a, score, messagec                    \        \        VR 7      ;_uu_ 4        V P                  W#VR7       RRR4       R#   + '       g   i     R# ; i)r   r   N)r   r   r  )rh   r  rL  r  r  s   &&&&&rr   test_nans_fail$TestPercentileOfScore.test_nans_fail  s.    :W55FF1F/ 6555s	   8A		r  c                    \         P                  ! . RO4      pVP                  V4      pV^
,          p. ROp\        V P	                  W#RR7      V4       R# )r~   r  r  N)r~   r\   rJ   r   r   r|   r  )r   r   rb   r   r  )rh   r  rL  r  resultss   &&   rr   test_ndTestPercentileOfScore.test_nd  sF     HH'(5!)+TVVAFV3W=ru   c                    R p\         P                  ! \        VR7      ;_uu_ 4        \        P                  ! \
        P                  ! R4      ^4       RRR4       R#   + '       g   i     R# ; i)z`a` must be 1-dimensional.r   Nr  )r   r   r   r`   r  r   r  r:  s   & rr   test_multidimensional_error1TestPercentileOfScore.test_multidimensional_error  sA     /]]:W55##BGGFOQ7 6555s   ,AA.	r   N))r  r%  )rc   #   r  r$  )r  r%  ))r  -   )rc   r%  r  )r  r  ))r  r  )rc   r  r  )r  r  ))r  r$  )rc   r$  r  )r  r$  )r~   rT   r  r  )r~   r|   _   r  )r~   r~   r\  r  )r~   rT   r  )r~   r|   r  )r~   r~   r\  r  )r~   r  r  r  ))r   rV  rJ   r\   r   )rJ   r\   r\   r   )r   r   r   r   r  r   r   r   r  r  r  r  r7  r  r  r  r  r   rg   casesr  r  r  r  r   r   r   s   @rr   r  r  3  ss    8 [[^ .< =6	=6 [[^ .< =6	=6 [[^ .< =6	=6 [[^ .< =6	=6 [[^ .< =7	=7 [[^ .< =7	=7 [[^ .< =7	=7 [[^v7H.I/5.G/7.I/57H.I.K LF	LF [[^v|.D/5z.B/7.D/5|.D.F GJ	GJ 2q"&&)BFF8Q/BFF8Y0HIAq6Aq"&&>BRVV3DEq!RVVni>q!fq!RVVnq"bffo>rvvrvv&	BFFBFFBFF3KLNE [[7?B @B 
1aBFF#Y	(	*	)aArvv.	(	*E [[8%@0 A0 [[W ' >>8 8ru   r  CaserD   )f_obsf_expr  rD   chi2r  mod_logcrc                   P  a  ] tR tRt o R t]P                  P                  R]4      ]P                  P                  R. RO4      R 4       4       t	R t
R t]P                  P                  R]4      ]P                  P                  R. RO4      R	 4       4       tR
 tR tR tRtV tR# )TestPowerDivergencei  c           
        VP                  R 4      P                  pVP                  WR7      pVe   VP                  W(R7      MTpVf   \        V4      p	M2Ve   VP                  W4      MV3p
V
^ ,          P                  V,          p	\
        P                  ! 4       ;_uu_ 4        \
        P                  ! RR\        4       \        P                  ! WVWER7      w  r\        WP                  WhR7      4       V^8X  g   VR8X  d7   \        P                  ! WVVR7      w  r\        WP                  WhR7      4       RRR4       \        P                   ! V4      p\        P                  P                  P!                  VV	^,
          V,
          4      p\        XVP                  WR7      4       R#   + '       g   i     L|; i)r   r;   Nr   zMean of empty slicer  r  rD   lambda_pearson)r  r  rD   )r^   r<   r'   broadcast_arraysr  r   r   r   r   r`   power_divergencer2   	chisquarer   distributionsr  sf)rh   r  r  r  rD   r  expected_statri   r<   num_obsrI  statrf  
expected_ps   &&&&&&&&      rr   check_power_divergence*TestPowerDivergence.check_power_divergence  s`   

2$$

5
.272C

5
.<enG;@;Lb))%7!8 Qiood+G$$&&##H.C^T,, %%)<GD D**]*"HI!|w)3//%4/35jjj&LM ' zz$((--00181t1CE
2::j:>?! '&s   %BF00G 	caselambda_statNc           	         Vw  rE\        W4      pV P                  VP                  VP                  VP                  VP
                  WFV4       R # r#  )rl  r  r  r  r  rD   rh   r  r  ri   r  attrr  s   &&&&   rr   rY  TestPowerDivergence.test_basic  sA     $+##DJJ

DIItyy$+B	@ru   c           
        \         ^ ,          p\         ^,          p\        P                  ! VP                  VP                  34      p\        P                  ! \        P                  ! VP                  4      \        P
                  ! VP                  4      ,          VP                  34      pVP                  V4      pVe   VP                  V4      MTpV P                  WE^ ^RVP                  VP                  .VR7       V P                  WE^ ^RVP                  VP                  .VR7       V P                  WE^ ^RVP                  VP                  .VR7       V P                  WE^ ^RVP                  VP                  .VR7       VP                  VP                  VP                  4      R4      pV P                  VR^ RRVP                  VR7       R# )r~   Nr  r=  log-likelihoodmod-log-likelihoodcressie-readr  )power_div_1d_casesr   vstackr  	ones_likerc   r  r^   r  r  r  r  r  rb   )rh   ri   case0case1r  r  f_obs_reshapes   &&     rr   r	  TestPowerDivergence.test_axis  s   "1%"1%		5;;45		2<<4RWWU[[5II ;;( ) 

5!%*%6

5!E##Q5::uzz2r 	$ 	; 	##Q%))UYY!7B 	$ 	@ 	##Q#emmU]]%C 	$ 	L 	##Q%((3 	$ 	<
 

2::ekk#:FC##dAt%** 	$ 	-ru   c                   \         ^ ,          p\         ^,          p\        P                  ! VP                  VP                  34      P                  p\        P                  ! \        P
                  ! VP                  4      \        P                  ! VP                  4      ,          VP                  34      P                  pVP                  VP                  .pVP                  R4      P                  pVP                  WGR7      pVP                  WWR7      pVP                  WgR7      pVP                  ^ .^..4      p\        P                  ! WEVR7      w  r\        W4       \        P                  ! WEVR,          R7      w  r\        P                  ! WEVR,          R7      w  rVP                  WP                  R3,          WP                  R3,          3^ R7      p\        W4       R# )	r~   r   r;   r  r   rC   Nr  r  )r  r   r  r  r   r  rc   r  r  r^   r<   r`   r  r2   r  r  )rh   ri   r  r  r  r  expected_chi2r<   r  r  rf  r  p0p1r  s   &&             rr   test_ddof_broadcasting*TestPowerDivergence.test_ddof_broadcasting7  sn    #1%"1%		5;;4577		2<<4RWWU[[5II ;;( ))* 	 UZZ0

2$$

5
.

5
.

=
> zzA3*%((DA, &&u$t*E&&u$t*EYY::q= 12jj!m3DEAYN
&ru   c                   Vw  rE\        W4      p\        P                  ! 4       ;_uu_ 4        V P                  VP                  VP
                  VP                  VP                  WFV4       R R R 4       R #   + '       g   i     R # ; ir#  )rl  r   r   r  r  r  r  rD   r  s   &&&&   rr   test_empty_cases$TestPowerDivergence.test_empty_casesY  s_     $+$$&&''

DJJ		499, '&&&s   ?A66B	c                j   \         ^ ,          P                  p\         ^ ,          P                  p\         ^ ,          P                  p\         ^ ,          P                  pVP                  R4      P                  pVP                  W&R7      p\        P                  ! W#VVRR7      pRp\        WxVR7       R# )r~   r   r;   r  r  r=  NrY  )
r  r  r  r  rD   r^   r<   r`   r  r   )	rh   ri   r  r  r  rD   r<   r   rA  s	   &&       rr   'test_power_divergence_result_attributes;TestPowerDivergence.test_power_divergence_result_attributesf  s    "1%++"1%++!!$))!!$))

2$$

5
. $$Ud*.	C,
C3ru   c           	     F   VP                  R R.RR..4      pVP                  RR.RR..4      p\        P                  ! W#R7      w  rE\        WAP                  RR	.4      4       \        WQP                  R
R.4      4       \	        V4      '       Ed   \        P                  ! W!P                  RR.4      R7      p\        VP                  VP                  VP                  VP                  .4      4       \        VP                  VP                  VP                  VP                  .4      4       \        P                  ! W#^R7      p\        VP                  VP                  VP                  VP                  .4      4       \        VP                  VP                  VP                  VP                  .4      4       VP                  RR.RR..4      p\        P                  ! W#R7      p\        VP                  VP                  V^ ,          VP                  .4      4       \        VP                  VP                  V^ ,          VP                  .4      4       R# Rp\        P                  ! \        VR7      ;_uu_ 4        \        P                  ! W!P                  RR.4      R7       RRR4       \        P                  ! \        VR7      ;_uu_ 4        \        P                  ! W#^R7       RRR4       R#   + '       g   i     LW; i  + '       g   i     R# ; i)r   r9        >@r=   r  g     A@      9@r  g̑m@gUUU@gj^4;?g4;?      N@)r  rD   For each axis slice...r   N)r^   r`   r  r2   r#   r3   r@  rg   r?  r   r   r   )rh   ri   r  r  r  r  r   r  s   &&      rr   test_power_divergence_gh_122822TestPowerDivergence.test_power_divergence_gh_12282t  s'   

S#Jc
34

RISz23++E?
jj*j)ABCjj*j)ABC((jj#s6LMCCMM2::rvvrvv6F+GHCJJ

BFFBFF3C(DE((!DCCMM2::rvvrvv6F+GHCJJ

BFFBFF3C(DE JJS	C:67E((<CCMM2::tAw6G+HICJJ

DGRVV3D(EF.Gz99&&uJJSz4JK :z99&&uB :9 :9999s   )K<L<L	L 	c           	        VP                  . RO4      pR pVP                  RVP                  ^ ,          R,           4      pVP                  VP	                  V4      VP	                  VP                  W4,          4      4      ,          4      pVP                  WSV,          ,           4      pVP                  W!P                  R3,          WaP                  R3,          34      P                  p\        P                   ! . R!NRNR"NRNR#NRNR$NR
NR%NRNR&NRNR'NRNRNRNRNRNRNRNRNRNRNRNR	NRNRNRNRNRNRNRN4      p\        P                  ! VR(4      p\        VP                  ^ ,          4       Fm  p\        W^ 3,          4      \        W^3,          4      r\        P                  ! VR),          VR*,          V	R7      w  r\        WP                  V
4      RR7       Ko  	  R# )+r  r=   r   r   r   r   g    @g     r@gffffffP@r   gLD@r7   g      A@g     =@r6   g     :@r   g8@gffffff7@gq=
ףp?g7@g333336@g6@gfffff6@g8@g     A@g     j@)r  g{Gzt?r  N)r  r        ,@g      1@r=   r  r   r   r   r   r   r   r  r   r   r   r   r   g*kqg      $r  r         r  rQ  r  )rZ   rJ   r  )r   r\   )r^   r   r  r  r]  expr  r  r   r   rb   r  r  r`   r  r2   )rh   ri   obsbetar  r  expected_countstable4table5r  r  r  rf  s   &&           rr   /test_power_divergence_against_cressie_read_dataCTestPowerDivergence.test_power_divergence_against_cressie_read_data  s    jj @ AIIb#))A,+,rvvc{RVVBFF46N%;;<&&a0 C

A.+JJM:< ==>Q 	     	 	
 
                       !  !" #" # & FG,v||A'A%*6Q$<%8%!t:M],,VD\6$<5<>GDD**]";$G	 (ru   r   ))Nr  r  r  )r\   r  r  r  r  r  r  r  rG  r  )r  r  r  r  r  )r   r   r   r   r  r   r   r   r  rY  r	  r  power_div_empty_casesr  r  r  r  r   r   r   s   @rr   r  r    s     @@ [[V%78[[]	./@	/ 9
@-: 'D [[V%:;[[]	./,	/ <
,4C:-H -Hru   r  c                   t   a  ] tR tRt o R tR t]P                  P                  RRR	.4      R 4       t	Rt
V tR# )
TestChisquarei  c                   VP                  R R.4      pVP                  RR.4      p\        V4      '       dx   \        P                  ! W#R7      p\	        VP
                  VP                  VP                  4      4       \	        VP                  VP                  VP                  4      4       R# \        \        RR7      ;_uu_ 4        \        P                  ! W#R7       RRR4       R#   + '       g   i     R# ; i)r   r9  r  r
  r	  r  r   N)
r^   r#   r`   r  r3   r@  rg   r?  r   r   )rh   ri   r  r  r   s   &&   rr   test_chisquare_12282a#TestChisquare.test_chisquare_12282a  s     

C:&

C:&//%5CCMM2::bff+=>CJJ

266(:;z1IJJ3 KJJJs   C**C;	c                J   \         P                  P                  R 4      p^
pVP                  RRVR7      pVP	                  V4      pVP                  V4      pVP                  WTP                  R7      p\        P                  ! WTRRR7      pWT,
          VP                  V4      ,          pVP                  V^,          4      p\        VP                  V4       \        VP                  W8P                  R7      4      p	\        VP                  V	P                  V4      4       R# )	l   *m{m r  rA  r  r;   F)r  r  	sum_checkNrZ   )r   r  r  rM  poissonr^   r<   r`   r  rI  r]  r2   r@  r   r?  r  )
rh   ri   r  r  lamrk   r   rm  r@  r^  s
   &&        rr   test_chisquare_12282b#TestChisquare.test_chisquare_12282b  s     ii##$45kk$1k-KKjjoJJq		J*ooauEW$FF1a4L	y1A__=>

BEE)$45ru   zn, dtypec           	        \        W24      pVP                  V^ .VR7      pVP                  V^,          V^,          .VR7      p\        P                  ! WE4      pVw  rx\	        WsP                  WP                  R4      P
                  R7      RR7       \        VP                  V4       \        VP                  V4       R# )r~   r;   r   vIh%<=r  N)	rl  r^   r`   r  r2   r<   r3   r@  r?  )	rh   r  r<   ri   r  r  r   r  rf  s	   &&&&     rr   #test_chiquare_data_types_attributes1TestChisquare.test_chiquare_data_types_attributes  s     "jj!Quj-jj!q&!q&)j7ooc'jj**R.2F2FjGeTt,

A&ru   r   N)r  rf  )i@B r  )r   r   r   r   r"  r(  r   r   r   r,  r   r   r   s   @rr   r   r     s;     46& [[Z.:L)MN
' O
'ru   r   c                     a  ] tR tRt o . RO. R	O. R
O. RO.tRt. RO. RO. RO. RO.tRt. RO. RO. RO. RO.tRt	]
P                  P                  R. RO4      ]
P                  P                  R]]3]]3]]	3.4      R 4       4       tR tRtV tR# )TestFriedmanChiSquarei  r<   Nz	data, refc                R   Vf   TM
\        WA4      pV Uu. uF  qTP                  WQR7      NK  	  pp\        P                  ! V!  p\	        VP
                  VP                  V^ ,          VR7      4       \	        VP                  VP                  V^,          VR7      4       R # u upi )Nr;   )rl  r^   r`   friedmanchisquarer2   r@  r?  )rh   r<   rG  r   ri   r   r   s   &&&&&  rr   test_against_references-TestFriedmanChiSquare.test_against_references  s     GB,><@AD5

5
.DA%%t,rzz#a&z'FG

BJJs1vUJ$CD Bs   B$c           	     :   R p\         P                  ! \        VR7      ;_uu_ 4        \        P                  ! VP                  V P                  ^ ,          4      VP                  V P                  ^,          4      4       RRR4       R#   + '       g   i     R# ; i)z At least 3 samples must be givenr   N)r   r   r   r`   r1  r^   r  rh   ri   r  s   && rr   test_too_few_samples*TestFriedmanChiSquare.test_too_few_samples  sY    4]]:W55##BJJtwwqz$:BJJtwwqz<RS 6555s   AB		B	r   )g"~j?g^I+?gI+?gjt?gMbX9?gʡE?x&?-?g?r   Gz?g+?gv?g9v?)g~jt?gx?gOn?r8  g"~j?n?gK7A`?r9  g7A`?r   bX9?gZd;O?g1Zd?gK?)gy&1?gzG?g`"?g!rh?g'1Z?gZd;O?grh|?gjt?gʡE?r   gzG?gS?gJ4?gʡE?)gK7?gh|?5?gl?g/$?g#~j?r;  gQ?      ?g      ?r   r<  g rh?g333333?
ףp=
?)gNt$@gBv?)r   r   r|   r   r|   r   rJ   r|   r   r   r   r   )rJ   rJ   r\   rJ   r   r\   rJ   r   rJ   r\   r\   r   )rJ   r   r   r   r   r   r   r   r   r\   rJ   r\   )r   r|   r   r   r   r   r   r   r   r   r   r   )g
__2@g5"]i2?)r   g#@      !@r  g$@)g333333@g@g@r  g@)r  r  rG   r   @)g!@g!@333333 @ffffff
@g333333"@)g\(\%@g+ԋ?r  )r   r   r   r   r  ref1r  ref2r  ref3r   r   r   r2  r6  r   r   r   s   @rr   r/  r/    s     
9:;:
;B 2D /
.
.
.
0B 4D %
#
#
#
%B 'D[[W&BC[[[B:DzB:*NOE P DET Tru   r/  c                      a  ] tR tRt o RtRR ltR tR t]P                  P                  R. RO4      ]P                  P                  R. RO4      R 4       4       tR	 tR
tV tR# )
TestKSTesti!  zLTests kstest and ks_1samp agree with K-S various sizes, alternatives, modes.c                   \         P                  ! VR W#R7      p\         P                  ! V\        P                  W#R7      p\        VP                  VP                  4       \        VP                  VP                  4       \        VP                  VP                  4       \        VP                  VP                  4       R# )rW  r  r  N)r`   kstestks_1sampr   ndtrr2   r@  r?  r3   statistic_locationstatistic_sign)rh   rk   r  r  r  r  result_1samps   &&&&&  rr   _test_kstest_and_ks1samp#TestKSTest._test_kstest_and_ks1samp%  s    a[L~~a2=J((,*@*@A|':':;1163L3LM--v/D/DEru   c                    VP                  R^^	4      pRp\        P                  ! V\        P                  4      p\        WCVR7       R# )r\   r=  NrZ   rY  )rA  r`   rJ  r   rL  r   )rh   ri   rk   rA  r   s   &&   rr   test_namedtuple_attributes%TestKSTest.test_namedtuple_attributes.  s7    KKAq!,
ll1gll+C3ru   c                8   VP                  R^^	4      pV P                  VR4       VP                  R^^	4      pV P                  VR4       . R	OpVP                  V4      pV P                  VR4       V P                  VRRR7       V P                  VRRR7       R# )
r\   r  r  r  r  r  NrZ   
r+  Q?g333333g(\?Q?g(\ſg{GzgHzG?g\(\gGz)rA  rP  r^   r   s   && rr   test_agree_with_ks_1samp#TestKSTest.test_agree_with_ks_1samp5  s    KKAq!%%a5KKR#%%a5NJJqM%%a5%%a%A%%ag%>ru   r  r  c                   \         P                  P                  R 4      pVP                  R4      w  rVVP                  V4      VP                  V4      re\        P
                  ! WVW!R7      p\        P                  ! WVW!R7      p\        VP                  VP                  4       \        VP                  VP                  4       \        VP                  VP                  4       \        VP                  VP                  4       R# )i)r  r  N)rJ   rN   rT   )r   r  r  r^   r`   rJ  ks_2sampr3   r@  r?  rM  rN  )	rh   r  r  ri   r  rk   rl   r   r   s	   &&&&     rr   test_agree_with_ks_2samp#TestKSTest.test_agree_with_ks_2sampD  s     ii##I.zz*%zz!}bjjm1ll1[HnnQ{Js}}5

CJJ/..0F0FG**C,>,>?ru   c                  a SP                  SP                  ) ^ ^SP                  .4      pV3R lp\        P                  ! W#4      p\        P                  ! W#RR7      pSP	                  VP
                  4      '       g   Q hSP	                  VP                  4      '       g   Q hSP	                  VP                  4      '       g   Q hSP	                  VP                  4      '       g   Q h\        VP
                  VP
                  4       \        VP                  VP                  4       \        VP                  VP                  4       \        VP                  VP                  4       R# )r~   c                 j   < SP                  SP                  V 4      V ) 4      SP                  ,          # r#  )atan2r  pi)rk   ri   s   &rr   cdf+TestKSTest.test_pm_inf_gh20386.<locals>.cdfU  s'    88BLLOaR025588ru   T)_no_decoN)
r^   r   r`   rJ  r  r@  r?  rM  rN  r3   )rh   ri   r  re  r   r   s   &f    rr   test_pm_inf_gh20386TestKSTest.test_pm_inf_gh20386Q  s    zzBFF7Aq"&&12	9ll4%ll4t4{{3==)))){{3::&&&&{{3112222{{3--....s}}5

CJJ/..0F0FG**C,>,>?ru   r   N)autorR   )rj  r  asymprm  )r   r   r   r   r8  rP  rS  r[  r   r   r   r_  rh  r   r   r   s   @rr   rG  rG  !  sh     VF4? [[X'AB[[],LM	@ N C	@@ @ru   rG  c                     a  ] tR tRt o RtRR/R lt]P                  P                  R. RO4      R 4       t	]P                  P                  R. RO4      R	 4       t
R
 t]P                  P                  R]P                  ]P                  .4      ]P                  P                  R. RO4      R 4       4       tR tRtV tR# )TestKSOneSampleie  zQ
Tests kstest and ks_samp 1-samples with K-S various sizes, alternatives, modes.
r  rj  c                   WgP                   8X  d   R MRp\        P                  ! V\        P                  W%R7      p	VP                  W6R7      p
VP                  WFR7      p\        V	P                  WR7       \        V	P                  WR7       R# )gvIh%,=h㈵>rI  r;   r  N)	r   r`   rK  r   rL  r^   r2   r@  r?  )rh   rk   r  expected_statisticexpected_probr  r<   ri   r  r   ref_statisticr_  s   &&&&&$$$    rr   _testOneTestKSOneSample._testOnek  se    +unnQ+Q

#5
CZZZ;
}@

J:ru   r<   Nc           
        Vf   \        V4      M
\        W!4      pVP                  R^^	VR7      pV P                  VRRRWR7       VP                  R^^	VR7      pV P                  VRRRWR7       . ROpVP	                  W1R7      pV P                  VRRR	WR7       V P                  VR
RRRWR7       V P                  VRRRRWR7       R # )Nr;   r  g|N?g7.s?r<   ri   g{CTp?g M<b*?gZL?g86J4?r  g+?r  )r  r<   ri   r  gHD?gr?rZ   rW  rX  )r&   rl  rA  rs  r^   rY  s   &&& rr   test_agree_with_r!TestKSOneSample.test_agree_with_rt  s     ). $72;MKKAqK.a&9;N! 	 	* KKR%K0a&9;O! 	 	* OJJqJ&a&79J! 	 	*a$57Hw! 	 	*a!24EG! 	 	*ru   c           
     (   Vf   \        V4      M
\        W4      p\        P                  P	                  R^dRR7      pVP                  W2R7      pV P                  VRRRRWR	7       V P                  VR
RRWR7       V P                  VRRRWR7       R # )Nr@  ih:)r*  r  r  r;   r  ghх?g7?rk  )r  ri   r<   r  g^h?)ri   r<   r  gv!ԉ}?g*z)?)r&   rl  r`   rW  r  r^   rs  )rh   ri   r<   rk   s   &&& rr   test_known_examples#TestKSOneSample.test_known_examples  s     ). $72;MJJNNs9NEJJqJ&a&9;O"r 	 	8a!46J 	 	*a$9;N 	 	*ru   c           	        \        \        P                  ! \        \        P                  ^R4      4      4       \        \        RR7      ;_uu_ 4        \        R^R4       RRR4       \        \        P                  ! \        R^R4      4      4       \        P                  ! . RO4      p\        \        VR	^4      P                  \        \        \        .R7       R#   + '       g   i     L; i)
r\   Tzn is not integral: 1.5r   r7   N)dtypesrZ   ))r  r\   Tr   )r  皙?Tr   )r  r~   Tr   )r  gTr   )          ?Tr   )r  r  Fr   )r  r6   Tgb?)r  r6   FgiNq>)r  g      ?Tgv(?)r        ?TgT	?)@  g\(\?Fr   )r  g      ?Fg(>)r  gQ?Fg@c?)r  g      ?FgX+ʳP?r  )r   r   rx  r   rg   r   r   r^   r   checkr  r  bool)rh   datasets   & rr   test_ks1samp_allpaths%TestKSOneSample.test_ks1samp_allpaths  s    D123:-EFFCD! GQ-./** 
 4 	'9a066sE4>P6Q= GFs   CC-	ksfunc*alternative, x6val, ref_location, ref_signc                   VP                  R 4      R,           p\        P                  ! V4      ^,          P                  V4      pR pV! VP	                  V4      WR7      p	VP
                  VP                  8X  d   RMRp
\        V	P                  VP	                  R4      V
R7       \        V	P                  VP	                  V4      4       \        V	P                  VP	                  WVP                  R7      4       R	# )
r   r6   c                     V R ,          # )r   r   r  s   &rr   re  /TestKSOneSample.test_location_sign.<locals>.cdf  s
    1s7Nru   r  rE  r	  rC  r  r;   N)r   rd   re   rf   r^   r<   r   r2   r@  r3   rM  rN  int8)rh   r  r  x6valref_locationref_signri   rk   re  r   r  s   &&&&&&&    rr   test_location_sign"TestKSOneSample.test_location_sign  s     IIcNS FF1IaLU#"RZZ]CA2::-u4rzz#TB..

<0HI**BJJxwwJ,OPru   c           	        \         P                  P                  ^*4      pVP                  ^
4      p\        P
                  ! \        RR7      ;_uu_ 4        \        P                  ! V\        P                  RR7       RRR4       \        P
                  ! \        RR7      ;_uu_ 4        \        P                  ! V\        P                  RRR7       RRR4       \        P
                  ! \        RR7      ;_uu_ 4        \        P                  ! V\        P                  RR	R7       RRR4       \        P
                  ! \        RR7      ;_uu_ 4        \        P                  ! V\        P                  P                  RR7       RRR4       R#   + '       g   i     EL; i  + '       g   i     L; i  + '       g   i     L; i  + '       g   i     R# ; i)
r  zInvalid value for methodr   any_other_stringr  Ntypor  rT  r  )r   r  r  r  r   r   r   r`   rK  r   rL  rJ  rW  re  )rh   r  rk   s   &  rr   test_invalid_method_raises*TestKSOneSample.test_invalid_method_raises  s    ii##B'#]]:-GHHNN1gll3EF I ]]:-GHHNN1gll6'02 I ]]:-GHHNN1gll6'-/ I ]]:-GHHLLEJJNN3EF IH IHH IH IH IHHs0   (F))F#>)F62G	F 	#F3	6G		G	r   r  ))r  r   r   r\   )r  r   r   rZ   )r  r   r   r\   )r  r   r   rZ   )r   r   r   r   r8  rs  r   r   r   rw  rz  r  r`   rJ  rK  r  r  r   r   r   s   @rr   rm  rm  e  s     ;; [[W&BC* D*( [[W&BC
* D
*!RF [[Xenn'EF[[I9:
Q: GQG Gru   rm  c                   d  a  ] tR tRt o RtR tR"R ltR tR tR t	R t
R	 tR
 tR t]P                  P                   R 4       t]P                  P                   R 4       tR tR t]P                  P*                  R 4       t]P                  P*                  R 4       t]P                  P                   R 4       tR t]P                  P                   ]! RR7      R 4       4       t]P                  P9                  R. R.3R.. 3. . 334      R 4       t]P                  P*                  R 4       t]P                  P9                  R. R#O4      R 4       t]P                  P9                  R] PB                  ] PD                  .4      ]P                  P9                  R. R$O4      R  4       4       t#R!t$V t%R# )%TestKSTwoSamplesi  z<Tests 2-samples with K-S various sizes, alternatives, modes.c                r    \        \        V4      VR4       V F  p\        VP                  VJ 4       K  	  R# )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.
znumber of warningsN)r   r  r   category)rh   	warn_listexpected_typeexpected_lenwarn_s   &&&& rr   _check_warnings TestKSTwoSamples._check_warnings  s0     	S^\3GHEENNm34 ru   c                  VP                  V4      VP                  V4      r!\        WR VR7      p\        P                  ! WW6R7      w  rVP                  WHR7      pVP                  WXR7      p\	        W4       \	        WRR7       R# )Tforce_floatingri   rV  r;   rh  rF  N)r^   r,   r`   r^  r2   )rh   r  r  r  rr  r_  r  ri   r<   res_statistic
res_pvalues   &&&&&&&$   rr   rs  TestKSTwoSamples._testOne  sn    BBBrdrB$)NN2;$R!

=
>ZZ
Z8
5
 	
U;ru   c           	     2   V P                  ^ .^.RRRVR7       V P                  ^ .^.RRRVR7       V P                  ^ .^.RRRVR7       V P                  ^.^ .RRRVR7       V P                  ^.^ .RRRVR7       V P                  ^.^ .RRRVR7       R# )	r~   r  r   r=  r  r6   r  r   N)rs  rQ  s   &&rr   	testSmallTestKSTwoSamples.testSmall  s    qcA3UCB?qcA3	5#"=qcA3sr:qcA3UCB?qcA3	5#"=qcA3sr:ru   c           	     ~   VP                  R R.4      pVR,           pVR,
          pVP                  . RO4      pV P                  W5RRR VR7       V P                  W5RRRVR7       V P                  W5RRRVR7       V P                  WERRRVR7       V P                  WERRR	VR7       V P                  WER^ R VR7       R
# )r   r   r  r  r=  r  rD  r  r{  rF  Nr  rW  rG  r^   rs  rh   ri   r  data1pdata1mr-  s   &&    rr   testTwoVsThreeTestKSTwoSamples.testTwoVsThree  s    

C:&

?+f['32FfYDfVWcbAf['32FfYDfVQ;ru   c           	     ~   VP                  R R.4      pVR,           pVR,
          pVP                  . RO4      pV P                  W5RR	R
VR7       V P                  W5RR	RVR7       V P                  W5RRRVR7       V P                  WERRRVR7       V P                  WERRRVR7       V P                  WER^ R VR7       R# )r   r   r  r  r=  r  r  Nr  r6   g?g?r  皙?r  rz  r@  r  r  s   &&    rr   testTwoVsFourTestKSTwoSamples.testTwoVsFour  s    

C:&

/0f['7rJfYBGfVWg"Ef['6bIfYBGfVQ;ru   c           	     t   VP                  ^^d^d4      pV^,           R,           pV^,           R,
          pV P                  W#RRRVR7       V P                  W#RRRVR7       V P                  W#R^ RVR7       V P                  W$RRRVR7       V P                  W$RRR	VR7       V P                  W$R^ RVR7       R
# )r\   rC  r  gz?r=  r  gz;.B?r  r   gn2IU?Nr  g{Gz?rA  rs  )rh   ri   x100	x100_2_p1	x100_2_m1s   &&   rr   test100_100TestKSTwoSamples.test100_100  s    {{1c3'1HsN	1HsN	d{I?PUWXdy)=OTVWdvq#"=d{IsrJdy)=NSUVdvq#"=ru   c           	        VP                  ^^d^d4      pVP                  ^^d^n4      pV^,           R,           pV^,           R,
          pV P                  W$RRRVR7       V P                  W$RRRVR7       V P                  W$R^ ^VR7       V P                  W%RRRVR7       V P                  W%RRR	VR7       V P                  W%RR
RVR7       R# )r\   rC  r  g@߿?r=  r  g}n?r  g]O);?gѫ);?r   r   Ng?g:[?r  )rh   ri   r  x110
x110_20_p1
x110_20_m1s   &&    rr   test100_110TestKSTwoSamples.test100_110(  s    {{1c3'{{1c3'BY_
BY_
d"$8R 	 	Ad	"$8R 	 	Adr 	 	#d"$8R 	 	Ad	"$8R 	 	Ad32 	 	'ru   c           	        VP                  ^.^,          ^.^,          ,           ^.^,          ,           ^.^,          ,           4      pV^,           pVP                  ^.^,          ^.^,          ,           ^.^
,          ,           ^.^,          ,           4      pVP                  ^.^
,          ^.^,          ,           ^.^
,          ,           ^.^,          ,           4      pV P                  W#RRRVR7       V P                  W#RRRVR7       V P                  W#RRRVR7       V P                  WERRR	VR7       V P                  WERRR
VR7       V P                  WERRRVR7       R# )rJ   r  g^dH?r=  r  gQl6y?r  r   r   g(f^?g!?g+ ϖ?Ng      ?gVdEVdE?gii?r  )rh   ri   x2233x3344x2356x3467s   &&    rr   testRepeatedValues#TestKSTwoSamples.testRepeatedValues;  s6   

A37aS1W,sQw6!q@A	

A37aS1W,sRx71#'AB

A38qcAg-b8A37BCeK9KPRSeIv7JrReFFCB?eK>PUWXeI{<NSUVeFJ8JrRru   c           	     6   VP                  . R	O4      pV P                  W"^,           RR
R VR7       V P                  W"^,           RR
RVR7       V P                  W"^,           RRR VR7       V P                  W"R,           RR
R VR7       V P                  W"R,           RR
RVR7       V P                  W"R,           RRR VR7       V P                  W"R,
          RR
R VR7       V P                  W"R,
          RRR VR7       V P                  W"R,
          RR
RVR7       R# )r   r  r=  r  r  r  r   r6   Nr  rW  r  )rh   ri   r-  s   && rr   testEqualSizesTestKSTwoSamples.testEqualSizesG  s    

?+e1Wk5#"Ee1WiDe1WfeRB?e3YUCBGe3Y	5$2Fe3YrbAe3YUCBGe3Y	5#"Ee3YtCru   c                (   R Rr2RV,          V,          ^,          ^,          pVP                  RRV4      V,
          pVP                  RRV4      pV P                  WVRRV,          V,          RRVR7       V P                  WVRRV,          V,          RR	VR7       V P                  WVR
RV,          V,          RR	VR7       V P                  WVRRV,          V,          RR	VR7       \        P                  ! 4       ;_uu_ 4        Rp\        P                  ! RV\
        4       V P                  WVR
RV,          V,          RRVR7       V P                  WVRRV,          V,          RRVR7       RRR4       \        P                  ! RR7      ;_uu_ 4       p\        P                  ! R4       V P                  WVRRV,          V,          RRVR7       V P                  V\
        ^4       RRR4       R#   + '       g   i     L; i  + '       g   i     R# ; i)r(  iX  r         i@r   r  g     @@rj  r  ri   rk  r  g2JE?r  g     @@gsW\nc?)ks_2samp: Exact calculation unsuccessful.r   r  NTrecordalwaysrA  rs  r   r   r   r   r  r  	rh   ri   n1n2deltark   rl   r  r!  s	   &&       rr   testMiddlingBoth!TestKSTwoSamples.testMiddlingBothS  s    cBBr	!AKKD"%-KKD"%aK"r)93!b 	 	*aK"r)93"r 	 	+aIv{R'79K"r 	 	+aFEBJO5G"r 	 	+$$&&AG##Hg~FMM!	6B;+;=O&2  /MM!
R9K&2  / ' $$D11Q!!(+MM!
R9K&2  /  NA6	 21 '& 211    A)G-AH -G=	 H	c                (   R Rr2RV,          V,          ^,          ^,          pVP                  RRV4      V,
          pVP                  RRV4      pV P                  WVRRV,          V,          RRVR7       V P                  WVRRV,          V,          RR	VR7       V P                  WVR
RV,          V,          RRVR7       V P                  WVRRV,          V,          RRVR7       \        P                  ! 4       ;_uu_ 4        Rp\        P                  ! RV\
        4       V P                  WVR
RV,          V,          RRVR7       V P                  WVRRV,          V,          RRVR7       RRR4       \        P                  ! RR7      ;_uu_ 4       p\        P                  ! R4       V P                  WVRRV,          V,          RRVR7       V P                  V\
        ^4       RRR4       R#   + '       g   i     L; i  + '       g   i     R# ; i)r  L  r   r  r   r  g     ȹ@rk  r  rj  r  gZ?r  g     @@g@J?r  r   r  NTr  r  r  r  s	   &&       rr   testMediumBothTestKSTwoSamples.testMediumBotho  s    tBBr	!AKKD"%-KKD"%aK"r)93"r 	 	+aK"r)93!b 	 	*aIv{R'79K"r 	 	+aFFRK"$46H"r 	 	+ $$&&AG##Hg~FMM!	6B;+;=O&2  /MM!b(8:L&2  / ' $$D11Q!!(+MM!b(8:L&2  /  NA6	 21 '& 211r  c           	     b   R ^nr2VR,          pRV,          V,          ^,          ^,          pVP                  RRV4      V,
          pVP                  RRV4      pV P                  WgRRV,          RVR	7       V P                  WgR
RV,          RVR	7       V P                  WgRRV,          RVR	7       R# )'  r  r   r  r         Y@r  g    `@g      <r=  r  g     @g򼉷?r  gimb:Nr  )rh   ri   r  r  lcmr  rk   rl   s   &&      rr   	testLargeTestKSTwoSamples.testLarge  s    BgBr	!AKKD"%-KKD"%aK38NSUVaIus{4GBOaFGcM3IbQru   c           
     "   \         P                  P                  R 4      pVP                  RR7      P	                  4       pVP                  RRR7      P	                  4       pV P                  W4RRRR	VR
7       V P                  W4RRRRVR
7       R# )@ i  r  i  r7   r  r+  r  gC?      <rk  r  r  Nr   r  r  r  r   rs  rh   ri   r  rk   rl   s   &&   rr   test_gh11184TestKSTwoSamples.test_gh11184  s    ii##F+JJDJ!((*JJDJ,335aK)<>T"r 	 	+aK)<>T"r 	 	+ru   c           
     ~   \         P                  P                  R 4      pVP                  RR7      P	                  4       pVP                  RRR7      P	                  4       pV P                  W4RRRR	VR
7       V P                  W4RRRRVR
7       V P                  W4RRRR	VR
7       V P                  W4RRRR	VR
7       R# )r  r  r  i'  r7   r  r  g 	_r!?gAD5rk  r  r  r  r  gלN#y7r  gvqw?Nr  r  s   &&   rr   test_gh11184_bigger$TestKSTwoSamples.test_gh11184_bigger  s     ii##F+JJEJ"))+JJEJ-446aK)<>T"r 	 	+aK)<>T"r 	 	+aI':<R"r 	 	+aF$79O"r 	 	+ru   c                   \         P                  P                  R 4      p\        RRR4       F  pVP	                  VP                  VR7      4      pVP	                  VP                  V^
,           RR7      4      p\        P                  ! WERR7      P                  p\        P                  ! WERR7      P                  p\        V^V,          4       \        V^V,          4       K  	  R	# )
r  r  i.  r  r6   r  r*  r  rV  rk  N)
r   r  r  r  r^   r  r`   r^  r?  r4   )rh   ri   r  rk   vals1vals2r  rk  s   &&      rr   test_gh12999TestKSTwoSamples.test_gh12999  s    ii##F+tUD)AJJszzz34EJJszzBSzABENN5g>EEENN5g>EEE5!e),5!e), *ru   c                   R Rr2VR,          pRV,          V,          ^,          ^,          pVP                  ^^V4      V,
          pVP                  ^^V4      pV P                  WgRRV,          RRVR7       V P                  WgRRV,          R	R
VR7       V P                  WgRRV,          RRVR7       V P                  WgRRV,          RVR7       V P                  WgRRV,          RVR7       \        P                  ! 4       ;_uu_ 4        Rp\        P                  ! RV\
        4       V P                  WgRRV,          RR
VR7       V P                  WgRRV,          RR
VR7       RRR4       R#   + '       g   i     R# ; i)r  i*  r  r   r  g     @gkHY?rk  r  gLɔ.?r  rj  r  g.LbG2?r=  r  r   gr?֎?r  r   N)rA  rs  r   r   r   r   )	rh   ri   r  r  r  r  rk   rl   r  s	   &&       rr   testLargeBothTestKSTwoSamples.testLargeBoth  sl    BgBr	!AKK3#e+KK3#aK6H"r 	 	+aK6H"r 	 	+aK6H!b 	 	*aIus{4F2NaFD3J0BrJ$$&&AG##Hg~FMM!	53;8J&2  /MM!s
4F&2  / '&&&s   	AE..E?	c                    Rp\         P                  ! VP                  ^^.4      VP                  ^.4      4      p\        W2VR7       R# )r@  r=  NrY  )r`   r^  r^   r   )rh   ri   rA  r   s   &&  rr   testNamedAttributes$TestKSTwoSamples.testNamedAttributes  s8    ,
nnRZZA/QCAC3ru   Tr  c           
     
   ^ RI HpHp V! ^^^^4       V! R^^R4       \        P                  ! RR7      ;_uu_ 4        \        \        VRR^^4       \        \        VRR^^4       R	R	R	4       R	#   + '       g   i     R	# ; i)
r~   )_count_paths_outside_method!_compute_outer_prob_inside_methodr  i  r   r  r  iK  rA  N)scipy.stats._stats_pyr  r 	  r   r  r   FloatingPointError)rh   r  r 	  s   &  rr   test_some_code_paths%TestKSTwoSamples.test_some_code_paths  sq    	

 	*!Q15#D!Q5[[)),.Ia,,.Ia, *)))s   )A11B	r  r   c           	        \        \        \        VR 7      ;_uu_ 4        \        P                  ! VP                  V^ ,          4      VP                  V^,          4      4      p\        VP                  VP                  \        P                  4      4       \        VP                  VP                  \        P                  4      4       RRR4       R#   + '       g   i     R# ; irg  )r"   r   r   r`   r^  r^   r3   r@  r   rg   r?  )rh   r  ri   r   s   &&& rr   test_argument_checking'TestKSTwoSamples.test_argument_checking  s     +3HRPP..DG!4bjja6IJCCMM2::bff+=>CJJ

266(:; QPPPs   B*CC$	c                j   \         P                  P                  R4      pRpVP                  \        P
                  P                  W2R7      P                  4       4      pV^,           p\        P                  ! WERRR7       \        P                  ! WERRR7       \        P                  ! WERRR7       R	# )
zEnsure gh-12218 is fixed.l   9B i    r  r  r  rk  rI  r  r  N)	r   r  r  r^   r`   rM  r  r   r^  )rh   ri   r  r  rvs1rvs2s   &&    rr   test_gh12218TestKSTwoSamples.test_gh12218  s    
 ii##J/zz%--+++FMMOPaxtywGtvGDt{Iru   r<   Nc           
        Vf   \        V4      M
\        W!4      p\        P                  P	                  RR7      pVP                  VP                  RR7      R,           VR7      pVP                  VP                  RR7      VR7      pRp\        \        WbR	7      ;_uu_ 4        \        P                  ! WER
R7      p\        VP                  VP                  RVR7      RR7       R R R 4       R #   + '       g   i     R # ; i)Ni{fr  iq  r  r6   r;   iq  z(ks_2samp: Exact calculation unsuccessfulr^  r  r  r   rh  rF  )r&   rl  r   r  r  r^   r"   r   r`   r^  r2   r?  )rh   r<   ri   r  r  r-  r  r   s   &&&     rr   test_warnings_gh_14019'TestKSTwoSamples.test_warnings_gh_14019  s    (- $72;M ii###2

3::3:/#5U
C

3::3:/u
=<w>>..6BCCJJ

2U
(C%P ?>>>s   #AC//D 	r  r  c                   VP                  R VP                  R7      p\        V4      p\        P                  ! V4      ^,          P                  V4      p\        P                  ! WxVR7      p	\        V	P                  VP                  RVP                  R7      RR7       V	P                  V8X  g   Q hV	P                  V8X  g   Q hR# )r   r;   r  rC  rE  r  N)r   r   r/   rd   re   rf   r`   r^  r2   r@  r^   rM  rN  )
rh   r  r  r  r  r  ri   rk   rl   r   s
   &&&&&&&   rr   r  #TestKSTwoSamples.test_location_sign  s     IIcI,AJFF1IaLU#nnQ{;rzz#RZZz'HuU%%555!!X---ru   r   )rj  r  ))r  皙@r	  r\   )r  ffffff@r   rZ   )r  r	  r	  r\   )r  r	  r   rZ   )&r   r   r   r   r8  r  rs  r  r  r  r  r  r  r  r   r   r  r  r  r  r  rn  r  r  r  r  r   r	  r   r	  r	  r	  r`   rJ  r^  r  r   r   r   s   @rr   r  r    s    F5<;
<<	>'&
S
D [[7 76 [[7 78	R+ [[+ + [[	- 	- [[/ /.4 [[d#, $ ,  [[VrB4jB4*r2h%GH< I< [[
J 
J [[W&BCQ DQ [[Xenn'EF[[I;<

.< G
.ru   r  z)ignore:divide by zero:RuntimeWarning:daskz(ignore:invalid value:RuntimeWarning:daskc                     a  ] tR tRt o R tR tR t]! RRR7      R 4       tR	 t	]! RR
R7      R 4       t
]! RRR7      R 4       tR t]! RRR7      ]P                  P                  R]P"                  ! R4      R3]P"                  ! R4      R3.4      R 4       4       tR t]! RRR7      ]P                  P                  R. R O4      R 4       4       tRtV tR# )!TestTTestReli"  c           
        VP                  R VP                  R7      pVP                  RVP                  R7      pVP                  ^^d^dVP                  R7      pVP                  RR^dVP                  R7      pVP                  VP                  ^^d^dVP                  R7      VP                  RR^dVP                  R7      .4      pVP                  VP                  RR^dVP                  R7      VP                  ^^d^dVP                  R7      .4      p\        P
                  ! WE^ R7      w  r\        W4       \        W4       \        P
                  ! VP                  VP                  ^ R7      w  r\        WP                  W") .4      4       \        WP                  W3.4      4       \        P
                  ! Wg^R7      w  r\        WP                  W") .4      4       \        WP                  W3.4      4       R# )u?r;   ȵI?r  KX@rC   N)r^   r   rA  r  r`   	ttest_relr2   r   )
rh   ri   trrO  r
	  r	  rvs1_2Drvs2_2Dre  rf  s
   &&        rr   test_ttest_rel_xpTestTTestRel.test_ttest_rel_xp&  s    ZZ+2::Z>ZZ+2::Z>{{1c3bjj{9{{4BJJ{?((Ac3bjj9D&#RZZ@
 ((D&#RZZ@AsCrzz:
 t2wyy'))!<88RI./88RH-.wa888RI./88RH-.ru   c           
     
   VP                   pVP                  R VR7      pVP                  RVR7      pVP                  W3) .4      VP                  WD.4      3pVP                  ^^d^dVR7      pVP                  RR^dVR7      pVP                  VP                  ^^d^dVR7      VP                  RR^dVR7      .4      pVP                  VP                  RR^dVR7      VP                  ^^d^dVR7      .4      p	\        P
                  ! Wg^ R7      w  r\        W4       \        W4       \        P
                  ! VP                  V	P                  ^ R7      w  r\        W^ ,          4       \        W^,          4       \        P
                  ! W^R7      w  r\        W^ ,          4       \        W^,          4       \        P                  ! 4       ;_uu_ 4        \        P                  ! RRR7      ;_uu_ 4        \        P                  ! RR\        4       \        P
                  ! VP                  R	4      VP                  R
4      4      w  rRRR4       RRR4       VP                  V
4      '       g   Q hVP                  V4      '       g   Q hRp\        P
                  ! Wg^ R7      p\        WVR7       VP                  WV.4      pVP                  WV	.4      p\        P
                  ! WRR7      w  r\        WP                  V^ ,          3^,          4      4       \        WP                  V^,          3^,          4      4       \        P
                  ! VP!                  V^^4      VP!                  V^^4      ^R7      w  r\        WP                  V^ ,          3^,          4      4       \        WP                  V^,          3^,          4      4       Rp\"        P$                  ! \&        VR7      ;_uu_ 4        \        P
                  ! WgRR7       RRR4       \        P
                  ! Wg^ RR7      w  r\        WP                  ^V^,          ,
          4      4       \        WP                  V4      4       \        P
                  ! Wg^ RR7      w  r\        WP                  V^,          4      4       \        WP                  V4      4       R#   + '       g   i     EL; i  + '       g   i     EL; i  + '       g   i     L; i)r	  r;   r	  r  r	  rC   r   rV  rX  r   r   Nr=  rr  r   rE  r  r  r  r  rY  rZ   )r   r^   r  rA  r`   r	  r2   r   r   r   r   r  r   r   rx  r   r  r   r   r   )rh   ri   r<   r	  rO  tprr
	  r	  r	  r	  re  rf  rA  r   rvs1_3Drvs2_3Dr  s   &&               rr   test_ttest_relTestTTestRel.test_ttest_relK  s   

 ZZ+5Z9ZZ+5Z9hhCy!288RH#55{{1c3e{4{{4E{:((BKK35KAKKfcKGI J((BKKfcKGKK35KAC D t2oogii;q6"q6"oogQ7q6"q6" $$&&HX>>##=~O??2::b>2::b>BDA ? '
 xx{{{xx{{{ -
oodq1C3 ((Gg67((Gg67wb988SVIaK0188SVIaK01r{{7Aq9!{{7Aq9$%' 	88SVIaK0188SVIaK01 M]]:W55OODG< 6 tvF::a"Q$h/0::b>*tyI::bd+,::b>*I ?>> '&&8 65s1   !S!3AS	S!7S5S	S!!S2	5T	c                   \         P                  P                  R 4      p\        P                  P                  ^^
RVR7      p\         P                  VR&   \        P                  P                  ^^
RVR7      \        P                  P                  RRVR7      ,           p\         P                  VR&   VP                  VP                  4       4      VP                  VP                  4       4      rC\        P                  ! W34      p\        VP                  VP                  VP                  4      4       \        VP                  VP                  VP                  4      4       R# )Na   rj  r(  r@  r+  r  r  N)r   r  r  r`   rW  r  rg   r^   r   r	  r3   r@  r?  )rh   ri   r  rk   rl   r   s   &&    rr   test_nan_policy_propagate&TestTTestRel.test_nan_policy_propagate  s    ii##H-JJNNq3NG#ZZ^^#C^HZZ^^#Cc^BC#zz!((*%rzz!((*'=1ooa#rzz"&&'9:

BJJrvv$67ru   Tz lazy->limited nan_policy supportr   c                   \         P                  P                  R 4      p\        P                  P                  ^^
RVR7      p\         P                  VR&   \        P                  P                  ^^
RVR7      \        P                  P                  RRVR7      ,           p\         P                  VR&   VP                  VP                  4       4      VP                  VP                  4       4      rC\        P                  ! W4RR7      p\        VP                  VP                  R4      4       \        VP                  VP                  R	4      4       R
p\        P                  ! \        VR7      ;_uu_ 4        \        P                  ! W4RR7       RRR4       Rp\        P                  ! \        VR7      ;_uu_ 4        \        P                  ! W4RR7       RRR4       R#   + '       g   i     LY; i  + '       g   i     R# ; i)r)	  r*	  rj  r(  r@  r+	  r   r   ghm#1?gry̧?r   r   r   Nrn  r   )r   r  r  r`   rW  r  rg   r^   r   r	  r2   r@  r?  r   r   r   )rh   ri   r  rk   rl   r   r  s   &&     rr   r  TestTTestRel.test_nan_policy  sS    ii##H-JJNNq3NG#ZZ^^#C^HZZ^^#Cc^BC#zz!((*%rzz!((*'=1ooav6rzz2E'FG

BJJ/A$BC1]]:W55OOAW5 6 .]]:W55OOAX6 65	 65 655s   8G"?G5"G2	5H	c           	        \        \        R VR7      ;_uu_ 4        \        P                  ! VP	                  . R	O4      VP	                  . R
O4      4      w  r#RRR4       \        VP                  X4      VP	                  VP                  4      4       \        XVP	                  R4      4       \        P                  ! RR7      ;_uu_ 4        \        P                  ! VP	                  . R	O4      VP	                  . R	O4      4      pRRR4       \        XP                  VP	                  VP                  4      4       \        VP                  VP	                  VP                  4      4       VP	                  R\        P                  .RR..4      p\        P                  ! WQP                  R4      4      p\        VP                  VP	                  ^ \        P                  .4      4       \        VP                  VP	                  ^\        P                  .4      4       VP                  ^4      pRp\        P                   ! \"        VR7      ;_uu_ 4        \        P                  ! VP%                  VR4      VP%                  VR4      4       RRR4       R#   + '       g   i     ELL; i  + '       g   i     EL; i  + '       g   i     R# ; i)r  r^  Nr   r   r  r   zXArray shapes are incompatible for broadcasting.|operands could not be broadcast togetherr   r~   r~   r~   r  rQ  r  )rK   r   rJ   r   r   )r"   r   r`   r	  r^   r3   absr   r   r  r@  rg   r?  rX  r   r   r   r   rb   )rh   ri   re  rf  r   ananrk   r  s   &&      rr   test_edge_casesTestTTestRel.test_edge_cases  s   /HRPP??2::i#8"**Y:OPDA Q 	q	2::bff#562::c?+[[**//"**Y"7I9NOC +rzz"&&'9:

BJJrvv$67zzB<#r34oodHHV$45rzz1bff+'>?

BJJ266{$;< IIbM>]]:W55OOBJJq&12::a3KL 65) QPP +** 655s#   <J:J.8KJ+	.J?	K	z"multidimensional 'omit' is np-onlyr  c           
     L   \         P                  ! \         P                  ! ^^d^d4      \         P                  ! RR^d4      .4      p\         P                  ! \         P                  ! RR^d4      \         P                  ! ^^d^d4      .4      p\         P                  VR&   \         P                  VR&   \         P                  R 4       p\
        P                  ! \        \        R7      ;_uu_ 4        \        P                  ! W#^ RR7      w  rVRRR4       \
        P                  ! \        \        R7      ;_uu_ 4        \        P                  ! W#^ RRR	7      w  rxRRR4       \        XXR
R7       \         P                  ! RR7      ;_uu_ 4        \        XV! VXR4      R
R7       RRR4       \
        P                  ! \        \        R7      ;_uu_ 4        \        P                  ! W#^ RRR	7      w  rxRRR4       \        WuR
R7       \         P                  ! RR7      ;_uu_ 4        \        XV! VXR4      R
R7       RRR4       R#   + '       g   i     ELD; i  + '       g   i     EL; i  + '       g   i     L; i  + '       g   i     L; i  + '       g   i     R# ; i)r\   r  r	  c                 l    V ^ 8  d   VR8X  g   V ^ 8  d   VR8X  d
   V^,          # ^V^,          ,
          # r~   r  r  r   re  rf  alts   &&&rr   	converterETestTTestRel.test_multidimensional_nan_policy_omit.<locals>.converter  s3    A#-QUsi7G1uA;ru   r   r   r   Nr  rl  rh  r  r   r  r  )r   :r  r$  N)r   :rw  r  N)r   r  rA  rg   	vectorizer   r  r   r   r`   r	  r
   r  )	rh   ri   r	  r	  r<	  r	  rO  re  rf  s	   &&       rr   %test_multidimensional_nan_policy_omit2TestTTestRel.test_multidimensional_nan_policy_omit  s   ((BKK34KKfc:< =((BKKfc:KK346 7FFFF 
	 
	
 \\,4EFF__WqVLFB G \\,4EFF??7Q.4&JDA G 	2E*[[**AyR8uE + \\,4EFF??7Q.4)MDA G 	E*[[**AyR;%H +* GFF GFF +* GF +**s<   -I9II,I?#JI	I)	,I<	?J	J#	"lazy -> limited nan_policy supportc                   VP                  VP                  R RR.4      pVP                  . R
O4      p\        P                  ! W#RR7      p\        P                  ! W2RR7      p\	        VP
                  VP
                  ) 4       \	        VP                  VP                  4       \        P                  ! VR,          VR,          4      p\	        VP
                  VP
                  4       \	        VP                  VP                  4       \	        VP
                  VP                  R4      RR7       \	        VP                  VP                  R4      RR7       R	# )r   r   r   r   r   r  -C6?rF  grh|?Nr   r   r   r   r  )r^   rg   r`   r	  r2   r@  r?  )rh   ri   rk   rl   r  r  r3s   &&     rr   test_ttest_rel_nan_2nd_arg'TestTTestRel.test_ttest_rel_nan_2nd_arg  s     JJS#./JJ+,__Qf5__Qf5r||m4		299- __QrUAbE*bll3		299- 	bjjoDA		2::f#5DAru   c                   \        \        \        VR 7      ;_uu_ 4        \        P                  ! VP                  . 4      VP                  . 4      4      pRRR4       \        X\        P                  P                  4      '       g   Q h\        VP                  VP                  VP                  4      4       \        VP                  VP                  VP                  4      4       R#   + '       g   i     L; irg  )r"   r   r   r`   r	  r^   rC  	_stats_pyTtestResultr3   r@  rg   r?  r7  s   && rr   #test_ttest_rel_empty_1d_returns_nan0TestTTestRel.test_ttest_rel_empty_1d_returns_nan  s     +3HRPP__RZZ^RZZ^DF Q&%//"="=>>>>(("**RVV*<=rzz"&&'9:	 QPs   6C))C9	
dask.arrayzdask fails this oner  b, expected_shapec                *   VP                  R4      pVP                  WP                  R7      p\        P                  ! 4       ;_uu_ 4        \        P
                  ! R\        \        4       \        P                  ! WARR7      pRRR4       \        X\        P                  P                  4      '       g   Q hVP                  W#P                  R7      p\        VP                   V4       \        VP"                  V4       R#   + '       g   i     L; ir   r;   r   rC   Nr?  )r   r\   r~   rZ   )r  r^   r<   r   r   r   r   r   r`   r	  rC  rI	  rJ	  r   rg   r3   r@  r?  )rh   rX  expected_shaperi   rL  r  expected_values   &&&&   rr   test_ttest_rel_axis_size_zero*TestTTestRel.test_ttest_rel_axis_size_zero  s     HHYJJqJ($$&&##H.CEWX__Q3F ' &%//"="=>>>>FFC((.9~6 '&   9DD	c                F   VP                  R4      pVP                  R4      p\        P                  ! W#^R7      p\        V\        P                  P
                  4      '       g   Q hVP                  P                  R8X  g   Q hVP                  P                  R8X  g   Q hR# r\   rC   N)r\   rK   r~   r|   rK   r\   r|   r~   )	r  r`   r	  rC  rI	  rJ	  r@  r  r?  )rh   ri   rL  rX  r  s   &&   rr    test_ttest_rel_nonaxis_size_zero-TestTTestRel.test_ttest_rel_nonaxis_size_zero  s    
 HHYHHYA.&%//"="=>>>>%%///}}""f,,,ru   r3  z*needs stdtrit; lazy->can't use root finderr  c                   \         P                  P                  R 4      p^
pVP                  VP	                  VR^R7      P                  4       4      pVP                  VP	                  V^^R7      P                  4       4      pRRR.RR\         P                  .R\         P                  ) R./p\        P                  ! WVVR7      pVP                  R	R
7      p	\        V	P                  VP                  Wq,          ^ ,          4      4       \        V	P                  VP                  Wq,          ^,          4      4       \        VP                  VP                  V^,
          4      4       R# )l   C67l:  r7   ry  r  gi{a?r  r  gC?r  r{  r|  NgZ?Yg8 )r   r  r  r^   r  r   r   r`   r	  r  r2   r  r  r~  )
rh   r  ri   r  r  rk   rl   r   r   r  s
   &&&       rr   test_ttest_rel_ci_1d!TestTTestRel.test_ttest_rel_ci_1d*  s    ii##$78JJszzqcz;BBDEJJszzqaqz9@@BC /1BC-rvv6!457 ooa<$$d$;

3+;A+> ?@C,<Q,?!@A

1Q30ru   r   Nr\   r|   r~   r  r\   r~   r~   r   r~   rm  )r   r   r   r   r 	  r&	  r,	  r   r  r5	  r?	  rF	  rK	  r   r   r   r   r  rS	  rZ	  r]	  r   r   r   s   @rr   r	  r	  "  s    #/J?+B8 .PQ7 R7,M2 d+OPI QI@ .RSB TB,; l+@A[[0!xx	2F;!xx	2F;=>7> B7
- k*VW[[],LM1 N X1ru   r	  ztest_fun, argsc                    a V ! V3R  lV 4       !  pRp\         P                  ! \        VR7      ;_uu_ 4        VP                  ^
R7       RRR4       R#   + '       g   i     R# ; i)c              3   F   <"   T F  pSP                  V4      x  K  	  R # 5ir#  )r^   ).0argri   s   & rr   	<genexpr>#test_ttest_ci_iv.<locals>.<genexpr>I  s     5RZZ__s   !r  r   r|  N)r   r   r   r  )test_funr  ri   r   r  s   &&f  rr   test_ttest_ci_ivri	  D  sK    
 55
6CDG	z	1	14 
2	1	1	1s   AA&	ri   c                `   a Sf   \        W4      MSoRV3R llpV! W4      V! W4      ,           # )Nc                    < SP                  V 4      p SP                  WR 7      pSP                  W^R7      pV P                  V,          pW#V3# )rC   r:  )r^   rc   r   r  )rk   rD   mur   nobsri   s   &&   rr   _stats_desc_stats.<locals>._statsR  sI    JJqMWWQW"ffQaf0wwt}}ru   r  )r!   )r  r  rD   ri   rn	  s   &&&d rr   _desc_statsrp	  O  s/    $&J	 BB "fR...ru   c                   r   a  ] tR tRt o R t]! RRR7      R 4       t]! RRR7      R 4       tR tR	 t	R
t
V tR# )TestTTestIndi\  c                b   VP                  R 4      pVP                  R4      pVP                  W") .4      pVP                  W3.4      pVP                  ^^i^d4      pVP                  ^^d^d4      pVP                  Wg.4      pVP                  Wv.4      p	\        P                  ! Wg^ R7      p
V
w  r\        W4       \        W4       \        P                  ! \        Wg4      !  p
V
w  r\        W4       \        W4       \        P                  ! VP                  V	P                  ^ R7      p
\        V
P                  V4       \        V
P                  V4       \        P                  ! \        VP                  V	P                  4      !  p
\        V
P                  V4       \        V
P                  V4       \        P                  ! W^R7      p
\        V
P                  V4       \        V
P                  V4       \        P                  ! \        W^R7      !  p
\        V
P                  V4       \        V
P                  V4       Rp\        P                  ! \        VR7      ;_uu_ 4        \        P                  ! WgRR7       RRR4       \        VP                  V	P                  4      p\        P                  ! \        VR7      ;_uu_ 4        \        P                  ! VRR/  RRR4       \        P                  ! WgR	R7      w  r\        V^V^,          ,
          4       \        W4       \        P                  ! WgR
R7      w  r\        W^,          4       \        W4       \        P                  ! VP                  V	P                  ^ R	R7      p\        VP                  V	P                  4      p\        P                  ! VRR	/ p\        VP                  VP                  4       \        VP                  VP                  4       \        P                  ! VP                  V	P                  ^ R	R7      p\        VP                  V	P                  4      p\        P                  ! VRR	/ p\        VP                  VP                  4       \        VP                  VP                  4       VP                  VP                  4      pVP                  VP!                  VP"                  ^ ,          4      ^ 8H  VV4      p\        P                  ! Wg^ R7      p
\%        V
P                  V4       \%        V
P                  V4       \        P                  ! \        Wg4      !  p
\%        V
P                  V4       \%        V
P                  V4       R#   + '       g   i     EL; i  + '       g   i     EL; i)	ru?g8Mѱ?rC   rr  r   rE  r  Nr  r  r  r  )r^   r  rA  r`   	ttest_indr2   ttest_ind_from_statsrp	  r   r@  r?  r   r   r   rg   r  r   r  r3   )rh   ri   r	  rO  tr_2Dpr_2Dr
	  r	  r	  r	  r   re  rf  r  r  r  r  rr  s   &&                rr   test_ttest_indTestTTestInd.test_ttest_ind^  s   ZZ*+ZZ+,"c#""{{1c3'{{1c3'((D<(((D<(oodq1((+d*ABoogii;u-

E*((+gii*KLu-

E*oogQ7u-

E*((+gQ*OPu-

E* M]]:W55OODg> 6 799gii0]]:W55&&DGD 6 tv>11:&tyAa4  wyy'))!P799gii0))4DVD7T[[1wyy'))!P799gii0))4DVD7T[[1 jj xx		$**Q-0A5sDAoodq1s+

C(((+d*ABs+

C(M 655 655s   'V	V	V	V.	TrA	  r   c           	     $   \         P                  P                  R 4      p\        P                  P                  ^^
RVR7      p\        P                  P                  ^^
RVR7      p\         P                  VR&   VP                  VP                  4       4      VP                  VP                  4       4      rC\        P                  ! W44      p\        VP                  VP                  VP                  4      4       \        VP                  VP                  VP                  4      4       \        \        RVR7      ;_uu_ 4        \        P                  ! VP                  . RO4      VP                  . R	O4      4      w  rgRRR4       \        XVP                  VP                  ) 4      4       \        XVP                  R4      4       VP                  . RO4      p\        P                  ! W34      p\        VP                  VP                  VP                  4      4       \        VP                  VP                  VP                  4      4       VP                  ^\         P                  .R
^..4      p\        P                  ! WP!                  R4      4      p\        VP                  VP                  ^ VP                  .4      4       \        VP                  VP                  ^VP                  .4      4       R#   + '       g   i     EL; i)iURr*	  rj  r(  r  r^  Nr   r1	  r  rZ   r  )r   r  r  r`   rW  r  rg   r^   r   ru	  r3   r@  r?  r"   r   r   rX  )	rh   ri   r  rk   rl   r   re  rf  r4	  s	   &&       rr   test_special_casesTestTTestInd.test_special_cases  s    ii##H-JJNNq3NGJJNNq3NG#zz!((*%rzz!((*'=1ooa#rzz"&&'9:

BJJrvv$67 /HRPP??2::i#8"**Y:OPDA Q2::rvvg./2::b>*JJy!ooa#rzz"&&'9:

BJJrvv$67 zzArvv;Q01oodHHV$45rzz1bff+'>?

BJJ266{$;< QPPs   <K>>L	c                   \         P                  P                  R 4      p\        P                  P                  ^^
RVR7      p\        P                  P                  ^^
RVR7      p\         P                  VR&   VP                  VP                  4       4      VP                  VP                  4       4      rC\        P                  ! W4RR7      p\        P                  ! W1P                  V4      ( ,          VRR7      p\        VP                  VP                  4       \        VP                  VP                  4       Rp\        P                  ! \         VR7      ;_uu_ 4        \        P                  ! W4RR7       R	R	R	4       R
p\        P                  ! \         VR7      ;_uu_ 4        \        P                  ! W4RR7       R	R	R	4       R	#   + '       g   i     LY; i  + '       g   i     R	# ; i)i#r*	  rj  r(  r   r   r   r   r   Nrn  r   )r   r  r  r`   rW  r  rg   r^   r   ru	  rx  r2   r@  r?  r   r   r   )rh   ri   r  rk   rl   r   r   r  s   &&      rr   test_ttest_ind_nan_policy&TestTTestInd.test_ttest_ind_nan_policy  s:   ii##H-JJNNq3NGJJNNq3NG#zz!((*%rzz!((*'=1ooav6ooa!oqVDs}}5

CJJ/1]]:W55OOAW5 6 .]]:W55OOAX6 65	 65 655s   #G*G G	 G1	c                   \         P                  ! ^^i^d4      p\         P                  ! ^^d^d4      p\         P                  ! W.4      p\         P                  ! W!.4      p\         P                  ! W3V.4      p\         P                  ! WDV.4      p\         P                  VR
&   \         P                  VR&   R p\         P
                  ! V4      p\        P                  ! WV^ RR7      w  r\        P                  ! WV^ RRR7      w  r\        WRR7       \        W! WR4      RR7       \        P                  ! WV^ RRR7      w  r\        WRR7       \        W! WR4      RR7       R	# )r|   c                 l    V ^ 8  d   VR8X  g   V ^ 8  d   VR8X  d
   V^,          # ^V^,          ,
          # r9	  r   r:	  s   &&&rr   convert5TestTTestInd.test_3D_nan_policy_omit.<locals>.convert  s3    A#-QUsi7G1uA;ru   r   r  r  )rD   r   r  rh  r  r  N)r   r   :rT   rw  N)r   r   :r   rV   N)	r   rA  r   r  rg   r>	  r`   ru	  r
   )rh   r
	  r	  r	  r	  r$	  r%	  r	  r<	  r	  rO  re  rf  s   &            rr   test_3D_nan_policy_omit$TestTTestInd.test_3D_nan_policy_omit  s   {{1c3'{{1c3'((D<(((D<())Ww78))Ww78!vv ff
	 LL)	fMwaF+13E*9RV45AwaF+46E*9RY7eDru   c           	        \         P                  ! 4       ;_uu_ 4        \        P                  ! R R7      ;_uu_ 4        \         P                  ! R R\
        4       \        P                  ! RR4      w  rRRR4       RRR4       \        P                  ! X4      '       g   Q h\        P                  ! X4      '       g   Q hR#   + '       g   i     LV; i  + '       g   i     La; i)r   r  rX  r   r   N)	r   r   r   r  r   r   r`   ru	  rx  )rh   re  rf  s   &  rr   test_ttest_ind_scalar"TestTTestInd.test_ttest_ind_scalar  s    $$&&H(E(E##=~O??2r*DA )F& xx{{{xx{{{ )F(E&&s"    C6C	4CCCC%	r   N)r   r   r   r   ry	  r   r|	  r	  r	  r	  r   r   r   s   @rr   rr	  rr	  \  sV     Q)f .RS= T=: .RS7 T7(E> ru   rr	  rM	  z&Dask doesn't support ResamplingMethodsr  z&ignore:Arguments...:DeprecationWarningc                     a  ] tR tRt o ^t]P                  P                  R4      t]P                  ! ]P                  ! ^],          ^,          4      ]P                  ^],          ^,          4      34      t]P                  ! ]P                  ! ]^,          4      ^d,           ]P                  ]^,          4      34      t]P                  ! ^
4      t]P                  ! ^
4      ^d,           t^^.t^^.t]P$                  P'                  ^^
R]R7      P)                  ^d^4      P*                  t]P$                  P'                  ^^^d]R7      tRR.tRR.t. ROt]]R^/]3]P*                  ]P*                  R^ /]3]R,          ]R,          RR/]^ ,          3]R,          P7                  4       ]R,          P7                  4       RR/]^ ,          3]]R^ R^/]3]]R]P                  P9                  ^ 4      R^/]3]]RR	/R3]]RRR^ /]3]]/ R3]]R]P                  P                  ^ 4      R^/]3.
t]P>                  PA                  R
. RO4      ]P>                  PA                  RRR.4      R 4       4       t!]P>                  PA                  R
. RO4      ]P>                  PA                  RRR.4      R 4       4       t"R t#Rt$V t%R# )TestTTestIndResamplingi  i
r(  rj  rD   Nr  	equal_varTr  r  c           	     p   \         P                  P                  R 4      pVP                  VP                  VR7      P	                  4       4      pVP                  VP                  ^R7      P	                  4       4      p\        RR7      p\         P                  P                  R4      p\        P                  ! RRV/VB p\        P                  ! WVRWR7      p	R p
\         P                  P                  R4      p\        P                  ! WV3V
3RRRVR	V/VB p\        V	P                  VP                  4       \        V	P                  VP                  4       \        P                  ! WVRVR
7      p\        V	P                  VP                  4       \        V	P                  VP                  RR7       R# )   Nvf_aB r    rd     N\Sju	 r  rD   r  r  c                 F    \         P                  ! WVR 7      P                  # rC   )r`   ru	  r@  )rk   rl   rD   s   &&&rr   r@  ATestTTestIndResampling.test_permutation_method.<locals>.statisticG  s    %//!T*J*T*T#Tru   rD   r  r  r  r  Nr   rZ   )r   r  r  r^   r   rF  r`   r  ru	  permutation_testr2   r@  r?  )rh   r  r  ri   r  rk   rl   r  r  r   r@  r   s   &&&&        rr   test_permutation_method.TestTTestIndResampling.test_permutation_method8  sZ    ii##$45JJszzuz-4467JJszzrz*1134#& ii##O4((;S;F;ooaT 	Uii##O4$$aVY HR HS H1<H@FH 	s}}5

CJJ/ ooaEs}}5

CJJT:ru   c                  a \         P                  P                  R 4      pSP                  VP                  VR7      P	                  4       4      pSP                  VP                  ^R7      P	                  4       4      p\        RR7      pV3R lp\         P                  P                  R4      pVP                  VP                  .p	\        P                  ! WV33RV	RVRRR	V/VB p
\         P                  P                  R4      pVP                  VP                  .p	\        P                  ! RRV	/VB p\        P                  ! WVRWR
7      p\        VP                  V
P                  4       \        VP                  V
P                  4       \        P                  ! RRR/VB p\        P                  ! WVRWR
7      p\        VP                  V
P                  4       \        VP                  V
P                  4       \        P                  ! WVRVR7      p
\        VP                  V
P                  4       \        VP                  V
P                  RR7       R# )r	  r  r	  r	  c                    < SP                  V 4      SP                  V4      r\        P                  ! WVR 7      P                  # r	  )r^   r`   ru	  r@  )rk   rl   rD   ri   s   &&&rr   r@  ATestTTestIndResampling.test_monte_carlo_method.<locals>.statistic^  s1    ::a="**Q-q??1d3===ru   r	  r  r@  rD   r  r	  r  r  rY  r  NrZ   r   )r   r  r  r^   r   rF  r  r`   monte_carlo_testr  ru	  r2   r@  r?  )rh   r  r  ri   r  rk   rl   r  r@  r  r   r  r   s   &&&f         rr   test_monte_carlo_method.TestTTestIndResampling.test_monte_carlo_methodT  s    ii##$45JJszzuz-4467JJszzrz*1134#&	> ii##O4""C$7$78$$aV H H	 HPR H1<H@FH ii##O4""C$7$78'':C:6:ooaTs}}5

CJJ/ ''FOFvFooaTs}}5

CJJ/ ooaEs}}5

CJJT:ru   c                   R p\         P                  ! \        VR7      ;_uu_ 4        VP                  . RO4      VP                  . RO4      rC\        P
                  ! W4RR7       RRR4       R#   + '       g   i     R# ; i)zC`method` must be an instance of `PermutationMethod`, an instance...r   	migratoryr  Nr  rn  )r   r   r   r^   r`   ru	  )rh   ri   r  rk   rl   s   &&   rr    test_resampling_input_validation7TestTTestIndResampling.test_resampling_input_validationz  sN    W]]:W55::i("**Y*?qOOA5 6555s   >A//B 	r   g'^P?gCq?g$R?)g=B?g?goŜ?gp?g'^?r  rZ   rW  r  )rV   )rJ   rV   )&r   r   r   r   Nr   r  r  r  r  r   rL  rX  r  b2r  b3r`   rW  r  rb   r   r
	  r	  p_dp_d_genp_d_bigr   r  paramsr   r   r   r	  r	  r	  r   r   r   s   @rr   r	  r	    s     	A ))

	
*C
		299QqS!V$cjj1a&89:A
		299QT?S(#**QT*:;<A 
2B	2	B QB
QB ::>>ar"  66=gc1oaa 	::>>ar#>FD<
 C~&G+G 
A{C 	
accFA;$	
4!D'FD>3q62	
4	1T7>>+fd^SVD	
A61-s3	
A		 5 5a 8&!DcJ	R+t$f-	tfb.!4g>	RS	
A		 5 5a 8&!DgN
F [[],LM[[Wug&67; 8 N;4 [[],LM[[Wug&67"; 8 N";H6 6ru   r	  c            
       r  a  ] tR tRt o ]P
                  P                  RRR// .RR.R7      ]P
                  P                  RRR	.RR
.R7      R 4       4       t]P
                  P                  ]P
                  P                  RRR// .RR.R7      ]P
                  P                  RRR	.RR
.R7      ]	! RRR7      ]	! RRR7      ]	! RRR7      R 4       4       4       4       4       4       t
]P
                  P                  RRR// .RR.R7      ]P
                  P                  RR^ .4      R 4       4       tRtV tR# )TestTTestIndCommoni  kwdstrimr@  basic)idsr	  TFunequal_varc                N   \         P                  P                  R 4      pVP                  VP                  R4      P	                  4       4      pVP                  VP                  R4      P	                  4       4      p\
        P                  ! WV3RR/VB p^^^rpWXRV	R^ R3,          pVR^ RV
R3,          p\
        P                  ! W3RR/VB p\        VP                  VRWR3,          VP                  4       \        VP                  VRWR3,          VP                  4       R# )   H-F rD   r   Nr|   r   r   rO   r\   r   r   r\   rK   rJ   r   r[   re  )
r   r  r  r^   r   r`   ru	  r2   r@  r?  )rh   r	  r	  ri   r  rL  rX  r   r  r  r  r  r	  r  s   &&&&          rr   test_ttest_many_dims_fast,TestTTestIndCommon.test_ttest_many_dims_fast  s     ii##J/JJszz"45<<>?JJszz/299;<ooa44t4 Qa!Q1a q!Q1}r7B7$7aA!m4dnnE

1aq=14;;?ru   array_api_strictzindexing in test is too fancyr  r3  rM	  ztoo slowc                   \         P                  P                  R 4      pVP                  R4      P                  4       pVP                  R4      P                  4       p\        P
                  ! VP                  V4      VP                  V4      3RR/VB p\         P                  ! \         P                  ! VR4      RR4      p\         P                  ! \         P                  ! VR	4      RR4      p	VP                  VP                  4       4      VP                  V	P                  4       4      rVP                  RR p
VP                  V
4      pVP                  V
4      p\        R V
 4       !   FH  pW,          pW,          p\        P
                  ! W3RR/VB pVP                  W&   VP                  W&   KJ  	  \        VP                  V4       \        VP                  V4       R# )
r	  rD   Nc              3   8   "   T F  p\        V4      x  K  	  R # 5ir#  )r  )rd	  r  s   & rr   rf	  ?TestTTestIndCommon.test_ttest_many_dims_full.<locals>.<genexpr>  s      95aq5s   r	  r	  r[   )r\   r\   r\   r\   rJ   r\   rZ   )r|   r\   r   r\   r\   r\   )r   r  r  r   r`   ru	  r^   r  tiler  rX  r   r@  r?  r2   )rh   r	  r	  ri   r  rL  rX  r   rk   rl   r  
statisticspvaluesindicesxiyir  s   &&&&             rr   test_ttest_many_dims_full,TestTTestIndCommon.test_ttest_many_dims_full  sn    ii##J/JJ)*113JJ'..0oobjjmRZZ]LLtL KK#56B?KK#56B?zz!((*%rzz!((*'=1XXe_
((5/ 95 9:GBB??2;;d;D"&..J#{{G ; 	z2

G,ru   rD   c           	     <   \         P                  P                  R 4      pVP                  ^
R
R7      P	                  R4      pVP                  ^
R
R7      P	                  R4      p\         P
                  V^ ,          ^,          ^&   \         P
                  V^,          ^ ,          ^&   VP                  VP                  4       4      VP                  VP                  4       4      reVP                  VP                  WV,           VR7      4      p\        P                  ! 4       ;_uu_ 4        \         P                  ! RR7      ;_uu_ 4        \        P                  ! RR\        4       \        P                  ! RR\        4       \        P                   ! WV3RV/VB pR	R	R	4       R	R	R	4       VP                  XP"                  4      p	\%        W4       VP                  VP&                  4      p
\%        W4       R	#   + '       g   i     Lf; i  + '       g   i     Lq; i)l   AoI;r  r  rC   r   r  z'invalid value encountered in less_equalr  rD   N)r|   r   rT   )r   r  r  r  r  rg   r^   r   rx  r]  r   r   r  r   r   r`   ru	  r?  r3   r@  )rh   r	  rD   ri   r  rL  rX  r  r   p_nansstatistic_nanss   &&&&       rr   test_nans_on_axis$TestTTestIndCommon.test_nans_on_axis  sr    ii##$9:LL*L-44W=LL*L-44W=VV!Q
VV!Q
zz!((*%rzz!((*'=1 88BFF15tF45 $$&&H(E(E##C^U##H.GX//!:T:T:C	 )F& #**%)#--01 )F(E&&s%   ( HAG8	H8HHH	r   NrZ   )r   r   r   r   r   r   r   r	  rn  r   r	  r	  r   r   r   s   @rr   r	  r	    sw     [[VvrlB%7"('!2  4[[[4-"-}!=  ?@?4@ [[[[VvrlB%7"('!2  4[[[4-"-}!=  ?(1PQk*IJl:6- 7 K R?4 -2 [[VvrlB%7"('!2  4[[Vb!W-2 .42ru   r	  c            	         a  ] tR tRt o . RO. RORRR.. RO. RORRR.. RO. RORRR.. RO. ROR	RR.. RO. ROR
R R..t]P                  P                  R]4      ]P                  P                  R. R!O4      R 4       4       tR t	R t
]P                  P                  RR"4      R 4       t]P                  P                  R. R#O4      R 4       tRtV tR# )$TestTTestTrimmedi  g9(?r@  gBM)u?g]@gyd?g	s5v@g{Gz?gn4aƂg?g!nu?za, b, pr, tr, trimr<   Nc                   Vf   TM
\        Wv4      pVP                  WR7      VP                  W&R7      r!\        P                  ! WVRR7      w  r\	        WP                  WFR7      4       \	        WP                  W6R7      4       R# )a8  
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
Nr;   Fr	  r	  )rl  r^   r`   ru	  r2   )
rh   rL  rX  rO  r	  r	  r<   ri   r@  r?  s
   &&&&&&&&  rr   test_ttest_compare_r%TestTTestTrimmed.test_ttest_compare_r  sm    B GB,>zz!z)2::a:+E1!OOAtuM		::b:#>?

2
 ;<ru   c                    . R	Op. R
OpVP                  V4      VP                  V4      r2\        P                  ! W#RRR7      w  rE\        WQP                  R4      RR7       \        WAP                  R4      RR7       R# )rV   g
ףp=
?Fr	  gRbv?r	  rF  gOi?N)rV   rR      r  r  ,   rV   rR   r	  r  r  r	  )      rR   rV   r$        r	  r	  rR   rV   r$  r	  r	  r^   r`   ru	  r2   rh   ri   rL  rX  r@  r?  s   &&    rr   test_compare_SAS!TestTTestTrimmed.test_compare_SAS  sb     =Dzz!}bjjm1 "OOAseL	

8 44@	::h#7dCru   c                    . ROp. ROpVP                  V4      VP                  V4      r2\        P                  ! W#RR7      w  rE\        WQP                  R4      4       \        WAP                  R4      4       R# )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
r@  r	  gR?Nr#  r#  r~  r   ffffff?r   r  r  rF  r	  r	  rH   g@rA  r  r6   r  g333333@r  r  g@r  g^r	  r	  s   &&    rr   test_equal_varTestTTestTrimmed.test_equal_var"  s]    & DCzz!}bjjm1!OOAr:	

+> ?@	::.@#ABru   z	alt,pr,trc                    . ROp. ROpVP                  V4      VP                  V4      re\        P                  ! WVRRVR7      w  rx\        WP                  V4      4       \        WtP                  V4      4       R# )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')
r@  F)r	  r	  r  Nr	  r	  r	  )	rh   r;	  rO  r	  ri   rL  rX  r@  r?  s	   &&&&&    rr   test_alternatives"TestTTestTrimmed.test_alternatives=  s`     DCzz!}bjjm1!OOArU8;=	

2/	::b>2ru   r	  c           	         R p\        \        VR7      ;_uu_ 4        \        P                  ! VP	                  ^^.4      VP	                  ^^.4      VR7       RRR4       R#   + '       g   i     R# ; i)z/Trimming percentage should be 0 <= `trim` < .5.r   r	  N)r   r   r`   ru	  r^   )rh   r	  ri   r   s   &&& rr   test_trim_bounds_error'TestTTestTrimmed.test_trim_bounds_errorQ  sM    A:U33OOBJJ1v.

Aq60BN 4333s   <A""A3	r   r  r~  r$  g@ge)      L@33333`@rV   33333^@(\P@N   fffffڇ@)r	  r	  r  r	  r	  g     S@r	  r	  r	  ^)g,
gm?giI@gE|x!gvwQ?glfPg_)gXG~?g=n@g8$mg({g]:m?gYv gb?gb	gSTß}g9yg.Ͽv?g޺i?g	'X@)g>_\?g hO?gau?@g׌g+?g:؜^gz54g7f?gDO%@g#xG@g\F?g4E?gnfg>P"ۿguX@gvR?g%?g2Ydzgo?gih?gn?gJ+br  ))r  gjΜ>5?r	  )r  gn4aƂW?r	  )gɿr6   r\   )r   r   r   r   r	  r   r   r   r	  r	  r	  r	  r	  r   r   r   s   @rr   r	  r	    s     
(;=PRTU	2O	/	5	4o	/	6	@	@	,b	2
H
  "34Db	JKF$ [[16:[[W&BC#= D ;#=JD C6 [[[N3	3  [[V\2O 3Oru   r	  c                   T  a  ] tR tRt o ]P
                  ! RR7      t. RO]R &   RRR]P                  ) R.]R!&   RRRR"]P                  .]R#&   . R%O]R&&   R$RR]P                  ) R	.]R'&   R$RR
R(]P                  .]R)&   R*^R]P                  ]P                  .]R+&   R*^R]P                  ]P                  .]R,&   R*^R]P                  ]P                  .]R-&   R.^R]P                  ]P                  .]R/&   R.^R]P                  ]P                  .]R0&   R.^R]P                  ]P                  .]R1&   ]	P                  P                  R. R2O4      ]	P                  P                  RRR.4      ]	P                  P                  R^ R.4      ]! RRR7      R 4       4       4       4       tRtV tR# )3TestTTestCIiX  )r  e13@gv7?gx_?g
/?-</&@g?JJ?gF?g{?gW"?gW"?g:T'?g0:5?g0:5?g be?r  r	  FTr	  r@  r3  rw  r  c                    V'       d   V'       d   \         P                  ! R 4       V'       d(   \        V4      '       g   \         P                  ! R4       \        P
                  P                  R4      pVP                  VP                  ^4      4      pVP                  VP                  ^4      4      p\        P                  ! WgVW#R7      pR^ R^R^/p	V P                  W,          \        V4      \        \        P                  ! V4      4      3,          p
V
w  rrpRp\        VP                  VP                  V4      VR7       \        VP                  VP                  V4      VR7       \        VP                   VP                  V4      VR7       V'       gb   VP#                  R	4      p\        VP$                  VP                  V4      VR7       \        VP&                  VP                  V4      VR7       R
# R
# )z3Discrepancy in `main`; needs further investigation.z*`trim` is only compatible with NumPy inputl   'Nk' )r  r	  r	  r  r  r  r  r  rE  N)r   xfailr$   r  r   r  r  r^   r`   ru	  r   r  ceilr2   r@  r~  r?  r  r  r  )rh   r  r	  r	  ri   r  rk   rl   r   r  r   r@  r~  r?  r  r  r  r  s   &&&&&             rr   test_confidence_interval$TestTTestCI.test_confidence_interval  se   
 LLNOKKDEii##$78JJszz"~&JJszz"~&ooa(1> $Q9a@ff\.IBGGDM@RRS+.(	vDrzz)'<4H

2T:

BJJv$6TB((-BBFFBJJsO$?BGGRZZ%5DA ru   r   N)r   rJ   rJ   r|   gͿ)r	  r	  gv7?gę"Ͽg8 {M(?r1	  r`	  gD .ɿ)rJ   r~   r~   reϿ)r	  r	  g?JJ?gbտg3Պ?)r~   r~   r\   )r\   r~   r\   gads2ѿ)rJ   r~   r\   gs"%ο)r~   r\   r~   r\   r\   r~   )rJ   r\   r~   g:п)r~   r\   r\   r  )rJ   r\   r\   rm  )r   r   r   r   r   r  r   r   rg   r   r   r   r   r	  r   r   r   s   @rr   r	  r	  X  s    < 	|$ANAgJibffWlKAgJiM266JAgJLAgJibffWlKAgJiL"&&IAgJ%r+=rvvrvvNAgJ%r+=rvvrvvNAgJ%r+=rvvrvvNAgJ%r+=rvvrvvNAgJ%r+=rvvrvvNAgJ%r+=rvvrvvNAgJ[[],LM[[[5$-8[[VaX.k*JKB L / 9 NBru   r	  c                   t  a  ] tR tRt o ]! ]P                  4      R 4       t]P                  P                  R4      ]P                  P                  R4      R 4       4       tR tR t]P                  P                  R]P                   ! R4      R3]P                   ! R4      R3.4      R	 4       tR
 tR t]P                  P)                  RRR7      ]P                  P                  R4      ]P                  P                  R4      ]P                  P+                  RRR7      R 4       4       4       4       tRtV tR# )TestTTestIndMorei  c                l   VP                  . RO4      pVP                  . RO4      pVP                  R4      pVP                  R4      p\        P                  ! W#RR7      w  rg\        We4       \        Wt4       \        P                  ! \        W#4      RR/ w  rg\        We4       \        Wt4       VP                  . RO4      pVP                  R4      pVP                  R4      p\        P                  ! W#RR7      w  rg\        We4       \        Wt4       \        P                  ! \        W#4      RR/ w  rg\        We4       \        Wt4       VP                  R4      pVP                  R4      pVP                  R4      pVP                  R	4      p	VP                  WU) .4      p
VP                  WD.4      pVP                  ^^d^4      pVP                  ^^d^d4      pVP                  ^^i^d4      pVP                  W.4      pVP                  W.4      p\        P                  ! W^ RR
7      w  rg\        We4       \        Wt4       \        P                  ! \        W4      RR/ w  rg\        We4       \        Wt4       \        P                  ! W^ RR
7      w  rg\        Wh4       \        Wy4       \        P                  ! \        W4      RR/ w  rg\        Wh4       \        Wy4       \        P                  ! VP                  VP                  ^ RR
7      p\        VP                  V
4       \        VP                  V4       \        VP                  VP                  4      p\        P                  ! VRR/ p\        VP                  V
4       \        VP                  V4       \        P                  ! VV^RR
7      p\        VP                  V
4       \        VP                  V4       \        VV^R7      p\        P                  ! VRR/ p\        VP                  V
4       \        VP                  V4       R# )r   g9(?Fr	  r	  gJ?rt	  g@[?gωұ?ghG?)rD   r	  rC   Nr  r	  ger  g9ʿ)r^   r`   ru	  r2   rv	  rp	  r  rA  r   r@  r?  )rh   ri   rL  rX  rO  r	  re  rf  	tr_uneq_n	pr_uneq_nrw	  rx	  rvs3r	  r
	  r	  r	  r   r  s   &&                 rr   test_ttest_ind_with_uneq_var-TestTTestIndMore.test_ttest_ind_with_uneq_var  sI    JJ|$JJ'ZZ+,ZZ-.qu5));q+<NNJJ'(ZZ+,ZZ-.qu5));q+<NN ZZ*+JJ23	ZZ+,JJ23	"c#""{{1c2&{{1c3'{{1c3'((D<(((D<(tUC));t+BTeTtUC%%));t+BTeT%%oogiieLu-

E*799gii0(($@%@u-

E*oogwQ%Hu-

E*7G!4(($@%@u-

E*ru   z0ignore:divide by zero encountered:RuntimeWarning/ignore:invalid value encountered:RuntimeWarningc           	        VP                  ^4      pVP                  ^4      p\        \        RVR7      ;_uu_ 4        \        P
                  ! W#RR7      w  rERRR4       \        XVP                  VP                  ) 4      4       \        XVP                  R4      4       \        P                  ! RR7      ;_uu_ 4        \        P
                  ! W"RR7      w  rE\        WAP                  VP                  4      4       \        WQP                  VP                  4      4       VP                  ^VP                  .R
^..4      p\        P
                  ! WaP                  R4      RR7      w  rE\        WAP                  R\        P                  .4      4       \        WQP                  R	\        P                  .4      4       RRR4       R#   + '       g   i     EL|; i  + '       g   i     R# ; i)r   r  r^  Fr	  Nr   r   r  r   rZ   r  )rX  r  r"   r   r`   ru	  r3   r^   r   r   r  rg   )rh   ri   rk   rl   re  rf  r4	  s   &&     rr   test_ttest_ind_zero_division-TestTTestIndMore.test_ttest_ind_zero_division  sM    HHQKGGAJ/HRPP??159DA Q 	2::rvvg./2::b>*[[X&&??159DAAzz"&&12Azz"&&12 ::266{RG45D??4&)9UKDAAzz2rvv,78Azz2rvv,78 '& QPP '&&s   G DGG	G0	c                   \         P                  R RR.p. R	Op\        P                  ! WRR7      p\        P                  ! W!RR7      p\	        VP
                  VP
                  ) RR7       \	        VP                  VP                  RR7       \        P                  ! W!R,          4      p\	        WERR7       \	        VR
RR7       R# )r   r   r   r   r   rE  rF  r  NrD	  )g8HgW@?)r   rg   r`   ru	  r
   r@  r?  )rh   rk   rl   r  r  rE	  s   &     rr   test_ttest_ind_nan_2nd_arg+TestTTestIndMore.test_ttest_ind_nan_2nd_arg  s    VVS#s# __Qf5__Qf5r||m%@		29959 __Q"&U+ 	G"	$ru   c                   \        \        \        VR 7      ;_uu_ 4        \        P                  ! VP                  . 4      VP                  . 4      4      pRRR4       \        X\        P                  P                  4      '       g   Q hVP                  VP                  4      R,          p\        VP                  V4       \        VP                  V4       R#   + '       g   i     L; i)r^  Nr   )r"   r   r   r`   ru	  r^   rC  rI	  rJ	  rg   r3   r@  r?  r  s   &&  rr   #test_ttest_ind_empty_1d_returns_nan4TestTTestIndMore.test_ttest_ind_empty_1d_returns_nan)  s     +3HRPP//"**R."**R.AC Q#u::;;;;jj $s+

C( QPs   6CC)	rN	  c                *   VP                  R4      pVP                  WP                  R7      p\        P                  ! 4       ;_uu_ 4        \        P
                  ! R\        \        4       \        P                  ! WARR7      pRRR4       \        X\        P                  P                  4      '       g   Q hVP                  W#P                  R7      p\        VP                   V4       \        VP"                  V4       R#   + '       g   i     L; irP	  )r  r^   r<   r   r   r   r   r   r`   ru	  rC  rI	  rJ	  r   rg   r3   r@  r?  )rh   rX  rQ	  ri   rL  r   rR	  s   &&&&   rr   test_ttest_ind_axis_size_zero.TestTTestIndMore.test_ttest_ind_axis_size_zero3  s     HHYJJqJ($$&&##H.CEWX//!R0C ' #u::;;;;FFC~6

N3 '&rU	  c                F   VP                  R4      pVP                  R4      p\        P                  ! W#^R7      p\        V\        P                  P
                  4      '       g   Q hVP                  P                  R8X  g   Q hVP                  P                  R8X  g   Q hR# rW	  	r  r`   ru	  rC  rI	  rJ	  r@  r  r?  rh   ri   rL  rX  r   s   &&   rr    test_ttest_ind_nonaxis_size_zero1TestTTestIndMore.test_ttest_ind_nonaxis_size_zeroE  s}    
 HHYHHYooa+#u::;;;;}}""V+++zz6)))ru   c                F   VP                  R4      pVP                  R4      p\        P                  ! W#^R7      p\        V\        P                  P
                  4      '       g   Q hVP                  P                  R8X  g   Q hVP                  P                  R8X  g   Q hR# )r\   rC   N)r\   rO   r~   rX	  rY	  r
  r
  s   &&   rr   2test_ttest_ind_nonaxis_size_zero_different_lengthsCTestTTestIndMore.test_ttest_ind_nonaxis_size_zero_different_lengthsQ  s}     HHYHHYooa+#u::;;;;}}""V+++zz6)))ru   TzTest uses ks_1samp)cpu_onlyr   r  r  Fznot used in this test yetr  c                x   \         P                  P                  R 4      pVP                  VP	                  RR7      4      pVP                  VP	                  RR7      4      pVP                  ^dR7      p\        P                  ! W4RRR7      p\         P                  ! VP                  4      p\        P                  ! V\        P
                  ! 4       P                  4      P                  R8  g   Q h\        \         P                  ! Wu4      VRR7       \        P                  ! WCRRR7      p\         P                  ! VP                  4      p\        P                  ! V\        P
                  ! 4       P                  4      P                  R8  g   Q h\        \         P                  ! Wu4      VRR7       VP                  . RO4      VP                  R.4      rC\        P                  ! W4RR7      p\        VP                  VP                  R	4      4       \        VP                  VP                  R
4      4       R# )l   bQ r  T)r	  rD   rC  r  rF  r7   r	  gzNd?gJ?N)r  rJ   )r  r\   rZ   )r   r   r=   )r   r  r  r^   r  rM  r`   ru	  r?  rK  re  r
   quantiler2   r@  )rh   ri   r  rk   rl   r  r   r?  s   &&      rr   test_ttest_uniform_pvalues+TestTTestIndMore.test_ttest_uniform_pvalues^  s}    ii##O4JJszzzz23JJszzzz23KKSK!ooad<CJJ'~~femmo&9&9:AACGGGF.=ooad<CJJ'~~femmo&9&9:AACGGGF.=
 zz,'SE):1ooad3rzz/'BC

BJJ~$>?ru   r   Nr_	  r  r`	  ra	  )r   r   r   r   r)   r`   rv	  r
  r   r   r   r
  r	
  r
  r   r   r  r
  r
  r
  r   uses_xp_capabilitiesr
  r   r   r   s   @rr   r	  r	    sD    u112N+ 3N+` [[: [[999,$*) [[0 hhy16:XXi0&9;<4<4
** [[!!48L!M[[ VW[[ WX[[%%e4O%P@ Q Y X N@ru   r	  c                   h   a  ] tR tRt o R t]P                  P                  R4      R 4       tRt	V t
R# )TestTTestIndFromStatsi~  c                   VP                  ^^.4      VP                  ^^.4      r2VP                  ^^.4      VP                  ^^.4      rTVP                  ^^.4      VP                  ^d^.4      rv\        P                  ! W$WcWW4       R# r\   N)r^   r`   rv	  )rh   ri   mean1mean2std1std2nobs1nobs2s   &&      rr   test_gh5686!TestTTestIndFromStats.test_gh5686  sw    zz1a&)2::q!f+=uZZA'QF);dzz3*-rzz3*/Eu""5dJru   r
  c           
        VP                  R 4      pVP                  R4      pVP                  VP                  4      p\        P                  ! W"W2W#RR7      p\	        VP
                  V4       \	        VP                  V4       R# )r   r   Fr	  N)r^   rg   r`   rv	  r3   r@  r?  )rh   ri   r3  sixrr  r   s   &&    rr   %test_ttest_ind_from_stats_inputs_zero;TestTTestIndFromStats.test_ttest_ind_from_stats_inputs_zero  se     zz"~jjnjj ((S38:s+

C(ru   r   N)r   r   r   r   r)
  r   r   r   r-
  r   r   r   s   @rr   r 
  r 
  ~  s0     K [[ QR) S)ru   r 
  c                   <  a  ] tR tRt o R t]P                  P                  ]P                  P                  R4      ]P                  P                  R4      R 4       4       4       t	]
! RRR7      R	 4       t]P                  P                  R
RR7      R 4       tRtV tR# )TestTTest1Sampi  c                    VP                  VR 8H  4      pVP                  VR8H  4      pV^ 8  V,          V^ 8  V,          ,          pVP                  Wr^,          ^V^,          ,
          4      # )r  r  )r^   r  )rh   re  rf  r;	  ri   r  r  r  s   &&&&&   rr   _convert_pvalue_alternative*TestTTest1Samp._convert_pvalue_alternative  sa     zz#-(**SI-.!et^Q' 12xxQ3AaC((ru   r  r  c           	     	   \         P                  P                  R 4      p^
^^rTp\        P                  P                  ^^
W4V3VR7      pVP                  V4      pVP                  ^WE34      p\        P                  ! Wg^ R7      w  r\        P                  ! VR^ R7      w  r\        P                  ! VR,          R4      w  r\        WRR7       \        VR,          VRR7       \        VP                  WE34       VP                  V^V34      p\        P                  ! Wg^R7      w  r\        P                  ! VR^R7      w  r\        P                  ! VR,          R4      w  r\        WRR7       \        VR,          VRR7       \        VP                  W534       VP                  W4^34      p\        P                  ! Wg^R7      w  r\        P                  ! VR^R7      w  r\        P                  ! VR,          R4      w  r\        WRR7       \        VR,          VRR7       \        VP                  W434       \        P                  ! VP                  . RO4      R4      w  r\        VP                  V4      VP                  VP                  4      4       \        WP                  R4      4       \        P                  ! VR,          R4      w  pp\        P                  ! VR,          RRR7      w  rV P                  VVRV4      p\        VV4       \        VV4       \        P                  ! VR,          RR	R7      w  rV P                  VVR	V4      p\        VV4       \        VV4       \         P                   ! R
R7      ;_uu_ 4        \        P                  ! VP                  . RO4      R4      p\        VP"                  VP                  VP$                  4      4       \        VP&                  VP                  VP$                  4      4       VP                  R\         P$                  .RR..4      p\        P                  ! VR4      p\        VP"                  VP                  RVP$                  .4      4       \        VP&                  VP                  RVP$                  .4      4       RRR4       R#   + '       g   i     R# ; i)i@rj  rC   r   rh  r  r   r  r  r  r   r
  N)r   r~   r~   r  )r~   r   r~   )r~   r~   r   )r   r   r   r   r   r   rQ  )r   r  r  r`   rW  r  r^   r  rZ  r2   r   r  r3   r3	  r   r2
  r  r@  rg   r?  )rh   ri   r  r  r  n3rvn1rz  r  r  r  p2r  p3re  rf  r	  rO  pcr   r4	  s   &&                   rr   test_ttest_1samp_new#TestTTest1Samp.test_ttest_1samp_new  s    ii##H-"bzz~~!2RRLs~Szz$ ''1b+&""4q9""4!4""4="5U+4"51RXXx(''2q"+&""4q9""4!4""4="5U+4"51RXXx(''21+&""4q9""4!4""4="5U+4"51RXXx(   L!92>q	2::bff#56::b>*""4="5B  g	J--b"iD22  gG--b"fbA22[[X&&##BJJ|$<bACCMM2::bff+=>CJJ

266(:; ::BFF|c2Y78D##D"-CCMM2::r266l+CDCJJ

B<(@A '&&&s   D%SS	Tr   r   c                L   \         P                  P                  R 4      p^^
^rTp\        P                  P                  ^^
W4V3VR7      pVP                  V4      pVP                  VR	&   \        P                  ! VR
,          RRR7      w  rx\        P                  ! VR
,          RRRR7      w  rV P                  WxRV4      p\        W4       \        W4       \        P                  ! VR
,          RRRR7      w  rV P                  WxRV4      p\        W4       \        W4       R# )l   U] rj  r   r   r   r  rl  r  N)r  :r\   r   N:r   rK   Nr5
  )r   r  r  r`   rW  r  r^   rg   rZ  r2
  r2   )rh   ri   r  r  r  r6
  r7
  r	  rO  re  rf  r:
  s   &&          rr   test_ttest_1samp_new_omit(TestTTest1Samp.test_ttest_1samp_new_omit  s    ii##J/Rzz~~!2RRLs~Szz$ ff]""4="H  gv-68--biD  gv-35--bfbAru   r3  rw  r  c                   \         P                  P                  R 4      pVP                  R	R7      pVP                  V4      pRpVP                  VP                  R
R7      4      p\        P
                  ! \        VR7      ;_uu_ 4        \        P                  ! W5RR7       RRR4       VP                  VP                  RR7      4      p\        P                  ! W5RR7      pVP                  P                  R8X  g   Q hVP                  4       w  rxVP                  VRR7      pVP                  VRR7      p\        P                  ! W7RR7      pVP                  VP                  RVP                  R7      VP                  P                  4      p	\!        VP                  V	4       \        P                  ! W8RR7      p\!        VP                  V	4       R#   + '       g   i     EL>; i)l   iI}pC r  z%`popmean.shape\[axis\]` must equal 1.r   )rz  rD   NrC   g?r;   )r\   rw  r  )r|   rJ   r  re  )r|   r\   r  )r|   r  )r   r  r  r^   r   r   r   r`   rZ  r@  r  r  r  r  r   r?  r2   )
rh   ri   r  rk   r  rz  r   r  r  r   s
   &&        rr   test_ttest_1samp_popmean_array-TestTTest1Samp.test_ttest_1samp_popmean_array  sj   
 ii##$78JJKJ(JJqM:**SZZZZ89]]:W55ar: 6 **SZZZZ89<}}""g---&&(NN12N&NN12N&26oobjjRZZj@#**BRBRS

C(26

C(! 655s   
GG)	r   N)r   r   r   r   r2
  r   r   r  r   r;
  r   r>
  rA
  r   r   r   s   @rr   r0
  r0
    s     ) [[[[ WX[[ VW:B X Y :Bx .WX Y, [[!!+6V!W) X)ru   r0
  c                      a  ] tR tRt o ]P
                  P                  R4      ]P
                  P                  R4      R 4       4       tR tR t	R t
R tR	 tR
 tR t]! RRR7      R 4       tRtV tR# )TestDescribei  r  r  c           
     6   \         P                  ! 4       ;_uu_ 4        \        P                  ! R R R7      ;_uu_ 4        \         P                  ! R R\
        4       \        P                  ! VP                  R4      R,          4      w  r#rErgRRR4       RRR4       X^8X  g   Q h\        X^ ,          VP                  R4      4       \        V^,          VP                  R4      4       \        XVP                  R4      4       \        XVP                  VP                  4      4       \        XVP                  VP                  4      4       \        XVP                  VP                  4      4       R#   + '       g   i     L; i  + '       g   i     L; i)r   rV  rX  r   Nr   )r   r   r   r  r   r   r`   describer^   r3   rg   )rh   ri   r  r  r  r   skkurts   &&      rr   test_describe_scalar!TestDescribe.test_describe_scalar  s	    $$&&kk(8<<##=~O$)NN2::b>"3E$F!A1 = '
 Avv1rzz#/1rzz#/2::c?+2::bff-.BJJrvv./bjj01 =< '&s#   !FAE5	F5F FF	c                   VP                  VP                  R4      VP                  RR4      34      p^pVP                  . RO4      VP                  . R	O4      3pVP                  . R
O4      pVP                  . RO4      pVP                  R.^,          4      pVP                  R.^,          4      p\        P
                  ! V4      w  rrrW8X  g   Q h\        V
^ ,          V^ ,          4       \        V
^,          V^,          4       \        W^VP                  VP                  4      P                  ,          R7       \        W^VP                  VP                  4      P                  ,          R7       \        W4       \        W4       \        P
                  ! VP                  ^R7      w  rrrW8X  g   Q h\        V
^ ,          V^ ,          4       \        V
^,          V^,          4       \        W^VP                  VP                  4      P                  ,          R7       \        W^VP                  VP                  4      P                  ,          R7       \        W4       \        W4       R# )r   r   H,p ?r  rC   Nr  rJ   r   r   r   r   r   r   r   r   r   ffffff?rQ
  rQ
  rQ
  )rF  rF  rF  rF  TUUUUU)r  r  r   r^   r`   rF
  r3   r2   r   r<   epsr   rh   ri   rk   ncmmcmcvcskckurtcr  r  r  r   rG
  rH
  s   &&             rr   test_describe_numbers"TestDescribe.test_describe_numbers#  s   IIrwwv(;<=zz*+RZZ8H-IJZZ,-ZZ,-jj-.23

./!34 %q 1qRww1s1v&1s1v&A(9(=(=$=>A(9(=(=$=> $ %qss ;qRww1s1v&1s1v&A(9(=(=$=>A(9(=(=$=> $ru   c                .   \         P                  ! R 4      p\         P                  V^	&   ^	Rr2RpRpRpR	p\        P                  ! VRR7      w  rrr\        W4       \        W4       \        W4       \        W4       \        W4       \        W^R7       R# )
r   r   r   r  r   r   r  N)r   r   gGz)r   r   rg   r`   rF
  r   r   )rh   rk   rU
  rV
  rW
  rX
  rY
  rZ
  r  r  r  r   rG
  rH
  s   &             rr   describe_nan_policy_omit_test*TestDescribe.describe_nan_policy_omit_test=  s    IIcNvv!jC# %qV DqRQRQQ!"*!$r:ru   c                   VP                  R 4      pVP                  V^	8H  VP                  V4      p\        V4      '       dG   \        P
                  ! \        RR7      ;_uu_ 4        \        P                  ! VRR7       RRR4       ME\        P
                  ! \        RR7      ;_uu_ 4        \        P                  ! VRR7       RRR4       \        P                  ! VRR7      w  r4rVrxVP                  VP                  4      R,          p	V^
8X  g   Q h\        V^ ,          V	4       \        V^,          V	4       \        WY4       \        Wi4       \        Wy4       \        W4       \        V4      '       d   V P                  4        MV\        V4      '       dF   \        P
                  ! \        RR7      ;_uu_ 4        \        P                  ! VRR7       RRR4       R	p
\        P
                  ! \        V
R7      ;_uu_ 4        \        P                  ! VR
R7       RRR4       R#   + '       g   i     ELk; i  + '       g   i     EL; i  + '       g   i     L; i  + '       g   i     R# ; i)r   not supported for lazy arraysr   r   r   Nr   r  r   r   r   r   )r   r  rg   r#   r   r   rg  r`   rF
  r   r^   r3   r$   r_
  )rh   ri   rk   r  r  r  r   rG
  rH
  r   r  s   &&         rr   test_describe_nan_policy_other+TestDescribe.test_describe_nan_policy_otherN  s   IIcNHHQT2661%y0OPPqW5 QP z1PQQqW5 R !&q[ IqRjj $Bww1s#1s# "B<<..01y0OPPqV4 Q 1]]:W55NN12 65/ QPP RQQ  QP 655s0   'H(-H<>II#(H9	<I	I 	#I4	c                t    \         P                  ! \        P                  ! R 4      4      pRp\	        W4       R# )r=   N)rm	  minmaxrc   variancer  r#  )r`   rF
  r   r   r   )rh   r  rA  s   &  rr   test_describe_result_attributes,TestDescribe.test_describe_result_attributesm  s'     		".S
F/ru   c                r   VP                  VP                  R4      VP                  RR4      34      p^pVP                  . RO4      VP                  . RO4      3pVP                  . R	O4      pVP                  . R
O4      pVP                  R.^,          4      pVP                  R.^,          4      p\        P
                  ! V^ R7      w  rrrW8X  g   Q h\        V
^ ,          V^ ,          4       \        V
^,          V^,          4       \        W4       \        W4       \        W4       \        W4       R# )r   r   rL
  r  Nr  rM
  rN
  rO
  rP
  )Q?rk
  rk
  rk
  rR
  r  r  r   r^   r`   rF
  r3   r2   rT
  s   &&             rr   test_describe_ddofTestDescribe.test_describe_ddoft  s    IIrwwv(;<=zz*+RZZ8H-IJZZ,-ZZ01jj-.23

./!34 %qq 9qRww1s1v&1s1v& $ru   c                B   VP                  VP                  R4      VP                  R	R4      34      p^pVP                  R4      VP                  R4      3pVP                  R4      pVP                  R4      pVP                  R4      pVP                  R
4      p\        P
                  ! VRR7      w  rrrW8X  g   Q h\        V
^ ,          V^ ,          4       \        V
^,          V^,          4       \        W4       \        W4       \        W4       \        W4       R# )r   r   r   rQ
  ga+?gE,p ?NrC   r  rM
  gUUUUUUrl
  rT
  s   &&             rr   test_describe_axis_none$TestDescribe.test_describe_axis_none  s    IIrwwv(;<= zz#

30ZZ*+ZZ+,jj+,

./ !&qt <qRww1s1v&1s1v& $ru   c                    R p\         P                  ! \        VR7      ;_uu_ 4        \        P                  ! VP                  . 4      4       RRR4       R#   + '       g   i     R# ; i)zThe input must not be empty.r   N)r   r   r   r`   rF
  r^   r5  s   && rr   test_describe_empty TestDescribe.test_describe_empty  s;    0]]:W55NN2::b>* 6555s   &AA(	r	  z"min/max don't accept boolean inputr  c                &   \         P                  P                  R 4      p^pVP                  VP                  V4      R8  4      p\        P
                  ! V4      p\        VP                  VP                  W1P                  R7      4       \        VP                  ^ ,          VP                  V4      4       \        VP                  ^,          VP                  V4      4       \        VRVR7      p\        VP                  VP                  V4      4       \        VP                  VP!                  V^R7      4       \        VP"                  \        P$                  ! V4      4       \        VP&                  \        P&                  ! V4      4       R# )l   Dr+\ r6   r;   Tr  rw   N)r   r  r  r^   r`   rF
  r3   rm	  int64rf
  r  r  r0   r2   rc   rg
  r   r  r  r#  )rh   ri   r  r  rk   r   s   &&    rr   test_boolean_input_gh24847'TestDescribe.test_boolean_input_gh24847  s    ii##$45JJszz!}s*+nnQ"**Qhh*"?@

1rvvay1

1rvvay1q"5"''!*-bffQ1f&=>ejjm4ennQ&78ru   r   N)r   r   r   r   r   r   r   rI
  r[
  r_
  rc
  rh
  rm
  rp
  rs
  r   rw
  r   r   r   s   @rr   rD
  rD
    s     [[ VW[[ WX2 Y X2%4;"3>0%"%,+
 (1UV9 W9ru   rD
  c                   n   a  ] tR tRt o ]P
                  P                  R. RO4      R 4       tR tRt	V t
R# )NormalityTestsi  r  c                   V P                   p\        \        V4      pVP                  V P                  ^ ,          4      pVP                  V P                  ^,          4      p/ pVR9   dU   VR9   d8   VR8X  d
   V^,          M^V^,          ,
          pVR8X  d
   ^V,
          MTpWR&   M\
        P                  ! R4       VP                  R	4      ^,          pV! V3/ VB p	V	w  r\        W4       \        W4       \        V	R
VR7       R# )r~   r  skewtestr  z,`alternative` not available for `normaltest`r=  N>   r  r  >   r|
  kurtosistest)re  rZ   r~   r\   rJ   r   re  rZ   r~   r\   rJ   r   re  rZ   r~   r\   rJ   r   re  rZ   r~   r\   rJ   r   rY  )		test_namerl  r`   r^   case_refr   r  r2   r   )rh   r  ri   r~
  rh	  rr  r_  r  rk   r   r  r  s   &&&         rr   test_against_RNormalityTests.test_against_R  s     NN	5),zz$--"23ZZa 01
--88-8F-BZ\*UV,
-6*-DQz\*
(3}%JKJJ./2q#F#$'!5
/C!8R@ru   c                   \        \        V P                  4      pVP                  R 4      pVP	                  VP
                  VP                  R7      pVP                  V^8H  WC4      p\        P                  ! RR7      ;_uu_ 4        V! V4      p\        VP                  V4       \        VP                  V4       RRR4       R#   + '       g   i     R# ; i)r  r;   r   r  N)rl  r`   r~
  r   r^   rg   r<   r  r   r  r3   r@  r?  )rh   ri   rh	  rk   rr  r   s   &&    rr   test_nanNormalityTests.test_nan  s    5$..1IIcNjjqwwj/HHQ"Wc%[[**1+CCMM3/CJJ, +***s   5CC	r   Nrm  )r   r   r   r   r   r   r   r
  r
  r   r   r   s   @rr   rz
  rz
    s6     [[],LMA NA<	- 	-ru   rz
  c                   4   a  ] tR tRt o RtRtR tR tRtV t	R# )TestSkewTesti  r|
  c                    \         P                  P                  ^R^{R7      pVP                  V4      p\         P                  ! VRR7      P
                  p\        WAP                  RVP                  R7      RR7       R	# )
r\   r  )rL  r  r  r  r  r   r;   g?̔>rF  N)r`   skewnormr  r^   r|
  r?  r2   r<   )rh   ri   r  a1_xpr  s   &&   rr   test_intuitiveTestSkewTest.test_intuitive  s[    ^^!%cB

2~~e;BBjjEKKj@tLru   c                   \         P                  ! VP                  R 4      4       VP                  R4      p\        \        \
        VR7      ;_uu_ 4        \         P                  ! V4      pVP                  VP                  4      p\        VP                  V4       \        VP                  V4       RRR4       R#   + '       g   i     R# ; i)r   r   r^  N)r`   r|
  r   r"   r   r   r^   rg   r3   r@  r?  rh   ri   rk   r   rr  s   &&   rr   "test_skewtest_too_few_observations/TestSkewTest.test_skewtest_too_few_observations  s     	ryy~&IIcN+3HRPP..#C**RVV$CCMM3/CJJ,	 QPPPs   AB==C	r   N)gvO?gbH`?)
r   r   r   r   r~
  r
  r
  r
  r   r   r   s   @rr   r
  r
    s     I9HM- -ru   r
  c                   :   a  ] tR tRt o RtRtR tR tR tRt	V t
R# )	TestKurtosisTesti  r}
  c                    \         P                  P                  R ^{R7      pVP                  V4      p\         P                  ! VRR7      P
                  p\        WAP                  RVP                  R7      RR7       R# )	r  r		  r  r  r   r;   rE  rF  N)r`   laplacer  r^   r}
  r?  r2   r<   )rh   ri   r  a2_xpr  s   &&   rr   r
  TestKurtosisTest.test_intuitive  s[    ]]E<

2!!%Y?FFjjEKKj@uMru   c           
     (   . ROp\         P                  ! \        V4       UUu. uF  w  r4\         P                  ! WC4      NK  	  upp4      pVP	                  WQP
                  R7      p\        P                  ! V4      ^,          R8  g   Q hR# u uppi )rO  r;   r  N)
rO  r~   :   rO   r~   )      r~   r~      )r   hstack	enumerater   r^   r   r`   r}
  )rh   ri   r  r  ri  rk   s   &&    rr   test_gh9033_regression'TestKurtosisTest.test_gh9033_regression  so     7II61BC1Brwwq}1BCDJJq

J+!!!$Q'$... Ds   !B
c                   \         P                  ! VP                  R 4      4       \        \        \
        VR7      ;_uu_ 4        \         P                  ! VP                  R4      4      pVP                  VP                  4      p\        VP                  V4       \        VP                  V4       RRR4       R#   + '       g   i     R# ; i)r=   r^  r   N)r`   r}
  r   r"   r   r   r^   rg   r3   r@  r?  r  s   &&  rr   &test_kurtosistest_too_few_observations7TestKurtosisTest.test_kurtosistest_too_few_observations  s     	299S>*+3HRPP$$RYYr]3C**RVV$CCMM3/CJJ,	 QPPP   A-B;;C	r   N)gglg2J@?)r   r   r   r   r~
  r
  r
  r
  r
  r   r   r   s   @rr   r
  r
    s$     I:HN/	- 	-ru   r
  c                   .   a  ] tR tRt o RtRtR tRtV tR# )TestNormalTesti  
normaltestc                   \         P                  ! VP                  R 4      4       \        \        \
        VR7      ;_uu_ 4        \         P                  ! VP                  R4      4      pVP                  VP                  4      p\        VP                  V4       \        VP                  V4       RRR4       R#   + '       g   i     R# ; i)r   r^  r   N)r`   r
  r   r"   r   r   r^   rg   r3   r@  r?  r  s   &&  rr   test_too_few_observations(TestNormalTest.test_too_few_observations  s    2' +3HRPP""299R=1C**RVV$CCMM3/CJJ,	 QPPPr
  r   N)gاc@g?)	r   r   r   r   r~
  r
  r
  r   r   r   s   @rr   r
  r
    s     I9H- -ru   r
  c                   t   a  ] tR tRt o ]P
                  P                  R. RO4      R 4       tR tR t	Rt
V tR# )	TestRankSumsi(  r  c                N   \         P                  P                  R 4      pVP                  R4      w  r4\        P                  ! W4VR7      p\        P
                  ! W4RVRR7      p\        VP                  VP                  4       \        VP                  VP                  4       R# )   hZ r  Fr  )use_continuityr  r  NrJ   rT   )
r   r  r  r`   ranksumsmannwhitneyur
   r?  r@  
zstatistic)rh   r  r  rk   rl   r  r  s   &&     rr   test_ranksums_is_mannwhitneyu*TestRankSums.test_ranksums_is_mannwhitneyu*  sv    ii##J/zz'" ~~a<!!!u.9,PT[[18ru   c                    \         P                  ! \        P                  ! ^
4      \        P                  ! ^
4      4      p\	        VR4       R# )rT   NrY  )r`   r
  r   r   r   )rh   r   s   & rr   test_ranksums_named_results(TestRankSums.test_ranksums_named_results6  s-    nnRYYr]BIIbM:C!89ru   c                    \        \        R R7      ;_uu_ 4        \        P                  ! \        P
                  ! ^
4      \        P
                  ! ^
4      RR7       RRR4       R#   + '       g   i     R# ; i)z`alternative` must be 'less'r   r   r  N)r   r   r`   r
  r   r   rs  s   &rr   r  "TestRankSums.test_input_validation:  s@    :-KLLNN299R="))B-XN MLLLs   AA&&A7	r   Nr  )r   r   r   r   r   r   r   r
  r
  r  r   r   r   s   @rr   r
  r
  (  s;     [[],LM	9 N	9:O Oru   r
  c                   P   a  ] tR tRt o R t]! RR7      R 4       tR tR tRt	V t
R	# )
TestJarqueBerai?  c                    . ROpVP                  V4      pVP                  RR.4      p\        P                  ! V4      p\        VP                  V^ ,          4       \        VP
                  V^,          4       R# )g8MK~?gq"?gPHK?N)g8MK~Ŀg/?g6r=TgGg!J_l )r^   r`   jarque_berar2   r@  r?  )rh   ri   rk   r   r   s   &&   rr   test_jarque_bera_against_R)TestJarqueBera.test_jarque_bera_against_RA  s^    
;JJqMjj*O<="s1v.

CF+ru   Tr  c                l   \         P                  P                  R 4      pVP                  RR7      p\        P
                  ! \        V4      4      ;pw  rV\        P
                  ! \        V4      4      ;pw  r\        P
                  ! VP                  ^R4      4      ;p
w  rYXu;8X  d?   Tu;8X  d7   VP                  u;8X  d%   VP                  u;8X  d   V
P                  8X  g   Q h Q hYiu;8X  d?   Tu;8X  d7   VP                  u;8X  d%   VP                  u;8X  d   V
P                  8X  g   Q h Q hR# )l   uT ri  r  P  N)r   r  r  r  r`   r
  rB  tuplerb   r@  r?  )rh   ri   r  rk   jb_test1JB1r  jb_test2JB2r8
  jb_test3JB3r9
  s   &&           rr   test_jarque_bera_array_like*TestJarqueBera.test_jarque_bera_array_likeN  s     ii##J/V,"..tAw7773"..uQx8873"..qyyE/BCC73bSbH$6$6b(:L:LbPXPbPbbbbbbV2VVHOOVxVVVVVru   c                l   VP                  . 4      p\        P                  ! \        \        R 7      ;_uu_ 4        \
        P                  ! V4      pVP                  VP                  4      p\        VP                  V4       \        VP                  V4       RRR4       R#   + '       g   i     R# ; i)r   N)r^   r   r  r   r   r`   r
  rg   r3   r@  r?  r
  s   &&   rr   %test_jarque_bera_too_few_observations4TestJarqueBera.test_jarque_bera_too_few_observations[  sp    JJrN\\,4IJJ##A&C**RVV$CCMM3/CJJ,	 KJJJs   AB""B3	c                   \         P                  P                  R R7      pVP                  VP                  RR7      4      p\        P
                  ! VRR7      p\        P
                  ! VP                  VR4      4      p\        VP                  VP                  4       \        VP                  VP                  4       \        P
                  ! V^R7      p\        P
                  ! VR,          4      w  rg\        P
                  ! VR,          4      w  r\        VP                  VP                  Wh.4      4       \        VP                  VP                  Wy.4      4       \        P
                  ! VP                  ^ R7      p
\        VP                  V
P                  4       \        VP                  V
P                  4       R# )	iKr	  r  NrC   )rJ   r  r  r  r  )r   r  r  r^   r`   r
  rb   r3   r@  r?  r2   r  r   )rh   ri   r  rk   r   r   s0r  s1r  resTs   &&         rr   r	  TestJarqueBera.test_axisd  s-   ii###3JJszzwz/0-

1e 45s}}5

CJJ/*""1T7+""1T7+rxx'9:

BHHbX$67  1-t~~6

DKK0ru   r   N)r   r   r   r   r
  r   r
  r
  r	  r   r   r   s   @rr   r
  r
  ?  s4     , d#
W $
W-1 1ru   r
  c                   &   a  ] tR tRt o R tRtV tR# )TestPointBiserialRix  c                .   . ^N^ N^N^N^N^N^ N^N^ N^ N^ N^N^N^ N^ N^ N^N^N^N^ N^ N^ N^ N^ N^ N^ N^ N^N^ N^ N^ N^ N^ NRNp. RNRNRNRNRNRNRNR	NR
NRNRNRNRNRNRNRNRNRNRNRNRNRNRNRNRNRNRNRNRNRNRNRNRNRNp\         P                  ! VP                  V4      VP                  V4      4      p\        V^ ,          VP                  R4      R R!7       R$p\	        WEVR"7       \        VP                  VP                  4       R## )%r\   r   g-@g+@g(@ry  r
  r	  g333333@ffffff@g333333@r  rB  皙	@r   r  ry   r  ffffff@ @333333?r7   ?r  r~  r  rD  r{  r6   r@  rC  g/"?ro  rF  r=  Nr>  )r`   pointbiserialrr^   r2   r   r3   r<  r@  )rh   ri   rk   rl   r   rA  s   &&    rr   test_pointbiserial%TestPointBiserialR.test_pointbiserialz  sf   Q q  1 Q q  1 Q q  1 Q q  1 Q q  1 Q q  1 Q q  1 Q *T *$ *t *D * *S * *S * *S * *S * ***** *!$*%(*),*-0*14*58*9<*=@*AD***** *!$*%(* ""2::a="**Q-@A

7 3$? /
C37ru   r   N)r   r   r   r   r
  r   r   r   s   @rr   r
  r
  x  s     8 8ru   r
  c                   8   a  ] tR tRt o R tR tR tR tRtV t	R# )TestObrientransformi  c                   VP                  . RO4      p\        P                  ! V4      pVP                  . RO4      p\        V^ ,          R,          V4       VP                  . RO4      p\        P                  ! V4      pVP                  . RO4      p\        V^ ,          R,          V4       \        P                  ! W%4      w  rx\        Ws^ ,          R,          4       \        W^ ,          R,          4       \        P                  ! W%V4      w  rxp	\        Ws^ ,          R,          4       \        W^ ,          R,          4       \        W^ ,          R,          4       R# )r~   r   N)r~   rJ   r   )r   r  r   )r~   r   r   rN   )r  r   r   r  )r^   r`   obrientransformr2   )
rh   ri   r  r  r  r  r  rL  rX  ri  s
   &&        rr   rY  TestObrientransform.test_basic  s    ZZ	"""2&::m,1a(+ZZ%""2&::011a(+ $$R,a58$a58$ ''3aa58$a58$a58$ru   c                    VP                  ^4      pVP                  . RO. RO.4      p\        P                  ! V^V,          4      p\	        V^ ,          VR,          4       \	        V^,          VR,          4       R# )r|   N)㪪@I?g&>ڿr
  r
  )5@㪪@gIr
  r
  r  r  )r   r^   r`   r
  r2   )rh   ri   r  r   r   s   &&   rr   test_something"TestObrientransform.test_something  sh     YYq\jjHJLM ##B"-AF,AF,ru   c                   \         P                  ! ^^4      p. ROp\         P                  ! W#4      p\         P                  ! . RO4      p\         P                  ! WS4      p\        P
                  ! VP                  V4      4      p\        V^ ,          R,          VP                  VP                  4       4      RR7       R# )r|   r   r  r  N)r|   rU   rN   r   rJ   rJ   )g_v	@g1%?g"u?gZӼ?g!u @gq&@)r   r   r  r^   r`   r
  r2   r   )rh   ri   valuesrepsrG  transformed_valuesr  r  s   &&      rr   test_reference"TestObrientransform.test_reference  s    1b!"yy&ZZ )B C99/6&&rzz$'78q	!bjj1B&C$Oru   c                   \         P                  P                  R 4      pVP                  ^
4      p\         P                  V^&   VP	                  V4      p\        V4      '       gG   \        P                  ! \        RR7      ;_uu_ 4        \        P                  ! VRR7       RRR4       ME\        P                  ! \        RR7      ;_uu_ 4        \        P                  ! VRR7       RRR4       \        P                  ! VRR7      ^ ,          p\        WAP                  W1P                  4      4       VP                  V4      p\        P                  ! VRR7      ^ ,          p\        P                  ! W5( ,          4      ^ ,          p\        WE,          W5,          4       \        WE( ,          V4       R#   + '       g   i     L; i  + '       g   i     L; i)	l   37j"Fzr   r   r   r   N#nan_policy='raise' is not supportedr  r   )r   r  r  rg   r^   r#   r   r   r   r`   r
  rg  r3   r  rx  r2   )rh   ri   r  rk   r   r  r   s   &&     rr   r  #TestObrientransform.test_nan_policy  s=   ii##$:;JJrNvv!JJqM Qz1PQQ%%aG< RQ y0UVV%%aG< W ##A+>qA\\!VV45 HHQK##A&9!<##AbE*1-%B% RQ WVs   
F5G5G	G	r   N)
r   r   r   r   rY  r
  r
  r  r   r   r   s   @rr   r
  r
    s     %.	-	P& &ru   r
  r<   r  r  c                    T;'       g    V P                   p\        V4      ^8X  d   Vw  rxMVw  rypV P                  WsR7      pV \        P                  8w  d   \        VRV R7      MRp
V P                  WR7      pVe   V P                  WSR7      MTp\        V4      ^8X  d   V3MVX	3pV! VRVRV/ p\        WVR7       R# )rJ   r;   Tr  NrD   r  r  )r   r  r^   r   r	  r,   r2   )ri   mean_funrD   r<   r  r  r  
array_liker  rf  dtype_referencerk   s   $$$$$$*     rr   check_equal_xmeanr
    s     RZZE
4yA~"
G!%
wJ4J+ &j"M+/ jjj8G292Ebjjj.7GINJ=QD$3T373AAT*ru   c                  8    \        V R \        P                  /VB # r
  )r
  r`   r  r  r  s   *,rr   check_equal_gmeanr
        dCU[[CFCCru   c                  8    \        V R \        P                  /VB # r
  )r
  r`   hmeanr
  s   *,rr   check_equal_hmeanr
    r
  ru   c                  8    \        V R \        P                  /VB # r
  )r
  r`   pmeanr
  s   *,rr   check_equal_pmeanr    r
  ru   c                   ~  a  ] tR tRt o ]P
                  P                  R4      R 4       t]P
                  P                  R. RO4      R 4       t	]P
                  P                  R4      ]P
                  P                  R4      R 4       4       t
]P
                  P                  R4      ]P
                  P                  R4      R	 4       4       tR
 tR t]P
                  P                  R4      R 4       tR t]P
                  P                  R4      R 4       t]! RRR7      R 4       tR tR tR t]R 4       tRtV tR# )	TestHMeani  r  c                ,    . ROp^ p\        W#VR7       R# r\   r=  N)r\   r~   rJ   )r
  rh   ri   rL  r  s   &&  rr   test_0TestHMean.test_0  s    !,ru   r<   Nc                t    Vf   TM
\        W!4      p. ROpRp\        W4WR7       . ROpRp\        W4WR7       R # )Ng=b#A@rv  r  r=  gQ?)rl  r
  rh   r<   ri   rL  r  s   &&&  rr   test_1dTestHMean.test_1d  s<    GB,> 6!E9:!E9ru   r  c                V    \         P                  ! ^^ .4      pRp\        W#VRR7       R# )r\   r   ri   r  Nr   r   r
  r  s   &&  rr   test_1d_with_zeroTestHMean.test_1d_with_zero  s'     HHaV!#6ru   c           	         \         P                  ! . RO4      pRp\        \        W1R7      ;_uu_ 4        \	        W!P
                  VRR7       RRR4       R#   + '       g   i     R# ; i)r\   z$The harmonic mean is only defined...r^  r   r  Nr\   r~   rZ   )r   r   r"   r   r
  rg   )rh   ri   rL  r  s   &&  rr   test_1d_with_negative_value%TestHMean.test_1d_with_negative_value  sC     HHZ 8w>>aBS9 ?>>>s   AA(	c                b    . RO. RO. RO.pRp\        \        P                  ! V4      W1R7       R# )rT   gmWUC@r=  NrT   r  r$  r%  r  r  r  r  r\  r  r  x   )r
  r   r   r  s   &&  rr   test_2dTestHMean.test_2d  s'    /1DE"((1+w6ru   c                h    . RO. RO. RO.p\         P                  ! . RO4      p\        W#^ VR7       R# )rT   rD   ri   Nr  r  r  )U|ϊ6@g4.=C@JQ6LsJ@E]P@r  r  s   &&  rr   test_2d_axis0TestHMean.test_2d_axis0  s+    /1DE((OP!14ru   c                h    . RO. RO. RO.p\         P                  ! . RO4      p\        W#^ VR7       R# )rT   r  NrT   r~   r$  r%  r  r  )r   r   r!  r"  r  r  s   &&  rr   test_2d_axis0_with_zero!TestHMean.test_2d_axis0_with_zero#  s+    .0CD((GH!14ru   c                h    . RO. RO. RO.p\         P                  ! . RO4      p\        W#^VR7       R# )rT   r  Nr  r  r  )g3333333@O@y@Y@r  r  s   &&  rr   test_2d_axis1TestHMean.test_2d_axis1)  s+    /1DE((<=!14ru   c                h    . RO. RO. RO.p\         P                  ! . RO4      p\        W#^VR7       R# )rT   r  Nr&  r  r  )r   r*  r+  r  r  s   &&  rr   test_2d_axis1_with_zero!TestHMean.test_2d_axis1_with_zero/  s+    .0CD((;<!14ru   T,array-likes only supported for NumPy backendr  c                    . ROp. ROpRp\         P                  ! W#R7      p\        V\        P                  ! V4      RR7       R# )rJ   r   r  ro  r  NrJ   rT   r   rT   r|   r   )r`   r
  r2   r   r^   rh   ri   rL  r  r  r   s   &&    rr   test_weights_1d_listTestHMean.test_weights_1d_list5  s8      kk!-RZZ0t<ru   c                    \         P                  ! . RO4      p\         P                  ! . RO4      p^p\        W$VRVR7       R# )rJ   ro  r  r  ri   Nr4  r5  )r   r^   r
  rh   ri   rL  r  r  s   &&   rr   test_weights_1dTestHMean.test_weights_1dE  s3     JJz"**Z(!gDRHru   c           	         \         P                  ! ^^.^
^.^^..4      p\         P                  ! ^
^.^^.^^..4      p\         P                  ! ^^.4      p\        W$^ VRVR7       R# )rJ   ro  rD   r  r  ri   Nr  r;  s   &&   rr   test_weights_2d_axis0TestHMean.test_weights_2d_axis0M  sh     HHq!fr1g1v./((RGaVaV45((Aq6"!1gDRPru   c           	         \         P                  ! . RO. RO.4      p\         P                  ! . RO. RO.4      p\         P                  ! ^^.4      p\        W$^VRVR7       R# )rJ   ro  r?  Nr4  rO   rO   rO   r5  r  r  r;  s   &&   rr   test_weights_2d_axis1TestHMean.test_weights_2d_axis1U  sL     HHj),-((J	23((Aq6"!1gDRPru   c           	         \         P                  ! . RO4      p\         P                  P                  . RO. ROR7      pRp\         P                  p\        W$VR\         P                  VR7       R# )	r   r   r  ro  r  r  r<   ri   Nr   r   r         E@r   r=   r   rI  r~   r~   r~   r\   )r   r   r	  r
  r   r;  s   &&   rr   test_weights_masked_1d_array&TestHMean.test_weights_masked_1d_array]  sO     HH'(%%++0|+DUU!gD "

r	3ru   r   Nrv
  rj  r   )r   r   r   r   r   r   r   r  r   r  r  r  r  r#  r'  r,  r/  r   r7  r<  r@  rD  r    rL  r   r   r   s   @rr   r  r    si    [[ WX- Y-
 [[W&KL
: M
: [[ WX[[ VW7 X Y7
 [[ VW[[ WX: Y X:75 [[ WX5 Y5
5 [[ WX5 Y5
 =
=	
=IQQ 3 3ru   r  c                   |  a  ] tR tRt o ]P
                  P                  R4      R 4       t]P
                  P                  R. RO4      R 4       t	R t
R tR	 tR
 t]P
                  P                  R4      R 4       t]P
                  P                  R4      R 4       t]! RRR7      R 4       tR t]R 4       tRtV tR# )	TestGMeanii  z<ignore:divide by zero encountered in log:RuntimeWarning:daskc                ,    . ROp^ p\        W#VR7       R# r  r
  r  s   &&  rr   r  TestGMean.test_0k  s     !,ru   r<   Nc                    Vf   TM
\        W!4      p. ROpRp\        W4WR7       . ROp\        ^R4      p\        W4VRVR7       R # )NgsŤF@rv  rh  )r<   r  ri   r  r=  r  )rl  r
  r   r  s   &&&  rr   r  TestGMean.test_1ds  sG    GB,> 6!E9w/!E"Eru   c                :    . RO. RO. RO.pRp\        W#VR7       R# )rT   g/,$qJ@r=  Nr  r  r  rR  r  s   &&  rr   r  TestGMean.test_2d  s    /1DE!,ru   c                    . RO. RO. RO.p\         P                  ! . RO4      p\        W#^ VR7       \        . R	O. R	O. R	O.4      p\        . R	O4      p\        W#^ RVR7       R# )
rT   r  rh  rD   r  ri   Nr  r  r  )g4@A@g5H@g`8|wЭN@gm~&+R@r=  r   r   r
  r  s   &&  rr   r#  TestGMean.test_2d_axis0  sT    /1DE((NO!14<|<=%!15R@ru   c                    . RO. RO. RO.p\         P                  ! . RO4      p\        W#^VR7       \        . R	O. R	O. R	O.4      p\        ^R
4      p\        WDV.4      p\        W#^RVR7       R# )rT   r  rh  rY  Nr  r  r  )gD" 6"6@gcP@g
ҧZ@r=  r  )r   r   r
  r   )rh   ri   rL  r  r   s   &&   rr   r,  TestGMean.test_2d_axis1  sd    /1DE((CD!14<|<=-)q	"!15R@ru   c                @    \        . RO4      pRp\        W#RVR7       R# )}Ô%ITr|  r+  r  ri   N)r_  r|  gu <7~)r   r
  r  s   &&  rr   test_large_valuesTestGMean.test_large_values  s    '(!5R8ru   c                    . ROpRp\         P                  ! RR7      ;_uu_ 4        \        W#VR7       RRR4       R#   + '       g   i     R# ; i)rT   r   r   r
  r=  N)
rT   r  r$  r%  r  r  r  r  r\  r~   )r   r  r
  r  s   &&  rr   test_1d_with_0TestGMean.test_1d_with_0  s6    
 4[[X&&aR0 '&&&s	   >A	z;ignore:invalid value encountered in log:RuntimeWarning:daskc                    . ROp\         P                  p\         P                  ! RR7      ;_uu_ 4        \        W#VR7       RRR4       R#   + '       g   i     R# ; i)rT   r   r  r=  N)
rT   r  r$  r%  r  r  r  r  r\  rZ   )r   rg   r  r
  r  s   &&  rr   test_1d_negTestGMean.test_1d_neg  s:    
 5&&[[**aR0 +***s   AA	Tr1  r  c                    . ROp. ROpRp\         P                  ! W#R7      p\        V\        P                  ! V4      RR7       R# )r\   GZ*oG8@r3  ro  r  Nr  rJ   r|   r   r   r   )r`   r  r2   r   r^   r6  s   &&    rr   r7  TestGMean.test_weights_1d_list  s8     !
 kk!-RZZ0t<ru   c                    \         P                  ! . RO4      p\         P                  ! . RO4      pRp\        W$VRVR7       R# )r\   rj  ro  r:  Nr  rk  rZ  r;  s   &&   rr   r<  TestGMean.test_weights_1d  s3     HH_%((?+!gDRHru   c           	         \         P                  ! . RO4      p\         P                  P                  . RO. ROR7      pRp\         P                  p\        W$VR\         P                  VR7       R# )	r   r  rj  ro  rG  N)r   r   r   r   r=   r   )r   r=   r   r   r   r=   )r~   r~   r~   r~   r~   r\   )r   r   r	  r
  r   r;  s   &&   rr   rL  &TestGMean.test_weights_masked_1d_array  sP     HH-.%%++6=O+PUU!gD "

r	3ru   r   rN  )r   r   r   r   r   r   r   r  r   r  r  r#  r,  ra  rd  rg  r   r7  r<  r    rL  r   r   r   s   @rr   rP  rP  i  s     [[F--
 [[W&KL
F M
F-A	A9
 [[F11 [[E11 ==	=I 3 3ru   rP  c            
       P  a  ] tR tRt o R tR tR t]P                  P                  R. RO4      R 4       t
]P                  P                  R4      ]P                  P                  R	4      R
 4       4       tR t]P                  P                  R^
^.^2^<.^Z^d..R3]P                  ! ^
^.^2^<.^Z^d..4      R3.4      R 4       t]P                  P                  R. R O. R!O. R"O.R3. R#O. R!O. R"O.R3.4      R 4       t]P                  P                  R. R O. R!O. R"O.R3. R#O. R!O. R"O.R3.4      R 4       tR t]! RRR7      R 4       t]R 4       t]P                  P                  R$. R%O4      R 4       tR t]P                  P                  RR&R. R'OR3.4      R 4       tRtV tR# )(	TestPMeani  c                x    \         P                  ! W,          4      V P                  ,          ^V,          ,          # r  )r   r]  r  rL  rf  s   &&rr   pmean_referenceTestPMean.pmean_reference  s$    qtqvv%1--ru   c                    \         P                  ! W V,          ,          4      \         P                  ! V4      ,          ^V,          ,          # r  )r   r]  )rL  rf  r  s   &&&rr   wpmean_referenceTestPMean.wpmean_reference  s-    wA~&8AaC@@ru   c                    \         P                  ! \        R R7      ;_uu_ 4        \        P                  ! VP                  . RO4      VP                  ^ .4      4       RRR4       R#   + '       g   i     R# ; i)zPower mean only defined forr   Nr  )r   r   r   r`   r  r^   rQ  s   &&rr   test_bad_exponentTestPMean.test_bad_exponent  sE    ]]:-JKKKK

9-rzz1#? LKKKs   9A((A9	r<   Nc                n   Vf   TM
\        W!4      p. RORrC\        P                  \        P                  ! V4      V4      p\        W4WQVR7       . RORrC\        P                  \        P                  ! V4      V4      p\        W4WQVR7       . RO^rC\        P                  ! R4      p\        W4WQVR7       R # )Nr  rv  r  g      r=  r  )rl  rr  ru  r   r   r  rI  )rh   r<   ri   rL  rf  r  s   &&&   rr   r  TestPMean.test_1d  s    GB,>8#1++BHHQK;!<8$1++BHHQK;!<Q1''9:!<ru   r  r  c                Z    \         P                  ! ^^ .4      Rr2Rp\        W#VRVR7       R# )r\   r   r`  NrZ   )r   r   r  )rh   ri   rL  rf  r  s   &&   rr   r  TestPMean.test_1d_with_zero  s+     xxA1!cb9ru   c           	         \         P                  ! . RO4      Rr2Rp\        \        WAR7      ;_uu_ 4        \	        W#VP
                  VR7       RRR4       R#   + '       g   i     R# ; i)r\   Gz?z!The power mean is only defined...r^  r=  Nr  )r   r   r"   r   r  rg   )rh   ri   rL  rf  r  s   &&   rr   r  %TestPMean.test_1d_with_negative_value  sC    xx
#T15w>>aBFFr2 ?>>>s   AA)	r6   c                t    \         P                  \        P                  ! V4      V4      p\	        WWCR 7       R# )r=  Nrr  ru  r   r   r  )rh   rL  rf  ri   r  s   &&&& rr   test_2d_axisnoneTestPMean.test_2d_axisnone  s(     ++BHHQK;!/ru   c                @   \        \        V^ ,          4      4       UUu. uF\  p\        P                  \        P
                  ! \        \        V4      4       Uu. uF  qQV,          V,          NK  	  up4      V4      NK^  	  ppp\        WV^ VR7       R# u upi u uppi )r~   r  N)r  r  rr  ru  r   r   r  )rh   rL  rf  ri   r  r  r  s   &&&&   rr   r#  TestPMean.test_2d_axis0  s     3qt9%	
 & %%5Q=9=aA$q''=9:A &	 	 
 	!aB7	 :
s   :BB1BBc                    V Uu. uF-  p\         P                  \        P                  ! V4      V4      NK/  	  pp\	        WV^VR7       R# u upi )r\   r  Nr  )rh   rL  rf  ri   a_r  s   &&&&  rr   r,  TestPMean.test_2d_axis1  sC     IJJ"9,,RXXb\1=J!aB7 Ks   3Ac           	         . RORr2. ROp\         P                  \        P                  ! V4      W44      p\	        W#WTRVR7       R# )rJ   ro  r:  Nr4  ރBr5  )rr  rx  r   r   r  rh   ri   rL  rf  r  r  s   &&    rr   r<  TestPMean.test_weights_1d$  s6    ;1,,RXXa[!E!tKru   Tr1  r  c                    . RORr2. ROp\         P                  \        P                  ! V4      W44      p\        P
                  ! W#VR7      p\        V\        P                  ! V4      RR7       R# )rJ   r3  ro  r  Nr4  r  r5  )rr  rx  r   r   r`   r  r2   r^   )rh   ri   rL  rf  r  r  r   s   &&     rr   r7  TestPMean.test_weights_1d_list*  sQ    
 ;1,,RXXa[!E kk!0RZZ0t<ru   c           
        \         P                  ! . RO4      ^r2\         P                  P                  . RO. ROR7      p\         P                  ! W$R7      p\         P                  p\	        W#WTR\         P
                  VR7       R# )	r   r  r3  ro  rG  NrH  rJ  rK  )r   r   r	  averager  r   r  s   &&    rr   rL  &TestPMean.test_weights_masked_1d_array8  sX    xx*+Q1%%++0|+D**Q0UU!t "

r	3ru   c           
        a VR 8X  d   V3R lpM\        \        V4      p\        P                  ! ^^.^
^.^^..4      p\        P                  ! ^
^.^^.^^..4      pV! WaVR7      p\	        VSWVRVR7       R# )rx  c                 2   < \         P                  V SV4      # r#  )rr  rx  )rL  rD   r  rf  s   &&&rr   fun&TestPMean.test_weights_2d.<locals>.funI  s     11!Q@@ru   )rD   r  ro  r?  N)rl  r`   r   r   r  )	rh   rD   fun_namerf  ri   r  rL  r  r  s	   &&&f&    rr   test_weights_2dTestPMean.test_weights_2dA  s     ))A %*CHHq!fr1g1v./((RGaVaV45aG4!QG$SUVru   c                    R p\         P                  ! \        VR7      ;_uu_ 4        \        P                  ! ^.\
        P                  4       RRR4       R#   + '       g   i     R# ; i)z*Power mean only implemented for finite `p`r   N)r   r   NotImplementedErrorr`   r  r   r   r:  s   & rr   test_infinite_p_gh23111!TestPMean.test_infinite_p_gh23111R  s>     ?]].g>>KKRVV$ ?>>>s   'AA)	zp, weights, ref缉ؗҬ<g9K@c                    VP                  . RO4      pVf   RMVP                  V4      p\        P                  ! WRVR7      p\        WqP                  V4      4       R# )      I@Nr3  )r  r  g     r@g     Q@r  r  r   )r^   r`   r  r2   )rh   ri   rf  r  r   rk   r!  r   s   &&&&&   rr   test_small_p_gh23407TestPMean.test_small_p_gh23407Z  sK    8 JJ1
 ODG)<kk!*ZZ_-ru   r   rN  rt  r  r  r  r  r&  )rD   r  rf  ))Nrx  g3E#@)r~   r  r~   )r\   r
  rZ   )r  NgDG@)r   r   r   r   r   r   r   )r   r   r   r   ru  rx  r{  r   r   r   r  r   r  r  r   r   r  r#  r,  r<  r   r7  r    rL  r  r  r  r   r   r   s   @rr   rr  rr    s/    .A@ [[W&KL= M= [[ VW[[ WX: Y X:
3 [[8b"XCy
)4	0
((RHr2hS	2
3S	9	;
0
0 [[-/B
CT	J,.A
BC	H	J
8
8 [[-/B
CT	J,.A
BC	H	J
8
8L ==	= 3 3 [[!		W	W% [[  -
 02CD)	
2	.32	.ru   rr  c            	          a  ] tR tRt o ]P
                  ! ^4      ^,           P                  4       tRt]P                  ! ]R4      P                  4       t
R t]! RRR7      R 4       t]R 4       t]P"                  P%                  R	4      ]P"                  P%                  R
4      ]P"                  P'                  R^ R]P(                  ]P*                  34      R 4       4       4       tR tR tR tR tR tR tR tRtV tR# )TestGSTDi  gv aZ@c                    \         P                  ! VP                  V P                  4      4      p\	        W!P                  V P
                  4      4       R # r#  )r`   r  r^   array_1dr2   gstd_array_1drh   ri   gstd_actuals   && rr   test_1d_arrayTestGSTD.test_1d_array  s4    jjDMM!:;ZZ0B0B%CDru   Tz$Only NumPy supports array-like inputr  c                    \         P                  ! \        V P                  4      4      p\	        W P
                  4       R # r#  )r`   r  r
  r  r
   r  r  s   && rr    test_1d_numeric_array_like_input)TestGSTD.test_1d_numeric_array_like_input  s'    jjt}}!56%7%78ru   c                    R p\         P                  ! \        \        3VR7      ;_uu_ 4        \        P
                  ! R4       RRR4       R#   + '       g   i     R# ; i)z.could not convert string to float|The DType...r   z*You cannot take the logarithm of a string.N)r   r   r   rg  r`   r  r5  s   && rr   #test_raises_error_non_numeric_input,TestGSTD.test_raises_error_non_numeric_input  s:    B]]J	2'BBJJCD CBBBs   AA	r  r  	bad_valuec                   VP                  V P                  V.,           4      p\        P                  ! V4      '       dX   \	        V4      '       gG   R p\
        P                  ! \        VR7      ;_uu_ 4        \        P                  ! V4      pRRR4       M\        P                  ! V4      p\        XVP                  \        P                  4      4       R#   + '       g   i     L;; i)z3The geometric standard deviation is only defined...r   N)r^   r  r   r  r#   r   r  r   r`   r  r3   rg   )rh   r  ri   rk   r  r   s   &&&   rr   test_returns_nan_invalid_value'TestGSTD.test_returns_nan_invalid_value  s     JJt}}	{23;;y!!-*:*:KGnG<<jjm =< **Q-CRZZ/0	 =<s   7CC(	c                    VP                  . ROVP                  ^^^..4      p\        P                  ! V^R7      p\	        W1P                  ^\
        P                  .4      4       R# )r\   rC   N)r\   r\   r\   r
  )r^   rg   r`   r  r2   r   )rh   ri   rL  r  s   &&  rr   test_propagates_nan_values#TestGSTD.test_propagates_nan_values  sK    JJ1a'89:jj+ZZBFF%<=ru   c                    VP                  V P                  4      p\        P                  ! W"P                  ^ ,          R7      p\        W1P                  VP                  4      4       R# )r~   r  N)r^   r  r`   r  r  r3   rg   r   s   &&  rr   )test_ddof_equal_to_number_of_observations2TestGSTD.test_ddof_equal_to_number_of_observations  s?    JJt}}%jj,ZZ/0ru   c                    VP                  V P                  4      p\        P                  ! VR R7      pVP                  V P                  4      p\        W44       R # )NrC   )r^   array_3dr`   r  r  r2   rh   ri   rk   r  r   s   &&   rr   test_3d_arrayTestGSTD.test_3d_array  s@    JJt}}%jj.jj++,)ru   c                    VP                  V P                  4      p\        P                  ! VRR7      pVP                  RR.4      p\	        W44       R# )r\   rC   g~d@g<
?Nrp  r^   r  r`   r  r2   r  s   &&   rr   test_3d_array_axis_type_tuple&TestGSTD.test_3d_array_axis_type_tuple  s?    JJt}}%jj0jj*j12)ru   c                    VP                  V P                  4      p\        P                  ! V^ R7      pVP                  . RO. RO. RO.4      p\	        W44       R# )r~   rC   N)gĶ?@gQӫ@g֌@gן7R@)g%Ȁ@ge@gv6B5 @g _?)gE ?gީ?gk̊Y?g%:?r  rh   ri   rk   r  gstd_desireds   &&   rr   test_3d_array_axis_0TestGSTD.test_3d_array_axis_0  sJ    JJt}}%jj+zzOOO#
 
 	2ru   c                    VP                  V P                  4      p\        P                  ! V^R7      pVP                  . RO. RO.4      p\	        W44       R# )r\   rC   N)gN@g85@g'ĥ?gwth?)gHnaX?gG?gBu?gE?r  r  s   &&   rr   test_3d_array_axis_1TestGSTD.test_3d_array_axis_1  sG    JJt}}%jj+zzLL#
  	2ru   c                    VP                  V P                  4      p\        P                  ! V^R7      pVP                  . RO. RO.4      p\	        W44       R# )rJ   rC   N)gt80?gі?g?)gۘ]~?gmT6(?g	G?r  r  s   &&   rr   test_3d_array_axis_2TestGSTD.test_3d_array_axis_2  sG    JJt}}%jj+zz??#
  	2ru   r   Nr2	  rZ   ) r   r   r   r   r   r   r   r  r  rb   r  r  r   r  r    r  r   r   r   r   r   rg   r  r  r  r  r  r  r  r  r   r   r   s   @rr   r  r    s     		)$q(002H"Mzz(I.557HE d+QR9 S9 E E
 [[ WX[[ VW[[[1b"&&"&&*AB1 C X Y1>
1
**333 3ru   r  c                   ,   a  ] tR tRt o R tR tRtV tR# )TestTrimi  c           
        \         P                  ! ^4      p\        \         P                  ! \        P
                  ! VR4      4      \         P                  ! ^
4      4       \        \         P                  ! \        P
                  ! VR4      4      \         P                  ! ^	4      4       \        \         P                  ! \        P
                  ! VRRR7      4      \         P                  ! ^^4      4       \        \         P                  ! \        P
                  ! VR	RR7      4      \         P                  ! ^^4      4       \        \        P
                  ! VR4      . 4       \        \        P
                  ! VRRR7      . 4       \        \        P
                  ! . R4      . 4       \        \        P
                  ! . R	RR7      . 4       \        \        P
                  ! . R
4      . 4       \         P                  ! ^4      P                  ^^4      p\         P                  ! ^^4      P                  ^^4      p^ p\        P
                  ! VRRVR7      p\        \         P                  ! WCR7      V4       ^p\        P
                  ! VP                  RRVR7      p\        \         P                  ! WCR7      VP                  4       R# )rU   rC  r@  left)tailr   )r  rD   rC   NtE]t?rG  )r   r   r   r  r`   trim1rb   r   )rh   rL  r   rD   trimmeds   &    rr   
test_trim1TestTrim.test_trim1  s   IIbMRWWU[[C01299R=ARWWU[[C01299Q<@RWWU[[Cf=>YYq"%	'RWWU[[E?@YYq"%	'U[[C("-U[[Cf5r: 	U[[S)2.U[[U8"=U[[T*B/ IIbM!!!Q'ii2&&q!,++a6=RWWW0#6++acc3V$?RWWW0#%%8ru   c           
        \         P                  ! ^4      p\        \         P                  ! \        P
                  ! VR4      4      \         P                  ! ^^4      4       \        \         P                  ! \        P
                  ! VR4      4      \         P                  ! . RO4      4       \        \         P                  ! \        P
                  ! \         P                  ! ^4      P                  ^^4      R4      4      \         P                  ! ^^4      P                  ^^4      4       \        \         P                  ! \        P
                  ! \         P                  ! ^4      P                  ^^4      P                  R4      4      \         P                  ! . RO. RO.4      4       \        \        \        P
                  \         P                  ! ^4      P                  ^^4      P                  R	4       \        \        P
                  ! . R4      . 4       \        \        P
                  ! . R4      . 4       \        \        P
                  ! . R	4      . 4       R# )
rU   r@  rC  Nr  )rJ   r   r   r|   r   rO   rK   rW  )rJ   rK   rR   r  )r   rN   rw  rQ   rG  )r   r   r   r  r`   trimbothr   rb   r   r   r   )rh   rL  s   & rr   test_trimbothTestTrim.test_trimboth  ss   IIbMRWWU^^Au56		!QHRWWU^^As34XX34	6RWWU^^BIIbM,A,A!Q,GMNYYq"%--a3	5RWWU^^BIIbM,A,A!Q,G,I,I,02 3XX~~>?	A 	j%..iim++Aq133T	; 	U^^B,b1U^^B.3U^^B-r2ru   r   N)r   r   r   r   r  r  r   r   r   s   @rr   r  r    s     983 3ru   r  c                   F   a  ] tR tRt o R t]! RRR7      R 4       tRtV tR# )	TestTrimMeani  c                x   \         P                  ! . RO4      p\         P                  ! ^4      P                  ^^4      VR3,          p\         P                  ! ^4      P                  ^^RR7      VR3,          p\	        \
        P                  ! VP                  V4      R4      VP                  . RO4      4       \	        \
        P                  ! VP                  V4      R4      VP                  . RO4      4       \         P                  ! . RO4      p\         P                  ! ^4      P                  ^^4      VR3,          p\	        \
        P                  ! VP                  V4      R4      VP                  . RO4      4       . ROp\	        \
        P                  ! VP                  V4      R4      VP                  R4      4       \	        \
        P                  ! VP                  . RO4      R4      VP                  R4      4       \         P                  P                  R4      pVP                  ^RR7      pVP                  V4      pR FN  p	\
        P                  ! VRV	R7      p
\
        P                  ! VP                  Wy^ 4      R4      p\	        W4       KP  	  \
        P                  ! VRR	R7      p
\
        P                  ! \        V4      R4      p\	        W4       \        P                  ! \        R
R7      ;_uu_ 4        \
        P                  ! VR4       R	R	R	4       R	#   + '       g   i     R	# ; i)r   r   Fr  ry   rS   r
  r  rC   NzProportion too big.r   r{  )r   r|   r~   r\   rJ   r   rW  )ry   r?  g      -@g     4@r  )r\   r~   r   rJ   )r   r   r  r  r  r  )rO   rU   rV   rQ   r
  r   r	  r\   r|   r~   r	  rT   r	  rN   r	  rw  r  r  rJ   rR   r   rW   rK   r   )r|   r   r   r\   rJ   r~   )r|   r   r   rO   )r~   r\   rJ   r   rZ   )r   r   r   rb   r3   r`   	trim_meanr^   r  r  r  r  r*   r   r   r   )rh   ri   idxr  r  idx4a4rL  r  rD   r  r  s   &&          rr   test_trim_meanTestTrimMean.test_trim_mean  s9   hh)*YYr]""1a(a0YYr]""1as"3CF;

2=

#9:	<

2=

#78	: xx%YYr]""1a(q1

2=

#@A	C%

1t<bjj>NO

3E(FM

3	) ii##J/LL,L/JJqM$D??1d6D??2;;q#:DADD' %
 q$T2x{D1#]]:-BCCOOAs# DCCCs   L((L9	rM	  zlazy -> no _axis_nan_policyr  c           	     ,   \         P                  ! \        R R7      ;_uu_ 4        \        \        P
                  ! VP                  . 4      R4      VP                  VP                  4      4       RRR4       \         P                  ! \        R R7      ;_uu_ 4        \        \        P
                  ! VP                  . 4      R4      VP                  VP                  4      4       RRR4       R#   + '       g   i     L; i  + '       g   i     R# ; i)r>  r   r   Nr{  )r   r  r   r3   r`   r  r^   rg   rQ  s   &&rr   r  TestTrimMean.test_empty_input/  s     \\,K@@EOOBJJrNC@"**RVVBTU A\\,K@@EOOBJJrNC@"**RVVBTU A@ A@@@@s   A
C/A
D/C?	D	r   N)	r   r   r   r   r  r   r  r   r   r   s   @rr   r  r    s*     %$N l+HIV JVru   r  c                      a  ] tR tRt o R tR tR tR tR t]	P                  P                  R. RO4      R	 4       tR
tV tR# )TestSigmaClipi8  c                   VP                  VP                  R R^4      VP                  RR^4      34      p^p\        P                  ! V4      w  rEpVP	                  V4      V8  g   Q hVP                  V4      V8  g   Q h\        WQP                  V4      W1P                  V4      ,          ,
          4       \        WaP                  V4      W1P                  V4      ,          ,           4       VP                  VP                  8X  g   Q hR# )r  rY   r   r9  N
r  rA  r`   	sigmaclipr  r  r3   rc   r   r  rh   ri   rL  factri  r  upps   &&     rr   test_sigmaclip1TestSigmaClip.test_sigmaclip1:  s    IIr{{3b1{{2sA.0 1ooa(vvay3vvay3WWQZ$vvay.89WWQZ$vvay.89ww!''!!!ru   c                ,   VP                  VP                  R R^4      VP                  RR^4      34      pRp\        P                  ! W#V4      w  rEpVP	                  V4      V8  g   Q hVP                  V4      V8  g   Q h\        WQP                  V4      W1P                  V4      ,          ,
          4       \        WaP                  V4      W1P                  V4      ,          ,           4       VP                  R8X  g   Q hVP                  R8X  g   Q hR# )r  rY   r   r9  r7   N)r   )$   r  r  s   &&     rr   test_sigmaclip2TestSigmaClip.test_sigmaclip2E  s    IIr{{3b1{{2sA.0 1ooat4vvay3vvay3WWQZ$vvay.89WWQZ$vvay.89ww$ww%ru   c                   VP                  VP                  R R^4      VP                  RR^4      34      pRp\        P                  ! W#V4      w  rEpVP	                  V4      V8  g   Q hVP                  V4      V8  g   Q h\        WQP                  V4      W1P                  V4      ,          ,
          4       \        WaP                  V4      W1P                  V4      ,          ,           4       \        WAP                  R R^4      4       R# )r  rY   ?N      Y      I)	r  rA  r`   r  r  r  r3   rc   r   r  s   &&     rr   test_sigmaclip3TestSigmaClip.test_sigmaclip3Q  s    IIr{{3b1{{5$24 5ooat4vvay3vvay3WWQZ$vvay.89WWQZ$vvay.89;;sD"56ru   c                    VP                  VP                  R R^4      VP                  RR^4      34      pRp\        P                  ! W#V4      pRp\	        WEVR7       R# )r  rY   r  r=  Nr  r  )clippedr  upper)r  rA  r`   r  r   )rh   ri   rL  r  r   rA  s   &&    rr    test_sigmaclip_result_attributes.TestSigmaClip.test_sigmaclip_result_attributes\  sX    IIr{{3b1{{5$24 5ooat,2
C3ru   c                v    VP                  ^
4      p\        \        P                  ! V4      ^ ,          V4       R# )rT   N)r  r3   r`   r  r   s   && rr   test_std_zeroTestSigmaClip.test_std_zerod  s'    GGBK*1-q1ru   r<   Nc                   Vf   R M
\        W!4      p\        P                  P                  R4      pVP                  ^
4      p\        P                  V^&   VP                  VP                  4       VR7      p\        V4      '       gG   \        P                  ! \        RR7      ;_uu_ 4        \        P                  ! VRR7       R R R 4       ME\        P                  ! \        RR7      ;_uu_ 4        \        P                  ! VRR7       R R R 4       \        P                  ! VRR7      p\        VP                  VP                  . VR7      4       \        VP                   VP                  VP                  VR7      4       \        VP"                  VP                  VP                  VR7      4       \        P                  ! VR	R7      p\        P                  ! WBP%                  V4      ( ,          4      p\        VP                  VP                  4       \        VP                   VP                   4       \        VP"                  VP"                  4       R #   + '       g   i     ELX; i  + '       g   i     ELl; i)
Nl   &G# r;   r   r   r   r   r
  r  r   )rl  r   r  r  rg   r^   r   r#   r   r   r   r`   r  rg  r3   r  r  r  rx  )rh   r<   ri   r  rk   r   r   s   &&&    rr   r  TestSigmaClip.test_nan_policyi  s   72+=ii##$9:JJrNvv!JJqxxzJ/ Qz1PQQg6 RQ y0UVVg6 W ooaK8RZZ%Z%@A		2::bffE:#BC		2::bffE:#BC ooaF3ooa!o.S[[1		399-		399-# RQQ WVVs   +I1I-I*	-I>	r   r  )r   r   r   r   r  r  r  r  r  r   r   r   r  r   r   r   s   @rr   r  r  8  sE     	"
 	742
 [[W&BC. D.ru   r  c                     a  ] tR tRt o R t]! RRR7      ]P                  P                  R^^.. 3^^.^3^^.^.3.4      R 4       4       t	]P                  P                  R4      R	 4       tR
 tR tR tR t]! RRR7      R 4       t]! RRR7      R 4       t]! RRR7      R 4       t]! RRR7      ]P                  P                  R. RO4      R 4       4       tRtV tR# )TestAlexanderGoverni  c                   . RO. RO. RO. RO.pV Uu. uF)  p\         P                  ! V\         P                  R7      NK+  	  ppV Uu. uF)  p\         P                  ! V\         P                  R7      NK+  	  ppV Uu. uF)  p\         P                  ! V\         P                  R7      NK+  	  ppV Uu. uF)  p\         P                  ! V\         P
                  R7      NK+  	  pp\        P                  ! V!  p\        P                  ! V!  p\        P                  ! V!  p	\        P                  ! V!  p
VP                  VP                  u;8X  d%   V	P                  u;8X  d   V
P                  8X  g   Q h Q hVP                  VP                  u;8X  d%   V	P                  u;8X  d   V
P                  8X  g   Q h Q hR# u upi u upi u upi u upi )rW   r;   N)	rW   rW   rW   rW   rW   rW   rW   rV   rV   )	rR   rW   rV   rV   rV   rV   rV   rU   rU   )	rR   rR   rW   rW   rW   rW   rW   rV   rV   )	rw  rR   rW   rW   rW   rV   rV   rV   rU   )
r   r^   r   r  rf  r   r`   alexandergovernr?  r@  )rh   r  re	  
args_int16
args_int32
args_uint8args_float64	res_int16	res_int32	res_uint8res_float64s   &          rr   test_compare_dtypes'TestAlexanderGovern.test_compare_dtypes  s   44446 BFF#bjjBHH5
FAEF#bjjBHH5
FAEF#bjjBHH5
FEIJTc

3bjj9TJ)):6	)):6	)):6	++\:  I$4$4 7  7$/$6$67 	8 7 	8 7##y':': =##='2'<'<= 	> = 	> = GFFJs   /F=/G=/G2/GrM	  z%Requires `_axis_nan_policy` decoratorr  r  c                   \        V4      pVP                  V^ ,          VR7      VP                  V^,          VR7      3p\        P                  ! \        \
        R7      ;_uu_ 4        \        P                  ! V!  pRRR4       \        XP                  VP                  VP                  4      4       \        VP                  VP                  VP                  4      4       R#   + '       g   i     Lp; i)r~   r;   r   N)r&   r^   r   r  r   r   r`   r  r3   r@  rg   r?  )rh   r  ri   r<   r   s   &&&  rr   test_too_small_inputs)TestAlexanderGovern.test_too_small_inputs  s     !$zz$q'z/DG51QQ\\,4IJJ''.C Krzz"&&'9:

BJJrvv$67 KJs   (C%%C5	r
  c                ^   VP                  R R.4      VP                  VP                  VP                  .4      3p\        P                  ! V!  p\	        VP
                  VP                  VP                  4      4       \	        VP                  VP                  VP                  4      4       R# )r   r   N)r^   r   r`   r  r3   r@  rg   r?  )rh   ri   samplesr   s   &&  rr   test_bad_inputs#TestAlexanderGovern.test_bad_inputs  sw     **b"X&

BFFBFF3C(DD##W-rzz"&&'9:

BJJrvv$67ru   c                   . RNRNRNRNRNRNRNRNRNRNRNRNRNRNR	NRNR NR
NRNRNRNRNRNR!NRNR"NR#NR$NR%NRNRNR&NRNR'NR(NR)NR*NR+NRNRNp. R,Op. R-OpVP                  V4      VP                  V4      VP                  V4      rCp\        P                  ! W#V4      p\        VP                  VP                  R4      4       \        VP
                  VP                  R4      4       R# ).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

gb?g^s/@gQ\
@gIF@gpZ>q@g7D @g(Q<.@gON/@gs&@gzp"@gzl @g57@g>m@*a"@g,ZbI@gX9
]?g1@g;@gL97vp,+@g"@glA! &@go@gFQ?gsdlT6?Ng5gl"k gRh g-ѪZng	Ogҷ%6gw)ܿg789Ŀg3gg"Jbgy:@j-g2A gp/rgfT|7g^v+g٘Yg0/g)g!_5*gqTEgg1$@g	a, @gˤ$g9%gZ@gT6U'@gS g6aGn*@gO@g\\"g^4w@g0gr~5@gDP=@gٿԙg:4R/gOx]r!@gU4 gta)B@g,Gg2$@gD%?gI,VL&g@gч
;g5@gWJc@gYjK@@)gj7#@g<e,g}m(@gr2g#^9gr/P@g:RO!g'9?g6F
?gcʱ'gt*&gg~*0Ώ7@gWzx>gIK'@g?Ң@g7C5@g;C>ϯUg7@g)n}.^@r^   r`   r  r2   r@  r?  )rh   ri   r  twoeightsolns   &&    rr   test_compare_r"TestAlexanderGovern.test_compare_r  s   :"! "#5 "7I "!"#4"6H" ""3"5F" #" %6" 8I" #	" %8	" :L	"
 #"
 %9"
 ;L" !" #4" 6G" !" #4" 6H" !" #6" 8J" #" %7" 9J" !" #6" 8I" #" %8" :L" "" $7" 9K" !"	I7 **S/2::c?BJJu<M%$$Su5

3H(IJRZZ0F%GHru   c                   . ROp. ROp. ROpVP                  V4      VP                  V4      VP                  V4      rCp\        P                  ! W#V4      p\        VP                  VP                  R4      RR7       \        VP
                  VP                  R4      RR7        \        VP                  VP                  R4      RR7       \        VP
                  VP                  R	4      RR7       R
# )z
Data taken from 'The Modification and Evaluation of the
Alexander-Govern Test in Terms of Power' by Kingsley Ochuko, T.,
Abdullah, S., Binti Zain, Z., & Soaad Syed Yahaya, S. (2015).
g]FxK@r  rF  g R8߱?rC	  g.UikL@>r  g&Zݱ?N)g{G&~@g(\E~@g=
ףp~@gfffff~@gp=
~@gףp=
k@gR@gQ2@g8@g`@g)\`@g=
ףp@g(\#@g@g=
ףpE@gQ@g
ףp=@gףp=
Մ@g33333#@)g=
ףpt@g{G&u@gq=
ףv@gQDy@g     X{@g\(P}@g]~@gt~@gzG~@g)\~@g1@gzG@)gGz8@g     @gp=
ב@g
ףp=@gGzԀ@gq=
ףր@gRi@g{Gt@gw@g=
ףp@gQS@gQ@g(\©@gQ-@gQ@r!  )rh   ri   youngmiddleoldr$  s   &&    rr   test_compare_scholar(TestAlexanderGovern.test_compare_scholar  s    08GZZ.

60BBJJsOs$$UC8

6(:FRZZ%8tD	: 	

8(<4HRZZ
%;$Gru   c                   . R
Op. ROpVP                  V4      VP                  V4      r2\        P                  ! W#4      p\        VP                  VP                  R4      RR7       \        VP
                  VP                  R4      RR7        \        VP                  VP                  R4      RR7       \        VP
                  VP                  R4      RR7       R	# )a  
Data taken from 'Robustness And Comparative Power Of WelchAspin,
Alexander-Govern And Yuen Tests Under Non-Normality And Variance
Heteroscedasticity', by Ayed A. Almoied. 2017. Page 34-37.
https://digitalcommons.wayne.edu/cgi/viewcontent.cgi?article=2775&context=oa_dissertations
g[z4?ro  rF  gOZ}?g9$?r(  r  g6MDd}?N)gQhgMOgY9gnSgdȿg}"Ogиp $?g8~4?g
p?gZ?g)J=?gX?gpG?gᛦ?gg?gh8en?g?guwg?gbX?g9SI?)gN]g	cZgAJigMg'gk,	Pg<gΈ׿g%zrKտg0BxҿgPNg:ǀgvꭁg?k~?gI\߇?gi?g<)?g0Xr?g
F%u?gsFZ*o?g7n:?r!  )rh   ri   r  r  r$  s   &&   rr   test_compare_scholar3)TestAlexanderGovern.test_compare_scholar3(  s    "! BBB$$R,

8(<4HRZZ%9E	8 	

9(=DIRZZ	%:Fru   c                J   VP                  . RO4      VP                  ^VP                  .4      3p\        P                  ! V!  p\	        VP
                  VP                  VP                  4      4       \	        VP                  VP                  VP                  4      4       R# )r   Nr  r^   rg   r`   r  r3   r?  r@  )rh   ri   r  r   s   &&  rr   r,	  -TestAlexanderGovern.test_nan_policy_propagate[  sm    zz*+RZZBFF-DD##T*

BJJrvv$67rzz"&&'9:ru   Tr   r   c                    VP                  . RO4      VP                  R VP                  .4      3p\        \        RR7      ;_uu_ 4        \        P
                  ! VRR/  RRR4       R#   + '       g   i     R# ; i)r   r   r   r   r   Nr  )r^   rg   r   r   r`   r  )rh   ri   r  s   && rr   test_nan_policy_raise)TestAlexanderGovern.test_nan_policy_raiseb  sX    zz*+RZZRVV-EE:-LMM!!4<G< NMMMs   
A++A<	c                   VP                  ^^^VP                  ^.4      VP                  ^VP                  ^^.4      3pVP                  . RO4      VP                  . RO4      3p\        P                  ! VRR/ p\        P                  ! V!  p\	        VP
                  VP
                  4       \	        VP                  VP                  4       R# )r\   r   r   Nr=  )r\   r	  r  r2  )rh   ri   args_nanargs_no_nanres_nan
res_no_nans   &&    rr   test_nan_policy_omit(TestAlexanderGovern.test_nan_policy_omith  s    ::q!Q23RZZBFFBPR@S5TTjj.

;0GG''EfE**K8

(9(9:)):+?+?@ru   z"lazy -> no data-dependent warningsc                   VP                  . RO4      pVP                  . RO4      p\        P                  ! \        RR7      ;_uu_ 4        \        P
                  ! W#4      pRRR4       \        XP                  VP                  VP                  4      4       \        VP                  VP                  VP                  4      4       R#   + '       g   i     Lp; i)r\  r  r   Nr_  r`  )
r^   r   r  r   r`   r  r3   r@  rg   r?  )rh   ri   r  r  r   s   &&   rr   rg  'TestAlexanderGovern.test_constant_inputq  s     ZZ-.ZZ-.\\.0LMM''/C Nrzz"&&'9:

BJJrvv$67 NMs   C

C	rD   Nc                   \         P                  P                  R 4      pRp\        ^4       Uu. uF  qSP                  V4      NK  	  ppV Uu. uF  qqP	                  V4      NK  	  pp\
        P                  ! VRV/ p	\
        P                  ! VRV/ p
\        V
P                  VP	                  V	P                  4      4       \        V
P                  VP	                  V	P                  4      4       R# u upi u upi )l   >030 rD   N)rU   rV   )
r   r  r  r  r^   r`   r  r2   r@  r?  )rh   ri   rD   r  r  r  r  sample
xp_samplesr   r   s   &&&        rr   test_2d_input!TestAlexanderGovern.test_2d_input{  s     ii##$89.3Ah7h::e$h77>?wVjj(w
?##W848##Z;d;rzz#--'@A

BJJszz$:; 8?s   C5C:r   r  )r   r   r   r   r  r   r   r   r   r  r   r  r%  r,  r/  r,	  r5  r<  rg  rC  r   r   r   s   @rr   r  r    s1    >* l+RS[[Vq!fb\QFA;!Q!$NO8 P T8 [[ QR8 S8BIH1Hf1Gf; .WX= Y=
 .WXA YA .RS8 T8 l+RS[[V\2	< 3 T	<ru   r  c            
         a  ] tR tRt o R t]P                  P                  R. R O4      R 4       t]P                  P                  R. R O4      R 4       t	R t
R tR	 tR
 t]P                  P                  R. R!O4      R 4       t]P                  P                  R4      ]P                  P                  R]P"                  ! . R"O4      ]P"                  ! . R#O4      ]P$                  R33]P"                  ! . R"O4      ]P"                  ! . R"O4      ]P&                  ]P&                  33.4      R 4       4       t]! RRR7      ]P                  P                  R. R$O4      R 4       4       t]! RRR7      R 4       tR t]! RRR7      R 4       tR t]! RRR7      ]P                  P                  RR. R%O3.4      R 4       4       t]! RRR7      R 4       tR tRtV tR# )&TestFOneWayi  c                    \         P                  ! VP                  ^ ^.4      VP                  ^ ^.4      4      w  r#\        W!P                  R4      4       \        W1P                  R4      4       R# )r~   r   r   Nr`   f_onewayr^   r3   rh   ri   r  rf  s   &&  rr   test_trivialTestFOneWay.test_trivial  sN    ~~bjj!Q0"**aV2DE::c?+::c?+ru   r<   Nc                   Vf   R M
\        W!4      p\        P                  ! VP                  ^ ^.VR7      VP                  ^^.VR7      4      w  r4\	        W2P                  RVR7      4       \        WBP                  RVR7      4       R # )Nr;   r   gf3þ?)rl  r`   rI  r^   r3   r2   )rh   r<   ri   r  rf  s   &&&  rr   rY  TestFOneWay.test_basic  su     72+=~~bjj!Quj= jj!Quj=?::c:78::l%:@Aru   c                D   Vf   R M
\        W!4      pVP                  . ROVR7      VP                  . ROVR7      VP                  . ROVR7      .p\        P                  ! VRR/ w  rE\	        WBP                  RVR7      4       \	        WRP                  RVR7      4       R # )	Nr;   r	  FgM;yQ?gWe?g(\5IgHz'D@gףp=
2g
ףp=A@g333333g)\(?gq=
ףp7@rF   g333333.@gRQ#@gzG'@r|  g=
ףp#@gGz."@gp=
ף
@)rl  r^   r`   rI  r2   )rh   r<   ri   r  r  rf  s   &&&   rr   test_unequal_varTestFOneWay.test_unequal_var  s     72+=::HPU:V::6e:D::>e:LN ~~w8%8& 	::&7u:EF::&7u:EFru   c                (   VP                  . RO4      VP                  . RO4      VP                  . RO4      .pRp\        P                  ! \        VR7      ;_uu_ 4        \        P
                  ! VRR/  RRR4       R#   + '       g   i     R# ; i)	g(\5I@z(Expected a boolean value for 'equal_var'r   r	  FalseNrP  rQ  rR  r^   r   r   rg  r`   rI  )rh   ri   r  r  s   &&  rr   test_equal_var_input_validation+TestFOneWay.test_equal_var_input_validation  si    ::HI::67::>?A =]]9G44NNG7w7 5444s   B  B	c                   \         P                  ! VP                  ^.4      VP                  ^.4      VP                  . RO4      4      w  r#\        W!P                  R4      4       \        W1P                  R4      4       R# )rJ   Nr2	  r{  r=  )r`   rI  r^   r2   rJ  s   &&  rr   test_known_exactTestFOneWay.test_known_exact  sX     ~~bjj!orzz1# jj35::c?+::c?+ru   c                    VP                  R R.VP                  R7      pVP                  RR.VP                  R7      p\        P                  ! W#4      w  rE\	        WAP                  R4      4       R# )    r;       gŸ?N)r^   uint16r`   rI  r2   )rh   ri   rL  rX  r  rf  s   &&    rr   test_large_integer_array$TestFOneWay.test_large_integer_array  sZ    JJSzJ3JJSzJ3~~a# 	::&9:;ru   c                    VP                  R R.VP                  R7      pVP                  RR.VP                  R7      p\        P                  ! W#4      pRp\	        WEVR7       R# )r^  r_  r;   r`  ra  r=  NrY  )r^   rb  r`   rI  r   )rh   ri   rL  rX  r   rA  s   &&    rr   test_result_attributes"TestFOneWay.test_result_attributes  sS    JJSzJ3JJSzJ3nnQ",
C3ru   	test_casec                   VR ,           pRp\         P                  P                  \         P                  P                  \         P                  P	                  \
        4      RV4      4      p\        V4      ;_uu_ 4       pVP                  4       P                  R4      pRRR4       XR,           Uu. uF+  pVP                  4       '       g   K  VP                  4       NK-  	  p	p\        P                  ! V^<R7      p
V
P                  w  rVP                  \        4      p\        P                  ! V4      p\!        V	^ ,          R
,          4      pV Uu. uF  qP#                  WV8H  ,          4      NK  	  pp\$        P&                  ! V!  pRpVV9   d   Rp\)        V^ ,          VP#                  WbP*                  R7      VR	7       R#   + '       g   i     EL1; iu upi u upi )z.datr  zdata/nist_anova
N:r%  r]  N)skiprowsrC	  r;   r  rZ   )SmLs07SmLs08SmLs09)ospathabspathjoindirname__file__openreadsplitstripr   loadtxtr   r  r  uniquer  r^   r`   rI  r2   r   )rh   rh  ri   filenamer  fnamer  contentline	certifieddatafrl   rk   catyr  xlistr   hard_tcs   &&&               rr   	test_nistTestFOneWay.test_nist  s[    v%RWW__X-F->!J K%[[AffhnnT*G .5en &nd

 "TZZ\n	 &

52.wwHHSMyy|)A,r"#0451A1fI&5nne$ 1DA

1JJ
 ?dK% [[& 6s   > G1GG"GG	r   za, b, expectedr   c                    \         P                  ! VP                  V4      VP                  V4      4      w  rV\        WTP                  V^ ,          4      4       \        WdP                  V^,          4      4       R# r~   NrH  )rh   rL  rX  r  ri   r  rf  s   &&&&&  rr   rg  TestFOneWay.test_constant_input  sP     ~~bjjmRZZ];::hqk23::hqk23ru   rM	  zneeds _axis_nan_policyr  rD   c                   \         P                  ! . RO. RO. RO. RO. RO.\         P                  R7      p\         P                  ! . RO. R	O. R
O. RO. RO. RO. RO. RO.\         P                  R7      p\         P                  ! . RO. RO. RO. RO.\         P                  R7      pVR9   d(   VP                  pVP                  pVP                  p^ pM^p\        P
                  ! VP                  V4      VP                  V4      VP                  V4      VR7      w  rxR F  p	\        P
                  ! \         P                  ! W9V4      \         P                  ! WIV4      \         P                  ! WYV4      4      w  r\        Wy,          VP                  V
4      4       \        W,          VP                  V4      4       K  	  R F  p	\        P
                  ! \         P                  ! W9V4      \         P                  ! WIV4      \         P                  ! WYV4      4      w  r\        Wy,          VP                  V
4      4       \        W,          VP                  V4      4       K  	  R# )r\   r;   rC   N)r\   r   r   r   )rJ   r|   r   r   )r   r   r   r   re  )r   r\   r|   r   )r   r   r|   r   )r   r   r|   r   )r\   r|   r|   r   )r|   r|   r|   r   )rJ   r   r|   r   )rK   rJ   r|   r   )rJ   rJ   r|   r   )r   r   r   r   )r   rJ   r   r   )r|   r   r   r   rZ   r\   rU  rV  )	r   r   r   r   r`   rI  r^   taker2   )rh   rD   ri   rL  rX  ri  	take_axisr  rf  r  fjpjs   &&&         rr   test_2d_inputsTestFOneWay.test_2d_inputs  s    HHl""""	$ ,.::	7
 HHl"""""""$ ,.::7 HHl"""$ ,.::7
 7?AAAII~~bjjmRZZ]BJJqM#')
 A^^BGGA)$<$&GGA)$<$&GGA)$<>FB AD"**R.1AD"**R.1  A^^BGGA)$<$&GGA)$<$&GGA)$<>FB AD"**R.1AD"**R.1 ru   c           
     *   VP                  ^VP                  RRVP                  R7      ,          R4      pVP                  ^VP                  RRVP                  R7      ,          R	4      pVP                  ^VP                  RR
VP                  R7      ,          R4      pVP                  V4      p\        P
                  ! W#V^R7      w  rVVP                  R8X  g   Q hVP                  R8X  g   Q h\        VP                  ^ ,          4       F  p\        VP                  ^,          4       F^  p\        P
                  ! W'RV3,          W7RV3,          WGRV3,          4      w  r\        WWx3,          4       \        WWx3,          4       K`  	  K  	  R# )r\   r   r;   rC   r   Ng     a@)r   r|   rO   g      l@)r   rK   rO   g     @\@)r   r   rO   )r   rO   )	rb   r   r   cosr`   rI  r  r  r2   )rh   ri   rL  rX  ri  r  rf  r  r  fijpijs   &&         rr   test_3d_inputsTestFOneWay.test_3d_inputs2  s7    JJq3
"**EEyQJJq3
"**EEyQJJq3
"**EEyQFF1I~~aAA.ww&   ww&   qwwqz"A1771:& >>!q!G*a1aj!q!G*MqtW-qtW- ' #ru   c           
        \        \        \        VR 7      ;_uu_ 4        \        P                  ! VP                  . RO4      VP                  . 4      VP                  . RO4      4      p\        VP                  VP                  VP                  4      4       \        VP                  VP                  VP                  4      4       RRR4       R#   + '       g   i     R# ; i)r^  Nr  )r   r=   r   r   )
r"   r   r   r`   rI  r^   r3   r@  rg   r?  r7  s   && rr   test_length0_1d_error!TestFOneWay.test_length0_1d_errorE  s    +3HRPP^^BJJ|$<bjjn$&JJ/?$@BFF,,bjj.@AFMM2::bff+=>	 QPPPs   B(CC"	c                   \        \        \        VR 7      ;_uu_ 4        ^pVP                  ^V34      pVP                  ^ V34      pVP                  ^V34      p\        P
                  ! W4V4      w  rgVP                  V3VP                  R7      p\        Wh4       \        Wx4       RRR4       R#   + '       g   i     R# ; i)r^  r?  N)	r"   r   r   r  r`   rI  r   rg   r3   )	rh   ri   ncolsrL  rX  ri  r  rf  r  s	   &&       rr   test_length0_2d_error!TestFOneWay.test_length0_2d_errorM  s    +3HRPPEE
#AE
#AE
#A>>!*DA77E877DA$A$ QPPPs   B	B22C	c                   VP                  ^
.4      VP                  ^.4      VP                  ^.4      VP                  ^.4      3p\        \        VR7      ;_uu_ 4        \        P                  ! V!  p\        VP                  VP                  VP                  4      4       \        VP                  VP                  VP                  4      4       RRR4       R#   + '       g   i     R# ; i)rT   r=  N)	r^   r"   r   r`   rI  r3   r@  rg   r?  )rh   ri   r  r  s   &&  rr   test_all_length_oneTestFOneWay.test_all_length_oneY  s    **bT"BJJt$4bjj"6F

TVSWHXX+33^^W-FF,,bjj.@AFMM2::bff+=> 4333s    A3CC.	zlazy->reduced input validationr  c                    V Uu. uF  q2P                  V4      NK  	  ppR p\        P                  ! \        VR7      ;_uu_ 4        \        P
                  ! V!   RRR4       R# u upi   + '       g   i     R# ; i)z$At least two samples are required...r   NrW  )rh   r  ri   re	  r  s   &&&  rr   test_too_few_inputsTestFOneWay.test_too_few_inputs`  sW     ,004C

3408]]9G44NND! 54 1444s   A%A**A;	c                    VP                  R4      pVP                  R4      p\        P                  ! \        4      ;_uu_ 4        \        P
                  ! W#^R7       RRR4       R#   + '       g   i     R# ; ir   rC   Nr  )r|   r   )r  r   r   r   r`   rI  rh   ri   rL  rX  s   &&  rr   test_axis_errorTestFOneWay.test_axis_errorh  sH    GGFOGGFO]]9%%NN1a( &%%%r  c                    VP                  R4      pVP                  R4      p\        P                  ! \        4      ;_uu_ 4        \        P
                  ! W#^R7       RRR4       R#   + '       g   i     R# ; ir  )r  r   r   r   r`   rI  r  s   &&  rr   test_bad_shapesTestFOneWay.test_bad_shapeso  sH    GGFOGGFO]]:&&NN1a( '&&&r  r   r  )SiRstvSmLs01SmLs02SmLs03AtmWtAgSmLs04SmLs05SmLs06rl  rm  rn  )r  r  r  rC  )re  rZ   r~   r\   r  ) r   r   r   r   rK  r   r   r   rY  rS  rX  r[  rc  rf  r  r   r   r   r   rg   rg  r   r  r  r  r  r  r  r  r  r   r   r   s   @rr   rF  rF    s
    , [[W&BCB DB [[W&BCG DG:8,<4 [[[	12L2L6 [[)[[-	,	)!4rvvrlC	,	,!7"&&"&&9IJ0
 4	 *
4 l+CD[[V^4+2 5 E+2Z l+CD. E.$? l+CD	% E	%? l+KL[[Vb9,%78" 9 M" l+KL) M)) )ru   rF  c                      a  ] tR tRt o R tR t]P                  P                  R. RO4      R 4       t	R t
R tR	 tR
 t]! RRR7      R 4       t]! RRR7      ]! RRR7      R 4       4       tR tR tRtV tR# )TestKruskaliv  c                    \         P                  ! ^.^.4      w  r\        VR4       \        V\         P                  P                  V^4      4       R# )r\   r   N)r`   kruskalr   r
   r  r  )rh   hrf  s   &  rr   test_array_likeTestKruskal.test_array_likex  s;    }}aS1#&Q5::==A./ru   c           	     <   VP                  ^.4      pVP                  ^.4      p\        P                  ! W#4      w  rE\        WAP                  R4      4       \	        V4      p\        WQP                  \        P
                  P                  ^^4      VR7      4       R# )r\   r   r;   N)r^   r`   r  r2   r&   r  r  )rh   ri   rk   rl   r  rf  r<   s   &&     rr   test_simpleTestKruskal.test_simple}  sn    JJsOJJsO}}Q"::c?+ $::ejjmmAq&9:GHru   r<   Nc           	     j   Vf   \        V4      M
\        W4      pVP                  . ROVR7      pVP                  . ROVR7      p\        P                  ! W44      w  rV\        WQP                  RVR7      4       \        WaP                  \        P                  P                  R^4      VR7      4       R # )Nr;   )r\   r   r|   rO   rN   )rJ   r   r   rK   rT   r  )r&   rl  r^   r`   r  r2   r  r  )rh   ri   r<   rk   rl   r  rf  s   &&&    rr   rY  TestKruskal.test_basic  s    (- $72;MJJeJ4JJ'uJ5}}Q"::d%:89::ejjmmD!&<E:JKru   c                .   ^.p^^.p\         P                  ! VP                  V4      VP                  V4      4      w  rERpRpVP                  Wg,          4      p\        WH4       \	        V\
        P                  ! VP                  R4      V4      4       R# )r\   r  r   Ng      ?)r`   r  r^   r3   r2   r   chdtrc	rh   ri   rk   rl   r  rf  h_uncorrr  r  s	   &&       rr   test_simple_tieTestKruskal.test_simple_tie  su    CF}}RZZ]BJJqM:*::ho. 	$7>>"**R.(CDru   c                ^   . ROp. ROp\         P                  ! VP                  V4      VP                  V4      4      w  rERp^\        ^4      R,          ,
          pVP                  Wg,          4      p\	        WH4       \	        V\
        P                  ! VP                  R4      V4      4       R# )r\   r   N)r\   r\   r\   rJ   )rJ   rJ   rJ   rJ   r     r`   r  r^   r  r2   r   r  r  s	   &&       rr   test_another_tieTestKruskal.test_another_tie  s}    }}RZZ]BJJqM:>5,-::::ho.$7>>"**R.(CDru   c                   . ROp. ROp^^.p\         P                  ! VP                  V4      VP                  V4      VP                  V4      4      w  rVRp^\        ^4      R,          ,
          pVP                  Wx,          4      p	\	        WY4       \	        V\
        P                  ! VP                  R4      V	4      4       R# )r\   r   Nr  rF  r=   r  r  )
rh   ri   rk   rl   rm  r  rf  r  r  r  s
   &&        rr   test_three_groupsTestKruskal.test_three_groups  s    F}}RZZ]BJJqM2::a=IG5,-::::ho.$7>>"**R.(CDru   c                   VP                  . RO4      pVP                  . RO4      pVP                  . VP                  R7      p\        P                  ! \        \
        R7      ;_uu_ 4        \        P                  ! W#V4      pRRR4       \        XP                  VP                  VP                  4      4       \        VP                  VP                  VP                  4      4       R#   + '       g   i     Lp; i)r\   r;   r   Nr  rF  )r^   r<   r   r  r   r   r`   r  r3   r@  rg   r?  )rh   ri   rk   rl   rm  r   s   &&    rr   r  TestKruskal.test_empty  s    JJy!JJy!JJrJ)\\,4IJJ--a(C Krzz"&&'9:

BJJrvv$67 KJs   ,C,,C<	r3  needs jax-ml/jax#34628r  c                |   VP                  R 4      p\        P                  ! V4      ^	,          P                  VP                  4      p\
        P                  ! W"4      p\        VP                  VP                  VP                  4      4       \        VP                  VP                  VP                  4      4       R# )r   N)r   rd   re   rf   rg   r`   r  r3   r@  r^   r?  r   s   &&  rr   r,	  %TestKruskal.test_nan_policy_propagate  ss    IIcNFF1IaLRVV$mmA!rzz"&&'9:

BJJrvv$67ru   r   cupyz,cupy returns small but nonzero res.statisticc                   VP                  R 4      p\        P                  ! V4      ^	,          P                  VP                  4      p\
        P                  ! W"RR7      p\        VP                  VP                  R4      4       \        VP                  VP                  R4      4       \        P                  ! \        RR7      ;_uu_ 4        \
        P                  ! W"RR7       RRR4       \        P                  ! \        R	R7      ;_uu_ 4        \
        P                  ! W"R
R7       RRR4       R#   + '       g   i     LW; i  + '       g   i     R# ; i)r   r   r   r   r   r   r   r   Nr   r   )r   rd   re   rf   rg   r`   r  r3   r@  r^   r?  r   r   r   r   s   &&  rr   test_nan_policy_omit_raise&TestKruskal.test_nan_policy_omit_raise  s     IIcNFF1IaLRVV$mmAV4rzz#7

BJJsO4]]:-LMMMM!73 N ]]:-KLLMM!84 ML NM MLLs   D6E	6E		E	c                R   R p\         P                  P                  R4      pVP                  VP	                  V4      4      pVP                  VP	                  V4      ^2,           4      p\
        P                  ! WE4      w  rg\        WqP                  RVP                  R7      4       R# )r
  ykr   r;   N)	r   r  r  r^   r  r`   r  r2   r<   )rh   ri   r  r  rk   rl   r  rf  s   &&      rr   test_large_samplesTestKruskal.test_large_samples  s{    ii##J/JJs**1-.JJs**1-23}}Q"::b:89ru   c                    R p\         P                  ! \        VR7      ;_uu_ 4        \        P                  ! 4        RRR4       R#   + '       g   i     R# ; i)z-Need at least two groups in stats.kruskal\(\)r   N)r   r   r   r`   r  r5  s   && rr   test_no_args_gh20661 TestKruskal.test_no_args_gh20661  s0    B]]:W55MMO 6555s   AA	r   )rj  r   N)r   r   r   r   r  r  r   r   r   rY  r  r  r  r  r   r,	  xfail_xp_backendsr  r  r  r   r   r   s   @rr   r  r  v  s     0
I [[W&BCL DLE
EE8 k*BC8 D8 k*STv&TU5 V U5: ru   r  c                     a  ] tR tRt o ]P
                  P                  R. RO4      R 4       t]P
                  P                  R. RORR3. RORR3.4      R	 4       t. ROt	]P
                  P                  R
. RO4      ]P
                  P                  R]	4      R 4       4       t
]P
                  P                  R]	4      R 4       t]P
                  P                  R]	4      ]P
                  P                  R. RO4      R 4       4       tRtV tR# )TestCombinePvaluesi  z+method, expected_statistic, expected_pvaluev!:@/?c                    . ROp\         P                  ! VP                  V4      VR7      p\        VP                  VP                  V4      4       \        VP
                  VP                  V4      4       R# r  r  Nr  r@  rF  )r`   combine_pvaluesr^   r2   r@  r?  )rh   ri   r  rp  r  rk   r   s   &&&&&  rr   test_reference_values(TestCombinePvalues.test_reference_values  sR     ##BJJqM&Arzz2D'EF

BJJ$?@ru   z,weights, expected_statistic, expected_pvaluega;?grmT?c                   VP                  . RO4      p\        P                  ! VRVP                  V4      R7      p\        VP                  VP                  V4      4       \        VP
                  VP                  V4      4       R# )r  stouffer)r  r  Nr  )r^   r`   r  r2   r@  r?  )rh   ri   r  rp  r  rk   r   s   &&&&&  rr   test_weighted_stouffer)TestCombinePvalues.test_weighted_stouffer    s^     JJ}%##Aj"**WBUVrzz2D'EF

BJJ$?@ru   ru  r  c                P   ^
^rT\         P                  P                  R4      pVR8X  dp   VP                  VP	                  VP                  V4      4      WE34      pVP                  VP                  VP                  RRV4      R4      VR,          .^R7      pMrVR8X  d0   VP                  VP                  RRV4      WT34      P                  pM<VR	8X  d6   VP                  VP	                  VP                  ^ ^WE3R
7      4      ^ R7      pTP	                  \        XP                  ^ ,          4       Uu. uF3  p\        \        P                  ! WxR3,          VR7      ^,          4      NK5  	  up4      p	VP!                  V	R,          V	RR ,
          ^ 8  4      '       g   Q hR# u upi )rT   l	   TrU(	"b-= singlerC  rE  r   r  rC   r  r  r  r  NrZ   r  )r   r  )r   r  r  r  r^   r  rb   rA  r   r  rM  r  r  r  r`   r  r  )
rh   ru  r  ri   r  r  r  pvaluessr  combined_pvaluess
   &&&&      rr   test_monotonicity$TestCombinePvalues.test_monotonicity   st    11ii##$KL hrzz#**Q-'@1&IHyy"**R[[c1-Ew"O"*5/"39: ! <Hr{{3Q'?!HJJH wwrzz#++a!+*HIPQwRH::8>>!,-'
- %''AvFqIJ-'
  vv&r*-=cr-BBaGHHHH	'
s   49F#c                    \         P                  ! VP                  . RO4      VR7      p\        VP                  V^ ,          4       \        VP
                  V^,          4       R# r  )r`   r  r^   r3   r@  r?  )rh   r  ri   r   s   &&& rr   rP  TestCombinePvalues.test_result,   sA    ##BJJ}$=fMs1v.

CF+ru   rD   Nc                   \         P                  P                  R 4      pVP                  VP                  RR7      4      pV^ 8X  d   VP                  MTp\
        P                  ! WRVR7      pVfl   VP                  VR4      p\
        P                  ! WQR7      p\        VP                  VP                  4       \        VP                  VP                  4       R# V^ 8X  d   VP                  MTpVR,          VR,          r\
        P                  ! WR7      p
\
        P                  ! WR7      p\        VP                  ^ ,          V
P                  4       \        VP                  ^,          VP                  4       \        VP                  ^ ,          V
P                  4       \        VP                  ^,          VP                  4       R# )	l   2'$FV r  )rD   r  Nr  r
  r  r  r  )r   r  r  r^   r   r`   r  rb   r2   r@  r?  )rh   r  rD   ri   r  rk   r   r   r  r  ref0rC  s   &&&&        rr   r	  TestCombinePvalues.test_axis2   s:    ii##O4JJszzwz/0AIACCA##A@<

1e$A''9CCMM3==9CJJ

3AIACCA4!D'B$$R7$$R7a($..9a($..9

1t{{3

1t{{3ru   r   ))r'  g'"x-@g({#H?)r  r  r  )r  gQ4gF?)tippettr  gL^i?)mudholkar_georgegzP@guKqN?r   r   r   )r   r   r   )r'  r  r  r  r  )r  r  r  r  )r   r   r   r   r   r   r   r  r  r  r  rP  r	  r   r   r   s   @rr   r  r    s*     [[5	GHAHA [[ 	7
)+>	?
)+=	>	@	AA	AA OG[[Y(CD[[Xw/I 0 EI0 [[Xw/, 0,
 [[Xw/[[V\24 3 04ru   r  c                   B   a  ] tR tRt o RtR tR tR tR tR t	Rt
V tR	# )
TestCdfDistanceValidationiL   z[
Test that _cdf_distance() (via wasserstein_distance()) raises ValueErrors
for bad inputs.
c           	         \        \        \        P                  ^.^.^.^^.4       \        \        \        P                  ^.^.^^ .4       R# r"
  r   r   r`   wasserstein_distancers  s   &rr   &test_distinct_value_and_weight_lengths@TestCdfDistanceValidation.test_distinct_value_and_weight_lengthsR   sG     	j%"<"<cA3aV	-j%"<"<qcA3AOru   c                    \        \        \        P                  ^ ^.^.^ ^ .4       \        \        \        P                  ^ ^.^.^^.^ .4       R# r  r  rs  s   &rr   test_zero_weight*TestCdfDistanceValidation.test_zero_weightY   sN     	j%"<"<!fqcAq6	+j%"<"<!fqcAq6A3	0ru   c           	     Z    \        \        \        P                  ^ ^.^^.^^.^R.4       R# )r~   NrZ   r  rs  s   &rr   test_negative_weights/TestCdfDistanceValidation.test_negative_weightsa   s-    j%"<"<!fq!fq!fq"g	7ru   c                    \        \        \        P                  . ^^.4       \        \        \        P                  ^.. 4       R# rS  r  rs  s   &rr   test_empty_distribution1TestCdfDistanceValidation.test_empty_distributionf   s2     	j%"<"<b1a&Ij%"<"<qc2Fru   c           	     x    \        \        \        P                  . RO^^.^\        P
                  ^.^^.4       R# r|  )r   r   r`   r  r   r   rs  s   &rr   test_inf_weight)TestCdfDistanceValidation.test_inf_weightl   s0    j%"<"<!Q!RVVQ!Q	Aru   r   N)r   r   r   r   r8  r   r  r  r	  r  r   r   r   s   @rr   r  r  L   s,     
P07
GA Aru   r  c                     a  ] tR tRt o RtR t]P                  P                  RR4      ]P                  P                  RR4      ]P                  P                  RR4      R 4       4       4       t	]P                  P                  RR4      ]P                  P                  R	R4      ]P                  P                  RR4      R
 4       4       4       t
]P                  P                  RR4      ]P                  P                  R	R4      ]P                  P                  RR4      R 4       4       4       tR t]P                  P                  RR4      ]P                  P                  R	R4      ]P                  P                  RR4      R 4       4       4       t]P                  P                  RR4      ]P                  P                  R	R4      ]P                  P                  RR 4      R 4       4       4       tR t]P                  P                  R. R!O4      ]P                  P                  R. R!O4      R 4       4       tRtV tR# )"TestWassersteinDistanceNDir   z7Tests for wasserstein_distance_nd() output values.
    c                &   . ROp. ROp\         P                  ! W4      p\        P                  ! . RO4      p\        P                  ! . RO4      p\        P                  ! WE,          4      \        P                  ! V4      ,          p\        W64       R# )r\   N)r  r  r  r  r  r  r  r  r  r  r  r   r\   r  )r  r  r~  r~  r~  r|   r\   r  r  r  r  r  r  )rO   r\   )r   r   r=   r   r   r   )r   gw@r   r  r   r   )r`   wasserstein_distance_ndr   r   r]  r
   )rh   r  r   r   flowdistr   s   &      rr   test_published_values/TestWassersteinDistanceND.test_published_valuesv   sg    
"" ++A1 xx01xx@AffT[!"&&,.!ru   n_valuer8  max_repeatsc                F   \         P                  P                  R 4      pVP                  ^W2\        R7      pVP                  W!3R7      p\         P
                  ! We^ R7      pVP                  \         P                  ! V4      4      p\         P
                  ! \         P                  ! \        V4      4      V4      p	\         P                  ! WR7      p
VP                  \        V4      4      pW{,          W,          r\        P                  ! WgW4      p\        V^ RR7       R# )l   na{F.aP	 )r  r<   r  rC   r3  rE  rF  N)r   r  r  r  r  r  r]  r   r  bincountr  r`   r  r
   )rh   r8  r  r  r  repeatsu_valuesv_values	v_weightsrange_repeat	u_weightsindexr   s   &&&&         rr   test_same_distribution_nD3TestWassersteinDistanceND.test_same_distribution_nD   s     ii##$<=,,q+3,G::G?:399XQ7JJrvvg/	yy3w<!8'BKK@	I/&oy/?)++H	UQU+ru   nunvc                   \         P                  P                  R 4      pVP                  W3R7      p\         P                  ! W#34      pVP                  VR7      pVP                  VR7      p\         P                  ! \         P
                  P                  V^R7      VR7      p	\        P                  ! WVWx4      p
\        W4       R# )   }&=:! r  rC   r3  N)
r   r  r  rX  r  r-  rW  r`   r  r
   )rh   r%  r&  r8  r  r  r  r!  r  r   r   s   &&&&       rr   test_collapse_nD*TestWassersteinDistanceND.test_collapse_nD   s     ii##$89::B::.88RJ'JJBJ'	JJBJ'	jjq99M++H	U!ru   c                (   \         P                  P                  R 4      pVP                  W3R7      pVP                  W#3R7      pVP                  VR7      pVP                  VR7      p\        P                  ! WVWx4      p	VP                  ^ V^R7      w  rVP                  W3R7      p\         P                  ! WZV^ R7      p\         P                  ! Wz\         P                  ! V4      ^ R7      p\        P                  ! WVWx4      p\        W4       R# )r(  r  rC   N)	r   r  r  r`   r  r  insertrX  r
   )rh   r%  r&  r8  r  r  r  r!  r  r   add_rownrows	add_valuer   s   &&&&          rr   test_zero_weight_nD-TestWassersteinDistanceND.test_zero_weight_nD   s    
 ii##$89::B::.::B::.JJBJ'	JJBJ'	++H	Ua!4JJUMJ2	99X	BIIi"((5/J	++H	U!ru   c                   ^^.^^..\         P                  \         P                  ) ..^^.r2p\        P                  ! WV4      p\	        V\         P                  4       \         P
                  ! RR7      ;_uu_ 4        \         P                  \         P                  ..\         P                  \         P                  ) ..r!\        P                  ! W4      p\	        V\         P                  4       RRR4       R#   + '       g   i     R# ; i)r\   r   r  N)r   r   r`   r  r   r  rg   )rh   uvvvuwdistances   &    rr   test_inf_values)TestWassersteinDistanceND.test_inf_values   s     !fq!f%"&&(9':QF00<Xrvv&[[**vvrvv&'266BFF7*;)<44R<H266* +***s   >A3C;;D	c                   \         P                  P                  R 4      pVP                  W3R7      pVP                  W#3R7      pVP                  VR7      pVP                  VR7      p\        P                  ! WVWx4      p	VP                  ^ V4      p
VP                  4       p\         P                  ! WZV^R7      p\         P                  ! WjV^R7      p\        P                  ! WVWx4      p\        W4       R# )l   Lq/ r  rC   N)r   r  r  r`   r  r  r,  r
   )rh   r%  r&  r8  r  r  r  r!  r  r   add_dimr/  r   s   &&&&         rr   test_multi_dim_nD+TestWassersteinDistanceND.test_multi_dim_nD   s    
 ii##$78::B::.::B::.JJBJ'	JJBJ'	++H	U,,q$'JJL	99X	B99X	B++H	U!ru   c                   \         P                  P                  R 4      pVP                  W3R7      pVP                  W#3R7      pVP                  VR7      pVP                  VR7      p\        P                  ! WVWx4      p	\        P
                  ! V4      p
V
P                  VR7      pVP                  VR7      p\        P                  ! W[,          V,           Wk,          V,           Wx4      p\        W4       R# )l   uAh~r  r  N)r   r  r  r`   r  ortho_groupr  r
   )rh   r%  r&  r8  r  r  r  r!  r  r   r  	transformshiftr   s   &&&&          rr   test_orthogonal_nD,TestWassersteinDistanceND.test_orthogonal_nD   s     ii##N3::B::.::B::.JJBJ'	JJBJ'	++H	U  &HH#H.	


%++H,@5,H)1)=)E)2? 	!ru   c                "   \         P                  P                  R 4      p\        P                  ! \
        RR7      ;_uu_ 4        VP                  RR7      pVP                  RR7      p\        P                  ! W#4      pRRR4       \        P                  ! \
        RR7      ;_uu_ 4        VP                  R	R7      pVP                  R
R7      p\        P                  ! W#4      pRRR4       \        P                  ! \
        RR7      ;_uu_ 4        VP                  RR7      pVP                  RR7      p\        P                  ! W#4      pRRR4       R#   + '       g   i     L; i  + '       g   i     L; i  + '       g   i     R# ; i)l   e36^[Y{lz Invalid input values. The inputsr   r  Nz Invalid input values. Dimensionsz(Invalid input values. If two-dimensional)r   rT   rw  )r   rJ   rO   )rw  )r   rw  r
  r  )r   r  r  r   r   r   r`   r  )rh   r  r  r  r  s   &    rr   test_error_code)TestWassersteinDistanceND.test_error_code   s   ii##$>?]]:-OPPzz{z3Hzzyz1H--hAA Q ]]:-OPPzzuz-Hzzwz/H--hAA Q ]]:<> >zzwz/Hzzvz.H--hAA	> > QP QP> > >s$   ;E+;E*;E=E'	*E:	=F	u_sizev_sizec                   \         P                  P                  R 4      pVP                  V^3R7      pVP                  V^3R7      pVP                  4       pVP                  4       p\        P
                  ! Wg4      p\        P                  ! WE4      p	\        P                  ! Wg4      p
\        W4       \        W4       VP                  VR7      pVP                  VR7      p\        P
                  ! WgW4      p\        P                  ! WEW4      p	\        P                  ! WgW4      p
\        W4       \        W4       R# )l   Ku@* r  N)r   r  r  r  r`   r  r  r
   )rh   rF  rG  r  r  r  u_values_flatv_values_flatd1d2d3r!  r  s   &&&          rr   test_optimization_vs_analytical9TestWassersteinDistanceND.test_optimization_vs_analytical!  s	    ii##K0::FA;:/::FA;:/ ( ( ''E**8>**=HJJFJ+	JJFJ+	''(1>**8(1>**=(1>ru   r   N)r   rw  r  )r   r   rO   r  )rK   rN   &   )rK   rV   r	  )r   r|   r  )rK   r
  r  )r\   rJ   r   )rT   rw  r  )r\   r   r|   )rO   rW   r	  )rJ   r   rO   )r\   rT   r  )r   r   r   r   r8  r  r   r   r   r#  r)  r0  r7  r;  rA  rD  rN  r   r   r   s   @rr   r  r  r   s    "& [[Y4[[VY/[[]G4, 5 0 5," [[T:.[[T;/[[VZ0" 1 0 /" [[T;/[[T;/[[VY/" 0 0 0" 	+ [[T<0[[T<0[[VY/" 0 1 1"" [[T;/[[T;/[[VY/" 0 0 0"$B  [[X{3[[X{3  4 4 ru   r  c                   N   a  ] tR tRt o RtR tR tR tR tR t	R t
R	 tR
tV tR# )TestWassersteinDistancei!!  z4Tests for wasserstein_distance() output values.
    c                F   \        \        P                  ! ^ ^.^ .^^.^.4      R4       \        \        P                  ! ^ ^.^ .^^.^.4      R4       \        \        P                  ! ^ ^.^ .^^.^.4      ^4       \        \        P                  ! . RO. RO4      ^4       R# )r~   r6   r  Nr  r  r
   r`   r  rs  s   &rr   r  #TestWassersteinDistance.test_simple%!  s     	&&1vsQFQC@	 	22FQC!Q!&	 	22FQC!Q!&	 	22y"	ru   c           	         \        \        P                  ! . RO. RO4      ^ 4       \        \        P                  ! . RO^^.. RO^^.4      ^ 4       R# r\   Nr  r  )r\   r\   r\   r   rX  )r   r`   r  rs  s   &rr   test_same_distribution.TestWassersteinDistance.test_same_distribution5!  sE     	U//	9EqI&&|aV'3aV=	ru   c           	     2   \        \        P                  ! ^ .^.4      ^4       \        \        P                  ! R.^.4      ^
4       \        \        P                  ! . RO. RO4      ^
4       \        \        P                  ! . RO. RO. RO. R	O4      R4       R# )
r~   ry   Nr  r  )rU   rV   rW   rR   rw  )rF   g@r
  )r
  rO   r  )r   r\   r\   )r\   r   r\   rT  rs  s   &rr   
test_shift"TestWassersteinDistance.test_shift>!  sz     	22A3<a@22B4!=rB&&8LM	 	&&'0)=	ru   c           
         \        \        P                  ! . RO. RO. RO. RO4      \        P                  ! . RO. RO. RO. RO4      4       R# r~   N)r~   r~   r\   r\   r\   r\   r|   )r~   r   r   r   r   r   r   )r\   r\   r\   r\   r\   r\   r\   )r|   r~   r\   )r~   r   r   r\   rJ   r   rT  rs  s   &rr   test_combine_weights,TestWassersteinDistance.test_combine_weightsK!  s?     	&&%'<%'<> &&y)'0)=		>ru   c                   \         P                  ! R^R4      p\         P                  ! V4      p\        \        P
                  ! W4      \         P                  ! \         P                  ! V4      4      4       \         P                  ! \        V4      4      pVRRR1,          p\        \        P
                  ! WW44      \         P                  ! \         P                  ! V4      VR7      4       R# )rT   rF  Nr3  rK  rZ   )
r   r   r  r
   r`   r  rc   r3	  r  r  )rh   r  r   r!  r  s   &    rr   test_collapse%TestWassersteinDistance.test_collapseU!  s     IIc2s#MM!&&q,GGBFF1I	  IIc!f%	ddO	&&qYBJJrvvay)4	6ru   c           
         \        \        P                  ! . RO^^.. RO^^.4      \        P                  ! ^^.^^.^^.^^.4      4       R# r\   N)r\   rJ   ri  r	  rT  rs  s   &rr   r  (TestWassersteinDistance.test_zero_weighte!  sN    &&~1v'01a&:&&1v1v1v1vF	Hru   c           	        \        \        P                  ! ^^\        P                  .^^.4      \        P                  4       \        \        P                  ! ^^\        P                  .\        P                  ) ^.4      \        P                  4       \        \        P                  ! ^\        P                  ) \        P                  .^^.4      \        P                  4       \
        P                  ! \        RR7      ;_uu_ 4        \        \        P                  ! ^^\        P                  .\        P                  ^.4      \        P                  4       RRR4       R#   + '       g   i     R# ; ir\   zinvalid valuer   N)	r   r`   r  r   r   r   r  r   rg   rs  s   &rr   r7  'TestWassersteinDistance.test_inf_valuesl!  s     	&&1bff~1v>FF	 	&&1bff~|DFF	 	&&BFF7BFF';aVDFF	 \\.@@**Aq"&&>BFFA;G A@@@   	AE$$E5	r   N)r   r   r   r   r8  r  rX  r[  r`  rc  r  r7  r   r   r   s   @rr   rR  rR  !!  s3      >6 H ru   rR  c                   H   a  ] tR tRt o RtR tR tR tR tR t	R t
R	tV tR
# )TestEnergyDistancei~!  z/Tests for energy_distance() output values.
    c                   \        \        P                  ! ^ ^.^ .^^.^.4      \        P                  ! ^4      R,          4       \        \        P                  ! ^ ^.^ .^^.^.4      \        P                  ! ^4      R,          4       \        \        P                  ! ^ ^.^ .^^.^.4      R4       \        \        P                  ! . RO. RO4      \        P                  ! ^4      R,          4       R# )r~   r6   r  r   Nr  r  r  r   r`   energy_distancer   rI  rs  s   &rr   r  TestEnergyDistance.test_simple!  s     	!!1a&1#1vs;GGAJO	 	E11FQC!Q!&GGAJ	 	E11FQC!Q!&	 	!!)Y7GGAJ**	,ru   c           	         \        \        P                  ! . RO. RO4      ^ 4       \        \        P                  ! . RO^^.. RO^^.4      ^ 4       R# rW  )r   r`   rp  rs  s   &rr   rX  )TestEnergyDistance.test_same_distribution!  sB     	U**9i@!D!!,Aq!fM	ru   c                    \        \        P                  ! ^ .^.4      \        P                  ! ^4      4       \        \        P                  ! R.^.4      \        P                  ! ^4      R,          4       R# )r~   Nr  gS[:XL	@ro  rs  s   &rr   r[  TestEnergyDistance.test_shift!  sS     	E111#s;RWWQZH!!2$,GGAJ	!ru   c           
         \        \        P                  ! . RO. RO. RO. RO4      \        P                  ! . RO. RO. RO. RO4      4       R# r^  r   r`   rp  rs  s   &rr   r`  'TestEnergyDistance.test_combine_weights!  s<     	!!"79N"79NP!!)Y	9M	Oru   c           
         \        \        P                  ! . RO^^.. RO^^.4      \        P                  ! ^^.^^.^^.^^.4      4       R# rf  rw  rs  s   &rr   r  #TestEnergyDistance.test_zero_weight!  sK    !!.1a&)aVL!!1a&1a&1a&1a&A	Cru   c           	        \        \        P                  ! ^^\        P                  .^^.4      \        P                  4       \        \        P                  ! ^^\        P                  .\        P                  ) ^.4      \        P                  4       \        \        P                  ! ^\        P                  ) \        P                  .^^.4      \        P                  4       \
        P                  ! \        RR7      ;_uu_ 4        \        \        P                  ! ^^\        P                  .\        P                  ^.4      \        P                  4       RRR4       R#   + '       g   i     R# ; iri  )	r   r`   rp  r   r   r   r  r   rg   rs  s   &rr   r7  "TestEnergyDistance.test_inf_values!  s     	U**Aq"&&>Aq6BBFFK!!1a.BFF7A,?FF	 	!!1rvvgrvv"6A?FF	 \\.@@%%q!RVVnrvvqkB A@@@rk  r   N)r   r   r   r   r8  r  rX  r[  r`  r  r7  r   r   r   s   @rr   rm  rm  ~!  s/     , !OC ru   rm  c            	       \  a  ] tR tRt o . ROt. ROtR tR tR tR t	R t
R t]P                  P                  RR	. /R
. /R
. R	. /.4      R 4       t]! RRR7      R 4       t]! RRR7      R 4       t]! RRR7      R 4       t]P                  P                  RRR.4      R 4       tR tRtV tR# )TestBrunnerMunzeli!  c                   VP                  V P                  4      VP                  V P                  4      r2\        P                  ! W#R R7      w  rE\        P                  ! W2RR7      w  rg\        P                  ! W#RR7      w  r\        P                  ! W2R R7      w  r\        WW4       \        W4       WY8w  g   Q h\        WAP                  R4      4       \        WaP                  R4      4       \        WP                  R4      4       \        WP                  R4      4       \        WQP                  R4      4       \        WP                  R4      4       R# )r  r  r  |	&	@gҏGg?g@-pL?N|	&	r^   r_   Yr`   brunnermunzelr2   )rh   ri   r_   r  u1r  u2r8
  u3r9
  u4p4s   &&          rr   test_brunnermunzel_one_sided.TestBrunnerMunzel.test_brunnermunzel_one_sided!  s    zz$&&!2::dff#51$$Qv>$$QyA$$QyA$$Qv>xxJJ'9:;JJ':;<JJ'9:;JJ':;<JJ'<=>JJ':;<ru   c                   VP                  V P                  4      VP                  V P                  4      r2\        P                  ! W#R R7      w  rE\        P                  ! W2R R7      w  rg\        WQP                  V4      4       \        WAP                  R4      4       \        WaP                  R4      4       \        WQP                  R4      4       R# )r  r  r   ҏGw?Nr  r  rh   ri   r_   r  r  r  r  r8
  s   &&      rr   test_brunnermunzel_two_sided.TestBrunnerMunzel.test_brunnermunzel_two_sided!  s    zz$&&!2::dff#51$$Q{C$$Q{CJJrN+JJ'9:;JJ':;<JJ'<=>ru   c                   VP                  V P                  4      VP                  V P                  4      r2\        P                  ! W#4      w  rE\        P                  ! W24      w  rg\        WW4       \        WAP                  R 4      4       \        WaP                  R4      4       \        WQP                  R4      4       R# )r  r  Nr  r  r  s   &&      rr   test_brunnermunzel_default,TestBrunnerMunzel.test_brunnermunzel_default!  s    zz$&&!2::dff#51$$Q*$$Q*JJ'9:;JJ':;<JJ'<=>ru   c           	     @   R pRpRpVR9  g   Q hRp\         P                  ! \        VR7      ;_uu_ 4        \        P                  ! VP                  V P                  4      VP                  V P                  4      W#V4       RRR4       R#   + '       g   i     R# ; i)rE  re  r  rr  r   N)r  r  r  r   r   r   r`   r  r^   r_   r  rh   ri   r  distributionr   r  s   &&    rr   $test_brunnermunzel_alternative_error6TestBrunnerMunzel.test_brunnermunzel_alternative_error!  sw     
"BBBBL]]:W55

466 2BJJtvv4F +:G 6555   ABB	c                   VP                  V P                  4      VP                  V P                  4      r2\        P                  ! W#R R7      w  rE\        P                  ! W2R R7      w  rg\        WQP                  V4      4       \        WAP                  R4      4       \        WaP                  R4      4       \        WQP                  R4      4       R# )r  r  r  g "H[?Nr  r  r  s   &&      rr   $test_brunnermunzel_distribution_norm6TestBrunnerMunzel.test_brunnermunzel_distribution_norm!  s    zz$&&!2::dff#51$$QA$$QAJJrN+JJ'9:;JJ':;<JJ'<=>ru   c           	     @   R pRpRpVR9  g   Q hRp\         P                  ! \        VR7      ;_uu_ 4        \        P                  ! VP                  V P                  4      VP                  V P                  4      W#V4       RRR4       R#   + '       g   i     R# ; i)r  rE  r  z&distribution should be 't' or 'normal'r   N)re  r  r  r  s   &&    rr   %test_brunnermunzel_distribution_error7TestBrunnerMunzel.test_brunnermunzel_distribution_error"  sv    ! 
?222:]]:W55

466 2BJJtvv4F +:G 6555r  kwarg_updaterl   rk   c                   R V P                   RV P                  /pVP                  V4       VP                  4        UUu/ uF!  w  rEWBP	                  V\        V4      R7      bK#  	  ppp\        \        \        VR7      ;_uu_ 4        \        P                  ! R/ VB w  rgRRR4       \        XVP	                  VP                  4      4       \        XVP	                  VP                  4      4       R# u uppi   + '       g   i     Lb; i)rk   rl   r;   r^  Nr   )r_   r  updateitemsr^   r&   r"   r   r   r`   r  r3   rg   )rh   r  ri   r  keyvalr@  r?  s   &&&     rr   test_brunnermunzel_empty_imput0TestBrunnerMunzel.test_brunnermunzel_empty_imput"  s     tvvsDFF+l#"(,,.2"0hc jj,<R,@jAA"0 	 2+3HRPP % 3 3 =f =I Q	2::bff#56

266 232PPs   'C4C::D
	r3  r  r  c                   VP                  ^^^^^^^^^^^^^^VP                  .4      pVP                  . RO4      p\        P                  ! W#RR7      w  rE\        P                  ! W2RR7      w  rg\	        WAP                  VP                  4      4       \	        WQP                  VP                  4      4       \	        WaP                  VP                  4      4       \	        WqP                  VP                  4      4       R# )r\   r  r   Nr   r   r   r   r\   rJ   r   r\   r\   r|   r   )r^   rg   r`   r  r3   r  s   &&      rr   &test_brunnermunzel_nan_input_propagate8TestBrunnerMunzel.test_brunnermunzel_nan_input_propagate"  s    JJ1aAq!Q1aAq"&&IJJJ9:$$QkB$$QkBJJrvv./JJrvv./JJrvv./JJrvv./ru   r   c                   VP                  ^^^^^^^^^^^^^^VP                  .4      pVP                  . RO4      pRpRpRpRp\        P                  ! \        VR7      ;_uu_ 4        \
        P                  ! W#WEV4       RRR4       \        P                  ! \        VR7      ;_uu_ 4        \
        P                  ! W2WEV4       RRR4       R#   + '       g   i     LW; i  + '       g   i     R# ; i)r\   r  re  r   r   r   Nr  )r^   rg   r   r   r   r`   r  )rh   ri   r_   r  r  r  r   r  s   &&      rr   "test_brunnermunzel_nan_input_raise4TestBrunnerMunzel.test_brunnermunzel_nan_input_raise$"  s    JJ1aAq!Q1aAq"&&IJJJ9:!
1]]:W55kL 6]]:W55kL 65 65555s   *C/C%C"	%C6	c                   VP                  ^^^^^^^^^^^^^^\        P                  .4      pVP                  . RO4      p\        P                  ! W#RR7      w  rE\        P                  ! W2RR7      w  rg\        WW4       \        WAP                  R4      4       \        WaP                  R4      4       \        WQP                  R4      4       R# )r\   r   r   r  r  Nr  r  )r^   r   rg   r`   r  r2   r  s   &&      rr   !test_brunnermunzel_nan_input_omit3TestBrunnerMunzel.test_brunnermunzel_nan_input_omit2"  s    JJ1aAq!Q1aAq"&&IJJJ9:$$Qf=$$Qf=JJ'9:;JJ':;<JJ'<=>ru   vectorized_callFTc                r   VP                  . RO4      pVP                  . RO4      pV'       d9   VP                  W334      P                  pVP                  WD34      P                  pRp\        \        WRR7      ;_uu_ 4        \
        P                  ! W4RR7       RRR4       R#   + '       g   i     R# ; i)zktests that a warning is emitted when p is nan
p-value with t-distributions can be nan (0/0) (see gh-15843)
zDp-value cannot be estimated|divide by zero|invalid value encounteredr^  re  r  Nr  r=   r   r   r   r   )r^   r  r   r"   r   r`   r  )rh   r  ri   rk   rl   r   s   &&&   rr   test_brunnermunzel_return_nan/TestBrunnerMunzel.test_brunnermunzel_return_nan>"  s    
 JJ|$JJ+,! ""A! ""ATs::37 ;:::s   B%%B6	c                (   VP                  . RO4      pVP                  . RO4      p\        \        RVR7      ;_uu_ 4        \        P                  ! W#RR7      w  rERRR4       \        XVP                  R4      4       R#   + '       g   i     L-; i)	z^tests that a p is 0 for datasets that cause p->nan
when t-distribution is used (see gh-15843)
zdivide by zeror^  r  r  Nr   r  r  )r^   r"   r   r`   r  r3   )rh   ri   rk   rl   r  rf  s   &&    rr   test_brunnermunzel_normal_dist0TestBrunnerMunzel.test_brunnermunzel_normal_distN"  sg     JJ|$JJ+,/?BGG&&q(CDA H2::b>* HGs   BB	r   N)r\   rJ   r\   r\   r\   r\   r\   r\   r\   r\   rJ   r   r\   r\   )r   r   r   r   r\   rJ   r   r\   r\   r|   r   )r   r   r   r   r_   r  r  r  r  r  r  r  r   r   r   r  r   r  r  r  r  r  r   r   r   s   @rr   r~  r~  !  s     	3A)A=&	?	?G?G [[^sBi#r/2BR.@.B C4C4 k*BC	0 D	0 k*STM UM k*ST	? U	? [[.>8 ?8	+ 	+ru   r~  c                     \         P                  ! WW#R 7      pVP                  VP                  .VP	                  4       O5# )r  rf  r  )r`   quantile_testr@  r?  r  )rk   r  rf  r  r   s   &&&& rr   quantile_test_referencer  Z"  s5    


a
CC==#**@s'>'>'@@@ru   c                   Z  a  ] tR tRt o RtR t]P                  P                  R. R%O. R&ORR]	P                  ) RR	.R
RR]	P                  R..4      R 4       t]P                  P                  R. R'O. R(O.4      R 4       t]P                  P                  RRR.4      ]P                  P                  RR	R.4      ]P                  P                  RRR.4      R 4       4       4       t]P                  P                  R. R)O4      R 4       t]! RRR7      ]P                  P                  R. R*O4      ]P                  P                  R^ ^.4      R 4       4       4       tR tR tR  t]P                  P)                  R!4      ]P                  P                  R". R+O4      R# 4       4       tR$tV tR# ),TestQuantileTesti`"  zYTest the non-parametric quantile test,
including the computation of confidence intervals
c           	        VP                  . RO4      pRp\        P                  ! \        VR7      ;_uu_ 4        \        P
                  ! VP                  W!P                  4      4       RRR4       Rp\        P                  ! \        VR7      ;_uu_ 4        \        P
                  ! VRR7       RRR4       Rp\        P                  ! \        VR7      ;_uu_ 4        \        P
                  ! V^R7       RRR4       VP                  RR
R	RVP                  .4      p\        P
                  ! W$R7      p\        VP                  VP                  WAP                  4      4       \        VP                  VP                  WAP                  4      4       \        VP                  VP                  WAP                  4      4       Rp\        P                  ! \        VR7      ;_uu_ 4        \        P
                  ! VRR7       RRR4       Rp\        P                  ! \        VR7      ;_uu_ 4        \        P
                  ! V^R7       RRR4       Rp\        P                  ! \        VR7      ;_uu_ 4        \        P
                  ! V^
R7       RRR4       Rp\        P                  ! \        VR7      ;_uu_ 4        \        P
                  ! W!P                  RR.4      RR7       RRR4       Rp\        P                  ! \        VR7      ;_uu_ 4        \        P
                  ! VRR7       RRR4       Rp\        P                  ! \        VR7      ;_uu_ 4        \        P
                  ! V4      P                  ^4       RRR4       R#   + '       g   i     EL-; i  + '       g   i     EL; i  + '       g   i     EL; i  + '       g   i     EL; i  + '       g   i     EL; i  + '       g   i     ELj; i  + '       g   i     EL&; i  + '       g   i     L; i  + '       g   i     R# ; i)r\   z `x` must be an array of numbers.r   Nz)`q` must be a scalar or array of numbers.F)r  z(`p` must be a scalar or array of floats.)rf  r   r   r7   z"`axis` must be an integer or None.ry   rC   z7`axis` is not compatible with the shapes of the inputs.z/If specified, `keepdims` must be True or False.r[  z"`keepdims` may be False only if...rC  r@  )rf  r"  z`alternative` must be one of...z	one-sidedr  r  r  rQ  )r^   r   r   r   r`   r  r  r  rg   r3   r@  r  statistic_typer?  r  )rh   ri   rk   r  rf  r   s   &&    rr   test_quantile_test_iv&TestQuantileTest.test_quantile_test_ivf"  s   JJy!4]]:W55		!WW 56 6 >]]:W55U+ 6 =]]:W55Q' 6 JJRS"&&12!!!)r||Avv'>?**BLLFF,CD

BLLFF$;<6]]:W55, 6 L]]:W55* 6 D]]:W55B/ 6 8]]:W55ZZc
%;eL 6 4]]:W55{; 6 I]]:W55"66q9 65O 655 655 655 655 655 655 655 65 655sk   0NN"N6*O
1O8O2?*PP&P-N	"N3	6O	
O	O/	2P	P	P*	-P>	zp, alpha, lb, ub, alternativeffffff?rE  r  g{gA?r  r  gg @r  c                0   VP                  VP                  R RR4      4      p\        P                  ! WqVR7      pVP	                  V4      p	\        V	P                  VP                  V4      4       \        V	P                  VP                  V4      4       R# )r   r  r  )rf  r  N)	r  r   r`   r  r  r2   r  r^   r  )
rh   rf  r  lbubr  ri   rk   r   r  s
   &&&&&&&   rr   test_R_ci_quantile#TestQuantileTest.test_R_ci_quantile"  sm      FF299S$-.!!!kB$$U+

2/B0ru   zq, p, alternative, refc                    VP                  VP                  R RR4      4      p\        P                  ! WaW#R7      p\	        VP
                  VP                  V4      4       R# )r   r  r  r  N)r  r   r`   r  r2   r?  r^   )rh   r  rf  r  r   ri   rk   r   s   &&&&&&  rr   test_R_pvalueTestQuantileTest.test_R_pvalue"  sG     FF299S$-.!!!AG

BJJsO4ru   r  
continuousdiscreter  r  c                
   \        ^\        V4      ,          \        V4      ,           V,          4      p\        P                  P	                  V4      pVR8X  d(   VP                  ^R7      w  rxVP                  ^dR7      p	M6VP                  ^^^dR7      p	VP                  4       pVP                  ^^4      pVP                  V	P                  4       4      \        V4      \        V4      rp	\        P                  ! WWrR7      p
V
P                  VR7      pVR8X  d4   VP                  WP                  P                  4      VP                  8*  pM2VP                  WP                  P                  4      VP                  8  pW,           F5  p\        P                  ! WWrR7      p
V
P                   ^V,
          8  d   K5  Q h	  W( ,           F5  p\        P                  ! WWrR7      p
V
P                   ^V,
          8  d   K5  Q h	  R# )rO   r  r  r  r|  r  N)r  r  r   r  r  r  r^   r   r  r`   r  r  r  r  r<   r  r?  )rh   r  r  r  ri   r  r  rf  r  r  r   r  i_insiderk   s   &&&&&         rr   test_pval_ci_match#TestQuantileTest.test_pval_ci_match"  s    As4yL3{#33U:;ii##D)<::1:%DA**#*&C,,q"3,/C

AQ#AJJszz|,eAha!!#aI$$e$< & yyggmm4?Hyyffll3rvv=HA%%c!MC::E	)))  YA%%c!MC::E	)))  ru   r<   Nc                &   Vf   \        V4      M
\        W!4      pVP                  . ROVR7      pRp\        P                  ! V^RRR7      p\        VP                  VP                  WAR7      RR7       VP                  ^;.^,          ^=.^h,          ,           VR7      p\        \        P                  ! R^pR	7      P                  ^R
7      4      p\        P                  ! V^<RRR7      p\        VP                  VP                  WAR7      RR7       R # )Nr;   geX?r  r  r  ro  r  r6   )rf  r  r  r  r  rF  )                                    rL        )
r&   rl  r^   r`   r  r2   r?  r  binomr^  )rh   r<   ri   rk   
pvalue_refr   s   &&&   rr   test_match_conover_examples,TestQuantileTest.test_match_conover_examples"  s     ). $72;MJJ ;BG  I
!!!sdL

BJJzJ$GdS JJtAve,EJ:5;;48818=>
!!!rSiH

BJJzJ$GeTru   r	  zindexing in _apply_over_batchr  rD   c                  a \         P                  P                  R 4      pVP                  RR7      pVP                  RR7      pWdP                  R	R7      R,          ,           pV3R lWVV3 4       w  rVp\        WWWaR7      pVw  rrV^ 8X  d$   VP                  VP                  VP                  rvp\
        P                  ! WWWaVR7      pVP                  VP                  rVP                  4       w  ppV^ 8X  d0   VP                  VP                  rVP                  VP                  pp\        W4       \        W4       \        VV4       \        VV4       R# )
l   z-\"}	r  r  c              3   b   <"   T F$  pSP                  VP                  4       4      x  K&  	  R # 5ir#  r^   r   rd	  r1  ri   s   & rr   rf	  9TestQuantileTest.test_multidimensional.<locals>.<genexpr>#  #     Ay2::cjjl++y   ,/r  )r  rf  r  rD   NrJ   r  r  r  )r   r  r  r  r   r`   r  r@  r?  r  r2   )rh   r  rD   ri   r  rk   rf  r  r   rr  r_  ref_lowref_highr   r  r  res_lowres_highs   &&&f              rr   test_multidimensional&TestQuantileTest.test_multidimensional#  s    ii##$56JJGJ$JJGJ$


',,AayAa%aAG7:4719cc133!A!!!ATR$'MM3::z33519(5: '		8::XG5
/))ru   c                  a \         P                  P                  R 4      pRpRpVP                  V4      pVP                  V4      pWbP                  V4      R,          ,           pV3R lWWV3 4       w  rWpSP                  ^ .VO54      pTp	\        P
                  ! WXVRR7      p
V
P                  4       p\        V
P                  SP                  V	4      4       \        V
P                  SP                  V	4      4       \        V
P                  V	4       \        VP                  V	4       \        VP                  V	4       SP                  ^ .VO54      pSP                  ^ .VO54      p	\        P
                  ! WVRR7      p
V
P                  4       p\        V
P                  SP                  V	4      4       \        V
P                  SP                  V	4      4       \        V
P                  V	4       \        VP                  V	4       \        VP                  V	4       SP                  VP                  ^ ,          ^ 34      pSP                  VSP                   4      p	\        P
                  ! WVRR7      p
V
P                  4       p\        V
P                  V	4       \        V
P                  V	4       \        V
P                  V	4       \        VP                  V	4       \        VP                  V	4       R# )   g3Dn&br  c              3   b   <"   T F$  pSP                  VP                  4       4      x  K&  	  R # 5ir#  r  r  s   & rr   rf	  2TestQuantileTest.test_zero_size.<locals>.<genexpr>'#  r  r  r  rf  rD   NrJ   r  r
  rZ   )r   r  r  r  r`   r  r  r3   r@  r^   r  r?  r  r  r  r   rg   )rh   ri   r  x_shapeqp_shaperk   r  rf  qp_zerooutr   r  x_zeros   &f           rr   r  TestQuantileTest.test_zero_size#  s9   ii##$67JJwJJx 

8$t++AayAa ((A>>*!!!'C$$&rzz#7**BJJsO<

C($% 1-w-(hh~H~&!!&<$$&rzz#7**BJJsO<

C($% 1771:q/*ggh'!!&<$$&s+**C0

C($%ru   c                  a \         P                  P                  R 4      pVP                  R4      pVP                  R4      pWBP                  R4      R,          ,           pV3R lW5V3 4       w  r5p\        P                  ! W4VRR7      pVP                  4       w  rx\        V4      p	SP                  VP                  VP                  4      R8  4      p
\        P                  ! V	4      V
,          P                  SP                  4      p	SP                  V
RR7      pSP                  V4      '       d   Q h\        P                  ! WVRR7      pVP                  4       w  rSP                  SP                  VP                   V,          4      4      '       g   Q hSP                  SP                  VP"                  V,          4      4      '       g   Q hSP                  SP                  VP$                  V,          4      4      '       g   Q hSP                  SP                  W,          4      4      '       g   Q hSP                  SP                  W,          4      4      '       g   Q h\'        VP                   V( ,          VP                   V( ,          4       \'        VP"                  V( ,          VP"                  V( ,          4       \'        VP$                  V( ,          VP$                  V( ,          4       \'        W( ,          W{( ,          4       \'        W( ,          W( ,          4       SP                  VP                  VP                  4      R8  4      pSP                  VP                  VP                  4      R8  4      pSP                  V4      '       g   Q hSP                  V4      '       g   Q h\        P                  ! V4      SP)                  WP                  4      ,          P                  SP                  4      p\        P                  ! V4      SP)                  VVP                  4      ,          P                  SP                  4      pSP                  VV,          4      pSP                  V4      '       d   Q h\        P                  ! W4VRR7      pVP                  4       w  rSP                  SP                  VP                   V,          4      4      '       g   Q hSP                  SP                  VP"                  V,          4      4      '       g   Q hSP                  SP                  VP$                  V,          4      4      '       g   Q hSP                  SP                  W,          4      4      '       g   Q hSP                  SP                  W,          4      4      '       g   Q h\'        VP                   V( ,          VP                   V( ,          4       \'        VP"                  V( ,          VP"                  V( ,          4       \'        VP$                  V( ,          VP$                  V( ,          4       \'        W( ,          W{( ,          4       \'        W( ,          W( ,          4       R# )	l   I8Z R r  c              3   b   <"   T F$  pSP                  VP                  4       4      x  K&  	  R # 5ir#  r  r  s   & rr   rf	  -TestQuantileTest.test_nans.<locals>.<genexpr>R#  r  r  r  rC   Nr  )rT   rT   rZ   )r   r  r  r`   r  r  r/   r^   r  rd   re   rf   rg   r  r  rx  r@  r  r?  r3   r  )rh   ri   r  rk   r  rf  r   r  r  x_i_nan	i_nan_outr   r  r  i_nan_qi_nan_ps   &f               rr   	test_nansTestQuantileTest.test_nansM#  s   ii##$45JJx JJx 

8$t++AayAa!!!AB7335QZ

3::agg.56VVBZ""266*FF5rF*	66)$$$$!!"QR8335vvbhhs}}Y789999vvbhhs11)<=>>>>vvbhhszz)456666vvbhhw123333vvbhhx234444yj13==)3LM**I:68J8JI:8VW

I:.

I:0FG
+WZ-@A,hz.BC**SZZ0478**SZZ0478vvgvvgFF1Iboogww78<<RVVDFF1Iboogqww78<<RVVDJJw01	66)$$$$!!!AB7335vvbhhs}}Y789999vvbhhs11)<=>>>>vvbhhszz)456666vvbhhw123333vvbhhx234444yj13==)3LM**I:68J8JI:8VW

I:.

I:0FG
+WZ-@A,hz.BCru   c                ,  a \         P                  P                  R 4      pRpRpVP                  V4      pVP                  V4      pWbP                  V4      R,          ,           pV3R lWWV3 4       w  rWp\        P                  ! WVVRR7      pVP                  4       w  r\        P                  ! \        V4      \        V4      \        V4      RR7      pVP                  4       w  r\        VP                  VP                  SP                  R3,          4       \        VP                  VP                  SP                  R3,          4       \        VP                  VP                  SP                  R3,          4       \        WSP                  R3,          4       \        WSP                  R3,          4       \        P                  ! VRRRR7      pVP                  4       w  r\        P                  ! \        V4      RRRR7      pVP                  4       w  r\        VP                  VP                  4       \        VP                  VP                  4       \        VP                  VP                  4       \        W4       \        W4       R# )	r  r  c              3   b   <"   T F$  pSP                  VP                  4       4      x  K&  	  R # 5ir#  r  r  s   & rr   rf	  2TestQuantileTest.test_axis_none.<locals>.<genexpr>#  r  r  Nr  r   r6   r  r
  )r   r  r  r`   r  r  r*   r3   r@  r  r  r?  )rh   ri   r  r  r  rk   r  rf  r   r  r  r   r  r  s   &f            rr   test_axis_noneTestQuantileTest.test_axis_none#  s   ii##$67JJwJJx 

8$t++AayAa!!!AD9335!!(1+!RVW335s}}RZZ]'CD**C,>,>rzz1},MN

CJJrzz1}$=>Q!782::q="9:!!!sc=335!!(1+$G335s}}5**C,>,>?

CJJ/)+ru   zignore:torch:UserWarningz,x_shape, qp_shape, axis, keepdims, res_shapec                .   \         P                  P                  R 4      pVP                  VP                  V4      4      VP                  VP                  V4      4      r\        P
                  ! WWVR7      p
V
P                  P                  V8X  g   Q hR# )r  )r  rf  rD   r"  N)r   r  r  r^   r`   r  r@  r  )rh   r  r  rD   r"  	res_shaperi   r  rk   qpr   s   &&&&&&&    rr   r  TestQuantileTest.test_keepdims#  sp     ii##$67

3::g./CJJx<P1Q2!!!RXN}}""i///ru   r   )rF  r  gX݊?g蜶?r  )r6   rE  g-tA?gP"?r  )r  rF  r  gͯI	?)r  r6   r  gɍQ%e?r  r  ))r  r  NFr   )r  r  NTr  )r  r  rZ   Fr  )r  r  rZ   Tr  )r   r   r   r   r8  r  r   r   r   r   r   r  r  r  r  r   r  r  r
  r  r   r  r   r   r   s   @rr   r  r  `"  s    ,:\ [['	G	F
rvvg/	8
s%rvvy	9	;11 [[ 	2	1	3
5
5 [[VlJ%?@[[]VY,?@[[WsDk2%* 3 A A%*N [[W&BCU DU> (1PQ[[],LM[[VaV,* - N R*2,&\2Dh,: [[ :;[[K N 0 <0ru   r  c                      a  ] tR tRt o R t]P                  P                  R4      t]P                  R(4      t	]P                  P                  R4      t]P                  R)4      t
RRRR]	3RR	RR]
3R
RRR. R*O. R+O. R,O.3RRRR. R-O. R.O. R/O.3RRRR. R0O. R0O. R0O. R0O. R0O. R0O. R0O. R0O. R1O. R2O. R2O. R2O. R2O. R2O.3^RRR. R3O. R3O. R3O. R3O. R3O. R3O. R3O. R3O. R3O. R3O. R3O. R3O. R3O. R4O. R5O. R5O. R5O. R5O. R5O. R5O. R5O.3^RRR. R6O. R6O. R7O.3RRRR. R8O. R8O. R8O. R8O. R8O. R8O. R9O. R:O. R:O. R:O.
3^RRR. R3O. R3O. R3O. R3O. R3O. R3O. R3O. R3O. R3O. R3O. R3O. R3O. R4O. R5O. R5O. R5O. R5O. R5O. R5O.3RRRR. R;O. R;O. R<O. R=O.3^uRRR. R3O. R3O. R3O. R3O. R3O. R3O. R3O. R3O. R3O. R4O. R5O.3.t]P                  P                  R]4      R 4       tRRRR. R>O. R?O.3RRRR. R>O. R>O. R>O. R@O. RAO.3R R!RR. RBO. RBO. RBO. RBO. RBO. RBO. RBO. RBO. RBO. RCO. RDO. RDO. RDO. RDO. RDO.3.t]P                  P                  R]4      ]P                  P#                  4       R" 4       4       tR# tR$ tR% tR&tV tR'# )ETestPageTrendTesti#  c                N    \         P                  P                  R 4      V n        R# )r  Nrr  rs  s   &rr   rt  TestPageTrendTest.setup_method#  s    99((4ru   l   /*s i2  g<n?Fr  iu  gRn@?i,0  gY@?i
  gXY?r  iL  gs˜?TgĞ?g>K=?i  gqူ?ga?i  g;?g[^6?zL, p, ranked, method, datac                    \         P                  ! WSVR 7      p\        WP                  4       \	        W&P
                  4       \        WFP                  4       R# )rankedr  Nr`   page_trend_testr   r@  r
   r?  r  rh   Lrf  r  r  rG  r   s   &&&&&& rr   test_accuracyTestPageTrendTest.test_accuracy#  s8    ##DGQ&::&VZZ(ru   i  g-lpW?i*  gh[?i  g&]2?c                    \         P                  ! WSVR 7      p\        WP                  4       \	        W&P
                  4       \        WFP                  4       R# r  r  r  s   &&&&&& rr   test_accuracy2 TestPageTrendTest.test_accuracy2
$  s:     ##DGQ&::&VZZ(ru   c                ,   \         P                  P                  R 4      p^
^r2\         P                  ! ^V^,           4      pVP	                  \         P                  ! V4      4      pVP                  W#34      p\
        P                  ! V^R7      p\
        P                  ! V4      p\
        P                  ! VRR7      p	\
        P                  ! VRR7      p
\
        P                  ! WtR7      p\
        P                  ! VRV3,          WE,          R7      p\        VP                  V	P                  4       \        VP                  V
P                  4       \        VP                  VP                  4       \        VP                  VP                  4       R# )i7
rC   Tr  F)predicted_ranksr   N)
r   r  r  r   r  r`   rankdatar  r   r@  )rh   r  r  r  r'  permrG  ranksr  r  r  res4res5s   &            rr   test_optionsTestPageTrendTest.test_options$  s   ii##I.21))Aqs+ryy|,zz1&!t!,$$U+$$U48$$T%8$$UL$$U1d7^5D5JLT^^T^^4T^^T^^4T^^T^^4T^^T^^4ru   c                   . R
O. RO. RO. RO. RO. RO.p\         P                  ! V4      P                  p\         P                  ! ^^4      p\        P
                  ! VRVRR7      p\        VP                  R4       \        VP                  R^R7       \        P
                  ! VRVRR7      p\        VP                  R4       \        VP                  R^R7       R	# )r  Fr  )r  r'  r  i  gy&1l?r^
  r  gHPsb?N)r  u   o   )rX  r\  k   )g      y   )rP        )         )   r  t   )
r   r   r   r   r`   r  r   r@  r   r?  )rh   rG  r'  r   s   &   rr   test_Ames_assay!TestPageTrendTest.test_Ames_assay$$  s    /Cxx~))Aq/##D4C+79 	S]]C(CJJ:##D4C+24 	S]]C(CJJ:ru   c                ,
   \        \        R R7      ;_uu_ 4        \        P                  ! R4       RRR4       \        \        R R7      ;_uu_ 4        \        P                  ! . 4       RRR4       \        \        R R7      ;_uu_ 4        \        P                  ! ^^.4       RRR4       \        \        R R7      ;_uu_ 4        \        P                  ! ^...4       RRR4       \        P
                  P                  R4      p\        \        RR7      ;_uu_ 4        \        P                  ! VP                  R4      4       RRR4       \        \        RR7      ;_uu_ 4        \        P                  ! VP                  R4      4       RRR4       Rp\        \        VR7      ;_uu_ 4        \        P                  ! . RO. RO.. ROR7       RRR4       \        \        VR7      ;_uu_ 4        \        P                  ! . RO. RO.. ROR7       RRR4       \        \        VR7      ;_uu_ 4        \        P                  ! . RO. RO.. ROR7       RRR4       \        \        VR7      ;_uu_ 4        \        P                  ! . RO. RO.RR7       RRR4       \        \        RR7      ;_uu_ 4        \        P                  ! . RO. RO.R	4       RRR4       \        \        RR7      ;_uu_ 4        \        P                  ! . RO. RO.R	4       RRR4       \        \        R
R7      ;_uu_ 4        \        P                  ! . RO^^\        P                  ..RR7       RRR4       \        \        RR7      ;_uu_ 4        \        P                  ! . RO. RO.RR7       RRR4       \        \        RR7      ;_uu_ 4        \        P                  ! . RO. RO.RR7       RRR4       R#   + '       g   i     EL; i  + '       g   i     EL; i  + '       g   i     ELx; i  + '       g   i     ELQ; i  + '       g   i     EL; i  + '       g   i     EL; i  + '       g   i     EL; i  + '       g   i     ELn; i  + '       g   i     EL@; i  + '       g   i     EL; i  + '       g   i     EL; i  + '       g   i     EL; i  + '       g   i     EL; i  + '       g   i     ELW; i  + '       g   i     R# ; i)z`data` must be a 2d array.r   Nl   s' zPage's L is only appropriatez+`predicted_ranks` must include each integer)rG  r'  r  z`data` is not properly rankedTz`data` contains NaNsFr&  z`method` must be inekki)rG  r  z`ranked` must be boolean.)rG  r  r  r  r  r  )r~  rJ   r   )r\   rJ   r   r   )r~   rJ   r   r_  )	r   r   r`   r  r   r  r  rg   rg  )rh   r  r  s   &  rr   r  'TestPageTrendTest.test_input_validation7$  sS   :-IJJ!!$' K:-IJJ!!"% K:-IJJ!!1a&) K:-IJJ!!QC5'* K ii##J/:-KLL!!#**V"45 M:-KLL!!#**V"45 M @:W55!!	9'=2;= 6 :W55!!	9'=2=? 6 :W55!!	9'=2>@ 6 :W55!!	9'=2;= 6
 :-LMM!!9i"8$? N:-LMM!!9i"8$? N :-CDD!!9q!RVVn"=).0 E :-BCC!!	9'=)/1 D 9,GHH!!	9'=)/1 IHY KJJJJJJJJJJJ
 MLLLLL
 655 655 655 655
 NMMMMM EDD DCC IHHs   O*O>
PP&&P:%&Q.!Q"0!Q62!R
4R4R23S2.SS.T*O;	>P	P#	&P7	:Q	Q	"Q3	6R	
R	R/	2S	S	S+	.S?	T	r  N)r   r  )rT      )r[  rY  I   r  rY  `   r$  ;   r
  r  rW  1   rV  +   r  rY     rH  r\  r  rP   r  r     r  )D   4   r  "   r	  r  A   X   rQ   rV  r]     rO  C   E   ^   r$  rI  r%  W   r  rG  r  rD  rG  )rV  rW   U   r  O   rV   \   V   rQ   @   r
  rX  rJ  r	  r
  Y   r{  rG  rG  r  6   rW   B   3   7   )r7   r   g @r|   r	  rU   )r|   r   r  rT   r  rQ   )r@  r
  rT   rV   rR   rw  )r   r   rJ   r\   )r   r   r\   rJ   r=  )r   rJ   r\   r  r  )r   r|   r   r   rJ   r\   )r\   r   r   r|   rJ   r   )r|   r   r   rJ   r\   )r   r\   r   rJ   r|   r  )rO   r   r|   r   r   rJ   r\   )r   r|   rO   r   r   rJ   r\   r  )
rT   rN   rK   rO   r   r|   r   r   rJ   r\   )
r\   rK   r   rO   r   r|   rN   r   rJ   rT   )
rN   rJ   rK   rO   r   r|   r   r   rT   r\   r  )rK   rO   r   r|   r   r   rJ   r\   )r\   r   r|   r   r   rO   rJ   rK   r  )r   r   r   r   rt  r   r  r  r  	data_3_25
data_10_26tsr   r   r   r   ts2r  r#  r-  r=  r  r   r   r   s   @rr   r  r  #  s    5" ))



+C

7#I
))



+CH%J .%y
A/5,

C$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DO*
BX [[92>) ?) #T7,,./ $dG,.M,.M,./ $dG%'?%'?%'?%'?%'?%'?%'?%'(C( [[93?[[)  @)5$;&01 01ru   r  c                   L  a  ] tR tRt o ]P
                  P                  R. RO4      R 4       tR tR t	R t
]P
                  P                  R. RO4      ]P
                  P                  R	^R
.4      R 4       4       t]P
                  P                  R	^R
.4      R 4       tRtV tR# )TestExpectileij$  r<   Nc                   Vf   TM
\        W!4      p\        P                  P                  ^*4      pVP	                  VP                  ^R7      P                  4       VR7      p\        \        P                  ! VR4      VP                  V4      4       VP	                  VP                  ^R7      P                  4       VR7      p\        \        P                  ! VRVR7      \        WER7      4       R # )Nr  r;   r6   r3  )rl  r   r  r  r^   r   r2   r`   	expectilerc   r   )rh   r<   ri   r  rk   r!  s   &&&   rr   test_same_as_meanTestExpectile.test_same_as_meanl$  s    GB,>ii##B'JJszzrz*1135JA3/<JJszzrz*1135JA3:HQ<RSru   c                    \         P                  P                  ^*4      pVP                  VP                  ^R7      4      p\	        \
        P                  ! V^ R7      VP                  V4      4       R# r  r  r  N)r   r  r  r^   r2   r`   re  r  rh   ri   r  rk   s   &&  rr   test_minimumTestExpectile.test_minimumw$  L    ii##B'JJszzrz*+3RVVAY?ru   c                    \         P                  P                  ^*4      pVP                  VP                  ^R7      4      p\	        \
        P                  ! V^R7      VP                  V4      4       R# ri  )r   r  r  r^   r2   r`   re  r  rk  s   &&  rr   test_maximumTestExpectile.test_maximum|$  rn  ru   c                r  a
 \         P                  P                  R 4      pR o
V
3R lp^
pVP                  V4      pVP                  4       pVP                  V4      p\        P                  ! VP                  V4      WaP                  V4      R7      pV! WVV4      p	\        WP                  V	4      4       R# )l   b@lN c                     \         P                  ! WR 7      p^V,
          WAV 8*  &   \         P                  ! WC,          W,
          ^,          ,          4      # )r?  )r   r  r]  )r  rL  r  r  r!  s   &&&& rr   r  'TestExpectile.test_weights.<locals>.fun$  s;    Q1AE	A1fI66!+
233ru   c                    < \         P                  ! V 4      \         P                  ! V 4      3p\        P                  ! SVWV3R 7      P
                  # ))bracketr  )r   r  r  r   minimize_scalarrk   )rL  r  r  rv  r  s   &&& rr   
expectile2.TestExpectile.test_weights.<locals>.expectile2$  sC    ffQi*G++C23G1DFFGaHru   r3  N)r   r  r  r`   re  r^   r2   )rh   ri   r  rx  r  rL  r  r  r   r   r  s   &&        @rr   test_weightsTestExpectile.test_weights$  s    
 ii##$78	4
	H
 JJqM

**Q-oobjjmUJJw<OP7+ZZ_-ru   r  r  rA  c                   \         P                  P                  ^*4      pVP                  VP	                  VR7      4      pR
 FD  p\        \        P                  ! VP                  W&R7      VR7      VP                  V4      4       KF  	  VP                  VP                  4       4      p\        \        P                  ! WV,           VR7      \        P                  ! WQR7      V,           4       \        \        P                  ! WV,
          VR7      \        P                  ! WQR7      V,
          4       \        \        P                  ! We,          VR7      V\        P                  ! WQR7      ,          4       VP                  VP                  V^
R7      4      pVR8X  d   R pMVR8  d   R pMR pV! \        P                  ! WW,           VR7      \        P                  ! WQR7      \        P                  ! WqR7      ,           4       VP                  VP	                  V^R7      4      p\        P                  ! WQR7      \        P                  ! WqR7      8:  g   Q hVP                  VP                  V^
R7      4      pR F  pV! \        P                  ! ^V,
          V,          Wg,          ,           VR7      ^V,
          \        P                  ! WQR7      ,          V\        P                  ! WqR7      ,          ,           4       K  	  \        \        P                  ! V) VR7      \        P                  ! V^V,
          R7      ) 4       R	# )ub  
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  r6   )r  r@  rj  r  c                     \        W4       R # r#  )r2   rL  rX  s   &&rr   	assert_op:TestExpectile.test_expectile_properties.<locals>.assert_op$  s
    %ru   c                     W8  g   Q hR # r#  r   r~  s   &&rr   r  r  $      uuru   c                     W8  g   Q hR # r#  r   r~  s   &&rr   r  r  $  r  ru   N)r  r   r6   )rC  r6   r  )r   r  r  r^   r  r2   r`   re  r   exponentiallogistic)	rh   r  r  ri   r  rk   ri  rl   r  s	   &&&&     rr   test_expectile_properties'TestExpectile.test_expectile_properties$  sV   $ ii##B'JJszzqz)*
  Aa >eL

1   JJs()OOAE/OOA+a/	
 	OOAE/OOA+a/	
 	OOAE///	
 JJs|||34C<& S[ 	OOAE/OOA+ooa-.	
 JJszzqaz01OOA+uq/NN	
N JJs|||34 A1a!#U;177EOOA334 ! 	OOQBe,__Qag..	
ru   c           	        \         P                  P                  ^*4      pVP                  VP	                  ^VR7      4      p. p\         P
                  ! R\         P                  ! R4      ^d4      p\         P                  ^ V^VRRR1,          ,
          ^3,           F3  pVP                  \        P                  ! V\        V4      R7      4       K5  	  VP                  VP                  VP                  V4      4      ^ 8  4      '       g   Q hR# )r  )rL  r  r6   Nrj  rW  rZ   )r   r  r  r^   paretologspacelog10r_rH  r`   re  r  r  diff)rh   r  ri   r  rk   e_list	alpha_seqr  s   &&&     rr   test_monotonicity_in_alpha(TestExpectile.test_monotonicity_in_alpha %  s    ii##B'JJszzAAz./KKRXXc]C8	UU1iYv2v->)>ABBEMM%//!5<@A Cvvbggbjj01A56666ru   r   r  )r@  g?r6   g/#    ?r  )r   r   r   r   r   r   r   rf  rl  rp  rz  r  r  r   r   r   s   @rr   rc  rc  j$  s     [[W&BCT DT@
@
.4 [[: [[S2t*-_
 ._
B [[S2t*-7 .7ru   rc  c            	         a  ] tR tRt o . RNRNRNRNRNRNRNRNRNR	NR
NRNRNRNRNRNRNRNRNRNRNRNRNRNRNRNRNRNRNRNRNRNRNR NR!NR"NR#NR$NR%NR&NR'NR(NRNR)NRNR*NR+NR,NR-NR.NR/NR0NR1NR'NR2NR3NR(NR4NR-NR5NR6NR7NR8NR9NR:NR;NR<NR=NR>NRNR4NR?NRNR@NRANR;NRNRBNRCNR3NRDNRENR	NRNRFNRGNRHNRINRNRNRJNRKNRENRLNRMNR%NRDNRNRNNRONRPNRKNR6NRQNRNRRNRSNRTNRUNRNRVNRNRWNRXNRNR)NR#NRYNRZNRNR[NtR. RnO]P                  ]P                  .tR\ t	]
R] 4       t]P                  P                  R^]^ Ro. . RpO..,           4      R_ 4       t]P                  P                  R`]4      Ra 4       t]P                  P                  Rb]4      Rc 4       t]P                  P                  Rd]4      Re 4       t]P                  P                  R^^^. RqO.4      ]P                  P                  RdRfRg.4      ]P                  P                  RhRfRg.4      Ri 4       4       4       tRj t]P                  P                  R`^ ^.4      Rk 4       tRltV tRm# )rTestLMomenti%  gףp=
?gp=
ף?r7   r
  rZ  r6   r   r  r
  g{Gz?g@g(\?gףp=
?g
ףp=
?g)\(?gRQ @r  g{Gz?gffffff @r>  g{Gz@gRQ?gGz@gQ?g(\?gzG?g
ףp=
?gQ?gffffff?gGz
@g(\?g(\?gQ?gq=
ףp?g(\@gQ?gGz	@r  gQ?gp=
ף @gp=
ף@gq=
ףp@gRQ?gGz?g
ףp=
?rD  gHzG?g      @gQ?r:  gRQ?gQ?gp=
ף?gQ@gGz?gQ?gp=
ף?gq=
ףp?r
  gp=
ף?gQ?g@g)\( @gzG@gGz?g(\?g333333?g)\(?gGz@r  g(\?g=
ףp=?gHzG?g{Gz?g
ףp=
?gףp=
?gQ?rQ
  g?g(\?gQ?gffffff@gQ?g?gGz?g{Gz?gHzG@g\(\?g(\?g(\?c           	        R p\         P                  ! \        VR7      ;_uu_ 4        \        P                  ! VP                  V P                  VP                  R7      4       RRR4       R#   + '       g   i     R# ; iz*`sample` must be an array of real numbers.r   r;   N)r   r   r   r`   lmomentr^   rG  r  r5  s   && rr   test_dtype_ivTestLMoment.test_dtype_iv%  sH    >]]:W55MM"**TYYbmm*DE 6555s   <A--A>	c           	         R p\         P                  ! \        VR7      ;_uu_ 4        \        P                  ! \
        P                  ! V P                  \        R7      4       RRR4       R#   + '       g   i     R# ; ir  )	r   r   r   r`   r  r   r   rG  r|  r:  s   & rr   test_dtype_iv_non_numeric%TestLMoment.test_dtype_iv_non_numeric"%  sB    >]]:W55MM"((499F;< 6555s   ;A,,A=	r  c                   \        V4      '       d:   \        P                  ! V4      '       d   V^8  d   \        P                  ! R4       Rp\        P
                  ! \        VR7      ;_uu_ 4        \        P                  ! VP                  V P                  4      VR7       RRR4       R#   + '       g   i     R# ; i)r\   z5Order is treated as an array; can't do value-based IVz*`order` must be a scalar or a non-empty...r   r  N)r-   r   isscalarr   r  r   r   r`   r  r^   rG  )rh   r  ri   r  s   &&& rr   test_order_ivTestLMoment.test_order_iv(%  sl    "::"++e,,KKOP>]]:W55MM"**TYY/u= 6555s   12B--B>	rD   c                    R p\         P                  ! \        VR7      ;_uu_ 4        \        P                  ! VP                  V P                  4      VR7       RRR4       R#   + '       g   i     R# ; i)z`axis` must be an integerr   rC   Nr   r   r   r`   r  r^   rG  )rh   rD   ri   r  s   &&& rr   test_axis_ivTestLMoment.test_axis_iv0%  sB    -]]:W55MM"**TYY/d; 6555   2A##A4	sortedc                    R p\         P                  ! \        VR7      ;_uu_ 4        \        P                  ! VP                  V P                  4      VR7       RRR4       R#   + '       g   i     R# ; i)z`sorted` must be True or False.r   )r  Nr  )rh   r  ri   r  s   &&& rr   test_sorted_ivTestLMoment.test_sorted_iv6%  sB    3]]:W55MM"**TYY/? 6555r  standardizec                    R p\         P                  ! \        VR7      ;_uu_ 4        \        P                  ! VP                  V P                  4      VR7       RRR4       R#   + '       g   i     R# ; i)z$`standardize` must be True or False.r   )r  Nr  )rh   r  ri   r  s   &&& rr   test_standardize_ivTestLMoment.test_standardize_iv<%  sB    8]]:W55MM"**TYY/[I 6555r  FT	presortedc                   VP                  . RO4      pV'       g4   \        P                  ! V4      R,          P                  V^,          4      pV'       d   \	        V P
                  4      MV P
                  pVP                  V4      p\        P                  ! WaW#R7      p\        WuVP                  V4      ^,
          ,          4       R# )0ʻEH?r  )r  r  N)r  gk?gz[?g+3 ?)	r^   rd   re   multiplyr  rG  r`   r  r2   )rh   r  r  r  ri   r   rG  r   s   &&&&&   rr   test_lmomentTestLMoment.test_lmomentB%  s     jj B C &&+b/**3q62C$-vdii 499zz$mmD[SE!21!456ru   c                    VP                   pVP                  V P                  4      p\        P                  ! VP                  W24      4      pVP                  \        P                  ! V4      V4      p\        WER R7       VP                  pVP                  . RO4      p\        P                  ! VP                  W24      4      p\        P                  ! VP                  V\        V4      4      4      p\        WE4       R# )rC	  r  Nr  )	rj  r^   rG  r`   r  r  r2   rv
  r&   )rh   ri   r<   rA  r   r   s   &&    rr   
test_dtypeTestLMoment.test_dtypeW%  s    

DII&mmBIIf45iif-u5t,O,mmBIIf45mmBIIf.>r.BCD!ru   c                   \         P                  P                  R 4      pVP                  RR7      p\        P                  ! VP                  V4      VR7      pVP                  \        P                  ! WAR7      4      p\        WV4       R# )l   jU3a r  rC   N)rT   rU   )r   r  r  r`   r  r^   r2   )rh   rD   ri   r  rk   r   r   s   &&&    rr   r	  TestLMoment.test_axisd%  s`     ii##$9:JJHJ%mmBJJqM5jjq45!ru   r   N)r\   rJ   r  rZ   r  r=  )r   r   r   r   rG  mathrg   r   not_integersr  r    r  r   r   r   r  r  r  r  r  r  r	  r   r   r   s   @rr   r  r  %  s    
ND 
N$ 
N 
Nc 
N3 
N 
Nc 
N3 
N 
ND 
N# 
N
N
N
N"
N$(
N*.
N04
N69
N;?
NAE
NGK
N
N
N
N"
N$(
N*.
N04
N6:
N<@
NBF
NHL
N 
N 
N 
N #
N %)
N +/
N 15
N 7;
N =A
N CG
N IM
N 	
N 	
N 	
N #	
N %)	
N +/	
N 14	
N 6:	
N <@	
N BF	
N HL	
N
 
N
 
N
 
N
 #
N
 %)
N
 +/
N
 15
N
 7;
N
 =A
N
 CG
N
 IM
N 
N 
N 
N "
N $(
N *.
N 04
N 6:
N <@
N BF
N HL
N 
N 
N 
N #
N %)
N +/
N 15
N 7;
N =@
N BF
N HL
N 
N 
N 
N #
N %)
N +/
N 15
N 7;
N =A
N CG
N IM
N 
N 
N 
N "
N $(
N *.
N 04
N 6:
N <@
N BF
N HL
N 
N 
N 
N #
N %)
N +/
N 15
N 7;
N =A
N CG
N IM
ND dhh9LF
 = =
 [[WlaR)5M&MN> O> [[V\2< 3<
 [[X|4@ 5@
 [[]L9J :J
 [[Wq!\&:;[[]UDM:[[[5$-87 9 ; <7$" [[VaV," -"ru   r  c                   H  a  ] tR tRt o ]P
                  P                  R. RO4      ]P
                  P                  RRR.4      ]P
                  P                  RRR.4      R 4       4       4       tR	 t]P
                  P                  R
4      ]P
                  P                  R4      R 4       4       t
]P
                  P                  R4      R 4       t]P
                  P                  R4      R 4       tR t]P
                  P                  R4      R 4       tR tR tRtV tR# )TestXP_Meanin%  rD   Nr  Tr"  Fc                   \         P                  P                  R 4      pVP                  R4      pVP                  V4      pR;rV'       d;   VP                  R4      pVP                  V4      p	\         P                  ! Wh4      w  rh\        WyW$R7      p
\         P                  ! WhW$R7      p\        WP                  V4      4       R# )   -fH N)r  rD   r"  rL  r{   )r   r  r  r^   r  r   r  r2   )rh   ri   rD   r  r"  r  rk   x_xpr!  w_xpr   r   s   &&&&&       rr   test_xp_mean_basicTestXP_Mean.test_xp_mean_basicp%  s     ii##N3JJy!zz!}

6"A::a=D&&q,DAtHjjDDZZ_-ru   c                    VP                  R 4      VP                  ^4      r2Rp\        P                  ! \        VR7      ;_uu_ 4        \        W#R7       RRR4       R#   + '       g   i     R# ; i)r   r  r   r3  N)r   rX  r   r   r   r   )rh   ri   rk   r!  r  s   &&   rr   test_non_broadcastable"TestXP_Mean.test_non_broadcastable%  sF    yy~rxx{1C]]:W55Q" 6555s   AA0	r  r  c                   VP                  . RO4      p\        VP                  . RO4      VR7      p\        W1P                  VP                  4      4       \        VP                  . RO4      VR7      p\        W1P                  \        P
                  ) 4      4       \        VP                  . RO4      VR7      p\        W1P                  \        P
                  4      4       R# )r   r3  Nr  r  )r   r   r   )r   r   r   )r^   r   r2   rg   r   r   )rh   ri   r  r   s   &&  rr   r|	  TestXP_Mean.test_special_cases%  s     **]+rzz,/AZZ/0rzz,/AZZ01rzz,/AZZ/0ru   r
  c                   VP                  R 4      pV^8H  pVP                  W1P                  V4      p\        V4      '       d<   \        P
                  ! \        RR7      ;_uu_ 4        \        VRR7       RRR4       M:\        P
                  ! \        RR7      ;_uu_ 4        \        VRR7       RRR4       \        V4      p\        VRR7      pVP                  VP                  4      p\        WF4       \        WV4       \        VRR7      pVP                  W#( ,          4      p\        Wv4       VP                  ^
4      pVP                  W1P                  V4      p\        VP                  R 4      VRR	7      pVP                  W#( ,          4      p\        Wv4       R#   + '       g   i     L; i  + '       g   i     EL; i)
r   rb
  r   r   r   Nr   r  r   )r  r   )r   r  rg   r#   r   r   rg  r   r   r^   r3   rc   r2   r  )	rh   ri   rk   r  r  r  r   r   r  s	   &&       rr   r  TestXP_Mean.test_nan_policy%  sO    IIcNQHHT661% y0OPPw/ QP z1PQQw/ R {k2jj "" qV,ggah! ''"+((41ryy~w6Jggah!/ QP RQQs   (F!#F4!F1	4G	c                h   VP                  R 4      pVP                  V^8H  VP                  V4      pRp\        \        W1R7      ;_uu_ 4        \        V\        P                  ,          RR7      pVP                  VP                  4      p\        WE4       RRR4       R#   + '       g   i     R# ; ir   zAfter omitting NaNs...r^  r   r   N)	r   r  rg   r"   r   r   r   r^   r3   rh   ri   rk   r  r   r   s   &&    rr   test_nan_policy_warns!TestXP_Mean.test_nan_policy_warns%  sz    IIcNHHQ!VRVVQ' +w>>1rvv:&9C**RVV$CC% ?>>>   A	B  B1	c                   R p\         P                  ! \        VR7      ;_uu_ 4        \        VP	                  . 4      4      pVP	                  VP
                  4      p\        W44       RRR4       Rp\         P                  ! \        VR7      ;_uu_ 4        \        VP	                  . .4      ^R7      pVP	                  VP
                  .4      p\        W44       RRR4       \        VP	                  . .4      ^ R7      pVP	                  . 4      p\        W44       R#   + '       g   i     L; i  + '       g   i     L^; iz,One or more sample arguments is too small...r   Nz2All axis-slices of one or more sample arguments...rC   )r   r  r   r   r^   rg   r3   rh   ri   r  r   r   s   &&   rr   r  TestXP_Mean.test_empty%  s    @\\,G<<2::b>*C**RVV$CC% =
 G\\,G<<2::rd+!4C**bffX&CC% =
 rzz2$'a0jjn! =< =<   ADAD1D.	1E	4ignore:overflow encountered in reduce:RuntimeWarningc           	        VP                  VP                  4      P                  p\        P                  ! W".\        P                  R 7      pVP	                  V4      p\        P
                  ! RR7      ;_uu_ 4        \        V4      p\        P                  ! V4      p\        P                  P                  V\        P                  4       \        WQP	                  V4      4       RRR4       \        WAP                  R 7      pVP	                  \        P                  ! \        P                  ! V\        P                  R 7      4      4      p\        WV4       R#   + '       g   i     L|; i)r;   r   )overN)r   rj  r  r   r^   r  r   rc   r!  r   r   r2   r   rh   ri   r  x_npr  r   r   s   &&     rr   r  TestXP_Mean.test_dtype%  s     hhrzz"&&zz3*BJJ7zz$ [[h''4.C''$-CJJ##C0CC1	 ( t::.jjD

!CDE! ('s   =A*EE+	c                    VP                  ^
4      pVP                  R4      p\        \        V4      \        V4      4       \        \        W2R7      \        W3R7      4       R# )rT   r   r3  N)r   r3   r   r   s   &&  rr   test_integerTestXP_Mean.test_integer%  sB    IIbMIIcNXa[1.0FGru   c                   \         P                  P                  R 4      pVP                  R4      w  r4rV\        VP	                  W4R,          ,           4      VP	                  WVR,          ,           4      R7      p\         P
                  ! W4R,          ,           WVR,          ,           R7      p\        WqP	                  V4      4       R# )r  r  r3  N)r   r  )r   r  r  r   r^   r  r2   )	rh   ri   r  rk   rl   wxwyr   r   s	   &&       rr   r   TestXP_Mean.test_complex_gh22404%  s~    ii##N3zz'*brzz!d(+RZZU
5KLjjrT22:6ZZ_-ru   r   Nr\   rZ   )re  rJ   )r   r   r   r   r   r   r   r  r  r   r|	  r  r  r  r  r  r  r   r   r   s   @rr   r  r  n%  s-    [[V%;<[[Yt5[[Z%7. 8 6 =. # [[ WX[[ VW1 X Y1 [[9""@ [[ QR	& S	&"" [[>"""H. .ru   r  c                     a  ] tR tRt o ]P
                  P                  R. RO4      ]P
                  P                  RRR.4      ]P
                  P                  R^ ^.4      ]P
                  P                  RR	R
.4      R 4       4       4       4       tR tR t	]P
                  P                  R4      R 4       t]! RR7      R 4       tR t]P
                  P                  R4      R 4       tR tR tRtV tR# )
TestXP_Vari%  rD   Nr"  FTrx   r   r  r   c           	     F   \         P                  P                  R 4      pVP                  R	4      p\         P                  pVR8X  dB   VP                  VP                  R7      R8  p	\         P
                  Wy&   \         P                  pVP                  V4      p
\        WW4VR7      p\        P                  ! 4       ;_uu_ 4        \        P                  ! RR\        4       V! WrW4R7      pRRR4       \        WP                  X4      4       R#   + '       g   i     L,; i)
r  r   r  r6   )rD   r"  rx   r   r   rX  )rD   r"  r  NrL  )r   r  r  r   r  rg   nanvarr^   r   r   r   r   r   r2   )rh   ri   rD   r"  rx   r   r  rk   var_refnan_maskr  r   r   s   &&&&&&       rr   test_xp_var_basicTestXP_Var.test_xp_var_basic%  s    
 ii##N3JJy!&&zzqwwz/#5H&&AKiiGzz!}d!+- $$&&##=~O!KC '
 	ZZ_- '&s   'DD 	c                    \        VP                  R R.4      ^R7      p\        W!P                  VP                  4      4       R# )r   r   rw   N)r   r^   r2   rg   )rh   ri   r   s   && rr   r|	  TestXP_Var.test_special_cases&  s0    bjj"b*q9ZZ/0ru   c                \   VP                  R 4      pV^8H  pVP                  W1P                  V4      p\        V4      p\        VRR7      pVP	                  VP                  4      p\        WF4       \        WV4       \        VRR7      pVP                  W#( ,          4      p\        Wv4       R# )r   r  r   r   N)r   r  rg   r   r^   r3   r   r2   )rh   ri   rk   r  r  r  r   r   s   &&      rr   r  TestXP_Var.test_nan_policy&  s    IIcNQHHT661% qzq[1jj "" aF+ffQuX!ru   r
  c                h   VP                  R 4      pVP                  V^8H  VP                  V4      pRp\        \        W1R7      ;_uu_ 4        \        V\        P                  ,          RR7      pVP                  VP                  4      p\        WE4       RRR4       R#   + '       g   i     R# ; ir  )	r   r  rg   r"   r   r   r   r^   r3   r  s   &&    rr   r   TestXP_Var.test_nan_policy_warns.&  sz    IIcNHHQ!VRVVQ' +w>>!bff*8C**RVV$CC% ?>>>r  )r   c           	         R p\         P                  ! \        VR7      ;_uu_ 4        \        VP	                  ^^VP
                  .4      RR7       RRR4       R#   + '       g   i     R# ; i)r   r   r   r   N)r   r   r   r   r^   rg   r5  s   && rr   r5   TestXP_Var.test_nan_policy_raise:&  sF     2]]:W55BJJ1bff~.7C 6555s   *AA,	c                   R p\         P                  ! \        VR7      ;_uu_ 4        \        VP	                  . 4      4      pVP	                  VP
                  4      p\        W44       RRR4       Rp\         P                  ! \        VR7      ;_uu_ 4        \        VP	                  . .4      ^R7      pVP	                  VP
                  .4      p\        W44       RRR4       \        VP	                  . .4      ^ R7      pVP	                  . 4      p\        W44       R#   + '       g   i     L; i  + '       g   i     L^; ir  )r   r  r   r   r^   rg   r3   r  s   &&   rr   r  TestXP_Var.test_emptyA&  s    @\\,G<<"**R.)C**RVV$CC% =
 G\\,G<<"**bT*3C**bffX&CC% =
 bjj"&Q/jjn! =< =<r  r  c           	     t   VP                  VP                  4      P                  p\        P                  ! W"^,          .\        P                  R7      pVP	                  V4      p\        V4      p\        P                  ! V4      p\        P                  P                  V\        P                  4       \        WQP	                  V4      4       \        WAP                  R7      pVP	                  \        P                  ! \        P                  ! V\        P                  R7      4      4      p\        WV4       R# )rJ   r;   N)r   rj  r  r   r^   r   r   r!  r   r   r2   r   r  s   &&     rr   r  TestXP_Var.test_dtypeR&  s     hhrzz"&&zz3A,bjj9zz$dmffTl


RVV,ZZ_- d**-jj

4rzz BCD!ru   c                    VP                  ^
4      pVP                  R4      p\        \        V4      \        V4      4       R# )rT   r   N)r   r3   r   r   s   &&  rr   r  TestXP_Var.test_integerd&  s.    IIbMIIcN
GAJ/ru   c                6   \         P                  P                  R 4      pVP                  R4      w  r4\        VP	                  W4R,          ,           4      4      p\         P
                  ! W4R,          ,           4      p\        WQP	                  V4      RR7       R# )r  r  F)check_dtypeNr  )r   r  r  r   r^   r   r2   r  s   &&     rr   r  TestXP_Var.test_complex_gh22404j&  se    ii##N3zz'"bjjrT*+ffQ2XZZ_%@ru   r   r  )r   r   r   r   r   r   r   r  r|	  r  r   r  r   r5  r  r  r  r  r   r   r   s   @rr   r  r  %  s    [[V%;<[[Z%7[[\Aq62[[\K+@A. B 3 8 =.,1
"" [[ QR	& S	& &D 'D"" [[>""0A Aru   r  Fprivatec           	      0   \         P                  P                  R 4      pVP                  RR7      pV P                  V4      p^p\	        W4V R7      w  rV\        WPP                  V4      4       \        Wd4       Rp\	        W4V R7      w  rV\        WPP                  VP                  4       4      4       \        V^ 4       ^p\	        VR,          W@R7      w  rV\        WPP                  \         P                  ! VR,          4      4      4       \        Wd4       R# )l   ~Mr  r  Nr2	  r1	  )	r   r  r  r^   r   r3   r   r  
atleast_1d)ri   r  r  rk   rD   x_outaxis_outs   &      rr   test_chk_asarrayr   r&  s    
))


.C		#B


2AD"1B7OEE::b>* D"1B7OEE::bhhj121D"1W:D@OEE::bmmBwK&@AB ru   r3  z!JAX+wilcoxon+'exact' incompatibler  c           	         \         P                  P                  R 4      pVP                  VP                  ^
4      4      pVP                  VP                  ^
4      4      pV \        P
                  8X  d   V3pM+V \        P                  8X  d   V\        P                  3pMW43pV ! VRR/ pV ! VRR/ p\        VP                  VP                  4       \        VP                  VP                  4       V P                   R2p\        P                  ! \        VR7      ;_uu_ 4        V ! VRRRR/  RRR4       R#   + '       g   i     R# ; i)l   0:00 r  r  r  z%\(\) got multiple values for argumentr   N)r   r  r  r^   r`   wilcoxonrK  r   rL  r3   r@  r?  r   r   r   rg  )	r  ri   r  rk   rl   r  r   r  errs	   &&       rr   test_rename_mode_methodr  &  s     ))

 4
5C


3::b>"A


3::b>"A
ennt		7<< v
t
$G
$C#7#DCMM4>>2CJJ,ll^@
AC	y	,	,T0'00 
-	,	,	,s   'D<<E	)	r\   rJ   r   r   r|   r   rO   rK   rN   )	r~   r~   r~   r~   r~   r~   r~   r~   r~   )	iiiiiiiii)	g~?g<?gEk?g?ge(?gq?g?g8C?gG?)	g   mBg   }Bg  yӅBg   Bg  @0Bg  yӕBg  BwBg   Bg   ^B)	r  g-=gAfc=g-=gdy=gAfc=gYLɞ=g-=gSSʣ=)	r6   r7   ry   r  rF   rG   rH   r  r?  )r   rK   rV   rK   r   rG  gfx(@)rJ   r
  rV   rJ   r   r  g{T4@)r   r|   rO   rN   r1	  r  )rk   r\   )r  r\   )rf  r\   )r8  r  ro  r   collectionsr   	itertoolsr   hypothesis.extra.numpyextranumpyrG  rL  rA  numpy.testingr   r   r   r   r   r	   r
   r   r   r   r   r   r   r   scipy.statsr`   scipy.stats._ksstatsr   scipy.special._testutilsr   scipyr   r   common_testsr   scipy.stats._axis_nan_policyr   r   r   r   r   r	  r   r   r   r   r   r   scipy._lib._utilr   scipy.conftestr    scipy._lib._array_apir!   r"   r#   r$   r%   r&   r'   r(   r)   r*   r+   r,   r-   r.   r/   r0   r1   scipy._lib._array_api_no_0dr2   r3   r4   scipy._external.array_api_extra	_externalarray_api_extrard   r5   lazy_xp_modulesr   r   r  r  r_   r;  r   r   r   r   r  r9   r   r   r:  r  rp  r  ry  rt  r  r  r.  r,  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r.  r   rX  rV  r  r  r  r  r  r#  r!  rN  	compositerR  rZ  rT  r  PowerDivCaser  r  r   r  r  r  r  r   r1  r/  rJ  rG  rK  rm  r^  r  r   r	  r	  r   ri	  rp	  ru	  rv	  rr	  r	  r	  r	  r	  r	  r 
  r0
  rF
  rD
  rz
  r|
  r
  r}
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r  r
  r  r  rP  r  rr  r  r  r  r  r  r  r  r  r  rI  rF  r  r  r  r  r  r  rR  rm  r  r~  r  r  r  r  re  rc  r  r  r  r  r
  r   r  r  r   ru   rr   <module>r     sV    	  "  % %  , , ,  * & &   ( - # -T TU U & .N N N N N
 Y X - - . ';;// KK11  

u% %(  !&(	 238
:DeLDeL3U;|T |T~p# p#f 5>>"M> M> #M>`DL DLNL6 L6^~5 ~5B 5##$A/ A/ %A/H 5##$XE XE %XEvP" P"f 5##$\3 \3 %\3~QX QXh 5==!4# 4 "4" 5==!"# " ""*rN rNj 5::I3 I3 I3X 599.. .. ..b 5::j" j" j"Z 5<< f& f& !f&R 5==!0 0 "0* 5--.IL IL /ILX 599X" X" X"v 5<< e" e" !e"P
1 
1 5::^" ^" ^"B 5>>"`"# `" #`"F    !< 5$$%Q) Q) &Q)hw8 w8t & }Dqt"&&+266$<78AbffSkMAbffTlN:;I	K }N"&&+"&&,6266#;FGAbffSkMBrvvd|O;asmKL1	2 }AAD"&&+266$<78AbffSkMAbffTlN:;I	K |<aaQa1+ : r!!Qa1
 rxxBr
+--!!" rxxBr
+!!# 4 5))*QH QH +QHh 5??#-' -' $-'` 5**++T +T ,+T\ 5<< >@ >@ !>@F 5>>"xG xG #xGt 5>>"~. ~. #~.B
 GHFG5??#\1 \1 $ H I\1~ )**RYYr]A,>?299R="))B-*HI,K L5L5
/d 
/ 5??E$>$>?m m @m` 5??#,'OPDEn6 n6 F Q $n6b 5??#R2 R2 $R2j 5??#}O }O $}O@ 5??#IB IB $IBX 5??#U@ U@ $U@p 5--.) ) /)( 5$$%z) z) &z)z 5>>"\9 \9 #\9~*- *-Z 5>>"-> - #-2 5%%&-~ - '-> 5##$-^ - %-O O. 5$$%51 51 &51p 5''(8 8 )8$ 5(()E& E& *E&P+ +D ++)-+$DDD 5;;s3 s3  s3l 5;;m3 m3  m3` 5;;a. a.  a.H 5::R3 R3 R3j.3 .3b 5??#.V .V $.Vb 5??#K. K. $K.\ 5(()~< ~< *~<B 5>>"i) i) #i)X 5==!n n "nb 5(()`4 `4 *`4F#A #ALl  l ^Z Zz? ?D 5&&'V+ V+ (V+r 8Xx0A 1A
 5&&'H0 H0 (H0V
{1 {1| 5??#^7 ^7 $^7B 5==!^" ^" "^"B 8J. J. J.Z 7rA rA rAj !!%	!:! ;!* +&IJ(((&	! 1 K1ru   