
    .iR                     |   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d Zej.                  j1                  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.                  j1                  dd d ejD                  gd d!ejD                  gg      d"        Z#d# Z$d$ Z%d% Z&ej.                  j1                  d&e      d'        Z'd( Z(d) Z)ej.                  j1                  d*e*d+ g      d,        Z+d- Z,ej.                  j1                  d.d/d0g      d1        Z-d2 Z.d3 Z/ej.                  j1                  d4d/i fd0d5difd0d5difg      d6        Z0d7 Z1ej.                  j1                  d8 eddg ejd                  d d9g      :       e ejf                  ddggdd ;      d<=      f e ejh                  d>ejj                  =       e
d?d>@      dAB       e ejf                  ddggd>d ;      d<=      fg      ej.                  j1                  dCdDdg      dE               Z6ej.                  j1                  dFd e ejn                  d>      d<=      fdGg      dH        Z8dI Z9ej.                  j1                  dJejt                  gdKgfejt                  ejv                  gdKdLgf ejx                  ejt                  g      dKgf ejx                  ejt                  ejv                  g      dKdLgfg      ej.                  j1                  dMd/i gd0d5digd0d5digg      dN               Z=ej.                  j1                  dOdPejt                  iejt                  ejv                  dQ edPejt                  i       eejt                  ejv                  dQ      g      ej.                  j1                  dMd/i gd0d5digd0d5digg      dR               Z>ej.                  j1                  dJej~                  gdSgfej                  ej~                  gdTdSgf ejx                  ej~                  g      dSgf ejx                  ej                  ej~                  g      dTdSgfg      dU        ZAej.                  j1                  dVdW g edXg dYi      fdZ g ed[gdXg:      fg      d\        ZBej.                  j1                  dOdPej~                  iej~                  ej                  dQ edPej~                  i       eej~                  ej                  dQ      g      d]        ZDej.                  j1                  dVdd^ i eg dY ej                  dgd_z   eFd_      g      :      fdd` i ed[gdg:      fg      da        ZGdb ZHdc ZIdd ZJy)e    N)	DataFrameIndex
MultiIndexSeriesconcat
date_rangetimedelta_range)series_transform_kernelsFcompat)paramsc                     | j                   S N)param)requests    b/var/www/app/trading-bot/venv/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   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_argsrS   N   sN    	{AWWSYYVW,F!9&&&fQi&&&r   zargs, kwargs, incrementr6   ar9   )r:   r       )r9   cr:      c                     dd}t        ddg      } |j                  |dg| i |}||z   }t        j                  ||       y )Nr   c                 $    | |z   d|z  z   d|z  z   S )N
   d   r6   )r   rT   brV   s       r   r   ztest_agg_args.<locals>.f\   s    1urAv~a''r   r9   r:   )r   r   r   )r   aggr%   r&   )rN   kwargs	incrementr   rD   r(   r)   s          r   test_agg_argsr`   V   sK    ( 	1vAQUU1a)$)&)F9}H68,r   c                      t        g d      } d	d}d
d}| j                  |ddd       | j                  ||gddd       | j                  ||dddd       y )Nr8   r   c                     | |z   |z   S r   r6   )r   rT   rV   s      r   foo1z.test_agg_mapping_func_deprecated.<locals>.foo1i       1uqyr   c                     | |z   |z   S r   r6   )r   r\   rV   s      r   foo2z.test_agg_mapping_func_deprecated.<locals>.foo2l   rd   r   r      )rV   )rT   r\   )r9   r   )r:   r   )r   r]   )rD   rc   rf   s      r    test_agg_mapping_func_deprecatedrh   e   s]    yA EE$1EEE4,1E"EE4
 !Q!E,r   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/2000rZ   r   c                 H    | j                   | j                  | j                  fS r   )hourdaymonthr!   s    r   funcz2test_series_apply_map_box_timestamps.<locals>.funcx   s    qww''r   z&Series' object has no attribute 'hour'matchr   )	r   r   pytestraisesAttributeErrorr#   r$   r%   r&   )r   r'   ro   msgr(   r)   s         r   $test_series_apply_map_box_timestampsrv   t   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__rm   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 )Nrx   z
