Fix deadlock when open_input() / open_output() fail.

Related to #20
vijos · Nov 28, 2017 · 130bcf9 · 130bcf9
1 parent b10efd9
commit 130bcf9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/jd4/case.py b/jd4/case.py
@@ -93,13 +93,13 @@ def __init__(self, open_input, open_output, time_ns, memory_bytes, score):
 
     def do_stdin(self, stdin_file):
         try:
-            with self.open_input() as src, open(stdin_file, 'wb') as dst:
+            with open(stdin_file, 'wb') as dst, self.open_input() as src:
                 dos2unix(src, dst)
         except BrokenPipeError:
             pass
 
     def do_stdout(self, stdout_file):
-        with self.open_output() as ans, open(stdout_file, 'rb') as out:
+        with open(stdout_file, 'rb') as out, self.open_output() as ans:
             return compare_stream(ans, out)
 
 class APlusBCase(CaseBase):