From 3fe4adc98d3cc3bfde4b89f7b0f0ddd2ae5d1d5f Mon Sep 17 00:00:00 2001
From: Ivan Grokhotkov <ivan@espressif.com>
Date: Sat, 2 May 2020 01:02:07 +0200
Subject: [PATCH] ttfw: allow relative imports in test scripts

---
 tools/ci/python_packages/tiny_test_fw/Utility/__init__.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/tools/ci/python_packages/tiny_test_fw/Utility/__init__.py b/tools/ci/python_packages/tiny_test_fw/Utility/__init__.py
index 7fec71828..75b85f6a2 100644
--- a/tools/ci/python_packages/tiny_test_fw/Utility/__init__.py
+++ b/tools/ci/python_packages/tiny_test_fw/Utility/__init__.py
@@ -61,11 +61,15 @@ def load_source(path):
         return __LOADED_MODULES[path]
     except KeyError:
         try:
+            dir = os.path.dirname(path)
+            sys.path.append(dir)
             from importlib.machinery import SourceFileLoader
             ret = SourceFileLoader(load_name, path).load_module()
         except ImportError:
             # importlib.machinery doesn't exists in Python 2 so we will use imp (deprecated in Python 3)
             import imp
             ret = imp.load_source(load_name, path)
+        finally:
+            sys.path.remove(dir)
         __LOADED_MODULES[path] = ret
         return ret