o
    ŀg                     @   sX   d dl Z d dlmZ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)	DataFrameDatetimeIndexSeries_testing
date_rangeerrorsread_hdf)_maybe_removeensure_clean_storec              	   C   s  t ddd|d}tdttd|di}t| }t|d |jd|dd	 |d}t	|| d
D ]}dD ]}t
t
|||d t
t
|||d ksMJ q7q3t ddd|d}ttj tdttd|di}	|d|	 W d    n1 szw   Y  |djd d d u sJ t|d tg dd| dd}
tdttd|
di}	|d|	 t ddd|d}tdttd|di}|d| W d    d S 1 sw   Y  d S )N2000-1-1   h)periodsfrequnitAindexdatatable)format)r   tzname)r   columns2002-1-1Dr   r   df2)z
2001-01-01z
2001-01-02z
2002-01-01zM8[])dtype)r   r   r   ranger
   r	   putgettmassert_frame_equalgetattrassert_produces_warningr   AttributeConflictWarningappend
get_storerinfor   )
setup_pathr   dtidfstoreresultattridxdti2r   dti3dti4df3 r5   c/var/www/html/myenv/lib/python3.10/site-packages/pandas/tests/io/pytables/test_retain_attributes.pytest_retain_index_attributes   sF   



"r7   c              
   C   sr  | | }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djj	dksxJ 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   r   wT)keymoder'   r   r   )r9   r'   foo)r9   z2001-1-1bar)r"   r%   r   r&   r   r   r   r   to_hdfr   r   r   )tmp_pathr*   pathr,   r   r0   idx2r5   r5   r6   test_retain_index_attributes2A   s.   rA   )pytestpandasr   r   r   r   r"   r   r   r   pandas.tests.io.pytables.commonr	   r
   mark
single_cpu
pytestmarkr7   rA   r5   r5   r5   r6   <module>   s    $	-