{och<ddlZddlZddlZddlmZmZmZmZddl m Z m Z ddl m Z mZmZmZmZmZmZddlmZmZmZddlmZddlmZej:j<gZej:jAdgd d Z!d Z"d Z#d Z$dZ%dZ&dZ'ej:jAde(dej:jAdejRjTej:jWdej:jYe xrexreddZ-ej:jYe ddZ.ej:jAdgdgdgdgdgdgd gd!ej^d"d#d$gd%ej^d"d#d$gg ej:jAd&d'dgd(Z0d)Z1d*Z2y)+N)PY311is_ci_environmentis_platform_linuxis_platform_little_endian)ClosedFileErrorPossibleDataLossError) DataFrameHDFStoreIndexSeries_testing date_rangeread_hdf) _maybe_removeensure_clean_storetables)pytables)Termmode)rr+awc ttjjdj dt t dttddd }d }||z }|d vr2tjt| 5t|| dddn2t|| 5}|jj|k(sJ ddd||z }|d vrr?)rAr@rBr-rDrFexpecteds rGtest_default_moderL]s  a 009d6l&1r< B j DIId3I' dD !FwwyH#++2259&(+c||z }t|d}ttjdtjt dd|d<d}t jt| 5|jd ddd|j|jrJ|jd |jsJt|d k(sJ|j|jrJt|d}ttjdtjt dd|d<|jd |jsJt|d k(sJ|jd k(sJ|j|jrJ|jd|jsJt|d k(sJ|jdk(sJ|j|jrJ|jd|jsJt|d k(sJ|jdk(sJ|j|jrJy#1swYxYw)Nrr,rr z 2020-01-01)r%r)zKRe-opening the file \[[\S]*\] with mode \[a\] will delete the current file!r*rrr)r r r1arangefloat64rr7r8ropencloseis_openlen_mode)rAr@rDrErCs rGtest_reopen_handlerXms j D T $E "BJJ'z,PR/SE#J   ,C 8 3 KKM}}  JJsO === u:?? KKM}}  T $E "BJJ'z,PR/SE#J  JJsO === u:?? ;;#   KKM}}  JJsO === u:?? ;;#   KKM}}  JJsO === u:?? ;;#   KKM}} }Qs .II#c tj|5}tdtjdj dzt tdtt tdDcgc]}d| c}t}t|d d d }||d <|jd||j}|r@|jjd|_ |jjd|_tj |d |tj |d||j#dddt$j&j)rJycc}w#1swY0xYw)N皙?xrrr r]i-r'r H5FD_COREr)rdriverdriver_core_backing_storer-df2r0)r> ensure_cleanr r1rQreshaper r5r6ranger appendrJr)r=r(r?rTosrDexists)r@rBrDir-rErKs rGtest_open_argsrjs3  $  "))C.((1 1$v,f5595a2aS5VD  s;! d  UB779 %^^2259HN'//66u=H  eDk84 eElH5 -2ww~~d## ##+6 sAE4- E/9C E4/E44E=ct|5}ttd|d<|j|jddddy#1swYyxYw)NrT)fsync)rr reflush)r@rEs rG test_flushrosJ J ' 5E!H%c   $    s :AAc tdtjdjdzt t dt t tdDcgc]}d| c}t }||z }|j|d d t|d }|j}|r@|jjd |_ |jjd |_ tj||t!j"|d 5}|j%ddD]8} | j&j(d k(sJ| j&j*dk(r8J ddd||z }|j|d dt|d }|j}|r@|jjd |_ |jjd |_ tj||t!j"|d 5}|j%ddD]5} | j&j(dk(sJ| j&j*5J ddd||z }|j|d t|d }|j}|r@|jjd |_ |jjd |_ tj||t!j"|d 5}|j%ddD]5} | j&j(dk(sJ| j&j*5J dddycc}w#1swYxYw#1swYxYw#1swYyxYw)NrZr[r\rr r]r^r'r- )r/ complevelr0rr,/dfLeafwhere classnamezlib)r/complibr)r/)r r1rQrdr r5r6rer;rrJr)r=r(r>r?r open_file walk_nodesfiltersrrry) rAr@rBrir-tmpfilerFrKh5filenodes rGtest_complibs_default_settingsrs   biin$$W--d6l&1uRy1!r!X1@ B#GIIg41I- gt $FwwyH!..u5#++2259&(+   ' ,2%%EV%D 2D<<))Q. ..<<''61 11 22 #GIIg4I0 gt $FwwyH!..u5#++2259&(+   ' ,0%%EV%D 0D<<))Q. ..<<''/ // 00 #GIIg4I gt $FwwyH!..u5#++2259&(+   ' ,0%%EV%D 0D<<))Q. ..<<''/ // 000Y2220000s? M#A M(%M(AM5M5 ANN(M25M?N c tdtjdjdzt t dt t tdDcgc]}d| c}t }||z }t|}|jd |d d |jd ||jtj|d5}|jddD]5}|jjdk(sJ|jj 5J|jddD]8}|jjd k(sJ|jj d k(r8J dddycc}w#1swYyxYw)NrZr[r\rr r]r^r'dfcrqblosc)rrryr-rr,rsrtrurz/dfc)r r1rQrdr r5r6rer rfrTrrzr{r|rrry)rAr@rir-r}rEr~rs rG'test_complibs_default_settings_overridersd  biin$$W--d6l&1uRy1!r!X1@ B #G W E LLaL9 LLr KKM   ' ,3%%EV%D 0D<<))Q. ..<<''/ // 0%%Ff%E 3D<<))Q. ..<<''72 22 3 33233s  E7AE<A E<*E<<Flvlrlibz!ignore:object name is not a validzSegfaulting in a CI environmentreasonctrGtr=|dk(r8|dk7r3|jtjj d|dt tjdtdtjdjtj }tjd stjd tjd stjd ||d|dz }|d|}|j!||||t#||}t%j&||tj(|d5}|j+d|zdD]V} | j,j.|k(sJ|dk(r| j,j0:J| j,j0|k(rVJ dddy#1swYyxYw)Nblosc2rz Fails for z on Linux and PY > 3.11rr\rr]r'lzozlzo not availablebzip2zbzip2 not available_z.h5)r/ryrrrr,/rtru)rr applymarkerr7markxfailr r1onesr5rQr=str_rwhich_lib_versionskipr;rr>r?rzr{r|rrry) rArrrequestr-r}gnamerFh5tablers rG test_complibsrs "$SAX KK  z#6M%N  O   $v,biim6J6J2776S B  # #E * '(  # #G , )*C5#c**Ge1SENEIIg5#I= gu %F&"%   ' ,3&&S5[F&K 3D<<))S0 00ax||++333||++s222  3333sAG-G GGz$reason platform is not little endianct|5}tdddtd}tj|j d<tj|j d<t |d|jd|d tj|d||jd g }|jdtd d }tj||dddy#1swYyxYw)Nfoobar)Ar#rlrO)rr)r#r-ascii)encodingr)r(z columns=A) rr rer1nanlocrrfr>r?reindexselectr)r@rEr-rKrFs rG test_encodingrBs J ' 05 U/uQx @vveT" T2 0 eDk2.::se:,dDw$GH fh/ 0 0 0s CC((C1val)E, 17rMabc)rrrr)sEE, 17rMrrr)rrrr)rMrrr)rrrr)ArMrrrrMrrrr!categorycd}d}d}|Dcgc]%}t|tr|j|n|'}}t||}||z } |j | |d||t | |} |dk(r8||j jvr|j j|g} n#|} n |j|tj} tj| | ycc}w)Nzlatin-1datar table)r/formatrnan_repr) isinstancebytesdecoder r;rcat categoriesremove_categoriesreplacer1rr>assert_series_equal) rAr@r!rencrr/xserrEretrs_nans rGtest_latin_encodingrSs CG CAD EAJq%0188C=a 7 EC E E "C z !EJJu#gWJM E3 D  cgg(( (GG--wi8EE GRVV,5$'# Fs*Cc p||z }tdtjdjdzt t dt t tdDcgc]}d| c}t }|j|d d d t|}d |jvsJ|jsJ|jd |jvsJ|jrJ||z }tjrNt|}d}tj t"|5t|ddd|jn0t|}t|}d |jvsJd |jvsJ|jsJ|jsJ|jd |jvsJ|jrJd |jvsJ|jsJ|jd |jvsJd |jvsJ|jrJ|jrJt|d }|j%d |t|}|j%d||jd |jvsJ|jrJ|jd |jvsJ|jrJt|d }|j%d |t|}|jd |jvsJ|jrJ|jd |jvsJ|jrJ||z }tdtjdjdzt t dt t tdDcgc]}d| c}t }|j|d d d t|}|jd}tj t&|5|j)dddtj t&|5d |vdddtj t&|5t+|dddtj t&|5|d dddtj t&|5|j-d dddtj t&|5|j/d dddtj t&|5|j%d|dddtj t&|5|j1d|dddtj t&|5|j3ddddtj t&|5|j5ddddtj t&|5|j-d dddd}tj t6|5|j8dddycc}w#1swYXxYwcc}w#1swYIxYw#1swY.xYw#1swY xYw#1swYxYw#1swYxYw#1swYxYw#1swYwxYw#1swYNxYw#1swY&xYw#1swYxYw#1swYxYw#1swYyxYw)NrZr[r\rr r]r^r'r-rr)r/rrCLOSEDzTThe file [\S]* is already opened\. Please close it before reopening in write mode\.r*r,rbz[\S]* file is not open!df3z''HDFStore' object has no attribute 'df')r r1rQrdr r5r6rer;r inforUrTr!_table_file_open_policy_is_strictr7r8r<rfrkeysrVrgetput get_storerremoveAttributeErrorr-) rAr@rDrir-rEstore1rCstore2s rGtest_multiple_open_closer{s j D  biin$$W--d6l&1uRy1!r!X1@ B IId3wI7 TNE 5::< '' ' === KKM uzz| ## #}}  j D11$ ) ]]:S 1  TN   $$v{{},,,v{{},,,~~~~~~ 6;;=(((>>!!v{{},,,~~~ 6;;=(((6;;=(((>>!!>>!!C( T2$ eR  6;;=(((>>!! 5::<'''==  C( T2$ 5::<'''==   6;;=(((>>!! j D  biin$$W--d6l&1uRy1!r!X1@ B IId3wI7 TNE KKM $C c 2  c 2   c 2 E  c 2 d  c 2 T c 2 $ c 2  UB  c 2 % c 2    c 2 U c 2 T 4C ~S 1 w2,  v2    s Z+ Z  ZZ;Z,# Z9[;[0[ %[-[:\\;\ 2 \, ZZ),Z69[[[ [*-[7:\\\ \),\5ctjd5}t|5}tj|t |k(sJ ddddddy#1swYxYw#1swYyxYw)Nzfoo.h5)r>rcr rgfspathr0)rDrEs rG test_fspathrsc  "1d d^ 1u99U#s4y0 00 111 1 111s" A$$AA$A! A$$A-)3rgnumpyr1r7 pandas.compatrrrr pandas.errorsrrpandasr r r r r r>rrpandas.tests.io.pytables.commonrrr pandas.iorpandas.io.pytablesrr single_cpu pytestmark parametrizerHrLrXrjrorrrer| all_complibsfilterwarningsskipifrrrrrrrMrGrs     #kk$$ % !673*83*l , 5p$: 40n3,b * ; ;<?@ I=#%=*;*= , 3 A=+ 3F!##,R 0 0 -(*5'- dD! rvvsD$/  :t"45(6 (2DN1rM