
    .ib                     J   d dl m Z  d dlZd dlZd dlZd dlmZ d dlmZ d dl	Z
d dl	mZmZmZmZmZ d dlmZ d dlmZ d dlmZ d dlmZ ej2                  d	        Z ej2                  d
 ej6                  dej8                  j:                        eg      d        Zd Zej8                  jA                  dd dg      ej8                  jA                  dddg      ej8                  jA                  dddg      d                      Z!d Z"d Z#ej8                  jA                  ddd edg      f edd      dfg      d        Z$d Z%ej8                  jA                  d ejL                  ejN                  g      d!        Z(d" Z)d# Z*d$ Z+ej8                  jA                  d g d%      d&        Z,d' Z-d( Z.d) Z/d* Z0d+ Z1d, Z2d- Z3d. Z4ej8                  jA                  dd dg      d/        Z5ej8                  jA                  dd dg      d0        Z6d1 Z7ej8                  jA                  dd dg      d2        Z8d3 Z9d4 Z:d5 Z;ej8                  jy                  d6      ej8                  jA                  d7d8d9g      ej8                  jA                  d d: d; gd<d=g>      ej8                  jA                  dddg      ej8                  jA                  dd dg      d?                                    Z=d@ Z>dA Z?dB Z@dC ZAdD ZBdE ZCdF ZDdG ZEdH ZFdI ZGdJ ZHdK ZIdL ZJdM ZKdN ZLej8                  jA                  dO edPdQgdRdSgg       edPdRdTdQdSdTg      g ed dgdUdVgg       ed dUdTddVdTg      gg      dW        ZMdX ZNdY ZOdZ ZPej8                  jA                  d[d\d]g      d^        ZQd_ ZRd` ZSda ZTdb ZUej8                  jA                  dcdddedej                  g      df        ZWdg ZXej8                  jA                  dhdidj idk edidUdVgi      fdidl id edidUdVgi      fdidm idk edidVi      fdidn id edidVi      fdidoej                  dp gidk edig dqig drs      fdidoej                  dt gid edig dqig drs      fdidu idk ediddgi      fdidv id edidi      fg      dw        ZZej8                  jA                  dxdidy ididz ididoej                  d{ gidid| ig      d}        Z[d~ Z\d Z]d Z^ej8                  jA                  dhd gdk eddUdVgi      fd gd eddUdVgi      fd gdk edidVgidgs      fd gd edidVgidgs      fdoej                  d gdk edig dqig drs      fdoej                  d gd edig dqig drs      fd d gdk edUdVgdVdVggdidigddgg      fd d gd edidUdVgiddg      fg      d        Z_d Z`d Zaej8                  jA                  dxd gd gdoej                  d gd d gg      d        Zbd Zcd Zdd Zed Zfej8                  jA                  dcddUg      d        Zgd Zhej8                  jA                  dg dddUgg      d        Zid Zjd Zkd Zld Zmd Znd Zod Zpej8                  jA                  dd d d gg d>      d        Zqd Zrd Zsd Ztd Zud Zvd Zwd Zxd Zyd Zzej8                  jA                  dd]d\g      d        Z{ej8                  jA                  dd]d\g      d        Z|d Z}ej8                  jA                  dd dg      ej8                  jA                  ddi fdi fdi fdddUifdddUdfddUddfdddUdVdfg      d               Z~ej8                  jA                  dg dĢ      dń        ZdƄ Zej8                  jA                  d edidgddgdʜ      g      ej8                  jA                  dg d̢      d̈́               Zej8                  jA                  dddddiej                  g      dЄ        Zdф Zd҄ Zdӄ ZdԄ ZdՄ Zdք Zdׄ Zd؄ Zdل Zdڄ Zej8                  jA                  dg dg dܢdg ddgdfg dݢg dܢddUgdgdfg      dބ        Zd߄ Zd Zd Zd Zd Zej8                  jA                  dddg      d        Zy)    )datetimeN)is_platform_arm)CategoricalDtype)	DataFrame
MultiIndexSeries	Timestamp
date_range)MockEngineDecorator)
zip_frames)Versionc                      t        t        j                  t        j                  dd      d      j	                  dd      dz   g d      } | S )	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    a/var/www/app/trading-bot/venv/lib/python3.12/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                 b    | j                   dk(  rt        j                  d       | j                   S )Nr%   )parampytestimportorskip)requests    r!   enginer-   '   s(     }}G$==r#   c                    |dk(  r;t         j                  j                  d      }|j                  j	                  |       t        j                  d      5  t        j                  | d         }| j                  t
        j                  |      d   }t        j                  ||       | j                  t
        j                  |      d   }t        j                  | d         }||k(  sJ | j                  d   }| j                  t
        j                  d	|
      }t        j                  | j                  |            }||   |k(  sJ |j                  | j                  u sJ 	 d d d        y # 1 sw Y   y xY w)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   sC   {{  (U V%		" 1S)*$$RWWV$<SA
vx0 ""2776":3?77;s+,!!!a ""2776"B77;>>!,-ayH$$$||{000001 1 1s   DE00E9r6   r   rawTFnopythonc           	         t        j                  d      }|dk(  rMt        |j                        t        d      k(  r,t	               r"t        j
                  d|j                          d|i}| j                  d |d|||      }| dz   }t        j                  ||       | j                  d	 dd
|||      }| dz   }t        j                  ||       |dk(  rt        j                  t        d      5  | j                  d d
|||       d d d        t        j                  t        j                  j                  d      5  | j                  d dd
|||       d d d        t        j                  t        j                  j                  d      5  | j                  d dd
|||       d d d        y y # 1 sw Y   xY w# 1 sw Y   gxY w# 1 sw Y   y xY w)Nr%   z0.61z&Segfaults on ARM platforms with numba rI   c                     | |z   S N xys     r!   <lambda>z!test_apply_args.<locals>.<lambda>W   s
    QU r#   r   )argsrH   r-   engine_kwargsr   c                     | |z   |z   S rL   rM   rO   abs      r!   rQ   z!test_apply_args.<locals>.<lambda>c   s    A	 r#      )rS   rX   rH   r-   rT   r   z missing a required argument: 'a'matchc                     | |z   S rL   rM   )rO   rW   s     r!   rQ   z!test_apply_args.<locals>.<lambda>q   s
    QU r#   )rX   rH   r-   rT   z-numba does not support keyword-only argumentsc                    | |z   |z   S rL   rM   rV   s      r!   rQ   z!test_apply_args.<locals>.<lambda>~   s    1q519 r#   c                     |d   |d   z   | z   S Nr   r   rM   )rX   rO   s     r!   rQ   z!test_apply_args.<locals>.<lambda>   s    adQqTkAo r#   )r*   r+   r   __version__r   skipr=   r>   assert_frame_equalraises	TypeErrorpderrorsNumbaUtilError)	rC   r6   rH   r-   rI   r%   rT   rD   rE   s	            r!   test_apply_argsrh   J   s    (E'E%%&'&/9<U=N=N<OPQ*M#  F QH&(+ !
#  F QH&(+]]9,NO 	"+  	 ]]II$$A
 	 ,+  		 ]]II$$A
 	 -+  		 	3 	 		 		 	s$   2F4G G4F= G	Gc                      t        g dg dd      } | j                  d       }|j                  dk(  sJ t        |d   j                  t
              sJ t        |d   j                  t
              sJ y )N)r   r   r   r   )r   r   Drj   c0c1c                 $    | j                  d      S )Ncategoryastype)tss    r!   rQ   z-test_apply_categorical_func.<locals>.<lambda>   s    :!6 r#   )   rY   rl   rm   )r   r=   shape
isinstancer   r   r    rD   s     r!   test_apply_categorical_funcrw      sj    	.6JK	LBXX67F<<6!!!fTl((*:;;;fTl((*:;;;r#   c                      t        dt        dd      gi      } | j                  d d      }t        j                  ||        y )Nr   
2013-01-01UTCtzc                     | S rL   rM   rO   s    r!   rQ   z*test_apply_axis1_with_ea.<locals>.<lambda>       a r#   r   r6   r   r	   r=   r>   rb   rE   rD   s     r!   test_apply_axis1_with_ear      s=    #	,5 ABCDH^^Ka^0F&(+r#   zdata, dtype)r   Nry   rz   r{   c                     t        | g| ggddg|      }|j                  d d      }t        j                  ||       y )NrW   )rA   r   c                     | S rL   rM   r~   s    r!   rQ   z0test_agg_axis1_duplicate_index.<locals>.<lambda>       A r#   r   r   r   aggr>   rb   )datar   rE   rD   s       r!   test_agg_axis1_duplicate_indexr      sA     4&4&)#s5IH\\+A\.F&(+r#   c                      t        t        dd      t        j                  t	        j
                  d      d      d      } | j                  d d	      }t        j                  ||        y )
N20130101r   periodss)unitr   r   c                     | S rL   rM   r~   s    r!   rQ   z/test_apply_mixed_datetimelike.<locals>.<lambda>   r   r#   r   r   )	r   r
   re   to_timedeltar   r   r=   r>   rb   r   s     r!   test_apply_mixed_datetimeliker      sW     J21C8	
