Запуск тестов в зависимости от результата предыдущей функции Selenium Python

Подскажие, как реализовать такой процесс запуска тестов (Pyton, Selenium Webdriver, использую pytest): есть функция, которая смотрит, какой элемент расположен на странице: если элемент А, то после этой функции должна запускаться функция (или тест), проверяющий именно элемент А, если же на сранице элемент В, тогда должна запускать другая функция(или тест), проверяющий именно элемент В. Заранее спасибо за помощь!