Pytest 使用 tempfile

在编写测试代码时，我们通常需要创建一些临时文件或目录来模拟测试数据。Python 提供了 tempfile 模块来管理临时文件和目录的创建和删除过程。Pytest 作为一个强大的测试框架，其也提供了与 tempfile 模块结合使用来管理临时文件和目录的方法。

创建临时文件和目录

使用 tempfile 模块，我们可以轻松地创建临时文件和目录，并在测试结束后自动删除它们。在 Pytest 中，我们可以使用 tmpdir fixture 来创建和管理临时目录。

import tempfile

def test_create_temp_file():
    with tempfile.TemporaryFile() as temp_file:
        # 进行测试操作
        pass

def test_create_temp_dir(tmpdir):
    # tmpdir 为 Pytest 自动生成的临时目录
    temp_file = tmpdir.join('temp_file')
    temp_file.write('test data')
    assert temp_file.read() == 'test data'

上述代码使用 tempfile.TemporaryFile 创建了一个临时文件，并使用 with 语句进行测试操作。测试结束后，使用 with 语句自动关闭文件并删除文件。使用 tmpdir fixture 创建了一个临时目录，并在临时目录中创建了一个临时文件，写入了测试数据，并验证了写入的数据是否成功。

使用 fixures 管理临时文件

当我们需要在多个测试用例中使用相同的临时文件时，Pytest 的 fixture 可以帮助我们共享临时文件。我们可以在 conftest.py 文件中定义一个 fixture 来自动管理临时文件的创建和删除过程，然后在测试用例中引用这个 fixture。

首先，在 conftest.py 文件中定义一个 fixture：

import pytest
import tempfile

@pytest.fixture(scope='session')
def temp_file():
    with tempfile.NamedTemporaryFile(delete=False) as f:
        # 可以进行一些初始化操作
        yield f.name  # 返回文件名
        os.remove(f.name)  # 删除临时文件

然后，在测试用例中引用该 fixture：

import os

def test_write_temp_file(temp_file):
    with open(temp_file, 'w') as f:
        f.write('test data')
    assert os.path.exists(temp_file)

def test_read_temp_file(temp_file):
    with open(temp_file, 'r') as f:
        data = f.read()
    assert data == 'test data'

上述代码使用 tempfile.NamedTemporaryFile 创建了一个具有唯一文件名的临时文件，并使用 yield 语句返回临时文件名，在测试结束后自动删除文件。在测试用例中引用这个 fixture，就可以共享同一个临时文件，进行读写操作，并进行断言验证。

总结

Pytest 的 tmpdir fixture 和 fixture 可以方便地管理临时文件和目录的创建和删除过程，可以帮助我们更高效地编写测试代码。同时，Python 的 tempfile 模块提供了丰富的接口供我们创建和管理临时文件和目录。