Python中的測試模塊--unittest與unittest.main()函數

問題背景

  在閱讀源碼的時候遇到了unittest.main()函數,網上查到用法,做些記錄

  需要導入其模塊

import unittest # 導入unittest模塊

  unittest模塊用於單元測試.其中測試類需要傳入參數unittest.TestCase

class UserCase(unittest.TestCase): # 需要傳入unittest.TestCase
    pass

  並且類中的函數需要以test開頭,方可執行.

class UserCase(unittest.TestCase):

    def testAddUser(self): #以test開頭
        print("add a user")

  最后是調用:unittest.main()

if __name__ == '__main__':
    unittest.main()

完整測試代碼

import unittest

class UserCase(unittest.TestCase):

    def testAddUser(self):
        print("add a user")

    def testDelUser(self):
        print("delete a user")

class haha(unittest.TestCase):
    def test_haha(self):
        print('我是一個測試')
    def a(self): # 這一個函數是不會執行的
        print('我不是一個測試')

if __name__ == '__main__':
    unittest.main()

參考

• unittest之入口main()