a
    é´«jû ã                   @   s~	  d dl Zd dlZd dlmZmZ d dlmZ d dlZ	d dlm
Z
mZmZmZmZmZmZmZmZ d dlmZ d dlmZ d dlmZ dd„ Zd	d
„ Zej dg d¢¡ejj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„ dd„ dd„ dd„ gg d¢d �d!d"„ ƒƒZej dg d¢¡ejj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„ d0d„ d1d„ d2d„ d3d„ gg d4¢d �d5d6„ ƒƒZej dg d¢¡d7d8„ ƒZ d9d:„ Z!ej d;e	 "d d<¡ed=d<gƒg¡d>d?„ ƒZ#d@dA„ Z$dBdC„ Z%dDdE„ Z&dFdG„ Z'dHdI„ Z(dJdK„ Z)ejjd;dLd„ dMd„ dNd„ dOd„ gg dP¢d �dQdR„ ƒZ*dSdT„ Z+dUdV„ Z,ej dWdXdYgidZdYid[dYig¡d\d]„ ƒZ-d^d_„ Z.d`da„ Z/dbdc„ Z0ddde„ Z1dfdg„ Z2ej dhdid„ djd„ dkd„ g¡dldm„ ƒZ3dndo„ Z4dpdq„ Z5drds„ Z6dtdu„ Z7ej dvd;dwg¡dxdy„ ƒZ8dzd{„ Z9d|d}„ Z:ejjd~dd„ d fd€d„ d<fd�d„ d fd‚d„ d fgg dƒ¢d �d„d…„ ƒZ;ej d†ed=d<gdYd‡�e
dYd=d<giƒg¡dˆd‰„ ƒZ<ej d†ed=d<gdYd‡�e
dYd=d<giƒg¡dŠd‹„ ƒZ=dŒd�„ Z>dŽd�„ Z?d�d‘„ Z@d’d“„ ZAej d”d•d=ejBfg¡d–d—„ ƒZCej d˜d™dšg¡d›dœ„ ƒZDej dd�d„ džd„ dŸd„ d d„ g¡d¡d¢„ ƒZEd£d¤„ ZFd¥d¦„ ZGd§d¨„ ZHej d©dYdªd=d«œd d=d d«œg¡d¬d­„ ƒZIej dd®d¯g¡d°d±„ ƒZJej d†ed=d<gƒe
dYd=d<giƒg¡d²d³„ ƒZKej d†ed=d<gƒe
dYd=d<giƒg¡d´dµ„ ƒZLd¶d·„ ZMd¸d¹„ ZNej dºeg d»¢dYd‡�i fe
dYg d»¢iƒd¼dYifg¡d½d¾„ ƒZOej dºeg d»¢dYd‡�i fe
dYg d»¢iƒd¼dYifg¡d¿dÀ„ ƒZPej dÁg dÂ¢¡dÃdÄ„ ƒZQdÅdÆ„ ZRdÇdÈ„ ZSej d†eg d»¢ƒe
dYg d»¢iƒg¡dÉdÊ„ ƒZTdËdÌ„ ZUdÍdÎ„ ZVdÏdÐ„ ZWej d©dÑdÒid;dÒig¡dÓdÔ„ ƒZXej dÕdÖd×g¡dØdÙ„ ƒZYdÚdÛ„ ZZdÜdÝ„ 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ìe`fdíg¡dîdï„ ƒZaej dàdádâg¡ej dhdðd„ dñd„ g¡dòdó„ ƒƒZbej dàdádâg¡ej dhdôd„ dõd„ g¡död÷„ ƒƒZcej dødùdúg¡ej dhdûd„ düd„ g¡dýdþ„ ƒƒZdej dhdÿ�d g¡�d�d„ ƒZe�d�d„ Zf�d�d„ Zg�d�d„ Zh�d	�d
„ Ziejjdàdádçg�d�dgd ��d�d„ ƒZj�d�d„ Zkej �ddYdYgg¡�d�d„ ƒZlej �d�d�dg¡ejjdàdádçg�d�dgd ��d�d„ ƒƒZmej �dd d=g¡ej �d�d�dg¡�d�d„ ƒƒZn�d�d „ Zo�d!�d"„ Zp�d#�d$„ Zq�d%�d&„ Zr�d'�d(„ Zsej ddì�d)g¡�d*�d+„ ƒZt�d,�d-„ Zu�d.�d/„ Zv�d0�d1„ Zw�d2�d3„ Zx�d4�d5„ Zy�d6�d7„ Zz�d8�d9„ Z{�d:�d;„ Z|�d<�d=„ Z}�d>�d?„ Z~dS (@  é    N)ÚHAS_PYARROWÚWARNING_CHECK_DISABLED)ÚSettingWithCopyWarning)	Ú	DataFrameÚIndexÚ
MultiIndexÚPeriodÚSeriesÚ	TimestampÚ
date_rangeÚoption_contextÚperiod_range)Ú	get_array)ÚVersionc                 C   sÔ   t g d¢g d¢g d¢dœƒ}| ¡ }|j|jus4J ‚|j|jusDJ ‚|j |j¡sVJ ‚|j |j¡shJ ‚t t|dƒt|dƒ¡r„J ‚| r´|jj	d j
 ¡ ržJ ‚|jj	d j
 ¡ r´J ‚d|jd< |jd dksÐJ ‚d S )	N©é   é   é   ©é   é   é   ©gš™™™™™¹?gš™™™™™É?g333333Ó?©ÚaÚbÚcr   r   r   ©r   r   )r   ÚcopyÚindexÚcolumnsÚis_ÚnpÚshares_memoryr   Ú_mgrÚblocksÚrefsÚhas_referenceÚiloc)Úusing_copy_on_writeÚdfÚdf_copy© r,   úb/var/www/html/laksyashrimp/venv/lib/python3.9/site-packages/pandas/tests/copy_view/test_methods.pyÚ	test_copy   s    
r.   c                 C   s¬  t g d¢g d¢g d¢dœƒ}|jdd�}| rr|j|jus<J ‚|j|jusLJ ‚|j |j¡s^J ‚|j |j¡s’J ‚n |j|ju s‚J ‚|j|ju s’J ‚t t|dƒt|dƒ¡s®J ‚| rÞ|jj	d j
 ¡ sÈJ ‚|jj	d	 j
 ¡ sÞJ ‚| �r@d|jd
< |jd
 d	k�sJ ‚t t|dƒt|dƒ¡�r J ‚t t|dƒt|dƒ¡�s¨J ‚nht |¡� d|jd
< W d   ƒ n1 �sl0    Y  |jd
 dk�sŠJ ‚t t|dƒt|dƒ¡�s¨J ‚d S )Nr   r   r   r   F©Údeepr   r   r   r   r   )r   r   r   r    r!   r"   r#   r   r$   r%   r&   r'   r(   ÚtmÚassert_cow_warning)r)   Úwarn_copy_on_writer*   r+   r,   r,   r-   Útest_copy_shallow0   s,    
 *r4   r   )TNFÚmethodc                 C   s   | j tj|d�S )N©r    r   )ÚrenameÚstrÚlower©r*   r   r,   r,   r-   Ú<lambda>Y   ó    r;   c                 C   s   | j ddg|d�S )Nr   r   r6   )Úreindexr:   r,   r,   r-   r;   Z   r<   c                 C   s   | j | |d�S ©N©r   ©Úreindex_liker:   r,   r,   r-   r;   [   r<   c                 C   s   | j | |d�d S ©Nr?   r   ©Úalignr:   r,   r,   r-   r;   \   r<   c                 C   s   | j g d¢d|d�S ©Nr   r   )Úaxisr   ©Úset_axisr:   r,   r,   r-   r;   ]   r<   c                 C   s   | j d|d�S ©NÚtest©r   r   ©Úrename_axisr:   r,   r,   r-   r;   ^   r<   c                 C   s   | j d|d�S )NrJ   r6   rL   r:   r,   r,   r-   r;   _   r<   c                 C   s   | j ddi|d�S )Nr   Úint64r?   ©Úastyper:   r,   r,   r-   r;   `   r<   c                 C   s   | j dd|d�S ©Nr   r?   ©Úswapaxesr:   r,   r,   r-   r;   b   r<   c                 C   s   | j dd|d�S ©Nr   r   r?   ©Útruncater:   r,   r,   r-   r;   c   r<   c                 C   s   | j |d�S r>   ©Úinfer_objectsr:   r,   r,   r-   r;   d   r<   c                 C   s   | j |d�S r>   ©Úto_timestampr:   r,   r,   r-   r;   e   r<   c                 C   s   | j d|d�S ©NÚD)Úfreqr   ©Ú	to_periodr:   r,   r,   r-   r;   f   r<   c                 C   s   | j d|d�S ©Nú
