def test_take_screenshot_with_suffix(browser, app_url):
    """Should add the suffix to the screenshot file name"""
    browser.visit(app_url)

    filename = browser.screenshot(suffix=".png")
    assert ".png" in filename[-4:]
Should add the suffix to the screenshot file name
tests/tests_webdriver/test_screenshot.py