为什么需要它们？

使用setup和teardown可以确保：

• 每个测试在独立、干净的环境中运行
• 测试之间不会相互影响
• 资源得到正确释放，避免内存泄漏
• 测试结果可重复且可靠

1. 方法级别的setup/teardown

在每个测试方法执行前后运行：

import unittest

class TestExample(unittest.TestCase):
    def setUp(self):
        """每个测试方法前执行"""
        self.data = [1, 2, 3, 4, 5]
        print("执行setUp - 准备测试数据")
    
    def tearDown(self):
        """每个测试方法后执行"""
        self.data = None
        print("执行tearDown - 清理测试数据")
    
    def test_sum(self):
        self.assertEqual(sum(self.data), 15)
    
    def test_length(self):
        self.assertEqual(len(self.data), 5)

运行结果：
执行setUp - 准备测试数据
执行tearDown - 清理测试数据
执行setUp - 准备测试数据
执行tearDown - 清理测试数据

2. 类级别的setup/teardown

在整个测试类执行前后运行：

import unittest

class TestDatabase(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        """整个测试类执行前执行一次"""
        cls.db_connection = create_db_connection()
        print("建立数据库连接")
    
    @classmethod
    def tearDownClass(cls):
        """整个测试类执行后执行一次"""
        cls.db_connection.close()
        print("关闭数据库连接")
    
    def test_query(self):
        result = self.db_connection.query("SELECT * FROM users")
        self.assertGreater(len(result), 0)
    
    def test_insert(self):
        # 测试插入操作
        pass

3. 模块级别的setup/teardown

在整个测试模块执行前后运行：

import unittest

def setUpModule():
    """模块中所有测试执行前运行"""
    print("=== 模块测试开始 ===")

def tearDownModule():
    """模块中所有测试执行后运行"""
    print("=== 模块测试结束 ===")

class TestFeatureA(unittest.TestCase):
    # 测试用例...

class TestFeatureB(unittest.TestCase):
    # 测试用例...

1. 函数级别的setup/teardown

def setup_function(function):
    """每个测试函数前执行"""
    print(f"\n为 {function.__name__} 准备资源")

def teardown_function(function):
    """每个测试函数后执行"""
    print(f"为 {function.__name__} 清理资源")

def test_addition():
    assert 1 + 2 == 3

def test_subtraction():
    assert 5 - 3 == 2

2. 类级别的setup/teardown

class TestCalculator:
    @classmethod
    def setup_class(cls):
        print("\n测试类开始 - 创建计算器实例")
        cls.calc = Calculator()
    
    @classmethod
    def teardown_class(cls):
        print("测试类结束 - 销毁计算器实例")
        del cls.calc
    
    def setup_method(self, method):
        print(f"\n准备测试: {method.__name__}")
    
    def teardown_method(self, method):
        print(f"清理测试: {method.__name__}")
    
    def test_multiplication(self):
        assert self.calc.multiply(3, 4) == 12
    
    def test_division(self):
        assert self.calc.divide(10, 2) == 5

3. 使用fixture实现setup/teardown

pytest推荐使用fixture来实现更灵活的资源管理：

import pytest

@pytest.fixture
def database_connection():
    """创建数据库连接fixture"""
    print("\n建立数据库连接")
    conn = create_db_connection()
    yield conn  # 测试执行点
    print("关闭数据库连接")
    conn.close()

def test_query_users(database_connection):
    users = database_connection.query("SELECT * FROM users")
    assert len(users) > 0

def test_insert_data(database_connection):
    # 测试插入操作
    pass

如何选择合适的作用域？

作用域	适用场景	执行频率
方法/函数级	需要完全隔离的测试	每个测试方法前后
类级	共享资源的测试类	测试类前后各一次
模块级	整个测试模块共享的资源	测试模块前后各一次
会话级	全局资源（如Docker容器）	整个测试会话前后各一次