a
    oÝEbï”  ã                   @   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	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,„ d-d,„ d.d,„ d/d,„ g¡d0d1„ ƒZ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)dJdK„ Z*dLdM„ Z+dNdO„ Z,dPdQ„ Z-dRdS„ Z.ej dTdUdVg¡dWdX„ ƒZ/dYdZ„ Z0d[d\„ Z1d]d^„ Z2d_d`„ Z3dadb„ Z4dcdd„ Z5dedf„ Z6dgdh„ Z7didj„ Z8dkdl„ Z9dmdn„ Z:dodp„ Z;ej dqe  <¡ e  <¡  ¡ e  <¡  =¡ g¡drds„ ƒZ>dtdu„ Z?ej dvdVdUg¡dwdx„ ƒZ@dydz„ ZAej d{e Bed|ƒ¡e Cd d}¡ejDd~d}dd€�e Eg d�¢¡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–„ ZJd—d˜„ ZKd™dš„ ZLd›dœ„ ZMd�dž„ ZNej dŸdUdVg¡d d¡„ ƒZOej d¢dVe	g d£¢g d¤¢ge
g d¥¢ePd¦�d!�gdUeddgejEd§d¨gdd©gdª�d«�gg¡d¬d­„ ƒZQd®d¯„ ZRd°d±„ ZSd²d³„ ZTej dŸdUdVg¡d´dµ„ ƒZUd¶d·„ ZVej d¸g d¢g d¹¢g¡dºd»„ ƒZWej d¼g d¢g d½¢gg d¢g d¾¢gg d¿¢g dÀ¢gg¡dÁdÂ„ ƒZXdS )Ãé    )ÚdateÚdatetime)ÚStringION)Ú	DataFrameÚIndexÚ
MultiIndexÚSeriesÚbdate_range)Ú
Int64Indexc                  C   sÂ   d} t jt| ƒd g d¢ddggd�}| d¡}| |jj¡ ¡ }| |jj¡ dd„ ¡}t	 
||¡ t jt| ƒd g d¢d	�}tg d
¢tdd�}tg 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.40649)r   ÚtimeÚvaluer   r   )ÚheaderÚnamesZparse_datesÚ	date_timec                 S   s   |   ¡ S ©N©Úidxmax©Úx© r   ú_/home/ja/django-apps/lartica_env/lib/python3.9/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Úname)z00:00ú02:00r   ©Úindexc                 S   s   | d | d   ¡  S )Nr   r   r   r   r   r   r   r   9   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 g d¢g d¢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©Úar-   Úbr.   r-   ©ç      ð?ç       @ç      @ç      @g      @©ÚkeyÚdatar5   r6   ©Úcolumnsé   Úfloat64r%   ©ÚaxisÚkeysc                 S   s   g | ]}t |ƒ‘qS r   ©Ústr©Ú.0r   r   r   r   Ú
<listcomp>E   r   z&test_apply_trivial.<locals>.<listcomp>©r<   c                    s   ˆ j dd … S ©Nr9   ©Úilocr   ©r(   r   r   r   F   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                     sn   t g d¢g d¢dœddgd�‰ tjˆ ˆ gddd	gd
�} ˆ jdd„ ˆ jD ƒdd� ‡ fdd„¡}t || ¡ d S )Nr,   r/   r4   r5   r6   r7   r9   r:   r%   r;   c                 S   s   g | ]}t |ƒ‘qS r   r>   r@   r   r   r   rB   S   r   z+test_apply_trivial_fail.<locals>.<listcomp>rC   c                    s   ˆ S r   r   r   rG   r   r   r   S   r   z)test_apply_trivial_fail.<locals>.<lambda>)r   r   rH   r!   rI   r"   r#   r$   rJ   r   rG   r   Útest_apply_trivial_failL   s    þ(rL   zdf, group_names)r9   r9   r9   é   é   )r-   r-   r-   r.   Úc©r-   r.   ©r9   rM   rN   ©r   r   r9   r9   )r   r9   r   r9   r9   r-   )r9   r9   r9   rM   rM   r9   r9   rM   é   rM   ©r9   rM   rN   r9   rM   rN   )é   é   é   é   rS   é	   )r-   ÚtwoZ
aaabbbcccc)
rN   rU   rN   rW   rV   rM   r9   rY   rV   rU   )
rU   r   rM   rM   rM   rX   rS   rW   rM   rS   )r-   ÚBÚC©r-   r.   rO   )rM   rM   rN   r7   )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 …= |   d¡ |¡ ˆ |ksNJ ‚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   ra   rc   r   r   Úf_nocopyˆ   s    z1test_group_apply_once_per_group.<locals>.f_nocopyc                    s   ˆ   | j¡ dS ©Nr   re   ra   rc   r   r   Úf_scalar�   s    z1test_group_apply_once_per_group.<locals>.f_scalarc                    s   ˆ   | j¡ d S r   re   ra   rc   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   ra   rc   r   r   Úf_constant_df—   s    z6test_group_apply_once_per_group.<locals>.f_constant_dfr-   )r!   r"   )r(   Zgroup_namesrd   rf   rh   ri   rj   Úfuncr   rc   r   Útest_group_apply_once_per_groupX   s    '
rl   c                 C   sV   d}t g d¢g d¢dœg d¢d�}| d¡ dd„ ¡ |  ¡ j d	¡}||ksRJ ‚d S )
NrM   )r   r   r   r   r9   r9   r9   r9   )Ú0Ú2Ú4Ú6Ú8Ú10Z12Z14)Úgroup_by_columnZtest_columnr   rs   c                 S   s   t dƒS )NÚfunction_called)ÚprintrG   r   r   r   r   ±   r   z2test_group_apply_once_per_group2.<locals>.<lambda>rt   )r   r!   r"   Z
readouterrÚoutÚcount)Zcapsysr)   r(   r*   r   r   r   Ú test_group_apply_once_per_group2£   s    þûrx   c                  C   sV   t g d¢tdƒdœƒ} dd„ }dd„ }|  d¡ |¡}|  d¡ |¡}t ||¡ d S )	N)r   r   r9   rN   )ÚAr.   c                 S   s   | S r   r   ra   r   r   r   Úslow¿   s    z,test_apply_fast_slow_identical.<locals>.slowc                 S   s   |   ¡ S r   ©r`   ra   r   r   r   ÚfastÂ   s    z,test_apply_fast_slow_identical.<locals>.fastry   )r   Úranger!   r"   r#   r$   )r(   rz   r|   Zfast_dfZslow_dfr   r   r   Útest_apply_fast_slow_identical¸   s    r~   rk   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©Údeepr{   r   r   r   r   r   Ð   r   c                 C   s   | j dd�S )NTr   r{   r   r   r   r   r   Ñ   r   c                 C   s<   t g d¢g d¢g d¢dœƒ}| d¡ | ¡}t ||¡ d S )N©r9   rM   rM   rM   ©r9   rM   rN   rU   )rV   rW   rX   rS   )Úgr-   r.   rƒ   ©r   r!   r"   r#   r$   )rk   r(   r*   r   r   r   Ú5test_groupby_apply_identity_maybecopy_index_identicalË   s    r…   c                  C   s–   t tj 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 )NrW   ©ÚonerZ   rZ   Úthreer‡   rZ   ©Úfoo1Zfoo2c                 S   s   | S r   r   r   r   r   r   r   è   r   z-test_apply_with_mixed_dtype.<locals>.<lambda>r9   rC   Úc1)r9   rM   rW   rW   rS   r1   Ú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   s4  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d„ ¡j}| dd„ ¡j}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�} | jddd� dd„ ¡j}t ||¡ d S )N)r.   r.   r-   rO   r-   r.   )r9   rM   r9   r9   rN   r9   rW   )Zitem_idÚuser_idr   r—   Tr�   FrM   )r   r9   rM   rU   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   rU   )©r9   r   ©r9   rM   )rM   r9   )rN   rU   rc   Úabcder9   rN   rU   rV   r   r   c                 S   s   | S r   r   r   r   r   r   r     r   )r   r}   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@J ‚| |¡}|jjd	ksZJ ‚| |¡}|jjd
kstJ ‚d S )Nry   r[   c                 S   s   |   ¡ }d|j_|S ©NÚstat)Údescriber   r   ©rb   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)ry   r[   r©   )ry   r[   N)r!   r"   r   r   )Zthree_groupÚgroupedr¬   r®   r°   r*   r•   Zresult3r   r   r   Ú test_apply_concat_preserve_names  s    



