def test_csv_gz(self):
with tmpfile(".csv.gz") as path:
df = UntypedDf(sample_data())
df.to_csv(path)
df2 = UntypedDf.read_csv(path)
assert list(df2.index.names) == [None]
assert set(df2.columns) == {"abc", "123", "xyz"}
def test_untyped_read_write_csv(self):
with tmpfile(".csv") as path:
for indices in [None, "abc", ["abc", "xyz"]]:
df = UntypedDf(sample_data())
if indices is not None:
df = df.set_index(indices)
df.to_csv(path)
df2 = UntypedDf.read_csv(path)
assert list(df2.index.names) == [None]
assert set(df2.columns) == {"abc", "123", "xyz"}
