U
    ¾mœd&æ  ã                   @   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 d dlmZ d dlmZ dd„ Zdd„ Zej d	d
ddg¡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ddd d!d"d#d$d%d&d'd(d)d*d+d,d-gd.�d/d0„ ƒƒZej d	d
ddg¡ejjdd1d„ d2d„ d3d„ d4d„ d5d„ d6d„ d7d„ d8d„ d9d„ d:d„ d;d„ d<d„ d=d„ d>d„ d?d„ d@d„ dAd„ gdBddd d!d"d#d%dCd&d'd(d)d*d+d,d-gd.�dDdE„ ƒƒZej d	d
ddg¡dFdG„ ƒZdHdI„ Zej dJe d dK¡edLdKgƒg¡dMdN„ ƒZdOdP„ ZdQdR„ ZdSdT„ Z dUdV„ Z!dWdX„ Z"dYdZ„ Z#d[d\„ Z$ej d]d^d_gid`d_idad_ig¡dbdc„ ƒZ%ddde„ Z&dfdg„ Z'dhdi„ Z(djdk„ Z)dldm„ Z*ej dndod„ dpd„ dqd„ g¡drds„ ƒZ+dtdu„ Z,dvdw„ Z-dxdy„ Z.dzd{„ Z/ej d|dJd}g¡d~d„ ƒZ0d€d�„ Z1d‚dƒ„ Z2ejjd„d…d„ d fd†d„ dKfd‡d„ d fdˆd„ d fgd‰dŠdd‹gd.�dŒd�„ ƒZ3ej dŽedLdKgd_d��ed_dLdKgiƒg¡d�d‘„ ƒZ4ej dŽedLdKgd_d��ed_dLdKgiƒg¡d’d“„ ƒZ5d”d•„ Z6d–d—„ Z7d˜d™„ Z8dšd›„ Z9ej dœd�dLe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¬„ Z>d­d®„ Z?d¯d°„ Z@ej d±d_d²dLd³œd dLd d³œg¡d´dµ„ ƒZAej dd¶d·g¡d¸d¹„ ƒZBej dŽedLdKgƒed_dLdKgiƒg¡dºd»„ ƒZCej dŽedLdKgƒed_dLdKgiƒg¡d¼d½„ ƒZDd¾d¿„ ZEdÀdÁ„ ZFej dÂedLdKdÃgd_d��i fed_dLdKdÃgiƒdÄd_ifg¡dÅdÆ„ ƒZGej dÂedLdKdÃgd_d��i fed_dLdKdÃgiƒdÄd_ifg¡dÇdÈ„ ƒZHej dÉdÊd dLg¡dËdÌ„ ƒZIdÍdÎ„ ZJdÏdÐ„ ZKej dŽedLdKdÃgƒed_dLdKdÃgiƒg¡dÑdÒ„ ƒZLdÓdÔ„ ZMdÕdÖ„ ZNd×dØ„ ZOej d±dÙdÚidJdÚig¡dÛdÜ„ ƒZPej dÝdÞdßg¡dàdá„ ƒZQdâdã„ ZRdädå„ ZSdædç„ ZTej dèdédêg¡dëdì„ ƒZUej dèdédêg¡dídî„ ƒZVej dèdïdðg¡dñ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þg¡dÿ�d „ ƒZ[�d�d„ Z\�d�d„ Z]�d�d„ Z^�d�d„ Z_ejjdèdédïg�d	�d
gd.��d�d„ ƒZ`ej �dd_d_gg¡�d�d„ ƒZaej �d�d�dg¡ejjdèdédïg�d	�d
gd.��d�d„ ƒƒZbej �dd dLg¡ej �d�d�dg¡�d�d„ ƒƒZc�d�d„ Zd�d�d„ Ze�d�d „ Zf�d!�d"„ Zgej d	d
dg¡�d#�d$„ ƒZh�d%�d&„ Zi�d'�d(„ Zj�d)�d*„ Zk�d+�d,„ ZldS (-  é    N)ÚSettingWithCopyWarning)Ú	DataFrameÚIndexÚ
MultiIndexÚPeriodÚSeriesÚ	TimestampÚ
date_rangeÚperiod_range)Ú	get_arrayc                 C   s–   t dddgdddgddd	gd
œƒ}| ¡ }t t|dƒt|dƒ¡rFt‚| rv|jjd j 	¡ r`t‚|jjd j 	¡ rvt‚d|j
d< |j
d dks’t‚d S ©Né   é   é   é   é   é   çš™™™™™¹?çš™™™™™É?ç333333Ó?©ÚaÚbÚcr   r   ©r   r   ©r   ÚcopyÚnpÚshares_memoryr   ÚAssertionErrorÚ_mgrÚblocksÚrefsZhas_referenceÚiloc©Úusing_copy_on_writeÚdfZdf_copy© r'   ú\/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/pandas/tests/copy_view/test_methods.pyÚ	test_copy   s    "
r)   c                 C   s  t dddgdddgddd	gd
œƒ}|jdd�}t t|dƒt|dƒ¡sJt‚| rz|jjd j 	¡ sdt‚|jjd j 	¡ szt‚| rÔd|j
d< |j
d dksšt‚t t|dƒt|dƒ¡r¶t‚t t|dƒt|dƒ¡sÒt‚n:d|j
d< |j
d dksðt‚t t|dƒt|dƒ¡�st‚d S )Nr   r   r   r   r   r   r   r   r   r   F©Údeepr   r   r   r   r   r$   r'   r'   r(   Útest_copy_shallow$   s    "

r,   r   TFÚmethodc                 C   s   | j tj|d�S )N©Úcolumnsr   )ÚrenameÚstrÚlower©r&   r   r'   r'   r(   Ú<lambda>B   ó    r4   c                 C   s   | j ddg|d�S )Nr   r   r.   )Úreindexr3   r'   r'   r(   r4   C   r5   c                 C   s   | j | |d�S ©N©r   ©Úreindex_liker3   r'   r'   r(   r4   D   r5   c                 C   s   | j | |d�d S ©Nr8   r   ©Úalignr3   r'   r'   r(   r4   E   r5   c                 C   s   | j dddgd|d�S ©Nr   r   r   Úindex)Úaxisr   ©Úset_axisr3   r'   r'   r(   r4   F   r5   c                 C   s   | j d|d�S ©NÚtest©r?   r   ©Úrename_axisr3   r'   r'   r(   r4   G   r5   c                 C   s   | j d|d�S )NrD   r.   rF   r3   r'   r'   r(   r4   H   r5   c                 C   s   | j ddi|d�S )Nr   Úint64r8   ©Úastyper3   r'   r'   r(   r4   I   r5   c                 C   s   | j dd|d�S ©Nr   r8   ©Úswapaxesr3   r'   r'   r(   r4   K   r5   c                 C   s   | j dd|d�S ©Nr   r   r8   ©Útruncater3   r'   r'   r(   r4   L   r5   c                 C   s   | j |d�S r7   ©Úinfer_objectsr3   r'   r'   r(   r4   M   r5   c                 C   s   | j |d�S r7   ©Úto_timestampr3   r'   r'   r(   r4   N   r5   c                 C   s   | j d|d�S ©NÚD)Úfreqr   ©Ú	to_periodr3   r'   r'   r(   r4   O   r5   c                 C   s   | j d|d�S ©Nú
US/Centralr8   ©Útz_localizer3   r'   r'   r(   r4   P   r5   c                 C   s   | j d|d�S rZ   ©Ú
tz_convertr3   r'   r'   r(   r4   Q   r5   c                 C   s   | j d|d�S ©NF)Úallows_duplicate_labelsr   ©Ú	set_flagsr3   r'   r'   r(   r4   R   r5   r0   r6   r:   r=   rB   Zrename_axis0Zrename_axis1rJ   rM   rP   rR   rT   rY   r]   r_   rc   )Zidsc           	      C   s&  d }d| j jjkr"tdddd�}nZd| j jjkr@tdddd�}n<d| j jjkr^tdddd�}nd| j jjkr|tdddd	d
�}tdddgdddgdddgdœ|d�}|||d�}|p¸|dk}| j jj d¡rÞ|sÞ|sÞ|dkrÞd}|�rt t	|dƒt	|dƒ¡�s"t
‚nt t	|dƒt	|dƒ¡�r"t
‚d S )NrT   ú
2012-01-01rV   r   ©rW   ÚperiodsrY   r]   r_   úEurope/Brussels©rW   rf   Útzr   r   r   r   r   r   r   r   r   ©r?   r8   Fzreindex-r   )ÚnodeÚcallspecÚidr
   r	   r   Ú
