o
    �¨ÊhÅ ã                   @   s^	  d dl Zd dl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„ 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„ d.d„ d/d„ d0d„ d1d„ gg d2¢d�d3d4„ ƒƒZej d	g d
¢¡d5d6„ ƒZd7d8„ Zej d9e d d:¡ed;d:gƒg¡d<d=„ ƒZd>d?„ Zd@dA„ Z dBdC„ Z!dDdE„ Z"dFdG„ Z#dHdI„ Z$ejjd9dJd„ dKd„ dLd„ dMd„ gg dN¢d�dOdP„ ƒZ%dQdR„ Z&dSdT„ Z'ej dUdVdWgidXdWidYdWig¡dZd[„ ƒZ(d\d]„ Z)d^d_„ Z*d`da„ Z+dbdc„ Z,ddde„ Z-ej dfdgd„ dhd„ did„ g¡djdk„ ƒZ.dldm„ Z/dndo„ Z0dpdq„ Z1drds„ Z2ej dtd9dug¡dvdw„ ƒZ3dxdy„ Z4dzd{„ Z5e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ƒ„ ƒZ6ej d„ed;d:gdWd…�edWd;d:giƒg¡d†d‡„ ƒZ7ej d„ed;d:gdWd…�edWd;d:giƒg¡dˆd‰„ ƒZ8dŠd‹„ Z9dŒd�„ Z:dŽd�„ Z;d�d‘„ Z<ej d’d“d;ej=fg¡d”d•„ ƒZ>ej d–d—d˜g¡d™dš„ ƒZ?ej dd›d„ dœd„ d�d„ džd„ g¡dŸd „ ƒZ@d¡d¢„ ZAd£d¤„ ZBd¥d¦„ ZCej d§dWd¨d;d©œd d;d d©œg¡dªd«„ ƒZDej dd¬d­g¡d®d¯„ ƒZEej d„ed;d:gƒedWd;d:giƒg¡d°d±„ ƒZFej d„ed;d:gƒedWd;d:giƒg¡d²d³„ ƒZGd´dµ„ ZHd¶d·„ ZIej d¸eg d¹¢dWd…�i fedWg d¹¢iƒdºdWifg¡d»d¼„ ƒZJej d¸eg d¹¢dWd…�i fedWg d¹¢iƒdºdWifg¡d½d¾„ ƒZKej d¿g dÀ¢¡dÁdÂ„ ƒZLdÃdÄ„ ZMdÅdÆ„ ZNej d„eg d¹¢ƒedWg d¹¢iƒg¡dÇdÈ„ ƒZOdÉdÊ„ ZPdËdÌ„ ZQdÍdÎ„ ZRej d§dÏdÐid9dÐig¡dÑdÒ„ ƒZSej dÓdÔdÕg¡dÖd×„ ƒZTdØdÙ„ ZUdÚdÛ„ ZVdÜdÝ„ ZWej dÞdßdàg¡dádâ„ ƒZXej dÞdßdàg¡dãdä„ ƒZYej dÞdådæg¡dçdè„ ƒZZej déd˜dêe[fdëg¡dìdí„ ƒZ\ej dÞdßdàg¡ej dfdîd„ dïd„ g¡dðdñ„ ƒƒZ]ej dÞdßdàg¡ej dfdòd„ dód„ g¡dôdõ„ ƒƒZ^ej död÷døg¡ej dfdùd„ dúd„ g¡dûdü„ ƒƒZ_ej dfdýdþg¡dÿ�d „ ƒZ`�d�d„ Za�d�d„ Zb�d�d„ Zc�d�d„ ZdejjdÞdßdåg�d	�d
gd��d�d„ ƒZe�d�d„ Zfej �ddWdWgg¡�d�d„ ƒZgej �d�d�dg¡ejjdÞdßdåg�d	�d
gd��d�d„ ƒƒZhej �dd d;g¡ej �d�d�dg¡�d�d„ ƒƒZi�d�d„ Zj�d�d „ Zk�d!�d"„ Zl�d#�d$„ Zm�d%�d&„ Znej d	dê�d'g¡�d(�d)„ ƒZo�d*�d+„ Zp�d,�d-„ Zq�d.�d/„ Zr�d0�d1„ Zs�d2�d3„ Zt�d4�d5„ Zu�d6�d7„ Zv�d8�d9„ Zw�d:�d;„ Zx�d<�d=„ ZydS (>  é    N)ÚSettingWithCopyWarning)	Ú	DataFrameÚIndexÚ
MultiIndexÚPeriodÚSeriesÚ	TimestampÚ
date_rangeÚoption_contextÚperiod_range)Ú	get_arrayc                 C   sÔ   t g d¢g d¢g d¢dœƒ}| ¡ }|j|jusJ ‚|j|jus"J ‚|j |j¡s+J ‚|j |j¡s4J ‚t t|dƒt|dƒ¡rBJ ‚| rZ|jj	d j
 ¡ rOJ ‚|jj	d j
 ¡ rZJ ‚d|jd< |jd dkshJ ‚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)   úU/var/www/html/env/lib/python3.10/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�}| r9|j|jusJ ‚|j|jus&J ‚|j |j¡s/J ‚|j |j¡s8J ‚n|j|ju sAJ ‚|j|ju sIJ ‚t t|dƒt|dƒ¡sWJ ‚| ro|jj	d j
 ¡ sdJ ‚|jj	d	 j
 ¡ soJ ‚| r�d|jd
< |jd
 d	ksJ ‚t t|dƒt|dƒ¡r�J ‚t t|dƒt|dƒ¡s›J ‚d S t |¡� d|jd
< W d   ƒ n1 s²w   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_shallow+   s.   
 ÿ r1   r   )TNFÚmethodc                 C   s   | j tj|d�S )N©r   r   )ÚrenameÚstrÚlower©r'   r   r)   r)   r*   Ú<lambda>T   ó    r8   c                 C   s   | j ddg|d�S )Nr   r   r3   )Úreindexr7   r)   r)   r*   r8   U   ó    c                 C   ó   | j | |d�S ©N©r   ©Úreindex_liker7   r)   r)   r*   r8   V   ó    c                 C   ó   | j | |d�d S ©Nr>   r   ©Úalignr7   r)   r)   r*   r8   W   r;   c                 C   ó   | j g d¢d|d�S ©Nr   r   )Úaxisr   ©Úset_axisr7   r)   r)   r*   r8   X   ó    c                 C   ó   | j d|d�S ©NÚtest©r   r   ©Úrename_axisr7   r)   r)   r*   r8   Y   rA   c                 C   rL   )NrN   r3   rP   r7   r)   r)   r*   r8   Z   rA   c                 C   ó   | j ddi|d�S )Nr   Úint64r>   ©Úastyper7   r)   r)   r*   r8   [   r;   c                 C   ó   | j dd|d�S ©Nr   r>   ©Úswapaxesr7   r)   r)   r*   r8   ]   r9   c                 C   ó   | j dd|d�S ©Nr   r   r>   ©Útruncater7   r)   r)   r*   r8   ^   r9   c                 C   ó   | j |d�S r=   ©Úinfer_objectsr7   r)   r)   r*   r8   _   ó    c                 C   r^   r=   ©Úto_timestampr7   r)   r)   r*   r8   `   ra   c                 C   rL   ©NÚD)Úfreqr   ©Ú	to_periodr7   r)   r)   r*   r8   a   rA   c                 C   rL   ©Nú
