
    Pib                        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j        d	             Z ej        d
 ej        dej        j                  eg          d             Zd Zej                             dd dg          ej                             dddg          ej                             dddg          d                                     Z!d Z"d Z#ej                             ddd edg          f edd          dfg          d             Z$d Z%ej                             d ej&        ej'        g          d!             Z(d" Z)d# Z*d$ Z+ej                             d g d%          d&             Z,d' Z-d( Z.d) Z/d* Z0d+ Z1d, Z2d- Z3d. Z4ej                             dd dg          d/             Z5ej                             dd dg          d0             Z6d1 Z7ej                             dd dg          d2             Z8d3 Z9d4 Z:d5 Z;ej        <                    d6          ej                             d7d8d9g          ej                             d d: d; gd<d=g>          ej                             dddg          ej                             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j                             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j                             d[d\d]g          d^             ZQd_ ZRd` ZSda ZTdb ZUej                             dcdddedejV        g          df             ZWdg ZXej                             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jY        dp gidk edig dqig drs          fdidoejY        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j                             dxdidy ididz ididoejY        d{ gidid| ig          d}             Z[d~ Z\d Z]d Z^ej                             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jY        d gdk edig dqig drs          fdoejY        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j                             dxd gd gdoejY        d gd d gg          d             Zbd Zcd Zdd Zed Zfej                             dcddUg          d             Zgd Zhej                             dg dddUgg          d             Zid Zjd Zkd Zld Zmd Znd Zod Zpej                             dd d d gg d>          d             Zqd Zrd Zsd Ztd Zud Zvd Zwd Zxd Zyd Zzej                             dd]d\g          d             Z{ej                             dd]d\g          d             Z|d Z}ej                             dd dg          ej                             ddi fdi fdi fdddUifdddUdfddUddfdddUdVdfg          d                         Z~ej                             dg dĢ          dń             ZdƄ Zej                             d edidgddgdʜ          g          ej                             dg d̢          d̈́                         Zej                             dddddiejV        g          dЄ             Zdф Zd҄ Zdӄ ZdԄ ZdՄ Zdք Zdׄ Zd؄ Zdل Zdڄ Zej                             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j                             dddg          d             ZdS )    )datetimeN)is_platform_arm)CategoricalDtype)	DataFrame
MultiIndexSeries	Timestamp
date_range)MockEngineDecorator)
zip_frames)Versionc                      t          t          j        t          j        dd          d                              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    w/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/pandas/tests/apply/test_frame_apply.pyint_frame_const_colr"      sY     

	!7+++Q//772>>B
 
 
B I    pythonnumba)marks)paramsc                 N    | 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                            d          }|j                            |           t          j        d          5  t          j        | d                   }|                     t
          j        |          d         }t          j
        ||           |                     t
          j        |          d         }t          j        | d                   }||k    sJ | j        d         }|                     t
          j        d	|
          }t          j        |                     |                    }||         |k    sJ |j        | j        u sJ 	 d d d            d S # 1 swxY w Y   d S )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   s   {  (U VV%%%		"	"	" 1 1S)**$$RWV$<<SA
vx000 ""276"::3?7;s+,,!!!!a ""276"BB7;>>!,,--ayH$$$$|{0000001 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1s   DE88E<?E<r6   r   rawTFnopythonc           	         t          j        d          }|dk    rOt          |j                  t          d          k    r*t	                      rt          j        d|j                    d|i}|                     d |d|||          }| dz   }t          j        ||           |                     d	 dd
|||          }| dz   }t          j        ||           |dk    rt          j	        t          d          5  |                     d d
|||           d d d            n# 1 swxY w Y   t          j	        t          j        j        d          5  |                     d dd
|||           d d d            n# 1 swxY w Y   t          j	        t          j        j        d          5  |                     d dd
|||           d d d            d S # 1 swxY w Y   d S d S )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9 : 	PU=NPPQQQ*M#   F QH&(+++ !!
#   F QH&(+++]9,NOOO 	 	""+    	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 ]I$A
 
 
 	 	 ,,+    		 	 	 	 	 	 	 	 	 	 	 	 	 	 	 ]I$A
 
 
 	 	 --+    		 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	3 s6   <D$$D(+D(E==FF-GGGc                      t          g dg dd          } |                     d           }|j        dk    sJ t          |d         j        t
                    sJ t          |d         j        t
                    sJ d S )N)r   r   r   r   )r   r   Drj   c0c1c                 ,    |                      d          S )Ncategoryastype)tss    r!   rQ   z-test_apply_categorical_func.<locals>.<lambda>   s    :!6!6 r#   )   rY   rl   rm   )r   r=   shape
isinstancer   r   r    rD   s     r!   test_apply_categorical_funcrw      s    	...6J6J6JKK	L	LBXX6677F<6!!!!fTl(*:;;;;;fTl(*:;;;;;;;r#   c                      t          dt          dd          gi          } |                     d d          }t          j        ||            d S )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      sT    #	,5 A A ABCDDH^^KKa^00F&(+++++r#   zdata, dtyper   Nry   rz   r{   c                     t          | g| ggddg|          }|                    d d          }t          j        ||           d S )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      sU     4&4&)#s5IIIH\\++A\..F&(+++++r#   c                      t          t          dd          t          j        t	          j        d          d          d          } |                     d d	          }t          j        ||            d S )
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      sy     J2221C888	
 	
 H ^^KKa^00F&(+++++r#   funcc                 b    t                      }|                    | |          }|j        sJ d S )Nr4   )r   r=   empty)r   r-   empty_framerD   s       r!   test_apply_emptyr      s8     ++KtF33F<<r#   c                 r   | d d         }|                     d |          }t          t          j        | j                  }t          j        ||           | j        d d g f         }|                     d d|          }t          t          j        | j                  }t          j        ||           d S )Nr   c                 *    |                                  S rL   r@   r~   s    r!   rQ   z(test_apply_float_frame.<locals>.<lambda>       QVVXX r#   r4   rA   c                 *    |                                  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    "1"oG]]--f]==FbfK$7888H68,,,oaaae$G]]--Af]EEFbfK$5666H68,,,,,r#   c                     t          dg          }|                    d d|           }t          j        ||           d S )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      sH    u%%%H^^,,1V^DDF&(+++++r#   c                  V   t                      } g }|                     |j        dd          }t          j        ||            |                     |j        dd          }t          g t          j                  }t          j        ||           t          g d          }|                    |j        dd          }t          j        ||           |                    |j        dd          }t          g t          j                  }t          j        ||           |g k    sJ d S )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qxaXFFF&+...qxaXFFFb
+++H68,,,888O""18!"JJF&/222""18!"JJFb
+++H68,,, 777777r#   )sumprodanyr3   c                    t          g d          }|                    t          t          |                     } t          ||                       }| dv r|                    t
                    }t          j        ||           d S )Nr   r   )r   r   )r   r=   getattrr   rq   floatr>   r?   )r   r    rD   rE   s       r!   test_apply_funcs_over_emptyr      s     
