import unittest import pak class TestMyModule(unittest.TestCase): def test_hello(self): self.assertEqual(pak.main(), "Hello, World!") if __name__ == '__main__': unittest.main()