r²   c                  C   sb   dd„ } t ddd�}ttj d¡|d�}| dd„ ¡}| | ¡}t|tƒsNJ ‚t	 
|j|j¡ d S )	Nc                 S   sN   t jdd�� t  | ¡}W d   ƒ n1 s,0    Y  t| | |  ¡  |dœƒS )NÚignore)Úinvalid)r   ZdemeanedÚlogged)r�   ZerrstateÚlogr   r’   )Zpiecerµ   r   r   r   Úf=  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   G  r   z,test_apply_series_to_frame.<locals>.<lambda>)r	   r   r�   r�   r‘   r!   r"   Ú
isinstancer   r#   r    r   )r·   ZdrÚtsr±   r*   r   r   r   Útest_apply_series_to_frame<  s    
r¾   c                 C   s4   |   ddg¡d  t¡}|jjd d… dks0J ‚d S )Nry   r[   r\   rM   ©ry   r[   )r!   r"   r­   r   r   ©r(   r*   r   r   r   Ú test_apply_series_yield_constantN  s    rÁ   c                 C   sl   |   ddg¡ t¡}t|tƒs"J ‚|jd u s0J ‚|   ddg¡ddg  t¡}t|tƒsZJ ‚|jd u shJ ‚d S )Nry   r[   r\   ÚD)r!   r"   r­   r¼   r   r   rÀ   r   r   r   Útest_apply_frame_yield_constantS  s    rÃ   c                 C   sH   |   ddg¡}| t¡}| ¡ d }t |j|j¡ t |j|j¡ d S )Nry   r[   r\   )	r!   r"   r­   rw   r#   r    r   Úassert_numpy_array_equalÚvalues©r(   r±   r*   r)   r   r   r   Útest_apply_frame_to_series^  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 )Nry   r[   Fr�   r\   r7   rÂ   )r!   r"   r­   rw   Úrenamer�   ÚnanÚdropr#   r    r   rÄ   rÅ   rÆ   r   r   r   Ú)test_apply_frame_not_as_index_column_namef  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†J ‚d S )Nc                 S   s   |   d¡d  ¡  ¡ d d… S )Nr[   r\   rM   )r!   ÚsumÚsort_valuesra   r   r   r   Útransq  s    z-test_apply_frame_concat_series.<locals>.transc                    s*   |   ˆ  | j¡d ¡}| ¡  ¡ d d… S )Nr[   rM   )r!   Úreindexr   rÌ   rÍ   )rb   r±   rG   r   r   Útrans2t  s    z.test_apply_frame_concat_series.<locals>.trans2r   rV   éè  ©ry   r[   r\   ry   r\   F)Zcheck_names)
r   r�   r�   Úrandintr‘   r!   r"   r#   r&   r   )rÎ   rÐ   r*   r£   r   rG   r   Útest_apply_frame_concat_seriesp  s    
ýÿrÔ   c                 C   s:   |   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>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 )Nry   éûÿÿÿ©rÍ   ra   r   r   r   r·   �  s    z%test_apply_multikey_corner.<locals>.f)r!   r"   r#   r$   Úloc)Útsframer±   r·   r*   r5   rb   r   r   r   Útest_apply_multikey_corner�  s
    