???	+	+	+BXXgb$''((F wr4  ""H??5))68,,,,,r#   c                  :   t          g d          } |                                 }t          d| j                  }t	          j        ||           | j                                        }t          g t          j                  }t	          j        ||           d S )Nr   r   r   r   r   )	r   nuniquer   r   r>   r?   Tr   r   r    rD   rE   s      r!   test_nunique_emptyr      s    	???	+	+	+BZZ\\Farz***H68,,,T\\^^Fb
+++H68,,,,,r#   c                  $   t          g dg dg dgg d          } |                     d d          }t          g d	g d          }t          j        ||           | j                            d
 d          }t          j        ||           d S )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     s    	IIIyyy)))4OOO	L	L	LBXXnn1X--Fiii11H68,,,TZZQZ//F68,,,,,r#   c                     |                      t          j        d          }t          |                                 g| j                  }t          j        ||           d S )N	broadcastr   r   )r=   r   r@   r   rA   r>   rb   rC   rD   rE   s      r!   test_apply_broadcast_scalarsr     sY    rwK@@F+**,,-[5FGGGH&(+++++r#   c                     |                      t          j        dd          }|                     d          }t          t                              | j        |                    }t          j        ||           d S )Nr   r   r   r   )	r=   r   r@   r   dictfromkeysr   r>   rb   rC   rD   mrE   s       r!   "test_apply_broadcast_scalars_axis1r     sk    rwQKHHFa  A{':A>>??H&(+++++r#   c                 *                           fddd          }t          t          t           j                                      }t          |gt           j                  z  d j         j                  }t          j        ||           d S )Nc                 `    t          t          t          j                                      S rL   )listrangelenr   rO   rC   s    r!   rQ   z4test_apply_broadcast_lists_columns.<locals>.<lambda>$  s"    $uS!455667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777   F
 	U3{*++,,--A	
c+#$$$#	  H &(+++++r#   c                 ,                           fdd          }t          t          t           j                                      }t          t                               j        |          d j                  }t          j
        ||           d S )Nc                 `    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33445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  s    5555;   F 	U3{())**++Ak)1--  H
 &(+++++r#   c                 f    | }|                     d dd          }t          j        ||           d S )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KKF&"%%%%%r#   c                     | }|                     d dd          }|                                }t          j        ||           d S )Nc                 B    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66 r#   r   r   r   )r=   copyr>   rb   r"   r    rD   rE   s       r!   'test_apply_broadcast_series_lambda_funcr   F  sS    	BXX66   F
 wwyyH&(+++++r#   c                 r    |dk    rt          j        d           d }|                     |||d           d S )Nr%   z)numba can't handle when UDF returns None.c                 X    t          | t          j                  sJ | j        dk    sJ d S Nr   ru   r   ndarrayndimr~   s    r!   _assert_rawz/test_apply_raw_float_frame.<locals>._assert_rawV  -    !RZ(((((v{{{{{{r#   Tr6   r-   rH   )r*   ra   r=   )rC   r6   r-   r   s       r!   test_apply_raw_float_framer   Q  sR    ?@@@   kVFFFFFr#   c                     |                      t          j        ||d          }|                      d |          }t          j        ||           d S )NTr   c                 4    | j                                         S rL   )valuesr@   r~   s    r!   rQ   z3test_apply_raw_float_frame_lambda.<locals>.<lambda>`  s    18==?? r#   r   )r=   r   r@   r>   r?   )rC   r6   r-   rD   rE   s        r!   !test_apply_raw_float_frame_lambdar   ]  sU    rwT&dKKF  !:!: FFH68,,,,,r#   c                 l    |                      d |d          }| dz  }t          j        ||           d S )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  s@    v4H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                    }|                    || |d           d S )Nr%   z(isinstance check doesn't work with numbac                 X    t          | t          j                  sJ | j        dk    sJ d S 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    >???  
 
x
)<<<XqcBhg666	
 	
 imm	
 	
 	
B HH[tFH=====r#   c                     | j         d         }|                     t          j        d          |         }t          j        |                     |                    }||k    sJ d S )Nr   r   r   )rA   r=   r   r@   rB   )rC   rF   rD   rE   s       r!   test_apply_axis1r    s`    !ArwQ//2Fw{~~a(())HXr#   c                      t          dgdgd          } | d d                             t          j        d          }t	          t          j                  }t          j        ||           d S )Nr  r  r   r   r   r   r   )r   r=   r   r@   r   r   r>   r?   r   s      r!   test_apply_mixed_dtype_cornerr    sg    	%u--	.	.BV\\"'\**F BJ'''H68,,,,,r#   c                  ^   t          dgdgd          } |                     d d          }t          dgt          d                    }t	          j        ||           |                     d d          }t          dgt          d                    }t	          j        ||           d S )	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ugU1XX...H68,,,XX&&QX//Fse588,,,H68,,,,,r#   zignore::RuntimeWarningaxrA   r   c                     | S rL   rM   r~   s    r!   rQ   rQ         q r#   c                 *    |                                  S rL   r   r~   s    r!   rQ   rQ     s    AFFHH r#   identityr@   )idsc                    t          di | g di}t          j        d          5   |t          j        g d                    }t	          |t          j                   }|                    ||||          }	|r8|                    |          }
t	          |	t                    sJ |	j	        |
u sJ nt	          |	t                     sJ d d d            d S # 1 swxY w Y   d S )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 
14400011%h
;;;$T&cBB 	1''--Hff-----<8+++++fi00000
1 
1 
1 
1 
1 
1 
1 
1 
1 
1 
1 
1 
1 
1 
1 
1 
1 
1s   B%CCCc                      t          g d          } |                     d d          }t          |t                     sJ d S )Nr   r   c                 *    |                                  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#    sK    ooo...G]]--;]GGFfi(((((((r#   c                     dd}|                      |d          }|                      d           }t          j        ||           d S )Nr   c                     | |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33F  11H&(+++++r#   c                     dd}|                      |d          }|                      d           }t          j        ||           d S )Nr   c                 0    |                                  |z   S rL   r   r&  s     r!   agg_and_addz:test_apply_with_args_kwds_agg_and_add.<locals>.agg_and_add  s    vvxx'!!r#   rY   r)  c                 0    |                                  dz   S r   r   r~   s    r!   rQ   z7test_apply_with_args_kwds_agg_and_add.<locals>.<lambda>  s    16688a< r#   r+  )r=   r>   r?   )rC   r/  rD   rE   s       r!   %test_apply_with_args_kwds_agg_and_addr1    s`    " " " " {A66F  !7!788H68,,,,,r#   c                     dd}|                      |dd          }|                      d           }t          j        ||           d S )Nr   c                     | |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   r5  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   r6  rD   rE   s       r!   -test_apply_with_args_kwds_subtract_and_divider9    sc    " " " " 2aHHF  !:!:;;H&(+++++r#   c                 d    |                      t                    }t          j        ||            d S 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                            d          f<   |                     d          }|                     t           j        d          }t          j        ||           d S )NrY   r   r   r   )	r   r   ilocr   get_locr@   r=   r>   r?   )rC   rE   rD   s      r!   test_apply_reduce_Seriesr?    sv    >@fKSSqS+-55c:::;Q''HrwQ//F68,,,,,r#   c                  z   t          ddgddggddgddg	          } |                     t          d
          }t          ddddddg| j                  }t          j        ||           |                     t          d          }t          ddddddg| j                  }t          j        ||           d S )Nr   rY   r   rs   rl   rm   i0i1r   rA   r   r   )rA  rB  r   rk   )r   r=   r   r   r   r>   r?   rA   )r   rD   rE   s      r!   test_apply_reduce_to_dictrD    s    q!fq!f%d|D$<PPPDZZ1Z%%Faq))!1+=+=>dlSSSH68,,,ZZ1Z%%Faq))!1+=+=>djQQQH68,,,,,r#   c                  8   t          t          j                            d                              d                    } |                     t          j        d          }t          d |                                 D             | j	                  }t          j        ||           |                     t          j        d          }t          d | j                                        D             | j                  j        }t          j        ||           d S )	NrY   )   r  r   r   c                 >    i | ]\  }}||                                 S rM   describe.0ivs      r!   
