49 lines
1.4 KiB
Python
49 lines
1.4 KiB
Python
from colorama import just_fix_windows_console, Fore, Style
|
|
just_fix_windows_console()
|
|
|
|
import asyncio
|
|
from pathlib import Path
|
|
from testr.file_data import DirectorySuite, ExecutableRunner
|
|
from testr.runner import StatusCode, TestOptions
|
|
import argparse
|
|
|
|
parser = argparse.ArgumentParser()
|
|
|
|
test_runner_group = parser.add_mutually_exclusive_group(required=True)
|
|
test_data_group = parser.add_mutually_exclusive_group(required=True)
|
|
|
|
test_data_group.add_argument("--testdir", help="Directory for test cases")
|
|
test_runner_group.add_argument("--exec", help="Executable to run test cases against")
|
|
|
|
parser.add_argument("--timelim", help="Time limit in seconds", type=float, default=5.0)
|
|
|
|
args = parser.parse_args()
|
|
|
|
|
|
code_styles = {
|
|
StatusCode.AC: Fore.GREEN,
|
|
StatusCode.WA: Fore.RED,
|
|
StatusCode.IR: Fore.YELLOW,
|
|
StatusCode.TLE: Fore.YELLOW,
|
|
}
|
|
|
|
|
|
async def main():
|
|
test_suite = DirectorySuite(Path(args.testdir))
|
|
test_runner = ExecutableRunner(Path(args.exec))
|
|
|
|
async for test_case in test_runner.run_test_suite(
|
|
test_suite,
|
|
TestOptions(
|
|
time_limit=args.timelim
|
|
)
|
|
):
|
|
print(
|
|
f"{test_case.test_data.name : <20} "
|
|
f"{code_styles.get(test_case.code, '')}{test_case.code.name : >4}{Style.RESET_ALL}"
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(main())
|