unit_test.py: Allow importing tiny-test-fw from IDF_PATH as well as TEST_FW_PATH

This commit is contained in:
Angus Gratton 2018-12-05 11:13:46 +11:00 committed by Angus Gratton
parent d8b625a32c
commit 0a27cfa850

View file

@ -25,13 +25,22 @@ import time
import argparse
import threading
# if we want to run test case outside `tiny-test-fw` folder,
# we need to insert tiny-test-fw path into sys path
test_fw_path = os.getenv("TEST_FW_PATH")
if test_fw_path and test_fw_path not in sys.path:
sys.path.insert(0, test_fw_path)
try:
import TinyFW
except ImportError:
# if we want to run test case outside `tiny-test-fw` folder,
# we need to insert tiny-test-fw path into sys path
test_fw_path = os.getenv("TEST_FW_PATH")
if test_fw_path and test_fw_path not in sys.path:
sys.path.insert(0, test_fw_path)
else:
# or try the copy in IDF
idf_path = os.getenv("IDF_PATH")
tiny_test_path = idf_path + "/tools/tiny-test-fw"
if os.path.exists(tiny_test_path):
sys.path.insert(0, tiny_test_path)
import TinyFW
import TinyFW
import IDF
import Utility
import Env