from .helper import unittest, PillowTestCase from PIL import GdImageFile TEST_GD_FILE = "Tests/images/hopper.gd" class TestFileGd(PillowTestCase): def test_sanity(self): im = GdImageFile.open(TEST_GD_FILE) self.assertEqual(im.size, (128, 128)) self.assertEqual(im.format, "GD") def test_bad_mode(self): self.assertRaises(ValueError, GdImageFile.open, TEST_GD_FILE, 'bad mode') def test_invalid_file(self): invalid_file = "Tests/images/flower.jpg" self.assertRaises(IOError, GdImageFile.open, invalid_file) if __name__ == '__main__': unittest.main()