From ca71f7bcbcb4fbf285e3193b7de17dcbc1f6806c Mon Sep 17 00:00:00 2001 From: Umang Date: Sun, 18 Aug 2019 12:41:07 -0500 Subject: [PATCH 1/2] testing groupby.transform('sum') --- pandas/tests/groupby/test_transform.py | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) diff --git a/pandas/tests/groupby/test_transform.py b/pandas/tests/groupby/test_transform.py index d3972e6ba9008..66c10ee163faa 100644 --- a/pandas/tests/groupby/test_transform.py +++ b/pandas/tests/groupby/test_transform.py @@ -1074,3 +1074,25 @@ def test_transform_lambda_with_datetimetz(): name="time", ) assert_series_equal(result, expected) + + +@pytest.mark.parametrize( + "input_df, expected_df", + [ + ( + DataFrame( + { + "A": [121, 121, 121, 121, 231, 231, 676], + "B": [1, 2, np.nan, 3, 3, np.nan, 4], + } + ), + 1, + ), + (DataFrame({"B": [6.0, 6.0, 6.0, 6.0, 3.0, 3.0, 4.0]}), 1), + ], +) +def test_groupby_transform_sum(input_df, expected_df): + # GH 27905 - Test sum in groupby.transform + df_transform = input_df.groupby("A")["B"].transform("sum") + df_transform = df_transform.to_frame() + assert all(df_transform == expected_df) From ec8e5aa0cc6bf6f3257be960c0d7ef2e4e672761 Mon Sep 17 00:00:00 2001 From: Umang Date: Sun, 18 Aug 2019 13:17:25 -0500 Subject: [PATCH 2/2] error fix --- pandas/tests/groupby/test_transform.py | 49 +++++++++++++++----------- 1 file changed, 29 insertions(+), 20 deletions(-) diff --git a/pandas/tests/groupby/test_transform.py b/pandas/tests/groupby/test_transform.py index 66c10ee163faa..f5eb58112fda3 100644 --- a/pandas/tests/groupby/test_transform.py +++ b/pandas/tests/groupby/test_transform.py @@ -1076,23 +1076,32 @@ def test_transform_lambda_with_datetimetz(): assert_series_equal(result, expected) -@pytest.mark.parametrize( - "input_df, expected_df", - [ - ( - DataFrame( - { - "A": [121, 121, 121, 121, 231, 231, 676], - "B": [1, 2, np.nan, 3, 3, np.nan, 4], - } - ), - 1, - ), - (DataFrame({"B": [6.0, 6.0, 6.0, 6.0, 3.0, 3.0, 4.0]}), 1), - ], -) -def test_groupby_transform_sum(input_df, expected_df): - # GH 27905 - Test sum in groupby.transform - df_transform = input_df.groupby("A")["B"].transform("sum") - df_transform = df_transform.to_frame() - assert all(df_transform == expected_df) +# @pytest.mark.parametrize( +# "input_df, expected_df", +# [ +# ( +# DataFrame( +# { +# "A": [121, 121, 121, 121, 231, 231, 676], +# "B": [1, 2, np.nan, 3, 3, np.nan, 4], +# } +# ), +# 1, +# ), +# (DataFrame({"B": [6.0, 6.0, 6.0, 6.0, 3.0, 3.0, 4.0]}), 1), +# ], +# ) +# def test_groupby_transform_sum(input_df, expected_df): +# # GH 27905 - Test sum in groupby.transform +# df_transform = input_df.groupby("A")["B"].transform("sum") +# df_transform = df_transform.to_frame() +# assert all(df_transform == expected_df) + + +def test_groupby_transform_sum(): + input_df = DataFrame( + {"A": [121, 121, 121, 121, 231, 231, 676], "B": [1, 2, np.nan, 3, 3, np.nan, 4]} + ) + expected = Series([6.0, 6.0, 6.0, 6.0, 3.0, 3.0, 4.0]) + result = input_df.groupby("A")["B"].transform("sum") + assert_series_equal(result, expected)