Merge pull request #1678 from radarhere/pillow_tests

Add tests for OLE file based formats
This commit is contained in:
Hugo 2016-01-23 11:10:20 +02:00
commit 29d23ee102
2 changed files with 12 additions and 2 deletions

View File

@ -6,11 +6,16 @@ from PIL import FpxImagePlugin
class TestFileFpx(PillowTestCase): class TestFileFpx(PillowTestCase):
def test_invalid_file(self): def test_invalid_file(self):
# Test an invalid OLE file
invalid_file = "Tests/images/flower.jpg" invalid_file = "Tests/images/flower.jpg"
self.assertRaises(SyntaxError, self.assertRaises(SyntaxError,
lambda: FpxImagePlugin.FpxImageFile(invalid_file)) lambda: FpxImagePlugin.FpxImageFile(invalid_file))
# Test a valid OLE file, but not an FPX file
ole_file = "Tests/images/test-ole-file.doc"
self.assertRaises(SyntaxError,
lambda: FpxImagePlugin.FpxImageFile(ole_file))
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()

View File

@ -6,11 +6,16 @@ from PIL import MicImagePlugin
class TestFileMic(PillowTestCase): class TestFileMic(PillowTestCase):
def test_invalid_file(self): def test_invalid_file(self):
# Test an invalid OLE file
invalid_file = "Tests/images/flower.jpg" invalid_file = "Tests/images/flower.jpg"
self.assertRaises(SyntaxError, self.assertRaises(SyntaxError,
lambda: MicImagePlugin.MicImageFile(invalid_file)) lambda: MicImagePlugin.MicImageFile(invalid_file))
# Test a valid OLE file, but not a MIC file
ole_file = "Tests/images/test-ole-file.doc"
self.assertRaises(SyntaxError,
lambda: MicImagePlugin.MicImageFile(ole_file))
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()