
    bMhq
                     t    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        gZd Zd Zd Zd ZdS )    N)	DataFrameHDFStoreIndexSeries
date_range)ensure_clean_storetablesc                    t          |           5 }t          t          j        dt          j                  t          dd                    |d<   t          t          d          dd t          d          D             	          |d
<   t          dt          j        d                              d          z  t          t          d          t                    t          d t          d          D             t                              |d<   t          |          dk    sJ h d}t          |                                          |k    sJ t          |          |k    sJ 	 d d d            d S # 1 swxY w Y   d S )N
   )dtypez
2020-01-01)periods)indexafloat64c                     g | ]}d | S )i_ .0is     b/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/pandas/tests/io/pytables/test_keys.py
<listcomp>ztest_keys.<locals>.<listcomp>   s    .K.K.KAxAxx.K.K.K    )r   r   bg?x   )      ABCDc                     g | ]}d | S )zi-r   r   s     r   r   ztest_keys.<locals>.<listcomp>   s    555aa555r   r   )columnsr   c   >   /a/b/c)r   r   nparanger   r   ranger   reshaper   listobjectlensetkeys)
setup_pathstoreexpecteds      r   	test_keysr2      s   	J	'	' &5Ib
+++:lTV3W3W3W
 
 
c
 "IIY.K.Kr.K.K.K
 
 
c
 ")C..((111$v,,f55555599555VDDD
 
 
c
 5zzQ%%%5::<<  H,,,,5zzX%%%%%!& & & & & & & & & & & & & & & & & &s   EE77E;>E;c                     G d dt           j                  } G d dt           j                  } G d dt           j                  }| |z  }t          j        |d          5 }|                    d	d
          }|                    |d|d           |                    |d|d           |                    |d|d           d d d            n# 1 swxY w Y   t          |          5 }t          |                    d                    dk    sJ h d}	t          |                    d                    |	k    sJ t          |                    d                    t                      k    sJ |	D ]1}
|	                    |
          }t          |j
                  dk    sJ 2	 d d d            d S # 1 swxY w Y   d S )Nc                   ,    e Zd Z ej                    ZdS )$test_non_pandas_keys.<locals>.Table1N)__name__
__module____qualname__r	   
Float32Colvalue1r   r   r   Table1r5   (           ""$$r   r;   c                   ,    e Zd Z ej                    ZdS )$test_non_pandas_keys.<locals>.Table2N)r6   r7   r8   r	   r9   value2r   r   r   Table2r>   +   r<   r   r@   c                   ,    e Zd Z ej                    ZdS )$test_non_pandas_keys.<locals>.Table3N)r6   r7   r8   r	   r9   value3r   r   r   Table3rB   .   r<   r   rD   w)mode/grouptable1zTable 1table2zTable 2table3zTable 3nativeincluder"   >   /group/table1/group/table2/group/table3pandas   )r	   IsDescription	open_filecreate_groupcreate_tabler   r,   r.   r-   getr    )tmp_pathr/   r;   r@   rD   pathh5filerH   r0   r1   namedfs               r   test_non_pandas_keysr^   '   s   % % % % %% % % %% % % % %% % % %% % % % %% % % % j D		$S	)	)	) @V##C11E8VY???E8VY???E8VY???	@ @ @ @ @ @ @ @ @ @ @ @ @ @ @
 
$ (55::h://00A5555FFF5::h://00H<<<<5::h://00CEE9999 	( 	(D4Brz??a'''''	(( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( (s&   $ACCC)C F77F;>F;c                     t          |           5 }t          j        t          d          5  |                    d           d d d            n# 1 swxY w Y   d d d            d S # 1 swxY w Y   d S )Nz@`include` should be either 'pandas' or 'native' but is 'illegal')matchillegalrM   )r   pytestraises
ValueErrorr.   )r/   r0   s     r   'test_keys_illegal_include_keyword_valuere   A   s    	J	'	' *5]T
 
 
 	* 	* JJyJ)))		* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	** * * * * * * * * * * * * * * * * *s4   A'AA'A	A'A	A''A+.A+c                    t          |           5 }t          t          d          t          d          d          }|                    d|           |                                dgk    sJ |j                            |j        j        dd           |                                dgk    sJ 	 d d d            d S # 1 swxY w Y   d S )N   )ABr]   z/dfsymlink)r   r   r(   putr.   _handlecreate_soft_linkroot)r/   r0   r]   s      r   test_keys_ignore_hdf_softlinkro   J   s    
J	'	' 	'5U1XXE!HH5566		$zz||w&&&&&&u}'99dKKK zz||w&&&&&	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	's   BB==CC)numpyr&   rb   rR   r   r   r   r   r   pandas.tests.io.pytables.commonr   r	   mark
single_cpu
pytestmarkr2   r^   re   ro   r   r   r   <module>ru      s                            
 k$%
& & &(( ( (4* * *' ' ' ' 'r   