Python 测试开发常用脚本大全

 1. 接口测试脚本（Requests + JSON）

使用 requests 发起 HTTP 请求，验证接口状态码和返回内容。

import requestsimport jsondef test_api(url, method='get', headers=None, params=None, payload=None):    try:        if method.lower() == 'get':            response = requests.get(url, params=params, headers=headers)        elif method.lower() == 'post':            response = requests.post(url, json=payload, headers=headers)        else:            raise ValueError("不支持的请求方法")        print(f"请求地址: {response.url}")        print(f"状态码: {response.status_code}")        print(f"响应内容: {json.dumps(response.json(), indent=2)}")        return response    except Exception as e:        print(f"请求失败: {e}")        return None# 示例调用if __name__ == "__main__":    url = "https://jsonplaceholder.typicode.com/posts"    res = test_api(url, method='get')‍

✅ 2. 数据驱动测试脚本（DDT）

使用 ddt 模块实现参数化测试，适用于多组测试数据的接口或功能验证。​​​​​​​

import unittestfrom ddt import ddt, data, unpack@ddtclass TestLogin(unittest.TestCase):    @data(        ("admin", "123456", True),        ("guest", "wrongpass", False),        ("", "", False)    )    @unpack    def test_login(self, username, password, expected):        result = self.login(username, password)        self.assertEqual(result, expected)    def login(self, username, password):        # 模拟登录逻辑        return username == "admin" and password == "123456"if __name__ == '__main__':    unittest.main()

📌 安装依赖：

pip install ddt‍

✅ 3. UI 自动化测试脚本（Selenium）

使用 selenium 实现浏览器自动化操作，适用于 Web 页面测试。​​​​​​​

from selenium import webdriverfrom selenium.webdriver.common.by import Byimport timedriver = webdriver.Chrome()try:    driver.get("https://www.example.com/login")    driver.find_element(By.ID, "username").send_keys("testuser")    driver.find_element(By.ID, "password").send_keys("password123")    driver.find_element(By.ID, "login-btn").click()    time.sleep(2)  # 等待页面加载    assert "欢迎页" in driver.title    print("登录测试通过")finally:    driver.quit()

📌 安装依赖：

pip install selenium

注意：需要安装对应的浏览器驱动（如 ChromeDriver）并添加到系统路径。

✅ 4. 性能测试脚本（Locust）

使用 locust 编写性能测试脚本，模拟并发用户访问接口。

​​​​​​​

from locust import HttpUser, task, betweenclass WebsiteUser(HttpUser):    wait_time = between(1, 3)  # 用户等待时间 1~3 秒    @task    def load_homepage(self):        self.client.get("/")    @task(3)    def get_posts(self):        self.client.get("/posts")

📌 运行方式：

locust -f locustfile.py

📌 安装依赖：

pip install locust‍

✅ 5. 日志记录模块（logging）

统一日志输出格式，便于调试和分析测试过程。​​​​​​​

import loggingimport os# 创建 logs 目录os.makedirs("logs", exist_ok=True)# 配置日志logging.basicConfig(    level=logging.INFO,    format='%(asctime)s [%(levelname)s] %(message)s',    handlers=[        logging.FileHandler("logs/test.log"),        logging.StreamHandler()    ])logger = logging.getLogger(__name__)# 示例使用logger.info("开始执行测试...")logger.warning("这是一个警告信息")logger.error("这是一个错误信息")‍

✅ 6. HTML 报告生成（Pytest + Allure）

结合 pytest 和 allure-pytest 生成美观的 HTML 测试报告。

示例脚本：​​​​​​​

import pytestdef add(a, b):    return a + b@pytest.mark.parametrize("a, b, expected", [(1, 1, 2), (2, 3, 5), (-1, 1, 0)])def test_add(a, b, expected):    assert add(a, b) == expected📌 执行命令生成报告：pytest --alluredir=./allure-results test_add.pyallure serve ./allure-results

📌 安装依赖：

pip install pytest allure-pytest

需要安装 Allure 命令行工具

✅ 7. CSV 文件读取与测试数据处理

用于从 CSV 中读取测试数据，常用于接口测试、数据驱动测试。​​​​​​​

import csvdef read_test_data(file_path):    with open(file_path, newline='', encoding='utf-8') as csvfile:        reader = csv.DictReader(csvfile)        for row in reader:            yield rowfor data in read_test_data('testdata.csv'):    print(data['username'], data['password'])

CSV 文件示例（testdata.csv）：

username,password

testuser1,123456

testuser2,wrongpass

,testempty‍

✅ 8. 邮件通知脚本（SMTP）

测试完成后发送邮件通知测试结果。​​​​​​​

import smtplibfrom email.mime.text import MIMETextfrom email.header import Headerdef send_email(subject, content, to_emails):    msg = MIMEText(content, 'plain', 'utf-8')    msg['Subject'] = Header(subject, 'utf-8')    msg['From'] = "your_email@example.com"    msg['To'] = ", ".join(to_emails)    try:        server = smtplib.SMTP("smtp.example.com", 587)        server.starttls()        server.login("your_email@example.com", "your_password")        server.sendmail(msg['From'], to_emails, msg.as_string())        print("邮件发送成功")    except Exception as e:        print("邮件发送失败:", e)    finally:        server.quit()# 示例调用send_email("自动化测试完成", "本次测试共执行了 10 条用例，全部通过。", ["dev@example.com"])

✅ 9. Git 操作脚本（自动化拉取代码）

在 CI/CD 中自动拉取最新代码进行测试。​​​​​​​

from git import Repodef pull_latest_code(repo_path):    repo = Repo(repo_path)    origin = repo.remotes.origin    origin.pull()    print(f"已更新到最新版本，当前分支: {repo.active_branch}")# 示例调用pull_latest_code("/path/to/your/repo")

📌 安装依赖：

pip install gitpython‍

✅ 10. 测试覆盖率统计（coverage.py）

统计测试代码对源码的覆盖情况。​​​​​​​

coverage run -m pytest test_my_module.pycoverage report -mcoverage html

📌 安装依赖：

pip install coverage‍

📌 小结：常用脚本分类表