Python:
编写测试
如何操作:
Python 自带了一个用于编写测试的内置模块,名为 unittest
。这是使用它来测试一个简单函数的方法:
import unittest
def add(a, b):
return a + b
class TestAddFunction(unittest.TestCase):
def test_add(self):
self.assertEqual(add(1, 2), 3)
self.assertEqual(add(-1, 1), 0)
self.assertNotEqual(add(10, 2), 12, "应该是 12")
if __name__ == '__main__':
unittest.main()
当你运行这个测试脚本时,你应该会看到输出,指示你的测试通过(或失败)。
若要进行更现代和更具表达性的测试,你可以使用第三方库,如 pytest
。首先,你需要使用 pip 安装它:
pip install pytest
然后,你可以以更简单的方式编写你的测试,无需继承任何内容:
# 将此保存在一个名为 test_with_pytest.py 的文件中
def add(a, b):
return a + b
def test_add():
assert add(1, 2) == 3
assert add(-1, 1) == 0
assert add(10, 2) != 12, "应该是 12"
要使用 pytest
运行你的测试,只需执行:
pytest test_with_pytest.py
你应该会看到 pytest 显示的测试结果输出。