U
    U‚gz¡  ã                   @   s~  d dl mZm Z  d dl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ejjde	dddddgdddddgdœƒdddgfe	d d ddgd dd dgdœƒd dgfe	ddgiƒdgfe	ddddddddgedƒdœƒddgfe	ddddddgddddddgdœƒdddgfe	edƒddddddddddg
dd ddddddddg
dœƒdddgfe	dddgdddggdddgd�ddgfgddd d!d"d#d$gd%�d&d'„ ƒZd(d)„ Zd*d+„ Zej d,d-d.„ d/d.„ d0d.„ d1d.„ g¡d2d3„ ƒZd4d5„ Zd6d7„ Zd8d9„ Z d:d;„ Z!d<d=„ Z"d>d?„ Z#d@dA„ Z$dBdC„ Z%dDdE„ Z&dFdG„ Z'dHdI„ Z(ej dJdKdLg¡dMdN„ ƒZ)dOdP„ Z*dQdR„ Z+dSdT„ Z,dUdV„ Z-dWdX„ Z.ej dYdKdLg¡dZd[„ ƒZ/d\d]„ Z0d^d_„ Z1d`da„ Z2dbdc„ Z3ddde„ Z4dfdg„ Z5dhdi„ Z6djdk„ Z7dldm„ Z8dndo„ Z9dpdq„ Z:drds„ Z;ej dte  <¡ e  <¡  ¡ e  <¡  =¡ g¡dudv„ ƒZ>dwdx„ Z?ej dydLdKg¡dzd{„ ƒZ@d|d}„ ZAej d~e Bedƒ¡e Cd d¡ejDd€dd�d‚�e Edƒd„d…g¡g¡d†d‡„ ƒZFej dˆd‰d.„ d dgddggfdŠd.„ d dhddhgfd‹d.„ dŒd�gfdŽd.„ d ddŒœdddŒœgfd�d.„ d d iddigd diddiggfg¡d�d‘„ ƒZGd’d“„ ZHej d”d•d.„ d–d.„ g¡d—d˜„ ƒZId™dš„ ZJej d›dœd.„ d�d.„ g¡ej dJdKdLg¡dždŸ„ ƒƒZKd d¡„ ZLd¢d£„ ZMd¤d¥„ ZNd¦d§„ ZOd¨d©„ ZPej dªdKdLg¡d«d¬„ ƒZQej d­dLe	dddgdddgge
dddgeRd®�d�gdKeddgejEd¯d°gddgd±�d²�gg¡d³d´„ ƒZSdµd¶„ ZTd·d¸„ ZUd¹dº„ ZVej dªdKdLg¡d»d¼„ ƒZWd½d¾„ ZXej d¿dddgdÀdÁdÂgg¡dÃdÄ„ ƒZYej dÅdddgdÆdÇdÈggdddgdÆdÈdÇggdddgdÆdÉdÈggg¡dÊdË„ ƒZZej dÌdKgi fg dÍdKifg¡dÎdÏ„ ƒZ[ej dÐdÑdg¡dÒdÓ„ ƒZ\ej dÔdÕdÖd.„ fdÕd×d.„ fdØdÙdÚg¡dÛdÜ„ ƒZ]dS )Ýé    )ÚdateÚdatetime)ÚStringION)Ú	DataFrameÚIndexÚ
MultiIndexÚSeriesÚbdate_range)Úget_groupby_method_argsc                  C   sÎ   d} t jt| ƒd dddgddggd�}| d¡}| |jj¡ ¡ }| |jj¡ dd„ ¡}t	 
||¡ t jt| ƒd dddgd	�}td
ddgtdd�}tdddg|d�}|jddd� dd„ ¡}t	 ||¡ d S )Na+  2011.05.16,00:00,1.40893
2011.05.16,01:00,1.40760
2011.05.16,02:00,1.40750
2011.05.16,03:00,1.40649
2011.05.17,02:00,1.40893
2011.05.17,03:00,1.40760
2011.05.17,04:00,1.40750
2011.05.17,05:00,1.40649
2011.05.18,02:00,1.40893
2011.05.18,03:00,1.40760
2011.05.18,04:00,1.40750
2011.05.18,05:00,1.40649r   ÚtimeÚvalue)ÚheaderÚnamesZparse_datesÚ	date_timec                 S   s   |   ¡ S ©N©Úidxmax©Úx© r   úC/tmp/pip-unpacked-wheel-u1j4u1ka/pandas/tests/groupby/test_apply.pyÚ<lambda>/   ó    z#test_apply_issues.<locals>.<lambda>)r   r   z
2011.05.16z
2011.05.17z
2011.05.18©ÚdtypeÚnamez00:00z02:00©ÚindexF©Ú
group_keysc                 S   s   | d | d   ¡  S )Nr   r   r   r   r   r   r   r   :   r   )ÚpdZread_csvr   Ú	set_indexÚgroupbyr   r   r   ÚapplyÚtmÚassert_frame_equalr   Úobjectr   Úassert_series_equal)ÚsÚdfÚexpectedÚresultZexp_idxr   r   r   Útest_apply_issues   s,    ü
  ÿÿr,   c                     sŽ   t dddddgdddddgdœd	d
