U
    ¾mœdE  ã                   @   sP   d dl Zd dlmZmZmZmZ d dlmZ	 dd„ Z
dd„ Zdd„ Zd	d
„ ZdS )é    N)Ú	DataFrameÚIndexÚ
RangeIndexÚSeriesc                  C   sv   t dddgdddgdœƒ} tjdddgdd	�}|| d
< t | d
 j|¡rLt‚d|d< t | d
 tdddgd
d�¡ d S )Né   é   é   é   é   é   ©ÚaÚbZint64)ZdtypeÚcr   ©Úname)	r   ÚnpÚarrayÚshares_memoryÚvaluesÚAssertionErrorÚtmÚassert_series_equalr   )ÚdfÚarr© r   ú\/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/pandas/tests/copy_view/test_setitem.pyÚtest_set_column_with_array   s    r   c                 C   s¤   t dddgdddgdœƒ}tdddgƒ}||d< | rNt |d j|j¡rft‚nt |d j|j¡rft‚d	|jd	< |jd	 d	ks‚t‚t |d tdddgdd
�¡ d S )Nr   r   r   r	   r
   r   r   r   r   r   )	r   r   r   r   r   r   Úilocr   r   )Úusing_copy_on_writer   Zserr   r   r   Útest_set_column_with_series   s    
r    c                 C   sÈ   t dddgdddgdœƒ}tdddgƒ}||d< t |d j|j¡rHt‚d	|jd	< t |d tdddgdd
�¡ t	ddƒ}|j}||d< t |d j|¡ržt‚d	|d	< t |d tdddgdd
�¡ d S )Nr   r   r   r	   r
   r   r   r   r   r   Úd)
r   r   r   r   r   r   r   r   r   r   )r   r   Úidxr   r   r   r   Útest_set_column_with_index2   s    

r#   c                 C   sª   t dddgdddgdœƒ}t dd	d
gdddgdœƒ}||ddg< | rbt |d j|d j¡r~t‚nt |d j|d j¡r~t‚d|jd< t |d tdd	d
gdd�¡ d S )Nr   r   r   r	   r
   r   r   é   é   é	   é
   é   é   )r   r!   r   r!   r   )r   r   r   )	r   r   r   r   r   r   r   r   r   )r   r   Zdf2r   r   r   Útest_set_columns_with_dataframeJ   s    
r*   )Únumpyr   Zpandasr   r   r   r   Zpandas._testingZ_testingr   r   r    r#   r*   r   r   r   r   Ú<module>   s   