python+unittest测试框架学习笔记

from study import unittestDemo
import unittest

if __name__ == "__main__":
    # 第一个unittest方法,使用TestSuite()
    suitt = unittest.TestSuite()
    suitt.addTests(map(unittestDemo.unitMymath, ["test_add_1", "test_add_2"]))
    unittestDemo.unittest.TextTestRunner().run(suitt)

    # 第二个unittest方法，使用TestLoader()
    suite = unittest.TestSuite()
    loader = unittest.TestLoader()
    suite1 = loader.loadTestsFromTestCase(unittestDemo.unitMymath)
    suite.addTest(suite1)
    unittestDemo.unittest.TextTestRunner().run(suite)

    # 第三个unittest方法，通过导入包来使用discover
    suitt = unittest.defaultTestLoader.discover(r"../study/", pattern="unittestD*.py")
    with open(r"../study/re.txt","a",encoding="utf-8") as f:
        runner = unittest.TextTestRunner(f,descriptions="单元测试执行报告",verbosity=2)
        runner.run(suitt)

    # 输出执行的测试用例的内容
    suitt = unittest.TestSuite()
    suitt.addTests(map(unittestDemo.unitMymath, ["test_add_1", "test_add_2"]))
    re = unittest.TestResult()
    suitt.run(re)
    print(re.__dict__)

	# 输出到html文件中
    result = unittest.defaultTestLoader.discover(r"../study/", pattern="unittestD*.py")
    filename = time.strftime("%Y-%m-%d-%H-%M-%S")+r".html" # 以时间格式命名
    path = os.path.dirname(__file__)+"\\"
    filename = path+filename
    print(filename)
    with open(filename,"wb") as f:
        runner = HTMLTestRunner(f,verbosity=2,title="单元测试报告",description="第一次运行结束")
        runner.run(result)