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)