US/Centralr>   ©Útz_localizer7   r)   r)   r*   r8   b   rA   c                 C   rL   ri   ©Ú
tz_convertr7   r)   r)   r*   r8   c   rA   c                 C   rL   ©NF)Úallows_duplicate_labelsr   ©Ú	set_flagsr7   r)   r)   r*   r8   d   rA   )r4   r:   r@   rE   rJ   Úrename_axis0Úrename_axis1rU   rY   r]   r`   rc   rh   rl   rn   rr   )Úidsc           
      C   sj  d }d| 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�}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 rud}tjt|d�� |||d�}W d   ƒ n1 sow   Y  n|||d�}|p€|du }	| j jj 	d¡r“|s“|s“|du r“d}	|	r¥t
 t|dƒt|dƒ¡s£J ‚d S t
 t|dƒt|dƒ¡r³J ‚d S )Nrc   ú
2012-01-01re   r   ©rf   Úperiodsrh   rl   rn   úEurope/Brussels©rf   rx   Útzr   r   r   r   ©r   rY   ú"'DataFrame.swapaxes' is deprecated©Úmatchr>   Fzreindex-r   )ÚnodeÚcallspecÚidr   r	   r   r.   Úassert_produces_warningÚFutureWarningÚ
startswithr   r    r   )
Úrequestr2   r   r&   Úusing_array_managerr   r'   ÚmsgÚdf2Úshare_memoryr)   r)   r*   Útest_methods_copy_keywordP   s0   - ÿ€  r‹   c                 C   rR   )Nr   éd   rO   ©r4   ©Úserr   r)   r)   r*   r8   ¡   r;   c                 C   s   | j d |d�S r=   r�   rŽ   r)   r)   r*   r8   ¢   rA   c                 C   s   | j | j|d�S )NrO   )r:   r   rŽ   r)   r)   r*   r8   £   r9   c                 C   r<   r=   r?   rŽ   r)   r)   r*   r8   ¤   rA   c                 C   rB   rC   rD   rŽ   r)   r)   r*   r8   ¥   r;   c                 C   rF   rG   rI   rŽ   r)   r)   r*   r8   ¦   rK   c                 C   rL   rM   rP   rŽ   r)   r)   r*   r8   §   rA   c                 C   rL   )NrS   r>   rT   rŽ   r)   r)   r*   r8   ¨   rA   c                 C   rZ   )Nr   r   r>   )Ú	swaplevelrŽ   r)   r)   r*   r8   ©   r9   c                 C   rV   rW   rX   rŽ   r)   r)   r*   r8   ª   r9   c                 C   rZ   r[   r\   rŽ   r)   r)   r*   r8   «   r9   c                 C   r^   r=   r_   rŽ   r)   r)   r*   r8   ¬   ra   c                 C   r^   r=   rb   rŽ   r)   r)   r*   r8   ­   ra   c                 C   rL   rd   rg   rŽ   r)   r)   r*   r8   ®   rA   c                 C   rL   ri   rk   rŽ   r)   r)   r*   r8   ¯   rA   c                 C   rL   ri   rm   rŽ   r)   r)   r*   r8   °   rA   c                 C   rL   ro   rq   rŽ   r)   r)   r*   r8   ±   rA   )zrename (dict)r4   r:   r@   rE   rJ   rs   rU   r�   rY   r]   r`   rc   rh   rl   rn   rr   c           	      C   sT  d }d| j jjv rtdddd�}n@d| j jjv r tdddd�}n1d| j jjv r/tdddd�}n"d| j jjv r?tdddd	d
�}nd| j jjv rQt 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 szw   Y  n|||d�}|p‹|du }|rœt t|ƒt|ƒ¡sšJ ‚d S t t|ƒt|ƒ¡r¨J ‚d S )Nrc   rv   re   r   rw   rh   rl   rn   ry   rz   r�   r   r   r|   rY   z'Series.swapaxes' is deprecatedr~   r>   F)r€   r�   r‚   r   r	   r   Úfrom_arraysr   r.   rƒ   r„   r   r    r   )	r†   r2   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�}| p|du p|d u }|o| }|r2t t|dƒt|dƒ¡s0J ‚d S 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 ¡  | r=t t|dƒt|dƒ¡s/J ‚t t|dƒt|dƒ¡s=J ‚d	|jd
< t t|dƒt|dƒ¡rPJ ‚| r`t t|dƒt|dƒ¡s`J ‚t	 
||¡ d S )Nr   r   r   r   )é
   é   é   r|   r   r   r   )r   r   )r   r   Úreset_indexr!   Ú_verify_integrityr   r    r   r%   r.   Ú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�}| r)t t|ƒt|ƒ¡s J ‚|j d¡r(J ‚nt t|ƒt|ƒ¡r5J ‚d|jd< t	 
||¡ d S )Nr   r   r|   T)Údropr   rŒ   )r   r   r›   r   r    r   r!   Ú_has_no_referencer%   r.   Ú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)Úobserved)r   Ú	set_indexÚgroupbyÚsumr.   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�}| r)t t|dƒt|dƒ¡s)J ‚d|jd	< t t|dƒt|dƒ¡r<J ‚| rLt t|d
ƒt|dƒ¡sLJ ‚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   r4   r5   Úupperr   r    r   r%   r.   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�}| ¡ }| r*t t|dƒt|dƒ¡s)J ‚nt t|dƒt|dƒ¡r8J ‚d|jd	< t t|dƒt|dƒ¡rKJ ‚| 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   r4   r5   r·   r   r   r    r   r%   r.   r�   )r&   r'   r‰   Údf2_origr³   r)   r)   r*   Ú!test_rename_columns_modify_parent8  s   
r»   c                 C   sÚ   t g d¢ddœƒ}| ¡ }dd„ }| |¡}t t|dƒt|dƒ¡s$J ‚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
ƒ¡skJ ‚d S )Nr   ç      ø?r”   c                 S   ó   | S ©Nr)   ©r'   r)   r)   r*   ÚtestfuncP  s   ztest_pipe.<locals>.testfuncr   r   r   r¶   r   )	r   r   Úpiper   r    r   r%   r.   r�   ©r&   r'   rŸ   rÀ   r‰   r³   r)   r)   r*   Ú	test_pipeL  s   

 rÃ   c                 C   sÐ   t g d¢ddœƒ}| ¡ }dd„ }| |¡}t t|dƒt|dƒ¡s$J ‚| r;t ||¡ t t|dƒt|dƒ¡r:J ‚nt g d¢ddœƒ}t ||¡ t t|dƒt|dƒ¡sXJ ‚t t|dƒt|dƒ¡sfJ ‚d S )	Nr   r¼   r”   c                 S   ó   d| j d< | S )NrŒ   r   ©r%   r¿   r)   r)   r*   rÀ   h  ó   
