Allow RuntimeError in test_boolean_assignment_value_mismatch

Author: honno

torch_np/tests/numpy_tests/core/test_indexing.py

@@ -241,9 +241,9 @@ def test_boolean_assignment_value_mismatch(self):
         def f(a, v):
             a[a > -1] = v
 
-        assert_raises(ValueError, f, a, [])
-        assert_raises(ValueError, f, a, [1, 2, 3])
-        assert_raises(ValueError, f, a[:1], [1, 2, 3])
+        assert_raises((ValueError, RuntimeError), f, a, [])
+        assert_raises((ValueError, RuntimeError), f, a, [1, 2, 3])
+        assert_raises((ValueError, RuntimeError), f, a[:1], [1, 2, 3])
 
     def test_boolean_assignment_needs_api(self):
         # See also gh-7666