<dictcomp>z2test_apply_differently_indexed.<locals>.<dictcomp>  s&    AAAda!QZZ\\AAAr#   r   r   c                 >    i | ]\  }}||                                 S rM   rH  rJ  s      r!   rN  z2test_apply_differently_indexed.<locals>.<dictcomp>  s&    CCCda!QZZ\\CCCr#   )r   r   randomdefault_rngstandard_normalr=   r   rI  itemsr   r>   rb   r   rA   r   s      r!   test_apply_differently_indexedrT    s    	29((++;;HEE	F	FBXXfoAX..FAAbhhjjAAA2:VVVH&(+++XXfoAX..FCCbdjjllCCCRXVVVXH&(+++++r#   c                     t          g dg dg dg dg dg dgg d          } d	 }|                     |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          } |                     |d
          }t          j        ||           d S )N)r   ABC02   )r   YUM0rF  )r   DEF0rF  )rY   ABC1rW  )rY   YUM1rF  )rY   DEF1rF  )rW   marketpositionr   c                     | d         S )Nr]  rM   )rs    r!   fztest_apply_bug.<locals>.f  s    {r#   r   r   i  rV  rW  rY   rX  rF  r   rY  rs   rZ  r   r[  r   r\  )r   r=   r   r>   r?   )	positionsra  rE   rD   s       r!   test_apply_bugrc    sN   OOOOOOOOOOOO	
 ,++
 
 
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                            d                              d          t          j                            d                              d          t          j                            d                              d          d          } |                     d d	          }t          j        ||            d S )
N)r  r  r  r  barre  re  re  r  r  r  )onerf  rf  tworf  rf  rf  rg  rg  rg  rf  )dullrh  shinyrh  rh  ri  ri  rh  ri  ri  ri  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   rP  rQ  rR  r=   r>   rb   r   s     r!   test_apply_convert_objectsrn    s           &&q))99"==&&q))99"==&&q))99"==U+	
 +	
- -H^ ^^KKa^00F&(+++++r#   c                     |                      d           }t          | j        | j                  }t          j        ||           d S )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_namert  P  sJ    //00Fk)1DEEEH68,,,,,r#   c                     |                      d d          }t          | j        | j                  }t          j        ||           d S )Nc                     | j         S rL   rq  r~   s    r!   rQ   z.test_apply_attach_name_axis1.<locals>.<lambda>W  rs  r#   r   r   r   )r=   r   rA   r>   r?   r   s      r!   test_apply_attach_name_axis1rw  V  sO    //a88Fk'{/@AAAH68,,,,,r#   c                     |                      d           }t          t          j        | j        t          | j                  df          | j        | j                  }t          j        ||           d S )Nc                 P    t          j        | j        t          |                     S rL   r   repeatrr  r   r~   s    r!   rQ   z6test_apply_attach_name_non_reduction.<locals>.<lambda>^      163q66)B)B r#   r   rA   r   )	r=   r   r   r   r   r   rA   r>   rb   r   s      r!   $test_apply_attach_name_non_reductionr~  \  sv    BBCCF
#c+*;&<&<a%@AA#  H
 &(+++++r#   c                                            d d          }t           fd                                 D                       } j        |_        t	          j        ||           d S )Nc                 P    t          j        | j        t          |                     S rL   rz  r~   s    r!   rQ   z<test_apply_attach_name_non_reduction_axis1.<locals>.<lambda>h  r|  r#   r   r   c              3   r   K   | ]1}t          j        |d          t          j                            V  2dS )r   N)r   r{  r   r   )rK  trC   s     r!   	<genexpr>z=test_apply_attach_name_non_reduction_axis1.<locals>.<genexpr>i  sO        67	!A$K/0011     r#   )r=   r   
itertuplesrA   r>   r?   r   s   `  r!   *test_apply_attach_name_non_reduction_axis1r  g  s    BBKKF    ;F;Q;Q;S;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          }|                    d d          }t          ddgddgddgg| ddg          }t	          j        ||d           d S )N)rW   rW   rX   )r   rF   rF   r   rY   r   rs   r   r   col1col2r}  c                 Z    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!ffSVV&D&DE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    "OOO___#EFFEAq6Aq6Aq6*%&&AQRRRAWWEEqIIF1a&1a&1a&1PUWWWH&(t<<<<<<r#   z	df, dictsr  re  spameggs)r   r   rY   r   c                    d }|                      |d          }|                      |d          }|                      |          }t          j        ||           t          j        ||            t          j        ||           d S )Nc                 *    |                                  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  s     
	B((28(44K88BH855L((2,,K;...,+++;.....r#   c                  6   t          dt          ddd          i          } |                     d           }t          j        ||            |                     d           }t          dt          ddd          i          }t          j        ||           d S )	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                 0    | t          j        d          z   S )N1day)re   	Timedeltar~   s    r!   rQ   z,test_apply_non_numpy_dtype.<locals>.<lambda>  s    BL$8$8 8 r#   z
2015-01-02)r   r
   r=   r>   rb   r   s      r!   test_apply_non_numpy_dtyper    s    	D*\1ARSSST	U	UBXXkk""F&"%%%XX8899F	z,6GHHHI H &(+++++r#   c                      t          dg did          } |                     d           }t          j        ||            d S )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    sL    	D.../z	B	B	BBXXkk""F&"%%%%%r#   c                      t          ddgddggddg          } t          ddggddgdg          }|                     dg          }t          j        ||           d S )	Nr   r   rY   r   rW   r   r  rC  r   r    rE   rD   s      r!   test_apply_dup_names_multi_aggr    so    	QFQF#c3Z	8	8	8B1a&C:eWEE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        ||           d S )Nc                 B    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    : : : 
28F##T&\\	:	:	:BWR__Za000F	///???OOOL H 68,,,,,r#   c                      t          dt          dd          gid          } |                     d           }t          j        ||            |                     d           }t          j        ||            d S )	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                 *    |                                  S rL   )r   r~   s    r!   rQ   z7test_apply_noreduction_tzaware_object.<locals>.<lambda>  s    affhh r#   r   r   s     r!   %test_apply_noreduction_tzaware_objectr    s    	6e,,,-.6K  H ^^KK((F&(+++^^..//F&(+++++r#   c                      t          dg di          } g fd}fd}||fD ]8}d d = |                     |d           t          | j                  k    sJ 9d S )NrW   r   c                 <                         | j                   d S rL   r   rr  r  namess    r!   reducing_functionz8test_apply_function_runs_once.<locals>.reducing_function  s    SXr#   c                 <                         | j                   | S rL   r  r  s    r!   non_reducing_functionz<test_apply_function_runs_once.<locals>.non_reducing_function  s    SX
r#   r   r   )r   r=   r   rA   )r    r  r  r   r  s       @r!   test_apply_function_runs_oncer    s     
C#	$	$BE         #$9: ' '!!!H
ARX&&&&&	' 'r#   c                    | dk    rt          j        d           t          dg di          }g fd}fd}||fD ]L}d d = |                    || dd	           t	          |j                                                  k    sJ Md S )
Nr%   z8appending to list outside of numba func is not supportedrW   r   c                 2                         |            d S rL   extendr  r   s    r!   r  z<test_apply_raw_function_runs_once.<locals>.reducing_function  s    cr#   c                 2                         |            | 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OO	C#	$	$BF         #$9: . .111I
f$Q777bdllnn-------	. .r#   c                      t          t          j        ddg          dg          } t          t          j        ddg          dgt                    }|                     d           }t          j        ||           d S )Ns   abcds   efghcolr   )r   r   c                 ,    |                      d          S )Nobjectrp   r~   s    r!   rQ   z-test_apply_with_byte_string.<locals>.<lambda>  s     2 2 r#   )r   r   r	  r  r=   r>   rb   r  s      r!   test_apply_with_byte_stringr    sz    	28Wg.//%	A	A	AB7G"455wfUUUH XX2233F&(+++++r#   valasd   c                      dd dddt           j        g}t          d|id          }|j                             fd          }t           fd|D             d	          }t          j        ||           d S )
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#   c                 H    g | ]}t          j        |          rd n|k    S )F)re   isnull)rK  rO   r  s     r!   
<listcomp>z1test_apply_category_equalness.<locals>.<listcomp>  s.    @@@")A,,	,AH@@@r#   rq  )r   r   r   rW   r=   r   r>   r?   )r  	df_valuesr    rD   rE   s   `    r!   test_apply_category_equalnessr    s     b%7I	C#:	6	6	6BTZZ****++F@@@@i@@@s  H 68,,,,,r#   c                  P   t          t          j                            d                              d                    } |                     t          j        j        d          j        }|dk    sJ |                     t          j        j        d          j        }|dk    sJ d S )NrY   )r  rY   r   r   )r   rY   )r   r   rP  rQ  r=   fftrt   rfftrv   s     r!   test_infer_row_shaper  
  s     
29((++227;;	<	<BXXbfjqX))/FWXXbfkX**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                 *    |                                  S rL   r   r~   s    r!   rQ   rQ          r#   c                 *    |                                  S rL   r  r~   s    r!   rQ   rQ     r  r#   r   c                 *    |                                  S rL   r  r~   s    r!   rQ   rQ         AEEGG r#   )r   r   r   )r   r   <lambda>r   c                 *    |                                  S rL   r  r~   s    r!   rQ   rQ   "  r  r#   c                     dS r   rM   r~   s    r!   rQ   rQ   &       r#   c                     dS r   rM   r~   s    r!   rQ   rQ   '  r  r#   c                     t          dddgi          }|                    | |          }t          j        ||           d S NrW   r   rY   by_rowr   r=   r>   assert_equalopsr  rE   r    rD   s        r!   test_dictlike_lambdar    sF    . 
C!Q=	!	!BXXc&X))FOFH%%%%%r#   r  c                     | dz   S r   rM   r~   s    r!   rQ   rQ   4  s
    A r#   c                 *    |                                  S rL   r  r~   s    r!   rQ   rQ   5  s     r#   c                 *    |                                  S rL   r  r~   s    r!   rQ   rQ   6  s     r#   c                     dS r   rM   r~   s    r!   rQ   rQ   7       r#   c                     t          dddgi          }t          j        t          d          5  |                    | d           d d d            d S # 1 swxY w Y   d S NrW   r   rY   zby_row=True not allowedrZ   Tr  r   r*   rc   
