from selenium.common.exceptions import TimeoutException
from selenium.webdriver.support.expected_conditions import presence_of_element_located
from selenium.webdriver.support.wait import WebDriverWait
def wait_el(driver, locator):
try:
WebDriverWait(driver, 40).until(
presence_of_element_located(locator),
message='Не дождались появления элемента'
)
return False
except TimeoutException:
return True
ну и как вариант, можно через хук pytest_collection_modifyitems также навешивать маркировки