
    Pi	                         d dl Zd dlZd dlmZmZmZmZmZ  ej	        d          Z
ej        j        gZd Zd Zd Zd ZdS )    N)	DataFrameHDFStoreIndexSeries
date_rangetablesc           
         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 S )N
   )dtypez
2020-01-01)periods)indexafloat64c                     g | ]}d | S )i_ .0is     v/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/pandas/tests/io/pytables/test_keys.py
<listcomp>ztest_keys.<locals>.<listcomp>   s    *G*G*G888*G*G*G    )r   r   bg?x   )      ABCDc                     g | ]}d | S )zi-r   r   s     r   r   ztest_keys.<locals>.<listcomp>   s    111!X!XX111r   r   )columnsr   c   >   /a/b/c)r   nparanger   r   ranger   reshaper   listobjectlensetkeys)temp_hdfstoreexpecteds     r   	test_keysr0      sZ   
	"BJ'''z,PR/S/S/S  M#  b		*G*GU2YY*G*G*G  M# #binn$$W---d6ll&11111uRyy111@@@  M# }""""!!!H}!!##$$0000}))))))r   c                     G d dt           j                  } G d dt           j                  } G d dt           j                  }t                               | 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                   6    e Zd Ze                                ZdS )$test_non_pandas_keys.<locals>.Table1N)__name__
__module____qualname__r   
Float32Colvalue1r   r   r   Table1r3   %           ""$$r   r9   c                   6    e Zd Ze                                ZdS )$test_non_pandas_keys.<locals>.Table2N)r4   r5   r6   r   r7   value2r   r   r   Table2r<   (   r:   r   r>   c                   6    e Zd Ze                                ZdS )$test_non_pandas_keys.<locals>.Table3N)r4   r5   r6   r   r7   value3r   r   r   Table3r@   +   r:   r   rB   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   )
temp_h5_pathr9   r>   rB   h5filerF   storer/   namedfs
             r   test_non_pandas_keysr\   $   s   % % % % %% % % %% % % % %% % % %% % % % %% % % % 
		,S		1	1 @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 F88F<?F<c                     t          j        t          d          5  |                     d           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illegalrK   )pytestraises
ValueErrorr-   )r.   s    r   'test_keys_illegal_include_keyword_valuerc   =   s    	P
 
 
 . . 	9---	. . . . . . . . . . . . . . . . . .s   A  AAc                 B   t          t          d          t          d          d          }|                     d|           |                                 dgk    sJ | j                            | j        j        dd           |                                 dgk    sJ d S )N   )ABr[   z/dfsymlink)r   r'   putr-   _handlecreate_soft_linkroot)r.   r[   s     r   test_keys_ignore_hdf_softlinkrm   E   s     
qa11	2	2BdBE7******=+@+EyRVWWW E7******r   )numpyr%   r`   rP   r   r   r   r   r   importorskipr   mark
single_cpu
pytestmarkr0   r\   rc   rm   r   r   r   <module>rs      s                      
	X	&	&k$%
* * *&( ( (2. . .+ + + + +r   