diff --git a/Tests/test_file_fpx.py b/Tests/test_file_fpx.py index 35119a612..01b285efe 100644 --- a/Tests/test_file_fpx.py +++ b/Tests/test_file_fpx.py @@ -6,11 +6,16 @@ from PIL import FpxImagePlugin class TestFileFpx(PillowTestCase): def test_invalid_file(self): + # Test an invalid OLE file invalid_file = "Tests/images/flower.jpg" - self.assertRaises(SyntaxError, 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__': unittest.main() diff --git a/Tests/test_file_mic.py b/Tests/test_file_mic.py index 0a574422a..f0a2752a9 100644 --- a/Tests/test_file_mic.py +++ b/Tests/test_file_mic.py @@ -6,11 +6,16 @@ from PIL import MicImagePlugin class TestFileMic(PillowTestCase): def test_invalid_file(self): + # Test an invalid OLE file invalid_file = "Tests/images/flower.jpg" - self.assertRaises(SyntaxError, 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__': unittest.main()