Skip to content

Commit 1e0335f

Browse files
TST: additional regression cases for slicing blockwise op (GH34421) (#34434)
1 parent 86f1cee commit 1e0335f

File tree

1 file changed

+12
-3
lines changed

1 file changed

+12
-3
lines changed

pandas/tests/frame/test_arithmetic.py

+12-3
Original file line numberDiff line numberDiff line change
@@ -1523,7 +1523,16 @@ def test_dataframe_blockwise_slicelike():
15231523
df2 = df1.copy()
15241524
df2.iloc[0, [1, 3, 7]] = np.nan
15251525

1526-
res = df1 + df2
1526+
df3 = df1.copy()
1527+
df3.iloc[0, [5]] = np.nan
15271528

1528-
expected = pd.DataFrame({i: df1[i] + df2[i] for i in df1.columns})
1529-
tm.assert_frame_equal(res, expected)
1529+
df4 = df1.copy()
1530+
df4.iloc[0, np.arange(2, 5)] = np.nan
1531+
df5 = df1.copy()
1532+
df5.iloc[0, np.arange(4, 7)] = np.nan
1533+
1534+
for left, right in [(df1, df2), (df2, df3), (df4, df5)]:
1535+
res = left + right
1536+
1537+
expected = pd.DataFrame({i: left[i] + right[i] for i in left.columns})
1538+
tm.assert_frame_equal(res, expected)

0 commit comments

Comments
 (0)