B Z0dV @sddlZddlmZddlZddlmZddlZddlZddl m Z m Z m Z m Z mZmZmZddlmZmZddlmZmZejjZddZd d Zd d Zd dZddZddZ ej!deddddde e"dgddZ#ddZ$ddZ%d d!Z&ej!d"d#d$gd%d&gd'd(Z'dS))N)BytesIO)catch_warnings)CategoricalIndex DataFrameHDFStore MultiIndex_testing date_rangeread_hdf)ensure_clean_pathensure_clean_store)Term_maybe_adjust_namec Cst}t|n}|d|d}tjt|d|jddgdWdQRXd}tjt|d|jddgdWdQRXWdQRXdS) Ndfzqcannot pass a column specification when reading a Fixed format store. this store must be selected in its entirety)matchA)columnszucannot pass a where specification when reading from a Fixed format store. this store must be selected in its entiretyz columns=A)where)tm makeDataFramer putpytestraises TypeErrorselect) setup_pathrstoremsgrV/var/www/html/venv/lib/python3.7/site-packages/pandas/tests/io/pytables/test_errors.pytest_pass_spec_to_storers  r c Cstddddgi}tddddgitddd d }t|L}|jd |d d td}tjt|d|jd |d ddWdQRXWdQRXdS)Naz1/1/2000)periods)indexframetable)formatz/incompatible kind in col [integer - datetime64])rT)r,append) rr r rreescaperrr)rZdf1df2rrrrr$test_table_index_incompatible_dtypes3s  r1c Cs t|z}dtdddfg}x^|D]V\}}t}|||<td|d}tjt |d| d||WdQRXq$WWdQRXt}d |d <d |d <tddd|d <| j dd}t|6}td}tjt |d| d|WdQRXWdQRXdS)Ndateir"r#[z&] is not implemented as a table column)rZdf1_fooZobj1barobj2Z datetime1T)datetimezjCannot serialize the column [datetime1] because its data contents are not [string] but [date] object dtypeZdf_unimplemented) r r7r2rrr.r/rrrr-Z _consolidate_convert)rrZdtypesnfrrrrr'test_unimplemented_dtypes_table_columns>s$ * r;c Cst|}tddt}d|d<d|j|jdddf<|jd|d d td }t j t |d  t WdQRXtd }t j t |d |ddWdQRXd}t j t|d |ddWdQRXWdQRXWdQRXt|X}ttjddtdtdddd}|j|dd ddt|dddt|dddWdQRXt|f}ttjddtdtdddd}|j|dd d d}t j t |d t|dddWdQRXWdQRXdS)NT)recordr4stringr5rr%rr+)r,z:__init__() missing 1 required positional argument: 'where')rzWcannot process expression [df.index[3]], [2000-01-06 00:00:00] is not a valid conditionz df.index[3]zinvalid syntaxzindex> ZABCDZ20130101)r()rr)dfq)r,Z data_columnsz0index>Timestamp('20130104') & columns=['A', 'B'])rz A>0 or C>0aThe passed where expression: A>0 or C>0\n\s*contains an invalid variable reference\n\s*all of the variable references must be a reference to\n\s*an axis \(e.g. 'index' or 'columns'\), or a data_column\n\s*The currently defined references are: index,columns\n)r rrZmakeTimeDataFramelocr)rr.r/rrrr ValueErrorr SyntaxErrorr rnprandomrandnlistr to_hdfr )rrrrpathr?rrrtest_invalid_terms]sD  *    rIc Csttjdd}tdtjdi}tdtjdi}tdtjdi}tddtftjdi}t|h}dtd}|||xD||||fD]4}t d } t j t | d |||WdQRXqWWdQRXdS) Nr>r"r!)r"r#)1r#rJr#Zdf_z?cannot match existing table structure for [0] on appending data)r)rrCrDrEobjectr rZrandsr-r.r/rrrA) rrr0Zdf3Zdf4Zdf5rnamedrrrr7test_append_with_diff_col_name_types_raises_value_errors  rNc Cshttjddtdtdd}t|4}d}tjt |d|j |dd d WdQRXWdQRXdS) Nr%r&abcdABCDE)r)rz)complib only supports \[.*\] compression.)rrZfoolib)Zcomplib) rrCrDrandrFrZ ensure_cleanrrrArG)rrrHrrrrtest_invalid_complibs   rRidxZ2019Dr$UTC)freqr(tzabcc Cs\t||g}td|dgd}t|,}tjtdd||dWdQRXWdQRXdS)Nrr!)r)rzSaving a MultiIndex)rr)rZ from_arraysrr rrNotImplementedErrorrG)rSrmirrHrrr&test_to_hdf_multiindex_extension_dtypes  r[c Cs6|ddd}d}tjt|dt|WdQRXdS)Niodataz"legacy_hdf/incompatible_dataset.h5z_Dataset\(s\) incompatible with Pandas data types, not table, or no datasets found in HDF5 file.)r)rrrAr )ZdatapathZ data_pathmessagerrrtest_unsuppored_hdf_file_errors r_c Csttjddtdtdd}t|v}d}tjt|dt |dWdQRX| |d t |d d }| d }tjt|dt |d WdQRXWdQRXdS) Nr%r&rOrP)r)rzFile [\S]* does not exist)rkeyrr)modez&The HDFStore must be open for reading.) rrCrDrQrFr rrIOErrorr rGrclose)rrrHrrrrrtest_read_hdf_errorss    rec Cs0d}tjt|dttddWdQRXdS)Nz5Support for generic buffers has not been implemented.)rr)rrrYr r)rrrr#test_read_hdf_generic_buffer_errorssrg bad_version)r"r#)r"Z12Z123c Cs.d}tjt|dtd|dWdQRXdS)Nz5Version is incorrect, expected sequence of 3 integers)rZvalues_block_0)version)rrrAr)rhrrrr)test_maybe_adjust_name_bad_version_raisessrj)(r7r\rr.warningsrnumpyrCrZpandasrrrrrrr r Zpandas.tests.io.pytables.commonr r Zpandas.io.pytablesr rmarkZsingleZ pytestmarkr r1r;rIrNrRZ parametrizerFr[r_rergrjrrrrs.  $  @