34 lines
1.2 KiB
Python
34 lines
1.2 KiB
Python
|
import pandas as pd
|
||
|
|
import pandas._testing as tm
|
||
|
|
|
||
|
|
|
||
|
|
def test_relabel_no_duplicated_method():
|
||
|
|
# 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():
|
||
|
|
# 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)
|