rÝ   c                  C   sL   t g d¢tdƒdœƒ} | jddd� dd„ ¡}|  g d	¢¡}t ||¡ d S )
N)	r9   r9   r9   rM   rM   rM   rN   rN   rN   rY   )r5   r   r5   F©Z
group_keysc                 S   s   | d d… S r˜   r   r   r   r   r   r   œ  r   z'test_apply_chunk_view.<locals>.<lambda>)r   r9   rN   rU   rW   rX   )r   r}   r!   r"   Ztaker#   r$   ©r(   r*   r)   r   r   r   Útest_apply_chunk_view˜  s    rà   c                  C   s@   t g d¢g d¢tdddƒdœƒ} |  ddg¡}| dd	„ ¡ d S )
N)
r9   r9   r9   r9   r9   r9   rM   rM   rM   rM   )
r   r   r   r9   r9   r9   r   r   r9   r9   rY   éÿÿÿÿ)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   r}   r!   r"   )r(   r±   r   r   r   Ú"test_apply_no_name_column_conflict¡  s    
ýÿ	rä   c                  C   sl   t g d¢t g d¢d¡t dd¡dœƒ} dd„ }|  d	¡ |¡}|  ¡ }t g d
¢d¡|d< t ||¡ d S )N©r0   r0   r0   r1   r1   r1   r]   rM   r0   ç      @©r¯   rO   Úvc                 S   s,   | d }||  ¡  | ¡ |  ¡   | d< | S ©Nrè   Úv2©ÚminÚmax©rb   rè   r   r   r   r·   ¸  s     z#test_apply_typecast_fail.<locals>.fr¯   ©ç        g      à?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  g d¢g d¢g¡} tg d¢t g d¢d¡t dd¡dœ| d	�}d
d„ }| d¡ |¡}| ¡ }t g d¢d¡|d< t	 
||¡ d S )N)r   r   r   r9   r9   r9   rT   rå   r]   rM   r0   ræ   rç   r   c                 S   s,   | d }||  ¡  | ¡ |  ¡   | d< | S ré   rë   rî   r   r   r   r·   Ð  s     z%test_apply_multiindex_fail.<locals>.fr¯   rï   rê   )r   Úfrom_arraysr   r�   rñ   rò   r!   r"   r`   r#   r$   )r   r(   r·   r*   r)   r   r   r   Útest_apply_multiindex_failÅ  s    
ýú	rõ   c                 C   s0   |   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>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   s\   t g d¢g d¢g d¢dœƒ} dd„ }dd„ }|  d	¡ |¡}|  d	¡ |¡}t ||¡ d S )
N)r¸   r¸   éÈ   i,  )r-   r.   rO   rO   r‚   )Ú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_seriesTFc                 C   s¦   t ddgddgddggddgg d¢d�}| rn| d¡d }|jdd	� d
d„ ¡}| ¡ }| ¡ }t ||¡ n4| d¡ dd„ ¡}| d¡}| d¡}t ||¡ d S )Nr   ÚpÚoÚXÚY)r9   rM   rM   ©r8   r   r   )Úlevelc                 S   s   | S r   r   r   r   r   r   r     r   z<test_apply_with_duplicated_non_sorted_axis.<locals>.<lambda>c                 S   s   | S r   r   r   r   r   r   r     r   )	r   r    r!   r"   Ú
sort_indexr#   r&   rÍ   r$   )rÿ   r(   Zserr*   r)   r   r   r   Ú*test_apply_with_duplicated_non_sorted_axis   s     ÿ

