Python+Pytest压力测试

这篇具有很好参考价值的文章主要介绍了Python+Pytest压力测试。希望对大家有所帮助。如果存在错误或未考虑完全的地方,请大家不吝赐教,您也可以点击"举报违法"按钮提交疑问。

在现代Web应用程序中,性能是至关重要的。为了确保应用程序能够在高负载下正常运行,我们需要进行性能测试。 今天,应小伙伴的提问, 老向老师来写一个Pytest进行压力测试的简单案例。 这个案例的测试网站我们就隐藏了,不过网站的基本情况是:

阿里框架:FastAdmin.net

1.程序说明

1.1 设置测试参数

首先,我做的第一件事情就是设置测试参数。代码如下

# 定义测试用例

def test_performance():

# 设置测试参数

url = 'http://www.a.com/'

num_threads = 20

num_requests = 200

timeout = 5

这里面,我设置了网站的URL, 线程数, 每个线程的请求次数,以及超时时间。 可以看到, 这里面我一共会做4000次请求。

1.2 初始化测试结果

这段代码我想不需要我多讲, 这里做一个提示:注意缩进, 这段代码仍然在测试用例test_performance内。

# 初始化测试结果

response_times = []

errors = 0

successes = 0

1.3 定义测试函数

接下来, 我定义了一个内部函数。这个函数就是在某一线程内完成设定次数的请求。

# 定义测试函数

def test_func():

nonlocal errors, successes

for _ in range(num_requests):

try:

start_time = time.time()

requests.get(url, timeout=timeout)

end_time = time.time()

response_time = end_time - start_time

response_times.append(response_time)

successes += 1

except requests.exceptions.RequestException:

errors += 1

1.4 创建线程、执行线程、等待

# 创建测试线程

threads = []

for _ in range(num_threads):

t = threading.Thread(target=test_func)

threads.append(t)

# 启动测试线程

for t in threads:

t.start()

# 等待测试线程结束

for t in threads:

t.join()

1.5 计算测试结果

# 计算测试结果

total_requests = num_threads * num_requests

throughput = successes / (sum(response_times) or 1)

concurrency = num_threads

error_rate = errors / (total_requests or 1)

cpu_usage = psutil.cpu_percent()

memory_usage = psutil.virtual_memory().percent

1.6 将测试结果写入文件

# 将测试结果写入文件

with open('performance_test_result.txt', 'w') as f:

f.write(f'总请求数:{total_requests}\n')

f.write(f'总时间:{sum(response_times):.2f}s\n')

f.write(f'吞吐量:{throughput:.2f} requests/s\n')

f.write(f'并发数:{concurrency}\n')

f.write(f'错误率:{error_rate:.2%}\n')

f.write(f'CPU利用率:{cpu_usage:.2f}%\n')

f.write(f'内存利用率:{memory_usage:.2f}%\n')文章来源地址https://www.toymoban.com/news/detail-528607.html

2.程序执行

2.1 直接执行

在PyCharm里面直接执行这段代码, 得出的结果是:

总请求数:4000

总时间:1837.65s

吞吐量:2.17 requests/s

并发数:20

错误率:0.12%

CPU利用率:4.10%

内存利用率:88.60%

2.2 加个装饰器然后出报告

如果在PyCharm里面直接执行上面的代码, 虽然我们把结果写在文件中,但是, 不好看呀。

所以呢,我再额外介绍一个方法,这个方法能够生成一个相对美观的测试报告出来。

2.2.1 声明压力测试

首先在定义用例的时候通过装饰器声明这是一个压力测试:

# 定义测试用例

@pytest.mark.performance

def test_performance():

# 设置测试参数

url = 'http://www.a.biz/'

num_threads = 20

2.2.2 在命令行中通过pytest命令执行测试

第二步, 在命令行中执行测试

-v 用于显示详细的测试结果

--html 用于指定输出报告的位置。 这个参数需要依赖包:pytest-html

$ pytest -v --html=report.html test_a.py

输出执行结果是:

======================== test session starts =================================

platform win32 -- Python 3.10.9, pytest-7.2.1, pluggy-1.0.0 -- D:\python-grp\miniconda_env\py3.10_playwright\python.exe

cachedir: .pytest_cache

