This commit is contained in:
Raymond Hettinger 2016-12-29 22:55:03 -07:00
commit 0ce64323cf
3 changed files with 4 additions and 0 deletions

View File

@ -26,6 +26,8 @@ choice = _sysrand.choice
def randbelow(exclusive_upper_bound):
"""Return a random int in the range [0, n)."""
if exclusive_upper_bound <= 0:
raise ValueError("Upper bound must be positive.")
return _sysrand._randbelow(exclusive_upper_bound)
DEFAULT_ENTROPY = 32 # number of bytes to return by default

View File

@ -70,6 +70,7 @@ class Random_Tests(unittest.TestCase):
for i in range(2, 10):
self.assertIn(secrets.randbelow(i), range(i))
self.assertRaises(ValueError, secrets.randbelow, 0)
self.assertRaises(ValueError, secrets.randbelow, -1)
class Token_Tests(unittest.TestCase):

View File

@ -369,6 +369,7 @@ Daniel Dittmar
Josip Djolonga
Walter Dörwald
Jaromir Dolecek
Brendan Donegan
Ismail Donmez
Robert Donohue
Marcos Donolo