o
    _~#gK                     @   sX  d dl mZm Z  d dlZd dlZd dlZd dlmZmZm	Z	m
Z
mZ d dlmZ d dlmZ dd Zdd Zd	d
 Zdd Zdd Zejjdeg dg ddg dfeg dg ddd dgfeddgidgfeg deddddgfeg dg ddg dfeedg dg d d!g d"feg dg d#gg d"d$ddgfgg d%d&d'd( Zd)d* Zd+d, Zejd-d.d/ d0d/ d1d/ d2d/ gd3d4 Zd5d6 Zd7d8 Zd9d: Z d;d< Z!d=d> Z"d?d@ Z#dAdB Z$dCdD Z%dEdF Z&dGdH Z'dIdJ Z(ejdKdLdMgdNdO Z)dPdQ Z*dRdS Z+dTdU Z,dVdW Z-dXdY Z.ejdZdLdMgd[d\ Z/d]d^ Z0d_d` Z1dadb Z2dcdd Z3dedf Z4dgdh Z5didj Z6dkdl Z7dmdn Z8dodp Z9dqdr Z:dsdt Z;ejdue < e <  e < = gdvdw Z>dxdy Z?ejdzdMdLgd{d| Z@d}d~ ZAejdeBedeCd dejDdddde	Eg dgdd ZFejddd/ d dgddggfdd/ d dhddhgfdd/ ddgfdd/ d dddddgfdd/ d d iddigd diddiggfgdd ZGdd ZHejddd/ dd/ gdd ZIdd ZJejddd/ dd/ gejdKdLdMgdd ZKdd ZLdd ZMdd ZNdd ZOdd ZPejddLdMgdd ZQejddMeg dg dgeg deRdd$gdLe
ddge	jEddgddgddggdd ZSdd ZTdd ZUdd ZVdd ZWejddLdMgdd ZXddĄ ZYejdg dg dƢgddȄ ZZejdg dg dʢgg dg dˢgg d̢g d͢ggddτ Z[ejddLgi fg ddLifgddӄ Z\ejdddgddׄ Z]ejdddd/ fddd/ fdddgdd Z^ejddLdMgdd Z_ejde`eaebgejddddggdd ZcdS )    )datedatetimeN)	DataFrameIndex
MultiIndexSeriesbdate_range)get_groupby_method_argsc                     s   t dttdd dgd } g   fdd}d}tjt|d | d	| W d    n1 s4w   Y  t dgd dgd d
t	
dddd}t d | d S )N   
   r   )indexcolumnsc                    s     |  d S N)appendgroupgroups /var/www/static.ux5.de/https/Moving-Object-Detection-with-OpenCV/env/lib/python3.10/site-packages/pandas/tests/groupby/test_apply.pystore      zFtest_apply_func_that_appends_group_to_list_without_copy.<locals>.store7DataFrameGroupBy.apply operated on the grouping columnsmatchr   )r   r   d   r   )r   listrangereset_indextmassert_produces_warningDeprecationWarninggroupbyapplypd
RangeIndexassert_frame_equal)dfr   msgexpected_valuer   r   r   7test_apply_func_that_appends_group_to_list_without_copy   s     r+   c                 C   sd   g d}t dg ditt|ddd}||jj }||jjdd }t	
|| d S )	Nz2011-05-16 00:00z2011-05-16 01:00z2011-05-16 02:00z2011-05-16 03:00z2011-05-17 02:00z2011-05-17 03:00z2011-05-17 04:00z2011-05-17 05:00z2011-05-18 02:00z2011-05-18 03:00z2011-05-18 04:00z2011-05-18 05:00value	/?ݓ?Q?j?r/   r0   r1   r2   r/   r0   r1   r2   	date_timenamer   c                 S      |   S r   idxmaxxr   r   r   <lambda>L       z'test_apply_index_date.<locals>.<lambda>)r   r   r%   to_datetimer#   r   r   r8   r$   r    r'   )using_infer_stringtsr(   expectedresultr   r   r   test_apply_index_date(   s   rB   c                 C   s   g d}t dd |D ddgd}g d|d< | rd	nt}tg d
|dd}tg d|d}d}tjt|d |jddddd }W d    n1 sNw   Y  t	|| d S )Nr,   c                 S   s   g | ]}|  qS r   )split).0rowr   r   r   
<listcomp>a       z0test_apply_index_date_object.<locals>.<listcomp>r   timer   r.   r-   zstring[pyarrow_numpy])z
2011-05-16z
2011-05-17z
2011-05-18dtyper5   )z00:0002:00rL   r   r   r   F
group_keysc                 S   s   | d | d    S )NrH   r-   r7   r9   r   r   r   r;   x   rG   z.test_apply_index_date_object.<locals>.<lambda>)
r   objectr   r   r    r!   r"   r#   r$   assert_series_equal)r>   r?   r(   rK   exp_idxr@   r)   rA   r   r   r   test_apply_index_date_objectP   s   
rR   c                    s   t g dg ddddgd | rdnd}tj jd	d   jd	d  gd	d
|gd}d}tjt|d  jdd  jD d	d}W d    n1 sLw   Y  |	 fdd}t
|| d S )NarT   brU   rT         ?       @      @      @g      @keydatar\   r]   rI   stringrO   r
   float64axiskeys+DataFrame.groupby with axis=1 is deprecatedr   c                 S      g | ]}t |qS r   strrD   r:   r   r   r   rF      rG   z&test_apply_trivial.<locals>.<listcomp>ra   c                    s    j dd  S Nr
   ilocr9   r(   r   r   r;          z$test_apply_trivial.<locals>.<lambda>)r   r%   concatrk   r    r!   FutureWarningr#   dtypesr$   r'   r>   rK   r@   r)   gbrA   r   rl   r   test_apply_trivial}   s   ,rs   c                    s   t g dg ddddgd | rdnd}tj  gd	d
