a
    Pfi                     @   s>   d dl m Z  d dlZd dlmZ d dlmZ G dd dZdS )    )datetimeN)	DataFramec                   @   s   e Zd Zdd Zdd ZdS )TestDataFrameAlterAxesc                 C   sL   t tdddtjdddg}|d}|d |_|d t|| d S )Ni        )tzinfo)tsfoor   )	r   r   pytzutcZ	set_indexindexpoptmZassert_frame_equal)selfdfexpected r   [/var/www/ai-form-bot/venv/lib/python3.9/site-packages/pandas/tests/frame/test_alter_axes.pytest_set_axis_setattr_index   s
    


z2TestDataFrameAlterAxes.test_set_axis_setattr_indexc                 C   sN   d|d< |  }g d|_tj|d |d dd tj|d |d dd d S )	NZtherehi)r	   barbazZquuxfoo2Cr   F)Zcheck_namesr   )copycolumnsr   Zassert_series_equal)r   Zfloat_framer   r   r   r   test_assign_columns   s
    
z*TestDataFrameAlterAxes.test_assign_columnsN)__name__
__module____qualname__r   r   r   r   r   r   r   	   s   r   )r   r
   Zpandasr   Zpandas._testingZ_testingr   r   r   r   r   r   <module>   s   