a
    Pfo
                     @   sh   d dl Zd dlZd dl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d	 Zd
d ZdS )    N)	DataFrameHDFStoreIndexSeries
date_range)ensure_clean_storetablesc                 C   s   t | }ttjdtjdtdddd|d< ttdddd	 tdD d
|d< tdtdd t	t
dtdt	dd	 tdD tdd|d< t|dksJ h d}t| |ksJ t||ksJ W d    n1 s0    Y  d S )N
   )dtypez
2020-01-01)Zperiods)indexafloat64c                 S   s   g | ]}d | qS )Zi_ .0ir   r   [/var/www/ai-form-bot/venv/lib/python3.9/site-packages/pandas/tests/io/pytables/test_keys.py
<listcomp>       ztest_keys.<locals>.<listcomp>)r
   r   bg?x   )      ZABCDc                 S   s   g | ]}d | qS )zi-r   r   r   r   r   r      r   r   )columnsr   c   >   z/bz/az/c)r   r   npZaranger   r   ranger   Zreshaper   listobjectlensetkeys)
setup_pathstoreexpectedr   r   r   	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 3nativeincluder   >   z/group/table3z/group/table2z/group/table1pandas   )r   ZIsDescription	open_fileZcreate_groupZcreate_tabler   r    r"   r!   getr   )Ztmp_pathr#   r+   r,   r-   pathZh5filer1   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illegalr3   )r   pytestZraises
ValueErrorr"   )r#   r$   r   r   r   'test_keys_illegal_include_keyword_valueA   s    
rA   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   r   putr"   Z_handleZcreate_soft_linkroot)r#   r$   r;   r   r   r   test_keys_ignore_hdf_softlinkJ   s    
rH   )numpyr   r?   r5   r   r   r   r   r   Zpandas.tests.io.pytables.commonr   r   markZ
single_cpuZ
pytestmarkr&   r<   rA   rH   r   r   r   r   <module>   s   	