[TEST][PR5] Implementing test infra (#1797)

Details: * Remove several `rum_*_pipeline` * Replace several `run_*_pipeline_and_compare` by one method * Remove `common.py` Tickets: * [159925](https://jira.devtools.intel.com/browse/CVS-159925) MERGE AFTER: * #1691 * #1702 * #1718 * #1722
openvinotoolkit · Feb 27, 2025 · e94089e · e94089e
1 parent 4316379
commit e94089e
Show file tree

Hide file tree

Showing 8 changed files with 152 additions and 322 deletions.
diff --git a/tests/python_tests/common.py b/tests/python_tests/common.py
diff --git a/tests/python_tests/data/test_dataset.py b/tests/python_tests/data/test_dataset.py
@@ -0,0 +1,21 @@
+# Copyright (C) 2024 Intel Corporation
+# SPDX-License-Identifier: Apache-2.0
+
+from typing import Tuple, List
+from openvino_genai import GenerationConfig
+from utils.generation_config import get_greedy, get_beam_search
+
+def get_test_dataset() -> Tuple[List[str], List[GenerationConfig]]:
+    prompts = [
+        "What is OpenVINO?",
+        "How are you?",
+        "What is your name?",
+        "Tell me something about Canada"
+    ]
+    generation_configs = [
+        get_greedy(),
+        get_beam_search(),
+        get_greedy(),
+        get_beam_search(),
+    ]
+    return (prompts, generation_configs)
diff --git a/tests/python_tests/ov_genai_test_utils.py b/tests/python_tests/ov_genai_test_utils.py