编写测试函数通常是为了确保代码的正确性和稳定性。以下是使用不同测试框架编写测试函数的方法：
使用Pytest框架
Pytest是一个流行的Python测试框架，它使得编写和运行测试变得非常简单。以下是使用Pytest编写测试函数的步骤：
安装Pytest
```bash
pip install pytest
```
编写测试函数
测试函数的名称必须以`test_`开头，这样Pytest才能自动发现并运行这些测试。在测试函数中，使用`assert`语句来判断被测试的代码是否符合预期结果。
```python
test_example.py
def test_addition（）:
assert 1 + 1 == 2
def test_subtraction（）:
assert 5 - 3 == 2
def test_multiplication（）:
assert 2 * 3 == 6
```
运行测试
在命令行中，导航到包含测试文件的目录，然后运行：
```bash
pytest
```
Pytest会自动发现并运行所有以`test_`开头的函数。
使用unittest框架
unittest是Python的标准库之一，用于编写和运行测试。以下是使用unittest编写测试函数的步骤：
编写测试类
创建一个继承自`unittest.TestCase`的测试类，并在类中定义测试方法。测试方法必须以`test_`开头。
```python
test_example.py
import unittest
class TestAddFunction（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）
def test_add_mixed_numbers（self）:
self.assertEqual（add（-1, 2）, 1）
```
运行测试
在命令行中，导航到包含测试文件的目录，然后运行：
```bash
python -m unittest test_example.py
```
或者，如果你在脚本中直接运行测试：
```python
if __name__ == '__main__':
unittest.main（）
```
使用自定义断言宏
有时你可能希望使用自定义的断言宏来简化测试代码。以下是一个简单的例子：
定义断言宏
```python
test_example.py
import unittest
TEST_ASSERT = unittest.TestCase（）.assertEqual
def test_addition（）:
TEST_ASSERT（1 + 1, 2）
def test_subtraction（）:
TEST_ASSERT（5 - 3, 2）
def test_multiplication（）:
TEST_ASSERT（2 * 3, 6）
```
总结
选择合适的测试框架（如Pytest或unittest）可以大大提高测试代码的编写效率和可维护性。无论使用哪种框架，关键是要确保测试函数以`test_`开头，并使用断言来验证代码的正确性。