Skip to content

Commit 520adf7

Browse files
committed
Address review comments
1 parent 5ab8581 commit 520adf7

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

pandas/tests/frame/test_indexing.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -542,15 +542,23 @@ def test_setitem_boolean(self):
542542
assert_frame_equal(df, expected)
543543

544544
def test_setitem_boolean_ndarary(self):
545+
# Test for issue #18582
546+
545547
df = self.frame.copy()
546548
mask = df > np.abs(df)
547549
expected = df.copy()
548550
expected.values[mask.values] = nan
551+
549552
# index with 2-d boolean ndarray
550553
actual = df.copy()
551554
actual[mask.values] = nan
552555
assert_frame_equal(actual, expected)
553556

557+
# index with boolean DataFrame
558+
actual = df.copy()
559+
actual[mask] = nan
560+
assert_frame_equal(actual, expected)
561+
554562
def test_setitem_cast(self):
555563
self.frame['D'] = self.frame['D'].astype('i8')
556564
assert self.frame['D'].dtype == np.int64

0 commit comments

Comments
 (0)