add timeout for validators

bin/program.py

@@ -5,6 +5,7 @@
 import threading
 
 from util import *
+from colorama import Fore
 
 EXTRA_LANGUAGES = '''
 checktestdata:
@@ -475,7 +476,7 @@ def run(self, bar, cwd, name, args=[]):
 
         if result.ok == -9:
             # Timeout -> stop retrying and fail.
-            bar.error(f'TIMEOUT after {timeout}s')
+            bar.log(f'TIMEOUT after {timeout}s',color=Fore.RED)
             return result
 
         if result.ok is not True:
@@ -490,7 +491,7 @@ def run(self, bar, cwd, name, args=[]):
                 stdout_path.rename(in_path)
         else:
             if not in_path.is_file():
-                bar.error(f'Did not write {name}.in and stdout is empty!')
+                bar.log(f'Did not write {name}.in and stdout is empty!',color=Fore.RED)
                 result.ok = False
                 return result
 

bin/tools.py

@@ -441,6 +441,9 @@ def build_parser():
         action='store_true',
         help='Skip sanity checks on testcases.',
     )
+    validate_parser.add_argument(
+        '--timeout', '-t', type=int, help='Override the default timeout. Default: 30.'
+    )
 
     # constraints validation
     constraintsparser = subparsers.add_parser(

bin/validate.py

@@ -98,6 +98,7 @@ def run(self, testcase, constraints=None, args=None):
                 expect=config.RTV_WA if testcase.bad_input else config.RTV_AC,
                 stdin=in_file,
                 cwd=cwd,
+                timeout=self.problem.settings.timeout,
             )
 
         if constraints is not None:
@@ -150,6 +151,7 @@ def run(self, testcase, run=None, constraints=None, args=None):
                     expect=config.RTV_WA if testcase.bad_output else config.RTV_AC,
                     stdin=ans_file,
                     cwd=cwd,
+                    timeout=self.problem.settings.timeout,
                 )
 
             if constraints is not None:
@@ -171,4 +173,5 @@ def run(self, testcase, run=None, constraints=None, args=None):
                 expect=config.RTV_AC,
                 stdin=out_file,
                 cwd=run.feedbackdir,
+                timeout=self.problem.settings.timeout,
             )