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

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

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

參考：查看地址

3、代碼實(shí)例

使用的IDE為 PyCharm,DEMO結(jié)構(gòu)如圖

1.簡單地一個(gè)實(shí)例

# Test002_Fail.py


# -*- coding:utf-8 -*-

import unittest

class Test002_Fail(unittest.TestCase):

  #測試用例前執(zhí)行
  def setUp(self):
    print 'Case Before'
    pass

  #測試用例后執(zhí)行
  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.當(dāng)case比較多的時(shí)候，可以使用Suite來管理多個(gè)case,suite 可以看做是一個(gè)case 的容器，可以裝很多的case

# SuiteDemo.py
# -*- coding:utf-8 -*-
#一個(gè)類，添加測試集
import unittest

class SuiteDemo(unittest.TestCase):


  #測試用例前執(zhí)行
  def setUp(self):
    print 'Case Before'
    pass

  #測試用例后執(zhí)行
  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'

#定義一個(gè)測試集合，方便添加Case

def suite():
  suiteTest = unittest.TestSuite()
  suiteTest.addTest(SuiteDemo("test_Case1"))
  suiteTest.addTest(SuiteDemo("test_Case2"))
  return suiteTest


#默認(rèn)運(yùn)行時(shí)通過 Suite 運(yùn)行

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

3.Suite 和 Case可以分開寫在不同的Python文件中，這樣方便區(qū)分case模塊和Suite管理模塊，在測試測試的時(shí)候只需要在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'

接下來為三個(gè)Suite模塊，都是先導(dǎo)入Case模塊，之后添加Suite,再在Suite里面添加Case，最后執(zhí)行的時(shí)候需要使用TestRunner,這個(gè)TestRunner 相當(dāng)于Suite 的執(zhí)行器

簡單地TestSuite:

# TestCase_Demo1.py

# -*- coding:utf-8 -*-

#集合寫入一個(gè)方法，main調(diào)用并啟動


import unittest
from TestCase_Demo import TestCaseDemo


#添加一個(gè)測試集合，并添加Case

def suite():
  suiteTest = unittest.TestSuite()
  suiteTest.addTest(TestCaseDemo('test_Case1'))
  return suiteTest


#指定并啟動測試集合,運(yùn)行集合方法

if __name__ == '__main__':

  runner = unittest.TextTestRunner()
  runner.run(suite())

也可以不定義Suite方法，直接在main方法中進(jìn)行Case的添加，之后執(zhí)行

# 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分組，用于區(qū)分不同的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 運(yùn)行'
  return suiteTest

def Suite2():
  suiteTest = unittest.TestSuite()
  suiteTest.addTest(TestCaseDemo("test_Case3"))
  suiteTest.addTest(TestCaseDemo("test_Case4"))
  print 'Suite2 運(yùn)行'
  return suiteTest

#包含所有的Suite

def AllSuite():
  allTest = unittest.TestSuite((Suite1(),Suite2()))
  return allTest

#運(yùn)行的時(shí)候，可以根據(jù)不同的要求，運(yùn)行不同的Suite,或者全部運(yùn)行，這樣就方便管理每次運(yùn)行的case

if __name__ == '__main__':
  runner = unittest.TextTestRunner()
  runner.run(AllSuite())

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