使用unittest进行测试，如果是需要实现上百个测试用例，把它们全部写在一个test.py文件中，文件会越来越臃肿，后期维护页麻烦。此时可以将这些用例按照测试功能进行拆分，分散到不同的测试文件中。
Python代码如下：
testadd.py：
from calculator import Math
import unittest
class TestAdd(unittest.TestCase):
def setUp(self):
print("test case start")
def tearDown(self):
print("test case end")
def test_add(self):
j = Math(2, 3)
self.assertEqual(j.add(), 5)
def test_add2(self):
j = Math(41, 76)
self.assertEqual(j.add(), 117)
if __name__ == '__main__':
unittest.main()
testsub.py：
from calculator import Math
import unittest
class TestSub(unittest.TestCase):
def setUp(self):
print("test case start")
def tearDown(self):
print("test case end")
def test_sub(self):
j = Math(2, 3)
self.assertEqual(j.sub(), -1)
def test_sub2(self):
j = Math(81, 76)
self.assertEqual(j.sub(), 5)
if __name__ == '__main__':
unittest.main()
runtest.py：
import unittest
test_dir = './'
discover = unittest.defaultTestLoader.discover(test_dir, pattern='test*.py')
if __name__ == '__main__':
runner = unittest.TextTestRunner()
runner.run(discover)
TestLoader
该类根据各种标准加载测试用例，并将它们返回给测试套件。正常情况下，不需要创建这个类的实例。unittest提供了可以共享的defaultTestLoader类，可以使用其子类和方法创建实例，discover()就是其中之一。
discover(start_dir,pattern='test*.py',top_level_dir=None)
找到指定目录下所有测试模块，并可递归查到子目录下的测试模块，只有匹配到文件名才能被加载。如果启动的不是顶层目录，那么顶层目录必须单独指定。
- start_dir：要测试的模块名或测试用例目录
- pattern='test*.py'：表示用例文件名的匹配原则。此处匹配文件名以“test”开头的“.py”类型的文件，幸好“*”表示任意多个字符
- top_level_dir=None：测试模块的顶层目录，如果没有顶层目录，默认为None