o
    թZh	                     @   sF   d dl Z d dlZd dlmZ e dZe jdd Z	G dd dZ
dS )    Ntablesc                 c   s    t jddt jdddt jddd}d}|ddd|d d	dd|d d
dd|d dddg}d}| d }t j|dd(}|jd||d}|D ]}| D ]	\}	}
|
|j|	< qP|j  qJW d   n1 sjw   Y  ||t	
|fV  dS )z4
    Use PyTables to create a simple HDF5 file.
    r   )pos         )Zc0c1c2g   $CAZaaaaaZbbbbbZccccci    Zdddddl    Zpandas_test_timeserieszwritten_with_pytables.h5w)mode/)namedescriptionN)r   Z	Time64ColZ	StringColZInt64Col	open_fileZcreate_tableitemsrowappendpdZ	DataFrame)Ztmp_pathZtable_schemat0Ztestsamplesobjnamepathftsamplekeyvalue r   [/var/www/html/lang_env/lib/python3.10/site-packages/pandas/tests/io/pytables/test_compat.pypytables_hdf5_file	   s,   


r   c                   @   s0   e Zd ZdZdd Zdd Zdd Zdd	 Zd
S )TestReadPyTablesHDF5z
    A group of tests which covers reading HDF5 files written by plain PyTables
    (not written by pandas).

    Was introduced for regression-testing issue 11188.
    c                 C   s0   |\}}}t j||d}|}tj||dd d S )N)r   TZcheck_index_type)r   read_hdftmassert_frame_equalselfr   r   r   dfresultexpectedr   r   r   test_read_complete2   s   
z'TestReadPyTablesHDF5.test_read_completec                 C   sB   |\}}}t j||dd}|dd  jdd}tj||dd d S )Nr   )r   startTZdropr    r   r!   Zreset_indexr"   r#   r$   r   r   r   test_read_with_start8      
z)TestReadPyTablesHDF5.test_read_with_startc                 C   sB   |\}}}t j||dd}|d d jdd}tj||dd d S )Nr   )r   stopTr+   r    r,   r$   r   r   r   test_read_with_stop?   r.   z(TestReadPyTablesHDF5.test_read_with_stopc                 C   sD   |\}}}t j||ddd}|dd jdd}tj||dd d S )Nr   r   )r   r*   r/   Tr+   r    r,   r$   r   r   r   test_read_with_startstopF   s   
z-TestReadPyTablesHDF5.test_read_with_startstopN)__name__
__module____qualname____doc__r)   r-   r0   r1   r   r   r   r   r   *   s    r   )ZpytestZpandasr   Zpandas._testingZ_testingr"   Zimportorskipr   Zfixturer   r   r   r   r   r   <module>   s    

 