unittest模块
1，基本使用
先准备好要测试的文件，如project.py
def add(a, b): return a + b def divide(a, b): return a / b def even(a): if a % 2 == 0: return True else: return False
再新建一个test.py文件，里面的测试方法必须以test开头，不然识别不了
import unittest from project import add, divide, even class Test_Project(unittest.TestCase): def setUp(self) -> None: """可用来准备测试前的环境""" print("每个测试方法执行前触发") def tearDown(self) -> None: """可用来还原测试后的环境""" print("每个测试方法执行后触发") def test_add(self): print("add") self.assertEqual(4, add(1, 3)) self.assertNotEqual(4, add(1, 2)) def test_divide(self): print("divide") self.assertEqual(3, divide(9, 3)) self.assertNotEqual(3, divide(4, 2)) def test_even(self): print("even") self.assertTrue(even(4)) self.assertTrue(even(9)) self.assertFalse(even(7)) if __name__ == '__main__': # 执行整个测试单元 unittest.main()
执行结果如下：..F表示测试的三个方法里面有一个失败，需要测试的方法不是安装写的先后顺序来执行的，是按照方法名的数字字母的排序来执行的。
每个测试方法执行前触发 add 每个测试方法执行后触发 每个测试方法执行前触发 devide 每个测试方法执行后触发 每个测试方法执行前触发 is_prime 每个测试方法执行后触发 ..F ====================================================================== FAIL: test_even (__main__.Test_Project) ---------------------------------------------------------------------- Traceback (most recent call last): File "D:/PycharmProjects/PM/test01/test.py", line 26, in test_even self.assertTrue(even(9)) AssertionError: False is not true ---------------------------------------------------------------------- Ran 3 tests in 0.000s FAILED (failures=1) Process finished with exit code 1
2，使用TestSuite来控制测试的执行顺序
将main方法替换成如下
if __name__ == '__main__': tests = [Test_Project("test_divide"), Test_Project("test_even"), Test_Project("test_add")] suite = unittest.TestSuite() suite.addTests(tests) runner = unittest.TextTestRunner() runner.run(suite)
3，如果只希望在所有测试方法执行前只准备一次环境或者只还原一次环境
class Test_Project(unittest.TestCase): @classmethod def setUpClass(cls) -> None: """可用来准备测试前的环境""" print("第一个测试方法执行前触发") @classmethod def tearDownClass(cls) -> None: """可用来还原测试后的环境""" print("最后一个测试方法执行后触发")
4，使用skip装饰器跳过某个case
被跳过的case执行结果会显示为S
@unittest.skipUnless(sys.platform.startswith("Linux"), "required Linux") def test_divide(self): print("divide") self.assertEqual(3, divide(9, 3)) self.assertNotEqual(3, divide(4, 2))
一共有三种skip装饰器和一种预期失败装饰器，就是不管执行结果是什么， 直接将case设置为失败
unittest.skip(reason):无条件跳过测试的装饰器，说明跳过的原因 unittest.skipIf(condition, reason)：如果条件为真，跳过测试的装饰器，reason说明跳过的理由 unittest.skipUnless(condition, reason):跳过测试的装饰器，除非条件为真，reason说明跳过的理由
unittest.expectedFailure()：测试标记为失败。不管执行结果是否失败，统一标记失败。
5，将执行结果输出
输出的结果的信息用TextTestRunner方法的参数表示
if __name__ == '__main__': tests = [Test_Project("test_divide"), Test_Project("test_even"), Test_Project("test_add")] suite = unittest.TestSuite() suite.addTests(tests) with open(r"report", "a") as f: # verbosity为0表示不输出用力的执行结果，为1表示一般输出，为2表示仔细输出，默认为1 runner = unittest.TextTestRunner(stream=f, verbosity=2) runner.run(suite)
输出为HTML格式
将HTMLTestRunner.py文件下载放到当前目录下： http://tungwaiyip.info/software/HTMLTestRunner.html
from HTMLTestRunner import HTMLTestRunner with open('report.html', 'a') as f: runner = HTMLTestRunner( stream=f, title='Test Report', verbosity=2 ) runner.run(suite)
6，自动加载测试用例
另起一个文件，跟test.py和project.py在同一个目录下
loader = unittest.TestLoader() suite = unittest.TestSuite()
# 加载一个类 loader.loadTestsFromTestCase(test.TestRpoject) suite.addTests(loader.loadTestsFromModule(test)) with open(r"report", "a") as f： runner = unittest.TextTestRunner(stream=f, verbosity=2) runner.run(suite)
discover方法: discover(start_dir, pattern='test*.py', top_level_dir=None)
# discover方法，自动查找test_dir目录下所有测试用例 test_dir = './test_case' discover = unittest.defaultTestLoader.discover(test_dir,pattern='test*.py') if __name__ == '__main__': runner = unittest.TextTestRunner() runner.run(discover)
