a
    Pf                     @   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}t|d |jd|dd	 |d}t	|| d
D ]6}dD ],}t
t
|||d t
t
|||d kspJ qpqht ddd|d}ttj4 tdttd|di}	|d|	 W d    n1 s0    Y  |djd d d u sJ 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    n1 s0    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[])Zdtype)r   r   r   ranger
   r	   putgettmZassert_frame_equalgetattrassert_produces_warningr   AttributeConflictWarningappendZ
get_storerinfor   )
setup_pathr   ZdtidfstoreresultattridxZdti2r   Zdti3Zdti4Zdf3 r-   h/var/www/ai-form-bot/venv/lib/python3.9/site-packages/pandas/tests/io/pytables/test_retain_attributes.pytest_retain_index_attributes   s@    

*
r/   c              
   C   sv  | | }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 d    n1 s0    Y  t
|ddjj	dksJ t tjL tdddd}d|_	tdttd|di}|j|dd	d W d    n1 sP0    Y  t
|djj	d u srJ d S )Nr   r   r   r   )r   r   r   r   wT)keymoder%   r   r   )r1   r%   Zfoo)r1   z2001-1-1bar)r!   r#   r   r$   r   r   r   r   Zto_hdfr   r   r   )Ztmp_pathr'   pathr(   r   r,   Zidx2r-   r-   r.   test_retain_index_attributes2A   s*    00r5   )ZpytestZpandasr   r   r   r   r!   r   r   r   Zpandas.tests.io.pytables.commonr	   r
   markZ
single_cpuZ
pytestmarkr/   r5   r-   r-   r-   r.   <module>   s
   $	-