ValueErrorr=   r  r    s     r!   test_dictlike_lambda_raisesr  1       
C!Q=	!	!B	z)B	C	C	C # #
T"""# # # # # # # # # # # # # # # # # #   AAAc                  "   t          ddgddggddg          } |                     d d          }t          d |                                 D                       }t	          j        ||           t          d	          t          d
          g| d<   |                     d d          }t	          j        ||           | d         | d         z                       d           }t          ddiddig          }t	          j        ||           d S )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   c                     g | ]}d diS )r   r   rM   rK  r  s     r!   r  z.test_with_dictlike_columns.<locals>.<listcomp>E  s    999AQx999r#   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?   r	   r   s      r!   test_with_dictlike_columnsr  A  s!   	QFQF#c3Z	8	8	8BXX66QX??F99999::H68,,, 	'(('((BtH XX66QX??F68,,, g3&&'9'9::FQx#q*++H68,,,,,r#   c                      t                      } g d| d<   g d| d<   t          j        g dd          | d<   |                     d	 d
          }t	          i i i g          }t          j        ||           d S )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  s    	B"??BxL+++B{OMMM  BvJ XXllX++Fr2rl##H68,,,,,r#   c                  X   t          ddgddggddg          } |                     d dd          }t          d	d
d
gi          }t          j        ||           t	          d          t	          d          g| d<   |                     d dd          }t          j        ||           d S )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	8	8	8BXX66QHXUUF#1v''H&(+++ 	'(('((BtH XX66QHXU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                 *    |                                  S rL   r  r~   s    r!   rQ   rQ   w      AEEGG r#   r  c                 *    |                                  S rL   r  r~   s    r!   rQ   rQ   x  r$  r#   c                 *    |                                  S rL   r  r~   s    r!   rQ   rQ   z      aeegg r#   c                 *    |                                  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                     dS Nr   rM   r~   s    r!   rQ   rQ     s     r#   r   c                     dS r   rM   r~   s    r!   rQ   rQ     r  r#   c                     dS r+  rM   r~   s    r!   rQ   rQ     r   r#   c                     t          dddgi          }|                    | |          }t          j        ||           d S r  r  r  s        r!   test_listlike_lambdar/  r  sF    > 
C!Q=	!	!BXXc&X))FOFH%%%%%r#   c            
         t          ddd          t          ddd          t          ddd          g} t          dddgi| d         | d         g          }|                    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        ||           d S )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                 .    |                      d          S Nrj   )freqshiftr~   s    r!   rQ   z7test_listlike_datetime_index_unsorted.<locals>.<lambda>  s    agg3g.?.? r#   Fr  r  r8  r  )codesr}  )r   r   r=   r   r   r   r>   rb   r   r    rD   rE   s       r!   %test_listlike_datetime_index_unsortedr:    s    tQ""HT1a$8$8(4A:N:NOF	C!Q=F1I(>	?	?	?BXX{{$?$?@XOOF
sc26]RVSM2ay&)VAY/SEJ<0!Q!Q8HII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          }|                    d	 d
 dd          }t          ddt          j        gdt          j        dgd| d         | d         | d         g          }t          j        ||           d S )Nr1  r   rY   r   rs   rW   rX   r   r   c                     | S rL   rM   r~   s    r!   rQ   z7test_dictlike_datetime_index_unsorted.<locals>.<lambda>  r  r#   c                 .    |                      d          S r4  r6  r~   s    r!   rQ   z7test_dictlike_datetime_index_unsorted.<locals>.<lambda>  s    !''s'*;*; r#   Fr  r  r8        @      @)r   r   r=   r   r   r>   rb   r9  s       r!   %test_dictlike_datetime_index_unsortedrA    s    tQ""HT1a$8$8(4A:N:NOF	!Qq!f--fQi5K	L	L	LBXXk ; ;<<U   F sBF#rvs#	
 	
 ay&)VAY/  H &(+++++r#   c                     | dz   S r   rM   r~   s    r!   rQ   rQ     
    1q5 r#   c                 *    |                                  S rL   r  r~   s    r!   rQ   rQ     s    15577 r#   c                 *    |                                  S rL   r  r~   s    r!   rQ   rQ     s    !%%'' r#   c                     | dz   S r   rM   r~   s    r!   rQ   rQ     rC  r#   c                     dS r+  rM   r~   s    r!   rQ   rQ     r   r#   c                     t          dddgi          }t          j        t          d          5  |                    | d           d d d            d S # 1 swxY w Y   d S r  r  r  s     r!   test_listlike_lambda_raisesrI    r   r  c            	      z   t          t          t          j                            d                              d                    g dt          ddd          d          } | dd	g                             t          d
          }t          d | dd	g         	                                D                       }t          j        ||           | ddg                             t          d
          }t          d | ddg         	                                D                       }t          j        ||           d S )NrY   rs   )rW   r   ofwordsz
2016-10-01h)r   r5  )rW   rX   rr   rW   rX   r   r   c                 "    g | ]}|d d         S r   rM   r  s     r!   r  z.test_with_listlike_columns.<locals>.<listcomp>  s     BBBquBBBr#   rr   c                 "    g | ]}|d d         S r   rM   r  s     r!   r  z.test_with_listlike_columns.<locals>.<listcomp>  s     CCCquCCCr#   )r   r   r   rP  rQ  rR  r
   r=   tupler  r>   r?   r   s      r!   test_with_listlike_columnsrQ    s-   		--a00@@CCDD---\13???	
 	

 
B c
^!!%a!00FBBb#sn&?&?&A&ABBBCCH68,,,d_""5q"11FCCb#to&@&@&B&BCCCDDH68,,,,,r#   c                  4   t          t          ddgdgg          t          dgddgg          d          } t          j        ddg          | _        |                     d	 d
          }t          g dgg| j                  }t          j        ||           d S )NrW   rX   qzr  rN   )rA  j0)rB  j1c                 ,      fd d         D             S )Nc                 (    g | ]}|d          v |S )rP   rM   )rK  elr  s     r!   r  zOtest_with_listlike_columns_returning_list.<locals>.<lambda>.<locals>.<listcomp>  s"    "K"K"K"B#c(NN2NNNr#   rO   rM   r  s   `r!   rQ   z;test_with_listlike_columns_returning_list.<locals>.<lambda>  s     "K"K"K"KC"K"K"K r#   r   r   r   )r   r   r   from_tuplesrA   r=   r>   r?   r   s      r!   )test_with_listlike_columns_returning_listr[    s    	#scU 3446C53PS*BU;V;VWW	X	XB%|\&BCCBHXXKKRSXTTFrC5k222H68,,,,,r#   c                     t          ddgddgt          d          t          d          gd          } |                     d d	
          }t          d |                                 D                       }t          j        ||           d S )Nr  r8  r  re  z2017-11-29 03:30:00z2017-11-29 03:45:00)numberstringr   c                     | j         | j        fS rL   r]  r^  r  s    r!   rQ   z1test_infer_output_shape_columns.<locals>.<lambda>  s    3:sz": r#   r   r   c                 *    g | ]}|j         |j        fS rM   r`  r  s     r!   r  z3test_infer_output_shape_columns.<locals>.<listcomp>  s!    EEE!(+EEEr#   r   r	   r=   r   r  r>   r?   r   s      r!   test_infer_output_shape_columnsrc    s     
Cjen/00/00	
 	
	
 	
B XX::XCCFEER]]__EEEFFH68,,,,,r#   c                     t          t          j                            d                              d          g d          } |                     d d          }t          d |                                 D                       }t          j	        ||           |                     d	 d          }t          d
 |                                 D                       }t          j	        ||           d S )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   c                     g | ]}g d S )r   rM   r  s     r!   r  z<test_infer_output_shape_listlike_columns.<locals>.<listcomp>  s    :::Qyyy:::r#   c                 
    ddgS Nr   rY   rM   r~   s    r!   rQ   z:test_infer_output_shape_listlike_columns.<locals>.<lambda>  
    A r#   c                     g | ]}d dgS r   rY   rM   r  s     r!   r  z<test_infer_output_shape_listlike_columns.<locals>.<listcomp>  s    777!1v777r#   
r   r   rP  rQ  rR  r=   r   r  r>   r?   r   s      r!   (test_infer_output_shape_listlike_columnsrm    s     

	a  0088///
 
 
B XX))X22F::"--//:::;;H68,,,XX&&QX//F77r}}77788H68,,,,,r#   c                     t          dg dit          d                    }|                     fdd          }t           fd|                                D             |j                  }t          j        ||           d S )	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   c                 8    g | ]}t          j                  S rM   rp  )rK  r  r  s     r!   r  zDtest_infer_output_shape_listlike_columns_np_func.<locals>.<listcomp>	  s!    ===rws||===r#   )r   r   r=   r   r  rA   r>   r?   )r  r    rD   rE   s   `   r!   0test_infer_output_shape_listlike_columns_np_funcrs    s     
C#4;;	7	7	7BXX....QX77F====R]]__===RXNNNH68,,,,,r#   c                  ^   t          t          d          t          d          t          d          t          d          gg dg dg dd          } d	 }|                     |d
          }t          d |                                 D                       }t          j        ||           d S )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                     dS )Nrk  rM   r~   s    r!   funzDtest_infer_output_shape_listlike_columns_with_timestamp.<locals>.fun  s    vr#   r   r   c                     g | ]}d S )rk  rM   r  s     r!   r  zKtest_infer_output_shape_listlike_columns_with_timestamp.<locals>.<listcomp>!  s    777!v777r#   rb  )r    rv  rD   rE   s       r!   7test_infer_output_shape_listlike_columns_with_timestamprx    s    	 ,'','','',''	 
	
 
	

 
B   XXcX""F77r}}77788H68,,,,,r#   lstr   c                 F    t          t          j                            d                              d          g d          }|                     fdd          }t           fd|                                D                       }t          j	        ||           d S )	NrY   )rs   r   r   r   c                     S rL   rM   )rO   ry  s    r!   rQ   z3test_consistent_coerce_for_shapes.<locals>.<lambda>-  s     r#   r   r   c                     g | ]}S rM   rM   )rK  r  ry  s     r!   r  z5test_consistent_coerce_for_shapes.<locals>.<listcomp>.  s    444qs444r#   rl  )ry  r    rD   rE   s   `   r!   !test_consistent_coerce_for_shapesr}  %  s     

	a  0088///
 
 
B XXmmmm!X,,F4444BMMOO44455H68,,,,,r#   c                    | }|                     d d          }|                     dddd          }t          j        ||           |                     d	 d          }|ddg         }t          j        ||           d S )
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*C*CDDD r#   r   r   r  r  r  r   r   c                 ,    t          ddgddg          S )Nr   rY   r  r  r   r  r~   s    r!   rQ   z'test_consistent_names.<locals>.<lambda>>  s    1vfg5F G G G r#   )r=   renamer>   rb   r   s       r!   test_consistent_namesr  2  s    	BXXDD1   F #))788 *  H &(+++XXGGaXPPF)*H&(+++++r#   c                     | }|                     d dd          }|                                }t          d          |_        t	          j        ||           d S )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_typer  C  sX     
BXX))xXHHFwwyyHQxxH&(+++++r#   c                     | }|                     d dd          }|ddg                                         }t          d          |_        t	          j        ||           d S )Nc                 
    ddgS rh  rM   r~   s    r!   rQ   z/test_result_type_shorter_list.<locals>.<lambda>R  ri  r#   r   r   r   r   r   rY   r  r   s       r!   test_result_type_shorter_listr  N  sd     
BXX&&QHXEEF3*~""$$HQxxH&(+++++r#   c                    |dk    r:t           j                            d          }|j                            |           | }|t
          u rPt          j        t          d          5  |                    d dd|	           d d d            d S # 1 swxY w Y   d S |                    d
 dd|	          }|	                                }t          j        ||           d S )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
    iii 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  sG    {  (R SS%%%	B$$$]S
 
 
 	 	 HH##!V    		 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 a[  
 
 7799
fh/////s   'BBBc                    |dk    r:t           j                            d          }|j                            |           | }g d|t
          u rRt          j        t          d          5  |                    fddd	|
           d d d            d S # 1 swxY w Y   d S |                    fddd	|
          }|	                                }t          j        ||           d S )Nr%   <numba Series constructor only support ndarrays not list datar/   r  r  r  r  rZ   c                 *    t          g d          S Nr   r   r  rO   r   s    r!   rQ   z8test_result_type_broadcast_series_func.<locals>.<lambda>  s    &'::: r#   r   r   r  c                 *    t          g d          S r  r  r  s    r!   rQ   z8test_result_type_broadcast_series_func.<locals>.<lambda>  s    fYYYg666 r#   r  r"   r-   r,   r7   r    rD   rE   r   s          @r!   &test_result_type_broadcast_series_funcr  p  sm    {  Q ! 
 
 	%%%	B'''G$$$]S
 
 
 		 		 HH::::'	    			 		 		 		 		 		 		 		 		 		 		 		 		 		 		 		 		 		 6666#	  
 
 7799
fh/////s   ,BBBc                    |dk    r:t           j                            d          }|j                            |           | }|                    d d|          }|                                }t          j        ||           d S )Nr%   r  r/   c                 2    t          g d| j                  S r  )r   rA   r~   s    r!   rQ   z0test_result_type_series_result.<locals>.<lambda>  s    yyy @ @ @ 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    s     {  Q ! 
 
 	%%%	BXX@@qQWXXXFwwyyH&(+++++r#   c                 *   |dk    r:t           j                            d          }|j                            |           | }g d|                    fdd|          }|                                }|_        t          j	        ||           d S )Nr%   z:no support in numba Series constructor for list of columnsr/   r  c                 *    t          g d          S r  r  r  s    r!   rQ   z<test_result_type_series_result_other_index.<locals>.<lambda>  s    yyy @ @ @ 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XXFwwyyHH&(+++++r#   boxc                      t          |           S rL   )r   r~   s    r!   rQ   rQ     s    tAww r#   c                      t          |           S rL   )rP  r~   s    r!   rQ   rQ     s    %(( r#   c                 .    t          j        | d          S )Nr   r   )r   r	  r~   s    r!   rQ   rQ     s    bhq6P6P6P r#   )r   rP  r	  c                 f    |}|                      fdd          }t           fd|                                D                       }t          j        ||           |                      fddd          }|dd	g                             d
dd          }t          j        ||           d S )Nc                      ddg          S rh  rM   rO   r  s    r!   rQ   z,test_consistency_for_boxed.<locals>.<lambda>      QF r#   r   r   c                 *    g | ]} d dg          S rk  rM   )rK  r  r  s     r!   r  z.test_consistency_for_boxed.<locals>.<listcomp>  s%    <<<qssAq6{{<<<r#   c                      ddg          S rh  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?   r  rb   )r  r"   r    rD   rE   s   `    r!   test_consistency_for_boxedr    s     
BXX++++!X44F<<<<BMMOO<<<==H68,,,XX++++!XJJF"C:.55AA>N>N5OOH&(+++++r#   c                    | dv rdnd}t          j        d          5  t          j        |          }t          j        |          }|                                }|                    t           j        |           }t          j        ||           |                    t           j        g|           }|                                }| dv r"t          j	        |j
        dgg          |_
        n!t          j	        |j        dgg          |_        t          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            d S # 1 swxY w Y   d S )	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 0{##%% ;;==""27"66
fh/// ""BG94"88;;==<)68KfX7VWWH'4k6G&5RSSHN
fh///
 ""BFBG#44"@@ufoJ???<)6$z6&:;   H (4"Z$89 HN 	fh///?0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0s   F#GGGc                      t          t          d          dd          } |                     ddg          }t          ddgddgdddgddg	          }t          j        ||           d S )
Nr   r   r  r  r   rs   r   r   rC  )r   r   r   r>   rb   r   s      r!   	test_demor    s    	q**	+	+BVVUEN##F!fAq6""S#Juen  H &(+++++r#   c                  2   t          t          d          dd          } |                     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        |                    |          |           d S )Nr   r   r  r  r   r?  r        @g      9@r   r   )r  r  r   rC  )r   r   r   r   r   r>   rb   reindex_liker   s      r!   test_demo_dict_aggr    s    	q**	+	+BVV5%.u~>>??FC RVT':;;c
###  H
 &--h77BBBBBr#   c                  4   dddgi} t          |           }|                    ddi          }t          ddi          }t          j        ||           |d                             ddi          }t          ddid          }t          j        ||           d S )Nrr  r  re  countrY   rq  )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                  p   t          g dg dg dd          } t          ddgddgd	d
gdddg          }|                     ddg          }t          j        ||           | g d                             ddg          }|g d                             ddg          }t          j        ||           d S )Nr   r  r8  r@  r  re  bazr   r   r   r        @re  	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55F (00%@@H&(+++++r#   c            	         t          g dg dg dt          dd          d          } d}t          j        t          |	          5  |                     d
dg           d d d            n# 1 swxY w Y   t          j        t          |	          5  | g d                             dd
g           d d d            d S # 1 swxY w Y   d S )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  ,  sh   
 &&&J222		
 	
 C 'C	y	,	,	,                                  
y	,	,	, 6 6   !%%uen5556 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6s$   A++A/2A/ B>>CCc                    | dv rdnd}|j         |                                         d d                                         \  }}t          j        |                    |           |                    |           |                    |           gd          }g d|_        | dv r|j	        n|}|
                    g d|           }t          j        ||           |d|di}|
                    ||           }t          |                    |          |                                         |                    |          |                                         g||g	          }t          j        ||           |dg|dgi}|
                    ||           }t!          |t          |                    |          |                                         gdg	          |t          |                    |          |                                         gdg	          i          }| d
v r|j	        n|}t          j        ||           |ddg|ddgi}|
                    ||           }t          j        |t          |                    |          |                                         |                    |          |                                         gddg	          |t          |                    |          |                                         |                    |          |                                         gddg	          id          }| d
v r|j	        n|}t          j        ||           d S )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8!<HHJJLE5 y$''OOO&&OOO&&	

   H .--H!\11xzzxH__333$_??F&(+++ 65%(D__T_--FOOJ''.3355OOJ''.2244	
 en  H 68,,, F8UUG,D__T_--F6;??:66u=BBDDEfXVVV6;??:66u=AACCDUGTTT	
 H "^33xzzH&(+++ FE?EE5>:D__T_--Fy6OOJ//6;;==OOJ//6::<< uo   6OOJ//6::<<OOJ//6::<< en  	
  #  H& "^33xzzH&(+++++r#   c                 *   | j         d                                         d d                                         \  }}d}dD ]Q}t          j        t
          |          5  |                     |df|df|           d d d            n# 1 swxY w Y   Rd S )	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??AALE5
;C Q Q].c::: 	Q 	QOO%uen4OPPP	Q 	Q 	Q 	Q 	Q 	Q 	Q 	Q 	Q 	Q 	Q 	Q 	Q 	Q 	QQ Qs   BB	B	c            	      ^   t          g dg dg dt          dd          d          } |                     d          }t          d	d
dt	          d          g| j                  }t          j        ||           |                     dg          }t          d	d
dt	          d          ggdg| j                  }t          j        ||           d}t          j
        t          |          5  |                     d           d d d            n# 1 swxY w Y   | g d                             d          }t          g dg d          }t          j        ||           d}t          j
        t          |          5  |                     dg           d d d            d S # 1 swxY w Y   d S )Nr   r  r  r   r   r   r  r  r   r  re  r   r}  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    sA   	 &&&J222		
 	

 
B VVE]]Fq#ui
&;&;<BJOOOH68,,,VVUG__F
S%:..	/0g
  H
 &(+++
&C	y	,	,	,  
u                $$U++F+++???CCCH68,,,
&C	y	,	,	,  
w                 s$   -DDD>F""F&)F&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          }|                                }t          j        ||           d S )NNrY   r   r  r@  r  Nre  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    s    
ll#rvs!3:N:N:NOO
 
B
 WRsGn--FsAhH68,,, WRsFm,,FsAhH68,,, gb#011Ggb#w&7w>OPP G a((a((a((	
 	
 H '7t<<<<'8==== WRg&&FxxzzH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	        ||           d S )Nr  r  r@  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     
ll#rvs!3:N:N:NOO
 
B
 WRf4000F|"(1+RZ888"(1+RX66668,,,,,r#   c                  h   t          g ddt          j        dgg dd          } d }|                     |          }t	          ddgdgdd	ggg d
          }t          j        ||           |                     |g          }|                    d          j        }t          j	        ||           d S )N)rY   rY   r   g      ?r  r   c                 h    t          |                                                                           S rL   )r   dropnar  )	group_cols    r!   r   z&test_agg_listlike_result.<locals>.func  s(    I$$&&--//000r#   rY   r   r  re  r   r   )
r   r   r   r   r   r>   r?   to_framer   rb   )r    r   rD   rE   s       r!   test_agg_listlike_resultr    s    	#rvs);BVBVBVWW	X	XB1 1 1 VVD\\F1vuuen5___MMMH68,,,VVTF^^F  ((*H&(+++++r#   zargs, kwargs)r   r      rk  rR   rX   rM   r<  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|R i |}t          j        ||           d S )Nr   c                 <    |                                  ||z   |z  z   S rL   r  )rO   rW   rX   r   s       r!   ra  ztest_agg_args_kwargs.<locals>.f  s    uuww!a%1$$r#   r   rY   rs   r   r  g      @r?  g       @)r   r   r   r   r>   r?   )r6   rS   kwargsra  r    rE   rD   s          r!   test_agg_args_kwargsr    s    % % % % 
QFQF#	$	$Bqyy3*%%3*%%RVAt-d---f--F68,,,,,r#   num_cols)rY   r   r   c                 l   |dk    r:t           j                            d          }|j                            |           t          j        g d          }|                                }t          d|t          |                     }|
                    d |           |j        |j        k    sJ d S )	Nr%   z*numba engine only supports numeric indicesr/   )z
1950-06-30z
1952-10-24z
1953-05-29r   r}  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=   r5  )r  r-   r,   r7   rA   originalr    s          r!   test_frequency_is_originalr    s     {  (T UU%%%GGGHHEzz||H	1E5??	;	;	;BHH[[H(((:&&&&&&r#   c                    | dk    r:t           j                            d          }|j                            |           t          dd          t          dd          t          dd          g}t          g d	|
          }|                    d d|           }t          ||          }t          j
        ||           d S )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   rq  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
 
				4	4	4BXX&&QvX>>FJZ888H68,,,,,r#   r    r   rF   r   method)r  r  r   c                     |dk    rd}nd}|sNt          j        t          |          5   t          | |                       d d d            d S # 1 swxY w Y   d S  t          | |                       d S )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00 	" 	"GB!!!	" 	" 	" 	" 	" 	" 	" 	" 	" 	" 	" 	" 	" 	" 	" 	" 	" 	" 	Fs   AAAr  r  c                     t          d| ggddg          }|                    d           }|j        }t          j        ||           d S )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  sU     
S#J<#s	4	4	4BXX''((FyH68,,,,,r#   c                  X   t          t          d          t          dd          d          } |                                 }d }|                                 }|dxx         dz  cc<   |                     |d          }t	          j        ||           t	          j        | |           d S )Nr  rF  r<  c                 T    | 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[  s9    hxs""""
r#   rW   r   r   )r   r   r   r=   r>   rb   )r    df_origr   rE   rD   s        r!   test_apply_mutatingr  V  s    	rr277	8	8BggiiG   wwyyHSMMMQMMM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          } |                     d d          }t          g g dt                    }t	          j        ||           d S )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   r<  r   )r   r=   r   r  r>   r?   r   s      r!   test_apply_empty_list_reducer  j  s    	QFQFQFQFQG<sCj	Q	Q	QBXXllX99FBR((777H68,,,,,r#   c                 N   | dk    r:t           j                            d          }|j                            |           t          ddggdz  ddg	          }|                    d
d d g|           }t          g dg ddg d          }t          j        ||           d S )Nr%   z;numba engine doesn't support list-likes/dict-like callablesr/   rs   r   r   r   r   r   r   c                 *    |                                  S rL   r  r~   s    r!   rQ   z,test_apply_no_suffix_index.<locals>.<lambda>{  s     r#   c                 *    |                                  S rL   r  r~   s    r!   rQ   z,test_apply_no_suffix_index.<locals>.<lambda>{  s    AEEGG 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*
5
5
5CYY002C2CDVYTTFll..6U6U6U  H &(+++++r#   c                     | dk    rt          j        d           t          dddgi          }|                    d | dd	          }t	          ddg          }t          j        ||           d S )
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    sy    6777	C$'	(	(BXXnnV!XFFFtUm$$H68,,,,,r#   c                      t          g dd          } d }d|ddg}|                     |          }t          g dg d	g d
dg d          }t          j        ||           d S )N))r   r   r   )rY   r   r   )r   r   r   )rs   r   rs   )r   r   r   )r   r   r   )att1att2att3r   c                 0    |                                  dz  S r   r  r   s    r!   sum_div2z4test_aggregation_func_column_order.<locals>.sum_div2  s    uuww{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#    s     
	
 	
 	
 )

 

 

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

 211  H &(+++++r#   c                 @   | dk    r:t           j                            d          }|j                            |           t          g dg dd          }|ddg                             d d	| 
          }t          g d          }t          j	        ||           d S )Nr%   z2numba engine not supporting duplicate index valuesr/   r  r   r<  rW   c                 8    | j         d         | j         d         z   S r_   )r=  r~   s    r!   rQ   z+test_apply_getitem_axis_1.<locals>.<lambda>  s    !&)afQi' 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3	4	4Bc
