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

flake8

Browse Source
This commit is contained in:
Hugo 2018-09-26 14:09:31 +03:00 committed by GitHub
parent 36baea18ee
commit 2389492f96
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
Tests/test_decompression_bomb.py
View File

@ -6,7 +6,6 @@ TEST_FILE = "Tests/images/hopper.ppm"
ORIGINAL_LIMIT = Image.MAX_IMAGE_PIXELS ORIGINAL_LIMIT = Image.MAX_IMAGE_PIXELS
from PIL.Image import DecompressionBombWarning, DecompressionBombError
class TestDecompressionBomb(PillowTestCase): class TestDecompressionBomb(PillowTestCase):
@ -67,7 +66,7 @@ class TestDecompressionCrop(PillowTestCase):
im = Image.new("RGB", (100, 100)) im = Image.new("RGB", (100, 100))
good_values = ((-9999, -9999, -9990, -9990), good_values = ((-9999, -9999, -9990, -9990),
(-999, -999, -990, -990)) (-999, -999, -990, -990))
warning_values = ((-160, -160, 99, 99), warning_values = ((-160, -160, 99, 99),
(160, 160, -99, -99)) (160, 160, -99, -99))
@ -76,14 +75,15 @@ class TestDecompressionCrop(PillowTestCase):
(99909, 99990, -99999, -99999)) (99909, 99990, -99999, -99999))
for value in good_values: for value in good_values:
self.assertEqual(im.crop(value).size, (9,9)) self.assertEqual(im.crop(value).size, (9, 9))
for value in warning_values: for value in warning_values:
self.assert_warning(DecompressionBombWarning, im.crop, value) self.assert_warning(Image.DecompressionBombWarning, im.crop, value)
for value in error_values: for value in error_values:
with self.assertRaises(DecompressionBombError): with self.assertRaises(Image.DecompressionBombError):
im.crop(value) im.crop(value)
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()