python-pillow/Pillow
1
1
Fork 0
mirror of https://github.com/python-pillow/Pillow.git synced 2025-02-03 21:24:31 +03:00
Code Issues Packages Projects Releases Wiki Activity

Updated error catching

Browse Source
This commit is contained in:
Andrew Murray 2019-12-28 23:39:50 +11:00
parent 5253b0cd57
commit 2543719e12
1 changed files with 6 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
Tests/test_imagegrab.py
View File

@ -36,15 +36,9 @@ class TestImageGrab:
@pytest.mark.skipif(not Image.core.HAVE_XCB, reason="requires XCB")
def test_grab_invalid_xdisplay(self):
exception = None
try:
with pytest.raises(IOError) as e:
ImageGrab.grab(xdisplay="error.test:0.0")
except Exception as e:
exception = e
assert isinstance(exception, IOError)
assert str(exception).startswith("X connection failed")
assert str(e.value).startswith("X connection failed")
def test_grabclipboard(self):
if sys.platform == "darwin":
@ -59,15 +53,11 @@ $bmp = New-Object Drawing.Bitmap 200, 200
)
p.communicate()
else:
exception = None
try:
with pytest.raises(NotImplementedError) as e:
ImageGrab.grabclipboard()
except IOError as e:
exception = e
assert isinstance(exception, IOError)
assert str(exception) == "ImageGrab.grabclipboard() is macOS and Windows only"
self.assertEqual(
str(e.value), "ImageGrab.grabclipboard() is macOS and Windows only"
)
return
im = ImageGrab.grabclipboard()