diff --git a/Dockerfile.tmpl b/Dockerfile.tmpl index 5f0815bd..15fffc47 100644 --- a/Dockerfile.tmpl +++ b/Dockerfile.tmpl @@ -513,6 +513,7 @@ RUN pip install flashtext && \ pip install ipympl==0.7.0 && \ pip install pandarallel && \ pip install onnx && \ + pip install tables && \ /tmp/clean-layer.sh # Download base easyocr models. diff --git a/test_pytables.py b/test_pytables.py new file mode 100644 index 00000000..20ec0b7b --- /dev/null +++ b/test_pytables.py @@ -0,0 +1,13 @@ +import unittest + +import pandas as pd +from pandas.testing import assert_frame_equal + + +class TestPandasPytables(unittest.TestCase): + + def test_rw_hd5(self): + want = pd.DataFrame([[1, 1.0, 'a']], columns=['x', 'y', 'z']) + want.to_hdf('./want.h5', 'data') + got = pd.read_hdf('./want.h5') + assert_frame_equal(want, got)