Pillow/Tests/test_file_wmf.py

76 lines
2.1 KiB
Python
Raw Normal View History

import pytest
from PIL import Image, WmfImagePlugin
2017-09-01 13:36:51 +03:00
from .helper import assert_image_similar_tofile, hopper
2020-03-28 04:51:28 +03:00
def test_load_raw():
# Test basic EMF open and rendering
with Image.open("Tests/images/drawing.emf") as im:
if hasattr(Image.core, "drawwmf"):
# Currently, support for WMF/EMF is Windows-only
im.load()
# Compare to reference rendering
assert_image_similar_tofile(im, "Tests/images/drawing_emf_ref.png", 0)
2020-03-28 04:51:28 +03:00
# Test basic WMF open and rendering
with Image.open("Tests/images/drawing.wmf") as im:
if hasattr(Image.core, "drawwmf"):
# Currently, support for WMF/EMF is Windows-only
im.load()
# Compare to reference rendering
assert_image_similar_tofile(im, "Tests/images/drawing_wmf_ref.png", 2.0)
2020-03-28 04:51:28 +03:00
def test_load():
with Image.open("Tests/images/drawing.emf") as im:
if hasattr(Image.core, "drawwmf"):
assert im.load()[0, 0] == (255, 255, 255)
2020-03-28 04:51:28 +03:00
def test_register_handler(tmp_path):
class TestHandler:
methodCalled = False
def save(self, im, fp, filename):
self.methodCalled = True
handler = TestHandler()
2020-03-27 14:15:04 +03:00
original_handler = WmfImagePlugin._handler
2020-03-28 04:51:28 +03:00
WmfImagePlugin.register_handler(handler)
im = hopper()
tmpfile = str(tmp_path / "temp.wmf")
im.save(tmpfile)
assert handler.methodCalled
# Restore the state before this test
2020-03-27 14:15:04 +03:00
WmfImagePlugin.register_handler(original_handler)
2020-03-28 04:51:28 +03:00
2021-05-06 13:08:54 +03:00
def test_load_float_dpi():
2020-03-28 04:51:28 +03:00
with Image.open("Tests/images/drawing.emf") as im:
2021-05-06 13:08:54 +03:00
assert im.info["dpi"] == 1423.7668161434979
2020-03-28 04:51:28 +03:00
def test_load_set_dpi():
with Image.open("Tests/images/drawing.wmf") as im:
assert im.size == (82, 82)
if hasattr(Image.core, "drawwmf"):
im.load(144)
assert im.size == (164, 164)
assert_image_similar_tofile(im, "Tests/images/drawing_wmf_ref_144.png", 2.1)
2020-03-28 04:51:28 +03:00
def test_save(tmp_path):
im = hopper()
for ext in [".wmf", ".emf"]:
tmpfile = str(tmp_path / ("temp" + ext))
with pytest.raises(OSError):
2020-03-28 04:51:28 +03:00
im.save(tmpfile)