U
    ¾mœdI  ã                   @   sX   d dl Zd dlmZ d dlmZ d dlmZ dd„ Z	dd„ Z
dd	„ Zd
d„ Zdd„ ZdS )é    N)Ú	DataFrame)Ú	get_arrayc                 C   sˆ   t ddddgiƒ}| ¡ }t|dƒ}|d d … }|jddd� t t|dƒ|¡rTt‚| r„|j d¡sht‚|j d¡sxt‚t	 
||¡ d S ©NÚaç      ø?é   é   T©ÚlowerZinplacer   ©r   Úcopyr   ÚclipÚnpÚshares_memoryÚAssertionErrorÚ_mgrÚ_has_no_referenceÚtmÚassert_frame_equal©Úusing_copy_on_writeÚdfZdf_copyÚarr_aÚview© r   úY/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/pandas/tests/copy_view/test_clip.pyÚtest_clip_inplace_reference   s    
r   c                 C   s    t ddddgiƒ}| ¡ }t|dƒ}|d d … }|jddd� | r†t t|dƒ|¡sXt‚|j d¡rht‚|j d¡rxt‚t	 
||¡ nt t|dƒ|¡rœt‚d S )Nr   r   r   r   r   Tr	   r   r   r   r   r   Ú!test_clip_inplace_reference_no_op   s    
r   c                 C   sX   t ddddgiƒ}t|dƒ}|jddd� t t|dƒ|¡r@t‚| rT|j d¡sTt‚d S r   )r   r   r   r   r   r   r   r   )r   r   r   r   r   r   Útest_clip_inplace(   s    
r   c                 C   sf   t ddddgiƒ}| ¡ }|jdd�}t t|dƒt|dƒ¡rBt‚| rV|j d¡sVt‚t	 
||¡ d S )Nr   r   r   r   ©r
   r   )r   r   r   r   r   r   r   r   r   r   r   )r   r   Zdf_origÚdf2r   r   r   Ú	test_clip4   s    r!   c                 C   sp   t ddddgiƒ}|jdd�}| rP|j d¡r2t‚t t|dƒt|dƒ¡slt‚nt t|dƒt|dƒ¡rlt‚d S )Nr   r   r   r   r   r   )r   r   r   r   r   r   r   r   )r   r   r    r   r   r   Útest_clip_no_op@   s    r"   )Únumpyr   Zpandasr   Zpandas._testingZ_testingr   Zpandas.tests.copy_view.utilr   r   r   r   r!   r"   r   r   r   r   Ú<module>   s   