Conditional compilation test can run from open source (#5629)

tests/conditional_compilation/test_utils.py

@@ -0,0 +1,88 @@
+#!/usr/bin/env python3
+# Copyright (C) 2021 Intel Corporation
+# SPDX-License-Identifier: Apache-2.0
+
+""" Utility functions for work with json test configuration file.
+"""
+import os
+import json
+import multiprocessing
+import sys
+from inspect import getsourcefile
+from pathlib import Path
+
+from install_pkg import get_openvino_environment  # pylint: disable=import-error
+from path_utils import get_lib_path  # pylint: disable=import-error
+from proc_utils import cmd_exec  # pylint: disable=import-error
+
+
+SESSION_INFO_FILE = "cc_tests.json"
+infer_tool = str((Path(getsourcefile(lambda: 0)) / ".." / "tools" / "infer_tool.py").resolve())
+
+
+def validate_path_arg(path: Path, is_dir=False):
+    """"Check if path argument is correct."""
+    if not path.exists():
+        raise ValueError(f"{path} does not exist.")
+
+    if is_dir and not path.is_dir():
+        raise ValueError(f"{path} is not a directory.")
+
+    return path
+
+
+def get_lib_sizes(path, libraries):
+    """ Function for getting lib sizes by lib names
+    """
+    assert Path.exists(path), f'Directory {path} isn\'t created'
+    result = {}
+    error_lib = []
+    for lib in libraries:
+        try:
+            result[lib] = Path(path).joinpath(get_lib_path(lib)).stat().st_size
+        except FileNotFoundError as error:
+            error_lib.append(str(error))
+    assert len(error_lib) == 0, 'Following libraries couldn\'t be found: \n{}'.format(
+        '\n'.join(error_lib))
+    return result
+
+
+def read_session_info(path: Path = Path(getsourcefile(lambda: 0)).parent / SESSION_INFO_FILE):
+    with open(path, 'r') as json_file:
+        cc_tests_ids = json.load(json_file)
+    return cc_tests_ids
+
+
+def write_session_info(path: Path = Path(getsourcefile(lambda: 0)).parent / SESSION_INFO_FILE,
+                       data: dict = None):
+    with open(path, "w") as json_file:
+        json.dump(data, json_file, indent=4)
+
+
+def run_infer(model, out_file, install_dir):
+    """ Function running inference
+    """
+    sys_executable = os.path.join(sys.prefix, 'python.exe') if sys.platform == "win32" \
+        else os.path.join(sys.prefix, 'bin', 'python')
+    return_code, output = cmd_exec(
+        [sys_executable,
+         infer_tool,
+         "-d=CPU", f"-m={model}", f"-r={out_file}"
+         ],
+        env=get_openvino_environment(install_dir),
+    )
+    return return_code, output
+
+
+def make_build(openvino_root_dir, build_dir, install_dir, cmake_additional_args=None, log=None):
+    """Parametrized build and install OpenVINO package."""
+    additional_args_line = " ".join(cmake_additional_args) + " " if cmake_additional_args else ""
+    nproc = multiprocessing.cpu_count()
+    cmd = (
+        f"cmake -DENABLE_PROFILING_ITT=ON -DCMAKE_BUILD_TYPE=Release "
+        f"-DPYTHON_EXECUTABLE={sys.executable} {additional_args_line}"
+        f"-S {openvino_root_dir} -B {build_dir} &&"
+        f"cmake --build {build_dir} -j{nproc} && "
+        f"cmake --install {build_dir} --prefix {install_dir}"
+    )
+    return cmd_exec([cmd], shell=True, log=log)