o
    _~#gn  ã                   @   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€„ Z5dd‚„ 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dg 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>ddž„ 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 ƒs»J ‚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   úŠ/var/www/static.ux5.de/https/Moving-Object-Detection-with-OpenCV/env/lib/python3.10/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r"|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   é   Úfloat32r8   ©r   r
   r   r   TÚEr	   ÚFr   r   )r
   r   r   rR   rS   r   r   r   r   r   r   )r   r(   ÚrandomÚdefault_rngÚ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.   üÿ

rW   c                  C   sX   t tdƒƒ} |  d¡ ttdƒ| dd}|jdksJ ‚|jdd ¡ }|jdks*J ‚d S )	Né
   é	   é   ÚIntCol)Údatar    r   Úint64r   ©Úlevel)ÚlistÚrangeÚappendr   r9   r   r   )r@   ÚserÚfr0   r0   r1   Útest_first_last_nth_dtypes2œ   s   
re   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   )r\   Únansr\   r   r   )r   r   Úobjectr   r*   rf   r"   Úassert_series_equalr   r%   r$   )r+   r,   r-   r0   r0   r1   Útest_first_last_nth_nan_dtype¦   s   
ri   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   ÚeÚaaaard   r.   )r6   rj   rd   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   üÿýrp   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   rh   r
   ©r+   Úgbr0   r0   r1   Útest_nthÇ   s   $
 (   $ru   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   ÚhamÚeggsÚporkgþI|î»ø?gçoB!²¿gHk:!4Àgn‹2d’Ý?gC8fÙ“Àà?g£W”†â?gV¿Òùð,ï¿gN'ÙêrŠð?gÃGÄ”H"é¿gû­(	‰æ?)ÚcolorÚfoodrM   rL   r|   r}   r   F)r_   rH   r   r   r   )r   r*   r   r$   r&   r"   r#   r   ©r+   r/   r-   r0   r0   r1   Ú	test_nth2Ü   s2   ûûöÿ
í
r   c                  C   sþ   t tj d¡ ddd¡dd} | d }| d }| |¡ ¡ }| |¡ dd	„ ¡}tj	||d
d |j
dks7J ‚|j
dks>J ‚||dk jd }|jd |ksPJ ‚|jd |ksYJ ‚tjtdd |j|d
djddd W d   ƒ d S 1 sxw   Y  d S )Nr   r	   rX   )éd   r   r]   r8   r   c                 S   s
   | j d S )Nr   )r&   )Úxr0   r0   r1   Ú<lambda>  s   
 ztest_nth3.<locals>.<lambda>F)Úcheck_nameszFor a DataFrame©Úmatch)ÚsortTr   )r   r(   rT   rU   Úintegersr   r   Úapplyr"   rh   r   r&   ÚpytestÚraisesÚ
ValueErrorr$   )r+   rc   rt   r-   Ú	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"   rh   )r+   rt   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   rx   )r   r	   r   r   )	r   r(   r)   r   r"   r#   r$   r&   r   rs   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   rq   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/30rn   )ÚpdÚ
date_ranger   r    ÚyearÚmonthr   r$   Úto_datetimeÚas_unitr"   r#   )r–   Úbusiness_datesr+   Úkeyr/   Ú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#   ©Ú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%   )r\   Úexpected_firstÚ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ÚcategoryÚ20130101r   r¯   )Úperiodsr§   r–   ©r	   r	   r   )ÚgroupÚcategory_stringÚ
datetimetzrµ   rj   r8   z
2013-01-03r¦   )r¶   r·   r	   r   r   r:   r·   )r   r`   Úastyper˜   r™   r   r;   r   ÚCategoricalr9   r   r   Údtr   r"   r#   )	r3   ÚtsÚalphar–   r¶   Údtir+   r/   r-   r0   r0   r1   Útest_first_last_tz_multi_column˜  s&   	ýÿ
ÿüù	r¾   r?   TFÚbooleanÚ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"   rh   Úaggr#   )	r?   rÁ   r+   r,   r@   Úexpected_seriesÚexpected_frameÚresult_seriesÚ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r)|| }|| }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 s1w   Y  |jd d …|f }t|| ƒ|ƒ}t ||¡ d S )Nrª   )r	   r   r   )r   r   r   )r   rO   rY   rÌ   r   ú+DataFrame.groupby with axis=1 is deprecatedr„   ©r   r   r	   r	   ©Úaxis)r   r"   Úassert_produces_warningÚFutureWarningr   r&   r;   r#   )rÐ   rÑ   Úexpected_colsr+   Úmsgr.   r-   r/   r0   r0   r1   Útest_groupby_head_tail_axis_1   s    ÿÿ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.   Úresult1Ú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   rj   r—   r6   rX   r7   rB   r~   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   r€   )r	   r6   é2   r	   r6   )r   rj   éÈ   )r   rk   é–   )r   r   r
   r   r   r   r   r   r   r   r   ©r   r(   r)   r   r$   r&   r"   r#   r~   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   rY   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"   rh   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   ÚNaT)r6   r7   rj   r6   r7   r±   r8   )rµ   r©   Ú
categoriesrµ   r   )r6   rj   r6   rá   )r©   rï   )r   rì   rí   r   )r   r(   Ú
datetime64r   r   r   r¸   r˜   ÚCategoricalDtyper   r    r"   r#   r~   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   )r€   r€   rä   r€   r€   )ÚappleÚorangeÚmangorõ   rô   )ÚjupiterÚmercuryÚmarsÚvenusrù   rQ   r   r±   r   r
   )Úby)ró   rõ   rõ   )rö   rø   rù   )rö   rø   r÷   rù   )r   r   ))r	   r€   )r	   rä   )r   r€   )Únames)r   r¸   r   r   r   r˜   rñ   r   Úfrom_tuplesr    r"   r#   r~   r0   r0   r1   Ú(test_first_multi_key_groupby_categorical©  s(   üÿÿþÿÿrý   )r   r%   r$   c                 C   s`   t dgƒ}t |||d|gg d¢djdd}| dkr"t|| ƒdƒ}nt|| ƒƒ }t ||¡ d S )NÚy)r   r   r   r   r   r:   r   r^   r$   r   )r   r   r;   r"   rh   )r3   r=   r-   r\   r/   r0   r0   r1   Ú%test_groupby_last_first_nth_with_noneÃ  s   
