python写 main 函数调用测试模板。测试用例放在 csv 文件中,从csv文件调用测试用例执行,结果导出到 csv 文件
时间: 2024-01-31 12:45:22 浏览: 104
以下是一个示例代码,其中假设测试用例文件名为 "test_cases.csv",结果文件名为 "test_results.csv":
```
import csv
def run_test_case(input1, input2):
# 这里放测试用例的代码
pass
def main():
# 打开测试用例文件
with open('test_cases.csv', 'r') as f:
# 读取测试用例数据
reader = csv.reader(f)
next(reader) # 跳过表头
# 遍历测试用例并执行
results = []
for row in reader:
input1 = row[0]
input2 = row[1]
expected_output = row[2]
# 执行测试用例
actual_output = run_test_case(input1, input2)
# 比较实际输出和期望输出
if actual_output == expected_output:
result = 'pass'
else:
result = 'fail'
# 记录结果
results.append([input1, input2, expected_output, actual_output, result])
# 将结果写入结果文件
with open('test_results.csv', 'w', newline='') as f:
writer = csv.writer(f)
writer.writerow(['Input 1', 'Input 2', 'Expected Output', 'Actual Output', 'Result'])
writer.writerows(results)
if __name__ == '__main__':
main()
```
在上面的代码中,我们定义了一个 `run_test_case` 函数用来执行单个测试用例,然后在 `main` 函数中读取测试用例文件,遍历测试用例并执行它们,最后将测试结果写入结果文件。注意在写结果文件时需要首先写表头。在写文件时使用了 `newline=''` 参数来避免写入空行。
阅读全文