Python _get_bento_service_event_propertiesの例

コード例 #1

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

コード例 #2

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

コード例 #3

ファイル: test_usage_stats.py プロジェクト: zidingz/BentoML

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

コード例 #4

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

コード例 #5

ファイル: test_usage_stats.py プロジェクト: leepand/BentoML

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

コード例 #6

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