11import pytest
22
3+ from tests .model_serving .model_server .components .constants import KSERVE_RUNTIME_PARAMS , SERVERLESS_ISVC_PARAMS
34from tests .model_serving .model_server .utils import verify_inference_response
45from utilities .constants import (
5- ModelAndFormat ,
6- ModelFormat ,
7- ModelInferenceRuntime ,
86 ModelStoragePath ,
9- ModelVersion ,
107 Protocols ,
118)
129from utilities .inference_utils import Inference
1310from utilities .manifests .openvino import OPENVINO_INFERENCE_CONFIG , OPENVINO_KSERVE_INFERENCE_CONFIG
1411
1512pytestmark = [pytest .mark .serverless , pytest .mark .modelmesh , pytest .mark .sanity ]
1613
17- KSERVE_RUNTIME_PARAMS = {
18- "runtime-name" : ModelInferenceRuntime .OPENVINO_KSERVE_RUNTIME ,
19- "model-format" : {ModelAndFormat .OPENVINO_IR : ModelVersion .OPSET1 },
20- }
21- KSERVE_ISVC_PARAMS = {
22- "name" : ModelFormat .OPENVINO ,
23- "model-version" : ModelVersion .OPSET1 ,
24- "model-dir" : ModelStoragePath .KSERVE_OPENVINO_EXAMPLE_MODEL ,
25- }
2614MODELMESH_ISVC_PARAMS = {
2715 "model-path" : ModelStoragePath .OPENVINO_EXAMPLE_MODEL ,
2816 "modelmesh-enabled" : True ,
2917}
3018
3119
3220@pytest .mark .parametrize (
33- "model_namespace, openvino_kserve_serving_runtime, ovms_serverless_inference_service , "
21+ "model_namespace, openvino_kserve_serving_runtime, ovms_kserve_inference_service , "
3422 "http_s3_openvino_model_mesh_inference_service" ,
3523 [
3624 pytest .param (
3725 {"name" : "serverless-model-mesh-openvino" , "modelmesh-enabled" : True },
3826 KSERVE_RUNTIME_PARAMS ,
39- KSERVE_ISVC_PARAMS ,
27+ SERVERLESS_ISVC_PARAMS ,
4028 MODELMESH_ISVC_PARAMS ,
4129 )
4230 ],
4331 indirect = True ,
4432)
4533class TestOpenVINOServerlessModelMesh :
4634 def test_serverless_openvino_created_before_model_mesh_ns_rest_inference (
47- self , ovms_serverless_inference_service , http_s3_openvino_model_mesh_inference_service
35+ self , ovms_kserve_inference_service , http_s3_openvino_model_mesh_inference_service
4836 ):
4937 """Verify that Serverless model can be queried when running with modelmesh inference service"""
5038 verify_inference_response (
51- inference_service = ovms_serverless_inference_service ,
39+ inference_service = ovms_kserve_inference_service ,
5240 inference_config = OPENVINO_KSERVE_INFERENCE_CONFIG ,
5341 inference_type = Inference .INFER ,
5442 protocol = Protocols .HTTPS ,
5543 use_default_query = True ,
5644 )
5745
5846 def test_model_mesh_openvino_created_after_serverless_in_namespace_rest_inference (
59- self , ovms_serverless_inference_service , http_s3_openvino_model_mesh_inference_service
47+ self , ovms_kserve_inference_service , http_s3_openvino_model_mesh_inference_service
6048 ):
6149 """Verify that modelmesh model can be queried when running with kserve inference service"""
6250 verify_inference_response (
@@ -70,20 +58,20 @@ def test_model_mesh_openvino_created_after_serverless_in_namespace_rest_inferenc
7058
7159@pytest .mark .parametrize (
7260 "model_namespace, http_s3_openvino_model_mesh_inference_service, openvino_kserve_serving_runtime, "
73- "ovms_serverless_inference_service , " ,
61+ "ovms_kserve_inference_service , " ,
7462 [
7563 pytest .param (
7664 {"name" : "model-mesh-serverless-openvino" , "modelmesh-enabled" : True },
7765 MODELMESH_ISVC_PARAMS ,
7866 KSERVE_RUNTIME_PARAMS ,
79- KSERVE_ISVC_PARAMS ,
67+ SERVERLESS_ISVC_PARAMS ,
8068 )
8169 ],
8270 indirect = True ,
8371)
8472class TestOpenVINOModelMeshServerless :
8573 def test_model_mesh_openvino_created_before_serverless_in_namespace_rest_inference (
86- self , http_s3_openvino_model_mesh_inference_service , ovms_serverless_inference_service
74+ self , http_s3_openvino_model_mesh_inference_service , ovms_kserve_inference_service
8775 ):
8876 """Verify that modelmesh model can be queried when running with kserve inference service"""
8977 verify_inference_response (
@@ -95,11 +83,11 @@ def test_model_mesh_openvino_created_before_serverless_in_namespace_rest_inferen
9583 )
9684
9785 def test_serverless_openvino_created_after_model_mesh_ns_rest_inference (
98- self , http_s3_openvino_model_mesh_inference_service , ovms_serverless_inference_service
86+ self , http_s3_openvino_model_mesh_inference_service , ovms_kserve_inference_service
9987 ):
10088 """Verify that Serverless model can be queried when running with modelmesh inference service"""
10189 verify_inference_response (
102- inference_service = ovms_serverless_inference_service ,
90+ inference_service = ovms_kserve_inference_service ,
10391 inference_config = OPENVINO_KSERVE_INFERENCE_CONFIG ,
10492 inference_type = Inference .INFER ,
10593 protocol = Protocols .HTTPS ,
0 commit comments