unittest框架提供了测试装置（Test Fixture）的功能，用于在测试用例执行前后进行准备和清理操作。测试装置包括setUp()方法和tearDown()方法，分别用于在每个测试用例执行前和执行后进行相应的操作。
下面是一个示例，演示如何使用测试装置：
import unittest class MyTestCase(unittest.TestCase): def setUp(self): # 在每个测试用例执行前进行准备操作 self.username = "testuser" self.password = "password" self.login() def tearDown(self): # 在每个测试用例执行后进行清理操作 self.logout() def test_login_success(self): # 执行登录成功的测试用例 self.assertTrue(self.is_logged_in()) def test_login_failure(self): # 执行登录失败的测试用例 self.assertFalse(self.is_logged_in()) def login(self): # 模拟登录操作 print("执行登录操作") def logout(self): # 模拟退出操作 print("执行退出操作") def is_logged_in(self): # 判断是否登录成功 return True # 假设登录成功 if __name__ == '__main__': unittest.main()
在上述示例中，setUp()方法在每个测试用例执行前被调用，用于进行登录操作，设置必要的测试数据。tearDown()方法在每个测试用例执行后被调用，用于执行退出操作，清理测试环境。
在测试用例中，可以直接使用setUp()方法中设置的数据和操作。这样可以确保每个测试用例在相同的环境下执行，并且不会相互干扰。
测试装置的使用可以提高测试的效率和可维护性，避免在每个测试用例中重复设置相同的数据和操作。同时，还可以在测试装置中进行其他初始化、配置和资源管理的操作，以适应不同的测试需求。
测试装置的几种使用方法：
-
setUp()和tearDown()方法：
- setUp()方法在每个测试用例执行前被调用，用于进行准备操作，如初始化数据、创建对象等。
- tearDown()方法在每个测试用例执行后被调用，用于进行清理操作，如释放资源、关闭连接等。
- 这两个方法在每个测试用例之间是相互独立的，每个测试用例都会执行一次对应的setUp()和tearDown()方法。
-
setUpClass()和tearDownClass()方法：
- setUpClass()方法在测试类执行前被调用，用于进行准备操作，如初始化类级别的数据、创建共享资源等。
- tearDownClass()方法在测试类执行后被调用，用于进行清理操作，如释放类级别的资源、关闭共享连接等。
- 这两个方法在整个测试类的生命周期中只会执行一次，而不是每个测试用例都执行。
示例代码如下所示：
import unittest class MyTestCase(unittest.TestCase): @classmethod def setUpClass(cls): # 在测试类执行前进行准备操作 print("执行测试类的准备操作") @classmethod def tearDownClass(cls): # 在测试类执行后进行清理操作 print("执行测试类的清理操作") def setUp(self): # 在每个测试用例执行前进行准备操作 print("执行测试用例的准备操作") def tearDown(self): # 在每个测试用例执行后进行清理操作 print("执行测试用例的清理操作") def test_case1(self): # 测试用例1 print("执行测试用例1") def test_case2(self): # 测试用例2 print("执行测试用例2") if __name__ == '__main__': unittest.main()