在Python中如何使用unittest.case进行测试数据的准备和清理

在Python中，unittest模块提供了一个TestCase类，用于编写测试用例。TestCase类中的方法用于准备测试数据、执行测试代码和清理测试数据。

首先，我们需要导入unittest和要测试的类或函数。假设我们要测试一个名为Calculator的类，该类具有add和subtract方法。

import unittest
from calculator import Calculator

接下来，我们需要定义一个继承自unittest.TestCase的测试类。在这个测试类中，我们可以定义一些特殊的方法来准备和清理测试数据。

class CalculatorTestCase(unittest.TestCase):
    def setUp(self):
        # 在每个测试方法执行前准备数据
        self.calculator = Calculator()
    
    def tearDown(self):
        # 在每个测试方法执行后清理数据
        del self.calculator

在setUp方法中，我们可以准备测试数据。通常情况下，我们会在这里初始化被测试类的实例。在tearDown方法中，我们可以清理测试数据。在这个例子中，我们简单地删除测试类的实例。

接下来，我们可以编写测试方法。测试方法以test开头，并且不带任何参数。在每个测试方法中，我们可以使用self.calculator来访问被测试类的实例。

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

在这个例子中，我们编写了两个测试方法test_add和test_subtract，分别用于测试add和subtract方法。在每个测试方法中，我们调用被测试类的方法，并使用断言方法self.assertEqual来验证结果是否符合预期。

最后，我们可以使用unittest的TestLoader和TextTestRunner来运行测试。

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

完整的测试代码如下所示：

import unittest
from calculator import Calculator

class CalculatorTestCase(unittest.TestCase):
    def setUp(self):
        self.calculator = Calculator()
    
    def tearDown(self):
        del self.calculator
    
    def test_add(self):
        result = self.calculator.add(2, 3)
        self.assertEqual(result, 5)
    
    def test_subtract(self):
        result = self.calculator.subtract(5, 3)
        self.assertEqual(result, 2)

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

在执行测试时，我们可以看到测试运行结果的输出：

..
----------------------------------------------------------------------
Ran 2 tests in 0.001s

OK

在这个例子中，我们使用了setUp方法准备了测试数据，使用了tearDown方法清理了测试数据，并使用了两个测试方法进行了测试。通过使用unittest模块的TestCase类，我们可以方便地进行测试数据的准备和清理，并且可以对测试结果进行断言。