startswithr   r   r   r   )	Úrequestr-   r   r%   Úusing_array_managerr?   r&   Údf2Úshare_memoryr'   r'   r(   Útest_methods_copy_keyword>   s$    -& rs   c                 C   s   | j ddi|d�S )Nr   éd   rE   ©r0   ©Úserr   r'   r'   r(   r4   ‰   r5   c                 C   s   | j d |d�S r7   ru   rv   r'   r'   r(   r4   Š   r5   c                 C   s   | j | j|d�S )NrE   )r6   r?   rv   r'   r'   r(   r4   ‹   r5   c                 C   s   | j | |d�S r7   r9   rv   r'   r'   r(   r4   Œ   r5   c                 C   s   | j | |d�d S r;   r<   rv   r'   r'   r(   r4   �   r5   c                 C   s   | j dddgd|d�S r>   rA   rv   r'   r'   r(   r4   Ž   r5   c                 C   s   | j d|d�S rC   rF   rv   r'   r'   r(   r4   �   r5   c                 C   s   | j d|d�S )NrH   r8   rI   rv   r'   r'   r(   r4   �   r5   c                 C   s   | j dd|d�S )Nr   r   r8   )Ú	swaplevelrv   r'   r'   r(   r4   ‘   r5   c                 C   s   | j dd|d�S rK   rL   rv   r'   r'   r(   r4   ’   r5   c                 C   s   | j dd|d�S rN   rO   rv   r'   r'   r(   r4   “   r5   c                 C   s   | j |d�S r7   rQ   rv   r'   r'   r(   r4   ”   r5   c                 C   s   | j |d�S r7   rS   rv   r'   r'   r(   r4   •   r5   c                 C   s   | j d|d�S rU   rX   rv   r'   r'   r(   r4   –   r5   c                 C   s   | j d|d�S rZ   r\   rv   r'   r'   r(   r4   —   r5   c                 C   s   | j d|d�S rZ   r^   rv   r'   r'   r(   r4   ˜   r5   c                 C   s   | j d|d�S r`   rb   rv   r'   r'   r(   r4   ™   r5   zrename (dict)rx   c                 C   s  d }d| j jjkr"tdddd�}n„d| j jjkr@tdddd�}nfd| j jjkr^tdddd�}nHd| j jjkr~tdddd	d
�}n(d| j jjkr¦t dddgdddgg¡}tdddg|d�}|||d�}|pÎ|dk}|rît 	t
|ƒt
|ƒ¡sìt‚nt 	t
|ƒt
|ƒ¡�rt‚d S )NrT   rd   rV   r   re   rY   r]   r_   rg   rh   rx   r   r   r   r   r   rj   r8   F)rk   rl   rm   r
   r	   r   Zfrom_arraysr   r   r   r   r   )ro   r-   r   r%   r?   rw   Úser2rr   r'   r'   r(   Ú test_methods_series_copy_keyword…   s"    +rz   c                 C   s†   t dddgdddgdœƒ}|j|d�}| p8|d	kp8|d k}|oB| }|rft t|d
ƒt|dƒ¡s‚t‚nt t|d
ƒt|dƒ¡r‚t‚d S )Nr   r   r   r   r   r   ©r   r   r8   Fr   r   )r   Ú	transposer   r   r   r   )r%   r   rp   r&   Úresultrr   r'   r'   r(   Útest_transpose_copy_keywordÇ   s    
r~   c                 C   sØ   t dddgdddgddd	gd
œdddgd�}| ¡ }| ¡ }|j ¡  | r‚t t|dƒt|dƒ¡sft‚t t|dƒt|dƒ¡s‚t‚d|j	d< t t|dƒt|dƒ¡r¨t‚| rÈt t|dƒt|dƒ¡sÈt‚t
 ||¡ d S )Nr   r   r   r   r   r   r   r   r   r   é
   é   é   rj   r   r   r   )r   r   )r   r   Úreset_indexr    Ú_verify_integrityr   r   r   r   r#   ÚtmÚassert_frame_equal©r%   r&   Údf_origrq   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@t‚|j d¡rjt‚nt t|ƒt|ƒ¡rjt‚d|j	d< t
 ||¡ d S )Nr   r   rj   T)Údropr   rt   )r   r   r‚   r   r   r   r   r    Ú_has_no_referencer#   r„   Úassert_series_equal©r%   r?   rw   Úser_origry   r'   r'   r(   Útest_reset_index_series_dropî   s    
rŽ   c                 C   sÜ   t dddgdddgddd	gd
œƒ}| ¡ }|jtjd�}| rXt t|dƒt|dƒ¡sXt‚d|j	d< t t|dƒt|dƒ¡r~t‚| ržt t|dƒt|dƒ¡sžt‚t dddgdddgddd	gdœƒ}t
 ||¡ t
 ||¡ d S )Nr   r   r   r   r   r   r   r   r   r   ©r/   ÚAr   r   r   ÚCr   )r�   ÚBr‘   )r   r   r0   r1   Úupperr   r   r   r   r#   r„   r…   ©r%   r&   r‡   rq   Úexpectedr'   r'   r(   Útest_rename_columnsý   s    "
"r–   c                 C   sú   t dddgdddgddd	gd
œƒ}|jtjd�}| ¡ }| rZt t|dƒt|dƒ¡svt‚nt t|dƒt|dƒ¡rvt‚d|j	d< t t|dƒt|dƒ¡rœt‚| r¼t t|dƒt|dƒ¡s¼t‚t dddgdddgddd	gd
œƒ}t
 ||¡ t
 ||¡ d S )Nr   r   r   r   r   r   r   r   r   r   r�   r�   r   r   r   r‘   r   )r   r0   r1   r“   r   r   r   r   r   r#   r„   r…   )r%   r&   rq   Zdf2_origr•   r'   r'   r(   Ú!test_rename_columns_modify_parent  s    "
"r—   c                 C   sÞ   t dddgddœƒ}| ¡ }dd„ }| |¡}t t|dƒt|dƒ¡sJt‚d	|jd
< | r‚t 	||¡ t t|dƒt|dƒ¡r¾t‚n<t d	ddgddœƒ}t 	||¡ t t|dƒt|dƒ¡s¾t‚t t|dƒt|dƒ¡sÚt‚d S )Nr   r   r   ç      ø?r{   c                 S   s   | S ©Nr'   ©r&   r'   r'   r(   Útestfunc'  s    ztest_pipe.<locals>.testfuncr   r   r   r   )
r   r   Úpiper   r   r   r   r#   r„   r…   ©r%   r&   r‡   r›   rq   r•   r'   r'   r(   Ú	test_pipe#  s    

rž   c                 C   sÔ   t dddgddœƒ}| ¡ }dd„ }| |¡}t t|dƒt|dƒ¡sJt‚| rxt ||¡ t t|d	ƒt|d	ƒ¡r´t‚n<t d
ddgddœƒ}t ||¡ t t|d	ƒt|d	ƒ¡s´t‚t t|dƒt|dƒ¡sÐt‚d S )Nr   r   r   r˜   r{   c                 S   s   d| j d< | S )Nrt   r   )r#   rš   r'   r'   r(   r›   ?  s    
z%test_pipe_modify_df.<locals>.testfuncr   r   rt   )	r   r   rœ   r   r   r   r   r„   r…   r�   r'   r'   r(   Útest_pipe_modify_df;  s    
rŸ   c                 C   sÎ   t dddgdddgddd	gd
œƒ}| ¡ }|jddgd�}| r\t t|dƒt|dƒ¡sxt‚nt t|dƒt|dƒ¡rxt‚d|jd< t t|dƒt|dƒ¡ržt‚| r¾t t|dƒt|dƒ¡s¾t‚t 	||¡ d S )Nr   r   r   r   r   r   r   r   r   r   r   r   r�   r   r   )
r   r   r6   r   r   r   r   r#   r„   r…   r†   r'   r'   r(   Útest_reindex_columnsR  s    "
r    c                 C   s  t dddgdddgddd	gd
œƒ}| ¡ }|jdd�}|j ¡  | r~t t|dƒt|dƒ¡s`t‚t t|dƒt|dƒ¡s¶t‚n8t t|dƒt|dƒ¡ršt‚t t|dƒt|dƒ¡r¶t‚d|j	d< t t|dƒt|dƒ¡rÜt‚| rüt t|dƒt|dƒ¡süt‚t
 ||¡ d S )Nr   r   r   r   r   r   r   r   r   r   r   r�   r   r   r   r   )r   r   r‰   r    rƒ   r   r   r   r   r#   r„   r…   r†   r'   r'   r(   Útest_drop_on_columnf  s    "

r¡   c                 C   s¶   t dddgdddgddd	gd
œƒ}| ¡ }| d¡}|j ¡  | r`t t|dƒt|dƒ¡s|t‚nt t|dƒt|dƒ¡r|t‚d|j	d< | r¦t t|dƒt|dƒ¡r¦t‚t
 ||¡ d S )Nr   r   r   r   r   r   r   r   r   r   rH   r   r   r   )r   r   Úselect_dtypesr    rƒ   r   r   r   r   r#   r„   r…   r†   r'   r'   r(   Útest_select_dtypesy  s    "


r£   Úfilter_kwargsÚitemsr   ÚlikeÚregexc                 C   s®   t dddgdddgddd	gd
œƒ}| ¡ }|jf |Ž}| rXt t|dƒt|dƒ¡stt‚nt t|dƒt|dƒ¡rtt‚| ržd|jd< t t|dƒt|dƒ¡ržt‚t 	||¡ d S r   )
r   r   Úfilterr   r   r   r   r#   r„   r…   )r%   r¤   r&   r‡   rq   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|t‚nt t|d	ƒt|d	ƒ¡r|t‚d|jd< | r¦t t|d
ƒt|d
ƒ¡r¦t‚t	 
||¡ d S )Nr   r   r   r   r   r   ú
2020-01-01ú
2020-01-03r   r   ©r?   r/   r   )rf   r   )r   r	   r   Úshiftr   r   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	ƒ¡rTt‚d S )Nr   r   r   r   r   r   rª   r«   r   r   r¬   r   ©rf   r@   )r   r	   r­   r   r   r   r   )r%   r&   rq   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Št‚nt t|d	ƒt|d	ƒ¡rŠt‚d|j	d< | r´t t|d	ƒt|d	ƒ¡r´t‚t
 ||¡ d S )Nr   r   r   r   r   r   rª   r«   r   r   r¬   ú
2020-01-02z
2020-01-04Z1D©rf   rW   r   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ƒ¡sNt‚d|jd< | rxt t|dƒt|dƒ¡rxt‚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   r   r   r#   Únanr„   r…   )r%   r&   rq   r•   r'   r'   r(   Útest_shift_columnsÖ  s"     ÿ
 ÿþr¶   c                 C   s  t dddgdddgddd	gd
œƒ}| ¡ }|d d … }| d¡}t |jt|dƒ¡sXt‚t t|dƒt|dƒ¡stt‚| ršd|jd< t |jt|dƒ¡ršt‚d|jd< | rÒt t|dƒt|dƒ¡rÄt‚t	 
||¡ n.t dddgdddgddd	gd
œƒ}t	 
||¡ d S )Nr   r   r   r   r   r   r   r   r   r   r   r   r   r   )r   r   Úpopr   r   Úvaluesr   r   r#   r„   r…   )r%   r&   r‡   Zview_originalr}   r•   r'   r'   r(   Útest_popé  s    "


"r¹   Úfuncc                 C   s
   |   |¡S r™   r<   ©ÚxÚyr'   r'   r(   r4     r5   c                 C   s   | j |jdd�S )Nr   ©r@   )r=   r   r»   r'   r'   r(   r4     r5   c                 C   s   | j |jjtddƒ dd�S )Nr   r   r¾   )r=   r   r#   Úslicer»   r'   r'   r(   r4     r5   c                 C   s²   t dddgddœƒ}| ¡ }|ddg  ¡ }|||ƒ\}}| r\t t|dƒt|dƒ¡sxt‚nt t|dƒt|dƒ¡rxt‚d|jd< | r¢t t|dƒt|dƒ¡r¢t‚t ||¡ d S )	Nr   r   r   r   r{   r   r   r   )	r   r   r   r   r   r   r#   r„   r…   )r%   rº   r&   r‡   Z
df_changedrq   Ú_r'   r'   r(   Útest_align_frameþ  s    	
rÁ   c                 C   sÜ   t ddgƒ}| ¡ }| ¡ }| |¡\}}| rXt |j|j¡sBt‚t |j|j¡s€t‚n(t |j|j¡rlt‚t |j|j¡r€t‚d|jd< d|jd< | rÀt |j|j¡r¬t‚t |j|j¡rÀt‚t 	||¡ t 	||¡ d S )Nr   r   r   )
r   r   r=   r   r   r¸   r   r#   r„   r‹   )r%   rw   r�   Z	ser_otherry   Zser_other_resultr'   r'   r(   Útest_align_series  s     

rÂ   c                 C   s    t dddgdddgdœƒ}| ¡ }|j|dd	�\}}t t|d
ƒt|d
ƒ¡sPt‚t t|dƒt|dƒ¡slt‚| rœd|jd< t 	||¡ d|jd< t 	||¡ d S )Nr   r   r   r   r   r   r{   Fr8   r   r   r   ©r   r   )
r   r   r=   r   r   r   r   Úlocr„   r…   )r%   r&   r‡   rq   Zdf3r'   r'   r(   Útest_align_copy_false-  s    

rÅ   c                 C   sØ   t dddgdddgdœƒ}tdddgdd	�}| ¡ }| ¡ }|j|d
dd�\}}t t|dƒt|dƒ¡slt‚t t|dƒt|dƒ¡sˆt‚t t|dƒt|dƒ¡s¤t‚| rÔd|jd< t	 
||¡ d|jd< t	 ||¡ d S )Nr   r   r   r   r   r   r{   r¼   ©ÚnameFr   )r   r@   r   r   rÃ   )r   r   r   r=   r   r   r   r   rÄ   r„   r…   r‹   )r%   r&   rw   r�   r‡   rq   ry   r'   r'   r(   Ú!test_align_with_series_copy_false=  s    

rÈ   c                 C   sæ   t dddgƒ}| ¡ }|d d …  ¡ }t |jt|dƒ¡s>t‚d|jd< | rrt |jt|dƒ¡rdt‚t	 
||¡ n| ¡ }d|jd< t	 
||¡ |d d …  ¡ }d|jd< | rÀt	 || ¡ ¡ n"| ¡  ¡ }d|jd< t	 ||¡ d S )Nr   r   r   r   r   )r   r   Zto_framer   r   r¸   r   r   r#   r„   r‹   r…   )r%   rw   r�   r&   r•   r'   r'   r(   Útest_to_frameP  s$    



rÉ   Úaxr/   c                 C   s¦   t dddgdddgdœƒ}| ¡ }| ||¡}| rPt t|dƒt|dƒ¡slt‚nt t|dƒt|dƒ¡rlt‚d	|jd
< | r–t t|dƒt|dƒ¡r–t‚t 	||¡ d S )Nr   r   r   r   r   r   r{   r   r   r   ©
r   r   rM   r   r   r   r   r#   r„   r…   )r%   rÊ   r&   r‡   rq   r'   r'   r(   Útest_swapaxes_noopr  s    
rÌ   c                 C   s°   t dddgdddgdœdd	d
gd�}| ¡ }| dd¡}| rZt t|dƒt|dƒ¡svt‚nt t|dƒt|dƒ¡rvt‚d|jd< | r t t|dƒt|dƒ¡r t‚t 	||¡ d S )Nr   r   r   r   r   r   r{   r¼   r½   Úzrj   r?   r/   r   r   r   rË   r†   r'   r'   r(   Útest_swapaxes_single_block„  s    $
rÎ   c                  C   sX   t ddgddœƒ} | jddd�} d| jd	< t ddgddgd
œddgd�}t | |¡ d S )Nr   r   r   r{   r?   r/   )Zaxis1Zaxis2rt   r   r´   r   r   rj   )r   rM   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*   r8   rš   r'   r'   r(   r4      r5   c                 C   s   |   ¡   ¡ S r™   )r‚   rš   r'   r'   r(   r4   ¡  r5   c                 C   s   | j tjd�j tjd�S )Nr�   )r0   r1   r“   r2   rš   r'   r'   r(   r4   ¢  r5   c                 C   s   | j dd�jdd�S )NFr*   Únumber)Úinclude)r   r¢   rš   r'   r'   r(   r4   £  r5   úshallow-copyr‚   r¢   c                 C   sœ   t dddgdddgddd	gd
œƒ}| ¡ }| o<| jjjdk}||ƒ}d|jd|f< |sdt ||¡ ||ƒ}d|jd< |s˜t |jd d …|d …f |¡ d S )Nr   r   r   r   r   r   r   r   r   r   rÒ   r   r   )r   r   rk   rl   rm   r#   r„   r…   )ro   r-   Úidxr%   r&   r‡   Zdf2_is_viewrq   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ƒ¡slt‚nt t|dƒt|dƒ¡rlt‚d|j	d< t t|dƒt|dƒ¡r’t‚t
 ||¡ d S )Nz2012-1-1rV   ©rW   z2012-1-2r   r   )r   r   r?   r   rT   r   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�}| rLt t|dƒt|dƒ¡sht‚nt t|dƒt|dƒ¡rht‚d|j	d< t t|dƒt|dƒ¡rŽt‚t
 ||¡ d S )Nú
2019-12-31ú
2020-12-31ÚYrÖ   r   r   )r   r   r?   r   rY   r   r   r   r   r#   r„   r×   rØ   r'   r'   r(   Útest_to_periodÍ  s    
rß   c                 C   s¨   t dddgdddgddd	gd
œƒ}| ¡ }| d¡}| rVt t|dƒt|dƒ¡srt‚nt t|dƒt|dƒ¡rrt‚d|jd< t t|dƒt|dƒ¡r˜t‚t 	||¡ d S )Nr   r   r   r   r   r   r   r   r   r   r   r   r   r´   r   )
r   r   Ú	set_indexr   r   r   r   r#   r„   r…   r†   r'   r'   r(   Útest_set_indexß  s    "

rá   c                  C   s@   t dddgddœƒ} |  d¡}| ¡ }d| jd< t ||¡ d S )Nr   r   r   r{   r   rt   r   )r   rà   r   r#   r„   r…   ©r&   r}   r•   r'   r'   r(   Ú4test_set_index_mutating_parent_does_not_mutate_indexð  s
    

rã   c                 C   sØ   t dddgdddgddd	gd
œƒ}| ¡ }| d¡}| rTt t|dƒt|dƒ¡sTt‚d|jd< t t|dƒt|dƒ¡rzt‚| ršt t|dƒt|dƒ¡sšt‚t dddgdddgddd	gdœƒ}t 	||¡ t 	||¡ d S )Nr   r   r   r   r   r   r   r   r   r   ZCoW_ÚCoW_ar   r   r   ÚCoW_cr   )rä   ZCoW_brå   )
r   r   Z
add_prefixr   r   r   r   r#   r„   r…   r”   r'   r'   r(   Útest_add_prefixù  s    "

ÿræ   c                 C   sØ   t dddgdddgddd	gd
œƒ}| ¡ }| d¡}| rTt t|dƒt|dƒ¡sTt‚d|jd< t t|dƒt|dƒ¡rzt‚| ršt t|dƒt|dƒ¡sšt‚t dddgdddgddd	gdœƒ}t 	||¡ t 	||¡ d S )Nr   r   r   r   r   r   r   r   r   r   Z_CoWÚa_CoWr   r   r   Úc_CoWr   )rç   Zb_CoWrè   )
r   r   Z
add_suffixr   r   r   r   r#   r„   r…   r”   r'   r'   r(   Útest_add_suffix  s    "

ÿré   z	axis, val)r   ç      @c                 C   s¨   t dddgd|dgddœƒ}| ¡ }|j|d�}| rRt t|d	ƒt|d	ƒ¡snt‚nt t|d	ƒt|d	ƒ¡rnt‚d
|jd< | r˜t t|d	ƒt|d	ƒ¡r˜t‚t 	||¡ d S )Nr   r   r   r   r   Údr   r¾   r   r   r   )
r   r   Údropnar   r   r   r   r#   r„   r…   )r%   r@   Úvalr&   r‡   rq   r'   r'   r(   Útest_dropna   s    
rî   rí   r   rê   c                 C   s~   t d|dgƒ}| ¡ }| ¡ }| r8t |j|j¡sLt‚nt |j|j¡rLt‚d|jd< | rnt |j|j¡rnt‚t 	||¡ d S )Nr   r   r   )
r   r   rì   r   r   r¸   r   r#   r„   r‹   )r%   rí   rw   r�   ry   r'   r'   r(   Útest_dropna_series1  s    
rï   c                 C   s   |   ¡ S r™   ©Úheadrš   r'   r'   r(   r4   E  r5   c                 C   s
   |   d¡S )Nr   rð   rš   r'   r'   r(   r4   F  r5   c                 C   s   |   ¡ S r™   ©Útailrš   r'   r'   r(   r4   G  r5   c                 C   s
   |   d¡S )Nr   rò   rš   r'   r'   r(   r4   H  r5   c                 C   sÒ   t dddgdddgdœƒ}| ¡ }| |ƒ}|j ¡  |rpt t|dƒt|dƒ¡sTt‚t t|d	ƒt|d	ƒ¡spt‚d
|jd< t t|d	ƒt|d	ƒ¡s–t‚|r¸t t|dƒt|dƒ¡rÂt‚n
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&   r‡   rq   r'   r'   r(   Útest_head_tailB  s    



rô   c                 C   sü   t ddgddddœƒ}| ¡ }| ¡ }| rdt t|dƒt|dƒ¡sFt‚t t|dƒt|dƒ¡sœt‚n8t t|dƒt|dƒ¡r€t‚t t|dƒt|dƒ¡rœt‚d|jd	< d
|jd< | rìt t|dƒt|dƒ¡rÐt‚t t|dƒt|dƒ¡rìt‚t 	||¡ d S )Nr   r   r   r¼   ©r   r   r   rë   r   r   r   r   rë   r´   )
r   r   rR   r   r   r   r   r#   r„   r…   r†   r'   r'   r(   Útest_infer_objects`  s    

rö   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
ƒ¡sŽt‚t |t|dƒ¡r¤t‚t |t|dƒ¡sºt‚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   rR   r   r#   r   r   r   )r%   r&   Úarr_aÚarr_bÚarr_dr'   r'   r(   Útest_infer_objects_no_referenceu  s,     ÿùÿ




r  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< | rÄt |t|d	ƒ¡r˜t‚t |t|d
ƒ¡r®t‚t |t|dƒ¡sÄt‚d S )Nr   r   r   rÜ   rÝ   r÷   rø   rõ   r   r   rë   r   r   r´   rû   rü   rý   )r%   r&   Úviewrþ   rÿ   r   r'   r'   r(   Útest_infer_objects_reference‘  s,     ÿüÿ





r  Úkwargsr   )ÚbeforeÚafterr@   c                 C   s¬   t dddgdddœƒ}| ¡ }|jf |Ž}|j ¡  | rVt t|dƒt|dƒ¡srt‚nt t|dƒt|dƒ¡rrt‚d|j	d< | rœt t|dƒt|dƒ¡rœt‚t
 ||¡ d S )Nr   r   r   r   r   r   r   )r   r   rP   r    rƒ   r   r   r   r   r#   r„   r…   ©r%   r  r&   r‡   rq   r'   r'   r(   Útest_truncate¬  s    