r  c                  C   sb   g d¢} g d¢}t ddgd | dœ|d�}t| |dd	�}d
d„ }| d¡j |¡}t ||¡ d S )Nr‚   ©r9   r9   rM   rM   ZGroup1ZGroup2rM   ©rb   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_helperrb   )r   r   r!   r   r"   r#   r&   )rÅ   Úindicesr(   r)   r  r*   r   r   r   Útest_apply_reindex_values  s    r  c                  C   sf   d} t jjdd| d�}t|t j | ¡g d¢| d  dœƒ}| d¡}d	d
„ }| |¡}d|v sbJ ‚d S )NrÑ   r   r¸   )Úsize)ÚfooÚbarZbazZquxrU   )r5   Úvalue1Zvalue2r5   c                 S   s   | d d | d< | S )Nr  rM   Úvalue3r   ©rƒ   r   r   r   r·   8  s    z"test_apply_corner_cases.<locals>.fr  )r�   r�   rÓ   r   r‘   r!   r"   )ÚNÚlabelsr(   r±   r·   r*   r   r   r   Útest_apply_corner_cases)  s    
ýÿ

r  c                  C   s:  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 g d¢g d¢t d¡gd dœƒ} dd„ }|  d¡ |¡d }| j	}| j
|_t ||¡ dd„ }t g d¢g d¢g d¢g d¢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 rg   rE   r   r   r   r   r   J  r   z;test_apply_numeric_coercion_when_datetime.<locals>.<lambda>c                 S   s
   | j d S rg   rE   r   r   r   r   r   L  r   r  )é
   é   é   )r  Ú3ro   z12:31:22rN   )ry   r[   ÚTc                 S   s   | j d dg S )Nr   r[   rE   r  r   r   r   Úget_BT  s    z8test_apply_numeric_coercion_when_datetime.<locals>.get_Bry   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 r†t| | jdk jjd ƒ|d< t| | jdk jjd ƒ|d< |S )	N)Úp1Úp2ÚuseTime)r   r   Ústep1r   r!  Ústep2r"  r#  )r   r%   r¢   ÚStater?   ÚMachinerÅ   ÚoTime)Ztoolrv   r   r   r   Úpredictions]  s    z>test_apply_numeric_coercion_when_datetime.<locals>.predictions)r[   r[   ry   ry   )r$  r%  r$  r%  )Ú z2016-09-19 05:24:33r*  z2016-09-19 23:59:04)Z23Z36LÚ36Rr+  )ÚKeyr&  r(  r'  r,  )r   r!   r"   r   Úto_datetimer  r#   r&   Ú	Timestampr[   ry   r   r`   r(  r!  )r(   r)   r*   r   r)  Údf1Údf2r   r   r   Ú)test_apply_numeric_coercion_when_datetime@  s8    ÿÿ	üÿr1  c                  C   s†   t g d¢t d¡gd dœƒ} | j| j | d< |  d¡ dd„ ¡}t g d¢t d	d
