Integrating Selenium Code with PyTest