H ^^Ka^0F&(+r#   funcc                 Z    t               }|j                  | |      }|j                  sJ y )Nr4   )r   r=   empty)r   r-   empty_framerD   s       r!   test_apply_emptyr      s,     +KtF3F<<<r#   c                 r   | d d }|j                  d |      }t        t        j                  | j                        }t        j                  ||       | j                  d d g f   }|j                  d d|      }t        t        j                  | j                        }t        j                  ||       y )Nr   c                 "    | j                         S rL   r@   r~   s    r!   rQ   z(test_apply_float_frame.<locals>.<lambda>       QVVX r#   r4   rA   c                 "    | j                         S rL   r   r~   s    r!   rQ   z(test_apply_float_frame.<locals>.<lambda>   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68,ooae$G]]-Af]EFbffK$5$56H68,r#   c                 t    t        dg      }|j                  d d|       }t        j                  ||       y )NrW   r   c                     | d   S )NrW   rM   r~   s    r!   rQ   z/test_apply_empty_except_index.<locals>.<lambda>   s
    af r#   r   r5   r   r=   r>   rb   )r-   rE   rD   s      r!   test_apply_empty_except_indexr      s3    u%H^^,1V^DF&(+r#   c                  R   t               } g }| j                  |j                  dd      }t        j                  ||        | j                  |j                  dd      }t        g t        j                        }t        j                  ||       t        g d      }|j                  |j                  dd      }t        j                  ||       |j                  |j                  dd      }t        g t        j                        }t        j                  ||       |g k(  sJ y )Nr   expandr6   result_typereducer   rW   rX   cr   )	r   r=   appendr>   rb   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68,8O""188!"JF&/2""188!"JFb

+H68, 7N7r#   )sumprodanyr3   c                     t        g d      }|j                  t        t        |             } t        ||              }| dv r|j	                  t
              }t        j                  ||       y )Nr   r   )r   r   )r   r=   getattrr   rq   floatr>   r?   )r   r    rD   rE   s       r!   test_apply_funcs_over_emptyr      sZ     
?	+BXXgb$'(F wr4 "H??5)68,r#   c                  0   t        g d      } | j                         }t        d| j                        }t	        j
                  ||       | j                  j                         }t        g t        j                        }t	        j
                  ||       y )Nr   r   r   r   r   )	r   nuniquer   r   r>   r?   Tr   r   r    rD   rE   s      r!   test_nunique_emptyr      sf    	?	+BZZ\Farzz*H68,TT\\^Fb

+H68,r#   c                     t        g dg dg dgg d      } | j                  d d      }t        g d	g d      }t        j                  ||       | j
                  j                  d
 d      }t        j                  ||       y )Nr   rY   r   rs      r         	   )rW   rW   r   r   c                     | d   S Nr   rM   r   s    r!   rQ   z/test_apply_standard_nonunique.<locals>.<lambda>  
    ! r#   r   r   )r   rs   r   c                     | d   S r   rM   r   s    r!   rQ   z/test_apply_standard_nonunique.<locals>.<lambda>  s
    !A$ r#   r   )r   r=   r   r>   r?   r   r   s      r!   test_apply_standard_nonuniquer     si    	Iy)4O	LBXXn1X-Fi1H68,TTZZQZ/F68,r#   c                     | j                  t        j                  d      }t        | j                         g| j                        }t        j                  ||       y )N	broadcastr   r   )r=   r   r@   r   rA   r>   rb   rC   rD   rE   s      r!   test_apply_broadcast_scalarsr     sI    rwwK@F+**,-[5F5FGH&(+r#   c                     | j                  t        j                  dd      }| j                  d      }t        t        j                  | j                  |            }t        j                  ||       y )Nr   r   r   r   )	r=   r   r@   r   dictfromkeysr   r>   rb   rC   rD   mrE   s       r!   "test_apply_broadcast_scalars_axis1r     s[    rwwQKHFa A{':':A>?H&(+r#   c                 "     j                   fddd      }t        t        t         j                                    }t        |gt         j                        z  d j                   j                        }t        j                  ||       y )Nc                 R    t        t        t        j                                    S rL   )listrangelenr   rO   rC   s    r!   rQ   z4test_apply_broadcast_lists_columns.<locals>.<lambda>$  s    $uS!4!4567 r#   r   r   r   r   )r   rA   r   )	r=   r   r   r   r   r   rA   r>   rb   r   s   `   r!   "test_apply_broadcast_lists_columnsr   !  s    7  F
 	U3{**+,-A	
c+##$$##	H &(+r#   c                      j                   fdd      }t        t        t         j                                    }t        t        j                   j                  |      d j                        }t        j                  ||       y )Nc                 R    t        t        t        j                                    S rL   )r   r   r   rA   r   s    r!   rQ   z2test_apply_broadcast_lists_index.<locals>.<lambda>4  s    $uS!2!2345 r#   r   r   r   )r   rA   )r=   r   r   r   rA   r   r   r   r   r>   rb   r   s   `   r!    test_apply_broadcast_lists_indexr   2  su    5;  F 	U3{(()*+Ak))1-H
 &(+r#   c                 ^    | }|j                  d dd      }t        j                  ||       y )Nc                 
    g dS Nr   rM   r~   s    r!   rQ   z7test_apply_broadcast_list_lambda_func.<locals>.<lambda>B      	 r#   r   r   r   r=   r>   rb   )r"   r    rD   s      r!   %test_apply_broadcast_list_lambda_funcr   ?  s,    	BXX){XKF&"%r#   c                 ~    | }|j                  d dd      }|j                         }t        j                  ||       y )Nc                 2    t        g dt        d            S )Nr   abcr   )r   r   r~   s    r!   rQ   z9test_apply_broadcast_series_lambda_func.<locals>.<lambda>I  s    &$u+6 r#   r   r   r   )r=   copyr>   rb   r"   r    rD   rE   s       r!   'test_apply_broadcast_series_lambda_funcr   F  s@    	BXX6  F
 wwyH&(+r#   c                 h    |dk(  rt        j                  d       d }| j                  |||d       y )Nr%   z)numba can't handle when UDF returns None.c                 ^    t        | t        j                        sJ | j                  dk(  sJ y Nr   ru   r   ndarrayndimr~   s    r!   _assert_rawz/test_apply_raw_float_frame.<locals>._assert_rawV  &    !RZZ(((vv{{r#   Tr6   r-   rH   )r*   ra   r=   )rC   r6   r-   r   s       r!   test_apply_raw_float_framer   Q  s5    ?@ kVFr#   c                     | j                  t        j                  ||d      }| j                  d |      }t        j                  ||       y )NTr   c                 6    | j                   j                         S rL   )valuesr@   r~   s    r!   rQ   z3test_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   ]  sE    rwwT&dKF  !: FH68,r#   c                 d    | j                  d |d      }| dz  }t        j                  ||       y )Nc                     | dz  S NrY   rM   r~   s    r!   rQ   z9test_apply_raw_float_frame_no_reduction.<locals>.<lambda>f  s
    Q r#   T)r-   rH   rY   r   )rC   r-   rD   rE   s       r!   'test_apply_raw_float_frame_no_reductionr   d  s2    v4HFQH&(+r#   c           
         |dk(  rt        j                  d       d }t        dddt        j                  dgdz  d	      t        j                  d
gdz  d	      dt        j
                  d            }|j                  || |d       y )Nr%   z(isinstance check doesn't work with numbac                 ^    t        | t        j                        sJ | j                  dk(  sJ y r   r   r~   s    r!   r   z4test_apply_raw_mixed_type_frame.<locals>._assert_rawp  r   r#         ?rY   foo
   float32r   r   int32)rW   rX   r   r  r  r   Tr   )r*   ra   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[tFH=r#   c                     | j                   d   }| j                  t        j                  d      |   }t        j                  | j	                  |            }||k(  sJ y )Nr   r   r   )rA   r=   r   r@   rB   )rC   rF   rD   rE   s       r!   test_apply_axis1r    sU    !ArwwQ/2Fww{~~a()HXr#   c                      t        dgdgd      } | d d j                  t        j                  d      }t	        t        j
                        }t        j                  ||       y )Nr  r  r   r   r   r   r   )r   r=   r   r@   r   r   r>   r?   r   s      r!   test_apply_mixed_dtype_cornerr    sQ    	%u-	.BV\\"''\*F BJJ'H68,r#   c                  (   t        dgdgd      } | j                  d d      }t        dgt        d            }t	        j
                  ||       | j                  d d      }t        dgt        d            }t	        j
                  ||       y )	Nr  r  r   c                     | d   S )Nr   rM   r~   s    r!   rQ   z8test_apply_mixed_dtype_corner_indexing.<locals>.<lambda>  
    # r#   r   r   r   c                     | d   S )Nr   rM   r~   s    r!   rQ   z8test_apply_mixed_dtype_corner_indexing.<locals>.<lambda>  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68,XX&QX/Fse58,H68,r#   zignore::RuntimeWarningaxrA   r   c                     | S rL   rM   r~   s    r!   rQ   rQ         q r#   c                 "    | j                         S rL   r   r~   s    r!   rQ   rQ     s    AFFH r#   identityr@   )idsc                    t        di | g di}t        j                  d      5   |t        j                  g d            }t	        |t        j
                         }|j                  ||||      }	|r3|j                  |      }
