a
    é´«jAn  ã                   @   s6  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	m
Z
 d dlmZ dd„ Zej dddg¡dd	„ ƒZej d
d dg¡dd„ ƒZej dddg¡dd„ ƒZej dddg¡ej dedddejgdœƒeddgiedgdd�d�fedejgdœed�eddgiedgdd�d�fg¡dd„ ƒƒZdd„ Zdd„ Zdd „ Zd!d"„ Zd#d$„ Zd%d&„ Zd'd(„ Zd)d*„ Zd+d,„ Zd-d.„ Z d/d0„ Z!ej d1d2ge	d3d4d5�d6gd7œd2ge	d3d4d5�d6gd7œd2ge	d3d4d5�d6gd7œfg d8¢e	d9d:d5�e	d3d4d5�e	d;d<d5�gg d=¢d7œd2d>ge	d9d:d5�e	d3d4d5�gd6d?gd7œd2d>ge	d;d<d5�e	d3d4d5�gd@d?gd7œfg¡dAdB„ ƒZ"ej dCde	dDdEd5�dgde	dFdEd5�dGgg¡dHdI„ ƒZ#ej dJej$dKdLgdMd�ej$d6d?gdNd�e %dOdPg¡ej&d6d?gdQdR�g¡ej dSg dT¢¡dUdV„ ƒƒZ'dWdX„ Z(ej dYdZdd gfdZd g fdZd6d d?gfdZd[g d\¢fd]dd6gfd]d g fd]d6d6d?gfd]d[g d\¢fg¡ej d^dg d2gd>gd2d>gg¡ej d_dKdLg¡d`da„ ƒƒƒZ)ej dbdZdd gfdZd g fdZd6d d?gfdZd[g d\¢fd]dd6gfd]d g fd]d6d6d?gfd]d[g d\¢fg¡dcdd„ ƒZ*dedf„ Z+dgdh„ Z,didj„ Z-ej dkg dl¢¡dmdn„ ƒZ.ej dkg dl¢¡dodp„ ƒZ/dqdr„ Z0dsdt„ Z1ej dg du¢¡dvdw„ ƒZ2ej dxe3dd@d?ƒg dy¢ge3ddzƒg d{¢ge3dd?ƒe3dzdƒgg d|¢gd d6e3dzdƒgg d|¢gg¡d}d~„ ƒZ4dd€„ Z5d�d‚„ Z6dƒd„„ Z7d…d†„ Z8d‡dˆ„ Z9d‰dŠ„ Z:ej d‹ddg dŒ¢e;d�ƒfdd6d d@ge;dŽƒfdd�g dŒ¢e;d�ƒfddg d�¢e;d‘ƒfd6dg d’¢e;d“ƒfd6dd6ge;d>ƒfddd?d”ge;d•ƒfdd?d”ge;d–ƒfg¡ej dd—d
g¡d˜d™„ ƒƒZ<ej =dš¡d›dœ„ ƒZ>d�dž„ Z?ej dŸdGdGgdGd gf¡ej dkg d¡¢¡d¢d£„ ƒƒZ@ej d¤e	d¥ƒe	d¦ƒfd§g¡d¨d©„ ƒZAdS )ªé    N)Ú	DataFrameÚIndexÚ
MultiIndexÚSeriesÚ	TimestampÚisnac                 C   s  |   d¡}| ¡ }| jddgg d¢f }tddgdd�|_| ¡ }t ||¡ | d¡}| jddg }t ||¡ | 	¡ }| jdd	gg d¢f }tddgdd�|_t ||¡ | d
¡}| j
dd	g }t ||¡ | d¡}| j
ddg }t ||¡ |d  ¡  |d  	¡  |d  d¡ |  ¡ } tj| j| d dkdf< |   d¡}t|d  ¡ d ƒ�s`J ‚t|d  	¡ d ƒ�szJ ‚t|d  d¡j
d ƒ�s˜J ‚tdtjgddgddggddgd�} |   d¡}| ¡ }| j
ddg  d¡}t ||¡ | j
ddg }|jddd�}t ||¡ d S )NÚAé   r   )ÚBÚCÚDÚbarÚfoo©Únameé   é   éÿÿÿÿé   é   r
   é   é   ©ÚcolumnsÚany©Údropna)ÚgroupbyÚfirstÚlocr   ÚindexÚ
sort_indexÚtmÚassert_frame_equalÚnthÚlastÚilocÚcopyÚnpÚnanr   r   Ú	set_index)ÚdfÚgroupedr   Úexpectedr$   r%   ÚgÚresult© r0   úd/var/www/html/laksyashrimp/venv/lib/python3.9/site-packages/pandas/tests/groupby/methods/test_nth.pyÚtest_first_last_nth   sH    




$
r2   Úmethodr   r%   c                 C   sŠ   t g d¢ddd|gdœƒ d¡}t|| ƒƒ }| dkr>ddg}nddg}tj||d jd	�}tddgdd
�}t d|i|d�}t ||¡ d S )N©r	   r	   r   r   r	   r   r   ©ÚaÚbr6   r   r7   ©Údtyper   ©r    )	r   r   Úgetattrr(   Úarrayr9   r   r"   r#   )r3   Únulls_fixtureÚgroupsr/   ÚvaluesÚidxr-   r0   r0   r1   Útest_first_last_with_na_objectB   s     
rA   r    r   c                 C   sb   t g d¢ddd|gdœƒ}| d¡}| | ¡}| dkrD|jddg n|jddg }t ||¡ d S )Nr4   r	   r   r   r5   r6   r   ©r   r   r$   r&   r"   r#   )r    r=   r+   r>   r/   r-   r0   r0   r1   Útest_nth_with_na_objectT   s
    

$rC   c                 C   s>   t  dgd gdœ¡}|jddd�}t|| ƒƒ }t ||¡ d S )Nr6   ©ÚidÚvaluerE   F©Úas_index)r   Ú	from_dictr   r;   r"   r#   )r3   r+   r>   r/   r0   r0   r1   Útest_first_last_with_None^   s    rJ   zdf, expectedr6   r   rD   rF   rE   r   r:   r8   c                 C   s"   t | d¡| ƒƒ }t ||¡ d S )NrE   )r;   r   r"   r#   )r3   r+   r-   r/   r0   r0   r1   Ú"test_first_last_with_None_expandedi   s    rK   c                  C   s
  t g d¢g d¢tj d¡ d¡tjtj d¡ d¡dd�dœƒ} d| d	< d
| d< |  d¡}| ¡ }| jd
dgg d¢f }t	ddgdd�|_
| ¡ }t ||¡ | ¡ }| jddgg d¢f }t	ddgdd�|_
| ¡ }t ||¡ | d
¡}| jddg }t ||¡ d S )N)r   r   r   r   r   r   r   r   )ÚonerL   ÚtwoÚthreerM   rM   rL   rN   r   é   Zfloat32r8   ©r   r
   r   r   TÚEr	   ÚFr   r   )r
   r   r   rQ   rR   r   r   r   r   r   r   )r   r(   ÚrandomÚdefault_rngZstandard_normalr<   r   r   r   r   r    r!   r"   r#   r%   r$   r&   )r+   r,   r   r-   r%   r$   r0   r0   r1   Útest_first_last_nth_dtypes}   s.    üÿ

rU   c                  C   sX   t tdƒƒ} |  d¡ ttdƒ| dd�}|jdks6J ‚|jdd� ¡ }|jdksTJ ‚d S )	Né
   é	   é   ZIntCol)Údatar    r   Úint64r   ©Úlevel)ÚlistÚrangeÚappendr   r9   r   r   )r@   ÚserÚfr0   r0   r1   Útest_first_last_nth_dtypes2œ   s    
rb   c                  C   s†   t dgtd gtd�dœƒ} |  d¡}|  d¡j}t |j ¡ |¡ t |j 	¡ |¡ | j}t |j 
d¡|¡ t |j 
d¡|¡ d S )Nr   r8   )rY   ÚnansrY   r   r   )r   r   Úobjectr   r*   rc   r"   Úassert_series_equalr   r%   r$   )r+   r,   r-   r0   r0   r1   Útest_first_last_nth_nan_dtype¦   s    
rf   c               
   C   s~   t tdƒddgtdƒddgdddgd	d
dgiƒ} |  d¡ ¡ }t g d¢gttdƒtdƒd	gƒtdgdd�d�}t ||¡ d S )Nz2012-01-01 00:00:00r6   r7   z2012-01-02 00:00:00ÚcÚdr   ÚeZaaaara   r.   )r6   rg   ra   z
2012-01-01z
2012-01-02r   ©r   r    )r   r   r   r   r   r"   r#   )Útestr/   r-   r0   r0   r1   Útest_first_strings_timestamps´   s    üÿýrl   c                  C   s´  t dtjgddgddggddgd�} |  d¡}t | d¡| jdd	g ¡ t | d¡| jdg ¡ t | d	¡| jg  ¡ t | d
¡| jdd	g ¡ t | d¡| jdg ¡ t | d¡| jg  ¡ t 	|j
 d¡| j
jdd	g ¡ t 	|j
 d¡| j
jdg ¡ t |dg  d¡| dg jdd	g ¡ t |jddd�| jdd	g ¡ t |jd
dd�| jdd	g ¡ t |jddd�| jd d… ¡ t |jd	dd�| jd d… ¡ d S )Nr	   r   r   r   r   r
   r   r   r   r   éþÿÿÿéýÿÿÿr   r   r   )r   r(   r)   r   r"   r#   r$   r&   r   re   r
   ©r+   Úgbr0   r0   r1   Útest_nthÇ   s    $
 (   rq   c               
   C   s¦   t ddddddœddddddœddd	d
ddœddddddœdœƒ ddg¡} | jddd� d¡}| jdg }t ||¡ | jddd� d¡}| jg  }t ||¡ d S )NÚgreenÚred©r   r	   r   r   r   ZhamÚeggsZporkgþI|î»ø?gçoB!²¿gHk:!4Àgn‹2d’Ý?gC8fÙ“Àà?g£W”†â?gV¿Òùð,ï¿gN'ÙêrŠð?gÃGÄ”H"é¿gû­�(	‰æ?)ÚcolorÚfoodrM   rL   rv   rw   r   F)r\   rH   r   r   r   )r   r*   r   r$   r&   r"   r#   r   ©r+   r/   r-   r0   r0   r1   Ú	test_nth2Ü   s2    ûûöÿí
ry   c                  C   sü   t tj d¡ ddd¡dd�} | d }| d }| |¡ ¡ }| |¡ dd	„ ¡}tj	||d
d� |j
dksnJ ‚|j
dks|J ‚||dk jd }|jd |ks J ‚|jd |ks²J ‚tjtdd��( |j|d
d�jddd� W d   ƒ n1 sî0    Y  d S )Nr   r	   rV   )éd   r   rZ   r8   r   c                 S   s
   | j d S )Nr   )r&   )Úxr0   r0   r1   Ú<lambda>  ó    ztest_nth3.<locals>.<lambda>F)Zcheck_nameszFor a DataFrame©Úmatch)ÚsortTr   )r   r(   rS   rT   Zintegersr   r   Úapplyr"   re   r   r&   ÚpytestÚraisesÚ
ValueErrorr$   )r+   r`   rp   r-   Z	expected2Úvr0   r0   r1   Ú	test_nth3ý   s    r†   c                  C   s^   t dtjgddgddggddgd�} |  d¡}|jjdd	d
�}| jjddg }t ||¡ d S )Nr	   r   r   r   r   r
   r   r   Úallr   r   )	r   r(   r)   r   r
   r$   r&   r"   re   )r+   rp   r/   r-   r0   r0   r1   Ú	test_nth4  s
    $
