Вот код, который я хочу поместить в базовую библиотеку, и я хочу вернуть значение, чтобы я мог использовать этот код в другом классе
@pytest.fixture(params=["chrome", "firefox"])
def browser(request):
if request.param == "chrome":
try:
driver = webdriver.Chrome(executable_path=r"/home/sunny/PycharmProjects/atomate/drivers/chromedriver")
except:
driver = webdriver.Chrome(executable_path=r"/Users/office/Documents/atomate/drivers/chromedriver mac")
elif request.param == "firefox":
try:
driver = webdriver.Firefox(
executable_path=r"/home/sunny/PycharmProjects/atomate/drivers/geckodriver")
except:
driver = webdriver.Firefox(executable_path=r"/Users/office/Documents/atomate/drivers/geckodriver mac")
# Initialize ChromeDriver
# driver = webdriver.Chrome(executable_path=r"/Users/office/Desktop/atomate/drivers/chromedriver mac")
# Wait implicitly for elements to be ready before attempting interactions
request.driver = driver
driver.implicitly_wait(10)
driver.maximize_window()
# Return the driver object at the end of setup
yield driver
# For cleanup, quit the driver
driver.quit()
Как создать Base Lib этого устройства и использовать каждый раз в тестовом сценарии.