单元测试请求重试python

import pytest
import requests

@pytest.fixture
def retry():
    max_retries = 3
    retry_delay = 1

    def _retry(func):
        def wrapper(*args, **kwargs):
            retries = 0
            while retries < max_retries:
                try:
                    return func(*args, **kwargs)
                except requests.exceptions.RequestException:
                    retries += 1
                    time.sleep(retry_delay)
            pytest.fail("Request failed after maximum retries")
        return wrapper
    return _retry

@retry()
def test_api_request():
    response = requests.get("https://example.com/api")
    assert response.status_code == 200