unittest--python单元测试模块（三）

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单元测试模块（二），如果想控制用例执行顺序需要按照文中方法进行调整。