Skip to content

Commit 86eaed9

Browse files
committed
port test from pandas-dev#34312
1 parent 5c85d1a commit 86eaed9

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

pandas/tests/frame/test_arithmetic.py

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1499,3 +1499,18 @@ def test_pow_nan_with_zero():
14991499

15001500
result = left["A"] ** right["A"]
15011501
tm.assert_series_equal(result, expected["A"])
1502+
1503+
1504+
def test_dataframe_series_extension_dtypes():
1505+
# https://github.com/pandas-dev/pandas/issues/34311
1506+
df = pd.DataFrame(np.random.randint(0, 100, (10, 3)), columns=["a", "b", "c"])
1507+
ser = pd.Series([1, 2, 3], index=["a", "b", "c"])
1508+
1509+
expected = df.to_numpy("int64") + ser.to_numpy("int64").reshape(-1, 3)
1510+
expected = pd.DataFrame(expected, columns=df.columns, dtype="Int64")
1511+
1512+
df_ea = df.astype("Int64")
1513+
result = df_ea + ser
1514+
tm.assert_frame_equal(result, expected)
1515+
result = df_ea + ser.astype("Int64")
1516+
tm.assert_frame_equal(result, expected)

0 commit comments

Comments
 (0)