docker-python/test_pytables.py at master · romazu/docker-python · GitHub

Sitelet https://github.com/romazu/docker-python/blob/master/test_pytables.py

13 lines (9 loc) · 346 Bytes

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)