Skip to content

Commit 50de326

Browse files
committed
fixed test
1 parent 5064217 commit 50de326

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

pandas/tests/extension/integer/test_integer.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -705,10 +705,11 @@ def test_groupby_mean_included():
705705
})
706706

707707
result = df.groupby("A").sum()
708+
# TODO(#22346): preserve Int64 dtype
708709
expected = pd.DataFrame({
709710
"B": np.array([1.0, 3.0]),
710-
"C": IntegerArray([1, 3], dtype="Int64")
711-
})
711+
"C": np.array([1, 3], dtype="int64")
712+
}, index=pd.Index(['a', 'b'], name='A'))
712713
tm.assert_frame_equal(result, expected)
713714

714715

0 commit comments

Comments
 (0)