Python Unittest原理及基本使用方法

更新時間：2020年11月06日 09:23:20 作者：南風丶輕語

這篇文章主要介紹了Python Unittest原理及基本使用方法,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下

基本使用步驟：

1、import unittest

2、新建一個Class繼承unittest.TestCase

3、定義一個以test開頭的函數(shù)作為測試用例【必須以test開頭】

4、執(zhí)行測試用例前都會先調(diào)用Setup函數(shù)【每個測試用例調(diào)用一次】或者setupClass函數(shù)【所有測試用例僅調(diào)用一次】

5、執(zhí)行完測試用例后都會調(diào)用tearDown函數(shù)【多次】和tearDownClass函數(shù)【僅一次】

PS：

1、不能重寫init()

2、函數(shù)必須以test開頭才能作為測試用例

3、setupClass()和tearDownClass()必須用@classmethod修飾

# -*- encoding=utf-8 -*-

import unittest
class Demo1(unittest.TestCase):
  def setUp(self):
    print '調(diào)用 setup 函數(shù)'

  @classmethod
  def setUpClass(cls):
    print '調(diào)用 setupClass 函數(shù)'

  def tearDown(self):
    print '調(diào)用 tearDown 函數(shù)'
  @classmethod
  def tearDownClass(cls):
    print '調(diào)用 tearDownClass 函數(shù)'

  @staticmethod
  def test_add():
    print '3 + 5 = {}'.format(3 + 5)

  @staticmethod
  def test_sub():
    print '3 - 5 = {}'.format(3 - 5)


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

運行結(jié)果

調(diào)用 setupClass 函數(shù)

調(diào)用 setup 函數(shù)

3 + 5 = 8

調(diào)用 tearDown 函數(shù)

調(diào)用 setup 函數(shù)

3 - 5 = -2

調(diào)用 tearDown 函數(shù)

調(diào)用 tearDownClass 函數(shù)

Ran 2 tests in 0.012s

OK

TestSuite【測試組】添加測試用例進行測試

# -*- encoding=utf-8 -*-


class Class1(object):
  def __init__(self):
    print '調(diào)用 Class1 初始化函數(shù)'
    self.name = '未定義姓名'
    self.age = '未定義年齡'

  def set_name(self, name):
    self.name = name

  def set_age(self, age):
    self.age = age

  def print_name(self):
    print 'Class1 的姓名是：{}'.format(self.name)

  def print_age(self):
    print 'Class1 的年齡是:{}'.format(self.age)


if __name__ == '__main__':
  pass

 

# -*- encoding=utf-8 -*-

import unittest

from Class1 import Class1


class CallClass1(unittest.TestCase):

  def setUp(self):
    self.class1 = Class1()
    pass

  def tearDown(self):
    self.class1 = None
    pass

  def test_set_name(self):
    self.class1.set_name('設置了名字')
    self.class1.print_name()

  def test_print_name(self):
    self.class1.print_name()


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

 

# -*- encoding=utf-8 -*-

import unittest
from Test_Demo2 import CallClass1

if __name__ == '__main__':
  testSuite = unittest.TestSuite()
  testSuite.addTest(CallClass1('test_set_name'))

  testSuite.addTests(
    [CallClass1('test_set_name'), CallClass1('test_print_name')])

  textTestRunner = unittest.TextTestRunner()
  textTestRunner.run(testSuite)

運行結(jié)果

調(diào)用 Class1 初始化函數(shù)

Class1 的姓名是：設置了名字

調(diào)用 Class1 初始化函數(shù)

Class1 的姓名是：設置了名字

調(diào)用 Class1 初始化函數(shù)

Class1 的姓名是：未定義姓名

...

----------------------------------------------------------------------

Ran 3 tests in 0.000s

OK

Process finished with exit code 0

記錄測試報告到txt文件

# -*- encoding=utf-8 -*-

import unittest
from Test_Demo2 import CallClass1

if __name__ == '__main__':
  testSuite = unittest.TestSuite()
  testSuite.addTest(CallClass1('test_set_name'))

  testSuite.addTests(
    [CallClass1('test_set_name'), CallClass1('test_print_name')])

  # textTestRunner = unittest.TextTestRunner()
  # textTestRunner.run(testSuite)

  # 將測試結(jié)果輸出到測試報告中
  with open('UnittestTextReport.txt', 'w') as f:
    runner = unittest.TextTestRunner(stream=f, verbosity=2)
    runner.run(testSuite)

記錄測試報告到html文件

# 將測試結(jié)果輸出到測試報告html中
with open('HTMLReport.html', 'w') as f:
  runner = HTMLTestRunner(stream=f,
              title='MathFunc Test Report',
              description='generated by HTMLTestRunner.',
              verbosity=2
              )
  runner.run(suite)

斷言

當斷言為真時，會執(zhí)行斷言之后的代碼如（在斷言之后打印成功提示信息）

若斷言為假，則之后的代碼不會被執(zhí)行，打應錯誤信息，之后提示信息不會被執(zhí)行；unittest中提示F，失敗，若捕獲到異常，則為E，代表錯誤；

補充：若斷言為假，則當前函數(shù)def，停止運行，但會運行之后的def函數(shù)

重點：try語句塊中不要使用assert一起使用，assert斷言為假時，會拋出異常，測試用例為失敗，如果放到try中，則會捕獲到異常，測試用例永遠標記為通過

斷言結(jié)果：

. 表示通過或者 pass

F False, 表示斷言沒有通過

E Error, 表示程序內(nèi)部發(fā)生了錯誤。

查找當前目錄下所有的測試用例【默認以test開頭的py文件，可以自己配置】

# -*- encoding=utf-8 -*-

import unittest
import os

if __name__ == '__main__':
  loader = unittest.defaultTestLoader.discover(os.getcwd())
  # tests = unittest.defaultTestLoader.discover(
  #   start_dir='.', # => 指定查找的目錄
  #   pattern='test*.py' # => 指定查找規(guī)則
  # )
  runner = unittest.TextTestRunner()
  runner.run(loader)

以上就是本文的全部內(nèi)容，希望對大家的學習有所幫助，也希望大家多多支持腳本之家。

您可能感興趣的文章: