unittest用例运行脚本

import unittest
import os
from HTMLTestRunner import HTMLTestRunner #导入HTMLTestRunner类
from 自己练习.testcases import test_login,test_register
#初始化加载器对象test_loader
loader=unittest.TestLoader()

#获取测试用例目录
dir_path = os.path.dirname(os.path.abspath(__file__))  #获取上级目录
case_path = os.path.join(dir_path,'testcases')

#收集所有用例
test_suit = loader.discover(case_path)   #discover方法第一个参数表示测试用例目录，第二个参数表示以test开头的模块

# #只收集登录、注册模块用例
# suit_login = loader.loadTestsFromModule(test_login)
# suit_register = loader.loadTestsFromModule(test_register)
# 
# #初始化一个TestSuite实例对象
# suit_total = unittest.TestSuite()
# suit_total.addTests([suit_login,suit_register])

#执行测试用例test_suit，无报告生成
runner = unittest.TextTestRunner()  #初始化一个执行器runner
runner.run(test_suit)

#生成txt格式测试报告
with open('test_report.txt','w', encoding='utf-8') as f:
    runner = unittest.TextTestRunner(f) #f作为参数传入
    runner.run(test_suit)

#生成html报告
with open('test_report.html','wb') as f:
    runner = HTMLTestRunner(f,title='python29期第一次测试报告',description='测试报告描述',tester='jojo') #f作为参数传入
    runner.run(test_suit)