【pytest】为什么不能使用__init__

示例代码分析

以下是一个简单的示例，演示了在 pytest 中使用 __init__ 方法导致的问题：

class TestClass:
    def __init__(self):
        self.x = 10

    def test_add(self):
        self.x += 5
        assert self.x == 15

def test_main():
    pytest.main()

在这个示例中，TestClass 类定义了一个 __init__ 方法，用于初始化类实例的 x 属性。在 test_add 测试方法中，x 属性的值被增加了 5，并断言其值等于 15。

然而，如果使用 pytest 运行该代码，则会抛出以下错误：

TypeError: cannot call __init__ on class TestClass; __init__ methods are not supported for test classes

原因分析

该错误的原因是 pytest 无法正确识别 TestClass 类。因为当 pytest 实例化 TestClass 类时，__init__ 方法会被自动调用，这会导致 x 属性的值被初始化为 10。然而，在 pytest 中，测试类的实例化是由框架自动完成的，用户通常不需要在 __init__ 方法中进行额外的初始化操作。

此外，__init__ 方法可能会掩盖测试类的实际测试逻辑。在该示例中，test_add 测试方法的目的是测试 x 属性是否被正确增加，但由于 __init__ 方法的存在，导致 x 属性的值在测试开始之前就已经被初始化为 10，这使得测试结果变得不可靠。

改进方案

为了避免上述问题，可以将 __init__ 方法中的初始化操作移到 setUp() 方法中，并使用 fixture 函数来提供测试所需的 x 属性值：

import pytest

@pytest.fixture
def x_fixture():
    return 10

class TestClass:
    def setUp(self, x):
        self.x = x

    def test_add(self, x_fixture):
        self.x += 5
        assert self.x == 15

def test_main():
    pytest.main()

在改进后的代码中，x_fixture 函数用于提供 x 属性的初始值，并将其作为参数传递给 setUp() 方法和 test_add 测试方法。这样可以确保 x 属性的值在每个测试用例之前都被正确初始化，并且不会影响测试类的实际测试逻辑。

总结

在 pytest 中不能使用 __init__ 方法的原因主要有以下几点：

1. pytest 的测试收集机制

pytest 采用的是自动发现机制来收集测试用例。它会递归遍历测试目录，并收集所有以 test 结尾的文件和目录中的测试函数和测试类。当遇到测试类时，pytest 会自动实例化该类并调用其所有以 test 结尾的方法作为测试用例。

2. init 方法的特殊性

__init__ 方法是 Python 中类的特殊方法，用于初始化类实例。当创建类实例时，__init__ 方法会被自动调用。在 pytest 中，如果测试类中定义了 __init__ 方法，那么当 pytest 实例化该类时，__init__ 方法会被调用。

3. pytest 与 init 方法的冲突

__init__ 方法通常用于初始化类实例的状态，例如分配成员变量的值等。然而，在 pytest 中，测试类的实例化是由 pytest 框架自动完成的，用户通常不需要在 __init__ 方法中进行额外的初始化操作。此外，如果测试类中定义了 __init__ 方法，那么可能会导致以下问题：

• __init__ 方法可能会掩盖测试类的实际测试逻辑。
• __init__ 方法可能会引入额外的副作用，影响测试结果的准确性。
• __init__ 方法可能会导致测试类难以被 pytest 正确识别。

4. 避免使用 init 方法的替代方案

为了避免上述问题，建议在 pytest 中不要使用 __init__ 方法。如果需要在测试类中进行初始化操作，可以使用以下替代方案：

• 使用 setUp() 和 tearDown() 方法：setUp() 方法会在每个测试用例之前运行，用于初始化测试环境；tearDown() 方法会在每个测试用例之后运行，用于清理测试环境。
• 使用类属性：可以使用类属性来存储测试类共享的状态。
• 使用 fixture 函数：可以使用 fixture 函数来提供测试所需的资源和环境。

总而言之，在 pytest 中不能使用 __init__ 方法主要是因为它与 pytest 的测试收集机制和测试类实例化过程存在冲突。为了避免出现问题，建议使用其他方法来替代 __init__ 方法进行初始化操作。

在 pytest 中，建议不要使用 __init__ 方法来初始化测试类。如果需要进行初始化操作，可以使用 setUp() 方法、类属性或 fixture 函数等替代方案。这样可以避免出现问题，并使测试代码更加清晰易懂。

如果你能看到最后，说明你认真看了，这篇文章是一道很常见的面试题，栽倒的人应该不是少数，记下来哈哈哈哈~~~