import unittest import mypackage class TestMain(unittest.TestCase): def test_main(self): mypackage.main() if __name__ == '__main__': unittest.main()