r  ZassignZdrop_duplicatesc                 C   s¨   t ddddgiƒ}| ¡ }t||ƒƒ }|j ¡  | rRt t|dƒt|dƒ¡snt‚nt t|dƒt|dƒ¡rnt‚d|j	d< | r˜t t|dƒt|dƒ¡r˜t‚t
 ||¡ d S )Nr   r   r   r   r   r   )r   r   Úgetattrr    rƒ   r   r   r   r   r#   r„   r…   )r%   r-   r&   r‡   rq   r'   r'   r(   Útest_assign_drop_duplicatesÄ  s    

r
  c                 C   sv   |  ¡ }| ddg¡}| r0t |j|j¡sDt‚nt |j|j¡rDt‚d|jd< | rft |j|j¡rft‚t ||¡ d S )Nr   r   )	r   Ztaker   r   r¸   r   r#   r„   r×   rØ   r'   r'   r(   Ú	test_takeÖ  s    
r  c                 C   s„   t dddd�|_| ¡ }| dd¡}| r>t |j|j¡sRt‚nt |j|j¡rRt‚d|jd< | rtt |j|j¡rtt‚t	 
||¡ d S )Nz
2018-04-09r   Z1D20minr²   z0:00z1:00r   )r	   r?   r   Zbetween_timer   r   r¸   r   r#   r„   r×   rØ   r'   r'   r(   Útest_between_timeç  s    
r  c                 C   s®   t ddgddœƒ}t dddgdœƒ}| ¡ }| |¡}| rXt t|dƒt|dƒ¡stt‚nt t|dƒt|dƒ¡rtt‚d|jd< | ržt t|dƒt|dƒ¡ržt‚t 	||¡ d S )Nr   r   r   r{   )r   r   r   r´   )
r   r   r:   r   r   r   r   r#   r„   r…   )r%   r&   Úotherr‡   rq   r'   r'   r(   Útest_reindex_likeø  s    