metadata: {'Python': '3.10.9', 'Platform': 'Windows-10-10.0.22624-SP0', 'Packages': {'pytest': '7.2.1', 'pluggy': '1.0.0'}, 'Plugins': {'allure-pytest': '2.12.0', 'base-url': '2.0.0', 'html': '3.2.0', 'metadata': '2.0.4', 'ordering': '0.6', 'playwright': '0.3.0'}, 'JAVA_HOME': 'D:\\java-grp\\jdk\\', 'Base URL': ''}

rootdir: E:\develop\python\pytest-training\test

plugins: allure-pytest-2.12.0, base-url-2.0.0, html-3.2.0, metadata-2.0.4, ordering-0.6, playwright-0.3.0

collected 1 item

test_a.py::test_performance PASSED [100%]

========================== warnings summary =================================

test_a.py:25

E:\develop\python\pytest-training\test\test_a.py:25: PytestUnknownMarkWarning: Unknown pytest.mark.performance - is this a typo? You can register custom marks to avoid this warning - for details, see https://docs.pytest.org/en/stable/how-to/mark.html

@pytest.mark.performance

-- Docs: https://docs.pytest.org/en/stable/how-to/capture-warnings.html

-- generated html file: file:///E:/develop/python/pytest-training/test/report.html --

================= 1 passed, 1 warning in 99.09s (0:01:39) ===================

(D:\python-grp\miniconda_env\py3.10_playwright) E:\develop\python\pytest-training\test>

3.案例缺陷

因为时间关系, 本案例今天没有时间在服务器端执行, 所以通过psutil库所取得CPU利用率和内存利用率时间并不对。 如果是在服务器端执行, 这两个数字才是对的。

如果要在本地获取服务器的CPU,内存,IO等情况,有一个监控神器:Prometheus。

4 完整源码

#!/usr/bin/env python

# -*- coding:utf-8 -*-

"""

#-----------------------------------------------------------------------------

# --- TDOUYA STUDIOS ---

#-----------------------------------------------------------------------------

#

# @Project : pytest-training

# @File : test_a.py

# @Author : tianxin.xp@gmail.com

# @Date : 2023/3/10 14:39

#

# 压力测试案例

#

#--------------------------------------------------------------------------"""

import threading

import time

import psutil

import pytest

import requests

# 定义测试用例

@pytest.mark.performance

def test_performance():

# 设置测试参数

url = 'http://www.tdouya.biz/'

num_threads = 20

num_requests = 200

timeout = 5

# 初始化测试结果

response_times = []

errors = 0

successes = 0

# 定义测试函数

def test_func():

nonlocal errors, successes

for _ in range(num_requests):

try:

start_time = time.time()

requests.get(url, timeout=timeout)

end_time = time.time()

response_time = end_time - start_time

response_times.append(response_time)

successes += 1

except requests.exceptions.RequestException:

errors += 1

# 创建测试线程

threads = []

for _ in range(num_threads):

t = threading.Thread(target=test_func)

threads.append(t)

# 启动测试线程

for t in threads:

t.start()

# 等待测试线程结束

for t in threads:

t.join()

# 计算测试结果

total_requests = num_threads * num_requests

throughput = successes / (sum(response_times) or 1)

concurrency = num_threads

error_rate = errors / (total_requests or 1)

cpu_usage = psutil.cpu_percent()

memory_usage = psutil.virtual_memory().percent

# 输出测试结果

print(f'总请求数:{total_requests}')

print(f'总时间:{sum(response_times):.2f}s')

print(f'吞吐量:{throughput:.2f} requests/s')

print(f'并发数:{concurrency}')

print(f'错误率:{error_rate:.2%}')

print(f'CPU利用率:{cpu_usage:.2f}%')

print(f'内存利用率:{memory_usage:.2f}%')

# 将测试结果写入文件

with open('performance_test_result.txt', 'w') as f:

f.write(f'总请求数:{total_requests}\n')

f.write(f'总时间:{sum(response_times):.2f}s\n')

f.write(f'吞吐量:{throughput:.2f} requests/s\n')

f.write(f'并发数:{concurrency}\n')

f.write(f'错误率:{error_rate:.2%}\n')

f.write(f'CPU利用率:{cpu_usage:.2f}%\n')

f.write(f'内存利用率:{memory_usage:.2f}%\n')

到了这里,关于Python+Pytest压力测试的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处: 如若内容造成侵权/违法违规/事实不符,请点击违法举报进行投诉反馈,一经查实,立即删除!

领支付宝红包 赞助服务器费用

