from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.common.keys import Keys
import time

# 初始化 WebDriver
driver = webdriver.Chrome()
try:
    # 打开百度首页
    driver.get("https://www.baidu.com")

    # 查找搜索输入框
    search_box = driver.find_element(By.ID, "kw")

    # 输入搜索关键字
    search_box.send_keys("小瑞")

    # 模拟按下回车键
    search_box.send_keys(Keys.RETURN)

    # 等待几秒钟以便加载搜索结果
    time.sleep(5)

    # 检查搜索结果
    results = driver.find_elements(By.CSS_SELECTOR, 'h3.t a')
    for result in results:
        print(result.text)

    # 断言确保找到结果
    assert len(results) > 0, "没有找到搜索结果"

finally:
    # 关闭浏览器
    driver.quit()