U
    sVcE                     @   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   G/tmp/pip-unpacked-wheel-xj8nt62q/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st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|rt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*   )Znumpyr   Zpandasr   r   r   r   Zpandas._testingZ_testingr   r   r    r#   r*   r   r   r   r   <module>   s   