如何在selenium中检查按钮的类是否发生了更改？

from selenium import webdriver

# 启动浏览器
driver = webdriver.Chrome()

# 打开目标网页
driver.get("https://example.com")

# 找到按钮元素
button = driver.find_element_by_id("button-id")

# 获取按钮的类属性值
initial_class = button.get_attribute("class")

# 执行导致按钮类发生更改的操作
# 例如点击其他按钮或执行其他操作

# 再次获取按钮的类属性值
updated_class = button.get_attribute("class")

# 比较初始类和更新后的类
if initial_class == updated_class:
    print("按钮的类未发生更改")
else:
    print("按钮的类发生了更改")

# 关闭浏览器
driver.quit()