Merge pull request #5404 from radarhere/feed

Changed error type to allow for incremental WebP parsing
This commit is contained in:
Andrew Murray 2021-12-29 07:33:56 +11:00 committed by GitHub
commit d8f2fb50a4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 14 additions and 1 deletions

View File

@ -82,6 +82,19 @@ class TestImageFile:
p.feed(data) p.feed(data)
assert (48, 48) == p.image.size assert (48, 48) == p.image.size
@skip_unless_feature("webp")
@skip_unless_feature("webp_anim")
def test_incremental_webp(self):
with ImageFile.Parser() as p:
with open("Tests/images/hopper.webp", "rb") as f:
p.feed(f.read(1024))
# Check that insufficient data was given in the first feed
assert not p.image
p.feed(f.read())
assert (128, 128) == p.image.size
@skip_unless_feature("zlib") @skip_unless_feature("zlib")
def test_safeblock(self): def test_safeblock(self):
im1 = hopper() im1 = hopper()

View File

@ -396,7 +396,7 @@ _anim_decoder_new(PyObject *self, PyObject *args) {
} }
PyObject_Del(decp); PyObject_Del(decp);
} }
PyErr_SetString(PyExc_RuntimeError, "could not create decoder object"); PyErr_SetString(PyExc_OSError, "could not create decoder object");
return NULL; return NULL;
} }