python-pillow/Pillow
1
1
Fork 0
mirror of https://github.com/python-pillow/Pillow.git synced 2025-07-04 11:53:32 +03:00
Code Issues Packages Projects Releases Wiki Activity

imagefile testing improvement final

Browse Source
This commit is contained in:
Deekshu Kare 2024-06-22 01:33:15 +02:00
parent 8e3cf21954
commit f206a1c617
1 changed files with 1 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
Tests/test_imagefile.py
View File

@ -235,25 +235,21 @@ class MockPyEncoder(ImageFile.PyEncoder):
def test_encode_to_file() -> None: def test_encode_to_file() -> None:
encoder = MockPyEncoder("RGBA") encoder = MockPyEncoder("RGBA")
# Case: _pushes_fd is False
with pytest.raises(NotImplementedError): with pytest.raises(NotImplementedError):
encoder.encode_to_file(None, None) encoder.encode_to_file(None, None)
# Case: _pushes_fd is True
encoder._pushes_fd = True encoder._pushes_fd = True
with pytest.raises(NotImplementedError): with pytest.raises(NotImplementedError):
encoder.encode_to_file(None, None) encoder.encode_to_file(None, None)
# Case: encode method called with buffer (no exception)
buffer = BytesIO(b"\x00" * 10) buffer = BytesIO(b"\x00" * 10)
encoder._pushes_fd = False encoder._pushes_fd = False
encoder.encode = lambda buffer: (1, 1, b"") encoder.encode = lambda buffer: (1, 1, b"")
try: try:
encoder.encode_to_file(buffer, None) encoder.encode_to_file(buffer, None)
except NotImplementedError: except NotImplementedError:
pass # NotImplementedError is expected pass
# Case: encode method raises NotImplementedError
encoder.encode = lambda buffer: (_ for _ in ()).throw(NotImplementedError) encoder.encode = lambda buffer: (_ for _ in ()).throw(NotImplementedError)
with pytest.raises(NotImplementedError): with pytest.raises(NotImplementedError):
encoder.encode_to_file(buffer, None) encoder.encode_to_file(buffer, None)