U
    ¾mœdz¡  ã                   @   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   úX/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/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)Z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   ©rZ   r   r\   r^   r   r   Úf_nocopyŒ   s    z1test_group_apply_once_per_group.<locals>.f_nocopyc                    s   ˆ   | j¡ dS ©Nr   r`   r\   r^   r   r   Úf_scalar‘   s    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 )Nr9   rP   )rZ   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_   ra   rc   rd   re   Úfuncr   r^   r   Útest_group_apply_once_per_group\   s    '
rh   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   ro   Fr   c                 S   s   t dƒS )NÚfunction_called)ÚprintrG   r   r   r   r   µ   r   z2test_group_apply_once_per_group2.<locals>.<lambda>rp   )r   r"   r#   Z
readouterrÚoutÚcountrf   )Zcapsysr*   r)   r+   r   r   r   Ú test_group_apply_once_per_group2¦   s    þûÿrt   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>.fastru   Fr   )r   Úranger"   r#   r$   r%   )r)   rv   rx   Zfast_dfZslow_dfr   r   r   Útest_apply_fast_slow_identical½   s    rz   rg   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©Údeeprw   r   r   r   r   r   Õ   r   c                 C   s   | j dd�S )NTr{   rw   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%   )rg   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   ry   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 )Nru   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)ru   rX   r¢   )ru   rX   N)r"   r#   r   r   rf   )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   rf   Ú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 )Nru   rX   rY   rM   ©ru   rX   )r"   r#   r¦   r   r   rf   ©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 )Nru   rX   rY   ÚD)r"   r#   r¦   rµ   r   rf   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 )Nru   rX   rY   )	r"   r#   r¦   rs   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 )Nru   rX   Fr†   rY   r7   r¼   )r"   r#   r¦   rs   Ú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   éè  ©ru   rX   rY   ru   rY   F)Zcheck_names)r   rˆ   r‰   ÚrandintrŠ   r"   r#   r$   r'   r   rf   )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 )Nru   éûÿÿÿ©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   ry   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   ry   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#   rf   )Ú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 rb   rE   r   r   r   r   r   V  r   z;test_apply_numeric_coercion_when_datetime.<locals>.<lambda>c                 S   s
   | j d S rb   rE   r   r   r   r   r   X  r   r  é
   é   é   Ú3rk   z12:31:22rN   )ru   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_Bru   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toolrr   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   ru   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 )Nru   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 )Nru   r  rX   r  r  rY   é(   é2   ÚNameÚValuez
2020-09-01z
2020-09-05rû   éZ   )r.  r/  TrÜ   )r   r    Ú
date_ranger"   rÆ   r!   r$   r%   )r6   r)   r+   r*   r   r   r   Ú test_apply_groupby_datetimeindexœ  s    "  
ÿr2  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   r6  r   r   r   Úfunc_with_date¸  s    z+test_time_field_bug.<locals>.func_with_dater-   ©ZbyrO   rM   r   r9  r:  )	r   ry   r"   r#   r   r   r   r$   r%   )r)   r8  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   rA  rB  T)rü   Úsortc                 S   s   t  | | ¡ d¡}|S )NrN   )rˆ   ÚarrayÚrepeat)r   rC  rr   r   r   r   ÚnoddyÔ  s    z7test_gb_apply_list_of_unequal_len_arrays.<locals>.noddyc                    s   ˆ | j | jƒS r   )r   rC  r   ©rG  r   r   r   Þ  r   z:test_gb_apply_list_of_unequal_len_arrays.<locals>.<lambda>)r   r!   r"   r#   )r)   Z
df_groupedr   rH  r   Ú(test_gb_apply_list_of_unequal_len_arraysÇ  s    üÿ
rI  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_funcrJ  r~   )Ztest_dfrK  r+   r*   r   r   r   Útest_groupby_apply_all_noneá  s
    rL  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   )rJ  Úvarsc                 S   s    | j d dk rd S | jddg S )Nr   rM   rÚ   )rò   rF   r   r   r   r   rK  ó  s    z0test_groupby_apply_none_first.<locals>.test_funcrJ  r^   r   )r   r"   r#   r   rØ   r$   r%   )	Ztest_df1Ztest_df2rK  r�   rŽ   Zindex1Úindex2Z	expected1Z	expected2r   r   r   Útest_groupby_apply_none_firstî  s    rO  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)   rJ  r+   r*   r   r   r   Ú%test_groupby_apply_return_empty_chunk  s    
 ÿþýrR  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Ì   ru   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   rS  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     rW  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    (rY  Ú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   r5  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%   )rZ  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_  r`  z
2015-02-24rX  r  )	r   r    r#  r"   r#   r   ÚDatetimeIndexr$   r'   )Ztdfrc  r+   r*   r   r   r   Ú)test_apply_series_return_dataframe_groups:  s:    ûûûñÿ  ÿre  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   )rf  r   g      4@g      9@rç   rX  r  é   )rY   r¼   r   r  )ru   rX   rY   r¼   ru   c                 S   s   |   ¡ S r   rS  r   r   r   r   r   s  r   z-test_apply_multi_level_name.<locals>.<lambda>)r    ZCategoricalÚCategoricalIndexr   r   rˆ   rê   r›   ry   r!   r"   r#   r$   r%   r   r   rf   )rð   r.   Zexpected_indexÚexpected_valuesr*   r)   r+   r   r   r   Útest_apply_multi_level_name`  s&     ÿ"ÿþrj  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   ÚredZdarkrm   r9  ÚgreenZstormyrM   Ú9iÞ  ÚblueÚbrightrN   rn   iÝ  ZcalmrR   ÚpotatoZobservationÚcolorZmoodZ	intensityZscorer7   c                 S   s
   | j d S rb   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    üù	þrr  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    ru  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   rv  r   r   r   r   r   Ÿ  r   c                 C   s   t | j ¡ ƒS r   )Útupler   rv  r   r   r   r   r      r   )r   r9   ©rM   rN   c                 C   s   t t| j ¡ ƒƒS r   )ÚdictÚ	enumerater   rv  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>)r{  r   rv  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 )Nru   rX   rJ  r7   rX  r   ©r   r"   r#   r   r   r$   r'   )Úfunctionri  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¹   ru   r/   r0   r1   rX  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   ©Ú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   rX  r   r~  )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   rw   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¹   ru   r   )r   r"   r#   rX   ru   r   Znlevelsrf   )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¹   ru   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   rw   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‹  rw   rG   r   r   r   r     r   z1test_apply_with_timezones_aware.<locals>.<lambda>c                 S   s   | ddg   ¡ S r‘  rw   rG   r   r   r   r     r   )	r    rd  r   r›   ry   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-   rX  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   ÚidxrX  r   ru   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%   ZlevelsÚtyperf   )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   ru   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   rQ  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   )ru   rý   r   r   r   r   r   ˆ  r   z(test_sort_index_groups.<locals>.<lambda>r–   r”   r—   ry  r•   r^   ru   r  )	r   ry   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   )rt  r²   r˜   )r   ÚvalsÚletFr   c                 S   s   | j dd … S rb   rE   r   r   r   r   r   ž  r   z;test_positional_slice_groups_datetimelike.<locals>.<lambda>)r   r    r1  ry   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Ì   ru   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   rU  rX  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-   rX  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š™™™™™É?rT  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 )Nru   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   rX  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   Únumpyrˆ   ZpytestZpandasr    r   r   r   r   r	   Zpandas._testingZ_testingr$   Zpandas.tests.groupbyr
   r,   rK   rL   ÚmarkZparametrizery   r›   rh   rt   rz   r   r�   r    r«   r¸   r»   r½   rÁ   rÅ   rÎ   rÐ   r×   rÙ   rÝ   rë   rì   rí   rõ   rÿ   r  r  r&  r+  r2  r=  rI  rL  rO  rR  rW  rY  Útodayr   r^  re  rj  rr  rh  Zinterval_rangeZperiod_rangerš   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   r   r   r   Ú<module>   s�  )*$(.ýÿø
(ðùì
,üþ	
%#


5	þ
&

üþ	
þþøþ

þ	
 ÿþ ÿþùþ


ýýúýýúýýúçþ)


	

ûþ
