def test_load(self):
dfs = DataFrameStorage(self.orig_filepath)
df = dfs.load()
# simple unittest assertions
self.assertEqual(len(df.index), 22283)
self.assertEqual(len(df.columns), 13)
def test_store_load(self):
# source file - some expression set assay table
orig_df = pandas.read_table(self.orig_filepath, sep=" ",
compression="gzip", index_col=0)
dfs = DataFrameStorage(self.test_filepath)
dfs.store(orig_df)
restored_df = dfs.load()
# using custom assert function,
# because ==operator for DataFrame doesn't return boolean
tm.assert_frame_equal(orig_df, restored_df)
