A Beginner's Guide to Writing Your First Selenium Python Web Test

In this post, we’ll take a look at how to perform software test automation using Selenium Python web bindings. We will write a script that “opens a web page, searches for “pycon”, and checks that search results are returned.” If you’re new to Selenium or Python, don’t worry! We’ll start with the basics and guide you through your first Selenium web test.

First, let’s make sure that you have the Selenium Python bindings installed. If not, please refer to our previous post on how to install them.

Once you have the necessary software installed, you can begin writing your first Selenium Python web test using your favorite Python IDE. Here’s a simple example script:


from selenium import webdriver
from selenium.webdriver.common.keys import Keys

driver = webdriver.Firefox()
driver.get("http://www.python.org")
assert "Python" in driver.title
elem = driver.find_element_by_name("q")
elem.clear()
elem.send_keys("pycon")
elem.send_keys(Keys.RETURN)
assert "No results found." not in driver.page_source
driver.close()

This code snippet is collected from official Selenium Python Document and The above script can be saved into a file (eg:- python_org_search.py), then it can be run like this from your terminal:

python python_org_search.py

Example Explained

Let’s go over the script line by line to understand what’s going on. We start by importing the necessary modules from Selenium:

from selenium import webdriver
from selenium.webdriver.common.keys import Keys

The selenium.webdriver module provides all the WebDriver implementations. Currently supported WebDriver implementations are Firefox, Chrome, IE and Remote. The Keys class provide keys in the keyboard like RETURN, F1, ALT etc.


from selenium import webdriver
from selenium.webdriver.common.keys import Keys

driver = webdriver.Firefox()

You can change the browser to Chrome, IE, or Remote if needed. The driver.get() method will navigate to a page specified by the URL. WebDriver will wait until the page has fully loaded before returning control to your script:

driver.get("http://www.python.org")

The next line is an assertion that confirms that the page title contains the word “Python”:

assert "Python" in driver.title

WebDriver offers a number of ways to find elements using one of the find_element_by_* methods. For example, the input text element can be located by its name attribute using find_element_by_name method.


elem = driver.find_element_by_name("q")

Next, we are sending keys, this is similar to entering keys using your keyboard. Special keys can be sent using Keys class imported from selenium.webdriver.common.keys.

To be safe, we’ll first clear any pre-populated text in the input field (e.g. “Search”) so it doesn’t affect our search results:


elem.clear()
elem.send_keys("pycon")
elem.send_keys(Keys.RETURN)

After submission of the page, you should get the result if there is any. To ensure that some results are found, make an assertion:


assert "No results found." not in driver.page_source

Finally, the browser window is closed. You can also call quit method instead of close. The quit will exit entire browser whereas close will close one tab, but if just one tab was open, by default most browser will exit entirely


driver.close()

To summarize, we’ve just written a simple script that opens a web page, searches for “pycon”, and checks that search results are returned.

If you want to learn more about Selenium Python, check out the Selenium Python repository on GitHub, or take a look at our Selenium Python crash course.

Selenium Python

Selenium Python Crash Course Source Code

About Content Creator:

A Seasoned Business Systems Analyst, Project Manager, and SQA Engineer with experience in driving digital changes within organizations. I specialize in understanding business needs and developing software solutions to improve processes and drive growth. I am skilled in managing projects, analyzing data, and ensuring quality in the final product. I am passionate about using my expertise to help organizations reach their goals and succeed. Let’s work together to improve your business and drive success.