diff options
| author | Raymond Hettinger <python@rcn.com> | 2010-09-07 10:06:56 (GMT) |
|---|---|---|
| committer | Raymond Hettinger <python@rcn.com> | 2010-09-07 10:06:56 (GMT) |
| commit | dc4872eefe2531e6627aa09818e95372a50e3b45 (patch) | |
| tree | 0b61130e5de9917744671797d0ffab1b2036898c /Lib/test/test_random.py | |
| parent | c324697bac35d7151c93f3c64009b3ec17a053e7 (diff) | |
| download | cpython-dc4872eefe2531e6627aa09818e95372a50e3b45.zip cpython-dc4872eefe2531e6627aa09818e95372a50e3b45.tar.gz cpython-dc4872eefe2531e6627aa09818e95372a50e3b45.tar.bz2 | |
Fix corner case for Random.choice() and add tests.
Diffstat (limited to 'Lib/test/test_random.py')
| -rw-r--r-- | Lib/test/test_random.py | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/Lib/test/test_random.py b/Lib/test/test_random.py index f5c0030..08edead 100644 --- a/Lib/test/test_random.py +++ b/Lib/test/test_random.py @@ -42,6 +42,13 @@ class TestBasicOps(unittest.TestCase): self.assertRaises(TypeError, self.gen.seed, 1, 2, 3, 4) self.assertRaises(TypeError, type(self.gen), []) + def test_choice(self): + choice = self.gen.choice + with self.assertRaises(IndexError): + choice([]) + self.assertEqual(choice([50]), 50) + self.assertIn(choice([25, 75]), [25, 75]) + def test_sample(self): # For the entire allowable range of 0 <= k <= N, validate that # the sample is of the correct length and contains only unique items |