U
    sVci                     @   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   F/tmp/pip-unpacked-wheel-xj8nt62q/pandas/tests/frame/test_alter_axes.pytest_set_axis_setattr_index   s
    


z2TestDataFrameAlterAxes.test_set_axis_setattr_indexc                 C   sT   d|d< |  }dddddg|_tj|d |d d	d
 tj|d |d d	d
 d S )NZtherehir	   barZbazZquuxZfoo2CF)Zcheck_names)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   