Ë
    ž…j'  ã                   ó†  — d dl mZ d dlZd dlZd dlZd dlm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 ej(                  j*                  gZd„ Zd„ Zd„ Zd	„ Zd
„ Zej(                  j9                   e«       d¬«      d„ «       Zd„ Zd„ Zd„ Z d„ Z!d„ Z"d„ Z#d„ Z$ej(                  jK                  dddg«      d„ «       Z&d„ Z'd„ Z(y)é    )ÚclosingN)Úis_platform_windows)Ú	DataFrameÚHDFStoreÚIndexÚSeriesÚ_testingÚ
date_rangeÚread_hdf)ÚTableIteratorc                 ó  — t        t        d«      t        d«      dœ«      }|j                  | d¬«       t        j                  t
        d¬«      5  t        | d«       d d d «       |j                  | d¬«       y # 1 sw Y   ŒxY w)Né   ©ÚaÚbÚk1©Úkeyú 'No object named k2 in the file'©ÚmatchÚk2)r   ÚrangeÚto_hdfÚpytestÚraisesÚKeyErrorr   ©Útemp_h5_pathÚdfs     úc/root/aria/tools/markitdown-venv/lib/python3.12/site-packages/pandas/tests/io/pytables/test_read.pyÚ!test_read_missing_key_close_storer"      si   € ä	œ˜q›¬¨a«Ñ1Ó	2€BØ‡I�Iˆl €IÔ%ä	�‰”xÐ'IÔ	Jñ %Ü�˜tÔ$÷%ð
 ‡I�Iˆl €IÕ%÷%ð %ús   ÁA8Á8Bc                 óì   — t        g g dœg ¬«      }|j                  | d¬«       t        j                  t        d¬«      5  t        | dd¬«       d d d «       |j                  | d¬«       y # 1 sw Y   ŒxY w)	N©ÚAÚB)Úindexr   r   zlist index out of ranger   r   ©Ústop)r   r   r   r   Ú
IndexErrorr   r   s     r!   Ú!test_read_index_error_close_storer+   &   se   € ä	˜ "Ñ%¨RÔ	0€BØ‡I�Iˆl €IÔ%ä	�‰”zÐ)CÔ	Dñ -Ü�˜t¨!Õ,÷-ð
 ‡I�Iˆl €IÕ%÷-ð -ús   ¿A*Á*A3c                 ó<  — t        t        d«      t        d«      dœ«      }|j                  | d¬«       t        | d«      5 }t	        j
                  t        d¬«      5  t        |d«       d d d «       t        |d«       d d d «       y # 1 sw Y   ŒxY w# 1 sw Y   y xY w)	Nr   r   r   r   Úrr   r   r   )r   r   r   r   r   r   r   r   )r   r    Ústores      r!   Ú"test_read_missing_key_opened_storer/   3   s‰   € ä	œ˜q›¬¨a«Ñ1Ó	2€BØ‡I�Iˆl €IÔ%ä	�, Ó	$ð ¨Ü�]‰]œ8Ð+MÔNñ 	"Ü�U˜DÔ!÷	"ô
 	�˜Ô÷ð ÷	"ð 	"ú÷ð ús$   Á BÁBÁ)BÂB	ÂBÂBc           	      óX
  — t        t        j                  j                  d«      j	                  d«      t        t        d«      «      t        ddd¬«      ¬«      }t        j                  t        d	¬
