断言是编程中常用的一种验证方法,也是测试代码中最重要的部分,用于验证某个条件是否为真,验证测试结果与预期结果是否一致。
unittest 提供了方便的断言方法,用于验证测试结果是否符合预期,若验证失败,则会抛出 AssertionError 异常,并输出指定的错误信息。本文将介绍 unittest 中常用的断言方法以及使用场景。
断言的类型
在 unittest 中,常用的断言方法可以分为以下5类:相等断言、包含断言、布尔断言、类型断言、异常断言。
1. 相等断言
相等断言用于验证两个值是否相等。
-
assertEqual(a, b, msg=None):验证 a 和 b 是否相等。
-
assertNotEqual(a, b, msg=None):验证 a 和 b 是否不相等。
代码实例:
import unittest
class TestEqual(unittest.TestCase):
def test_equal(self):
self.assertEqual(1 + 2, 3)
在上述代码中,test_equal 方法使用 assertEqual 方法验证 1 + 2 是否等于 3。
2. 包含断言
包含断言用于验证某个值是否包含在另一个值中。
-
assertIn(a, b, msg=None):验