US/Centralr?   ©Útz_localizer:   r,   r,   r-   r;   g   r<   c                 C   s   | j d|d�S r`   ©Ú
tz_convertr:   r,   r,   r-   r;   h   r<   c                 C   s   | j d|d�S ©NF)Úallows_duplicate_labelsr   ©Ú	set_flagsr:   r,   r,   r-   r;   i   r<   )r7   r=   rA   rD   rH   Úrename_axis0Zrename_axis1rP   rS   rV   rX   rZ   r_   rc   re   ri   )Úidsc           
      C   sx  d }d| j jjv r"tdddd�}nZd| j jjv r@tdddd�}n<d| j jjv r^tdddd�}nd| j jjv r|tdddd	d
�}tg d¢g d¢g d¢dœ|d�}d| j jjv rêd}tjt|d�� |||d�}W d   ƒ qö1 sÞ0    Y  n|||d�}|�p|du }	| j jj 	d¡�r0|�s0|�s0|du �r0d}	|	�rVt
 t|dƒt|dƒ¡�stJ ‚nt
 t|dƒt|dƒ¡�rtJ ‚d S )NrZ   ú
2012-01-01r\   r   ©r]   Úperiodsr_   rc   re   úEurope/Brussels©r]   rn   Útzr   r   r   r   ©r   rS   ú"'DataFrame.swapaxes' is deprecated©Úmatchr?   Fzreindex-r   )ÚnodeÚcallspecÚidr   r   r   r1   Úassert_produces_warningÚFutureWarningÚ
startswithr"   r#   r   )
Úrequestr5   r   r)   Úusing_array_managerr   r*   ÚmsgÚdf2Úshare_memoryr,   r,   r-   Útest_methods_copy_keywordU   s,    - , r�   c                 C   s   | j ddi|d�S )Nr   éd   rK   ©r7   ©Úserr   r,   r,   r-   r;   ¦   r<   c                 C   s   | j d |d�S r>   rƒ   r„   r,   r,   r-   r;   §   r<   c                 C   s   | j | j|d�S )NrK   )r=   r   r„   r,   r,   r-   r;   ¨   r<   c                 C   s   | j | |d�S r>   r@   r„   r,   r,   r-   r;   ©   r<   c                 C   s   | j | |d�d S rB   rC   r„   r,   r,   r-   r;   ª   r<   c                 C   s   | j g d¢d|d�S rE   rG   r„   r,   r,   r-   r;   «   r<   c                 C   s   | j d|d�S rI   rL   r„   r,   r,   r-   r;   ¬   r<   c                 C   s   | j d|d�S )NrN   r?   rO   r„   r,   r,   r-   r;   ­   r<   c                 C   s   | j dd|d�S )Nr   r   r?   )Ú	swaplevelr„   r,   r,   r-   r;   ®   r<   c                 C   s   | j dd|d�S rQ   rR   r„   r,   r,   r-   r;   ¯   r<   c                 C   s   | j dd|d�S rT   rU   r„   r,   r,   r-   r;   °   r<   c                 C   s   | j |d�S r>   rW   r„   r,   r,   r-   r;   ±   r<   c                 C   s   | j |d�S r>   rY   r„   r,   r,   r-   r;   ²   r<   c                 C   s   | j d|d�S r[   r^   r„   r,   r,   r-   r;   ³   r<   c                 C   s   | j d|d�S r`   rb   r„   r,   r,   r-   r;   ´   r<   c                 C   s   | j d|d�S r`   rd   r„   r,   r,   r-   r;   µ   r<   c                 C   s   | j d|d�S rf   rh   r„   r,   r,   r-   r;   ¶   r<   )zrename (dict)r7   r=   rA   rD   rH   rj   rP   r†   rS   rV   rX   rZ   r_   rc   re   ri   c           	      C   s\  d }d| j jjv r"tdddd�}n€d| j jjv r@tdddd�}nbd| j jjv r^tdddd�}nDd| j jjv r~tdddd	d
�}n$d| j jjv r¢t g d¢g d¢g¡}tg d¢|d�}d| j jjv �rd}tj	t
|d�� |||d�}W d   ƒ n1 sö0    Y  n|||d�}|�p|du }|�r>t t|ƒt|ƒ¡�sXJ ‚nt t|ƒt|ƒ¡�rXJ ‚d S )NrZ   rl   r\   r   rm   r_   rc   re   ro   rp   r†   r   r   rr   rS   z'Series.swapaxes' is deprecatedrt   r?   F)rv   rw   rx   r   r   r   Zfrom_arraysr	   r1   ry   rz   r"   r#   r   )	r|   r5   r   r)   r   r…   r~   Úser2r€   r,   r,   r-   Ú test_methods_series_copy_keyword¢   s*    +,rˆ   c                 C   s‚   t g d¢g d¢dœƒ}|j|d�}| p4|du p4|d u }|o>| }|rbt t|dƒt|dƒ¡s~J ‚nt t|dƒt|dƒ¡r~J ‚d S )Nr   r   ©r   r   r?   Fr   r   )r   Ú	transposer"   r#   r   )r)   r   r}   r*   Úresultr€   r,   r,   r-   Útest_transpose_copy_keywordê   s    
rŒ   c                 C   sÐ   t g d¢g d¢g d¢dœg d¢d�}| ¡ }| ¡ }|j ¡  | rzt t|dƒt|dƒ¡s^J ‚t t|dƒt|dƒ¡szJ ‚d	|jd
< t t|dƒt|dƒ¡r J ‚| rÀt t|dƒt|dƒ¡sÀJ ‚t	 
||¡ d S )Nr   r   r   r   )é
   é   é   rr   r   r   r   )r   r   )r   r   Úreset_indexr$   Ú_verify_integrityr"   r#   r   r(   r1   Úassert_frame_equal©r)   r*   Údf_origr   r,   r,   r-   Útest_reset_indexû   s    ÿ

r•   r   r   r   c                 C   s„   t ddg|d�}| ¡ }|jdd�}| rRt t|ƒt|ƒ¡s@J ‚|j d¡rjJ ‚nt t|ƒt|ƒ¡rjJ ‚d|jd< t	 
||¡ d S )Nr   r   rr   T)Údropr   r‚   )r	   r   r�   r"   r#   r   r$   Ú_has_no_referencer(   r1   Úassert_series_equal©r)   r   r…   Úser_origr‡   r,   r,   r-   Útest_reset_index_series_drop  s    
r›   c                  C   sb   t g d¢g d¢g d¢dœƒ} |  d¡} | d }| j|dd� ¡ }| jddd� ¡ }t ||¡ d S )	N©r   r   r   Úd©r   r   r   r   )r   r   r   r   ©ÚAÚBÚCr    r¢   T)Zobserved)r   Ú	set_indexÚgroupbyÚsumr1   r’   )r*   Úkeyr‹   Úexpectedr,   r,   r-   Ú'test_groupby_column_index_in_references   s    ÿ
r¨   c                 C   sÐ   t g d¢g d¢g d¢dœƒ}| ¡ }|jtjd�}| rRt t|dƒt|dƒ¡sRJ ‚d|jd	< t t|dƒt|dƒ¡rxJ ‚| r˜t t|d
ƒt|dƒ¡s˜J ‚t g d¢g d¢g d¢dœƒ}t	 
||¡ t	 
||¡ d S )Nr   r   r   r   ©r    r    r   r   r   r¢   r   ©r   r   r   rŸ   )r   r   r7   r8   Úupperr"   r#   r   r(   r1   r’   ©r)   r*   r”   r   r§   r,   r,   r-   Útest_rename_columns+  s    
r­   c                 C   sî   t g d¢g d¢g d¢dœƒ}|jtjd�}| ¡ }| rTt t|dƒt|dƒ¡spJ ‚nt t|dƒt|dƒ¡rpJ ‚d|jd	< t t|dƒt|dƒ¡r–J ‚| r¶t t|d
ƒt|dƒ¡s¶J ‚t g d¢g d¢g d¢dœƒ}t	 
||¡ t	 
||¡ d S )Nr   r   r   r   r©   r    r   r   r   r¢   r   rª   )r   r7   r8   r«   r   r"   r#   r   r(   r1   r’   )r)   r*   r   Zdf2_origr§   r,   r,   r-   Ú!test_rename_columns_modify_parent=  s    
r®   c                 C   sÚ   t g d¢ddœƒ}| ¡ }dd„ }| |¡}t t|dƒt|dƒ¡sHJ ‚d|jd< | r€t ||¡ t t|dƒt|dƒ¡rºJ ‚n:t g d	¢ddœƒ}t ||¡ t t|dƒt|dƒ¡sºJ ‚t t|d
ƒt|d
ƒ¡sÖJ ‚d S )Nr   ç      ø?r‰   c                 S   s   | S ©Nr,   ©r*   r,   r,   r-   ÚtestfuncU  s    ztest_pipe.<locals>.testfuncr   r   r   rª   r   )	r   r   Úpiper"   r#   r   r(   r1   r’   ©r)   r*   r”   r²   r   r§   r,   r,   r-   Ú	test_pipeQ  s    

rµ   c                 C   sÐ   t g d¢ddœƒ}| ¡ }dd„ }| |¡}t t|dƒt|dƒ¡sHJ ‚| rvt ||¡ t t|dƒt|dƒ¡r°J ‚n:t g d¢ddœƒ}t ||¡ t t|dƒt|dƒ¡s°J ‚t t|dƒt|dƒ¡sÌJ ‚d S )	Nr   r¯   r‰   c                 S   s   d| j d< | S )Nr‚   r   ©r(   r±   r,   r,   r-   r²   m  s    
z%test_pipe_modify_df.<locals>.testfuncr   r   ©r‚   r   r   )r   r   r³   r"   r#   r   r1   r’   r´   r,   r,   r-   Útest_pipe_modify_dfi  s    
r¸   c                 C   sÈ   t g d¢g d¢g d¢dœƒ}| ¡ }|jddgd�}| rVt t|dƒt|dƒ¡srJ ‚nt t|dƒt|dƒ¡rrJ ‚d|jd	< t t|dƒt|dƒ¡r˜J ‚| r¸t t|dƒt|dƒ¡s¸J ‚t ||¡ d S )
Nr   r   r   r   r   r   r©   r   r   )	r   r   r=   r"   r#   r   r(   r1   r’   r“   r,   r,   r-   Útest_reindex_columns€  s    
r¹   c                 C   s   | S r°   r,   ©Úidxr,   r,   r-   r;   —  r<   c                 C   s   |   ¡ S r°   )Úviewrº   r,   r,   r-   r;   ˜  r<   c                 C   s   |   ¡ S r°   r?   rº   r,   r,   r-   r;   ™  r<   c                 C   s   t | ƒS r°   )Úlistrº   r,   r,   r-   r;   š  r<   )Z	identicalr¼   r   Úvaluesc                 C   sÊ   t g d¢g d¢g d¢dœƒ}| ¡ }|j| |jƒd�}|rXt t|dƒt|dƒ¡stJ ‚nt t|dƒt|dƒ¡rtJ ‚d|jd< t t|dƒt|dƒ¡ršJ ‚|rºt t|d	ƒt|d	ƒ¡sºJ ‚t 	||¡ d S )
Nr   r   r   r   rr   r   r   r   r   )
r   r   r=   r   r"   r#   r   r(   r1   r’   )r   r)   r*   r”   r   r,   r,   r-   Útest_reindex_rows”  s    
r¿   c                 C   s  t g d¢g d¢g d¢dœƒ}| ¡ }|jdd�}|j ¡  | rxt t|dƒt|dƒ¡sZJ ‚t t|dƒt|dƒ¡s°J ‚n8t t|dƒt|dƒ¡r”J ‚t t|dƒt|dƒ¡r°J ‚d	|jd
< t t|dƒt|dƒ¡rÖJ ‚| röt t|dƒt|dƒ¡söJ ‚t	 
||¡ d S )Nr   r   r   r   r   r©   r   r   r   r   )r   r   r–   r$   r‘   r"   r#   r   r(   r1   r’   r“   r,   r,   r-   Útest_drop_on_column²  s    

rÀ   c                 C   s°   t g d¢g d¢g d¢dœƒ}| ¡ }| d¡}|j ¡  | rZt t|dƒt|dƒ¡svJ ‚nt t|dƒt|dƒ¡rvJ ‚d|jd< | r t t|dƒt|dƒ¡r J ‚t	 
||¡ d S )	Nr   r   r   r   rN   r   r   r   )r   r   Úselect_dtypesr$   r‘   r"   r#   r   r(   r1   r’   r“   r,   r,   r-   Útest_select_dtypesÅ  s    


rÂ   Úfilter_kwargsÚitemsr   ÚlikeÚregexc                 C   s¬   t g d¢g d¢g d¢dœƒ}| ¡ }|jf i |¤Ž}| rVt t|dƒt|dƒ¡srJ ‚nt t|dƒt|dƒ¡rrJ ‚| rœd|jd< t t|dƒt|dƒ¡rœJ ‚t ||¡ d S )Nr   r   r   r   r   r   r   )	r   r   Úfilterr"   r#   r   r(   r1   r’   )r)   rÃ   r*   r”   r   r,   r,   r-   Útest_filterÙ  s    
rÈ   c                 C   sÒ   t ddgddgddggtddƒd	d
gd�}| ¡ }|jdd�}| r`t t|d	ƒt|d	ƒ¡s|J ‚nt t|d	ƒt|d	ƒ¡r|J ‚d|jd< | rÂt t|d	ƒt|d	ƒ¡r¦J ‚t t|d
ƒt|d
ƒ¡sÂJ ‚t 	||¡ d S )Nr   r   r   r   r   r   ú
2020-01-01ú
2020-01-03r   r   ©r   r    r   )rn   r   )
r   r   r   Úshiftr"   r#   r   r(   r1   r’   r“   r,   r,   r-   Útest_shift_no_opî  s    ý
rÍ   c                 C   sX   t ddgddgddggtddƒd	d
gd�}|jddd�}t t|d	ƒt|d	ƒ¡rTJ ‚d S )Nr   r   r   r   r   r   rÉ   rÊ   r   r   rË   r   ©rn   rF   )r   r   rÌ   r"   r#   r   )r)   r*   r   r,   r,   r-   Útest_shift_index  s    ýrÏ   c                 C   sÄ   t ddgddgddggtddƒd	d
gd�}| ¡ }tddƒ|_|jddd�}| rnt t|d	ƒt|d	ƒ¡sŠJ ‚nt t|d	ƒt|d	ƒ¡rŠJ ‚d|jd< | r´t t|d	ƒt|d	ƒ¡r´J ‚t	 
||¡ d S )Nr   r   r   r   r   r   rÉ   rÊ   r   r   rË   ú
2020-01-02z
2020-01-04Z1D©rn   r]   r   r   )r   r   r   r   rÌ   r"   r#   r   r(   r1   r’   r“   r,   r,   r-   Útest_shift_rows_freq  s    ý
rÒ   c                 C   sÜ   t ddgddgddggtddƒd	�}|jddd
�}t t|dƒt|dƒ¡sNJ ‚t |¡� d|jd< W d   ƒ n1 sx0    Y  | rØt t|dƒt|dƒ¡r¢J ‚t tj	dgtj	dgtj	dggtddƒd	�}t 
||¡ d S )Nr   r   r   r   r   r   rÉ   rÐ   r©   rÎ   r   r   )r   r   rÌ   r"   r#   r   r1   r2   r(   Únanr’   )r)   r3   r*   r   r§   r,   r,   r-   Útest_shift_columns#  s     ÿ(ÿþrÔ   c                 C   s"  t g d¢g d¢g d¢dœƒ}| ¡ }|d d … }| d¡}t |jt|dƒ¡sRJ ‚t t|dƒt|dƒ¡snJ ‚| r”d|jd< t |jt|dƒ¡r”J ‚t 	|¡� d|jd< W d   ƒ n1 s¾0    Y  | röt t|dƒt|dƒ¡rèJ ‚t 
||¡ n(t g d¢g d	¢g d¢dœƒ}t 
||¡ d S )
Nr   r   r   r   r   r   r   r   )r   r   r   )r   r   Úpopr"   r#   r¾   r   r(   r1   r2   r’   )r)   r3   r*   r”   Zview_originalr‹   r§   r,   r,   r-   Útest_pop7  s     

