def test_1lev():
grouped_df = pandas.DataFrame(data={
"lev1": ["A", "A", "B", "B"],
"lev2": ["X", "Y", "X", "Y"],
})
levels = get_agg_series(grouped_df, [["lev1"]])
expected_levels = ["A", "B"]
assert sorted(levels) == sorted(expected_levels)
levels = get_agg_series(grouped_df, [["lev2"]])
expected_levels = ["X", "Y"]
assert sorted(levels) == sorted(expected_levels)
def test_parent_child():
grouped_df = pandas.DataFrame(data={
"lev1": ["A", "A", "B"],
"lev2": ["X", "Y", "Z"],
})
levels = get_agg_series(grouped_df, [["lev1", "lev2"]])
expected_levels = ["A_X", "A_Y", "B_Z"]
assert sorted(levels) == sorted(expected_levels)
def test_hierarchichal():
hier_df = pandas.DataFrame(data={
"lev1": ["A", "A", "A", "B", "B"],
"lev2": ["X", "Y", "Z", "X", "Y"],
})
levels = get_agg_series(hier_df, [["lev1"], ["lev1", "lev2"]])
expected_levels = ["A", "B", "A_X", "A_Y", "A_Z", "B_X", "B_Y"]
assert sorted(levels) == sorted(expected_levels)