gd�‰ tjˆ jdd … ˆ jdd … gdddgd�} ˆ jdd„ ˆ jD ƒdd� ‡ fdd„¡}t || ¡ d S )NÚaÚbç      ð?ç       @ç      @ç      @ç      @©ÚkeyÚdatar5   r6   ©Úcolumnsé   Úfloat64r&   ©ÚaxisÚkeysc                 S   s   g | ]}t |ƒ‘qS r   ©Ústr©Ú.0r   r   r   r   Ú
<listcomp>G   s     z&test_apply_trivial.<locals>.<listcomp>©r<   c                    s   ˆ j dd … S ©Nr9   ©Úilocr   ©r)   r   r   r   H   r   z$test_apply_trivial.<locals>.<lambda>)	r   r    ÚconcatrF   r"   Údtypesr#   r$   r%   ©r*   r+   r   rG   r   Útest_apply_trivial?   s    þ,
ÿrK   c                     s|   t dddddgdddddgdœd	d
gd�‰ tjˆ ˆ gdddgd�} ˆ jdd„ ˆ jD ƒddd� ‡ fdd„¡}t || ¡ d S )Nr-   r.   r/   r0   r1   r2   r3   r4   r5   r6   r7   r9   r:   r&   r;   c                 S   s   g | ]}t |ƒ‘qS r   r>   r@   r   r   r   rB   U   s     z+test_apply_trivial_fail.<locals>.<listcomp>T)r<   r   c                    s   ˆ S r   r   r   rG   r   r   r   V   r   z)test_apply_trivial_fail.<locals>.<lambda>)r   r    rH   r"   rI   r#   r$   r%   rJ   r   rG   r   Útest_apply_trivial_failN   s    þ
ÿrL   zdf, group_namesr9   é   é   r-   r.   Úc©r-   r.   é   é   é   é   é   é	   )r-   ÚtwoZ
aaabbbcccc)r-   ÚBÚCr7   ZGH2936zGH7739 & GH10519ZGH10519ZGH2656ZGH12155ZGH20084ZGH21417)Úidsc                    s‚   g ‰ ‡ fdd„}‡ fdd„}‡ fdd„}‡ fdd„}‡ fd	d
„}|||||fD ].}ˆ d d …= | j ddd� |¡ ˆ |ksNt‚qNd S )Nc                    s   ˆ   | j¡ |  ¡ S r   )Úappendr   Úcopy©Úgroup©r   r   r   Úf_copy‡   s    z/test_group_apply_once_per_group.<locals>.f_copyc                    s   ˆ   | j¡ | S r   ©r[   r   r]   r_   r   r   Úf_nocopyŒ   s    z1test_group_apply_once_per_group.<locals>.f_nocopyc                    s   ˆ   | j¡ dS ©Nr   ra   r]   r_   r   r   Úf_scalar‘   s    z1test_group_apply_once_per_group.<locals>.f_scalarc                    s   ˆ   | j¡ d S r   ra   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 )Nr9   rP   )r[   r   r   r]   r_   r   r   Úf_constant_dfš   s    z6test_group_apply_once_per_group.<locals>.f_constant_dfr-   Fr   )r"   r#   ÚAssertionError)r)   Zgroup_namesr`   rb   rd   re   rf   Úfuncr   r_   r   Útest_group_apply_once_per_group\   s    '
ri   c              
   C   s~   d}t ddddddddgdddddd	d
dgdœdddddd	d
dgd�}|jddd� dd„ ¡ |  ¡ j d¡}||kszt‚d S )NrM   r   r9   Ú0Ú2Ú4Ú6Ú8Ú10Z12Z14)Úgroup_by_columnZtest_columnr   rp   Fr   c                 S   s   t dƒS )NÚfunction_called)ÚprintrG   r   r   r   r   µ   r   z2test_group_apply_once_per_group2.<locals>.<lambda>rq   )r   r"   r#   Z
readouterrÚoutÚcountrg   )Zcapsysr*   r)   r+   r   r   r   Ú test_group_apply_once_per_group2¦   s    þûÿru   c                  C   s`   t dddgtdƒdœƒ} dd„ }dd„ }| jd	d
d� |¡}| jd	d
d� |¡}t ||¡ d S )Nr   r9   rN   )ÚAr.   c                 S   s   | S r   r   r]   r   r   r   ÚslowÄ   s    z,test_apply_fast_slow_identical.<locals>.slowc                 S   s   |   ¡ S r   ©r\   r]   r   r   r   ÚfastÇ   s    z,test_apply_fast_slow_identical.<locals>.fastrv   Fr   )r   Úranger"   r#   r$   r%   )r)   rw   ry   Zfast_dfZslow_dfr   r   r   Útest_apply_fast_slow_identical½   s    r{   rh   c                 C   s   | S r   r   r   r   r   r   r   Ó   r   r   c                 C   s   | d d … S r   r   r   r   r   r   r   Ô   r   c                 C   s   | j dd�S )NF©Údeeprx   r   r   r   r   r   Õ   r   c                 C   s   | j dd�S )NTr|   rx   r   r   r   r   r   Ö   r   c                 C   sL   t ddddgddddgddddgd	œƒ}|jd
dd� | ¡}t ||¡ d S )Nr9   rM   rN   rR   rS   rT   rU   rQ   )Úgr-   r.   r~   Fr   ©r   r"   r#   r$   r%   )rh   r)   r+   r   r   r   Ú5test_groupby_apply_identity_maybecopy_index_identicalÐ   s    (r€   c                  C   s¤   t tj d¡ddddddgdœƒ} | jdd„ dd	�j}| j}t ||¡ t d
dddddgiƒ} | jd | d< |  	d¡ 
¡  ¡ j}| j	ddd� 
¡ j}t ||¡ d S )NrT   ÚonerW   Úthree©Úfoo1Zfoo2c                 S   s   | S r   r   r   r   r   r   r   í   r   z-test_apply_with_mixed_dtype.<locals>.<lambda>r9   rC   Úc1rM   rQ   r0   Úc2F©Úas_index)r   ÚnpÚrandomÚrandnr#   rI   r$   r'   r…   r"   ÚmeanÚreset_indexr†   )r)   r+   r*   Úresult1Úresult2r   r   r   Útest_apply_with_mixed_dtypeå   s    
þÿr�   c                  C   sR  t ddddddgddddddgtdƒdœƒ} | jd	d
d�}| jd	dd�}| d¡j}| d¡j}tddddgƒ}t ||¡ t ||¡ | dd„ ¡j}| dd„ ¡j}t	 
ddddg¡}ddddg}	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�} | jdddd� dd„ ¡j}t ||¡ d S ) Nr.   r-   rO   r9   rM   rN   rT   )Zitem_idÚuser_idr   r‘   Tr‡   Fr   rR   c                 S   s
   |   d¡S ©NrM   ©Úheadr   r   r   r   r     r   z-test_groupby_as_index_apply.<locals>.<lambda>c                 S   s
   |   d¡S r’   r“   r   r   r   r   r     r   )r   r   )r   rM   ©r9   r9   ©rM   rR   ©r9   r   ©r9   rM   )rM   r9   )rN   rR   r_   ÚabcderS   r   )rˆ   r   c                 S   s   | S r   r   r   r   r   r   r     r   )r   rz   r"   r”   r   r   r$   Úassert_index_equalr#   r   Úfrom_tuplesÚlist)r)   Zg_asZg_not_asZres_asZ
res_not_asÚexpZres_as_applyZres_not_as_applyZexp_not_as_applyÚtpZexp_as_applyÚindÚresr   r   r   Útest_groupby_as_index_applyù   s0    ýÿ*r¡   c                 C   sx   |   ddg¡}dd„ }dd„ }dd„ }| |¡}|jjd	ks@t‚| |¡}|jjd	ksZt‚| |¡}|jjd
kstt‚d S )Nrv   rX   c                 S   s   |   ¡ }d|j_|S ©NÚstat)Údescriber   r   ©r^   r+   r   r   r   Údesc!  s    z.test_apply_concat_preserve_names.<locals>.descc                 S   s$   |   ¡ }d|j_|d t| ƒ… }|S r¢   )r¤   r   r   Úlenr¥   r   r   r   Údesc2&  s    z/test_apply_concat_preserve_names.<locals>.desc2c                 S   s0   |   ¡ }dt| ƒd›�|j_|d t| ƒ… }|S )NZstat_Úd)r¤   r§   r   r   r¥   r   r   r   Údesc3-  s    z/test_apply_concat_preserve_names.<locals>.desc3)rv   rX   r£   )rv   rX   N)r"   r#   r   r   rg   )Zthree_groupÚgroupedr¦   r¨   rª   r+   r�   Zresult3r   r   r   Ú test_apply_concat_preserve_names  s    



r¬   c                  C   st   dd„ } t ddd�}ttj d¡|d�}|jdd„ d	d
�}| | ¡}t|tƒsRt	‚t
|dƒr`t	‚t |j|j¡ d S )Nc              	   S   s:   t jdd�� t  | ¡}W 5 Q R X t| | |  ¡  |dœƒS )NÚignore)Úinvalid)r   ZdemeanedÚlogged)r‰   ZerrstateÚlogr   rŒ   )Zpiecer¯   r   r   r   ÚfB  s
    ÿz%test_apply_series_to_frame.<locals>.fz1/1/2000éd   )Úperiodsr   c                 S   s   | j S r   ©Úmonthr   r   r   r   r   L  r   z,test_apply_series_to_frame.<locals>.<lambda>Fr   r   )r	   r   r‰   rŠ   r‹   r"   r#   Ú
isinstancer   rg   Úhasattrr$   rš   r   )r±   ZdrÚtsr«   r+   r   r   r   Útest_apply_series_to_frameA  s    
r¹   c                 C   s4   |   ddg¡d  t¡}|jjd d… dks0t‚d S )Nrv   rX   rY   rM   ©rv   rX   )r"   r#   r§   r   r   rg   ©r)   r+   r   r   r   Ú test_apply_series_yield_constantT  s    r¼   c                 C   sl   |   ddg¡ t¡}t|tƒs"t‚|jd ks0t‚|   ddg¡ddg  t¡}t|tƒsZt‚|jd ksht‚d S )Nrv   rX   rY   ÚD)r"   r#   r§   r¶   r   rg   r   r»   r   r   r   Útest_apply_frame_yield_constantY  s    r¾   c                 C   sH   |   ddg¡}| t¡}| ¡ d }t |j|j¡ t |j|j¡ d S )Nrv   rX   rY   )	r"   r#   r§   rt   r$   rš   r   Úassert_numpy_array_equalÚvalues©r)   r«   r+   r*   r   r   r   Útest_apply_frame_to_seriesd  s
    
rÂ   c                 C   s^   | j ddgdd�}| t¡}| ¡ jdtjid�jdd�}t 	|j
|j
¡ t |j|j¡ d S )Nrv   rX   Fr‡   rY   r7   r½   )r"   r#   r§   rt   Úrenamer‰   ÚnanÚdropr$   rš   r   r¿   rÀ   rÁ   r   r   r   Ú)test_apply_frame_not_as_index_column_namel  s
    
rÆ   c                     sŠ   dd„ } ‡ fdd„}t tj ddd¡tj ddd¡tj d¡dœƒ‰ ˆ  d	¡ | ¡}ˆ  d	¡d
  |¡}tj||dd� |j	d
ks†t
‚d S )Nc                 S   s    |   d¡d  ¡  ¡ jd d… S )NrX   rY   rM   )r"   ÚsumÚsort_valuesrF   r]   r   r   r   Útransw  s    z-test_apply_frame_concat_series.<locals>.transc                    s,   |   ˆ  | j¡d ¡}| ¡  ¡ jd d… S )NrX   rM   )r"   Úreindexr   rÇ   rÈ   rF   )r^   r«   rG   r   r   Útrans2z  s    z.test_apply_frame_concat_series.<locals>.trans2r   rS   éè  ©rv   rX   rY   rv   rY   F)Zcheck_names)r   r‰   rŠ   Úrandintr‹   r"   r#   r$   r'   r   rg   )rÉ   rË   r+   r�   r   rG   r   Útest_apply_frame_concat_seriesv  s    
ýÿrÏ   c                 C   s>   | j dd„ dd�}| dd„ ¡}| dd„ ¡}t ||¡ d S )Nc                 S   s   | j S r   r´   r   r   r   r   r   �  r   z&test_apply_transform.<locals>.<lambda>Fr   c                 S   s   | d S r’   r   r   r   r   r   r   Ž  r   c                 S   s   | d S r’   r   r   r   r   r   r   �  r   )r"   r#   Ú	transformr$   r'   )r¸   r«   r+   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   s   | j S r   ©Úyearr   r   r   r   r   ”  r   z,test_apply_multikey_corner.<locals>.<lambda>c                 S   s   | j S r   r´   r   r   r   r   r   ”  r   c                 S   s   |   d¡dd … S )Nrv   éûÿÿÿ©rÈ   r]   r   r   r   r±   –  s    z%test_apply_multikey_corner.<locals>.f)r"   r#   r$   r%   Úloc)Útsframer«   r±   r+   r5   r^   r   r   r   Útest_apply_multikey_corner“  s
    
rØ   r   TFc              
   C   sŒ   t dddddddddg	tdƒdœƒ}|jd| d� dd	„ ¡}| d
