Python Automation Tutorial: A Comprehensive Guide With Examples and Best Practices

Writing Your First Python Automation Test

The fastest way to learn Python test automation is to run a real test. The example below uses pytest and Selenium to open the Selenium Playground and assert the page state. Install the dependencies with pip install pytest selenium, save the file as test_first.py, then run it with pytest test_first.py.

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

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

def test_playground_title(driver):
    driver.get("https://www.testmuai.com/selenium-playground/")
    assert "Selenium Playground" in driver.title
    driver.find_element(By.LINK_TEXT, "Simple Form Demo").click()
    assert "simple-form-demo" in driver.current_url

The test above runs on one local browser. To run the same script across real browsers and operating systems, point the WebDriver at the TestMu AI cloud grid with LT:Options. This is the Selenium 4 pattern for turning a single script into cross-browser coverage.

from selenium import webdriver

options = webdriver.ChromeOptions()
options.set_capability("browserName", "Chrome")
options.set_capability("browserVersion", "latest")
options.set_capability("LT:Options", {
    "platform": "Windows 11",
    "build": "Python Automation Tutorial",
    "name": "First cloud test",
    "user": "YOUR_USERNAME",
    "accessKey": "YOUR_ACCESS_KEY",
})

driver = webdriver.Remote(
    command_executor="https://hub.lambdatest.com/wd/hub",
    options=options,
)
driver.get("https://www.testmuai.com/selenium-playground/")
print(driver.title)
driver.quit()

Best Practices for Python Test Automation

• Prefer explicit waits over time.sleep(): Wait for a specific element condition with WebDriverWait so tests stay fast and stop failing on timing differences between browsers.
• Use the Page Object Model: Keep locators and page actions in dedicated classes so a UI change is fixed in one place instead of across every test.
• Externalize test data and credentials: Read URLs, logins, and grid keys from config files or environment variables rather than hardcoding them in the test body.
• Parameterize with pytest: Drive one test function with many inputs using @pytest.mark.parametrize instead of copy-pasting near-identical tests.
• Run in parallel on the cloud: Split the suite across browser combinations with pytest-xdist and a cloud grid to cut total run time and widen coverage.