import pytest
from wasd.wd import Browser


# Define custom action here
class MyExtendedBrowser(Browser):
    def my_super_fn(self):
        print('Hello, World!')


@pytest.fixture(scope='session')
def _browser():
    browser = MyExtendedBrowser()
    yield browser
    browser.close_driver()
