diff options
author | Victor Stinner <vstinner@python.org> | 2020-04-17 20:54:38 (GMT) |
---|---|---|
committer | GitHub <noreply@github.com> | 2020-04-17 20:54:38 (GMT) |
commit | 87502ddd710eb1f030b8ff5a60b05becea3f474f (patch) | |
tree | 695a6abbaa45adf48766011f144050ce9b6b3780 /Lib/test/test_zlib.py | |
parent | 223221b290db00ca1042c77103efcbc072f29c90 (diff) | |
download | cpython-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.py | 20 |
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 |