r  c                 C   sz   t dddgƒ}| ¡ }| ¡ }| r8t |j|j¡sLt‚nt |j|j¡rLt‚d|jd< t |j|j¡rjt‚t 	||¡ d S )Nr   r   r   r   )
r   r   Z
sort_indexr   r   r¸   r   r#   r„   r‹   ©r%   rw   r�   ry   r'   r'   r(   Útest_sort_index
  s    
r  zobj, kwargsr   Zbyc                 C   sˆ   |  ¡ }|jf |Ž}| r6t t|dƒt|dƒ¡sRt‚nt t|dƒt|dƒ¡rRt‚d|jd< t t|dƒt|dƒ¡rxt‚t ||¡ d S )Nr   r   ©	r   Zsort_valuesr   r   r   r   r#   r„   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ƒ¡sDt‚d|jd< | r|t t|dƒt|dƒ¡rnt‚t ||¡ nt t|dƒt|dƒ¡s˜t‚d S )NÚinplaceTr   r   r  )r%   rÕ   r  rp   rÙ   r  r'   r'   r(   Útest_sort_values_inplace.  s    
r  Údecimalséÿÿÿÿc                 C   s  t ddgddœƒ}| ¡ }|j|d�}| rŠt t|dƒt|dƒ¡sFt‚|dkrlt t|dƒt|dƒ¡sˆt‚q¦t t|dƒt|dƒ¡r¦t‚nt t|dƒt|dƒ¡r¦t‚d	|jd
< d|jd< | röt t|dƒt|dƒ¡rÚt‚t t|dƒt|dƒ¡röt‚t 	||¡ d S )Nr   r   r   r{   )r  r   r   r   rë   r´   r   r   )
r   r   Úroundr   r   r   r   r#   r„   r…   )r%   r  r&   r‡   rq   r'   r'   r(   Ú
test_roundB  s    

