Refactor python test mn_options_test.py

SeleniumHQ · Oct 21, 2016 · 702f1a9 · 702f1a9
1 parent fff22cd
commit 702f1a9
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/py/test/selenium/webdriver/marionette/mn_options_tests.py b/py/test/selenium/webdriver/marionette/mn_options_tests.py
@@ -25,14 +25,12 @@
 from selenium.webdriver.firefox.firefox_binary import FirefoxBinary
 from selenium.webdriver.firefox.firefox_profile import FirefoxProfile
 from selenium.webdriver.firefox.options import Log, Options
-from selenium.webdriver import Firefox
 
 
 @pytest.fixture
-def driver(capabilities):
-    driver = Firefox(firefox_options=Options(), capabilities=capabilities)
-    yield driver
-    driver.quit()
+def driver_kwargs(driver_kwargs):
+    driver_kwargs['firefox_options'] = Options()
+    return driver_kwargs
 
 
 class TestIntegration(object):