«      5  | j                  dd«       d d d «       | j                  d|«       t        j                  t        t        j                  d«      ¬
«      5  | j                  dd«       d d d «       t        j                  d«      }t        j                  t        |¬
«      5  | j                  dddg¬«       d d d «       | j                  dd«      }t!        j"                  |j$                  t'        |j(                  «      j$                  «       t+        |t&        «      sJ ‚t        j                  d«      }t        j                  t,        |¬
«      5  | j                  dd«       d d d «       |j/                  «       }d|d<   | j                  d|dg¬«       | j                  dd«      }t!        j"                  |j$                  |d   j$                  «       |j/                  «       }d|d<   t        j0                  |j2                  |j(                  dd df<   | j                  d|dg¬«       | j                  dd«      }t!        j"                  |j$                  |d   j$                  «       | j                  ddd¬«      }t!        j"                  |j$                  |d   j$                  dd  «       | j                  ddd¬«      }t!        j"                  |j$                  |d   j$                  dd  «       | j                  ddd¬«      }t!        j"                  |j$                  |d   j$                  d d «       | j                  ddd¬«      }t!        j"                  |j$                  |d   j$                  d d «       | j                  dddd¬«      }t!        j"                  |j$                  |d   j$                  dd «       | j                  dddd¬«      }t!        j"                  |j$                  |d   j$                  dd «       t        t        j                  j                  d«      j	                  d«      ddœ«      }| j                  d|d ¬«       |d   }| j                  dd«      }t!        j4                  ||«       y # 1 sw Y   �ŒsxY w# 1 sw Y   �Œ%xY w# 1 sw Y   �ŒäxY w# 1 sw Y   �ŒExY w)!Nr   )é
   é   ÚABCDz
2000-01-01r1   r&   )ÚperiodsÚfreq)Úcolumnsr'   zNo object named df in the filer   r    r'   z%'column [foo] not found in the table'Úfooz:select_column() got an unexpected keyword argument 'where'zindex>5)ÚwherezScolumn [values_block_0] can not be extracted individually; it is not data indexableÚvalues_block_0ÚstringÚdf2)Údata_columnsr2   é   Údf3)Ústartéþÿÿÿr(   )r?   r)   r$   Údf4T)r   ÚnpÚrandomÚdefault_rngÚstandard_normalr   Úlistr
   r   r   r   Úselect_columnÚappendÚreÚescapeÚ	TypeErrorÚtmÚassert_almost_equalÚvaluesr   r'   Ú
isinstanceÚ
ValueErrorÚcopyÚnanÚlocÚassert_series_equal)Útemp_hdfstorer    ÚmsgÚresultr;   r>   rA   Úexpecteds           r!   Útest_read_columnrY   A   sa  € Ü	Ü
�	‰	×Ñ˜aÓ ×0Ñ0°Ó9Ü”d˜6“lÓ#Ü˜¨r¸Ô<ô
€Bô 
�‰”xÐ'GÔ	Hñ 3Ø×#Ñ# D¨'Ô2÷3ð ×Ñ˜˜rÔ"ä	�‰ÜœŸ	™	Ð"IÓJô
ñ 1ð 	×#Ñ# D¨%Ô0÷1ô
 �)‰)ÐPÓ
