Python 单元测试是确保代码质量的重要手段,它可以帮助我们验证代码的正确性。下面是一些常用的 Python 单元测试方法。
断言方法
在 Python 中,我们通常使用 assert
语句来进行断言。以下是一些常用的断言方法:
assertEqual(a, b)
:检查 a 和 b 是否相等。assertNotEqual(a, b)
:检查 a 和 b 是否不相等。assertTrue(condition)
:检查条件是否为真。assertFalse(condition)
:检查条件是否为假。
测试框架
Python 有许多测试框架,其中最流行的是 unittest
和 pytest
。
unittest
:Python 标准库中的测试框架,易于上手。pytest
:一个成熟、功能丰富的测试框架,提供丰富的断言方法。
示例
以下是一个使用 unittest
的简单示例:
import unittest
class TestSum(unittest.TestCase):
def test_sum(self):
self.assertEqual(sum([1, 2, 3]), 6, "1+2+3 should be 6")
if __name__ == '__main__':
unittest.main()
Python 单元测试