
    A>ib                        S SK J r   S SKrS SKrS SKrS SKJr  S SKJr  S SK	r
S SK	JrJrJrJrJr  S SKJr  S SKJr  S SKJr  S SKJr  \R2                  S	 5       r\R2                  " S
\R6                  " S\R8                  R:                  S9\/S9S 5       rS r\R8                  RA                  SS S/5      \R8                  RA                  SSS/5      \R8                  RA                  SSS/5      S 5       5       5       r!S r"S r#\R8                  RA                  SSS\" S/5      4\" SSS9S4/5      S 5       r$S r%\R8                  RA                  S \RL                  \RN                  /5      S! 5       r(S" r)S# r*S$ r+\R8                  RA                  S / S%Q5      S& 5       r,S' r-S( r.S) r/S* r0S+ r1S, r2S- r3S. r4\R8                  RA                  SS S/5      S/ 5       r5\R8                  RA                  SS S/5      S0 5       r6S1 r7\R8                  RA                  SS S/5      S2 5       r8S3 r9S4 r:S5 r;\R8                  Ry                  S65      \R8                  RA                  S7S8S9/5      \R8                  RA                  S S: S; /S<S=/S>9\R8                  RA                  SSS/5      \R8                  RA                  SS S/5      S? 5       5       5       5       5       r=S@ r>SA r?SB r@SC rASD rBSE rCSF rDSG rESH rFSI rGSJ rHSK rISL rJSM rKSN rL\R8                  RA                  SO\" SPSQ/SRSS//5      \" SPSRST.SQSSST./5      /\" S S/SUSV//5      \" S SUST.SSVST./5      //5      SW 5       rMSX rNSY rOSZ rP\R8                  RA                  S[S\S]/5      S^ 5       rQS_ rRS` rSSa rTSb rU\R8                  RA                  ScSdSeS\R                  /5      Sf 5       rWSg rX\R8                  RA                  ShSiSj 0Sk\" SiSUSV/05      4SiSl 0S\" SiSUSV/05      4SiSm 0Sk\" SiSV05      4SiSn 0S\" SiSV05      4SiSo\R                  Sp /0Sk\" Si/ SqQ0/ SrQSs94SiSo\R                  St /0S\" Si/ SqQ0/ SrQSs94SiSu 0Sk\" SiSS/05      4SiSv 0S\" SiS05      4/5      Sw 5       rZ\R8                  RA                  SxSiSy 0SiSz 0SiSo\R                  S{ /0SiS| 0/5      S} 5       r[S~ r\S r]S r^\R8                  RA                  ShS /Sk\" SSUSV/05      4S /S\" SSUSV/05      4S /Sk\" SiSV/0S/Ss94S /S\" SiSV/0S/Ss94So\R                  S /Sk\" Si/ SqQ0/ SrQSs94So\R                  S /S\" Si/ SqQ0/ SrQSs94S S /Sk\" SUSV/SVSV//SiSi/SS//S94S S /S\" SiSUSV/0SS/5      4/5      S 5       r_S r`S ra\R8                  RA                  SxS /S /So\R                  S /S S //5      S 5       rbS rcS rdS reS rf\R8                  RA                  ScSSU/5      S 5       rgS rh\R8                  RA                  S/ SQSSU//5      S 5       riS rjS rkS rlS rmS rnS roS rp\R8                  RA                  SS S S // SQS>9S 5       rqS rrS rsS rtS ruS rvS rwS rxS ryS rz\R8                  RA                  SS]S\/5      S 5       r{\R8                  RA                  SS]S\/5      S 5       r|S r}\R8                  RA                  SS S/5      \R8                  RA                  SS0 4S0 4S0 4SSSU04SSSUS.4SSUSS.4SSSUSVS.4/5      S 5       5       r~\R8                  RA                  S/ SQ5      S 5       rS r\R8                  RA                  S\" SiS/SS/S.5      /5      \R8                  RA                  S/ SQ5      S 5       5       r\R8                  RA                  SSSSSi\R                  /5      S 5       rS rS rS rS rS rS rS rS rS rS r\R8                  RA                  S/ SQ/ SQS./ SQS/S.4/ SQ/ SQS.SU/S/S.4/5      S 5       rS rS rS rS rS r\R8                  RA                  SSS/5      S 5       rg)    )datetimeN)is_platform_arm)CategoricalDtype)	DataFrame
MultiIndexSeries	Timestamp
date_range)MockEngineDecorator)
zip_frames)Versionc            	          [        [        R                  " [        R                  " SSS9S5      R	                  SS5      S-   / SQS9n U $ )	z
Fixture for DataFrame of ints which are constant per column

Columns are ['A', 'B', 'C'], with values (per column): [1, 2, 3]
   int64dtype      ABCcolumns)r   nptilearangereshape)dfs    b/var/www/html/land-tabula/venv/lib/python3.13/site-packages/pandas/tests/apply/test_frame_apply.pyint_frame_const_colr"      sC     

		!7+Q/772>B
