How to use WebDriverFactory file in test_suit.py

[ajay kumar]

from selenium import webdriver
from pathlib import Path
import os


level_up = Path(__file__).parents[1]
driver_dir = os.path.join(level_up, 'driver')

baseURL = "http://the-internet.herokuapp.com/"

class WebDriverFactory():


    def __init__(self, browser):
        self.browser = browser


    def getWebDriverInstance(self):

        if self.browser == "iexplorer":
            driver = webdriver.Ie()

        elif self.browser == "firefox":
            driver = webdriver.Firefox(executable_path = driver_dir + "\\geckodriver.exe")

        elif self.browser == "chrome":
            chromedriver = driver_dir + "\\chromedriver.exe"
            os.environ["webdriver.chrome.driver"] = chromedriver
            driver = webdriver.Chrome(chromedriver)
            driver.set_window_size(1440, 900)
        else:
            driver = webdriver.Firefox(executable_path = driver_dir + "\\geckodriver.exe")
        # Setting Driver Implicit Time out for An Element
        driver.implicitly_wait(3)
        # Maximize the window
        driver.maximize_window()
        # Loading browser with App URL
        driver.get(baseURL)
        return driver

here is my WebDriverFatory code please help me out!