
    bMhW                     4   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                            dddg          d             Zd Ze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                            d"d#d#ej#        gd#d$ej#        gg          d%             Z$d& Z%d' Z&d( Z'ej                            d)e          d*             Z(d+ Z)d, Z*ej                            d-e+d. g          d/             Z,d0 Z-ej                            d1d2d3g          d4             Z.d5 Z/d6 Z0ej                            d7d2i fd3d8difd3d8difg          d9             Z1d: Z2ej                            d; eddg ej3        d d<g          =           e ej4        ddggdd >          d?@          f e ej5        dAej6        @           e
dBdAC          dDE           e ej4        ddggdAd >          d?@          fg          ej                            dFddg          dG                         Z7ej                            dHd e ej8        dA          d?@          fdIg          dJ             Z9dK Z:ej                            dLej;        gdMgfej;        ej<        gdMdNgf ej=        ej;        g          dMgf ej=        ej;        ej<        g          dMdNgfg          ej                            dOd2i gd3d8digd3d8digg          dP                         Z>ej                            dQdRej;        iej;        ej<        dS edRej;        i           eej;        ej<        dS          g          ej                            dOd2i gd3d8digd3d8digg          dT                         Z?ej                            dLej@        gdUgfejA        ej@        gdVdUgf ej=        ej@        g          dUgf ej=        ejA        ej@        g          dVdUgfg          dW             ZBej                            dXdY g edZg d[i          fd\ g ed]gdZg=          fg          d^             ZCej                            dQdRej@        iej@        ejD        dS edRej@        i           eej@        ejD        dS          g          d_             ZEej                            dXdd` i eg d[ ejF        dgdaz   eGda          g          =          fddb i ed]gdg=          fg          dc             ZHdd ZIde ZJdf ZKdS )g    N)	DataFrameIndex
MultiIndexSeriesconcat
date_rangetimedelta_range)series_transform_kernelsFcompat)paramsc                     | j         S N)param)requests    d/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/pandas/tests/apply/test_series_apply.pyby_rowr      s
    =    c                      t          t          ddd                    } fd}|                    |           }|                    d           }t	          j        ||           t          g d          }t	          j        ||           d S )	Nz	1 day 1 s   h)periodsfreqc                 b    r|                                  n| j                                         S r   )total_secondsdtxr   s    r   fz(test_series_map_box_timedelta.<locals>.f   s+    $*Dq   0B0B0D0DDr   r   c                 *    |                                  S r   )r   r   s    r   <lambda>z/test_series_map_box_timedelta.<locals>.<lambda>    s    !2!2 r   )g    @g    @g    @)r   r	   applymaptmassert_series_equal)r   serr   resultexpecteds   `    r   test_series_map_box_timedeltar*      s    
acBBB
C
CCE E E E E YYqY((Fww2233H68,,,11122H68,,,,,r   c                 6   |                      t          j        |          }t          j        d          5  t          j        |           }d d d            n# 1 swxY w Y   t	          j        ||           |                      t          j        |          }t          j        |           }t	          j        ||           t          t          dt          g d                    }|                     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        
          }|                     d |          }t	          j        ||           d S )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   '   s   ""276"::F		"	"	" , ,7?++, , , , , , , , , , , , , , ,68,,, ""26&"99Fvo&&H68,,, 	V%uRe/D/D/DEEEA	
V	,	,B1b!!! B;;;;7bh7bh6RW 	YYYbj111A	
V	,	,B1b!!!!!s   AAAc                      t          ddg          } d }|                     |d          }|                     |          }t          j        ||           d S )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   s_    1vA   WWQxW((FuuQxxH68,,,,,r   convert_dtypeTc                    t          t          j                            d                              d                    }d }t          j        t                    5  |                    || d           d d d            d S # 1 swxY w Y   d S )Nr:   
   c                 *    | dk    r| nt           j        S )Nr   )r=   nanr!   s    r   funcz1test_apply_convert_dtype_deprecated.<locals>.funcR   s    EEqqrv%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   s    
