
    OwgW                        d dl Zd dlZd dlZd dlmZmZmZmZm	Z	m
Z
mZ d dlmZ d dlmZ  ej"                  ddg      d        Zd Zd	 Zd
 Zej,                  j/                  dddg      d        Zd Zej,                  j/                  ddi d fdddidfdi dfdddidfg      d        Zd Zd Zd Zd Zd Zd Z d  Z!d! Z"ej,                  j/                  d"d#d#ejF                  gd#d$ejF                  gg      d%        Z$d& Z%d' Z&d( Z'ej,                  j/                  d)e      d*        Z(d+ Z)d, Z*ej,                  j/                  d-e+d. g      d/        Z,d0 Z-ej,                  j/                  d1d2d3g      d4        Z.d5 Z/d6 Z0ej,                  j/                  d7d2i fd3d8difd3d8difg      d9        Z1d: Z2ej,                  j/                  d; eddg ejf                  d d<g      =       e ejh                  ddggdd >      d?@      f e ejj                  dAejl                  @       e
dBdAC      dDE       e ejh                  ddggdAd >      d?@      fg      ej,                  j/                  dFddg      dG               Z7ej,                  j/                  dHd e ejp                  dA      d?@      fdIg      dJ        Z9dK Z:ej,                  j/                  dLejv                  gdMgfejv                  ejx                  gdMdNgf ejz                  ejv                  g      dMgf ejz                  ejv                  ejx                  g      dMdNgfg      ej,                  j/                  dOd2i gd3d8digd3d8digg      dP               Z>ej,                  j/                  dQdRejv                  iejv                  ejx                  dS edRejv                  i       eejv                  ejx                  dS      g      ej,                  j/                  dOd2i gd3d8digd3d8digg      dT               Z?ej,                  j/                  dLej                  gdUgfej                  ej                  gdVdUgf ejz                  ej                  g      dUgf ejz                  ej                  ej                  g      dVdUgfg      dW        ZBej,                  j/                  dXdY g edZg d[i      fd\ g ed]gdZg=      fg      d^        ZCej,                  j/                  dQdRej                  iej                  ej                  dS edRej                  i       eej                  ej                  dS      g      d_        ZEej,                  j/                  dXdd` i eg d[ ej                  dgdaz   eGda      g      =      fddb i ed]gdg=      fg      dc        ZHdd ZIde ZJdf ZKy)g    N)	DataFrameIndex
MultiIndexSeriesconcat
date_rangetimedelta_range)series_transform_kernelsFcompat)paramsc                     | j                   S N)param)requests    [/var/www/horilla/myenv/lib/python3.12/site-packages/pandas/tests/apply/test_series_apply.pyby_rowr      s    ==    c                      t        t        ddd            } fd}|j                  |       }|j                  d       }t	        j
                  ||       t        g d      }t	        j
                  ||       y )	Nz	1 day 1 s   h)periodsfreqc                 \    r| j                         S | j                  j                         S r   )total_secondsdtxr   s    r   fz(test_series_map_box_timedelta.<locals>.f   s$    $*q D0B0B0DDr   r   c                 "    | j                         S r   )r   r   s    r   <lambda>z/test_series_map_box_timedelta.<locals>.<lambda>    s    !2 r   )g    @g    @g    @)r   r	   applymaptmassert_series_equal)r   serr   resultexpecteds   `    r   test_series_map_box_timedeltar*      sj    
acB
CCE YYqY(Fww23H68,12H68,r   c                 T   | j                  t        j                  |      }t        j                  d      5  t        j                  |       }d d d        t	        j
                  |       | j                  t        j                  |      }t        j                  |       }t	        j
                  ||       t        t        dt        g d            }|j                  d |      }t	        j
                  ||       ||usJ |j                  |j                  u sJ |j                  |j                  k(  sJ |j                  |j                  k(  sJ t        g d	t        j                  
      }|j                  d |      }t	        j
                  ||       y # 1 sw Y   QxY w)Nr   ignoreallfoobarname)dtyper2   indexc                     | S r    r!   s    r   r"   ztest_apply.<locals>.<lambda>4       1 r         r   r4   r3   c                     | S r   r6   r!   s    r   r"   ztest_apply.<locals>.<lambda>?   r7   r   )r#   npsqrterrstater%   r&   expr   objectr   r4   r3   r2   float64)datetime_seriesr   r(   r)   srss         r   
