Python高级用法：使用unittest进行单元测试

艾醒(AiXing-w)

于 2024-01-10 10:25:05 发布

阅读量945

点赞数 14

分类专栏： python高级用法文章标签： python 单元测试开发语言

本文链接：https://blog.youkuaiyun.com/DuLNode/article/details/135495010

版权

本文介绍了如何使用Python的unittest模块对is_prime函数进行单元测试，包括创建测试用例、使用assertTrue和assertFalse等断言方法，以及比较了旧式test_suite函数的使用方式。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

前言

Python中含有一个单元测试模块unittest，我们可以使用其中的工具来测试代码。在编写测试用例后，可以查看一系列输入是否得到了预期的输出。单元测试专注于测试软件单元，软件单元可以被理解为应用程序代码的最小可测试部分。

代码

import unittest


def is_prime(number):
    if number < 0 or number in (0, 1):
        return False
    for element in range(2, number):
        if number % element == 0:
            return False
    return True


class MyTests(unittest.TestCase):
    def test_is_prime(self):
        self.assertTrue(is_prime(5))
        self