與java不同，python學(xué)習(xí)到單元測試這個(gè)概念更多的情況下是在學(xué)習(xí)自動(dòng)化測試框架的時(shí)候接觸到的，而且使用的單元測試框架也五花八門。但其中最有名的可能就是unittest了。如果說unittest的名號(hào)還不夠響亮的話，你可以看看他的曾用名——PyUnit，沒錯(cuò)，他也是xUnit家族的一員，他家最出名的Junit在業(yè)界內(nèi)聲明顯赫，現(xiàn)在你知道unittest有多厲害了吧？今天我們就來簡單的了解一下unittest的使用，也借此機(jī)會(huì)學(xué)習(xí)一下unittest單元測試怎么寫吧！
unittest引入
unittest框架現(xiàn)在已經(jīng)被python官方“收編”，unittest作為python標(biāo)準(zhǔn)包支持的庫，在使用時(shí)只需要直接引入即可，不需要使用?pip?進(jìn)行安裝。
如何寫單元測試代碼
單元測試代碼可以簡單的分成三步，分別為?Given?，?When?，?Then?。?Given?代表初始狀態(tài)或者前置條件（可以簡單的理解為數(shù)據(jù)輸入）；?When?代表行為發(fā)生，也就是進(jìn)行測試動(dòng)作；Then?則是斷言結(jié)果。
所以大部分情況下我們可以把單元測試的代碼分為三塊：
- 第一塊代碼負(fù)責(zé)測試前的數(shù)據(jù)處理，比如輸入數(shù)據(jù)等。
- 第二塊代碼負(fù)責(zé)調(diào)用相關(guān)的功能模塊進(jìn)行測試。
- 第三塊代碼負(fù)責(zé)將測試的結(jié)果與我們期待的結(jié)果進(jìn)行比對(duì)，從而得出測試結(jié)果（通過或者失敗）。
代碼實(shí)戰(zhàn)
以一個(gè)簡單的計(jì)算功能為例，以下是這個(gè)計(jì)算模塊的方法（文件名為calculator.py）：
def add(a, b):
return a+b
def substract(a, b):
return a-b
我們?cè)O(shè)計(jì)了一些簡單的測試代碼（代碼解讀詳見注釋）：
from calculator import * # 導(dǎo)入目標(biāo)模塊
import unittest # 導(dǎo)入unittest包
class TestCount(unittest.TestCase):
# 測試代碼寫在一個(gè)測試類里面
def test_add(self):
# 我們要測的是一個(gè)add函數(shù)，他傳入兩個(gè)參數(shù)
x = 5
y = 6
# 輸入測試數(shù)據(jù)
res = add(x, y) # 將數(shù)據(jù)傳遞給被測函數(shù)，然后將結(jié)果存在res中
self.assertEqual(res, 11) # 使用斷言判斷res的值是否等于期待的值（這里期待為11）
# 如果res的值等于期待的值，則測試通過，反之測試失敗
def test_substract(self):
# 這是另一個(gè)測試代碼，使用簡寫的方式將第一二步合在一起
res = substract(11, 6)
self.assertEqual(res, 5)
以上代碼只是用來介紹如何使用unittest，測試用例設(shè)計(jì)得比較隨意。實(shí)際上測試用例設(shè)計(jì)需要考慮很多，要通過測試用例來證明目標(biāo)代碼的功能健全，所以對(duì)于一個(gè)簡單的相加函數(shù)而言，這樣的測試用例略顯簡單了。詳細(xì)的測試用例設(shè)計(jì)小編以后會(huì)繼續(xù)分享，關(guān)注W3C技術(shù)頭條，學(xué)習(xí)如何編寫更好的測試用例！
python作為一種代碼易于理解的語言，實(shí)際上單元測試只在比較正規(guī)和大型的項(xiàng)目上會(huì)比較常見，unittest的更常見用法是與selenium集成為一個(gè)測試框架，由unittest提供比較好的用例管理和斷言，由selenium進(jìn)行功能測試。這時(shí)候unittest已經(jīng)不算是單元測試工具了（是用例管理工具）。
小結(jié)
對(duì)于初學(xué)者而言，以上簡單的測試案例已經(jīng)足夠入門了。但實(shí)際上在測試的過程中還會(huì)遇到很多對(duì)功能的要求，比如說功能代碼的執(zhí)行速度是否符合要求；如何聚合測試；代碼拋出異常如何跳過等等，這些都是unittest的學(xué)習(xí)內(nèi)容，而對(duì)于完整的unittest學(xué)習(xí)，一篇文章是講不完的，所以小編后續(xù)也會(huì)根據(jù)unittest的部分功能進(jìn)行逐一介紹。以上就是python單元測試怎么寫的全部內(nèi)容了，各位讀者是否滿意呢？
