如何使用PyTest捕获测试用例失败时的屏幕截图

pip install pytest pytest-selenium

import pytest
from selenium import webdriver

@pytest.fixture(scope="module")
def browser():
    driver = webdriver.Chrome()
    yield driver
    driver.quit()

def test_example(browser):
    browser.get("https://www.example.com")
    assert "Example Domain" in browser.title

import pytest
from selenium import webdriver

@pytest.fixture(scope="module")
def browser():
    driver = webdriver.Chrome()
    yield driver
    driver.quit()

def pytest_runtest_makereport(item, call):
    if call.when == "call" and call.failed:
        try:
            browser = item.funcargs['browser']
            browser.save_screenshot('screenshot.png')
        except Exception as e:
            print(f"Failed to capture screenshot: {e}")

def test_example(browser):
    browser.get("https://www.example.com")
    assert "Example Domain" in browser.title