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

1.页面：

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的值