o
    %j                     @   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[])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-   p/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/pandas/tests/io/pytables/test_retain_attributes.pytest_retain_index_attributes   sF   



"r/   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   )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.   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    $	-