|gd}d}tjt|d  jdd  jD d	dd}W d    n1 sCw   Y  | fdd}t	|| d S )NrS   rV   r[   r\   r]   rI   r^   rO   r
   r_   r`   rc   r   c                 S   rd   r   re   rg   r   r   r   rF      rG   z+test_apply_trivial_fail.<locals>.<listcomp>T)ra   rN   c                    s    S r   r   r9   rl   r   r   r;          z)test_apply_trivial_fail.<locals>.<lambda>)
r   r%   rn   r    r!   ro   r#   rp   r$   r'   rq   r   rl   r   test_apply_trivial_fail   s   ru   zdf, group_names)r
   r
   r
         )rT   rT   rT   rU   crT   rU   r
   rv   rw   r   r   r
   r
   )r   r
   r   r
   r
   rT   )r
   r
   r
   rv   rv   r
   r
   rv      rv   r
   rv   rw   r
   rv   rw   )            r|   	   )rT   two
aaabbbcccc)
rw   r~   rw   r   r   rv   r
   r   r   r~   )
r~   r   rv   rv   rv   r   r|   r   rv   r|   )rT   BCrT   rU   rx   )rv   rv   rw   rI   )GH2936zGH7739 & GH10519GH10519GH2656GH12155GH20084GH21417)idsc           	   	      s   g   fdd} fdd} fdd} fdd} fd	d
}|||||fD ]0} d d = d}t jt|d | jddd| W d    n1 sLw   Y   |ksWJ q'd S )Nc                    s     | j |  S r   )r   r5   copyr   namesr   r   f_copy   s   z/test_group_apply_once_per_group.<locals>.f_copyc                    s     | j | S r   r   r5   r   r   r   r   f_nocopy      z1test_group_apply_once_per_group.<locals>.f_nocopyc                    s     | j dS Nr   r   r   r   r   r   f_scalar   r   z1test_group_apply_once_per_group.<locals>.f_scalarc                    s     | j d S r   r   r   r   r   r   f_none   s   z/test_group_apply_once_per_group.<locals>.f_nonec                    s     | j tdgdgdS )Nr
   ry   )r   r5   r   r   r   r   r   f_constant_df   s   z6test_group_apply_once_per_group.<locals>.f_constant_dfr   r   rT   FrM   )r    r!   r"   r#   r$   )	r(   group_namesr   r   r   r   r   funcr)   r   r   r   test_group_apply_once_per_group   s   '
r   c                 C   s   d}t g dg ddg dd}d}tjt|d |jdd	d
dd  W d    n1 s1w   Y  |  jd}||ksDJ d S )Nrv   )r   r   r   r   r
   r
   r
   r
   )02468101214)group_by_columntest_columnr   r   r   r   FrM   c                 S   s   t dS )Nfunction_called)printrl   r   r   r   r;      r<   z2test_group_apply_once_per_group2.<locals>.<lambda>r   )	r   r    r!   r"   r#   r$   
readouterroutcount)capsysr@   r(   r)   rA   r   r   r    test_group_apply_once_per_group2   s   r   c                  C   s   t g dtdd} dd }dd }d}tjt|d	 | jd
dd|}W d    n1 s1w   Y  tjt|d	 | jd
dd|}W d    n1 sRw   Y  t|| d S )N)r   r   r
   rw   )ArU   c                 S      | S r   r   r   r   r   r   slow     z,test_apply_fast_slow_identical.<locals>.slowc                 S   r6   r   r   r   r   r   r   fast  s   z,test_apply_fast_slow_identical.<locals>.fastr   r   r   FrM   )r   r   r    r!   r"   r#   r$   r'   )r(   r   r   r)   fast_dfslow_dfr   r   r   test_apply_fast_slow_identical  s   r   r   c                 C   r   r   r   r9   r   r   r   r;     rt   r;   c                 C   s   | d d  S r   r   r9   r   r   r   r;         c                 C      | j ddS )NFdeepr   r9   r   r   r   r;     r   c                 C   r   )NTr   r   r9   r   r   r   r;      r   c                 C   sr   t g dg dg dd}d}tjt|d |jddd	| }W d    n1 s,w   Y  t|| d S )
Nr
   rv   rv   rv   r
   rv   rw   r~   )r   r   r   r|   )grT   rU   r   r   r   FrM   r   r    r!   r"   r#   r$   r'   )r   r(   r)   rA   r   r   r   5test_groupby_apply_identity_maybecopy_index_identical  s   r   c                  C   s   t tjddg dd} | jdd ddj}| j}t|| t d	g d
i} | j	d | d< | 
d  j}| j
ddd j}t|| d S )Nrv   r   oner   r   threer   r   foo1foo2c                 S   r   r   r   r9   r   r   r   r;   9  rt   z-test_apply_with_mixed_dtype.<locals>.<lambda>r
   rh   c1)r
   rv   r   r   r|   rX   c2Fas_index)r   nprandomdefault_rngstandard_normalr$   rp   r    rP   r   r#   meanr   r   )r(   rA   r@   result1result2r   r   r   test_apply_with_mixed_dtype1  s   r   c                  C   s  t g dg dtdd} | jddd}| jddd}|d	j}|d	j}tg d
}t|| t|| d}tjt	|d |
dd j}W d    n1 sVw   Y  tjt	|d |
dd j}W d    n1 suw   Y  tg d}	g d}
tj|
dd gd}t|| t||	 ttd}t dd	gd	dgddgddgd	dgg|d} d}tjt	|d | jdddd
dd j}W d    n1 sw   Y  t|| d S )N)rU   rU   rT   rx   rT   rU   )r
   rv   r
   r
   rw   r
   r   )item_iduser_idrH   r   Tr   Frv   )r   r
   rv   r~   r   r   c                 S   
   |  dS Nrv   headr9   r   r   r   r;   Z     
 z-test_groupby_as_index_apply.<locals>.<lambda>c                 S   r   r   r   r9   r   r   r   r;   \  r   ))r   r   )r   rv   r
   r
   rv   r~   )r
   r   r
   rv   )rv   r
   )rw   r~   r   abcder
   rw   r~   r   r   r   )r   rN   c                 S   r   r   r   r9   r   r   r   r;   k  rt   )r   r   r#   r   r   r   r    assert_index_equalr!   r"   r$   r   from_tuplesr   )r(   g_asg_not_asres_as
res_not_asexpr)   res_as_applyres_not_as_applyexp_not_as_applytpexp_as_applyindresr   r   r   test_groupby_as_index_applyE  s@   *r   c           	      C   s  |  ddg}dd }dd }dd }d	}tjt|d
 ||}W d    n1 s,w   Y  |jjdks9J d	}tjt|d
 ||}W d    n1 sRw   Y  |jjdks_J d	}tjt|d
 ||}W d    n1 sxw   Y  |jjdksJ d S )Nr   r   c                 S   s   |   }d|j_|S Nstat)describer   r5   r   rA   r   r   r   descr  s   z.test_apply_concat_preserve_names.<locals>.descc                 S   s$   |   }d|j_|d t|  }|S r   )r   r   r5   lenr   r   r   r   desc2w  s   z/test_apply_concat_preserve_names.<locals>.desc2c                 S   s0   |   }dt| d|j_|d t|  }|S )Nstat_d)r   r   r   r5   r   r   r   r   desc3~  s   z/test_apply_concat_preserve_names.<locals>.desc3r   r   )r   r   r   )r   r   N)r#   r    r!   r"   r$   r   r   )	three_groupgroupedr   r   r   r)   rA   r   result3r   r   r    test_apply_concat_preserve_nameso  s&   
r   c                  C   sz   dd } t ddd}ttjdd|d}|jdd	 d
d}|| }t|t	s,J t
|dr3J t|j|j d S )Nc                 S   sN   t jdd t | }W d    n1 sw   Y  t| | |   |dS )Nignore)invalid)r-   demeanedlogged)r   errstatelogr   r   )piecer   r   r   r   f  s   z%test_apply_series_to_frame.<locals>.fz1/1/2000r   )periodsrv   r   c                 S      | j S r   monthr9   r   r   r   r;         z,test_apply_series_to_frame.<locals>.<lambda>FrM   r5   )r   r   r   r   r   r   r#   r$   
isinstancer   hasattrr    r   r   )r  drr?   r   rA   r   r   r   test_apply_series_to_frame  s   
r  c                 C   s4   |  ddgd t}|jjd d dksJ d S )Nr   r   r   rv   r   r   )r#   r$   r   r   r   )r(   rA   r   r   r    test_apply_series_yield_constant  s   r  c                 C   s   d}t jt|d | ddgt}W d    n1 sw   Y  t|ts*J |jd u s1J | ddgddg t}t|tsFJ |jd u sMJ d S )Nr   r   r   r   r   D)	r    r!   r"   r#   r$   r   r  r   r5   )r(   r)   rA   r   r   r   test_apply_frame_yield_constant  s   r  c                 C   sz   |  ddg}d}tjt|d |t}W d    n1 s w   Y  | d }t|j|j t	|j
|j
 d S )Nr   r   r   r   r   )r#   r    r!   r"   r$   r   r   r   r   assert_numpy_array_equalvaluesr(   r   r)   rA   r@   r   r   r   test_apply_frame_to_series  s   r  c                 C   s   | j ddgdd}d}tjt|d |t}W d    n1 s"w   Y  | jdtj	idj
d	d}t|j|j t|j|j d S )
Nr   r   Fr   r   r   r   rI   r  )r#   r    r!   r"   r$   r   r   renamer   nandropr   r   r  r  r  r   r   r   )test_apply_frame_not_as_index_column_name  s   r  c                     s   dd }  fdd}t tjddddtjddddtjddd	 d
}tjt|d  	d
| }W d    n1 sGw   Y   	dd 
|}tj||dd |jdkseJ d S )Nc                 S   s    |  dd   jd d S )Nr   r   rv   )r#   sumsort_valuesrk   r   r   r   r   trans  s    z-test_apply_frame_concat_series.<locals>.transc                    s,   |   | jd }|  jd d S )Nr   rv   )r#   reindexr   r  r  rk   )r   r   rl   r   r   trans2  s   z.test_apply_frame_concat_series.<locals>.trans2rv   r   r     r   r   r   r   r   r   r   F)check_names)r   r   r   r   integersr   r    r!   r"   r#   r$   rP   r5   )r  r  r)   rA   r   r   rl   r   test_apply_frame_concat_series  s   r!  c                 C   s>   | j dd dd}|dd }|dd }t|| d S )Nc                 S   r  r   r  r9   r   r   r   r;     r  z&test_apply_transform.<locals>.<lambda>FrM   c                 S      | d S r   r   r9   r   r   r   r;     r<   c                 S   r"  r   r   r9   r   r   r   r;     r<   )r#   r$   	transformr    rP   )r?   r   rA   r@   r   r   r   test_apply_transform  s   r$  c                 C   sP   |  dd dd g}dd }||}|D ]\}}t|j| || qd S )Nc                 S   r  r   yearr9   r   r   r   r;     r  z,test_apply_multikey_corner.<locals>.<lambda>c                 S   r  r   r  r9   r   r   r   r;     r  c                 S   s   |  ddd  S )Nr   r  r   r   r   r   r    s   z%test_apply_multikey_corner.<locals>.f)r#   r$   r    r'   loc)tsframer   r  rA   r\   r   r   r   r   test_apply_multikey_corner  s   
r+  rN   TFc                 C   s   t g dtdd}d}tjt|d |jd| ddd	 }W d    n1 s+w   Y  |g d
}| rHtj	g d|j
gdd gd|_
t|| d S )N)	r
   r
   r
   rv   rv   rv   rw   rw   rw   r   )r\   r-   r   r   r\   rM   c                 S   s   | j d d S r   rj   r9   r   r   r   r;     rm   z'test_apply_chunk_view.<locals>.<lambda>)r   r
   rw   r~   r   r   )r
   r
   rv   rv   rw   rw   r   )r   r   r    r!   r"   r#   r$   taker   from_arraysr   r'   )rN   r(   r)   rA   r@   r   r   r   test_apply_chunk_view  s   r.  c                  C   st   t g dg dtdddd} | ddg}d}tjt|d	 |d
d  W d    d S 1 s3w   Y  d S )N)
r
   r
   r
   r
   r
   r
   rv   rv   rv   rv   )
r   r   r   r
   r
   r
   r   r   r
   r
   r   )r5   name2r-   r5   r0  r   r   c                 S      | j dddS )Nr-   Tinplacer(  r9   r   r   r   r;     rm   z4test_apply_no_name_column_conflict.<locals>.<lambda>)r   r   r#   r    r!   r"   r$   )r(   r   r)   r   r   r   "test_apply_no_name_column_conflict  s   
	"r4  c                  C   s   t g dtg ddtddd} dd }d	}tjt|d
 | jddd|}W d    n1 s6w   Y  | 	 }tg dd|d< t
|| d S )NrW   rW   rW   rX   rX   rX   r   rv   rW         @r   rx   vc                 S   ,   | d }||   | |    | d< | S Nr8  v2minmaxr   r8  r   r   r   r  '      z#test_apply_typecast_fail.<locals>.fr   r   r   FrM           g      ?r
   r;  )r   r   tilearanger    r!   r"   r#   r$   r   r'   )r(   r  r)   rA   r@   r   r   r   test_apply_typecast_fail  s   
rE  c                  C   s   t g dg dg} tg dtg ddtddd| d	}d
d }d}tjt|d |j	ddd
|}W d    n1 sCw   Y  | }tg dd|d< t|| d S )N)r   r   r   r
   r
   r
   r}   r5  r   rv   rW   r6  r7  r   c                 S   r9  r:  r<  r?  r   r   r   r  A  r@  z%test_apply_multiindex_fail.<locals>.fr   r   r   FrM   rA  r;  )r   r-  r   r   rC  rD  r    r!   r"   r#   r$   r   r'   )r   r(   r  r)   rA   r@   r   r   r   test_apply_multiindex_fail6  s    
	rF  c                 C   s4   | j dd dddd }| d }t|| d S )Nc                 S   r  r   r%  r9   r   r   r   r;   Q  r  z#test_apply_corner.<locals>.<lambda>FrM   c                 S   r"  r   r   r9   r   r   r   r;   Q  r<   rv   )r#   r$   r    r'   )r*  rA   r@   r   r   r   test_apply_cornerP  s   rG  c                  C   s   t g dg dg dd} dd }dd }d	}tjt|d
 | d|}W d    n1 s2w   Y  tjt|d
 | d|}W d    n1 sQw   Y  t|| d S )N)r   r      i,  )rT   rU   rx   rx   r   )id_fieldcategoryr-   c                 S   s$   | j d dkr|  S | | jdk S Nr   r
   rx   )shaper   rJ  r9   r   r   r   filt1b  s   z&test_apply_without_copy.<locals>.filt1c                 S   s    | j d dkr	| S | | jdk S rK  )rL  rJ  r9   r   r   r   filt2h  s   z&test_apply_without_copy.<locals>.filt2r   r   rI  r   )r]   rM  rN  r)   r@   rA   r   r   r   test_apply_without_copyV  s    rO  test_seriesc                 C   s   t ddgddgddggddgg dd}| r9|dd }|jdd	d
dd }| }| }t|| d S d}tjt|d |jdd	ddd }W d    n1 sYw   Y  |	d}|	d}t
|| d S )Nr:   poXY)r
   rv   rv   r   r   r   F)levelrN   c                 S   r   r   r   r9   r   r   r   r;   ~  rt   z<test_apply_with_duplicated_non_sorted_axis.<locals>.<lambda>r   r   rM   c                 S   r   r   r   r9   r   r   r   r;     rt   )r   	set_indexr#   r$   
sort_indexr    rP   r!   r"   r  r'   )rP  r(   serrA   r@   r)   r   r   r   *test_apply_with_duplicated_non_sorted_axisv  s     

rZ  c                  C   sf   g d} g d}t ddgd | d|d}t| |dd	}d
d }|jdddj|}t|| d S )Nr   r
   r
   rv   rv   Group1Group2rv   r   r-   r   r-   r   r5   c                 S   s"   |  t| j | j d S ri   )r  r   rD  r   r=  r>  r9   r   r   r   reindex_helper  s   "z1test_apply_reindex_values.<locals>.reindex_helperr   FrM   )r   r   r#   r-   r$   r    rP   )r  indicesr(   r@   r`  rA   r   r   r   test_apply_reindex_values  s   rb  c                  C   s   d} t jdjdd| d}t|t jd| g d| d  d}|jd	d