US/Eastern)r   ry   zM8[ns, US/Eastern]rz   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[us]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 )
Nrx   Mr   ry   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        g t        t        d            d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 )Nrx   ry   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   )rl   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   )rO   r   r   r   s    r   r   z test_apply_datetimetz.<locals>.f   s$    "s144y4ADDGG4r      )r   tz_localizer   r#   r%   r&   rR   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
&4b	?&A&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 'lowerrp   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   CategoricalrR   r   rr   rs   rt   r#   r%   r&   assert_categorical_equalr   r3   rA   )r   using_infer_stringr   r'   ru   r(   r@   s          r   test_apply_categoricalr      s@   ^^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!FF!#	: 	:s   !F		Fseriesz1-1z1-2c                 v   t        | d      }|s;d}t        j                  t        |      5  |j	                  d |       d d d        y |j	                  d |      }|j                  t              }t        g dd      }|j                  t              }t        j                  ||       y # 1 sw Y   y xY w)	Ncategoryrz   z('Series' object has no attribute 'split'rp   c                 *    | j                  d      d   S )N-r   )rP   r!   s    r   r"   z8test_apply_categorical_with_nan_values.<locals>.<lambda>   s    aggcl1o r   r   c                 X    t        j                  |       r| j                  d      d   S dS )Nr   r   F)r   notnarP   r!   s    r   r"   z8test_apply_categorical_with_nan_values.<locals>.<lambda>   s     "((1+qwws|A 5 r   )1r   F)	r   rr   rs   rt   r#   astyperA   r%   r&   )r   r   rD   ru   r(   r)   s         r   &test_apply_categorical_with_nan_valuesr      s     	vZ(A8]]>5 	>GG-fG=	>WWHQWWXF]]6"F'z:Hv&H68,	>s   B//B8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   rg      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   rM   )r   rg   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shiftz is successful on any dtype)reasonr   )fillnangroupTransform function failed|z4not supported between instances of 'type' and 'type'zunsupported operand typerp   r   r   B)applymarkerrr   markxfailr   rA   
ValueError	TypeErrorjoinrs   	transform)r   r   r'   errorru   s        r   test_transform_partial_failurer  1  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>.noopW  s    r   c                     t         r   )r  )r}   s    r   
raising_opz=test_transform_partial_failure_valueerror.<locals>.raising_opZ  s    r   r   r   rp   r   )r   rA   rr   rs   r  r  )r	  r  r'   ru   s       r   )test_transform_partial_failure_valueerrorr  U  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   r]   r%   r&   rD   r(   r)   s      r   	test_demor  m  s    uQxwX6AUUE5>"Fq!fUENBH68,UUE5>"Fqc%x8H68,r   ro   c                     t        |       S r   )rO   r!   s    r   r"   r"   z  s
    Q r   c                     | j                  ||      }|r*| j                  ||      }t        j                  ||       y |t	        |       k(  sJ y )Nr   )engine)r#   r$   r%   r&   rO   )r   ro   r   r  r(   r)   s         r   (test_apply_map_evaluate_lambdas_the_samer  z  sT       f 5F $$T&$9
vx0]++++r   c                 z    | j                  d       }|t        u sJ | j                  t              }|t        u sJ y )Nc                     t        |       S r   )r   r!   s    r   r"   z+test_agg_evaluate_lambdas.<locals>.<lambda>  s
    a r   )r]   r   r   )r   r(   s     r   test_agg_evaluate_lambdasr    s?    01FVt$FVr   op_namer]   r#   c                      t        | |      d       }|dk(  r(t        | | dz  d      }t        j                  ||       y t	        | | dz  gddg      }t        j
                  ||       y )Nc                 *    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    &!QT3,7 r   r#   r:   )r   r  r   r  r  )getattrr   r%   r   r   r&   )rC   r  r(   r)   s       r   test_with_nested_seriesr"    sr     /W_g.7F '??A;MNO
