python的简单unittest例子

import unittest
from BeautifulReport import BeautifulReport

class TestDemo(unittest.TestCase):
    def setUp(self):
        print("setup")

   def tearDown(self):
        print("tearDown")

   def test1(self):
        print("test1")

def suite():
    test_suite = unittest.TestSuite()
   #loader = unittest.TestLoader()  # 创建一个用例加载对象
   #suite.addTest(loader.loadTestsFromTestCase(test_demo1.TestDemo))# 类名
   #suite.addTest(loader.loadTestsFromModule(test_demo2))# 模块
   #return suite
   #pattern='test*.py' 规定测试集文件开头命名为test，也可以是pattern='test_*.py'
   #discover方法找到path 目录下所有文件到的测试用例组装到测试套件
   #因此可以直接通过run()方法执行discover
   #runner = unittest.TextTestRunner()
   #runner.run(discover)
   discover = unittest.defaultTestLoader.discover(".", pattern='test_demo1.py', top_level_dir=None)# 查找文件名称
   print(discover)
   for suite in discover:  # 如果存在，循环取出元素组内容，命名为suite
       #print(suite)
       for test_name in suite:  # 从discover中取出test_name，使用addTest添加到测试集
           #print(test_name)
           test_suite.addTest(test_name)
   #print(test_suite)
   return test_suite
   
if __name__=="__main__":
   br = BeautifulReport(suite())
   br.report(filename='testdemoreport.html', description='测试报告', log_path='.', report_dir='.')