B
    Z0d/                 @   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   s  t dttdtdddddi}t| J}t|d |jd|dd	 |d}t	|| xBd
D ]:}x4dD ],}t
t
|||d t
t
|||d ksntqnW qdW tdd4 t dttdtdddddi}|d| W d Q R X |djd d d kst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 d Q R X d S )NA   z2000-1-1H)periodsfreq)indexdatatable)format)r   tzname)r   columnsT)recordz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)   a/var/www/html/venv/lib/python3.7/site-packages/pandas/tests/io/pytables/test_retain_attributes.pytest_retain_index_attributes   s8     




 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 d 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 d Q R X t|d	j	jd ksFt
W d Q R X d S )NT)r   r   r   z2000-1-1r   )r   r   )r   r   w)moder!   z2002-1-1r   )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*    r0   )warningsr   ZpytestZpandas._libs.tslibsr   Zpandasr   r   r   r   r   r   Zpandas.tests.io.pytables.commonr	   r
   r   markZsingleZ
pytestmarkr+   filterwarningsr0   r)   r)   r)   r*   <module>   s   6