U
|
ý°d. ã @ sb d dl mZ d dlmZ d dlZd dlZd dlZd dlm Z d dl
|
m Z d dl Z d dl mZmZmZmZmZmZ d dlmZmZ d dlmZ d d lm Z ej j Z d
|
d Z d d Z!dd Z"dd Z#ej j$e dddd Z%dd Z&dd Z'dd Z(d d Z)d d Z*d d! Z+d"d# Z,d$d% Z-d&d' Z.d(d) Z/e 0d*¡d+d, Z1ej 2d-d.d/g¡d0d1 Z3d2d3 Z4dS )4é )Úclosing)ÚPathN)Ú Timestamp)Úis_platform_windows)Ú DataFrameÚHDFStoreÚIndexÚSeriesÚ_testingÚread_hdf)Ú _maybe_removeÚensure_clean_store)Ú_test_decorators)Ú TableIteratorc C s^ | | }t tdtdd}| |d¡ tjtdd t|d W 5 Q R X | |d¡ d S )Né ©ÚaÚbÚk1ú 'No object named k2 in the file'©ÚmatchÚk2)r ÚrangeÚto_hdfÚpytestÚraisesÚKeyErrorr )Útmp_pathÚ
|
setup_pathÚpathÚdf© r" úYd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/io/pytables/test_read.pyÚ!test_read_missing_key_close_store s r$ c
|
C sr | | }t tdtdd}| |d¡ t|d4}tjtdd t|d W 5 Q R X t|d W 5 Q R X d S )Nr r r Úrr r r )r r r r r r r r )r r r r! Ústorer" r" r# Ú"test_read_missing_key_opened_store- s r' c
|
C s t ¡ }t| ô}t|d tjtdd | dd¡ W 5 Q R X | d|¡ tjtt
|
d¡d | dd¡ W 5 Q R X t
|
d¡}tjt |d |jdddgd W 5 Q R X | dd¡}t |j t|jj ¡ t|tsâtt
|
d
|
¡}tjt|d | dd ¡ W 5 Q R X | ¡ }d|d < |jd |d gd | d d ¡}t |j |d j ¡ | ¡ }d|d < tj|j|jdd
d f< |jd|d gd | dd ¡}t |j |d j ¡ |jdd dd}t |j |d j dd
¡ |jdd dd}t |j |d j dd
¡ |jdd dd}t |j |d j d d
¡ |jdd dd}t |j |d j d d
¡ |jdd ddd}t |j |d j dd
¡ |jdd ddd}t |j |d j dd
¡ ttj d¡dd}|jd|dd |d }| dd¡}t ||¡ W 5 Q R X d S ) Nr! z No object named df in the filer Úindexz%'column [foo] not found in the table'Zfooz:select_column() got an unexpected keyword argument 'where'zindex>5)ÚwherezScolumn [values_block_0] can not be extracted individually; it is not data indexableZvalues_block_0ÚstringÚdf2)Z data_columnsé é Údf3r )Ústartéþÿÿÿ)Ústop)r/ r1 é
|
)ÚAÚBÚdf4Tr4 )ÚtmZmakeTimeDataFramer r r r r Z select_columnÚappendÚreÚescapeÚ TypeErrorZassert_almost_equalÚvaluesr r( Ú
|
isinstanceÚAssertionErrorÚ
|
ValueErrorÚcopyÚnpÚnanÚlocr ÚrandomZrandnÚassert_series_equal) r r! r& ÚmsgÚresultr+ r. r5 Úexpectedr" r" r# Útest_read_column< sd
|
ÿ
|
ÿ rH c C s: t | ddddd }|d }t|ts,tW 5 Q R X d S )NÚioÚdataz legacy_hdf/pytables_native.h5r% ©Úmodezdetector/readout)r r< r r= )Údatapathr& Zd2r" r" r# Útest_pytables_native_read s
|
ÿrN z#native2 read fails oddly on windows)Úreasonc C sD t | dddddd$}t| |d }t|ts6tW 5 Q R X d S )NrI rJ Ú
|
legacy_hdfzpytables_native2.h5r% rK Údetector)r Ústrr< r r= )rM r& Úd1r" r" r# Útest_pytables_native2_read s ÿrT c C sf t | ddddddF}| d¡}tdd d
|
d ggd d dd gtdgddd}t ||¡ W 5 Q R X d S )NrI rJ rP zlegacy_table_fixed_py2.h5r% rK r! é r é ÚDr3 r4 ÚCÚABCZ
|
INDEX_NAME)Úname©Úcolumnsr( )r Úselectr r r6 Úassert_frame_equal©rM r& rF rG r" r" r# Ú'test_legacy_table_fixed_format_read_py2 s ÿ
|
|
ýr` c C sZ t | dddddd:}| d¡}ttdggd gtd
|
gd }t ||¡ W 5 Q R X d S ) NrI rJ rP z"legacy_table_fixed_datetime_py2.h5r% rK r! z2020-02-06T18:00r3 Údater[ )r r] r r r r6 r^ r_ r" r" r# Ú0test_legacy_table_fixed_format_read_datetime_py2« s þ
|
|
ýrb c C sR t | dddddd}| d¡}W 5 Q R X tdd gd
|
d gd }t ||¡ d S ) NrI rJ rP zlegacy_table_py2.h5r% rK Útabler r r rV r )r r] r r6 r^ r_ r" r" r# Útest_legacy_table_read_py2» s ÿrd c C s t tj dd¡tdtdd}d|j_|jddd }| | }|j|d
|
d d t |d
|
}t
|
|d d &}t |d
|
}t ||¡ |j stW 5 Q R X d S )Nr, é ÚabcdÚABCDE©r( r\ ÚlettersÚET©Úkeysr7 r! ÚwrK r% )r r@ rC ÚrandÚlistr( rZ Ú set_indexr r r r6 r^ Zis_openr= )r r r! r Údirectr& Úindirectr" r" r# Útest_read_hdf_open_storeÇ s
|
|
rs c C sj t tj dd¡ddddgtdd}| | }|j|d d
|
d d t|d }|jjj d ksZt
|
t ||¡ d S ) Nr, re r rU r rV rg rh r! rm rc ©rL Úformat) r r@ rC rn ro r r r( Ú_dataÚbaser= r6 r^ )r r r! r r+ r" r" r# Ú test_read_hdf_index_not_view× s $
|
rx c C s¨ t tj dd¡tdtdd}d|j_|jddd }| | }|j|d
|
d d d t |d
|
}t |d
|
dd}t
|
|j , t |t stt| ¡ }t ||¡ W 5 Q R X d S )Nr, re rf rg rh ri rj Trk r! rm Útrt )Úiterator)r r@ rC rn ro r( rZ rp r r r r& r< r r= ÚnextÚ__iter__r6 r^ )r r r! r rq rz rr r" r" r# Útest_read_hdf_iteratorå s
|
r} c C s t tj dd¡tdtdd}| | }|j|ddd t|}t ||¡ |j|d dd d
|
}t j
|
t |d t| W 5 Q R X d S ) Nr, re rf rg rh r! r rK r+ ú?key must be provided when HDF5 file contains multiple datasets.r ) r r@ rC rn ro r r r6 r^ r r r> ©r r r! r ZrereadrE r" r" r# Útest_read_nokeyô s r c C s t tdttdddd}| | }|j|dddd t|}t ||¡ |j|d
|
ddd d }tj t
|
|d t| W 5 Q R X d S ) Nre ZabacdÚcategory)Zdtype)ÚiÚcr! r rc rt r+ r~ r ) r r r ro r r r6 r^ r r r> r r" r" r# Útest_read_nokey_table s r c C sH | | }t |}| ¡ t d¡}tjt|d t| W 5 Q R X d S )Nz]Dataset(s) incompatible with Pandas data types, not table, or no datasets found in HDF5 file.r )r Úcloser8 r9 r r r> r )r r r r& rE r" r" r# Útest_read_nokey_empty s ÿr c C sZ t tj dd¡tdtdd}| | }t|}|j|ddd t|d}t ||¡ d S ) Nr, re rf rg rh r! r rK )
|
r r@ rC rn ro r r r r6 r^ )r r rG ÚfilenameÚpath_objÚactualr" r" r# Útest_read_from_pathlib_path s ÿ
|
r zpy.pathc C sf ddl m} ttj dd¡tdtdd}| | }||}|j|dd d
|
t|d}t
|
||¡ d S ) Nr )Úlocalr, re rf rg rh r! r rK ) Zpy.pathr r r@ rC rn ro r r r6 r^ )r r Z LocalPathrG r r r r" r" r# Útest_read_from_py_localpath. s ÿ
|
r ru Zfixedrc c C s> t ¡ }| | }|j|d|d t|ddd}t ||¡ d S )NrJ )Úkeyru r% )r rL )r6 ZmakeFloatSeriesr r rD )r ru r Zseriesr rF r" r" r# Útest_read_hdf_series_mode_r? s
|
r c C s\ t dddgtjdddgddd }t| d
|
d d d dd}|d }t ||¡ W 5 Q R X d S )Ng ð?r rV z
|
2015-01-01z
|
2015-01-02z
|
2015-01-05r4 )Úfreq)r( rI rJ rP z*periodindex_0.20.1_x86_64_darwin_2.7.13.h5r% rK Úp)r ÚpdZ PeriodIndexr r6 r^ )rM rG r& rF r" r" r# Ú test_read_py2_hdf_file_in_py3K s þ ÿür )5Ú
|
contextlibr Úpathlibr r8 Únumpyr@ r Zpandas._libs.tslibsr Z pandas.compatr Zpandasr r r r r r
|
r6 r Z pandas.tests.io.pytables.commonr r Z pandas.utilr ÚtdZpandas.io.pytablesr ÚmarkZ
|
single_cpuZ
|
pytestmarkr$ r' rH rN ZskipifrT r` rb rd rs rx r} r r r r Z
|
skip_if_nor Z parametrizer r r" r" r" r# Ú<module> sB N
|
|
|
|
