Need advice on how to refactor my driver instantiation to work with device farm

[alex-osuna]

I am currently using the following architecture to instantiate drivers and pass them to the tests. This allows me to run tests in parallel with 6 devices. I am looking for a way to handle the tests when only 5 devices or less are available. I think device farm fits the bill, but I am looking for ideas on how to refactor the driver instantiation to make it work. I appreciate any advice

My conftest.py looks like this:
The NewDriver() class which is in a separate file has the appium driver implementation. I am using this implementation https://github.com/testdevlab/Py-TestUI

def appium_driver():
    driver_1 = NewDriver() \
        .set_app_path(...).set_device_udid(0).set_appium_port(4725)...
    yield driver_1
    driver_1.get_driver().set_network_connection(6)
    driver_1.quit()
    
def appium_driver_2():
    driver_1 = NewDriver() \
        .set_app_path(...).set_device_udid(1).set_appium_port(4730)...
    yield driver_2
    driver_2.get_driver().set_network_connection(6)
    driver_2.quit()

def appium_driver_3()
etc

I pass those drivers to the tests:

class Test1():
def mytest(appium_driver, appium_driver_2):
...
class Test2():
def anothertest(appium_driver_3, appium_driver_4)
...etc