mirror of
https://github.com/python/cpython.git
synced 2025-12-08 06:10:17 +00:00
bpo-44439: BZ2File.write() / LZMAFile.write() handle buffer protocol correctly (GH-26764) (GH-26845)
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.
(cherry picked from commit bc6c12c72a)
Co-authored-by: Ma Lin <animalize@users.noreply.github.com>
This commit is contained in:
parent
cf739332bd
commit
01858fbe31
7 changed files with 55 additions and 9 deletions
|
|
@ -592,6 +592,15 @@ def test_prepend_error(self):
|
|||
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
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue