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單元測試架構