d}dd }d}tjt	|d |
|}W d    n1 sGw   Y  d|v sRJ d S )Nr  rv   r   r   )size)foobarbazquxr~   )r\   value1value2r\   FrM   c                 S   s   | d d | d< | S )Nrh  rv   value3r   r   r   r   r   r    s   z"test_apply_corner_cases.<locals>.fr   r   rj  )r   r   r   r   r   r   r#   r    r!   r"   r$   )Nlabelsr(   r   r  r)   rA   r   r   r   test_apply_corner_cases  s   rn  c                  C   s2  t ddgdgd ddgd} d}tjt|d | d	gd
d }W d    n1 s.w   Y  t| j| _d}tjt|d | d	gdd }W d    n1 sYw   Y  t	|d |d  t g dg dt
dgd d} dd }d}tjt|d | d|d }W d    n1 sw   Y  | j}| j|_t	|| dd }t g dg dg dg dd}| }t|j|_d}tjt|d |d|j}W d    n1 sw   Y  tjt|d |d|j}W d    n	1 sw   Y  t	|| d S )Nr
   rv   z
2017-03-02rd  inf)NumberDateStrr   r   rp  c                 S   
   | j d S r   rj   r9   r   r   r   r;     r   z;test_apply_numeric_coercion_when_datetime.<locals>.<lambda>c                 S   rs  r   rj   r9   r   r   r   r;     r   rr  )r         )rd  3r   z12:31:22rw   )r   r   Tc                 S   s   | j d dg S )Nr   r   rj   rk  r   r   r   get_B  s   z8test_apply_numeric_coercion_when_datetime.<locals>.get_Br   r   c                 S   s   t g dtd}dt| jv rt| | jdk jjd |d< dt| jv rCt| | jdk jjd |d< t| | jdk jjd |d< |S )	N)p1p2useTime)r   rK   step1r   ry  step2rz  r{  )r   rO   r   Staterf   Machiner  oTime)toolr   r   r   r   predictions  s   z>test_apply_numeric_coercion_when_datetime.<locals>.predictions)r   r   r   r   )r|  r}  r|  r}  ) z2016-09-19 05:24:33r  z2016-09-19 23:59:04)2336L36Rr  )Keyr~  r  r  r  )r   r    r!   r"   r#   r$   r%   r=   rq  rP   	Timestampr   r   r   r   r  ry  )r(   r)   r@   rA   rx  r  df1df2r   r   r   )test_apply_numeric_coercion_when_datetime  sT   	r  c                  C   s   t g dtdgd d} | j| j | d< d}tjt|d | dd	d
 }W d    n1 s5w   Y  t g dt	ddgd tdgd d