dddddg¡}| r|tjddddddg|jgdd gd�|_t 	||¡ d S )Nr9   rM   rN   rV   )r5   r   r5   r   c                 S   s   | j d d… S r’   rE   r   r   r   r   r   £  r   z'test_apply_chunk_view.<locals>.<lambda>r   rR   rT   rU   r_   )
r   rz   r"   r#   Ztaker   Úfrom_arraysr   r$   r%   )r   r)   r+   r*   r   r   r   Útest_apply_chunk_viewž  s    $ ÿrÚ   c                  C   s`   t ddddddddddg
ddddddddddg
tdddƒdœƒ} |  ddg¡}| d	d
„ ¡ d S )Nr9   rM   r   rV   éÿÿÿÿ)r   Úname2r   r   rÜ   c                 S   s   | j ddd�S )Nr   T©ZinplacerÕ   r   r   r   r   r   ¸  r   z4test_apply_no_name_column_conflict.<locals>.<lambda>)r   rz   r"   r#   )r)   r«   r   r   r   Ú"test_apply_no_name_column_conflict­  s    
ýÿ	rÞ   c                  C   s|   t ddddddgt dddgd¡t dd¡dœƒ} d	d
„ }| jddd� |¡}|  ¡ }t dddgd¡|d< t ||¡ d S )Nr/   r0   r-   r.   rO   rM   ç      @©r©   rO   Úvc                 S   s,   | d }||  ¡  | ¡ |  ¡   | d< | S ©Nrá   Úv2©ÚminÚmax©r^   rá   r   r   r   r±   Ä  s     z#test_apply_typecast_fail.<locals>.fr©   Fr   ç        ç      à?r9   rã   )	r   r‰   ÚtileÚaranger"   r#   r\   r$   r%   )r)   r±   r+   r*   r   r   r   Útest_apply_typecast_fail»  s    
ýÿrì   c               	   C   s¦   t  ddddddgddddddgg¡} tddddddgt ddd	gd¡t dd
¡dœ| d�}dd„ }|jddd� |¡}| ¡ }t dddgd¡|d< t	 
||¡ d S )Nr   r9   rM   rN   r/   r0   r-   r.   rO   rß   rà   r   c                 S   s,   | d }||  ¡  | ¡ |  ¡   | d< | S râ   rä   rç   r   r   r   r±   Ü  s     z%test_apply_multiindex_fail.<locals>.fr©   Fr   rè   ré   rã   )r   rÙ   r   r‰   rê   rë   r"   r#   r\   r$   r%   )r   r)   r±   r+   r*   r   r   r   Útest_apply_multiindex_failÑ  s    &
ýú	rí   c                 C   s4   | j dd„ dd� dd„ ¡}| d }t ||¡ d S )Nc                 S   s   | j S r   rÒ   r   r   r   r   r   ê  r   z#test_apply_corner.<locals>.<lambda>Fr   c                 S   s   | d S r’   r   r   r   r   r   r   ê  r   rM   )r"   r#   r$   r%   )r×   r+   r*   r   r   r   Útest_apply_corneré  s    rî   c                  C   sh   t ddddgddddgddd	d
gdœƒ} dd„ }dd„ }|  d¡ |¡}|  d¡ |¡}t ||¡ d S )Nr²   éÈ   i,  r-   r.   rO   r9   rM   rN   rR   )Úid_fieldÚcategoryr   c                 S   s(   | j d dkr|  ¡ S | | jdk S d S ©Nr   r9   rO   )Úshaper\   rñ   r   r   r   r   Úfilt1û  s    z&test_apply_without_copy.<locals>.filt1c                 S   s$   | j d dkr| S | | jdk S d S rò   )ró   rñ   r   r   r   r   Úfilt2  s    z&test_apply_without_copy.<locals>.filt2rð   r   )r6   rô   rõ   r*   r+   r   r   r   Útest_apply_without_copyï  s    


ýÿrö   Útest_seriesc                 C   s®   t ddgddgddggddgdddgd�}| rr| d¡d }|jd	d
d� dd„ ¡}| ¡ }| ¡ }t ||¡ n8|jdd
d� dd„ ¡}| d¡}| d¡}t ||¡ d S )Nr   ÚpÚoÚXÚYr9   rM   ©r8   r   r   F)Úlevelr   c                 S   s   | S r   r   r   r   r   r   r     r   z<test_apply_with_duplicated_non_sorted_axis.<locals>.<lambda>r   c                 S   s   | S r   r   r   r   r   r   r     r   )	r   r!   r"   r#   Ú
sort_indexr$   r'   rÈ   r%   )r÷   r)   Úserr+   r*   r   r   r   Ú*test_apply_with_duplicated_non_sorted_axis  s      ÿ

r   c                  C   sn   ddddg} ddddg}t ddgd | dœ|d�}t| |d	d
�}dd„ }|jddd�j |¡}t ||¡ d S )Nr9   rM   rN   rR   ZGroup1ZGroup2©r^   r   r   r   ©r   r   c                 S   s"   |   t | j ¡ | j ¡ d ¡¡S rD   )rÊ   r‰   rë   r   rå   ræ   r   r   r   r   Úreindex_helper-  s    z1test_apply_reindex_values.<locals>.reindex_helperr^   Fr   )r   r   r"   r   r#   r$   r'   )rÀ   Úindicesr)   r*   r  r+   r   r   r   Útest_apply_reindex_values#  s    r  c                  C   sn   d} t jjdd| d�}t|t j | ¡ddddg| d	  d
œƒ}|jddd�}dd„ }| |¡}d|ksjt‚d S )NrÌ   r   r²   )ÚsizeÚfooÚbarZbazZquxrR   )r5   Úvalue1Zvalue2r5   Fr   c                 S   s   | d d | d< | S )Nr	  rM   Úvalue3r   ©r~   r   r   r   r±   D  s    z"test_apply_corner_cases.<locals>.fr
  )r‰   rŠ   rÎ   r   r‹   r"   r#   rg   )ÚNÚlabelsr)   r«   r±   r+   r   r   r   Útest_apply_corner_cases5  s    