rˆ   c                  C   s0  t dtjgddgddgddgddggddgd	�} |  d¡}t | d
¡| jd
dg ¡ t | d
g¡| jd
dg ¡ t | d
dg¡| jg d¢ ¡ t | d
dg¡| jg d¢ ¡ t | g d¢¡| jg d¢ ¡ t | g d¢¡| jg d¢ ¡ t | dg¡| jdg ¡ t | ddg¡| jg  ¡ d S )Nr	   r   r   r   r   r   r   r
   r   r   )r   r	   r   r   r   )r   r   r   r   ©r   r	   r   rt   )r   r	   r   r   )	r   r(   r)   r   r"   r#   r$   r&   r   ro   r0   r0   r1   Ú	test_nth5  s    0
    rŠ   c                 C   s‚   t jddd| d�}td|ddgd�}|jj|jjg}|j|d	d
� g d¢¡}t  g d¢¡ 	| ¡}tdddg|d�}t
 ||¡ d S )Nz4/1/2014z	6/30/2014r
   )ÚstartÚendÚfreqÚunitr	   r6   r7   ©r    r   FrG   )r   r   rm   r   )z2014/4/1z2014/4/4z	2014/4/29z	2014/4/30z2014/5/1z2014/5/6z	2014/5/29z	2014/5/30z2014/6/2z2014/6/5z	2014/6/27z	2014/6/30rj   )ÚpdÚ
