python selenium unitest框架

Python中的selenium unittest框架结合了Selenium WebDriver和Python的unittest测试框架，可以用于自动化测试Web应用程序。以下是一个示例代码：

import unittest
from selenium import webdriver

class Test(unittest.TestCase):
    def setUp(self):
        self.driver = webdriver.Chrome()

    def test_search(self):
        self.driver.get("https://www.google.com/")
        self.driver.find_element_by_name("q").send_keys("selenium")
        self.driver.find_element_by_name("btnK").click()
        assert "selenium" in self.driver.page_source

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

if __name__ == "__main__":
    unittest.main()

在这个示例中，我们使用了Chrome浏览器和Google搜索引擎。setUp()方法用于在每个测试之前启动浏览器。test_search()方法模拟了在Google搜索栏中输入“selenium”并单击搜索按钮的操作，并断言“selenium”是否在搜索结果页面中。tearDown()方法在每个测试之后关闭浏览器。最后，我们使用unittest.main()方法运行测试。