版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/weixin_36485376/article/details/81057961
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作为前缀开头的方法。