a
    d                     @   sX   d dl Z d dlmZmZmZ d dlmZmZ e j	j
Zdd Zdd Zdd	 Zd
d ZdS )    N)	DataFrameHDFStore_testing)ensure_clean_storetablesc                 C   s   t | p}t |d< t |d< t |d< t|dks>J h d}t| |ksZJ t||ksjJ W d    n1 s~0    Y  d S )Nabc   >   z/az/cz/b)r   tmZmakeTimeSeriesZmakeStringSeriesZmakeDataFramelensetkeys)
setup_pathstoreexpected r   k/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/pandas/tests/io/pytables/test_keys.py	test_keys   s    
r   c                 C   sT  G dd dt j}G dd dt j}G dd dt j}| | }t j|ddL}|d	d
}||d|d ||d|d ||d|d W d    n1 s0    Y  t|}t|jdddksJ h d}	t|jdd|	ksJ t|jddt ksJ |	D ]$}
|	|
}t|j
dks
J q
W d    n1 sF0    Y  d S )Nc                   @   s   e Zd Ze ZdS )z$test_non_pandas_keys.<locals>.Table1N)__name__
__module____qualname__r   
Float32ColZvalue1r   r   r   r   Table1   s   r   c                   @   s   e Zd Ze ZdS )z$test_non_pandas_keys.<locals>.Table2N)r   r   r   r   r   Zvalue2r   r   r   r   Table2    s   r   c                   @   s   e Zd Ze ZdS )z$test_non_pandas_keys.<locals>.Table3N)r   r   r   r   r   Zvalue3r   r   r   r   Table3#   s   r   w)mode/groupZtable1zTable 1Ztable2zTable 2Ztable3zTable 3Znativeincluder
   >   z/group/table1z/group/table3z/group/table2pandas   )r   ZIsDescription	open_fileZcreate_groupZcreate_tabler   r   r   r   getcolumns)Ztmp_pathr   r   r   r   pathZh5filer   r   r   namedfr   r   r   test_non_pandas_keys   s"    .

r*   c              	   C   sf   t | J}tjtdd |jdd W d    n1 s:0    Y  W d    n1 sX0    Y  d S )Nz@`include` should be either 'pandas' or 'native' but is 'illegal')matchillegalr    )r   pytestZraises
ValueErrorr   )r   r   r   r   r   'test_keys_illegal_include_keyword_value6   s    
r/   c                 C   s   t | j}ttdtdd}|d| | dgks>J |j|jjdd | dgksdJ W d    n1 sx0    Y  d S )N   )ABr)   z/dfsymlink)r   r   rangeputr   _handleZcreate_soft_linkroot)r   r   r)   r   r   r   test_keys_ignore_hdf_softlink?   s    
r8   )r-   r"   r   r   r   r   Zpandas.tests.io.pytables.commonr   r   markZ
single_cpuZ
pytestmarkr   r*   r/   r8   r   r   r   r   <module>   s   	