From 097e93234bc1747449492da8d08b101b64b8ebdc Mon Sep 17 00:00:00 2001 From: Joris Van den Bossche Date: Thu, 28 May 2020 20:15:47 +0200 Subject: [PATCH] TST: additional regression cases for slicing blockwise op (GH34421) --- pandas/tests/frame/test_arithmetic.py | 15 ++++++++++++--- 1 file changed, 12 insertions(+), 3 deletions(-) diff --git a/pandas/tests/frame/test_arithmetic.py b/pandas/tests/frame/test_arithmetic.py index 993b644c6993b..e7b7f3e524d44 100644 --- a/pandas/tests/frame/test_arithmetic.py +++ b/pandas/tests/frame/test_arithmetic.py @@ -1523,7 +1523,16 @@ def test_dataframe_blockwise_slicelike(): df2 = df1.copy() df2.iloc[0, [1, 3, 7]] = np.nan - res = df1 + df2 + df3 = df1.copy() + df3.iloc[0, [5]] = np.nan - expected = pd.DataFrame({i: df1[i] + df2[i] for i in df1.columns}) - tm.assert_frame_equal(res, expected) + df4 = df1.copy() + df4.iloc[0, np.arange(2, 5)] = np.nan + df5 = df1.copy() + df5.iloc[0, np.arange(4, 7)] = np.nan + + for left, right in [(df1, df2), (df2, df3), (df4, df5)]: + res = left + right + + expected = pd.DataFrame({i: left[i] + right[i] for i in left.columns}) + tm.assert_frame_equal(res, expected)