add exception check for numpy reshape (#16180)

apache · Sep 18, 2019 · 1e058a3 · 1e058a3
1 parent 995b477
commit 1e058a3
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/tests/python/unittest/test_exc_handling.py b/tests/python/unittest/test_exc_handling.py
@@ -190,6 +190,15 @@ def check_resize():
     assert_raises(MXNetError, check_resize)
 
 
+@with_seed()
+def test_np_reshape_exception():
+    a = mx.np.ones((10, 10))
+    a.reshape((-1,)).asnumpy()  # Check no-raise
+    assert_raises(MXNetError, lambda: a.reshape((1,)))
+    assert_raises(MXNetError, lambda: mx.np.reshape(a, (1,)))
+    assert_raises(MXNetError, lambda: mx.np.reshape(a, (-1, 3)))
+
+
 if __name__ == '__main__':
     import nose
     nose.runmodule()