在python中使用单元测试unittest方式

 更新时间:2026年01月23日 08:47:55   作者:babybin  
单元测试是保障代码质量和功能的重要工具,通过它们可以验证代码的正确性和可靠性,本篇文章介绍了如何使用Python的unittest模块进行单元测试,并通过实例展示了如何编写、运行和扩展测试

为什么单元测试如此重要?

在我们开始冒险之前,先来思考一下为什么单元测试如此重要。

嗯,首先,它们让我们的代码更可靠,保证了我们的函数和方法按照预期运行。而且,单元测试还是代码质量的守护者,一旦引入新的功能或者修复一个Bug,只需运行测试,就能够确保不会对现有的功能产生负面影响。

第一步:准备好你的工具

在我们的冒险开始之前,确保你已经准备好了最重要的武器:Python和unittest模块。

当然,你也可以选择其他测试框架,但unittest是Python内建的,简单而有效。

# unitest_example1.py

import unittest

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

第二步:让测试跑起来!

现在,让我们看看这个测试是如何运作的。

简单的测试函数,通过 assertEqual 断言我们的结果是否和期望一致。运行测试:

python -m unittest unittest_example1.py

第三步:享受成功的喜悦

你听到了吗?那是代码在默默地说:“我没有Bug!” 这种感觉,就像是踏入夏日冰淇淋店的凉意,清新又令人愉悦。

第四步:扩展你的冒险

单元测试的美妙之处在于,一旦你开始,你就会迷上它。不断扩展你的测试套件,覆盖更多的代码路径。尝试参数化测试,用更少的代码覆盖更多的场景。

注意: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

结语:单元测试是一场愉快的冒险!

在我们的技术冒险中,我们发现了单元测试的美妙之处。它不仅保护了我们的代码,还让我们充满信心地迎接新的挑战。

附录

完整代码:

#!/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 断言检查结果是否等于 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()

以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。

相关文章

  • pycharm导入第三方库的两种方法(永不报错)

    pycharm导入第三方库的两种方法(永不报错)

    这篇文章主要介绍了pycharm导入第三方库的两种方法(永不报错),本文通过图文并茂的形式给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
    2022-11-11
  • python学习之matplotlib绘制散点图实例

    python学习之matplotlib绘制散点图实例

    这篇文章主要介绍了python学习之matplotlib绘制散点图实例,具有一定借鉴价值,需要的朋友可以参考下。
    2017-12-12
  • pytorch实现线性拟合方式

    pytorch实现线性拟合方式

    今天小编就为大家分享一篇pytorch实现线性拟合方式,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2020-01-01
  • 用Python手把手教你实现2048小游戏

    用Python手把手教你实现2048小游戏

    感觉好久没和大家一起写小游戏玩了,今天恰巧有空.这次我们来用Python做个2048小游戏吧.废话不多说,文中有非常详细的代码示例,需要的朋友可以参考下
    2021-06-06
  • 浅谈python为什么不需要三目运算符和switch

    浅谈python为什么不需要三目运算符和switch

    下面小编就为大家带来一篇浅谈python为什么不需要三目运算符和switch。小编觉得挺不错的,现在就分享给大家,也给大家做个参考。一起跟随小编过来看看吧
    2016-06-06
  • Python while循环使用else语句代码实例

    Python while循环使用else语句代码实例

    这篇文章主要介绍了Python while循环使用else语句代码实例,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2020-02-02
  • Pandas实现复制dataframe中的每一行

    Pandas实现复制dataframe中的每一行

    这篇文章主要介绍了Pandas实现复制dataframe中的每一行方式,
    2024-02-02
  • Linux上安装Python的PIL和Pillow库处理图片的实例教程

    Linux上安装Python的PIL和Pillow库处理图片的实例教程

    这里我们来看一下在Linux上安装Python的PIL和Pillow库处理图片的实例教程,包括一个使用Pillow库实现批量转换图片的例子:
    2016-06-06
  • PyQt5信号与槽机制案例详解

    PyQt5信号与槽机制案例详解

    信号和槽是一种高级接口,应用于对象之间的通信,它是 QT 的核心特性,也是 QT 区别于其它工具包的重要地方,所有继承qwidget的控件都支持信号与槽机制,本文给大家介绍下PyQt5信号与槽机制的相关知识,感兴趣的朋友一起看看吧
    2022-03-03
  • 使用Python读取.nc文件的方法详解

    使用Python读取.nc文件的方法详解

    .nc文件,即NetCDF(Network Common Data Form)文件,是一种用于存储科学数据的文件格式,本文主要为大家介绍了两种常见的读取方法,希望对大家有所帮助
    2024-03-03

最新评论