
    MhW                     v   S SK rS SKrS SKrS SKJrJrJrJrJ	r	J
r
Jr  S SKJr  S SKJr  \R"                  " SS/S9S 5       rS rS	 rS
 r\R,                  R/                  SSS/5      S 5       rS r\R,                  R/                  SS0 S 4SSS0S4S0 S4SSS0S4/5      S 5       rS rS rS rS rS rS r S  r!S! r"\R,                  R/                  S"S#S#\RF                  /S#S$\RF                  //5      S% 5       r$S& r%S' r&S( r'\R,                  R/                  S)\5      S* 5       r(S+ r)S, r*\R,                  R/                  S-\+S. /5      S/ 5       r,S0 r-\R,                  R/                  S1S2S3/5      S4 5       r.S5 r/S6 r0\R,                  R/                  S7S20 4S3S8S04S3S8S04/5      S9 5       r1S: r2\R,                  R/                  S;\" SS/\Rf                  " S S</5      S=9\" \Rh                  " SS//SS S>9S?S@94\" \Rj                  " SA\Rl                  S@9\
" SBSASC9SDSE9\" \Rh                  " SS//SAS S>9S?S@94/5      \R,                  R/                  SFSS/5      SG 5       5       r7\R,                  R/                  SHS\" \Rp                  " SA5      S?S@94SI/5      SJ 5       r9SK r:\R,                  R/                  SL\Rv                  /SM/4\Rv                  \Rx                  /SMSN/4\Rz                  " \Rv                  /5      SM/4\Rz                  " \Rv                  \Rx                  /5      SMSN/4/5      \R,                  R/                  SOS20 /S3S8S0/S3S8S0//5      SP 5       5       r>\R,                  R/                  SQSR\Rv                  0\Rv                  \Rx                  SS.\" SR\Rv                  05      \" \Rv                  \Rx                  SS.5      /5      \R,                  R/                  SOS20 /S3S8S0/S3S8S0//5      ST 5       5       r?\R,                  R/                  SL\R                  /SU/4\R                  \R                  /SVSU/4\Rz                  " \R                  /5      SU/4\Rz                  " \R                  \R                  /5      SVSU/4/5      SW 5       rB\R,                  R/                  SXSY /\" SZ/ S[Q05      4S\ /\" S]/SZ/S=94/5      S^ 5       rC\R,                  R/                  SQSR\R                  0\R                  \R                  SS.\" SR\R                  05      \" \R                  \R                  SS.5      /5      S_ 5       rE\R,                  R/                  SXSS` 0\" / S[Q\R                  " S/Sa-  \G" Sa5      /5      S=94SSb 0\" S]/S/S=94/5      Sc 5       rHSd rISe rJSf rKg)g    N)	DataFrameIndex
MultiIndexSeriesconcat
date_rangetimedelta_range)series_transform_kernelsFcompat)paramsc                     U R                   $ N)param)requests    V/var/www/html/env/lib/python3.13/site-packages/pandas/tests/apply/test_series_apply.pyby_rowr      s    ==    c                    ^  [        [        SSSS95      nU 4S jnUR                  UT S9nUR                  S 5      n[        R
                  " X45        [        / SQ5      n[        R
                  " X45        g )	Nz	1 day 1 s   h)periodsfreqc                 f   > T(       a  U R                  5       $ U R                  R                  5       $ r   )total_secondsdtxr   s    r   f(test_series_map_box_timedelta.<locals>.f   s$    $*q D0B0B0DDr   r   c                 "    U R                  5       $ r   )r   r   s    r   <lambda>/test_series_map_box_timedelta.<locals>.<lambda>    s    !2r   )g    @g    @g    @)r   r	   applymaptmassert_series_equal)r   serr   resultexpecteds   `    r   test_series_map_box_timedeltar,      sf    
acB
CCE YYqY(Fww23H6,12H6,r   c           	      <   U R                  [        R                  US9n[        R                  " SS9   [        R                  " U 5      nS S S 5        [        R
                  " UW5        U R                  [        R                  US9n[        R                  " U 5      n[        R
                  " X#5        [        [        S[        / SS9S9nUR                  S US9n[        R
                  " XE5        XELd   eUR                  UR                  L d   eUR                  UR                  :X  d   eUR                  UR                  :X  d   e[        / S	Q[        R                  S
9nUR                  S US9n[        R
                  " XE5        g ! , (       d  f       GNI= f)Nr    ignoreallfoobarname)dtyper4   indexc                     U $ r    r"   s    r   r#   test_apply.<locals>.<lambda>4       1r         r   r6   r5   c                     U $ r   r8   r"   s    r   r#   r9   ?   r:   r   )r%   npsqrterrstater'   r(   expr   objectr   r6   r5   r4   float64)datetime_seriesr   r*   r+   srss         r   
test_applyrI   '   sA   ""2776":F		"77?+ 
#68, ""266&"9Fvvo&H6, 	V%uRe/DEA	
V	,B1! ;;77bhh77bhh66RWW 	Ybjj1A	
V	,B1!/ 
#	"s   F
Fc                      [        SS/5      n S nU R                  USS9nU R                  U5      n[        R                  " X#5        g )Nr<   r=   c                     X S-   4$ Nr<   r8   r"   s    r   r   3test_apply_map_same_length_inference_bug.<locals>.fF   s    q5zr   r   r    )r   r%   r&   r'   r(   )rG   r   r*   r+   s       r   (test_apply_map_same_length_inference_bugrN   C   sC    1vA WWQxW(FuuQxH6,r   convert_dtypeTc                     [        [        R                  R                  S5      R	                  S5      5      nS n[
        R                  " [        5         UR                  X SS9  S S S 5        g ! , (       d  f       g = f)Nr=   
   c                 2    U S:  a  U $ [         R                  $ )Nr   )r@   nanr"   s    r   func1test_apply_convert_dtype_deprecated.<locals>.funcR   s    Eq%rvv%r   r   )rO   r   )	r   r@   randomdefault_rngstandard_normalr'   assert_produces_warningFutureWarningr%   )rO   r)   rT   s      r   #test_apply_convert_dtype_deprecatedr[   N   sW    
&&q)99"=
>C& 
	#	#M	2		$H	E 
3	2	2s   A//
A=c                      [        S/5      n U R                  [        R                  SS9nUS   SS/:X  d   e[	        US   [
        5      (       d   eg )Nzfoo,bar),argsr   r1   r2   )r   r%   strsplit
isinstancelist)rG   r*   s     r   test_apply_argsrd   Y   sQ    	{AWWSYYVW,F!9&&&fQi&&&&r   zargs, kwargs, incrementr8   ar<   )r=   r       )r<   cr=      c                     SS jn[        SS/5      nSn[        R                  " [        US9   UR                  " US/U Q70 UD6nS S S 5        XB-   n[        R
                  " WU5        g ! , (       d  f       N*= f)Nr   c                 "    X-   SU-  -   SU-  -   $ )NrQ   d   r8   )r   re   brg   s       r   r   test_agg_args.<locals>.fg   s    urAv~a''r   r<   r=   zhin Series.agg cannot aggregate and has been deprecated. Use Series.transform to keep behavior unchanged.match)r   r   r   )r   r'   rY   rZ   aggr(   )r_   kwargs	incrementr   rG   msgr*   r+   s           r   test_agg_argsrt   a   sq    ( 	1vA	;  
	#	#M	=q!-d-f- 
>}H68, 
>	=s   A**
A8c                     [        / SQ5      n SS jnSS jnSn[        R                  " [        US9   U R	                  USSSS	9  S S S 5        [        R                  " [        US9   U R	                  X/SSSS	9  S S S 5        [        R                  " [        US9   U R	                  XS
.SSSS	9  S S S 5        g ! , (       d  f       Nz= f! , (       d  f       NV= f! , (       d  f       g = f)Nr;   r   c                     X-   U-   $ r   r8   )r   re   rg   s      r   foo1.test_agg_mapping_func_deprecated.<locals>.foo1y       uqyr   c                     X-   U-   $ r   r8   )r   rl   rg   s      r   foo2.test_agg_mapping_func_deprecated.<locals>.foo2|   ry   r   z+using .+ in Series.agg cannot aggregate andrn   r      )rg   )re   rl   )r<   r   )r=   r   )r   r'   rY   rZ   rp   )rG   rw   r{   rs   s       r    test_agg_mapping_func_deprecatedr~   u   s    yA 8C		#	#M	=	dAqA 
>		#	#M	=	tlAqA& 
>		#	#M	=	D$aa0 
>	=	 
>	=	=	=	=	=s#   B:'CC:
C
C
C*c                 4   [        [        SSS95      nS nU (       d4  Sn[        R                  " [        US9   UR                  X S9  S S S 5        g UR                  X S9nUR                  U5      n[        R                  " XE5        g ! , (       d  f       g = f)Nz1/1/2000rQ   r   c                 H    U R                   U R                  U R                  4$ r   )hourdaymonthr"   s    r   rT   2test_series_apply_map_box_timestamps.<locals>.func   s    qww''r   z&Series' object has no attribute 'hour'rn   r    )	r   r   pytestraisesAttributeErrorr%   r&   r'   r(   )r   r)   rT   rs   r*   r+   s         r   $test_series_apply_map_box_timestampsr      s}    
J3
4C( 6]]>5IIdI* 6YYtY+Fwwt}H6, 65s   B		
Bc                      [         R                  " S5      [         R                  " S5      /n [        U SS9nUR                  S:X  d   eUR	                  S SS9n[        S	S
/5      n[
        R                  " X#5        g )N
2011-01-01
2011-01-02zM8[ns]r5   zdatetime64[ns]c                 d    [        U 5      R                   SU R                   SU R                   3$ N_type__name__r   tzr"   s    r   r#   %test_apply_box_dt64.<locals>.<lambda>   (    a!1!1 2!AEE7!ADD6Br   r   r    Timestamp_1_NoneTimestamp_2_Nonepd	Timestampr   r5   r%   r'   r(   valsr)   resrC   s       r   test_apply_box_dt64r      so    LL&\(BCD
X
&C99((((
))B8)
TC
$&89
:C3$r   c                      [         R                  " SSS9[         R                  " SSS9/n [        U SS9nUR                  S:X  d   eUR	                  S S	S
9n[        SS/5      n[
        R                  " X#5        g )Nr   z
US/Eastern)r   r   zM8[ns, US/Eastern]r   zdatetime64[ns, US/Eastern]c                 d    [        U 5      R                   SU R                   SU R                   3$ r   r   r"   s    r   r#   'test_apply_box_dt64tz.<locals>.<lambda>   r   r   r   r    zTimestamp_1_US/EasternzTimestamp_2_US/Easternr   r   s       r   test_apply_box_dt64tzr      sy    
\l3
\l3D 1
2C994444
))B8)
TC
*,DE
FC3$r   c                      [         R                  " S5      [         R                  " S5      /n [        U 5      nUR                  S:X  d   eUR	                  S SS9n[        SS/5      n[
        R                  " X#5        g )	Nz1 daysz2 daysztimedelta64[ns]c                 J    [        U 5      R                   SU R                   3$ r   )r   r   daysr"   s    r   r#   %test_apply_box_td64.<locals>.<lambda>   s    a!1!1 2!AFF8<r   r   r    Timedelta_1Timedelta_2)r   	Timedeltar   r5   r%   r'   r(   r   s       r   test_apply_box_td64r      sj    LL"BLL$:;D
,C99))))
))<X)
NC
-/
0C3$r   c                      [         R                  " SSS9[         R                  " SSS9/n [        U 5      nUR                  S:X  d   eUR	                  S SS9n[        S	S	/5      n[
        R                  " X#5        g )
Nr   Mr   r   z	Period[M]c                 J    [        U 5      R                   SU R                   3$ r   )r   r   freqstrr"   s    r   r#   'test_apply_box_period.<locals>.<lambda>   s    a!1!1 2!AII;?r   r   r    Period_M)r   Periodr   r5   r%   r'   r(   r   s       r   test_apply_box_periodr      sm    IIl-ryyC/PQD
,C99###
))?)
QC
*j)
*C3$r   c                    ^  [        SSSS9R                  S5      n[        USS9nUR                  S T S	9n[        SS
SS9R                  S5      n[        USS9n[        R
                  " X55        UR                  U 4S jT S	9n[        [        [        S5      5      S/-   ST (       a  SOSS9n[        R
                  " X55        U 4S jnUR                  UT S	9nT (       a%  [        S/S-  SS9n[        R
                  " X55        g US:X  d   eg )Nr   r   r   r   z
Asia/TokyoXXr3   c                 D    U [         R                  R                  5       -   $ r   )r   offsetsDayr"   s    r   r#   'test_apply_datetimetz.<locals>.<lambda>   s    q2::>>#33r   r    z
2011-01-03c                 V   > T(       a  U R                   $ U R                  R                   $ r   )r   r   r   s    r   r#   r      s    qvv>QTTYY>r      r   int64int32r4   r5   c                 z   > T(       a  [        U R                  5      $ [        U R                  R                  5      $ r   )r`   r   r   r   s    r   r    test_apply_datetimetz.<locals>.f   s$    "s144y4ADDGG4r      )r   tz_localizer   r%   r'   r(   rc   range)r   valuesrG   r*   
exp_valuesrC   r   s   `      r   test_apply_datetimetzr      s   l=II,WFvD!AWW3FWCFL,SAMMJ $
'C6'WW>vWNF
eBiA3&TFPW
XC6'5 WWQvW&Fl^b(t4
v+%%%r   c                     [         R                  " [        S5      [        S5      SS9n[        US[        S5      S9nU (       dM  Sn[        R
                  " [        US	9   UR                  S
 U S9  S S S 5        UR                  S U S9S:X  d   eg UR                  S U S9n[         R                  " [        S5      [        S5      SS9n[        US[        S5      S9n[        R                  " XV5        [        R                  " UR                  UR                  5        UR                  S 5      n[        S/S-  S[        S5      S9n[        R                  " XV5        U(       d  UR                  [        :X  d   eg g ! , (       d  f       GN= f)NABBABCDDCBAT)
categoriesorderedr   abcdefg)r4   r6   z&Series' object has no attribute 'lowerrn   c                 "    U R                  5       $ r   lowerr"   s    r   r#   (test_apply_categorical.<locals>.<lambda>   s
    	r   r    c                     gNAr8   r"   s    r   r#   r      s    3r   r   c                 "    U R                  5       $ r   r   r"   s    r   r#   r      s
    r   abbabcddcbac                     gr   r8   r"   s    r   r#   r      s    r      )r   Categoricalrc   r   r   r   r   r%   r'   r(   assert_categorical_equalr   r5   rD   )r   using_infer_stringr   r)   rs   r*   rC   s          r   test_apply_categoricalr      s6   ^^DOVdSF
d$y/
:C6]]>5II)&I9 6yyvy6#===YY*6Y:F ^^DOVdSF
d$y/
:C6'szz:YY}%F
#T)_
=C6');6<<6!XXAX!# 65s   E>>
Fseriesz1-1z1-2c                    [        U SS9nU(       d6  Sn[        R                  " [        US9   UR	                  S US9  S S S 5        g UR	                  S US9nUR                  [        5      n[        SS[        R                  /SS9nUR                  [        5      n[        R                  " XE5        g ! , (       d  f       g = f)	Ncategoryr   z('Series' object has no attribute 'split'rn   c                 *    U R                  S5      S   $ N-r   ra   r"   s    r   r#   8test_apply_categorical_with_nan_values.<locals>.<lambda>  s    aggcl1or   r    c                 *    U R                  S5      S   $ r   r   r"   s    r   r#   r     s    qwws|Ar   1)r   r   r   r   r%   astyperD   r@   rS   r'   r(   )r   r   rG   rs   r*   r+   s         r   &test_apply_categorical_with_nan_valuesr      s     	vZ(A8]]>5GG-fG= 6WW.vW>F]]6"FsC(
;Hv&H6, 65s   B55
Cc                     [        / [        SSS9[        S9nUR                  S U S9n[        R
                  " X!5        g )Nz
2018-01-01r   )startr   r>   c                     U $ r   r8   r"   s    r   r#   Etest_apply_empty_integer_series_with_datetime_index.<locals>.<lambda>  s    qr   r    )r   r   intr%   r'   r(   )r   rG   r*   s      r   3test_apply_empty_integer_series_with_datetime_indexr     s7    r,B#NAWW[W0F6%r   c                      [        [        R                  " / SQ5      S/S9n [        / SQS/S9nS nUS   R                  X 4S9n[	        / SQSS	S
9n[
        R                  " X45        g )N)r<   r=   r   r}      Numbers)columns)r=   r   r=   r<   r=   Indicesc                 *    US   R                   U    nU$ )Nr   )iloc)	targetRowtargetDFvals      r   retrieve+test_apply_dataframe_iloc.<locals>.retrieve  s    y!&&y1
r   r^   )r   r}   r   r=   r   uint64r   )r   r@   r   r%   r   r'   r(   )uintDFindexDFr   r*   r+   s        r   test_apply_dataframe_ilocr     sd    ryy1I;GF)=G Y%%hY%?FoIXFH6,r   c                    [         R                  " SS9   [         R                  " U 5      n[         R                  " U 5      nU R	                  [         R                  US9nUR                  5       n[        R                  " XE5        U R	                  [         R                  /US9nUR                  5       R                  5       nS/Ul	        [        R                  " XE5        U R	                  S/US9n[        R                  " XE5        [        X#/SS9nSS/Ul	        U R	                  [         R                  [         R                  /US9n[        R                  " XE5        [        X#/SS9nSS	/Ul	        UR                  5       R                  S
5      nU R	                  [         R                  [         R                  S.US9n[        R                  " UR                  U5      U5        S S S 5        g ! , (       d  f       g = f)Nr.   r/   r    rA   r<   axisabsoluter1   r2   r   )r1   r2   )r@   rB   rA   absr%   copyr'   r(   to_framer   assert_frame_equalr   unstackrenamereindex_like)string_seriesr   f_sqrtf_absr*   r+   s         r   test_transformr    s    
	"'}% $$RWWV$<;;=
v0 $$bggYv$>??$))+"8
f/$$fXf$=
f/
 6/2"J/$$bggrvv%6v$F
f/ 6/2!5>##%,,X6$$RWWRVV%DV$T
v228<hGA 
#	"	"s   GG11
G?opc                    U S;   a/  UR                  [        R                  R                  U  S3S95        [	        S[
        /-  5      nU S;   a	  [        nSnO[        nSR                  SS	/5      n[        R                  " X4S
9   UR                  U S/5        S S S 5        [        R                  " X4S
9   UR                  U SS.5        S S S 5        [        R                  " X4S
9   UR                  U /S/S.5        S S S 5        [        R                  " X4S
9   UR                  U S/U /S.5        S S S 5        g ! , (       d  f       N= f! , (       d  f       N= f! , (       d  f       Ne= f! , (       d  f       g = f)N)ffillbfillpadbackfillshiftz is successful on any dtype)reasonr   )fillnangroupTransform function failed|z4not supported between instances of 'type' and 'type'zunsupported operand typern   r  r   B)applymarkerr   markxfailr   rD   
ValueError	TypeErrorjoinr   	transform)r  r   r)   errorrs   s        r   test_transform_partial_failurer%  E  s;    
;;KKt+F%GH	

 fX
C	!!)hhF*
 
u	(r7m$ 
) 
u	(BW-. 
) 
u	(RDy12 
) 
u	(RM56 
)	( 
)	( 
)	( 
)	( 
)	(s0   D:5E&EE-:
E
E
E*-
E;c                  t   S n S n[        S[        /-  5      nSn[        R                  " [        US9   UR                  X/5        S S S 5        [        R                  " [        US9   UR                  XS.5        S S S 5        [        R                  " [        US9   UR                  U/U /S.5        S S S 5        [        R                  " [        US9   UR                  X/U /S.5        S S S 5        g ! , (       d  f       N= f! , (       d  f       N= f! , (       d  f       Ni= f! , (       d  f       g = f)Nc                     U $ r   r8   r"   s    r   noop7test_transform_partial_failure_valueerror.<locals>.noopk  s    r   c                     [         er   )r   )r   s    r   
raising_op=test_transform_partial_failure_valueerror.<locals>.raising_opn  s    r   r   r  rn   r  )r   rD   r   r   r   r#  )r(  r+  r)   rs   s       r   )test_transform_partial_failure_valueerrorr-  i  s     fX
C
%C	z	-t() 
. 
z	-J23 
. 
z	-ZLv67 
. 
z	-T.dV<= 
.	- 
.	- 
.	- 
.	- 
.	-s/   C6)DDD)6
D
D
D&)
D7c                     [        [        S5      SSS9n U R                  SS/5      n[        SS/SS/SS	9n[        R                  " X5        U R                  S
S05      n[        S/S
/SS	9n[        R                  " X5        g )N   r   r   )r5   r4   minmaxr   r   r6   r4   r1   )r   r   rp   r'   r(   rG   r*   r+   s      r   	test_demor4    s|    uQxwX6AUUE5>"Fq!fUENBH6,UUE5>"Fqc%x8H6,r   rT   c                     [        U 5      $ r   )r`   r"   s    r   r#   r#     s    Qr   c                     U R                  XS9nU(       a(  U R                  U5      n[        R                  " X45        g U[	        U 5      :X  d   eg )Nr    )r%   r&   r'   r(   r`   )r  rT   r   r*   r+   s        r   (test_apply_map_evaluate_lambdas_the_samer7    sK        5F $$T*
v0]++++r   c                    [         R                  " [        5         U R                  S 5      nS S S 5        [	        W[
        5      (       a  [        U5      [        U 5      :X  d   e[         R                  " [        5         U R                  [        5      nS S S 5        [	        U[
        5      (       a  [        U5      [        U 5      :X  d   eg ! , (       d  f       N= f! , (       d  f       NO= f)Nc                     [        U 5      $ r   )r   r"   s    r   r#   +test_agg_evaluate_lambdas.<locals>.<lambda>  s    T!Wr   )r'   rY   rZ   rp   rb   r   lenr   )r  r*   s     r   test_agg_evaluate_lambdasr<    s     
	#	#M	2""#45 
3ff%%#f+]9K*KKK		#	#M	2""4( 
3ff%%#f+]9K*KKK*K 
3	2 
3	2s   C?C
C
C,op_namerp   r%   c                    SnUS:X  a  [         OS n[        R                  " X2S9   [        X5      " S 5      nS S S 5        [	        X S-  S.5      n[        R
                  " WU5        [        R                  " [         US9   U R                  S 5      nS S S 5        [        R
                  " XE5        g ! , (       d  f       N= f! , (       d  f       N6= f)Nzcannot aggregaterp   rn   c                 "    [        X S-  /SS/S9$ Nr=   r   x^2r6   r   r"   s    r   r#   )test_with_nested_series.<locals>.<lambda>  s    faAYsEl;r   r=   )r   rA  c                 "    [        X S-  /SS/S9$ r@  rC  r"   s    r   r#   rD    s    vqQ$iU|/Tr   )rZ   r'   rY   getattrr   r  rp   )rF   r=  rs   warningr*   r+   s         r   test_with_nested_seriesrH    s     C&%/mTG		#	#G	72;
 
8
 7IJKH&(+		#	#M	= $$%TU 
>&+ 
8	7 
>	=s   B2 C2
C 
Cc                     U R                  5       nU R                  SSSSS SS SS	.5      n[        R                  " X!5        g )
Ncountmeanstdr0  c                 $    U R                  S5      $ )Ng      ?quantiler"   s    r   r#   )test_replicate_describe.<locals>.<lambda>      QZZ-r   medianc                 $    U R                  S5      $ )Ng      ?rN  r"   s    r   r#   rP    rQ  r   r1  )rJ  rK  rL  r0  z25%z50%z75%r1  )describer%   r'   r(   )r  r+   r*   s      r   test_replicate_describerU    sM    %%'H  --		
F 6,r   c                     U R                  SS/5      n[        U R                  5       U R                  5       /SS/U R                  S9n[
        R                  " X5        g )NsumrK  r3   )rp   r   rW  rK  r4   r'   r(   )r  r*   r+   s      r   test_reducerX    sZ    v/F				m0023	H
 6,r   z	how, kwdsr   c                 0   [        / SQ5      n[        X 5      " S	0 UD6nUR                  nX4:X  d   e[        X 5      " / SQ40 UD6n[        SSSS.5      n[        R                  " X45        [        X 5      " SSSS.40 UD6n[        R                  " X45        g )
N)r<   r=   Nsize)rZ  rJ  rK        @       @g      ?rJ  rK  )rZ  )r   rF  rZ  r'   r(   )howkwdsrG   r*   r+   s        r   test_non_callable_aggregatesr_    s     	|A Q_,t,FvvH Q_6?$?FsS#>?H6,Q_fwOXSWXF6,r   c                     [        S/S-  5      nUR                  SS S /U S9n[        / SQ/ SQS	9n[        R                  " X#5        g )
Nr}   r   rW  c                 "    U R                  5       $ r   rW  r"   s    r   r#   3test_series_apply_no_suffix_index.<locals>.<lambda>  s
    quuwr   c                 "    U R                  5       $ r   rb  r"   s    r   r#   rc    s
    !%%'r   r    )   re  re  )rW  <lambda>rf  rB  )r   r%   r'   r(   )r   rG   r*   r+   s       r   !test_series_apply_no_suffix_indexrg    sG    sQwAWWe.0AB6WRFl*IJH6,r   zdti,expl    ,b/ rB  r  r   r   rQ   
2020-01-01r   tsr2  awarec                     U(       a  U R                  S5      R                  nOU R                  n[        U5      R                  S 5      n[        R
                  " XA5        g )NUTCc                     [        SS/5      $ )Nr<   r=   rC  r"   s    r   r#   Ctest_apply_series_on_date_time_index_aware_series.<locals>.<lambda>  s    61a&>r   )r   r6   r   r%   r'   r  )dtirC   rj  r6   r*   s        r   1test_apply_series_on_date_time_index_aware_seriesrp    sG    * &,,		E]  !9:F&&r   zby_row, expected)Fr<   c           	          [        [        R                  " S[        R                  S9[	        SSSS9S9n[        UR
                  5      R                  S U S9n[        R                  " X15        g )	NrQ   r   rh  rl  )r   r   rB  c                     grL   r8   r"   s    r   r#   Ctest_apply_scalar_on_date_time_index_aware_series.<locals>.<lambda>  s    !r   r    )	r   r@   arangerE   r   r6   r%   r'   assert_equal)r   r+   r   r*   s       r   1test_apply_scalar_on_date_time_index_aware_seriesrv    sX     
		"BJJ're<F FLL!''F'CFOOF%r   c                    SS/n[         R                  " U5      n[        U5      R                  [         R                  U S9n[        R
                  " [        U5      U5        S[        R                  [         R                  [         R                  /n[         R                  " U5      n[        U5      nUR                  [         R                  U S9n[        R
                  " [        U5      U5        g )Nz00:00:01z00:00:02r    )	r   to_timedeltar   r%   r'   r(   r@   rS   NaT)r   list_of_valid_stringsre   rl   list_of_stringsr)   s         r   test_apply_to_timedeltar|  #  s    '4
-.A$%++BOOF+KA6!9a(!266266266:O
(A

!C		"//&	1A6!9a(r   z
ops, namesrW  rK  zhow, kwargsc           
      p   [        [        X!5       VVs0 s H  u  pVXV" U 5      _M     snn5      nSUl        US:X  a  [        OS nSSR	                  U5       S3n	[
        R                  " XS9   [        X5      " U40 UD6n
S S S 5        [
        R                  " W
U5        g s  snnf ! , (       d  f       N,= f)Nr   rp   zusing Series.[r  ]rn   )	r   zipr4   rZ   r"  r'   rY   rF  r(   )r  opsnamesr]  rq   r4   r  r+   warnrs   r*   s              r   test_apply_listlike_reducerr  1  s     s5O84tR..OPHHM5L=dD388E?+1
-C		#	#D	4,S;F; 
568, P 
5	4s   B!
.B''
B5r  r   r  c           
      h   [        UR                  5        VVs0 s H  u  pVXV" U 5      _M     snn5      nU R                  Ul        US:X  a  [        OS nSn	[        R
                  " XS9   [        X5      " U40 UD6n
S S S 5        [        R                  " W
U5        g s  snnf ! , (       d  f       N,= f)Nrp   zusing Series.[sum|mean]rn   )r   itemsr4   rZ   r'   rY   rF  r(   )r  r  r]  rq   r   r4   r  r+   r  rs   r*   s              r   test_apply_dictlike_reducerr  I  s     syy{K{84tR..{KLH!&&HM5L=dD
#C		#	#D	4,S;F; 
568, L 
5	4s   B
*B##
B1rA   r  c           	          [         R                  " SS9   [        U Vs/ s H
  oD" U 5      PM     snSS9nX%l        U R	                  XS9n[
        R                  " Xe5        S S S 5        g s  snf ! , (       d  f       g = f)Nr.   r/   r<   r  r    )r@   rB   r   r   r%   r'   r  )r  r  r  r   r  r+   r*   s          r   test_apply_listlike_transformerr  a  sk     
	"s;s2m,s;!D $$S$8
f/	 
#	"; 
#	"s   
A/A*
1A/*A//
A=zops, expectedc                     U $ r   r8   r"   s    r   r#   r#   v  s    Ar   rf  r;   c                 "    U R                  5       $ r   rb  r"   s    r   r#   r#   w  s
    AEEGr   r/  c                 h    [        / SQ5      nUR                  XS9n[        R                  " XA5        g Nr;   r    r   r%   r'   ru  )r  r+   r   r)   r*   s        r   test_apply_listlike_lambdar  s  s*     
CYYsY*FOOF%r   c                 @   [         R                  " SS9   [        UR                  5        VVs0 s H  u  p4X4" U 5      _M     snn5      nU R                  Ul        U R                  XS9n[        R                  " Xe5        S S S 5        g s  snnf ! , (       d  f       g = f)Nr.   r/   r    )r@   rB   r   r  r4   r%   r'   r(   )r  r  r   r4   r  r+   r*   s          r   test_apply_dictlike_transformerr    s}     
	"399;O;xt4M!22;OP%**$$S$8
v0	 
#	"O 
#	"s   BB	>B	B
Bc                     U $ r   r8   r"   s    r   r#   r#     s    Ar   r   c                 "    U R                  5       $ r   rb  r"   s    r   r#   r#     s
    r   c                 h    [        / SQ5      nUR                  XS9n[        R                  " XB5        g r  r  )r  r   r+   r)   r*   s        r   test_apply_dictlike_lambdar    s*     
CYYsY*FOOF%r   c           	      |   [        S[        S5      0[        [        S5      SS95      nUR                  R	                  S 5      n[        S[
        R                  [
        R                  /SS[
        R                  // SQ/[        [        S5      SS9[        [        S5      SS9S	9n[        R                  " X#5        g )
Nr   r   r3   c           	      Z    [        [        U S-   5      [        [        U S-   5      SS95      $ )Nr<   yr3   )r   r   r   r"   s    r   r#   0test_apply_retains_column_name.<locals>.<lambda>  s#    &q1uuU1q5\PS7T"Ur                 ?)r  r  r\  r  )r   r6   )	r   r   r   r   r%   r@   rS   r'   r  )r   dfr*   r+   s       r   test_apply_retains_column_namer    s    	Cq?E%($=	>BTTZZUVF
rvvrvv	c266 2ODeAhS)E!H3'H
 &+r   c                      [        SS[        // SQS9n U R                  [        5      n[        [        [
        [        // SQS9n[        R                  " X5        g )Nr   string)re   rl   rg   rB  )r   floatr%   r   r   r`   r'   r(   r3  s      r   test_apply_typer    sC    8U#?;AWWT]FsC&o>H6,r   c                      [        / SQ/ SQ/5      n U R                  S 5      n[        SS/SS/SS	/[        R                  S
/S.5      n[
        R                  " X5        g )Nr;   )r}   r   r/  r   c                     [        U 5      $ r   rC  r"   s    r   r#   6test_series_apply_unpack_nested_data.<locals>.<lambda>  s    r   r  g      @r\  g      @r[  g      @r   )r   r<   r=   r   )r   r%   r   r@   rS   r'   r  )r)   r*   r+   s      r   $test_series_apply_unpack_nested_datar    sV    
)\*
+CYY*+Fc3ZS#JC:266ST+VWH&+r   )Lnumpyr@   r   pandasr   r   r   r   r   r   r   r	   pandas._testing_testingr'   pandas.tests.apply.commonr
   fixturer   r,   rI   rN   r  parametrizer[   rd   rt   r~   r   r   r   r   r   r   r   rS   r   r   r   r  r%  r-  r4  r`   r7  r<  rH  rU  rX  r_  rg  DatetimeIndexrepeatrt  rE   rp  onesrv  r|  rW  rK  arrayr  r  rA   r  r  r  rC   r  from_arraysr   r  r  r  r  r8   r   r   <module>r     s         > x() *- "8- 4-8F 9F' "a[2Qx#fb"%5sAh7LM-	- 1&-$%	%%%&6Y4 UE266$:UE266<R#ST- U- &
-#HL 78 7 9 7F>0
- #'7!89	, :	,
L UG$45, 6,$-$- R[7Xx01Gh=N3OP-	-&-  Aq6!1!11k2B!CDbii!Q!!4GD	

 		"BJJ/ r:
 bii!Q"15WE	
" 4-0' 1#$' (F2772;g$FGT&&) 
&&E7
&&"''	UFO,	266(	eW%	266277#	$ufo6	 R[7Xx01Gh=N3OP-	- 		bffff277#RVV}RVV"''*+	 R[7Xx01Gh=N3OP-	- 
''VH
&&"''	Z01	277)	vh'	266277#	$z6&:;	00 
	:y"9:;
	faS=>&& 		bggggBFF#RWW~RWW266*+	11  +9J$:$:SEAIuQx;P$QR	
  	!61#cU#;<	&	&	,-,r   