z%test_pipe_modify_df.<locals>.testfuncr   r   ©rŒ   r   r   )r   r   rÁ   r   r    r   r.   r�   rÂ   r)   r)   r*   Útest_pipe_modify_dfd  s   
 rÈ   c                 C   sÈ   t g d¢g d¢g d¢dœƒ}| ¡ }|jddgd�}| r+t t|dƒt|dƒ¡s*J ‚nt t|dƒt|dƒ¡r9J ‚d|jd	< t t|dƒt|dƒ¡rLJ ‚| 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ž   r)   r)   r*   Útest_reindex_columns{  s   
rÉ   c                 C   r½   r¾   r)   ©Úidxr)   r)   r*   r8   ’  s    c                 C   ó   |   ¡ S r¾   )ÚviewrÊ   r)   r)   r*   r8   “  ó    c                 C   rÌ   r¾   r>   rÊ   r)   r)   r*   r8   ”  rÎ   c                 C   s   t | ƒS r¾   )ÚlistrÊ   r)   r)   r*   r8   •  rÎ   )Ú	identicalrÍ   r   Úvaluesc                 C   sÊ   t g d¢g d¢g d¢dœƒ}| ¡ }|j| |jƒd�}|r,t t|dƒt|dƒ¡s+J ‚nt t|dƒt|dƒ¡r:J ‚d|jd< t t|dƒt|dƒ¡rMJ ‚|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�   )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 ¡  | r<t t|dƒt|dƒ¡s-J ‚t t|dƒt|dƒ¡s;J ‚nt t|dƒt|dƒ¡rJJ ‚t t|dƒt|dƒ¡rXJ ‚d	|jd
< t t|dƒt|dƒ¡rkJ ‚| 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%   r.   r�   rž   r)   r)   r*   Útest_drop_on_column­  s   

rÓ   c                 C   s°   t g d¢g d¢g d¢dœƒ}| ¡ }| d¡}|j ¡  | r-t t|dƒt|dƒ¡s,J ‚nt t|dƒt|dƒ¡r;J ‚d|jd< | rPt t|dƒt|dƒ¡rPJ ‚t	 
||¡ d S )	Nr   r   r   r   rS   r   r   r   )r   r   Úselect_dtypesr!   rœ   r   r    r   r%   r.   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di |¤Ž}| r+t t|dƒt|dƒ¡s*J ‚nt t|dƒt|dƒ¡r9J ‚| rNd|jd< t t|dƒt|dƒ¡rNJ ‚t ||¡ d S )	Nr   r   r   r   r   r   r   r)   )	r   r   Úfilterr   r    r   r%   r.   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�}| r0t t|d	ƒt|d	ƒ¡s/J ‚nt t|d	ƒt|d	ƒ¡r>J ‚d|jd< | rat t|d	ƒt|d	ƒ¡rSJ ‚t t|d
ƒt|d
ƒ¡saJ ‚t 	||¡ d S )Nr   r   r   r   r   r   ú
2020-01-01ú
2020-01-03r   r   ©r   r   r   )rx   r   )
r   r	   r   Úshiftr   r    r   r%   r.   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	ƒ¡r*J ‚d S )Nr   r   r   r   r   r   rÜ   rÝ   r   r   rÞ   r   ©rx   rH   )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�}| r7t t|d	ƒt|d	ƒ¡s6J ‚nt t|d	ƒt|d	ƒ¡rEJ ‚d|jd< | rZt t|d	ƒt|d	ƒ¡rZJ ‚t	 
||¡ d S )Nr   r   r   r   r   r   rÜ   rÝ   r   r   rÞ   ú
2020-01-02z
2020-01-04Ú1D©rx   rf   r   r   )r   r	   r   r   rß   r   r    r   r%   r.   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ƒ¡s'J ‚t |¡� d|jd< W d   ƒ n1 s<w   Y  | rnt t|dƒt|dƒ¡rQJ ‚t tj	dgtj	dgtj	dggtddƒd	�}t 
||¡ d S d S )Nr   r   r   r   r   r   rÜ   rã   rµ   rá   r   r   )r   r	   rß   r   r    r   r.   r/   r%   Únanr�   )r&   r0   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ƒ¡s)J ‚t t|dƒt|dƒ¡s7J ‚| rJd|jd< t |jt|dƒ¡rJJ ‚t 	|¡� d|jd< W d   ƒ n1 s_w   Y  | r|t t|dƒt|dƒ¡rtJ ‚t 
||¡ d S 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%   r.   r/   r�   )r&   r0   r'   rŸ   Úview_originalr–   r³   r)   r)   r*   Útest_pop2  s"   

ÿrë   Úfuncc                 C   s
   |   |¡S r¾   rD   ©ÚxÚyr)   r)   r*   r8   K  ó   
 c                 C   s   | j |jdd�S )Nr   ©rH   )rE   r   rí   r)   r)   r*   r8   L  r9   c                 C   s   | j |jjtddƒ dd�S )Nr   r   rñ   )rE   r   r%   Úslicerí   r)   r)   r*   r8   M  s    c                 C   s°   t g d¢ddœƒ}| ¡ }|ddg  ¡ }|||ƒ\}}| r-t t|dƒt|dƒ¡s,J ‚nt t|dƒt|dƒ¡r;J ‚d|jd< | rPt t|dƒt|dƒ¡rPJ ‚t ||¡ d S )Nr   r   r”   r   r   r   ©r   r   r   r    r   r%   r.   r�   )r&   rì   r'   rŸ   Ú
df_changedr‰   Ú_r)   r)   r*   Útest_align_frameH  s   	
rö   c                 C   sÜ   t ddgƒ}| ¡ }| ¡ }| |¡\}}| r,t |j|j¡s!J ‚t |j|j¡s+J ‚nt |j|j¡r6J ‚t |j|j¡r@J ‚d|jd< d|jd< | r`t |j|j¡rVJ ‚t |j|j¡r`J ‚t ||¡ t ||¡ d S )Nr   r   r   )	r   r   rE   r   r    rÑ   r%   r.   r£   )r&   r�   r¥   Ú	ser_otherr’   Úser_other_resultr)   r)   r*   Útest_align_seriesa  s    

