eh3rddlmZddlmZddlZddlZddlZddlm Z ddl m Z ddl Z ddl mZmZmZmZmZmZmZddlmZmZddlmZdd lmZej<j>Z d Z!d Z"d Z#d Z$dZ%ej<jMe ddZ'dZ(dZ)dZ*dZ+dZ,dZ-dZ.dZ/dZ0dZ1ejdddZ3ej<jiddd gd!Z5ej<jmd"ej<jmd#d$Z7d%Z8y)&)closing)PathN) Timestamp)is_platform_windows) DataFrameHDFStoreIndexSeries_testing date_rangeread_hdf) _maybe_removeensure_clean_store)_test_decorators) TableIteratorc||z }ttdtdd}|j|dtjt d5t |dddd|j|dy#1swYxYw)Nabk1key 'No object named k2 in the file'matchk2)rrangeto_hdfpytestraisesKeyErrorr tmp_path setup_pathpathdfs `/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/pandas/tests/io/pytables/test_read.py!test_read_missing_key_close_storer) ss j D qa1 2BIIdI x'I Jt IIdI s  A==Bc||z }tggdg}|j|dtjtd5t |ddddd|j|dy#1swYxYw) NABindexrrzlist index out of rangerrstop)rrr r! IndexErrorr r#s r(!test_read_index_error_close_storer3.so j D "%R 0BIIdI z)C D%t!$% IIdI %%s A//A8cF||z }ttdtdd}|j|dt|d5}t j t d5t|ddddt|ddddy#1swYxYw#1swYyxYw) Nrrrrrrrr)rrrrr r!r"r )r$r%r&r'stores r("test_read_missing_key_opened_storer7<s j D qa1 2BIIdI $  ]]8+M N " UD ! "   " "s$B! B .B B BB c  ttjjdj dt t dttddd }t|5}t|d tjtd 5|jd d ddd|jd |tjtt!j"d 5|jd ddddt!j"d}tjt$| 5|jd d dgddd|jd d }t'j(|j*t-|j.j*t1|t,sJt!j"d}tjt2| 5|jd dddd|j5}d|d<|jd|dg|jdd}t'j(|j*|dj*|j5}d|d<tj6|j8|j.dddf<|jd|dg|jdd}t'j(|j*|dj*|jddd}t'j(|j*|dj*dd|jddd}t'j(|j*|dj*dd|jddd}t'j(|j*|dj*dd|jddd}t'j(|j*|dj*dd|jdddd}t'j(|j*|dj*dd|jdddd}t'j(|j*|dj*ddttjjdj ddd}|jd |d!|d}|jd d}t'j:||dddy#1swY{xYw#1swY-xYw#1swYxYw#1swYMxYw#1swYyxYw)"Nr) ABCDdtypez 2000-01-01r9r-)periodsfreqcolumnsr/r'zNo object named df in the filerr/z%'column [foo] not found in the table'fooz:select_column() got an unexpected keyword argument 'where'zindex>5)wherezScolumn [values_block_0] can not be extracted individually; it is not data indexablevalues_block_0stringdf2) data_columnsr:df3)startr0)rJr1r+df4T)rnprandom default_rngstandard_normalr listobjectr rrr r!r" select_columnappendreescape TypeErrortmassert_almost_equalvaluesr r/ isinstance ValueErrorcopynanlocassert_series_equal) r%r'r6msgresultrFrIrLexpecteds r(test_read_columnrdKs  a 009d6l&1r< B J 'H15eT" ]]8+K L /   g . /  T2 ]] BII&MN  -   e , - iiTU ]]9C 0 B   gi[  A B$$T73 v}}fRXX.>.E.EF&&)))ii ' ]]:S 1 8   &6 7 8ggiH  UCxj 9$$UH5 v}}c(m.B.BCggiH ,.FF !A() UCxj 9$$UH5 v}}c(m.B.BC$$UHA$> v}}c(m.B.B12.FG$$UHB$? v}}c(m.B.B23.GH$$UH1$= v}}c(m.B.B2A.FG$$UH2$> v}}c(m.B.B3B.GH$$UHAB$G v}}c(m.B.B1R.HI$$UHBQ$G v}}c(m.B.B2a.HIbii33A6FFrJQVWX UCd 3s8$$UC0 vx0QH1H1  / /  - -  B B 8 8=H1H1so*(UT%AU-T+8U8T8BU'U:LUT( #U+T5 0U8U =UU UUct|dddd5}|d}dddttsJy#1swYxYw)Niodatazlegacy_hdf/pytables_native.h5r5modezdetector/readout)rr[r)datapathr6d2s r(test_pytables_native_readrlsN v>?c '  % &' b) $$ $ ''s 7Az#native2 read fails oddly on windows)reasonct|ddddd5}t||d}dddttsJy#1swYxYw)Nrfrg legacy_hdfzpytables_native2.h5r5rhdetector)rstrr[r)rjr6d1s r(test_pytables_native2_readrssV v|-BC#   E :   b) $$ $ s AA ct|ddddd5}|jd}dddtgdggd td gd  }t j |y#1swY?xYw)Nrfrgrozlegacy_table_fixed_py2.h5r5rhr')rD)r,r-CrwABC INDEX_NAME)namer@)rselectrr rXassert_frame_equalrjr6rbrcs r('test_legacy_table_fixed_format_read_py2rsw v|-HIPS $ d#$ $UG,/H (F+$$s A''A0cttdggdgtdgd}t|dddd d 5}|j d }dddt j |y#1swY xYw) Nz2020-02-06T18:00r,datezM8[ns])rAr/r=rfrgroz"legacy_table_fixed_datetime_py2.h5r5rhr')rrr rr|rXr}rjrcr6rbs r(0test_legacy_table_fixed_format_read_datetime_py2rs & ' ()VHo H v|-QR  $ d# $ (F+ $$s A--A6ct|ddddd5}|jd}dddtdd gd d gd }tj|y#1swY2xYw) Nrfrgrozlegacy_table_py2.h5r5rhtablerrrrvr)rr|rrXr}r~s r(test_legacy_table_read_py2rsl v|-BC# ' g&' SzA78H(F+ ''s AA#cttjjdjdt dt d}d|j _|jdd }||z }|j|d d t|d }t|d 5}t|d }tj|||jsJ dddy#1swYyxYw)Nrr:abcdABCDEr/rAlettersETkeysrTr'wrrir5rh)rrMrNrOrQr/r{ set_indexrr rrXr}is_open)r$r%r'r&directr6indirects r(test_read_hdf_open_storers  a ''/6lW  B BHHM 3t ,B j DIId3I' dD !F $S !UE4( fh/}}}s #1CC'cJttjjdjdgdt d}||z }|j |ddd t |d}|jjjJtj||y) Nrr)rrurrvrrr'rrrriformat) rrMrNrOrQrr r/_databaserXr})r$r%r'r&rFs r(test_read_hdf_index_not_viewrs  a ''/W  B j DIId3wI7 4 C 99??   '' '"c"c"ttjjdjdt dt d}d|j _|jdd }||z }|j|d d d t|d }t|d d}t|j5t|tsJt|j}dddt!j"|y#1swY xYw)NrrrrrrrTrr'rtr)iterator)rrMrNrOrQr/r{rrr rr6r[rnext__iter__rXr})r$r%r'r&rrrs r(test_read_hdf_iteratorrs  a ''/6lW  B BHHM 3t ,B j DIId3sI3 dD !FdT2H  -(M222))+,-&(+--s :,DDcttjjdjdt dt d}||z }|j |ddt |}tj|||j |d dd }tjt| 5t |dddy#1swYyxYw) Nrrrrrr'rrrF?key must be provided when HDF5 file contains multiple datasets.r) rrMrNrOrQrr rXr}r r!r\r$r%r'r&rereadras r(test_read_nokeyrs  a ''/6lW  B j DIId3I' d^F"f%IIdCI( KC z -s 8 C  Cctttdttddd}||z }|j |ddd t |}t j|||j |d dd d }tjt| 5t |dddy#1swYyxYw) Nrabacdcategoryr<)icr'rrrrFrr) rrr rQrr rXr}r r!r\rs r(test_read_nokey_tabler)s qtG}J(OP QB j DIId3wI7 d^F"f%IIdCI8 KC z -s  B..B7c||z }t|}|jtjd}t j t |5t|dddy#1swYyxYw)Nz]Dataset(s) incompatible with Pandas data types, not table, or no datasets found in HDF5 file.r)rcloserUrVr r!r\r )r$r%r&r6ras r(test_read_nokey_emptyr8s] j D TNE KKM )) ' C z -s  A&&A/c*ttjjdjdt dt d}||z }t |}|j |ddt|d }tj||y) Nrrrrrr'rrr) rrMrNrOrQrrr rXr})r$r%rcfilenamepath_objactuals r(test_read_from_pathlib_pathrDsz a ''/6lW H *$HH~H OOH$SO1 hD )F(F+rzpy.pathc0ddlm}ttjj dj dt dt d}||z }||}|j|dd t|d }tj||y) Nr)localrrrrrr'rrr) py.pathrrrMrNrOrQrr rXr})r$r% LocalPathrcrrrs r(test_read_from_py_localpathrTs+ a ''/6lW H *$H"H OOH$SO1 hD )F(F+rrfixedrcttdtj}||z }|j |d|t |dd}t j||y)Nr9r<rgrrr5r)r rrMfloat64rr rXr`)r$rr%seriesr&rbs r(test_read_hdf_series_mode_rrgsR E"IRZZ 0F j D MM$F6M2 dS 1F66*rz8ignore:Period with BDay freq is deprecated:FutureWarningz3ignore:PeriodDtype\[B\] is deprecated:FutureWarningctgdtjgdd}t|dddd d 5}|d }dddt j |y#1swY xYw) N)g?rrv)z 2015-01-01z 2015-01-02z 2015-01-05r-)r?r.rfrgroz*periodindex_0.20.1_x86_64_darwin_2.7.13.h5r5rhp)rpd PeriodIndexrrXr}rs r(test_read_py2_hdf_file_in_py3rssvnnGcRH  &,(T    s &(+s A""A+cXtjdtdgdi}||z }|j|ddt j dd5t |dd }dddtdgdid tdgd  }tj|y#1swY>xYw)Npyarrowr)rrNrgrrzfuture.infer_stringTr5rzstring[pyarrow_numpy]r<)r=rA) r importorskiprrroption_contextr r rXr})r$r%r'r&rbrcs r(test_read_infer_stringrs  " C)* +B j DIIdwI/  0$ 76$F56 %se#:;H &(+66s B  B))9 contextlibrpathlibrrUnumpyrMr pandas._libs.tslibsr pandas.compatrpandasrrrr r r rXr r pandas.tests.io.pytables.commonrr pandas.utilrtdpandas.io.pytablesrmark single_cpu pytestmarkr)r3r7rdrlskipifrsrrrrrrrrrr skip_if_nor parametrizerfilterwarningsrrrr(rsO  )-/, [[ # #    O1d%')2WX%Y% ,," ,(#$,&*   , y,,$GW#56+7+WXRS,TY,4 ,r