mirror of
https://github.com/python-pillow/Pillow.git
synced 2024-09-21 11:28:58 +03:00
50 lines
1.1 KiB
Python
50 lines
1.1 KiB
Python
from helper import unittest, PillowTestCase
|
|
|
|
import sys
|
|
|
|
try:
|
|
from PIL import ImageGrab
|
|
|
|
class TestImageGrab(PillowTestCase):
|
|
|
|
def test_grab(self):
|
|
im = ImageGrab.grab()
|
|
self.assert_image(im, im.mode, im.size)
|
|
|
|
def test_grab2(self):
|
|
im = ImageGrab.grab()
|
|
self.assert_image(im, im.mode, im.size)
|
|
|
|
except ImportError:
|
|
class TestImageGrab(PillowTestCase):
|
|
def test_skip(self):
|
|
self.skipTest("ImportError")
|
|
|
|
|
|
class TestImageGrabImport(PillowTestCase):
|
|
|
|
def test_import(self):
|
|
# Arrange
|
|
exception = None
|
|
|
|
# Act
|
|
try:
|
|
from PIL import ImageGrab
|
|
ImageGrab.__name__ # dummy to prevent Pyflakes warning
|
|
except Exception as e:
|
|
exception = e
|
|
|
|
# Assert
|
|
if sys.platform == 'win32':
|
|
self.assertIsNone(exception, None)
|
|
else:
|
|
self.assertIsInstance(exception, ImportError)
|
|
self.assertEqual(exception.message,
|
|
"ImageGrab is Windows only")
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|
|
|
|
# End of file
|