test_applyrF   '   sP   ""2776":F		" ,77?+,68, ""266&"9Fvvo&H68, 	V%uRe/DEA	
V	,B1b! B;;77bhh77bhh66RWW 	Ybjj1A	
V	,B1b!/, ,s   FF'c                      t        ddg      } d }| j                  |d      }| j                  |      }t        j                  ||       y )Nr9   r:   c                     | | dz   fS Nr9   r6   r!   s    r   r   z3test_apply_map_same_length_inference_bug.<locals>.fF   s    1q5zr   r   r   )r   r#   r$   r%   r&   )rD   r   r(   r)   s       r   (test_apply_map_same_length_inference_bugrJ   C   sE    1vA WWQxW(FuuQxH68,r   convert_dtypeTc                     t        t        j                  j                  d      j	                  d            }d }t        j                  t              5  |j                  || d       d d d        y # 1 sw Y   y xY w)Nr:   
   c                 0    | dkD  r| S t         j                  S )Nr   )r=   nanr!   s    r   funcz1test_apply_convert_dtype_deprecated.<locals>.funcR   s    Eq%rvv%r   r   )rK   r   )	r   r=   randomdefault_rngstandard_normalr%   assert_produces_warningFutureWarningr#   )rK   r'   rP   s      r   #test_apply_convert_dtype_deprecatedrV   N   sf    
&&q)99"=
>C& 
	#	#M	2 F		$mH	EF F Fs   A22A;c                      t        dg      } | j                  t        j                  d      }|d   ddgk(  sJ t	        |d   t
              sJ y )Nzfoo,bar),argsr   r/   r0   )r   r#   strsplit
isinstancelist)rD   r(   s     r   test_apply_argsr_   Y   sN    	{AWWSYYVW,F!9&&&fQi&&&r   zargs, kwargs, incrementr6   ar9   )r:   r       )r9   cr:      c                     dd}t        ddg      }d}t        j                  t        |      5   |j                  |dg| i |}d d d        ||z   }t        j
                  |       y # 1 sw Y   %xY w)Nr   c                 $    | |z   d|z  z   d|z  z   S )NrM   d   r6   )r   r`   brb   s       r   r   ztest_agg_args.<locals>.fg   s    1urAv~a''r   r9   r:   zhin Series.agg cannot aggregate and has been deprecated. Use Series.transform to keep behavior unchanged.match)r   r   r   )r   r%   rT   rU   aggr&   )rZ   kwargs	incrementr   rD   msgr(   r)   s           r   test_agg_argsrn   a   s|    ( 	1vA	;  
	#	#M	= .q!-d-f-.9}H68,. .s   A++A4c                     t        g d      } dd}dd}d}t        j                  t        |      5  | j	                  |ddd	       d d d        t        j                  t        |      5  | j	                  ||gddd	       d d d        t        j                  t        |      5  | j	                  ||d
