Added testcase for Compare with IO

Author: lin-toto

cyaron/compare.py

@@ -115,7 +115,7 @@ def program(cls, *programs, **kwargs):
         if std_program is not None:
             def get_std():
                 with open(os.dup(input.input_file.fileno()), 'r', newline='\n') as input_file:
-                    content =  make_unicode(subprocess.check_output(std_program, shell=(not list_like(std_program)), stdin=input.input_file, universal_newlines=True))
+                    content = make_unicode(subprocess.check_output(std_program, shell=(not list_like(std_program)), stdin=input.input_file, universal_newlines=True))
                     input_file.seek(0)
                 return content
             if job_pool is not None:

cyaron/tests/compare_test.py

@@ -92,6 +92,26 @@ def test_fulltext_program(self):
         correct_out = 'python correct.py: Correct \npython incorrect.py: !!!INCORRECT!!! Hash mismatch: read 53c234e5e8472b6ac51c1ae1cab3fe06fad053beb8ebfd8977b010655bfdd3c3, expected 4355a46b19d348dc2f57c046f8ef63d4538ebb936000f3c9ee954a27460dd865'
         self.assertEqual(result, correct_out)
 
+    def test_file_input(self):
+        with open("correct.py", "w") as f:
+            f.write("print(input())")
+
+        with open("std.py", "w") as f:
+            f.write("print(input())")
+
+        io = None
+        with captured_output() as (out, err):
+            io = IO()
+
+        io.input_writeln("233")
+
+        with captured_output() as (out, err):
+            Compare.program("python correct.py", std_program="python std.py", input=io, grader="NOIPStyle")
+
+        result = out.getvalue().strip()
+        correct_out = 'python correct.py: Correct'
+        self.assertEqual(result, correct_out)
+
     def test_concurrent(self):
         programs = ['test{}.py'.format(i) for i in range(16)]
         for fn in programs: