首页 > 编程语言 > 详细

python单元测试框架

时间:2015-05-28 21:21:54      阅读:303      评论:0      收藏:0      [点我收藏+]

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

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!