ddd	       d d d        y # 1 sw Y   xY w# 1 sw Y   RxY w# 1 sw Y   y xY w)Nr8   r   c                     | |z   |z   S r   r6   )r   r`   rb   s      r   foo1z.test_agg_mapping_func_deprecated.<locals>.foo1y       1uqyr   c                     | |z   |z   S r   r6   )r   rg   rb   s      r   foo2z.test_agg_mapping_func_deprecated.<locals>.foo2|   rr   r   z+using .+ in Series.agg cannot aggregate andrh   r      )rb   )r`   rg   )r9   r   )r:   r   )r   r%   rT   rU   rj   )rD   rq   rt   rm   s       r    test_agg_mapping_func_deprecatedrv   u   s    yA 8C		#	#M	= 	dAqA		#	#M	= '	tTlAqA&'		#	#M	= 1	Dt$aa01 1	 ' '1 1s#   C	,C'C!	CC!C*c                 :   t        t        dd            }d }| s:d}t        j                  t        |      5  |j                  ||        d d d        y |j                  ||       }|j                  |      }t        j                  ||       y # 1 sw Y   y xY w)Nz1/1/2000rM   r   c                 H    | j                   | j                  | j                  fS r   )hourdaymonthr!   s    r   rP   z2test_series_apply_map_box_timestamps.<locals>.func   s    qww''r   z&Series' object has no attribute 'hour'rh   r   )	r   r   pytestraisesAttributeErrorr#   r$   r%   r&   )r   r'   rP   rm   r(   r)   s         r   $test_series_apply_map_box_timestampsr      s    
J3
4C( 6]]>5 	+IId6I*	+YYtFY+Fwwt}H68,	+s   BBc                     t        j                  d      t        j                  d      g} t        | d      }|j                  dk(  sJ |j	                  d d      }t        d	d
g      }t        j                  ||       y )N
2011-01-01
2011-01-02zM8[ns]r3   zdatetime64[ns]c                 d    t        |       j                   d| j                   d| j                   S N_type__name__r{   tzr!   s    r   r"   z%test_apply_box_dt64.<locals>.<lambda>   *    a!1!1 2!AEE7!ADD6B r   r   r   Timestamp_1_NoneTimestamp_2_Nonepd	Timestampr   r3   r#   r%   r&   valsr'   resr@   s       r   test_apply_box_dt64r      sq    LL&\(BCD
X
&C99((((
))B8)
TC
$&89
:C3$r   c                  
   t        j                  dd      t        j                  dd      g} t        | d      }|j                  dk(  sJ |j	                  d d	
      }t        ddg      }t        j                  ||       y )Nr   z
US/Eastern)r   r   zM8[ns, US/Eastern]r   zdatetime64[ns, US/Eastern]c                 d    t        |       j                   d| j                   d| j                   S r   r   r!   s    r   r"   z'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      s{    
\l3
\l3D 1
2C994444
))B8)
TC
*,DE
FC3$r   c                      t        j                  d      t        j                  d      g} t        |       }|j                  dk(  sJ |j	                  d d      }t        ddg      }t        j                  ||       y )	Nz1 daysz2 daysztimedelta64[ns]c                 J    t        |       j                   d| j                   S r   )r   r   daysr!   s    r   r"   z%test_apply_box_td64.<locals>.<lambda>   s     a!1!1 2!AFF8< r   r   r   Timedelta_1Timedelta_2)r   	Timedeltar   r3   r#   r%   r&   r   s       r   test_apply_box_td64r      sl    LL"BLL$:;D
,C99))))
))<X)
NC
-/
0C3$r   c                     t        j                  dd      t        j                  dd      g} t        |       }|j                  dk(  sJ |j	                  d d      }t        d	d	g      }t        j                  ||       y )
Nr   Mr   r   z	Period[M]c                 J    t        |       j                   d| j                   S r   )r   r   freqstrr!   s    r   r"   z'test_apply_box_period.<locals>.<lambda>   s     a!1!1 2!AII;? r   r   r   Period_M)r   Periodr   r3   r#   r%   r&   r   s       r   test_apply_box_periodr      so    IIl-ryyC/PQD
,C99###
))?)
QC
*j)
*C3$r   c                 <    t        ddd      j                  d      }t        |d      }|j                  d  	      }t        dd
d      j                  d      }t        |d      }t	        j
                  ||       |j                   fd 	      }t        t        t        d            dgz   d rdnd      }t	        j
                  ||        fd}|j                  | 	      } r(t        dgdz  d      }t	        j
                  ||       y |dk(  sJ y )Nr   r   r   r   z
