基于TestLoader提供的discover方法实现用例执行
当用例达达到数十条后,在runtest.py文件中通过addTest()添加/删除测试用例就非常麻烦。此时可以使用discover方法找到目录下所有测试模块,并可以递归查到子目录下的测试模块,只有匹配到文件名才能被加载。如果启动的不是顶层目录,那么顶层目录必须单独指定。
discover(start_dir, pattern="test*.py", top_level_dir=None)
#start_dir:要测试的模块名或测试用例目录
#pattern="test*.py":表示用例文件名的匹配原则。此处匹配以test开头的.py文件
#top_level_dir=None:测试模块的顶层目录,如果没有顶层目录,则默认为None
calculator.py
class Count:
def __init__(self,a,b):
self.a = int(a)
self.b = int(b)
def add(self):
return self.a + self.b
def sub(self):
return self.a - self.b
testadd.py
from calculator import Count
import unittest
class TestAdd(unittest.TestCase):
def setUp(self):
print("test add start")
def test_add(self):
j = Count(2,3)
self.assertEqual(j.add(),5)
print("test_add执行完毕")
def test_add2(self):
j = Count(54,49)
self.assertEqual(j.add(),103)
print("test_add2执行完毕")
def tearDown(self):
print("test add end")
if __name__=="__main__":
unittest.main()
testsub.py
from calculator import Count
import unittest
class TestSub(unittest.TestCase):
def setUp(self):
print("test sub start")
def test_sub(self):
j = Count(4,1)
self.assertEqual(j.sub(),3)
print("test_sub执行完毕")
def test_sub2(self):
j = Count(5,9)
self.assertEqual(j.sub(),-4)
print("test_sub2执行完毕")
def tearDown(self):
print("test sub end")
if __name__=="__main__":
unittest.main()
runtest.py
import unittest
test_dir="./"
discover=unittest.defaultTestLoader.discover(test_dir,pattern="test[a-z]{3}.py")
if __name__=="__main__":
runner = unittest.TextTestRunner()
runner.run(discover)