(rÖ   Úfuncc                 C   s
   |   |¡S r°   rC   ©ÚxÚyr,   r,   r-   r;   P  r<   c                 C   s   | j |jdd�S )Nr   ©rF   )rD   r   rØ   r,   r,   r-   r;   Q  r<   c                 C   s   | j |jjtddƒ dd�S )Nr   r   rÛ   )rD   r   r(   ÚslicerØ   r,   r,   r-   r;   R  r<   c                 C   s°   t g d¢ddœƒ}| ¡ }|ddg  ¡ }|||ƒ\}}| rZt t|dƒt|dƒ¡svJ ‚nt t|dƒt|dƒ¡rvJ ‚d|jd< | r t t|dƒt|dƒ¡r J ‚t ||¡ d S )Nr   r   r‰   r   r   r   ©r   r   r"   r#   r   r(   r1   r’   )r)   r×   r*   r”   Z
df_changedr   Ú_r,   r,   r-   Útest_align_frameM  s    	
rß   c                 C   sÜ   t ddgƒ}| ¡ }| ¡ }| |¡\}}| rXt |j|j¡sBJ ‚t |j|j¡s€J ‚n(t |j|j¡rlJ ‚t |j|j¡r€J ‚d|jd< d|jd< | rÀt |j|j¡r¬J ‚t |j|j¡rÀJ ‚t ||¡ t ||¡ d S )Nr   r   r   )	r	   r   rD   r"   r#   r¾   r(   r1   r˜   )r)   r…   rš   Z	ser_otherr‡   Zser_other_resultr,   r,   r-   Útest_align_seriesf  s     

rà   c                 C   sœ   t g d¢g d¢dœƒ}| ¡ }|j|dd�\}}t t|dƒt|dƒ¡sLJ ‚t t|dƒt|dƒ¡shJ ‚| r˜d|jd	< t ||¡ d|jd	< t ||¡ d S )
Nr   r   r‰   Fr?   r   r   r   ©r   r   )	r   r   rD   r"   r#   r   Úlocr1   r’   )r)   r*   r”   r   Zdf3r,   r,   r-   Útest_align_copy_false|  s    

