diff options
| author | Serhiy Storchaka <storchaka@gmail.com> | 2015-07-24 06:05:59 (GMT) |
|---|---|---|
| committer | Serhiy Storchaka <storchaka@gmail.com> | 2015-07-24 06:05:59 (GMT) |
| commit | c19bb3279cbb42c84a24255db48ee2d385bb8167 (patch) | |
| tree | 72fa9802d7f4c2b0078fbdff5c078e06af3b06e0 /Lib/test | |
| parent | 5b718d7f4fc3ad28ffb774aa96324a81f5f3e742 (diff) | |
| parent | 178f0b6ddcae040c235188b49c1f6defc76588ca (diff) | |
| download | cpython-c19bb3279cbb42c84a24255db48ee2d385bb8167.zip cpython-c19bb3279cbb42c84a24255db48ee2d385bb8167.tar.gz cpython-c19bb3279cbb42c84a24255db48ee2d385bb8167.tar.bz2 | |
Issue #24620: Random.setstate() now validates the value of state last element.
Diffstat (limited to 'Lib/test')
| -rw-r--r-- | Lib/test/test_random.py | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/Lib/test/test_random.py b/Lib/test/test_random.py index e648045..4b5232f 100644 --- a/Lib/test/test_random.py +++ b/Lib/test/test_random.py @@ -338,6 +338,11 @@ class MersenneTwister_TestBasicOps(TestBasicOps, unittest.TestCase): self.assertRaises(TypeError, self.gen.setstate, (2, ('a',)*625, None)) # Last element s/b an int also self.assertRaises(TypeError, self.gen.setstate, (2, (0,)*624+('a',), None)) + # Last element s/b between 0 and 624 + with self.assertRaises((ValueError, OverflowError)): + self.gen.setstate((2, (1,)*624+(625,), None)) + with self.assertRaises((ValueError, OverflowError)): + self.gen.setstate((2, (1,)*624+(-1,), None)) # Little trick to make "tuple(x % (2**32) for x in internalstate)" # raise ValueError. I cannot think of a simple way to achieve this, so |