есть вот такое решение через декоратор:
   def decorator_screenshot(func):
       def wrapper(*args, **kwargs):
           try:
              func(*args, **kwargs)
           except AssertionError:
               file_name = f'{datetime.today().strftime("%Y-%m-%d_%H:%M")}.png'.replace("/", "_").replace("::", "__")
               driver.save_screenshot(file_name)
               raise
       return wrapper

Получается просто добавляю ещё 1 except?

Если знаете тип ошибки - да

если не знаете - на любой жксепшн снимайте экран и пишите в лог

def assert_wrapper(func):
   def func_wrapper(*args, **kwargs):
       try:
           return func(*args, **kwargs)
       except AssertionError as ae:
           logger.error(f"Assertion failed: {ae}")
           raise
       except Exception as e:
           logger.exception(str(e))
           logger.error(f"Error: {e}")
           raise

   update_wrapper(func_wrapper, func)
   return func_wrapper
типа такого, добавьте соотв снятие скриншота

и соотв в конфтеест
def pytest_runtest_call(item):
   item._obj = assert_wrapper(item._obj)

я как-то делал несколько лет назад через class EventListeners(AbstractEventListener):

что-то вроде такого

class ScreenshotListener(AbstractEventListener):
    def on_exception(self, exception, driver):
        driver.get_screenshot_as_file("C:/Error.png")

class ScreenshotListener(AbstractEventListener):
    def on_exception(self, exception, driver):
        driver.get_screenshot_as_file("C:/Error.png")

это если ловить строго селениум эксепшены. Враппер выше ловит и логирует любые.

Всем привет
Новичок в автоматизация
Пишу тест на python unittest
У меня есть стек кейсов которые я должен выполнить
Но я не могу этого сделать так при запуске теста у меня шаги повторяються(так как наследую класс)
Вопрос как мне сделать что бы разбить test suit на шаги, что бы присутствовали анотации перед каждой функции?

Я угадаю страну с трёх попыток

Первая К?

Не понял

Что за стек кейсов? Какой тест сьют ты хочешь разбить? Что такое шаг в твоём понимании?

Что ты вы очень много впихнули в одно действие.
Давайте по порядку:
1. У вас уже готовые тестовые случаи для автоматизации? Или у вас есть тестовые случаи для ручного тестирования и их вы пытаетесь перевести в автоматизацию? Или у вас нет ничего и вам нужно "написать автотесты"?

2. Разбиение шагов в тестовом случае это задача не тривиальная. Я обычно делаю два шага: выполнить действие, начать проверку. Безусловно перед этим множество фикстур, состояний и прочего.

3. Аннотации перед каждой функцией? Что для вас аннотация? У вас есть репортинг (например, Allure)? Если так, то либо через контекстный менеджер allure.step(""), либо оборачивать  декоратором. Первый вариант - просто описание, второй - описание + входные параметры в функцию

4. Ну и напоследок, если нет четкого понимания как строить тесты и как разбивать, то:
- выполните ручное тестирование и подробно запишите шаги;
- унифицируйте шаги (общие блоки можно вынести)
- напишите шаблон тестового модуля, где через заглушки (можно просто принтами) выобощначаете порядок действий;
- напишите полноценный тестовый модуль

как то так.

Пардон за цитирование годовой давности. Но путь был выбран схожий. Сейчас есть желание переписать большую часть. Как правильнее сделать - менять потихоньку имеющийся или написать на его основе новый?

Нет универсального ответа.

не факт что новый получится лучше и чище

есть эталон к чему стоит стремиться?

Нет

Пустая директория - не эталон?

Эталон автоматизатора - дворник Петрович, потому что он не написал ни единой строки кода