o Tj<@sddlZddlZddlZddlmZmZmZmZddl m Z m Z ddl m Z mZmZmZmZmZmZddlmZmZmZddlmZddlmZejjgZej dgd d d Z!d d Z"ddZ#ddZ$ddZ%ddZ&ddZ'ej de(dej dej)j*ej+dejj,e oeoeddddZ-ejj,e d dd!d"Z.ej d#gd$gd%gd&gd'gd(gd)gd*ej/d+d,d-gd.ej/d+d,d-gg ej d/d0dgd1d2Z0d3d4Z1d5d6Z2dS)7N)PY311is_ci_environmentis_platform_linuxis_platform_little_endian)ClosedFileErrorPossibleDataLossError) DataFrameHDFStoreIndexSeries_testing date_rangeread_hdf) _maybe_removeensure_clean_storetables)pytables)Termmode)rr+awc CsZttjddttdtdtddddd }d }||}|d vrAt j t |d t ||d Wdn1s;wYnt ||d }|j j|ksPJWdn1sZwY||}|d vrt j t |d t ||d }Wdn1swYWdn1swYnt ||d }|j j|ksJWdn1swY||}|d vrt j t |d |j|d|dWdn1swY|j|dddn|j|d|d|dvrd}t j t|d t|d|d WddS1s wYdSt|d|d }|r%|jd|_t||dS)N ABCDdtype 2000-01-01rBperiodsfreqcolumnsindexz[\S]* does not exist)rrmatchrdfkeyrr)rzNmode w is not allowed while performing a read. Allowed modes are r, r\+ and a.str)rnprandom default_rngstandard_normalr listobjectr pytestraisesOSErrorr _handlerto_hdf ValueErrorrr&astypetmassert_frame_equal) setup_pathtmp_pathrusing_infer_stringr+msgpathstoreresultrEn/home/dinkstrade/pdmp-scanner/venv/lib/python3.10/site-packages/pandas/tests/io/pytables/test_file_handling.py test_mode&sX  $rGcCs~ttjddttdtdtddddd }||}|j |d d d t |d }| }|r7|j d |_ t||dS)Nrrrrr rr!r"r%r+rr,r.)rr/r0r1r2r r3r4r r9rcopyr&r;r<r=)r?r>r@r+rBrDexpectedrErErFtest_default_mode]s  rJcCs||}t|dd}ttjdtjdtdddd|d<d}tjt|d  | d Wdn1s5wY| |j rCJ| d |j sMJt |d ksUJ| |j r^Jt|dd}ttjdtjdtdddd|d<| d |j sJt |d ksJ|j d ksJ| |j rJ| d|j sJt |d ksJ|j dksJ| |j rJ| d|j sJt |d ksJ|j dksJ| |j rJdS)Nrr*rrz 2020-01-01)r#r'zKRe-opening the file \[[\S]*\] with mode \[a\] will delete the current file!r(rrr)r r r/arangefloat64r r5r6ropencloseis_openlen_mode)r?r>rBrCrArErErFtest_reopen_handlemsN                rTcCst|b}tdtddttdtdtddt dDtdd }t |d d d d }||d<| d|| }|rL|j d|_ |jd|_t|d|t|d||Wdn1sjwYtj|rwJdS)N皙?xrrrcSg|]}d|qSzi-rE.0irErErF z"test_open_args..rXr%r H5FD_COREr)rdriverdriver_core_backing_storer+df2r.)r< ensure_cleanrr/rMreshaper r3r4ranger appendrHr'r;r&r=rPosrBexists)r>r@rBr+rCrIrErErFtest_open_argss(   rjcCsRt|}ttd|d<||jddWddS1s"wYdS)NrT)fsync)rr rfflush)r>rCrErErF test_flushs "rnc CsttdtddttdtdtddtdDtdd }||}|j|d d d t |d }| }|rE|j d |_ |j d |_ t||tj|dd"}|jdddD]}|jjd ksdJ|jjdkslJqZWdn1swwY||}|j|d ddt |d }| }|r|j d |_ |j d |_ t||tj|dd"}|jdddD]}|jjdksJ|jjdusJqWdn1swY||}|j|d dt |d }| }|r|j d |_ |j d |_ t||tj|dd&}|jdddD]}|jjdksJ|jjdus%JqWddS1s3wYdS)NrUrVrWrrcSrYrZrEr[rErErFr^r_z2test_complibs_default_settings..rXr%r+ )r- complevelr.rr*/dfLeafwhere classnamezlib)r-complibr)r-)rr/rMrer r3r4rfr9rrHr'r;r&r<r=r open_file walk_nodesfiltersrprw) r?r>r@r+tmpfilerDrIh5filenoderErErFtest_complibs_default_settingss^      $r~cCstdtddttdtdtddtdDtdd }||}t|}|j d |d d d | d|| t j |dd=}|j dddD]}|jjdksTJ|jjdus\JqJ|j dddD]}|jjd ksnJ|jjd ksvJqdWddS1swYdS)NrUrVrWrrcSrYrZrEr[rErErFr^r_z;test_complibs_default_settings_override..rXr%dfcroblosc)rprwr+rr*rqrrrsrz/dfc)rr/rMrer r3r4rfr rgrPrrxryrzrprw)r?r>r+r{rCr|r}rErErF'test_complibs_default_settings_overrides& "rlvlrlibz!ignore:object name is not a validzSegfaulting in a CI environmentreasonc CsTtrtr|dkr|dkr|tjjd|ddttdt dt d tj d }t d s8td t d sBtd ||d|d}|d|}|j||||dt||}t||t j|dd2}|jd|ddD]} | jj|ksJ|dkr| jjdusJqx| jj|ksJqxWddS1swYdS)Nblosc2rz Fails for z on Linux and PY > 3.11rrWrrXr%lzozlzo not availablebzip2zbzip2 not available_z.h5)r-rwrprr*/rrrs)rr applymarkerr5markxfailrr/onesr3rMr;str_rwhich_lib_versionskipr9rr<r=rxryrzrprw) r?rrrequestr+r{gnamerDh5tabler}rErErF test_complibss0       "rz$reason platform is not little endiancCst|L}tdddtdd}tj|jd<tj|jd<t|d|jd|d d t |d||j d gd }| dt d d d }t ||WddS1sSwYdS)Nfoobar)Ar!rkrK)rr)r!r+ascii)encodingr)r&z columns=A) rrrfr/nanlocrrgr<r=reindexselectr)r>rCr+rIrDrErErF test_encodingBs    "rval)E, 17abc)rrrr)sEE, 17rrrr)rrrr)rrrr)rrrr)Arrrrrrrrrcategoryc sdd}d}fdd|D}t||d}||}|j||d|dt||}|d kr=||jjvr:|j|g} n |} n||tj} t | |dS) Nzlatin-1datacs$g|]}t|tr|n|qSrE) isinstancebytesdecode)r\xencrErFr^gs$z'test_latin_encoding..rtable)r-formatrnan_repr) r r9rcat categoriesremove_categoriesreplacer/rr<assert_series_equal) r?r>rrrr-serrCretrs_nanrErrFtest_latin_encodingSs   rcCs||}tdtddttdtdtddtdDtdd }|j|d d d d t |}d| vs9J|j s>J| d| vsJJ|j rOJ||}t jr|t |}d}tjt|d t |Wdn1srwY| nt |}t |}d| vsJd| vsJ|j sJ|j sJ| d| vsJ|j rJd| vsJ|j sJ| d| vsJd| vsJ|j rJ|j rJt |d d}|d |t |}|d|| d| vsJ|j rJ| d| vsJ|j rJt |d d}|d |t |}| d| vs2J|j r8J| d| vsEJ|j rKJ||}tdtddttdtdtddtdDtdd }|j|d d d d t |}| d}tjt|d |Wdn 1swYtjt|d d |vWdn 1swYtjt|d t|Wdn 1swYtjt|d |d Wdn 1swYtjt|d |d Wdn 1s wYtjt|d |d Wdn 1s'wYtjt|d|d|Wdn 1sEwYtjt|d|d|Wdn 1scwYtjt|d |dWdn 1swYtjt|d |dWdn 1swYtjt|d |d Wdn 1swYd}tjt|d |jWddS1swYdS)NrUrVrWrrcSrYrZrEr[rErErFr^r_z,test_multiple_open_close..rXr%r+rr)r-rrCLOSEDzTThe file [\S]* is already opened\. Please close it before reopening in write mode\.r(r*rccSrYrZrEr[rErErFr^r_z[\S]* file is not open!df3z''HDFStore' object has no attribute 'df')rr/rMrer r3r4rfr9r inforQrPr!_table_file_open_policy_is_strictr5r6r:rgrkeysrRrgetput get_storerremoveAttributeErrorr+)r?r>rBr+rCstore1rAstore2rErErFtest_multiple_open_close{s                            $rc Cs~td0}t|}t|t|ksJWdn1s wYWddSWddS1s8wYdS)Nzfoo.h5)r<rdr rhfspathr.)rBrCrErErF test_fspaths  "r)3rhnumpyr/r5 pandas.compatrrrr pandas.errorsrrpandasrr r r r r<r rpandas.tests.io.pytables.commonrrr pandas.iorpandas.io.pytablesrr single_cpu pytestmark parametrizerGrJrTrjrnr~rrfrz all_complibsfilterwarningsskipifrrrrrrrErErErFs`$    687 #