rã   c                 C   sÒ   t g d¢g d¢dœƒ}tg d¢dd�}| ¡ }| ¡ }|j|ddd�\}}t t|d	ƒt|d	ƒ¡sfJ ‚t t|d
ƒt|d
ƒ¡s‚J ‚t t|dƒt|dƒ¡sžJ ‚| rÎd|jd< t 	||¡ d|jd< t 
||¡ d S )Nr   r   r‰   rÙ   ©ÚnameFr   )r   rF   r   r   rá   )r   r	   r   rD   r"   r#   r   râ   r1   r’   r˜   )r)   r*   r…   rš   r”   r   r‡   r,   r,   r-   Ú!test_align_with_series_copy_falseŒ  s    

ræ   c                 C   s:  t g d¢ƒ}| ¡ }|d d …  ¡ }t |jt|dƒ¡s<J ‚t |¡� d|j	d< W d   ƒ n1 sf0    Y  | ršt |jt|dƒ¡rŒJ ‚t 
||¡ n| ¡ }d|j	d< t 
||¡ |d d …  ¡ }t |¡� d|j	d< W d   ƒ n1 sò0    Y  | �rt || ¡ ¡ n"| ¡  ¡ }d|j	d< t ||¡ d S )Nr   r   r   )r	   r   Úto_framer"   r#   r¾   r   r1   r2   r(   r˜   r’   )r)   r3   r…   rš   r*   r§   r,   r,   r-   Útest_to_frameŸ  s(    (
(
rè   Úaxr    c                 C   sÔ   t g d¢g d¢dœƒ}| ¡ }d}tjt|d�� | ||¡}W d   ƒ n1 sR0    Y  | r~t t|dƒt|dƒ¡sšJ ‚nt t|dƒt|dƒ¡ršJ ‚d|j	d< | rÄt t|dƒt|dƒ¡rÄJ ‚t 
||¡ d S )	Nr   r   r‰   rs   rt   r   r   r   ©r   r   r1   ry   rz   rS   r"   r#   r   r(   r’   )r)   ré   r*   r”   r~   r   r,   r,   r-   Útest_swapaxes_noopÃ  s    *
rë   c                 C   sÜ   t g d¢g d¢dœg d¢d�}| ¡ }d}tjt|d�� | dd	¡}W d   ƒ n1 sZ0    Y  | r†t t|d
ƒt|dƒ¡s¢J ‚nt t|d
ƒt|dƒ¡r¢J ‚d|j	d< | rÌt t|d
ƒt|dƒ¡rÌJ ‚t 
||¡ d S )Nr   r   r‰   )rÙ   rÚ   Úzrr   rs   rt   r   r    rÙ   r   r   r   rê   )r)   r*   r”   r~   r   r,   r,   r-   Útest_swapaxes_single_block×  s    *
rí   c                  C   sŠ   t ddgddœƒ} d}tjt|d�� | jddd	�} W d   ƒ n1 sH0    Y  d
| jd< t d
dgddgdœddgd�}t | |¡ d S )Nr   r   r   r‰   rs   rt   r   r    )Zaxis1Zaxis2r‚   r   ©r   r   r   r   rr   )r   r1   ry   rz   rS   r(   r’   )r*   r~   r§   r,   r,   r-   Útest_swapaxes_read_only_arrayê  s    ,
rï   zmethod, idxc                 C   s   | j dd�j dd�S )NFr/   r?   r±   r,   r,   r-   r;   ÷  r<   c                 C   s   |   ¡   ¡ S r°   )r�   r±   r,   r,   r-   r;   ø  r<   c                 C   s   | j tjd�j tjd�S )Nr©   )r7   r8   r«   r9   r±   r,   r,   r-   r;   ù  r<   c                 C   s   | j dd�jdd�S )NFr/   Únumber)Úinclude)r   rÁ   r±   r,   r,   r-   r;   ú  r<   )úshallow-copyr�   r7   rÁ   c           	      C   sò   t g d¢g d¢g d¢dœƒ}| ¡ }| o6| jjjdk}||ƒ}t |oJ|¡� d|jd|f< W d   ƒ n1 sr0    Y  |sŒt ||¡ ||ƒ}t |ož|¡� d|jd< W d   ƒ n1 sÂ0    Y  |sît |jd d …|d …f |¡ d S )Nr   r   r   r   rò   r   r   )	r   r   rv   rw   rx   r1   r2   r(   r’   )	r|   r5   r»   r)   r3   r*   r”   Zdf2_is_viewr   r,   r,   r-   Útest_chained_methodsô  s    ,(ró   Úobjrä   c                 C   s¢   t tddd�tddd�gƒ|_| ¡ }| ¡ }| rPt t|dƒt|dƒ¡slJ ‚nt t|dƒt|dƒ¡rlJ ‚d|jd< t t|dƒt|dƒ¡r’J ‚t	 
||¡ d S )Nz2012-1-1r\   ©r]   z2012-1-2r   r   )r   r   r   r   rZ   r"   r#   r   r(   r1   Úassert_equal©r)   rô   Úobj_origÚobj2r,   r,   r-   Útest_to_timestamp  s    
rú   c                 C   sž   t tdƒtdƒgƒ|_| ¡ }|jdd�}| rLt t|dƒt|dƒ¡shJ ‚nt t|dƒt|dƒ¡rhJ ‚d|jd< t t|dƒt|dƒ¡rŽJ ‚t	 
||¡ d S )Nú
2019-12-31ú
2020-12-31ÚYrõ   r   r   )r   r
   r   r   r_   r"   r#   r   r(   r1   rö   r÷   r,   r,   r-   Útest_to_period&  s    
rþ   c                 C   s¢   t g d¢g d¢g d¢dœƒ}| ¡ }| d¡}| rPt t|dƒt|dƒ¡slJ ‚nt t|dƒt|dƒ¡rlJ ‚d|jd< t t|d	ƒt|d	ƒ¡r’J ‚t ||¡ d S )
Nr   r   r   r   r   r   r   rî   r   )	r   r   r£   r"   r#   r   r(   r1   r’   r“   r,   r,   r-   Útest_set_index8  s    

rÿ   c                  C   s>   t g d¢ddœƒ} |  d¡}| ¡ }d| jd< t ||¡ d S )Nr   r   r‰   r   r‚   r   )r   r£   r   r(   r1   r’   ©r*   r‹   r§   r,   r,   r-   Ú4test_set_index_mutating_parent_does_not_mutate_indexI  s
    

r  c                 C   sÌ   t g d¢g d¢g d¢dœƒ}| ¡ }| d¡}| rNt t|dƒt|dƒ¡sNJ ‚d|jd	< t t|dƒt|dƒ¡rtJ ‚| r”t t|d
ƒt|dƒ¡s”J ‚t g d¢g d¢g d¢dœƒ}t ||¡ t ||¡ d S )Nr   r   r   r   ZCoW_ÚCoW_ar   r   r   ÚCoW_cr   rª   )r  ZCoW_br  )	r   r   Z
add_prefixr"   r#   r   r(   r1   r’   r¬   r,   r,   r-   Útest_add_prefixR  s    

ÿr  c                 C   sÌ   t g d¢g d¢g d¢dœƒ}| ¡ }| d¡}| rNt t|dƒt|dƒ¡sNJ ‚d|jd	< t t|dƒt|dƒ¡rtJ ‚| r”t t|d
ƒt|dƒ¡s”J ‚t g d¢g d¢g d¢dœƒ}t ||¡ t ||¡ d S )Nr   r   r   r   Z_CoWÚa_CoWr   r   r   Úc_CoWr   rª   )r  Zb_CoWr  )	r   r   Z
add_suffixr"   r#   r   r(   r1   r’   r¬   r,   r,   r-   Útest_add_suffixg  s    

ÿr  z	axis, val)r   ç      @c                 C   s¦   t g d¢d|dgddœƒ}| ¡ }|j|d�}| rPt t|dƒt|dƒ¡slJ ‚nt t|dƒt|dƒ¡rlJ ‚d|jd	< | r–t t|dƒt|dƒ¡r–J ‚t ||¡ d S )
Nr   r   r   r�   r   rÛ   r   r   r   )	r   r   Údropnar"   r#   r   r(   r1   r’   )r)   rF   Úvalr*   r”   r   r,   r,   r-   Útest_dropnay  s    
r  r
  r   r  c                 C   s~   t d|dgƒ}| ¡ }| ¡ }| r8t |j|j¡sLJ ‚nt |j|j¡rLJ ‚d|jd< | rnt |j|j¡rnJ ‚t ||¡ d S )Nr   r   r   )	r	   r   r	  r"   r#   r¾   r(   r1   r˜   )r)   r
  r…   rš   r‡   r,   r,   r-   Útest_dropna_seriesŠ  s    
