Element screenshot in Selenium Python - Test Execution Trace

Test Overview

This test opens a webpage, finds a specific element by its ID, takes a screenshot of that element, and verifies the screenshot file is created successfully.

Test Code - Selenium

Selenium Python

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

def test_element_screenshot():
    driver = webdriver.Chrome()
    try:
        driver.get('https://example.com')
        element = driver.find_element(By.ID, 'logo')
        screenshot_path = 'element_screenshot.png'
        element.screenshot(screenshot_path)
        time.sleep(1)  # wait to ensure file is saved
        assert os.path.exists(screenshot_path), f"Screenshot file {screenshot_path} was not created."
    finally:
        driver.quit()

Execution Trace - 6 Steps

Step	Action	System State	Assertion	Result
1	Test starts and Chrome browser opens	Chrome browser window is open, ready to navigate	—	PASS
2	Browser navigates to 'https://example.com'	Example.com homepage is loaded in the browser	—	PASS
3	Find element with ID 'logo' using driver.find_element(By.ID, 'logo')	Element with ID 'logo' is located on the page	Element is found without exception	PASS
4	Take screenshot of the element and save as 'element_screenshot.png'	Screenshot file 'element_screenshot.png' is created in the test folder	Screenshot file exists on disk	PASS
5	Assert that screenshot file exists using os.path.exists	File system contains 'element_screenshot.png'	assert os.path.exists('element_screenshot.png') passes	PASS
6	Close browser and end test	Browser window is closed, test ends cleanly	—	PASS

Failure Scenario

Failing Condition: Element with ID 'logo' is not found on the page

Execution Trace Quiz - 3 Questions

What does the test verify after taking the element screenshot?

A That the entire page was saved as a screenshot

B That the element is clickable

C That the screenshot file was created successfully

D That the element is visible on the screen

Key Result

Always verify that the element you want to interact with exists before performing actions like taking screenshots. This prevents test failures and helps diagnose issues quickly.

Practice

1. What does the element.screenshot('file.png') method do in Selenium Python?

easy

A. Captures a screenshot of only the specific web element.

B. Captures a screenshot of the entire browser window.

C. Saves the HTML source of the element to a file.

D. Highlights the element on the page without saving an image.

Solution

1. Step 1: Understand the method purpose

   The element.screenshot() method captures an image of the specific element, not the whole page.

2. Step 2: Compare with other screenshot methods

   driver.save_screenshot() captures the full page, but element.screenshot() targets only one element.

3. Final Answer:

   Captures a screenshot of only the specific web element. -> Option A

4. Quick Check:

   element.screenshot() = element image capture [OK]

Hint: Remember: element.screenshot saves only that element's image [OK]

Common Mistakes:

• Confusing element screenshot with full page screenshot
• Thinking it saves HTML instead of image
• Using driver.screenshot instead of element.screenshot

2. Which of the following is the correct syntax to save a screenshot of a web element stored in variable button as 'button.png'?

easy

A. driver.screenshot('button.png')

B. button.save_screenshot('button.png')

C. button.screenshot('button.png')

D. button.capture('button.png')

Solution

1. Step 1: Identify the correct method for element screenshot

   The method to save an element's screenshot is element.screenshot(filename).

2. Step 2: Check the variable and filename usage

   Since the element is stored in button, calling button.screenshot('button.png') is correct syntax.

3. Final Answer:

   button.screenshot('button.png') -> Option C

4. Quick Check:

   element.screenshot(filename) = correct syntax [OK]

Hint: Use element.screenshot('file.png') exactly to save element image [OK]

Common Mistakes:

• Using driver.screenshot() for element instead of element.screenshot()
• Using non-existent methods like save_screenshot on element
• Wrong method name like capture()

3. What will happen if you run this code snippet?

element = driver.find_element(By.ID, 'logo')
element.screenshot('logo.png')
print('Screenshot saved')

medium

A. Saves a screenshot of the whole page and prints confirmation.

B. Saves a screenshot of the element with ID 'logo' and prints confirmation.

C. Throws an error because screenshot method is not valid for elements.

D. Prints 'Screenshot saved' but does not save any image.

Solution

1. Step 1: Locate element and call screenshot

   The code finds the element with ID 'logo' and calls element.screenshot('logo.png'), which saves the element's image.

2. Step 2: Confirm output message

   After saving, it prints 'Screenshot saved' confirming the action.

3. Final Answer:

   Saves a screenshot of the element with ID 'logo' and prints confirmation. -> Option B

4. Quick Check:

   element.screenshot() saves image + print confirms [OK]

Hint: element.screenshot saves image; print confirms success [OK]

Common Mistakes:

• Assuming screenshot method is invalid for elements
• Thinking it saves full page instead of element
• Ignoring the print statement output

4. You wrote this code but no screenshot file is created. What is the likely error?

element = driver.find_element(By.CLASS_NAME, 'submit-btn')
element.screenshot()

medium

A. File path is invalid but filename is provided.

B. Incorrect locator type used for find_element.

C. screenshot() method does not exist for elements.

D. Missing filename argument in screenshot() method.

Solution

1. Step 1: Check screenshot method usage

   The screenshot() method requires a filename argument to save the image.

2. Step 2: Identify missing filename

   Since no filename is given, no file is created, causing the issue.

3. Final Answer:

   Missing filename argument in screenshot() method. -> Option D

4. Quick Check:

   screenshot() needs filename to save file [OK]

Hint: Always provide filename in element.screenshot('file.png') [OK]

Common Mistakes:

• Calling screenshot() without filename argument
• Assuming locator type is wrong without checking
• Thinking screenshot() method is invalid for elements

5. You want to verify visually that a specific button is rendered correctly on different browsers by saving its screenshot. Which approach is best to automate this using Selenium Python?

hard

A. Locate the button element and use element.screenshot('button.png') to save its image for comparison.

B. Use driver.save_screenshot('button.png') to capture the whole page and crop manually later.

C. Extract the button's HTML and compare it as text across browsers.

D. Use JavaScript to highlight the button but do not save any image.

Solution

1. Step 1: Identify the goal - visual verification of a button

   To check how the button looks, capturing its image directly is best.

2. Step 2: Choose the method that captures only the button

   element.screenshot() saves the button's image exactly, making comparison easy and automated.

3. Final Answer:

   Locate the button element and use element.screenshot('button.png') to save its image for comparison. -> Option A

4. Quick Check:

   element.screenshot() = best for visual element checks [OK]

Hint: Use element.screenshot() for focused visual testing [OK]

Common Mistakes:

• Capturing full page screenshot and cropping manually
• Comparing HTML text instead of visual appearance
• Not saving any image for visual verification