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)

python单元测试框架

标签：

原文地址：http://www.cnblogs.com/tuochao/p/4536867.html