eh9ddlZddlZddlZddlmZmZmZmZddl m Z m Z ddl m Z mZmZmZmZmZmZddlmZmZmZddlmZddlmZej:j<Zej: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'e0gd(Z1d)Z2d*Z3y)+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-rArCs rDtest_default_moderG[s}  a 009d6l&1r< B j DIId3I' dD !F&"%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 r0arangefloat64rr6r7ropencloseis_openlen_mode)r?r>rArBr@s rDtest_reopen_handlerShs 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|tj|d |tj|d||jdddtjj!rJycc}w#1swY0xYw)N皙?xrrr rXi-r'r H5FD_COREr)rdriverdriver_core_backing_storer-df2)r< ensure_cleanr r0rLreshaper r4r5ranger appendr=rOosrAexists)r>rAir-rBs rDtest_open_argsres  $  "))C.((1 1$v,f5595a2aS5VD  s;! d  UB eDk2. eElB/ #(ww~~d## ##!6 sAD"- D9A:D"D""D+ct|5}ttd|d<|j|jddddy#1swYyxYw)NrT)fsync)rr r`flush)r>rBs rD test_flushrjsJ J ' 5E!H%c   $    s :AAc .tdtjdjdzt t dt t tdDcgc]}d| c}t }||z }|j|d d t|d }tj||tj|d 5}|jddD]8}|jj d k(sJ|jj"dk(r8J ddd||z }|j|d dt|d }tj||tj|d 5}|jddD]5}|jj dk(sJ|jj"5J ddd||z }|j|d t|d }tj||tj|d 5}|jddD]5}|jj dk(sJ|jj"5J dddycc}w#1swY]xYw#1swYxYw#1swYyxYw)NrUrVrWrr rXrYr'r- )r/ complevelrr,/dfLeafwhere classnamezlib)r/complibr)r/)r r0rLr_r r4r5r`r:rr<r=r open_file walk_nodesfiltersrmrt)r?r>rdr-tmpfilerCh5filenodes rDtest_complibs_default_settingsr{si  biin$$W--d6l&1uRy1!r!X1@ B#GIIg41I- gt $F&"%   ' ,2%%EV%D 2D<<))Q. ..<<''61 11 22 #GIIg4I0 gt $F&"%   ' ,0%%EV%D 0D<<))Q. ..<<''/ // 00 #GIIg4I gt $F&"%   ' ,0%%EV%D 0D<<))Q. ..<<''/ // 000A2220000s? I-A I2I21AI?:I?AJ  J 2I<?J Jc 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)NrUrVrWrr rXrYr'dfcrlblosc)rmrtr-rr,rnrorprz/dfc)r r0rLr_r r4r5r`r rarOrrurvrwrmrt)r?r>rdr-rxrBryrzs rD'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.11rrWrrXr'lzozlzo not availablebzip2zbzip2 not available_z.h5)r/rtrmrr,/rorp)rr applymarkerr6markxfailr r0onesr4rLastypestr_rwhich_lib_versionskipr:rr<r=rurvrwrmrt) r?rrrequestr-rxgnamerCh5tablerzs rD 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#rgrJ)rr)r#r-ascii)encodingr)r(z columns=A) rr r`r0nanlocrrar<r=reindexselectr)r>rBr-expectedrCs rD test_encodingr,s J ' 05 U/uQx @vveT" T2 0 eDk2.::se:,dDw$GH fh/ 0 0 0s CC((C1val)E, 17rHabc)rrrr)sEE, 17rHrrr)rrrr)rHrrr)rrrr)ArHrrrrHrrrr!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_categoriesreplacer0rr<assert_series_equal) r?r>r!rencrr/xserrBretrs_nans rDtest_latin_encodingr=s 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)NrUrVrWrr rXrYr'r-rr)r/rrCLOSEDzTThe file [\S]* is already opened\. Please close it before reopening in write mode\.r*r,r]z[\S]* file is not open!df3z''HDFStore' object has no attribute 'df')r r0rLr_r r4r5r`r:r inforPrOr!_table_file_open_policy_is_strictr6r7r;rarkeysrQrgetput get_storerremoveAttributeErrorr-) r?r>rArdr-rBstore1r@store2s rDtest_multiple_open_closeres 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<r^r rbfspathstr)rArBs rD test_fspathrsc  "1d d^ 1u99U#s4y0 00 111 1 111s" A$$AA$A! A$$A-)4rbnumpyr0r6 pandas.compatrrrr pandas.errorsrrpandasr r r r r r<rrpandas.tests.io.pytables.commonrrr pandas.iorpandas.io.pytablesrr single_cpu pytestmark parametrizerErGrSrerjr{rr`rw all_complibsfilterwarningsskipifrrrr5rrrrHrDrs     # [[ # # !671*81*h &5p$0 (0V3,b * ; ;<?@ I=#%=*;*= , 3 A=+ 3F!##,R 0 0 -(*5'- dD! rvvsD$/  :v"67(8 (2DN1rH