¡gd t d¡gd dœƒ d¡}t 	||¡ d S )NrÒ   z2017-02-01 00:00:00rN   )Úclientidr   Útime_delta_zeror2  c                 S   s   t | j ¡ | j ¡ dœƒS )N)Úclientid_ager   )r   r3  rì   r   )Zddfr   r   r   r   �  s   ÿz?test_apply_aggregating_timedelta_and_datetime.<locals>.<lambda>r   rÂ   )r2  r4  r   )
r   r�   Z
datetime64r   r!   r"   Ztimedelta64r    r#   r$   rß   r   r   r   Ú-test_apply_aggregating_timedelta_and_datetimeu  s$    þÿ
ÿýÿúr5  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 )Nry   r  r[   r  r  r\   é(   é2   ÚNameÚValuez
2020-09-01z
2020-09-05r  rÒ   )r  r7  éZ   )r8  r9  Trã   )r   r   Ú
date_ranger!   rÌ   r    r#   r$   )r6   r(   r*   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	„ }| 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   §  r   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   r@  r   r   r   Úfunc_with_date¬  s    z+test_time_field_bug.<locals>.func_with_dater-   ©ZbyrO   rM   r   rC  rD  )	r   r}   r!   r"   r   r   r   r#   r$   )r(   rB  rE  Zdfg_no_conversionZdfg_no_conversion_expectedZdfg_conversionZdfg_conversion_expectedr   r   r   Útest_time_field_bug¡  s    rG  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)r-   r-   r-   r.   r.   r.   r-   r-   r-   r.   r.   r.   )rO   rO   r¯   r¯   r¯   ÚerO   rO   r¯   r¯   r¯   rH  )gš™™™™™ñ?rM   rN   rU   rV   rW   rM   rU   rW   rS   r9   rM   )gffffff@rS   rY   r  é   é   rS   rX   rW   rV   rU   rN   )Úgroup1Úgroup2Úweightr   rK  rL  T)r  Úsortc                 S   s   t  | | ¡ d¡}|S )NrN   )r�   ÚarrayÚrepeat)r   rM  rv   r   r   r   ÚnoddyÉ  s    z7test_gb_apply_list_of_unequal_len_arrays.<locals>.noddyc                    s   ˆ | j | jƒS r   )r   rM  r   ©rQ  r   r   r   Ó  r   z:test_gb_apply_list_of_unequal_len_arrays.<locals>.<lambda>)r   r    r!   r"   )r(   Z
df_groupedr   rR  r   Ú(test_gb_apply_list_of_unequal_len_arrays»  s    üÿ
rS  c                  C   sD   t g d¢g d¢dœƒ} dd„ }|  d¡ |¡}t ƒ }t ||¡ d S )NrR   )rS   rX   rU   rV   )ÚgroupsZrandom_varsc                 S   s   d S r   r   r   r   r   r   Ú	test_funcÛ  s    z.test_groupby_apply_all_none.<locals>.test_funcrT  r„   )Ztest_dfrU  r*   r)   r   r   r   Útest_groupby_apply_all_noneÖ  s
    rV  c            	      C   sà   t g d¢g d¢dœƒ} t g d¢g d¢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 )N)r9   r9   r9   rM   )r   r9   rM   rN   )rT  Úvarsr�   c                 S   s    | j d dk rd S | jddg S )Nr   rM   rá   )rû   rF   r   r   r   r   rU  è  s    z0test_groupby_apply_none_first.<locals>.test_funcrT  r9   r   rM   rc   rN   r   )r   r!   r"   r   rô   r#   r$   )	Ztest_df1Ztest_df2rU  r”   r•   Zindex1Úindex2Z	expected1Z	expected2r   r   r   Útest_groupby_apply_none_firstã  s    rY  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   rb   rb   c                 S   s   | | j dk d S )Nr9   r   )r   ra   r   r   r   r   û  r   z7test_groupby_apply_return_empty_chunk.<locals>.<lambda>r   rc   )r   r   )	r   r!   r"   r   r   Úfrom_productrÊ   r#   r&   )r(   rT  r*   r)   r   r   r   Ú%test_groupby_apply_return_empty_chunk÷  s    