ýÿ
r  c                  C   sN  t ddgdgd ddgdœƒ} |  dg¡ dd	„ ¡}t | j¡| _|  dg¡ d
d	„ ¡}t |d |d ¡ t dddgdddgt d¡gd dœƒ} dd„ }|  d¡ |¡d }| j	}| j
|_t ||¡ dd„ }t ddddgddddgddddgdd d!d!gd"œƒ}| ¡ }t |j¡|_| d#¡ |¡j}| d#¡ |¡j}t ||¡ d S )$Nr9   rM   z
2017-03-02r  Úinf)ÚNumberÚDateÚStrr  c                 S   s
   | j d S rc   rE   r   r   r   r   r   V  r   z;test_apply_numeric_coercion_when_datetime.<locals>.<lambda>c                 S   s
   | j d S rc   rE   r   r   r   r   r   X  r   r  é
   é   é   Ú3rl   z12:31:22rN   )rv   rX   ÚTc                 S   s   | j d dg S )Nr   rX   rE   r  r   r   r   Úget_B`  s    z8test_apply_numeric_coercion_when_datetime.<locals>.get_Brv   rX   c                 S   sŒ   t dddgtd�}dt| jƒkr>t| | jdk jjd ƒ|d< dt| jƒkrˆt| | jdk jjd ƒ|d< t| | jdk jjd ƒ|d< |S )NÚp1Úp2ZuseTime)r   r   Ústep1r   Ústep2)r   r&   rœ   ÚStater?   ÚMachinerÀ   ÚoTime)Ztoolrs   r   r   r   Úpredictionsi  s    z>test_apply_numeric_coercion_when_datetime.<locals>.predictionsr  r  Ú z2016-09-19 05:24:33z2016-09-19 23:59:04Z23Z36LZ36R)ÚKeyr  r  r  r"  )r   r"   r#   r    Úto_datetimer  r$   r'   Ú	TimestamprX   rv   r   r\   r  r  )r)   r*   r+   r  r   Údf1Údf2r   r   r   Ú)test_apply_numeric_coercion_when_datetimeL  s8    ÿ"ÿ	



üÿr'  c                  C   sŠ   t dddgt d¡gd dœƒ} | j| j | d< |  d¡ d	d
„ ¡}t dddgt dd¡gd t d¡gd dœƒ d¡}t 	||¡ d S )Nrv   rX   rY   z2017-02-01 00:00:00rN   )Úclientidr   Útime_delta_zeror(  c                 S   s   t | j ¡ | j ¡ dœƒS )N)Úclientid_ager   )r   r)  rå   r   )Zddfr   r   r   r   �  s   ÿz?test_apply_aggregating_timedelta_and_datetime.<locals>.<lambda>r   r½   )r(  r*  r   )
r   r‰   Z
datetime64r   r"   r#   Ztimedelta64r!   r$   r%   ©r)   r+   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 dddgdddgdœƒ}|jd	dd� t ||¡ d S )Nrv   r  rX   r  r  rY   é(   é2   ÚNameÚValuez
2020-09-01z
2020-09-05rü   éZ   )r/  r0  TrÝ   )r   r    Ú
date_ranger"   rÇ   r!   r$   r%   )r6   r)   r+   r*   r   r   r   Ú test_apply_groupby_datetimeindexœ  s    "  
ÿr3  c                  C   s®   t ddd„ tdƒD ƒdœƒ} dd„ }dd	„ }| jd
gd� |¡}t ddidgd�}d
|j_| jd
gd� |¡}t tdddƒddœdgd�}d
|j_t ||¡ t ||¡ d S )Nr9   c                 S   s   g | ]}t  ¡ ‘qS r   )r   Únow)rA   Únnr   r   r   rB   ³  s     z'test_time_field_bug.<locals>.<listcomp>r  rP   c                 S   s   t ddiƒS )NrO   rM   ©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éß  r9   rM   ©r.   rO   )r   r   r7  r   r   r   Úfunc_with_date¸  s    z+test_time_field_bug.<locals>.func_with_dater-   ©ZbyrO   rM   r   r:  r;  )	r   rz   r"   r#   r   r   r   r$   r%   )r)   r9  r<  Zdfg_no_conversionZdfg_no_conversion_expectedZdfg_conversionZdfg_conversion_expectedr   r   r   Útest_time_field_bug­  s    r>  c                     s°   t ddddddddddddgddddddddddddgdddd	d
ddd	ddddgdddddddddd
d	dgdœƒ} |  ddg¡} | jddgdd�}dd„ ‰ | ‡ fdd„¡ d S )Nr-   r.   rO   r©   Úegš™™™™™ñ?rM   rN   rR   rS   rT   rQ   r9   gffffff@rV   r  é   é   rU   )Úgroup1Úgroup2Úweightr   rB  rC  T)rý   Úsortc                 S   s   t  | | ¡ d¡}|S )NrN   )r‰   ÚarrayÚrepeat)r   rD  rs   r   r   r   ÚnoddyÔ  s    z7test_gb_apply_list_of_unequal_len_arrays.<locals>.noddyc                    s   ˆ | j | jƒS r   )r   rD  r   ©rH  r   r   r   Þ  r   z:test_gb_apply_list_of_unequal_len_arrays.<locals>.<lambda>)r   r!   r"   r#   )r)   Z
df_groupedr   rI  r   Ú(test_gb_apply_list_of_unequal_len_arraysÇ  s    üÿ
rJ  c                  C   sL   t ddddgddddgdœƒ} dd	„ }|  d
¡ |¡}t ƒ }t ||¡ d S )Nr   r9   rQ   rU   rR   rS   )ÚgroupsZrandom_varsc                 S   s   d S r   r   r   r   r   r   Ú	test_funcæ  s    z.test_groupby_apply_all_none.<locals>.test_funcrK  r   )Ztest_dfrL  r+   r*   r   r   r   Útest_groupby_apply_all_noneá  s
    rM  c            	      C   sð   t ddddgddddgdœƒ} t ddddgddddgdœƒ}dd„ }|  d¡ |¡}| d¡ |¡}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 )Nr9   rM   r   rN   )rK  Úvarsc                 S   s    | j d dk rd S | jddg S )Nr   rM   rÛ   )ró   rF   r   r   r   r   rL  ó  s    z0test_groupby_apply_none_first.<locals>.test_funcrK  r_   r   )r   r"   r#   r   rÙ   r$   r%   )	Ztest_df1Ztest_df2rL  rŽ   r�   Zindex1Úindex2Z	expected1Z	expected2r   r   r   Útest_groupby_apply_none_firstî  s    rP  c                  C   sl   t ddgddgdœƒ} |  d¡}| dd„ ¡}tdgd	tjddgdggdd gd
� d¡d�}t ||¡ d S )Nr   r9   ZfilledÚempty)r   r^   r^   c                 S   s   | | j dk d S )Nr9   r   )r   r]   r   r   r   r     r   z7test_groupby_apply_return_empty_chunk.<locals>.<lambda>r   r_   )r   r   )	r   r"   r#   r   r   Úfrom_productrÅ   r$   r'   )r)   rK  r+   r*   r   r   r   Ú%test_groupby_apply_return_empty_chunk  s    
 ÿþýrS  c                  C   s€   t d ¡ dddgdddgdœƒ} | jd	d
d�}| dd„ ¡}t dddgdddgdœƒ}t ||¡ | dd„ ¡}t ||¡ d S )Nza a br9   rM   rN   rR   rT   rS   rÍ   rv   Fr   c                 S   s   | |   ¡  S r   ©rÇ   r   r   r   r   r     r   z-test_apply_with_mixed_types.<locals>.<lambda>gUUUUUUÕ?gUUUUUUå?çš™™™™™Ù?g333333ã?r/   ©rX   rY   c                 S   s   | |   ¡  S r   rT  r   r   r   r   r     r   )r   Úsplitr"   rÐ   r$   r%   r#   )r)   r~   r+   r*   r   r   r   Útest_apply_with_mixed_types  s     rX  c                  C   sh   t dddgitddgƒd�} |  d¡ dd„ ¡}ttdgƒtdgƒgtddgdd�d�}t ||¡ d S )Nr-   r9   rM   r   c                 S   s   | j S r   r   r  r   r   r   r   !  r   z*test_func_returns_object.<locals>.<lambda>©r   )r   r   r"   r#   r   r$   r'   r+  r   r   r   Útest_func_returns_object  s    (rZ  Úgroup_column_dtlikec                 C   sT   t dg| gdœƒ}| d¡ dd„ ¡}t dgtdgddd�d	gd
�}t ||¡ d S )Nr  rP   r-   c                 S   s   t dgdgd�S )NÚspamé*   r   r6  r   r   r   r   r   2  r   z+test_apply_datetime_issue.<locals>.<lambda>r\  r&   r   r]  r7   ©r   r"   r#   r   r$   r%   )r[  r)   r+   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	¡ |¡d
 }tdgtjdgd	d�d
d�}t ||¡ d S )Nz2015-02-24 00:00:00)r   r9   rM   rN   rR   zsome UA stringzanother UA stringZ17661101)ÚdayZ	userAgentÚuserIdc                 S   s   t dd„ |  ¡ D ƒƒS )Nc                 S   s    i | ]\}}||  ¡ jd  “qS )r   )Zvalue_countsr   )rA   rO   r(   r   r   r   Ú
<dictcomp>W  s      zYtest_apply_series_return_dataframe_groups.<locals>.most_common_values.<locals>.<dictcomp>)r   ÚitemsrG   r   r   r   Úmost_common_valuesV  s    zEtest_apply_series_return_dataframe_groups.<locals>.most_common_valuesr`  ra  z
2015-02-24rY  r  )	r   r    r$  r"   r#   r   ÚDatetimeIndexr$   r'   )Ztdfrd  r+   r*   r   r   r   Ú)test_apply_series_return_dataframe_groups:  s:    ûûûñÿ  ÿrf  rñ   c                 C   sà   ddgd }| rLt j|dddgd�}t jdddgdddgdd�}dd	d
g}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¡}| 
d¡ dd„ ¡}t ||¡ |jjddgksÜt‚d S )Nr9   rM   rS   rN   )Ú
categoriesrX   )rg  r   g      4@g      9@rè   rY  r  é   )rY   r½   r   r  )rv   rX   rY   r½   rv   c                 S   s   |   ¡ S r   rT  r   r   r   r   r   s  r   z-test_apply_multi_level_name.<locals>.<lambda>)r    ZCategoricalÚCategoricalIndexr   r   r‰   rë   rœ   rz   r!   r"   r#   r$   r%   r   r   rg   )rñ   r.   Zexpected_indexÚexpected_valuesr*   r)   r+   r   r   r   Útest_apply_multi_level_name`  s&     ÿ"ÿþrk  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dddddgd�} |  d¡ dd„ ¡j}tt 	d¡t
t
tjt
gdddddgd�}t ||¡ d S )Nià  r9   ZredZdarkrn   r:  ZgreenZstormyrM   Ú9iÞ  ZblueZbrightrN   ro   iÝ  ZcalmrR   ZpotatoZobservationÚcolorZmoodZ	intensityZscorer7   c                 S   s
   | j d S rc   rE   r  r   r   r   r   ƒ  r   z;test_groupby_apply_datetime_result_dtypes.<locals>.<lambda>zdatetime64[ns]r   )r   Zfrom_recordsr    r$  r"   r#   rI   r   r‰   r   r&   Zint64r$   r'   )r6   r+   r*   r   r   r   Ú)test_groupby_apply_datetime_result_dtypesx  s    üù	þrn  r   ÚabcZ2020r½   )r³   Úfreq)r-   r   )r-   r9   )r.   r   c                 C   sF   t dddgdddgdœ| d�}|jddd� d	d
