import os

from selenium import webdriver

class SeleniumTest:
    def __init__(self, src, drive_path):
        self.driver = webdriver.Chrome(executable_path=drive_path)

        self.driver.get(src)

    def __del__(self):
        self.driver.close()

    def assert_title(self, title):
        assert title in self.driver.title


if __name__ == '__main__':
    drive_path = "%s/chromedriver" % os.getcwd()

    st = SeleniumTest("https://python.org", drive_path)
    st.assert_title("Python")