在python中使用單元測試unittest方式

更新時間：2026年01月23日 08:47:55 作者：babybin

單元測試是保障代碼質(zhì)量和功能的重要工具,通過它們可以驗證代碼的正確性和可靠性,本篇文章介紹了如何使用Python的unittest模塊進行單元測試,并通過實例展示了如何編寫、運行和擴展測試

為什么單元測試如此重要？

在我們開始冒險之前，先來思考一下為什么單元測試如此重要。

嗯，首先，它們讓我們的代碼更可靠，保證了我們的函數(shù)和方法按照預期運行。而且，單元測試還是代碼質(zhì)量的守護者，一旦引入新的功能或者修復一個Bug，只需運行測試，就能夠確保不會對現(xiàn)有的功能產(chǎn)生負面影響。

第一步：準備好你的工具

在我們的冒險開始之前，確保你已經(jīng)準備好了最重要的武器：Python和unittest模塊。

當然，你也可以選擇其他測試框架，但unittest是Python內(nèi)建的，簡單而有效。

# unitest_example1.py

import unittest

class MyTestCase(unittest.TestCase):
    def test_hello_world(self):
        result = "Hello, World!"
        self.assertEqual(result, "Hello, World!")

第二步：讓測試跑起來！

現(xiàn)在，讓我們看看這個測試是如何運作的。

簡單的測試函數(shù)，通過 assertEqual 斷言我們的結(jié)果是否和期望一致。運行測試：

python -m unittest unittest_example1.py

第三步：享受成功的喜悅

你聽到了嗎？那是代碼在默默地說：“我沒有Bug！” 這種感覺，就像是踏入夏日冰淇淋店的涼意，清新又令人愉悅。

第四步：擴展你的冒險

單元測試的美妙之處在于，一旦你開始，你就會迷上它。不斷擴展你的測試套件，覆蓋更多的代碼路徑。嘗試參數(shù)化測試，用更少的代碼覆蓋更多的場景。

注意：parameterized.expand 并不是 unittest 模塊中的一部分，它實際上來自于 parameterized 模塊，該模塊并不是 Python 標準庫的一部分。

為了使用 parameterized.expand，您需要安裝 parameterized 模塊。

pip install parameterized

#!/usr/bin/env python
# coding=utf-8
# @Time    : 2024/1/31 15:40

# unittest_examples2.py

import unittest

from parameterized import parameterized

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

class AdvancedTestCase(unittest.TestCase):
    @parameterized.expand([
        ("case1", 1, 2, 3),
        ("case2", -1, 2, 1),
        ("case3", 0, 0, 0),
        ("case4", 0, 0, 2),
    ])
    def test_addition(self, name, a, b, expected):
        result = add_numbers(a, b)
        self.assertEqual(result, expected)


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

# python -m unittest unittest_examples2.py
# python3 -m unittest unittest_examples2.py

我特意加了一個錯誤的case4，運行代碼：

python3 -m unittest unitest_example2.py

結(jié)語：單元測試是一場愉快的冒險！

在我們的技術(shù)冒險中，我們發(fā)現(xiàn)了單元測試的美妙之處。它不僅保護了我們的代碼，還讓我們充滿信心地迎接新的挑戰(zhàn)。

附錄

完整代碼：

#!/usr/bin/env python
# unitest_example1.py

import unittest


class MyTestCase(unittest.TestCase):
    def setUp(self):
        # 在測試前的準備工作
        self.data = [1, 2, 3]

    def test_something(self):
        # 測試代碼
        result = 1 + 1

        # 使用 assert 斷言檢查結(jié)果是否等于 2
        self.assertEqual(result, 2)

        # 使用 assert 斷言檢查列表中是否包含特定元素
        self.assertIn(2, self.data)

        # 使用 assert 斷言檢查條件是否為真
        self.assertTrue(result == 2)

        # 使用 assert 斷言檢查條件是否為假
        self.assertFalse(result != 2)

        # 使用 assert 斷言檢查兩個值是否不相等
        self.assertNotEqual(result, 3)

        # 使用 assert 斷言檢查某個值是否為 None
        self.assertIsNone(None)

        # 使用 assert 斷言檢查某個值是否不為 None
        self.assertIsNotNone(result)

        # 使用 assert 斷言檢查字符串是否以特定前綴開頭
        self.assertTrue("hello".startswith("he"))

    def tearDown(self):
        # 在測試后的清理工作
        del self.data


def run():
    loader = unittest.TestLoader()
    suite = loader.loadTestsFromTestCase(MyTestCase)


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

以上為個人經(jīng)驗，希望能給大家一個參考，也希望大家多多支持腳本之家。

您可能感興趣的文章: