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

pandas-dev · Nov 21, 2017 · 683dab7 · 683dab7
1 parent a98dd0e
commit 683dab7
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)