a
    Pf                     @   s6   d dl Z d dlmZmZ d dlmZ G dd dZdS )    N)	DataFrameSeriesc                   @   s$   e Zd Zdd Zdd Zdd ZdS )TestDataFrameSetItemc                 C   sx   t g dg dg}t ddgddggdd}|d	d
g| t tddgddtddgddddgd}t|| d S )N                              Int64Zdtyper   r   r   r   r   r   r   r   isetitemr   tmZassert_frame_equalselfdfrhsexpected r   a/var/www/ai-form-bot/venv/lib/python3.9/site-packages/pandas/tests/frame/methods/test_isetitem.pytest_isetitem_ea_df   s    z(TestDataFrameSetItem.test_isetitem_ea_dfc                 C   sh   t g dg dg}t dgdggdd}|d| t dd	gdd
gtddgddd}t|| d S )Nr   r	   r   r   r   r   r   r   r
   r   r   r   r   r   r   r   "test_isetitem_ea_df_scalar_indexer   s    z7TestDataFrameSetItem.test_isetitem_ea_df_scalar_indexerc                 C   s   t ddgddgddgd}| }tjtdd	  |ddg| W d    n1 sX0    Y  | }tjtd
d	& |ddg|dg  W d    n1 s0    Y  d S )Nr   r   r   r
   r   r   )abcz'Got 2 positions but value has 3 columns)matchz'Got 2 positions but value has 1 columnsr    )r   copypytestZraises
ValueErrorr   )r   r   valuer   r   r    test_isetitem_dimension_mismatch)   s    .z5TestDataFrameSetItem.test_isetitem_dimension_mismatchN)__name__
__module____qualname__r   r   r(   r   r   r   r   r   
   s   r   )r%   Zpandasr   r   Zpandas._testingZ_testingr   r   r   r   r   r   <module>   s   