Appium с Python - класс настройки для веб-драйвера

Question

Программирование для меня новость, и, возможно, я упускаю что-то очевидное.

Я хотел бы создать отдельный файл и класс для настройки моего веб-драйвера для тестов Appium, но я получил такие ошибки, как:

in test_login main_page = MainPage(self.driver)
AttributeError: 'test_Login_iOS' object has no attribute 'driver'

В настоящее время у меня есть два файла: один с тестовым набором и другой с методами для шагов теста:

1. test_Login_iOS.py:

   Класс test_Login_iOS (unittest.TestCase):

   def setUp (self):

   logging.info("WebDriver request initiated. Waiting for response, this may take a while.")
   
   # choose desired capabilities from desired_capabilities.py
   
   desired_capabilities = DesiredCapabilities.desired_capabilities_for_iOS_iPad
   
   self.driver = webdriver.Remote("http://localhost:4723/wd/hub", desired_capabilities)
   
   self.driver.implicitly_wait(15)  # seconds
   

   def test_login (self):

   logging.info("starting Test Case 1: login into active account")
   welcome_page = WelcomePage(self.driver)
   welcome_page.click_login_button()
   login_page = LoginPage(self.driver)
   

и второй файл, page_ios.py:

class BasePage(unittest.TestCase):
"""
:type driver: appium.webdriver.Remote
"""

def __init__(self, driver):

    super().__init__()
    self.driver = driver

Когда я добавляю новый тестовый пример, я должен добавить к нему тот же метод setUp, что и в предыдущем тестовом примере, поэтому я хотел бы создать новый класс "Setup", который можно было бы использовать в нескольких тестовых случаях.

Цель состоит в том, чтобы переместить метод setUp в отдельный файл и новый класс.