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")
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")