Element-level screenshot in Selenium Python - Practice Problems & Coding Challenges

Challenge - 5 Problems

❓ Predict Output

intermediate

What is the output of this element-level screenshot code?

Consider the following Selenium Python code snippet that takes a screenshot of a specific element on a webpage. What will be the result of running this code?

Selenium Python

from selenium import webdriver
from selenium.webdriver.common.by import By

browser = webdriver.Chrome()
browser.get('https://example.com')
element = browser.find_element(By.TAG_NAME, 'h1')
element.screenshot('header.png')
browser.quit()
print('Screenshot saved')

A Screenshot saved (and a file named 'header.png' is created with the element's image)

B No file is created; code runs without error and prints 'Screenshot saved'

C Raises NoSuchElementException because 'h1' tag does not exist

D SyntaxError due to incorrect method usage

💡 Hint

Remember that element.screenshot() saves the image of the element to the given filename.

The code finds the first

element on the page and saves its screenshot as 'header.png'. Then it prints confirmation. No errors occur if the element exists.

❓ locator

intermediate

Which locator is best to capture an element for screenshot?

You want to take a screenshot of a button with text 'Submit' on a webpage. Which locator is the best choice to find this button uniquely and reliably?

A By.XPATH, "//button[text()='Submit']" - finds button by exact text

B By.CLASS_NAME, 'btn' - finds any element with class 'btn'

C By.TAG_NAME, 'button' - finds the first button element

D By.CSS_SELECTOR, 'div > button' - finds any button inside a div

💡 Hint

Choose the locator that uniquely identifies the button by its visible text.

Using XPath with exact text ensures you get the correct 'Submit' button, avoiding other buttons or elements.

❓ assertion

advanced

Which assertion correctly verifies the element screenshot file exists after saving?

After taking an element-level screenshot saved as 'element.png', which assertion correctly checks the file was created?

Selenium Python

import os

# Assume element.screenshot('element.png') was called before

A assert os.path.isfile('element.png') == false

B assert os.path.exists('element.png') == false

C assert os.path.isfile('element.png')

D assert os.path.isdir('element.png')

💡 Hint

Check if the file exists and is a file, not a directory.

os.path.isfile returns true if the path exists and is a file. This confirms the screenshot file was created.

🔧 Debug

advanced

Why does this element screenshot code raise an error?

This Selenium Python code raises an error. Identify the cause.

Selenium Python

from selenium import webdriver
from selenium.webdriver.common.by import By

browser = webdriver.Chrome()
browser.get('https://example.com')
element = browser.find_element(By.ID, 'nonexistent')
element.screenshot('elem.png')
browser.quit()

A TimeoutException because page did not load

B NoSuchElementException because element with ID 'nonexistent' does not exist

C TypeError because screenshot method requires a file object, not a string

D WebDriverException due to invalid browser driver

💡 Hint

Check if the element locator matches any element on the page.

The element with ID 'nonexistent' is not found, so find_element raises NoSuchElementException before screenshot is called.

❓ framework

expert

How to integrate element-level screenshot in pytest for failure debugging?

You want to automatically take a screenshot of a specific element when a pytest test fails. Which pytest fixture setup correctly implements this?

Selenium Python

import pytest
from selenium import webdriver
from selenium.webdriver.common.by import By

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

@pytest.hookimpl(tryfirst=True, hookwrapper=True)
def pytest_runtest_makereport(item, call, when):
    # Implementation to capture element screenshot on failure
    outcome = yield
    rep = outcome.get_result()
    if rep.when == 'call' and rep.failed():
        driver = item.funcargs['browser']
        try:
            elem = driver.find_element(By.ID, 'target')
            elem.screenshot('fail_elem.png')
        except Exception:
            pass

A Use pytest.mark.parametrize to pass screenshot filename to tests

B Use pytest.skip to skip tests that fail to capture screenshots

C Use pytest.fixture to capture screenshot after each test regardless of result

D Use pytest_runtest_makereport hook to check test failure and capture element screenshot

💡 Hint

Look for a hook that runs after each test and can detect failure.

pytest_runtest_makereport hook runs after each test phase and can detect failure to trigger screenshot capture.