ÿþýr\  c                  C   st   t d ¡ g d¢g d¢dœƒ} |  d¡}| dd„ ¡}t g d¢g d	¢d
œƒ}t ||¡ | dd„ ¡}t ||¡ d S )Nza a brQ   ©rU   rW   rV   rÒ   ry   c                 S   s   | |   ¡  S r   ©rÌ   r   r   r   r   r     r   z-test_apply_with_mixed_types.<locals>.<lambda>)gUUUUUUÕ?gUUUUUUå?r9   )çš™™™™™Ù?g333333ã?r0   ©r[   r\   c                 S   s   | |   ¡  S r   r^  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    
rb  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     ÿrd  Ú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   r?  r   r   r   r   r   )  r   z+test_apply_datetime_issue.<locals>.<lambda>rf  r%   r   rg  r7   ©r   r!   r"   r   r#   r$   )re  r(   r*   r)   r   r   r   Útest_apply_datetime_issue  s    
ÿri  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   rU   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>N  r   zYtest_apply_series_return_dataframe_groups.<locals>.most_common_values.<locals>.<dictcomp>)r   Ú	iteritemsrG   r   r   r   Úmost_common_valuesM  s    zEtest_apply_series_return_dataframe_groups.<locals>.most_common_valuesrj  rk  z
2015-02-24rc  r
  )	r   r   r.  r!   r"   r   ÚDatetimeIndexr#   r&   )Ztdfrn  r*   r)   r   r   r   Ú)test_apply_series_return_dataframe_groups1  s6    ûûûñÿÿrp  rù   c                 C   sÐ   ddgd }| r<t j|g d¢d�}t jddgg d¢dd�}ntddgdd�}tt d	¡|ttd	ƒƒttd	ƒƒd
