Skip to content

Commit 7bf75b0

Browse files
TST : add test for groupby aggregation dtype (#43915)
1 parent e2d0288 commit 7bf75b0

File tree

1 file changed

+22
-0
lines changed

1 file changed

+22
-0
lines changed

pandas/tests/groupby/aggregate/test_aggregate.py

+22
Original file line numberDiff line numberDiff line change
@@ -1327,6 +1327,28 @@ def func(ser):
13271327
tm.assert_frame_equal(res, expected)
13281328

13291329

1330+
def test_groupby_aggregate_directory(reduction_func):
1331+
# GH#32793
1332+
if reduction_func in ["corrwith", "nth"]:
1333+
return None
1334+
1335+
obj = DataFrame([[0, 1], [0, np.nan]])
1336+
1337+
result_reduced_series = obj.groupby(0).agg(reduction_func)
1338+
result_reduced_frame = obj.groupby(0).agg({1: reduction_func})
1339+
1340+
if reduction_func in ["size", "ngroup"]:
1341+
# names are different: None / 1
1342+
tm.assert_series_equal(
1343+
result_reduced_series, result_reduced_frame[1], check_names=False
1344+
)
1345+
else:
1346+
tm.assert_frame_equal(result_reduced_series, result_reduced_frame)
1347+
tm.assert_series_equal(
1348+
result_reduced_series.dtypes, result_reduced_frame.dtypes
1349+
)
1350+
1351+
13301352
def test_group_mean_timedelta_nat():
13311353
# GH43132
13321354
data = Series(["1 day", "3 days", "NaT"], dtype="timedelta64[ns]")

0 commit comments

Comments
 (0)