„ ¡}t ||¡ d S )Nr9   rM   r   r  r   r^   Fr   c                 S   s   | S r   r   r   r   r   r   r   —  r   z8test_apply_index_has_complex_internals.<locals>.<lambda>r   )r   r)   r+   r   r   r   Ú&test_apply_index_has_complex_internals‹  s    rq  zfunction, expected_valuesc                 C   s
   | j  ¡ S r   )r   Úto_listr   r   r   r   r   ž  r   c                 C   s   t | j ¡ ƒS r   )Úsetr   rr  r   r   r   r   r   Ÿ  r   c                 C   s   t | j ¡ ƒS r   )Útupler   rr  r   r   r   r   r      r   )r   r9   ©rM   rN   c                 C   s   t t| j ¡ ƒƒS r   )ÚdictÚ	enumerater   rr  r   r   r   r   r   ¢  r   c                 C   s   dd„ t | j ¡ ƒD ƒS )Nc                 S   s   g | ]\}}||i‘qS r   r   )rA   ÚnÚir   r   r   rB   ¦  s     z<lambda>.<locals>.<listcomp>)rw  r   rr  r   r   r   r   r   ¦  r   c                 C   sN   t ddddgdgd�}| d¡ | ¡}t|tddgdd�d�}t ||¡ d S )Nrv   rX   rK  r7   rY  r   ©r   r"   r#   r   r   r$   r'   )Úfunctionrj  r)   r+   r*   r   r   r   Ú1test_apply_function_returns_non_pandas_non_scalar›  s    r|  c                  C   sr   dd„ } t ddddgdddtjgd	œƒ}| d
