1. 程式人生 > >Python Unittest 自動化單元測試框架Demo

Python Unittest 自動化單元測試框架Demo

  1. 環境準備
    首先確定已經安裝有Python,之後通過安裝PyUnit,Python版本比較新的已經整合有PyUnit(PyUnit 提供了一個圖形測試介面UnittestGUI.py)
    參考: https://wiki.python.org/moin/PyUnit

  2. 程式碼例項
    使用的IDE為 PyCharm,DEMO結構如圖
    這裡寫圖片描述


    1.簡單地一個例項
    
    # Test002_Fail.py
    
    
    # -*- coding:utf-8 -*-
    
    import  unittest
    
    class Test002_Fail(unittest.TestCase):
    
        #測試用例前執行
        def
    setUp(self):
    print 'Case Before' pass #測試用例後執行 def tearDown(self): print 'Case After' pass #測試用例1 def test_Case1(self): a = 3 b = 2 self.assertEqual(a+b,4,'Result Fail') #測試用例2 def test_Case2(self): a = 2 b = 3
    self.assertEqual(a*b,7,'Result Fail') if __name__ == '__main__': unittest.main()

    2.當case比較多的時候,可以使用Suite來管理多個case,suite 可以看做是一個case 的容器,可以裝很多的case
    
    # SuiteDemo.py
    
    
    # -*- coding:utf-8 -*-
    
    
    #一個類,新增測試集
    
    
    import  unittest
    
    class SuiteDemo(unittest.TestCase):
    
    
        #測試用例前執行
        def setUp
    (self):
    print 'Case Before' pass #測試用例後執行 def tearDown(self): print 'Case After' pass def test_Case1(self): a = 3 b = 2 self.assertEqual(a+b,5,'Result Fail') print 'Case1' def test_Case2(self): a = 2 b = 3 self.assertEqual(a*b,6,'Result Fail') print 'Case2' #定義一個測試集合,方便新增Case def suite(): suiteTest = unittest.TestSuite() suiteTest.addTest(SuiteDemo("test_Case1")) suiteTest.addTest(SuiteDemo("test_Case2")) return suiteTest #預設執行時通過 Suite 執行 if __name__ == '__main__': unittest.main(defaultTest='suite')

    3.Suite 和 Case可以分開寫在不同的Python檔案中,這樣方便區分case模組和Suite管理模組,在測試測試的時候只需要在case模組裡面新增case,之後新增到Suite模組即可了

    case模組

    
    # TestCaseDemo
    
    
    # -*- coding:utf-8 -*-
    
    
    import unittest
    
    
    class TestCaseDemo(unittest.TestCase):
        def setUp(self):
            pass
    
        def tearDown(self):
            pass
    
        def test_Case1(self):
            print 'test_Case1'
    
    
        def test_Case2(self):
            print 'test_Case2'
    
    
        def test_Case3(self):
            print 'test_Case3'
    
    
        def test_Case4(self):
            print 'test_Case4'
    

    接下來為三個Suite模組,都是先匯入Case模組,之後新增Suite,再在Suite裡面新增Case,最後執行的時候需要使用TestRunner,這個TestRunner 相當於Suite 的執行器

    簡單地TestSuite:

    
    # TestCase_Demo1.py
    
    
    # -*- coding:utf-8 -*-
    
    
    #集合寫入一個方法,main呼叫並啟動
    
    
    
    import unittest
    from TestCase_Demo import TestCaseDemo
    
    
    #新增一個測試集合,並新增Case
    
    def suite():
        suiteTest = unittest.TestSuite()
        suiteTest.addTest(TestCaseDemo('test_Case1'))
        return suiteTest
    
    
    
    #指定並啟動測試集合,執行集合方法
    
    if __name__ == '__main__':
    
        runner = unittest.TextTestRunner()
        runner.run(suite())
    

    也可以不定義Suite方法,直接在main方法中進行Case的新增,之後執行

    
    # TestCase_Demo2.py
    
    
    # -*- coding:utf-8 -*-
    
    
    #集合可以寫在main
    
    
    import unittest
    from TestCase_Demo import TestCaseDemo
    
    
    #指定並啟動測試集合
    
    if __name__ == '__main__':
    
        #新增測試集合Case,並啟動
        suiteTest = unittest.TestSuite()
        suiteTest.addTest(TestCaseDemo('test_Case1'))
        suiteTest.addTest(TestCaseDemo('test_Case2'))
    
        #直接啟動集合
        runner = unittest.TextTestRunner()
        runner.run(suiteTest)

    可以把不同的Case 新增道不同的Suite裡面,同樣,可以給Suite分組,用於區分不同的Case,並且,Suite 之間還可以包含

    
    # TestCase_Demo3.py
    
    
    # -*- coding:utf-8 -*-
    
    
    #包含集合,多集合
    
    
    
    import unittest
    from TestCase_Demo import TestCaseDemo
    
    
    
    #新增不同的合集
    
    def Suite1():
        suiteTest = unittest.TestSuite()
        suiteTest.addTest(TestCaseDemo("test_Case1"))
        suiteTest.addTest(TestCaseDemo("test_Case2"))
        print 'Suite1 執行'
        return suiteTest
    
    
    def Suite2():
        suiteTest = unittest.TestSuite()
        suiteTest.addTest(TestCaseDemo("test_Case3"))
        suiteTest.addTest(TestCaseDemo("test_Case4"))
        print 'Suite2 執行'
        return suiteTest
    
    
    
    #包含所有的Suite
    
    def AllSuite():
        allTest = unittest.TestSuite((Suite1(),Suite2()))
        return allTest
    
    
    
    #執行的時候,可以根據不同的要求,執行不同的Suite,或者全部執行,這樣就方便管理每次執行的case
    
    if __name__ == '__main__':
        runner = unittest.TextTestRunner()
        runner.run(AllSuite())
    
  3. 小結
    PyUnit 是Python的一個單元測試框架,使用它可以方便的管理自動化執行的case.
    在Python 中,每一個方法 只要方法名使用test_ 開頭的都被作為是一個用例,都可以看做成一個case,這裡介紹了使用Suite管理Case,並且分組執行Case,還有其他很多的功能,請參考:
    https://docs.python.org/2.7/library/unittest.html