Python unittest模块的相关用法

待测源码类：

widget.py

<br /> #!/usr/bin/env python</p> <p>class Widget:<br /> def __init__(self, size = (40, 40)):<br /> self._size = size</p> <p> def getSize(self):<br /> return self._size</p> <p> def resize(self, width, height):<br /> if width == 0 or height < 0:<br /> raise ValueError, "illegal size"<br /> self._size = (width, height)</p> <p> def dispose(self):<br /> pass<br />

测试代码类：

widget_unittest.py

<br /> #!/usr/bin/env python</p> <p>import sys<br /> import unittest<br /> from widget import Widget</p> <p>class WidgetTestCase(unittest.TestCase):<br /> def setUp(self):<br /> self.widget = Widget() </p> <p> def tearDown(self):<br /> self.widget = None</p> <p> def testSize(self):<br /> self.assertEqual(self.widget.getSize(), (40, 40))</p> <p> def testReSize(self):<br /> self.widget.resize(20,30)<br /> self.assertEqual(self.widget.getSize(), (20, 30))</p> <p>def suite():<br /> suite = unittest.TestSuite()<br /> suite.addTest(WidgetTestCase("testSize"))<br /> suite.addTest(WidgetTestCase("testReSize"))<br /> return suite</p> <p>if __name__ == "__main__":<br /> unittest.TextTestRunner().run(suite())<br /> # If you use IDE to execute test script in an external console,<br /> # uncomment following code line.<br /> # while True<br /> # if raw_input("Type y or Y to close!") in ("y","Y"): sys.exit(0)<br />

运行结果：

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")