diff --git a/Lib/bz2.py b/Lib/bz2.py index cc71ae025ae..cbf5233b065 100644 --- a/Lib/bz2.py +++ b/Lib/bz2.py @@ -75,11 +75,11 @@ def __init__(self, filename=None, mode="r", buffering=None, elif mode in ("w", "wb"): mode = "wb" mode_code = _MODE_WRITE - self._compressor = BZ2Compressor() + self._compressor = BZ2Compressor(compresslevel) elif mode in ("a", "ab"): mode = "ab" mode_code = _MODE_WRITE - self._compressor = BZ2Compressor() + self._compressor = BZ2Compressor(compresslevel) else: raise ValueError("Invalid mode: {!r}".format(mode)) diff --git a/Lib/test/test_bz2.py b/Lib/test/test_bz2.py index 5a7d342afde..d0032387e97 100644 --- a/Lib/test/test_bz2.py +++ b/Lib/test/test_bz2.py @@ -224,6 +224,13 @@ def testWriteChunks10(self): with open(self.filename, 'rb') as f: self.assertEqual(self.decompress(f.read()), self.TEXT) + def testWriteNonDefaultCompressLevel(self): + expected = bz2.compress(self.TEXT, compresslevel=5) + with BZ2File(self.filename, "w", compresslevel=5) as bz2f: + bz2f.write(self.TEXT) + with open(self.filename, "rb") as f: + self.assertEqual(f.read(), expected) + def testWriteLines(self): with BZ2File(self.filename, "w") as bz2f: self.assertRaises(TypeError, bz2f.writelines)