python assert能用于单元测试吗 - 问答

是的，Python的assert语句可以用于单元测试。在Python中，assert语句用于检查一个条件是否为真。如果条件为假，则会引发一个AssertionError异常。在编写单元测试时，我们通常使用assert语句来验证代码的输出是否符合预期。

以下是一个简单的示例，展示了如何在单元测试中使用assert语句：

import unittest

def add(a, b):
    return a + b

class TestAddition(unittest.TestCase):
    def test_add(self):
        self.assertEqual(add(1, 2), 3)
        self.assertEqual(add(-1, 1), 0)
        self.assertEqual(add(0, 0), 0)
        # 使用assert语句检查异常情况
        with self.assertRaises(TypeError):
            add("1", 2)

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

在这个例子中，我们定义了一个add函数，用于计算两个数的和。然后，我们创建了一个名为TestAddition的单元测试类，该类继承自unittest.TestCase。在test_add方法中，我们使用assertEqual方法来验证add函数的输出是否符合预期。此外，我们还使用了assertRaises方法来检查当传递错误类型的参数时，add函数是否会引发TypeError异常。

0 赞

0 踩