date_ranger   r    ÚyearÚmonthr   r$   Úto_datetimeÚas_unitr"   r#   )rŽ   Zbusiness_datesr+   Úkeyr/   Zexpected_datesr-   r0   r0   r1   Útest_nth_bdays*  s    ÿÿñr—   c                 C   s6   |   ddg¡}| d¡}| jg d¢ }t ||¡ d S )Nr   r
   r   ©r   r   r   r   )r   r$   r&   r"   r#   ©Zthree_groupr,   r/   r-   r0   r0   r1   Útest_nth_multi_grouperF  s    
rš   z#data, expected_first, expected_lastr   z2012-02-01 14:00:00z
US/Central©Útzr	   ©rE   Útimer   )r   r
   r   z2012-01-01 13:00:00zAmerica/New_Yorkz2012-03-01 12:00:00zEurope/London©r	   r   r   r
   r   r   c                 C   sÌ   t | ƒ}|jddd� ¡ }t |ƒ}g d¢}t || || ¡ |jddd�d  ¡ }t ||ddg ¡ |jddd� ¡ }t |ƒ}g d¢}t || || ¡ |jddd�d  ¡ }t ||ddg ¡ d S )NrE   FrG   r�   rž   )r   r   r   r"   r#   r%   )rY   Zexpected_firstZexpected_lastr+   r/   r-   Úcolsr0   r0   r1   Útest_first_last_tzO  s    6r¡   zmethod, ts, alphaz
