import os import subprocess import sys from unittest import TestCase class TestMain(TestCase): def test_main(self): out = subprocess.check_output([sys.executable, "-m", "PIL"]).decode("utf-8") lines = out.splitlines() self.assertEqual(lines[0], "-" * 68) self.assertTrue(lines[1].startswith("Pillow ")) self.assertEqual(lines[2], "-" * 68) self.assertTrue(lines[3].startswith("Python modules loaded from ")) self.assertTrue(lines[4].startswith("Binary modules loaded from ")) self.assertEqual(lines[5], "-" * 68) self.assertTrue(lines[6].startswith("Python ")) jpeg = ( os.linesep + "-" * 68 + os.linesep + "JPEG image/jpeg" + os.linesep + "Extensions: .jfif, .jpe, .jpeg, .jpg" + os.linesep + "Features: open, save" + os.linesep + "-" * 68 + os.linesep ) self.assertIn(jpeg, out)