
    Pi:                     j   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 d dl	m
Z
mZ d dlmZmZmZmZmZmZmZ d dlmZ d dlmZ  ej        d          Zej        j        gZej                            dg d	          d
             Zd Z d Z!d Z"d Z#d Z$d Z%ej                            d e&d                    ej                            dej'        j(                  ej        )                    d          d                                     Z*ej        +                     e             d          d             Z,ej                            dg dg dg dg dg dg dg d ej-        d!d"d#gd$ej-        d!d"d#gg	          ej                            d%d&dg          d'                         Z.d( Z/d) Z0dS )*    N)is_platform_linuxis_platform_little_endianis_platform_mac)ClosedFileErrorPossibleDataLossError)	DataFrameHDFStoreIndexSeries_testing
date_rangeread_hdf)pytables)Termtablesmode)rr+awc           	         t          t          j                            d                              d          t          t          d          t                    t          ddd          	          }d
}t          j
                     d}|dv rEt          j        t          |          5  t          ||           d d d            n# 1 swxY w Y   n<t          | |          5 }|j        j        |k    sJ 	 d d d            n# 1 swxY w Y   |dv r^t          j        t          |          5  t          ||          5 }	 d d d            n# 1 swxY w Y   d d d            n# 1 swxY w Y   n<t          | |          5 }|j        j        |k    sJ 	 d d d            n# 1 swxY w Y   |dv rdt          j        t          |          5  |                    |d|           d d d            n# 1 swxY w Y   |                    | dd           n|                    | d|           |dv rJd}t          j        t$          |          5  t'          | d|           d d d            d S # 1 swxY w Y   d S t'          | d|          }|r|j                            d          |_        t-          j        ||           d S )N   
      ABCDdtype
2000-01-01r   Bperiodsfreqcolumnsindexz[\S]* does not exist.h5)r   r   matchr   dfkeyr   r   )r   zNmode w is not allowed while performing a read. Allowed modes are r, r\+ and a.str)r   nprandomdefault_rngstandard_normalr
   listobjectr   uuiduuid4pytestraisesOSErrorr	   _handler   to_hdf
ValueErrorr   r%   astypetmassert_frame_equal)temp_h5_pathr   using_infer_stringr+   msgdoesnt_existstoreresults           /home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/pandas/tests/io/pytables/test_file_handling.py	test_moderG   "   s   	
	a  0099d6ll&111r<<<
 
 
B
 "Cjll'''L {]7#... 	. 	.\----	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. l... 	.%=%-----	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. {]7#... 	 	,T222 e              	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 l... 	.%=%-----	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. {]7#... 	9 	9IIl4I888	9 	9 	9 	9 	9 	9 	9 	9 	9 	9 	9 	9 	9 	9 	9
		,Ds	3333
		,Dt	444 u}}/ 	 ]:S111 	4 	4\4d3333	4 	4 	4 	4 	4 	4 	4 	4 	4 	4 	4 	4 	4 	4 	4 	4 	4 	4 ,4888 	2**511BJ
fb)))))s   +C		CC&DD
D
0E'EE'E	E'E	E''E+.E+F$$F(+F(G33G7:G7I00I47I4c           	         t          t          j                            d                              d          t          t          d          t                    t          ddd          	          }|	                    | d
d           t          | d
          }|                                }|r|j                            d          |_        t          j        ||           d S )Nr   r   r   r   r   r   r    r!   r$   r+   r   r,   r.   )r   r/   r0   r1   r2   r
   r3   r4   r   r;   r   copyr%   r=   r>   r?   )r@   rA   r+   rE   expecteds        rF   test_default_moderK   U   s    	
	a  0099d6ll&111r<<<
 
 
B
 IIl3I///lD))FwwyyH :#+22599&(+++++    c                    t          | d          }t          t          j        dt          j                  t          dd                    |d<   d}t          j        t          |	          5  |	                    d
           d d d            n# 1 swxY w Y   |
                                 |j        rJ |	                    d
           |j        sJ t          |          dk    sJ |
                                 |j        rJ t          | d          }t          t          j        dt          j                  t          dd                    |d<   |	                    d           |j        sJ t          |          dk    sJ |j        dk    sJ |
                                 |j        rJ |	                    d           |j        sJ t          |          dk    sJ |j        dk    sJ |
                                 |j        rJ |	                    d           |j        sJ t          |          dk    sJ |j        dk    sJ |
                                 |j        rJ d S )Nr   r*   r   r   z
2020-01-01)r"   r&   zKRe-opening the file \[[\S]*\] with mode \[a\] will delete the current file!r(   r   r   r      )r	   r   r/   arangefloat64r   r7   r8   r   opencloseis_openlen_mode)r@   rD   rB   s      rF   test_reopen_handlerW   d   s   \,,,E
	"BJ'''z,PR/S/S/S  E#J
	 
 