r  c                 C   s   |   ¡ S r°   ©Úheadr±   r,   r,   r-   r;   ž  r<   c                 C   s
   |   d¡S )Nr   r  r±   r,   r,   r-   r;   Ÿ  r<   c                 C   s   |   ¡ S r°   ©Útailr±   r,   r,   r-   r;      r<   c                 C   s
   |   d¡S )Nr   r  r±   r,   r,   r-   r;   ¡  r<   c                 C   s@  t g d¢g d¢dœƒ}| ¡ }| |ƒ}|j ¡  |rlt t|dƒt|dƒ¡rPJ ‚t t|dƒt|dƒ¡rlJ ‚t |¡� d|j	d< W d   ƒ n1 s–0    Y  |rÞt t|dƒt|dƒ¡rÀJ ‚t t|dƒt|dƒ¡rÜJ ‚nRt t|dƒt|dƒ¡súJ ‚t |¡� d|j	d< W d   ƒ n1 �s&0    Y  t 
||¡ d S )	Nr   r   r‰   r   r   r   r   r   )r   r   r$   r‘   r"   r#   r   r1   r2   r(   r’   )r5   r)   r3   r*   r”   r   r,   r,   r-   Útest_head_tail›  s     

(*r  c                 C   s0  t ddgtddgtd�dddœƒ}| ¡ }| ¡ }| r’t t|dƒt|dƒ¡sRJ ‚|rtt t|dƒt|dƒ¡r�J ‚qÊt t|dƒt|dƒ¡sÊJ ‚n8t t|dƒt|dƒ¡r®J ‚t t|dƒt|dƒ¡rÊJ ‚d	|j	d
< d|j	d< | �r t t|dƒt|dƒ¡�rJ ‚t t|dƒt|dƒ¡�r J ‚t 
||¡ d S )Nr   r   rÙ   rÚ   ©Údtyperœ   r   r   r   r   r�   rî   )r   r	   Úobjectr   rX   r"   r#   r   r1   r(   r’   )r)   Úusing_infer_stringr*   r”   r   r,   r,   r-   Útest_infer_objects¾  s$    ÿ

r  c              	   C   sò   t ddgtddgtd�dttdƒtdƒgdd�td	d
gtd�dœƒ}| ¡ }t|dƒ}t|dƒ}t|dƒ}d|jd< d|jd< tdƒ|jd< | rît |t|dƒ¡s¦J ‚|rÂt	 |t|dƒ¡sØJ ‚nt |t|dƒ¡rØJ ‚t |t|dƒ¡sîJ ‚d S )Nr   r   rÙ   rÚ   r  rû   rü   r  rì   Úw)r   r   r   r�   Úer   r   r�   r   r   rî   ú
2018-12-31©r   r   )
r   r	   r  r
   rX   r   r(   r"   r#   r1   )r)   r  r*   Úarr_aÚarr_bÚarr_dr,   r,   r-   Útest_infer_objects_no_referenceØ  s.    ÿùÿ




r  c                 C   sÜ   t ddgtddgtd�dttdƒtdƒgdd�d	œƒ}|d d … }| ¡ }t|d
ƒ}t|dƒ}t|dƒ}d|jd< d|jd< tdƒ|jd< | rØt |t|d
ƒ¡r¤J ‚|r¬t	rÂt |t|dƒ¡rÂJ ‚t |t|dƒ¡sØJ ‚d S )Nr   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,   r,   r-   Útest_infer_objects_referenceú  s,    ÿüÿ





r  Úkwargsr   )ÚbeforeÚafterrF   c                 C   s®   t g d¢dddœƒ}| ¡ }|jf i |¤Ž}|j ¡  | rXt t|dƒt|dƒ¡stJ ‚nt t|dƒt|dƒ¡rtJ ‚d|jd< | ržt t|dƒt|dƒ¡ržJ ‚t	 
||¡ d S )Nr   r   r   r   r   r   r   )r   r   rV   r$   r‘   r"   r#   r   r(   r1   r’   ©r)   r   r*   r”   r   r,   r,   r-   Útest_truncate  s    

r$  ZassignZdrop_duplicatesc                 C   s¦   t dg d¢iƒ}| ¡ }t||ƒƒ }|j ¡  | rPt t|dƒt|dƒ¡slJ ‚nt t|dƒt|dƒ¡rlJ ‚d|jd< | r–t t|dƒt|dƒ¡r–J ‚t	 
||¡ d S )Nr   r   r   r   )r   r   Úgetattrr$   r‘   r"   r#   r   r(   r1   r’   )r)   r5   r*   r”   r   r,   r,   r-   Útest_assign_drop_duplicates.  s    

r&  c                 C   sv   |  ¡ }| ddg¡}| r0t |j|j¡sDJ ‚nt |j|j¡rDJ ‚d|jd< | rft |j|j¡rfJ ‚t ||¡ d S )Nr   r   )r   Ztaker"   r#   r¾   r(   r1   rö   r÷   r,   r,   r-   Ú	test_take@  s    
r'  c                 C   s„   t dddd�|_| ¡ }| dd¡}| r>t |j|j¡sRJ ‚nt |j|j¡rRJ ‚d|jd< | rtt |j|j¡rtJ ‚t 	||¡ d S )Nz
2018-04-09r   Z1D20minrÑ   z0:00z1:00r   )
r   r   r   Zbetween_timer"   r#   r¾   r(   r1   rö   r÷   r,   r,   r-   Útest_between_timeQ  s    
r(  c                 C   s®   t ddgddœƒ}t dddgdœƒ}| ¡ }| |¡}| rXt t|dƒt|dƒ¡stJ ‚nt t|dƒt|dƒ¡rtJ ‚d|jd< | ržt t|dƒt|dƒ¡ržJ ‚t ||¡ d S )Nr   r   r   r‰   )r   r   r   rî   )	r   r   rA   r"   r#   r   r(   r1   r’   )r)   r*   Úotherr”   r   r,   r,   r-   Útest_reindex_likeb  s    

r*  c                 C   sx   t g d¢ƒ}| ¡ }| ¡ }| r6t |j|j¡sJJ ‚nt |j|j¡rJJ ‚d|jd< t |j|j¡rhJ ‚t ||¡ d S )Nr   r   )	r	   r   Z
sort_indexr"   r#   r¾   r(   r1   r˜   ©r)   r…   rš   r‡   r,   r,   r-   Útest_sort_indext  s    
r,  zobj, kwargsr   Zbyc                 C   sŒ   |  ¡ }|jf i |¤Ž}| r:t t|dƒt|dƒ¡sVJ ‚nt t|dƒt|dƒ¡rVJ ‚d|jd< t t|dƒt|dƒ¡r|J ‚t ||¡ d S )Nr   r   )r   Úsort_valuesr"   r#   r   r(   r1   rö   )r)   rô   r   rø   rù   r,   r,   r-   Útest_sort_values…  s    
r.  c                 C   sÆ   |  ¡ }|d d … }|jf ddi|¤Ž t t|dƒt|dƒ¡sDJ ‚t |¡� d|jd< W d   ƒ n1 sn0    Y  | r¦t t|dƒt|dƒ¡r˜J ‚t ||¡ nt t|dƒt|dƒ¡sÂJ ‚d S )NÚinplaceTr   r   )	r   r-  r"   r#   r   r1   r2   r(   rö   )r)   rô   r   r3   rø   r¼   r,   r,   r-   Útest_sort_values_inplace˜  s    (r0  Údecimals)éÿÿÿÿr   r   c                 C   s:  t ddgddœƒ}| ¡ }|j|d�}| rœt t|dƒt|dƒ¡sFJ ‚|dkr~ttjƒtdƒk r~t t|d	ƒt|d	ƒ¡sšJ ‚qÔt t|d	ƒt|d	ƒ¡rÔJ ‚n8t t|dƒt|dƒ¡r¸J ‚t t|d	ƒt|d	ƒ¡rÔJ ‚d
|j	d< d|j	d< | �r*t t|dƒt|dƒ¡�rJ ‚t t|d	ƒt|d	ƒ¡�r*J ‚t 
||¡ d S )Nr   r   r   r‰   )r1  r   r   z
2.4.0.dev0r   r�   rî   r   r   )r   r   Úroundr1   r#   r   r   r"   Ú__version__r(   r’   )r)   r3   r1  r*   r”   r   r,   r,   r-   Ú
test_round­  s     

