Python单元测试框架-unittest

本文主要是介绍Python单元测试框架-unittest，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

参考
https://docs.python.org/2/library/unittest.html#organizing-tests

import unittestclass 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 stringwith self.assertRaises(TypeError):s.split(2)if __name__ == '__main__':unittest.main()

首先是一个基本的例子，
这个类中有三个方法，以test开头，这种命名会作为测试用例。会通知runner

assertEqual() 检查相等
assertTrue() 检查条件为真
assertFalse() 检查条件为假
assertRaises() 检查会抛出某种异常

这里例子虽然简单，其中特性可以满足我们每天进行的测试了。

做下面的命令行测试时候，请注意Python的版本，2.7.10的不报错，可是2.6.6就会报错。

[root@-01-VM-back-test- ut]# /usr/local/bin/python -m unittest test.TestStringMethod.test_upper
test_isupper (test.TestStringMethod) ... ok
test_split (test.TestStringMethod) ... ok
test_upper (test.TestStringMethod) ... ok----------------------------------------------------------------------
Ran 3 tests in 0.000sOK
.
----------------------------------------------------------------------
Ran 1 test in 0.000sOK

[root@back-test- ut]# python -m unittest discover
...
----------------------------------------------------------------------
Ran 3 tests in 0.000sOK

测试用例case是基本的测试块，而测试用例又是TestCase的实例，

最基本的 TestCase 类必须重写runTest()方法

import unittestclass DefaultWidgetSizeTestCase(unittest.TestCase):def runTest(self):widget = Widget('The widget')self.assertEqual(widget.size(), (50, 50), 'incorrect default size')

在上个例子里，每次写测试用例，都需要自己先构造一个widget很麻烦， unittest提供了 setUp()方法，来帮助我们自动地在每次运行测试用例前，做一下准备工作，像下面这样：

import unittestclass SimpleWidgetTestCase(unittest.TestCase):#准备工作def setUp(self):self.widget = Widget('The widget')#清理工作    def tearDown(self):self.widget.dispose()self.widget = None#01
class DefaultWidgetSizeTestCase(SimpleWidgetTestCase):def runTest(self):self.assertEqual(self.widget.size(), (50,50),'incorrect default size')
#02
class WidgetResizeTestCase(SimpleWidgetTestCase):def runTest(self):self.widget.resize(100,150)self.assertEqual(self.widget.size(), (100,150),'wrong size after resize')

在写01 ，02 两个测试用例时候，我们继承了SimpleWidgetTestCase 这个类，在SimpleWidgetTestCase这个类中有setUp()方法，来帮我们自动的每次做好准备的工作。

类似的，有一个tearDown() 方法，来做每次测试后的清理工作。

这样的测试环境，在unittest中叫测试运行的fixture，类似上下文。

但是如果这样的测试用例很多，SimpleWidgetTestCase 这个类就要被重载许多次，这很累，所以下面这个机制，可以省一点时间。就是把这些都写在一个类里面。

import unittestclass WidgetTestCase(unittest.TestCase):def setUp(self):self.widget = Widget('The widget')def tearDown(self):self.widget.dispose()self.widget = Nonedef test_default_size(self):self.assertEqual(self.widget.size(), (50,50),'incorrect default size')def test_resize(self):self.widget.resize(100,150)self.assertEqual(self.widget.size(), (100,150),'wrong size after resize')

这里我们没有提供runTest() 的方法，没关系，实例会运行每个test_*()的方法。所以只要名字写的规范就可以了。

跳过测试用例,使用装饰器@unittest.skip()，如下：

class MyTestCase(unittest.TestCase):@unittest.skip("demonstrating skipping")def test_nothing(self):self.fail("shouldn't happen")@unittest.skipIf(mylib.__version__ < (1, 3),"not supported in this library version")def test_format(self):# Tests that work for only a certain version of the library.pass@unittest.skipUnless(sys.platform.startswith("win"), "requires Windows")def test_windows_support(self):# windows specific testing codepass