diff --git a/Tests/test_file_gribstub.py b/Tests/test_file_gribstub.py index 79025d3b4..ea43d13c0 100644 --- a/Tests/test_file_gribstub.py +++ b/Tests/test_file_gribstub.py @@ -1,23 +1,47 @@ from helper import unittest, PillowTestCase, hopper -from PIL import GribStubImagePlugin +from PIL import GribStubImagePlugin, Image + +TEST_FILE = "Tests/images/WAlaska.wind.7days.grb" class TestFileGribStub(PillowTestCase): + def test_open(self): + # Act + im = Image.open(TEST_FILE) + + # Assert + self.assertEqual(im.format, "GRIB") + + # Dummy data from the stub + self.assertEqual(im.mode, "F") + self.assertEqual(im.size, (1, 1)) + def test_invalid_file(self): + # Arrange invalid_file = "Tests/images/flower.jpg" + # Act / Assert self.assertRaises(SyntaxError, lambda: GribStubImagePlugin.GribStubImageFile(invalid_file)) def test_save(self): + # Arrange im = hopper() - tmpfile = self.tempfile("temp.grib") + + # Act / Assert: stub cannot save without an implemented handler self.assertRaises(IOError, lambda: im.save(tmpfile)) + def test_load(self): + # Arrange + im = Image.open(TEST_FILE) + + # Act / Assert: stub cannot load without an implemented handler + self.assertRaises(IOError, lambda: im.load()) + if __name__ == '__main__': unittest.main()