用python写网站压力测试

在Python中，可以使用多个库和框架来进行网站压力测试。其中，最常用的是`locust`和`pytest`。

下面是一个使用`locust`进行网站压力测试的示例代码：

from locust import HttpUser, TaskSet, task, between

class WebsiteUser(HttpUser):
    wait_time = between(5, 15)

    @task
    def index(self):
        self.client.get("/")

    @task
    def about(self):
        self.client.get("/about")

在上面的示例中，`WebsiteUser`类继承自`HttpUser`类，表示将使用`locust`进行HTTP请求。`wait_time`属性指定了每个用户等待时间的范围。`@task`装饰器定义了一个任务，这里定义了两个任务，分别是访问首页和关于页面。

使用`locust`进行压力测试时，可以使用以下命令启动测试：

locust -f test_file.py

其中，`test_file.py`是包含上述代码的Python文件。

另外，也可以使用`pytest`进行网站压力测试。下面是一个使用`pytest`进行压力测试的示例代码：

import pytest
import requests

@pytest.mark.parametrize("url", [
    "https://www.example.com",
    "https://www.example.com/about",
])
def test_load(url):
    response = requests.get(url)
    assert response.status_code == 200

在上面的示例中，`pytest`使用`pytest.mark.parametrize`装饰器定义了一个参数化测试，其中`url`参数包含了要测试的网址。`test_load`函数发送HTTP请求并检查响应状态码是否为200。

使用`pytest`进行压力测试时，可以使用以下命令启动测试：

pytest test_file.py -n 4

其中，`-n 4`表示使用4个进程运行测试。