t	        |	t              sJ |	j                  |
u sJ t	        |	t               sJ d d d        y # 1 sw Y   y xY w)Nr   r1   r2   f8r   r   rM   )
r   r   r;   r  ru   r   r=   _get_agg_axisr   rA   )r  r   rH   r6   r-   r,   r    test_resis_reductionrD   agg_axiss              r!   test_apply_empty_infer_typer    s     
	+b/*	+B		" 
1401%h

;;$T&cB''-Hff---<<8+++fi000
1 
1 
1s   BCCc                  l    t        g d      } | j                  d d      }t        |t               sJ y )Nr   r   c                 "    | j                         S rL   r   r~   s    r!   rQ   z7test_apply_empty_infer_type_broadcast.<locals>.<lambda>  r   r#   r   r   )r   r=   ru   )r   rD   s     r!   %test_apply_empty_infer_type_broadcastr"    s/    o.G]]-;]GFfi(((r#   c                     dd}| j                  |d      }| j                  d       }t        j                  ||       y )Nc                     | |z   S rL   rM   rO   howmuchs     r!   add_somez4test_apply_with_args_kwds_add_some.<locals>.add_some  s    7{r#   rY   r&  c                     | dz   S r   rM   r~   s    r!   rQ   z4test_apply_with_args_kwds_add_some.<locals>.<lambda>  s
    1q5 r#   r   r   )rC   r'  rD   rE   s       r!   "test_apply_with_args_kwds_add_somer+    s>     x3F  1H&(+r#   c                     dd}| j                  |d      }| j                  d       }t        j                  ||       y )Nc                 (    | j                         |z   S rL   r   r%  s     r!   agg_and_addz:test_apply_with_args_kwds_agg_and_add.<locals>.agg_and_add  s    vvx'!!r#   rY   r(  c                 (    | j                         dz   S r   r   r~   s    r!   rQ   z7test_apply_with_args_kwds_agg_and_add.<locals>.<lambda>  s    1668a< r#   r*  )r=   r>   r?   )rC   r.  rD   rE   s       r!   %test_apply_with_args_kwds_agg_and_addr0    s?    " {A6F  !78H68,r#   c                     dd}| j                  |dd      }| j                  d       }t        j                  ||       y )Nc                     | |z
  |z  S rL   rM   )rO   subdivides      r!   subtract_and_dividezJtest_apply_with_args_kwds_subtract_and_divide.<locals>.subtract_and_divide  s    C6!!r#   )rY   rY   )rS   r4  c                     | dz
  dz  S )N       @rM   r~   s    r!   rQ   z?test_apply_with_args_kwds_subtract_and_divide.<locals>.<lambda>  s    AGs? r#   rR   r   )rC   r5  rD   rE   s       r!   -test_apply_with_args_kwds_subtract_and_divider8    sB    " 2aHF  !:;H&(+r#   c                 Z    | j                  t              }t        j                  ||        y rL   )r=   r   r>   rb   )rC   rD   s     r!   test_apply_yield_listr:    s"    t$F&+.r#   c                    t         j                  | j                  d d d| j                  j	                  d      f<   | j                  d      }| j                  t         j
                  d      }t        j                  ||       y )NrY   r   r   r   )	r   r   ilocr   get_locr@   r=   r>   r?   )rC   rE   rD   s      r!   test_apply_reduce_Seriesr>    sl    >@ffKSqS+--55c::;Q'HrwwQ/F68,r#   c                  d   t        ddgddggddgddg	      } | j                  t        d
      }t        ddddddg| j                        }t        j                  ||       | j                  t        d      }t        ddddddg| j                        }t        j                  ||       y )Nr   rY   r   rs   rl   rm   i0i1r   rA   r   r   )r@  rA  r   rk   )r   r=   r   r   r   r>   r?   rA   )r   rD   rE   s      r!   test_apply_reduce_to_dictrC    s    q!fq!f%d|D$<PDZZ1Z%Faq)!1+=>dllSH68,ZZ1Z%Faq)!1+=>djjQH68,r#   c            	         t        t        j                  j                  d      j	                  d            } | j                  t        j                  d      }t        | j                         D ci c]  \  }}||j                          c}}| j                        }t        j                  ||       | j                  t        j                  d      }t        | j                  j                         D ci c]  \  }}||j                          c}}| j                        j                  }t        j                  ||       y c c}}w c c}}w )NrY   )   r  r   r   r   r   )r   r   randomdefault_rngstandard_normalr=   r   describeitemsr   r>   rb   r   rA   )r    rD   ivrE   s        r!   test_apply_differently_indexedrM    s    	299((+;;HE	FBXXfooAX.FbhhjAda!QZZ\/A2::VH&(+XXfooAX.FbddjjlCda!QZZ\/CRXXVXXH&(+ B Ds   1E	
9E
c                     t        g dg dg dg dg dg dgg d      } d	 }| j                  |d
      }t        t        dd
d
      ddgt        dd
d      ddgt        dd
d      ddgt        dd
d      ddgt        dd
d      ddgt        dd
d      ddggg d      } | j                  |d
      }t        j                  ||       y )N)r   ABC02   )r   YUM0rE  )r   DEF0rE  )rY   ABC1rP  )rY   YUM1rE  )rY   DEF1rE  )rW   marketpositionr   c                     | d   S )NrV  rM   )rs    r!   fztest_apply_bug.<locals>.f  s    {r#   r   r   i  rO  rP  rY   rQ  rE  r   rR  rs   rS  r   rT  r   rU  )r   r=   r   r>   r?   )	positionsrZ  rE   rD   s       r!   test_apply_bugr\    s    	
 ,
I qq)HdAq!62.dAq!62.dAq!62.dAq!62.dAq!62.dAq!62.	
 ,
I __QQ_'F68,r#   c            
         t        g dg dg dt        j                  j                  d      j	                  d      t        j                  j                  d      j	                  d      t        j                  j                  d      j	                  d      d      } | j                  d d	      }t        j                  ||        y )
N)r  r  r  r  barr^  r^  r^  r  r  r  )oner_  r_  twor_  r_  r_  r`  r`  r`  r_  )dullra  shinyra  ra  rb  rb  ra  rb  rb  rb  rY      )r   r   r   rj   EFc                     | S rL   rM   r~   s    r!   rQ   z,test_apply_convert_objects.<locals>.<lambda>L  r   r#   r   r   )r   r   rF  rG  rH  r=   r>   rb   r   s     r!   test_apply_convert_objectsrg    s     &&q)99"=&&q)99"=&&q)99"=U+	
-H^ ^^Ka^0F&(+r#   c                     | j                  d       }t        | j                  | j                        }t        j                  ||       y )Nc                     | j                   S rL   namer~   s    r!   rQ   z(test_apply_attach_name.<locals>.<lambda>Q  
     r#   r   )r=   r   r   r>   r?   r   s      r!   test_apply_attach_namerm  P  s;    /0Fk))1D1DEH68,r#   c                     | j                  d d      }t        | j                  | j                        }t        j                  ||       y )Nc                     | j                   S rL   rj  r~   s    r!   rQ   z.test_apply_attach_name_axis1.<locals>.<lambda>W  rl  r#   r   r   r   )r=   r   rA   r>   r?   r   s      r!   test_apply_attach_name_axis1rp  V  s@    /a8Fk''{/@/@AH68,r#   c                     | j                  d       }t        t        j                  | j                  t        | j                        df      | j                  | j                        }t        j                  ||       y )Nc                 T    t        j                  | j                  t        |             S rL   r   repeatrk  r   r~   s    r!   rQ   z6test_apply_attach_name_non_reduction.<locals>.<lambda>^      1663q6)B r#   r   rA   r   )	r=   r   r   r   r   r   rA   r>   rb   r   s      r!   $test_apply_attach_name_non_reductionrw  \  se    BCF
##c+*;*;&<a%@A##H
 &(+r#   c                       j                  d d      }t         fd j                         D              } j                  |_        t	        j
                  ||       y )Nc                 T    t        j                  | j                  t        |             S rL   rs  r~   s    r!   rQ   z<test_apply_attach_name_non_reduction_axis1.<locals>.<lambda>h  ru  r#   r   r   c              3   v   K   | ]0  }t        j                  |d    t        j                               2 yw)r   N)r   rt  r   r   ).0trC   s     r!   	<genexpr>z=test_apply_attach_name_non_reduction_axis1.<locals>.<genexpr>i  s0      67		!A$K//01s   69)r=   r   
itertuplesrA   r>   r?   r   s   `  r!   *test_apply_attach_name_non_reduction_axis1r  g  sX    BKF ;F;Q;Q;S H !&&HN68,r#   c                      t        j                  g dg dg      } t        ddgddgddgg| d	d
g      }|j                  d d      }t        ddgddgddgg| ddg      }t	        j
                  ||d       y )N)rW   rW   rX   )r   rF   rF   r   rY   r   rs   r   r   col1col2rv  c                 B    t        t        |       t        |       d      S )N)minmax)r   r  r  r~   s    r!   rQ   z(test_apply_multi_index.<locals>.<lambda>s  s    vc!fSV&DE r#   r  r  T
check_like)r   from_arraysr   r=   r>   rb   )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  r^  spameggs)r   r   rY   r   c                     d }| j                  |d      }| j                  |d      }| j                  |      }t        j                  ||       t        j                  ||        t        j                  ||       y )Nc                 "    | j                         S rL   )to_dictr~   s    r!   rQ   z!test_apply_dict.<locals>.<lambda>  s    199; r#   r   r   r   )r=   r>   r?   rb   )r    dictsfnreduce_truereduce_falsereduce_nones         r!   test_apply_dictr  x  sj     
B((28(4K88BH85L((2,K;.,+;.r#   c                     t        dt        ddd      i      } | j                  d       }t        j                  ||        | j                  d       }t        dt        ddd      i      }t        j                  ||       y )	Ndtz
2015-01-01r   zEurope/Brussels)r   r|   c                     | S rL   rM   r~   s    r!   rQ   z,test_apply_non_numpy_dtype.<locals>.<lambda>       r#   c                 2    | t        j                  d      z   S )N1day)re   	Timedeltar~   s    r!   rQ   z,test_apply_non_numpy_dtype.<locals>.<lambda>  s    BLL$8 8 r#   z
2015-01-02)r   r
   r=   r>   rb   r   s      r!   test_apply_non_numpy_dtyper    sv    	D*\1ARST	UBXXk"F&"%XX89F	z,6GHIH &(+r#   c                  v    t        dg did      } | j                  d       }t        j                  ||        y )Nr  )rW   rX   r   rW   ro   r   c                     | S rL   rM   r~   s    r!   rQ   z5test_apply_non_numpy_dtype_category.<locals>.<lambda>  r  r#   r   rv   s     r!   #test_apply_non_numpy_dtype_categoryr    s2    	D./z	BBXXk"F&"%r#   c                      t        ddgddggddg      } t        ddggddgdg      }| j                  dg      }t        j                  ||       y )	Nr   r   rY   r   rW   r   r  rB  r   r    rE   rD   s      r!   test_apply_dup_names_multi_aggr    sX    	QFQF#c3Z	8B1a&C:eWEHVVUG_F&(+r#   opr=   r   c                     d }t        t        j                  d      t        d            } t	        ||       |d      }t        g dg dg dg dg      }t        j                  ||       y )Nc                 0    d| d   z  d| d   z  d| d   z  gS )NrY   r   r   r   rM   rows    r!   
apply_listz3test_apply_nested_result_axis_1.<locals>.apply_list  s'    CHa#c(lACL99r#   )rs   rs   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    s^    : 
288F#T&\	:BWR_Za0F	/?OLH 68,r#   c                      t        dt        dd      gid      } | j                  d       }t        j                  ||        | j                  d       }t        j                  ||        y )	Nr  2020rz   r{   zdatetime64[ns, UTC]r   c                     | S rL   rM   r~   s    r!   rQ   z7test_apply_noreduction_tzaware_object.<locals>.<lambda>  r   r#   c                 "    | j                         S rL   )r   r~   s    r!   rQ   z7test_apply_noreduction_tzaware_object.<locals>.<lambda>  s    affh r#   r   r   s     r!   %test_apply_noreduction_tzaware_objectr    s`    	6e,-.6KH ^^K(F&(+^^./F&(+r#   c                      t        dg di      } g fd}fd}||fD ]4  }d d = | j                  |d       t        | j                        k(  r4J  y )NrW   r   c                 <    j                  | j                         y rL   r   rk  r  namess    r!   reducing_functionz8test_apply_function_runs_once.<locals>.reducing_function  s    SXXr#   c                 >    j                  | j                         | S rL   r  r  s    r!   non_reducing_functionz<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    sg     
C#	$BE #$9: '!H
ARXX&&&	'r#   c                    | dk(  rt        j                  d       t        dg di      }g fd}fd}||fD ]D  }d d = |j                  || dd	       t	        |j
                  j                               k(  rDJ  y )
Nr%   z8appending to list outside of numba func is not supportedrW   r   c                 (    j                  |        y rL   extendr  r   s    r!   r  z<test_apply_raw_function_runs_once.<locals>.reducing_function  s    cr#   c                 *    j                  |        | S rL   r  r  s    r!   r  z@test_apply_raw_function_runs_once.<locals>.non_reducing_function  s    c
r#   Tr   )r-   rH   r6   )r*   ra   r   r=   r   rW   to_list)r-   r    r  r  r   r   s        @r!   !test_apply_raw_function_runs_oncer    s    NO	C#	$BF #$9: .1I
f$Q7bddlln----	.r#   c                      t        t        j                  ddg      dg      } t        t        j                  ddg      dgt              }| j	                  d       }t        j                  ||       y )Ns   abcds   efghcolr   )r   r   c                 $    | j                  d      S )Nobjectrp   r~   s    r!   rQ   z-test_apply_with_byte_string.<locals>.<lambda>  s     2 r#   )r   r   r  r  r=   r>   rb   r  s      r!   test_apply_with_byte_stringr    s]    	288Wg./%	AB7G"45wfUH XX23F&(+r#   valasd   c                 ,    dd dddt         j                  g}t        d|id      }|j                  j	                   fd      }t        |D cg c]  }t        j                  |      rdn| k(    c}d	      }t        j                  ||       y c c}w )
Nr  r  cderW   ro   r   c                     | k(  S rL   rM   )rO   r  s    r!   rQ   z/test_apply_category_equalness.<locals>.<lambda>  s    !s( r#   Frj  )
r   r   r   rW   r=   r   re   isnullr>   r?   )r  	df_valuesr    rD   rO   rE   s   `     r!   test_apply_category_equalnessr    s     b%7I	C#:	6BTTZZ*+F6?@"))A,AH	,@sH 68, 	As   #Bc                  b   t        t        j                  j                  d      j                  d            } | j	                  t        j
                  j
                  d      j                  }|dk(  sJ | j	                  t        j
                  j                  d      j                  }|dk(  sJ y )NrY   )r  rY   r   r   )r   rY   )r   r   rF  rG  r=   fftrt   rfftrv   s     r!   test_infer_row_shaper  
  s     
299((+227;	<BXXbffjjqX)//FWXXbffkkX*00FVr#   zops, by_row, expectedrW   c                     | dz   S r   rM   r~   s    r!   rQ   rQ     
    Q r#   compatc                     | dz   S r   rM   r~   s    r!   rQ   rQ     r  r#   c                 "    | j                         S rL   r   r~   s    r!   rQ   rQ          r#   c                 "    | j                         S rL   r  r~   s    r!   rQ   rQ     r  r#   r   c                 "    | j                         S rL   r  r~   s    r!   rQ   rQ         AEEG r#   )r   r   r   )r   r   <lambda>r   c                 "    | j                         S rL   r  r~   s    r!   rQ   rQ   "  r  r#   c                      yr   rM   r~   s    r!   rQ   rQ   &      r#   c                      yr   rM   r~   s    r!   rQ   rQ   '  r  r#   c                 t    t        dddgi      }|j                  | |      }t        j                  ||       y NrW   r   rY   by_rowr   r=   r>   assert_equalopsr  rE   r    rD   s        r!   test_dictlike_lambdar    s6    . 
C!Q=	!BXXc&X)FOOFH%r#   r  c                     | dz   S r   rM   r~   s    r!   rQ   rQ   4  s
    A r#   c                 "    | j                         S rL   r  r~   s    r!   rQ   rQ   5  s     r#   c                 "    | j                         S rL   r  r~   s    r!   rQ   rQ   6  s     r#   c                      yr   rM   r~   s    r!   rQ   rQ   7  r  r#   c                     t        dddgi      }t        j                  t        d      5  |j	                  | d       d d d        y # 1 sw Y   y xY wNrW   r   rY   zby_row=True not allowedrZ   Tr  r   r*   rc   
ValueErrorr=   r  r    s     r!   test_dictlike_lambda_raisesr  1  J     
C!Q=	!B	z)B	C #
T"# # #   AAc                     t        ddgddggddg      } | j                  d d      }t        | j                         D cg c]  }dd	i c}      }t	        j
                  ||       t        d
      t        d      g| d<   | j                  d d      }t	        j
                  ||       | d   | d   z   j                  d       }t        dd	idd	ig      }t	        j
                  ||       y c c}w )Nr   rY   rW   rX   r   c                     d| d   | d   z   iS Nr   rW   rX   rM   r~   s    r!   rQ   z,test_with_dictlike_columns.<locals>.<lambda>D      afqvo 6 r#   r   r   r   2017-05-01 00:00:002017-05-02 00:00:00r>   c                     d| d   | d   z   iS r  rM   r~   s    r!   rQ   z,test_with_dictlike_columns.<locals>.<lambda>L  r  r#   c                 
    d| iS )Nr   rM   r~   s    r!   rQ   z,test_with_dictlike_columns.<locals>.<lambda>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9AQx9:H68, 	'('(BtH XX6QX?F68, g3&&'9:FQx#q*+H68, :s   C/c                      t               } g d| d<   g d| d<   t        j                  g dd      | d<   | j                  d	 d
      }t	        i i i g      }t        j                  ||       y )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                     i S rL   rM   r~   s    r!   rQ   z:test_with_dictlike_columns_with_datetime.<locals>.<lambda>^       r#   r   r   )r   re   to_datetimer=   r   r>   r?   r   s      r!   (test_with_dictlike_columns_with_datetimer  U  sg    	B"BxL+B{OMBvJ XXlX+Fr2rl#H68,r#   c                  *   t        ddgddggddg      } | j                  d dd      }t        d	d
d
gi      }t        j                  ||       t	        d      t	        d      g| d<   | j                  d dd      }t        j                  ||       y )Nr   rY   rW   rX   r   c                     d| d   | d   z   iS r  rM   r~   s    r!   rQ   z7test_with_dictlike_columns_with_infer.<locals>.<lambda>f  r  r#   r   r   r   r   r  r  r>   c                     d| d   | d   z   iS r  rM   r~   s    r!   rQ   z7test_with_dictlike_columns_with_infer.<locals>.<lambda>n  r  r#   )r   r=   r>   rb   r	   r   s      r!   %test_with_dictlike_columns_with_inferr  c  s    	QFQF#c3Z	8BXX6QHXUF#1v'H&(+ 	'('(BtH XX6QHXUF&(+r#   c                     | dz   S r   rM   r~   s    r!   rQ   rQ   u  
    AE r#   )rW   r  c                     | dz   S r   rM   r~   s    r!   rQ   rQ   v  r  r#   c                 "    | j                         S rL   r  r~   s    r!   rQ   rQ   w      AEEG r#   r  c                 "    | j                         S rL   r  r~   s    r!   rQ   rQ   x  r  r#   c                 "    | j                         S rL   r  r~   s    r!   rQ   rQ   z      aeeg r#   c                 "    | j                         S rL   r  r~   s    r!   rQ   rQ     r  r#   c                     | dz   S r   rM   r~   s    r!   rQ   rQ     s
    q1u r#   c                      yNr   rM   r~   s    r!   rQ   rQ     r  r#   r   c                      yr   rM   r~   s    r!   rQ   rQ     r  r#   c                      yr!  rM   r~   s    r!   rQ   rQ     r  r#   c                 t    t        dddgi      }|j                  | |      }t        j                  ||       y r  r  r  s        r!   test_listlike_lambdar%  r  s6    > 
C!Q=	!BXXc&X)FOOFH%r#   c            
         t        ddd      t        ddd      t        ddd      g} t        dddgi| d   | d   g      }|j                  d d	 gd
      }t        ddgdt        j                  gt        j                  dgg| d   | d   | d   gt        dgdggddgddgg            }t        j                  ||       y )N  r   rY   r   rW   r   r   c                     | S rL   rM   r~   s    r!   rQ   z7test_listlike_datetime_index_unsorted.<locals>.<lambda>  s     r#   c                 &    | j                  d      S Nrj   )freqshiftr~   s    r!   rQ   z7test_listlike_datetime_index_unsorted.<locals>.<lambda>  s    agg3g.? r#   Fr  r  r7  r  )codesrv  )r   r   r=   r   r   r   r>   rb   r   r    rD   rE   s       r!   %test_listlike_datetime_index_unsortedr0    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                  `   t        ddd      t        ddd      t        ddd      g} t        ddgddgd| d   | d   g      }|j                  d	 d
 dd      }t        ddt        j                  gdt        j                  dgd| d   | d   | d   g      }t        j                  ||       y )Nr'  r   rY   r   rs   rW   rX   r   r   c                     | S rL   rM   r~   s    r!   rQ   z7test_dictlike_datetime_index_unsorted.<locals>.<lambda>  s     r#   c                 &    | j                  d      S r*  r,  r~   s    r!   rQ   z7test_dictlike_datetime_index_unsorted.<locals>.<lambda>  s    !''s'*; r#   Fr  r  r7        @      @)r   r   r=   r   r   r>   rb   r/  s       r!   %test_dictlike_datetime_index_unsortedr7    s    tQ"HT1a$8(4A:NOF	!Qq!f-fQi5K	LBXX ;<U  F sBFF#rvvs#	
 ay&)VAY/H &(+r#   c                     | dz   S r   rM   r~   s    r!   rQ   rQ     
    1q5 r#   c                 "    | j                         S rL   r  r~   s    r!   rQ   rQ     s    1557 r#   c                 "    | j                         S rL   r  r~   s    r!   rQ   rQ     s    !%%' r#   c                     | dz   S r   rM   r~   s    r!   rQ   rQ     r9  r#   c                      yr!  rM   r~   s    r!   rQ   rQ     r  r#   c                     t        dddgi      }t        j                  t        d      5  |j	                  | d       d d d        y # 1 sw Y   y xY wr  r  r  s     r!   test_listlike_lambda_raisesr?    r  r  c            	      @   t        t        t        j                  j	                  d      j                  d            g dt        ddd      d      } | dd	g   j                  t        d
      }t        | dd	g   j                         D cg c]  }|d
d  	 c}      }t        j                  ||       | ddg   j                  t        d
      }t        | ddg   j                         D cg c]  }|d
d  	 c}      }t        j                  ||       y c c}w c c}w )NrY   rs   )rW   r   ofwordsz
2016-10-01h)r   r+  )rW   rX   rr   rW   rX   r   r   rr   )r   r   r   rF  rG  rH  r
   r=   tupler~  r>   r?   r  s       r!   test_with_listlike_columnsrE    s   			--a0@@CD-\13?	

B c
^!!%a!0Fb#sn&?&?&ABquBCH68,d_""5q"1Fb#to&@&@&BCquCDH68, C Ds   D,Dc                     t        t        ddgdgg      t        dgddgg      d      } t        j                  ddg      | _        | j                  d	 d
      }t        g dgg| j                        }t        j                  ||       y )NrW   rX   qzr|  rN   )r@  j0)rA  j1c                 B    | d   D cg c]  }|| d   v s| c}S c c}w )NrO   rP   rM   )r  els     r!   rQ   z;test_with_listlike_columns_returning_list.<locals>.<lambda>  s"    C"K"B#c(N2"K "Ks   r   r   r   )r   r   r   from_tuplesrA   r=   r>   r?   r   s      r!   )test_with_listlike_columns_returning_listrN    s    	#scU 346C53PS*BU;VW	XB%%|\&BCBHXXKRSXTFrC5k2H68,r#   c                  *   t        ddgddgt        d      t        d      gd      } | j                  d d	
      }t        | j	                         D cg c]  }|j
                  |j                  f c}      }t        j                  ||       y c c}w )Nr  r7  r  r^  z2017-11-29 03:30:00z2017-11-29 03:45:00)numberstringr   c                 2    | j                   | j                  fS rL   )rP  rQ  r  s    r!   rQ   z1test_infer_output_shape_columns.<locals>.<lambda>  s    3::szz": r#   r   r   )	r   r	   r=   r   r~  rP  rQ  r>   r?   r  s       r!   test_infer_output_shape_columnsrS    s     
Cjen/0/0	
	
B XX:XCFR]]_E!((+EFH68, Fs   Bc                     t        t        j                  j                  d      j	                  d      g d      } | j                  d d      }t        | j                         D cg c]  }g d c}      }t        j                  ||       | j                  d	 d      }t        | j                         D cg c]  }ddg c}      }t        j                  ||       y c c}w c c}w )
NrY   )r   r   r   r   c                 
    g dS r   rM   r~   s    r!   rQ   z:test_infer_output_shape_listlike_columns.<locals>.<lambda>  r   r#   r   r   r   c                 
    ddgS Nr   rY   rM   r~   s    r!   rQ   z:test_infer_output_shape_listlike_columns.<locals>.<lambda>  
    A r#   
r   r   rF  rG  rH  r=   r   r~  r>   r?   r  s       r!   (test_infer_output_shape_listlike_columnsrZ    s     

		a 008/
B XX)X2F"--/:Qy:;H68,XX&QX/Fr}}7!1v78H68, ; 8s   'C#:C(c                 (    t        dg dit        d            }|j                   fdd      }t        |j	                         D cg c]  }t        j                          c}|j                        }t        j                  ||       y c c}w )NrW   r   r   r   c                 .    t        j                        S rL   )r   ones)r  r  s    r!   rQ   zBtest_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^    sk     
C#4;	7BXX.QX7FR]]_=rwws|=RXXNH68, >s   	Bc                  0   t        t        d      t        d      t        d      t        d      gg dg dg dd      } d	 }| j                  |d
      }t        | j	                         D cg c]  }d c}      }t        j                  ||       y c c}w )Nz
2010-02-01z
2010-02-04z
2010-02-05z
2010-02-06)r   r   rs   r   )r   r   rs   rY   )r   rY   r   rs   )rW   rX   r   rF   c                      y)Nr   rY   rM   r~   s    r!   funzDtest_infer_output_shape_listlike_columns_with_timestamp.<locals>.fun  s    r#   r   r   ra  )r   r	   r=   r   r~  r>   r?   )r    rb  rD   r|  rE   s        r!   7test_infer_output_shape_listlike_columns_with_timestamprc    s    	 ,',',','	 
	

B XXcX"Fr}}7!v78H68, 8s   ,	Blstr   c                 ,    t        t        j                  j                  d      j	                  d      g d      }|j                   fdd      }t        |j                         D cg c]  }  c}      }t        j                  ||       y c c}w )NrY   )rs   r   r   r   c                     S rL   rM   )rO   rd  s    r!   rQ   z3test_consistent_coerce_for_shapes.<locals>.<lambda>-  s     r#   r   r   rY  )rd  r    rD   r|  rE   s   `    r!   !test_consistent_coerce_for_shapesrg  %  sr     

		a 008/
B XXm!X,FBMMO4qs45H68, 5s   *	Bc                     | }|j                  d d      }| j                  dddd      }t        j                  ||       |j                  d	 d      }|ddg   }t        j                  ||       y )
Nc                 $    t        g dg d      S )Nr   )testothercolsr   r   r~   s    r!   rQ   z'test_consistent_names.<locals>.<lambda>7  s    &*CD r#   r   r   rj  rk  rl  r   r   c                 $    t        ddgddg      S )Nr   rY   rj  rk  r   rm  r~   s    r!   rQ   z'test_consistent_names.<locals>.<lambda>>  s    1vfg5F G r#   )r=   renamer>   rb   r   s       r!   test_consistent_namesrp  2  s    	BXXD1  F #))78 * H &(+XXGaXPF)*H&(+r#   c                     | }|j                  d dd      }|j                         }t        d      |_        t	        j
                  ||       y )Nc                 
    g dS r   rM   r~   s    r!   rQ   z"test_result_type.<locals>.<lambda>H  r   r#   r   r   r   r   r=   r   r   r   r>   rb   r   s       r!   test_result_typert  C  sE     
BXX)xXHFwwyHQxH&(+r#   c                     | }|j                  d dd      }|ddg   j                         }t        d      |_        t	        j
                  ||       y )Nc                 
    ddgS rW  rM   r~   s    r!   rQ   z/test_result_type_shorter_list.<locals>.<lambda>R  rX  r#   r   r   r   r   r   rY   rs  r   s       r!   test_result_type_shorter_listrw  N  sP     
BXX&QHXEF3*~""$HQxH&(+r#   c                    |dk(  r;t         j                  j                  d      }|j                  j	                  |       | }|t
        u r;t        j                  t        d      5  |j                  d dd|	       d d d        y |j                  d
 dd|	      }|j                         }t        j                  ||       y # 1 sw Y   y xY w)Nr%   z(numba engine doesn't support list returnr/   ?result_type='broadcast' only implemented for the default enginerZ   c                 
    g dS r   rM   r~   s    r!   rQ   z,test_result_type_broadcast.<locals>.<lambda>e  s    ) r#   r   r   r6   r   r-   c                 
    g dS r   rM   r~   s    r!   rQ   z,test_result_type_broadcast.<locals>.<lambda>j  s    i r#   r*   r7   r8   r9   r:   r   rc   NotImplementedErrorr=   r   r>   rb   )r"   r,   r-   r7   r    rD   rE   s          r!   test_result_type_broadcastr  X  s     {{  (R S%	B$$]]S
 	 HH#!V  		 	 a[  
 779
fh/	 	s   &CCc                    |dk(  r;t         j                  j                  d      }|j                  j	                  |       | }g d|t
        u r=t        j                  t        d      5  |j                  fddd	|
       d d d        y |j                  fddd	|
      }|j                         }t        j                  ||       y # 1 sw Y   y xY w)Nr%   <numba Series constructor only support ndarrays not list datar/   rk  r  r  ry  rZ   c                 "    t        g d      S Nr   r   rm  rO   r   s    r!   rQ   z8test_result_type_broadcast_series_func.<locals>.<lambda>  s    &': r#   r   r   r{  c                 "    t        g d      S r  rm  r  s    r!   rQ   z8test_result_type_broadcast_series_func.<locals>.<lambda>  s    fYg6 r#   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
fh/%		 		s   +CCc                     |dk(  r;t         j                  j                  d      }|j                  j	                  |       | }|j                  d d|      }|j                         }t        j                  ||       y )Nr%   r  r/   c                 4    t        g d| j                        S r  )r   rA   r~   s    r!   rQ   z0test_result_type_series_result.<locals>.<lambda>  s    y @ r#   r   r5   )	r*   r7   r8   r9   r:   r=   r   r>   rb   )r"   r-   r,   r7   r    rD   rE   s          r!   test_result_type_series_resultr    sr     {{  Q ! 
 	%	BXX@qQWXXFwwyH&(+r#   c                    |dk(  r;t         j                  j                  d      }|j                  j	                  |       | }g d|j                  fdd|      }|j                         }|_        t        j                  ||       y )Nr%   z:no support in numba Series constructor for list of columnsr/   r  c                 "    t        g d      S r  rm  r  s    r!   rQ   z<test_result_type_series_result_other_index.<locals>.<lambda>  s    y @ r#   r   r5   )
r*   r7   r8   r9   r:   r=   r   r   r>   rb   r  s          @r!   *test_result_type_series_result_other_indexr    s     {{  O ! 
 	%	B'GXX@qQWXXFwwyHH&(+r#   boxc                     t        |       S rL   )r   r~   s    r!   rQ   rQ     s
    tAw r#   c                     t        |       S rL   )rD  r~   s    r!   rQ   rQ     s
    %( r#   c                 0    t        j                  | d      S )Nr   r   )r   r  r~   s    r!   rQ   rQ     s    bhhq6P r#   )r   rD  r  c           	      X    |}|j                   fdd      }t        |j                         D cg c]  }  ddg       c}      }t        j                  ||       |j                   fddd      }|dd	g   j                  d
dd      }t        j                  ||       y c c}w )Nc                      ddg      S rW  rM   rO   r  s    r!   rQ   z,test_consistency_for_boxed.<locals>.<lambda>      QF r#   r   r   rY   c                      ddg      S rW  rM   r  s    r!   rQ   z,test_consistency_for_boxed.<locals>.<lambda>  r  r#   r   r   r   r   r   r   r   )r=   r   r~  r>   r?   ro  rb   )r  r"   r    rD   r|  rE   s   `     r!   test_consistency_for_boxedr    s     
BXX+!X4FBMMO<qsAq6{<=H68,XX+!XJF"C:.55AA>N5OH&(+ =s   B'c                    | dv rdnd}t        j                  d      5  t        j                  |      }t        j                  |      }|j	                         }|j                  t         j                  |       }t        j                  ||       |j                  t         j                  g|       }|j	                         }| dv r(t        j                  |j                  dgg      |_
        n't        j                  |j                  dgg      |_        t        j                  ||       |j                  t         j                  t         j                  g|       }t        ||g|      }| dv r)t        j                  |j                  ddgg      |_
        n(t        j                  |j                  ddgg      |_        t        j                  ||       d d d        y # 1 sw Y   y xY w)	N   rA   r   r   r   r1   r2   r   r<   absolute)r   r;   absr<   r   r=   r>   rb   r   from_productr   rA   r   )r6   rC   
other_axisf_absf_sqrtrE   rD   s          r!   test_agg_transformr    s   l*J		" 0{#% ;;=""277"6
fh/ ""BGG94"8;;=<)668K8KfX7VWH'44k6G6G&5RSHN
fh/
 ""BFFBGG#44"@ufoJ?<)66$$z6&:; H (44""Z$89HN 	fh/?0 0 0s   F8G  G)c                      t        t        d      dd      } | j                  ddg      }t        ddgddgdddgddg	      }t        j                  ||       y )
Nr   r   r  r  r   rs   r   r   rB  )r   r   r   r>   rb   r   s      r!   	test_demor    s_    	q*	+BVVUEN#F!fAq6"S#JuenH &(+r#   c                     t        t        d      dd      } | j                  ddgddgd      }t        ddt        j                  gdt        j                  d	gdd
dgg d      }t        j                  |j                  |      |       y )Nr   r   r  r  r   r5  r        @g      9@r   r   )r  r  r   rB  )r   r   r   r   r   r>   rb   reindex_liker   s      r!   test_demo_dict_aggr    s    	q*	+BVV5%.u~>?FC RVVT':;c
#H
 &--h7Br#   c                     dddgi} t        |       }|j                  ddi      }t        ddi      }t        j                  ||       |d   j                  ddi      }t        ddid      }t        j                  ||       y )Nrk  r  r^  countrY   rj  )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68, Z^^VW-.Fvqk/H68,r#   c                  >   t        g dg dg dd      } t        ddgddgd	d
gdddg      }| j                  ddg      }t        j                  ||       | g d   j                  ddg      }|g d   j	                  ddg      }t        j                  ||       y )Nr   r  r7  r6  r  r^  bazr   r   r   r        @r^  	foobarbazr  r   r   )r   r   r   )r   r   r>   rb   reindex)mdfrE   rD   s      r!   test_agg_multiple_mixedr    s    
 &	
C Qs%	

 enH WWeU^$F&(+!%%uen5F (00%@H&(+r#   c            	      `   t        g dg dg dt        dd      d      } d}t        j                  t        |	      5  | j                  d
dg       d d d        t        j                  t        |	      5  | g d   j                  dd
g       d d d        y # 1 sw Y   FxY w# 1 sw Y   y xY w)Nr   r  r  r   r   r   r   r   r   rj   does not support operationrZ   r  r   )rj   r   r   r   )r   r
   r*   rc   rd   r   )r  msgs     r!   test_agg_multiple_mixed_raisesr  ,  s    
 &J2		
C 'C	y	,    
y	, 6 !%%uen56 6   6 6s   B6B$B!$B-c                    | dv rdnd}|j                   |   j                         d d j                         \  }}t        j                  |j                  |       |j                  |       |j                  |       gd      }g d|_        | dv r|j                  n|}|j                  g d|       }t        j                  ||       |d|di}|j                  ||       }t        |j                  |      |   j                         |j                  |      |   j                         g||g	      }t        j                  ||       |dg|dgi}|j                  ||       }t!        |t        |j                  |      |   j                         gdg	      |t        |j                  |      |   j                         gdg	      i      }| d
v r|j                  n|}t        j                  ||       |ddg|ddgi}|j                  ||       }t        j                  |t        |j                  |      |   j                         |j                  |      |   j                         gddg	      |t        |j                  |      |   j                         |j                  |      |   j                         gddg	      id      }| d
v r|j                  n|}t        j                  ||       y )Nr  r   r   rY   r   )r@   r  r   r@   r   r   >   r   r   r  )axesuniquesort_valuesre   concatr@   r  r   r   r   r   r>   rb   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JLE5 yy$'OOO&OOO&	

 H .H!\1xzzxH__3$_?F&(+ 65%(D__T_-FOOJ'.335OOJ'.224	
 enH 68, 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                    | j                   d   j                         d d j                         \  }}d}dD ]>  }t        j                  t
        |      5  | j                  |df|df|       d d d        @ y # 1 sw Y   KxY w)	Nr   rY   z/Named aggregation is not supported when axis=1.)r   r   rZ   r   r  )row1row2r6   )r  r  r  r*   rc   r~  r   )rC   r  r  r  r6   s        r!   "test_named_agg_reduce_axis1_raisesr    s    ##A&--/3??ALE5
;C Q]].c: 	QOO%uen4OP	Q 	QQ	Q 	Qs   A88B	c            	         t        g dg dg dt        dd      d      } | j                  d      }t        d	d
dt	        d      g| j
                        }t        j                  ||       | j                  dg      }t        d	d
dt	        d      ggdg| j
                        }t        j                  ||       d}t        j                  t        |      5  | j                  d       d d d        | g d   j                  d      }t        g dg d      }t        j                  ||       d}t        j                  t        |      5  | j                  dg       d d d        y # 1 sw Y   xY w# 1 sw Y   y xY w)Nr   r  r  r   r   r   r  r  r   r  r^  r   rv  r  rZ   r   r   )r   r  r  )r   r
   r   r   r	   r   r>   r?   rb   r*   rc   rd   )r    rD   rE   r  s       r!   test_nuiscance_columnsr    sM   	 &J2		

B VVE]Fq#ui
&;<BJJOH68,VVUG_F
S%:.	/0g

H
 &(+
&C	y	, 
u  $$U+F+?CH68,
&C	y	, 
w   s   E)E5)E25E>howc                    t        g ddt        j                  dgg dd      } t        ||       ddi      }t	        ddi      }t        j                  ||        t        ||       dd	i      }t	        dd
i      }t        j                  ||        t        ||       dd	g      } t        ||       dd	gdd	gdd	gd      }t        dd
ddd
ddd
dd      }t        j                  ||d       t        j                  ||d        t        ||       d      }|j                         }t        j                  ||       y )NNrY   r   r  r6  r  Nr^  r   r   r  rY   sizer   )r  r  Tr  )	r   r   r   r   r   r>   r?   rb   r  )r  r    rD   rE   result1result2s         r!   test_non_callable_aggregatesr    sW    
#rvvs!3:NO
B
 WRsGn-FsAhH68, WRsFm,FsAhH68, gb#01Ggb#w&7w>OPG a(a(a(	
H '7t<'8= WRg&FxxzH68,r#   c                 :   t        g ddt        j                  dgg dd      } t        ||       d|      }|dv r%t	        |j
                  d	   |j                  
      }n$t	        |j
                  d   |j                  
      }t        j                  ||       y )Nr  r  r6  r  r   r  r   )r   rA   r   r   r   )
r   r   r   r   r   rt   r   rA   r>   r?   )r  r6   r    rD   rE   s        r!   test_size_as_strr    s     
#rvvs!3:NO
B
 WRf40F|"((1+RZZ8"((1+RXX668,r#   c                  T   t        g ddt        j                  dgg dd      } d }| j                  |      }t	        ddgdgdd	ggg d
      }t        j                  ||       | j                  |g      }|j                  d      j                  }t        j                  ||       y )N)rY   rY   r   g      ?r  r   c                 P    t        | j                         j                               S rL   )r   dropnar  )	group_cols    r!   r   z&test_agg_listlike_result.<locals>.func  s    I$$&--/00r#   rY   r   r  r^  r   r   )
r   r   r   r   r   r>   r?   to_framer   rb   )r    r   rD   rE   s       r!   test_agg_listlike_resultr    s    	#rvvs);BVW	XB1 VVD\F1vuuen5_MH68,VVTF^F  (**H&(+r#   zargs, kwargs)r   r      ra  rR   rX   rM   r2  r   c                     dd}t        ddgddgg      }| dk(  rt        ddg      }nt        d	d
g      } |j                  || g|i |}t        j                  ||       y )Nr   c                 4    | j                         ||z   |z  z   S rL   r  )rO   rW   rX   r   s       r!   rZ  ztest_agg_args_kwargs.<locals>.f  s    uuw!a%1$$r#   r   rY   rs   r   r  g      @r5  g       @)r   r   r   r   r>   r?   )r6   rS   kwargsrZ  r    rE   rD   s          r!   test_agg_args_kwargsr    so    % 
QFQF#	$Bqy3*%3*%RVVAt-d-f-F68,r#   num_cols)rY   r   r   c                 ^   |dk(  r;t         j                  j                  d      }|j                  j	                  |       t        j                  g d      }|j                         }t        d|t        |             }|j                  d |       |j                  |j                  k(  sJ y )	Nr%   z*numba engine only supports numeric indicesr/   )z
1950-06-30z
1952-10-24z
1953-05-29r   rv  c                     | S rL   rM   r~   s    r!   rQ   z,test_frequency_is_original.<locals>.<lambda>$  r  r#   r4   )r*   r7   r8   r9   r:   re   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                 ^   | dk(  r;t         j                  j                  d      }|j                  j	                  |       t        dd      t        dd      t        dd      g}t        g d	|
      }|j                  d d|       }t        ||      }t        j                  ||       y )Nr%   z0numba engine doesn't support non-numeric indexesr/   z2019-03-15 12:34:31.909000+0000rz   r{   z2019-03-15 12:34:34.359000+0000z2019-03-15 12:34:34.660000+0000r   r   rY   )r   rA   c                     | j                   S rL   rj  r~   s    r!   rQ   z.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JZ8H68,r#   r    r   rF   r   method)r  r  r   c                     |dk(  rd}nd}|s6t        j                  t        |      5   t        | |              d d d        y  t        | |              y # 1 sw Y   y xY w)Nr   z>can only concatenate str \(not "int"\) to str|does not supportz4not supported between instances of 'str' and 'float'rZ   )r*   rc   rd   r   )r    r  using_infer_stringr  s       r!   test_mixed_column_raisesr  =  s`     OD]]9C0 	"GB!	" 	" 	F	" 	"s   AAr  r  c                     t        d| ggddg      }|j                  d       }|j                  }t        j                  ||       y )Nr  rW   rX   r   c                     | j                   S rL   r   r~   s    r!   rQ   z"test_apply_dtype.<locals>.<lambda>P  s
     r#   )r   r=   dtypesr>   r?   )r  r    rD   rE   s       r!   test_apply_dtyper  L  sB     
S#J<#s	4BXX'(FyyH68,r#   c                  $   t        t        d      t        dd      d      } | j                         }d }| j                         }|dxx   dz  cc<   | j                  |d      }t	        j
                  ||       t	        j
                  | |       y )Nr  rE  r2  c                 l    | j                   }| j                  dxx   dz  cc<   | j                   |usJ | S )NrW   r   )_mgrr   )r  mgrs     r!   r   z!test_apply_mutating.<locals>.func[  s4    hhxxs"""
r#   rW   r   r   )r   r   r   r=   r>   rb   )r    df_origr   rE   rD   s        r!   test_apply_mutatingr  V  sx    	rr27	8BggiG wwyHSMQMXXdX#F&(+"g&r#   c                      t        ddgddgddgddgd	d
ggddg      } | j                  d d      }t        g g dt              }t	        j
                  ||       y )Nr   rY   r   rs   r   r   r   r   r   r  rW   rX   r   c                     g S rL   rM   r~   s    r!   rQ   z.test_apply_empty_list_reduce.<locals>.<lambda>n  r  r#   r   r   r2  r   )r   r=   r   r  r>   r?   r   s      r!   test_apply_empty_list_reducer  j  sd    	QFQFQFQFQG<sCj	QBXXlX9FBR(7H68,r#   c                 6   | dk(  r;t         j                  j                  d      }|j                  j	                  |       t        ddggdz  ddg	      }|j                  d
d d g|       }t        g dg ddg d      }t        j                  ||       y )Nr%   z;numba engine doesn't support list-likes/dict-like callablesr/   rs   r   r   r   r   r   r   c                 "    | j                         S rL   r  r~   s    r!   rQ   z,test_apply_no_suffix_index.<locals>.<lambda>{  s     r#   c                 "    | j                         S rL   r  r~   s    r!   rQ   z,test_apply_no_suffix_index.<locals>.<lambda>{  s    AEEG r#   r4   )r  r  r  )   r  r  r   )r   r  r  r   )	r*   r7   r8   r9   r:   r   r=   r>   rb   )r-   r,   r7   pdfrD   rE   s         r!   test_apply_no_suffix_indexr  s  s    {{  P ! 
 	%
aVHqL3*
5CYY02CDVYTF.6UH &(+r#   c                     | dk(  rt        j                  d       t        dddgi      }|j                  d | dd	      }t	        ddg      }t        j                  ||       y )
Nr%   z No object dtype support in numbar   aabbbc                     | d   S r   rM   r~   s    r!   rQ   z/test_apply_raw_returns_string.<locals>.<lambda>  r   r#   r   T)r-   r6   rH   )r*   ra   r   r=   r   r>   r?   r-   r    rD   rE   s       r!   test_apply_raw_returns_stringr    s^    67	C$'	(BXXnV!XFFtUm$H68,r#   c                      t        g dd      } d }d|ddg}| j                  |      }t        g dg d	g d
dg d      }t        j                  ||       y )N))r   r   r   )rY   r   r   )r   r   r   )rs   r   rs   )r   r   r   )r   r   r   )att1att2att3r   c                 (    | j                         dz  S r   r  r   s    r!   sum_div2z4test_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   r
  r  r  r   r   )r    r
  aggsrD   rE   s        r!   "test_aggregation_func_column_orderr    si     
	
 )

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

 2H &(+r#   c                 "   | dk(  r;t         j                  j                  d      }|j                  j	                  |       t        g dg dd      }|ddg   j                  d d	| 
      }t        g d      }t        j                  ||       y )Nr%   z2numba engine not supporting duplicate index valuesr/   r  r   r2  rW   c                 @    | j                   d   | j                   d   z   S r_   )r<  r~   s    r!   rQ   z+test_apply_getitem_axis_1.<locals>.<lambda>  s    !&&)affQi' r#   r   r5   )r   rY   rs   )
r*   r7   r8   r9   r:   r   r=   r   r>   r?   )r-   r,   r7   r    rD   rE   s         r!   test_apply_getitem_axis_1r    s    {{  G ! 
 	%	3	4Bc
^!!'a " F i H68,r#   c                      d } t        dg di      }t        j                  t        d      5  |j	                  | g       d d d        y # 1 sw Y   y xY w)Nc                 L    t        j                  d       | j                         S )NHello, World!)warningswarnr   r~   s    r!   expected_warningzDtest_nuisance_depr_passes_through_warnings.<locals>.expected_warning  s    o&uuwr#   rW   r   r  rZ   )r   r>   assert_produces_warningUserWarningr   )r  r    s     r!   *test_nuisance_depr_passes_through_warningsr    sM     
C#	$B		#	#K	G #
 !"# # #s   A

Ac            	         t        ddt        gdt        ddd      t        j                  gdg d      } | j                  t        d	
      }t        t        t        d      }t        j                  ||       | j                  t        d
      }t        t        t        t        d      }t        j                  ||       y )Nr   rQ  g      ?i  r   )r  r  r   r   r   r   )
r   r   r   r   r   r=   typer   r>   r?   r   s      r!   test_apply_typer    s    	Xu%htQ6JBFF/ST
B XXdX#Fvv67H68, XXdX#FFf=>H68,r#   c                     t        ddgddgd      }|j                  d      j                  d d|       }t        g t        j
                        }t        j                  ||       y )	Nr   rY   r   r   r2  c                 &    t        | d   | d         S )NrW   rX   )r  r~   s    r!   rQ   z/test_apply_on_empty_dataframe.<locals>.<lambda>  s    AcFAcF(; r#   r5   r   )r   headr=   r   r   r   r>   r?   r  s       r!   test_apply_on_empty_dataframer    sY    	!Qq!f-	.BWWQZ;!FSFb

+H68,r#   c                      t        ddgddgd      } | j                  d       }t        ddggddggd      }t        j                  ||       y )Nr   rY   r   r2  c                     | j                   gS rL   )r   r~   s    r!   rQ   z(test_apply_return_list.<locals>.<lambda>  s    
 r#   r   r   s      r!   test_apply_return_listr"    sS    	!Qq!f-	.BXX*+FAx1vh78H&(+r#   ztest, constant)r   r   r   )rY   rY   rY   c                     t        |       }t        |ddgd      }ddd}|j                  |      }t        j                  ||       y )NrW   rX   r  )r   rA   r   r  r2  r  )rj  constantdf1rE   aggregationrD   s         r!   test_unique_agg_type_is_seriesr'    sG     D/C8C:XFH x0KWW[!F68,r#   c                  6   t        g dg dg dd      } t        g d      }t        j                  | j	                  d      |       | j                  dd      }t        j                  ||       | j                  dd      }t        j                  ||       y )	N)r   rY   r   )r   rY   r   )r   r   r   r   )TTFr   r   r   )r   r   r>   r?   r   r=   r  s      r!   /test_any_apply_keyword_non_zero_axis_regressionr)    sz    	C	DB)*H266q6>84XXe!X$F68,XXeQF68,r#   c                     t        dg di      } dd}dd}| j                  |ddd      }| d	z   }t        j                  ||       | j                  ||gddd      }t        d
d
gddgddggddgddgg      }t        j                  ||       | j                  d|iddd      }t        g ddg      }t        j                  ||       y )NrO   r   r   c                     | |z   |z   S rL   rM   )rO   rW   r   s      r!   foo1z.test_agg_mapping_func_deprecated.<locals>.foo1      1uqyr#   c                     | |z   |z   S rL   rM   )rO   rX   r   s      r!   foo2z.test_agg_mapping_func_deprecated.<locals>.foo2  r-  r#   r   rs   )r   r   r   r   r  r,  r/  r   )rY   r   rs   )r   r   )rY   r   r   )r    r,  r/  rD   rE   s        r!    test_agg_mapping_func_deprecatedr0    s    	C#	$B VVD!Q!V$FAvH&(+VVT4L!Q!V,F
Q!Q"b"c3Z&&9I,JH &(+ VVS$KAV+FSE2H&(+r#   c                     t        t        j                  d      j                  dd      ddg      } | j	                  t        j
                  d      }t        d	d	d
t              }t        j                  ||       | j	                  t        j
                  gd      }t        d	d	d
dg      }t        j                  ||       y )Nr   r   rY   r   r   r   r   )ddofr7  r   r   stdr   )r   r   r   r   r   r3  r   r   r>   r?   rb   r   s      r!   test_agg_stdr4  '  s    	299Q<''1-Sz	BBVVBFFV#FCc*%8H68,VVRVVH1V%Fs-eW=H&(+r#   c                     t        g dg dg dgg d      } | j                  dt        j                  gi      }t        ddgidg	      }t	        j
                  ||       | j                  dt        j                  i      }t        ddi      }t	        j                  ||       | j                  dt        j                  t        j                  gi      }t        dd
dgiddg	      }t	        j
                  ||       y )Nr   r   r   r   r   r   r   r  r   r5  r6  r@   )	r   r   r   r  r>   rb   r   r?   r@   r   s      r!   test_agg_np_sizer6  3  s    	Iy)4o	NBVVS277)$%F#sF84H&(+VVS"''N#FsAhH68,VVS277BGG,-.F#Sz*662BCH&(+r#   c                      t        g ddt        j                  dgg dd      } g d| _        | j	                  ddi      }| d   j                         }t        j                  ||       y )	Nr  r  r6  r  r   )r   r   r   r   r  )r   r   r   r   r   r  r>   r?   r   s      r!   (test_agg_dist_like_and_nonunique_columnsr8  D  s\    	#rvvs!3:NO
B !BJVVS'N#F#w}}H68,r#   engine_nameunknown   c                     t        j                  t        d      5  t               j	                  d |        d d d        y # 1 sw Y   y xY w)NzUnknown engine rZ   c                     | S rL   rM   r~   s    r!   rQ   z#test_wrong_engine.<locals>.<lambda>S  r   r#   r4   )r*   rc   r  r   r=   )r9  s    r!   test_wrong_enginer>  P  s;    	z):	; ;+k:; ; ;s   AA)r   r  numpyr   r*   pandas.compatr   pandas.core.dtypes.dtypesr   pandasre   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   parametrizerh   rw   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  filterwarningsr  r"  r+  r0  r8  r:  r>  rC  rM  r\  rg  rm  rp  rw  r  r  r  r  r  r  r  r  r  r  r  r   r  r  r   r  r  r  r  r  r%  r0  r7  r?  rE  rN  rS  rZ  r^  rc  rg  rp  rt  rw  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  r  r"  r'  r)  r0  r4  r6  r8  r>  rM   r#   r!   <module>rL     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#   