Skip to content

Commit 142e3ab

Browse files
committed
Add headless firefox test
1 parent 93e3cdc commit 142e3ab

File tree

1 file changed

+16
-5
lines changed

1 file changed

+16
-5
lines changed

q2_feature_table/tests/test_summarize.py

Lines changed: 16 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,8 @@
1818
import pandas as pd
1919
import numpy as np
2020
from selenium import webdriver
21-
from selenium.webdriver.chrome.options import Options
21+
from selenium.webdriver.chrome.options import Options as ChromeOptions
22+
from selenium.webdriver.firefox.options import Options as FirefoxOptions
2223
from selenium.webdriver.common.by import By
2324

2425
import qiime2
@@ -512,11 +513,21 @@ def test_vega_spec_nandling(self):
512513

513514
self.assertEqual(spec['data'][0]['values'], exp)
514515

515-
def test_summarize_viz(self):
516-
chrome_options = Options()
517-
chrome_options.add_argument("--headless=new")
518-
driver = webdriver.Chrome(options=chrome_options)
516+
def test_summarize_viz_chrome(self):
517+
chrome_options = ChromeOptions()
518+
chrome_options.add_argument("-headless")
519519

520+
with webdriver.Chrome(options=chrome_options) as driver:
521+
self._selenium_test(driver)
522+
523+
def test_summarize_viz_firefox(self):
524+
firefox_options = FirefoxOptions()
525+
firefox_options.add_argument("-headless")
526+
527+
with webdriver.Firefox(options=firefox_options) as driver:
528+
self._selenium_test(driver)
529+
530+
def _selenium_test(self, driver):
520531
table = biom.Table(np.array([[0, 1, 3],
521532
[1, 1, 2],
522533
[400, 450, 500],

0 commit comments

Comments
 (0)