diff options
| author | Ma Lin <animalize@users.noreply.github.com> | 2021-06-22 07:04:23 (GMT) |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2021-06-22 07:04:23 (GMT) |
| commit | bc6c12c72a9536acc96e7b9355fd69d1083a43c1 (patch) | |
| tree | 6037ba6a1f6d777581124df4395a0b8a13472706 /Lib/test/test_gzip.py | |
| parent | 92c2e91580521ba5c85aa3205a0211df5b48689b (diff) | |
| download | cpython-bc6c12c72a9536acc96e7b9355fd69d1083a43c1.zip cpython-bc6c12c72a9536acc96e7b9355fd69d1083a43c1.tar.gz cpython-bc6c12c72a9536acc96e7b9355fd69d1083a43c1.tar.bz2 | |
bpo-44439: BZ2File.write() / LZMAFile.write() handle buffer protocol correctly (GH-26764)
No longer use len() to get the length of the input data. For some buffer protocol objects,
the length obtained by using len() is wrong.
Diffstat (limited to 'Lib/test/test_gzip.py')
| -rw-r--r-- | Lib/test/test_gzip.py | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/Lib/test/test_gzip.py b/Lib/test/test_gzip.py index 446b61a..7b51e45 100644 --- a/Lib/test/test_gzip.py +++ b/Lib/test/test_gzip.py @@ -592,6 +592,15 @@ class TestGzip(BaseTest): with gzip.open(self.filename, "rb") as f: f._buffer.raw._fp.prepend() + def test_issue44439(self): + q = array.array('Q', [1, 2, 3, 4, 5]) + LENGTH = len(q) * q.itemsize + + with gzip.GzipFile(fileobj=io.BytesIO(), mode='w') as f: + self.assertEqual(f.write(q), LENGTH) + self.assertEqual(f.tell(), LENGTH) + + class TestOpen(BaseTest): def test_binary_modes(self): uncompressed = data1 * 50 |