mvt/tests/utils.py

24 lines
539 B
Python

import os
def get_artifact(fname):
"""
Return the artifact path in the artifact folder
"""
DATA_FOLDER = os.path.join(os.path.dirname(__file__), "artifacts")
fpath = os.path.join(DATA_FOLDER, fname)
if os.path.isfile(fpath):
return fpath
return
def get_artifact_folder():
return os.path.join(os.path.dirname(__file__), "artifacts")
def init_setup():
"""
init data to have a clean state before testing
"""
try:
del os.environ['MVT_STIX2']
except KeyError:
pass