在Python中执行单元测试,你可以使用内置的`unittest`模块,或者选择第三方库如`pytest`。以下是使用`unittest`模块进行单元测试的基本步骤和示例代码:
创建测试类和测试方法
定义一个继承自`unittest.TestCase`的测试类。
在测试类中编写测试方法,方法名通常以`test_`开头。
编写测试用例
在测试方法中使用`self.assertEqual`、`self.assertTrue`等方法来验证代码的行为是否符合预期。
运行测试
可以在命令行中直接运行测试脚本,或者在IDE中通过测试运行器执行。
下面是一个简单的示例,展示了如何使用`unittest`进行单元测试:
```python
my_module.py
def add(a, b):
return a + b
test_my_module.py
import unittest
from my_module import add
class TestAddition(unittest.TestCase):
def test_add_positive_numbers(self):
self.assertEqual(add(1, 2), 3)
def test_add_negative_numbers(self):
self.assertEqual(add(-1, -2), -3)
if __name__ == '__main__':
unittest.main()
在命令行中运行测试脚本:
```sh
python test_my_module.py
这将执行所有的测试用例,并显示测试结果。