詳解Python之unittest單元測(cè)試代碼

更新時(shí)間：2018年01月24日 14:06:34 作者：JeremyWYL

本篇文件主要介紹了詳解Python之unittest測(cè)試代碼，小編覺得挺不錯(cuò)的，現(xiàn)在分享給大家，也給大家做個(gè)參考。一起跟隨小編過來看看吧

前言

編寫函數(shù)或者類時(shí)，還可以為其編寫測(cè)試。通過測(cè)試，可確定代碼面對(duì)各種輸入都能夠按要求的那樣工作。

本次我將介紹如何使用Python模塊unittest中的工具來測(cè)試代碼。

測(cè)試函數(shù)

首先我們先編寫一個(gè)簡單的函數(shù)，它接受姓、名、和中間名三個(gè)參數(shù)，并返回完整的姓名：

names.py

def get_fullname(firstname,lastname,middel=''):
  '''創(chuàng)建全名'''
  if middel:
    full_name = firstname + ' ' + middel + ' ' + lastname
    return full_name.title()
  else:
    full_name = firstname + ' ' + lastname
    return full_name.title()

然后再當(dāng)前目錄下編寫調(diào)用函數(shù)程序

get_name.py

from names import get_fullname

message = "Please input 'q' to quit."
print(message)

while True:
  first = input("Please input your firstname: ")
  if first == 'q':
    break
  last = input("Please input your lastname: ")
  if last == 'q':
    break
  middels = input("Please input your middel name or None: ")
  if last == 'q':
    break
  formant_name = get_fullname(first,last,middels)
  print("\tYour are fullname is: " + formant_name.title())

調(diào)用結(jié)果：

Please input 'q' to quit.
Please input your firstname: xiao
Please input your lastname: peng
Please input your middel or None:
Your are fullname is: Xiao Peng
Please input your firstname: xiao
Please input your lastname: peng
Please input your middel or None: you
Your are fullname is: Xiao You Peng
Please input your firstname: q
進(jìn)程已結(jié)束,退出代碼0

創(chuàng)建測(cè)試程序

創(chuàng)建測(cè)試用例的語法需要一段時(shí)間才能習(xí)慣，但測(cè)試用例創(chuàng)建后，再針對(duì)函數(shù)的單元測(cè)試就很簡單了。先導(dǎo)入模塊unittest以及要測(cè)試的函數(shù)，再創(chuàng)建一個(gè)繼承函數(shù)unittest.TestCase的類，

并編寫一系列方法對(duì)函數(shù)行為的不同方便進(jìn)行測(cè)試。

下面介紹測(cè)試上面names.py函數(shù)是否能夠正確的獲取姓名：

Test_get_name.py

import unittest
from names import get_fullname

class NamesTestCase(unittest.TestCase):
  '''定義測(cè)試類'''
  def test_get_name2(self):
    '''測(cè)試2個(gè)字的名字'''
    formatied_name2 = get_fullname('xiao','pengyou')
    self.assertEqual(formatied_name2,'Xiao Pengyou')
  def test_get_name3(self):
    '''測(cè)試3個(gè)字的名字'''
    formatied_name3 = get_fullname('xiao','peng',middel='you')
    self.assertEqual(formatied_name3,'Xiao Peng You')
if __name__ == '__init__':
  unittest.main()

測(cè)試結(jié)果：

Ran 2 tests in 0.034s

OK

兩個(gè)測(cè)試單元測(cè)試通過測(cè)試！

在當(dāng)前的大目錄下會(huì)生成一個(gè)測(cè)試報(bào)告，可以通過瀏覽器進(jìn)行打開查看。

由圖可知，兩個(gè)測(cè)試通過，并顯示測(cè)試的時(shí)間！??！

unittest.TestCase的各種斷言方法

unittest各種斷言方法

方法	用途
assertEqual(a,b)	核實(shí)a == b
assertNotEqual(a,b)	核實(shí)a != b
assertTrue(x)	核實(shí)x為True
assertFalse(x)	核實(shí)x為False
assertIn(item,list)	核實(shí)item在list中
assertNotIn(item,list)	核實(shí)item不在list中