如何将Python测试中的多个assert语句合并为一个?
解决方案
你可以通过手动索引遍历两个列表的方式,实现和原多个assert语句完全一致的效果,而且不需要使用任何内置函数:
def test_function(): l1 = [A.a, B.b, C.c, D.d, E.e, F.f, G.g] l2 = [1, 2, 3, 4, 5, 6, 7] index = 0 # 已知两个列表长度都是7,直接循环对应次数 while index < 7: assert l1[index] == l2[index], f"位置{index}不匹配:{l1[index]} != {l2[index]}" index += 1
补充说明:
- 这个写法和你原来的7条
assert语句逻辑完全等价,每个索引位置的元素对应原代码中的一组断言 - 相比你之前尝试的字典嵌套循环写法,避免了不必要的双重遍历和条件判断,逻辑更清晰,执行效率也更高
- 如果不想硬编码长度
7,也可以用try-except处理索引越界(同样不依赖内置函数):
def test_function(): l1 = [A.a, B.b, C.c, D.d, E.e, F.f, G.g] l2 = [1, 2, 3, 4, 5, 6, 7] index = 0 while True: try: val1 = l1[index] val2 = l2[index] except IndexError: break assert val1 == val2, f"位置{index}不匹配:{val1} != {val2}" index += 1
内容的提问来源于stack exchange,提问作者jhonD
相关产品推荐
相关产品推荐

