U md@sXddlZddlmZmZmZddlmZmZej j Z ddZ ddZ dd Zd d ZdS) N) DataFrameHDFStore_testing)ensure_clean_storetablesc Cszt|h}t|d<t|d<t|d<t|dks>tdddh}t||ks\tt||ksltW5QRXdS)Nabcz/az/bz/c) rtmZmakeTimeSeriesZmakeStringSeriesZ makeDataFramelenAssertionErrorsetkeys) setup_pathstoreexpectedr[/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/pandas/tests/io/pytables/test_keys.py test_keyss     rc Cs&Gdddtj}Gdddtj}Gdddtj}||}tj|ddB}|d d }||d |d ||d |d||d|dW5QRXt|}t|jdddkstdddh} t |jdd| kstt |jddt kst| D] } | | } t| j dkstqW5QRXdS)Nc@seZdZeZdS)z$test_non_pandas_keys..Table1N)__name__ __module__ __qualname__r Float32ColZvalue1rrrrTable1src@seZdZeZdS)z$test_non_pandas_keys..Table2N)rrrrrZvalue2rrrrTable2 src@seZdZeZdS)z$test_non_pandas_keys..Table3N)rrrrrZvalue3rrrrTable3#srw)mode/groupZtable1zTable 1Ztable2zTable 2Ztable3zTable 3nativeincluder z /group/table1z /group/table2z /group/table3pandas) rZ IsDescription open_fileZ create_groupZ create_tablerr rr rgetcolumns) Ztmp_pathrrrrpathZh5filer rrnamedfrrrtest_non_pandas_keyss"    r,c Cs>t|,}tjtdd|jddW5QRXW5QRXdS)Nz@`include` should be either 'pandas' or 'native' but is 'illegal')matchillegalr")rpytestZraises ValueErrorr)rrrrr'test_keys_illegal_include_keyword_value6s  r1c Csrt|`}ttdtdd}|d||dgks>t|j|jjdd|dgksdtW5QRXdS)N)ABr+z/dfsymlink) rrrangeputrr Z_handleZcreate_soft_linkroot)rrr+rrrtest_keys_ignore_hdf_softlink?s   r9)r/r$rrrr Zpandas.tests.io.pytables.commonrrmarkZ single_cpuZ pytestmarkrr,r1r9rrrrs