Q€CÜ	�‰”y¨Ô	,ñ FØ×#Ñ# D¨'¸)¸Ð#ÔE÷Fð ×(Ñ(¨¨wÓ7€FÜ×Ñ˜6Ÿ=™=¬&°·±Ó*:×*AÑ*AÔBÜ�fœfÔ%Ð%Ð%ô �)‰)ð	#ó€Cô 
�‰”z¨Ô	-ñ <Ø×#Ñ# DÐ*:Ô;÷<ð �'‰'‹)€CØ€Cˆ�MØ×Ñ˜ °8°*ÐÔ=Ø×(Ñ(¨°Ó9€FÜ×Ñ˜6Ÿ=™=¨#¨h©-×*>Ñ*>Ô?ð �'‰'‹)€CØ€Cˆ�MÜ(*¯©€C‡G�GˆC�I‰I�a˜ˆN˜HÐ$Ñ%Ø×Ñ˜ °8°*ÐÔ=Ø×(Ñ(¨°Ó9€FÜ×Ñ˜6Ÿ=™=¨#¨h©-×*>Ñ*>Ô?ð ×(Ñ(¨°ÀÐ(ÓB€FÜ×Ñ˜6Ÿ=™=¨#¨h©-×*>Ñ*>¸q¸rÐ*BÔCà×(Ñ(¨°ÀÐ(ÓC€FÜ×Ñ˜6Ÿ=™=¨#¨h©-×*>Ñ*>¸r¸sÐ*CÔDà×(Ñ(¨°¸qÐ(ÓA€FÜ×Ñ˜6Ÿ=™=¨#¨h©-×*>Ñ*>¸rÀÐ*BÔCà×(Ñ(¨°¸rÐ(ÓB€FÜ×Ñ˜6Ÿ=™=¨#¨h©-×*>Ñ*>¸sÀÐ*CÔDà×(Ñ(¨°ÀÈÐ(ÓK€FÜ×Ñ˜6Ÿ=™=¨#¨h©-×*>Ñ*>¸qÀÐ*DÔEà×(Ñ(¨°ÀÈÐ(ÓK€FÜ×Ñ˜6Ÿ=™=¨#¨h©-×*>Ñ*>¸rÀ!Ð*DÔEô œ"Ÿ)™)×/Ñ/°Ó2×BÑBÀ2ÓFÈUÑSÓ
T€CØ×Ñ˜ °$ÐÔ7Ø�3‰x€HØ×(Ñ(¨°Ó4€FÜ×Ñ˜6 8Õ,÷E3ñ 3ú÷
1ñ 1ú÷Fñ Fú÷<ñ <ús0   Á4S8ÃTÄTÇ	TÓ8TÔTÔTÔT)c                 ó†   — t         | ddd«      d¬«      5 }|d   }d d d «       t        t        «      sJ ‚y # 1 sw Y   ŒxY w)NÚioÚdatazlegacy_hdf/pytables_native.h5r-   ©Úmodezdetector/readout)r   rO   r   )Údatapathr.   Úd2s      r!   Útest_pytables_native_readra   �   sN   € Ü	Ù��vÐ>Ó?Àcô
ð 'à	ØÐ%Ñ&ˆ÷'ô �bœ)Ô$Ð$Ñ$÷	'ð 'ús	   –7·A z#native2 read fails oddly on windows)Úreasonc                 óž   — t         | dddd«      d¬«      5 }t        |«       |d   }d d d «       t        t        «      sJ ‚y # 1 sw Y   ŒxY w)Nr[   r\   Ú
legacy_hdfzpytables_native2.h5r-   r]   Údetector)r   ÚstrrO   r   )r_   r.   Úd1s      r!   Útest_pytables_native2_readrh   ˜   sV   € ä	Ù��v˜|Ð-BÓCÈ#ô
ð à	ÜˆEŒ
Ø�:Ñˆ÷	ô
 �bœ)Ô$Ð$Ñ$÷ð ús   —AÁAc                 óÊ  — t        t        j                  j                  d«      j                  d«      t	        d«      t	        d«      ¬«      }d|j
                  _        |j                  dd¬	«      }|j                  | d
d¬«       t        | d
«      }t        | d¬«      5 }t        |d
«      }t        j                  ||«       |j                  sJ ‚	 d d d «       y # 1 sw Y   y xY w)Nr   ©r2   é   ÚabcdÚABCDE©r'   r6   ÚlettersÚET©ÚkeysrH   r    Úw©r   r^   r-   r]   )r   rB   rC   rD   rF   r'   ÚnameÚ	set_indexr   r   r   rL   Úassert_frame_equalÚis_open)r   Úusing_infer_stringr    Údirectr.   Úindirects         r!   Útest_read_hdf_open_storer|   ¢   s¿   € ô 
Ü
�	‰	×Ñ˜aÓ ×'Ñ'¨Ó/Ü�6‹lÜ�W“ô
€Bð
 €B‡H�H„MØ	�‰˜3 tˆÓ	,€Bà‡I�Iˆl ¨3€IÔ/Ü�l DÓ)€FÜ	�, SÔ	)ð ¨UÜ˜E 4Ó(ˆÜ
×Ñ˜f hÔ/Ø�}Š}Ð‰}÷÷ ñ ús   Â1CÃC"c                 ó@  — t        t        j                  j                  d«      j                  d«      g d¢t	        d«      ¬«      }|j                  | ddd¬	«       t        | d«      }|j                  j                  j                  �J ‚t        j                  ||«       y )
Nr   rj   )r   é   r   é   rm   rn   r    rs   Útable©r   r^   Úformat)r   rB   rC   rD   rF   r   r   r'   Ú_dataÚbaserL   rw   )r   r    r;   s      r!   Útest_read_hdf_index_not_viewr…   µ   s�   € ô 
Ü
�	‰	×Ñ˜aÓ ×'Ñ'¨Ó/ÚÜ�W“ô
€Bð ‡I�Iˆl ¨3°w€IÔ?ä
�< Ó
&€CØ�9‰9�?‰?×ÑÐ'Ð'Ð'Ü×Ñ˜"˜cÕ"ó    c                 ó  — t        t        j                  j                  d«      j                  d«      t	        d«      t	        d«      ¬«      }d|j
                  _        |j                  dd¬	«      }|j                  | d