¡ | ¡}tddgdgtjggtdddgd
d�d�}t ||¡ d S )Nc                 S   s   | d j  ¡ S )NrX   )rÀ   Úflattenr]   r   r   r   Úfctµ  s    z4test_apply_function_returns_numpy_array.<locals>.fctr-   r.   Únoner9   rM   rN   rº   rv   r/   r0   r1   rY  r   )	r   r‰   rÄ   r"   r#   r   r   r$   r'   )r~  r)   r+   r*   r   r   r   Ú'test_apply_function_returns_numpy_array³  s      ÿr€  r{  c                 C   s   | j S r   r   ©Zgrr   r   r   r   Á  r   c                 C   s   | j d d S rD   r   r�  r   r   r   r   Á  r   c                 C   s‚   t ddddddddddg
dgd�}| d¡ | ¡}ttdddd	gƒtdddd
gƒtddgƒgtdddgdd�d�}t ||¡ d S )Nr9   rM   rN   Úidr7   r   rR   rU   rV   rS   rT   rQ   rY  r   rz  )r{  r)   r+   r*   r   r   r   Ú test_apply_function_index_returnÁ  s    "(þrƒ  c                  C   sh   t ddddddgddddddgd	œƒ} | jd
dd� dd„ ¡}t dddgdddgd	œƒ}t ||¡ d S )Nr�   rW   r‚   r9   rM   rR   rS   rT   rƒ   r„   Fr‡   c                 S   s   |   ¡ S r   )rŒ   r   r   r   r   r   Õ  r   zAtest_apply_function_with_indexing_return_column.<locals>.<lambda>r1   r2   r   r+  r   r   r   Ú/test_apply_function_with_indexing_return_columnÍ  s    þÿþÿr„  Úudfc                 C   s   |   ¡ S r   rx   r   r   r   r   r   á  r   c                 C   s   |   ¡  dd„ ¡S )Nc                 S   s   | d S rD   r   )Úyr   r   r   r   á  r   z<lambda>.<locals>.<lambda>)r\   rÃ   r   r   r   r   r   á  r   c                 C   sŒ   t ddgddgdœƒ}|jd| d� |¡}|jj|j| d� |¡}| rh|jjdksVt‚|jjdksˆt‚n |jjdksxt‚|jjdksˆt‚d S )Nr-   r.   r9   rM   rº   rv   r   )r   r"   r#   rX   rv   r   Znlevelsrg   )r   r…  r)   Z	df_resultZseries_resultr   r   r   Útest_apply_result_typeß  s    	r‡  c                  C   sZ   t dddgdddgdœƒ} | jddd� dd	„ ¡}| jddd� d
d	„ ¡}t ||¡ d S )NrM   r9   rN   rº   rv   Fr   c                 S   s   | S r   r   r   r   r   r   r   ø  r   z4test_result_order_group_keys_false.<locals>.<lambda>c                 S   s   |   ¡ S r   rx   r   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	œƒ}|jd
dd� dd„ ¡}|jd
dd� dd„ ¡}t ||¡ d S )Nz
2001-01-01rM   z
2001-01-02z
2001-01-03ÚUTC)ÚtzrN   rT   )r   r†  Útr   Fr   c                 S   s   | ddg   ¡ S ©Nr   r†  rx   rG   r   r   r   r     r   z1test_apply_with_timezones_aware.<locals>.<lambda>c                 S   s   | ddg   ¡ S rŒ  rx   rG   r   r   r   r     r   )	r    re  r   rœ   rz   r"   r#   r$   r%   )ÚdatesZindex_no_tzZindex_tzr%  r&  rŽ   r�   r   r   r   Útest_apply_with_timezones_awareý  s    
  rŽ  c              	   C   sÀ   t ddddddgddddddgd	d
ddddgdœƒ}t ddgddgddgdœtddgdd�d�}|jdd�}| t¡}t ||¡ |jdd�}t| |ƒ}t|| ƒ|Ž }| t¡}t ||¡ d S )Néc   éX   r9   rM   rN   rR   rS   rT   r  r  r  r-  r.  é<   ©r-   r.   rO   i  i)  é   é–   r-   rY  r   r=  )	r   r   r"   r#   rÇ   r$   r%   r
   Úgetattr)Zreduction_funcr)   r*   Úgrpr+   ÚargsÚ_r   r   r   Ú;test_apply_is_unchanged_when_other_methods_are_called_first  s$    ýÿþ


r™  c               	   C   sä   t ddddgtdddƒtdddƒtdddƒtdddƒgddddgd	œtd
dddgdd�d�} |  ddg¡}| dd„ ¡}| jdddg }| ¡ }t |dddg ¡|_	|j
dd�}t ||¡ |j	jd D ]}t|ƒtksÊt‚qÊd S )Nr-   r.   iä  r9   r  rM   rN   rR   rÍ   r²   ée   éf   ég   ÚidxrY  r   rv   rX   c                 S   s
   |   d¡S rD   r“   r   r   r   r   r   :  r   zRtest_apply_with_date_in_multiindex_does_not_convert_to_timestamp.<locals>.<lambda>r   r7   )r   r   r   r"   r#   rF   r�   r   Z
from_framer   rÅ   r$   r%   ÚlevelsÚtyperg   )r)   r–  r+   r*   Úvalr   r   r   Ú@test_apply_with_date_in_multiindex_does_not_convert_to_timestamp(  s(    




ü
ø
õr¡  c                  C   sz   t tj ddg¡t ddgddgg¡d�} | jjddd	� d
d„ ¡}| jddd	� dd„ ¡}t 	||j¡ t 	|| ¡ d S )NrT   rR   rv   rX   r9   rM   r7   r   ©r<   rý   c                 S   s   | j ddd�S )Nr   r¢  ©Z	droplevelr   r   r   r   r   R  r   zDtest_apply_by_cols_equals_apply_by_rows_transposed.<locals>.<lambda>c                 S   s   | j ddd�S )Nr9   r   r¢  r£  r   r   r   r   r   T  r   )
r   r‰   rŠ   r   rR  r  r"   r#   r$   r%   )r)   Zby_rowsZby_colsr   r   r   Ú2test_apply_by_cols_equals_apply_by_rows_transposedF  s    þÿr¤  Údropnac                 C   sx   t dddddgdtjtjddgdœtd	ƒd
�}|jd| dd� dd„ ¡}| rT| ¡ n|jdddddg }t 	||¡ d S )Nr9   rM   rN   rR   rS   r-   r.   )Úcolr^   Zxxyxzr   r^   F)r¥  r   c                 S   s   | S r   r   r   r   r   r   r   e  r   z5test_apply_dropna_with_indexed_same.<locals>.<lambda>r   )
r   r‰   rÄ   rœ   r"   r#   r¥  rF   r$   r%   )r¥  r)   r+   r*   r   r   r   Ú#test_apply_dropna_with_indexed_sameZ  s    þû r§  zas_index, expected)r   r•   )rM   rM   r_   r   c                 C   sT   t ddddgddddgddddgdœƒ}|jddg| d� dd„ ¡}t ||¡ d S )	Nr9   rM   r’  r-   r.   r‡   c                 S   s   dS rD   r   r   r   r   r   r   ~  r   z5test_apply_as_index_constant_lambda.<locals>.<lambda>)r   r"   r#   r$   Zassert_equal)rˆ   r*   r)   r+   r   r   r   Ú#test_apply_as_index_constant_lambdaj  s    (r¨  c                  C   s†   t dddddgdddd	d
gdddddgdœtdƒd�} |  d¡ dd„ ¡}ttddƒtjdddddgdd gd�dd�}t ||¡ d S )Nr9   rM   rN   rR   rS   rT   rU   rQ   rV   r   rÍ   r   rY   c                 S   s
   | j  ¡ S r   )rv   rþ   r   r   r   r   r   ˆ  r   z(test_sort_index_groups.<locals>.<lambda>r—   r•   r˜   ru  r–   r_   rv   r  )	r   rz   r"   r#   r   r   r›   r$   r'   r+  r   r   r   Útest_sort_index_groups‚  s    (þ ÿûr©  c                  C   sX   t tjdddd�tdƒtdƒdœƒ} | j| j| jjjgdd� 	d	d