Asia/TokyoXXr1   c                 D    | t         j                  j                         z   S r   )r   offsetsDayr!   s    r   r"   z'test_apply_datetimetz.<locals>.<lambda>   s    q2::>>#33 r   r   z
2011-01-03c                 L    r| j                   S | j                  j                   S r   )rz   r   r   s    r   r"   z'test_apply_datetimetz.<locals>.<lambda>   s    qvv QTTYY r      r   int64int32r2   r3   c                 p    rt        | j                        S t        | j                  j                        S r   )r[   r   r   r   s    r   r   z test_apply_datetimetz.<locals>.f   s$    "s144y4ADDGG4r      )r   tz_localizer   r#   r%   r&   r^   range)r   valuesrD   r(   
exp_valuesr@   r   s   `      r   test_apply_datetimetzr      s   l=II,WFvD!AWW3FWCFL,SAMMJ $
'C63'WW>vWNF
eBiA3&TFPW
XC63'5 WWQvW&Fl^b(t4
vs+%%%r   c                 ,   t        j                  t        d      t        d      d      }t        |dt        d            }| sTd}t	        j
                  t        |	      5  |j                  d
 |        d d d        |j                  d |       dk(  sJ y |j                  d |       }t        j                  t        d      t        d      d      }t        |dt        d            }t        j                  ||       t        j                  |j                  |j                         |j                  d       }t        dgdz  dt        d            }t        j                  ||       |s|j                  t        k(  sJ y y # 1 sw Y   xY w)NABBABCDDCBAT)
categoriesorderedr   abcdefg)r2   r4   z&Series' object has no attribute 'lowerrh   c                 "    | j                         S r   lowerr!   s    r   r"   z(test_apply_categorical.<locals>.<lambda>   s    	 r   r   c                      yNAr6   r!   s    r   r"   z(test_apply_categorical.<locals>.<lambda>       r   r   c                 "    | j                         S r   r   r!   s    r   r"   z(test_apply_categorical.<locals>.<lambda>   s     r   abbabcddcbac                      yr   r6   r!   s    r   r"   z(test_apply_categorical.<locals>.<lambda>   r   r      )r   Categoricalr^   r   r}   r~   r   r#   r%   r&   assert_categorical_equalr   r3   rA   )r   using_infer_stringr   r'   rm   r(   r@   s          r   test_apply_categoricalr      sA   ^^DOVdSF
d$y/
:C6]]>5 	:II)&I9	:yyvy6#===YY*6Y:F ^^DOVdSF
d$y/
:C63'szz:YY}%F
#T)_
=C63');6<<6!XXAX!#	: 	:s   !F		Fseriesz1-1z1-2c                    t        | d      }|s;d}t        j                  t        |      5  |j	                  d |       d d d        y |j	                  d |      }|j                  t              }t        ddt        j                  gd      }|j                  t              }t        j                  ||       y # 1 sw Y   y xY w)	Ncategoryr   z('Series' object has no attribute 'split'rh   c                 *    | j                  d      d   S N-r   r\   r!   s    r   r"   z8test_apply_categorical_with_nan_values.<locals>.<lambda>  s    aggcl1o r   r   c                 *    | j                  d      d   S r   r   r!   s    r   r"   z8test_apply_categorical_with_nan_values.<locals>.<lambda>  s    qwws|A r   1)r   r}   r~   r   r#   astyperA   r=   rO   r%   r&   )r   r   rD   rm   r(   r)   s         r   &test_apply_categorical_with_nan_valuesr      s     	vZ(A8]]>5 	>GG-fG=	>WW.vW>F]]6"FsC(
;Hv&H68,	>s   B>>Cc                     t        g t        dd      t              }|j                  d |       }t	        j
                  ||       y )Nz
2018-01-01r   )startr   r;   c                     | S r   r6   r!   s    r   r"   zEtest_apply_empty_integer_series_with_datetime_index.<locals>.<lambda>  s    q r   r   )r   r   intr#   r%   r&   )r   rD   r(   s      r   3test_apply_empty_integer_series_with_datetime_indexr     s9    r,B#NAWW[W0F61%r   c                      t        t        j                  g d      dg      } t        g ddg      }d }|d   j                  || f      }t	        g ddd	
      }t        j                  ||       y )N)r9   r:   r   ru      Numbers)columns)r:   r   r:   r9   r:   Indicesc                 *    |d   j                   |    }|S )Nr   )iloc)	targetRowtargetDFvals      r   retrievez+test_apply_dataframe_iloc.<locals>.retrieve  s    y!&&y1
r   rY   )r   ru   r   r:   r   uint64r   )r   r=   r   r#   r   r%   r&   )uintDFindexDFr   r(   r)   s        r   test_apply_dataframe_ilocr     sh    ryy1I;GF)=G Y%%hfY%?FoIXFH68,r   c                 &   t        j                  d      5  t        j                  |       }t        j                  |       }| j	                  t         j                  |      }|j                         }t        j                  ||       | j	                  t         j                  g|      }|j                         j                         }dg|_	        t        j                  ||       | j	                  dg|      }t        j                  ||       t        ||gd      }ddg|_	        | j	                  t         j                  t         j                  g|      }t        j                  ||       t        ||gd      }dd	g|_	        |j                         j                  d
      }| j	                  t         j                  t         j                  d|      }t        j                  |j                  |      |       d d d        y # 1 sw Y   y xY w)Nr,   r-   r   r>   r9   axisabsoluter/   r0   r   )r/   r0   )r=   r?   r>   absr#   copyr%   r&   to_framer   assert_frame_equalr   unstackrenamereindex_like)string_seriesr   f_sqrtf_absr(   r)   s         r   test_transformr     s    
	"  H'}% $$RWWV$<;;=
vx0 $$bggYv$>??$))+"8
fh/$$fXf$=
fh/
 65/2"J/$$bggrvv%6v$F
fh/ 65/2!5>##%,,X6$$RWWRVV%DV$T
v228<hGA H  H  Hs   G'HHopc                    | dv r2|j                  t        j                  j                  |  d             t	        dt
        gz        }| dv r	t        }d}nt        }dj                  dd	g      }t        j                  ||
      5  |j                  | dg       d d d        t        j                  ||
      5  |j                  | dd       d d d        t        j                  ||
      5  |j                  | gdgd       d d d        t        j                  ||
      5  |j                  | dg| gd       d d d        y # 1 sw Y   xY w# 1 sw Y   xY w# 1 sw Y   YxY w# 1 sw Y   y xY w)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 typerh   r  r   B)applymarkerr}   markxfailr   rA   
ValueError	TypeErrorjoinr~   	transform)r   r   r'   errorrm   s        r   test_transform_partial_failurer  E  sa    
;;KKt+F%GH	

 fX
C	!!)hhF*
 
uC	( %r7m$% 
uC	( /BW-./ 
uC	( 3RDy123 
uC	( 7RM567 7% %/ /3 37 7s0   E:E.E$E)EEE&)E2c                  b   d } d }t        dt        gz        }d}t        j                  t        |      5  |j                  | |g       d d d        t        j                  t        |      5  |j                  || d       d d d        t        j                  t        |      5  |j                  |g| gd       d d d        t        j                  t        |      5  |j                  | |g| gd       d d d        y # 1 sw Y   xY w# 1 sw Y   xY w# 1 sw Y   ]xY w# 1 sw Y   y xY w)Nc                     | S r   r6   r!   s    r   noopz7test_transform_partial_failure_valueerror.<locals>.noopk  s    r   c                     t         r   )r  )r   s    r   
raising_opz=test_transform_partial_failure_valueerror.<locals>.raising_opn  s    r   r   r  rh   r	  )r   rA   r}   r~   r  r  )r  r  r'   rm   s       r   )test_transform_partial_failure_valueerrorr  i  s	    fX
C
%C	z	- *tZ()* 
z	- 4JT234 
z	- 8ZLv678 
z	- >T:.dV<=> >* *4 48 8> >s/   D.D&D D%D
DD"%D.c                     t        t        d      dd      } | j                  ddg      }t        ddgddgd	      }t        j                  ||       | j                  d
di      }t        dgd
gd	      }t        j                  ||       y )N   r   r   )r3   r2   minmaxr   r   r4   r2   r/   )r   r   rj   r%   r&   rD   r(   r)   s      r   	test_demor     s    uQxwX6AUUE5>"Fq!fUENBH68,UUE5>"Fqc%x8H68,r   rP   c                     t        |       S r   )r[   r!   s    r   r"   r"     s
    Q r   c                     | j                  ||      }|r(| j                  |      }t        j                  ||       y |t	        |       k(  sJ y )Nr   )r#   r$   r%   r&   r[   )r   rP   r   r(   r)   s        r   (test_apply_map_evaluate_lambdas_the_samer#    sO       f 5F $$T*
vx0]++++r   c                    t        j                  t              5  | j                  d       }d d d        t	        t
              rt        |      t        |       k(  sJ t        j                  t              5  | j                  t              }d d d        t	        |t
              rt        |      t        |       k(  sJ y # 1 sw Y   xY w# 1 sw Y   ?xY w)Nc                     t        |       S r   )r   r!   s    r   r"   z+test_agg_evaluate_lambdas.<locals>.<lambda>  s
    T!W r   )r%   rT   rU   rj   r]   r   lenr   )r   r(   s     r   test_agg_evaluate_lambdasr'    s     
	#	#M	2 6""#456ff%#f+]9K*KKK		#	#M	2 )""4()ff%#f+]9K*KKK*K6 6) )s   B?7C?CCop_namerj   r#   c                    d}|dk(  rt         nd }t        j                  ||      5   t        | |      d       }d d d        t	        | | dz  d      }t        j
                  |       t        j                  t         |      5  | j                  d       }d d d        t        j
                  ||       y # 1 sw Y   }xY w# 1 sw Y   ,xY w)Nzcannot aggregaterj   rh   c                 *    t        | | dz  gddg      S Nr:   r   x^2r4   r   r!   s    r   r"   z)test_with_nested_series.<locals>.<lambda>  s    faAYsEl; r   r:   )r   r,  c                 *    t        | | dz  gddg      S r+  r.  r!   s    r   r"   z)test_with_nested_series.<locals>.<lambda>  s    vq!Q$iU|/T r   )rU   r%   rT   getattrr   r   rj   )rC   r(  rm   warningr(   r)   s         r   test_with_nested_seriesr2    s     C&%/mTG		#	#G3	7 
2'2;


 7IJKH&(+		#	#M	= V $$%TUV&(+
 
V Vs   B7C7C Cc                     | j                         }| j                  ddddd dd dd	      }t        j                  ||       y )
Ncountmeanstdr  c                 $    | j                  d      S )Ng      ?quantiler!   s    r   r"   z)test_replicate_describe.<locals>.<lambda>      QZZ- r   medianc                 $    | j                  d      S )Ng      ?r8  r!   s    r   r"   z)test_replicate_describe.<locals>.<lambda>  r:  r   r  )r4  r5  r6  r  z25%z50%z75%r  )describer#   r%   r&   )r   r)   r(   s      r   test_replicate_describer>    sO    %%'H  --		
F 68,r   c                     | j                  ddg      }t        | j                         | j                         gddg| j                        }t        j                  ||       y )Nsumr5  r1   )rj   r   r@  r5  r2   r%   r&   )r   r(   r)   s      r   test_reducerA    s\    v/F				m0023	H
 68,r   z	how, kwdsr   c                 6   t        g d      } t        ||       d	i |}|j                  }||k(  sJ  t        ||       g dfi |}t        dddd      }t        j                  ||        t        ||       ddddfi |}t        j                  ||       y )
N)r9   r:   Nsize)rC  r4  r5        @       @g      ?r4  r5  )rC  )r   r0  rC  r%   r&   )howkwdsrD   r(   r)   s        r   test_non_callable_aggregatesrH    s     	|A WQ_,t,FvvHX WQ_6?$?FsS#>?H68,WQ_fwOXSWXF68,r   c                     t        dgdz        }|j                  dd d g|       }t        g dg d	      }t        j                  ||       y )
Nru   r   r@  c                 "    | j                         S r   r@  r!   s    r   r"   z3test_series_apply_no_suffix_index.<locals>.<lambda>  s    quuw r   c                 "    | j                         S r   rK  r!   s    r   r"   z3test_series_apply_no_suffix_index.<locals>.<lambda>  s    !%%' r   r   )   rM  rM  )r@  <lambda>rN  r-  )r   r#   r%   r&   )r   rD   r(   r)   s       r   !test_series_apply_no_suffix_indexrO    sI    sQwAWWe.0AB6WRFl*IJH68,r   zdti,expl    ,b/ r-  r   r   r   rM   
2020-01-01rx   tsr  awarec                     |r| j                  d      j                  }n| j                  }t        |      j                  d       }t	        j
                  ||       y )NUTCc                     t        ddg      S )Nr9   r:   r.  r!   s    r   r"   zCtest_apply_series_on_date_time_index_aware_series.<locals>.<lambda>  s    61a&> r   )r   r4   r   r#   r%   r   )dtir@   rR  r4   r(   s        r   1test_apply_series_on_date_time_index_aware_seriesrW    sI    * &,,		E]  !9:F&#&r   zby_row, expected)Fr9   c                     t        t        j                  dt        j                        t	        ddd            }t        |j
                        j                  d |       }t        j                  ||       y )	NrM   r   rP  rT  )r   r   r-  c                      yrI   r6   r!   s    r   r"   zCtest_apply_scalar_on_date_time_index_aware_series.<locals>.<lambda>  r   r   r   )	r   r=   arangerB   r   r4   r#   r%   assert_equal)r   r)   r   r(   s       r   1test_apply_scalar_on_date_time_index_aware_seriesr\    sZ     
		"BJJ're<F FLL!''F'CFOOFH%r   c                    ddg}t        j                  |      }t        |      j                  t         j                  |       }t	        j
                  t        |      |       dt        j                  t         j                  t         j                  g}t        j                  |      }t        |      }|j                  t         j                  |       }t	        j
                  t        |      |       y )Nz00:00:01z00:00:02r   )	r   to_timedeltar   r#   r%   r&   r=   rO   NaT)r   list_of_valid_stringsr`   rg   list_of_stringsr'   s         r   test_apply_to_timedeltarb  #  s    '4
