cpython-lru_cache-weakref: Lib/test/test

annotate Lib/test/test_random.py @ 58765:2ba7677bfa95 legacy-trunk

testCopysign was defined twice in test_math; combine the definitions

author	Mark Dickinson <dickinsm@gmail.com>
date	Sat, 06 Feb 2010 23:11:25 +0000
parents	ff040880d1be
children	70e115c34f97

rev	line source
26829 f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	1 #!/usr/bin/env python
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	2
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	3 import unittest
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	4 import random
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	5 import time
29586 8b163370372d SF bug #759889: Pickling of Random is broken Raymond Hettinger <python@rcn.com> parents: 28944 diff changeset	6 import pickle
30649 8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	7 import warnings
51737 00f7be02ae1f Inline coefficients in gamma(). Add reflection formula. Add comments. Raymond Hettinger <python@rcn.com> parents: 49099 diff changeset	8 from math import log, exp, sqrt, pi, fsum, sin
58669 ff040880d1be #7092: Silence more py3k warnings. Patch by Florent Xicluna. Ezio Melotti <ezio.melotti@gmail.com> parents: 58460 diff changeset	9 from functools import reduce
24364 aa0d2d51fbde Get rid of relative imports in all unittests. Now anything that Barry Warsaw <barry@python.org> parents: 23380 diff changeset	10 from test import test_support
26829 f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	11
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	12 class TestBasicOps(unittest.TestCase):
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	13 # Superclass with tests common to all generators.
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	14 # Subclasses must arrange for self.gen to retrieve the Random instance
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	15 # to be tested.
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	16
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	17 def randomlist(self, n):
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	18 """Helper function to make a list of random numbers"""
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	19 return [self.gen.random() for i in xrange(n)]
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	20
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	21 def test_autoseed(self):
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	22 self.gen.seed()
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	23 state1 = self.gen.getstate()
30114 9f6e2dee1892 SF bug #778964: bad seed in python 2.3 random Raymond Hettinger <python@rcn.com> parents: 29586 diff changeset	24 time.sleep(0.1)
26829 f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	25 self.gen.seed() # diffent seeds at different times
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	26 state2 = self.gen.getstate()
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	27 self.assertNotEqual(state1, state2)
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	28
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	29 def test_saverestore(self):
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	30 N = 1000
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	31 self.gen.seed()
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	32 state = self.gen.getstate()
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	33 randseq = self.randomlist(N)
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	34 self.gen.setstate(state) # should regenerate the same sequence
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	35 self.assertEqual(randseq, self.randomlist(N))
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	36
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	37 def test_seedargs(self):
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	38 for arg in [None, 0, 0L, 1, 1L, -1, -1L, 1020, -(1020),
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	39 3.14, 1+2j, 'a', tuple('abc')]:
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	40 self.gen.seed(arg)
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	41 for arg in [range(3), dict(one=1)]:
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	42 self.assertRaises(TypeError, self.gen.seed, arg)
32562 a160b9669db1 Add some tests for corner cases. Raymond Hettinger <python@rcn.com> parents: 30929 diff changeset	43 self.assertRaises(TypeError, self.gen.seed, 1, 2)
a160b9669db1 Add some tests for corner cases. Raymond Hettinger <python@rcn.com> parents: 30929 diff changeset	44 self.assertRaises(TypeError, type(self.gen), [])
26829 f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	45
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	46 def test_jumpahead(self):
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	47 self.gen.seed()
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	48 state1 = self.gen.getstate()
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	49 self.gen.jumpahead(100)
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	50 state2 = self.gen.getstate() # s/b distinct from state1
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	51 self.assertNotEqual(state1, state2)
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	52 self.gen.jumpahead(100)
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	53 state3 = self.gen.getstate() # s/b distinct from state2
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	54 self.assertNotEqual(state2, state3)
23380 9be01b7b6c80 random.gauss() uses a piece of hidden state used by nothing else, Tim Peters <tim.peters@gmail.com> parents: diff changeset	55
56527 93350acb5425 Silence py3k warning claiming to affect the random module Antoine Pitrou <solipsis@pitrou.net> parents: 55002 diff changeset	56 # Silence py3k warnings
93350acb5425 Silence py3k warning claiming to affect the random module Antoine Pitrou <solipsis@pitrou.net> parents: 55002 diff changeset	57 with test_support.check_warnings():
93350acb5425 Silence py3k warning claiming to affect the random module Antoine Pitrou <solipsis@pitrou.net> parents: 55002 diff changeset	58 self.assertRaises(TypeError, self.gen.jumpahead) # needs an arg
93350acb5425 Silence py3k warning claiming to affect the random module Antoine Pitrou <solipsis@pitrou.net> parents: 55002 diff changeset	59 self.assertRaises(TypeError, self.gen.jumpahead, "ick") # wrong type
93350acb5425 Silence py3k warning claiming to affect the random module Antoine Pitrou <solipsis@pitrou.net> parents: 55002 diff changeset	60 self.assertRaises(TypeError, self.gen.jumpahead, 2.3) # wrong type
93350acb5425 Silence py3k warning claiming to affect the random module Antoine Pitrou <solipsis@pitrou.net> parents: 55002 diff changeset	61 self.assertRaises(TypeError, self.gen.jumpahead, 2, 3) # too many
26829 f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	62
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	63 def test_sample(self):
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	64 # For the entire allowable range of 0 <= k <= N, validate that
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	65 # the sample is of the correct length and contains only unique items
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	66 N = 100
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	67 population = xrange(N)
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	68 for k in xrange(N+1):
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	69 s = self.gen.sample(population, k)
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	70 self.assertEqual(len(s), k)
30929 54ff68e973ee * Migrate set() and frozenset() from the sandbox. Raymond Hettinger <python@rcn.com> parents: 30649 diff changeset	71 uniq = set(s)
26829 f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	72 self.assertEqual(len(uniq), k)
55002 6768b2350c57 convert usage of fail* to assert* Benjamin Peterson <benjamin@python.org> parents: 51737 diff changeset	73 self.assertTrue(uniq <= set(population))
27081 3199e974f55a Test an edge case for sample(). Raymond Hettinger <python@rcn.com> parents: 26829 diff changeset	74 self.assertEqual(self.gen.sample([], 0), []) # test edge case N==k==0
26829 f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	75
27279 9de064c7a247 * Migrate sample distribution test from random.py to test_random.py. Raymond Hettinger <python@rcn.com> parents: 27094 diff changeset	76 def test_sample_distribution(self):
9de064c7a247 * Migrate sample distribution test from random.py to test_random.py. Raymond Hettinger <python@rcn.com> parents: 27094 diff changeset	77 # For the entire allowable range of 0 <= k <= N, validate that
9de064c7a247 * Migrate sample distribution test from random.py to test_random.py. Raymond Hettinger <python@rcn.com> parents: 27094 diff changeset	78 # sample generates all possible permutations
9de064c7a247 * Migrate sample distribution test from random.py to test_random.py. Raymond Hettinger <python@rcn.com> parents: 27094 diff changeset	79 n = 5
9de064c7a247 * Migrate sample distribution test from random.py to test_random.py. Raymond Hettinger <python@rcn.com> parents: 27094 diff changeset	80 pop = range(n)
9de064c7a247 * Migrate sample distribution test from random.py to test_random.py. Raymond Hettinger <python@rcn.com> parents: 27094 diff changeset	81 trials = 10000 # large num prevents false negatives without slowing normal case
9de064c7a247 * Migrate sample distribution test from random.py to test_random.py. Raymond Hettinger <python@rcn.com> parents: 27094 diff changeset	82 def factorial(n):
27668 8e5de43529b7 Add refcount test. Raymond Hettinger <python@rcn.com> parents: 27279 diff changeset	83 return reduce(int.__mul__, xrange(1, n), 1)
27279 9de064c7a247 * Migrate sample distribution test from random.py to test_random.py. Raymond Hettinger <python@rcn.com> parents: 27094 diff changeset	84 for k in xrange(n):
33597 355b15e4a4da Use floor division operator. Raymond Hettinger <python@rcn.com> parents: 33481 diff changeset	85 expected = factorial(n) // factorial(n-k)
27279 9de064c7a247 * Migrate sample distribution test from random.py to test_random.py. Raymond Hettinger <python@rcn.com> parents: 27094 diff changeset	86 perms = {}
9de064c7a247 * Migrate sample distribution test from random.py to test_random.py. Raymond Hettinger <python@rcn.com> parents: 27094 diff changeset	87 for i in xrange(trials):
9de064c7a247 * Migrate sample distribution test from random.py to test_random.py. Raymond Hettinger <python@rcn.com> parents: 27094 diff changeset	88 perms[tuple(self.gen.sample(pop, k))] = None
9de064c7a247 * Migrate sample distribution test from random.py to test_random.py. Raymond Hettinger <python@rcn.com> parents: 27094 diff changeset	89 if len(perms) == expected:
9de064c7a247 * Migrate sample distribution test from random.py to test_random.py. Raymond Hettinger <python@rcn.com> parents: 27094 diff changeset	90 break
9de064c7a247 * Migrate sample distribution test from random.py to test_random.py. Raymond Hettinger <python@rcn.com> parents: 27094 diff changeset	91 else:
9de064c7a247 * Migrate sample distribution test from random.py to test_random.py. Raymond Hettinger <python@rcn.com> parents: 27094 diff changeset	92 self.fail()
9de064c7a247 * Migrate sample distribution test from random.py to test_random.py. Raymond Hettinger <python@rcn.com> parents: 27094 diff changeset	93
30346 aaf6b6a8510b SF bug #801342: Bug (documentation or real, your choice) in random.sample. Raymond Hettinger <python@rcn.com> parents: 30114 diff changeset	94 def test_sample_inputs(self):
aaf6b6a8510b SF bug #801342: Bug (documentation or real, your choice) in random.sample. Raymond Hettinger <python@rcn.com> parents: 30114 diff changeset	95 # SF bug #801342 -- population can be any iterable defining __len__()
30929 54ff68e973ee * Migrate set() and frozenset() from the sandbox. Raymond Hettinger <python@rcn.com> parents: 30649 diff changeset	96 self.gen.sample(set(range(20)), 2)
30346 aaf6b6a8510b SF bug #801342: Bug (documentation or real, your choice) in random.sample. Raymond Hettinger <python@rcn.com> parents: 30114 diff changeset	97 self.gen.sample(range(20), 2)
aaf6b6a8510b SF bug #801342: Bug (documentation or real, your choice) in random.sample. Raymond Hettinger <python@rcn.com> parents: 30114 diff changeset	98 self.gen.sample(xrange(20), 2)
aaf6b6a8510b SF bug #801342: Bug (documentation or real, your choice) in random.sample. Raymond Hettinger <python@rcn.com> parents: 30114 diff changeset	99 self.gen.sample(str('abcdefghijklmnopqrst'), 2)
aaf6b6a8510b SF bug #801342: Bug (documentation or real, your choice) in random.sample. Raymond Hettinger <python@rcn.com> parents: 30114 diff changeset	100 self.gen.sample(tuple('abcdefghijklmnopqrst'), 2)
37360 98251780236d Another crack at bug #1460340: make random.sample(dict) Tim Peters <tim.peters@gmail.com> parents: 37293 diff changeset	101
98251780236d Another crack at bug #1460340: make random.sample(dict) Tim Peters <tim.peters@gmail.com> parents: 37293 diff changeset	102 def test_sample_on_dicts(self):
98251780236d Another crack at bug #1460340: make random.sample(dict) Tim Peters <tim.peters@gmail.com> parents: 37293 diff changeset	103 self.gen.sample(dict.fromkeys('abcdefghijklmnopqrst'), 2)
98251780236d Another crack at bug #1460340: make random.sample(dict) Tim Peters <tim.peters@gmail.com> parents: 37293 diff changeset	104
37293 1ca012ce427e SF bug #1460340: random.sample can raise KeyError Raymond Hettinger <python@rcn.com> parents: 33597 diff changeset	105 # SF bug #1460340 -- random.sample can raise KeyError
1ca012ce427e SF bug #1460340: random.sample can raise KeyError Raymond Hettinger <python@rcn.com> parents: 33597 diff changeset	106 a = dict.fromkeys(range(10)+range(10,100,2)+range(100,110))
37360 98251780236d Another crack at bug #1460340: make random.sample(dict) Tim Peters <tim.peters@gmail.com> parents: 37293 diff changeset	107 self.gen.sample(a, 3)
98251780236d Another crack at bug #1460340: make random.sample(dict) Tim Peters <tim.peters@gmail.com> parents: 37293 diff changeset	108
98251780236d Another crack at bug #1460340: make random.sample(dict) Tim Peters <tim.peters@gmail.com> parents: 37293 diff changeset	109 # A followup to bug #1460340: sampling from a dict could return
98251780236d Another crack at bug #1460340: make random.sample(dict) Tim Peters <tim.peters@gmail.com> parents: 37293 diff changeset	110 # a subset of its keys or of its values, depending on the size of
98251780236d Another crack at bug #1460340: make random.sample(dict) Tim Peters <tim.peters@gmail.com> parents: 37293 diff changeset	111 # the subset requested.
98251780236d Another crack at bug #1460340: make random.sample(dict) Tim Peters <tim.peters@gmail.com> parents: 37293 diff changeset	112 N = 30
98251780236d Another crack at bug #1460340: make random.sample(dict) Tim Peters <tim.peters@gmail.com> parents: 37293 diff changeset	113 d = dict((i, complex(i, i)) for i in xrange(N))
98251780236d Another crack at bug #1460340: make random.sample(dict) Tim Peters <tim.peters@gmail.com> parents: 37293 diff changeset	114 for k in xrange(N+1):
98251780236d Another crack at bug #1460340: make random.sample(dict) Tim Peters <tim.peters@gmail.com> parents: 37293 diff changeset	115 samp = self.gen.sample(d, k)
98251780236d Another crack at bug #1460340: make random.sample(dict) Tim Peters <tim.peters@gmail.com> parents: 37293 diff changeset	116 # Verify that we got ints back (keys); the values are complex.
98251780236d Another crack at bug #1460340: make random.sample(dict) Tim Peters <tim.peters@gmail.com> parents: 37293 diff changeset	117 for x in samp:
55002 6768b2350c57 convert usage of fail* to assert* Benjamin Peterson <benjamin@python.org> parents: 51737 diff changeset	118 self.assertTrue(type(x) is int)
37360 98251780236d Another crack at bug #1460340: make random.sample(dict) Tim Peters <tim.peters@gmail.com> parents: 37293 diff changeset	119 samp.sort()
98251780236d Another crack at bug #1460340: make random.sample(dict) Tim Peters <tim.peters@gmail.com> parents: 37293 diff changeset	120 self.assertEqual(samp, range(N))
30346 aaf6b6a8510b SF bug #801342: Bug (documentation or real, your choice) in random.sample. Raymond Hettinger <python@rcn.com> parents: 30114 diff changeset	121
26829 f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	122 def test_gauss(self):
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	123 # Ensure that the seed() method initializes all the hidden state. In
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	124 # particular, through 2.2.1 it failed to reset a piece of state used
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	125 # by (and only by) the .gauss() method.
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	126
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	127 for seed in 1, 12, 123, 1234, 12345, 123456, 654321:
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	128 self.gen.seed(seed)
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	129 x1 = self.gen.random()
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	130 y1 = self.gen.gauss(0, 1)
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	131
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	132 self.gen.seed(seed)
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	133 x2 = self.gen.random()
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	134 y2 = self.gen.gauss(0, 1)
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	135
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	136 self.assertEqual(x1, x2)
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	137 self.assertEqual(y1, y2)
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	138
29586 8b163370372d SF bug #759889: Pickling of Random is broken Raymond Hettinger <python@rcn.com> parents: 28944 diff changeset	139 def test_pickling(self):
8b163370372d SF bug #759889: Pickling of Random is broken Raymond Hettinger <python@rcn.com> parents: 28944 diff changeset	140 state = pickle.dumps(self.gen)
8b163370372d SF bug #759889: Pickling of Random is broken Raymond Hettinger <python@rcn.com> parents: 28944 diff changeset	141 origseq = [self.gen.random() for i in xrange(10)]
8b163370372d SF bug #759889: Pickling of Random is broken Raymond Hettinger <python@rcn.com> parents: 28944 diff changeset	142 newgen = pickle.loads(state)
8b163370372d SF bug #759889: Pickling of Random is broken Raymond Hettinger <python@rcn.com> parents: 28944 diff changeset	143 restoredseq = [newgen.random() for i in xrange(10)]
8b163370372d SF bug #759889: Pickling of Random is broken Raymond Hettinger <python@rcn.com> parents: 28944 diff changeset	144 self.assertEqual(origseq, restoredseq)
26829 f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	145
44122 97210d863d12 Issue #1727780: Support loading pickles of random.Random objects created Martin v. Löwis <martin@v.loewis.de> parents: 42584 diff changeset	146 def test_bug_1727780(self):
97210d863d12 Issue #1727780: Support loading pickles of random.Random objects created Martin v. Löwis <martin@v.loewis.de> parents: 42584 diff changeset	147 # verify that version-2-pickles can be loaded
97210d863d12 Issue #1727780: Support loading pickles of random.Random objects created Martin v. Löwis <martin@v.loewis.de> parents: 42584 diff changeset	148 # fine, whether they are created on 32-bit or 64-bit
97210d863d12 Issue #1727780: Support loading pickles of random.Random objects created Martin v. Löwis <martin@v.loewis.de> parents: 42584 diff changeset	149 # platforms, and that version-3-pickles load fine.
97210d863d12 Issue #1727780: Support loading pickles of random.Random objects created Martin v. Löwis <martin@v.loewis.de> parents: 42584 diff changeset	150 files = [("randv2_32.pck", 780),
97210d863d12 Issue #1727780: Support loading pickles of random.Random objects created Martin v. Löwis <martin@v.loewis.de> parents: 42584 diff changeset	151 ("randv2_64.pck", 866),
97210d863d12 Issue #1727780: Support loading pickles of random.Random objects created Martin v. Löwis <martin@v.loewis.de> parents: 42584 diff changeset	152 ("randv3.pck", 343)]
97210d863d12 Issue #1727780: Support loading pickles of random.Random objects created Martin v. Löwis <martin@v.loewis.de> parents: 42584 diff changeset	153 for file, value in files:
97210d863d12 Issue #1727780: Support loading pickles of random.Random objects created Martin v. Löwis <martin@v.loewis.de> parents: 42584 diff changeset	154 f = open(test_support.findfile(file),"rb")
97210d863d12 Issue #1727780: Support loading pickles of random.Random objects created Martin v. Löwis <martin@v.loewis.de> parents: 42584 diff changeset	155 r = pickle.load(f)
97210d863d12 Issue #1727780: Support loading pickles of random.Random objects created Martin v. Löwis <martin@v.loewis.de> parents: 42584 diff changeset	156 f.close()
97210d863d12 Issue #1727780: Support loading pickles of random.Random objects created Martin v. Löwis <martin@v.loewis.de> parents: 42584 diff changeset	157 self.assertEqual(r.randrange(1000), value)
97210d863d12 Issue #1727780: Support loading pickles of random.Random objects created Martin v. Löwis <martin@v.loewis.de> parents: 42584 diff changeset	158
26829 f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	159 class WichmannHill_TestBasicOps(TestBasicOps):
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	160 gen = random.WichmannHill()
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	161
32562 a160b9669db1 Add some tests for corner cases. Raymond Hettinger <python@rcn.com> parents: 30929 diff changeset	162 def test_setstate_first_arg(self):
a160b9669db1 Add some tests for corner cases. Raymond Hettinger <python@rcn.com> parents: 30929 diff changeset	163 self.assertRaises(ValueError, self.gen.setstate, (2, None, None))
a160b9669db1 Add some tests for corner cases. Raymond Hettinger <python@rcn.com> parents: 30929 diff changeset	164
26829 f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	165 def test_strong_jumpahead(self):
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	166 # tests that jumpahead(n) semantics correspond to n calls to random()
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	167 N = 1000
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	168 s = self.gen.getstate()
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	169 self.gen.jumpahead(N)
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	170 r1 = self.gen.random()
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	171 # now do it the slow way
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	172 self.gen.setstate(s)
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	173 for i in xrange(N):
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	174 self.gen.random()
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	175 r2 = self.gen.random()
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	176 self.assertEqual(r1, r2)
23380 9be01b7b6c80 random.gauss() uses a piece of hidden state used by nothing else, Tim Peters <tim.peters@gmail.com> parents: diff changeset	177
26829 f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	178 def test_gauss_with_whseed(self):
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	179 # Ensure that the seed() method initializes all the hidden state. In
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	180 # particular, through 2.2.1 it failed to reset a piece of state used
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	181 # by (and only by) the .gauss() method.
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	182
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	183 for seed in 1, 12, 123, 1234, 12345, 123456, 654321:
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	184 self.gen.whseed(seed)
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	185 x1 = self.gen.random()
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	186 y1 = self.gen.gauss(0, 1)
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	187
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	188 self.gen.whseed(seed)
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	189 x2 = self.gen.random()
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	190 y2 = self.gen.gauss(0, 1)
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	191
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	192 self.assertEqual(x1, x2)
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	193 self.assertEqual(y1, y2)
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	194
30649 8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	195 def test_bigrand(self):
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	196 # Verify warnings are raised when randrange is too large for random()
49099 ff96a313cd1c warnings.catch_warnings() now returns a list or None instead of the custom Brett Cannon <bcannon@gmail.com> parents: 48101 diff changeset	197 with warnings.catch_warnings():
40455 480789b3294d Add test.test_support.guard_warnings_filter . This function returns a context Brett Cannon <bcannon@gmail.com> parents: 37360 diff changeset	198 warnings.filterwarnings("error", "Underlying random")
480789b3294d Add test.test_support.guard_warnings_filter . This function returns a context Brett Cannon <bcannon@gmail.com> parents: 37360 diff changeset	199 self.assertRaises(UserWarning, self.gen.randrange, 2**60)
30649 8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	200
33481 eacdc29606d8 SF #1027105: HardwareRandom should be renamed OSRandom Raymond Hettinger <python@rcn.com> parents: 33409 diff changeset	201 class SystemRandom_TestBasicOps(TestBasicOps):
eacdc29606d8 SF #1027105: HardwareRandom should be renamed OSRandom Raymond Hettinger <python@rcn.com> parents: 33409 diff changeset	202 gen = random.SystemRandom()
33329 97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	203
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	204 def test_autoseed(self):
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	205 # Doesn't need to do anything except not fail
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	206 self.gen.seed()
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	207
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	208 def test_saverestore(self):
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	209 self.assertRaises(NotImplementedError, self.gen.getstate)
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	210 self.assertRaises(NotImplementedError, self.gen.setstate, None)
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	211
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	212 def test_seedargs(self):
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	213 # Doesn't need to do anything except not fail
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	214 self.gen.seed(100)
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	215
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	216 def test_jumpahead(self):
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	217 # Doesn't need to do anything except not fail
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	218 self.gen.jumpahead(100)
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	219
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	220 def test_gauss(self):
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	221 self.gen.gauss_next = None
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	222 self.gen.seed(100)
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	223 self.assertEqual(self.gen.gauss_next, None)
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	224
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	225 def test_pickling(self):
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	226 self.assertRaises(NotImplementedError, pickle.dumps, self.gen)
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	227
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	228 def test_53_bits_per_float(self):
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	229 # This should pass whenever a C double has 53 bit precision.
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	230 span = 2 ** 53
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	231 cum = 0
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	232 for i in xrange(100):
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	233 cum \|= int(self.gen.random() * span)
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	234 self.assertEqual(cum, span-1)
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	235
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	236 def test_bigrand(self):
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	237 # The randrange routine should build-up the required number of bits
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	238 # in stages so that all bit positions are active.
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	239 span = 2 ** 500
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	240 cum = 0
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	241 for i in xrange(100):
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	242 r = self.gen.randrange(span)
55002 6768b2350c57 convert usage of fail* to assert* Benjamin Peterson <benjamin@python.org> parents: 51737 diff changeset	243 self.assertTrue(0 <= r < span)
33329 97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	244 cum \|= r
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	245 self.assertEqual(cum, span-1)
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	246
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	247 def test_bigrand_ranges(self):
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	248 for i in [40,80, 160, 200, 211, 250, 375, 512, 550]:
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	249 start = self.gen.randrange(2 ** i)
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	250 stop = self.gen.randrange(2 ** (i-2))
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	251 if stop <= start:
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	252 return
55002 6768b2350c57 convert usage of fail* to assert* Benjamin Peterson <benjamin@python.org> parents: 51737 diff changeset	253 self.assertTrue(start <= self.gen.randrange(start, stop) < stop)
33329 97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	254
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	255 def test_rangelimits(self):
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	256 for start, stop in [(-2,0), (-(260)-2,-(260)), (260,260+2)]:
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	257 self.assertEqual(set(range(start,stop)),
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	258 set([self.gen.randrange(start,stop) for i in xrange(100)]))
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	259
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	260 def test_genrandbits(self):
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	261 # Verify ranges
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	262 for k in xrange(1, 1000):
55002 6768b2350c57 convert usage of fail* to assert* Benjamin Peterson <benjamin@python.org> parents: 51737 diff changeset	263 self.assertTrue(0 <= self.gen.getrandbits(k) < 2**k)
33329 97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	264
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	265 # Verify all bits active
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	266 getbits = self.gen.getrandbits
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	267 for span in [1, 2, 3, 4, 31, 32, 32, 52, 53, 54, 119, 127, 128, 129]:
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	268 cum = 0
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	269 for i in xrange(100):
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	270 cum \|= getbits(span)
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	271 self.assertEqual(cum, 2**span-1)
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	272
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	273 # Verify argument checking
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	274 self.assertRaises(TypeError, self.gen.getrandbits)
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	275 self.assertRaises(TypeError, self.gen.getrandbits, 1, 2)
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	276 self.assertRaises(ValueError, self.gen.getrandbits, 0)
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	277 self.assertRaises(ValueError, self.gen.getrandbits, -1)
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	278 self.assertRaises(TypeError, self.gen.getrandbits, 10.1)
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	279
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	280 def test_randbelow_logic(self, _log=log, int=int):
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	281 # check bitcount transition points: 2i and 2(i+1)-1
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	282 # show that: k = int(1.001 + _log(n, 2))
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	283 # is equal to or one greater than the number of bits in n
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	284 for i in xrange(1, 1000):
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	285 n = 1L << i # check an exact power of two
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	286 numbits = i+1
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	287 k = int(1.00001 + _log(n, 2))
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	288 self.assertEqual(k, numbits)
55002 6768b2350c57 convert usage of fail* to assert* Benjamin Peterson <benjamin@python.org> parents: 51737 diff changeset	289 self.assertTrue(n == 2**(k-1))
33329 97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	290
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	291 n += n - 1 # check 1 below the next power of two
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	292 k = int(1.00001 + _log(n, 2))
58460 17891566a478 use assert[Not]In where appropriate Ezio Melotti <ezio.melotti@gmail.com> parents: 58198 diff changeset	293 self.assertIn(k, [numbits, numbits+1])
55002 6768b2350c57 convert usage of fail* to assert* Benjamin Peterson <benjamin@python.org> parents: 51737 diff changeset	294 self.assertTrue(2k > n > 2(k-2))
33329 97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	295
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	296 n -= n >> 15 # check a little farther below the next power of two
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	297 k = int(1.00001 + _log(n, 2))
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	298 self.assertEqual(k, numbits) # note the stronger assertion
55002 6768b2350c57 convert usage of fail* to assert* Benjamin Peterson <benjamin@python.org> parents: 51737 diff changeset	299 self.assertTrue(2k > n > 2(k-1)) # note the stronger assertion
33329 97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	300
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	301
26829 f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	302 class MersenneTwister_TestBasicOps(TestBasicOps):
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	303 gen = random.Random()
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	304
32562 a160b9669db1 Add some tests for corner cases. Raymond Hettinger <python@rcn.com> parents: 30929 diff changeset	305 def test_setstate_first_arg(self):
a160b9669db1 Add some tests for corner cases. Raymond Hettinger <python@rcn.com> parents: 30929 diff changeset	306 self.assertRaises(ValueError, self.gen.setstate, (1, None, None))
a160b9669db1 Add some tests for corner cases. Raymond Hettinger <python@rcn.com> parents: 30929 diff changeset	307
a160b9669db1 Add some tests for corner cases. Raymond Hettinger <python@rcn.com> parents: 30929 diff changeset	308 def test_setstate_middle_arg(self):
a160b9669db1 Add some tests for corner cases. Raymond Hettinger <python@rcn.com> parents: 30929 diff changeset	309 # Wrong type, s/b tuple
a160b9669db1 Add some tests for corner cases. Raymond Hettinger <python@rcn.com> parents: 30929 diff changeset	310 self.assertRaises(TypeError, self.gen.setstate, (2, None, None))
a160b9669db1 Add some tests for corner cases. Raymond Hettinger <python@rcn.com> parents: 30929 diff changeset	311 # Wrong length, s/b 625
a160b9669db1 Add some tests for corner cases. Raymond Hettinger <python@rcn.com> parents: 30929 diff changeset	312 self.assertRaises(ValueError, self.gen.setstate, (2, (1,2,3), None))
a160b9669db1 Add some tests for corner cases. Raymond Hettinger <python@rcn.com> parents: 30929 diff changeset	313 # Wrong type, s/b tuple of 625 ints
a160b9669db1 Add some tests for corner cases. Raymond Hettinger <python@rcn.com> parents: 30929 diff changeset	314 self.assertRaises(TypeError, self.gen.setstate, (2, ('a',)*625, None))
a160b9669db1 Add some tests for corner cases. Raymond Hettinger <python@rcn.com> parents: 30929 diff changeset	315 # Last element s/b an int also
a160b9669db1 Add some tests for corner cases. Raymond Hettinger <python@rcn.com> parents: 30929 diff changeset	316 self.assertRaises(TypeError, self.gen.setstate, (2, (0,)*624+('a',), None))
a160b9669db1 Add some tests for corner cases. Raymond Hettinger <python@rcn.com> parents: 30929 diff changeset	317
26829 f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	318 def test_referenceImplementation(self):
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	319 # Compare the python implementation with results from the original
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	320 # code. Create 2000 53-bit precision random floats. Compare only
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	321 # the last ten entries to show that the independent implementations
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	322 # are tracking. Here is the main() function needed to create the
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	323 # list of expected random numbers:
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	324 # void main(void){
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	325 # int i;
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	326 # unsigned long init[4]={61731, 24903, 614, 42143}, length=4;
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	327 # init_by_array(init, length);
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	328 # for (i=0; i<2000; i++) {
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	329 # printf("%.15f ", genrand_res53());
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	330 # if (i%5==4) printf("\n");
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	331 # }
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	332 # }
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	333 expected = [0.45839803073713259,
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	334 0.86057815201978782,
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	335 0.92848331726782152,
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	336 0.35932681119782461,
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	337 0.081823493762449573,
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	338 0.14332226470169329,
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	339 0.084297823823520024,
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	340 0.53814864671831453,
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	341 0.089215024911993401,
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	342 0.78486196105372907]
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	343
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	344 self.gen.seed(61731L + (24903L<<32) + (614L<<64) + (42143L<<96))
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	345 actual = self.randomlist(2000)[-10:]
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	346 for a, e in zip(actual, expected):
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	347 self.assertAlmostEqual(a,e,places=14)
23380 9be01b7b6c80 random.gauss() uses a piece of hidden state used by nothing else, Tim Peters <tim.peters@gmail.com> parents: diff changeset	348
26829 f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	349 def test_strong_reference_implementation(self):
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	350 # Like test_referenceImplementation, but checks for exact bit-level
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	351 # equality. This should pass on any box where C double contains
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	352 # at least 53 bits of precision (the underlying algorithm suffers
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	353 # no rounding errors -- all results are exact).
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	354 from math import ldexp
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	355
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	356 expected = [0x0eab3258d2231fL,
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	357 0x1b89db315277a5L,
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	358 0x1db622a5518016L,
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	359 0x0b7f9af0d575bfL,
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	360 0x029e4c4db82240L,
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	361 0x04961892f5d673L,
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	362 0x02b291598e4589L,
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	363 0x11388382c15694L,
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	364 0x02dad977c9e1feL,
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	365 0x191d96d4d334c6L]
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	366 self.gen.seed(61731L + (24903L<<32) + (614L<<64) + (42143L<<96))
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	367 actual = self.randomlist(2000)[-10:]
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	368 for a, e in zip(actual, expected):
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	369 self.assertEqual(long(ldexp(a, 53)), e)
23380 9be01b7b6c80 random.gauss() uses a piece of hidden state used by nothing else, Tim Peters <tim.peters@gmail.com> parents: diff changeset	370
26829 f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	371 def test_long_seed(self):
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	372 # This is most interesting to run in debug mode, just to make sure
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	373 # nothing blows up. Under the covers, a dynamically resized array
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	374 # is allocated, consuming space proportional to the number of bits
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	375 # in the seed. Unfortunately, that's a quadratic-time algorithm,
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	376 # so don't make this horribly big.
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	377 seed = (1L << (10000 * 8)) - 1 # about 10K bytes
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	378 self.gen.seed(seed)
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	379
30649 8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	380 def test_53_bits_per_float(self):
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	381 # This should pass whenever a C double has 53 bit precision.
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	382 span = 2 ** 53
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	383 cum = 0
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	384 for i in xrange(100):
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	385 cum \|= int(self.gen.random() * span)
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	386 self.assertEqual(cum, span-1)
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	387
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	388 def test_bigrand(self):
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	389 # The randrange routine should build-up the required number of bits
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	390 # in stages so that all bit positions are active.
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	391 span = 2 ** 500
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	392 cum = 0
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	393 for i in xrange(100):
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	394 r = self.gen.randrange(span)
55002 6768b2350c57 convert usage of fail* to assert* Benjamin Peterson <benjamin@python.org> parents: 51737 diff changeset	395 self.assertTrue(0 <= r < span)
30649 8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	396 cum \|= r
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	397 self.assertEqual(cum, span-1)
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	398
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	399 def test_bigrand_ranges(self):
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	400 for i in [40,80, 160, 200, 211, 250, 375, 512, 550]:
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	401 start = self.gen.randrange(2 ** i)
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	402 stop = self.gen.randrange(2 ** (i-2))
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	403 if stop <= start:
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	404 return
55002 6768b2350c57 convert usage of fail* to assert* Benjamin Peterson <benjamin@python.org> parents: 51737 diff changeset	405 self.assertTrue(start <= self.gen.randrange(start, stop) < stop)
30649 8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	406
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	407 def test_rangelimits(self):
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	408 for start, stop in [(-2,0), (-(260)-2,-(260)), (260,260+2)]:
30929 54ff68e973ee * Migrate set() and frozenset() from the sandbox. Raymond Hettinger <python@rcn.com> parents: 30649 diff changeset	409 self.assertEqual(set(range(start,stop)),
54ff68e973ee * Migrate set() and frozenset() from the sandbox. Raymond Hettinger <python@rcn.com> parents: 30649 diff changeset	410 set([self.gen.randrange(start,stop) for i in xrange(100)]))
30649 8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	411
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	412 def test_genrandbits(self):
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	413 # Verify cross-platform repeatability
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	414 self.gen.seed(1234567)
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	415 self.assertEqual(self.gen.getrandbits(100),
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	416 97904845777343510404718956115L)
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	417 # Verify ranges
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	418 for k in xrange(1, 1000):
55002 6768b2350c57 convert usage of fail* to assert* Benjamin Peterson <benjamin@python.org> parents: 51737 diff changeset	419 self.assertTrue(0 <= self.gen.getrandbits(k) < 2**k)
30649 8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	420
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	421 # Verify all bits active
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	422 getbits = self.gen.getrandbits
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	423 for span in [1, 2, 3, 4, 31, 32, 32, 52, 53, 54, 119, 127, 128, 129]:
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	424 cum = 0
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	425 for i in xrange(100):
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	426 cum \|= getbits(span)
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	427 self.assertEqual(cum, 2**span-1)
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	428
32562 a160b9669db1 Add some tests for corner cases. Raymond Hettinger <python@rcn.com> parents: 30929 diff changeset	429 # Verify argument checking
a160b9669db1 Add some tests for corner cases. Raymond Hettinger <python@rcn.com> parents: 30929 diff changeset	430 self.assertRaises(TypeError, self.gen.getrandbits)
a160b9669db1 Add some tests for corner cases. Raymond Hettinger <python@rcn.com> parents: 30929 diff changeset	431 self.assertRaises(TypeError, self.gen.getrandbits, 'a')
a160b9669db1 Add some tests for corner cases. Raymond Hettinger <python@rcn.com> parents: 30929 diff changeset	432 self.assertRaises(TypeError, self.gen.getrandbits, 1, 2)
a160b9669db1 Add some tests for corner cases. Raymond Hettinger <python@rcn.com> parents: 30929 diff changeset	433 self.assertRaises(ValueError, self.gen.getrandbits, 0)
a160b9669db1 Add some tests for corner cases. Raymond Hettinger <python@rcn.com> parents: 30929 diff changeset	434 self.assertRaises(ValueError, self.gen.getrandbits, -1)
a160b9669db1 Add some tests for corner cases. Raymond Hettinger <python@rcn.com> parents: 30929 diff changeset	435
30649 8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	436 def test_randbelow_logic(self, _log=log, int=int):
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	437 # check bitcount transition points: 2i and 2(i+1)-1
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	438 # show that: k = int(1.001 + _log(n, 2))
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	439 # is equal to or one greater than the number of bits in n
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	440 for i in xrange(1, 1000):
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	441 n = 1L << i # check an exact power of two
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	442 numbits = i+1
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	443 k = int(1.00001 + _log(n, 2))
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	444 self.assertEqual(k, numbits)
55002 6768b2350c57 convert usage of fail* to assert* Benjamin Peterson <benjamin@python.org> parents: 51737 diff changeset	445 self.assertTrue(n == 2**(k-1))
30649 8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	446
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	447 n += n - 1 # check 1 below the next power of two
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	448 k = int(1.00001 + _log(n, 2))
58460 17891566a478 use assert[Not]In where appropriate Ezio Melotti <ezio.melotti@gmail.com> parents: 58198 diff changeset	449 self.assertIn(k, [numbits, numbits+1])
55002 6768b2350c57 convert usage of fail* to assert* Benjamin Peterson <benjamin@python.org> parents: 51737 diff changeset	450 self.assertTrue(2k > n > 2(k-2))
30649 8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	451
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	452 n -= n >> 15 # check a little farther below the next power of two
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	453 k = int(1.00001 + _log(n, 2))
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	454 self.assertEqual(k, numbits) # note the stronger assertion
55002 6768b2350c57 convert usage of fail* to assert* Benjamin Peterson <benjamin@python.org> parents: 51737 diff changeset	455 self.assertTrue(2k > n > 2(k-1)) # note the stronger assertion
30649 8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	456
40488 96d533efbe7d Bug #1590891: random.randrange don't return correct value for big number Raymond Hettinger <python@rcn.com> parents: 40455 diff changeset	457 def test_randrange_bug_1590891(self):
96d533efbe7d Bug #1590891: random.randrange don't return correct value for big number Raymond Hettinger <python@rcn.com> parents: 40455 diff changeset	458 start = 1000000000000
96d533efbe7d Bug #1590891: random.randrange don't return correct value for big number Raymond Hettinger <python@rcn.com> parents: 40455 diff changeset	459 stop = -100000000000000000000
96d533efbe7d Bug #1590891: random.randrange don't return correct value for big number Raymond Hettinger <python@rcn.com> parents: 40455 diff changeset	460 step = -200
96d533efbe7d Bug #1590891: random.randrange don't return correct value for big number Raymond Hettinger <python@rcn.com> parents: 40455 diff changeset	461 x = self.gen.randrange(start, stop, step)
55002 6768b2350c57 convert usage of fail* to assert* Benjamin Peterson <benjamin@python.org> parents: 51737 diff changeset	462 self.assertTrue(stop < x <= start)
40488 96d533efbe7d Bug #1590891: random.randrange don't return correct value for big number Raymond Hettinger <python@rcn.com> parents: 40455 diff changeset	463 self.assertEqual((x+stop)%step, 0)
96d533efbe7d Bug #1590891: random.randrange don't return correct value for big number Raymond Hettinger <python@rcn.com> parents: 40455 diff changeset	464
51737 00f7be02ae1f Inline coefficients in gamma(). Add reflection formula. Add comments. Raymond Hettinger <python@rcn.com> parents: 49099 diff changeset	465 def gamma(z, sqrt2pi=(2.0pi)*0.5):
00f7be02ae1f Inline coefficients in gamma(). Add reflection formula. Add comments. Raymond Hettinger <python@rcn.com> parents: 49099 diff changeset	466 # Reflection to right half of complex plane
00f7be02ae1f Inline coefficients in gamma(). Add reflection formula. Add comments. Raymond Hettinger <python@rcn.com> parents: 49099 diff changeset	467 if z < 0.5:
00f7be02ae1f Inline coefficients in gamma(). Add reflection formula. Add comments. Raymond Hettinger <python@rcn.com> parents: 49099 diff changeset	468 return pi / sin(pi*z) / gamma(1.0-z)
00f7be02ae1f Inline coefficients in gamma(). Add reflection formula. Add comments. Raymond Hettinger <python@rcn.com> parents: 49099 diff changeset	469 # Lanczos approximation with g=7
00f7be02ae1f Inline coefficients in gamma(). Add reflection formula. Add comments. Raymond Hettinger <python@rcn.com> parents: 49099 diff changeset	470 az = z + (7.0 - 0.5)
00f7be02ae1f Inline coefficients in gamma(). Add reflection formula. Add comments. Raymond Hettinger <python@rcn.com> parents: 49099 diff changeset	471 return az ** (z-0.5) / exp(az) * sqrt2pi * fsum([
00f7be02ae1f Inline coefficients in gamma(). Add reflection formula. Add comments. Raymond Hettinger <python@rcn.com> parents: 49099 diff changeset	472 0.9999999999995183,
00f7be02ae1f Inline coefficients in gamma(). Add reflection formula. Add comments. Raymond Hettinger <python@rcn.com> parents: 49099 diff changeset	473 676.5203681218835 / z,
00f7be02ae1f Inline coefficients in gamma(). Add reflection formula. Add comments. Raymond Hettinger <python@rcn.com> parents: 49099 diff changeset	474 -1259.139216722289 / (z+1.0),
00f7be02ae1f Inline coefficients in gamma(). Add reflection formula. Add comments. Raymond Hettinger <python@rcn.com> parents: 49099 diff changeset	475 771.3234287757674 / (z+2.0),
00f7be02ae1f Inline coefficients in gamma(). Add reflection formula. Add comments. Raymond Hettinger <python@rcn.com> parents: 49099 diff changeset	476 -176.6150291498386 / (z+3.0),
00f7be02ae1f Inline coefficients in gamma(). Add reflection formula. Add comments. Raymond Hettinger <python@rcn.com> parents: 49099 diff changeset	477 12.50734324009056 / (z+4.0),
00f7be02ae1f Inline coefficients in gamma(). Add reflection formula. Add comments. Raymond Hettinger <python@rcn.com> parents: 49099 diff changeset	478 -0.1385710331296526 / (z+5.0),
00f7be02ae1f Inline coefficients in gamma(). Add reflection formula. Add comments. Raymond Hettinger <python@rcn.com> parents: 49099 diff changeset	479 0.9934937113930748e-05 / (z+6.0),
00f7be02ae1f Inline coefficients in gamma(). Add reflection formula. Add comments. Raymond Hettinger <python@rcn.com> parents: 49099 diff changeset	480 0.1659470187408462e-06 / (z+7.0),
00f7be02ae1f Inline coefficients in gamma(). Add reflection formula. Add comments. Raymond Hettinger <python@rcn.com> parents: 49099 diff changeset	481 ])
27094 e841c873a937 Move the statistical tests for four distributions into the unittest suite. Raymond Hettinger <python@rcn.com> parents: 27092 diff changeset	482
27092 7dab27731be8 Add a test case. Raymond Hettinger <python@rcn.com> parents: 27081 diff changeset	483 class TestDistributions(unittest.TestCase):
7dab27731be8 Add a test case. Raymond Hettinger <python@rcn.com> parents: 27081 diff changeset	484 def test_zeroinputs(self):
7dab27731be8 Add a test case. Raymond Hettinger <python@rcn.com> parents: 27081 diff changeset	485 # Verify that distributions can handle a series of zero inputs'
7dab27731be8 Add a test case. Raymond Hettinger <python@rcn.com> parents: 27081 diff changeset	486 g = random.Random()
7dab27731be8 Add a test case. Raymond Hettinger <python@rcn.com> parents: 27081 diff changeset	487 x = [g.random() for i in xrange(50)] + [0.0]*5
7dab27731be8 Add a test case. Raymond Hettinger <python@rcn.com> parents: 27081 diff changeset	488 g.random = x[:].pop; g.uniform(1,10)
7dab27731be8 Add a test case. Raymond Hettinger <python@rcn.com> parents: 27081 diff changeset	489 g.random = x[:].pop; g.paretovariate(1.0)
7dab27731be8 Add a test case. Raymond Hettinger <python@rcn.com> parents: 27081 diff changeset	490 g.random = x[:].pop; g.expovariate(1.0)
7dab27731be8 Add a test case. Raymond Hettinger <python@rcn.com> parents: 27081 diff changeset	491 g.random = x[:].pop; g.weibullvariate(1.0, 1.0)
7dab27731be8 Add a test case. Raymond Hettinger <python@rcn.com> parents: 27081 diff changeset	492 g.random = x[:].pop; g.normalvariate(0.0, 1.0)
7dab27731be8 Add a test case. Raymond Hettinger <python@rcn.com> parents: 27081 diff changeset	493 g.random = x[:].pop; g.gauss(0.0, 1.0)
7dab27731be8 Add a test case. Raymond Hettinger <python@rcn.com> parents: 27081 diff changeset	494 g.random = x[:].pop; g.lognormvariate(0.0, 1.0)
7dab27731be8 Add a test case. Raymond Hettinger <python@rcn.com> parents: 27081 diff changeset	495 g.random = x[:].pop; g.vonmisesvariate(0.0, 1.0)
7dab27731be8 Add a test case. Raymond Hettinger <python@rcn.com> parents: 27081 diff changeset	496 g.random = x[:].pop; g.gammavariate(0.01, 1.0)
7dab27731be8 Add a test case. Raymond Hettinger <python@rcn.com> parents: 27081 diff changeset	497 g.random = x[:].pop; g.gammavariate(1.0, 1.0)
7dab27731be8 Add a test case. Raymond Hettinger <python@rcn.com> parents: 27081 diff changeset	498 g.random = x[:].pop; g.gammavariate(200.0, 1.0)
7dab27731be8 Add a test case. Raymond Hettinger <python@rcn.com> parents: 27081 diff changeset	499 g.random = x[:].pop; g.betavariate(3.0, 3.0)
46029 f1e51785947f Issue 1681432: Add triangular distribution the random module. Raymond Hettinger <python@rcn.com> parents: 44122 diff changeset	500 g.random = x[:].pop; g.triangular(0.0, 1.0, 1.0/3.0)
27092 7dab27731be8 Add a test case. Raymond Hettinger <python@rcn.com> parents: 27081 diff changeset	501
27094 e841c873a937 Move the statistical tests for four distributions into the unittest suite. Raymond Hettinger <python@rcn.com> parents: 27092 diff changeset	502 def test_avg_std(self):
e841c873a937 Move the statistical tests for four distributions into the unittest suite. Raymond Hettinger <python@rcn.com> parents: 27092 diff changeset	503 # Use integration to test distribution average and standard deviation.
e841c873a937 Move the statistical tests for four distributions into the unittest suite. Raymond Hettinger <python@rcn.com> parents: 27092 diff changeset	504 # Only works for distributions which do not consume variates in pairs
e841c873a937 Move the statistical tests for four distributions into the unittest suite. Raymond Hettinger <python@rcn.com> parents: 27092 diff changeset	505 g = random.Random()
e841c873a937 Move the statistical tests for four distributions into the unittest suite. Raymond Hettinger <python@rcn.com> parents: 27092 diff changeset	506 N = 5000
e841c873a937 Move the statistical tests for four distributions into the unittest suite. Raymond Hettinger <python@rcn.com> parents: 27092 diff changeset	507 x = [i/float(N) for i in xrange(1,N)]
e841c873a937 Move the statistical tests for four distributions into the unittest suite. Raymond Hettinger <python@rcn.com> parents: 27092 diff changeset	508 for variate, args, mu, sigmasqrd in [
e841c873a937 Move the statistical tests for four distributions into the unittest suite. Raymond Hettinger <python@rcn.com> parents: 27092 diff changeset	509 (g.uniform, (1.0,10.0), (10.0+1.0)/2, (10.0-1.0)**2/12),
46029 f1e51785947f Issue 1681432: Add triangular distribution the random module. Raymond Hettinger <python@rcn.com> parents: 44122 diff changeset	510 (g.triangular, (0.0, 1.0, 1.0/3.0), 4.0/9.0, 7.0/9.0/18.0),
27094 e841c873a937 Move the statistical tests for four distributions into the unittest suite. Raymond Hettinger <python@rcn.com> parents: 27092 diff changeset	511 (g.expovariate, (1.5,), 1/1.5, 1/1.5**2),
e841c873a937 Move the statistical tests for four distributions into the unittest suite. Raymond Hettinger <python@rcn.com> parents: 27092 diff changeset	512 (g.paretovariate, (5.0,), 5.0/(5.0-1),
e841c873a937 Move the statistical tests for four distributions into the unittest suite. Raymond Hettinger <python@rcn.com> parents: 27092 diff changeset	513 5.0/((5.0-1)*2(5.0-2))),
e841c873a937 Move the statistical tests for four distributions into the unittest suite. Raymond Hettinger <python@rcn.com> parents: 27092 diff changeset	514 (g.weibullvariate, (1.0, 3.0), gamma(1+1/3.0),
e841c873a937 Move the statistical tests for four distributions into the unittest suite. Raymond Hettinger <python@rcn.com> parents: 27092 diff changeset	515 gamma(1+2/3.0)-gamma(1+1/3.0)**2) ]:
e841c873a937 Move the statistical tests for four distributions into the unittest suite. Raymond Hettinger <python@rcn.com> parents: 27092 diff changeset	516 g.random = x[:].pop
e841c873a937 Move the statistical tests for four distributions into the unittest suite. Raymond Hettinger <python@rcn.com> parents: 27092 diff changeset	517 y = []
e841c873a937 Move the statistical tests for four distributions into the unittest suite. Raymond Hettinger <python@rcn.com> parents: 27092 diff changeset	518 for i in xrange(len(x)):
e841c873a937 Move the statistical tests for four distributions into the unittest suite. Raymond Hettinger <python@rcn.com> parents: 27092 diff changeset	519 try:
e841c873a937 Move the statistical tests for four distributions into the unittest suite. Raymond Hettinger <python@rcn.com> parents: 27092 diff changeset	520 y.append(variate(*args))
e841c873a937 Move the statistical tests for four distributions into the unittest suite. Raymond Hettinger <python@rcn.com> parents: 27092 diff changeset	521 except IndexError:
e841c873a937 Move the statistical tests for four distributions into the unittest suite. Raymond Hettinger <python@rcn.com> parents: 27092 diff changeset	522 pass
e841c873a937 Move the statistical tests for four distributions into the unittest suite. Raymond Hettinger <python@rcn.com> parents: 27092 diff changeset	523 s1 = s2 = 0
e841c873a937 Move the statistical tests for four distributions into the unittest suite. Raymond Hettinger <python@rcn.com> parents: 27092 diff changeset	524 for e in y:
e841c873a937 Move the statistical tests for four distributions into the unittest suite. Raymond Hettinger <python@rcn.com> parents: 27092 diff changeset	525 s1 += e
e841c873a937 Move the statistical tests for four distributions into the unittest suite. Raymond Hettinger <python@rcn.com> parents: 27092 diff changeset	526 s2 += (e - mu) ** 2
e841c873a937 Move the statistical tests for four distributions into the unittest suite. Raymond Hettinger <python@rcn.com> parents: 27092 diff changeset	527 N = len(y)
e841c873a937 Move the statistical tests for four distributions into the unittest suite. Raymond Hettinger <python@rcn.com> parents: 27092 diff changeset	528 self.assertAlmostEqual(s1/N, mu, 2)
e841c873a937 Move the statistical tests for four distributions into the unittest suite. Raymond Hettinger <python@rcn.com> parents: 27092 diff changeset	529 self.assertAlmostEqual(s2/(N-1), sigmasqrd, 2)
e841c873a937 Move the statistical tests for four distributions into the unittest suite. Raymond Hettinger <python@rcn.com> parents: 27092 diff changeset	530
26829 f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	531 class TestModule(unittest.TestCase):
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	532 def testMagicConstants(self):
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	533 self.assertAlmostEqual(random.NV_MAGICCONST, 1.71552776992141)
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	534 self.assertAlmostEqual(random.TWOPI, 6.28318530718)
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	535 self.assertAlmostEqual(random.LOG4, 1.38629436111989)
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	536 self.assertAlmostEqual(random.SG_MAGICCONST, 2.50407739677627)
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	537
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	538 def test__all__(self):
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	539 # tests validity but not completeness of the __all__ list
55002 6768b2350c57 convert usage of fail* to assert* Benjamin Peterson <benjamin@python.org> parents: 51737 diff changeset	540 self.assertTrue(set(random.__all__) <= set(dir(random)))
26829 f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	541
40682 77e0f4718a7c Bug #1486663: don't reject keyword arguments for subclasses of builtin Georg Brandl <georg@python.org> parents: 40488 diff changeset	542 def test_random_subclass_with_kwargs(self):
77e0f4718a7c Bug #1486663: don't reject keyword arguments for subclasses of builtin Georg Brandl <georg@python.org> parents: 40488 diff changeset	543 # SF bug #1486663 -- this used to erroneously raise a TypeError
77e0f4718a7c Bug #1486663: don't reject keyword arguments for subclasses of builtin Georg Brandl <georg@python.org> parents: 40488 diff changeset	544 class Subclass(random.Random):
77e0f4718a7c Bug #1486663: don't reject keyword arguments for subclasses of builtin Georg Brandl <georg@python.org> parents: 40488 diff changeset	545 def __init__(self, newarg=None):
77e0f4718a7c Bug #1486663: don't reject keyword arguments for subclasses of builtin Georg Brandl <georg@python.org> parents: 40488 diff changeset	546 random.Random.__init__(self)
77e0f4718a7c Bug #1486663: don't reject keyword arguments for subclasses of builtin Georg Brandl <georg@python.org> parents: 40488 diff changeset	547 Subclass(newarg=1)
77e0f4718a7c Bug #1486663: don't reject keyword arguments for subclasses of builtin Georg Brandl <georg@python.org> parents: 40488 diff changeset	548
77e0f4718a7c Bug #1486663: don't reject keyword arguments for subclasses of builtin Georg Brandl <georg@python.org> parents: 40488 diff changeset	549
27668 8e5de43529b7 Add refcount test. Raymond Hettinger <python@rcn.com> parents: 27279 diff changeset	550 def test_main(verbose=None):
33406 483c0b4b6332 SF bug #1022010: Import random fails Raymond Hettinger <python@rcn.com> parents: 33329 diff changeset	551 testclasses = [WichmannHill_TestBasicOps,
26829 f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	552 MersenneTwister_TestBasicOps,
27092 7dab27731be8 Add a test case. Raymond Hettinger <python@rcn.com> parents: 27081 diff changeset	553 TestDistributions,
33406 483c0b4b6332 SF bug #1022010: Import random fails Raymond Hettinger <python@rcn.com> parents: 33329 diff changeset	554 TestModule]
483c0b4b6332 SF bug #1022010: Import random fails Raymond Hettinger <python@rcn.com> parents: 33329 diff changeset	555
33409 8e23bd3d5b1a Fulfill Martin's request to use try/except rather than a "look before Raymond Hettinger <python@rcn.com> parents: 33406 diff changeset	556 try:
33481 eacdc29606d8 SF #1027105: HardwareRandom should be renamed OSRandom Raymond Hettinger <python@rcn.com> parents: 33409 diff changeset	557 random.SystemRandom().random()
33409 8e23bd3d5b1a Fulfill Martin's request to use try/except rather than a "look before Raymond Hettinger <python@rcn.com> parents: 33406 diff changeset	558 except NotImplementedError:
8e23bd3d5b1a Fulfill Martin's request to use try/except rather than a "look before Raymond Hettinger <python@rcn.com> parents: 33406 diff changeset	559 pass
8e23bd3d5b1a Fulfill Martin's request to use try/except rather than a "look before Raymond Hettinger <python@rcn.com> parents: 33406 diff changeset	560 else:
33481 eacdc29606d8 SF #1027105: HardwareRandom should be renamed OSRandom Raymond Hettinger <python@rcn.com> parents: 33409 diff changeset	561 testclasses.append(SystemRandom_TestBasicOps)
33406 483c0b4b6332 SF bug #1022010: Import random fails Raymond Hettinger <python@rcn.com> parents: 33329 diff changeset	562
28944 faf2cb5070b4 Apply the simplified test_support boilerplate. Raymond Hettinger <python@rcn.com> parents: 28937 diff changeset	563 test_support.run_unittest(*testclasses)
26829 f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	564
27668 8e5de43529b7 Add refcount test. Raymond Hettinger <python@rcn.com> parents: 27279 diff changeset	565 # verify reference counting
8e5de43529b7 Add refcount test. Raymond Hettinger <python@rcn.com> parents: 27279 diff changeset	566 import sys
8e5de43529b7 Add refcount test. Raymond Hettinger <python@rcn.com> parents: 27279 diff changeset	567 if verbose and hasattr(sys, "gettotalrefcount"):
28937 c5f18db1c966 Simplify ref count test. Raymond Hettinger <python@rcn.com> parents: 27996 diff changeset	568 counts = [None] * 5
c5f18db1c966 Simplify ref count test. Raymond Hettinger <python@rcn.com> parents: 27996 diff changeset	569 for i in xrange(len(counts)):
28944 faf2cb5070b4 Apply the simplified test_support boilerplate. Raymond Hettinger <python@rcn.com> parents: 28937 diff changeset	570 test_support.run_unittest(*testclasses)
28937 c5f18db1c966 Simplify ref count test. Raymond Hettinger <python@rcn.com> parents: 27996 diff changeset	571 counts[i] = sys.gettotalrefcount()
27668 8e5de43529b7 Add refcount test. Raymond Hettinger <python@rcn.com> parents: 27279 diff changeset	572 print counts
8e5de43529b7 Add refcount test. Raymond Hettinger <python@rcn.com> parents: 27279 diff changeset	573
26829 f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	574 if __name__ == "__main__":
27668 8e5de43529b7 Add refcount test. Raymond Hettinger <python@rcn.com> parents: 27279 diff changeset	575 test_main(verbose=True)

Mercurial > cpython-lru_cache-weakref

annotate Lib/test/test_random.py @ 58765:2ba7677bfa95 legacy-trunk