mirror of
https://github.com/python-pillow/Pillow.git
synced 2024-09-21 03:18:57 +03:00
48 lines
1.2 KiB
Python
48 lines
1.2 KiB
Python
from .helper import PillowTestCase, hopper
|
|
|
|
from PIL import Image
|
|
from PIL import ImageShow
|
|
|
|
|
|
class TestImageShow(PillowTestCase):
|
|
def test_sanity(self):
|
|
dir(Image)
|
|
dir(ImageShow)
|
|
|
|
def test_register(self):
|
|
# Test registering a viewer that is not a class
|
|
ImageShow.register("not a class")
|
|
|
|
# Restore original state
|
|
ImageShow._viewers.pop()
|
|
|
|
def test_show(self):
|
|
class TestViewer(ImageShow.Viewer):
|
|
methodCalled = False
|
|
|
|
def show_image(self, image, **options):
|
|
self.methodCalled = True
|
|
return True
|
|
|
|
viewer = TestViewer()
|
|
ImageShow.register(viewer, -1)
|
|
|
|
for mode in ("1", "I;16", "LA", "RGB", "RGBA"):
|
|
im = hopper(mode)
|
|
self.assertTrue(ImageShow.show(im))
|
|
self.assertTrue(viewer.methodCalled)
|
|
|
|
# Restore original state
|
|
ImageShow._viewers.pop(0)
|
|
|
|
def test_viewer(self):
|
|
viewer = ImageShow.Viewer()
|
|
|
|
self.assertIsNone(viewer.get_format(None))
|
|
|
|
self.assertRaises(NotImplementedError, viewer.get_command, None)
|
|
|
|
def test_viewers(self):
|
|
for viewer in ImageShow._viewers:
|
|
viewer.get_command("test.jpg")
|