rù   c                 C   s    t g d¢g d¢dœƒ}| ¡ }|j|dd�\}}t t|dƒt|dƒ¡s&J ‚t t|dƒt|dƒ¡s4J ‚| rNd|jd	< t ||¡ d|jd	< t ||¡ d S d S )
Nr   r   r”   Fr>   r   r   r   ©r   r   )	r   r   rE   r   r    r   Úlocr.   r�   )r&   r'   rŸ   r‰   Údf3r)   r)   r*   Útest_align_copy_falsew  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	ƒ¡s3J ‚t t|d
ƒt|d
ƒ¡sAJ ‚t t|dƒt|dƒ¡sOJ ‚| rid|jd< t 	||¡ d|jd< t 
||¡ d S d S )Nr   r   r”   rî   ©ÚnameFr   )r   rH   r   r   rú   )r   r   r   rE   r   r    r   rû   r.   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 s3w   Y  | rMt |jt|dƒ¡rFJ ‚t 
||¡ n| ¡ }d|j	d< t 
||¡ |d d …  ¡ }t |¡� d|j	d< W d   ƒ n1 syw   Y  | rŠt || ¡ ¡ d S | ¡  ¡ }d|j	d< t ||¡ d S )Nr   r   r   )r   r   Úto_framer   r    rÑ   r   r.   r/   r%   r£   r�   )r&   r0   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 s)w   Y  | r?t t|dƒt|dƒ¡s>J ‚nt t|dƒt|dƒ¡rMJ ‚d|j	d< | rbt t|dƒt|dƒ¡rbJ ‚t 
||¡ d S )	Nr   r   r”   r}   r~   r   r   r   ©r   r   r.   rƒ   r„   rY   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 s-w   Y  | rCt t|d
ƒt|dƒ¡sBJ ‚nt t|d
ƒt|dƒ¡rQJ ‚d|j	d< | rft t|d
ƒt|dƒ¡rfJ ‚t 
||¡ d S )Nr   r   r”   )rî   rï   Úzr|   r}   r~   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 s$w   Y  d
| jd< t d
dgddgdœddgd�}t | |¡ d S )Nr   r   r   r”   r}   r~   r   r   )Úaxis1Úaxis2rŒ   r   ©r   r   r   r   r|   )r   r.   rƒ   r„   rY   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*   r8   ò  rK   c                 C   s   |   ¡   ¡ S r¾   )r›   r¿   r)   r)   r*   r8   ó  ra   c                 C   s   | j tjd�j tjd�S )Nrµ   )r4   r5   r·   r6   r¿   r)   r)   r*   r8   ô  s    c                 C   s   | j dd�jdd�S )NFr,   Únumber)Úinclude)r   rÔ   r¿   r)   r)   r*   r8   õ  rK   )úshallow-copyr›   r4   rÔ   c           	      C   sö   t g d¢g d¢g d¢dœƒ}| ¡ }| o| jjjdk}||ƒ}t |o%|¡� d|jd|f< W d   ƒ n1 s9w   Y  |sFt ||¡ ||ƒ}t |oO|¡� d|jd< W d   ƒ n1 saw   Y  |syt |jd d …|d …f |¡ d S d S )Nr   r   r   r   r  r   r   )	r   r   r€   r�   r‚   r.   r/   r%   r�   )	r†   r2   rË   r&   r0   r'   rŸ   Ú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ƒ|_| ¡ }| ¡ }| r(t t|dƒt|dƒ¡s'J ‚nt t|dƒt|dƒ¡r6J ‚d|jd< t t|dƒt|dƒ¡rIJ ‚t	 
||¡ d S )Nz2012-1-1re   ©rf   z2012-1-2r   r   )r   r   r   r   rc   r   r    r   r%   r.   Ú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�}| r&t t|dƒt|dƒ¡s%J ‚nt t|dƒt|dƒ¡r4J ‚d|jd< t t|dƒt|dƒ¡rGJ ‚t	 
||¡ d S )Nú
2019-12-31ú
2020-12-31ÚYr  r   r   )r   r   r   r   rh   r   r    r   r%   r.   r  r  r)   r)   r*   Útest_to_period!  s   
r  c                 C   s¢   t g d¢g d¢g d¢dœƒ}| ¡ }| d¡}| r(t t|dƒt|dƒ¡s'J ‚nt t|dƒt|dƒ¡r6J ‚d|jd< t t|d	ƒt|d	ƒ¡rIJ ‚t ||¡ d S )
Nr   r   r   r   r   r   r   r
  r   )	r   r   r¯   r   r    r   r%   r.   r�   rž   r)   r)   r*   Útest_set_index3  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%   r.   r�   ©r'   r–   r³   r)   r)   r*   Ú4test_set_index_mutating_parent_does_not_mutate_indexD  s
   

r  c                 C   óÌ   t g d¢g d¢g d¢dœƒ}| ¡ }| d¡}| r't t|dƒt|dƒ¡s'J ‚d|jd	< t t|dƒt|dƒ¡r:J ‚| rJt t|d
ƒt|dƒ¡sJJ ‚t g d¢g d¢g d¢dœƒ}t ||¡ t ||¡ d S )Nr   r   r   r   ÚCoW_ÚCoW_ar   r   r   ÚCoW_cr   r¶   )r!  ÚCoW_br"  )	r   r   Ú
add_prefixr   r    r   r%   r.   r�   r¸   r)   r)   r*   Útest_add_prefixM  s   

ÿr%  c                 C   r  )Nr   r   r   r   Ú_CoWÚa_CoWr   r   r   Úc_CoWr   r¶   )r'  Úb_CoWr(  )	r   r   Ú
add_suffixr   r    r   r%   r.   r�   r¸   r)   r)   r*   Útest_add_suffixb  s   

ÿr+  z	axis, val)r   ç      @c                 C   s¦   t g d¢d|dgddœƒ}| ¡ }|j|d�}| r(t t|dƒt|dƒ¡s'J ‚nt t|dƒt|dƒ¡r6J ‚d|jd	< | rKt t|dƒt|dƒ¡rKJ ‚t ||¡ d S )
Nr   r   r   r¨   r   rñ   r   r   r   )	r   r   Údropnar   r    r   r%   r.   r�   )r&   rH   Úvalr'   rŸ   r‰   r)   r)   r*   Útest_dropnat  s   
r/  r.  r   r,  c                 C   s~   t d|dgƒ}| ¡ }| ¡ }| rt |j|j¡sJ ‚n
t |j|j¡r&J ‚d|jd< | r7t |j|j¡r7J ‚t ||¡ d S )Nr   r   r   )	r   r   r-  r   r    rÑ   r%   r.   r£   )r&   r.  r�   r¥   r’   r)   r)   r*   Útest_dropna_series…  s   
r0  c                 C   rÌ   r¾   ©Úheadr¿   r)   r)   r*   r8   ™  rÎ   c                 C   ó
   |   d¡S )Nr   r1  r¿   r)   r)   r*   r8   š  rð   c                 C   rÌ   r¾   ©Útailr¿   r)   r)   r*   r8   ›  rÎ   c                 C   r3  )Nr   r4  r¿   r)   r)   r*   r8   œ  rð   c                 C   s>  t g d¢g d¢dœƒ}| ¡ }| |ƒ}|j ¡  |r6t t|dƒt|dƒ¡r(J ‚t t|dƒt|dƒ¡r6J ‚t |¡� d|j	d< W d   ƒ n1 sKw   Y  |rot t|dƒt|dƒ¡r`J ‚t t|dƒt|dƒ¡rnJ ‚n(t t|dƒt|dƒ¡s}J ‚t |¡� d|j	d< W d   ƒ n1 s’w   Y  t 
||¡ d S )	Nr   r   r”   r   r   r   r   r   )r   r   r!   rœ   r   r    r   r.   r/   r%   r�   )r2   r&   r0   r'   rŸ   r‰   r)   r)   r*   Útest_head_tail–  s$   

ÿÿr6  c                 C   sü   t ddgddddœƒ}| ¡ }| ¡ }| r2t t|dƒt|dƒ¡s#J ‚t t|dƒt|dƒ¡s1J ‚nt t|dƒt|dƒ¡r@J ‚t t|dƒt|dƒ¡rNJ ‚d|jd	< d
|jd< | rvt t|dƒt|dƒ¡rhJ ‚t t|dƒt|dƒ¡rvJ ‚t ||¡ d S )Nr   r   r   rî   r§   r   r   r   r   r¨   r
  )	r   r   r`   r   r    r   r%   r.   r�   rž   r)   r)   r*   Útest_infer_objects¹  s   

r7  c                 C   sÂ   t ddgddttdƒtdƒgdd�dd	œƒ}| ¡ }t|d
ƒ}t|dƒ}t|dƒ}d|jd< d|jd< tdƒ|jd< | r]t |t|d
ƒ¡sGJ ‚t |t|dƒ¡rRJ ‚t |t|dƒ¡s_J ‚d S d S )Nr   r   r   r  r  Úobject©Údtyper   )r   r   r   r¨   Úer   r¨   r   r   r
  ú
