pytest使用parametrize读取Excel执行用例后如何更新同文件对应行数据

实现方案 #

你只需要调整读取Excel的逻辑，给每条用例绑定对应的Excel行号即可，具体操作如下：

第一步：调整Excel读取逻辑，带出对应行号

修改你的Utility.readExcel方法，让它返回的每个元素是(行号, 行数据)的元组，注意Excel的行计数从1开始，如果你的Sheet第1行是表头，那第一条测试数据对应的行号就是2。

如果不想改原有读取方法，也可以直接用enumerate给读取到的用例列表加索引，再换算成实际Excel行号：

# 先读取所有测试数据
test_data = list(Utility.readExcel("Input.xlsx", "Sheet1"))
# 索引+偏移量得到实际Excel行号，这里偏移量2是假设第1行是表头，数据从第2行开始，可根据实际情况调整
@pytest.mark.parametrize('idx, input', enumerate(test_data))

第二步：测试用例执行完成后写入对应行

把行号作为参数传入测试用例，执行完成后调用写Excel的方法写入结果即可，完整示例代码如下：

import pytest
import openpyxl # 推荐用openpyxl处理xlsx格式读写，不会破坏原有文件格式

test_data = list(Utility.readExcel("Input.xlsx", "Sheet1"))

# 封装写Excel的公共方法，也可以放到你的Utility类里
def write_excel_result(file_path, sheet_name, row_num, col_name, value):
    wb = openpyxl.load_workbook(file_path)
    sheet = wb[sheet_name]
    # 先找到对应列的索引，假设表头在第1行
    col_idx = None
    for cell in sheet[1]:
        if cell.value == col_name:
            col_idx = cell.column_letter
            break
    if col_idx:
        sheet[f"{col_idx}{row_num}"] = value
        wb.save(file_path)

@pytest.mark.parametrize('idx, input', enumerate(test_data))
def test_demo(idx, input):
    # 你的原有测试逻辑，得到测试结果
    test_result = "通过" if 1+1==2 else "失败"
    # 换算实际Excel行号，按需调整偏移量
    excel_row = idx + 2
    # 写入结果，这里"测试结果"是你Excel里要写入的列名，按需修改
    write_excel_result("Input.xlsx", "Sheet1", excel_row, "测试结果", test_result)

注意事项

• 不要用多进程/多线程跑这批用例，同时写入同一个Excel会出现文件冲突、内容覆盖的问题，如果必须并行，建议先把所有用例的行号、结果存在临时列表里，全部用例执行完成后统一写一次Excel
• 写文件时要确保Excel没有被其他程序打开，否则会抛出权限错误
• 可以把写结果的逻辑放到测试用例的teardown环节，保证不管用例执行成功、失败还是跳过，都能正常写入结果

内容的提问来源于stack exchange，提问作者Yerigeri