U
    Mf3                     @   sx   d dl mZ d dlZd dlmZ d dlmZmZmZ	m
Z
mZ d dlmZmZmZ ejjZdd Zejdd	d
 ZdS )    )catch_warningsN)	Timestamp)	DataFrameSeries_testing
date_rangeread_hdf)_maybe_removeensure_clean_pathensure_clean_storec                 C   sv  t dttdtdddddi}t| @}t|d |jd|dd	 |d}t	|| d
D ]6}dD ],}t
t
|||d t
t
|||d ksjtqjqbtdd4 t dttdtdddddi}|d| W 5 Q R X |djd d d kstt|d t dttdtdtdtdgdi}|d| t dttdtdddddi}|d| W 5 Q R X d S )NA   2000-1-1HZperiodsfreqindexdatatable)format)r   tzname)r   columnsTrecord2002-1-1Dr   r   df2Z20010101Z20010102Z20020101)r   r   ranger   r   r	   putgettmZassert_frame_equalgetattrAssertionErrorr   appendZ
get_storerinfor   )
setup_pathdfstoreresultattridxr   Zdf3 r-   S/tmp/pip-unpacked-wheel-eb6vo0j3/pandas/tests/io/pytables/test_retain_attributes.pytest_retain_index_attributes   sV    

    
 r/   z9ignore:\nthe :pandas.io.pytables.AttributeConflictWarningc                 C   sT  t | @}tdd tdttdtdddddi}|j|d	d
dd tdttdtdddddi}|j|d	dd tdddd}d|_tdttd|di}|j|d	d
dd W 5 Q R X t|d	j	jdkst
tddB tdddd}d|_tdttd|di}|j|d	dd W 5 Q R X t|d	j	jd ksFt
W 5 Q R X d S )NTr   r   r   r   r   r   r   r   w)moder%   r   r   )r%   Zfooz2001-1-1bar)r
   r   r   r   r   r   Zto_hdfr   r   r   r$   )r'   pathr(   r   r,   Zidx2r-   r-   r.   test_retain_index_attributes2M   s>        r4   )warningsr   ZpytestZpandas._libs.tslibsr   Zpandasr   r   r   r"   r   r   Zpandas.tests.io.pytables.commonr	   r
   r   markZ
single_cpuZ
pytestmarkr/   filterwarningsr4   r-   r-   r-   r.   <module>   s   6