相关文章

  • python实现接口压力测试

    直接上代码: 输出20次压测结果如下:

    2024年02月17日
    浏览(39)
  • Python测试框架 Pytest —— mock使用(pytest-mock)

    安装:pip install pytest-mock 这里的mock和unittest的mock基本上都是一样的,唯一的区别在于pytest.mock需要导入mock对象的详细路径。 先将需要模拟的天气接口,以及需要模拟的场景的代码写好,然后在进行遵循pytest的用例规范进行书写关于mock的测试用例 通过上述代码,提供pytest中

    2024年02月09日
    浏览(46)
  • Python实现短信循环压力测试教程

    一、安装依赖库 在使用短信循环压力测试之前,需要先安装需要的依赖库——selenium和webdriver。其中,selenium是Python的一个第三方库,能够模拟浏览器的行为,进行自动化测试等操作。webdriver是浏览器的驱动程序,可以实现Python与浏览器之间的交互。 pip install selenium 安装完成

    2024年02月08日
    浏览(85)
  • Python测试工具-Pytest使用详解

    Pytest是一个全功能Python测试工具,支持第三方扩展插件,能够使用其开展单元测试和复杂功能测试。可以和selenium、requests、appium等模块结合使用实现WEB UI、API、APP自动化测试。 详见参考指南文档:https://docs.pytest.org/en/7.1.x/# PDF文档 : https://media.readthedocs.org/pdf/pytest/latest/pyt

    2024年02月02日
    浏览(40)
  • Python单元测试pytest捕获日志输出

    使用pytest进行单元测试时,遇到了需要测试日志输出的情况,查看了文档 https://docs.pytest.org/en/latest/how-to/capture-stdout-stderr.html https://docs.pytest.org/en/latest/how-to/logging.html 然后试了一下,捕捉logger.info可以用caplog,获取print输出可用capsys,Demo如下: - a.py - test_a.py - 验证:

    2024年04月10日
    浏览(41)
  • Python基础介绍 —— 使用pytest进行测试!

    Pytest 是 Python 的一种单元测试框架,与 Python 自带的 unittest 测试框架类似,但是比 unittest 框架使用起来更简洁,效率更高。 Pytest 是 Python 的一种单元测试框架,与 Python 自带的 unittest 测试框架类似,但是比 unittest 框架使用起来更简洁,效率更高。 适合从简单的单元到复杂的

    2024年02月05日
    浏览(44)
  • Python单元测试框架之pytest -- 断言

    对于测试来讲,不管是功能测试,自动化测试,还是单元测试。一般都会预设一个正确的预期结果,而在测试执行的过程中会得到一个实际的结果。测试的成功与否就是拿实际的结果与预期的结果进行比较。这个比的过程实际就是断言(assert)。 在unittest单元测试框架中提供

    2024年02月11日
    浏览(64)
  • Python代理池健壮性测试 - 压力测试和异常处理

    大家好!在构建一个可靠的Python代理池时,除了实现基本功能外,我们还需要进行一系列健壮性测试来确保其能够稳定运行,并具备应对各种异常情况的能力。本文将介绍如何使用压力测试工具以及合适的异常处理机制来提升Python代理池的可用性,并为您提供高质量、专业度

    2024年02月11日
    浏览(34)
  • python+pytest接口自动化之测试函数、测试类/测试方法的封装

    今天呢,笔者想和大家聊聊python+pytest接口自动化中将代码进行封装,只有将测试代码进行封装,才能被测试框架识别执行。 例如单个接口的请求代码如下: 假设我们需要将上面这段代码编写成测试框架能执行的测试用例,仅仅只是这样写代码显然是不够的,还需要进行如下

    2024年03月09日
    浏览(65)
  • Python 面试:单元测试unit testing & 使用pytest

    calc.py test_calc.py employee.py test_employee.py 输出为: setupClass setUp test_apply_raise tearDown .setUp test_email tearDown .setUp test_fullname tearDown .teardownClass Ran 3 tests in 0.001s OK employee.py test_employee.py 输出为: setupClass setUp test_apply_raise tearDown .setUp test_email tearDown .setUp test_fullname tearDown .setUp tearDown

    2024年02月10日
    浏览(45)

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

请作者喝杯咖啡吧~博客赞助

支付宝扫一扫领取红包,优惠每天领

二维码1

领取红包

二维码2

领红包