Add time_tests dir path to sys.path (openvinotoolkit#5498)

vixadd · May 6, 2021 · 4790c79 · 4790c79
1 parent a411af1
commit 4790c79
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 1 deletion.
diff --git a/tests/time_tests/README.md b/tests/time_tests/README.md
@@ -34,6 +34,5 @@ cmake .. -DInferenceEngineDeveloperPackage_DIR=$(realpath ../../../build) && mak
 
 2. Run several configurations using `pytest`:
 ``` bash
-export PYTHONPATH=./:$PYTHONPATH
 pytest ./test_runner/test_timetest.py --exe ../../bin/intel64/Release/timetest_infer
 ```
diff --git a/tests/time_tests/scripts/run_timetest.py b/tests/time_tests/scripts/run_timetest.py
@@ -22,6 +22,9 @@
 from pathlib import Path
 from pprint import pprint
 
+TIME_TESTS_DIR = os.path.dirname(os.path.dirname(__file__))
+sys.path.append(TIME_TESTS_DIR)
+
 from test_runner.utils import filter_timetest_result
 
 

diff --git a/tests/time_tests/test_runner/conftest.py b/tests/time_tests/test_runner/conftest.py
@@ -29,6 +29,9 @@
 from pathlib import Path
 from jsonschema import validate, ValidationError
 
+TIME_TESTS_DIR = os.path.dirname(os.path.dirname(__file__))
+sys.path.append(TIME_TESTS_DIR)
+
 from scripts.run_timetest import check_positive_int
 from test_runner.utils import upload_timetest_data, metadata_from_manifest, get_os_name, get_os_version, \
     get_cpu_info, DATABASE, DB_COLLECTIONS