参考
https://docs.python.org/2/library/unittest.html#organizing-tests
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()
首先是一个基本的例子，
这个类中有三个方法，以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.000s
OK
.
----------------------------------------------------------------------
Ran 1 test in 0.000s
OK
[root@back-test- ut]# python -m unittest discover
...
----------------------------------------------------------------------
Ran 3 tests in 0.000s
OK
测试用例case是基本的测试块， 而测试用例 又是TestCase的实例，
最基本的 TestCase 类必须重写runTest()方法
import unittest
class DefaultWidgetSizeTestCase(unittest.TestCase):
def runTest(self):
widget = Widget('The widget')
self.assertEqual(widget.size(), (50, 50), 'incorrect default size')
在上个例子里，每次写测试用例，都需要 自己 先构造一个widget很麻烦， unittest提供了 setUp()方法，来帮助我们自动地在每次运行测试用例前，做一下准备工作，像下面这样：
import unittest
class 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 unittest
class WidgetTestCase(unittest.TestCase):
def setUp(self):
self.widget = Widget('The widget')
def tearDown(self):
self.widget.dispose()
self.widget = None
def 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 code
pass
也可以跳过整个类。