&&q))99"==
>
>C& & & 
	#M	2	2 F F		$mH	EEEF F F F F F F F F F F F F F F F F Fs   BB	Bc                      t          dg          } |                     t          j        d          }|d         ddgk    sJ t	          |d         t
                    sJ d S )Nzfoo,bar),argsr   r/   r0   )r   r#   strsplit
isinstancelist)rD   r(   s     r   test_apply_argsr_   Y   sa    	{AWWSYVW,,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| R i |}d d d            n# 1 swxY w Y   ||z   }t          j        ||           d S )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AAc                     t          g d          } dd}dd}d}t          j        t          |          5  |                     |dd	d
           d d d            n# 1 swxY w Y   t          j        t          |          5  |                     ||gdd	d
           d d d            n# 1 swxY w Y   t          j        t          |          5  |                     ||ddd	d
           d d d            d S # 1 swxY w Y   d S )Nr8   r9   r   c                     | |z   |z   S r   r6   )r   r`   rb   s      r   foo1z.test_agg_mapping_func_deprecated.<locals>.foo1y       1uqyr   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   yyyA       8C		#M	=	=	=  	dAqA              		#M	=	=	= ' '	tTlAqA&&&' ' ' ' ' ' ' ' ' ' ' ' ' ' '		#M	=	=	= 1 1	Dt$$aa0001 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1s5   AA!$A!B++B/2B/C;;C?C?c                 j   t          t          dd                    }d }| sNd}t          j        t          |          5  |                    ||            d d d            n# 1 swxY w Y   d S |                    ||           }|                    |          }t          j        ||           d S )Nz1/1/2000rM   r   c                 *    | j         | j        | j        fS r   )hourdaymonthr!   s    r   rP   z2test_series_apply_map_box_timestamps.<locals>.func   s    qw''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33
4
4C( ( (  6]>555 	+ 	+IId6I***	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+YYtFY++Fwwt}}H68,,,,,s   A%%A),A)c                     t          j        d          t          j        d          g} t          | d          }|j        dk    sJ |                    d d          }t          d	d
g          }t          j        ||           d S )N
2011-01-01
2011-01-02zM8[ns]r3   zdatetime64[ns]c                 N    t          |           j         d| j         d| j         S N_type__name__r{   tzr!   s    r   r"   z%test_apply_box_dt64.<locals>.<lambda>   )    a!1BBAEBBADBB r   r   r   Timestamp_1_NoneTimestamp_2_Nonepd	Timestampr   r3   r#   r%   r&   valsr'   resr@   s       r   test_apply_box_dt64r      s    L&&\(B(BCD
X
&
&
&C9(((((
))BB8)
T
TC
$&89
:
:C3$$$$$r   c                     t          j        dd          t          j        dd          g} t          | d          }|j        dk    sJ |                    d d	
          }t          ddg          }t          j        ||           d S )Nr   z
US/Eastern)r   r   zM8[ns, US/Eastern]r   zdatetime64[ns, US/Eastern]c                 N    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33
\l333D 1
2
2
2C944444
))BB8)
T
TC
*,DE
F
FC3$$$$$r   c                  
   t          j        d          t          j        d          g} t          |           }|j        dk    sJ |                    d d          }t          ddg          }t          j        ||           d S )	Nz1 daysz2 daysztimedelta64[ns]c                 >    t          |           j         d| j         S r   )r   r   daysr!   s    r   r"   z%test_apply_box_td64.<locals>.<lambda>   s    a!1<<AF<< r   r   r   Timedelta_1Timedelta_2)r   	Timedeltar   r3   r#   r%   r&   r   s       r   test_apply_box_td64r      s    L""BL$:$:;D
