一、python unittest简介

unittest是python下的单元测试框架，是java JUnit的python版本，跟其它语言下的单元测试框架风格类似，unittest支持自动化测试、共享setup和teardown代码、测试聚合成集、独立于报告框架。unittest模块提供了一个丰富的工具集用于构建和执行用例，先看一个入门的例子

import unittest

class TestStringMethods(unittest.TestCase):

    def test_upper(self):
        self.assertEqual('foo'.upper(), 'FOO')

    def test_isupper(self):
        self.assertTrue('FOO'.isupper())
        self.assertFalse('Foo'.isupper())

    def test_split(self):
        s = 'hello world'
        self.assertEqual(s.split(), ['hello', 'world'])
        # check that s.split fails when the separator is not a string
        with self.assertRaises(TypeError):
            s.split(2)

if __name__ == '__main__':
    unittest.main()

可以通过继承unittest.TestCase创建一个测试用例TestStringMethods，在这个用例中定义了测试函数，这些函数名字都以”test”开头，在执行测试用例TestStringMethods时，这些方法会被自动调用。每个测试函数中都调用了assertTrue()和assertFalse()方法检查预期结果，或者使用assertRaises()确认产生了一个特定异常。现在来看一下这段代码的运行结果：

...
----------------------------------------------------------------------
Ran 3 tests in 0.000s

OK

有时我们需要在用例执行前后做一些工作如初始化和清理，这就需要实现setUp()和tearDown()方法

import unittest

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

    def test_1(self):
        print("test_1")

    def test_2(self):
        print("test_2")

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

if __name__ == '__main__':
    unittest.main()

运行结果：

setUp()
.test_1
tearDown()
setUp()
.test_2
tearDown()
----------------------------------------------------------------------
Ran 2 tests in 0.000s

OK

注：如果setUp()执行成功(没有异常发生)，那么无论测试方法是否通过，tearDown()都会被执行
根据所测的特性测试用例被组合在一起，通过调用unittest.main()，unittest测试框架会自动收集所有模块的测试用例然后执行。

import unittest

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

    def test_Widget(self):
        print("test_Widget")

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

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

    def test_func(self):
        print("test_func")

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


if __name__ == '__main__':
    unittest.main()

如果想构建自已的用例集，只需要这么做：

import unittest

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

    def test_Widget(self):
        print("test_Widget")

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

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

    def test_func(self):
        print("test_func")

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

def suite():
    suite = unittest.TestSuite()
    suite.addTest(FuncTestCase('test_func'))
    return suite

if __name__ == '__main__':
    runner=unittest.TextTestRunner()
    runner.run(suite())

二、unittest中相关类和函数

在unittest中 TestCase类的实例代表逻辑测试单元，这个类通常被当作测试类的基类使用， TestCase类实现了许多测试相关的接口，主要是以下三组方法：

setUp()
#在每个测试方法之前执行，这个方法引发的异常会被认为是错误，而非测试失败，默认实现是不做任何事
tearDown()
#在每个测试方法之后执行，即使测试方法抛出异常tearDown()方法仍会执行，并且只有setUp()成功执行时tearDown()才会执行，
#同样这个方法引发的异常会被认为是错误，而非测试失败。默认实现是不做任何事
setUpClass()
#在一个测试类的所有测试方法执行之前执行，相当于google test中的SetUpTestCase()方法,setUpClass()必须被装饰成一个classmethod()
@classmethod
def setUpClass(cls):
    ...
tearDownClass()
#在一个测试类的所有测试方法执行之后执行，相当于google test中的TearDownTestCase()方法，tearDownClass()必须被装饰成一个classmethod()
@classmethod
def tearDownClass(cls):
    ...

Python unittest单元测试框架

二、unittest中相关类和函数

猜你喜欢