2018-12-31©r   r   ©r   r   r   r`   r   r%   r   r    )r&   r'   Úarr_aÚarr_bÚarr_dr)   r)   r*   Útest_infer_objects_no_referenceÎ  s.   ÿùÿ




ürB  c                 C   sÌ   t ddg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< | rbt |t|d	ƒ¡rLJ ‚t |t|d
ƒ¡rWJ ‚t |t|dƒ¡sdJ ‚d S d S )Nr   r   r   r  r  r8  r9  r§   r   r   r¨   r   r   r
  r<  r=  r>  )r&   r'   rÍ   r?  r@  rA  r)   r)   r*   Útest_infer_objects_referenceê  s.   ÿüÿ





ýrC  Úkwargsr   )ÚbeforeÚafterrH   c                 C   s®   t g d¢dddœƒ}| ¡ }|jdi |¤Ž}|j ¡  | r,t t|dƒt|dƒ¡s+J ‚nt t|dƒt|dƒ¡r:J ‚d|jd< | rOt t|dƒt|dƒ¡rOJ ‚t	 
||¡ d S )	Nr   r   r   r   r   r   r   r)   )r   r   r]   r!   rœ   r   r    r   r%   r.   r�   ©r&   rD  r'   rŸ   r‰   r)   r)   r*   Útest_truncate  s   

rH  ÚassignÚdrop_duplicatesc                 C   s¦   t dg d¢iƒ}| ¡ }t||ƒƒ }|j ¡  | r(t t|dƒt|dƒ¡s'J ‚nt t|dƒt|dƒ¡r6J ‚d|jd< | rKt t|dƒt|dƒ¡rKJ ‚t	 
||¡ d S )Nr   r   r   r   )r   r   Úgetattrr!   rœ   r   r    r   r%   r.   r�   )r&   r2   r'   rŸ   r‰   r)   r)   r*   Útest_assign_drop_duplicates  s   

rL  c                 C   sv   |  ¡ }| ddg¡}| rt |j|j¡sJ ‚n
t |j|j¡r"J ‚d|jd< | r3t |j|j¡r3J ‚t ||¡ d S )Nr   r   )r   Útaker   r    rÑ   r%   r.   r  r  r)   r)   r*   Ú	test_take/  s   
rN  c                 C   s„   t dddd�|_| ¡ }| dd¡}| rt |j|j¡sJ ‚n
t |j|j¡r)J ‚d|jd< | r:t |j|j¡r:J ‚t 	||¡ d S )Nz
2018-04-09r   Ú1D20minrå   z0:00z1:00r   )
r	   r   r   Úbetween_timer   r    rÑ   r%   r.   r  r  r)   r)   r*   Útest_between_time@  s   
rQ  c                 C   s®   t ddgddœƒ}t dddgdœƒ}| ¡ }| |¡}| r,t t|dƒt|dƒ¡s+J ‚nt t|dƒt|dƒ¡r:J ‚d|jd< | rOt t|dƒt|dƒ¡rOJ ‚t ||¡ d S )Nr   r   r   r”   )r   r   r   r
  )	r   r   r@   r   r    r   r%   r.   r�   )r&   r'   ÚotherrŸ   r‰   r)   r)   r*   Útest_reindex_likeQ  s   

rS  c                 C   sx   t g d¢ƒ}| ¡ }| ¡ }| rt |j|j¡sJ ‚n
t |j|j¡r%J ‚d|jd< t |j|j¡r4J ‚t ||¡ d S )Nr   r   )	r   r   Ú
sort_indexr   r    rÑ   r%   r.   r£   ©r&   r�   r¥   r’   r)   r)   r*   Útest_sort_indexc  s   
rV  zobj, kwargsr   Úbyc                 C   sŒ   |  ¡ }|jdi |¤Ž}| rt t|dƒt|dƒ¡sJ ‚nt t|dƒt|dƒ¡r+J ‚d|jd< t t|dƒt|dƒ¡r>J ‚t ||¡ d S )Nr   r   r)   )r   Úsort_valuesr   r    r   r%   r.   r  )r&   r  rD  r  r  r)   r)   r*   Útest_sort_valuest  s   
rY  c                 C   sÈ   |  ¡ }|d d … }|jdddi|¤Ž t t|dƒt|dƒ¡s"J ‚t |¡� d|jd< W d   ƒ n1 s7w   Y  | rTt t|dƒt|dƒ¡rLJ ‚t ||¡ d S t t|dƒt|dƒ¡sbJ ‚d S )NÚinplaceTr   r   r)   )	r   rX  r   r    r   r.   r/   r%   r  )r&   r  rD  r0   r  rÍ   r)   r)   r*   Útest_sort_values_inplace‡  s   ÿ r[  Údecimals)éÿÿÿÿr   r   c                 C   s"  t ddgddœƒ}| ¡ }|j|d�}| rEt t|dƒt|dƒ¡s#J ‚|dkr6t t|dƒt|dƒ¡s5J ‚n+t t|dƒt|dƒ¡rDJ ‚nt t|dƒt|dƒ¡rSJ ‚t t|dƒt|dƒ¡raJ ‚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”   )r\  r   r   r   r¨   r
  r   r   )	r   r   Úroundr   r    r   r%   r.   r�   )r&   r0   r\  r'   rŸ   r‰   r)   r)   r*   Ú
test_roundœ  s    

r_  c                 C   sº   t jg d¢ddgd�}tdg d¢i|d�}| ¡ }|jddgd�}| r2t t|dƒt|dƒ¡s1J ‚nt t|dƒt|dƒ¡r@J ‚d	|jd
< | rUt t|dƒt|dƒ¡rUJ ‚t	 
||¡ d S )N©©r   r   ©r   r   ©r   r   )r   r   ÚoneÚtwo©Únamesr   r©   r|   ©Úorderr   r   )r   Úfrom_tuplesr   r   Úreorder_levelsr   r    r   r%   r.   r�   ©r&   r   r'   rŸ   r‰   r)   r)   r*   Útest_reorder_levels¶  s   ÿ
rm  c                 C   sž   t jg d¢ddgd�}tg d¢|d�}| ¡ }|jddgd�}| r,t |j|j¡s+J ‚n
t |j|j¡r6J ‚d|jd< | rGt |j|j¡rGJ ‚t	 
||¡ d S )	Nr`  rd  re  rf  r©   r|   rh  r   )r   rj  r   r   rk  r   r    rÑ   r%   r.   r£   r¤   r)   r)   r*   Útest_series_reorder_levelsÉ  s   ÿ
rn  c                 C   sŒ   t jg d¢ddgd�}||_| ¡ }| ¡ }| r#t |j|j¡s"J ‚n
t |j|j¡r-J ‚d|jd< | r>t |j|j¡r>J ‚t	 
||¡ d S )N©ra  rb  rc  rd  re  rf  r   )r   rj  r   r   r�   r   r    rÑ   r%   r.   r  )r&   r  r   r  r  r)   r)   r*   Útest_swaplevelÜ  s   
rp  c                 C   sª   t g d¢g d¢g d¢dœƒ}| ¡ }|jg d¢dd�}| r,t t|dƒt|dƒ¡s+J ‚nt t|dƒt|dƒ¡r:J ‚d|jd	< t t|dƒt|dƒ¡rMJ ‚t ||¡ d S )
Nr   r   r   r   r   rñ   r   r   r   )	r   r   rJ   r   r    r   r%   r.   r�   rž   r)   r)   r*   Útest_frame_set_axisî  s   
rq  c                 C   sv   t g d¢ƒ}| ¡ }|jg d¢dd�}| rt ||¡sJ ‚nt ||¡r&J ‚d|jd< t ||¡r3J ‚t ||¡ d S )Nr   r   r   rñ   r   )r   r   rJ   r   r    r%   r.   r£   rU  r)   r)   r*   Útest_series_set_axisÿ  s   
rr  c                 C   s´   t g d¢ƒ}| ¡ }|jdd�}t ||¡sJ ‚t |¡� d|jd< W d   ƒ n1 s-w   Y  | rDt ||¡r<J ‚t ||¡ d S t ||¡sLJ ‚t g d¢ƒ}t ||¡ d S )Nr   F)rp   r   r¶   )	r   r   rr   r   r    r.   r/   r%   r£   ©r&   r0   r�   r¥   r’   r³   r)   r)   r*   Útest_set_flags  s   ÿrt  ÚmapperrN   c                 C   s°   t dg d¢itg d¢dd�d�}| ¡ }|jdi |¤Ž}| r-t t|dƒt|dƒ¡s,J ‚nt t|dƒt|dƒ¡r;J ‚d|jd< | rPt t|dƒt|dƒ¡rPJ ‚t 	||¡ d S )Nr   r©   rþ   r|   r   r   r)   )
r   r   r   rQ   r   r    r   r%   r.   r�   rG  r)   r)   r*   Útest_rename_axis#  s    
rv  zfunc, tz)rn   zEurope/Berlin)rl   Nc                 C   sŽ   t ddgtddd|d�d�}| ¡ }t||ƒdƒ}| r&t |j|j¡s%J ‚n
t |j|j¡r0J ‚d|jd< t |j|j¡r?J ‚t 	||¡ d S )	Nr   r   z2014-08-01 09:00Úh)Ústartrf   rx   r{   r|   rj   r   )
r   r	   r   rK  r   r    rÑ   r%   r.   r£   )r&   rì   r{   r�   r¥   r’   r)   r)   r*   Útest_tz_convert_localize4  s   ÿ
ry  c                 C   s  t jg d¢ddgd�}tg d¢g d¢g d¢dœ|d	�}| ¡ }| d
¡}| rCt t|dƒt|dƒ¡s4J ‚t t|dƒt|dƒ¡sBJ ‚nt t|dƒt|dƒ¡rQJ ‚t t|dƒt|dƒ¡r_J ‚d
|jd< t t|dƒt|dƒ¡rrJ ‚| r‚t t|dƒt|dƒ¡s‚J ‚t	 
||¡ d S )Nro  rd  re  rf  r   r   ©é   é   é	   r   r|   r   r   r   r   r   )r   rj  r   r   Ú	droplevelr   r    r   r%   r.   r�   rl  r)   r)   r*   Útest_droplevelJ  s    

