Python apiの例

コード例 #1

ファイル: test_service.py プロジェクト: yashika51/BentoML

def test_custom_api_name():
    # these names should work:
    bentoml.api(input=DataframeInput(), api_name="a_valid_name")(lambda x: x)
    bentoml.api(input=DataframeInput(), api_name="AValidName")(lambda x: x)
    bentoml.api(input=DataframeInput(), api_name="_AValidName")(lambda x: x)
    bentoml.api(input=DataframeInput(),
                api_name="a_valid_name_123")(lambda x: x)

    with pytest.raises(InvalidArgument) as e:
        bentoml.api(input=DataframeInput(),
                    api_name="a invalid name")(lambda x: x)
    assert str(e.value).startswith("Invalid API name")

    with pytest.raises(InvalidArgument) as e:
        bentoml.api(input=DataframeInput(),
                    api_name="123_a_invalid_name")(lambda x: x)
    assert str(e.value).startswith("Invalid API name")

    with pytest.raises(InvalidArgument) as e:
        bentoml.api(input=DataframeInput(),
                    api_name="a-invalid-name")(lambda x: x)
    assert str(e.value).startswith("Invalid API name")

コード例 #2

ファイル: test_service.py プロジェクト: vincycode7/BentoML

def test_custom_api_name():
    # these names should work:
    bentoml.api(bentoml.handlers.DataframeHandler,
                api_name="a_valid_name")(lambda x: x)
    bentoml.api(bentoml.handlers.DataframeHandler,
                api_name="AValidName")(lambda x: x)
    bentoml.api(bentoml.handlers.DataframeHandler,
                api_name="_AValidName")(lambda x: x)
    bentoml.api(bentoml.handlers.DataframeHandler,
                api_name="a_valid_name_123")(lambda x: x)

    with pytest.raises(ValueError) as e:
        bentoml.api(bentoml.handlers.DataframeHandler,
                    api_name="a invalid name")(lambda x: x)
    assert str(e.value).startswith("Invalid API name")

    with pytest.raises(ValueError) as e:
        bentoml.api(bentoml.handlers.DataframeHandler,
                    api_name="123_a_invalid_name")(lambda x: x)
    assert str(e.value).startswith("Invalid API name")

    with pytest.raises(ValueError) as e:
        bentoml.api(bentoml.handlers.DataframeHandler,
                    api_name="a-invalid-name")(lambda x: x)
    assert str(e.value).startswith("Invalid API name")