如何在TestCase中重写tearDown()方法以便在每个测试方法之后清理环境

在编写测试用例（TestCase）时，有时候我们需要在每个测试方法执行完成后清理一些环境，比如关闭数据库连接、删除测试数据等。为了实现这样的需求，可以重写tearDown()方法。

tearDown()方法是在每个测试方法执行完成后自动调用的方法，用于清理测试环境。以下是重写tearDown()方法的步骤。

步骤一：创建一个继承自unittest.TestCase的测试类。

import unittest

class MyTestCase(unittest.TestCase):
    ...

步骤二：重写setUp()方法，用于初始化测试环境，在每个测试方法执行前调用。

def setUp(self):
    # 初始化测试环境代码

步骤三：重写tearDown()方法，用于清理测试环境，在每个测试方法执行后调用。

def tearDown(self):
    # 清理测试环境代码

下面是一个具体的例子，假设我们要测试一个名为Calculator的计算器类，其中有一个加法方法add()用于计算两个数的和。在每个测试方法执行后，我们希望清理计算器对象，以便下一个测试方法可以重新创建一个新的计算器对象。

import unittest

class Calculator:
    def add(self, a, b):
        return a + b

class CalculatorTestCase(unittest.TestCase):
    def setUp(self):
        self.calculator = Calculator()

    def tearDown(self):
        self.calculator = None

    def test_add(self):
        result = self.calculator.add(2, 3)
        self.assertEqual(result, 5)

    def test_add_negative_numbers(self):
        result = self.calculator.add(-2, -3)
        self.assertEqual(result, -5)

if __name__ == "__main__":
    unittest.main()

以上代码中，setUp()方法在每个测试方法执行前创建一个名为calculator的Calculator对象。tearDown()方法在每个测试方法执行后将calculator对象置为None，以清理测试环境。在test_add()和test_add_negative_numbers()方法中，我们分别进行了加法测试，并验证了测试结果是否符合预期。

通过重写tearDown()方法，我们可以在每个测试方法执行后进行环境的清理，以确保测试的独立性和准确性。