Update execute.py

execute.py

@@ -70,22 +70,24 @@ def unsafe_execute(check_program, output, result, timeout):
         chdir = os.chdir
 
         # Disable functionalities that can make destructive changes to the test.
-        # reliability_guard()
+        reliability_guard()
 
         # Run program.
-        try:
-            exec_globals = {}
-            with time_limit(timeout):
-                with stdoutIO() as s:
-                    exec(check_program, exec_globals)
-                    if s.getvalue().strip() == output[0].strip():
-                        result.append("passed")
-                    else:
-                        result.append("not passed")
-        except TimeoutException:
-            result.append("timed out")
-        except BaseException as e:
-            result.append(f"failed: {e}")
+        
+        with stdoutIO() as s:
+            try:
+                exec_globals = {}
+                with swallow_io():
+                    with time_limit(timeout):
+                            exec(check_program, exec_globals)
+            except TimeoutException:
+                result.append("timed out")
+            except BaseException as e:
+                result.append(f"failed: {e}")
+        if s.getvalue().strip() == output.strip():
+            result.append("passed")
+        else:
+            result.append("not passed")
 
         # Needed for cleaning up.
         shutil.rmtree = rmtree
@@ -100,6 +102,14 @@ def stdoutIO(stdout=None):
     sys.stdout = stdout
     yield stdout
     sys.stdout = old
+ 
+@contextlib.contextmanager
+def swallow_io():
+    stream = WriteOnlyStringIO()
+    with contextlib.redirect_stdout(stream):
+        with contextlib.redirect_stderr(stream):
+            with redirect_stdin(stream):
+                yield
 
 
 @contextlib.contextmanager
@@ -229,7 +239,7 @@ def reliability_guard(maximum_memory_bytes=None):
 
     import subprocess
 
-    #subprocess.Popen = None  # type: ignore
+    subprocess.Popen = None  # type: ignore
 
     __builtins__["help"] = None