d}t|| d S )Nr  z2017-02-01 00:00:00rw   )clientidr   time_delta_zeror   r   r  c                 S   s   t | j | j dS )N)clientid_ager   )r   r  r=  r   )ddfr   r   r   r;     s    z?test_apply_aggregating_timedelta_and_datetime.<locals>.<lambda>r   r  )r  r  r   )r   r   
datetime64r   r    r!   r"   r#   r$   timedelta64rW  r'   r(   r)   rA   r@   r   r   r   -test_apply_aggregating_timedelta_and_datetime  s*   
r  c                  C   s~   ddgddgddgddgddgg} t | d	d
gtddd}|d	 }t g dg dd}|jd	dd t|| d S )Nr   r   r   rt  ru  r   (   2   NameValuez
2020-09-01z
2020-09-05rU  r  )r   r  Z   )r  r  Tr2  )r   r%   
date_ranger#   r  rW  r    r'   )r]   r(   rA   r@   r   r   r    test_apply_groupby_datetimeindex  s   "r  c                  C   s  t ddd tdD d} dd }dd	 }d
}tjt|d | jdgd|}W d    n1 s4w   Y  t ddidgd}d|j_d
}tjt|d | jdgd|}W d    n1 sdw   Y  t t	
dddddddgd}d|j_t|| t|| d S )Nr
   c                 S   s   g | ]}t  qS r   )r   now)rD   nnr   r   r   rF   ,  rG   z'test_time_field_bug.<locals>.<listcomp>r   ry   c                 S   s   t ddiS )Nrx   rv   r   batchr   r   r   func_with_no_date.  s   z.test_time_field_bug.<locals>.func_with_no_datec                 S   s   t tdddddS )N  r
   rv   rU   rx   )r   r   r  r   r   r   func_with_date1     z+test_time_field_bug.<locals>.func_with_dater   r   rT   byrx   rv   r   r  nsr  )r   r   r    r!   r"   r#   r$   r   r5   r%   r  as_unitr'   )r(   r  r  r)   dfg_no_conversiondfg_no_conversion_expecteddfg_conversiondfg_conversion_expectedr   r   r   test_time_field_bug&  s&   r  c                     s`   t g dg dg dg dd} | ddg} | jddgdd	}d