r  c                 C   sÂ   t jddddgddgd�}tdd	d
ddgi|d�}| ¡ }|jddgd�}| rlt t|dƒt|dƒ¡sˆt‚nt t|dƒt|dƒ¡rˆt‚d|j	d< | r²t t|dƒt|dƒ¡r²t‚t
 ||¡ d S )N©r   r   ©r   r   ©r   r   ©r   r   ÚoneÚtwo©Únamesr   r   r   r   r   rj   ©Úorderr   r   )r   Úfrom_tuplesr   r   Úreorder_levelsr   r   r   r   r#   r„   r…   ©r%   r?   r&   r‡   rq   r'   r'   r(   Útest_reorder_levels[  s    
 ÿ
r&  c                 C   s¦   t jddddgddgd�}tdd	d
dg|d�}| ¡ }|jddgd�}| r`t |j|j¡stt‚nt |j|j¡rtt‚d|j	d< | r–t |j|j¡r–t‚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¸   r   r#   r„   r‹   rŒ   r'   r'   r(   Útest_series_reorder_levelsn  s    
 ÿ
r'  c                 C   sŽ   t jdddgddgd�}||_| ¡ }| ¡ }| rHt |j|j¡s\t‚nt |j|j¡r\t‚d|j	d< | r~t |j|j¡r~t‚t
 ||¡ d S )Nr  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_swaplevel�  s    
r(  c                 C   s²   t dddgdddgddd	gd
œƒ}| ¡ }|jdddgdd�}| r`t t|dƒt|dƒ¡s|t‚nt t|dƒt|dƒ¡r|t‚d|jd< t t|dƒt|dƒ¡r¢t‚t 	||¡ d S )Nr   r   r   r   r   r   r   r   r   r   r   r   r   r?   r¾   r   r   )
r   r   rB   r   r   r   r   r#   r„   r…   r†   r'   r'   r(   Útest_frame_set_axis“  s    "
r)  c                 C   sz   t dddgƒ}| ¡ }|jdddgdd�}| r@t ||¡sPt‚nt ||¡rPt‚d	|jd	< t ||¡rjt‚t ||¡ d S )
Nr   r   r   r   r   r   r?   r¾   r   )	r   r   rB   r   r   r   r#   r„   r‹   r  r'   r'   r(   Útest_series_set_axis¤  s    
r*  c                 C   sŒ   t dddgƒ}| ¡ }|jdd�}t ||¡s2t‚d|jd< | r^t ||¡rPt‚t ||¡ n*t ||¡snt‚t dddgƒ}t ||¡ d S )Nr   r   r   F)ra   r   )	r   r   rc   r   r   r   r#   r„   r‹   ©r%   rw   r�   ry   r•   r'   r'   r(   Útest_set_flagsµ  s    
r,  ZmapperrD   c                 C   s´   t dddddgitddddgdd�d�}| ¡ }|jf |Ž}| r^t t|dƒt|dƒ¡szt‚nt t|dƒt|dƒ¡rzt‚d|jd	< | r¤t t|dƒt|dƒ¡r¤t‚t	 
||¡ d S )
Nr   r   r   r   r   rÆ   rj   r   r   )r   r   r   rG   r   r   r   r   r#   r„   r…   r  r'   r'   r(   Útest_rename_axisÇ  s    (
r-  zfunc, tz)r_   zEurope/Berlin)r]   Nc                 C   sŽ   t ddgtddd|d�d�}| ¡ }t||ƒdƒ}| rLt |j|j¡s`t‚nt |j|j¡r`t‚d|jd< t |j|j¡r~t‚t	 
||¡ d S )	Nr   r   z2014-08-01 09:00ÚH)ÚstartrW   rf   ri   rj   r[   r   )r   r	   r   r	  r   r   r¸   r   r#   r„   r‹   )r%   rº   ri   rw   r�   ry   r'   r'   r(   Útest_tz_convert_localizeØ  s     ÿ
r0  c                 C   sÄ   t jdddgddgd�}tddd	gd
ddgdddgdœ|d�}| ¡ }| d¡}| rrt t|dƒt|dƒ¡sŽt‚nt t|dƒt|dƒ¡rŽt‚d|j	d< t t|dƒt|dƒ¡r´t‚t
 ||¡ d S )Nr  r  r  r  r  r  r   r   r   r   r   r   é   é   é	   r   rj   r   r   r   )r   r#  r   r   Z	droplevelr   r   r   r   r#   r„   r…   r%  r'   r'   r(   Útest_droplevelî  s    &

