
    I j:                        S SK r S SKrS SKrS SKrS SKJrJrJr  S SK	J
r
Jr  S SKJrJrJrJrJrJrJr  S SKJr  S SKJr  \R2                  " S5      r\R6                  R8                  /r\R6                  R=                  S/ S	Q5      S
 5       rS r S r!S r"S r#S r$S r%\R6                  R=                  S\&" S5      5      \R6                  R=                  S\RN                  RP                  5      \R6                  RS                  S5      S 5       5       5       r*\R6                  RW                  \" 5       (       + SS9S 5       r,\R6                  R=                  S/ SQ/ SQ/ SQ/ SQ/ SQ/ SQ/ S Q\RZ                  S!S"S#/S$\RZ                  S!S"S#//	5      \R6                  R=                  S%S&S/5      S' 5       5       r.S( r/S) r0g)*    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           
      (   [        [        R                  R                  S5      R	                  S5      [        [        S5      [        S9[        SSSS9S	9nS
n[        R                  " 5        S3nUS;   a,  [        R                  " [        US9   [        XQS9  S S S 5        O/[        XS9 nUR                  R                   U:X  d   e S S S 5        US;   a6  [        R                  " [        US9   [        XQS9 n S S S 5        S S S 5        O/[        XS9 nUR                  R                   U:X  d   e S S S 5        US;   aE  [        R                  " [        US9   UR#                  USUS9  S S S 5        UR#                  U SSS9  OUR#                  U SUS9  US;   a0  Sn[        R                  " [$        US9   ['        U SUS9  S S S 5        g ['        U SUS9nU(       a   UR(                  R+                  S5      Ul        [,        R.                  " Xs5        g ! , (       d  f       GNU= f! , (       d  f       GNg= f! , (       d  f       GNF= f! , (       d  f       GN = f! , (       d  f       GN2= f! , (       d  f       GN= f! , (       d  f       g = f)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           |/root/GenerationalWealth/GenerationalWealth/venv/lib/python3.13/site-packages/pandas/tests/io/pytables/test_file_handling.py	test_moderG   "   s	   	
		a 009d6l&1r<
B
 "Cjjl^3'L {]]7#.\- /. l.%==%%--- / {]]7#.,2e 3 /. l.%==%%--- / {]]7#.IIl4I8 /
		,Ds	3
		,Dt	4 u}/ 	 ]]:S1\4d3 21 ,48**51BJ
f)I /. /. 32 /. /.
 /. 21s`   
H+H)0
I:H;<III19J
H&)
H8;
I
	I
I
I.1
J 
Jc           
         [        [        R                  R                  S5      R	                  S5      [        [        S5      [        S9[        SSSS9S	9nUR                  U S
SS9  [        U S
5      nUR                  5       nU(       a   UR                  R                  S5      Ul        [        R                  " X45        g )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d6l&1r<
B
 IIl3I/lD)FwwyH#++2259&+    c                    [        U SS9n[        [        R                  " S[        R                  S9[        SSS9S9US'   Sn[        R                  " [        US	9   UR                  S
5        S S S 5        UR                  5         UR                  (       a   eUR                  S
5        UR                  (       d   e[        U5      S:X  d   eUR                  5         UR                  (       a   e[        U SS9n[        [        R                  " S[        R                  S9[        SSS9S9US'   UR                  S5        UR                  (       d   e[        U5      S:X  d   eUR                  S:X  d   eUR                  5         UR                  (       a   eUR                  S5        UR                  (       d   e[        U5      S:X  d   eUR                  S:X  d   eUR                  5         UR                  (       a   eUR                  S5        UR                  (       d   e[        U5      S:X  d   eUR                  S:X  d   eUR                  5         UR                  (       a   eg ! , (       d  f       GN	= f)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
		"BJJ'z,PR/SE#J
	 
 
,C	8

3 
9 
KKM}} 
JJsO===u:??	KKM}}\,E
		"BJJ'z,PR/SE#J
 
JJsO===u:??;;#	KKM}} 
JJsO===u:??;;#	KKM}} 
JJsO===u:??;;#	KKM}}}Q 
9	8s   I00
I?c                    [         R                  " 5        S3n[        S[        R                  " S5      R                  S5      -  [        [        S5      [        S9[        [        S5       Vs/ s H  nSU 3PM
     sn[        S9S	9n[        US
SSS9nX4S'   UR                  SU5        UR                  5       nU (       a@  UR                  R                  S5      Ul        UR                  R                  S5      Ul        [         R"                  " US   U5        [         R"                  " US   U5        UR%                  5         [&        R(                  R+                  U5      (       a   eg s  snf )Nr'   皙?x      r   r   r   r\   i-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_writtenir+   rD   rJ   s         rF   test_open_argsrj      s'   ZZ\N#&K	biin$$W--d6l&1uRy1y!r!Xy1@
B #kQE $K	LLwwyH!..u5#++2259%+x0%,1	KKM ww~~k*****+ 2s   -E,c                     [        U SS9 n[        [        S5      5      US'   UR                  5         UR                  SS9  S S S 5        g ! , (       d  f       g = f)Nr   r*      r   T)fsync)r	   r   rc   flushr@   rD   s     rF   
test_flushrp      sC    	,S	)UE!H%c
$ 
*	)	)s   7A
Ac                    [        S[        R                  " S5      R                  S5      -  [	        [        S5      [        S9[	        [        S5       Vs/ s H  nSU 3PM
     sn[        S9S9nUR                  U S	S
S9  [        U S	5      nUR                  5       nU(       a@  UR                  R                  S5      Ul        UR                  R                  S5      Ul        [        R                  " XE5        [         R#                  U SS9 nUR%                  SSS9 H;  nUR&                  R(                  S
:X  d   eUR&                  R*                  S:X  a  M;   e   S S S 5        UR                  U S	SS9  [        U S	5      nUR                  5       nU(       a@  UR                  R                  S5      Ul        UR                  R                  S5      Ul        [        R                  " XE5        [         R#                  U SS9 nUR%                  SSS9 H8  nUR&                  R(                  S:X  d   eUR&                  R*                  c  M8   e   S S S 5        UR                  U S	S9  [        U S	5      nUR                  5       nU(       a@  UR                  R                  S5      Ul        UR                  R                  S5      Ul        [        R                  " XE5        [         R#                  U SS9 nUR%                  SSS9 H8  nUR&                  R(                  S:X  d   eUR&                  R*                  c  M8   e   S S S 5        g s  snf ! , (       d  f       GN= f! , (       d  f       GN= f! , (       d  f       g = f)NrY   rZ   r[   r   r   r\   r]   r$   r+   	   )r-   	complevelr.   r   r*   /dfLeafwhere	classnamezlib)r-   complibr   )r-   )r   r/   rP   rb   r
   r3   r4   rc   r;   r   rI   r&   r=   r%   r>   r?   r   	open_file
walk_nodesfiltersrs   rz   )r@   rA   ri   r+   rE   rJ   h5filenodes           rF   test_complibs_default_settingsr      s   	biin$$W--d6l&1uRy1y!r!Xy1@
B IIlI2lD)FwwyH!..u5#++2259&+			,S		1V%%EV%DD<<))Q...<<''6111 E 
2 IIlfI5lD)FwwyH!..u5#++2259&+			,S		1V%%EV%DD<<))Q...<<''/// E 
2 IIlI%lD)FwwyH!..u5#++2259&+			,S		1V%%EV%DD<<))Q...<<''/// E 
2	1S 2 
2	1 
2	1 
2	1s?   MAMMAM#M#6AM5?M5
M #
M25
Nc                    [        S[        R                  " S5      R                  S5      -  [	        [        S5      [        S9[	        [        S5       Vs/ s H  nSU 3PM
     sn[        S9S9n[        U 5      nUR                  S	US
SS9  UR                  SU5        UR                  5         [        R                  U SS9 nUR                  SSS9 H8  nUR                  R                  S:X  d   eUR                  R                   c  M8   e   UR                  SSS9 H;  nUR                  R                  S
:X  d   eUR                  R                   S:X  a  M;   e   S S S 5        g s  snf ! , (       d  f       g = f)NrY   rZ   r[   r   r   r\   r]   r$   dfcrr   blosc)rs   rz   r+   r   r*   rt   ru   rv   r   z/dfc)r   r/   rP   rb   r
   r3   r4   rc   r	   rd   rS   r   r{   r|   r}   rs   rz   )r@   ri   r+   rD   r~   r   s         rF   'test_complibs_default_settings_overrider      sP   	biin$$W--d6l&1uRy1y!r!Xy1@
B
 \"E	LLaL9	LLr	KKM			,S		1V%%EV%DD<<))Q...<<''/// E %%Ff%ED<<))Q...<<''7222 F	 
2	1 2 
2	1s    E(AE-AE-E--
E;lvlr   libz!ignore:object name is not a validc                 L   [        5       (       a<  US:X  a6  US:w  a0  UR                  [        R                  R	                  SU S3S95        [        [        R                  " S5      [        S5      [        R                  " S5      R                  [        R                  5      S	9n[        R                  S
5      (       d  [        R                  " S5        [        R                  S5      (       d  [        R                  " S5        X SU S3-  nU SU 3nUR                  XVX!S9  [!        XV5      n["        R$                  " Xt5        ['        5       n[        R)                  USS9 n	U	R+                  SU-   SS9 H  n
U
R,                  R.                  U:X  d   eUS:X  a  U
R,                  R0                  b   eM@  U(       a,  US:X  a&  U
R,                  R0                  nXS4;   d   U5       eMs  U
R,                  R0                  U:X  a  M   e   S S S 5        g ! , (       d  f       g = f)Nblosc2r   z
Fails for z	 on Linuxreasonr[   r   r\   r$   lzozlzo not availablebzip2zbzip2 not available_r'   )r-   rz   rs   r   r*   /ru   rv   zblosc2:blosclz)r   applymarkerr7   markxfailr   r/   onesr3   rP   r=   str_r   which_lib_versionskipr;   r   r>   r?   r   r{   r|   r}   rs   rz   )tmp_pathr   r   requestr+   tmpfilegnamerE   is_mach5tabler   ress               rF   test_complibsr     s   
 sh3!8FKK--z#i5P-QR	
$v,biim6J6J2776S
B
 ##E**'(##G,,)*5#c**Ge1SENE IIg#I=g%F&%F 
		'		,&&S5[F&KD<<))S000ax||++333C8Oll**$455:s:5||++s222 L 
-	,	,s   %BHH
H#z$reason platform is not little endianr   c                 z   [        SSS.[        S5      S9n[        R                  UR                  S'   [        R                  UR                  S'   U R                  SUS	S
9  [        R                  " U S   U5        UR                  S/S9nU R                  S[        SS	S
95      n[        R                  " X25        g )Nfoobar)Ar    rl   rN   )r   r   )   r    r+   ascii)encodingr   )r%   z	columns=A)r   rc   r/   nanlocrd   r>   r?   reindexselectr   )temp_hdfstorer+   rJ   rE   s       rF   test_encodingr   0  s     
U+58	<BVVBFF6NVVBFF6NrG4--r2zz3%z(H!!$[7(K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                    SnSnSnU Vs/ s H+  n[        U[        5      (       a  UR                  U5      OUPM-     nn[        X!S9nUR	                  XSX4S9  [        X5      nUS:X  a9  XGR                  R                  ;   a  UR                  R                  U/5      n	O#Un	O UR                  U[        R                  5      n	[        R                  " X5        g s  snf )Nzlatin-1 datar   table)r-   formatr   nan_repr   )
isinstancebytesdecoder   r;   r   cat
categoriesremove_categoriesreplacer/   r   r>   assert_series_equal)
r@   r   r   encr   r-   xserretrs_nans
             rF   test_latin_encodingr   ?  s      CG
CAD
EAJq%00188C=a7C
E

"CJJ|WsJTL&D 
gg(((GG--wi8EEGRVV,5'! Fs   2Cc                 ,   [        S[        R                  " S5      R                  S5      -  [	        [        S5      [        S9[	        [        S5       Vs/ s H  nSU 3PM
     sn[        S9S9nUR                  U S	S
SS9  [        U 5      nSUR                  5       ;  d   eUR                  (       d   eUR                  5         SUR                  5       ;   d   eUR                  (       a   e[        R                  (       aL  [        U 5      nSn[        R                   " ["        US9   [        U 5        S S S 5        UR                  5         GOp[        U 5      n[        U 5      nSUR                  5       ;  d   eSUR                  5       ;  d   eUR                  (       d   eUR                  (       d   eUR                  5         SUR                  5       ;   d   eUR                  (       a   eSUR                  5       ;  d   eUR                  (       d   eUR                  5         SUR                  5       ;   d   eSUR                  5       ;   d   eUR                  (       a   eUR                  (       a   e[        U S
S9nUR%                  S	U5        [        U 5      nUR%                  SU5        UR                  5         SUR                  5       ;   d   eUR                  (       a   eUR                  5         SUR                  5       ;   d   eUR                  (       a   e[        U S
S9nUR%                  S	U5        [        U 5      nUR                  5         SUR                  5       ;   d   eUR                  (       a   eUR                  5         SUR                  5       ;   d   eUR                  (       a   e[        S[        R                  " S5      R                  S5      -  [	        [        S5      [        S9[	        [        S5       Vs/ s H  nSU 3PM
     sn[        S9S9nUR                  U S	S
SS9  [        U 5      nUR                  5         Sn[        R                   " [&        US9   UR)                  5         S S S 5        [        R                   " [&        US9   S	U;     S S S 5        [        R                   " [&        US9   [+        U5        S S S 5        [        R                   " [&        US9   US	     S S S 5        [        R                   " [&        US9   UR-                  S	5        S S S 5        [        R                   " [&        US9   UR/                  S	5        S S S 5        [        R                   " [&        US9   UR%                  SU5        S S S 5        [        R                   " [&        US9   UR1                  SU5        S S S 5        [        R                   " [&        US9   UR3                  S5        S S S 5        [        R                   " [&        US9   UR5                  S5        S S S 5        [        R                   " [&        US9   UR-                  S	5        S S S 5        Sn[        R                   " [6        US9   UR8                    S S S 5        g s  snf ! , (       d  f       GNy= fs  snf ! , (       d  f       GN>= f! , (       d  f       GN)= f! , (       d  f       GN= f! , (       d  f       GN= f! , (       d  f       GN= f! , (       d  f       GN= f! , (       d  f       GN= f! , (       d  f       GNs= f! , (       d  f       GNR= f! , (       d  f       GN1= f! , (       d  f       GN= f! , (       d  f       g = f)NrY   rZ   r[   r   r   r\   r]   r$   r+   r   r   )r-   r   r   CLOSEDzTThe file [\S]* is already opened\.  Please close it before reopening in write mode\.r(   r*   ra   z[\S]* file is not open!df3z''HDFStore' object has no attribute 'df')r   r/   rP   rb   r
   r3   r4   rc   r;   r	   inforT   rS   r   !_table_file_open_policy_is_strictr7   r8   r<   rd   r   keysrU   r   getput
get_storerremoveAttributeErrorr+   )r@   ri   r+   rD   store1rB   store2s          rF   test_multiple_open_closer   f  s~    
biin$$W--d6l&1uRy1y!r!Xy1@
B
 IIl3wI? \"E5::<'''===	KKMuzz|###}}11,') 	 ]]:S1\" 2 	 ,','v{{},,,v{{},,,~~~~~~6;;=(((>>!!v{{},,,~~~6;;=(((6;;=(((>>!!>>!! C0T2,'eR 6;;=(((>>!!5::<'''==   C0T2,'5::<'''==  6;;=(((>>!! 
biin$$W--d6l&1uRy1y!r!Xy1@
B
 IIl3wI?\"E	KKM
$C	c	2

 
3 
c	2 
3 
c	2E
 
3 
c	2d 
3 
c	2T 
3 
c	2		$ 
3 
c	2UB 
3 
c	2		% 
3 
c	2 
3 
c	2U 
3 
c	2T 
3 4C	~S	1 
2	1o 2( 21r 2 
3	2 
3	2 
3	2 
3	2 
3	2 
3	2 
3	2 
3	2 
3	2 
3	2 
3	2 
2	1s   Z#*Z(Z:7Z?)[[#=[5$\\
\+>\=2]%]!]3^(
Z7?
[
[ #
[25
\
\
\(+
\:=
]
]!
]03
^
^c                     [        U 5       n[        R                  " U5      [        U 5      :X  d   e S S S 5        g ! , (       d  f       g = f)N)r	   re   fspathr.   ro   s     rF   test_fspathr     s4    	,	5yy3|#4444 
 		s	   &<
A
)1re   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   rj   rp   r   r   rc   r}   all_complibsfilterwarningsr   skipifr   r   r   r   r    rL   rF   <module>r      s   	    

    #			X	&kk$$%
 !67/* 8/*d,3l+: 10h3* b	* ; ;<?@#3 A = +#3L !##,R  	,	, 	-(*5'-	dD!	rvvsD$/
 :t"45( 6(0~B5rL   