U
|
ý°d
ã @ sX d dl Zd dlZd dlmZmZ d dlmZ d dl m
|
Z
|
m Z e d¡ G dd dZ dS )é N)Ú DataFrameÚSeries)ÚHDFStoreÚread_hdfZtablesc @ s e Zd Zdd Zdd ZdS )ÚTestHDFStoreSubclassc C s ddgddgd}t j|tjd}t|tjd}|d }| |d¡ t|d}t ||¡ |d }t|}| d|¡ W 5 Q R X t|d}t ||¡ d S ) Né é é é )ÚaÚb©Zdtypeútemp.h5Zdf)
|
ÚtmZSubclassedDataFrameÚnpÚintpr Úto_hdfr Zassert_frame_equalr Úput)ÚselfÚtmp_pathÚdataZsdfÚexpectedÚpathÚresultÚstore© r ú]d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/io/pytables/test_subclass.pyÚ%test_supported_for_subclass_dataframe s
|
|
|
z:TestHDFStoreSubclass.test_supported_for_subclass_dataframec C s dddg}t j|tjd}t|tjd}|d }| |d¡ t|d}t ||¡ |d }t|}| d|¡ W 5 Q R X t|d}t ||¡ d S )Nr r r r r Zser)
|
r ZSubclassedSeriesr r r r r Zassert_series_equalr r )r r r Zsserr r r r r r r Ú"test_supported_for_subclass_series% s
|
|
|
|
z7TestHDFStoreSubclass.test_supported_for_subclass_seriesN)Ú__name__Ú
|
__module__Ú __qualname__r r r r r r r s r )Únumpyr ZpytestZpandasr r Zpandas._testingZ_testingr Zpandas.io.pytablesr r Z importorskipr r r r r Ú<module> s
|
|
