selenium+python,如何判断<a>标签类按钮,是否被选中

1.页面:

img

img

2.信息:
页面中,选中的按钮,class="medical_form_choose_current",未选中时 class无值。

3.需求:
selenium+python如何实现,选择其中一个按钮后,判断该按钮是否已经被选中。

示例 :

driver.find_element_by_xpath('//*[@id="predict-tab"]')..get_attribute("class")

.get_attribute("class") 这个方法可以获取标签属性,拿到之后可以去判断class的值