U
    NZf                     @   s`   d dl Z d dlmZ d dlmZmZmZmZm	Z	m
Z
 d dlmZmZ e jjZdd Zdd ZdS )	    N)	Timestamp)	DataFrameSeries_testing
date_rangeerrorsread_hdf)_maybe_removeensure_clean_storec                 C   sx  t dttdtdddddi}t| B}t|d |jd|dd	 |d}t	|| d
D ]6}dD ],}t
t
|||d t
t
|||d ksjtqjqbttj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   columns2002-1-1Dr   r   df2Z20010101Z20010102Z20020101)r   r   ranger   r
   r	   putgettmZassert_frame_equalgetattrAssertionErrorassert_produces_warningr   AttributeConflictWarningappendZ
get_storerinfor   )
setup_pathdfstoreresultattridxr   Zdf3 r,   S/tmp/pip-unpacked-wheel-nbcvw55c/pandas/tests/io/pytables/test_retain_attributes.pytest_retain_index_attributes   sV    

    
 r.   c              
   C   sJ  | | }t tj 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 tj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d S )Nr   r   r   r   r   r   r   wT)moder$   r   r   )r$   Zfooz2001-1-1bar)r   r"   r   r#   r   r   r   r   Zto_hdfr   r   r   r!   )Ztmp_pathr&   pathr'   r   r+   Zidx2r,   r,   r-   test_retain_index_attributes2J   s*    r3   )ZpytestZpandas._libs.tslibsr   Zpandasr   r   r   r   r   r   r   Zpandas.tests.io.pytables.commonr	   r
   markZ
single_cpuZ
pytestmarkr.   r3   r,   r,   r,   r-   <module>   s    5