TST: Add tests for Index.putmask with an invalid mask

pandas-dev · Nov 25, 2017 · 86280fe · 86280fe
1 parent 2243252
commit 86280fe
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/pandas/tests/indexes/common.py b/pandas/tests/indexes/common.py
@@ -996,3 +996,16 @@ def test_searchsorted_monotonic(self, indices):
             # non-monotonic should raise.
             with pytest.raises(ValueError):
                 indices._searchsorted_monotonic(value, side='left')
+
+    def test_putmask_with_wrong_mask(self):
+        # GH18368
+        index = self.create_index()
+
+        with pytest.raises(ValueError):
+            index.putmask(np.ones(len(index) + 1, np.bool), 1)
+
+        with pytest.raises(ValueError):
+            index.putmask(np.ones(len(index) - 1, np.bool), 1)
+
+        with pytest.raises(ValueError):
+            index.putmask('foo', 1)