r  c                 C   sÆ   t dg d¢iƒ}| ¡ }| ¡ }t |jt|dƒ¡sJ ‚t |¡� d|j	d< W d   ƒ n1 s1w   Y  | rLt |jt|dƒ¡rDJ ‚t 
||¡ d S t |jt|dƒ¡sXJ ‚|jd dksaJ ‚d S )Nr   r   r   rú   )r   r   Úsqueezer   r    rÑ   r   r.   r/   r%   r�   rû   )r&   r0   r'   rŸ   Úseriesr)   r)   r*   Útest_squeezeb  s   ÿr‚  c              
   C   sÜ   t g d¢g d¢g d¢dœƒ}| ¡ }tdƒD ]U}| ¡ D ]N\}}t t||ƒt||ƒ¡s.J ‚t |¡� d|j	d< W d   ƒ n1 sCw   Y  | r_t t||ƒt||ƒ¡rXJ ‚t 
||¡ q|jd|f dksjJ ‚qqd S )Nr   r   rz  r   r   r   )r   r   Úranger×   r   r    r   r.   r/   r%   r�   rû   )r&   r0   r'   rŸ   Úirÿ   r�   r)   r)   r*   Ú
test_itemsv  s   ÿôÿr…  r:  rS   ÚInt64c                 C   sÄ   t ddgdddœ|d�}|d d … }| ¡ }t |¡� d|||k< W d   ƒ n1 s,w   Y  | rIt t|dƒt|dƒ¡rAJ ‚t ||¡ d S t t|dƒt|dƒ¡sWJ ‚|jd dks`J ‚d S )Nr   r   r   r9  r   r   r   )	r   r   r.   r/   r   r    r   r�   r%   )r&   r:  r0   r'   rÍ   rŸ   r)   r)   r*   Útest_putmaskŒ  s   ÿr‡  c                 C   sP   t ddgdddœ|d�}t|dƒ}d|||k< | r$t |t|dƒ¡s&J ‚d S d S )Nr   r   r   r9  r   r   ©r   r   r   r    ©r&   r:  r'   r?  r)   r)   r*   Útest_putmask_no_reference�  s   
ÿrŠ  Úfloat64ÚFloat64c                 C   sZ   t ddgddœ|d�}t|dƒ}t dddgiƒ|||k< | r)t |t|dƒ¡s+J ‚d S d S )Nr¼   r   r”   r9  r   r,  r   rˆ  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�}|r=t ¡ � |||< W d   ƒ n1 s7w   Y  ntj|dd	�� |||< W d   ƒ n1 sSw   Y  | r t t|d
ƒt|d
ƒ¡rhJ ‚t t|dƒt|dƒ¡|u sxJ ‚t t|dƒt|dƒ¡s†J ‚|j	 
d¡|us�J ‚|j	 
d¡r˜J ‚t ||¡ d S |dkrÉt t|d
ƒt|d
ƒ¡s²J ‚t t|dƒt|dƒ¡sÀJ ‚|jd dksËJ ‚d S d S )Nr   r   r¼   r   )TFFÚabcrµ   zincompatible dtyper~   r   r   r   r   r   )r   r   rÏ   r.   r/   rƒ   r   r    r   r!   r¢   r�   r%   )	r&   r.  ÚexpÚwarnr0   r'   rÍ   rŸ   Úindexerr)   r)   r*   Ú"test_putmask_dont_copy_some_blocks±  s6   ÿ

ÿ€
ÿ ür’  c                 C   s   |   | dkd¡S ©Nr   r˜   ©Úwhere©r�   r)   r)   r*   r8   ×  r9   c                 C   s   |   | dkd¡S r“  ©Úmaskr–  r)   r)   r*   r8   Ø  r9   c                 C   sŒ   t g d¢|d�}| ¡ }||ƒ}| rt t|ƒt|ƒ¡sJ ‚nt t|ƒt|ƒ¡r+J ‚d|jd< | r>t t|ƒt|ƒ¡r>J ‚t ||¡ d S )Nr   r9  r˜   r   )r   r   r   r    r   r%   r.   r£   ©r&   r:  rì   r�   r¥   r–   r)   r)   r*   Útest_where_mask_noopÓ  s   	
rš  c                 C   s   |   | dk d¡S r“  r”  r–  r)   r)   r*   r8   ð  r9   c                 C   s   |   | dkd¡S r“  r—  r–  r)   r)   r*   r8   ñ  r9   c                 C   sH   t g d¢|d�}| ¡ }||ƒ}t t|ƒt|ƒ¡rJ ‚t ||¡ d S )Nr   r9  )r   r   r   r    r   r.   r£   r™  r)   r)   r*   Útest_where_maskì  s
   	r›  z
dtype, val)rS   g      %@)r†  r˜   c                 C   s   |   | dk |¡S ©Nr   r”  ©r'   r.  r)   r)   r*   r8     r9   c                 C   s   |   | dk|¡S rœ  r—  r�  r)   r)   r*   r8     r9   c                 C   sÀ   t g d¢g d¢dœ|d�}| ¡ }|||ƒ}| r5t t|dƒt|dƒ¡s&J ‚t t|dƒt|dƒ¡r4J ‚nt t|dƒt|dƒ¡rCJ ‚d|jd< | rXt t|dƒt|dƒ¡rXJ ‚t ||¡ d S )	Nr   )éüÿÿÿéûÿÿÿiúÿÿÿr”   r9  r   r   r˜   r
  ró   )r&   r:  r.  rì   r'   rŸ   r–   r)   r)   r*   Ú%test_where_mask_noop_on_single_columnþ  s   	

r   r˜  r•  c              	   C   s$  t g d¢ddœƒ}| ¡ }| rft ¡ � t|d |ƒ|d dkddd� W d   ƒ n1 s.w   Y  t ||¡ t ¡ � t|dg |ƒ|d dkddd� W d   ƒ n1 sYw   Y  t ||¡ d S tjtd	d
�� t|d |ƒ|d dkddd� W d   ƒ n1 sˆw   Y  t d ¡�. tdd ƒ� t|dg |ƒ|d dkddd� W d   ƒ n1 s´w   Y  W d   ƒ n1 sÃw   Y  t d ¡�: tdd ƒ� t||d dk |ƒ|d dkddd� W d   ƒ n1 sòw   Y  W d   ƒ d S W d   ƒ d S 1 �sw   Y  d S )N)r   r   r   r   r”   r   r   r   T©rZ  úinplace methodr~   úmode.chained_assignment)	r   r   r.   Úraises_chained_assignment_errorrK  r�   rƒ   r„   r
   )r&   rì   r'   rŸ   r)   r)   r*   Útest_chained_where_mask  s6   
"ÿ
$ÿ"ÿ$ÿ€ÿ*ÿÿ$ÿr¥  c                 C   s¦   t dg d¢itdddd�d�}| ¡ }|jdd�}| r*t t|dƒt|dƒ¡s)J ‚nt t|dƒt|dƒ¡r8J ‚d	|jd
< t t|dƒt|dƒ¡rKJ ‚t 	||¡ d S )Nr   )g        Nç       @ç      @z1/1/2000r   Úminrå   r|   r  r   r   )
r   r	   r   Úasfreqr   r    r   r%   r.   r�   rž   r)   r)   r*   Útest_asfreq_noop1  s   
þ
rª  c                 C   sR   t dddœg d¢d�}| ¡ }| ¡ D ]	\}}d|jd< q| r't ||¡ d S d S )Nr   r   r”   r   r|   rŒ   )r   r   Úiterrowsr%   r.   r�   )r&   r'   rŸ   rõ   Úsubr)   r)   r*   Útest_iterrowsE  s   ÿr­  c                 C   sÆ   t ddtjdgiƒ}|d d … }| ¡ }t |¡� |jdd� W d   ƒ n1 s*w   Y  t |¡� d|jd< W d   ƒ n1 sDw   Y  | rSt ||¡ d S t dg d¢iƒ}t ||¡ d S )	Nr   r¼   r   Tr¡  ç      Y@r   )r®  r¼   r   )	r   r   rç   r   r.   r/   Úffillr%   r�   )r&   r0   r'   rÍ   r³   r)   r)   r*   Útest_interpolate_creates_copyO  s   ÿÿr°  c                 C   s  t g d¢g d¢g d¢dœƒ}| ¡ }|jd d�}| dt g d¢¡¡ | rBt t|dƒt|dƒ¡s3J ‚t t|d	ƒt|d	ƒ¡sAJ ‚nt t|dƒt|dƒ¡rPJ ‚t t|d	ƒt|d	ƒ¡r^J ‚d
|jd< t 	||¡ | r{t t|dƒt|dƒ¡syJ ‚d S t t|dƒt|dƒ¡r‰J ‚d S )Nr   r   rz  r   r,   r   )r]  éþÿÿÿéýÿÿÿr   r   r   rú   )
r   r   Úisetitemr   Úarrayr    r   rû   r.   r�   rž   r)   r)   r*   Útest_isetitema  s   
  rµ  zsingle-blockzmixed-blockc                 C   sö   t g d¢tjg d¢|d�dœƒ}tg d¢ƒ}| ¡ }| d|¡ | r7t t|dƒt|ƒ¡s/J ‚|j 	d¡r7J ‚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   r9  r”   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*   Útest_isetitem_seriesw  s    
 
 r¶  c                 C   sè   t g d¢dddœƒ}t g d¢ddœƒ}| ddg|¡ | rBt t|dƒt|dƒ¡s+J ‚t t|d	ƒt|d	ƒ¡s9J ‚|j d¡rAJ ‚nt t|dƒt|dƒ¡rPJ ‚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%   r.   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ƒ¡s$J ‚d|jd< t t|dƒt|dƒ¡r7J ‚t ||¡ d S |rKt	|t
ƒrHtnd }n	t	|tƒrRtnd }tddƒ�" t |¡� d|jd< W d   ƒ n1 sow   Y  W d   ƒ n1 s~w   Y  t	|tƒr�t ||¡ d S |jd dks™J ‚d S )	Nr   r   r”   r   r   Úchained_assignmentr�  r   )r   r   Úgetr   r    r   r%   r.   r�   Ú
isinstancer5   r„   rÏ   r   r
   rƒ   )r&   r0   r²   r'   rŸ   r–   r�  r)   r)   r*   Útest_get£  s(   

ÿ€ÿ
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s3|rAt t|d	ƒt|
ƒ¡s@J ‚n
| rK|
j d
¡sKJ ‚| sQ|rW|sWd
|
j	d
< nP|rxt
 |p`|dk¡� d
