
    Mh#+                     n   S SK Jr  S SKrS SKrS SKrS SKJr  S SKJ	r	J
r
  S SKJr  S SKJrJr  \R"                  R%                  S/ SQ5      \R"                  R%                  S\R&                  " / 0 S	S
9\R&                  " S/0 SS
9\R&                  " / SS0SS
9\R&                  " / SS0SS
9\R&                  " SS/SS0SS
9/5      \R"                  R%                  SSS/5      S 5       5       5       r\R"                  R%                  S/ SQ5      S 5       r\R"                  R%                  S/ SQ5      \R"                  R%                  SSS/5      S 5       5       r\R"                  R%                  S/ SQ5      \R"                  R%                  SSS/5      S 5       5       r\R"                  R%                  S\" \R0                  " \
" \R2                  S 9S!S"\R4                  4S#\R4                  4S$S%S&\R4                  4S'S(\R4                  4S)\R4                  4S*\R4                  4/
5      \R0                  " \
" \R4                  SS+S,/5      / S-Q5      \R0                  " \
" S.R7                  5       5      / S/Q5      5      5      S0 5       r\R"                  R%                  S\" \R0                  " \
" \R2                  S 9S1\
" / \R2                  S 94S2\
" / \R2                  S 94/5      \R0                  " \
" \R4                  SS+S,/5      S1\
" \R4                  SS+S3/5      4S2\
" \R4                  SS,S3/5      4/5      \R0                  " \
" S.R7                  5       5      S2\
" / S4Q5      4/5      5      5      S5 5       r\R"                  R%                  S6\" \R0                  " \	" 5       S7\
" S8S 94S"\
" S8S 94S#\
" S8S 94S9\
" \S 94S:\
" \S 94S&\
" S8S 94S;\
" S8S 94S(\
" S8S 94S)\
" S8S 94S*\
" S8S 94/
5      \R0                  " \	" \R4                  S/SS+//5      S7\
" S<S,/5      4S"\
" S<S+/5      4S#\
" S<S/5      4S9\
" SS/5      4S:\
" SS/5      4S&\
" SS=/5      4S;\
" S<S+/5      4S(\
" \R4                  S>/5      4S)\
" \R4                  S?/5      4S*\
" SS=/5      4/
5      5      5      S@ 5       r\R"                  R%                  S6\" \R0                  " \	" 5       S1\	" 5       4S2\	" 5       4/5      \R0                  " \	" \R4                  S/SS+//5      S1\	" \R4                  S/SS+//5      4S2\	" \R4                  S/SS,//5      4/5      5      5      SA 5       r \R"                  R%                  S\5      SB 5       r!\R"                  R%                  S\5      SC 5       r"\R"                  R%                  SD/ SEQ5      SF 5       r#g)G    )chainN)	is_number)	DataFrameSeries)frame_transform_kernelsseries_transform_kernelsfunc)summeanminmaxstdz	args,kwdsno_args_or_kwds)id   axis_from_argsaxisaxis_from_kwdsnumeric_onlyToptional_kwdsargs_and_kwdshowaggapplyc                    [        U5      S:  a7  US:X  a1  U R                  [        R                  R	                  [
        SS95        [        X5      " U/UQ70 UD6n[        X5      " U0 UD6n[        R                  " Xg5        g )Nr   r   z>agg/apply signature mismatch - agg passes 2nd argument to funcraisesreason)	lenapplymarkerpytestmarkxfail	TypeErrorgetattrtmassert_series_equal)requestfloat_framer	   argskwdsr   resultexpecteds           M/var/www/html/env/lib/python3.13/site-packages/pandas/tests/apply/test_str.pytest_apply_with_string_funcsr/      s     4y1}KK #  	
 [&t;d;d;F{)4848H6,    argc                 T    U R                  U5      n[        X5      " 5       nX#:X  d   eg )N)r   r%   )datetime_seriesr1   r,   r-   s       r.   test_with_string_argsr4   .   s+    ""3'F,.Hr0   op)r   medianr   varc                     [        SS/SS/S.5      n[        X!5      " U 5      nU S;   a  SS0O0 n[        [        [        U 5      " U4SS	0UD6UR                  S
9n[
        R                  " X55        g )Nr            )ab)r   r7   ddofr   r   )index)r   r%   r   npcolumnsr&   r'   )r5   r   r)   r,   kwargsr-   s         r.   test_apply_np_reducerrC   5   sv     1a&1v67K[&r*F.0fa[bFB6!6v6k>Q>QH 6,r0   )absceilcoscumsumexplogsqrtsquare	transformc                    SU R                   S'   S nUS;   a  [        n[        R                  " USS9   [	        X5      " U5      n[	        [
        U5      " U 5      nS S S 5        [        R                  " WW5        g ! , (       d  f       N&= f)Ng      )r   r   )rI   rJ   F)check_stacklevel)ilocRuntimeWarningr&   assert_produces_warningr%   r@   assert_frame_equal)r)   r5   r   warnr,   r-   s         r.   test_apply_np_transformerrT   C   sy     "KTD	_		#	#D5	A *2.2r?;/	 
B
 &(+ 
B	As   (A;;
B	zseries, func, expected)dtype)r
   r   r   r   allT)anyFr   )prodr   r   r7   r6   r9   r:   )
)r
      )r   r:   )r   r   rV   rX   T)r   r9   )rY   rZ   )r   r   )r7   r   )r6   r9   za b c))r
   abc)r   c)r   r<   rV   r[   c                 *   [        U[        5      (       a  S O[        n[        R                  " USS9   U R                  U5      nS S S 5        [        U5      (       a  [        R                  " WUSS9(       d   eg WU:X  d   eg ! , (       d  f       NE= f)Nis currently using Series.*matchT)	equal_nan)	
isinstancestrFutureWarningr&   rQ   r   r   r@   iscloseseriesr	   r-   rS   r,   s        r.   test_agg_cython_table_seriesri   Y   sv    b dC((4mD		#	#D0M	ND! 
Ozz&(d;;;!!! 
O	Ns   B
BcumprodrG   rZ   )r<   abr\   c                     [        U[        5      (       a  S O[        n[        R                  " USS9   U R                  U5      nS S S 5        [        R                  " WU5        g ! , (       d  f       N&= f)Nr_   r`   rc   rd   re   r&   rQ   r   r'   rg   s        r.   &test_agg_cython_table_transform_seriesrn      sV    4 dC((4mD		#	#D0M	ND! 
O68, 
O	Ns   A%%
A3zdf, func, expectedr
   float64rW   rX   rY   g      ?g      ?g۞?g      ?c                     [        U[        5      (       a  S O[        n[        R                  " USS9   U R                  XS9nS S S 5        [        R                  " WU5        g ! , (       d  f       N&= f)Nis currently using DataFrame.*r`   r   rm   dfr	   r-   r   rS   r,   s         r.   test_agg_cython_table_frameru      s[    N dC((4mD		#	#D0P	Q( 
R 68, 
R	Qs   A##
A1c                    US;   a  UR                  S5      n[        U[        5      (       a  S O[        n[        R
                  " USS9   U R                  XS9nS S S 5        [        R                  " WU5        g ! , (       d  f       N&= f)N)rA   r   ro   rq   r`   rr   )astyperc   rd   re   r&   rQ   r   rR   rs   s         r.   %test_agg_cython_table_transform_framerx      sp    & ~??9-dC((4mD		#	#D0P	Q( 
R &(+ 
R	Qs   
A::
Bc                    US:X  a1  U R                  [        R                  R                  [        SS95        US:X  a  S/O/ n[
        R                  " UR                  S   5      nUS:X  a  [        OS nSn[        R                  " XVS9   UR                  U5      R                  " U/UQ76 nS S S 5        UR                  " US/UQ76 n[        R                  " UW5        g ! , (       d  f       N:= f)	Nngroupngroup not valid for NDFramer   fillna        r   z"SeriesGroupBy.fillna is deprecatedr`   )r    r!   r"   r#   
ValueErrorr@   onesshapere   r&   rQ   groupbyrL   r'   )	r(   string_seriesr5   r*   r   rS   msgr-   r,   s	            r.   $test_transform_groupby_kernel_seriesr      s     
X~KKZ8VW	
 (NC5D77=&&q)*D(N=D
.C		#	#D	4 ((.88CdC 
5$$R2T2F68, 
5	4s   
#C!!
C/c                 V   US:X  a1  U R                  [        R                  R                  [        SS95        US:X  a  S/O/ nUS;   a&  [
        R                  " UR                  S   5      nSnO%[
        R                  " UR                  S	   5      nS
n[        R                  " [        US9   UR                  XQS9nS S S 5        US:X  a  [        OS nSn	[        R                  " XS9   WR                  " U/UQ76 n
S S S 5        UR                  " X1/UQ76 n[        R                  " UW
5        SUR                  ;  d   eUS   R                  5       US'   [!        UR"                  R$                  5      S	:  d   eUS;   a$  [
        R                  " UR                  S   5      nO#[
        R                  " UR                  S	   5      n[        R                  " [        US9   UR                  XQS9nS S S 5        US:X  a  [        OS nSn	[        R                  " XS9   WR                  " U/UQ76 nS S S 5        UR                  " X1/UQ76 n[        R                  " UW5        g ! , (       d  f       GN= f! , (       d  f       GN= f! , (       d  f       N= f! , (       d  f       Nn= f)Nrz   r{   r   r|   r}   )r   r?   r   z5The 'axis' keyword in DataFrame.groupby is deprecatedr   z+DataFrame.groupby with axis=1 is deprecatedr`   rr   z%DataFrameGroupBy.fillna is deprecatedEA)r    r!   r"   r#   r~   r@   r   r   r&   rQ   re   r   rL   rR   rA   copyr   _mgrarrays)r(   r   r)   r5   r*   r   r   gbrS   op_msgr-   r,   gb2	expected2result2s                  r.   #test_transform_groupby_kernel_framer     s=   	X~KKZ8VW	
 (NC5D|ww{((+,Eww{((+,;		#	#M	=   1 
> (N=D4F		#	#D	7<<*T* 
8 ""23d3F&(+ k)))))"3',,.K{&&'!+++|ww{((+,ww{((+,		#	#M	=!!$!2 
>(N=D4F		#	#D	7MM",t,	 
8##B4t4G'9-7 
>	=
 
8	7 
>	= 
8	7s0   -I%)I7"J	J%
I47
J	
J
J(method)rD   shift
pct_changerG   rankc                     [        SSS/05      nUR                  U 5      n[        R                  " U 5      " U5      n[        R
                  " X#5        g )Nr   r9   )r   rL   operatormethodcallerr&   rR   )r   rt   r,   r-   s       r.   test_transform_method_namer   @  sG     
C"a>	"B\\&!F$$V,R0H&+r0   )$	itertoolsr   r   numpyr@   r!   pandas.core.dtypes.commonr   pandasr   r   pandas._testing_testingr&   pandas.tests.apply.commonr   r   r"   parametrizeparamr/   r4   rC   rT   get_cython_table_paramsro   nansplitri   rn   boolru   rx   r   r   r    r0   r.   <module>r      sq       /  !EFR 12aS"!12R&!)9:R.$/ODaY 6?K	  01- 2	 G-  DE F ?@ 01	- 2 A	- 
J g 67, 8,$ 	
""$ 266"	
 	""BFFAq!$%	
 	""7==?#		
?),Z
"[,Z
" 	
""$F2RZZ896"BJJ78	
 	""BFFAq!$%FBFFAq!#45662661a"345	
 	""7==?#x8J1K&L%M	
,--,- 	
""KY/0Y/0Y/0T*+T*+i01i01Y/0Y/06	23	
 	""{QF+,Qx()Qx()Qx()d|,-d|,-C)*a)*123}-.61c(+,	
!"F-G"F- 	
""K9ik2Xy{4KL	
 	""{QF+,I{QF&;<=9rvvqkAq6%:;<	
	,, 78- 9-" 67+. 8+.\ #ST, U,r0   