r5  c                 C   sº   t jg d¢ddgd�}tdg d¢i|d�}| ¡ }|jddgd�}| rdt t|dƒt|dƒ¡s€J ‚nt t|dƒt|dƒ¡r€J ‚d	|jd
< | rªt t|dƒt|dƒ¡rªJ ‚t	 
||¡ d S )N©©r   r   ©r   r   ©r   r   )r   r   ÚoneÚtwo©Únamesr   rž   rr   ©Úorderr   r   )r   Úfrom_tuplesr   r   Úreorder_levelsr"   r#   r   r(   r1   r’   ©r)   r   r*   r”   r   r,   r,   r-   Útest_reorder_levelsÈ  s    ÿ
rC  c                 C   sž   t jg d¢ddgd�}tg d¢|d�}| ¡ }|jddgd�}| rXt |j|j¡slJ ‚nt |j|j¡rlJ ‚d|jd< | rŽt |j|j¡rŽJ ‚t	 
||¡ d S )	Nr6  r:  r;  r<  rž   rr   r>  r   )r   r@  r	   r   rA  r"   r#   r¾   r(   r1   r˜   r™   r,   r,   r-   Útest_series_reorder_levelsÛ  s    ÿ
rD  c                 C   sŒ   t jg d¢ddgd�}||_| ¡ }| ¡ }| rFt |j|j¡sZJ ‚nt |j|j¡rZJ ‚d|jd< | r|t |j|j¡r|J ‚t	 
||¡ d S )N©r7  r8  r9  r:  r;  r<  r   )r   r@  r   r   r†   r"   r#   r¾   r(   r1   rö   )r)   rô   r   rø   rù   r,   r,   r-   Útest_swaplevelî  s    
rF  c                 C   sª   t g d¢g d¢g d¢dœƒ}| ¡ }|jg d¢dd�}| rXt t|dƒt|dƒ¡stJ ‚nt t|dƒt|dƒ¡rtJ ‚d|jd	< t t|dƒt|dƒ¡ršJ ‚t ||¡ d S )
Nr   r   r   r   r   rÛ   r   r   r   )	r   r   rH   r"   r#   r   r(   r1   r’   r“   r,   r,   r-   Útest_frame_set_axis   s    
rG  c                 C   sv   t g d¢ƒ}| ¡ }|jg d¢dd�}| r<t ||¡sLJ ‚nt ||¡rLJ ‚d|jd< t ||¡rfJ ‚t ||¡ d S )Nr   r   r   rÛ   r   )r	   r   rH   r"   r#   r(   r1   r˜   r+  r,   r,   r-   Útest_series_set_axis  s    
rH  c                 C   s²   t g d¢ƒ}| ¡ }|jdd�}t ||¡s0J ‚t |¡� d|jd< W d   ƒ n1 sZ0    Y  | r†t ||¡rxJ ‚t ||¡ n(t ||¡s–J ‚t g d¢ƒ}t ||¡ d S )Nr   F)rg   r   rª   )	r	   r   ri   r"   r#   r1   r2   r(   r˜   ©r)   r3   r…   rš   r‡   r§   r,   r,   r-   Útest_set_flags"  s    (rJ  ZmapperrJ   c                 C   s°   t dg d¢itg d¢dd�d�}| ¡ }|jf i |¤Ž}| rZt t|dƒt|dƒ¡svJ ‚nt t|dƒt|dƒ¡rvJ ‚d|jd< | r t t|dƒt|dƒ¡r J ‚t 	||¡ d S )Nr   rž   rä   rr   r   r   )
r   r   r   rM   r"   r#   r   r(   r1   r’   r#  r,   r,   r-   Útest_rename_axis5  s     
rK  zfunc, tz)re   zEurope/Berlin)rc   Nc                 C   sŽ   t ddgtddd|d�d�}| ¡ }t||ƒdƒ}| rLt |j|j¡s`J ‚nt |j|j¡r`J ‚d|jd< t |j|j¡r~J ‚t 	||¡ d S )	Nr   r   z2014-08-01 09:00Úh)Ústartr]   rn   rq   rr   ra   r   )
r	   r   r   r%  r"   r#   r¾   r(   r1   r˜   )r)   r×   rq   r…   rš   r‡   r,   r,   r-   Útest_tz_convert_localizeF  s    ÿ
rN  c                 C   s  t jg d¢ddgd�}tg d¢g d¢g d¢dœ|d	�}| ¡ }| d
¡}| r†t t|dƒt|dƒ¡shJ ‚t t|dƒt|dƒ¡s¾J ‚n8t t|dƒt|dƒ¡r¢J ‚t t|dƒt|dƒ¡r¾J ‚d
|jd< t t|dƒt|dƒ¡räJ ‚| �rt t|dƒt|dƒ¡�sJ ‚t	 
||¡ d S )NrE  r:  r;  r<  r   r   ©é   é   é	   r   rr   r   r   r   r   r   )r   r@  r   r   Z	droplevelr"   r#   r   r(   r1   r’   rB  r,   r,   r-   Útest_droplevel\  s     

rS  c                 C   sÄ   t dg d¢iƒ}| ¡ }| ¡ }t |jt|dƒ¡s8J ‚t |¡� d|j	d< W d   ƒ n1 sb0    Y  | r–t |jt|dƒ¡rˆJ ‚t 
||¡ n*t |jt|dƒ¡s®J ‚|jd dksÀJ ‚d S )Nr   r   r   rá   )r   r   Zsqueezer"   r#   r¾   r   r1   r2   r(   r’   râ   )r)   r3   r*   r”   Zseriesr,   r,   r-   Útest_squeezet  s    (rT  c              
   C   sÜ   t g d¢g d¢g d¢dœƒ}| ¡ }tdƒD ]ª}| ¡ D ]œ\}}t t||ƒt||ƒ¡s\J ‚t |¡� d|j	d< W d   ƒ n1 s†0    Y  | r¾t t||ƒt||ƒ¡r°J ‚t 
||¡ q8|jd|f dks8J ‚q8q,d S )Nr   r   rO  r   r   r   )r   r   ÚrangerÄ   r"   r#   r   r1   r2   r(   r’   râ   )r)   r3   r*   r”   Úirå   r…   r,   r,   r-   Ú
test_itemsˆ  s    (rW  r  rN   ÚInt64c                 C   sÂ   t ddgdddœ|d�}|d d … }| ¡ }t |¡� d|||k< W d   ƒ n1 sX0    Y  | r�t t|dƒt|dƒ¡r‚J ‚t ||¡ n.t t|dƒt|dƒ¡s¬J ‚|jd dks¾J ‚d S )Nr   r   r   r  r   r   r   )	r   r   r1   r2   r"   r#   r   r’   r(   )r)   r  r3   r*   r¼   r”   r,   r,   r-   Útest_putmaskž  s    *rY  c                 C   sL   t ddgdddœ|d�}t|dƒ}d|||k< | rHt |t|dƒ¡sHJ ‚d S )Nr   r   r   r  r   r   ©r   r   r"   r#   ©r)   r  r*   r  r,   r,   r-   Útest_putmask_no_reference¯  s
    
r\  Zfloat64ZFloat64c                 C   sV   t ddgddœ|d�}t|dƒ}t dddgiƒ|||k< | rRt |t|dƒ¡sRJ ‚d S )Nr¯   r   r‰   r  r   r  r   rZ  r[  r,   r,   r-   Ú$test_putmask_aligns_rhs_no_reference¹  s
    
r]  zval, exp, warnT)r   FNc           	      C   s¤  t ddgdddœƒ}|d d … }| ¡ }t g d¢g d¢gtdƒd�}|rzt ¡ � |||< W d   ƒ q°1 sn0    Y  n6tj|dd	�� |||< W d   ƒ n1 s¦0    Y  | �rFt t|d
ƒt|d
ƒ¡rÒJ ‚t t|dƒt|dƒ¡|u sòJ ‚t t|dƒt|dƒ¡�sJ ‚|j	 
d¡|u�s&J ‚|j	 
d¡�r8J ‚t ||¡ nZ|dk�r t t|d
ƒt|d
ƒ¡�snJ ‚t t|dƒt|dƒ¡�sŒJ ‚|jd dk�s J ‚d S )Nr   r   r¯   r   )TFFÚabcr©   zincompatible dtypert   r   r   r   r   r   )r   r   r½   r1   r2   ry   r"   r#   r   r$   r—   r’   r(   )	r)   r
  ÚexpÚwarnr3   r*   r¼   r”   Zindexerr,   r,   r-   Ú"test_putmask_dont_copy_some_blocksÃ  s,    ÿ
(& 
ra  c                 C   s   |   | dkd¡S ©Nr   r�   ©Úwhere©r…   r,   r,   r-   r;   é  r<   c                 C   s   |   | dkd¡S rb  ©Úmaskre  r,   r,   r-   r;   ê  r<   c                 C   sŒ   t g d¢|d�}| ¡ }||ƒ}| r>t t|ƒt|ƒ¡sVJ ‚nt t|ƒt|ƒ¡rVJ ‚d|jd< | r|t t|ƒt|ƒ¡r|J ‚t ||¡ d S )Nr   r  r�   r   )r	   r   r"   r#   r   r(   r1   r˜   ©r)   r  r×   r…   rš   r‹   r,   r,   r-   Útest_where_mask_noopå  s    	
ri  c                 C   s   |   | dk d¡S rb  rc  re  r,   r,   r-   r;     r<   c                 C   s   |   | dkd¡S rb  rf  re  r,   r,   r-   r;     r<   c                 C   sH   t g d¢|d�}| ¡ }||ƒ}t t|ƒt|ƒ¡r8J ‚t ||¡ d S )Nr   r  )r	   r   r"   r#   r   r1   r˜   rh  r,   r,   r-   Útest_where_maskþ  s
    	rj  z
