Python start_test Exemples

Exemple #1

Fichier : mxnet_multiple_models_test.py Projet : vijinkp/mxnet-model-server

def test_multiple_mxnet_integ(tmpdir):
    tmpdir = str(tmpdir)
    utils._download_file(
        tmpdir, "https://s3.amazonaws.com/model-server/inputs/kitten.jpg")
    model_urls = utils.filtered_urls(mxnet_models, utils.mxnet_model_urls)
    utils.start_test(tmpdir, model_urls, port='8080', model_type='mxnet')
    utils.cleanup(tmpdir)

Exemple #2

Fichier : onnx_export_integ_test.py Projet : vijinkp/mxnet-model-server

def test_onnx_integ(tmpdir, model_data):
    tmpdir = str(tmpdir)
    model_urls = utils.filtered_urls([model_data], utils.onnx_model_urls)
    utils._download_file(
        tmpdir, "https://s3.amazonaws.com/model-server/inputs/kitten.jpg")
    utils.start_test(tmpdir, model_urls, port='8080', model_type='onnx')
    utils.cleanup(tmpdir)