基于unittest的基类/派生类测试转pytest的等价实现问询

1. 在pytest中是否仍需使用类？ #

pytest不强制要求使用类，它支持更简洁的函数式测试写法，但完全兼容unittest的类式测试结构：

• 如果你的旧代码依赖类的继承、类属性来复用测试逻辑，可以直接保留类结构，pytest能正常识别并执行Test开头的类中的test_前缀方法。
• 若想简化代码，也可以将类中的测试方法拆成独立函数，配合pytest的fixture、参数化等特性实现复用。

2. 针对基类/派生类测试场景的pytest实现模式 #

针对你示例中通过基类复用通用测试、派生类通过类属性控制测试逻辑的场景，pytest有几种适配方式：

方式一：直接兼容原有继承结构 #

这是最低成本的迁移方式，无需大幅修改代码：

• 确保基类（如示例中的Base、Base2）不以Test开头，pytest默认不会将其识别为测试类，只会执行Test开头的派生类（Test1、Test2、Test3）。
• 派生类继承基类后，会自动继承基类的测试方法和类属性，派生类的类属性会覆盖基类的对应属性，逻辑和原unittest完全一致。

你的示例代码可以直接在pytest中运行，Test1会执行test_1和test_2，并使用自身的val_1/val_2/val_3覆盖Base的默认值。

方式二：用fixture替代类属性（更贴合pytest风格） #

如果想摆脱类的束缚，改用pytest的fixture来管理测试参数：

import pytest

# 定义参数化fixture，提供不同的测试参数组合
@pytest.fixture(params=[
    {"val_1":1, "val_2":2, "val_3":3, "val_4":5, "val_5":0},
    {"val_1":2, "val_2":3, "val_3":4, "val_4":5, "val_5":0},
    {"val_1":1, "val_2":2, "val_3":3, "val_4":0, "val_5":0},
])
def test_params(request):
    return request.param

# 通用测试函数
def test_1(test_params):
    val_1 = test_params["val_1"]
    # 测试逻辑
    pass

def test_2(test_params):
    pass

def test_3(test_params):
    if test_params["val_4"] != 0:
        pytest.skip("仅在val_4=0的场景执行")
    # 测试逻辑
    pass

方式三：结合参数化与类（兼顾原有结构与pytest特性） #

如果想保留类的结构，同时利用pytest的参数化简化派生类：

import pytest

class BaseTests:
    val_1 = None
    val_2 = None
    val_3 = None
    val_4 = 5
    val_5 = 0

    def test_1(self):
        assert self.val_1 is not None

    def test_2(self):
        pass

# 用参数化装饰器生成多个测试类
@pytest.mark.parametrize("val_1, val_2, val_3, val_4, val_5", [
    (1,2,3,5,0),
    (2,3,4,5,0),
    (1,2,3,0,0),
])
class TestDerived(BaseTests):
    def __init__(self, val_1, val_2, val_3, val_4, val_5):
        self.val_1 = val_1
        self.val_2 = val_2
        self.val_3 = val_3
        self.val_4 = val_4
        self.val_5 = val_5

    def test_3(self):
        if self.val_4 == 0:
            # 测试逻辑
            pass

这种方式可以避免定义多个派生类，用参数化一次性生成所有测试场景。

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