import pandas as pd
import pandas._testing as tm


class TestNamedAggregation:
    def test_relabel_no_duplicated_method(self):
        # this is to test there is no duplicated method used in agg
        df = pd.DataFrame({"A": [1, 2, 1, 2], "B": [1, 2, 3, 4]})

        result = df["A"].agg(foo="sum")
        expected = df["A"].agg({"foo": "sum"})
        tm.assert_series_equal(result, expected)

        result = df["B"].agg(foo="min", bar="max")
        expected = df["B"].agg({"foo": "min", "bar": "max"})
        tm.assert_series_equal(result, expected)

        result = df["B"].agg(foo=sum, bar=min, cat="max")
        expected = df["B"].agg({"foo": sum, "bar": min, "cat": "max"})
        tm.assert_series_equal(result, expected)

    def test_relabel_duplicated_method(self):
        # this is to test with nested renaming, duplicated method can be used
        # if they are assigned with different new names
        df = pd.DataFrame({"A": [1, 2, 1, 2], "B": [1, 2, 3, 4]})

        result = df["A"].agg(foo="sum", bar="sum")
        expected = pd.Series([6, 6], index=["foo", "bar"], name="A")
        tm.assert_series_equal(result, expected)

        result = df["B"].agg(foo=min, bar="min")
        expected = pd.Series([1, 1], index=["foo", "bar"], name="B")
        tm.assert_series_equal(result, expected)