scriptshifter/test/unittest/__init__.py

from importlib import reload
from os import path, environ
from tempfile import gettempdir

import scriptshifter
from scriptshifter import tables


UNITTEST_CONFIG_DIR = path.join(
        path.dirname(path.realpath(__file__)), "tables", "data")

# Reload main SS modules after changing environment variables.
environ["TXL_DB_PATH"] = path.join(gettempdir(), "scriptshifter_unittest.db")
reload(scriptshifter)
environ["TXL_CONFIG_TABLE_DIR"] = UNITTEST_CONFIG_DIR
reload(tables)