使用unittest里面的discover()方法组织测试用例
import os
import unittest
directory = os.getcwd()
# 测试用例的目录
organize = unittest.defaultTestLoader.discover(directory, pattern='test*.py')
"""
discover()方法有3个参数
start_dir表示测试的模块名称或者测试用例的目录
pattern='test*.py'表示测试用例文件名的匹配规则,请注意文件名以test开头
top_level_dir=NONE表示测试模块的顶层目录
"""
if __name__ == '__main__':
runner = unittest.TextTestRunner()
runner.run(organize)
# discover()方法会自动根据测试目录匹配测试文件,并将测试用例添加到测试套件中