,C	8	8	8  

3               
KKMMM} 
JJsOOO==u::????	KKMMM}\,,,E
	"BJ'''z,PR/S/S/S  E#J
 
JJsOOO==u::????;#	KKMMM} 
JJsOOO==u::????;#	KKMMM} 
JJsOOO==u::????;#	KKMMM}s   0BBBc           
         t          j                     d}t          dt          j        d                              d          z  t          t          d          t                    t          d t          d          D             t                    	          }t          |d
dd          }||d<   |                    d|           |                                }| r>|j                            d          |_        |j                            d          |_        t!          j        |d         |           t!          j        |d         |           |                                 t&          j                            |          rJ d S )Nr'   皙?x      r   r   r   c                     g | ]}d | S zi- .0is     rF   
<listcomp>z"test_open_args.<locals>.<listcomp>       111!X!XX111rL   r\   r$   r   	H5FD_COREr   )r   driverdriver_core_backing_storer+   df2r.   )r5   r6   r   r/   rP   reshaper
   r3   r4   ranger	   appendrI   r&   r=   r%   r>   r?   rS   ospathexists)rA   not_writtenr+   rD   rJ   s        rF   test_open_argsrp      sc   Z\\&&&K	binn$$W---d6ll&11111uRyy111@@@
 
 
B #kQ  E E$K	LLwwyyH :!..u55#+22599%+x000%,111	KKMMM w~~k*******rL   c                     t          | d          5 }t          t          d                    |d<   |                                 |                    d           d d d            d S # 1 swxY w Y   d S )Nr   r*      r   T)fsync)r	   r   rj   flushr@   rD   s     rF   
test_flushrv      s    	,S	)	)	)  UE!HH%%c
$                                   s   A
A))A-0A-c           
         t          dt          j        d                              d          z  t	          t          d          t                    t	          d t          d          D             t                              }|                    | d	d
           t          | d	          }|
                                }|r>|j                            d          |_        |j                            d          |_        t          j        ||           t                               | d          5 }|                    dd          D ]&}|j        j        d
k    sJ |j        j        dk    sJ '	 d d d            n# 1 swxY w Y   |                    | d	d           t          | d	          }|
                                }|r>|j                            d          |_        |j                            d          |_        t          j        ||           t                               | d          5 }|                    dd          D ]"}|j        j        dk    sJ |j        j        J #	 d d d            n# 1 swxY w Y   |                    | d	           t          | d	          }|
                                }|r>|j                            d          |_        |j                            d          |_        t          j        ||           t                               | d          5 }|                    dd          D ]"}|j        j        dk    sJ |j        j        J #	 d d d            d S # 1 swxY w Y   d S )NrY   rZ   r[   r   r   c                     g | ]}d | S r^   r_   r`   s     rF   rc   z2test_complibs_default_settings.<locals>.<listcomp>   rd   rL   r\   r$   r+   	   )r-   	complevelr.   r   r*   /dfLeafwhere	classnamezlib)r-   complibr   )r-   )r   r/   rP   ri   r
   r3   r4   rj   r;   r   rI   r&   r=   r%   r>   r?   r   	open_file
walk_nodesfiltersrz   r   )r@   rA   r+   rE   rJ   h5filenodes          rF   test_complibs_default_settingsr      s8   	binn$$W---d6ll&11111uRyy111@@@
 
 
B IIlI222lD))FwwyyH :!..u55#+22599&(+++			,S		1	1 2V%%EV%DD 	2 	2D<)Q....<'611111	22 2 2 2 2 2 2 2 2 2 2 2 2 2 2 IIlfI555lD))FwwyyH :!..u55#+22599&(+++			,S		1	1 0V%%EV%DD 	0 	0D<)Q....<'////	00 0 0 0 0 0 0 0 0 0 0 0 0 0 0 IIlI%%%lD))FwwyyH :!..u55#+22599&(+++			,S		1	1 0V%%EV%DD 	0 	0D<)Q....<'////	00 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0s6   4?F  FF8;J  JJ7;N  NNc           
         t          dt          j        d                              d          z  t	          t          d          t                    t	          d t          d          D             t                              }t          |           }|	                    d	|d
d           |	                    d|           |
                                 t                              | d          5 }|                    dd          D ]"}|j        j        dk    sJ |j        j        J #|                    dd          D ]&}|j        j        d
