如何在Python中使用Unittest自動化單元測試框架

這篇文章將為大家詳細講解有關如何在Python中使用Unittest自動化單元測試框架，文章內容質量較高，因此小編分享給大家做個參考，希望大家閱讀完這篇文章后對相關知識有一定的了解。

首先確定已經安裝有Python，之后通過安裝PyUnit，Python版本比較新的已經集成有PyUnit(PyUnit 提供了一個圖形測試界面UnittestGUI.py) 

參考：查看地址

3、代碼實例

使用的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())

關于如何在Python中使用Unittest自動化單元測試框架就分享到這里了，希望以上內容可以對大家有一定的幫助，可以學到更多知識。如果覺得文章不錯，可以把它分享出去讓更多的人看到。