Skip to content

Commit 2a3f814

Browse files
committed
Add test
1 parent fddaa99 commit 2a3f814

File tree

1 file changed

+17
-0
lines changed

1 file changed

+17
-0
lines changed

pandas/tests/groupby/test_function.py

+17
Original file line numberDiff line numberDiff line change
@@ -1636,3 +1636,20 @@ def test_apply_to_nullable_integer_returns_float(values, function):
16361636
result = groups.agg([function])
16371637
expected.columns = MultiIndex.from_tuples([("b", function)])
16381638
tm.assert_frame_equal(result, expected)
1639+
1640+
1641+
def test_groupby_sum_below_mincount_nullable_integer():
1642+
# https://github.com/pandas-dev/pandas/issues/32861
1643+
df = pd.DataFrame({"a": [0, 1, 2], "b": [0, 1, 2], "c": [0, 1, 2]}, dtype="Int64")
1644+
grouped = df.groupby("a")
1645+
idx = pd.Index([0, 1, 2], dtype=object, name="a")
1646+
1647+
result = grouped["b"].sum(min_count=2)
1648+
expected = pd.Series([np.nan] * 3, index=idx, name="b")
1649+
tm.assert_series_equal(result, expected)
1650+
1651+
result = grouped.sum(min_count=2)
1652+
expected = pd.DataFrame(
1653+
{"b": [pd.NA] * 3, "c": [pd.NA] * 3}, dtype="Int64", index=idx
1654+
)
1655+
tm.assert_frame_equal(result, expected)

0 commit comments

Comments
 (0)