Python单元测试：如何在setUp中保留临时目录上下文管理器？

核心问题是：在setUp()里用with tempfile.TemporaryDirectory() as dir_name时，上下文会在setUp()执行完毕后立即退出，触发临时目录的销毁，导致后续测试方法访问时目录已不存在。下面是两种针对unittest的可靠解决方案：

方案1：手动管理临时目录实例（单测试方法级别） #

创建TemporaryDirectory实例并保存为测试类的属性，在tearDown()里显式调用cleanup()方法，这样目录会在每个测试方法执行期间保持存在，测试结束后再清理：

import tempfile
import unittest
import os

class YourTestClass(unittest.TestCase):
    def setUp(self):
        # 创建临时目录实例，绑定到类属性
        self.temp_dir = tempfile.TemporaryDirectory()
        self.temp_dir_path = self.temp_dir.name

    def tearDown(self):
        # 显式清理临时目录，避免代码检查器警告
        self.temp_dir.cleanup()

    def test_file_write_and_read(self):
        # 使用临时目录路径操作文件
        test_file = os.path.join(self.temp_dir_path, "test_data.txt")
        # 写入测试内容
        with open(test_file, "w") as f:
            f.write("sample content")
        # 回读验证
        with open(test_file, "r") as f:
            content = f.read()
        self.assertEqual(content, "sample content")

这种方式和你原来的setUp()/tearDown()逻辑对齐，既满足代码检查器对显式资源清理的要求，又能保证临时目录在测试方法执行期间可用。

方案2：类级别共享临时目录（多测试方法复用） #

如果你的测试方法之间可以共享同一个临时目录（不需要每次测试都重置目录状态），可以用类级别的setUpClass()和tearDownClass()来创建/清理，减少重复开销：

import tempfile
import unittest
import os

class YourTestClass(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        # 整个测试类只创建一次临时目录
        cls.temp_dir = tempfile.TemporaryDirectory()
        cls.temp_dir_path = cls.temp_dir.name

    @classmethod
    def tearDownClass(cls):
        # 测试类所有方法执行完毕后再清理
        cls.temp_dir.cleanup()

    def test_first_file_operation(self):
        file_path = os.path.join(self.temp_dir_path, "first_test.txt")
        with open(file_path, "w") as f:
            f.write("first test")
        self.assertTrue(os.path.exists(file_path))

    def test_second_file_operation(self):
        # 可以访问上一个测试创建的文件（如果需要的话）
        file_path = os.path.join(self.temp_dir_path, "first_test.txt")
        with open(file_path, "r") as f:
            content = f.read()
        self.assertEqual(content, "first test")

为什么原来的with写法不行？ #

tempfile.TemporaryDirectory的上下文管理器会在with块结束时自动调用cleanup()，而setUp()方法执行完成后，with块就退出了，临时目录被立即销毁，后续测试方法自然无法访问。

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