U
|
ý°dE ã @ sX d dl Z d dlmZmZmZ d dlmZmZ e j j
|
Z dd Z dd Z dd Zd
|
d ZdS ) é N)Ú DataFrameÚHDFStoreÚ_testing)Úensure_clean_storeÚtablesc C sz t | h}t ¡ |d< t ¡ |d< t ¡ |d< t|dks>tdddh}t| ¡ |ks\tt||ksltW 5 Q R X d S )NÚaÚbÚcé z/az/bz/c) r ÚtmZmakeTimeSeriesZmakeStringSeriesZ makeDataFrameÚlenÚAssertionErrorÚsetÚkeys)Ú
|
setup_pathÚstoreÚexpected© r úYd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/io/pytables/test_keys.pyÚ test_keys s
|
|
r c C s& G dd dt j}G dd dt j}G dd dt j}| | }t j|ddB}| d d
|
¡}| |d |d ¡ | |d |d¡ | |d|d¡ W 5 Q R X t|}t|jdddks¶tdddh} t |jdd| ksØtt |jddt ksòt| D ] }
|
|
|
|
|
¡} t| j dksötqöW 5 Q R X d S )Nc @ s e Zd Ze ¡ ZdS )z$test_non_pandas_keys.<locals>.Table1N)Ú__name__Ú
|
__module__Ú __qualname__r Ú
|
Float32ColZvalue1r r r r ÚTable1 s r c @ s e Zd Ze ¡ ZdS )z$test_non_pandas_keys.<locals>.Table2N)r r r r r Zvalue2r r r r ÚTable2 s r c @ s e Zd Ze ¡ ZdS )z$test_non_pandas_keys.<locals>.Table3N)r r r r r Zvalue3r r r r ÚTable3# s r Úw)Úmodeú/ÚgroupZtable1zTable 1Ztable2zTable 2Ztable3zTable 3Znative©Úincluder
|
z /group/table1z /group/table2z /group/table3Úpandasé ) r Z IsDescriptionÚ open_fileZ create_groupZ create_tabler r r r r ÚgetÚcolumns) Ztmp_pathr r r r ÚpathZh5filer r r ÚnameÚdfr r r Útest_non_pandas_keys s"
|
|
|
r+ c
|
C s> t | ,}tjtdd |jdd W 5 Q R X W 5 Q R X d S )Nz@`include` should be either 'pandas' or 'native' but is 'illegal')ÚmatchÚillegalr! )r ÚpytestZraisesÚ
|
ValueErrorr )r r r r r Ú'test_keys_illegal_include_keyword_value6 s
|
þr0 c C sr t | `}ttdtdd}| d|¡ | ¡ dgks>t|j |jjdd¡ | ¡ dgksdtW 5 Q R X d S )Né )ÚAÚBr* z/dfÚsymlink) r r ÚrangeÚputr r Ú_handleZcreate_soft_linkÚroot)r r r* r r r Ú test_keys_ignore_hdf_softlink? s
|
r9 )r. r# r r r r Z pandas.tests.io.pytables.commonr r ÚmarkZ
|
single_cpuZ
|
pytestmarkr r+ r0 r9 r r r r Ú<module> s
|
