Pillow/Tests/test_file_wmf.py

58 lines
1.7 KiB
Python
Raw Normal View History

2017-03-01 12:20:18 +03:00
from helper import unittest, PillowTestCase, hopper
2017-09-01 13:36:51 +03:00
from PIL import Image
2017-09-01 13:36:51 +03:00
from PIL import WmfImagePlugin
2017-04-20 14:14:23 +03:00
class TestFileWmf(PillowTestCase):
def test_load_raw(self):
# Test basic EMF open and rendering
im = Image.open('Tests/images/drawing.emf')
if hasattr(Image.core, "drawwmf"):
# Currently, support for WMF/EMF is Windows-only
im.load()
# Compare to reference rendering
imref = Image.open('Tests/images/drawing_emf_ref.png')
imref.load()
self.assert_image_similar(im, imref, 0)
# Test basic WMF open and rendering
im = Image.open('Tests/images/drawing.wmf')
if hasattr(Image.core, "drawwmf"):
# Currently, support for WMF/EMF is Windows-only
im.load()
# Compare to reference rendering
imref = Image.open('Tests/images/drawing_wmf_ref.png')
imref.load()
self.assert_image_similar(im, imref, 2.0)
2017-09-01 13:36:51 +03:00
def test_register_handler(self):
class TestHandler:
methodCalled = False
def save(self, im, fp, filename):
self.methodCalled = True
handler = TestHandler()
WmfImagePlugin.register_handler(handler)
im = hopper()
tmpfile = self.tempfile("temp.wmf")
im.save(tmpfile)
self.assertTrue(handler.methodCalled)
# Restore the state before this test
WmfImagePlugin.register_handler(None)
2017-03-01 12:20:18 +03:00
def test_save(self):
im = hopper()
for ext in [".wmf", ".emf"]:
tmpfile = self.tempfile("temp"+ext)
2017-09-01 14:05:40 +03:00
self.assertRaises(IOError, im.save, tmpfile)
2017-03-01 12:20:18 +03:00
if __name__ == '__main__':
unittest.main()