def test_wrap_multilayer(self):
# not fully supported yet, but let's check that it's reasonable
rows = ["yes", "no", "maybe"]
cols = [("animal", "cat"), ("animal", "armadillo", ("person", "matt"))]
cols = pd.MultiIndex.from_tuples(cols)
df = pd.DataFrame(np.zeros((3, 2)), rows, cols)
df = TypedDfs.wrap(df)
assert df.column_names() == [
("animal", "cat", np.nan),
("animal", "armadillo", ("person", "matt")),
]
def test_wrap(self):
df = pd.DataFrame({})
df2 = TypedDfs.wrap(df)
assert not isinstance(df, AbsDf)
assert isinstance(df2, BaseDf)