set chrome options for webdriver

Author: emilykl

tests/test_callbacks.py

@@ -5,14 +5,13 @@
 from selenium.webdriver.common.by import By
 
 
-
-
 def test_callback(sample_app, start_app):
-
     # Start the app in another thread
     start_app(sample_app)
 
-    driver = webdriver.Chrome()
+    options = webdriver.ChromeOptions()
+    options.add_argument("--headless")
+    driver = webdriver.Chrome(options=options)
     driver.get("http://127.0.0.1:5000")
 
     # Set text in the input with id "input_sample"
@@ -21,8 +20,10 @@ def test_callback(sample_app, start_app):
     time.sleep(2)
 
     # Check that it is showing up as the graph title
-    graph = driver.find_element(By.ID, "graph-component-sample").find_element(By.CLASS_NAME, "g-gtitle")
+    graph = driver.find_element(By.ID, "graph-component-sample").find_element(
+        By.CLASS_NAME, "g-gtitle"
+    )
     assert graph.text == "Hello, world!!"
 
     # Close the driver
-    driver.quit()
+    driver.quit()