^!!''a "  F iii  H68,,,,,r#   c                      d } t          dg di          }t          j        t          d          5  |                    | g           d d d            d S # 1 swxY w Y   d S )Nc                 R    t          j        d           |                                 S )NHello, World!)warningswarnr   r~   s    r!   expected_warningzDtest_nuisance_depr_passes_through_warnings.<locals>.expected_warning  s    o&&&uuwwr#   rW   r   r)  rZ   )r   r>   assert_produces_warningUserWarningr   )r,  r    s     r!   *test_nuisance_depr_passes_through_warningsr/    s       
C#	$	$B		#K	G	G	G # #
 !"""# # # # # # # # # # # # # # # # # #s   AAAc            	         t          ddt          gdt          ddd          t          j        gdg d          } |                     t          d	
          }t          t          t          d          }t          j	        ||           |                     t          d
          }t          t          t          t          d          }t          j	        ||           d S )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_typer2    s    	Xu%htQ6J6JBF/STToo
 
 
B XXdX##Fvv6677H68,,, XXdX##FFf==>>H68,,,,,r#   c                     t          ddgddgd          }|                    d                              d d|           }t          g t          j                  }t          j        ||           d S )	Nr   rY   r   r   r<  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_dataframer6    st    	!Qq!f--	.	.BWWQZZ;;!FSSFb
+++H68,,,,,r#   c                      t          ddgddgd          } |                     d           }t          ddggddggd          }t          j        ||           d S )Nr   rY   r   r<  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_listr9    sn    	!Qq!f--	.	.BXX**++FAx1vh7788H&(+++++r#   ztest, constant)r   r   r   )rY   rY   rY   c                     t          |           }t          |ddgd          }ddd}|                    |          }t          j        ||           d S )NrW   rX   r  )r   rA   r   r  r<  r  )r  constantdf1rE   aggregationrD   s         r!   test_unique_agg_type_is_seriesr>    s^     D//C8C:XFFFH x00KWW[!!F68,,,,,r#   c                  Z   t          g dg dg dd          } t          g d          }t          j        |                     d          |           |                     dd          }t          j        ||           |                     dd          }t          j        ||           d S )	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@    s    	CC	D	DB)))**H266q6>>8444XXe!X$$F68,,,XXeQF68,,,,,r#   c                     t          dg di          } dd}dd}|                     |ddd	
          }| dz   }t          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        ||           |                     d|iddd	
          }t          g ddg          }t          j        ||           d S )NrO   r   r   r   c                     | |z   |z   S rL   rM   )rO   rW   r   s      r!   foo1z.test_agg_mapping_func_deprecated.<locals>.foo1      1uqyr#   rY   c                     | |z   |z   S rL   rM   )rO   rX   r   s      r!   foo2z.test_agg_mapping_func_deprecated.<locals>.foo2  rD  r#   r   rs   )r   r   r   r   r  rC  rF  r   )rY   r   rs   )r   r   )rY   r   r   )r    rC  rF  rD   rE   s        r!    test_agg_mapping_func_deprecatedrG    s3   	C#	$	$B       VVD!Q!V$$FAvH&(+++VVT4L!Q!V,,F
Q!Q"b"c3Z&&9I,J  H &(+++ VVS$KAV++FSE222H&(+++++r#   c                     t          t          j        d                              dd          ddg          } |                     t          j        d          }t          d	d	d
t                    }t          j	        ||           |                     t          j        gd          }t          d	d	d
dg          }t          j
        ||           d S )Nr   r   rY   r   r   r   r   )ddofr8  r   r   stdr   )r   r   r   r   r   rJ  r   r   r>   r?   rb   r   s      r!   test_agg_stdrK  '  s    	29Q<<''1--Sz	B	B	BBVVBFV##FCc**%888H68,,,VVRVH1V%%Fs--eW===H&(+++++r#   c                     t          g dg dg dgg d          } |                     dt          j        gi          }t          ddgidg	          }t	          j        ||           |                     dt          j        i          }t          ddi          }t	          j        ||           |                     dt          j        t          j        gi          }t          dd
dgiddg	          }t	          j        ||           d S )Nr   r   r   r   r   r   r   r  r   r?  r@  r@   )	r   r   r   r  r>   rb   r   r?   r@   r   s      r!   test_agg_np_sizerM  3  s   	IIIyyy)))4ooo	N	N	NBVVS27)$%%F#sF8444H&(+++VVS"'N##FsAhH68,,,VVS27BG,-..F#Sz*662BCCCH&(+++++r#   c                      t          g ddt          j        dgg dd          } g d| _        |                     ddi          }| d                                         }t          j        ||           d S )	Nr  r  r@  r  r   )r   r   r   r   r  )r   r   r   r   r   r  r>   r?   r   s      r!   (test_agg_dist_like_and_nonunique_columnsrO  D  s~    	ll#rvs!3:N:N:NOO
 
B !BJVVS'N##F#w}}H68,,,,,r#   engine_nameunknown   c                     t          j        t          d          5  t                                          d |            d d d            d S # 1 swxY w Y   d S )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=   )rP  s    r!   test_wrong_enginerU  P  s    	z):	;	;	; ; ;++k:::; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ;s   %A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,  r1  r9  r;  r?  rD  rT  rc  rn  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:  rA  rI  rQ  r[  rc  rm  rs  rx  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/  r2  r6  r9  r>  r@  rG  rK  rM  rO  rU  rM   r#   r!   <module>rc     s               ) ) ) ) ) ) 6 6 6 6 6 6                        ; ; ; ; ; ; 0 0 0 0 0 0 ' ' ' ' ' ' 
 
 
 WFK$:;;;    1 1 1, !Q((u..dE]33D D 43 /. )(DN< < <, , , $$aS))*YY|-N-N-NPT,UV , ,	 ,
, 
, 
, "'27!344  54	- 	- 	-, , ,  , !>!>!>??- - @?-
- 
- 
-- - -, , ,, , ,, , ,"
, 
, 
,& & &, , , !Q((G G )(G !Q((- - )(-, , , !Q((> > )(>,  - - -- - - 455344
[[,,-J3G    u..!Q((1 1 )( /.  54 651 ) ) ), , ,- - -, , ,/ / /
- - -
- 
- 
-	, 	, 	,- - -D1, 1, 1,h- - -- - -, , ,- - -= = =  Iu~'7899F&))u+@+@ABB	
 
QFQF#	$	$ff!llaLL-I&J&JK	 		/ 	/	 		/
, 
, 
,& & &, , , /00
- 
- 10
-, , ,' ' '(. . .,, , , D"& 9::
- 
- ;:
-$   
	99cAq6]+C+CD
			3A-(@(@A
  	!8VVS!H-=-=>
  	!5&&#q*:*:;5"&"3"345IsIII&.H.H.HIII	
 5"&"3"345IsIII&.H.H.HIII	

 {{	Xyy#1v'?'?@
{{	UFFC8$4$45! *& &+ *& 		oo	 	ubf//01	kk	 # # #- - -(- - -, , , 
/	Hii1BQF0K&L&LM
/	E99.?!Q-H#I#IJ

	h		3*ZL(Q(Q(QR

	eYYaSz*%N%N%NOBF--.IsIII&.H.H.HIII	
 BF--.IsIII&.H.H.HIII	
 _kk*I1v1v&#sj*=U0VWWW	
 [++&IsQFmj*%=>>	
) :& &; :&
, 
, 
,, , ," 					))*	++&	 # # #- - -&- - -- - -$- - -  A''- - ('-- - -0 QF 344	- 	- 54	-, , ,", , ,, , ,0 0 000 0 0B, , ,, , ,$ 	**,P,PQ"""   

, 
, 

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