Python自动单元测试实例

Python unittest模块的相关用法

待测源码类：widget.py

#!/usr/bin/env python

class Widget:
def __init__(self, size = (40, 40)):
self._size = size

def getSize(self):
return self._size

def resize(self, width, height):
if width == 0  or height < 0:
raise ValueError, "illegal size"
self._size = (width, height)

def dispose(self):
pass

测试代码类：widget_unittest.py

#!/usr/bin/env python

import sys
import unittest
from widget import Widget

class WidgetTestCase(unittest.TestCase):
def setUp(self):
self.widget = Widget()

def tearDown(self):
self.widget = None

def testSize(self):
self.assertEqual(self.widget.getSize(), (40, 40))

def testReSize(self):
self.widget.resize(20,30)
self.assertEqual(self.widget.getSize(), (20, 30))

def suite():
suite = unittest.TestSuite()
suite.addTest(WidgetTestCase("testSize"))
suite.addTest(WidgetTestCase("testReSize"))
return suite

if __name__ == "__main__":
unittest.TextTestRunner().run(suite())
# If you use IDE to execute test script in an external console,
# uncomment following code line.
# while True
#     if raw_input("Type y or Y to close!") in ("y","Y"): sys.exit(0)

运行结果：python widget_unittest.py

[work@db-testing-com06-vm3.db01.baidu.com test]$ python widget_unittest.py
..
----------------------------------------------------------------------
Ran 2 tests in 0.000s

OK

动态方法不再覆盖runTest()方法，而是为测试类编写多个测试方法（按习惯这些方法通常以test开头），在创建TestCase子类的实例时必须给出测试方法的名称，来为PyUnit测试框架指明运行该测试用例时究竟应该调用测试类中的哪个方法.

如果用于测试的类中所有的测试方法都以test开，Python程序员甚至可以用PyUnit模块提供的makeSuite()方法来构造一个TestSuite：

def suite():
return unittest.makeSuite(WidgetTestCase, "test")