2013-01-01ú
US/Easternz
2013-01-02r7   c           	      C   s¨   t tdƒƒ d¡}tjddd|d�}tg d¢||dœƒ}t| d	¡| ƒƒ }ttj|d
g|j	d�|t
ddd�gdœtddgd	d�d�}|d j |¡|d< t ||¡ d S )NÚabcÚcategoryZ20130101r   r¢   )Zperiodsrœ   rŽ   ©r	   r	   r   )ÚgroupÚcategory_stringÚ
datetimetzr¦   rg   r8   z
2013-01-03r›   )r§   r¨   r	   r   r   r:   r¨   )r   r]   Úastyper�   r‘   r   r;   r   ZCategoricalr9   r   r   Údtr•   r"   r#   )	r3   ÚtsÚalpharŽ   r§   Zdtir+   r/   r-   r0   r0   r1   Útest_first_last_tz_multi_column˜  s&    	ýÿ
ÿüù	r­   r?   TFÚbooleanZInt64z
2020-01-01z
2020-02-01r   )rŽ   Úfunction)r   r%   ÚminÚmaxc           	      C   s„   t ddg| dœƒ}| d¡}tddgdd�}t| d|d�}t d| i|d�}t|d |ƒƒ }t ||¡ | d|i¡}t ||¡ d S )	Nr	   r   r5   r6   r   r7   )r   r    r:   )	r   r   r   r   r;   r"   re   Zaggr#   )	r?   r¯   r+   r,   r@   Zexpected_seriesZexpected_frameZresult_seriesZresult_framer0   r0   r1   Ú+test_first_last_extension_array_keeps_dtype¸  s    
r²   c                  C   sR   t g d¢g d¢g d¢dœƒ} |  ddg¡}| d¡}| jg d¢ }t ||¡ d S )	N)r   r   r   r   r   r   r   r   r   r   r   )rL   rL   rL   rM   rL   rL   rL   rM   rM   rM   rL   )Údullr³   Úshinyr³   r³   r´   r´   r³   r´   r´   r´   ©r   r
   r   r   r
   r   r˜   rB   r™   r0   r0   r1   Ú test_nth_multi_index_as_expectedÒ  s    åÿ+
