Pillow/Tests/check_j2k_overflow.py

from PIL import Image
from helper import unittest, PillowTestCase


class TestJ2kEncodeOverflow(PillowTestCase):
    def test_j2k_overflow(self):

        im = Image.new('RGBA', (1024, 131584))
        target = self.tempfile('temp.jpc')
        try:
            im.save(target)
            self.assertTrue(False, "Expected IOError, save succeeded?")
        except IOError as err:
            self.assertTrue(True, "IOError is expected")
        except Exception as err:
            self.assertTrue(False, "Expected IOError, got %s" % type(err))

if __name__ == '__main__':
    unittest.main()