Python get_fake_model_metadata_requestの例

コード例 #1

ファイル: test_get_model_metadata.py プロジェクト: xiaoyubing/OpenVINO-model-server

def test_get_model_metadata_correct_response(get_grpc_service_for_predict):
    request = get_fake_model_metadata_request(model_name='test',
                                              metadata_field='signature_'
                                              'def',
                                              version=1)
    grpc_server = get_grpc_service_for_predict
    rpc = grpc_server.invoke_unary_unary(
        PREDICT_SERVICE.methods_by_name['GetModelMetadata'], (), request, None)
    rpc.initial_metadata()
    response, trailing_metadata, code, details = rpc.termination()
    assert grpc.StatusCode.OK == code

コード例 #2

ファイル: test_get_model_metadata.py プロジェクト: xiaoyubing/OpenVINO-model-server

def test_get_model_metadata_wrong_metadata_field(get_grpc_service_for_predict):
    wrong_metadata = "wrong"
    request = get_fake_model_metadata_request(model_name='test',
                                              metadata_field=wrong_metadata,
                                              version=1)
    grpc_server = get_grpc_service_for_predict
    rpc = grpc_server.invoke_unary_unary(
        PREDICT_SERVICE.methods_by_name['GetModelMetadata'], (), request, None)
    rpc.initial_metadata()
    response, trailing_metadata, code, details = rpc.termination()
    assert grpc.StatusCode.INVALID_ARGUMENT == code