o
    թZho
                     @   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 | m}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sSJ h d}t| |ksaJ t||ksiJ W d    d S 1 stw   Y  d S )N
   )dtypez
2020-01-01)Zperiods)indexafloat64c                 S      g | ]}d | qS )Zi_ .0ir   r   Y/var/www/html/lang_env/lib/python3.10/site-packages/pandas/tests/io/pytables/test_keys.py
<listcomp>       ztest_keys.<locals>.<listcomp>)r
   r   bg?x   )      ZABCDc                 S   r   )zi-r   r   r   r   r   r      r   r   )columnsr   c   >   z/az/cz/b)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   sN  G dd dt j}G dd dt j}G dd dt j}| | }t j|dd&}|d	d
}||d|d ||d|d ||d|d W d    n1 sOw   Y  t|E}t|jdddkseJ h d}	t|jdd|	ksuJ t|jddt ksJ |	D ]}
|	|
}t|j
dksJ qW d    d S 1 sw   Y  d S )Nc                   @      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(       r-   c                   @   r(   )z$test_non_pandas_keys.<locals>.Table2N)r)   r*   r+   r   r,   Zvalue2r   r   r   r   Table2+   r.   r/   c                   @   r(   )z$test_non_pandas_keys.<locals>.Table3N)r)   r*   r+   r   r,   Zvalue3r   r   r   r   Table3.   r.   r0   w)mode/groupZtable1zTable 1Ztable2zTable 2Ztable3zTable 3nativeincluder   >   z/group/table1z/group/table2z/group/table3pandas   )r   ZIsDescription	open_fileZcreate_groupZcreate_tabler   r!   r#   r"   getr   )Ztmp_pathr$   r-   r/   r0   pathZh5filer4   r%   r&   namedfr   r   r   test_non_pandas_keys'   s(   

"r?   c              	   C   sx   t | .}tjtdd |jdd W d    n1 sw   Y  W d    d S W d    d S 1 s5w   Y  d S )Nz@`include` should be either 'pandas' or 'native' but is 'illegal')matchillegalr6   )r   pytestZraises
ValueErrorr#   )r$   r%   r   r   r   'test_keys_illegal_include_keyword_valueA   s   
"rD   c                 C   s   t | 6}ttdtdd}|d| | dgksJ |j|jjdd | dgks2J W d    d S 1 s=w   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   
"rK   )numpyr   rB   r8   r   r   r   r   r   Zpandas.tests.io.pytables.commonr   r   markZ
single_cpuZ
pytestmarkr'   r?   rD   rK   r   r   r   r   <module>   s    	