Python--标准测试工具unittst-优快云博客

本文链接：https://blog.youkuaiyun.com/weixin_36485376/article/details/81057961

本文介绍Python中unittest单元测试框架的基本使用方法。通过一个简单的测试案例，展示了如何定义测试类及测试方法，unittest框架如何扫描并执行这些测试。

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

unittest是最常见的Python单元测试框架，它提供了一个名为TestCase的基类，它有一组非常广泛的方法来验证函数调用和语句的输出。
该模块是为了编写单元测试而创建的，但是只要测试使用用户接口，验收测试也可以用它来编写。

看一个为primes.py模块编写的简单测试用例

import unittest
from primes import is_primes

class PrimesTests(unittest.TestCase):
    def test_is_prime(self):
        self.assertTrue(is_primes(5))
        self.assertTrue(is_primes(7))

        self.assertFalse(is_primes(8))
        self.assertFalse(is_primes(0))
        self.assertFalse(is_primes(1))

        self.assertFalse(is_primes(-1))
        self.assertFalse(is_primes(-3))
        self.assertFalse(is_primes(-6))

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

unittest.main()函数是一个通用程序，它允许将整个模块作为一个测试套件执行，执行效果如下：

Ran 1 test in 0.001s

OK

Process finished with exit code 0

unittest.main()函数扫描当前模块的上下文，并查找TestCase类的子类，它实力化这些子类，然后运行所有以test作为前缀开头的方法。