U
    c                     @   s6   d dl Zd dlmZ d dlmZmZ G dd dZdS )    N)is_float_dtype)	DataFrameisnac                   @   s$   e Zd Zdd Zdd Zdd ZdS )TestSetValuec                 C   s>   |j D ]2}|jD ]&}|||d || | dkstqqd S )N   )indexcolumns
_set_valueAssertionError)selffloat_frameidxcol r   N/tmp/pip-unpacked-wheel-g7fro6k3/pandas/tests/frame/indexing/test_set_value.pytest_set_value   s    

zTestSetValue.test_set_valuec                 C   s  | ddd}|d kst|jd dks,t|dddks@td|jd< |dddks^t| }| ddd |d jtjkst| }| ddd	 |d jtjkst| }| ddd
 t	|d stt
|d dg st| ddd |jd dkstd S )NfoobarBr   )r   quxr   bazZsamT   )r   r   )r	   r
   r   
_get_valueloccopyZdtypenpZobject_r   r   Zdropall)r   r   resr   r   r   test_set_value_resize   s$    
z"TestSetValue.test_set_value_resizec                 C   s,  t tjddtdtdd}| }|ddd t|jt|jdg ksTt	| }d|j
d< t|jt|jdg kst	| }|ddd t|jt|jdg kst	t|jt|jdg kst	| }d|j
d	< t|jt|jdg kst	t|jt|jdg ks(t	d S )
N   ABC)r   r   C   g      ?)r!   r"   D)r!   r#   )r   r   randomZrandnrangelistr   r	   r   r
   r   r   )r   Zdf_origZdfr   r   r   &test_set_value_with_index_dtype_change,   s     

 z3TestSetValue.test_set_value_with_index_dtype_changeN)__name__
__module____qualname__r   r   r'   r   r   r   r   r      s   r   )Znumpyr   Zpandas.core.dtypes.commonr   Zpandasr   r   r   r   r   r   r   <module>   s   