þýrÿ   zarg, expected_rows)r   r	   r   r   rq   )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	   rq   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   ƒ d S 1 sw   Y  d S )NzInvalid indexr„   g…ëQ¸	@)r‰   rŠ   Ú	TypeErrorr$   ©r  r0   r0   r1   Útest_invalid_argumentó  s   "ÿr  c                 C   sF   t jtdd |  td d dƒ¡ W d   ƒ d S 1 sw   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	   rx   )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/w   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Û   rt   r/   r-   r0   r0   r1   Ú!test_groupby_nth_with_column_axis	  s   þúÿ
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   rX   é   )r   r   r   r	   r	   )r   r	   r	   r   r   Úcolr:   F)r_   Úobserved©r   r	   r   rÕ   )r   r	   r   r—   )r   r˜   ÚCategoricalIndexÚIntervalr   ra   Úlenr   r$   r"   r#   )Ú
idx_resultÚ	df_resultr/   Úval_expectedÚidx_expectedr-   r0   r0   r1   Útest_groupby_nth_interval  s"   þûþûr  z.start, stop, expected_values, expected_columnsrx   ÚABCDEÚADrY   r  ÚABD)r	   r   r   ÚBCEr   ÚCErR   Úcallc           	         s    t tdƒgtdƒgd}d}tjt|d |jg d¢dd‰ W d   ƒ n1 s)w   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©rt   r0   r1   r‚   H  s    z2test_nth_slices_with_column_axis.<locals>.<lambda>c                    s   ˆ j | |… S r  )r$   r"  r$  r0   r1   r‚   I  s    )r!  r    )r   ra   r`   r"   rØ   rÙ   r   r#   )	r“   r#  Úexpected_valuesÚexpected_columnsr3   r+   rÛ   r/   r-   r0   r$  r1   Ú test_nth_slices_with_column_axis2  s   ÿ

þýý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   rj   ÚXÚYr   r	   rÝ   r   ©	r   r(   r)   r   rÎ   r"   r#   rÏ   r$   ©r+   r-   r/   r0   r0   r1   Útest_head_tail_dropna_trueO  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   rj   r(  r)  r   Fr   r	   rÝ   r   r*  r+  r0   r0   r1   Útest_head_tail_dropna_falsec  s   &&r-  Ú	selectionrj   )r   r   Nc                 C   sˆ   t g d¢tjddgg d¢dœƒ}| d¡|  }|jd|d}|d	ks,|d
kr1| ddgkr1ddg}nddg}|j|| f }t ||¡ d S )Nr´   r   r   )r   r   r   rá   r6   r   r   r   r   r7   rj   r	   r   )r   r(   r)   r   r$   r   r"   Úassert_equal)r.  r   r+   rt   r/   Úlocsr-   r0   r0   r1   Útest_nth_after_selectionr  s   
ýÿ
r1  r\   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#   )r\   r+   r,   r/   r-   r0   r0   r1   Ú#test_groupby_nth_int_like_precision‡  s
   

r2  )BÚnumpyr(   r‰   Úpandasr˜   r   r   r   r   r   r   Úpandas._testingÚ_testingr"   r2   ÚmarkÚparametrizerA   rC   rJ   r)   rg   rK   rW   re   ri   rp   ru   r   rŽ   r   r’   r¡   r¥   r®   r¾   r<   rœ   Ú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-  r1  r2  r0   r0   r0   r1   Ú<module>   s`    2

	

þþûþ
!		
ý
ý
ýõ


ýù


þú	

þúîîþ
1þþ
üþ	4

øþ

øþ



üþ
	øþÿ
þûþ