r¶   zop, n, expected_rowsÚheadr   r‰   Útailr   rH   c           	      C   sp   t ddgddgddggddgd�}|jd|d	�}|j| }|d urR|| }|| }t|| ƒ|ƒ}t ||¡ d S )
Nr	   r   r   r   r   r   r
   r   rG   )r   r   r&   r;   r"   r#   )	ÚopÚnÚexpected_rowsr   rH   r+   r.   r-   r/   r0   r0   r1   Útest_groupby_head_tail  s    "
r¼   zop, n, expected_colsc                 C   sœ   t g d¢g d¢g d¢g d¢gg d¢d�}d}tjt|d��" |jg d	¢d
d�}W d   ƒ n1 sb0    Y  |jd d …|f }t|| ƒ|ƒ}t ||¡ d S )NrŸ   )r	   r   r   )r   r   r   )r   rO   rW   rµ   r   ú+DataFrame.groupby with axis=1 is deprecatedr~   ©r   r   r	   r	   ©Zaxis)r   r"   Úassert_produces_warningÚFutureWarningr   r&   r;   r#   )r¹   rº   Zexpected_colsr+   Úmsgr.   r-   r/   r0   r0   r1   Útest_groupby_head_tail_axis_1   s     ÿ0rÃ   c                  C   s  t ddgddgddggddgd�} | jd	dg }|  d¡}|jdd
�}| d	¡}t || ¡ t ||¡ |  d¡}|jdd
�}| d	¡}t || ¡ t ||¡ |  d¡}| d	¡}|jdd
�}t ||¡ t || ¡ |  d¡}| d	¡}|jdd
�}t ||¡ t || ¡ d S )Nr	   r   r   r   r   r   r
   r   r   ©rº   )r   r&   r   r·   r$   r"   r#   r¸   )r+   r-   r.   Zresult1Zresult2r0   r0   r1   Útest_group_selection_cache:  s,    "







rÅ   c                  C   sn   t dgg d¢d�} |  d¡ d¡}| jd d… }t ||¡ |  ddg¡ d¡}| jd d… }t ||¡ d S )Nr   ©r6   r7   rg   r�   r6   rV   r7   rB   rx   r0   r0   r1   Útest_nth_emptyX  s    rÇ   c                  C   sŽ   t g d¢g d¢ddtjgg d¢g d¢gg d¢d�} |  d	¡ d
¡}| jd
dg }t ||¡ |  d	¡jddd�}| jddg }t ||¡ d S )N)r	   r7   rz   )r	   r6   é2   r	   r6   )r   rg   éÈ   )r   rh   é–   )r   r   r
   r   r   r   r   r   r   r   r   ©r   r(   r)   r   r$   r&   r"   r#   rx   r0   r0   r1   Útest_nth_column_orderd  s    $þrÌ   r   )Nr   r‡   c                 C   sZ   t tjdtjdtjgg d¢g d¢dœƒ}| d¡jd| d�}|jdd	g }t ||¡ d S )
