def test_get_bento_service_event_properties_with_no_artifact(): class ExampleBentoService(bentoml.BentoService): pass properties = _get_bento_service_event_properties(ExampleBentoService()) assert "input_types" not in properties assert properties["artifact_types"] assert 'NO_ARTIFACT' in properties["artifact_types"] assert properties["env"] is not None
def test_get_bento_service_event_properties(bento_service): properties = _get_bento_service_event_properties(bento_service) assert 'PickleArtifact' in properties["artifact_types"] assert 'DataframeInput' in properties["input_types"] assert 'ImageInput' in properties["input_types"] assert 'JsonInput' in properties["input_types"] assert len(properties["input_types"]) == 4 assert properties["env"] is not None
def test_get_bento_service_event_properties(bento_service): properties = _get_bento_service_event_properties(bento_service) assert 'PickleArtifact' in properties["artifact_types"] assert 'DataframeInput' in properties["input_types"] assert 'ImageInput' in properties["input_types"] assert 'JsonInput' in properties["input_types"] assert len(properties["input_types"]) == 4 assert properties["env"] is not None assert properties["env"]["conda_env"]["channels"] == [ "conda-forge", "defaults" ]
def test_get_bento_service_event_properties(bento_service): properties = _get_bento_service_event_properties(bento_service) assert 'PickleArtifact' in properties["artifact_types"] assert 'DataframeHandler' in properties["handler_types"] assert 'ImageHandler' in properties["handler_types"] assert 'JsonHandler' in properties["handler_types"] assert len(properties["handler_types"]) == 3 # Disabling fastai related tests to fix travis build # assert 'FastaiImageHandler' in properties["handler_types"] # assert len(properties["handler_types"]) == 4 assert properties["env"] is not None assert properties["env"]["conda_env"]["channels"] == ["defaults"]
def test_get_bento_service_event_properties(bento_service): properties = _get_bento_service_event_properties(bento_service) assert 'PickleArtifact' in properties["artifact_types"] assert 'DataframeHandler' in properties["handler_types"] assert 'ImageHandler' in properties["handler_types"] assert 'JsonHandler' in properties["handler_types"] if six.PY3: assert 'FastaiImageHandler' in properties["handler_types"] assert len(properties["handler_types"]) == 4 else: assert len(properties["handler_types"]) == 3 assert properties["env"] is not None
def test_get_bento_service_event_properties(bento_service): properties = _get_bento_service_event_properties(bento_service) assert 'PickleArtifact' in properties["artifact_types"] assert 'DataframeInput' in properties["input_types"] assert 'ImageInput' in properties["input_types"] assert 'LegacyImageInput' in properties["input_types"] assert 'JsonInput' in properties["input_types"] assert 'LegacyJsonInput' in properties["input_types"] assert len(properties["input_types"]) == 5 # Disabling fastai related tests to fix travis build # assert 'FastaiImageInput' in properties["input_types"] # assert len(properties["input_types"]) == 4 assert properties["env"] is not None assert properties["env"]["conda_env"]["channels"] == [ "conda-forge", "defaults" ]