python單元測試如何寫

今天小編給大家分享一下python單元測試如何寫的相關知識點，內容詳細，邏輯清晰，相信大部分人都還太了解這方面的知識，所以分享這篇文章給大家參考一下，希望大家閱讀完這篇文章后有所收獲，下面我們一起來了解一下吧。

unittest引入

unittest框架現在已經被python官方“收編”，unittest作為python標準包支持的庫，在使用時只需要直接引入即可，不需要使用pip進行安裝。

如何寫單元測試代碼

單元測試代碼可以簡單的分成三步，分別為Given，When，Then。Given代表初始狀態或者前置條件（可以簡單的理解為數據輸入）；When代表行為發生，也就是進行測試動作；Then則是斷言結果。

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

1. 第一塊代碼負責測試前的數據處理，比如輸入數據等。

2. 第二塊代碼負責調用相關的功能模塊進行測試。

3. 第三塊代碼負責將測試的結果與我們期待的結果進行比對，從而得出測試結果（通過或者失敗）。

代碼實戰

以一個簡單的計算功能為例，以下是這個計算模塊的方法（文件名為calculator.py）：

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


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

我們設計了一些簡單的測試代碼（代碼解讀詳見注釋）：

from calculator import *  # 導入目標模塊
import unittest  # 導入unittest包


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

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

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

 以上代碼只是用來介紹如何使用unittest，測試用例設計得比較隨意。實際上測試用例設計需要考慮很多，要通過測試用例來證明目標代碼的功能健全，所以對于一個簡單的相加函數而言，這樣的測試用例略顯簡單了。詳細的測試用例設計小編以后會繼續分享，關注W3C技術頭條，學習如何編寫更好的測試用例！

python作為一種代碼易于理解的語言，實際上單元測試只在比較正規和大型的項目上會比較常見，unittest的更常見用法是與selenium集成為一個測試框架，由unittest提供比較好的用例管理和斷言，由selenium進行功能測試。這時候unittest已經不算是單元測試工具了（是用例管理工具）。

以上就是“python單元測試如何寫”這篇文章的所有內容，感謝各位的閱讀！相信大家閱讀完這篇文章都有很大的收獲，小編每天都會為大家更新不同的知識，如果還想學習更多的知識，請關注億速云行業資訊頻道。