refactor and prettify tests
driver to self.driver in unittest module