def test_pivottable(self):
df = superstore()
pt = pd.pivot_table(df, values='Discount', index=['Country', 'Region'], columns='Category')
table = Table(pt)
columns = table.columns()
assert "Country" in columns
assert "Region" in columns
def test_rowpivots(self):
df = superstore(100)
df_pivoted = df.set_index(['Country', 'Region'])
table = Table(df_pivoted)
columns = table.columns()
assert table.size() == 100
assert "Country" in columns
assert "Region" in columns
def test_table_columnar(self):
data = {"a": [1, 2, 3], "b": [4, 5, 6]}
tbl = Table(data)
assert tbl.columns() == ["a", "b"]
assert tbl.size() == 3
assert tbl.schema() == {
"a": int,
"b": int
}
def test_table_dataframe_selective_column_order(self):
d = [{
"a": 1,
"b": 2,
"c": 3,
"d": 4
}, {
"a": 3,
"b": 4,
"c": 5,
"d": 6
}]
data = pd.DataFrame(d, columns=["b", "c", "a"])
tbl = Table(data)
assert tbl.size() == 2
assert tbl.columns() == ["index", "b", "c", "a"]