Nr6   r7   ©r   r   r   r   rO   )r	   r   r   r   rW   rÆ   r   r   r	   r   rË   ©r   r+   r/   r-   r0   r0   r1   Útest_nth_nan_in_groupert  s    ýÿrÏ   c                 C   s\   t tjdtjdtjgg d¢dœƒ}| d¡d jd| d�}|d jddg }t ||¡ d S )	Nr6   r7   rÍ   r5   r   r   r	   r   )r   r(   r)   r   r$   r&   r"   re   rÎ   r0   r0   r1   Útest_nth_nan_in_grouper_series„  s    þÿrÐ   c                  C   sŽ   t g d¢dt d¡g tg d¢dd�dœƒ} |  d¡ ¡ }t d	t d¡g tg d
¢ƒ t g d¢¡¡dœƒ}t	g d¢dd�|_
t ||¡ d S )N)r   r   ÚsecondÚthirdrÒ   r   ZNaT)r6   r7   rg   r6   r7   r¤   r8   )r¦   rž   Ú
categoriesr¦   r   )r6   rg   r6   rÆ   )rž   rÓ   )r   rÑ   rÒ   r   )r   r(   Z
datetime64r   r   r   r©   r�   ÚCategoricalDtyper   r    r"   r#   rx   r0   r0   r1   Ú,test_first_categorical_and_datetime_data_nat“  s     ýÿÿþÿrÕ   c                  C   s–   t g d¢g d¢g d¢g d¢dœƒ} |  ddi¡} | jdd	gd
� ¡ }t g d¢tg d¢ƒ t g d¢¡¡dœƒ}tjg d¢dd	gd�|_	t
 ||¡ d S )N)r	   r	   r	   r   r   )rz   rz   rÉ   rz   rz   )ÚappleÚorangeÚmangorØ   r×   )ÚjupiterÚmercuryÚmarsÚvenusrÜ   rP   r   r¤   r   r
   )Zby)rÖ   rØ   rØ   )rÙ   rÛ   rÜ   )rÙ   rÛ   rÚ   rÜ   )r   r   ))r	   rz   )r	   rÉ   )r   rz   )Únames)r   r©   r   r   r   r�   rÔ   r   Úfrom_tuplesr    r"   r#   rx   r0   r0   r1   Ú(test_first_multi_key_groupby_categorical©  s(    üÿÿþÿÿrß   )r   r%   r$   c                 C   sf   t dgtd�}t |||d|gg d¢td�jdd�}| dkrJt|| ƒdƒ}nt|| ƒƒ }t ||¡ d S )	NÚyr8   )r   r   r   r   r   )r    r9   r   r[   r$   r   )r   rd   r   r;   r"   re   )r3   r=   r-   rY   r/   r0   r0   r1   Ú%test_groupby_last_first_nth_with_noneÃ  s    ýürá   zarg, expected_rows)r   r	   r   r   rm   )r   r   r   ©r   r	   r   r   r   r   r   c                 C   s:   |j | }|  |¡}| j| }t ||¡ t ||¡ d S ©N)r$   r&   r"   r#   )Úslice_test_dfÚslice_test_groupedÚargr»   r/   Ú
equivalentr-   r0   r0   r1   Ú
test_sliceÕ  s
    


rè   c                 C   sT   |j dddd …f }|  ddtdd ƒg¡}| jg d¢ }t ||¡ t ||¡ d S )Nr   r	   rm   râ   )r$   Úslicer&   r"   r#   )rä   rå   r/   rç   r-   r0   r0   r1   Útest_nth_indexedé  s
    rê   c                 C   s<   t jtdd�� |  d¡ W d   ƒ n1 s.0    Y  d S )NzInvalid indexr~   g…ëQ¸	@)r‚   rƒ   Ú	TypeErrorr$   ©rå   r0   r0   r1   Útest_invalid_argumentô  s    rí   c                 C   sD   t jtdd��" |  td d dƒ¡ W d   ƒ n1 s60    Y  d S )NzInvalid stepr~   r   )r‚   rƒ   r„   r$   ré   rì   r0   r0   r1   Útest_negative_stepû  s    rî   c                 C   s2   |  t ddg¡¡}| jg d¢ }t ||¡ d S )Nr   r	   rt   )r$   r(   r<   r&   r"   r#   )rä   rå   r/   r-   r0   r0   r1   Útest_np_ints  s    rï   c                  C   s˜   t g d¢g d¢gddgg d¢d�} d}tjt|d��$ | j| jd	 d	d
�}W d   ƒ n1 s^0    Y  | d¡}| jd d …ddgf }t ||¡ d S )N)r   r   r   )rO   rO   r   Úzrà   )r   r
   r   r�   r½   r~   r	   r¿   r   r   )r   r"   rÀ   rÁ   r   r&   r$   r#   )r+   rÂ   rp   r/   r-   r0   r0   r1   Ú!test_groupby_nth_with_column_axis
  s    þú2
