Pillow/Tests/test_image_load.py

from helper import unittest, PillowTestCase, tearDownModule, lena

from PIL import Image

import os


class TestImageLoad(PillowTestCase):

    def test_sanity(self):

        im = lena()

        pix = im.load()

        self.assertEqual(pix[0, 0], (223, 162, 133))

    def test_close(self):
        im = Image.open("Images/lena.gif")
        im.close()
        self.assertRaises(ValueError, lambda: im.load())
        self.assertRaises(ValueError, lambda: im.getpixel((0, 0)))

    def test_contextmanager(self):
        fn = None
        with Image.open("Images/lena.gif") as im:
            fn = im.fp.fileno()
            os.fstat(fn)

        self.assertRaises(OSError, lambda: os.fstat(fn))

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

# End of file