Merge pull request #397 from certik/fix2189

certik · certik · commit 7d30cada0e3e · 2012-09-02T09:35:43.000-07:00
BUG: Add a test for Ticket #2189
diff --git a/numpy/core/tests/test_regression.py b/numpy/core/tests/test_regression.py
@@ -1737,5 +1737,15 @@ def test_refcount_error_in_clip(self):
         # Check the final string:
         assert_(y == "[0 0]")
 
+    def test_searchsorted_wrong_dtype(self):
+        # Ticket #2189, it used to segfault, so we check that it raises the
+        # proper exception.
+        a = np.array([('a', 1)], dtype='S1, int')
+        assert_raises(TypeError, np.searchsorted, a, 1.2)
+        # Ticket #2066, similar problem:
+        dtype = np.format_parser(['i4', 'i4'], [], [])
+        a = np.recarray((2, ), dtype)
+        assert_raises(TypeError, np.searchsorted, a, 1)
+
 if __name__ == "__main__":
     run_module_suite()