rñ   c               	   C   sô   t t t dd¡t dd¡g¡t t dd¡t dd¡g¡gg d¢g d¢gƒ} tdtt| ƒƒi| d	�}|jddgd
d� d¡}g d¢}t t t dd¡t dd¡g¡t t dd¡t dd¡g¡gg d¢g d¢gƒ}t||dgd�}t	 
||¡ d S )Nr   r	   r   rV   é   )r   r   r   r	   r	   )r   r	   r	   r   r   Úcolr:   F)r\   Zobserved©r   r	   r   r¾   )r   r	   r   r�   )r   r�   ZCategoricalIndexZIntervalr   r^   Úlenr   r$   r"   r#   )Z
idx_resultZ	df_resultr/   Zval_expectedZidx_expectedr-   r0   r0   r1   Útest_groupby_nth_interval  s"    þûþûrö   z.start, stop, expected_values, expected_columnsrt   ÚABCDEZADrW   rô   ZABD)r	   r   r   ZBCEr   ZCErQ   Úcallc           	         s    t tdƒgtdƒgd�}d}tjt|d��" |jg d¢dd�‰ W d   ƒ n1 sR0    Y  ‡ fd	d
„‡ fdd
„dœ| | |ƒ}t |g|gd�}t ||¡ d S )Nr   r÷   r   r½   r~   )r   r   r   r   r   r	   r¿   c                    s   ˆ   t| |ƒ¡S rã   )r$   ré   ©r‹   Ústop©rp   r0   r1   r|   I  r}   z2test_nth_slices_with_column_axis.<locals>.<lambda>c                    s   ˆ j | |… S rã   )r$   rù   rû   r0   r1   r|   J  r}   )rø   r    )r   r^   r]   r"   rÀ   rÁ   r   r#   )	r‹   rú   Zexpected_valuesZexpected_columnsr3   r+   rÂ   r/   r-   r0   rû   r1   Ú test_nth_slices_with_column_axis3  s    0

þýýrü   z<ignore:invalid value encountered in remainder:RuntimeWarningc                  C   s®   t ddgdtjgdtjgdtjggddgd�} t ddggddgd�}|  ddg¡jdd	�}t ||¡ |  ddg¡jdd	�}t ||¡ |  ddg¡jd
d	�}t ||¡ d S )Nr6   rð   r7   rg   ÚXÚYr   r	   rÄ   r   ©	r   r(   r)   r   r·   r"   r#   r¸   r$   ©r+   r-   r/   r0   r0   r1   Útest_head_tail_dropna_trueP  s    &ÿr  c                  C   sÂ   t ddgdtjgdtjggddgd�} t ddgdtjgdtjggddgd�}| jddgdd	�jd
d�}t ||¡ | jddgdd	�jd
d�}t ||¡ | jddgdd	�jdd�}t ||¡ d S )Nr6   rð   r7   rg   rý   rþ   r   Fr   r	   rÄ   r   rÿ   r   r0   r0   r1   Útest_head_tail_dropna_falsed  s    &&r  Ú	selectionrg   )r   r‡   Nc                 C   sˆ   t g d¢tjddgg d¢dœƒ}| d¡|  }|jd|d�}|d	ksX|d
krb| ddgkrbddg}nddg}|j|| f }t ||¡ d S )Nr¥   r   r   )r   r   r   rÆ   r6   r   r   r   r‡   r7   rg   r	   r   )r   r(   r)   r   r$   r   r"   Zassert_equal)r  r   r+   rp   r/   Zlocsr-   r0   r0   r1   Útest_nth_after_selections  s    
ýÿ
r  rY   z2011-01-15 12:50:28.502376z2011-01-20 12:50:28.593448)l    m;-L¼l    m;-L¼c                 C   sJ   t ddg| dœƒ}| d¡}| d¡}t d| d gdœƒ}t ||¡ d S )Nr	   r5   r6   r   )r   r   r$   r"   r#   )rY   r+   r,   r/   r-   r0   r0   r1   Ú#test_groupby_nth_int_like_precisionˆ  s
    

r  )BÚnumpyr(   r‚   Zpandasr�   r   r   r   r   r   r   Zpandas._testingZ_testingr"   r2   ÚmarkZparametrizerA   rC   rJ   r)   rd   rK   rU   rb   rf   rl   rq   ry   r†   rˆ   rŠ   r—   rš   r¡   r­   r<   r”   Zto_timedeltar²   r¶   r¼   rÃ   rÅ   rÇ   rÌ   rÏ   rÐ   rÕ   rß   rá   ré   rè   rê   rí   rî   rï   rñ   rö   r]   rü   Úfilterwarningsr  r  r  r  r0   r0   r0   r1   Ú<module>   s^   2

	

þþûþ
!		
ý
ý
ýõ


ýù


þú	

þúîîþ1
þþ
üþ	4

øþ

øþ



üþ	
øþÿ
þûþ
