### 如何为新功能编写 Python 自动化测试代码 在开发过程中，为新增功能编写自动化测试代码是确保代码质量的重要环节。Python 提供了丰富的工具和框架来支持这一过程。以下是一些关键步骤和示例代码，帮助开发者为新功能编写自动化测试。 #### 使用 pytest 框架编写单元测试 `pytest` 是一个强大的测试框架，能够简化测试用例的编写和运行。它支持参数化测试、插件扩展等功能，适合用于单元测试和集成测试。 ```python # 示例：为一个简单的计算器功能编写单元测试 import pytest def add(a, b): return a + b @pytest.mark.parametrize("a, b, expected", [ (1, 2, 3), (-1, -1, -2), (0, 0, 0) ]) def test_add(a, b, expected): assert add(a, b) == expected ``` 上述代码展示了如何使用 `pytest` 的参数化功能测试 `add` 函数的不同输入场景[^2]。 #### 集成 FlaskRESTful API 测试 如果新增功能涉及 RESTful API，可以结合 `Flask` 和 `Flask-Testing` 扩展进行测试。以下是一个示例： ```python from flask import Flask, jsonify from flask.testing import FlaskClient app = Flask(__name__) @app.route('/api/add', methods=['POST']) def api_add(): data = request.json result = data['a'] + data['b'] return jsonify({'result': result}) class TestAPI: def setup_method(self): self.app = app.test_client() def test_api_add(self): response = self.app.post('/api/add', json={'a': 5, 'b': 10}) assert response.status_code == 200 assert response.json['result'] == 15 ``` 此示例演示了如何使用 `Flask-Testing` 模块测试一个简单的加法 API[^1]。 #### 移动应用自动化测试 对于移动应用的新功能，可以使用 `Appium` 和 `Python` 编写自动化测试脚本。以下是一个基本示例： ```python from appium import webdriver desired_caps = { 'platformName': 'Android', 'deviceName': 'emulator-5554', 'appPackage': 'com.example.app', 'appActivity': '.MainActivity' } driver = webdriver.Remote('http://localhost:4723/wd/hub', desired_caps) def test_login(): username_field = driver.find_element_by_id('username') password_field = driver.find_element_by_id('password') login_button = driver.find_element_by_id('login_button') username_field.send_keys('test_user') password_field.send_keys('password123') login_button.click() # 验证登录成功 welcome_message = driver.find_element_by_id('welcome_message') assert welcome_message.text == 'Welcome, test_user!' ``` 这段代码展示了如何使用 `Appium` 测试 Android 应用中的登录功能[^3]。 #### 性能测试 对于需要性能测试的新功能，可以使用 `locust` 工具模拟高并发场景。以下是一个简单的示例： ```python from locust import HttpUser, task, between class WebsiteUser(HttpUser): wait_time = between(1, 5) @task def load_page(self): self.client.get("/api/add?a=1&b=2") @task def stress_test(self): for i in range(100): self.client.post("/api/add", json={'a': i, 'b': i+1}) ``` 此示例展示了如何使用 `locust` 对 RESTful API 进行负载测试[^2]。 ---