-.A$%++BOOF+KA6!9a(!266266266:O
(A

!C		"//&	1A6!9a(r   z
ops, namesr@  r5  zhow, kwargsc           
      j   t        t        ||      D ci c]  \  }}| ||        c}}      }d|_        |dk(  rt        nd }ddj	                  |       d}	t        j                  ||	      5   t        | |      |fi |}
d d d        t        j                  
|       y c c}}w # 1 sw Y   &xY w)Nr   rj   zusing Series.[r  ]rh   )	r   zipr2   rU   r  r%   rT   r0  r&   )r   opsnamesrF  rk   r2   r   r)   warnrm   r(   s              r   test_apply_listlike_reducerri  1  s     s5#O84tR..OPHHM5L=dD388E?+1
-C		#	#D	4 <,,S;F;<68, P< <s   B#
0B))B2rf  r   r	  c           
      `   t        |j                         D ci c]  \  }}| ||        c}}      }| j                  |_        |dk(  rt        nd }d}	t	        j
                  ||	      5   t        | |      |fi |}
d d d        t	        j                  
|       y c c}}w # 1 sw Y   &xY w)Nrj   zusing Series.[sum|mean]rh   )r   itemsr2   rU   r%   rT   r0  r&   )r   rf  rF  rk   r   r2   r   r)   rh  rm   r(   s              r   test_apply_dictlike_reducerrl  I  s     syy{K84tR..KLH!&&HM5L=dD
#C		#	#D	4 <,,S;F;<68, L< <s   B
+B$$B-r>   r   c           	         t        j                  d      5  t        |D cg c]
  } ||        c}d      }||_        | j	                  ||      }t        j                  ||       d d d        y c c}w # 1 sw Y   y xY w)Nr,   r-   r9   r   r   )r=   r?   r   r   r#   r%   r   )r   rf  rg  r   r   r)   r(   s          r   test_apply_listlike_transformerrn  a  sv     
	" 0s;2m,;!D $$S$8
fh/	0 0;0 0s   
A7A2
9A72A77B zops, expectedc                     | S r   r6   r!   s    r   r"   r"   v  s    A r   rN  r8   c                 "    | j                         S r   rK  r!   s    r   r"   r"   w  s    AEEG r   r  c                 p    t        g d      }|j                  | |      }t        j                  ||       y Nr8   r   r   r#   r%   r[  )rf  r)   r   r'   r(   s        r   test_apply_listlike_lambdart  s  s.     
CYYs6Y*FOOFH%r   c                 @   t        j                  d      5  t        |j                         D ci c]  \  }}| ||        c}}      }| j                  |_        | j                  ||      }t        j                  ||       d d d        y c c}}w # 1 sw Y   y xY w)Nr,   r-   r   )r=   r?   r   rk  r2   r#   r%   r&   )r   rf  r   r2   r   r)   r(   s          r   test_apply_dictlike_transformerrv    s     
	" 1399;OxtR4M!22OP%**$$S$8
vx0	1 1O1 1s   BBABBBc                     | S r   r6   r!   s    r   r"   r"     s    A r   r   c                 "    | j                         S r   rK  r!   s    r   r"   r"     s     r   c                 p    t        g d      }|j                  | |      }t        j                  ||       y rr  rs  )rf  r   r)   r'   r(   s        r   test_apply_dictlike_lambdarz    s.     
CYYs6Y*FOOFH%r   c           	         t        dt        d      it        t        d      d            }|j                  j	                  d       }t        dt
        j                  t
        j                  gddt
        j                  gg dgt        t        d      d      t        t        d      d      	      }t        j                  ||       y )
Nr   r   r1   c           	      `    t        t        | dz         t        t        | dz         d            S )Nr9   yr1   )r   r   r   r!   s    r   r"   z0test_apply_retains_column_name.<locals>.<lambda>  s%    &q1uuU1q5\PS7T"U r                 ?)r~  r  rE  r}  )r   r4   )	r   r   r   r   r#   r=   rO   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                      t        ddt        gg d      } | j                  t              }t        t        t
        t        gg d      }t        j                  ||       y )Nr   string)r`   rg   rb   r-  )r   floatr#   r   r   r[   r%   r&   r  s      r   test_apply_typer    sE    8U#?;AWWT]FsC&o>H68,r   c                      t        g dg dg      } | j                  d       }t        ddgddgdd	gt        j                  d
gd      }t        j                  ||       y )Nr8   )ru   r   r  r   c                     t        |       S r   r.  r!   s    r   r"   z6test_series_apply_unpack_nested_data.<locals>.<lambda>  s
     r   r  g      @rE  g      @rD  g      @r   )r   r9   r:   r   )r   r#   r   r=   rO   r%   r   )r'   r(   r)   s      r   $test_series_apply_unpack_nested_datar    sX    
)\*
+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*   rF   rJ   r  parametrizerV   r_   rn   rv   r   r   r   r   r   r   r   rO   r   r   r   r   r  r  r   r[   r#  r'  r2  r>  rA  rH  rO  DatetimeIndexrepeatrZ  rB   rW  onesr\  rb  r@  r5  arrayri  rl  r>   r   rn  rt  r@   rv  from_arraysr   rz  r  r  r  r6   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!1!11k2B!CDibii!Q!!4GD	

 		"BJJ/ r:
 ibii!Q"15WE	
" 4-0' 1#$' (F72772;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  +9$:J$:$:SEAIuQx;P$QR	
  	!61#cU#;<	&	&	,-,r   