Add test

Author: neriyaco

tests/base.py

@@ -0,0 +1,11 @@
+import io
+import contextlib
+from unittest import TestCase
+
+class BaseTestCase(TestCase):
+    def assertStdout(self, expected_output, func):
+        with io.StringIO() as buf, contextlib.redirect_stdout(buf):
+            func()
+            output = buf.getvalue().strip()
+            self.assertEqual(output, expected_output)
+            

tests/simple.py tests/hello_world.pytests/simple.py renamed to tests/hello_world.py

@@ -0,0 +1,7 @@
+from .base import BaseTestCase
+
+class TestHelloWorld(BaseTestCase):
+    def test_hello_world(self):
+        def hello_world():
+            from . import hello_world
+        self.assertStdout('Hello, World!', hello_world)