
    Pi              
          d dl Z d dlmZ d dlZd dlZd dlZd dlZd dlm	Z	m
Z
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	 Zd
 Zd Zd Zd Zej                            d eddddd           e	 e d                    g          d             Z!d Z"d Z#d Z$ej                            dddg ddg          d             Z%dS )     N)BytesIO)CategoricalIndex	DataFrameHDFStoreIndex
MultiIndex
date_rangeread_hdf)Term_maybe_adjust_namec           
      t   t          dt          j        d                              d          z  t	          t          d          t                    t	          d t          d          D             t                              }|                     d	|           d
}t          j
        t          |          5  |                     d	dg           d d d            n# 1 swxY w Y   d}t          j
        t          |          5  |                     d	dg           d d d            d S # 1 swxY w Y   d S )N皙?x         ABCDdtypec                     g | ]}d | S zi- .0is     x/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/pandas/tests/io/pytables/test_errors.py
<listcomp>z,test_pass_spec_to_storer.<locals>.<listcomp>       111!X!XX111    r   columnsindexdfzqcannot pass a column specification when reading a Fixed format store. this store must be selected in its entiretymatchA)r!   zucannot pass a where specification when reading from a Fixed format store. this store must be selected in its entiretyz	columns=Awhere)r   nparangereshaper   listobjectrangeputpytestraises	TypeErrorselecttemp_hdfstorer#   msgs      r   test_pass_spec_to_storerr7      s   	binn$$W---d6ll&11111uRyy111@@@
 
 
B dB	=  
y	,	,	, 2 2TC51112 2 2 2 2 2 2 2 2 2 2 2 2 2 2	D  
y	,	,	, 8 8T+7778 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8s$   :CC#&C#D--D14D1c                 h   t          dg di          }t          dg dit          ddd                    }|                     d	|d
           t          j        d          }t          j        t          |          5  |                     d	|d
d           d d d            d S # 1 swxY w Y   d S )Na)         )r         z1/1/2000r<   nsperiodsunit)r"   frametableformatz3incompatible kind in col [integer - datetime64[ns]]r$   T)rF   append)r   r	   r/   reescaper0   r1   r2   )r5   df1df2r6   s       r   $test_table_index_incompatible_dtypesrL   1   s    
S)))$
%
%C
	iii
:qt L L L  C gs7333
)I
J
JC	y	,	,	, E E'3wtDDDE E E E E E E E E E E E E E E E E Es    B''B+.B+c                 &   dt          j        ddd          fg}|D ]\  }}t          dt          j        d                              d          z  t          t          d          t          	          t          d
 t          d          D             t          	                    }|||<   t          j        d| d          }t          j        t          |          5  |                     d| |           d d d            n# 1 swxY w Y   d S )Ndate  r:   r;   r   r   r   r   r   c                     g | ]}d | S r   r   r   s     r   r   z;test_unimplemented_dtypes_table_columns.<locals>.<listcomp>E   s    555aa555r   r   r    [z&] is not implemented as a table columnr$   df1_)datetimerN   r   r)   r*   r+   r   r,   r-   r.   rH   rI   r0   r1   r2   rG   )r5   dtypesnfr#   r6   s         r   'test_unimplemented_dtypes_table_columnsrW   =   sV   x}T1a0012F  	1 	11")C..((111$v,,f55555599555VDDD
 
 

 1iEAEEEFF]9C000 	1 	1  R000	1 	1 	1 	1 	1 	1 	1 	1 	1 	1 	1 	1 	1 	1 	1	1 	1s   DD		D		c           
         t          dt          j        d                              d          z  t	          t          d          t                    t	          d t          d          D             t                              }d	|d
<   d|d<   t          j	        ddd          |d<   |
                                }d                    t          j        d          t          j        d          g          }t          j        t           |          5  |                     d|           d d d            d S # 1 swxY w Y   d S )Nr   r   r   r   r   c                     g | ]}d | S r   r   r   s     r   r   z<test_unimplemented_dtypes_table_columns2.<locals>.<listcomp>R   r   r   r   r    fooobj1barobj2rO   r:   r;   	datetime1|zjCannot serialize the column [datetime1]
