（python）Selenium - 多浏览器运行

from selenium import webdriver
import os

class Base:
    def setup(self):
        browser = os.getenv(‘browser‘)
        if browser == ‘firefox‘: # 调用firefox浏览器
            self.driver =webdriver.Firefox()
        elif browser == ‘headless‘ :
            self.driver = webdriver.PhantomJS()
        else:
            self.driver = webdriver.Chrome()
            self.driver.maximize_window() # 窗口最大化
            self.driver.implicitly_wait(3) # 隐式等待3秒

def teardown(self):
    self.driver.quit()