catfood/CompilerSysY
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add timer to test script

Browse Source
This commit is contained in:
ridethepig 2023-06-18 19:02:23 +08:00
parent c97f3dfdac
commit 4470c97411
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
scripts/mytester.py
View File

@ -2,6 +2,8 @@ import argparse
import re import re
import os import os
import subprocess import subprocess
import time
from functools import wraps
from pretty_print import Print_C from pretty_print import Print_C
# "([1-4]\d|51).*\.sy" # "([1-4]\d|51).*\.sy"
@ -19,7 +21,15 @@ def case_collector(re_selector, dir):
testcases.append(os.path.splitext(filename)[0]) testcases.append(os.path.splitext(filename)[0])
return sorted(testcases) return sorted(testcases)
def func_timer(function):
@wraps(function)
def function_timer(*args, **kwargs):
t0 = time.time()
result = function(*args, **kwargs)
t1 = time.time()
print('[Time elapsed {time:.3f}s]'.format(time=t1 - t0))
return result
return function_timer
class Compiler: class Compiler:
def __init__(self, scheme, target_dir, testcases): def __init__(self, scheme, target_dir, testcases):
@ -145,7 +155,7 @@ class Compiler:
if self.count_error >= error_tolerance: if self.count_error >= error_tolerance:
Print_C.print_error(f"Test script stopped due to {self.count_error} errors") Print_C.print_error(f"Test script stopped due to {self.count_error} errors")
return return
@func_timer
def run_case(self, testcase, kase=0): def run_case(self, testcase, kase=0):
bin = self.bin_template.format(testcase=testcase) bin = self.bin_template.format(testcase=testcase)
if not os.path.exists(bin): if not os.path.exists(bin):