œƒ 	ddg¡}| 
d¡ dd„ ¡}tddgddgdœ|d�}t ||¡ |jjddgksÌJ ‚d S )Nr9   rM   rV   rQ   )Ú
categoriesr[   )rq  r   rc  r  )ry   r[   r\   rÂ   ry   c                 S   s   |   ¡ S r   r^  r   r   r   r   r   c  r   z-test_apply_multi_level_name.<locals>.<lambda>r  é   )r\   rÂ   r   )r   ZCategoricalÚCategoricalIndexr   r   r�   rò   r¢   r}   r    r!   r"   r#   r$   r   r   )rù   r.   Zexpected_indexr(   r*   r)   r   r   r   Útest_apply_multi_level_nameW  s    "ÿþrt  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¡ dd„ ¡j}tt 	d¡t
t
tjt
gg d¢d�}t ||¡ d S )Nià  r9   ÚredZdarkrq   rC  ÚgreenZstormyrM   Ú9iÞ  ÚblueÚbrightrN   rr   iÝ  ZcalmrU   Úpotato)ZobservationÚcolorZmoodZ	intensityZscorer7   r{  c                 S   s
   | j d S rg   rE   r  r   r   r   r   t  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_dtypesi  s    üù	þr|  r   ÚabcrN   Z2020rÂ   )r¹   Úfreq))r-   r   )r-   r9   )r.   r   c                 C   s>   t g d¢g d¢dœ| d�}| d¡ dd„ ¡}t ||¡ d S )N)r9   r9   rM   )r   r9   r   r	  r   rb   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    r  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   r€  r   r   r   r   r   �  r   c                 C   s   t | j ¡ ƒS r   )Útupler   r€  r   r   r   r   r   ‘  r   )r   r9   ©rM   rN   c                 C   s   dd„ t | j ¡ ƒD ƒS )Nc                 S   s   i | ]\}}||“qS r   r   ©rA   ÚnÚir   r   r   rl  “  r   z<lambda>.<locals>.<dictcomp>©Ú	enumerater   r€  r   r   r   r   r   “  r   c                 C   s   dd„ t | j ¡ ƒD ƒS )Nc                 S   s   g | ]\}}||i‘qS r   r   r„  r   r   r   rB   —  r   z<lambda>.<locals>.<listcomp>r‡  r   r   r   r   r   —  r   c                 C   sJ   t g d¢dgd�}| d¡ | ¡}t|tddgdd�d�}t ||¡ d S )N)ry   ry   r[   r[   rT  r7   ry   r[   rc  r   ©r   r!   r"   r   r   r#   r&   )ÚfunctionZexpected_valuesr(   r*   r)   r   r   r   Ú1test_apply_function_returns_non_pandas_non_scalarŒ  s    r‹  c                  C   sl   dd„ } t g d¢dddtjgdœƒ}| d¡ | ¡}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Å   Úflattenra   r   r   r   Úfct¦  s    z4test_apply_function_returns_numpy_array.<locals>.fct)r-   r-   r.   Únoner9   rM   rN   r¿   ry   r0   r1   r2   )r-   r.   rŽ  rc  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   sh   t g d¢dgd�}| d¡ | ¡}ttg d¢ƒtg d¢ƒtddgƒgtg d¢dd	�d
�}t ||¡ d S )N)
r9   rM   rM   rM   r9   rM   rN   r9   rN   r9   Úidr7   )r   rU   rX   rY   )r9   rM   rN   rV   rW   rS   rQ   rc  r   r‰  )rŠ  r(   r*   r)   r   r   r   Ú test_apply_function_index_return²  s     þr“  c                  C   s‚   t g d¢g d¢dœƒ} tjtdd��( | jddd� d	d
„ ¡}W d   ƒ n1 sR0    Y  t g d¢g d¢dœƒ}t ||¡ d S )Nr†   )r9   rM   rU   rU   rV   rW   r‰   zSelect only valid)ÚmatchrŠ   Fr�   c                 S   s   |   ¡ S r   )r’   r   r   r   r   r   Ç  r   zAtest_apply_function_with_indexing_return_column.<locals>.<lambda>)r‡   rˆ   rZ   )r2   r3   r3   )r   r#   Zassert_produces_warningÚFutureWarningr!   r"   r$   rß   r   r   r   Ú/test_apply_function_with_indexing_return_column¾  s    þÿ6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   rW   )r   ÚyÚtr   FrÞ   c                 S   s   | ddg   ¡ S ©Nr   r™  r{   rG   r   r   r   r   Ô  r   z1test_apply_with_timezones_aware.<locals>.<lambda>c                 S   s   | ddg   ¡ S r›  r{   rG   r   r   r   r   Õ  r   )	r   ro  r   r¢   r}   r!   r"   r#   r$   )ÚdatesZindex_no_tzZindex_tzr/  r0  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	d
gdœtddgdd�d�}|jdd�}| t¡}t ||¡ |jdd�}dg|gdœ | g ¡}t|| ƒ|Ž }| t¡}t ||¡ d S )N)éc   rž  rž  éX   rŸ  rŸ  )r9   rM   rN   rU   rV   rW   )r  r  r  r6  r7  é<   r]   i  i)  é   rW   é–   r   rŸ  rž  r-   rc  r   rF  r   )ZnthZcorrwith)	r   r   r!   r"   rÌ   r#   r$   ÚgetÚ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 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d„ ¡}| jg d¢ }| ¡ }t |g d¢ ¡|_	|j
d	d�}t ||¡ |j	jd D ]}t|ƒtu sºJ ‚qºd S )N)r-   r-   r-   r.   iä  r9   r  rM   r‚   rÒ   )r¸   ée   éf   ég   Úidxrc  r   ry   r[   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   rM   rN   )ry   r[   r¬  r7   )r   r   r   r!   r"   rF   r“   r   Z
from_framer   rÊ   r#   r$   ZlevelsÚtype)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 )NrW   rU   ry   r[   r9   rM   r7   r   ©r<   r  c                 S   s   | j ddd�S )Nr   r°  ©Z	droplevelr   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   #  r   )
r   r�   r�   r   r[  r  r!   r"   r#   r$   )r(   Zby_rowsZby_colsr   r   r   Ú2test_apply_by_cols_equals_apply_by_rows_transposed  s    þÿr²  Údropnac                 C   sj   t g d¢dtjtjddgdœtdƒd�}|jd| d� d	d
„ ¡}| rL| ¡ n|jg d¢ }t 	||¡ d S )N©r9   rM   rN   rU   rV   r-   r.   )Úcolrb   Zxxyxzr   rb   ©r³  c                 S   s   | S r   r   r   r   r   r   r   4  r   z5test_apply_dropna_with_indexed_same.<locals>.<lambda>)r   rN   r9   rM   rU   )
r   r�   rÉ   r¢   r!   r"   r³  rF   r#   r$   )r³  r(   r*   r)   r   r   r   Ú#test_apply_dropna_with_indexed_same)  s    þûr·  zas_index, expected)r9   r9   r9   )rM   rM   r9   )r-   r.   N)r   r›   )rM   rM   r.   rc   r   c                 C   sH   t g d¢g d¢g d¢dœƒ}|jddg| d� dd„ ¡}t ||¡ d S )	Nr  )r9   r9   r9   r9   r]   r-   r.   r�   c                 S   s   dS rD   r   r   r   r   r   r   M  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_lambda9  s    r¸  c                  C   sn   t g d¢g d¢g d¢dœtdƒd�} |  d¡ dd	„ ¡}ttd
dƒtjg d¢dd gd�dd�}t ||¡ d S )Nr´  )rW   rX   rS   rY   r   )r9   r9   r9   rM   rM   rÒ   rV   r   r\   c                 S   s
   | j  ¡ S r   )ry   r  r   r   r   r   r   W  r   z(test_sort_index_groups.<locals>.<lambda>r9   rW   )r�   r›   rž   rƒ  rœ   rc   ry   r
  )	r   r}   r!   r"   r   r   r¡   r#   r&   rß   r   r   r   Útest_sort_index_groupsQ  s    þÿûr¹  c                  C   sT   t tjdddd�tdƒtdƒdœƒ} |  | j| jjjg¡ 	dd„ ¡}t
 || ¡ d S )	Nz
