o
    ôT·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 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   ƒ d S 1 sÕw   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   úr/home/dinkstrade/pdmp-scanner/venv/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 s§w   Y  t
|dƒjj	d u s·J ‚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    $	-