Skip to content

Commit 14727ac

Browse files
committed
fixup! Add test for DataFrame.agg with *args, **kwargs
1 parent e5d8880 commit 14727ac

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

pandas/tests/frame/apply/test_frame_apply.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -1483,13 +1483,13 @@ def f(x, a, b, c=3):
14831483
df = pd.DataFrame([[1, 2], [3, 4]])
14841484

14851485
if axis == 0:
1486-
expected = pd.DataFrame([5.0, 7.0])
1486+
expected = pd.Series([5.0, 7.0])
14871487
else:
1488-
expected = pd.DataFrame([4.0, 8.0])
1488+
expected = pd.Series([4.0, 8.0])
14891489

14901490
result = df.agg(f, axis, *args, **kwargs)
14911491

1492-
tm.assert_frame_equal(result, expected)
1492+
tm.assert_series_equal(result, expected)
14931493

14941494
@pytest.mark.parametrize("num_cols", [2, 3, 5])
14951495
def test_frequency_is_original(self, num_cols):

0 commit comments

Comments
 (0)