Flask 测试指南：unittest 和 pytest 示例

Flask 可以使用 unittest 或 pytest 进行测试。以下分别介绍两种测试方法的步骤：

使用 unittest 进行测试

创建测试文件，例如 test.py。
导入 Flask 和 unittest 模块：

from flask import Flask
import unittest

创建测试类，继承 unittest.TestCase：

class TestFlask(unittest.TestCase):
    def setUp(self):
        self.app = Flask(__name__)
        self.app.config['TESTING'] = True
        self.client = self.app.test_client()
    
    def tearDown(self):
        pass

编写测试方法：

def test_index(self):
    response = self.client.get('/'
    self.assertEqual(response.status_code, 200)
    self.assertIn(b'Hello, World!', response.data)

在测试文件中加入 main 方法，运行测试：

if __name__ == '__main__':
    unittest.main()

使用 pytest 进行测试

安装 pytest 模块：

pip install pytest

创建测试文件，例如 test.py。
编写测试方法：

def test_index():
    app = Flask(__name__)
    app.config['TESTING'] = True
    client = app.test_client()
    response = client.get('/'
    assert response.status_code == 200
    assert b'Hello, World!' in response.data

在命令行中进入测试文件所在目录，运行 pytest：

pytest test.py

注意：

以上示例代码仅供参考，请根据实际情况进行修改。
为了更好地理解测试过程，建议参考 Flask 官方文档：https://flask.palletsprojects.com/en/2.2.x/testing/