r4  c                 C   sœ   t ddddgiƒ}| ¡ }| ¡ }t |jt|dƒ¡s:t‚d|jd< | rnt |jt|dƒ¡r`t‚t	 
||¡ n*t |jt|dƒ¡s†t‚|jd dks˜t‚d S )Nr   r   r   r   r   rÃ   )r   r   Zsqueezer   r   r¸   r   r   r#   r„   r…   rÄ   )r%   r&   r‡   Zseriesr'   r'   r(   Útest_squeeze  s    
r5  c                 C   s¸   t dddgdddgddd	gd
œƒ}| ¡ }tdƒD ]€}| ¡ D ]r\}}t t||ƒt||ƒ¡sbt‚d|jd< | ršt t||ƒt||ƒ¡rŒt‚t	 
||¡ q>|jd|f dks>t‚q>q2d S )Nr   r   r   r   r   r   r1  r2  r3  r   r   )r   r   Úranger¥   r   r   r   r   r#   r„   r…   rÄ   )r%   r&   r‡   ÚirÇ   rw   r'   r'   r(   Ú
test_items  s    "
r8  rù   rH   ÚInt64c                 C   s˜   t ddgdddœ|d�}|d d … }| ¡ }d|||k< | rft t|dƒt|dƒ¡rXt‚t ||¡ n.t t|dƒt|dƒ¡s‚t‚|jd dks”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_putmask)  s    r:  c                 C   sL   t ddgdddœ|d�}t|dƒ}d|||k< | rHt |t|dƒ¡sHt‚d S )Nr   r   r   rø   r   r   ©r   r   r   r   r   ©r%   rù   r&   rþ   r'   r'   r(   Útest_putmask_no_reference9  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ƒ¡sRt‚d S )Nr˜   r   r{   rø   r   rê   r   r;  r<  r'   r'   r(   Ú$test_putmask_aligns_rhs_no_referenceC  s
    
r>  zval, exp)rê   T)r   Fc                 C   s<  t ddgdddœƒ}|d d … }| ¡ }t dddgdddggtdƒd�}|||< | rÞt t|d	ƒt|d	ƒ¡rpt‚t t|d
ƒt|d
ƒ¡|ks�t‚t t|dƒt|dƒ¡s¬t‚|j d¡|k	sÀt‚|j d¡rÐt‚t	 
||¡ nZ|dk�r8t t|d	ƒt|d	ƒ¡�st‚t t|dƒt|dƒ¡�s$t‚|jd dk�s8t‚d S )Nr   r   r˜   r   TFÚabcr�   r   r   r   r   r   )r   r   Úlistr   r   r   r   r    rŠ   r„   r…   r#   )r%   rí   Úexpr&   r  r‡   Zindexerr'   r'   r(   Ú"test_putmask_dont_copy_some_blocksM  s&     ÿ 
rB  c                 C   s–   t dddg|d�}| ¡ }| |dkd¡}| rHt t|ƒt|ƒ¡s`t‚nt t|ƒt|ƒ¡r`t‚d|jd< | r†t t|ƒt|ƒ¡r†t‚t 	||¡ d S ©Nr   r   r   rø   r   r   )
r   r   Úwherer   r   r   r   r#   r„   r‹   ©r%   rù   rw   r�   r}   r'   r'   r(   Útest_where_noopf  s    
rF  c                 C   sR   t dddg|d�}| ¡ }| |dk d¡}t t|ƒt|ƒ¡rBt‚t ||¡ d S rC  )	r   r   rD  r   r   r   r   r„   r‹   rE  r'   r'   r(   Ú
test_wherex  s
    rG  z
dtype, val)rH   g      %@)r9  r   c                 C   sÊ   t dddgdddgdœ|d�}| ¡ }| |d	k |¡}| rtt t|d
ƒt|d
ƒ¡sVt‚t t|dƒt|dƒ¡r�t‚nt t|d
ƒt|d
ƒ¡r�t‚d|jd< | rºt t|d
ƒt|d
ƒ¡rºt‚t 	||¡ d S )Nr   r   r   éüÿÿÿéûÿÿÿiúÿÿÿr{   rø   r   r   r   r   r´   )
r   r   rD  r   r   r   r   r#   r„   r…   )r%   rù   rí   r&   r‡   r}   r'   r'   r(   Ú test_where_noop_on_single_columnƒ  s    
rJ  c                 C   sª   t ddd ddgitdddd�d	�}| ¡ }|jdd
�}| rXt t|dƒt|dƒ¡stt‚nt t|dƒt|dƒ¡rtt‚d|jd< t t|dƒt|dƒ¡ršt‚t	 
||¡ d S )Nr   g        ç       @ç      @z1/1/2000r   ÚTr²   rj   rÖ   r   r   )r   r	   r   Zasfreqr   r   r   r   r#   r„   r…   r†   r'   r'   r(   Útest_asfreq_noop–  s    þ
rN  c                 C   sP   t dddœdddgd�}| ¡ }| ¡ D ]\}}d|jd< q(| rLt ||¡ d S )Nr   r   r{   r   r   rj   rt   )r   r   Ziterrowsr#   r„   r…   )r%   r&   r‡   rÀ   Úsubr'   r'   r(   Útest_iterrowsª  s    rP  c                 C   sr   t ddtjdgiƒ}|d d … }| ¡ }|jdd� d|jd< | rPt ||¡ nt ddddgiƒ}t ||¡ d S )Nr   r˜   r   T)r  g      Y@r   )r   r   rµ   r   Zffillr#   r„   r…   )r%   r&   r  r•   r'   r'   r(   Útest_interpolate_creates_copy´  s    
rQ  c                 C   s  t dddgdddgddd	gd
œƒ}| ¡ }|jd d�}| dt dddg¡¡ | rŒt t|dƒt|dƒ¡snt‚t t|dƒt|dƒ¡sÄt‚n8t t|dƒt|dƒ¡r¨t‚t t|dƒt|dƒ¡rÄt‚d|jd< t	 
||¡ | rüt t|dƒt|dƒ¡sút‚nt t|dƒt|dƒ¡�rt‚d S )Nr   r   r   r   r   r   r1  r2  r3  r   r*   r  éþÿÿÿéýÿÿÿr   r   r   rÃ   )r   r   Úisetitemr   Úarrayr   r   r   rÄ   r„   r…   r†   r'   r'   r(   Útest_isetitemÄ  s    "
rV  zsingle-blockzmixed-blockc                 C   sö   t dddgtjdddg|d�dœƒ}td	d
dgƒ}| ¡ }| d|¡ | rdt t|dƒt|ƒ¡rdt‚d|j	d< t
 ||¡ t dddgtjdddg|d�dœƒ}td	d
dgƒ}| d|¡ d|j	d< t d	d
dgtjdddg|d�dœƒ}t
 ||¡ d S )Nr   r   r   r   r   r   rø   r{   r1  r2  r3  r   r   rÃ   )r   r   rU  r   r   rT  r   r   r   rÄ   r„   r‹   r…   )r%   rù   r&   rw   r�   r•   r'   r'   r(   Útest_isetitem_seriesÚ  s    $
$
$rW  Úkeyc              
   C   sø   t dddgdddgdœƒ}| ¡ }| |¡}| r€t t|dƒt|dƒ¡sLt‚d	|jd	< t t|dƒt|dƒ¡rrt‚t 	||¡ ntt
|tƒrŽtnd }t d
d¡�& t |¡� d	|jd	< W 5 Q R X W 5 Q R X t
|tƒrât 	||¡ n|jd d	ksôt‚d S )Nr   r   r   r   r   r   r{   r   r   Úchained_assignmentÚwarnr   )r   r   Úgetr   r   r   r   r#   r„   r…   Ú
isinstancer@  r   ÚpdÚoption_contextÚassert_produces_warning)r%   rX  r&   r‡   r}   rZ  r'   r'   r(   Útest_getõ  s    


r`  z	axis, keyr   )r   r   c           
   
   C   s"  |dko| }|p|o|dk}t dddgdddgtjdd	d