k    sJ |j        j        dk    sJ '	 d d d            d S # 1 swxY w Y   d S )NrY   rZ   r[   r   r   c                     g | ]}d | S r^   r_   r`   s     rF   rc   z;test_complibs_default_settings_override.<locals>.<listcomp>   rd   rL   r\   r$   dfcry   blosc)rz   r   r+   r   r*   r{   r|   r}   r   z/dfc)r   r/   rP   ri   r
   r3   r4   rj   r	   rk   rS   r   r   r   r   rz   r   )r@   r+   rD   r   r   s        rF   'test_complibs_default_settings_overrider      s   	binn$$W---d6ll&11111uRyy111@@@
 
 
B
 \""E	LLaL999	LLr	KKMMM			,S		1	1 3V%%EV%DD 	0 	0D<)Q....<'////%%Ff%EE 	3 	3D<)Q....<'722222	3	3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3s   5A9E<<F F lvlr   libz!ignore:object name is not a validc                 >   t                      rC|dk    r=|dk    r7|                    t          j                            d| d                     t          t          j        d          t          d          t          j	        d          
                    t          j                  	          }t                              d
          st          j        d           t                              d          st          j        d           | | d| dz  }| d| }|                    ||||           t!          ||          }t#          j        ||           t'                      }t                              |d          5 }	|	                    d|z   d          D ]`}
|
j        j        |k    sJ |dk    r|
j        j        J )|r#|dk    r|
j        j        }||dfv s
J |            N|
j        j        |k    sJ a	 d d d            d S # 1 swxY w Y   d S )Nblosc2r   z
Fails for z	 on Linuxreasonr[   r   r\   r$   lzozlzo not availablebzip2zbzip2 not available_r'   )r-   r   rz   r   r*   /r|   r}   zblosc2:blosclz)r   applymarkerr7   markxfailr   r/   onesr3   rP   r=   str_r   which_lib_versionskipr;   r   r>   r?   r   r   r   r   rz   r   )tmp_pathr   r   requestr+   tmpfilegnamerE   is_mach5tabler   ress               rF   test_complibsr     s   
  Ssh3!88FK--5P#5P5P5P-QQRRR	
$v,,bimm6J6J276S6S
 
 
B
 ##E** )'(((##G,, +)***C**#****GNNSNNE IIg5#I===gu%%F&"%%%F 
		'		,	, 	3&&S5[F&KK 	3 	3D<)S0000axx|+3333 3C8OOl*s$45555s5555|+s22222	3	3 	3 	3 	3 	3 	3 	3 	3 	3 	3 	3 	3 	3 	3 	3 	3 	3 	3s   A<HHHz$reason platform is not little endianr   c                    t          dddt          d                    }t          j        |j        d<   t          j        |j        d<   |                     d|d	
           t          j        | d         |           |                    dg          }| 	                    dt          dd	
                    }t          j        ||           d S )Nfoobar)Ar    rr   rN   )r   r   )   r    r+   ascii)encodingr   )r%   z	columns=A)r   rj   r/   nanlocrk   r>   r?   reindexselectr   )temp_hdfstorer+   rJ   rE   s       rF   test_encodingr   0  s     
U++588	<	<	<BVBF6NVBF6NrG444--r222zz3%z((H!!$[7(K(K(KLLF&(+++++rL   val)   E, 17rL      a   b   c)r   r   r   r   )s   EE, 17rL   r   r   r   )r      r   r   r   )rL   r   r   r   )r   r   r   r   )   ArL   r   r   r   rL   r   r   r   r   categoryc                 v   dd}d}fd|D             }t          ||          }|                    | |d|           t          | |          }|dk    r-||j        j        v r|j                            |g          }n#|}n |                    |t          j                  }t          j
        ||           d S )	Nzlatin-1 datac                 h    g | ].}t          |t                    r|                              n|/S r_   )
