U
|
ý°dX ã @ s d dl Z d dlmZ d dlZd dlmZ d dlZd dlZd dl m
|
Z
|
m Z m Z m Z mZmZmZ d dlmZ d dlmZmZ ejjZdd Zd d
|
Zd d Z d d Z dd Z dd Z ej deddddde
|
e!dg¡dd Z"d d Z#d d Z$d d! Z%ej d"d#d$g d%d&g¡d'd( Z&dS ))é N)ÚBytesIO)Úcatch_warnings)ÚCategoricalIndexÚ DataFrameÚHDFStoreÚ
|
MultiIndexÚ_testingÚ
|
date_rangeÚread_hdf)Úensure_clean_store)ÚTermÚ_maybe_adjust_namec
|
C s t ¡ }t| n}| d|¡ d}tjt|d |jddgd W 5 Q R X d}tjt|d |jddgd W 5 Q R X W 5 Q R X d S ) NÚdfzqcannot pass a column specification when reading a Fixed format store. this store must be selected in its entirety©ÚmatchÚA)Ú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 ÚputÚpytestÚraisesÚ TypeErrorÚselect)Ú
|
setup_pathr ÚstoreÚmsg© r ú[d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/io/pytables/test_errors.pyÚtest_pass_spec_to_storer s
|
ÿÿr! c
|
C s t ddddgi}t ddddgitddd d
|
}t| L}|jd |d d t d¡}tjt|d |jd |d dd W 5 Q R X W 5 Q R X d S )NÚaé é é é é é z1/1/2000©Úperiods)ÚindexÚframeÚtable©Úformatz/incompatible kind in col [integer - datetime64]r T)r/ Úappend) r r r r ÚreÚescaper r r )r Zdf1Údf2r r r r r Ú$test_table_index_incompatible_dtypes/ s
|
|
r4 c C sþ t | v}dt ddd¡fg}|D ]V\}}t ¡ }|||< t d| d¡}tjt |d |
|
d| |¡ W 5 Q R X q"W 5 Q R X t ¡ }d |d
|
< d |d < t ddd¡|d < | ¡ }t | 6}t d¡}tjt |d |
|
d|¡ W 5 Q R X W 5 Q R X d S )NÚdateiÑ r# r$ ú[z&] is not implemented as a table columnr Zdf1_ÚfooZobj1ÚbarÚobj2Z datetime1zjCannot serialize the column [datetime1]
|
because its data contents are not [string] but [date] object dtypeZdf_unimplemented) r Údatetimer5 r r r1 r2 r r r r0 Z _consolidate)r r ZdtypesÚnÚfr r r r r Ú'test_unimplemented_dtypes_table_columns: s&
|
(
|
ÿr= c C s¬ t |Ø}tdd t ¡ }d|d< d|j|jdd
df< |jd|d d
|
t d ¡}t j
|
t |d t W 5 Q R X t d ¡}t j
|
t |d | dd¡ W 5 Q R X d}t j
|
t|d | dd¡ W 5 Q R X W 5 Q R X W 5 Q R X | | }ttj dd¡tdtdddd}|j|dd dd t|ddd t|ddd | | }ttj dd¡tdtdddd}|j|dd d
|
d}t j
|
t |d t|ddd W 5 Q R X d S ) NT)Úrecordr7 Ústringr8 r r& r r- r. z:__init__() missing 1 required positional argument: 'where'r zWcannot process expression [df.index[3]], [2000-01-06 00:00:00] is not a valid conditionz df.index[3]zinvalid syntaxzindex>é
|
ÚABCDZ20130101r) )r r+ Údfq)r/ Z data_columnsz0index>Timestamp('20130104') & columns=['A', 'B']r z
|
A>0 or C>0a The 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 r r ZmakeTimeDataFrameÚlocr+ r r1 r2 r r r r Ú
|
ValueErrorr Ú SyntaxErrorr ÚnpÚrandomÚrandnÚlistr Úto_hdfr
|
)Útmp_pathr r r r ÚpathrB r r r Útest_invalid_termsW sN
|
ÿÿ*
|
ý
|
ýÿrM c
|
C sÞ t tj dd¡}t dtj d¡i}t dtj d¡i}t dtj d¡i}t ddtftj d¡i}t| d}dt d¡ }| ||¡ ||||fD ]4}t
|
d ¡} t j t | d
|
| ||¡ W 5 Q R X qW 5 Q R X d S ) Nr@ r# r" ©r# r$ )Ú1r$ rO r$ Zdf_z?cannot match existing table structure for [0] on appending datar )r rF rG rH Úobjectr r Zrandsr0 r1 r2 r r rD )
|
r r r3 Zdf3Zdf4Zdf5r ÚnameÚdr r r r Ú7test_append_with_diff_col_name_types_raises_value_error s
|
ÿrS c
|
C sh t tj dd¡tdtdd}t | ¡4}d}tjt |d |j
|
|dd d
|
W 5 Q R X W 5 Q R X d S ) Nr& r' ÚabcdÚABCDE©r+ r z)complib only supports \[.*\] compression.r r Zfoolib)Zcomplib) r rF rG ÚrandrI r Z ensure_cleanr r rD rJ )r r rL r r r r Útest_invalid_complib§ s
|
rX ÚidxZ2019ÚDr% ÚUTC)Úfreqr* ÚtzÚabcc C sP t | | g¡}td|dgd}|| }tjtdd | |d¡ W 5 Q R X d S )Nr r" rV zSaving a MultiIndexr r )r Z from_arraysr r r ÚNotImplementedErrorrJ )rY rK r Úmir rL r r r Ú&test_to_hdf_multiindex_extension_dtype¯ s
|
ra c C s6 | ddd}d}t jt|d t| W 5 Q R X d S )NÚioÚdataz"legacy_hdf/incompatible_dataset.h5z_Dataset\(s\) incompatible with Pandas data types, not table, or no datasets found in HDF5 file.r )r r rD r
|
)ÚdatapathZ data_pathÚmessager r r Ú test_unsuppored_hdf_file_error¿ s
|
ÿrf c C s t tj dd¡tdtdd}|| }d}tjt|d t|d W 5 Q R X | |d ¡ t
|
|d
|
d }| ¡ d }tjt|d t|d W 5 Q R X d S ) Nr& r' rT rU rV zFile [\S]* does not existr Úkeyr Úr)Úmodez&The HDFStore must be open for reading.) r rF rG rW rI r r ÚOSErrorr
|
rJ r Úclose)r rK r rL r r r r r Útest_read_hdf_errorsË s rl c C s0 d} t jt| d ttdd W 5 Q R X d S )Nz5Support for generic buffers has not been implemented.r ó r )r r r_ r
|
r )r r r r Ú#test_read_hdf_generic_buffer_errorsÜ s rn Ú bad_versionrN )r# Z12Z123c C s. d}t jt|d td| d W 5 Q R X d S )Nz5Version is incorrect, expected sequence of 3 integersr Zvalues_block_0)Úversion)r r rD r )ro r r r r Ú)test_maybe_adjust_name_bad_version_raisesâ s rq )'r: rb r r1 Úwarningsr ÚnumpyrF r Zpandasr r r r r r r r
|
Z pandas.tests.io.pytables.commonr Zpandas.io.pytablesr r ÚmarkZ
|
single_cpuZ
|
pytestmarkr! r4 r= rM rS rX Z parametrizerI ra rf rl rn rq r r r r Ú<module> s6 $ =
|
þþ
|
|
