python的单元测试框架pyUnit和java的junit非常类似,以下是对模块中的方法进行单元测试的例子:
实现代码:输入一个整数组成的列表,判断其中是否有重复的数字,如果有返回True,否则返回False
class Solution:
# @param {integer[]} nums
# @return {boolean}
def containsDuplicate(self, nums):
return len(nums) > len(set(nums))
单元测试代码如下,有两种方法,一种是把不同的测试用例写成测试类的不同方法,分别调用。这样的程序在运行时需要选择Run As->python unit test。或者在main()函数中用unittest.main()来调用。另一种方法是在main()函数中用TestSuite类把测试用例组装成测试套,然后一起运行,这样在运行时可以按照python程序的类型来运行,而不是按照单元测试的类型来运行。
import unittest
from Solution import Solution
class MyTest(unittest.TestCase):
def setUp(self):
self.testclass = Solution()
def testCase1(self):
self.assertEqual(False, self.testclass.containsDuplicate([1,2,3,4,5,6]), 'test failed!')
def testCase2(self):
self.assertEqual(True, self.testclass.containsDuplicate([1,2,3,3,5,6]), 'test failed!')
def testCase3(self):
self.assertEqual(True, self.testclass.containsDuplicate([1,1,1,3,5,6]), 'test failed!')
def tearDown(self):
self.testclass = None
if __name__ == '__main__':
#方法一,直接调用,这样会运行所有test打头的测试用例
#unittest.main()
#方法二,把指定的测试用例组装成测试套,然后一起运行
suite = unittest.TestSuite()
suite.addTest(MyTest('testCase1'))
suite.addTest(MyTest('testCase2'))
suite.addTest(MyTest('testCase3'))
#也可以不用上边的分别组装的方法,这行代码可以返回所有test打头的测试用例套,和方法一的效果类似
#suite = unittest.TestLoader().loadTestsFromTestCase(MyTest)
runner = unittest.TextTestRunner()
runner.run(suite)
参考文档: