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"]