运行文件夹中的所有std输入测试文件并验证结果？(python)

import os
import subprocess

# 获取当前文件夹路径
folder_path = os.getcwd()

# 列出文件夹中的所有文件
files = os.listdir(folder_path)

# 遍历文件列表
for file in files:
    # 判断文件是否为txt文件
    if os.path.splitext(file)[1] == '.txt':
        # 打开txt文件并读取内容
        with open(file, 'r') as f:
            input_data = f.read()

        # 运行Python脚本并获取输出结果
        output = subprocess.check_output(['python', 'your_script.py'], input=input_data, universal_newlines=True)

        # 获取预期结果
        with open(os.path.splitext(file)[0] + '.out', 'r') as f:
            expected_output = f.read()

        # 比较输出结果和预期结果
        assert output.strip() == expected_output.strip(), 'Test failed for file: {}'.format(file)

print('All tests passed!')