58 lines
1.7 KiB
Python
58 lines
1.7 KiB
Python
import os
|
||
import subprocess
|
||
import sys
|
||
import time
|
||
|
||
class Color:
|
||
GREEN = '\033[92m'
|
||
RED = '\033[91m'
|
||
RESET = '\033[0m'
|
||
|
||
def test_compiler(test_dir, should_fail):
|
||
total_files = 0
|
||
passed_files = 0
|
||
has_failed_tests = False
|
||
|
||
print(f'开始测试 {test_dir}：')
|
||
for file in os.listdir(test_dir):
|
||
if file.endswith('.ac'):
|
||
total_files += 1
|
||
file_path = os.path.join(test_dir, file)
|
||
|
||
start_time = time.time()
|
||
result = subprocess.run(['./acpa', file_path], capture_output=True, text=True)
|
||
elapsed_time = time.time() - start_time
|
||
|
||
if 'runtime error' in result.stderr or result.returncode != 0:
|
||
print(f' {Color.RED}测试失败:{Color.RESET} 在编译 {file_path} 时发生运行时错误')
|
||
has_failed_tests = True
|
||
continue
|
||
|
||
if should_fail:
|
||
if 'error' not in result.stderr:
|
||
print(f' {Color.RED}测试失败:{Color.RESET} {file_path} 应该编译失败，但是没有发现错误')
|
||
has_failed_tests = True
|
||
else:
|
||
print(f' {Color.GREEN}测试通过:{Color.RESET} {file_path}')
|
||
passed_files += 1
|
||
else:
|
||
if 'error' in result.stderr:
|
||
print(f' {Color.RED}测试失败:{Color.RESET} {file_path} 应该编译通过，但是发现错误。')
|
||
has_failed_tests = True
|
||
else:
|
||
print(f' {Color.GREEN}测试通过:{Color.RESET} {file_path}')
|
||
passed_files += 1
|
||
|
||
print(f' - 编译时间: {elapsed_time:.3f} 秒')
|
||
|
||
print(f'在 {test_dir} 中共有 {total_files} 个测试文件，通过了 {passed_files} 个。\n')
|
||
return has_failed_tests
|
||
|
||
if __name__ == '__main__':
|
||
has_failed_tests = test_compiler('tests/ok/', should_fail=False)
|
||
has_failed_tests |= test_compiler('tests/fail/', should_fail=True)
|
||
|
||
if has_failed_tests:
|
||
sys.exit(1)
|
||
|