fh/?OQ,>?U|T
vx0r   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      ?r(  r!   s    r   r"   z)test_replicate_describe.<locals>.<lambda>  r*  r   r  )r$  r%  r&  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sumr%  r1   )r]   r   r0  r%  r2   r%   r&   )r   r(   r)   s      r   test_reducer1    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)r3  r$  r%        @       @g      ?r$  r%  )r3  )r   r!  r3  r%   r&   )howkwdsrD   r(   r)   s        r   test_non_callable_aggregatesr8    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 )
Nrg   r   r0  c                 "    | j                         S r   r0  r!   s    r   r"   z3test_series_apply_no_suffix_index.<locals>.<lambda>  s    quuw r   c                 "    | j                         S r   r;  r!   s    r   r"   z3test_series_apply_no_suffix_index.<locals>.<lambda>  s    !%%' r   r   )   r=  r=  )r0  <lambda>r>  r  )r   r#   r%   r&   )r   rD   r(   r)   s       r   !test_series_apply_no_suffix_indexr?    sI    sQwAWWe.0AB6WRFl*IJH68,r   zdti,expl    ,b/ r  r   r   rz   rZ   
2020-01-01rj   tsr  awareTc                     |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@   rB  r4   r(   s        r   1test_apply_series_on_date_time_index_aware_seriesrG    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 )	NrZ   rz   r@  rD  )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_seriesrL    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=   nanNaT)r   list_of_valid_stringsrT   r\   list_of_stringsr'   s         r   test_apply_to_timedeltarS    s    '4
-.A$%++BOOF+KA6!9a(!266266266:O
(A

!C		"//&	1A6!9a(r   z
ops, namesr0  r%  zhow, kwargsc           
          t        t        ||d      D ci c]  \  }}| ||        c}}      }d|_         t        | |      |fi |}t	        j
                  ||       y c c}}w )NT)strictr   )r   zipr2   r!  r%   r&   )	r   opsnamesr6  r^   r2   r   r)   r(   s	            r   test_apply_listlike_reducerrY    sj     14UC1MNXT2r- 	 NH HM(W]C(77F68,	 	Os   A#
rW  r   r   c           
          t        |j                         D ci c]  \  }}| ||        c}}      }| j                  |_         t        | |      |fi |}t	        j
                  ||       y c c}}w r   )r   itemsr2   r!  r%   r&   )	r   rW  r6  r^   r   r2   r   r)   r(   s	            r   test_apply_dictlike_reducerr\  ,  sh     syy{K84tR..KLH!&&HM(W]C(77F68, Ls   A/
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   rW  rX  r   r   r)   r(   s          r   test_apply_listlike_transformerr^  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   r>  r8   c                 "    | j                         S r   r;  r!   s    r   r"   r"   W  s    AEEG r   r  c                 p    t        g d      }|j                  | |      }t        j                  ||       y Nr8   r   r   r#   r%   rK  )rW  r)   r   r'   r(   s        r   test_apply_listlike_lambdard  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   r[  r2   r#   r%   r&   )r   rW  r   r2   r   r)   r(   s          r   test_apply_dictlike_transformerrf  a  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"   w  s    A r   r   c                 "    | j                         S r   r;  r!   s    r   r"   r"   z  s     r   c                 p    t        g d      }|j                  | |      }t        j                  ||       y rb  rc  )rW  r   r)   r'   r(   s        r   test_apply_dictlike_lambdarj  s  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                 ?)rn  ro  r5  rm  )r   r4   )	r   r   r   r   r#   r=   rO  r%   r   )r   dfr(   r)   s       r   test_apply_retains_column_namerq    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)rT   r\   rV   r  )r   floatr#   r   r   rO   r%   r&   r  s      r   test_apply_typeru    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   )rg   r   r  r   c                     t        |       S r   r   r!   s    r   r"   z6test_series_apply_unpack_nested_data.<locals>.<lambda>  s
     r   ro  g      @r5  g      @r4  g      @r   )r   r9   r:   r   )r   r#   r   r=   rO  r%   r   )r'   r(   r)   s      r   $test_series_apply_unpack_nested_datarx    sX    
)\*
+CYY*+Fc3ZS#JC:266ST+VWH&(+r   )Knumpyr=   rr   pandasr   r   r   r   r   r   r   r	   pandas._testing_testingr%   pandas.tests.apply.commonr
   fixturer   r*   rF   rJ   rS   r   parametrizer`   rh   rv   r   r   r   r   r   r   rO  r   r   r   r   r  r  r  rO   r  r  r"  r.  r1  r8  r?  DatetimeIndexrepeatrJ  rB   rG  onesrL  rS  r0  r%  arrayrY  r\  r>   r   r^  rd  r@   rf  from_arraysr   rj  rq  ru  rx  r6   r   r   <module>r     s         > x() *- "8-' "a[2Qx#fb"%5sAh7LM-	---$%	%%%&6G4 UE266$:UE266<R#ST- U- &
-#HL 78 7 9 7F>0
- #'7!89	, :	, UG$451 61-$- 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   