selenium的官方文档参考

2. Getting Started — Selenium Python Bindings 2 documentationhttps://selenium-python.readthedocs.io/getting-started.html

用例的关键要素

导入依赖、创建driver、执行自动化步骤、断言

#导入依赖
from selenium import  webdriver


def test_search():
   #创建driver
    driver = webdriver.Chrome()
   #执行自动化测试步骤
    driver.get("http://www.baidu.com")
    search=driver.find_element_by_id("su").get_attribute("value")
   #断言
    assert  search=="百度一下"

下面是上段代码的优化版本

from  selenium import  webdriver
from selenium.webdriver.common.by import By
from time import sleep


class Test1:
    def setup_method(self,method):
        self.driver=webdriver.Chrome()
        #窗口最大化
        self.driver.maximize_window()
        #隐式等待5秒 就是会等到五秒找元素，如果2秒就找到了，他就不在等待5秒了
        self.driver.implicitly_wait(5)
       
    def teardown_method(self,method):
       #方法运行完毕后，关掉浏览器，回收资源
        self.driver.quit()

    def test_search(self):
        self.driver.get("https://www.baidu.com/")
        #强制等待5秒：不管是否找到元素，在此等到五秒，该方式不推荐
        sleep(5)
        #方法内设置窗口大小，上方设置的就不用了
        self.driver.set_window_size(1218, 1026)
        search = self.driver.find_element_by_id("su").get_attribute("value")
        assert search == "百度一下"