refine error message of randint (#25613)

fix_copy_if_different
zhupengyang 5 years ago committed by GitHub
parent 9b46fe0440
commit cd933c0aa2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -57,6 +57,7 @@ class TestRandintOpError(unittest.TestCase):
self.assertRaises(TypeError, paddle.randint, 5, shape=np.array([2]))
self.assertRaises(TypeError, paddle.randint, 5, dtype='float32')
self.assertRaises(ValueError, paddle.randint, 5, 5)
self.assertRaises(ValueError, paddle.randint, -5)
class TestRandintOp_attr_tensorlist(OpTest):

@ -114,6 +114,10 @@ def randint(low=0, high=None, shape=[1], dtype=None, name=None):
"""
if high is None:
if low <= 0:
raise ValueError(
"If high is None, low must be greater than 0, but received low = {0}.".
format(low))
high = low
low = 0
if dtype is None:

Loading…
Cancel
Save