|
j	d
< W d   ƒ n1 srw   Y  n/tddƒ�" t
 t¡� d
|
j	d
< W d   ƒ n1 s“w   Y  W d   ƒ n1 s¢w   Y  | s¯|s·|d
kr·t
 ||	¡ d S |j	d d
ksÀJ ‚d S )NrS   r   r   r   rz  r9  r   rñ   r   r   r¹  r�  r   )r   r   r´  r   Úxsr    r   r!   r¢   r%   r.   r/   r
   rƒ   r   r�   )r&   r0   r‡   rH   r²   r:  Úsingle_blockÚis_viewr'   rŸ   r–   r)   r)   r*   Útest_xsÀ  s6    ÿÿ€ÿ€ÿrÀ  rH   z
key, level)Úl1r   rc  c              	   C   s4  t  d¡ dd¡}tjddgg d¢gddgd	�}t||td
ƒd�}|dkr+| ¡  ¡ }| ¡ }	|j	|||d�}
|dkrOt  
t||jd ƒt|
|
jd ƒ¡sOJ ‚|rZ|dkrWtnd }n	| sa|sat}nd }tddƒ�" t |¡� d|
jd< W d   ƒ n1 s~w   Y  W d   ƒ n1 s�w   Y  t ||	¡ d S )Né   r   r   rÁ  Úl2r   Úlev1Úlev2rf  rŽ  rÞ   r   )ÚlevelrH   r   r¹  r�  r   )r   ÚarangeÚreshaper   Úfrom_productr   rÏ   r•   r   r½  r    r   r   r„   r   r
   r.   rƒ   r%   r�   )r&   r0   r‡   r²   rÆ  rH   Úarrr   r'   rŸ   r–   r�  r)   r)   r*   Útest_xs_multiindexå  s.   ÿÿ€ÿ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 s4w   Y  t g d¢g d¢dœƒ}t ||¡ | rpt ||¡ t t|d	ƒt|d	ƒ¡s`J ‚t t|dƒt|dƒ¡rnJ ‚d S t ||¡ d S )