dd¬«       t        | d
«      }t        | d
d¬«      }t        |j                  «      5  t        |t        «      sJ ‚t        |j                  «       «      }d d d «       t!        j"                  |«       y # 1 sw Y   Œ xY w)Nr   rj   rl   rm   rn   ro   rp   Trq   r    rs   Útr�   )Úiterator)r   rB   rC   rD   rF   r'   ru   rv   r   r   r   r.   rO   r   ÚnextÚ__iter__rL   rw   )r   r    rz   r‰   r{   s        r!   Útest_read_hdf_iteratorrŒ   Æ   sÚ   € Ü	Ü
�	‰	×Ñ˜aÓ ×'Ñ'¨Ó/Ü�6‹lÜ�W“ô
€Bð
 €B‡H�H„MØ	�‰˜3 tˆÓ	,€Bà‡I�Iˆl ¨3°s€IÔ;Ü�l DÓ)€FÜ˜ d°TÔ:€HÜ	�—‘Ó	 ñ -Ü˜(¤MÔ2Ð2Ð2Ü˜×)Ñ)Ó+Ó,ˆ÷-ô ×Ñ˜& (Õ+÷-ð -ús   Â5,D Ä D	c                 ó¨  — t        t        j                  j                  d«      j                  d«      t	        d«      t	        d«      ¬«      }|j                  | dd¬«       t        | «      }t        j                  ||«       |j                  | d	d¬«       d
}t        j                  t        |¬«      5  t        | «       d d d «       y # 1 sw Y   y xY w)Nr   rj   rl   rm   rn   r    r   rt   r;   ú?key must be provided when HDF5 file contains multiple datasets.r   )r   rB   rC   rD   rF   r   r   rL   rw   r   r   rP   ©r   r    ÚrereadrV   s       r!   Útest_read_nokeyr‘   Ø   s§   € ä	Ü
�	‰	×Ñ˜aÓ ×'Ñ'¨Ó/Ü�6‹lÜ�W“ô
€Bð ‡I�Iˆl ¨3€IÔ/Ü�lÓ#€FÜ×Ñ˜"˜fÔ%Ø‡I�Iˆl ¨C€IÔ0à
K€CÜ	�‰”z¨Ô	-ñ Ü�Ô÷÷ ñ ús   Â3CÃCc                 ój  — t        t        d«      t        t        d«      d¬«      dœ«      }|j	                  | ddd¬	«       t        | «      }t        j                  ||«       |j	                  | d
dd¬	«       d}t        j                  t        |¬«      5  t        | «       d d d «       y # 1 sw Y   y xY w)Nrk   ÚabacdÚcategory©Údtype)ÚiÚcr    r   r€   r�   r;   rŽ   r   )r   r   r   rF   r   r   rL   rw   r   r   rP   r�   s       r!   Útest_read_nokey_tabler™   ì   s’   € ä	œ˜q›¬¬t°G«}ÀJÔ(OÑPÓ	Q€Bà‡I�Iˆl ¨3°w€IÔ?Ü�lÓ#€FÜ×Ñ˜"˜fÔ%Ø‡I�Iˆl ¨C¸€IÔ@à
K€CÜ	�‰”z¨Ô	-ñ Ü�Ô÷÷ ñ ús   ÂB)Â)B2c                 óÚ   — t        | «      }|j                  «        t        j                  d«      }t	        j
                  t        |¬«      5  t        | «       d d d «       y # 1 sw Y   y xY w)Nz]Dataset(s) incompatible with Pandas data types, not table, or no datasets found in HDF5 file.r   )r   ÚcloserI   rJ   r   r   rP   r   )r   r.   rV   s      r!   Útest_read_nokey_emptyrœ   ú   sU   € Ü�\Ó"€EØ	‡K�K„MÜ