dtype, val)rN   g      %@)rX  r�   c                 C   s   |   | dk |¡S ©Nr   rc  ©r*   r
  r,   r,   r-   r;     r<   c                 C   s   |   | dk|¡S rk  rf  rl  r,   r,   r-   r;     r<   c                 C   sÀ   t g d¢g d¢dœ|d�}| ¡ }|||ƒ}| rjt t|dƒt|dƒ¡sLJ ‚t t|dƒt|dƒ¡r†J ‚nt t|dƒt|dƒ¡r†J ‚d|jd< | r°t t|dƒt|dƒ¡r°J ‚t ||¡ d S )	Nr   )éüÿÿÿéûÿÿÿiúÿÿÿr‰   r  r   r   r�   rî   rÝ   )r)   r  r
  r×   r*   r”   r‹   r,   r,   r-   Ú%test_where_mask_noop_on_single_column  s    	

ro  rg  rd  c              	   C   s"  t g d¢ddœƒ}| ¡ }| rÌt ¡ �0 t|d |ƒ|d dkddd� W d   ƒ n1 s\0    Y  t ||¡ t ¡ �2 t|dg |ƒ|d dkddd� W d   ƒ n1 s²0    Y  t ||¡ �nRtjtsØtnd d	d
��0 t|d |ƒ|d dkddd� W d   ƒ n1 �s0    Y  t d ¡�^ t	dd ƒ�2 t|dg |ƒ|d dkddd� W d   ƒ n1 �st0    Y  W d   ƒ n1 �s”0    Y  t d ¡�d t	dd ƒ�8 t||d dk |ƒ|d dkddd� W d   ƒ n1 �sô0    Y  W d   ƒ n1 �s0    Y  d S )N)r   r   r   r   r‰   r   r   r   T©r/  úinplace methodrt   úmode.chained_assignment)
r   r   r1   Úraises_chained_assignment_errorr%  r’   ry   r   rz   r   )r)   r×   r*   r”   r,   r,   r-   Útest_chained_where_mask*  s(    
>
@
þ@brt  c                 C   s¦   t dg d¢itdddd�d�}| ¡ }|jdd�}| rTt t|dƒt|dƒ¡spJ ‚nt t|dƒt|dƒ¡rpJ ‚d	|jd
< t t|dƒt|dƒ¡r–J ‚t 	||¡ d S )Nr   )g        Nç       @ç      @z1/1/2000r   ÚminrÑ   rr   rõ   r   r   )
r   r   r   Zasfreqr"   r#   r   r(   r1   r’   r“   r,   r,   r-   Útest_asfreq_noopF  s    
þ
rx  c                 C   sN   t dddœg d¢d�}| ¡ }| ¡ D ]\}}d|jd< q&| rJt ||¡ d S )Nr   r   r‰   r   rr   r‚   )r   r   Ziterrowsr(   r1   r’   )r)   r*   r”   rÞ   Úsubr,   r,   r-   Útest_iterrowsZ  s    rz  c                 C   sÄ   t ddtjdgiƒ}|d d … }| ¡ }t |¡� |jdd� W d   ƒ n1 sT0    Y  t |¡� d|jd< W d   ƒ n1 sˆ0    Y  | r¤t ||¡ nt dg d¢iƒ}t ||¡ d S )	Nr   r¯   r   Trp  ç      Y@r   )r{  r¯   r   )	r   r"   rÓ   r   r1   r2   Zffillr(   r’   )r)   r3   r*   r¼   r§   r,   r,   r-   Útest_interpolate_creates_copyd  s    *(r|  c                 C   s  t g d¢g d¢g d¢dœƒ}| ¡ }|jd d�}| dt g d¢¡¡ | r„t t|dƒt|dƒ¡sfJ ‚t t|d	ƒt|d	ƒ¡s¼J ‚n8t t|dƒt|dƒ¡r J ‚t t|d	ƒt|d	ƒ¡r¼J ‚d
|jd< t 	||¡ | rôt t|dƒt|dƒ¡sòJ ‚nt t|dƒt|dƒ¡�rJ ‚d S )Nr   r   rO  r   r/   r   )r2  éþÿÿÿéýÿÿÿr   r   r   rá   )
r   r   Úisetitemr"   Úarrayr#   r   râ   r1   r’   r“   r,   r,   r-   Útest_isetitemv  s    
r�  zsingle-blockzmixed-blockc                 C   sö   t g d¢tjg d¢|d�dœƒ}tg d¢ƒ}| ¡ }| d|¡ | rnt t|dƒt|ƒ¡s^J ‚|j 	d¡rnJ ‚d|j
d< t ||¡ t g d¢tjg d¢|d�dœƒ}tg d¢ƒ}| d|¡ d|j
d< t g d¢tjg d¢|d�dœƒ}t ||¡ d S )	Nr   r   r  r‰   rO  r   r   rá   )r   r"   r€  r	   r   r  r#   r   r$   r—   râ   r1   r˜   r’   )r)   r  r*   r…   rš   r§   r,   r,   r-   Útest_isetitem_seriesŒ  s     
 
 r‚  c                 C   sè   t g d¢dddœƒ}t g d¢ddœƒ}| ddg|¡ | r„t t|dƒt|dƒ¡sVJ ‚t t|d	ƒt|d	ƒ¡srJ ‚|j d¡r¼J ‚n8t t|dƒt|dƒ¡r J ‚t t|d	ƒt|d	ƒ¡r¼J ‚| ¡ }d
|jd< d
|jd< t	 
||¡ d S )Nr   r   r   r   r   r‰   r   r   r   r‚   r   rî   )r   r  r"   r#   r   r$   r—   r   r(   r1   r’   )r)   r*   Úrhsr§   r,   r,   r-   Útest_isetitem_frame§  s    

r„  r¦   c              	   C   s6  t g d¢g d¢dœƒ}| ¡ }| |¡}| r|t t|dƒt|dƒ¡sHJ ‚d|jd< t t|dƒt|dƒ¡rnJ ‚t ||¡ n¶|r”t	|t
ƒrŽtnd }nt	|tƒr¢tnd }tddƒ�D t |¡� d|jd< W d   ƒ n1 sÜ0    Y  W d   ƒ n1 sú0    Y  t	|tƒ�rt ||¡ n|jd dk�s2J ‚d S )	Nr   r   r‰   r   r   Úchained_assignmentr`  r   )r   r   Úgetr"   r#   r   r(   r1   r’   Ú
isinstancer8   rz   r½   r   r   ry   )r)   r3   r¦   r*   r”   r‹   r`  r,   r,   r-   Útest_get¸  s"    

Frˆ  z	axis, keyr   )r   r   c              	   C   sŽ  |dko| }|p|o|dk}t g d¢g d¢tjg d¢|d�dœƒ}| ¡ }	|j||d�}
|dksf|r‚t t|d	ƒt|
ƒ¡s–J ‚n| r–|
j d
¡s–J ‚| s¢|r®|s®d
|
j	d
< n¤|rðt
 |pÀ|dk¡� d
|
j	d
< W d   ƒ n1 sä0    Y  nbtddƒ�F t
 t¡� d
