Skip to content

Commit 6d36c45

Browse files
Add a test for agg std (pandas-dev#52371)
* Add a test for agg std * move the test as requested and added one more test for df.agg([np.std]) * fix second test * fix format * Update test_aggregate.py * Update test_frame_apply.py
1 parent a174f42 commit 6d36c45

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

pandas/tests/apply/test_frame_apply.py

+12
Original file line numberDiff line numberDiff line change
@@ -1665,3 +1665,15 @@ def foo2(x, b=2, c=0):
16651665
columns=MultiIndex.from_tuples([("x", "foo1"), ("x", "foo2")]),
16661666
)
16671667
tm.assert_frame_equal(result, expected)
1668+
1669+
1670+
def test_agg_std():
1671+
df = DataFrame(np.arange(6).reshape(3, 2), columns=["A", "B"])
1672+
1673+
result = df.agg(np.std)
1674+
expected = Series({"A": 2.0, "B": 2.0}, dtype=float)
1675+
tm.assert_series_equal(result, expected)
1676+
1677+
result = df.agg([np.std])
1678+
expected = DataFrame({"A": 2.0, "B": 2.0}, index=["std"])
1679+
tm.assert_frame_equal(result, expected)

0 commit comments

Comments
 (0)