如何在Selenium中一次编写Assert方法并多次验证文本？

pip install selenium

from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC

def assert_text(driver, locator, expected_text, timeout=10):
    """
    自定义断言函数，用于验证页面元素的文本。
    
    :param driver: WebDriver实例
    :param locator: 元素定位器，例如(By.ID, 'element_id')
    :param expected_text: 预期的文本
    :param timeout: 等待元素出现的超时时间，默认为10秒
    """
    try:
        element = WebDriverWait(driver, timeout).until(
            EC.presence_of_element_located(locator)
        )
        assert element.text == expected_text, f"文本验证失败，预期文本为：{expected_text}，实际文本为：{element.text}"
        print("文本验证成功！")
    except AssertionError as e:
        print(e)
    except Exception as e:
        print(f"发生异常：{e}")

# 使用示例
driver = webdriver.Chrome()  # 假设你已经配置好了WebDriver
driver.get("http://example.com")

# 验证多个元素的文本
assert_text(driver, (By.ID, "element1"), "Expected Text 1")
assert_text(driver, (By.XPATH, "//div[@class='example']/p[2]"), "Expected Text 2")
assert_text(driver, (By.CSS_SELECTOR, ".another-class"), "Expected Text 3")

driver.quit()