g|d�dœƒ}| ¡ }|j||d�}	|dksl|rˆt t|dƒt|	ƒ¡sœt‚n| rœ|	j 	d¡sœt‚| s¤|r°d|	j
d< n8t dd¡�& t t¡� d|	j
d< W 5 Q R X W 5 Q R X | sü|�s
|dk�r
t ||¡ n|j
d dk�st‚d S )NrH   r   r   r   r   r   r   r1  r2  r3  rø   r   r¾   r   r   rY  rZ  r   )r   r   rU  r   Úxsr   r   r   r    rŠ   r#   r]  r^  r„   r_  r   r…   )
r%   rp   r@   rX  rù   Zsingle_blockZis_viewr&   r‡   r}   r'   r'   r(   Útest_xs  s&    &ÿrb  r@   z
key, level)Úl1r   r  c              
   C   sø   t  d¡ dd¡}tjddgdddggdd	gd
�}t||tdƒd�}|dkrX| ¡  ¡ }| ¡ }|j	|||d�}	|dkr t  
t||jd ƒt|	|	jd ƒ¡s t‚| s¬|s¬tnd }
t dd¡�& t |
¡� d|	jd< W 5 Q R X W 5 Q R X t ||¡ d S )Né   r   r   rc  Úl2r   r   Zlev1Zlev2r  r?  r¬   )Úlevelr@   r   rY  rZ  r   )r   ZarangeZreshaper   Zfrom_productr   r@  r|   r   ra  r   r   r/   r   r   r]  r^  r„   r_  r#   r…   )r%   rp   rX  rf  r@   Zarrr?   r&   r‡   r}   rZ  r'   r'   r(   Útest_xs_multiindex/  s,      ÿÿÿýrg  c                 C   sÌ   t dddgdddgdœƒ}t dd	gid
gd�}| ¡ }|d d … }| |¡ t dddgdd	dgdœƒ}t ||¡ | r¼t ||¡ t t|dƒt|dƒ¡sžt‚t t|dƒt|dƒ¡rÈt‚nt ||¡ d S )Nç      ð?rK  rL  g      @g      @g      @r{   r   ç      Y@r   rj   r   )	r   r   Úupdater„   r…   r   r   r   r   )r%   Zdf1rq   Zdf1_origr  r•   r'   r'   r(   Útest_update_frameL  s    
rk  c                 C   sx   t dddgƒ}t dgdgd�}| ¡ }|d d … }| |¡ t dddgƒ}t ||¡ | rht ||¡ nt ||¡ d S )Nrh  rK  rL  ri  r   rj   )r   r   rj  r„   r‹   )r%   Zser1ry   Z	ser1_origr  r•   r'   r'   r(   Útest_update_series_  s    
rl  c                  C   sF   t dddgƒ} t| ƒ}| d9 } t t| ƒ|¡s2t‚t |t| ƒ¡ d S ©Nr   r   r   )r   r   r   r   r   r„   Zassert_numpy_array_equal)rw   Údatar'   r'   r(   Útest_inplace_arithmetic_seriesp  s
    ro  c                 C   sp   t dddgƒ}| ¡ }|d d … }|d9 }| rTt t|ƒt|ƒ¡rFt‚t ||¡ nt t|ƒt|ƒ¡slt‚d S rm  )r   r   r   r   r   r   r„   r‹   )r%   rw   r�   r  r'   r'   r(   Ú-test_inplace_arithmetic_series_with_referencex  s    rp  c                 C   sŒ   t dddgddœƒ}| ¡ }|j|d�}|s0|r4| rRt t|dƒt|dƒ¡snt‚nt t|dƒt|dƒ¡rnt‚d|jd	< | rˆt 	||¡ d S )
