Pytest：如何在脚本中使用--basetemp及优雅管理测试文件

1. 获取--basetemp指定的临时根目录 #

要在测试脚本中获取通过--basetemp参数指定的临时根目录，可以借助pytest的request fixture访问配置项：

class TestGraphProcessing:
    def test_get_basetemp(self, request):
        basetemp_dir = request.config.getoption("--basetemp")
        # basetemp_dir为Path对象或字符串，可直接用于文件路径操作
        print(f"临时根目录路径: {basetemp_dir}")

即使未手动指定--basetemp，pytest会自动生成默认临时目录，通过上述方式同样能获取其路径。

2. 更优雅的临时文件创建与清理方案 #

推荐直接使用pytest内置的tmp_path或tmp_path_factory fixture，它们会自动完成临时目录的创建、测试后的文件清理，完全无需手动调用unlink()。

方式一：tmp_path（单测试用例专属临时目录） #

tmp_path是每个测试用例独立的临时目录（类型为pathlib.Path），可直接在测试中生成文件路径：

class TestGraphProcessing:
    # 调整参数化逻辑，仅传递graph对象，文件路径由tmp_path生成
    @pytest.mark.parametrize("graph", YOUR_GRAPHS_LIST)
    def test_save_graph(self, graph, tmp_path):
        # 在临时目录下生成目标文件路径
        path_to_save = tmp_path / "test_graph_output.json"
        save_graph(graph, path_to_save)
        assert path_to_save.is_file()
        # 测试结束后，tmp_path下的所有文件会被自动清理，无需手动处理

如果需要保留原参数化中的文件名，可将参数调整为(graph, filename)，再通过tmp_path / filename生成完整路径。

方式二：tmp_path_factory（跨测试用例共享临时目录） #

若多个测试用例需要共享同一临时目录，可通过tmp_path_factory创建：

class TestGraphProcessing:
    @pytest.fixture(scope="class")
    def shared_temp_dir(self, tmp_path_factory):
        # 创建一个类级别的共享临时目录
        return tmp_path_factory.mktemp("shared_graph_files")

    @pytest.mark.parametrize("graph, filename", GRAPH_TO_FILENAME)
    def test_save_shared_graph(self, graph, filename, shared_temp_dir):
        path_to_save = shared_temp_dir / filename
        save_graph(graph, path_to_save)
        assert path_to_save.is_file()

优势说明 #

• 自动清理：测试结束后pytest会自动删除临时目录下所有内容，无需手动维护删除逻辑
• 隔离性：tmp_path为每个测试提供独立目录，避免测试间的文件干扰
• 简洁性：无需编写路径创建、清理的冗余代码，让测试逻辑更聚焦业务验证

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