Element screenshot in Selenium Python - Practice Problems & Coding Challenges

Challenge - 5 Problems

❓ Predict Output

intermediate

What is the output of this Selenium element screenshot code?

Consider the following Python Selenium code snippet that takes a screenshot of a web element. What will be the result after 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()

A A file named 'header.png' is saved containing the screenshot of the first <h1> element on the page.

B The entire page screenshot is saved as 'header.png', ignoring the element locator.

C Raises a NoSuchElementException because 'h1' tag does not exist on example.com.

D SyntaxError due to incorrect method usage for taking element screenshot.

💡 Hint

Remember that element.screenshot() saves only the specific element's image, not the whole page.

The method element.screenshot(filename) saves a screenshot of only the located element. Since example.com has an

tag, the file 'header.png' will be created with that element's image.

❓ assertion

intermediate

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

You have saved an element screenshot as 'logo.png'. Which assertion correctly checks that the file was created?

Selenium Python

import os

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

A assert os.path.isfile('logo.png') == False

B assert os.path.isfile('logo.png')

C assert os.path.isdir('logo.png')

D assert os.path.exists('logo.png') == False

💡 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 is the correct way to assert the screenshot file was created.

🔧 Debug

advanced

Why does this element screenshot code raise an error?

Identify the cause of the error in this Selenium Python code snippet:

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('nonexistent.png')
browser.quit()

A NoSuchElementException is raised because the element with ID 'nonexistent' does not exist.

B FileNotFoundError because 'nonexistent.png' path is invalid.

C TypeError because screenshot() requires a file object, not a string filename.

D TimeoutException because the page took too long to load.

💡 Hint

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

The code tries to find an element with ID 'nonexistent' which does not exist on example.com, causing NoSuchElementException before screenshot() is called.

❓ framework

advanced

Which Selenium WebDriver method is best to capture a screenshot of a specific element?

You want to capture only a single element's screenshot in Selenium Python. Which method should you use?

A driver.get_screenshot_as_file('element.png')

B driver.save_screenshot('element.png')

C element.screenshot('element.png')

D element.get_screenshot_as_base64()

💡 Hint

Only the element object has a screenshot method that saves just that element.

element.screenshot(filename) saves the screenshot of the element only. driver.save_screenshot() saves the whole page.

🧠 Conceptual

expert

What is a key limitation of element screenshot in Selenium WebDriver?

Which of the following is a known limitation when taking screenshots of elements using Selenium WebDriver?

A Element screenshots automatically scroll the page to the bottom before capturing.

B Element screenshots always include the entire page content regardless of element size.

C Element screenshots can only be saved in JPEG format, not PNG.

D Element screenshots may be blank if the element is not visible or outside the viewport.

💡 Hint

Think about what happens if the element is hidden or off-screen.

If the element is hidden or outside the visible area, the screenshot may be blank or incomplete because Selenium captures only visible pixels.