unittest测试套件.py

import unittest


class TestCase1(unittest.TestCase):
    def test_case_01(self):
        self.assertTrue(1)

    def test_case_02(self):
        self.assertTrue(0)


if __name__ == '__main__':
    # 1、实例化用例对象:
    case1 = TestCase1("test_case_01")
    case2 = TestCase1("test_case_02")

    # 高阶函数：
    case = map(TestCase1, ["test_case_01", "test_case_02"])

    # 2、创建测试套件容器：
    suite_obj = unittest.TestSuite()
    suite_obj.addTest(case1)
    suite_obj.addTest(case2)

    # 2.1for循环两个用例：
    for i in [case1, case2]:
        suite_obj.addTest(i)

    # 2.2复数添加：
    suite_obj.addTests([case1, case2])

    #添加高阶函数：
    suite_obj.addTests(case)

    # 3、找个执行器执行测试套件中的用例：
    unittest.TextTestRunner().run(suite_obj)