test_arg_parser.py: Added passing OS

artanokhov · Sep 11, 2024 · 493d9ed · 493d9ed
1 parent 5e13551
commit 493d9ed
Showing 1 changed file with 13 additions and 8 deletions.
diff --git a/utils/tests/test_arg_parser.py b/utils/tests/test_arg_parser.py
@@ -55,35 +55,37 @@
     pytest.param(
         {
             'mode': 'build',
-            'package_url': 'openvino_genai_ubuntu22_2024.5.0.0.dev20240905_x86_64.tar.gz',
+            'package_url': 'openvino_genai_ubuntu22_2024.5.0.0_x86_64.tar.gz',
+            'os': 'ubuntu22',
             'source': 'local',
         },
         {
             'device': ['cpu', 'gpu'],
-            'dockerfile_name': 'openvino_cg_dev_2022.1.0.dockerfile',
+            'dockerfile_name': 'openvino_cg_dev_2024.5.0.0.dockerfile',
             'python': 'python38',
-            'tags': ['ubuntu18_dev:2022.1.0.320', 'ubuntu18_dev:latest'],
+            'tags': ['ubuntu22_dev:2024.5.0.0', 'ubuntu22_dev:latest'],
             'distribution': 'dev',
             'install_type': 'copy',
-            'product_version': '2022.1.0',
+            'product_version': '2024.5.0.0',
         },
         id='parse product_version and distribution from package_url',
     ),
     pytest.param(
         {
             'mode': 'build',
-            'package_url': 'openvino_genai_ubuntu22_2024.5.0.0.dev20240905_x86_64.tar.gz',
+            'package_url': 'openvino_genai_ubuntu22_2024.5.0.0_x86_64.tar.gz',
+            'os': 'ubuntu22',
             'source': 'local',
             'tag_postfix': '_qqq',
         },
         {
             'device': ['cpu', 'gpu'],
-            'dockerfile_name': 'openvino_cg_dev_2024.3.0.dockerfile',
+            'dockerfile_name': 'openvino_cg_dev_2024.5.0.0.dockerfile',
             'python': 'python38',
-            'tags': ['ubuntu18_dev:2022.1.0.320', 'ubuntu18_dev:latest', 'ubuntu18_dev:2024.5.0.0.dev20240905_qqq'],
+            'tags': ['ubuntu22_dev:2024.5.0.0', 'ubuntu22_dev:latest', 'ubuntu22_dev:2024.5.0.0_qqq'],
             'distribution': 'dev',
             'install_type': 'copy',
-            'product_version': '2024.5.0.0.dev20240905',
+            'product_version': '2024.5.0.0',
         },
         id='check tag postfix',
     ),
@@ -161,6 +163,7 @@
         {
             'mode': 'build',
             'package_url': 'openvino_genai_ubuntu22_2024.5.0.0.dev20240905_x86_64.tar.gz',
+            'os': 'ubuntu22',
             'source': 'local',
         },
         {
@@ -179,6 +182,7 @@
             'mode': 'build',
             'package_url': 'openvino_genai_ubuntu22_2024.5.0.0.tar.gz',
             'source': 'local',
+            'os': 'ubuntu22',
             'tags': ['my_tag:latest'],
             'device': ['cpu', 'gpu'],
         },
@@ -237,6 +241,7 @@
             'mode': 'all',
             'package_url': 'openvino_genai_ubuntu22_2024.5.0.0_x86_64.tar.gz',
             'source': 'local',
+            'os': 'ubuntu22',
             'registry': 'https://deploy',
         },
         {