mirror of
https://github.com/python-pillow/Pillow.git
synced 2025-07-10 16:22:22 +03:00
Improve the error message when importing ImageGrab on non-Windows (#901)
This commit is contained in:
parent
1ecc3195d9
commit
99887a5467
|
@ -17,6 +17,9 @@
|
||||||
|
|
||||||
from PIL import Image
|
from PIL import Image
|
||||||
|
|
||||||
|
import sys
|
||||||
|
if sys.platform != "win32":
|
||||||
|
raise ImportError("ImageGrab is Windows only")
|
||||||
|
|
||||||
try:
|
try:
|
||||||
# built-in driver (1.1.3 and later)
|
# built-in driver (1.1.3 and later)
|
||||||
|
|
|
@ -1,5 +1,8 @@
|
||||||
from helper import unittest, PillowTestCase
|
from helper import unittest, PillowTestCase
|
||||||
|
|
||||||
|
import exceptions
|
||||||
|
import sys
|
||||||
|
|
||||||
try:
|
try:
|
||||||
from PIL import ImageGrab
|
from PIL import ImageGrab
|
||||||
|
|
||||||
|
@ -19,6 +22,28 @@ except ImportError:
|
||||||
self.skipTest("ImportError")
|
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 exception:
|
||||||
|
pass
|
||||||
|
|
||||||
|
# Assert
|
||||||
|
if sys.platform == 'win32':
|
||||||
|
self.assertIsNone(exception, None)
|
||||||
|
else:
|
||||||
|
self.assertIsInstance(exception, exceptions.ImportError)
|
||||||
|
self.assertEqual(exception.message,
|
||||||
|
"ImageGrab is Windows only")
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
unittest.main()
|
unittest.main()
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue
Block a user