„ ¡}t
 || ¡ d S )Nz
2010-01-01Z12HrS   )rp  r³   r™   )r   ÚvalsÚletFr   c                 S   s   | j dd … S rc   rE   r   r   r   r   r   ž  r   z;test_positional_slice_groups_datetimelike.<locals>.<lambda>)r   r    r2  rz   rœ   r"   r«  r   Údtr#   r$   r%   rJ   r   r   r   Ú)test_positional_slice_groups_datetimelike“  s    ýÿ ÿþr­  c                  C   sx   t dddgdddgdddg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 )Nr-   r.   r9   rM   rN   rR   rT   rS   rÍ   rv   rX   rY   c                 S   s   |   t¡ ¡ |  ¡  S r   )ZastypeÚfloatræ   rå   r   r   r   r   r   ¦  r   z7test_groupby_apply_shape_cache_safety.<locals>.<lambda>r/   rè   r0   rV  rY  r   r^  )r)   Úgbr+   r*   r   r   r   Ú%test_groupby_apply_shape_cache_safety¢  s    "
 ÿr°  c                 C   sf   t ddddgddddgddtjtjgdœƒ}|jd| d�}| dd	„ ¡}| d
d	„ ¡}t ||¡ d S )Nr9   rM   r   rS   )r–  r†  Úzr–  )r¥  c                 S   s   |   dd¡S )Nr9   r±  )Únlargest)Zgrp_dfr   r   r   r   µ  r   ztest_apply_na.<locals>.<lambda>c                 S   s   | j ddd� d¡S )Nr±  F)Z	ascendingr9   )rÈ   r”   r   r   r   r   r   ¶  r   )r   r‰   rÄ   r"   r#   r$   r%   )r¥  r)   Zdfgrpr+   r*   r   r   r   Útest_apply_na®  s    &ÿr³  c                  C   s    t ddddgddddgtjddddgdd�dœƒ d	d
g¡ dd„ ¡} t ddtjddd�gddtjddd�ggd	d
dgtjddgd	d
gd�d�}t | |¡ d S )Nr9   rM   r!  rN   rR   r(   )Úunitr’  r-   r.   c                 S   s
   | j d S ©NrÛ   rE   rG   r   r   r   r   Å  r   z8test_apply_empty_string_nan_coerce_bug.<locals>.<lambda>rO   )r9   r!  )rM   r!  r_   rü   )	r   r    r#  r"   r#   r   r›   r$   r%   )r+   r*   r   r   r   Ú&test_apply_empty_string_nan_coerce_bugº  s"    

ýÿùøÿ&ýr¶  Úindex_valuesr/   r0   r1   c                 C   sj   t dddgdddgdœt| ƒd�}t d	d
ddgitdddgdd�d�}| d¡ dd„ ¡}t ||¡ d S )NZaaZa2Úa3r9   rM   rN   rP   r   Úb_meanr0   r1   r/   r-   rY  c                 S   s   t | d  ¡ gdgd�S )Nr.   r¹  r   )r   rŒ   rG   r   r   r   r   à  r   z0test_apply_index_key_error_bug.<locals>.<lambda>)r   r   r"   r#   r$   r%   )r·  r+   r*   r   r   r   Útest_apply_index_key_error_bugÏ  s     þû ÿü
ÿrº  zarg,idxgš™™™™™¹?g333333Ó?gš™™™™™É?rU  c                 C   s8   t d| i|d�}|jddd� dd„ ¡}t ||¡ d S )Nr¦  r   Fr   c                 S   s   | S r   r   r   r   r   r   r     r   z5test_apply_nonmonotonic_float_index.<locals>.<lambda>r   )Úargr�  r*   r+   r   r   r   Ú#test_apply_nonmonotonic_float_indexå  s    +r¼  zargs, kwargsÚnumeric_onlyc                 C   s6   |   d¡}|jd|ž|Ž}|jdd�}t ||¡ d S )Nrv   rÇ   T)r½  )rÇ   )r"   r#   rÇ   r$   r%   )r)   r—  Úkwargsr¯  r+   r*   r   r   r   Útest_apply_str_with_args  s    
r¿  r   Z	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 )	Nr9   rM   rY  r-   Fr   c                 S   s   | S r   r   r   r   r   r   r   "  r   z1test_result_name_when_one_group.<locals>.<lambda>)r   r"   r#   r$   r'   )r   rÿ   r+   r*   r   r   r   Útest_result_name_when_one_group  s    rÀ  z
method, opr#   c                 C   s
   | j d S rµ  )rÀ   ©r¯  r   r   r   r   +  r   c                 C   s   | d j d S )Nr.   r   rE   rÁ  r   r   r   r   ,  r   )ÚaggZ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 )	NrP   r-   Tr   r.   r:   r   )r   r   r   )r   r"   r•  r   r   r$   r'   )ÚmethodÚopZempty_dfr¯  r^   r+   r*   r   r   r   Útest_empty_df(  s    
   ÿrÆ  )^r   r   Úior   Znumpyr‰   ZpytestZpandasr    r   r   r   r   r	   Zpandas._testingZ_testingr$   Zpandas.tests.groupbyr
   r,   rK   rL   ÚmarkZparametrizerz   rœ   ri   ru   r{   r€   r�   r¡   r¬   r¹   r¼   r¾   rÂ   rÆ   rÏ   rÑ   rØ   rÚ   rÞ   rì   rí   rî   rö   r   r  r  r'  r,  r3  r>  rJ  rM  rP  rS  rX  rZ  Útodayr   r_  rf  rk  rn  ri  Zinterval_rangeZperiod_ranger›   rq  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   Ú<module>   s�  )*$(.ýÿø
(ðùì
,üþ	
%#


5	þ
&

üþ	
þþøþ

þ	
 ÿþ ÿþùþ


ýýúýýúýýúçþ)


	

ûþ