isinstancebytesdecode)ra   xencs     rF   rc   z'test_latin_encoding.<locals>.<listcomp>S  s6    
E
E
EAJq%007188C===a
E
E
ErL   r   table)r-   formatr   nan_repr   )r   r;   r   cat
categoriesremove_categoriesreplacer/   r   r>   assert_series_equal)	r@   r   r   r   r-   serretrs_nanr   s	           @rF   test_latin_encodingr   ?  s      CG
C
E
E
E
E
E
E
EC
E
"
"
"CJJ|WsGJTTTL#&&D 
cg(((G--wi88EEEEGRV,,5$'''''rL   c           
         t          dt          j        d                              d          z  t	          t          d          t                    t	          d t          d          D             t                              }|                    | d	d
d           t          |           }d|
                                vsJ |j        sJ |                                 d|
                                v sJ |j        rJ t          j        rit          |           }d}t          j        t"          |          5  t          |            d d d            n# 1 swxY w Y   |                                 nbt          |           }t          |           }d|
                                vsJ d|
                                vsJ |j        sJ |j        sJ |                                 d|
                                v sJ |j        rJ d|
                                vsJ |j        sJ |                                 d|
                                v sJ d|
                                v sJ |j        rJ |j        rJ t          | d
          }|                    d	|           t          |           }|                    d|           |                                 d|
                                v sJ |j        rJ |                                 d|
                                v sJ |j        rJ t          | d
          }|                    d	|           t          |           }|                                 d|
                                v sJ |j        rJ |                                 d|
                                v sJ |j        rJ t          dt          j        d                              d          z  t	          t          d          t                    t	          d t          d          D             t                              }|                    | d	d
d           t          |           }|                                 d}t          j        t&          |          5  |                                 d d d            n# 1 swxY w Y   t          j        t&          |          5  d	|v  d d d            n# 1 swxY w Y   t          j        t&          |          5  t+          |           d d d            n# 1 swxY w Y   t          j        t&          |          5  |d	          d d d            n# 1 swxY w Y   t          j        t&          |          5  |                    d	           d d d            n# 1 swxY w Y   t          j        t&          |          5  |                    d	           d d d            n# 1 swxY w Y   t          j        t&          |          5  |                    d|           d d d            n# 1 swxY w Y   t          j        t&          |          5  |                    d|           d d d            n# 1 swxY w Y   t          j        t&          |          5  |                    d           d d d            n# 1 swxY w Y   t          j        t&          |          5  |                    d           d d d            n# 1 swxY w Y   t          j        t&          |          5  |                    d	           d d d            n# 1 swxY w Y   d}t          j        t6          |          5  |j         d d d            d S # 1 swxY w Y   d S )NrY   rZ   r[   r   r   c                     g | ]}d | S r^   r_   r`   s     rF   rc   z,test_multiple_open_close.<locals>.<listcomp>l  rd   rL   r\   r$   r+   r   r   )r-   r   r   CLOSEDzTThe file [\S]* is already opened\.  Please close it before reopening in write mode\.r(   r*   rh   c                     g | ]}d | S r^   r_   r`   s     rF   rc   z,test_multiple_open_close.<locals>.<listcomp>  rd   rL   z[\S]* file is not open!df3z''HDFStore' object has no attribute 'df')r   r/   rP   ri   r
   r3   r4   rj   r;   r	   inforT   rS   r   !_table_file_open_policy_is_strictr7   r8   r<   rk   r   keysrU   r   getput
get_storerremoveAttributeErrorr+   )r@   r+   rD   store1rB   store2s         rF   test_multiple_open_closer   f  s
    
binn$$W---d6ll&11111uRyy111@@@
 
 
B
 IIl3wI??? \""E5::<<''''==	KKMMMuzz||####}1 :",'') 	 ]:S111 	# 	#\"""	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	 ,'',''v{{}},,,,v{{}},,,,~~~~6;;==((((>!!!v{{}},,,,~~6;;==((((6;;==((((>!!!>!!! C000T2,''eR   6;;==((((>!!!5::<<''''=    C000T2,''5::<<''''=   6;;==((((>!!! 
binn$$W---d6ll&11111uRyy111@@@
 
 
B
 IIl3wI???\""E	KKMMM
$C	c	2	2	2  

               
c	2	2	2                 
c	2	2	2  E


               
c	2	2	2  d               
c	2	2	2  T               
c	2	2	2  		$               
c	2	2	2    UB                              
c	2	2	2  		%               
c	2	2	2                                  
c	2	2	2  U               
c	2	2	2  T               4C	~S	1	1	1                   s   =EE E;SS #S TTT9UUU;	VVV6WWW>X  X$'X$Y))Y-0Y-Z22Z69Z6[::[>[> ]]	](^

^^2___c                     t          |           5 }t          j        |          t          |           k    sJ 	 d d d            d S # 1 swxY w Y   d S )N)r	   rl   fspathr.   ru   s     rF   test_fspathr     s    	,		 55y3|#4#4444445 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5s   (AA
A
)1rl   r5   numpyr/   r7   pandas.compatr   r   r   pandas.errorsr   r   pandasr   r	   r
   r   r   r>   r   r   	pandas.ior   pandas.io.pytablesr   importorskipr   r   
single_cpu
pytestmarkparametrizerG   rK   rW   rp   rv   r   r   rj   r   all_complibsfilterwarningsr   skipifr   r   r   r   r   r_   rL   rF   <module>r      sY   				               
       
                        # # # # # #		X	&	&k$%
 !6!6!677/* /* 87/*d, , ,3 3 3l+ + +:     10 10 10h3 3 3* b		** ;<<?@@#3 #3 A@ =< +*#3L !!###,R   	, 	, 	, 	---(((***555'''---	dD!	rvsD$/
  :t"455( ( 65 (0~ ~ ~B5 5 5 5 5rL   