python unittest单元测试框架-2discover

基于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)

原文地址：https://www.cnblogs.com/csj2018/p/10023839.html