d  | fdd d S )N)rT   rT   rT   rU   rU   rU   rT   rT   rT   rU   rU   rU   )rx   rx   r   r   r   erx   rx   r   r   r   r  )g?rv   rw   r~   r   r   rv   r~   r   r|   r
   rv   )gffffff@r|   r   r         r|   r   r   r   r~   rw   )group1group2weightr-   r  r  T)rV  sortc                 S   s   t | | d}|S )Nrw   )r   arrayrepeat)r-   r  r   r   r   r   noddyS  s   z7test_gb_apply_list_of_unequal_len_arrays.<locals>.noddyc                    s    | j | jS r   )r-   r  r9   r  r   r   r;   ]  rm   z:test_gb_apply_list_of_unequal_len_arrays.<locals>.<lambda>)r   rW  r#   r$   )r(   
df_groupedr   r  r   (test_gb_apply_list_of_unequal_len_arraysF  s   
r  c                  C   sv   t g dg dd} dd }d}tjt|d | d|}W d    n1 s+w   Y  t  }t|| d S )	Nr{   )r|   r   r~   r   )r   random_varsc                 S   s   d S r   r   r9   r   r   r   	test_funce  r   z.test_groupby_apply_all_none.<locals>.test_funcr   r   r   r   )test_dfr  r)   rA   r@   r   r   r   test_groupby_apply_all_none`  s   r  c            
      C   s@  t g dg dd} t g dg dd}dd }d}tjt|d | d	|}W d    n1 s6w   Y  tjt|d |d	|}W d    n1 sUw   Y  tjd
d
gddggd	d gd}tjddgd
dggd	d gd}t d
d
gddgd|d}t ddgd
dgd|d}	t|| t||	 d S )N)r
   r
   r
   rv   )r   r
   rv   rw   )r   varsr   c                 S   s    | j d dk r	d S | jddg S )Nr   rv   r/  )rL  rk   r9   r   r   r   r  t  s   z0test_groupby_apply_none_first.<locals>.test_funcr   r   r   r
   r   rv   r   rw   r   )	r   r    r!   r"   r#   r$   r   r-  r'   )
test_df1test_df2r  r)   r   r   index1index2	expected1	expected2r   r   r   test_groupby_apply_none_firsto  s    r  c                  C   s   t ddgddgd} | d}d}tjt|d |d	d
 }W d    n1 s+w   Y  tdgdtjddgdggdd gd	dd}t
|| d S )Nr   r
   filledempty)r-   r   r   r   r   c                 S   s   | | j dk d S )Nr
   r-   )r-   r   r   r   r   r;         z7test_groupby_apply_return_empty_chunk.<locals>.<lambda>r-   r   )r5   r   )r   r#   r    r!   r"   r$   r   r   from_productr  rP   )r(   r   r)   rA   r@   r   r   r   %test_groupby_apply_return_empty_chunk  s   
r  c                  C   sx   t d g dg dd} | jddd}|dd	 }t g d
g dd}t|| |dd	 }t|| d S )Nza a brz   r~   r   r   r  r   FrM   c                 S      | |    S r   r  r9   r   r   r   r;     r   z-test_apply_with_mixed_types.<locals>.<lambda>)gUUUUUU?gUUUUUU?r
   )皙?g333333?rW   r   r   c                 S   r  r   r  r9   r   r   r   r;     r   )r   rC   r#   r#  r    r'   r$   )r(   r   rA   r@   r   r   r   test_apply_with_mixed_types  s   r  c                  C   s   t dddgitddgd} d}tjt|d | ddd }W d    n1 s,w   Y  ttdgtdggtddgdd	d}t|| d S )
NrT   r
   rv   r   r   r   c                 S   r  r   r   rk  r   r   r   r;     r  z*test_func_returns_object.<locals>.<lambda>r4   )	r   r   r    r!   r"   r#   r$   r   rP   r  r   r   r   test_func_returns_object  s   (r  group_column_dtlikec                 C   s   t dg| gd}d}tjt|d |ddd }W d    n1 s'w   Y  |r0dnd	}t d
gtdg|dddgd}t|| d S )Nrd  ry   r   r   rT   c                 S   s   t dgdgdS )Nspam*   r   r  r9   r   r   r   r;         z+test_apply_datetime_issue.<locals>.<lambda>r^   rO   r  rJ   r  rI   )r   r    r!   r"   r#   r$   r   r'   )r  r>   r(   r)   rA   rK   r@   r   r   r   test_apply_datetime_issue  s   
r  c               	   C   s   t tdtdtdtdtddddddddddddddd} dd }d	}tjt|d
 | d|d }W d    n1 sKw   Y  tdgtj	dgdddd}t
|| d S )Nz2015-02-24 00:00:00)r   r
   rv   rw   r~   zsome UA stringzanother UA string17661101)day	userAgentuserIdc                 S   s   t dd |  D S )Nc                 S   s    i | ]\}}||  jd  qS )r   )value_countsr   )rD   rx   sr   r   r   
<dictcomp>  s     zYtest_apply_series_return_dataframe_groups.<locals>.most_common_values.<locals>.<dictcomp>)r   itemsrl   r   r   r   most_common_values  r  zEtest_apply_series_return_dataframe_groups.<locals>.most_common_valuesr   r   r  r  z
2015-02-24r4   r_  )r   r%   r  r    r!   r"   r#   r$   r   DatetimeIndexrP   )tdfr  r)   rA   r@   r   r   r   )test_apply_series_return_dataframe_groups  s<   r  rJ  c                 C   s   ddgd }| r"t j|g dd}t jg dg ddd}g d}ntddgdd	}d
dg}t||d|d}ttd|ttdttdd	ddg}|j
ddddd }t|| |jjddgkslJ d S )Nr
   rv   r   rz   )
categoriesr   )r  r5   )rt     r   r4   rt  r  )r   r  r   r   )r   r   r   r  r   F)observedc                 S   r6   r   r  r9   r   r   r   r;     r<   z-test_apply_multi_level_name.<locals>.<lambda>)r%   CategoricalCategoricalIndexr   r   r   rD  r   r   rW  r#   r$   r    r'   r   r   )rJ  rU   expected_indexexpected_valuesr@   r(   rA   r   r   r   test_apply_multi_level_name  s$   

"
r  c              	   C   s   t jtdddddddftdddddd	d
ftdddddddftdddddddfgg dd}d}tjt|d |ddd j	}W d    n1 sTw   Y  | r]dnt
}ttd||tj|gg dd}t|| d S )Ni  r
   reddarkr   r  greenstormyrv   9i  bluebrightrw   r   i  calmr~   potato)observationcolormood	intensityscorerI   r   r   r  c                 S   rs  r   rj   rk  r   r   r   r;     r   z;test_groupby_apply_datetime_result_dtypes.<locals>.<lambda>r^   zdatetime64[ns]r   )r   from_recordsr%   r  r    r!   r"   r#   r$   rp   rO   r   r   rK   int64rP   )r>   r]   r)   rA   rK   r@   r   r   r   )test_groupby_apply_datetime_result_dtypes  s$   	r  r   abcrw   2020r  )r  freq))rT   r   )rT   r
   )rU   r   c                 C   st   t g dg dd| d}d}tjt|d |jddd	d
d }W d    n1 s-w   Y  t|| d S )Nr
   r
   rv   )r   r
   r   r^  r   r   r   r   FrM   c                 S   r   r   r   r9   r   r   r   r;   &  rt   z8test_apply_index_has_complex_internals.<locals>.<lambda>r   )r   r(   r)   rA   r   r   r   &test_apply_index_has_complex_internals  s   r  zfunction, expected_valuesc                 C   
   | j  S r   )r   to_listr9   r   r   r   r;   -  r   c                 C      t | j S r   )setr   r  r9   r   r   r   r;   .  rm   c                 C   r  r   )tupler   r  r9   r   r   r   r;   /  rm   )r   r
   rv   rw   c                 C   s   t t| j S r   )dict	enumerater   r  r9   r   r   r   r;   1  r  c                 C   s   dd t | j D S )Nc                 S   s   g | ]\}}||iqS r   r   )rD   nir   r   r   rF   5      z<lambda>.<locals>.<listcomp>)r	  r   r  r9   r   r   r   r;   5  r  c                 C   s|   t g ddgd}d}tjt|d |d| }W d    n1 s%w   Y  t|tddgddd	}t|| d S )
N)r   r   r   r   r   rI   r   r   r   r   r4   r   	r   r    r!   r"   r#   r$   r   r   rP   )functionr  r(   r)   rA   r@   r   r   r   1test_apply_function_returns_non_pandas_non_scalar*  s   r  c                  C   s   dd } t g ddddtjgd}d}tjt|d	 |d
| }W d    n1 s.w   Y  tddgdgtjggt	g dd
dd}t
|| d S )Nc                 S   s   | d j  S )Nr   )r  flattenr   r   r   r   fctF  r   z4test_apply_function_returns_numpy_array.<locals>.fct)rT   rT   rU   noner
   rv   rw   r  r   r   r   rW   rX   rY   )rT   rU   r  r4   r   )r   r   r  r    r!   r"   r#   r$   r   r   rP   )r  r(   r)   rA   r@   r   r   r   'test_apply_function_returns_numpy_arrayD  s    r  r  c                 C   r  r   r   grr   r   r   r;   T  r  c                 C   s   | j d d S ri   r   r  r   r   r   r;   T  rm   c                 C   s   t g ddgd}d}tjt|d |d| }W d    n1 s%w   Y  ttg dtg dtdd	ggtg d
ddd}t|| d S )N)
r
   rv   rv   rv   r
   rv   rw   r
   rw   r
   idrI   r   r   )r   r~   r   r   )r
   rv   rw   r   r   r|   rz   r4   r   r  )r  r(   r)   rA   r@   r   r   r    test_apply_function_index_returnT  s    r  c                  C   sT   t g dg dd} | jddddd }t g d	g d
d}t|| d S )Nr   )r
   rv   r~   r~   r   r   r   r   Fr   c                 S   r6   r   )r   r9   r   r   r   r;   j  r<   zAtest_apply_function_with_indexing_return_column.<locals>.<lambda>)r   r   r   )rY   rZ   rZ   )r   r#   r$   r    r'   )r(   rA   r@   r   r   r   /test_apply_function_with_indexing_return_columnb  s   r  udfc                 C   r6   r   r   r9   r   r   r   r;   v  r<   c                 C   s   |   dd S )Nc                 S   s   | d S ri   r   )yr   r   r   r;   v  r<   z<lambda>.<locals>.<lambda>)r   r  r9   r   r   r   r;   v  r  c                 C   s   t ddgddgd}d}tjt|d |jd| d	|}W d    n1 s)w   Y  |jj|j| d	|}| rN|jj	dksDJ |jj	dksLJ d S |jj	dksVJ |jj	dks^J d S )
NrT   rU   r
   rv   r  r   r   r   rM   )
r   r    r!   r"   r#   r$   r   r   r   nlevels)rN   r  r(   r)   	df_resultseries_resultr   r   r   test_apply_result_typet  s   	r  c                  C   s   t g dg dd} d}tjt|d | jdddd	d
 }W d    n1 s+w   Y  tjt|d | jddddd
 }W d    n1 sNw   Y  t|| d S )N)rv   r
   rv   rz   r  r   r   r   FrM   c                 S   r   r   r   r9   r   r   r   r;     rt   z4test_result_order_group_keys_false.<locals>.<lambda>c                 S   r6   r   r   r9   r   r   r   r;     r<   r   r  r   r   r   "test_result_order_group_keys_false  s   r  c                  C   s  dgd dgd  dgd  } t | }t j| dd}tttdd td|d	}tttdd td|d	}d
}tjt|d |jddd	dd }W d    n1 s[w   Y  tjt|d |jddd	dd }W d    n1 s~w   Y  t
|| d S )Nz
2001-01-01rv   z
2001-01-02z
2001-01-03UTC)tzrw   r   )r:   r  tr   r   r:   FrM   c                 S      | ddg   S Nr:   r  r   rl   r   r   r   r;     r  z1test_apply_with_timezones_aware.<locals>.<lambda>c                 S   r#  r$  r   rl   r   r   r   r;     r  )r%   r  r   r   r   r    r!   r"   r#   r$   r'   )datesindex_no_tzindex_tzr  r  r)   r   r   r   r   r   test_apply_with_timezones_aware  s"   
  r(  c                 C   s  t g dg dg dd}t ddgddgd	td
dgddd}|jdd}d}tjt|dd |jtdd}W d    n1 sCw   Y  t|| |jdd}t	| |}t
|| | }tjt|dd |jtdd}W d    n1 szw   Y  t|| d S )N)c   r)  r)  X   r*  r*  )r
   rv   rw   r~   r   r   )r   rt  ru  r  r  <   r      r      r+  r  r*  r)  rT   r4   r   r  :The behavior of DataFrame.sum with axis=None is deprecatedF)r   check_stacklevelinclude_groups)r   r   r#   r    r!   ro   r$   r  r'   r	   getattr)reduction_funcr(   r@   grpr)   rA   args_r   r   r   ;test_apply_is_unchanged_when_other_methods_are_called_first  s.   
r7  c               	   C   s  t g dtdddtdddtdddtdddgg ddtg dd	d
d} | ddg}d}tjt|d |dd }W d    n1 sJw   Y  | jg d }|	 }t
|g d |_|jd	d}t|| |jjd D ]
}t|tu sJ qvd S )N)rT   rT   rT   rU   i  r
   r   rv   r   r  )r   e   f   g   idxr4   r   r   r   r   r   c                 S   r   ri   r   r9   r   r   r   r;     r   zRtest_apply_with_date_in_multiindex_does_not_convert_to_timestamp.<locals>.<lambda>)r   rv   rw   )r   r   r;  rI   )r   r   r   r#   r    r!   r"   r$   rk   r   r   
from_framer   r  r'   levelstype)r(   r4  r)   rA   r@   valr   r   r   @test_apply_with_date_in_multiindex_does_not_convert_to_timestamp  s0   




r@  c                  C   s   t tjdddgtddgddggd} d}tjt|d	 | j	j
d
d
d}W d    n1 s4w   Y  |dd }d}tjt|d	 | j
dd
d}W d    n1 s[w   Y  |dd }t||j	 t||  d S )Nrv   r   r~   r   r   r
   rI   z5The 'axis' keyword in DataFrame.groupby is deprecatedr   r   ra   rV  c                 S   s   | j dddS )Nr   rA  	droplevelr9   r   r   r   r;     rm   zDtest_apply_by_cols_equals_apply_by_rows_transposed.<locals>.<lambda>rc   c                 S   r1  )Nr
   r   rA  rB  r9   r   r   r   r;     rm   )r   r   r   r   r   r  r    r!   ro   rw  r#   r$   r'   )r(   r)   rr   by_rowsgb2by_colsr   r   r   2test_apply_by_cols_equals_apply_by_rows_transposed  s    rG  dropnac                 C   s   t g ddtjtjddgdtdd}d}tjt|d |jd	| d
ddd }W d    n1 s5w   Y  | r@|	 n|j
g d }t|| d S )Nr
   rv   rw   r~   r   rT   rU   )colr   xxyxzr   r   r   r   F)rH  rN   c                 S   r   r   r   r9   r   r   r   r;     rt   z5test_apply_dropna_with_indexed_same.<locals>.<lambda>)r   rw   r
   rv   r~   )r   r   r  r   r    r!   r"   r#   r$   rH  rk   r'   )rH  r(   r)   rA   r@   r   r   r   #test_apply_dropna_with_indexed_same  s   rL  zas_index, expected)r
   r
   r
   )rv   rv   r
   )rT   rU   N)rK   r   )rv   rv   rU   r   r   c                 C   sz   t g dg dg dd}d}tjt|d |jddg| dd	d
 }W d    n1 s0w   Y  t|| d S )Nr[  )r
   r
   r
   r
   r   r   r   rT   rU   r   c                 S   s   dS ri   r   r9   r   r   r   r;   -  rt   z5test_apply_as_index_constant_lambda.<locals>.<lambda>)r   r    r!   r"   r#   r$   assert_equal)r   r@   r(   r)   rA   r   r   r   #test_apply_as_index_constant_lambda  s   rN  c                  C   s   t g dg dg ddtdd} d}tjt|d | d	d
d }W d    n1 s0w   Y  ttddtj	g dd	d gddd}t
|| d S )NrI  )r   r   r|   r   r   )r
   r
   r
   rv   rv   r  r   r   r   r   r   c                 S   r  r   )r   rX  r9   r   r   r   r;   9  r   z(test_sort_index_groups.<locals>.<lambda>r
   r   )r   r   r   r  r   r   r   r_  )r   r   r    r!   r"   r#   r$   r   r   r   rP   r  r   r   r   test_sort_index_groups1  s    rO  c                  C   s   t tjddddtdtdd} d}tjt|d | j| j	| j
jj
gd	d
dd }W d    n1 s8w   Y  t||  d S )Nz
2010-01-0112hr   )r  r  r   )r   valsletr   r   FrM   c                 S   s   | j dd  S r   rj   r9   r   r   r   r;   Q  rm   z;test_positional_slice_groups_datetimelike.<locals>.<lambda>)r   r%   r  r   r   r    r!   r"   r#   rR  r   dtr$   r'   )r@   r)   rA   r   r   r   )test_positional_slice_groups_datetimelikeD  s   
rT  c                  C   sr   t g dg dg dd} | d}|ddg dd	 }t d
dgddgdtddgddd}t|| d S )N)rT   rT   rU   rz   r  r  r   r   r   c                 S   s   |  t |   S r   )astypefloatr>  r=  r9   r   r   r   r;   Y  s    z7test_groupby_apply_shape_cache_safety.<locals>.<lambda>rW   rB  rX   r  rT   rU   r4   r   )r   r#   r$   r   r    r'   )r(   rr   rA   r@   r   r   r   %test_groupby_apply_shape_cache_safetyU  s   
rW  c                  C   s~   t g dg dg dd} | dddg }|dd	 }tjg d
g dgdd gd}tg d|dd}t|| d S )N)rT   rU   rT   rU   )aaacrY  ad)rX  rX  rX  rY  )rT   b1b2rT   r[  r\  c                 S   s   |    S r   )unstackr  r9   r   r   r   r;   k  r   z3test_groupby_apply_to_series_name.<locals>.<lambda>)rT   rT   rU   rU   rU   )rX  rY  rY  rZ  rX  )arraysr   )rw   r
   rv   r
   r
   r   r_  )	r   	from_dictr#   r$   r   r-  r   r    rP   )r(   r4  rA   expected_idxr@   r   r   r   !test_groupby_apply_to_series_namea  s   ra  c                 C   s   t g dg dddtjtjgd}|jd| d}d}tjt|d	 |d
d }W d    n1 s4w   Y  tjt|d	 |dd }W d    n1 sRw   Y  t|| d S )Nr[  )r
   r   rv   r   r
   rv   )r4  r  zr4  )rH  r   r   c                 S   s   |  ddS )Nr
   rb  )nlargest)grp_dfr   r   r   r;   ~  r   ztest_apply_na.<locals>.<lambda>c                 S   s   | j ddddS )Nrb  F)	ascendingr
   )r  r   r9   r   r   r   r;     rG   )	r   r   r  r#   r    r!   r"   r$   r'   )rH  r(   dfgrpr)   rA   r@   r   r   r   test_apply_nau  s   rg  c                  C   s   d} t jt| d% tg dg dtjg ddddd	d
gdd }W d    n1 s1w   Y  tddtjdddgddtjdddggg dtj	ddgd	d
gdd}t 
|| d S )Nr   r   r[  )r  r  r  r  r   r  )unitr   rT   rU   c                 S   rs  Nr/  rj   rl   r   r   r   r;     r   z8test_apply_empty_string_nan_coerce_bug.<locals>.<lambda>r
   r  rv   r~   )r
   r  )rv   r  r   rU  )r    r!   r"   r   r%   r=   r#   r$   r   r   r'   )r)   rA   r@   r   r   r   &test_apply_empty_string_nan_coerce_bug  s$   

&rj  index_values)rW   rX   rY   c                 C   s   t g dg ddt| d}t dg ditg ddd	d}d
}tjt|d |ddd }W d    n1 s=w   Y  t|| d S )N)rX  a2a3rz   ry   r   b_mean)rX   rY   rW   )rl  rm  rX  rT   r4   r   r   c                 S   s   t | d  gdgdS )NrU   rn  r   )r   r   rl   r   r   r   r;     r  z0test_apply_index_key_error_bug.<locals>.<lambda>)r   r   r    r!   r"   r#   r$   r'   )rk  rA   r@   r)   r   r   r   test_apply_index_key_error_bug  s$   
ro  zarg,idx)皙?333333?皙?)rp  rr  rq  )r
   r~   rw   )rp  r  rr  c                 C   sj   t d| i|d}d}tjt|d |jddddd }W d    n1 s(w   Y  t|| d S )	NrJ  r   r   r   FrM   c                 S   r   r   r   r9   r   r   r   r;     rt   z5test_apply_nonmonotonic_float_index.<locals>.<lambda>r   )argr;  r@   r)   rA   r   r   r   #test_apply_nonmonotonic_float_index  s   +rt  zargs, kwargsnumeric_onlyc                 C   s>   |  d}|jdg|R i |}|jdd}t|| d S )Nr   r  T)ru  )r#   r$   r  r    r'   )r(   r5  kwargsrr   rA   r@   r   r   r   test_apply_str_with_args  s   
rw  r5   	some_namec                 C   sL   t ddg| d}|jddgdddd }t ddg| d}t|| d S )	Nr
   rv   r4   rT   FrM   c                 S   r   r   r   r9   r   r   r   r;     rt   z1test_result_name_when_one_group.<locals>.<lambda>)r   r#   r$   r    rP   )r5   rY  rA   r@   r   r   r   test_result_name_when_one_group  s   ry  z
method, opr$   c                 C   rs  ri  )r  rr   r   r   r   r;     r   c                 C   s   | d j d S )NrU   r   rj   rz  r   r   r   r;     rm   )aggskew)r{  prod)r{  r  c              	   C   s^   t g g d}|jddd}t|d}t|| |}tg ddtg dddd}t|| d S )	Nry   rT   TrM   rU   r_   rJ   )r5   rK   r   )r   r#   r2  r   r   r    rP   )methodopempty_dfrr   r   rA   r@   r   r   r   test_empty_df  s   
r  r1  c                 C   s   t g dg dd}|d}| rtnd }d}tj||d |jdd | d	}W d    n1 s3w   Y  t d
d
gddgdtdd
gddd}| sR|dg }t|| d S )Nr   )rw   r~   r   ry   rT   r   r   c                 S   r6   r   r  r9   r   r   r   r;     r<   z%test_include_groups.<locals>.<lambda>r0  rv   r   r   r
   r4   r   rU   )r   r#   r"   r    r!   r$   r   r'   )r1  r(   rr   warnr)   rA   r@   r   r   r   test_include_groups  s   
&
r  r  rb   jimjoec                    s  t jd}t|dddddgd}|d|d	< || }|j |tur(d nt	}d
}t
j||ddd ||}W d    n1 sEw   Y  t|j| d}d|j d| d}	|j|dfksgJ |	 fddd}t
jt|d |}
W d    n1 sw   Y  t
||
 t
jt|d |fdd}W d    n1 sw   Y  t
|| |tkr|  }
|
j| ddd t
j||
dd t
t| ddt| dd d S )Nrv   r
   r   )r   rv   r  r  rI   r   jolier.  F)r   r/  raise_on_extra_warnings)subsetzinvalid frame shape: z (expected (z, 3))rw   c                    s   t t | ddS )Nr   rh   )r2  r   r9   )fnamer   r   r;   7  r  z%test_builtins_apply.<locals>.<lambda>r   r   c                    s    | S r   r   r9   )npfuncr   r   r;   >  r<   T)r3  r  )check_dtyper   rh   )r   r   r   r   r   r   r#   __name__r  ro   r    r!   r$   r   drop_duplicatesrL  r"   r'   r{  r   rW  rP   r2  )rb   r  rsr(   rr   r  r)   rA   ngroups
assert_msgr@   r  r   )r  r  r   test_builtins_apply  s<   
(r  )dr   r   numpyr   pytestpandasr%   r   r   r   r   r   pandas._testing_testingr    pandas.tests.groupbyr	   r+   rB   rR   rs   ru   markparametrizer   r   r   r   r   r   r   r   r   r  r  r  r  r  r!  r$  r+  r.  r4  rE  rF  rG  rO  rZ  rb  rn  r  r  r  r  r  r  r  r  r  r  todayrH   r  r  r  r  r  interval_rangeperiod_ranger   r  r  r  r  r  r  r  r(  r7  r@  rG  rL  rO   rN  rO  rT  rW  ra  rg  rj  ro  rt  rw  ry  r  r  r>  r=  r  r  r   r   r   r   <module>   sj   (-"

.
	*)

 
> 
(


		


  

	


)	

	




