bpo-40286: Use random.randbytes() in tests (GH-19575)
diff --git a/Lib/test/test_zipfile.py b/Lib/test/test_zipfile.py
index 643c5b4..29d98c8 100644
--- a/Lib/test/test_zipfile.py
+++ b/Lib/test/test_zipfile.py
@@ -16,7 +16,7 @@
from tempfile import TemporaryFile
-from random import randint, random, getrandbits
+from random import randint, random, randbytes
from test.support import script_helper
from test.support import (TESTFN, findfile, unlink, rmtree, temp_dir, temp_cwd,
@@ -33,9 +33,6 @@
('ziptest2dir/ziptest3dir/_ziptest3', 'azsxdcfvgb'),
('ziptest2dir/ziptest3dir/ziptest4dir/_ziptest3', '6y7u8i9o0p')]
-def getrandbytes(size):
- return getrandbits(8 * size).to_bytes(size, 'little')
-
def get_files(test):
yield TESTFN2
with TemporaryFile() as f:
@@ -324,7 +321,7 @@
# than requested.
for test_size in (1, 4095, 4096, 4097, 16384):
file_size = test_size + 1
- junk = getrandbytes(file_size)
+ junk = randbytes(file_size)
with zipfile.ZipFile(io.BytesIO(), "w", self.compression) as zipf:
zipf.writestr('foo', junk)
with zipf.open('foo', 'r') as fp:
@@ -2423,8 +2420,8 @@
class TestsWithMultipleOpens(unittest.TestCase):
@classmethod
def setUpClass(cls):
- cls.data1 = b'111' + getrandbytes(10000)
- cls.data2 = b'222' + getrandbytes(10000)
+ cls.data1 = b'111' + randbytes(10000)
+ cls.data2 = b'222' + randbytes(10000)
def make_test_archive(self, f):
# Create the ZIP archive