N©ç      ð?r¦  r§  )ç      @g      @ç      @r”   r   ç      Y@r   r|   )rÎ  rÐ  rÏ  r   )	r   r   r.   r/   Úupdater�   r   r    r   )r&   r0   Údf1r‰   Ú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 … }|r4t ¡ � | |¡ W d   ƒ n1 s.w   Y  n| |¡ t g d¢ƒ}t ||¡ | rOt ||¡ d S t ||¡ d S )NrÌ  rÐ  r   r|   )rÍ  rÐ  r§  )r   r   r.   r/   rÑ  r£   )r&   r0   Úser1r’   Ú	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 s,w   Y  t ||¡ t ¡ � |dg  | ¡ ¡ W d   ƒ n1 sPw   Y  t ||¡ d S tjt	dd�� |d  |¡ W d   ƒ n1 svw   Y  t d ¡�' t
dd ƒ� |dg  | ¡ ¡ W d   ƒ n1 s›w   Y  W d   ƒ n1 sªw   Y  t d ¡�3 t
dd ƒ� ||d dk  | ¡ ¡ W d   ƒ n1 sÒw   Y  W d   ƒ d S W d   ƒ d S 1 sêw   Y  d S )	Nr   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*   Útest_update_chained_assignment/  s8   
ÿ
ÿÿÿ€ÿÿÿ"ÿrØ  c                 C   sx   t g d¢ƒ}| ¡ }t|ƒ}|d9 }| r(t t|ƒ|¡rJ ‚t |t|ƒ¡ d S t t|ƒ|¡s2J ‚t |t|ƒ¡ d S ©Nr   r   )r   r   r   r   r    r.   Úassert_numpy_array_equal)r&   r�   r¥   Údatar)   r)   r*   Útest_inplace_arithmetic_seriesH  s   rÜ  c                 C   sš   t g d¢ƒ}| ¡ }|d d … }t |¡� |d9 }W d   ƒ n1 s$w   Y  | r?t t|ƒt|ƒ¡r7J ‚t ||¡ d S t t|ƒt|ƒ¡sKJ ‚d S rÙ  )r   r   r.   r/   r   r    r   r£   )r&   r0   r�   r¥   rÍ   r)   r)   r*   Ú-test_inplace_arithmetic_series_with_referenceZ  s   
ÿrÝ  Fc                 C   sŽ   t g d¢ddœƒ}| ¡ }|j|d�}|s|r| r(t t|dƒt|dƒ¡s'J ‚nt t|dƒt|dƒ¡r6J ‚d|jd< | rEt ||¡ d S d S )	Nr   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*   Útest_transposei  s   
ÿrÞ  c                 C   s^   t g d¢ddœƒ}| ¡ }|j}t t|dƒt|dƒ¡rJ ‚d|jd< | r-t ||¡ d S d S )Nr   r¼   r”   r   r   rŒ   r   )	r   r   ÚTr   r    r   r%   r.   r�   ©r&   r'   rŸ   r–   r)   r)   r*   Útest_transpose_different_dtypesy  s   
ÿrá  c                 C   s:   t dg d¢idd�}|j}t t|dƒt|dƒ¡rJ ‚d S )Nr   r   r†  r9  r   )r   rß  r   r    r   )r&   r'   r–   r)   r)   r*   Útest_transpose_ea_single_column„  s    râ  c                 C   sn   t g d¢ddœƒ}| ¡ }dd„ }t |¡� | |¡ W d   ƒ n1 s&w   Y  | r5t ||¡ d S d S )Nr   r   r”   c                 S   rÄ   ©NrŒ   r   rÅ   r–  r)   r)   r*   rì   �  rÆ   z"test_transform_frame.<locals>.func)r   r   r.   r/   Ú	transformr�   )r&   r0   r'   rŸ   rì   r)   r)   r*   Útest_transform_frame‹  s   ÿÿrå  c                 C   sh   t g d¢ƒ}| ¡ }dd„ }t |¡� | |¡ W d   ƒ n1 s#w   Y  | r2t ||¡ d S d S )Nr   c                 S   rÄ   rã  rÅ   r–  r)   r)   r*   rì   �  rÆ   z#test_transform_series.<locals>.func)r   r   r.   r/   rä  r£   )r&   r0   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   r|   )r   Úcountr%   r   r.   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 w   Y  t t|ƒt|ƒ¡s1J ‚| r;|j	 
d¡r;J ‚t |¡� d|jd< W d   ƒ n1 sPw   Y  | r_t ||¡ d S t g d¢ƒ}t ||¡ d S )Nr   zis deprecatedr~   r   rŒ   rÇ   )r   r   r.   rƒ   r„   rÍ   r   r    r   r!   r¢   r/   r%   r£   rs  r)   r)   r*   Útest_series_view¯  s   
ÿÿré  c                 C   s˜   t dg d¢iƒ}tg d¢ƒ}| ¡ }|jd|dd� | r2t t|ƒt|dƒ¡s)J ‚|j d¡r1J ‚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%   r.   r£   )r&   r'   r�   r¥   r)   r)   r*   Útest_insert_seriesÃ  s   
rí  c                 C   s|   t g d¢ddœƒ}| ¡ }| d¡}| r#t t|dƒt|dƒ¡s"J ‚nt t|dƒt|dƒ¡r1J ‚d|jd< t ||¡ d S )Nr   r   r”   úc = a+br   rŒ   r   )	r   r   Úevalr   r    r   r%   r.   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ƒ¡s(J ‚t |¡� d|jd	< W d   ƒ n1 s=w   Y  | rLt 	||¡ d S d S )
Nr   r   r”   rî  Tr¡  r   rŒ   r   )
r   r   rï  r   r    r   r.   r/   r%   r�   )r&   r0   r'   rŸ   Ú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 s*w   Y  | r8t ||¡ n	|jd	 d
ksAJ ‚t ddgddgdœƒ}| ¡ }t d ¡� |j|dd� W d   ƒ n1 sgw   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   r.   r/   Úapplyr�   rû   rƒ   )r&   r0   r'   rŸ   rä  r)   r)   r*   Útest_apply_modify_rowî  s   ÿÿrõ  )zÚnumpyr   ÚpytestÚpandas.errorsr   ÚpandasÚpdr   r   r   r   r   r   r	   r
   r   Úpandas._testingÚ_testingr.   Úpandas.tests.copy_view.utilr   r+   r1   ÚmarkÚparametrizer‹   r“   r—   r    Ú
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/  r0  r6  r7  rB  rC  rH  rL  rN  rQ  rS  rV  rY  r[  r_  rm  rn  rp  rq  rr  rt  rv  ry  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*   Ú<module>   sú   ,%ïë)#ïë)
 
üø

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

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