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
def test_format_model_uri_input_exception(): model_uri = 1 with pytest.raises(ValueError): _Job._format_model_uri_input(model_uri)
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