B I    pythonnumba)marks)paramsc                 f    U R                   S:X  a  [        R                  " S5        U R                   $ )Nr%   )parampytestimportorskip)requests    r!   enginer-   '   s(     }}G$==r#   c                    US:X  a8  [         R                  R                  SS9nUR                  R	                  U5        [
        R                  " SS9   [
        R                  " U S   5      nU R                  [
        R                  US9S   n[        R                  " XE5        U R                  [
        R                  US9S   n[
        R                  " U S   5      nXE:X  d   eU R                  S   nU R                  [
        R                  S	US
9n[
        R                  " U R                  U5      5      nXF   U:X  d   eUR                  U R                  L d   e S S S 5        g ! , (       d  f       g = f)Nr%   z+numba engine not supporting numpy ufunc yetreasonignoreallr   r-   r   r   axisr-   )r*   markxfailnode
add_markerr   errstatesqrtapplytmassert_series_equalmeanindexxs)float_framer-   r,   r7   resultexpectedds          r!   
test_applyrG   4   s4   {{  (U V%		"S)*$$RWWV$<SA
v0 ""2776":3?77;s+,!!!a ""2776"B77;>>!,-yH$$$||{00000 
#	"	"s   D
E''
E5r6   r   rawTFnopythonc           
         [         R                  " S5      nUS:X  aT  [        UR                  5      [        S5      :X  a2  [	        5       (       a#  [         R
                  " SUR                   35        SU0nU R                  S USUUUS9nU S-   n[        R                  " Xx5        U R                  S	 SS
UUUS9nU S-   n[        R                  " Xx5        US:X  a  [         R                  " [        SS9   U R                  S S
UUUS9  S S S 5        [         R                  " [        R                  R                  SS9   U R                  S SS
UUUS9  S S S 5        [         R                  " [        R                  R                  SS9   U R                  S SS
UUUS9  S S S 5        g g ! , (       d  f       N= f! , (       d  f       Nl= f! , (       d  f       g = f)Nr%   z0.61z&Segfaults on ARM platforms with numba rI   c                 
    X-   $ N xys     r!   <lambda>!test_apply_args.<locals>.<lambda>W   s    QUr#   r   )argsrH   r-   engine_kwargsr   c                     X-   U-   $ rL   rM   rO   abs      r!   rQ   rR   c   s
    	r#      )rT   rY   rH   r-   rU   r   z missing a required argument: 'a'matchc                 
    X-   $ rL   rM   )rO   rX   s     r!   rQ   rR   q   s    QUr#   )rY   rH   r-   rU   z-numba does not support keyword-only argumentsc                    X-   U-   $ rL   rM   rW   s      r!   rQ   rR   ~   s
    1519r#   c                     US   US   -   U -   $ Nr   r   rM   )rY   rO   s     r!   rQ   rR      s    adQqTkAor#   )r*   r+   r   __version__r   skipr=   r>   assert_frame_equalraises	TypeErrorpderrorsNumbaUtilError)	rC   r6   rH   r-   rI   r%   rU   rD   rE   s	            r!   test_apply_argsri   J   s    (E'E%%&'&/9<U=N=N<OPQ*M#  F QH&+ !
#  F QH&+]]9,NO"+   P ]]II$$A
 ,+  	
 ]]II$$A
 -+  	
 
3 OO
 

 
s$   4F)>F:	G)
F7:
G
Gc                      [        / SQ/ SQS.5      n U R                  S 5      nUR                  S:X  d   e[        US   R                  [
        5      (       d   e[        US   R                  [
        5      (       d   eg )N)r   r   r   r   )r   r   Drk   c0c1c                 $    U R                  S5      $ )Ncategoryastype)tss    r!   rQ   -test_apply_categorical_func.<locals>.<lambda>   s    :!6r#   )   rZ   rm   rn   )r   r=   shape
isinstancer   r   r    rD   s     r!   test_apply_categorical_funcry      sp    	.6JK	LBXX67F<<6!!!fTl((*:;;;;fTl((*:;;;;r#   c                  ~    [        S[        SSS9/05      n U R                  S SS9n[        R                  " X5        g )Nr   
2013-01-01UTCtzc                     U $ rL   rM   rO   s    r!   rQ   *test_apply_axis1_with_ea.<locals>.<lambda>       ar#   r   r6   r   r	   r=   r>   rc   rE   rD   s     r!   test_apply_axis1_with_ear      s;    #	,5 ABCDH^^Ka^0F&+r#   zdata, dtype)r   Nr{   r|   r}   c                 t    [        U /U //SS/US9nUR                  S SS9n[        R                  " X25        g )NrX   )rA   r   c                     U $ rL   rM   r   s    r!   rQ   0test_agg_axis1_duplicate_index.<locals>.<lambda>       Ar#   r   r   r   aggr>   rc   )datar   rE   rD   s       r!   test_agg_axis1_duplicate_indexr      s?     4&4&)#s5IH\\+A\.F&+r#   c                      [        [        SSS9[        R                  " [        R
                  " S5      SS9S.5      n U R                  S SS	9n[        R                  " X5        g )
N20130101r   periodss)unitr   r   c                     U $ rL   rM   r   s    r!   rQ   /test_apply_mixed_datetimelike.<locals>.<lambda>   r   r#   r   r   )	r   r
   rf   to_timedeltar   r   r=   r>   rc   r   s     r!   test_apply_mixed_datetimeliker      sU     J21C8	
H ^^Ka^0F&+r#   funcc                 \    [        5       nUR                  XS9nUR                  (       d   eg )Nr4   )r   r=   empty)r   r-   empty_framerD   s       r!   test_apply_emptyr      s*     +Kt3F<<<r#   c                 Z   U S S nUR                  S US9n[        [        R                  U R                  S9n[
        R                  " X45        U R                  S S 2/ 4   nUR                  S SUS9n[        [        R                  U R                  S9n[
        R                  " X45        g )Nr   c                 "    U R                  5       $ rL   r@   r   s    r!   rQ   (test_apply_float_frame.<locals>.<lambda>   
    QVVXr#   r4   rA   c                 "    U R                  5       $ rL   r   r   s    r!   rQ   r      r   r#   r   r5   )	r=   r   r   nanr   r>   r?   locrA   )rC   r-   no_rowsrD   rE   no_colss         r!   test_apply_float_framer      s    "1oG]]-f]=FbffK$7$78H6,ooae$G]]-Af]EFbffK$5$56H6,r#   c                 h    [        S/S9nUR                  S SU S9n[        R                  " X!5        g )NrX   r   c                     U S   $ )NrX   rM   r   s    r!   rQ   /test_apply_empty_except_index.<locals>.<lambda>   s    afr#   r   r5   r   r=   r>   rc   )r-   rE   rD   s      r!   test_apply_empty_except_indexr      s1    u%H^^,1V^DF&+r#   c                  *   [        5       n / nU R                  UR                  SSS9n[        R                  " X 5        U R                  UR                  SSS9n[        / [        R                  S9n[        R                  " X#5        [        / SQS9nUR                  UR                  SSS9n[        R                  " X$5        UR                  UR                  SSS9n[        / [        R                  S9n[        R                  " X#5        U/ :X  d   eg )Nr   expandr6   result_typereducer   rX   rY   cr   )	r   r=   appendr>   rc   r   r   float64r?   )r   rO   rD   rE   empty_with_colss        r!   test_apply_with_reduce_emptyr      s    +K
AqxxaXFF&.qxxaXFFb

+H6,8O""188!"JF&2""188!"JFb

+H6, 7N7r#   )sumprodanyr3   c                     [        / SQS9nUR                  [        [        U 5      5      n[        X5      " 5       nU S;   a  UR	                  [
        5      n[        R                  " X#5        g )Nr   r   )r   r   )r   r=   getattrr   rr   floatr>   r?   )r   r    rD   rE   s       r!   test_apply_funcs_over_emptyr      sT     
?	+BXXgb$'(Fr "H??5)6,r#   c                     [        / SQS9n U R                  5       n[        SU R                  S9n[        R
                  " X5        U R                  R                  5       n[        / [        R                  S9n[        R
                  " X5        g )Nr   r   r   r   r   )	r   nuniquer   r   r>   r?   Tr   r   r    rD   rE   s      r!   test_nunique_emptyr      sb    	?	+BZZ\Farzz*H6,TT\\^Fb

+H6,r#   c                      [        / SQ/ SQ/ SQ// SQS9n U R                  S SS9n[        / S	Q/ SQ5      n[        R                  " X5        U R
                  R                  S
 SS9n[        R                  " X5        g )Nr   rZ   r   ru      r         	   )rX   rX   r   r   c                     U S   $ Nr   rM   r   s    r!   rQ   /test_apply_standard_nonunique.<locals>.<lambda>      !r#   r   r   )r   ru   r   c                     U S   $ r   rM   r   s    r!   rQ   r     s    !A$r#   r   )r   r=   r   r>   r?   r   r   s      r!   test_apply_standard_nonuniquer     se    	Iy)4O	LBXXn1X-Fi1H6,TTZZQZ/F6,r#   c                     U R                  [        R                  SS9n[        U R                  5       /U R                  S9n[
        R                  " X5        g )N	broadcastr   r   )r=   r   r@   r   rA   r>   rc   rC   rD   rE   s      r!   test_apply_broadcast_scalarsr     sG    rwwK@F+**,-[5F5FGH&+r#   c                     U R                  [        R                  SSS9nU R                  SS9n[        [        R                  U R                  U5      5      n[        R                  " X5        g )Nr   r   r   r   )	r=   r   r@   r   dictfromkeysr   r>   rc   rC   rD   mrE   s       r!   "test_apply_broadcast_scalars_axis1r     sY    rwwQKHFa A{':':A>?H&+r#   c                   ^  T R                  U 4S jSSS9n[        [        [        T R                  5      5      5      n[        U/[        T R                  5      -  ST R                  T R                  S9n[        R                  " X5        g )Nc                 R   > [        [        [        TR                  5      5      5      $ rL   )listrangelenr   rO   rC   s    r!   rQ   4test_apply_broadcast_lists_columns.<locals>.<lambda>$  s    $uS!4!4567r#   r   r   r   r   )r   rA   r   )	r=   r   r   r   r   r   rA   r>   rc   r   s   `   r!   "test_apply_broadcast_lists_columnsr   !  s    7  F
 	U3{**+,-A	
c+##$$##	H &+r#   c                   ^  T R                  U 4S jSS9n[        [        [        T R                  5      5      5      n[        [        R                  T R                  U5      ST R                  S9n[        R                  " X5        g )Nc                 R   > [        [        [        TR                  5      5      5      $ rL   )r   r   r   rA   r   s    r!   rQ   2test_apply_broadcast_lists_index.<locals>.<lambda>4  s    $uS!2!2345r#   r   r   r   )r   rA   )r=   r   r   r   rA   r   r   r   r   r>   rc   r   s   `   r!    test_apply_broadcast_lists_indexr   2  ss    5;  F 	U3{(()*+Ak))1-H
 &+r#   c                 X    U nUR                  S SSS9n[        R                  " X!5        g )Nc                 
    / SQ$ Nr   rM   r   s    r!   rQ   7test_apply_broadcast_list_lambda_func.<locals>.<lambda>B      	r#   r   r   r   r=   r>   rc   )r"   r    rD   s      r!   %test_apply_broadcast_list_lambda_funcr   ?  s*    	BXX){XKF&%r#   c                 x    U nUR                  S SSS9nUR                  5       n[        R                  " X#5        g )Nc                 ,    [        / SQ[        S5      S9$ )Nr   abcr   )r   r   r   s    r!   rQ   9test_apply_broadcast_series_lambda_func.<locals>.<lambda>I  s    &$u+6r#   r   r   r   )r=   copyr>   rc   r"   r    rD   rE   s       r!   'test_apply_broadcast_series_lambda_funcr   F  s>    	BXX6  F
 wwyH&+r#   c                 d    US:X  a  [         R                  " S5        S nU R                  X1USS9  g )Nr%   z)numba can't handle when UDF returns None.c                 j    [        U [        R                  5      (       d   eU R                  S:X  d   eg Nr   rw   r   ndarrayndimr   s    r!   _assert_raw/test_apply_raw_float_frame.<locals>._assert_rawV  )    !RZZ((((vv{{r#   Tr6   r-   rH   )r*   rb   r=   )rC   r6   r-   r  s       r!   test_apply_raw_float_framer  Q  s3    ?@ kVFr#   c                     U R                  [        R                  XSS9nU R                  S US9n[        R                  " X45        g )NTr  c                 6    U R                   R                  5       $ rL   )valuesr@   r   s    r!   rQ   3test_apply_raw_float_frame_lambda.<locals>.<lambda>`  s    188==?r#   r   )r=   r   r@   r>   r?   )rC   r6   r-   rD   rE   s        r!   !test_apply_raw_float_frame_lambdar
  ]  sA    rwwTdKF  !: FH6,r#   c                 ^    U R                  S USS9nU S-  n[        R                  " X#5        g )Nc                     U S-  $ NrZ   rM   r   s    r!   rQ   9test_apply_raw_float_frame_no_reduction.<locals>.<lambda>f  s    Qr#   T)r-   rH   rZ   r   )rC   r-   rD   rE   s       r!   'test_apply_raw_float_frame_no_reductionr  d  s0    v4HFQH&+r#   c                    US:X  a  [         R                  " S5        S n[        SSS[        R                  " S/S-  SS	9[        R                  " S
/S-  SS	9S.[        R
                  " S5      S9nUR                  X USS9  g )Nr%   z(isinstance check doesn't work with numbac                 j    [        U [        R                  5      (       d   eU R                  S:X  d   eg r   r   r   s    r!   r  4test_apply_raw_mixed_type_frame.<locals>._assert_rawp  r  r#         ?rZ   foo
   float32r   r   int32)rX   rY   r   r  r  r   Tr  )r*   rb   r   r   arrayr   r=   )r6   r-   r  r    s       r!   test_apply_raw_mixed_type_framer  k  s    >?
 
xx
)<XXqcBhg6	
 iim	
B HH[FH=r#   c                     U R                   S   nU R                  [        R                  SS9U   n[        R                  " U R	                  U5      5      nX#:X  d   eg )Nr   r   r   )rA   r=   r   r@   rB   )rC   rF   rD   rE   s       r!   test_apply_axis1r    sS    !ArwwQ/2Fww{~~a()Hr#   c                      [        S/S/S.5      n U S S R                  [        R                  SS9n[	        [        R
                  S9n[        R                  " X5        g )Nr  r  r   r   r   r   r   )r   r=   r   r@   r   r   r>   r?   r   s      r!   test_apply_mixed_dtype_cornerr    sO    	%u-	.BV\\"''\*F BJJ'H6,r#   c                     [        S/S/S.5      n U R                  S SS9n[        S/[        S5      S9n[        R
                  " X5        U R                  S SS9n[        S/[        S5      S9n[        R
                  " X5        g )	Nr  r  r   c                     U S   $ )Nr   rM   r   s    r!   rQ   8test_apply_mixed_dtype_corner_indexing.<locals>.<lambda>      #r#   r   r   r   c                     U S   $ )Nr   rM   r   s    r!   rQ   r     r!  r#   )r   r=   r   r   r>   r?   r   s      r!   &test_apply_mixed_dtype_corner_indexingr#    s{    	%u-	.BXX&QX/FugU1X.H6,XX&QX/Fse58,H6,r#   zignore::RuntimeWarningaxrA   r   c                     U $ rL   rM   r   s    r!   rQ   rQ         qr#   c                 "    U R                  5       $ rL   r   r   s    r!   rQ   rQ     s
    AFFHr#   identityr@   )idsc                    [        S0 U / SQ0D6n[        R                  " SS9   U" [        R                  " / SS95      n[	        U[        R
                  5      (       + nUR                  XXBS9n	U(       a:  UR                  U5      n
[	        U	[        5      (       d   eU	R                  U
L d   eO[	        U	[         5      (       d   eS S S 5        g ! , (       d  f       g = f)Nr   r1   r2   f8r   r  rM   )
r   r   r;   r  rw   r   r=   _get_agg_axisr   rA   )r$  r   rH   r6   r-   r,   r    test_resis_reductionrD   agg_axiss              r!   test_apply_empty_infer_typer0    s     
	+b/*	+B		"401%h

;;$&B''-Hff----<<8+++fi0000 
#	"	"s   B#C
Cc                  j    [        / SQS9n U R                  S SS9n[        U[         5      (       d   eg )Nr   r   c                 "    U R                  5       $ rL   r   r   s    r!   rQ   7test_apply_empty_infer_type_broadcast.<locals>.<lambda>  r   r#   r   r   )r   r=   rw   )r   rD   s     r!   %test_apply_empty_infer_type_broadcastr4    s2    o.G]]-;]GFfi((((r#   c                 ~    SS jnU R                  USS9nU R                  S 5      n[        R                  " X#5        g )Nc                 
    X-   $ rL   rM   rO   howmuchs     r!   add_some4test_apply_with_args_kwds_add_some.<locals>.add_some  s
    {r#   rZ   r8  c                     U S-   $ r  rM   r   s    r!   rQ   4test_apply_with_args_kwds_add_some.<locals>.<lambda>  s    1q5r#   r   r   )rC   r9  rD   rE   s       r!   "test_apply_with_args_kwds_add_somer?    s<     x3F  1H&+r#   c                 ~    SS jnU R                  USS9nU R                  S 5      n[        R                  " X#5        g )Nc                 (    U R                  5       U-   $ rL   r   r7  s     r!   agg_and_add:test_apply_with_args_kwds_agg_and_add.<locals>.agg_and_add  s    vvx'!!r#   rZ   r;  c                 (    U R                  5       S-   $ r  r   r   s    r!   rQ   7test_apply_with_args_kwds_agg_and_add.<locals>.<lambda>  s    1668a<r#   r>  )r=   r>   r?   )rC   rB  rD   rE   s       r!   %test_apply_with_args_kwds_agg_and_addrF    s=    " {A6F  !78H6,r#   c                     SS jnU R                  USSS9nU R                  S 5      n[        R                  " X#5        g )Nc                     X-
  U-  $ rL   rM   )rO   subdivides      r!   subtract_and_divideJtest_apply_with_args_kwds_subtract_and_divide.<locals>.subtract_and_divide  s    6!!r#   )rZ   rZ   )rT   rJ  c                     U S-
  S-  $ )N       @rM   r   s    r!   rQ   ?test_apply_with_args_kwds_subtract_and_divide.<locals>.<lambda>  s    AGs?r#   rS   r   )rC   rK  rD   rE   s       r!   -test_apply_with_args_kwds_subtract_and_dividerP    s@    " 2aHF  !:;H&+r#   c                 Z    U R                  [        5      n[        R                  " X5        g rL   )r=   r   r>   rc   )rC   rD   s     r!   test_apply_yield_listrR    s     t$F&.r#   c                     [         R                  U R                  S S S2U R                  R	                  S5      4'   U R                  SS9nU R                  [         R
                  SS9n[        R                  " X!5        g )NrZ   r   r   r   )	r   r   ilocr   get_locr@   r=   r>   r?   )rC   rE   rD   s      r!   test_apply_reduce_SeriesrV    sj    >@ffKSqS+--55c::;Q'HrwwQ/F6,r#   c                  F   [        SS/SS//SS/SS/S	9n U R                  [        S
S9n[        SSS.SSS./U R                  S9n[
        R                  " X5        U R                  [        SS9n[        SSS.SSS./U R                  S9n[
        R                  " X5        g )Nr   rZ   r   ru   rm   rn   i0i1r   rA   r   r   )rX  rY  r   rl   )r   r=   r   r   r   r>   r?   rA   )r   rD   rE   s      r!   test_apply_reduce_to_dictr[    s    q!fq!f%d|D$<PDZZ1Z%Faq)!1+=>dllSH6,ZZ1Z%Faq)!1+=>djjQH6,r#   c            	         [        [        R                  R                  S5      R	                  S5      5      n U R                  [        R                  SS9n[        U R                  5        VVs0 s H  u  p#X#R                  5       _M     snnU R                  S9n[        R                  " X5        U R                  [        R                  SS9n[        U R                  R                  5        VVs0 s H  u  p#X#R                  5       _M     snnU R                  S9R                  n[        R                  " X5        g s  snnf s  snnf )NrZ   )   r  r   r   r   r   )r   r   randomdefault_rngstandard_normalr=   r   describeitemsr   r>   rc   r   rA   )r    rD   ivrE   s        r!   test_apply_differently_indexedre    s    	299((+;;HE	FBXXfooAX.FbhhjAjda!ZZ\/jA2::VH&+XXfooAX.FbddjjlClda!ZZ\/lCRXXVXXH&+ B Ds   .D?
1E
c                  ~   [        / SQ/ SQ/ SQ/ SQ/ SQ/ SQ// SQS9n S	 nU R                  US
S9n[        [        SS
S
5      SS/[        SS
S5      SS/[        SS
S5      SS/[        SS
S5      SS/[        SS
S5      SS/[        SS
S5      SS/// SQS9n U R                  US
S9n[        R                  " X25        g )N)r   ABC02   )r   YUM0r]  )r   DEF0r]  )rZ   ABC1rh  )rZ   YUM1r]  )rZ   DEF1r]  )rX   marketpositionr   c                     U S   $ )Nrn  rM   )rs    r!   ftest_apply_bug.<locals>.f  s    {r#   r   r   i  rg  rh  rZ   ri  r]  r   rj  ru   rk  r   rl  r   rm  )r   r=   r   r>   r?   )	positionsrr  rE   rD   s       r!   test_apply_bugru    s    	
 ,
I qq)HdAq!62.dAq!62.dAq!62.dAq!62.dAq!62.dAq!62.	
 ,
I __QQ_'F6,r#   c            
         [        / SQ/ SQ/ SQ[        R                  R                  S5      R	                  S5      [        R                  R                  S5      R	                  S5      [        R                  R                  S5      R	                  S5      S.5      n U R                  S SS	9n[        R                  " X5        g )
N)r  r  r  r  barrw  rw  rw  r  r  r  )onerx  rx  tworx  rx  rx  ry  ry  ry  rx  )dullrz  shinyrz  rz  r{  r{  rz  r{  r{  r{  rZ      )r   r   r   rk   EFc                     U $ rL   rM   r   s    r!   rQ   ,test_apply_convert_objects.<locals>.<lambda>L  r   r#   r   r   )r   r   r^  r_  r`  r=   r>   rc   r   s     r!   test_apply_convert_objectsr    s     &&q)99"=&&q)99"=&&q)99"=U+	
-H^ ^^Ka^0F&+r#   c                     U R                  S 5      n[        U R                  U R                  S9n[        R                  " X5        g )Nc                     U R                   $ rL   namer   s    r!   rQ   (test_apply_attach_name.<locals>.<lambda>Q      r#   r   )r=   r   r   r>   r?   r   s      r!   test_apply_attach_namer  P  s9    /0Fk))1D1DEH6,r#   c                     U R                  S SS9n[        U R                  U R                  S9n[        R                  " X5        g )Nc                     U R                   $ rL   r  r   s    r!   rQ   .test_apply_attach_name_axis1.<locals>.<lambda>W  r  r#   r   r   r   )r=   r   rA   r>   r?   r   s      r!   test_apply_attach_name_axis1r  V  s>    /a8Fk''{/@/@AH6,r#   c                     U R                  S 5      n[        [        R                  " U R                  [        U R                  5      S45      U R                  U R                  S9n[        R                  " X5        g )Nc                 V    [         R                  " U R                  [        U 5      5      $ rL   r   repeatr  r   r   s    r!   rQ   6test_apply_attach_name_non_reduction.<locals>.<lambda>^      1663q6)Br#   r   rA   r   )	r=   r   r   r   r   r   rA   r>   rc   r   s      r!   $test_apply_attach_name_non_reductionr  \  sc    BCF
##c+*;*;&<a%@A##H
 &+r#   c                    ^  T R                  S SS9n[        U 4S jT R                  5        5       5      nT R                  Ul        [        R
                  " X5        g )Nc                 V    [         R                  " U R                  [        U 5      5      $ rL   r  r   s    r!   rQ   <test_apply_attach_name_non_reduction_axis1.<locals>.<lambda>h  r  r#   r   r   c              3   ~   >#    U  H2  n[         R                  " US    [        TR                  5      5      v   M4     g7f)r   N)r   r  r   r   ).0trC   s     r!   	<genexpr>=test_apply_attach_name_non_reduction_axis1.<locals>.<genexpr>i  s2      ;Sa		!A$K//011;Ss   :=)r=   r   
itertuplesrA   r>   r?   r   s   `  r!   *test_apply_attach_name_non_reduction_axis1r  g  sV    BKF ;F;Q;Q;S H !&&HN6,r#   c                      [         R                  " / SQ/ SQ/5      n [        SS/SS/SS//U S	S
/S9nUR                  S S5      n[        SS/SS/SS//U SS/S9n[        R
                  " X#SS9  g )N)rX   rX   rY   )r   rF   rF   r   rZ   r   ru   r   r   col1col2r  c                 B    [        [        U 5      [        U 5      S.5      $ )N)minmax)r   r  r  r   s    r!   rQ   (test_apply_multi_index.<locals>.<lambda>s  s    vc!fSV&DEr#   r  r  T
check_like)r   from_arraysr   r=   r>   rc   )rA   r   rD   rE   s       r!   test_apply_multi_indexr  p  s    ""O_#EFEAq6Aq6Aq6*%&&AQRAWWEqIF1a&1a&1a&1PUWH&t<r#   z	df, dictsr  rw  spameggs)r   r   rZ   r   c                     S nU R                  USS9nU R                  USS9nU R                  U5      n[        R                  " X15        [        R                  " X@5        [        R                  " XQ5        g )Nc                 "    U R                  5       $ rL   )to_dictr   s    r!   rQ   !test_apply_dict.<locals>.<lambda>  s
    199;r#   r   r   r   )r=   r>   r?   rc   )r    dictsfnreduce_truereduce_falsereduce_nones         r!   test_apply_dictr  x  sd     
B((28(4K88BH85L((2,K;.,+;.r#   c            	          [        S[        SSSS905      n U R                  S 5      n[        R                  " X5        U R                  S 5      n[        S[        SSSS905      n[        R                  " X5        g )	Ndtz
2015-01-01r   zEurope/Brussels)r   r~   c                     U $ rL   rM   r   s    r!   rQ   ,test_apply_non_numpy_dtype.<locals>.<lambda>      r#   c                 4    U [         R                  " S5      -   $ )N1day)rf   	Timedeltar   s    r!   rQ   r    s    BLL$8 8r#   z
2015-01-02)r   r
   r=   r>   rc   r   s      r!   test_apply_non_numpy_dtyper    sr    	D*\1ARST	UBXXk"F&%XX89F	z,6GHIH &+r#   c                  p    [        S/ SQ0SS9n U R                  S 5      n[        R                  " X5        g )Nr  )rX   rY   r   rX   rp   r   c                     U $ rL   rM   r   s    r!   rQ   5test_apply_non_numpy_dtype_category.<locals>.<lambda>  r  r#   r   rx   s     r!   #test_apply_non_numpy_dtype_categoryr    s0    	D./z	BBXXk"F&%r#   c                      [        SS/SS//SS/S9n [        SS//SS/S/S9nU R                  S/5      n[        R                  " X!5        g )	Nr   r   rZ   r   rX   r   r  rZ  r   r    rE   rD   s      r!   test_apply_dup_names_multi_aggr    sV    	QFQF#c3Z	8B1a&C:eWEHVVUG_F&+r#   opr=   r   c                     S n[        [        R                  " S5      [        S5      S9n[	        X 5      " USS9n[        / SQ/ SQ/ SQ/ SQ/5      n[        R                  " X45        g )Nc                 0    SU S   -  SU S   -  SU S   -  /$ )NrZ   r   r   r   rM   rows    r!   
apply_list3test_apply_nested_result_axis_1.<locals>.apply_list  s'    CHa#c(lACL99r#   )ru   ru   ABCDr   r   r   )        r  r  )r   r   zerosr   r   r   r>   r?   )r  r  r    rD   rE   s        r!   test_apply_nested_result_axis_1r    sX    : 
288F#T&\	:BR_Za0F	/?OLH 6,r#   c                      [        S[        SSS9/0SS9n U R                  S 5      n[        R                  " X5        U R                  S 5      n[        R                  " X5        g )	Nr  2020r|   r}   zdatetime64[ns, UTC]r   c                     U $ rL   rM   r   s    r!   rQ   7test_apply_noreduction_tzaware_object.<locals>.<lambda>  r   r#   c                 "    U R                  5       $ rL   )r   r   s    r!   rQ   r    s
    affhr#   r   r   s     r!   %test_apply_noreduction_tzaware_objectr    s\    	6e,-.6KH ^^K(F&+^^./F&+r#   c                     ^ [        S/ SQ05      n / mU4S jnU4S jnX4 H3  nTS S 2	 U R                  USS9  T[        U R                  5      :X  a  M3   e   g )NrX   r   c                 <   > TR                  U R                  5        g rL   r   r  r  namess    r!   reducing_function8test_apply_function_runs_once.<locals>.reducing_function  s    SXXr#   c                 >   > TR                  U R                  5        U $ rL   r  r  s    r!   non_reducing_function<test_apply_function_runs_once.<locals>.non_reducing_function  s    SXX
r#   r   r   )r   r=   r   rA   )r    r  r  r   r  s       @r!   test_apply_function_runs_oncer    s_     
C#	$BE #:!H
ARXX&&&	 ;r#   c                 
  ^ U S:X  a  [         R                  " S5        [        S/ SQ05      n/ mU4S jnU4S jnX#4 HB  nTS S 2	 UR                  X@SSS	9  T[	        UR
                  R                  5       5      :X  a  MB   e   g )
Nr%   z8appending to list outside of numba func is not supportedrX   r   c                 (   > TR                  U 5        g rL   extendr  r  s    r!   r  <test_apply_raw_function_runs_once.<locals>.reducing_function  s    cr#   c                 *   > TR                  U 5        U $ rL   r  r  s    r!   r  @test_apply_raw_function_runs_once.<locals>.non_reducing_function  s    c
r#   Tr   )r-   rH   r6   )r*   rb   r   r=   r   rX   to_list)r-   r    r  r  r   r  s        @r!   !test_apply_raw_function_runs_oncer    s{    NO	C#	$BF #:1I
$Q7bddlln----	 ;r#   c                      [        [        R                  " SS/5      S/S9n [        [        R                  " SS/5      S/[        S9nU R	                  S 5      n[
        R                  " X!5        g )Ns   abcds   efghcolr   )r   r   c                 $    U R                  S5      $ )Nobjectrq   r   s    r!   rQ   -test_apply_with_byte_string.<locals>.<lambda>  s     2r#   )r   r   r  r  r=   r>   rc   r  s      r!   test_apply_with_byte_stringr    s[    	288Wg./%	AB7G"45wfUH XX23F&+r#   valasd   c                 4  ^  SS SSS[         R                  /n[        SU0SS9nUR                  R	                  U 4S j5      n[        U Vs/ s H%  n[        R                  " U5      (       a  SOUT :H  PM'     snSS	9n[        R                  " X55        g s  snf )
Nr  r  cderX   rp   r   c                    > U T:H  $ rL   rM   )rO   r  s    r!   rQ   /test_apply_category_equalness.<locals>.<lambda>  s	    !s(r#   Fr  )
r   r   r   rX   r=   r   rf   isnullr>   r?   )r  	df_valuesr    rD   rO   rE   s   `     r!   test_apply_category_equalnessr    s     b%7I	C#:	6BTTZZ*+F6?@i"))A,,AH	,i@sH 6, 	As   ,Bc                  Z   [        [        R                  R                  S5      R                  S5      5      n U R	                  [        R
                  R
                  SS9R                  nUS:X  d   eU R	                  [        R
                  R                  SS9R                  nUS:X  d   eg )NrZ   )r  rZ   r   r   )r   rZ   )r   r   r^  r_  r=   fftrv   rfftrx   s     r!   test_infer_row_shaper  
  s     
299((+227;	<BXXbffjjqX)//FWXXbffkkX*00FVr#   zops, by_row, expectedrX   c                     U S-   $ r   rM   r   s    r!   rQ   rQ         Qr#   compatc                     U S-   $ r   rM   r   s    r!   rQ   rQ     r  r#   c                 "    U R                  5       $ rL   r   r   s    r!   rQ   rQ     
    r#   c                 "    U R                  5       $ rL   r  r   s    r!   rQ   rQ     r  r#   r   c                 "    U R                  5       $ rL   r  r   s    r!   rQ   rQ     
    AEEGr#   )r   r   r   )r   r   <lambda>r   c                 "    U R                  5       $ rL   r  r   s    r!   rQ   rQ   "  r	  r#   c                     gr   rM   r   s    r!   rQ   rQ   &      r#   c                     gr   rM   r   s    r!   rQ   rQ   '  r  r#   c                 l    [        SSS/05      nUR                  XS9n[        R                  " XB5        g NrX   r   rZ   by_rowr   r=   r>   assert_equalopsr  rE   r    rD   s        r!   test_dictlike_lambdar    s2    . 
C!Q=	!BXXcX)FOOF%r#   r  c                     U S-   $ r   rM   r   s    r!   rQ   rQ   4  s    Ar#   c                 "    U R                  5       $ rL   r  r   s    r!   rQ   rQ   5  s
    r#   c                 "    U R                  5       $ rL   r  r   s    r!   rQ   rQ   6  s
    r#   c                     gr   rM   r   s    r!   rQ   rQ   7  s    r#   c                     [        SSS/05      n[        R                  " [        SS9   UR	                  U SS9  S S S 5        g ! , (       d  f       g = fNrX   r   rZ   zby_row=True not allowedr[   Tr  r   r*   rd   
ValueErrorr=   r  r    s     r!   test_dictlike_lambda_raisesr!  1  B     
C!Q=	!B	z)B	C
T" 
D	C	C   A
Ac                     [        SS/SS//SS/S9n U R                  S SS9n[        U R                  5        Vs/ s H  nSS	0PM	     sn5      n[        R
                  " X5        [        S
5      [        S5      /U S'   U R                  S SS9n[        R
                  " X5        U S   U S   -   R                  S 5      n[        SS	0SS	0/5      n[        R
                  " X5        g s  snf )Nr   rZ   rX   rY   r   c                     SU S   U S   -   0$ Nr   rX   rY   rM   r   s    r!   rQ   ,test_with_dictlike_columns.<locals>.<lambda>D      afqvo 6r#   r   r   r   2017-05-01 00:00:002017-05-02 00:00:00r>   c                     SU S   U S   -   0$ r&  rM   r   s    r!   rQ   r'  L  r(  r#   c                 
    SU 0$ )Nr   rM   r   s    r!   rQ   r'  P  s    #qr#   )r   r=   r   r  r>   r?   r	   r    rD   r  rE   s       r!   test_with_dictlike_columnsr.  A  s    	QFQF#c3Z	8BXX6QX?F9AQx9:H6, 	'('(BtH XX6QX?F6, g3&&'9:FQx#q*+H6, :s   C)c                      [        5       n / SQU S'   / SQU S'   [        R                  " / SQSS9U S'   U R                  S	 S
S9n[	        0 0 0 /5      n[
        R                  " X5        g )N)XYZauthor)BBCNBCN24	publisher)z17-10-2010 07:15:30z13-05-2011 08:20:35z15-01-2013 09:09:09T)dayfirstdatec                     0 $ rL   rM   r   s    r!   rQ   :test_with_dictlike_columns_with_datetime.<locals>.<lambda>^      r#   r   r   )r   rf   to_datetimer=   r   r>   r?   r   s      r!   (test_with_dictlike_columns_with_datetimer>  U  se    	B"BxL+B{OMBvJ XXlX+Fr2rl#H6,r#   c                     [        SS/SS//SS/S9n U R                  S SSS9n[        S	S
S
/05      n[        R                  " X5        [	        S5      [	        S5      /U S'   U R                  S SSS9n[        R                  " X5        g )Nr   rZ   rX   rY   r   c                     SU S   U S   -   0$ r&  rM   r   s    r!   rQ   7test_with_dictlike_columns_with_infer.<locals>.<lambda>f  r(  r#   r   r   r   r   r)  r*  r>   c                     SU S   U S   -   0$ r&  rM   r   s    r!   rQ   rA  n  r(  r#   )r   r=   r>   rc   r	   r   s      r!   %test_with_dictlike_columns_with_inferrC  c  s    	QFQF#c3Z	8BXX6QHXUF#1v'H&+ 	'('(BtH XX6QHXUF&+r#   c                     U S-   $ r   rM   r   s    r!   rQ   rQ   u      AEr#   )rX   r
  c                     U S-   $ r   rM   r   s    r!   rQ   rQ   v  rE  r#   c                 "    U R                  5       $ rL   r  r   s    r!   rQ   rQ   w  
    AEEGr#   r
  c                 "    U R                  5       $ rL   r  r   s    r!   rQ   rQ   x  rH  r#   c                 "    U R                  5       $ rL   r  r   s    r!   rQ   rQ   z  
    aeegr#   c                 "    U R                  5       $ rL   r  r   s    r!   rQ   rQ     rK  r#   c                     U S-   $ r   rM   r   s    r!   rQ   rQ     s    q1ur#   c                     gNr   rM   r   s    r!   rQ   rQ     s    r#   r   c                     gr  rM   r   s    r!   rQ   rQ     s    qr#   c                     grO  rM   r   s    r!   rQ   rQ         Ar#   c                 l    [        SSS/05      nUR                  XS9n[        R                  " XB5        g r  r  r  s        r!   test_listlike_lambdarT  r  s2    > 
C!Q=	!BXXcX)FOOF%r#   c            
      l   [        SSS5      [        SSS5      [        SSS5      /n [        SSS/0U S   U S   /S9nUR                  S S	 /S
S9n[        SS/S[        R                  /[        R                  S//U S   U S   U S   /[        S/S//SS/SS//S9S9n[        R                  " X#5        g )N  r   rZ   r   rX   r   r   c                     U $ rL   rM   r   s    r!   rQ   7test_listlike_datetime_index_unsorted.<locals>.<lambda>  s    r#   c                      U R                  SS9$ Nrk   )freqshiftr   s    r!   rQ   rX    s    agg3g.?r#   Fr  r  rN  r
  )codesr  )r   r   r=   r   r   r   r>   rc   r  r    rD   rE   s       r!   %test_listlike_datetime_index_unsortedr`    s    tQ"HT1a$8(4A:NOF	C!Q=F1I(>	?BXX{$?@XOF
sc266]RVVSM2ay&)VAY/SEJ<0!Q!Q8HIH
 &+r#   c                  N   [        SSS5      [        SSS5      [        SSS5      /n [        SS/SS/S.U S   U S   /S9nUR                  S	 S
 S.SS9n[        SS[        R                  /S[        R                  S/S.U S   U S   U S   /S9n[
        R                  " X#5        g )NrV  r   rZ   r   ru   rX   rY   r   r   c                     U $ rL   rM   r   s    r!   rQ   7test_dictlike_datetime_index_unsorted.<locals>.<lambda>  s    r#   c                      U R                  SS9$ rZ  r\  r   s    r!   rQ   rd    s    !''s'*;r#   Fr  r  rN        @      @)r   r   r=   r   r   r>   rc   r_  s       r!   %test_dictlike_datetime_index_unsortedrh    s    tQ"HT1a$8(4A:NOF	!Qq!f-fQi5K	LBXX ;<U  F sBFF#rvvs#	
 ay&)VAY/H &+r#   c                     U S-   $ r   rM   r   s    r!   rQ   rQ         1q5r#   c                 "    U R                  5       $ rL   r  r   s    r!   rQ   rQ     s
    1557r#   c                 "    U R                  5       $ rL   r  r   s    r!   rQ   rQ     s
    !%%'r#   c                     U S-   $ r   rM   r   s    r!   rQ   rQ     rj  r#   c                     grO  rM   r   s    r!   rQ   rQ     rR  r#   c                     [        SSS/05      n[        R                  " [        SS9   UR	                  U SS9  S S S 5        g ! , (       d  f       g = fr  r  r   s     r!   test_listlike_lambda_raisesrp    r"  r#  c            
      6   [        [        [        R                  R	                  S5      R                  S5      5      / SQ[        SSSS9S.5      n U SS	/   R                  [        S
S9n[        U SS	/   R                  5        Vs/ s H  o"S
S  PM	     sn5      n[        R                  " X5        U SS/   R                  [        S
S9n[        U SS/   R                  5        Vs/ s H  o"S
S  PM	     sn5      n[        R                  " X5        g s  snf s  snf )NrZ   ru   )rX   r   ofwordsz
2016-10-01h)r   r[  )rX   rY   rs   rX   rY   r   r   rs   )r   r   r   r^  r_  r`  r
   r=   tupler  r>   r?   r-  s       r!   test_with_listlike_columnsrv    s   			--a0@@CD-\13?	

B c
^!!%a!0Fb#sn&?&?&AB&Au&ABCH6,d_""5q"1Fb#to&@&@&BC&Bu&BCDH6, C Ds   D%Dc                     [        [        SS/S//5      [        S/SS//5      S.5      n [        R                  " SS/5      U l        U R                  S	 S
S9n[        / S//U R                  S9n[        R                  " X5        g )NrX   rY   qzr  rN   )rX  j0)rY  j1c                 L    U S    Vs/ s H  oU S   ;   d  M  UPM     sn$ s  snf )NrO   rP   rM   )r  els     r!   rQ   ;test_with_listlike_columns_returning_list.<locals>.<lambda>  s"    C"K"#c(N2"K"Ks   !!r   r   r   )r   r   r   from_tuplesrA   r=   r>   r?   r   s      r!   )test_with_listlike_columns_returning_listr    s    	#scU 346C53PS*BU;VW	XB%%|\&BCBHXXKRSXTFrC5k2H6,r#   c                  (   [        SS/SS/[        S5      [        S5      /S.5      n U R                  S S	S
9n[        U R	                  5        Vs/ s H  o"R
                  UR                  4PM     sn5      n[        R                  " X5        g s  snf )Nr  rN  r  rw  z2017-11-29 03:30:00z2017-11-29 03:45:00)numberstringr   c                 2    U R                   U R                  4$ rL   )r  r  r  s    r!   rQ   1test_infer_output_shape_columns.<locals>.<lambda>  s    3::szz":r#   r   r   )	r   r	   r=   r   r  r  r  r>   r?   r-  s       r!   test_infer_output_shape_columnsr    s     
Cjen/0/0	
	
B XX:XCFR]]_E_!((+_EFH6, Fs   !Bc                     [        [        R                  R                  S5      R	                  S5      / SQS9n U R                  S SS9n[        U R                  5        Vs/ s H  n/ SQPM	     sn5      n[        R                  " X5        U R                  S	 SS9n[        U R                  5        Vs/ s H  nSS/PM	     sn5      n[        R                  " X5        g s  snf s  snf )
NrZ   )r   r   r   r   c                 
    / SQ$ r   rM   r   s    r!   rQ   :test_infer_output_shape_listlike_columns.<locals>.<lambda>  r   r#   r   r   r   c                 
    SS/$ Nr   rZ   rM   r   s    r!   rQ   r        Ar#   
r   r   r^  r_  r`  r=   r   r  r>   r?   r-  s       r!   (test_infer_output_shape_listlike_columnsr    s     

		a 008/
B XX)X2F"--/:/Qy/:;H6,XX&QX/Fr}}7!1v78H6, ; 8s   !C 4C%c                    ^  [        S/ SQ0[        S5      S9nUR                  U 4S jSS9n[        UR	                  5        Vs/ s H  n[
        R                  " T 5      PM     snUR                  S9n[        R                  " X$5        g s  snf )NrX   r   r   r   c                 0   > [         R                  " T5      $ rL   )r   ones)r  r  s    r!   rQ   Btest_infer_output_shape_listlike_columns_np_func.<locals>.<lambda>  s    "''#,r#   r   r   )
r   r   r=   r   r  r   r  rA   r>   r?   )r  r    rD   r  rE   s   `    r!   0test_infer_output_shape_listlike_columns_np_funcr    sm     
C#4;	7BXX.QX7FR]]_=_rwws|_=RXXNH6, >s    Bc                  0   [        [        S5      [        S5      [        S5      [        S5      // SQ/ SQ/ SQS.5      n S	 nU R                  US
S9n[        U R	                  5        Vs/ s H  nSPM     sn5      n[
        R                  " X$5        g s  snf )Nz
2010-02-01z
2010-02-04z
2010-02-05z
2010-02-06)r   r   ru   r   )r   r   ru   rZ   )r   rZ   r   ru   )rX   rY   r   rF   c                     g)Nr   rZ   rM   r   s    r!   funDtest_infer_output_shape_listlike_columns_with_timestamp.<locals>.fun  s    r#   r   r   r  )r   r	   r=   r   r  r>   r?   )r    r  rD   r  rE   s        r!   7test_infer_output_shape_listlike_columns_with_timestampr    s    	 ,',',','	 
	

B XXcX"Fr}}7!v78H6, 8s   )Blstr   c                 (  ^  [        [        R                  R                  S5      R	                  S5      / SQS9nUR                  U 4S jSS9n[        UR                  5        Vs/ s H  nT PM     sn5      n[        R                  " X$5        g s  snf )NrZ   )ru   r   r   r   c                    > T$ rL   rM   )rO   r  s    r!   rQ   3test_consistent_coerce_for_shapes.<locals>.<lambda>-  s    r#   r   r   r  )r  r    rD   r  rE   s   `    r!   !test_consistent_coerce_for_shapesr  %  st     

		a 008/
B XXm!X,FBMMO4OqsO45H6, 5s   %Bc                     U nUR                  S SS9nU R                  SSSS.S9n[        R                  " X#5        UR                  S	 SS9nUSS/   n[        R                  " X#5        g )
Nc                     [        / SQ/ SQS9$ )Nr   )testothercolsr   r   r   s    r!   rQ   'test_consistent_names.<locals>.<lambda>7  s    &*CDr#   r   r   r  r  r  r   r   c                     [        SS/SS/S9$ )Nr   rZ   r  r  r   r  r   s    r!   rQ   r  >  s    1vfg5F Gr#   )r=   renamer>   rc   r   s       r!   test_consistent_namesr  2  s    	BXXD1  F #))78 * H &+XXGaXPF)*H&+r#   c                     U nUR                  S SSS9nUR                  5       n[        S5      Ul        [        R
                  " X#5        g )Nc                 
    / SQ$ r   rM   r   s    r!   rQ   "test_result_type.<locals>.<lambda>H  r   r#   r   r   r   r   r=   r   r   r   r>   rc   r   s       r!   test_result_typer  C  sC     
BXX)xXHFwwyHQxH&+r#   c                     U nUR                  S SSS9nUSS/   R                  5       n[        S5      Ul        [        R
                  " X#5        g )Nc                 
    SS/$ r  rM   r   s    r!   rQ   /test_result_type_shorter_list.<locals>.<lambda>R  r  r#   r   r   r   r   r   rZ   r  r   s       r!   test_result_type_shorter_listr  N  sN     
BXX&QHXEF3*~""$HQxH&+r#   c                    US:X  a8  [         R                  R                  SS9nUR                  R	                  U5        U nU[
        L a6  [         R                  " [        SS9   UR                  S SSUS	9  S S S 5        g UR                  S
 SSUS	9nUR                  5       n[        R                  " XV5        g ! , (       d  f       g = f)Nr%   z(numba engine doesn't support list returnr/   ?result_type='broadcast' only implemented for the default enginer[   c                 
    / SQ$ r   rM   r   s    r!   rQ   ,test_result_type_broadcast.<locals>.<lambda>e  s    )r#   r   r   r6   r   r-   c                 
    / SQ$ r   rM   r   s    r!   rQ   r  j  s    ir#   r*   r7   r8   r9   r:   r   rd   NotImplementedErrorr=   r   r>   rc   )r"   r,   r-   r7   r    rD   rE   s          r!   test_result_type_broadcastr  X  s     {{  (R S%	B$$]]S
 HH#!V  	
 
 a[  
 779
f/
 
s   #B::
Cc                   ^ US:X  a8  [         R                  R                  SS9nUR                  R	                  U5        U n/ SQmU[
        L a9  [         R                  " [        SS9   UR                  U4S jSS	US
9  S S S 5        g UR                  U4S jSS	US
9nUR                  5       n[        R                  " XV5        g ! , (       d  f       g = f)Nr%   <numba Series constructor only support ndarrays not list datar/   r  r  r  r  r[   c                    > [        / SQTS9$ Nr   r   r  rO   r   s    r!   rQ   8test_result_type_broadcast_series_func.<locals>.<lambda>  s    &':r#   r   r   r  c                    > [        / SQTS9$ r  r  r  s    r!   rQ   r    s    fYg6r#   r  r"   r-   r,   r7   r    rD   rE   r   s          @r!   &test_result_type_broadcast_series_funcr  p  s     {{  Q ! 
 	%	B'G$$]]S
 HH:'	  	
 
 6#	  
 779
f/%
 
s   (C
Cc                     US:X  a8  [         R                  R                  SS9nUR                  R	                  U5        U nUR                  S SUS9nUR                  5       n[        R                  " XV5        g )Nr%   r  r/   c                 .    [        / SQU R                  S9$ r  )r   rA   r   s    r!   rQ   0test_result_type_series_result.<locals>.<lambda>  s    y @r#   r   r5   )	r*   r7   r8   r9   r:   r=   r   r>   rc   )r"   r-   r,   r7   r    rD   rE   s          r!   test_result_type_series_resultr    sp     {{  Q ! 
 	%	BXX@qQWXXFwwyH&+r#   c                   ^ US:X  a8  [         R                  R                  SS9nUR                  R	                  U5        U n/ SQmUR                  U4S jSUS9nUR                  5       nTUl        [        R                  " XV5        g )Nr%   z:no support in numba Series constructor for list of columnsr/   r  c                    > [        / SQTS9$ r  r  r  s    r!   rQ   <test_result_type_series_result_other_index.<locals>.<lambda>  s    y @r#   r   r5   )
r*   r7   r8   r9   r:   r=   r   r   r>   rc   r  s          @r!   *test_result_type_series_result_other_indexr    s~     {{  O ! 
 	%	B'GXX@qQWXXFwwyHH&+r#   boxc                     [        U 5      $ rL   )r   r   s    r!   rQ   rQ     s    tAwr#   c                     [        U 5      $ rL   )ru  r   s    r!   rQ   rQ     s    %(r#   c                 ,    [         R                  " U SS9$ )Nr   r   )r   r  r   s    r!   rQ   rQ     s    bhhq6Pr#   )r   ru  r  c           	      P  ^  UnUR                  U 4S jSS9n[        UR                  5        Vs/ s H  nT " SS/5      PM     sn5      n[        R                  " X55        UR                  U 4S jSSS9nUSS	/   R                  S
SS.S9n[        R                  " X55        g s  snf )Nc                    > T" SS/5      $ r  rM   rO   r  s    r!   rQ   ,test_consistency_for_boxed.<locals>.<lambda>      QFr#   r   r   rZ   c                    > T" SS/5      $ r  rM   r  s    r!   rQ   r    r  r#   r   r   r   r   r   r   r   )r=   r   r  r>   r?   r  rc   )r  r"   r    rD   r  rE   s   `     r!   test_consistency_for_boxedr    s     
BXX+!X4FBMMO<OqsAq6{O<=H6,XX+!XJF"C:.55AA>N5OH&+ =s   B#c                    U S;   a  SOSn[         R                  " SS9   [         R                  " U5      n[         R                  " U5      nUR	                  5       nUR                  [         R                  U S9n[        R                  " Xe5        UR                  [         R                  /U S9nUR	                  5       nU S;   a)  [        R                  " UR                  S//5      Ul
        O([        R                  " UR                  S//5      Ul        [        R                  " Xe5        UR                  [         R                  [         R                  /U S9n[        X4/US9nU S;   a*  [        R                  " UR                  SS//5      Ul
        O)[        R                  " UR                  SS//5      Ul        [        R                  " Xe5        S S S 5        g ! , (       d  f       g = f)	N   rA   r   r   r   r1   r2   r   r<   absolute)r   r;   absr<   r   r=   r>   rc   r   from_productr   rA   r   )r6   rC   
other_axisf_absf_sqrtrE   rD   s          r!   test_agg_transformr    s   l*J		"{#% ;;=""277"6
f/ ""BGG94"8;;=<)668K8KfX7VWH'44k6G6G&5RSHN
f/
 ""BFFBGG#44"@uoJ?<)66$$z6&:; H (44""Z$89HN 	f/? 
#	"	"s   F5G
G+c                      [        [        S5      SS.5      n U R                  SS/5      n[        SS/SS/S.SS/SS/S	9n[        R                  " X5        g )
Nr   r   r  r  r   ru   r   r   rZ  )r   r   r   r>   rc   r   s      r!   	test_demor    s]    	q*	+BVVUEN#F!fAq6"S#JuenH &+r#   c                     [        [        S5      SS.5      n U R                  SS/SS/S.5      n[        SS[        R                  /S[        R                  S	/S.S
S// SQS9n[
        R                  " UR                  U5      U5        g )Nr   r   r  r  r   rf  r        @g      9@r   r   )r  r  r   rZ  )r   r   r   r   r   r>   rc   reindex_liker   s      r!   test_demo_dict_aggr    s    	q*	+BVV5%.u~>?FC RVVT':;c
#H
 &--h7Br#   c                     SSS/0n [        U 5      nUR                  SS05      n[        SS05      n[        R                  " X#5        US   R                  SS05      n[        SS0SS9n[        R                  " X#5        g )Nr  r  rw  countrZ   r  )r   r   r   r>   r?   )r   r    rD   rE   s       r!   !test_agg_with_name_as_column_namer     s    UEN#D	4B VVVW%&Fvqk"H6, Z^^VW-.Fvqk/H6,r#   c                  8   [        / SQ/ SQ/ SQS.5      n [        SS/SS/S	S
/S.SS/S9nU R                  SS/5      n[        R                  " X!5        U / SQ   R                  SS/5      nU/ SQ   R	                  SS/5      n[        R                  " X!5        g )Nr   r  rN  rg  r  rw  bazr   r   r   r        @rw  	foobarbazr  r   r   )r   r   r   )r   r   r>   rc   reindex)mdfrE   rD   s      r!   test_agg_multiple_mixedr    s    
 &	
C Qs%	

 enH WWeU^$F&+!%%uen5F (00%@H&+r#   c            
      f   [        / SQ/ SQ/ SQ[        SSS9S.5      n Sn[        R                  " [        US	9   U R                  S
S/5        S S S 5        [        R                  " [        US	9   U / SQ   R                  SS
/5        S S S 5        g ! , (       d  f       NI= f! , (       d  f       g = f)Nr   r  r  r   r   r   r   r   r   rk   does not support operationr[   r  r   )rk   r   r   r   )r   r
   r*   rd   re   r   )r  msgs     r!   test_agg_multiple_mixed_raisesr  ,  s    
 &J2		
C 'C	y	, 
- 
y	, !%%uen5 
-	, 
-	, 
-	,s   B/B"
B"
B0c                    U S;   a  SOSnUR                   U   R                  5       S S R                  5       u  p4[        R                  " UR                  U S9UR                  U S9UR                  U S9/SS9n/ SQUl        U S;   a  UR                  OUnUR                  / SQU S9n[        R                  " Xe5        USUS0nUR                  XpS9n[        UR                  U5      U   R                  5       UR                  U5      U   R                  5       /X4/S	9n[        R                  " Xe5        US/US/0nUR                  XpS9n[!        U[        UR                  U5      U   R                  5       /S/S	9U[        UR                  U5      U   R                  5       /S/S	905      nU S
;   a  UR                  OUn[        R                  " Xe5        USS/USS/0nUR                  XpS9n[        R                  " U[        UR                  U5      U   R                  5       UR                  U5      U   R                  5       /SS/S	9U[        UR                  U5      U   R                  5       UR                  U5      U   R                  5       /SS/S	90SS9nU S
;   a  UR                  OUn[        R                  " Xe5        g )Nr  r   r   rZ   r   )r@   r  r   r@   r   r   >   r   r   r  )axesuniquesort_valuesrf   concatr@   r  r   r   r   r   r>   rc   r   r   r?   r   )r6   rC   r  name1name2rE   rD   r   s           r!   test_agg_reducer  @  s   l*J##J/668!<HHJLE yy$'OOO&OOO&	

 H .H!\1xzzxH__3$_?F&+ 65%(D__T_-FOOJ'.335OOJ'.224	
 nH 6, F8UUG,D__T_-F6;??:6u=BBDEfXV6;??:6u=AACDUGT	
H "^3xzzH&+ FE?EE5>:D__T_-Fyy6OOJ/6;;=OOJ/6::< uo 6OOJ/6::<OOJ/6::< en	
  #H& "^3xzzH&+r#   c                    U R                   S   R                  5       S S R                  5       u  pSnS H:  n[        R                  " [
        US9   U R                  US4US4US9  S S S 5        M<     g ! , (       d  f       MN  = f)	Nr   rZ   z/Named aggregation is not supported when axis=1.)r   r   r[   r   r  )row1row2r6   )r  r  r  r*   rd   r  r   )rC   r  r  r  r6   s        r!   "test_named_agg_reduce_axis1_raisesr    st    ##A&--/3??ALE
;C]].c:OO%uen4OP ;: ::s   A44
B	c            
         [        / SQ/ SQ/ SQ[        SSS9S.5      n U R                  S5      n[        S	S
S[	        S5      /U R
                  S9n[        R                  " X5        U R                  S/5      n[        S	S
S[	        S5      //S/U R
                  S9n[        R                  " X5        Sn[        R                  " [        US9   U R                  S5        S S S 5        U / SQ   R                  S5      n[        / SQ/ SQS9n[        R                  " X5        Sn[        R                  " [        US9   U R                  S/5        S S S 5        g ! , (       d  f       N= f! , (       d  f       g = f)Nr   r  r  r   r   r   r  r  r   r  rw  r   r  r  r[   r   r   )r   r  r  )r   r
   r   r   r	   r   r>   r?   rc   r*   rd   re   )r    rD   rE   r  s       r!   test_nuiscance_columnsr    s7   	 &J2		

B VVE]Fq#ui
&;<BJJOH6,VVUG_F
S%:.	/0g

H
 &+
&C	y	,
u 
-  $$U+F+?CH6,
&C	y	,
w 
-	, 
-	, 
-	,s   E=E*
E'*
E8howc                 r   [        / SQS[        R                  S// SQS.5      n[        X5      " SS05      n[	        SS05      n[
        R                  " X#5        [        X5      " SS	05      n[	        SS
05      n[
        R                  " X#5        [        X5      " SS	/5      n[        X5      " SS	/SS	/SS	/S.5      n[        SS
S.SS
S.SS
S.S.5      n[
        R                  " XESS9  [
        R                  " XSSS9  [        X5      " S5      nUR                  5       n[
        R                  " X#5        g )NNrZ   r   r  rg  r  Nrw  r   r   r  rZ   sizer   )r  r
  Tr  )	r   r   r   r   r   r>   r?   rc   r  )r  r    rD   rE   result1result2s         r!   test_non_callable_aggregatesr    s9    
#rvvs!3:NO
B
 RsGn-FsAhH6, RsFm,FsAhH6, b01Gbw&7w>OPG a(a(a(	
H 't<'= Rg&FxxzH6,r#   c                 *   [        / SQS[        R                  S// SQS.5      n[        X 5      " SUS9nUS;   a"  [	        UR
                  S	   UR                  S
9nO![	        UR
                  S   UR                  S
9n[        R                  " X45        g )Nr  r  rg  r	  r   r
  r   )r   rA   r   r   r   )
r   r   r   r   r   rv   r   rA   r>   r?   )r  r6   r    rD   rE   s        r!   test_size_as_strr    s}     
#rvvs!3:NO
B
 Rf40F|"((1+RZZ8"((1+RXX66,r#   c                  N   [        / SQS[        R                  S// SQS.5      n S nU R                  U5      n[	        SS/S/SS	/// SQS
9n[
        R                  " X#5        U R                  U/5      nUR                  S5      R                  n[
        R                  " X#5        g )N)rZ   rZ   r   g      ?r	  r   c                 P    [        U R                  5       R                  5       5      $ rL   )r   dropnar  )	group_cols    r!   r   &test_agg_listlike_result.<locals>.func  s    I$$&--/00r#   rZ   r   r  rw  r   r   )
r   r   r   r   r   r>   r?   to_framer   rc   )r    r   rD   rE   s       r!   test_agg_listlike_resultr    s    	#rvvs);BVW	XB1 VVD\F1vuuen5_MH6,VVTF^F  (**H&+r#   zargs, kwargs)r   r      r  rS   rY   rM   rb  r   c                     SS jn[        SS/SS//5      nU S:X  a  [        SS/5      nO[        S	S
/5      nUR                  " X0/UQ70 UD6n[        R                  " Xe5        g )Nr   c                 2    U R                  5       X-   U-  -   $ rL   r  )rO   rX   rY   r   s       r!   rr  test_agg_args_kwargs.<locals>.f  s    uuw!%1$$r#   r   rZ   ru   r   r  g      @rf  g       @)r   r   r   r   r>   r?   )r6   rT   kwargsrr  r    rE   rD   s          r!   test_agg_args_kwargsr    si    % 
QFQF#	$Bqy3*%3*%VVA-d-f-F6,r#   num_cols)rZ   r   r   c                 R   US:X  a8  [         R                  R                  SS9nUR                  R	                  U5        [
        R                  " / SQ5      nUR                  5       n[        SU[        U 5      S9nUR                  S US9  UR                  UR                  :X  d   eg )	Nr%   z*numba engine only supports numeric indicesr/   )z
1950-06-30z
1952-10-24z
1953-05-29r   r  c                     U $ rL   rM   r   s    r!   rQ   ,test_frequency_is_original.<locals>.<lambda>$  r&  r#   r4   )r*   r7   r8   r9   r:   rf   DatetimeIndexr   r   r   r=   r[  )r  r-   r,   r7   rA   originalr    s          r!   test_frequency_is_originalr$    s     {{  (T U%GHEzz|H	1E5?	;BHH[H(::&&&r#   c                 4   U S:X  a8  [         R                  R                  SS9nUR                  R	                  U5        [        SSS9[        SSS9[        SSS9/n[        / S	QUS
9nUR                  S SU S9n[        X3S9n[        R                  " XV5        g )Nr%   z0numba engine doesn't support non-numeric indexesr/   z2019-03-15 12:34:31.909000+0000r|   r}   z2019-03-15 12:34:34.359000+0000z2019-03-15 12:34:34.660000+0000r   r   rZ   )r   rA   c                     U R                   $ rL   r  r   s    r!   rQ   .test_apply_datetime_tz_issue.<locals>.<lambda>7  s    r#   r   r5   )rA   r   )r*   r7   r8   r9   r:   r	   r   r=   r   r>   r?   )r-   r,   r7   
timestampsr    rD   rE   s          r!   test_apply_datetime_tz_issuer*  (  s     {{  E ! 
 	% 	3>3>3>J
 
		4BXX&QvX>FJ8H6,r#   r    r   rF   r   method)r  r  r   c                     US:X  a  SnOSnU(       d3  [         R                  " [        US9   [        X5      " 5         S S S 5        g [        X5      " 5         g ! , (       d  f       g = f)Nr   z>can only concatenate str \(not "int"\) to str|does not supportz4not supported between instances of 'str' and 'float'r[   )r*   rd   re   r   )r    r+  using_infer_stringr  s       r!   test_mixed_column_raisesr.  =  sO     OD]]9C0B! 10 	 10s   A
A%r  r  c                     [        SU //SS/S9nUR                  S 5      nUR                  n[        R                  " X#5        g )Nr  rX   rY   r   c                     U R                   $ rL   r   r   s    r!   rQ   "test_apply_dtype.<locals>.<lambda>P  s    r#   )r   r=   dtypesr>   r?   )r  r    rD   rE   s       r!   test_apply_dtyper3  L  s@     
S#J<#s	4BXX'(FyyH6,r#   c                     [        [        S5      [        SS5      S.5      n U R                  5       nS nU R                  5       nUS==   S-  ss'   U R                  USS9n[        R
                  " XC5        [        R
                  " X5        g )Nr  r]  rb  c                 n    U R                   nU R                  S==   S-  ss'   U R                   ULd   eU $ )NrX   r   )_mgrr   )r  mgrs     r!   r   !test_apply_mutating.<locals>.func[  s4    hhxxs"""
r#   rX   r   r   )r   r   r   r=   r>   rc   )r    df_origr   rE   rD   s        r!   test_apply_mutatingr:  V  st    	rr27	8BggiG wwyHSMQMXXdX#F&+"&r#   c                      [        SS/SS/SS/SS/S	S
//SS/S9n U R                  S SS9n[        / / S.[        S9n[        R
                  " X5        g )Nr   rZ   r   ru   r   r   r   r   r   r  rX   rY   r   c                     / $ rL   rM   r   s    r!   rQ   .test_apply_empty_list_reduce.<locals>.<lambda>n  r<  r#   r   r   rb  r   )r   r=   r   r  r>   r?   r   s      r!   test_apply_empty_list_reducer>  j  sb    	QFQFQFQFQG<sCj	QBXXlX9FBR(7H6,r#   c                     U S:X  a8  [         R                  R                  SS9nUR                  R	                  U5        [        SS//S-  SS/S	9nUR                  S
S S /U S9n[        / SQ/ SQS./ SQS9n[        R                  " XE5        g )Nr%   z;numba engine doesn't support list-likes/dict-like callablesr/   ru   r   r   r   r   r   r   c                 "    U R                  5       $ rL   r  r   s    r!   rQ   ,test_apply_no_suffix_index.<locals>.<lambda>{  s
    r#   c                 "    U R                  5       $ rL   r  r   s    r!   rQ   rA  {  s
    AEEGr#   r4   )r  r  r  )   rC  rC  r   )r   r
  r
  r   )	r*   r7   r8   r9   r:   r   r=   r>   rc   )r-   r,   r7   pdfrD   rE   s         r!   test_apply_no_suffix_indexrE  s  s    {{  P ! 
 	%
aVHqL3*
5CYY02CDVYTF.6UH &+r#   c                     U S:X  a  [         R                  " S5        [        SSS/05      nUR                  S U SSS	9n[	        SS/5      n[
        R                  " X#5        g )
Nr%   z No object dtype support in numbar   aabbbc                     U S   $ r   rM   r   s    r!   rQ   /test_apply_raw_returns_string.<locals>.<lambda>  r   r#   r   T)r-   r6   rH   )r*   rb   r   r=   r   r>   r?   r-   r    rD   rE   s       r!   test_apply_raw_returns_stringrL    s\    67	C$'	(BXXnV!XFFtUm$H6,r#   c                      [        / SQSS9n S nSUSS/nU R                  U5      n[        / SQ/ S	Q/ S
QS./ SQS9n[        R                  " X45        g )N))r   r   r   )rZ   r   r   )r   r   r   )ru   r   ru   )r   r   r   )r   r   r   )att1att2att3r   c                 (    U R                  5       S-  $ r  r  r   s    r!   sum_div24test_aggregation_func_column_order.<locals>.sum_div2  s    uuw{r#   r   r  r  )g      5@g      %@r  r  )g      2@g      "@r  r  )g      1@g      !@r  r  )r   rR  r  r  r   r   )r    rR  aggsrD   rE   s        r!   "test_aggregation_func_column_orderrU    sg     
	
 )

B 8We,DVVD\F*))	

 2H &+r#   c                    U S:X  a8  [         R                  R                  SS9nUR                  R	                  U5        [        / SQ/ SQS.5      nUSS/   R                  S S	U S
9n[        / SQ5      n[        R                  " XE5        g )Nr%   z2numba engine not supporting duplicate index valuesr/   r&  r   rb  rX   c                 @    U R                   S   U R                   S   -   $ r`   )rT  r   s    r!   rQ   +test_apply_getitem_axis_1.<locals>.<lambda>  s    !&&)affQi'r#   r   r5   )r   rZ   ru   )
r*   r7   r8   r9   r:   r   r=   r   r>   r?   )r-   r,   r7   r    rD   rE   s         r!   test_apply_getitem_axis_1rY    s    {{  G ! 
 	%	3	4Bc
^!!'a " F i H6,r#   c                      S n [        S/ SQ05      n[        R                  " [        SS9   UR	                  U /5        S S S 5        g ! , (       d  f       g = f)Nc                 N    [         R                  " S5        U R                  5       $ )NHello, World!)warningswarnr   r   s    r!   expected_warningDtest_nuisance_depr_passes_through_warnings.<locals>.expected_warning  s    o&uuwr#   rX   r   r\  r[   )r   r>   assert_produces_warningUserWarningr   )r_  r    s     r!   *test_nuisance_depr_passes_through_warningsrc    sE     
C#	$B		#	#K	G
 !" 
H	G	Gs   A
Ac            	      r   [        SS[        /S[        SSS5      [        R                  /S./ SQS9n U R                  [        S	S
9n[        [        [        S.5      n[        R                  " X5        U R                  [        SS
9n[        [        [        [        S.5      n[        R                  " X5        g )Nr   r  g      ?i  r   )r  r  r   r   r   r   )
r   r   r   r   r   r=   typer   r>   r?   r   s      r!   test_apply_typerf    s    	Xu%htQ6JBFF/ST
B XXdX#Fvv67H6, XXdX#FFf=>H6,r#   c                     [        SS/SS/S.5      nUR                  S5      R                  S SU S9n[        / [        R
                  S9n[        R                  " X#5        g )	Nr   rZ   r   r   rb  c                 &    [        U S   U S   5      $ )NrX   rY   )r  r   s    r!   rQ   /test_apply_on_empty_dataframe.<locals>.<lambda>  s    AcFAcF(;r#   r5   r   )r   headr=   r   r   r   r>   r?   rK  s       r!   test_apply_on_empty_dataframerk    sW    	!Qq!f-	.BWWQZ;!FSFb

+H6,r#   c                      [        SS/SS/S.5      n U R                  S 5      n[        SS//SS//S.5      n[        R                  " X5        g )Nr   rZ   r   rb  c                     U R                   /$ rL   )r  r   s    r!   rQ   (test_apply_return_list.<locals>.<lambda>  s
    
r#   r   r   s      r!   test_apply_return_listro    sQ    	!Qq!f-	.BXX*+FAx1vh78H&+r#   ztest, constant)r   r   r   )rZ   rZ   rZ   c                     [        U 5      n[        USS/SS9nSSS.nUR                  U5      n[        R                  " XS5        g )NrX   rY   r  )r   rA   r   r  rb  r  )r  constantdf1rE   aggregationrD   s         r!   test_unique_agg_type_is_seriesrt    sE     D/C8C:XFH x0KWW[!F6,r#   c                  ,   [        / SQ/ SQ/ SQS.5      n [        / SQ5      n[        R                  " U R	                  SS9U5        U R                  SSS9n[        R                  " X!5        U R                  SS5      n[        R                  " X!5        g )	N)r   rZ   r   )r   rZ   r   )r   r   r   r   )TTFr   r   r   )r   r   r>   r?   r   r=   r  s      r!   /test_any_apply_keyword_non_zero_axis_regressionrv    sv    	C	DB)*H266q6>84XXe!X$F6,XXeQF6,r#   c                     [        S/ SQ05      n SS jnSS jnU R                  USSSS9nU S	-   n[        R                  " X45        U R                  X/SSSS9n[        S
S
/SS/SS//SS/SS//S9n[        R                  " X45        U R                  SU0SSSS9n[        / SQS/S9n[        R                  " X45        g )NrO   r   r   c                     X-   U-   $ rL   rM   )rO   rX   r   s      r!   foo1.test_agg_mapping_func_deprecated.<locals>.foo1      uqyr#   c                     X-   U-   $ rL   rM   )rO   rY   r   s      r!   foo2.test_agg_mapping_func_deprecated.<locals>.foo2  r{  r#   r   ru   )r   r   r   r   r  ry  r}  r   )rZ   r   ru   )r   r   )rZ   r   r   )r    ry  r}  rD   rE   s        r!    test_agg_mapping_func_deprecatedr    s    	C#	$B VVD!Q!V$FAvH&+VVTL!Q!V,F
Q!Q"b"c3Z&&9I,JH &+ VVS$KAV+FSE2H&+r#   c                  t   [        [        R                  " S5      R                  SS5      SS/S9n U R	                  [        R
                  SS9n[        S	S	S
.[        S9n[        R                  " X5        U R	                  [        R
                  /SS9n[        S	S	S
.S/S9n[        R                  " X5        g )Nr   r   rZ   r   r   r   r   )ddofrN  r   r   stdr   )r   r   r   r   r   r  r   r   r>   r?   rc   r   s      r!   test_agg_stdr  '  s    	299Q<''1-Sz	BBVVBFFV#FCc*%8H6,VVRVVH1V%Fs-eW=H&+r#   c                     [        / SQ/ SQ/ SQ// SQS9n U R                  S[        R                  /05      n[        SS/0S/S	9n[        R
                  " X5        U R                  S[        R                  05      n[        SS05      n[        R                  " X5        U R                  S[        R                  [        R                  /05      n[        SS
S/0SS/S	9n[        R
                  " X5        g )Nr   r   r   r   r   r   r   r
  r   rf  rg  r@   )	r   r   r   r
  r>   rc   r   r?   r@   r   s      r!   test_agg_np_sizer  3  s    	Iy)4o	NBVVS277)$%F#sF84H&+VVS"''N#FsAhH6,VVS277BGG,-.F#Sz*662BCH&+r#   c                      [        / SQS[        R                  S// SQS.5      n / SQU l        U R	                  SS05      nU S   R                  5       n[        R                  " X5        g )	Nr  r  rg  r	  r   )r   r   r   r   r  )r   r   r   r   r   r  r>   r?   r   s      r!   (test_agg_dist_like_and_nonunique_columnsr  D  sZ    	#rvvs!3:NO
B !BJVVS'N#F#w}}H6,r#   engine_nameunknown   c                     [         R                  " [        SS9   [        5       R	                  S U S9  S S S 5        g ! , (       d  f       g = f)NzUnknown engine r[   c                     U $ rL   rM   r   s    r!   rQ   #test_wrong_engine.<locals>.<lambda>S  r   r#   r4   )r*   rd   r  r   r=   )r  s    r!   test_wrong_enginer  P  s2    	z):	;+k: 
<	;	;s	   =
A)r   r]  numpyr   r*   pandas.compatr   pandas.core.dtypes.dtypesr   pandasrf   r   r   r   r	   r
   pandas._testing_testingr>   pandas.tests.apply.conftestr   pandas.tests.frame.commonr   pandas.util.versionr   fixturer"   r)   r7   
single_cpur-   rG   parametrizeri   ry   r   r   r   r<   r@   r   r   r   r   r   r   r   r   r   r   r   r   r   r  r
  r  r  r  r  r#  filterwarningsr0  r4  r?  rF  rP  rR  rV  r[  re  ru  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r   r  r  r   r  r!  r.  r>  rC  rT  r`  rh  rp  rv  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r$  r*  r.  r3  r:  r>  rE  rL  rU  rY  rc  rf  rk  ro  rt  rv  r  r  r  r  r  rM   r#   r!   <module>r     s       ) 6    ; 0 ' 
 
 WFKK$:$:;1, !Q(u.dE]3D 4 / )DN<, $aS)*Y|-NPT,UV,	,
, "''277!34 5	-,, !>?- @-
--,,,"
,&, !Q(G )G !Q(- )-, !Q(> )>,-- 4534
[,-J3G   u.!Q(1 ) / 5 61 ),-,/
-
-	,-D1,h--,-=  u~'789&)u+@AB	
 
QFQF#	$f!laL-I&JK		/		/
,&, /0
- 1
-,'(.,, D"&& 9:
- ;
-$ 
	9cAq6]+CD
		3A-(@A
 	!8VS!H-=>
 	!5&#q*:;5"&&"345sI&.HI	
 5"&&"345sI&.HI	

 {	Xy#1v'?@
{	UFC8$45!*&+*& 		o	 	ubff/01	k	##-(-, 
	Hi1BQF0K&LM
	E9.?!Q-H#IJ
	h	3*ZL(QR
	eYaSz*%NOBFF-.sI&.HI	
 BFF-.sI&.HI	
 k*1v1v&#sj*=U0VW	
 +&sQFmj*%=>	
):&;:&
,," 				)*	+&	##-&--$-  A'- (--0 QF 34	- 5	-,",,000B,,$ 	*,PQ"  

,

,"0J,	C- ,86(B,JQ!H  01)- 2)-X  01- 2-,  !Q(	B	R		Qx	11	11	111%&- )-  Y/	' 0	'-* 	d3**M NOP#89
 : Q
 CsBFF ;<- =-'(-, -,>-#"-$-, y	)!+EFy	)!A3+?@--
-,6	,,"	- B8; 9;r#   