|
j	d
< W d   ƒ n1 �s(0    Y  W d   ƒ n1 �sH0    Y  | �sh|�sv|d
k�rvt
 ||	¡ n|j	d d
k�sŠJ ‚d S )NrN   r   r   r   rO  r  r   rÛ   r   r   r…  r`  r   )r   r"   r€  r   Úxsr#   r   r$   r—   r(   r1   r2   r   ry   r   r’   )r)   r3   r}   rF   r¦   r  Zsingle_blockZis_viewr*   r”   r‹   r,   r,   r-   Útest_xsÕ  s,     ÿ*JrŠ  rF   z
key, level)Úl1r   r9  c              	   C   s6  t  d¡ dd¡}tjddgg d¢gddgd	�}t||td
ƒd�}|dkrV| ¡  ¡ }| ¡ }	|j	|||d�}
|dkržt  
t||jd ƒt|
|
jd ƒ¡sžJ ‚|r´|dkr®tnd }n| sÂ|sÂt}nd }tddƒ�D t |¡� d|
jd< W d   ƒ n1 sü0    Y  W d   ƒ n1 �s0    Y  t ||	¡ d S )Né   r   r   r‹  Úl2r   Zlev1Zlev2r<  r^  rË   r   )ÚlevelrF   r   r…  r`  r   )r"   ZarangeZreshaper   Zfrom_productr   r½   rŠ   r   r‰  r#   r   r    rz   r   r   r1   ry   r(   r’   )r)   r3   r}   r¦   rŽ  rF   Zarrr   r*   r”   r‹   r`  r,   r,   r-   Útest_xs_multiindexú  s(    ÿHr�  c                 C   sî   t g d¢g d¢dœƒ}t ddgidgd�}| ¡ }|d d … }t |¡� | |¡ W d   ƒ n1 sh0    Y  t g d¢g d¢dœƒ}t ||¡ | rÞt ||¡ t t|d	ƒt|d	ƒ¡sÀJ ‚t t|dƒt|dƒ¡rêJ ‚nt ||¡ d S )
N©ç      ð?ru  rv  )ç      @g      @ç      @r‰   r   ç      Y@r   rr   )r’  r”  r“  r   )	r   r   r1   r2   Úupdater’   r"   r#   r   )r)   r3   Zdf1r   Zdf1_origr¼   r§   r,   r,   r-   Útest_update_frame  s    (r–  c                 C   s¬   t g d¢ƒ}t dgdgd�}| ¡ }|d d … }|rht ¡ � | |¡ W d   ƒ qr1 s\0    Y  n
| |¡ t g d¢ƒ}t ||¡ | rœt ||¡ nt ||¡ d S )Nr�  r”  r   rr   )r‘  r”  rv  )r	   r   r1   r2   r•  r˜   )r)   r3   Zser1r‡   Z	ser1_origr¼   r§   r,   r,   r-   Útest_update_series/  s    
*
r—  c              	   C   sà  t dg d¢iƒ}tdgdgd�}| ¡ }| rºt ¡ � |d  |¡ W d   ƒ n1 sX0    Y  t ||¡ t ¡ �$ |dg  | ¡ ¡ W d   ƒ n1 s 0    Y  t ||¡ �n"tjt	sÆt
nd dd�� |d  |¡ W d   ƒ n1 sô0    Y  t d ¡�P tdd ƒ�$ |dg  | ¡ ¡ W d   ƒ n1 �s@0    Y  W d   ƒ n1 �s`0    Y  t d ¡�V tdd ƒ�* ||d dk  | ¡ ¡ W d   ƒ n1 �s²0    Y  W d   ƒ n1 �sÒ0    Y  d S )	Nr   r   r”  r   rr   rq  rt   rr  )r   r	   r   r1   rs  r•  r’   rç   ry   r   rz   r   )r)   r*   r‡   r”   r,   r,   r-   Útest_update_chained_assignmentD  s*    
,
2
þ,Tr˜  c                 C   sv   t g d¢ƒ}| ¡ }t|ƒ}|d9 }| rNt t|ƒ|¡r<J ‚t |t|ƒ¡ n$t t|ƒ|¡sbJ ‚t |t|ƒ¡ d S ©Nr   r   )r	   r   r   r"   r#   r1   Zassert_numpy_array_equal)r)   r…   rš   Údatar,   r,   r-   Útest_inplace_arithmetic_series`  s    r›  c                 C   s˜   t g d¢ƒ}| ¡ }|d d … }t |¡� |d9 }W d   ƒ n1 sH0    Y  | r|t t|ƒt|ƒ¡rnJ ‚t ||¡ nt t|ƒt|ƒ¡s”J ‚d S r™  )r	   r   r1   r2   r"   r#   r   r˜   )r)   r3   r…   rš   r¼   r,   r,   r-   Ú-test_inplace_arithmetic_series_with_referencer  s    &rœ  Fc                 C   sŠ   t g d¢ddœƒ}| ¡ }|j|d�}|s.|r2| rPt t|dƒt|dƒ¡slJ ‚nt t|dƒt|dƒ¡rlJ ‚d|jd< | r†t ||¡ d S )	Nr   r   r‰   r?   r   r   r‚   r   )	r   r   rŠ   r"   r#   r   r(   r1   r’   )r)   r   r}   r*   r”   r‹   r,   r,   r-   Útest_transpose�  s    
r�  c                 C   sZ   t g d¢ddœƒ}| ¡ }|j}t t|dƒt|dƒ¡r<J ‚d|jd< | rVt ||¡ d S )Nr   r¯   r‰   r   r   r‚   r   )	r   r   ÚTr"   r#   r   r(   r1   r’   ©r)   r*   r”   r‹   r,   r,   r-   Útest_transpose_different_dtypes‘  s    
r   c                 C   s:   t dg d¢idd�}|j}t t|dƒt|dƒ¡r6J ‚d S )Nr   r   rX  r  r   )r   rž  r"   r#   r   )r)   r*   r‹   r,   r,   r-   Útest_transpose_ea_single_columnœ  s    r¡  c                 C   sj   t g d¢ddœƒ}| ¡ }dd„ }t |¡� | |¡ W d   ƒ n1 sL0    Y  | rft ||¡ d S )Nr   r   r‰   c                 S   s   d| j d< | S ©Nr‚   r   r¶   re  r,   r,   r-   r×   §  s    
z"test_transform_frame.<locals>.func)r   r   r1   r2   Ú	transformr’   )r)   r3   r*   r”   r×   r,   r,   r-   Útest_transform_frame£  s    (r¤  c                 C   sd   t g d¢ƒ}| ¡ }dd„ }t |¡� | |¡ W d   ƒ n1 sF0    Y  | r`t ||¡ d S )Nr   c                 S   s   d| j d< | S r¢  r¶   re  r,   r,   r-   r×   µ  s    
z#test_transform_series.<locals>.func)r	   r   r1   r2   r£  r˜   )r)   r3   r…   rš   r×   r,   r,   r-   Útest_transform_series±  s    (r¥  c                  C   sH   t ddgddœƒ} |  ¡ }d|jd< tddgddgd	�}t ||¡ d S )
Nr   r   r   r‰   r‚   r   r   r   rr   )r   Úcountr(   r	   r1   r˜   r   r,   r,   r-   Útest_count_read_only_array¿  s
    
r§  c                 C   sØ   t g d¢ƒ}| ¡ }tjtdd�� | ¡ }W d   ƒ n1 s@0    Y  t t|ƒt|ƒ¡sbJ ‚| rv|j	 
d¡rvJ ‚t |¡� d|jd< W d   ƒ n1 s 0    Y  | r¼t ||¡ nt g d¢ƒ}t ||¡ d S )Nr   zis deprecatedrt   r   r‚   r·   )r	   r   r1   ry   rz   r¼   r"   r#   r   r$   r—   r2   r(   r˜   rI  r,   r,   r-   Útest_series_viewÇ  s    &(r¨  c                 C   s˜   t dg d¢iƒ}tg d¢ƒ}| ¡ }|jd|dd� | rdt t|ƒt|dƒ¡sRJ ‚|j d¡r~J ‚nt t|ƒt|dƒ¡r~J ‚d|j	d< t
 ||¡ d S )Nr   r   r   r   )râ   ÚvalueÚcolumnr‚   rî   )r   r	   r   Úinsertr"   r#   r   r$   r—   r(   r1   r˜   )r)   r*   r…   rš   r,   r,   r-   Útest_insert_seriesÛ  s    
r¬  c                 C   s|   t g d¢ddœƒ}| ¡ }| d¡}| rFt t|dƒt|dƒ¡sbJ ‚nt t|dƒt|dƒ¡rbJ ‚d|jd< t ||¡ d S )Nr   r   r‰   úc = a+br   r‚   r   )	r   r   Úevalr"   r#   r   r(   r1   r’   rŸ  r,   r,   r-   Ú	test_evalê  s    

r¯  c                 C   s˜   t g d¢ddœƒ}| ¡ }|d d … }|jddd� t t|dƒt|dƒ¡sPJ ‚t |¡� d|jd	< W d   ƒ n1 sz0    Y  | r”t 	||¡ d S )
Nr   r   r‰   r­  Trp  r   r‚   r   )
r   r   r®  r"   r#   r   r1   r2   r(   r’   )r)   r3   r*   r”   Zdf_viewr,   r,   r-   Útest_eval_inplaceø  s    (r°  c                 C   sè   t ddgddgdœƒ}| ¡ }dd„ }t |¡� |j|dd� W d   ƒ n1 sT0    Y  | rpt ||¡ n|jd	 d
ks‚J ‚t ddgddgdœƒ}| ¡ }t d ¡� |j|dd� W d   ƒ n1 sÎ0    Y  t ||¡ d S )Nr   r   r   r   )r    r¡   c                 S   s   d| d< | S )Nr‚   r¡   r,   )Úrowr,   r,   r-   r£    s    z(test_apply_modify_row.<locals>.transformrÛ   )r   r¡   r‚   r   r   )r   r   r1   r2   Úapplyr’   râ   ry   )r)   r3   r*   r”   r£  r,   r,   r-   Útest_apply_modify_row  s    ,,r³  )Únumpyr"   ZpytestZpandas.compatr   r   Zpandas.errorsr   ZpandasÚpdr   r   r   r   r	   r
   r   r   r   Zpandas._testingZ_testingr1   Zpandas.tests.copy_view.utilr   Zpandas.util.versionr   r.   r4   ÚmarkZparametrizer�   rˆ   rŒ   r•   Z
RangeIndexr›   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ÿ   r  r  r  rÓ   r  r  r  r  r  r  r$  r&  r'  r(  r*  r,  r.  r0  r5  rC  rD  rF  rG  rH  rJ  rK  rN  rS  rT  rW  rY  r\  r]  rz   ra  ri  rj  ro  rt  rx  rz  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ü  ,%ïë)#ïë)
 
üø

ÿ
ýþ
$





üø

(
(
	

üþ	
"

þþ

$
$
*þ
*þ

$

ÿ


	
	ÿ
þþþþ
þþ
ÿÿ!