在Python中使用Selenium创建类

from selenium import webdriver

class SeleniumTest:
    def __init__(self):
        self.driver = webdriver.Chrome()  # 使用Chrome浏览器驱动

    def open_url(self, url):
        self.driver.get(url)

    def click_element(self, locator):
        element = self.driver.find_element(*locator)
        element.click()

    def input_text(self, locator, text):
        element = self.driver.find_element(*locator)
        element.clear()
        element.send_keys(text)

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

test = SeleniumTest()
test.open_url("https://www.example.com")
test.click_element(("id", "submit-button"))
test.input_text(("name", "username"), "testuser")
test.close_browser()