summaryrefslogtreecommitdiffstats
path: root/Lib/test/test_zlib.py
diff options
context:
space:
mode:
authorVictor Stinner <vstinner@python.org>2020-04-17 20:54:38 (GMT)
committerGitHub <noreply@github.com>2020-04-17 20:54:38 (GMT)
commit87502ddd710eb1f030b8ff5a60b05becea3f474f (patch)
tree695a6abbaa45adf48766011f144050ce9b6b3780 /Lib/test/test_zlib.py
parent223221b290db00ca1042c77103efcbc072f29c90 (diff)
downloadcpython-87502ddd710eb1f030b8ff5a60b05becea3f474f.zip
cpython-87502ddd710eb1f030b8ff5a60b05becea3f474f.tar.gz
cpython-87502ddd710eb1f030b8ff5a60b05becea3f474f.tar.bz2
bpo-40286: Use random.randbytes() in tests (GH-19575)
Diffstat (limited to 'Lib/test/test_zlib.py')
-rw-r--r--Lib/test/test_zlib.py20
1 files changed, 2 insertions, 18 deletions
diff --git a/Lib/test/test_zlib.py b/Lib/test/test_zlib.py
index f828b4c..02509cd 100644
--- a/Lib/test/test_zlib.py
+++ b/Lib/test/test_zlib.py
@@ -134,8 +134,7 @@ class BaseCompressTestCase(object):
# Generate 10 MiB worth of random, and expand it by repeating it.
# The assumption is that zlib's memory is not big enough to exploit
# such spread out redundancy.
- data = b''.join([random.getrandbits(8 * _1M).to_bytes(_1M, 'little')
- for i in range(10)])
+ data = random.randbytes(_1M * 10)
data = data * (size // len(data) + 1)
try:
compress_func(data)
@@ -488,7 +487,7 @@ class CompressObjectTestCase(BaseCompressTestCase, unittest.TestCase):
# others might simply have a single RNG
gen = random
gen.seed(1)
- data = genblock(1, 17 * 1024, generator=gen)
+ data = gen.randbytes(17 * 1024)
# compress, sync-flush, and decompress
first = co.compress(data)
@@ -825,20 +824,6 @@ class CompressObjectTestCase(BaseCompressTestCase, unittest.TestCase):
self.assertEqual(dco.decompress(gzip), HAMLET_SCENE)
-def genblock(seed, length, step=1024, generator=random):
- """length-byte stream of random data from a seed (in step-byte blocks)."""
- if seed is not None:
- generator.seed(seed)
- randint = generator.randint
- if length < step or step < 2:
- step = length
- blocks = bytes()
- for i in range(0, length, step):
- blocks += bytes(randint(0, 255) for x in range(step))
- return blocks
-
-
-
def choose_lines(source, number, seed=None, generator=random):
"""Return a list of number lines randomly chosen from the source"""
if seed is not None:
@@ -847,7 +832,6 @@ def choose_lines(source, number, seed=None, generator=random):
return [generator.choice(sources) for n in range(number)]
-
HAMLET_SCENE = b"""
LAERTES