2010-01-01Z12HrV   )r~  r¹   rŸ   )r   ÚvalsÚletc                 S   s   | j dd … S rg   rE   r   r   r   r   r   l  r   z;test_positional_slice_groups_datetimelike.<locals>.<lambda>)r   r   r;  r}   r¢   r!   r»  r   Údtr"   r#   r$   rJ   r   r   r   Ú)test_positional_slice_groups_datetimelikeb  s    ýÿÿr½  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)r-   r-   r.   rQ   r]  rÒ   ry   r[   r\   c                 S   s   |   t¡ ¡ |  ¡  S r   )ZastypeÚfloatrí   rì   r   r   r   r   r   u  r   z7test_groupby_apply_shape_cache_safety.<locals>.<lambda>r0   rð   r1   r`  r-   r.   rc  r   rh  )r(   Úgbr*   r)   r   r   r   Ú%test_groupby_apply_shape_cache_safetyq  s    
ÿrÀ  c                 C   s^   t g d¢g d¢ddtjtjgdœƒ}|jd| d�}| dd	„ ¡}| d
d	„ ¡}t ||¡ d S )Nr  )r9   r   rM   rV   r9   rM   )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 g d¢g d¢tjg d¢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g d¢tjddgddgd�d�}t | |¡ d S )Nr  )r*  r*  r*  r*  r‚   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>r9   r*  rM   rU   )r9   r*  )rM   r*  rc   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_values)r0   r1   r2   c                 C   sb   t g d¢g d¢dœt| ƒd�}t dg d¢itg d¢dd	�d�}| d¡ d
d„ ¡}t ||¡ d S )N)ÚaaÚa2Úa3rQ   rP   r   Úb_mean)r1   r2   r0   )rÈ  rÉ  rÇ  r-   rc  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,idx)çš™™™™™¹?ç333333Ó?çš™™™™™É?)rÌ  rÎ  rÍ  )r9   rU   rN   )rÌ  r_  rÎ  c                 C   s4   t d| i|d�}| d¡ dd„ ¡}t ||¡ d S )Nrµ  r   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Ð  )Yr   r   Úior   Únumpyr�   ZpytestZpandasr   r   r   r   r   r	   Zpandas._testingZ_testingr#   Zpandas.core.apir
   r+   rK   rL   ÚmarkZparametrizer}   r¢   rl   rx   r~   r…   r–   r§   r²   r¾   rÁ   rÃ   rÇ   rË   rÔ   rÖ   rÝ   rà   rä   ró   rõ   rö   rþ   r  r  r  r1  r5  r<  rG  rS  rV  rY  r\  rb  rd  Útodayr   ri  rp  rt  r|  rs  Zinterval_rangeZperiod_ranger¡   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þ
&

üþ	
þþøþ


ÿþÿþùþ


úúúçþ)