from selenium.common.exceptions import NoSuchElementException try: if driver.find_element_by_class_name(test_element_class_name).is_displayed(): print('found') except NoSuchElementException: pass