Minimal testing script for the Selenium Python package

selenium.py

#!/usr/bin/python3

import os

from selenium import webdriver
from selenium.webdriver.chrome.options import Options

chrome_options = Options()
chrome_options.add_argument('--headless')
chrome_options.add_argument('--window-size=1920x1080')

# Download the chrome driver from
# https://sites.google.com/a/chromium.org/chromedriver/downloads
# and put it in the current directory:
chrome_driver = os.getcwd() + '/chromedriver'

driver = webdriver.Chrome(
    chrome_options=chrome_options,
    executable_path=chrome_driver
)

driver.get('https://www.google.com')

driver.get_screenshot_as_file('capture.png')

driver.close()
driver.quit()