Exemplos de get_fake_model_metadata_request em Python

Exemplo n.º 1

Arquivo: test_get_model_metadata.py Projeto: 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

Exemplo n.º 2

Arquivo: test_get_model_metadata.py Projeto: 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