,,C9)))))
))<<X)
N
NC
-/
0
0C3$$$$$r   c                     t          j        dd          t          j        dd          g} t          |           }|j        dk    sJ |                    d d          }t          d	d	g          }t          j        ||           d S )
Nr   Mr   r   z	Period[M]c                 >    t          |           j         d| j         S r   )r   r   freqstrr!   s    r   r"   z'test_apply_box_period.<locals>.<lambda>   s    a!1??AI?? r   r   r   Period_M)r   Periodr   r3   r#   r%   r&   r   s       r   test_apply_box_periodr      s    Il---ryC/P/P/PQD
,,C9####
))??)
Q
QC
*j)
*
*C3$$$$$r   c                     t          ddd                              d          }t          |d          }|                    d  	          }t          dd
d                              d          }t          |d          }t	          j        ||           |                     fd 	          }t          t          t          d                    dgz   d rdnd          }t	          j        ||            fd}|                    | 	          } r,t          dgdz  d          }t	          j        ||           d S |dk    sJ d S )Nr   r   r   r   z
Asia/TokyoXXr1   c                 D    | t           j                                        z   S r   )r   offsetsDayr!   s    r   r"   z'test_apply_datetimetz.<locals>.<lambda>   s    q2:>>#3#33 r   r   z
2011-01-03c                 .    r| j         n| j        j         S r   )rz   r   r   s    r   r"   z'test_apply_datetimetz.<locals>.<lambda>   s    >qvvQTY r      r   int64int32r2   r3   c                 b    rt          | j                  nt          | j        j                  S r   )r[   r   r   r   s    r   r   z test_apply_datetimetz.<locals>.f   s$    "4s14yyyADG4r      )r   tz_localizer   r#   r%   r&   r^   range)r   valuesrD   r(   
exp_valuesr@   r   s   `      r   test_apply_datetimetzr      s   l===II,WWFvD!!!AWW33FWCCFL,SAAAMM J $
'
'
'C63'''WW>>>>vWNNF
eBiiA3&TF9WPW
X
X
XC63'''5 5 5 5 5 WWQvW&&F &l^b(t444
vs+++++%%%%%%r   c                    t          j        t          d          t          d          d          }t          |dt          d                    }| smd}t	          j        t          |	          5  |                    d
 |            d d d            n# 1 swxY w Y   |                    d |           dk    sJ d S |                    d |           }t          j        t          d          t          d          d          }t          |dt          d                    }t          j	        ||           t          j
        |j        |j                   |                    d           }t          dgdz  dt          d                    }t          j	        ||           |s|j        t          k    snd S J d S )NABBABCDDCBAT)
categoriesorderedr   abcdefg)r2   r4   z&Series' object has no attribute 'lowerrh   c                 *    |                                  S r   lowerr!   s    r   r"   z(test_apply_categorical.<locals>.<lambda>   s    		 r   r   c                     dS NAr6   r!   s    r   r"   z(test_apply_categorical.<locals>.<lambda>   s    3 r   r   c                 *    |                                  S r   r   r!   s    r   r"   z(test_apply_categorical.<locals>.<lambda>   s     r   abbabcddcbac                     dS r   r6   r!   s    r   r"   z(test_apply_categorical.<locals>.<lambda>   s     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      s   ^DOOVdSSSF
d$y//
:
:
:C 6]>555 	: 	:II))&I999	: 	: 	: 	: 	: 	: 	: 	: 	: 	: 	: 	: 	: 	: 	:yyvy66#====YY**6Y::F ^DOOVdSSSF
d$y//
:
:
:C63'''sz:::YY}}%%F
#T)__
=
=
=C63''');F6<6!!!!!!!!!s   0BBBseriesz1-1z1-2c                    t          | d          }|sOd}t          j        t          |          5  |                    d |           d d d            n# 1 swxY w Y   d S |                    d |          }|                    t                    }t          ddt          j        gd          }|                    t                    }t          j
        ||           d S )	Ncategoryr   z('Series' object has no attribute 'split'rh   c                 8    |                      d          d         S N-r   r\   r!   s    r   r"   z8test_apply_categorical_with_nan_values.<locals>.<lambda>  s    aggcll1o r   r   c                 8    |                      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55 	> 	>GG--fG===	> 	> 	> 	> 	> 	> 	> 	> 	> 	> 	> 	> 	> 	> 	>WW..vW>>F]]6""FsC(
;;;Hv&&H68,,,,,s   AAAc                     t          g t          dd          t                    }|                    d |           }t	          j        ||           d S )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     sS    r,BBB#NNNAWW[[W00F61%%%%%r   c                     t          t          j        g d          dg          } t          g ddg          }d }|d                             || f          }t	          g ddd	
          }t          j        ||           d S )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     s    ry11I;GGGF)===G   Y%%hfY%??FoooIXFFFH68,,,,,r   c                 X   t          j        d          5  t          j        |           }t          j        |           }|                     t           j        |          }|                                }t          j        ||           |                     t           j        g|          }|                                                                }dg|_	        t          j
        ||           |                     dg|          }t          j
        ||           t          ||gd          }ddg|_	        |                     t           j        t           j        g|          }t          j
        ||           t          ||gd          }dd	g|_	        |                                                    d
          }|                     t           j        t           j        d|          }t          j        |                    |          |           d d d            d S # 1 swxY w Y   d S )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     sN    
	"	"	"  H  H''}%% $$RWV$<<;;==
vx000 $$bgYv$>>??$$))++"8
fh///$$fXf$==
fh///
 65/222"J/$$bgrv%6v$FF
fh/// 65/222!5>##%%,,X66$$RWRV%D%DV$TT
v228<<hGGGA H  H  H  H  H  H  H  H  H  H  H  H  H  H  H  H  H  Hs   G<HH#&H#opc                 <   | dv r6|                     t          j                            |  d                     t	          dt
          gz            }| dv r
t          }d}nt          }d                    dd	g          }t          j	        ||
          5  |
                    | dg           d d d            n# 1 swxY w Y   t          j	        ||
          5  |
                    | dd           d d d            n# 1 swxY w Y   t          j	        ||
          5  |
                    | gdgd           d d d            n# 1 swxY w Y   t          j	        ||
          5  |
                    | dg| gd           d d d            d S # 1 swxY w Y   d S )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  s    
;;;K%G%G%GHH	
 	
 	

 fX

C	!!!)hhF*
 
 
uC	(	(	( % %r7m$$$% % % % % % % % % % % % % % % 
uC	(	(	( / /BW--.../ / / / / / / / / / / / / / / 
uC	(	(	( 3 3RDy112223 3 3 3 3 3 3 3 3 3 3 3 3 3 3 
uC	(	(	( 7 7RM556667 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7sH   B99B= B=C??DD EEE(FFFc                     d } d }t          dt          gz            }d}t          j        t          |          5  |                    | |g           d d d            n# 1 swxY w Y   t          j        t          |          5  |                    || d           d d d            n# 1 swxY w Y   t          j        t          |          5  |                    |g| gd           d d d            n# 1 swxY w Y   t          j        t          |          5  |                    | |g| gd           d d d            d S # 1 swxY w Y   d S )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  sg        fX

C
%C	z	-	-	- * *tZ()))* * * * * * * * * * * * * * * 
z	-	-	- 4 4JT223334 4 4 4 4 4 4 4 4 4 4 4 4 4 4 
z	-	-	- 8 8ZLv667778 8 8 8 8 8 8 8 8 8 8 8 8 8 8 
z	-	-	- > >T:.dV<<===> > > > > > > > > > > > > > > > > >sG   A  A$'A$B++B/2B/C88C<?C<EEEc                  H   t          t          d          dd          } |                     ddg          }t          ddgddgd	          }t          j        ||           |                     d
di          }t          dgd
gd	          }t          j        ||           d S )N   r   r   )r3   r2   minmaxr   r   r4   r2   r/   )r   r   rj   r%   r&   rD   r(   r)   s      r   	test_demor    s    uQxxwX666AUUE5>""Fq!fUENBBBH68,,,UUE5>""Fqc%x888H68,,,,,r   rP   c                      t          |           S r   )r[   r!   s    r   r"   r"     s    Q r   c                     |                      ||          }|r,|                     |          }t          j        ||           d S |t	          |           k    sJ d S )Nr   )r#   r$   r%   r&   r[   )r   rP   r   r(   r)   s        r   (test_apply_map_evaluate_lambdas_the_samer"    so       f 55F , $$T**
vx00000]++++++++r   c                    t          j        t                    5  |                     d           }d d d            n# 1 swxY w Y   t	          |t
                    r t          |          t          |           k    sJ t          j        t                    5  |                     t                    }d d d            n# 1 swxY w Y   t	          |t
                    r t          |          t          |           k    sJ d S )Nc                      t          |           S r   )r   r!   s    r   r"   z+test_agg_evaluate_lambdas.<locals>.<lambda>  s    T!WW r   )r%   rT   rU   rj   r]   r   lenr   )r   r(   s     r   test_agg_evaluate_lambdasr&    s\    
	#M	2	2 6 6""#4#4556 6 6 6 6 6 6 6 6 6 6 6 6 6 6ff%%K#f++]9K9K*K*K*K*K		#M	2	2 ) )""4(() ) ) ) ) ) ) ) ) ) ) ) ) ) )ff%%K#f++]9K9K*K*K*K*K*K*Ks!   =AAB??CCop_namerj   r#   c                    d}|dk    rt           nd }t          j        ||          5   t          | |          d           }d d d            n# 1 swxY w Y   t	          | | dz  d          }t          j        ||           t          j        t           |          5  |                     d           }d d d            n# 1 swxY w Y   t          j        ||           d S )Nzcannot aggregaterj   rh   c                 2    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                 2    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/T/T r   )rU   r%   rT   getattrr   r   rj   )rC   r'  rm   warningr(   r)   s         r   test_with_nested_seriesr1    s    C&%//mmTG		#G3	7	7	7 
 
2'22;;
 

 
 
 
 
 
 
 
 
 
 
 
 
 
 

 7IJJKKH&(+++		#M	=	=	= V V $$%T%TUUV V V V V V V V V V V V V V V&(+++++s#   AAACC	Cc                     |                                  }|                     ddddd dd dd	          }t          j        ||           d S )
Ncountmeanstdr  c                 ,    |                      d          S )Ng      ?quantiler!   s    r   r"   z)test_replicate_describe.<locals>.<lambda>      QZZ-- r   medianc                 ,    |                      d          S )Ng      ?r7  r!   s    r   r"   z)test_replicate_describe.<locals>.<lambda>  r9  r   r  )r3  r4  r5  r  z25%z50%z75%r  )describer#   r%   r&   )r   r)   r(   s      r   test_replicate_describer=    sk    %%''H  ----		
 		
 F 68,,,,,r   c                     |                      ddg          }t          |                                 |                                 gddg| j                  }t          j        ||           d S )Nsumr4  r1   )rj   r   r?  r4  r2   r%   r&   )r   r(   r)   s      r   test_reducer@    su    v//F					m00223	  H
 68,,,,,r   z	how, kwdsr   c                 T   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        ||           d S )
N)r9   r:   Nsize)rB  r3  r4        @       @g      ?r3  r4  )rB  )r   r/  rB  r%   r&   )howkwdsrD   r(   r)   s        r   test_non_callable_aggregatesrG    s     	|||A WQ__,,t,,FvHX WQ__666??$??FsS#>>??H68,,,WQ__fwOOXXSWXXF68,,,,,r   c                     t          dgdz            }|                    dd d g|           }t          g dg d	          }t          j        ||           d S )
Nru   r   r?  c                 *    |                                  S r   r?  r!   s    r   r"   z3test_series_apply_no_suffix_index.<locals>.<lambda>  s    quuww r   c                 *    |                                  S r   rJ  r!   s    r   r"   z3test_series_apply_no_suffix_index.<locals>.<lambda>  s    !%%'' r   r   )   rL  rL  )r?  <lambda>rM  r,  )r   r#   r%   r&   )r   rD   r(   r)   s       r   !test_series_apply_no_suffix_indexrN    sm    sQwAWWe..0A0AB6WRRFlll*I*I*IJJJH68,,,,,r   zdti,expl    ,b/ r,  r   r   r   rM   
2020-01-01rx   tsr  awarec                     |r|                      d          j        }n| j        }t          |                              d           }t	          j        ||           d S )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@   rQ  r4   r(   s        r   1test_apply_series_on_date_time_index_aware_seriesrV    s_    *  &&,	E]]  !9!9::F&#&&&&&r   zby_row, expected)Fr9   c                    t          t          j        dt          j                  t	          ddd                    }t          |j                                      d |           }t          j        ||           d S )	NrM   r   rO  rS  )r   r   r,  c                     dS rI   r6   r!   s    r   r"   zCtest_apply_scalar_on_date_time_index_aware_series.<locals>.<lambda>  s    ! 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[    s|     
	"BJ'''re<<<  F FL!!''F'CCFOFH%%%%%r   c                    ddg}t          j        |          }t          |                              t           j        |           }t	          j        t          |          |           dt          j        t           j        t           j        g}t          j        |          }t          |          }|                    t           j        |           }t	          j        t          |          |           d S )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_timedeltara  #  s    '4
-..A$%%++BOF+KKA6!99a(((!262626:O
((A

!
!C		"/&	11A6!99a(((((r   z
ops, namesr?  r4  zhow, kwargsc                 l    t           fdt          ||          D                       }d|_        |dk    rt          nd }dd                    |           d}t          j        ||          5   t           |          |fi |}d d d            n# 1 swxY w Y   t          j        ||           d S )Nc                 .    i | ]\  }}| |          S r6   r6   .0r2   r   r   s      r   
<dictcomp>z/test_apply_listlike_reducer.<locals>.<dictcomp>@  s)    OOO84tRR..OOOr   r   rj   zusing Series.[r  ]rh   )	r   zipr2   rU   r  r%   rT   r/  r&   )	r   opsnamesrE  rk   r)   warnrm   r(   s	   `        r   test_apply_listlike_reducerrl  1  s    OOOOs5#OOOPPHHM5LL==dD
-388E??
-
-
-C		#D	4	4	4 < <,,,S;;F;;< < < < < < < < < < < < < < <68,,,,,s   0BBBri  r   r  c                 P    t           fd|                                D                       } j        |_        |dk    rt          nd }d}t	          j        ||          5   t           |          |fi |}d d d            n# 1 swxY w Y   t	          j        ||           d S )Nc                 .    i | ]\  }}| |          S r6   r6   rd  s      r   rf  z/test_apply_dictlike_reducer.<locals>.<dictcomp>X  s)    KKK84tRR..KKKr   rj   zusing Series.[sum|mean]rh   )r   itemsr2   rU   r%   rT   r/  r&   )	r   ri  rE  rk   r   r)   rk  rm   r(   s	   `        r   test_apply_dictlike_reducerrp  I  s     KKKKsyy{{KKKLLH!&HM5LL==dD
#C		#D	4	4	4 < <,,,S;;F;;< < < < < < < < < < < < < < <68,,,,,s   "BB
B
r>   r   c                     t          j        d          5  t           fd|D             d          }||_                             ||          }t          j        ||           d d d            d S # 1 swxY w Y   d S )Nr,   r-   c                 &    g | ]} |          S r6   r6   )re  r   r   s     r   
<listcomp>z3test_apply_listlike_transformer.<locals>.<listcomp>m  s#    ;;;22m,,;;;r   r9   r   r   )r=   r?   r   r   r#   r%   r   )r   ri  rj  r   r)   r(   s   `     r   test_apply_listlike_transformerrt  a  s     
	"	"	" 0 0;;;;s;;;!DDD $$S$88
