Skip to content

Commit 370d109

Browse files
committed
BUG-46369: CR on test
1 parent 4ee3184 commit 370d109

File tree

1 file changed

+6
-8
lines changed

1 file changed

+6
-8
lines changed

pandas/tests/groupby/test_apply.py

+6-8
Original file line numberDiff line numberDiff line change
@@ -1319,13 +1319,11 @@ def test_apply_str_with_args(df, args, kwargs):
13191319
tm.assert_frame_equal(result, expected)
13201320

13211321

1322-
def test_result_name_when_one_group():
1322+
@pytest.mark.parametrize("name", ["some_name", None])
1323+
def test_result_name_when_one_group(name):
13231324
# GH 46369
1324-
names = ("some_name", None)
1325+
ser = Series([1, 2], name=name)
1326+
result = ser.groupby(["a", "a"], group_keys=False).apply(lambda x: x)
1327+
expected = Series([1, 2], name=name)
13251328

1326-
for name in names:
1327-
ser = Series([0, 1], name=name)
1328-
result = ser.groupby(["a", "a"]).apply(lambda x: x).name
1329-
expected = name
1330-
1331-
assert result == expected
1329+
tm.assert_series_equal(result, expected)

0 commit comments

Comments
 (0)