如何在PyTest中为每个测试重新实例化StringIO变量

你的问题核心在于全局变量output在所有测试用例间共享状态，导致test_2会继承test_1写入的内容，最终断言失败。要解决这个问题，我们可以利用PyTest的fixture机制，为每个测试函数创建独立的StringIO实例，同时调整代码结构避免依赖全局变量。

推荐解决方案：使用Fixture + 依赖注入 #

这种方式既符合测试最佳实践，也让你的代码更具可维护性：

1. 定义function级别的fixture：默认情况下，fixture的scope是function，会为每个测试函数生成全新的实例。
2. 修改函数，让output作为参数传入：摆脱对全局变量的依赖，让函数更灵活。
3. 在测试中注入fixture：每个测试都会拿到专属的StringIO实例。

修改后的完整代码如下：

from io import StringIO
import pytest

# 定义fixture，每个测试都会得到新的StringIO实例
@pytest.fixture
def output():
    return StringIO()

# 修改函数，接受output作为参数
def info_to_string(output, text):
    output.write(text)

def write_to_file(output, some_text):
    for row in some_text:
        info_to_string(output, row)

def test_1(output):
    write_to_file(output, ['hello', 'there', 'what', 'is', 'up'])
    assert output.getvalue() == "hellotherewhatisup"

def test_2(output):
    write_to_file(output, ['nothing', 'much'])
    assert output.getvalue() == "nothingmuch"

备选方案：替换全局变量（不推荐，但兼容现有代码） #

如果你暂时不想修改函数的参数结构，可以用PyTest的monkeypatch fixture来替换全局的output变量，为每个测试重新赋值：

from io import StringIO
import pytest

# 保留原有的全局变量和函数结构
output = StringIO()

def info_to_string(text):
    output.write(text)

def write_to_file(some_text):
    for row in some_text:
        info_to_string(row)

def test_1(monkeypatch):
    # 为当前测试替换全局的output为新实例
    monkeypatch.setattr(__name__, 'output', StringIO())
    write_to_file(['hello', 'there', 'what', 'is', 'up'])
    assert output.getvalue() == "hellotherewhatisup"

def test_2(monkeypatch):
    monkeypatch.setattr(__name__, 'output', StringIO())
    write_to_file(['nothing', 'much'])
    assert output.getvalue() == "nothingmuch"

不过这种方式依赖全局变量，代码耦合度较高，长期来看还是推荐第一种依赖注入的方案。

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