如何在selenium和pytest中访问日期选择器(python)

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
import pytest

@pytest.mark.parametrize("date", ["2022-01-01", "2022-02-01", "2022-03-01"])
def test_date_picker(date):
    # 打开浏览器
    driver = webdriver.Chrome()
    driver.get("https://example.com")  # 替换为实际的日期选择器所在的网页地址

    # 等待日期选择器加载完成
    date_picker = WebDriverWait(driver, 10).until(
        EC.presence_of_element_located((By.ID, "date-picker"))  # 替换为实际的日期选择器元素定位方式和值
    )

    # 清除日期输入框的内容
    date_input = driver.find_element(By.ID, "date-input")  # 替换为实际的日期输入框元素定位方式和值
    date_input.clear()

    # 输入日期
    date_input.send_keys(date)

    # 点击确认按钮
    confirm_button = driver.find_element(By.ID, "confirm-button")  # 替换为实际的确认按钮元素定位方式和值
    confirm_button.click()

    # 断言日期是否正确选择
    assert date_input.get_attribute("value") == date

    # 关闭浏览器
    driver.quit()

pytest test_example.py