fh///	0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0s   AA55A9<A9zops, expectedc                     | S r   r6   r!   s    r   r"   r"   v  s    A r   rM  r8   c                 *    |                                  S r   rJ  r!   s    r   r"   r"   w  s    AEEGG r   r  c                     t          g d          }|                    | |          }t          j        ||           d S Nr8   r   r   r#   r%   rZ  )ri  r)   r   r'   r(   s        r   test_apply_listlike_lambdarz  s  sB     

CYYs6Y**FOFH%%%%%r   c                 .    t          j        d          5  t           fd|                                D                       } j        |_                             ||          }t          j        ||           d d d            d S # 1 swxY w Y   d S )Nr,   r-   c                 .    i | ]\  }}| |          S r6   r6   rd  s      r   rf  z3test_apply_dictlike_transformer.<locals>.<dictcomp>  s)    OOOxtR4M!2!2OOOr   r   )r=   r?   r   ro  r2   r#   r%   r&   )r   ri  r   r)   r(   s   `    r   test_apply_dictlike_transformerr}    s     
	"	"	" 1 1OOOO399;;OOOPP%*$$S$88
vx000	1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1s   A&B

BBc                     | S r   r6   r!   s    r   r"   r"     s    A r   r   c                 *    |                                  S r   rJ  r!   s    r   r"   r"     s     r   c                     t          g d          }|                    | |          }t          j        ||           d S rx  ry  )ri  r   r)   r'   r(   s        r   test_apply_dictlike_lambdar    sB     