Nr   r   r   r{   r8   r   r   rt   r   )
r   r   r|   r   r   r   r   r#   r„   r…   )r%   r   rp   r&   r‡   r}   r'   r'   r(   Útest_transpose„  s    
rq  c                 C   s\   t dddgddœƒ}| ¡ }|j}t t|dƒt|dƒ¡r>t‚d|jd	< | rXt 	||¡ d S )
Nr   r   r   r˜   r{   r   r   rt   r   )
r   r   rM  r   r   r   r   r#   r„   r…   )r%   r&   r‡   r}   r'   r'   r(   Útest_transpose_different_dtypes”  s    
rr  c                 C   s<   t ddddgidd�}|j}t t|dƒt|dƒ¡r8t‚d S )Nr   r   r   r   r9  rø   r   )r   rM  r   r   r   r   )r%   r&   r}   r'   r'   r(   Útest_transpose_ea_single_columnŸ  s    rs  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{   rt   r   r   r   rj   )r   Úcountr#   r   r„   r‹   râ   r'   r'   r(   Útest_count_read_only_array¦  s
    
ru  c                 C   s„   t dddgƒ}| ¡ }| ¡ }t t|ƒt|ƒ¡s6t‚| rJ|j d¡rJt‚d|j	d< | rft
 ||¡ nt dddgƒ}t
 ||¡ d S )Nr   r   r   r   rt   )r   r   r  r   r   r   r   r    rŠ   r#   r„   r‹   r+  r'   r'   r(   Útest_series_view®  s    
rv  )mÚnumpyr   ZpytestZpandas.errorsr   Zpandasr]  r   r   r   r   r   r   r	   r
   Zpandas._testingZ_testingr„   Zpandas.tests.copy_view.utilr   r)   r,   ÚmarkZparametrizers   rz   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  r&  r'  r(  r)  r*  r,  r-  r0  r4  r5  r8  r:  r=  r>  rB  rF  rG  rJ  rN  rP  rQ  rV  rW  r`  rb  rg  rk  rl  ro  rp  rq  rr  rs  ru  rv  r'   r'   r'   r(   Ú<module>   sô  (
ïïë)ïïë)
 
 ÿ
ýþ
"




ü
ø

(
(
	

üþ	


þþ

$
$
.þ
.þ

(

 ÿ


	
	




  
ÿ  
ÿ