because its data contents are not [string] but [date] object dtypez+[date] is not implemented as a table columnr$   df_unimplemented)r   r)   r*   r+   r   r,   r-   r.   rS   rN   _consolidatejoinrH   rI   r0   r1   r2   rG   r4   s      r   (test_unimplemented_dtypes_table_columns2rc   M   sz   	binn$$W---d6ll&11111uRyy111@@@
 
 
B
 BvJBvJmD!Q//B{O			B ((ID  ICDD	
 C 
y	,	,	, 5 5/4445 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5s   D88D<?D<c           
      V   t          t          j                            d                              d          t          t          d          t                    t          dddd	          
          }d|d<   d|j	        |j
        dd         df<   |                     d|d           t          j        d          }t          j        t           |          5  t#                       d d d            n# 1 swxY w Y   t          j        d          }t          j        t$          |          5  |                     dd           d d d            n# 1 swxY w Y   d}t          j        t(          |          5  |                     dd           d d d            d S # 1 swxY w Y   d S )Nr;   
   r   r   r   z
2000-01-01rf   Br?   )rA   freqrB   r    rZ   stringr\   r   r   r#   rD   rE   z:__init__() missing 1 required positional argument: 'where'r$   zWcannot process expression [df.index[3]], [2000-01-06 00:00:00] is not a valid conditionzdf.index[3]zinvalid syntaxzindex>)r   r)   randomdefault_rngstandard_normalr   r,   r-   r	   locr"   r/   rH   rI   r0   r1   r2   r   
ValueErrorr3   SyntaxErrorr4   s      r   test_invalid_termsrp   g   sW   	
	a  0099d6ll&111r$GGG
 
 
B
 BxL&+BF28AaC=("#dBw/// )P
Q
QC	y	,	,	,                 )	9 C 
z	-	-	- 2 2T=1112 2 2 2 2 2 2 2 2 2 2 2 2 2 2 C	{#	.	.	. - -T8,,,- - - - - - - - - - - - - - - - - -s6   C55C9<C9/EEE:FF"%F"c           	      >   t          t          j                            d                              d          t          d          t          ddd                    }|                    | d	d
d           t          | d	d           t          | d	d           d S )Nr;   re   r   20130101rf   r?   r@   r    dfqrD   T)keyrF   data_columnsz0index>Timestamp('20130104') & columns=['A', 'B']r'   
A>0 or C>0)	r   r)   rj   rk   rl   r,   r	   to_hdfr
   )temp_h5_pathrs   s     r   test_invalid_terms_from_docsry      s    

	a  0099VRd;;;  C
 JJ|wTJJJJ e#U    \5555555r   c           	         t          t          j                            d                              d          t          d          t          ddd                    }|                    | d	d
           d}t          j	        t          |          5  t          | d	d           d d d            d S # 1 swxY w Y   d S )Nr;   re   r   rr   rf   r?   r@   r    rs   rD   )rt   rF   a  The passed where expression: A>0 or C>0\n\s*contains an invalid variable reference\n\s*all of the variable references must be a reference to\n\s*an axis \(e.g. 'index' or 'columns'\), or a data_column\n\s*The currently defined references are: index,columns\nr$   rv   r'   )r   r)   rj   rk   rl   r,   r	   rw   r0   r1   rn   r
   )rx   rs   r6   s      r   test_invalid_terms_referencer{      s   

	a  0099VRd;;;  C
 JJ|wJ777	A  
z	-	-	- : :uL9999: : : : : : : : : : : : : : : : : :s   B55B9<B9c                    t          t          j                            d                              d                    }t          dt          j                            d                              d          i          }t          dt          j                            d                              d          i          }t          dt          j                            d                              d          i          }t          ddt
          ft          j                            d                              d          i          }d}|                     ||           ||||fD ]_}t          j        d	          }t          j
        t          |
          5  |                     ||           d d d            n# 1 swxY w Y   `d S )Nr;   )rf   r:   r9   rf   r:   r;   )1r;   r~   df_diff_valerrorz?cannot match existing table structure for [0] on appending datar$   )r   r)   rj   rk   rl   r-   rG   rH   rI   r0   r1   rn   )	r5   r#   rK   df3df4df5namedr6   s	            r   7test_append_with_diff_col_name_types_raises_value_errorr      s   	29((++;;GDD	E	EB
