
    MhH                     0   S SK r S SKrS SKrS SKJr  S SKrS SKJrJ	r	J
r
JrJr  S SKJr  S SKJr   " S S5      r\R&                  R)                  S/ SQ5      S	 5       r\R&                  R)                  S
/ SQ5      \R&                  R)                  SSS\R,                  /5      \R&                  R)                  SS/SS//5      S 5       5       5       r\R&                  R1                  S5      \R&                  R)                  S\\\\/5      S 5       5       rg)    N)lib)	DataFrameIndexSeries	Timestamp
date_range)get_groupby_method_argsc                      \ rS rSr\R
                  S 5       r\R                  R                  SSS/5      S 5       r	\R                  R                  SSS/5      S	 5       r
\R                  R                  SS
S/5      S 5       r\R                  R                  SSS/5      S 5       r\R                  R                  SSS/5      S 5       r\R                  R                  SSS/5      S 5       rS rSrg)TestNumericOnly   c                     [        / SQ/ SQ/ SQ[        S5      [        [        S5      5      R                  S5      / SQ[	        SSS	9[	        SSS
S9[
        R                  " SSSS9S.	/ SQS9nU$ )N)   r      )r   r      )g      @g      @      @abccategory)      	   20130101r   )periods
US/Eastern)r   tzz1 ss)r   freq)	groupintfloatstringcategory_stringcategory_intdatetime
datetimetz	timedeltacolumns)r   listr   astyper   pdtimedelta_range)selfdfs     X/var/www/html/env/lib/python3.13/site-packages/pandas/tests/groupby/test_numeric_only.pyr-   TestNumericOnly.df   sq     " (u+#)$u+#6#=#=j#I )&z1=(Q<P//qsK


0 	    methodmeanmedianc                    [        / SQ5      nUR                  S5      n[        SS/SS/[        R                  " S5      [        R                  " S5      /S	S
/[        S5      [        S5      /[        SSS9[        SSS9/S.[        SS/SS9/ SQS9n[        XB5      " SS9n[        R                  " UR                  U5      U5        UR                  nU R                  XXs5        g )Nr   r   r"   r   g      @r   g      @r   z1.5s3sg      ?r   z2013-01-01 12:00:00z2013-01-03 00:00:00r   )r   )r"   r   r%   r   r#   r$   r   r   )name)indexr'   Tnumeric_only)r   groupbyr   r*   	Timedeltar   getattrtmassert_frame_equalreindex_liker'   _check)r,   r-   r1   expected_columns_numericgbexpectedresultexpected_columnss           r.   test_averagesTestNumericOnly.test_averages5   s     $))I#J ZZ !$as ll62BLL4FGQx3434
 3E3E AW-!
. $$7
f11(;XF#++B 0Kr0   minmaxc                 F    [        / SQ5      nUnU R                  XX45        g )N)r   r   r    r"   r#   r$   r%   r   rA   r,   r-   r1   rF   rB   s        r.   test_extremaTestNumericOnly.test_extremaY   s+    
 !

 $4 B 0Kr0   firstlastc                 F    [        / SQ5      nUnU R                  XX45        g )N)r   r   r    r!   r"   r#   r$   r%   rL   rM   s        r.   test_first_lastTestNumericOnly.test_first_lastm   s)     	
 $4 B 0Kr0   sumcumsumc                     [        / SQ5      n[        / SQ5      nUS:X  a  [        / SQ5      nU R                  XXC5        g )Nr5   )r   r   r    r"   r%   rV   )r   r   r"   r%   rL   )r,   r-   r1   rB   rF   s        r.   test_sum_cumsumTestNumericOnly.test_sum_cumsum   sA    #()I#J  C
 X$%RSB 0Kr0   prodcumprodc                 F    [        / SQ5      nUnU R                  XX45        g )Nr5   rL   rM   s        r.   test_prod_cumprod!TestNumericOnly.test_prod_cumprod   s#     !AB#3 B 0Kr0   cummincummaxc                 F    [        / SQ5      nUnU R                  XX45        g )N)r   r   r"   r#   r$   r%   rL   rM   s        r.   test_cummin_cummax"TestNumericOnly.test_cummin_cummax   s)     !S

 $4 B 0Kr0   c                 H   UR                  S5      nUR                  S5      (       a  [        O[        nUS;   a_  SR	                  SSU S3[
        R                  " SU S	35      S
/5      n[        R                  " XgS9   [        XR5      " 5         S S S 5        OUS;   ar  SR	                  S[
        R                  " SU S	35      [
        R                  " SU S35      /5      n[        R                  " XgS9   [        XR5      " 5         S S S 5        O1[        XR5      " 5       n[        R                  " UR                  U5        US;  ax  SR	                  SSS
SU S3[
        R                  " SU S	35      [
        R                  " SU S35      /5      n[        R                  " XgS9   [        XR5      " SS9  S S S 5        g [        XR5      " SS9n[        R                  " UR                  U5        g ! , (       d  f       N= f! , (       d  f       N= f! , (       d  f       g = f)Nr   cum)rI   rJ   r_   r`   rV   r[   |zCategorical is not orderedzCannot perform z with non-ordered Categoricalagg function failed [how->,dtype->object]*function is not implemented for this dtypematch)rU   r2   r3   rZ   z-category type does not support sum operationsz,dtype->string]rP   rQ   zcategory type does not supportFr9   )r;   
startswithNotImplementedError	TypeErrorjoinreescapepytestraisesr=   r>   assert_index_equalr'   )	r,   r-   r1   rF   rB   rC   	exceptionmsgrE   s	            r.   rA   TestNumericOnly._check   s   ZZ  ,2+<+<U+C+C'	LL((0%fX-JKII :6(/RS@C y4#% 5488((CII :6(/RSII :6(/RSC y4#% 54 R(*F!!&..2JK**((04@%fX-JKII :6(/RSII :6(/RS	C y4#7 54 R(e<F!!&..2BC? 54 54" 54s$   ?G17H'H1
G?
H
H! N)__name__
__module____qualname____firstlineno__rs   fixturer-   markparametrizerG   rN   rS   rX   r]   rb   rA   __static_attributes__ry   r0   r.   r   r      s(    ^^ : [[X'9:!L ;!LF [[Xu~6L 7L& [[X'89L :L" [[Xx'89	L :	L [[X	':;L <L [[X(';<	L =	L1Dr0   r   r:   )TFNc                    US;   a  [         R                  " S5        US;   a.  SnU R                  [         R                  R	                  US95        [        [        R                  R                  S5      R                  S5      / SQS	9nS
US'   / SQnUR                  U5      n[        Xq5      n[        X5      n	SS0n
Ub  X*S'   SnSnSU S3nUb  X;   a  SnUS;   aM  [         R                  " [        US9   [        R                   " ["        US9   U" U	0 U
D6  S S S 5        S S S 5        g [         R                  " [        US9   U" U	0 U
D6  S S S 5        g X;  a-  Sn[         R                  " [        US9   U" U	0 U
D6  S S S 5        g Ub  U(       d  US;  a  SnU(       a!  SS Kn[        UR&                  R(                  4nO[        n[         R                  " USSR+                  U5       S3S9   [        R                   " ["        US9   U" U	0 U
D6  S S S 5        S S S 5        g [        R                   " ["        US9   U" U	0 U
D6nS S S 5        U(       a  UR-                  SS	9R.                  OUR.                  n[        UU5      " U	6 R.                  nUS:X  a  U(       d  UR1                  [2        5      n[        R4                  " WU5        g ! , (       d  f       GN= f! , (       d  f       g = f! , (       d  f       g = f! , (       d  f       g = f! , (       d  f       GN= f! , (       d  f       g = f! , (       d  f       N= f)N)idxmaxidxminz5idxmax and idx_min tested in test_idxmin_idxmax_axis1)corrwithskewzDGH#47723 groupby.corrwith and skew do not correctly implement axis=1)reasonr   )
      )ABCDr&   xE)
r   r   r   r   r   r   r   r   r   r   axisr   r:   )r[   rV   difffillna
pct_changerankshift)r[   rV   r   r   r   r   r`   r_   r   r   r   zDataFrameGroupBy.z with axis=1 is deprecated1got an unexpected keyword argument 'numeric_only')r[   rV   rj   z)got an unexpected keyword argument 'axis')r   r   )z"not supported between instances ofz2can't multiply sequence by non-int of type 'float'unsupported operand typezhas no kernelr   (rf   )r   )rs   skipapplymarkerr   xfailr   nprandomdefault_rngstandard_normalr;   r=   r	   rt   ro   r>   assert_produces_warningFutureWarningpyarrowr   ArrowNotImplementedErrorrp   dropTr)   r   assert_equal)requestgroupby_funcr:   using_infer_stringrw   r-   groupsrC   r1   argskwargsno_argshas_axiswarn_msgmsgspaerrsrE   df_expectedrD   s                       r.   test_axis1_numeric_onlyr      s   ++KL++TFKK--S-9:	
		a 009CW
B BsG+F	F	BR&F"<4Da[F!-~ UGH #<.0JKHL$;A00y4//XND+F+ O 54 y4'' 54		%9]]9C0D#F# 10 
l M 9
  rvv>>?DD]]4388D>*:!'<=++MJ'' K >= ''XFT,V,F G 1=bggcg*,,"$$;5t<>>7"<  u-H
)W ON 54 54 10. KJ >= GFsl   7K	KK	K06	L0L$
	LL$=	L5
K	K
K-0
K>
L
L!	L$$
L25
Mzkernel, has_arg))allF)anyF)bfillF)corrT)r   T)covT)r`   T)r_   T)r[   T)rV   T)r   F)ffillF)r   F)rP   T)r   T)r   T)rQ   T)rJ   T)r2   T)r3   T)rI   T)nthF)nuniqueF)r   F)rZ   T)quantileT)semT)r   T)stdT)rU   T)varTTFkeysa1a2c                    [        SS/SS/SS/S[        /-  S.5      n[        X5      nU[        R                  L a  0 OSU0nUR                  U5      n[        Xp5      nU(       a   USL a  U" U0 UD6n	SU	R                  ;  d   eg U S	;   d  U S
;   a[  U[        R                  L aH  U S:X  a  [        OS n
Sn[        R                  " XS9   U" U0 UD6n	S S S 5        SW	R                  ;   d   eg U(       a  USLd   eU R                  S5      (       a  [        O[        nSR                  SSSSS[        R                   " SU  S35      /5      nU S:X  a  SnOU S:X  a  Sn["        R$                  " XS9   U" U0 UD6  S S S 5        g U(       d>  U[        R                  La+  ["        R$                  " [        SS9   U" U0 UD6  S S S 5        g U S;   d   eU[        R                  L d   e["        R$                  " [        SS9   U" U0 UD6  S S S 5        g ! , (       d  f       GNE= f! , (       d  f       g = f! , (       d  f       g = f! , (       d  f       g = f)Nr   r         )r   r   a3br:   Tr   rl   )r   r   r   r   r   r   r   r   %DataFrameGroupBy.fillna is deprecatedrj   re   rf   znot allowed for this dtype+cannot be performed against 'object' dtypeszmust be a string or a.* numberr   ri   rg   rh   r   z8'<' not supported between instances of 'type' and 'type'r   z8'>' not supported between instances of 'type' and 'type'r   )r   r   )r   objectr	   r   
no_defaultr;   r=   r'   r   r>   r   rm   rn   ro   rp   rq   rr   rs   rt   )kernelhas_argr:   r   r-   r   r   rC   r1   rE   warnrw   rv   s                r.   test_numeric_onlyr   *  sB   T 
1a&Aq!f1x<P	QB"6.D3>>1R7UF	D	BR F<4'((&..((( 	## RR. !'( 2}5''8T,V,F 9fnn$$$	4'''
 ,2+<+<U+C+C'	hh,=0*<		6vhoNO

 XLCxLC]]90D#F# 10S^^;]]P
 D#F#
 

 ////s~~---]]9,GHD#F# IHK 984 10
 
 IHs0   		H1	H"6	H3>	I
H"
H03
I
Iz4ignore:Downcasting object dtype arrays:FutureWarningdtypec                    / SQn[        / SQU S9nUR                  U5      nUS:X  a  [        XQ5      (       a   eg [        XQ5      n[        / SQ5      nUR                  U5      n[        X5      n	[	        X5      n
SnSnX;   ai  U [
        L a`  US:X  a  SnOS	nUS
:X  a  [        OS nSn[        R                  " XS9   [        R                  " [        US9   U" U
6   S S S 5        S S S 5        OU [
        L a  US
:X  a  [        OS nSn[        R                  " XS9   U" U
6 nS S S 5        [        R                  " XS9   U	" U
6 nS S S 5        X;   a  WR                  [
        5      n[        R                  " WW5        SnUU;  a-  Sn[        R                  " [        US9   U" U
SS06  S S S 5        g U [
        L a>  SR                  / SQ5      n[        R                  " [        US9   U" U
SS06  S S S 5        g U [        :X  aR  US:X  aL  Sn[        R                  " [        US9   U" U
SS06nU" U
SS06nS S S 5        [        R                  " WW5        g U" U
SS06nU" U
SS06n[        R                  " UU5        g ! , (       d  f       GN= f! , (       d  f       GN!= f! , (       d  f       GN= f! , (       d  f       GNv= f! , (       d  f       g = f! , (       d  f       g = f! , (       d  f       N= f)N)r   r   r   )r   r   r   )r   r   )r   r   r`   r_   r[   rV   r   )rP   rQ   r   r   r   r   rU   r   r   r   r2   r3   rI   rJ   rZ   r   r   r   z!is not supported for object dtyper   r   rj   z"SeriesGroupBy.fillna is deprecated)rP   rQ   rJ   r2   r3   rI   rZ   r   r   r   r   rU   r   r`   r_   r[   rV   r   r:   Trf   )z@SeriesGroupBy.sem called with numeric_only=True and dtype objectz=Series.skew does not allow numeric_only=True with non-numericz7cum(sum|prod|min|max) is not supported for object dtypezCCannot use numeric_only=True with SeriesGroupBy\..* and non-numericz-Allowing bool dtype in SeriesGroupBy.quantileF)r   r;   hasattrr=   r	   r   r   r>   r   rs   rt   ro   r)   assert_series_equalrp   bool)r   r   r   grouperserrC   r1   expected_serexpected_gbexpected_methodr   fails_on_numeric_object
obj_resultrw   r   r   rE   rD   has_numeric_onlys                      r.   "test_deprecate_numeric_only_seriesr     s    G
%
(C	W	Bz!2,,,,R&F)$L&&w/Kk8O"<5DJ* .5F?:%?C5C , 8}d:''=y4 5 >= 
& , 8}d7''=T]F >''=&-H >%v.H
vx0& ++A]]9C0D,t, 10	&hh
 ]]9C0D,t, 10	$<:5=''SAT55Ft8%8H B 	vx01D144e4
vx0s 54 >= >===6 10 10 BAs`   ;JI:JJ;J0	K$	K!K$:
J		J
J
J-0
J?
K
K!$
K2)rq   numpyr   rs   pandas._libsr   pandasr*   r   r   r   r   r   pandas._testing_testingr>   pandas.tests.groupbyr	   r   r   r   r   r   r   filterwarningsr   r   r   r   r   ry   r0   r.   <module>r      s#   	       8{D {D| )<=T* >T*n  #H $s~~)FG4&4,!78@$ 9 HI#L@$F RS4eV"<=t1 > Tt1r0   