CYYs6Y**FOFH%%%%%r   c           	         t          dt          d          it          t          d          d                    }|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        ||           d S )
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>  s2    &q1uuU1q5\\PS7T7T7T"U"U r                 ?)r  r  rD  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TZZUUVVF
rvrv	c26 2OOODeAhhS)))E!HH3'''  H
 &(+++++r   c                      t          ddt          gg d          } |                     t                    }t          t          t
          t          gg d          }t          j        ||           d S )Nr   string)r`   rg   rb   r,  )r   floatr#   r   r   r[   r%   r&   r  s      r   test_apply_typer    sd    8U#???;;;AWWT]]FsC&ooo>>>H68,,,,,r   c                      t          g dg dg          } |                     d           }t          ddgddgdd	gt          j        d
gd          }t          j        ||           d S )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      @rD  g      @rC  g      @r   )r   r9   r:   r   )r   r#   r   r=   rO   r%   r   )r'   r(   r)   s      r   $test_series_apply_unpack_nested_datar    sv    
)))\\\*
+
+CYY**++Fc3ZS#JC:26ST+VVW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&  r1  r=  r@  rG  rN  DatetimeIndexrepeatrY  rB   rV  onesr[  ra  r?  r4  arrayrl  rp  r>   r   rt  rz  r@   r}  from_arraysr   r  r  r  r  r6   r   r   <module>r     s~	                                   > > > > > > x()))  *)- - - " " "8- - - 4-88F F 98F' ' ' "a[2Qx#fb"%5sAh7LM - -	 - 1 1 1&- - -$% % %	% 	% 	%% % %% % %& & &6G G G4 UE26$:UE26<R#STT- - UT- & & &
- 
- 
-#H #H #HL 788 7  7 98 7F> > >0
- 
- 
- #'7'7!899	, 	, :9	,
L 
L 
L UG$455, , 65,$- - -$- - - R[7Xx01Gh=N3OP - -	 -&- - -  FAq6!1!11k2B!C!CDDDIibi!Q!!444GDDD	

 F	"BJ/// jr:::  
 Iibi!Q"1555WEEE	
 " 4-00' ' 10# $' (FF7272;;g$F$F$FGT & & &) ) ) 
&E7
&"'	UFO,	26(		eW%	2627#	$	$ufo6	  R[7Xx01Gh=N3OP - -	  - 		bff27##RV}RV"'**++	  R[7Xx01Gh=N3OP - -	  - 
'VH
&"'	Z01	27)		vh'	2627#	$	$z6&:;	 0 0 0 
+		:yyy"9::;

	ffaS===> & & & 		bggBF##RW~RW26**++	 1 1 1  ++F999$:J$:SEAIuuQxx;P$Q$QRRR	
   	!661#cU#;#;#;<	 	& &	 	&	, 	, 	,- - -, , , , ,r   