S")//22BB2FFG
H
HC
VRY22155EEbIIJ
K
KC
Xry44Q77GGKKL
M
MC
c1f%ry'<'<Q'?'?'O'OPR'S'ST
U
UCDr"""3S! * *iM
 
 ]:S111 	* 	*  q)))	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	*	* *s   F>>G	G	c                 Z   t          t          j                            d                              d          t	          d          t	          d                    }d}t          j        t          |          5  |                    | dd	
           d d d            d S # 1 swxY w Y   d S )Nr;   r   r=   abcdABCDEr"   r!   z)complib only supports \[.*\] compression.r$   r#   foolib)rt   complib)	r   r)   rj   rk   r,   r0   r1   rn   rw   )rx   r#   r6   s      r   test_invalid_complibr      s    	
	a  ''//6llW
 
 
B
 7C	z	-	-	- < <
		,D(	;;;< < < < < < < < < < < < < < < < < <s   :B  B$'B$idx2019Dr<   UTCr?   )rh   rA   tzrB   abcc                     t          j        | | g          }t          d|dg          }t          j        t
          d          5  |                    |d           d d d            d S # 1 swxY w Y   d S )Nr   r9   r   zSaving a MultiIndexr$   r#   rt   )r   from_arraysr   r0   r1   NotImplementedErrorrw   )r   rx   mir#   s       r   &test_to_hdf_multiindex_extension_dtyper      s     
	c
	+	+B	1B	.	.	.B	*2G	H	H	H * *
		,D	)))* * * * * * * * * * * * * * * * * *s   A**A.1A.c                      | ddd          }d}t          j        t          |          5  t          |           d d d            d S # 1 swxY w Y   d S )Niodataz"legacy_hdf/incompatible_dataset.h5z_Dataset\(s\) incompatible with Pandas data types, not table, or no datasets found in HDF5 file.r$   )r0   r1   rn   r
   )datapath	data_pathmessages      r   test_unsuppored_hdf_file_errorr      s    v'KLLI	8 
 
z	1	1	1                   s   AAAc                 t   t          t          j                            d                              d          t	          d          t	          d                    }d}t          j        t          |          5  t          t          j
                     dd	           d d d            n# 1 swxY w Y   |                    | d
           t          | d          }|                                 d}t          j        t          |          5  t          |d
           d d d            d S # 1 swxY w Y   d S )Nr;   r   r   r   r   zFile [\S]* does not existr$   z.h5rt   r#   r   r)modez&The HDFStore must be open for reading.)r   r)   rj   rk   r,   r0   r1   OSErrorr
   uuiduuid4rw   r   close)rx   r#   r6   stores       r   test_read_hdf_errorsr      s   	
	a  ''//6llW
 
 
B 'C	wc	*	*	* . .DJLL%%%u---. . . . . . . . . . . . . . . IIlI%%%\,,,E	KKMMM
2C	wc	*	*	*                   s$   :%B++B/2B/D--D14D1c                      d} t          j        t          |           5  t          t	          d          d           d d d            d S # 1 swxY w Y   d S )Nz5Support for generic buffers has not been implemented.r$   r   r#   )r0   r1   r   r
   r   )r6   s    r   #test_read_hdf_generic_buffer_errorsr      s    
AC	*#	6	6	6 % %t$$$% % % % % % % % % % % % % % % % % %s   A		AAbad_versionr}   )r:   12123c                     d}t          j        t          |          5  t          d|            d d d            d S # 1 swxY w Y   d S )Nz5Version is incorrect, expected sequence of 3 integersr$   values_block_0)version)r0   r1   rn   r   )r   r6   s     r   )test_maybe_adjust_name_bad_version_raisesr      s    
AC	z	-	-	- B B+[AAAAB B B B B B B B B B B B B B B B B Bs   =AA)&rS   r   r   rH   r   numpyr)   r0   pandasr   r   r   r   r   r	   r
   pandas.io.pytablesr   r   mark
single_cpu
pytestmarkr7   rL   rW   rc   rp   ry   r{   r   r   parametrizer,   r   r   r   r   r   r   r   r   <module>r      s>          				                              
 k$%
8 8 8,	E 	E 	E1 1 1 5 5 54- - -:6 6 6 : : :(* * *$< < < 	
6Q5tDDDe%% * * *	 	 	  (% % % r4(GHHB B IHB B Br   