| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152 |
- import numpy as np
- import pytest
- from pandas import (
- DataFrame,
- Series,
- )
- import pandas._testing as tm
- from pandas.io.pytables import (
- HDFStore,
- read_hdf,
- )
- pytest.importorskip("tables")
- class TestHDFStoreSubclass:
- # GH 33748
- def test_supported_for_subclass_dataframe(self, tmp_path):
- data = {"a": [1, 2], "b": [3, 4]}
- sdf = tm.SubclassedDataFrame(data, dtype=np.intp)
- expected = DataFrame(data, dtype=np.intp)
- path = tmp_path / "temp.h5"
- sdf.to_hdf(path, "df")
- result = read_hdf(path, "df")
- tm.assert_frame_equal(result, expected)
- path = tmp_path / "temp.h5"
- with HDFStore(path) as store:
- store.put("df", sdf)
- result = read_hdf(path, "df")
- tm.assert_frame_equal(result, expected)
- def test_supported_for_subclass_series(self, tmp_path):
- data = [1, 2, 3]
- sser = tm.SubclassedSeries(data, dtype=np.intp)
- expected = Series(data, dtype=np.intp)
- path = tmp_path / "temp.h5"
- sser.to_hdf(path, "ser")
- result = read_hdf(path, "ser")
- tm.assert_series_equal(result, expected)
- path = tmp_path / "temp.h5"
- with HDFStore(path) as store:
- store.put("ser", sser)
- result = read_hdf(path, "ser")
- tm.assert_series_equal(result, expected)
|
