o
    ҷh                     @   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   s  t dttdtdddddi}t| }t|d |jd|dd	 |d}t	|| d
D ]}dD ]}t
t
|||d t
t
|||d ksJJ q4q0ttj t dttdtdddddi}|d| W d    n1 stw   Y  |djd d d u sJ 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    d S 1 sw   Y  d S )NA   2000-1-1Hperiodsfreqindexdatatable)format)r   tzname)r   columns2002-1-1Dr   r   df2200101012001010220020101)r   r   ranger   r
   r	   putgettmassert_frame_equalgetattrassert_produces_warningr   AttributeConflictWarningappend
get_storerinfor   )
setup_pathdfstoreresultattridxr   df3 r2   b/var/www/html/venv/lib/python3.10/site-packages/pandas/tests/io/pytables/test_retain_attributes.pytest_retain_index_attributes   sR   




"r4   c              
   C   sp  | | }t tjZ 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    n1 sgw   Y  t
|djj	dkswJ t tj& tdddd}d|_	tdttd|di}|j|dd	d W d    n1 sw   Y  t
|djj	d u sJ d S )Nr   r   r   r   r   r   r   wT)moder(   r   r   )r(   fooz2001-1-1bar)r#   r&   r   r'   r   r   r    r   to_hdfr   r   r   )tmp_pathr+   pathr,   r   r0   idx2r2   r2   r3   test_retain_index_attributes2J   s.   r=   )pytestpandas._libs.tslibsr   pandasr   r   r   r#   r   r   r   pandas.tests.io.pytables.commonr	   r
   mark
single_cpu
pytestmarkr4   r=   r2   r2   r2   r3   <module>   s     5