First test structure

This commit is contained in:
tek
2021-12-16 12:50:12 +01:00
committed by Donncha Ó Cearbhaill
parent 28d57e7178
commit 513e2cc704
16 changed files with 318 additions and 0 deletions
+23
View File
@@ -0,0 +1,23 @@
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