Change the output checking method in test_func.py. (#18)

* Change the output checking method in test_func.py.

Use redirect_stdout method in test_func.py.
Change the `venv/` into `venv*/` in .gitignore.

* Delete a package that is not used in test_func.py

.gitignore

@@ -125,7 +125,7 @@ celerybeat.pid
 .env
 .venv
 env/
-venv/
+venv*/
 ENV/
 env.bak/
 venv.bak/

tests/test_func.py

@@ -1,6 +1,8 @@
 # -*- coding: utf-8 -*-
+from contextlib import redirect_stdout
+import io
+import unittest
 import os
-import sys
 import shutil
 import unittest
 from OiRunner import BetterRunner
@@ -20,15 +22,12 @@ def tearDown(self):
 
     def test_modify_file(self):
         for i in range(1, 3):
-            out = sys.stdout
             with self.assertRaises(SystemExit):
-                with open("~temp", "w") as f:
-                    sys.stdout = f
+                with io.StringIO() as buf, redirect_stdout(buf):
                     self.func._modify_file(f"empty{i}.in", "in")
-            self.assertFalse(os.path.exists("~tmp"))
-            with open("~temp", "r") as f:
-                self.assertEqual(f.read(), f"error:empty{i}.in is empty.\n")
-            sys.stdout = out
+                    self.assertFalse(os.path.exists("~tmp"))
+                    output = buf.getvalue()
+                    self.assertEqual(output, f"error:empty{i}.in is empty.\n")
 
         ret_code = self.func._modify_file("a.in", "in")
         self.assertEqual(ret_code, 3)