cpython-lru_cache-weakref: Lib/test/test

annotate Lib/test/test_random.py @ 78069:a8aa77458e12 3.2

Add c_size_t and c_ssize_t to the ctypes summary table.

author	Antoine Pitrou <solipsis@pitrou.net>
date	Thu, 12 Jul 2012 20:31:50 +0200
parents	f4f56efa2870
children	41e97652a9f9 62d9f1285f51

rev	line source
59522 fcfe324663a6 convert shebang lines: python -> python3 Benjamin Peterson <benjamin@python.org> parents: 58415 diff changeset	1 #!/usr/bin/env python3
26829 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
59615 b93cfafcdcad Merged revisions 78093 via svnmerge from Georg Brandl <georg@python.org> parents: 59522 diff changeset	8 from math import log, exp, pi, fsum, sin
47201 9c930a32cebb #2621 rename test.test_support to test.support Benjamin Peterson <benjamin@python.org> parents: 46034 diff changeset	9 from test import 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	10
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	11 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	12 # 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	13 # 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	14 # 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	15
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	16 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	17 """Helper function to make a list of random numbers"""
41642 a6eb6acfe04a Merged revisions 55007-55179 via svnmerge from Guido van Rossum <guido@python.org> parents: 40775 diff changeset	18 return [self.gen.random() for i in range(n)]
26829 f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	19
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	20 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	21 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	22 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	23 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	24 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	25 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	26 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	27
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	28 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	29 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	30 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	31 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	32 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	33 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	34 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	35
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	36 def test_seedargs(self):
40644 1c26505c1de6 Rip out 'long' and 'L'-suffixed integer literals. Guido van Rossum <guido@python.org> parents: 40582 diff changeset	37 for arg in [None, 0, 0, 1, 1, -1, -1, 1020, -(1020),
26829 f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	38 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	39 self.gen.seed(arg)
41642 a6eb6acfe04a Merged revisions 55007-55179 via svnmerge from Guido van Rossum <guido@python.org> parents: 40775 diff changeset	40 for arg in [list(range(3)), dict(one=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	41 self.assertRaises(TypeError, self.gen.seed, arg)
64510 65642f085c55 Document which part of the random module module are guaranteed. Raymond Hettinger <python@rcn.com> parents: 59615 diff changeset	42 self.assertRaises(TypeError, self.gen.seed, 1, 2, 3, 4)
32562 a160b9669db1 Add some tests for corner cases. Raymond Hettinger <python@rcn.com> parents: 30929 diff changeset	43 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	44
64517 e83af141dffd Fix corner case for Random.choice() and add tests. Raymond Hettinger <python@rcn.com> parents: 64512 diff changeset	45 def test_choice(self):
e83af141dffd Fix corner case for Random.choice() and add tests. Raymond Hettinger <python@rcn.com> parents: 64512 diff changeset	46 choice = self.gen.choice
e83af141dffd Fix corner case for Random.choice() and add tests. Raymond Hettinger <python@rcn.com> parents: 64512 diff changeset	47 with self.assertRaises(IndexError):
e83af141dffd Fix corner case for Random.choice() and add tests. Raymond Hettinger <python@rcn.com> parents: 64512 diff changeset	48 choice([])
e83af141dffd Fix corner case for Random.choice() and add tests. Raymond Hettinger <python@rcn.com> parents: 64512 diff changeset	49 self.assertEqual(choice([50]), 50)
e83af141dffd Fix corner case for Random.choice() and add tests. Raymond Hettinger <python@rcn.com> parents: 64512 diff changeset	50 self.assertIn(choice([25, 75]), [25, 75])
e83af141dffd Fix corner case for Random.choice() and add tests. Raymond Hettinger <python@rcn.com> parents: 64512 diff changeset	51
26829 f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	52 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	53 # 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	54 # 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	55 N = 100
41642 a6eb6acfe04a Merged revisions 55007-55179 via svnmerge from Guido van Rossum <guido@python.org> parents: 40775 diff changeset	56 population = range(N)
a6eb6acfe04a Merged revisions 55007-55179 via svnmerge from Guido van Rossum <guido@python.org> parents: 40775 diff changeset	57 for k in range(N+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	58 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	59 self.assertEqual(len(s), k)
30929 54ff68e973ee * Migrate set() and frozenset() from the sandbox. Raymond Hettinger <python@rcn.com> parents: 30649 diff changeset	60 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	61 self.assertEqual(len(uniq), k)
55593 0bb232078b8c Merged revisions 73715 via svnmerge from Georg Brandl <georg@python.org> parents: 51738 diff changeset	62 self.assertTrue(uniq <= set(population))
27081 3199e974f55a Test an edge case for sample(). Raymond Hettinger <python@rcn.com> parents: 26829 diff changeset	63 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	64
27279 9de064c7a247 * Migrate sample distribution test from random.py to test_random.py. Raymond Hettinger <python@rcn.com> parents: 27094 diff changeset	65 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	66 # 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	67 # 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	68 n = 5
9de064c7a247 * Migrate sample distribution test from random.py to test_random.py. Raymond Hettinger <python@rcn.com> parents: 27094 diff changeset	69 pop = range(n)
9de064c7a247 * Migrate sample distribution test from random.py to test_random.py. Raymond Hettinger <python@rcn.com> parents: 27094 diff changeset	70 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	71 def factorial(n):
39627 e86c0b29b4a2 Kill reduce(). A coproduction of John Reese, Jacques Frechet, and Alex M. Guido van Rossum <guido@python.org> parents: 37879 diff changeset	72 if n == 0:
e86c0b29b4a2 Kill reduce(). A coproduction of John Reese, Jacques Frechet, and Alex M. Guido van Rossum <guido@python.org> parents: 37879 diff changeset	73 return 1
e86c0b29b4a2 Kill reduce(). A coproduction of John Reese, Jacques Frechet, and Alex M. Guido van Rossum <guido@python.org> parents: 37879 diff changeset	74 return n * factorial(n - 1)
41642 a6eb6acfe04a Merged revisions 55007-55179 via svnmerge from Guido van Rossum <guido@python.org> parents: 40775 diff changeset	75 for k in range(n):
33597 355b15e4a4da Use floor division operator. Raymond Hettinger <python@rcn.com> parents: 33481 diff changeset	76 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	77 perms = {}
41642 a6eb6acfe04a Merged revisions 55007-55179 via svnmerge from Guido van Rossum <guido@python.org> parents: 40775 diff changeset	78 for i in range(trials):
27279 9de064c7a247 * Migrate sample distribution test from random.py to test_random.py. Raymond Hettinger <python@rcn.com> parents: 27094 diff changeset	79 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	80 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	81 break
9de064c7a247 * Migrate sample distribution test from random.py to test_random.py. Raymond Hettinger <python@rcn.com> parents: 27094 diff changeset	82 else:
9de064c7a247 * Migrate sample distribution test from random.py to test_random.py. Raymond Hettinger <python@rcn.com> parents: 27094 diff changeset	83 self.fail()
9de064c7a247 * Migrate sample distribution test from random.py to test_random.py. Raymond Hettinger <python@rcn.com> parents: 27094 diff changeset	84
30346 aaf6b6a8510b SF bug #801342: Bug (documentation or real, your choice) in random.sample. Raymond Hettinger <python@rcn.com> parents: 30114 diff changeset	85 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	86 # 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	87 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	88 self.gen.sample(range(20), 2)
41642 a6eb6acfe04a Merged revisions 55007-55179 via svnmerge from Guido van Rossum <guido@python.org> parents: 40775 diff changeset	89 self.gen.sample(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	90 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	91 self.gen.sample(tuple('abcdefghijklmnopqrst'), 2)
aaf6b6a8510b SF bug #801342: Bug (documentation or real, your choice) in random.sample. Raymond Hettinger <python@rcn.com> parents: 30114 diff changeset	92
37879 f731859e9b3b Merge p3yk branch with the trunk up to revision 45595. This breaks a fair Thomas Wouters <thomas@python.org> parents: 33597 diff changeset	93 def test_sample_on_dicts(self):
44665 91928d99ebb0 Take Tim's advice and have random.sample() support only sequences and sets. Raymond Hettinger <python@rcn.com> parents: 44662 diff changeset	94 self.assertRaises(TypeError, self.gen.sample, dict.fromkeys('abcdef'), 2)
37879 f731859e9b3b Merge p3yk branch with the trunk up to revision 45595. This breaks a fair Thomas Wouters <thomas@python.org> parents: 33597 diff changeset	95
26829 f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	96 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	97 # 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	98 # 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	99 # 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	100
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	101 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	102 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	103 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	104 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	105
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	106 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	107 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	108 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	109
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	110 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	111 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	112
29586 8b163370372d SF bug #759889: Pickling of Random is broken Raymond Hettinger <python@rcn.com> parents: 28944 diff changeset	113 def test_pickling(self):
8b163370372d SF bug #759889: Pickling of Random is broken Raymond Hettinger <python@rcn.com> parents: 28944 diff changeset	114 state = pickle.dumps(self.gen)
41642 a6eb6acfe04a Merged revisions 55007-55179 via svnmerge from Guido van Rossum <guido@python.org> parents: 40775 diff changeset	115 origseq = [self.gen.random() for i in range(10)]
29586 8b163370372d SF bug #759889: Pickling of Random is broken Raymond Hettinger <python@rcn.com> parents: 28944 diff changeset	116 newgen = pickle.loads(state)
41642 a6eb6acfe04a Merged revisions 55007-55179 via svnmerge from Guido van Rossum <guido@python.org> parents: 40775 diff changeset	117 restoredseq = [newgen.random() for i in range(10)]
29586 8b163370372d SF bug #759889: Pickling of Random is broken Raymond Hettinger <python@rcn.com> parents: 28944 diff changeset	118 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	119
44129 b637349336fd Merged revisions 59275-59303 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 43285 diff changeset	120 def test_bug_1727780(self):
b637349336fd Merged revisions 59275-59303 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 43285 diff changeset	121 # verify that version-2-pickles can be loaded
b637349336fd Merged revisions 59275-59303 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 43285 diff changeset	122 # fine, whether they are created on 32-bit or 64-bit
b637349336fd Merged revisions 59275-59303 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 43285 diff changeset	123 # platforms, and that version-3-pickles load fine.
b637349336fd Merged revisions 59275-59303 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 43285 diff changeset	124 files = [("randv2_32.pck", 780),
b637349336fd Merged revisions 59275-59303 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 43285 diff changeset	125 ("randv2_64.pck", 866),
b637349336fd Merged revisions 59275-59303 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 43285 diff changeset	126 ("randv3.pck", 343)]
b637349336fd Merged revisions 59275-59303 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 43285 diff changeset	127 for file, value in files:
47201 9c930a32cebb #2621 rename test.test_support to test.support Benjamin Peterson <benjamin@python.org> parents: 46034 diff changeset	128 f = open(support.findfile(file),"rb")
44129 b637349336fd Merged revisions 59275-59303 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 43285 diff changeset	129 r = pickle.load(f)
b637349336fd Merged revisions 59275-59303 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 43285 diff changeset	130 f.close()
64512 c2f8418a0e14 Issues #7889, #9025 and #9379: Improvements to the random module. Raymond Hettinger <python@rcn.com> parents: 64510 diff changeset	131 self.assertEqual(int(r.random()*1000), value)
c2f8418a0e14 Issues #7889, #9025 and #9379: Improvements to the random module. Raymond Hettinger <python@rcn.com> parents: 64510 diff changeset	132
c2f8418a0e14 Issues #7889, #9025 and #9379: Improvements to the random module. Raymond Hettinger <python@rcn.com> parents: 64510 diff changeset	133 def test_bug_9025(self):
c2f8418a0e14 Issues #7889, #9025 and #9379: Improvements to the random module. Raymond Hettinger <python@rcn.com> parents: 64510 diff changeset	134 # Had problem with an uneven distribution in int(n*random())
c2f8418a0e14 Issues #7889, #9025 and #9379: Improvements to the random module. Raymond Hettinger <python@rcn.com> parents: 64510 diff changeset	135 # Verify the fix by checking that distributions fall within expectations.
c2f8418a0e14 Issues #7889, #9025 and #9379: Improvements to the random module. Raymond Hettinger <python@rcn.com> parents: 64510 diff changeset	136 n = 100000
c2f8418a0e14 Issues #7889, #9025 and #9379: Improvements to the random module. Raymond Hettinger <python@rcn.com> parents: 64510 diff changeset	137 randrange = self.gen.randrange
c2f8418a0e14 Issues #7889, #9025 and #9379: Improvements to the random module. Raymond Hettinger <python@rcn.com> parents: 64510 diff changeset	138 k = sum(randrange(6755399441055744) % 3 == 2 for i in range(n))
c2f8418a0e14 Issues #7889, #9025 and #9379: Improvements to the random module. Raymond Hettinger <python@rcn.com> parents: 64510 diff changeset	139 self.assertTrue(0.30 < k/n < .37, (k/n))
44129 b637349336fd Merged revisions 59275-59303 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 43285 diff changeset	140
33481 eacdc29606d8 SF #1027105: HardwareRandom should be renamed OSRandom Raymond Hettinger <python@rcn.com> parents: 33409 diff changeset	141 class SystemRandom_TestBasicOps(TestBasicOps):
eacdc29606d8 SF #1027105: HardwareRandom should be renamed OSRandom Raymond Hettinger <python@rcn.com> parents: 33409 diff changeset	142 gen = random.SystemRandom()
33329 97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	143
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	144 def test_autoseed(self):
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	145 # 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	146 self.gen.seed()
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	147
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	148 def test_saverestore(self):
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	149 self.assertRaises(NotImplementedError, self.gen.getstate)
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	150 self.assertRaises(NotImplementedError, self.gen.setstate, None)
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	151
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	152 def test_seedargs(self):
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	153 # 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	154 self.gen.seed(100)
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	155
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	156 def test_gauss(self):
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	157 self.gen.gauss_next = None
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	158 self.gen.seed(100)
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	159 self.assertEqual(self.gen.gauss_next, None)
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	160
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	161 def test_pickling(self):
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	162 self.assertRaises(NotImplementedError, pickle.dumps, self.gen)
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	163
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	164 def test_53_bits_per_float(self):
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	165 # 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	166 span = 2 ** 53
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	167 cum = 0
41642 a6eb6acfe04a Merged revisions 55007-55179 via svnmerge from Guido van Rossum <guido@python.org> parents: 40775 diff changeset	168 for i in range(100):
33329 97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	169 cum \|= int(self.gen.random() * span)
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	170 self.assertEqual(cum, span-1)
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	171
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	172 def test_bigrand(self):
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	173 # 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	174 # 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	175 span = 2 ** 500
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	176 cum = 0
41642 a6eb6acfe04a Merged revisions 55007-55179 via svnmerge from Guido van Rossum <guido@python.org> parents: 40775 diff changeset	177 for i in range(100):
33329 97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	178 r = self.gen.randrange(span)
55593 0bb232078b8c Merged revisions 73715 via svnmerge from Georg Brandl <georg@python.org> parents: 51738 diff changeset	179 self.assertTrue(0 <= r < span)
33329 97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	180 cum \|= r
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	181 self.assertEqual(cum, span-1)
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	182
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	183 def test_bigrand_ranges(self):
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	184 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	185 start = self.gen.randrange(2 ** i)
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	186 stop = self.gen.randrange(2 ** (i-2))
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	187 if stop <= start:
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	188 return
55593 0bb232078b8c Merged revisions 73715 via svnmerge from Georg Brandl <georg@python.org> parents: 51738 diff changeset	189 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	190
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	191 def test_rangelimits(self):
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	192 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	193 self.assertEqual(set(range(start,stop)),
41642 a6eb6acfe04a Merged revisions 55007-55179 via svnmerge from Guido van Rossum <guido@python.org> parents: 40775 diff changeset	194 set([self.gen.randrange(start,stop) for i in range(100)]))
33329 97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	195
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	196 def test_genrandbits(self):
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	197 # Verify ranges
41642 a6eb6acfe04a Merged revisions 55007-55179 via svnmerge from Guido van Rossum <guido@python.org> parents: 40775 diff changeset	198 for k in range(1, 1000):
55593 0bb232078b8c Merged revisions 73715 via svnmerge from Georg Brandl <georg@python.org> parents: 51738 diff changeset	199 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	200
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	201 # Verify all bits active
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	202 getbits = self.gen.getrandbits
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	203 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	204 cum = 0
41642 a6eb6acfe04a Merged revisions 55007-55179 via svnmerge from Guido van Rossum <guido@python.org> parents: 40775 diff changeset	205 for i in range(100):
33329 97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	206 cum \|= getbits(span)
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	207 self.assertEqual(cum, 2**span-1)
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	208
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	209 # Verify argument checking
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	210 self.assertRaises(TypeError, self.gen.getrandbits)
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	211 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	212 self.assertRaises(ValueError, self.gen.getrandbits, 0)
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	213 self.assertRaises(ValueError, self.gen.getrandbits, -1)
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	214 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	215
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	216 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	217 # 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	218 # 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	219 # is equal to or one greater than the number of bits in n
41642 a6eb6acfe04a Merged revisions 55007-55179 via svnmerge from Guido van Rossum <guido@python.org> parents: 40775 diff changeset	220 for i in range(1, 1000):
40644 1c26505c1de6 Rip out 'long' and 'L'-suffixed integer literals. Guido van Rossum <guido@python.org> parents: 40582 diff changeset	221 n = 1 << i # check an exact power of two
33329 97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	222 numbits = i+1
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	223 k = int(1.00001 + _log(n, 2))
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	224 self.assertEqual(k, numbits)
42188 81930eaae715 Patch by Christian Heimes to change self.assert_(x == y) into Guido van Rossum <guido@python.org> parents: 41642 diff changeset	225 self.assertEqual(n, 2**(k-1))
33329 97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	226
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	227 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	228 k = int(1.00001 + _log(n, 2))
58415 f6a626f34840 use assert[Not]In where appropriate Benjamin Peterson <benjamin@python.org> parents: 55593 diff changeset	229 self.assertIn(k, [numbits, numbits+1])
55593 0bb232078b8c Merged revisions 73715 via svnmerge from Georg Brandl <georg@python.org> parents: 51738 diff changeset	230 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	231
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	232 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	233 k = int(1.00001 + _log(n, 2))
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	234 self.assertEqual(k, numbits) # note the stronger assertion
55593 0bb232078b8c Merged revisions 73715 via svnmerge from Georg Brandl <georg@python.org> parents: 51738 diff changeset	235 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	236
97fca735c8da Teach the random module about os.urandom(). Raymond Hettinger <python@rcn.com> parents: 32562 diff changeset	237
26829 f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	238 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	239 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	240
64510 65642f085c55 Document which part of the random module module are guaranteed. Raymond Hettinger <python@rcn.com> parents: 59615 diff changeset	241 def test_guaranteed_stable(self):
65642f085c55 Document which part of the random module module are guaranteed. Raymond Hettinger <python@rcn.com> parents: 59615 diff changeset	242 # These sequences are guaranteed to stay the same across versions of python
65642f085c55 Document which part of the random module module are guaranteed. Raymond Hettinger <python@rcn.com> parents: 59615 diff changeset	243 self.gen.seed(3456147, version=1)
65642f085c55 Document which part of the random module module are guaranteed. Raymond Hettinger <python@rcn.com> parents: 59615 diff changeset	244 self.assertEqual([self.gen.random().hex() for i in range(4)],
65642f085c55 Document which part of the random module module are guaranteed. Raymond Hettinger <python@rcn.com> parents: 59615 diff changeset	245 ['0x1.ac362300d90d2p-1', '0x1.9d16f74365005p-1',
65642f085c55 Document which part of the random module module are guaranteed. Raymond Hettinger <python@rcn.com> parents: 59615 diff changeset	246 '0x1.1ebb4352e4c4dp-1', '0x1.1a7422abf9c11p-1'])
65642f085c55 Document which part of the random module module are guaranteed. Raymond Hettinger <python@rcn.com> parents: 59615 diff changeset	247 self.gen.seed("the quick brown fox", version=2)
65642f085c55 Document which part of the random module module are guaranteed. Raymond Hettinger <python@rcn.com> parents: 59615 diff changeset	248 self.assertEqual([self.gen.random().hex() for i in range(4)],
66786 f4f56efa2870 Update whatsnew. Salt the random number seed. Raymond Hettinger <python@rcn.com> parents: 64524 diff changeset	249 ['0x1.1239ddfb11b7cp-3', '0x1.b3cbb5c51b120p-4',
f4f56efa2870 Update whatsnew. Salt the random number seed. Raymond Hettinger <python@rcn.com> parents: 64524 diff changeset	250 '0x1.8c4f55116b60fp-1', '0x1.63eb525174a27p-1'])
64510 65642f085c55 Document which part of the random module module are guaranteed. Raymond Hettinger <python@rcn.com> parents: 59615 diff changeset	251
32562 a160b9669db1 Add some tests for corner cases. Raymond Hettinger <python@rcn.com> parents: 30929 diff changeset	252 def test_setstate_first_arg(self):
a160b9669db1 Add some tests for corner cases. Raymond Hettinger <python@rcn.com> parents: 30929 diff changeset	253 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	254
a160b9669db1 Add some tests for corner cases. Raymond Hettinger <python@rcn.com> parents: 30929 diff changeset	255 def test_setstate_middle_arg(self):
a160b9669db1 Add some tests for corner cases. Raymond Hettinger <python@rcn.com> parents: 30929 diff changeset	256 # Wrong type, s/b tuple
a160b9669db1 Add some tests for corner cases. Raymond Hettinger <python@rcn.com> parents: 30929 diff changeset	257 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	258 # Wrong length, s/b 625
a160b9669db1 Add some tests for corner cases. Raymond Hettinger <python@rcn.com> parents: 30929 diff changeset	259 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	260 # Wrong type, s/b tuple of 625 ints
a160b9669db1 Add some tests for corner cases. Raymond Hettinger <python@rcn.com> parents: 30929 diff changeset	261 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	262 # Last element s/b an int also
a160b9669db1 Add some tests for corner cases. Raymond Hettinger <python@rcn.com> parents: 30929 diff changeset	263 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	264
26829 f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	265 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	266 # 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	267 # 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	268 # 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	269 # 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	270 # 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	271 # 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	272 # 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	273 # 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	274 # 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	275 # 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	276 # 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	277 # 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	278 # }
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	279 # }
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	280 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	281 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	282 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	283 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	284 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	285 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	286 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	287 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	288 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	289 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	290
40644 1c26505c1de6 Rip out 'long' and 'L'-suffixed integer literals. Guido van Rossum <guido@python.org> parents: 40582 diff changeset	291 self.gen.seed(61731 + (24903<<32) + (614<<64) + (42143<<96))
26829 f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	292 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	293 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	294 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	295
26829 f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	296 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	297 # 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	298 # 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	299 # 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	300 # 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	301 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	302
40644 1c26505c1de6 Rip out 'long' and 'L'-suffixed integer literals. Guido van Rossum <guido@python.org> parents: 40582 diff changeset	303 expected = [0x0eab3258d2231f,
1c26505c1de6 Rip out 'long' and 'L'-suffixed integer literals. Guido van Rossum <guido@python.org> parents: 40582 diff changeset	304 0x1b89db315277a5,
1c26505c1de6 Rip out 'long' and 'L'-suffixed integer literals. Guido van Rossum <guido@python.org> parents: 40582 diff changeset	305 0x1db622a5518016,
1c26505c1de6 Rip out 'long' and 'L'-suffixed integer literals. Guido van Rossum <guido@python.org> parents: 40582 diff changeset	306 0x0b7f9af0d575bf,
1c26505c1de6 Rip out 'long' and 'L'-suffixed integer literals. Guido van Rossum <guido@python.org> parents: 40582 diff changeset	307 0x029e4c4db82240,
1c26505c1de6 Rip out 'long' and 'L'-suffixed integer literals. Guido van Rossum <guido@python.org> parents: 40582 diff changeset	308 0x04961892f5d673,
1c26505c1de6 Rip out 'long' and 'L'-suffixed integer literals. Guido van Rossum <guido@python.org> parents: 40582 diff changeset	309 0x02b291598e4589,
1c26505c1de6 Rip out 'long' and 'L'-suffixed integer literals. Guido van Rossum <guido@python.org> parents: 40582 diff changeset	310 0x11388382c15694,
1c26505c1de6 Rip out 'long' and 'L'-suffixed integer literals. Guido van Rossum <guido@python.org> parents: 40582 diff changeset	311 0x02dad977c9e1fe,
1c26505c1de6 Rip out 'long' and 'L'-suffixed integer literals. Guido van Rossum <guido@python.org> parents: 40582 diff changeset	312 0x191d96d4d334c6]
1c26505c1de6 Rip out 'long' and 'L'-suffixed integer literals. Guido van Rossum <guido@python.org> parents: 40582 diff changeset	313 self.gen.seed(61731 + (24903<<32) + (614<<64) + (42143<<96))
26829 f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	314 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	315 for a, e in zip(actual, expected):
40644 1c26505c1de6 Rip out 'long' and 'L'-suffixed integer literals. Guido van Rossum <guido@python.org> parents: 40582 diff changeset	316 self.assertEqual(int(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	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_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	319 # 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	320 # 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	321 # 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	322 # 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	323 # so don't make this horribly big.
40644 1c26505c1de6 Rip out 'long' and 'L'-suffixed integer literals. Guido van Rossum <guido@python.org> parents: 40582 diff changeset	324 seed = (1 << (10000 * 8)) - 1 # about 10K bytes
26829 f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	325 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	326
30649 8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	327 def test_53_bits_per_float(self):
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	328 # 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	329 span = 2 ** 53
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	330 cum = 0
41642 a6eb6acfe04a Merged revisions 55007-55179 via svnmerge from Guido van Rossum <guido@python.org> parents: 40775 diff changeset	331 for i in range(100):
30649 8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	332 cum \|= int(self.gen.random() * span)
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	333 self.assertEqual(cum, span-1)
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	334
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	335 def test_bigrand(self):
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	336 # 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	337 # 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	338 span = 2 ** 500
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	339 cum = 0
41642 a6eb6acfe04a Merged revisions 55007-55179 via svnmerge from Guido van Rossum <guido@python.org> parents: 40775 diff changeset	340 for i in range(100):
30649 8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	341 r = self.gen.randrange(span)
55593 0bb232078b8c Merged revisions 73715 via svnmerge from Georg Brandl <georg@python.org> parents: 51738 diff changeset	342 self.assertTrue(0 <= r < span)
30649 8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	343 cum \|= r
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	344 self.assertEqual(cum, span-1)
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	345
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	346 def test_bigrand_ranges(self):
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	347 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	348 start = self.gen.randrange(2 ** i)
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	349 stop = self.gen.randrange(2 ** (i-2))
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	350 if stop <= start:
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	351 return
55593 0bb232078b8c Merged revisions 73715 via svnmerge from Georg Brandl <georg@python.org> parents: 51738 diff changeset	352 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	353
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	354 def test_rangelimits(self):
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	355 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	356 self.assertEqual(set(range(start,stop)),
41642 a6eb6acfe04a Merged revisions 55007-55179 via svnmerge from Guido van Rossum <guido@python.org> parents: 40775 diff changeset	357 set([self.gen.randrange(start,stop) for i in range(100)]))
30649 8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	358
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	359 def test_genrandbits(self):
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	360 # Verify cross-platform repeatability
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	361 self.gen.seed(1234567)
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	362 self.assertEqual(self.gen.getrandbits(100),
40644 1c26505c1de6 Rip out 'long' and 'L'-suffixed integer literals. Guido van Rossum <guido@python.org> parents: 40582 diff changeset	363 97904845777343510404718956115)
30649 8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	364 # Verify ranges
41642 a6eb6acfe04a Merged revisions 55007-55179 via svnmerge from Guido van Rossum <guido@python.org> parents: 40775 diff changeset	365 for k in range(1, 1000):
55593 0bb232078b8c Merged revisions 73715 via svnmerge from Georg Brandl <georg@python.org> parents: 51738 diff changeset	366 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	367
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	368 # Verify all bits active
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	369 getbits = self.gen.getrandbits
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	370 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	371 cum = 0
41642 a6eb6acfe04a Merged revisions 55007-55179 via svnmerge from Guido van Rossum <guido@python.org> parents: 40775 diff changeset	372 for i in range(100):
30649 8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	373 cum \|= getbits(span)
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	374 self.assertEqual(cum, 2**span-1)
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	375
32562 a160b9669db1 Add some tests for corner cases. Raymond Hettinger <python@rcn.com> parents: 30929 diff changeset	376 # Verify argument checking
a160b9669db1 Add some tests for corner cases. Raymond Hettinger <python@rcn.com> parents: 30929 diff changeset	377 self.assertRaises(TypeError, self.gen.getrandbits)
a160b9669db1 Add some tests for corner cases. Raymond Hettinger <python@rcn.com> parents: 30929 diff changeset	378 self.assertRaises(TypeError, self.gen.getrandbits, 'a')
a160b9669db1 Add some tests for corner cases. Raymond Hettinger <python@rcn.com> parents: 30929 diff changeset	379 self.assertRaises(TypeError, self.gen.getrandbits, 1, 2)
a160b9669db1 Add some tests for corner cases. Raymond Hettinger <python@rcn.com> parents: 30929 diff changeset	380 self.assertRaises(ValueError, self.gen.getrandbits, 0)
a160b9669db1 Add some tests for corner cases. Raymond Hettinger <python@rcn.com> parents: 30929 diff changeset	381 self.assertRaises(ValueError, self.gen.getrandbits, -1)
a160b9669db1 Add some tests for corner cases. Raymond Hettinger <python@rcn.com> parents: 30929 diff changeset	382
30649 8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	383 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	384 # 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	385 # 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	386 # is equal to or one greater than the number of bits in n
41642 a6eb6acfe04a Merged revisions 55007-55179 via svnmerge from Guido van Rossum <guido@python.org> parents: 40775 diff changeset	387 for i in range(1, 1000):
40644 1c26505c1de6 Rip out 'long' and 'L'-suffixed integer literals. Guido van Rossum <guido@python.org> parents: 40582 diff changeset	388 n = 1 << i # check an exact power of two
30649 8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	389 numbits = i+1
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	390 k = int(1.00001 + _log(n, 2))
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	391 self.assertEqual(k, numbits)
42188 81930eaae715 Patch by Christian Heimes to change self.assert_(x == y) into Guido van Rossum <guido@python.org> parents: 41642 diff changeset	392 self.assertEqual(n, 2**(k-1))
30649 8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	393
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	394 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	395 k = int(1.00001 + _log(n, 2))
58415 f6a626f34840 use assert[Not]In where appropriate Benjamin Peterson <benjamin@python.org> parents: 55593 diff changeset	396 self.assertIn(k, [numbits, numbits+1])
55593 0bb232078b8c Merged revisions 73715 via svnmerge from Georg Brandl <georg@python.org> parents: 51738 diff changeset	397 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	398
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	399 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	400 k = int(1.00001 + _log(n, 2))
8841d34a2c37 SF bug #812202: randint is always even Raymond Hettinger <python@rcn.com> parents: 30346 diff changeset	401 self.assertEqual(k, numbits) # note the stronger assertion
55593 0bb232078b8c Merged revisions 73715 via svnmerge from Georg Brandl <georg@python.org> parents: 51738 diff changeset	402 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	403
40582 67287ca4e1e8 Merged revisions 53005-53303 via svnmerge from Thomas Wouters <thomas@python.org> parents: 39627 diff changeset	404 def test_randrange_bug_1590891(self):
67287ca4e1e8 Merged revisions 53005-53303 via svnmerge from Thomas Wouters <thomas@python.org> parents: 39627 diff changeset	405 start = 1000000000000
67287ca4e1e8 Merged revisions 53005-53303 via svnmerge from Thomas Wouters <thomas@python.org> parents: 39627 diff changeset	406 stop = -100000000000000000000
67287ca4e1e8 Merged revisions 53005-53303 via svnmerge from Thomas Wouters <thomas@python.org> parents: 39627 diff changeset	407 step = -200
67287ca4e1e8 Merged revisions 53005-53303 via svnmerge from Thomas Wouters <thomas@python.org> parents: 39627 diff changeset	408 x = self.gen.randrange(start, stop, step)
55593 0bb232078b8c Merged revisions 73715 via svnmerge from Georg Brandl <georg@python.org> parents: 51738 diff changeset	409 self.assertTrue(stop < x <= start)
40582 67287ca4e1e8 Merged revisions 53005-53303 via svnmerge from Thomas Wouters <thomas@python.org> parents: 39627 diff changeset	410 self.assertEqual((x+stop)%step, 0)
67287ca4e1e8 Merged revisions 53005-53303 via svnmerge from Thomas Wouters <thomas@python.org> parents: 39627 diff changeset	411
51738 78e311f277a7 Inline coefficients in gamma(). Add reflection formula. Add comments. Raymond Hettinger <python@rcn.com> parents: 48606 diff changeset	412 def gamma(z, sqrt2pi=(2.0pi)*0.5):
78e311f277a7 Inline coefficients in gamma(). Add reflection formula. Add comments. Raymond Hettinger <python@rcn.com> parents: 48606 diff changeset	413 # Reflection to right half of complex plane
78e311f277a7 Inline coefficients in gamma(). Add reflection formula. Add comments. Raymond Hettinger <python@rcn.com> parents: 48606 diff changeset	414 if z < 0.5:
78e311f277a7 Inline coefficients in gamma(). Add reflection formula. Add comments. Raymond Hettinger <python@rcn.com> parents: 48606 diff changeset	415 return pi / sin(pi*z) / gamma(1.0-z)
78e311f277a7 Inline coefficients in gamma(). Add reflection formula. Add comments. Raymond Hettinger <python@rcn.com> parents: 48606 diff changeset	416 # Lanczos approximation with g=7
78e311f277a7 Inline coefficients in gamma(). Add reflection formula. Add comments. Raymond Hettinger <python@rcn.com> parents: 48606 diff changeset	417 az = z + (7.0 - 0.5)
78e311f277a7 Inline coefficients in gamma(). Add reflection formula. Add comments. Raymond Hettinger <python@rcn.com> parents: 48606 diff changeset	418 return az ** (z-0.5) / exp(az) * sqrt2pi * fsum([
78e311f277a7 Inline coefficients in gamma(). Add reflection formula. Add comments. Raymond Hettinger <python@rcn.com> parents: 48606 diff changeset	419 0.9999999999995183,
78e311f277a7 Inline coefficients in gamma(). Add reflection formula. Add comments. Raymond Hettinger <python@rcn.com> parents: 48606 diff changeset	420 676.5203681218835 / z,
78e311f277a7 Inline coefficients in gamma(). Add reflection formula. Add comments. Raymond Hettinger <python@rcn.com> parents: 48606 diff changeset	421 -1259.139216722289 / (z+1.0),
78e311f277a7 Inline coefficients in gamma(). Add reflection formula. Add comments. Raymond Hettinger <python@rcn.com> parents: 48606 diff changeset	422 771.3234287757674 / (z+2.0),
78e311f277a7 Inline coefficients in gamma(). Add reflection formula. Add comments. Raymond Hettinger <python@rcn.com> parents: 48606 diff changeset	423 -176.6150291498386 / (z+3.0),
78e311f277a7 Inline coefficients in gamma(). Add reflection formula. Add comments. Raymond Hettinger <python@rcn.com> parents: 48606 diff changeset	424 12.50734324009056 / (z+4.0),
78e311f277a7 Inline coefficients in gamma(). Add reflection formula. Add comments. Raymond Hettinger <python@rcn.com> parents: 48606 diff changeset	425 -0.1385710331296526 / (z+5.0),
78e311f277a7 Inline coefficients in gamma(). Add reflection formula. Add comments. Raymond Hettinger <python@rcn.com> parents: 48606 diff changeset	426 0.9934937113930748e-05 / (z+6.0),
78e311f277a7 Inline coefficients in gamma(). Add reflection formula. Add comments. Raymond Hettinger <python@rcn.com> parents: 48606 diff changeset	427 0.1659470187408462e-06 / (z+7.0),
78e311f277a7 Inline coefficients in gamma(). Add reflection formula. Add comments. Raymond Hettinger <python@rcn.com> parents: 48606 diff changeset	428 ])
27094 e841c873a937 Move the statistical tests for four distributions into the unittest suite. Raymond Hettinger <python@rcn.com> parents: 27092 diff changeset	429
27092 7dab27731be8 Add a test case. Raymond Hettinger <python@rcn.com> parents: 27081 diff changeset	430 class TestDistributions(unittest.TestCase):
7dab27731be8 Add a test case. Raymond Hettinger <python@rcn.com> parents: 27081 diff changeset	431 def test_zeroinputs(self):
7dab27731be8 Add a test case. Raymond Hettinger <python@rcn.com> parents: 27081 diff changeset	432 # Verify that distributions can handle a series of zero inputs'
7dab27731be8 Add a test case. Raymond Hettinger <python@rcn.com> parents: 27081 diff changeset	433 g = random.Random()
41642 a6eb6acfe04a Merged revisions 55007-55179 via svnmerge from Guido van Rossum <guido@python.org> parents: 40775 diff changeset	434 x = [g.random() for i in range(50)] + [0.0]*5
27092 7dab27731be8 Add a test case. Raymond Hettinger <python@rcn.com> parents: 27081 diff changeset	435 g.random = x[:].pop; g.uniform(1,10)
7dab27731be8 Add a test case. Raymond Hettinger <python@rcn.com> parents: 27081 diff changeset	436 g.random = x[:].pop; g.paretovariate(1.0)
7dab27731be8 Add a test case. Raymond Hettinger <python@rcn.com> parents: 27081 diff changeset	437 g.random = x[:].pop; g.expovariate(1.0)
7dab27731be8 Add a test case. Raymond Hettinger <python@rcn.com> parents: 27081 diff changeset	438 g.random = x[:].pop; g.weibullvariate(1.0, 1.0)
7dab27731be8 Add a test case. Raymond Hettinger <python@rcn.com> parents: 27081 diff changeset	439 g.random = x[:].pop; g.normalvariate(0.0, 1.0)
7dab27731be8 Add a test case. Raymond Hettinger <python@rcn.com> parents: 27081 diff changeset	440 g.random = x[:].pop; g.gauss(0.0, 1.0)
7dab27731be8 Add a test case. Raymond Hettinger <python@rcn.com> parents: 27081 diff changeset	441 g.random = x[:].pop; g.lognormvariate(0.0, 1.0)
7dab27731be8 Add a test case. Raymond Hettinger <python@rcn.com> parents: 27081 diff changeset	442 g.random = x[:].pop; g.vonmisesvariate(0.0, 1.0)
7dab27731be8 Add a test case. Raymond Hettinger <python@rcn.com> parents: 27081 diff changeset	443 g.random = x[:].pop; g.gammavariate(0.01, 1.0)
7dab27731be8 Add a test case. Raymond Hettinger <python@rcn.com> parents: 27081 diff changeset	444 g.random = x[:].pop; g.gammavariate(1.0, 1.0)
7dab27731be8 Add a test case. Raymond Hettinger <python@rcn.com> parents: 27081 diff changeset	445 g.random = x[:].pop; g.gammavariate(200.0, 1.0)
7dab27731be8 Add a test case. Raymond Hettinger <python@rcn.com> parents: 27081 diff changeset	446 g.random = x[:].pop; g.betavariate(3.0, 3.0)
46034 0a485593ae1c Merged revisions 61724-61725,61731-61735,61737,61739,61741,61743-61744,61753,61761,61765-61767,61769,61773,61776-61778,61780-61783,61788,61793,61796,61807,61813 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 44665 diff changeset	447 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	448
27094 e841c873a937 Move the statistical tests for four distributions into the unittest suite. Raymond Hettinger <python@rcn.com> parents: 27092 diff changeset	449 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	450 # 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	451 # 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	452 g = random.Random()
e841c873a937 Move the statistical tests for four distributions into the unittest suite. Raymond Hettinger <python@rcn.com> parents: 27092 diff changeset	453 N = 5000
41642 a6eb6acfe04a Merged revisions 55007-55179 via svnmerge from Guido van Rossum <guido@python.org> parents: 40775 diff changeset	454 x = [i/float(N) for i in range(1,N)]
27094 e841c873a937 Move the statistical tests for four distributions into the unittest suite. Raymond Hettinger <python@rcn.com> parents: 27092 diff changeset	455 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	456 (g.uniform, (1.0,10.0), (10.0+1.0)/2, (10.0-1.0)**2/12),
46034 0a485593ae1c Merged revisions 61724-61725,61731-61735,61737,61739,61741,61743-61744,61753,61761,61765-61767,61769,61773,61776-61778,61780-61783,61788,61793,61796,61807,61813 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 44665 diff changeset	457 (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	458 (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	459 (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	460 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	461 (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	462 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	463 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	464 y = []
41642 a6eb6acfe04a Merged revisions 55007-55179 via svnmerge from Guido van Rossum <guido@python.org> parents: 40775 diff changeset	465 for i in range(len(x)):
27094 e841c873a937 Move the statistical tests for four distributions into the unittest suite. Raymond Hettinger <python@rcn.com> parents: 27092 diff changeset	466 try:
e841c873a937 Move the statistical tests for four distributions into the unittest suite. Raymond Hettinger <python@rcn.com> parents: 27092 diff changeset	467 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	468 except IndexError:
e841c873a937 Move the statistical tests for four distributions into the unittest suite. Raymond Hettinger <python@rcn.com> parents: 27092 diff changeset	469 pass
e841c873a937 Move the statistical tests for four distributions into the unittest suite. Raymond Hettinger <python@rcn.com> parents: 27092 diff changeset	470 s1 = s2 = 0
e841c873a937 Move the statistical tests for four distributions into the unittest suite. Raymond Hettinger <python@rcn.com> parents: 27092 diff changeset	471 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	472 s1 += e
e841c873a937 Move the statistical tests for four distributions into the unittest suite. Raymond Hettinger <python@rcn.com> parents: 27092 diff changeset	473 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	474 N = len(y)
43285 755fa5187d1b Forgot to run the tests after making the places and msg argument keyword-only. Jeffrey Yasskin <jyasskin@gmail.com> parents: 42629 diff changeset	475 self.assertAlmostEqual(s1/N, mu, places=2)
755fa5187d1b Forgot to run the tests after making the places and msg argument keyword-only. Jeffrey Yasskin <jyasskin@gmail.com> parents: 42629 diff changeset	476 self.assertAlmostEqual(s2/(N-1), sigmasqrd, places=2)
27094 e841c873a937 Move the statistical tests for four distributions into the unittest suite. Raymond Hettinger <python@rcn.com> parents: 27092 diff changeset	477
26829 f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	478 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	479 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	480 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	481 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	482 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	483 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	484
f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	485 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	486 # tests validity but not completeness of the __all__ list
55593 0bb232078b8c Merged revisions 73715 via svnmerge from Georg Brandl <georg@python.org> parents: 51738 diff changeset	487 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	488
40737 6ebad2a6b990 Merged revisions 53451-53537 via svnmerge from Thomas Wouters <thomas@python.org> parents: 40644 diff changeset	489 def test_random_subclass_with_kwargs(self):
6ebad2a6b990 Merged revisions 53451-53537 via svnmerge from Thomas Wouters <thomas@python.org> parents: 40644 diff changeset	490 # SF bug #1486663 -- this used to erroneously raise a TypeError
6ebad2a6b990 Merged revisions 53451-53537 via svnmerge from Thomas Wouters <thomas@python.org> parents: 40644 diff changeset	491 class Subclass(random.Random):
6ebad2a6b990 Merged revisions 53451-53537 via svnmerge from Thomas Wouters <thomas@python.org> parents: 40644 diff changeset	492 def __init__(self, newarg=None):
6ebad2a6b990 Merged revisions 53451-53537 via svnmerge from Thomas Wouters <thomas@python.org> parents: 40644 diff changeset	493 random.Random.__init__(self)
6ebad2a6b990 Merged revisions 53451-53537 via svnmerge from Thomas Wouters <thomas@python.org> parents: 40644 diff changeset	494 Subclass(newarg=1)
6ebad2a6b990 Merged revisions 53451-53537 via svnmerge from Thomas Wouters <thomas@python.org> parents: 40644 diff changeset	495
6ebad2a6b990 Merged revisions 53451-53537 via svnmerge from Thomas Wouters <thomas@python.org> parents: 40644 diff changeset	496
27668 8e5de43529b7 Add refcount test. Raymond Hettinger <python@rcn.com> parents: 27279 diff changeset	497 def test_main(verbose=None):
44662 c25760c21397 Remove defunct parts of the random module Raymond Hettinger <python@rcn.com> parents: 44129 diff changeset	498 testclasses = [MersenneTwister_TestBasicOps,
27092 7dab27731be8 Add a test case. Raymond Hettinger <python@rcn.com> parents: 27081 diff changeset	499 TestDistributions,
33406 483c0b4b6332 SF bug #1022010: Import random fails Raymond Hettinger <python@rcn.com> parents: 33329 diff changeset	500 TestModule]
483c0b4b6332 SF bug #1022010: Import random fails Raymond Hettinger <python@rcn.com> parents: 33329 diff changeset	501
33409 8e23bd3d5b1a Fulfill Martin's request to use try/except rather than a "look before Raymond Hettinger <python@rcn.com> parents: 33406 diff changeset	502 try:
33481 eacdc29606d8 SF #1027105: HardwareRandom should be renamed OSRandom Raymond Hettinger <python@rcn.com> parents: 33409 diff changeset	503 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	504 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	505 pass
8e23bd3d5b1a Fulfill Martin's request to use try/except rather than a "look before Raymond Hettinger <python@rcn.com> parents: 33406 diff changeset	506 else:
33481 eacdc29606d8 SF #1027105: HardwareRandom should be renamed OSRandom Raymond Hettinger <python@rcn.com> parents: 33409 diff changeset	507 testclasses.append(SystemRandom_TestBasicOps)
33406 483c0b4b6332 SF bug #1022010: Import random fails Raymond Hettinger <python@rcn.com> parents: 33329 diff changeset	508
47201 9c930a32cebb #2621 rename test.test_support to test.support Benjamin Peterson <benjamin@python.org> parents: 46034 diff changeset	509 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	510
27668 8e5de43529b7 Add refcount test. Raymond Hettinger <python@rcn.com> parents: 27279 diff changeset	511 # verify reference counting
8e5de43529b7 Add refcount test. Raymond Hettinger <python@rcn.com> parents: 27279 diff changeset	512 import sys
8e5de43529b7 Add refcount test. Raymond Hettinger <python@rcn.com> parents: 27279 diff changeset	513 if verbose and hasattr(sys, "gettotalrefcount"):
28937 c5f18db1c966 Simplify ref count test. Raymond Hettinger <python@rcn.com> parents: 27996 diff changeset	514 counts = [None] * 5
41642 a6eb6acfe04a Merged revisions 55007-55179 via svnmerge from Guido van Rossum <guido@python.org> parents: 40775 diff changeset	515 for i in range(len(counts)):
47201 9c930a32cebb #2621 rename test.test_support to test.support Benjamin Peterson <benjamin@python.org> parents: 46034 diff changeset	516 support.run_unittest(*testclasses)
28937 c5f18db1c966 Simplify ref count test. Raymond Hettinger <python@rcn.com> parents: 27996 diff changeset	517 counts[i] = sys.gettotalrefcount()
40775 4cdf21a0dbbc Fix most trivially-findable print statements. Guido van Rossum <guido@python.org> parents: 40737 diff changeset	518 print(counts)
27668 8e5de43529b7 Add refcount test. Raymond Hettinger <python@rcn.com> parents: 27279 diff changeset	519
26829 f64c96922708 SF patch 658251: Install a C implementation of the Mersenne Twister as the Raymond Hettinger <python@rcn.com> parents: 24364 diff changeset	520 if __name__ == "__main__":
27668 8e5de43529b7 Add refcount test. Raymond Hettinger <python@rcn.com> parents: 27279 diff changeset	521 test_main(verbose=True)

Mercurial > cpython-lru_cache-weakref

annotate Lib/test/test_random.py @ 78069:a8aa77458e12 3.2