示例#1
0
def test_format_model_uri_input_string():
    model_uri = MODEL_URI

    model_uri_input = _Job._format_model_uri_input(model_uri)

    assert model_uri_input.config["DataSource"]["S3DataSource"][
        "S3Uri"] == model_uri
示例#2
0
def test_format_model_uri_input_exception():
    model_uri = 1

    with pytest.raises(ValueError):
        _Job._format_model_uri_input(model_uri)
示例#3
0
def test_format_model_uri_input_local_file():
    model_uri_input = _Job._format_model_uri_input(LOCAL_MODEL_NAME)

    assert model_uri_input.config["DataSource"]["FileDataSource"][
        "FileUri"] == LOCAL_MODEL_NAME