Selenium basic test using headless firefox and arbitrary binary and driver

selenium_test_firefox.py

from selenium import webdriver
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.firefox.options import Options
from selenium.webdriver.firefox.firefox_binary import FirefoxBinary

# https://askubuntu.com/questions/870530/how-to-install-geckodriver-in-ubuntu
# https://stackoverflow.com/questions/25713824/setting-path-to-firefox-binary-on-windows-with-selenium-webdriver
# http://selenium-python.readthedocs.io/getting-started.html

binary = FirefoxBinary(PATH_TO_BINARY)
options = Options()
options.set_headless(headless=True)
driver = webdriver.Firefox(
    firefox_binary=binary,
    executable_path=r'PATH_TO_DRIVER',
    firefox_options=options)

print("Headless Firefox Initialized")
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.quit()