python單元測(cè)試怎么寫？怎么使用unittest進(jìn)行單元測(cè)試

與java不同，python學(xué)習(xí)到單元測(cè)試這個(gè)概念更多的情況下是在學(xué)習(xí)自動(dòng)化測(cè)試框架的時(shí)候接觸到的，而且使用的單元測(cè)試框架也五花八門。但其中最有名的可能就是unittest了。如果說(shuō)unittest的名號(hào)還不夠響亮的話，你可以看看他的曾用名——PyUnit，沒(méi)錯(cuò)，他也是xUnit家族的一員，他家最出名的Junit在業(yè)界內(nèi)聲明顯赫，現(xiàn)在你知道unittest有多厲害了吧？今天我們就來(lái)簡(jiǎn)單的了解一下unittest的使用，也借此機(jī)會(huì)學(xué)習(xí)一下unittest單元測(cè)試怎么寫吧！

unittest引入

unittest框架現(xiàn)在已經(jīng)被python官方“收編”，unittest作為python標(biāo)準(zhǔn)包支持的庫(kù)，在使用時(shí)只需要直接引入即可，不需要使用?pip?進(jìn)行安裝。

如何寫單元測(cè)試代碼

單元測(cè)試代碼可以簡(jiǎn)單的分成三步，分別為?Given?，?When?，?Then?。?Given?代表初始狀態(tài)或者前置條件（可以簡(jiǎn)單的理解為數(shù)據(jù)輸入）；?When?代表行為發(fā)生，也就是進(jìn)行測(cè)試動(dòng)作；Then?則是斷言結(jié)果。

所以大部分情況下我們可以把單元測(cè)試的代碼分為三塊：

1. 第一塊代碼負(fù)責(zé)測(cè)試前的數(shù)據(jù)處理，比如輸入數(shù)據(jù)等。
2. 第二塊代碼負(fù)責(zé)調(diào)用相關(guān)的功能模塊進(jìn)行測(cè)試。
3. 第三塊代碼負(fù)責(zé)將測(cè)試的結(jié)果與我們期待的結(jié)果進(jìn)行比對(duì)，從而得出測(cè)試結(jié)果（通過(guò)或者失敗）。

代碼實(shí)戰(zhàn)

以一個(gè)簡(jiǎn)單的計(jì)算功能為例，以下是這個(gè)計(jì)算模塊的方法（文件名為calculator.py）：

def add(a, b):
    return a+b


def substract(a, b):
    return a-b

我們?cè)O(shè)計(jì)了一些簡(jiǎn)單的測(cè)試代碼（代碼解讀詳見(jiàn)注釋）：

from calculator import *  # 導(dǎo)入目標(biāo)模塊
import unittest  # 導(dǎo)入unittest包


class TestCount(unittest.TestCase):
    #  測(cè)試代碼寫在一個(gè)測(cè)試類里面

    def test_add(self):
        # 我們要測(cè)的是一個(gè)add函數(shù)，他傳入兩個(gè)參數(shù)
        x = 5
        y = 6
        # 輸入測(cè)試數(shù)據(jù)
        res = add(x, y)  # 將數(shù)據(jù)傳遞給被測(cè)函數(shù)，然后將結(jié)果存在res中
        self.assertEqual(res, 11)  # 使用斷言判斷res的值是否等于期待的值（這里期待為11）
        # 如果res的值等于期待的值，則測(cè)試通過(guò)，反之測(cè)試失敗

    def test_substract(self):
        # 這是另一個(gè)測(cè)試代碼，使用簡(jiǎn)寫的方式將第一二步合在一起
        res = substract(11, 6)
        self.assertEqual(res, 5)

 以上代碼只是用來(lái)介紹如何使用unittest，測(cè)試用例設(shè)計(jì)得比較隨意。實(shí)際上測(cè)試用例設(shè)計(jì)需要考慮很多，要通過(guò)測(cè)試用例來(lái)證明目標(biāo)代碼的功能健全，所以對(duì)于一個(gè)簡(jiǎn)單的相加函數(shù)而言，這樣的測(cè)試用例略顯簡(jiǎn)單了。詳細(xì)的測(cè)試用例設(shè)計(jì)小編以后會(huì)繼續(xù)分享，關(guān)注W3C技術(shù)頭條，學(xué)習(xí)如何編寫更好的測(cè)試用例！

python作為一種代碼易于理解的語(yǔ)言，實(shí)際上單元測(cè)試只在比較正規(guī)和大型的項(xiàng)目上會(huì)比較常見(jiàn)，unittest的更常見(jiàn)用法是與selenium集成為一個(gè)測(cè)試框架，由unittest提供比較好的用例管理和斷言，由selenium進(jìn)行功能測(cè)試。這時(shí)候unittest已經(jīng)不算是單元測(cè)試工具了（是用例管理工具）。

小結(jié)

對(duì)于初學(xué)者而言，以上簡(jiǎn)單的測(cè)試案例已經(jīng)足夠入門了。但實(shí)際上在測(cè)試的過(guò)程中還會(huì)遇到很多對(duì)功能的要求，比如說(shuō)功能代碼的執(zhí)行速度是否符合要求；如何聚合測(cè)試；代碼拋出異常如何跳過(guò)等等，這些都是unittest的學(xué)習(xí)內(nèi)容，而對(duì)于完整的unittest學(xué)習(xí)，一篇文章是講不完的，所以小編后續(xù)也會(huì)根據(jù)unittest的部分功能進(jìn)行逐一介紹。以上就是python單元測(cè)試怎么寫的全部?jī)?nèi)容了，各位讀者是否滿意呢？