mirrors/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-07-24 11:44:31 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 8

bpo-44439: BZ2File.write() / LZMAFile.write() handle buffer protocol correctly (GH-26764)

Browse source 
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.
This commit is contained in:
Ma Lin 2021-06-22 15:04:23 +08:00 • committed by GitHub
parent 92c2e91580
commit bc6c12c72a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
7 changed files with 55 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

9
Lib/test/test_lzma.py
View file

@ -1,4 +1,5 @@
import _compression
import array
from io import BytesIO, UnsupportedOperation, DEFAULT_BUFFER_SIZE
import os
import pathlib
@ -1231,6 +1232,14 @@ class FileTestCase(unittest.TestCase):
self.assertTrue(d2.eof)
self.assertEqual(out1 + out2, entire)
def test_issue44439(self):
q = array.array('Q', [1, 2, 3, 4, 5])
LENGTH = len(q) * q.itemsize
with LZMAFile(BytesIO(), 'w') as f:
self.assertEqual(f.write(q), LENGTH)
self.assertEqual(f.tell(), LENGTH)
class OpenTestCase(unittest.TestCase):