How to use remove_output_dir method in SeleniumLibrary

1#!/usr/bin/env python
2
3import os
4import shutil
5import sys
6from os.path import abspath, dirname, join
7
8from pytest import main as py_main
9
10
11CURDIR = dirname(abspath(__file__))
12SRC = join(CURDIR, os.pardir, 'src')
13
14
15def remove_output_dir():
16    output_dir = os.path.join(CURDIR, 'output_dir')
17    if os.path.exists(output_dir):
18        shutil.rmtree(output_dir)
19    os.mkdir(output_dir)
20
21
22def run_unit_tests():
23    sys.path.insert(0, SRC)
24    try:
25        result = py_main(['--rootdir=%s' % CURDIR, '-p', 'no:cacheprovider', CURDIR])
26    finally:
27        sys.path.pop(0)
28    return result
29
30
31if __name__ == '__main__':
32    remove_output_dir()
33    sys.exit(run_unit_tests())
34

Source: run.py

#!/usr/bin/env python

import os
import shutil
import sys
from os.path import abspath, dirname, join

from pytest import main as py_main


CURDIR = dirname(abspath(__file__))
SRC = join(CURDIR, os.pardir, 'src')


def remove_output_dir():
    output_dir = os.path.join(CURDIR, 'output_dir')
    if os.path.exists(output_dir):
        shutil.rmtree(output_dir)
    os.mkdir(output_dir)


def run_unit_tests():
    sys.path.insert(0, SRC)
    try:
        result = py_main(['--rootdir=%s' % CURDIR, '-p', 'no:cacheprovider', CURDIR])
    finally:
        sys.path.pop(0)
    return result


if __name__ == '__main__':
    remove_output_dir()
    sys.exit(run_unit_tests())