alex/esp-idf
1
0
Fork 0
mirror of https://github.com/espressif/esp-idf.git synced 2025-08-10 04:43:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

ttfw: allow relative imports in test scripts

Browse Source
This commit is contained in:
Ivan Grokhotkov
2020-05-02 01:02:07 +02:00
parent 7d74cc4255
commit 3fe4adc98d
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tools/ci/python_packages/tiny_test_fw/Utility/__init__.py
View File

@@ -61,11 +61,15 @@ def load_source(path):
return __LOADED_MODULES[path] return __LOADED_MODULES[path]
except KeyError: except KeyError:
try: try:
dir = os.path.dirname(path)
sys.path.append(dir)
from importlib.machinery import SourceFileLoader from importlib.machinery import SourceFileLoader
ret = SourceFileLoader(load_name, path).load_module() ret = SourceFileLoader(load_name, path).load_module()
except ImportError: except ImportError:
# importlib.machinery doesn't exists in Python 2 so we will use imp (deprecated in Python 3) # importlib.machinery doesn't exists in Python 2 so we will use imp (deprecated in Python 3)
import imp import imp
ret = imp.load_source(load_name, path) ret = imp.load_source(load_name, path)
finally:
sys.path.remove(dir)
__LOADED_MODULES[path] = ret __LOADED_MODULES[path] = ret
return ret return ret