�)‰)ð	'ó€Cô 
�‰”z¨Ô	-ñ Ü�Ô÷÷ ñ ús   ÁA!Á!A*c                 ó
  — t        t        j                  j                  d«      j                  d«      t	        d«      t	        d«      ¬«      }|j                  | dd¬«       t        | d¬	«      }t        j                  ||«       y )
Nr   rj   rl   rm   rn   r    r   rt   r   )	r   rB   rC   rD   rF   r   r   rL   rw   )r   rX   Úactuals      r!   Útest_read_from_pathlib_pathrŸ     sg   € äÜ
�	‰	×Ñ˜aÓ ×'Ñ'¨Ó/Ü�6‹lÜ�W“ô€Hð ‡O�O�L d°€OÔ5Ü�l¨Ô-€Fä×Ñ˜( FÕ+r†   r‚   Úfixedr€   c                 ó¼   — t        t        d«      t        j                  ¬«      }|j	                  | d|¬«       t        | dd¬«      }t        j                  ||«       y )Nr1   r•   r\   ©r   r‚   r-   rt   )r   r   rB   Úfloat64r   r   rL   rT   )r   r‚   ÚseriesrW   s       r!   Útest_read_hdf_series_mode_rr¥     sH   € ô
 ”E˜"“I¤R§Z¡ZÔ0€FØ
‡M�M�, F°6€MÔ:Ü�l¨°SÔ9€FÜ×Ñ˜6 6Õ*r†   c           
      ó¬  — t        dg d¢i«      }|j                  | dd¬«       t        j                  dd«      5  t	        | dd¬	«      }d d d «       t        dg d¢it        j
                  t        j                  ¬
«      t        dgt        j
                  t        j                  ¬
«      ¬«      ¬«      }t        j                  |«       y # 1 sw Y   Œ‚xY w)Nr   )r   r   Nr\   r€   r¢   zfuture.infer_stringTr-   rt   )Úna_valuer•   )r–   r6   )r   r   ÚpdÚoption_contextr   ÚStringDtyperB   rR   r   rL   rw   )r   r    rW   rX   s       r!   Útest_read_infer_stringr«     s¦   € ä	�CÒ)Ð*Ó	+€BØ‡I�Iˆl ¨w€IÔ7Ü	×	Ñ	Ð0°$Ó	7ñ >Ü˜,¨F¸Ô=ˆ÷>äØ	ÒÐÜ�n‰n¤b§f¡fÔ-Ü�s�e¤2§>¡>¼2¿6¹6Ô#BÔCô€Hô
 ×Ñ˜& (Õ+÷>ð >ús   ºC
Ã
Cc                 ó”   — t        ddgd¬«      }| j                  d|«       | j                  d«      }t        j                  ||«       y )Nz
2001-01-01z
2002-02-02zdatetime64[s]r•   Údf_s)r   ÚputÚgetrL   rw   )rU   r­   Údf_fromstores      r!   Ú$test_hdfstore_read_datetime64_unit_sr±   ,  sC   € ä�l LÐ1¸ÔI€DØ×Ñ�f˜dÔ#Ø ×$Ñ$ VÓ,€LÜ×Ñ˜$ Õ-r†   ))Ú
contextlibr   rI   ÚnumpyrB   r   Úpandas.compatr   Úpandasr¨   r   r   r   r   r	   rL   r
   r   Úpandas.io.pytablesr   ÚmarkÚ
single_cpuÚ
pytestmarkr"   r+   r/   rY   ra   Úskipifrh   r|   r…   rŒ   r‘   r™   rœ   rŸ   Úparametrizer¥   r«   r±   © r†   r!   ú<module>r½      sÝ   ðÝ Û 	ã Û å -ã ÷÷ ñ õ -à�k‰k×$Ñ$Ð%€
ò
&ò
&òòL-ò^%ð ‡�×ÑÑ'Ó)Ð2WÐÓXñ%ó Yð%òò&#ò",ò$ò(òò,ð ‡�×Ñ˜ G¨WÐ#5Ó6ñ+ó 7ð+ò,ó.r†   