import unittest
#加载测试文件
#import test_add
#import testsub
#构造测试集
#suite=unittest.TestSuite()
#suite.addTest(test_add.TestAdd("test_add"))
#suite.addTest(test_add.TestAdd("test_add2"))
#suite.addTest(testsub.TestSub("test_sub"))
#suite.addTest(testsub.TestSub("test_sub2"))
#定义测试用例的目录为当前目录
test_dir='./'
discover =unittest.defaultTestLoader.discover(test_dir,pattern='test*.py')
if __name__ =='__main__':
#执行测试
runner=unittest.TextTestRunner()
#runner.run(suite)
#加载测试文件
#import test_add
#import testsub
#构造测试集
#suite=unittest.TestSuite()
#suite.addTest(test_add.TestAdd("test_add"))
#suite.addTest(test_add.TestAdd("test_add2"))
#suite.addTest(testsub.TestSub("test_sub"))
#suite.addTest(testsub.TestSub("test_sub2"))
#定义测试用例的目录为当前目录
test_dir='./'
discover =unittest.defaultTestLoader.discover(test_dir,pattern='test*.py')
if __name__ =='__main__':
#执行测试
runner=unittest.TextTestRunner()
#runner.run(suite)
runner.run(discover)
被注释的代码:对上一博客内容的测试用例进行拆分,拆分后的目录结构
testpro/
|--count.py
|--testadd.py
|--testsub.py
|--runtest.py
discover()方法会自动根据测试目录(test_dir)匹配查找测试用例文件(test*.py)并将查找到的测试用例组装到测试套件中,因此,可以直接通过run()方法执行discover,大大简化了测试用例的查找与执行。