mirrors/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-08-04 00:48:58 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 8

bpo-41773: Raise exception for non-finite weights in random.choices(). (GH-22441)

Browse source
This commit is contained in:
Ram Rachum 2020-09-29 04:32:10 +03:00 • committed by GitHub
parent e8acc355d4
commit b0dfc75816
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
4 changed files with 23 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

16
Lib/test/test_random.py
View file

@ -324,6 +324,22 @@ class TestBasicOps:
with self.assertRaises(ValueError):
self.gen.choices('AB', [0.0, 0.0])
def test_choices_negative_total(self):
with self.assertRaises(ValueError):
self.gen.choices('ABC', [3, -5, 1])
def test_choices_infinite_total(self):
with self.assertRaises(ValueError):
self.gen.choices('A', [float('inf')])
with self.assertRaises(ValueError):
self.gen.choices('AB', [0.0, float('inf')])
with self.assertRaises(ValueError):
self.gen.choices('AB', [-float('inf'), 123])
with self.assertRaises(ValueError):
self.gen.choices('AB', [0.0, float('nan')])
with self.assertRaises(ValueError):
self.gen.choices('AB', [float('-inf'), float('inf')])
def test_gauss(self):
# Ensure that the seed() method initializes all the hidden state. In
# particular, through 2.2.1 it failed to reset a piece of state used