python单元测试框架

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)

 

参考文档:

http://wenku.baidu.com/link?url=Wxk_RCRtQyjPTar_5GDVMZrGKKsnluBrt8utjjhOpXguK0r88koVBgAv8vNhMHpJsLQFToSyzRvvLLa9RUu8-1T1vTQCV1mFaKVLbDOBIfC

posted on 2015-05-28 20:37  tuo_chao  阅读(265)  评论(0编辑  收藏  举报

导航