unittest–python单元测试–批量执行以及生成测试报告
在实际情况中,尤其是针对一个项目的测试,经常会有多个测试文件,此时就需要我们能够批量执行;并针对项目的测试结果,生成可视化、易读性好的测试报告
例如,针对之前提到的calculator.py中的Count类,我们针对add和sub两个函数分别建立测试文件:
testadd.py
# -*- coding: utf-8 -*-
import unittest
from calculator import Count
def setUpClass():
print "TestAdd Module start >>>>>>"
def tearDownClass():
print "TestAdd Module end <<<<<<<<"
class TestAdd(unittest.TestCase):
"""Test Count add function"""
@classmethod
def setUpClass(cls):
print "test add class start ++++++++"
@classmethod
def tearDownClass(cls):
print "test add class end ----------"
def setUp(self):
print "test case start ...."
def tearDown(self):
print "test case end ......"
def test_add1(self):
c = Count(2, 3)
self.assertEqual(c.add(), 5)
def test_add2(self):
c = Count(27, 31)
self.assertEqual(c.add(), 58)
testsub.py
# -*- coding: utf-8 -*-
import unittest
from test1 import Count
def setUpModule():
print u"sub函数测试模块开始 >>>>>>>>>>>>"
def tearDownModule():
print u"sub函数测试模块结束 <<<<<<<<<<<<"
class TestSub(unittest.TestCase):
u"""sub函数测试类"""
@classmethod
def setUpClass(cls):
print u"sub函数测试类开始"
@classmethod
def tearDownClass(cls):
print u"sub函数测试类结束"
def setUp(self):
print "test case start"
def tearDown(self):
print "test case end"
def test_sub(self):
u"""测试sub函数"""
c = Count(2, 3)
self.assertEqual(c.sub(), -1)
def test_sub2(self):
u"""测试sub函数"""
c = Count(71, 46)
self.assertEqual(c.sub(), 25)
除此之外,以后可能还会有testmultiplication.py,testdivision.py等等一系列测试文件,针对每一个文件都需要去执行一次显得有点子蠢。。所以出现了一种可以批量执行的测试用例(测试用例文件)的方法,具体代码和解释说明如下
# -*- coding: utf-8 -*-
import unittest
from HTMLTestRunner import HTMLTestRunner
test_dir = "./" # 指定测试用例文件存放位置
# 在test_dir目录下搜索符和test*.py这个模式的文件
discover = unittest.defaultTestLoader.discover(start_dir=test_dir, pattern="test*.py") 合
with open("result.html", "wb") as fp:
runner = HTMLTestRunner(stream=fp, title=u"测试", description=u"Count类测试")
runner.run(discover) # 批量执行文件
执行后生成的测试报告如下:
注意
通过该方法执行测试用例时,用例执行顺序与使用unittest.main()执行用例时的顺序一致,具体可见之前的unittest的介绍unittest–python单元测试模块(二),如果想控制用例执行顺序需要按照文中方法进行调整。