当编写Python代码时,可以使用自测试(unit testing)来验证代码的正确性。Python提供了unittest模块,它是Python标准库中用于编写和运行单元测试的框架。
以下是一个简单的例子,演示如何使用unittest模块进行自测试:
import unittest
def add_numbers(a, b):
return a + b
class TestAddNumbers(unittest.TestCase):
def test_add_positive_numbers(self):
result = add_numbers(3, 4)
self.assertEqual(result, 7)
def test_add_negative_numbers(self):
result = add_numbers(-2, -5)
self.assertEqual(result, -7)
def test_add_zero(self):
result = add_numbers(10, 0)
self.assertEqual(result, 10)
if __name__ == '__main__':
unittest.main()