def test_parse_models_uri_with_latest(uri, expected_name): (name, version, stage) = _parse_model_uri(uri) assert name == expected_name assert version is None assert stage is None
def test_parse_models_uri_invalid_input(uri): with pytest.raises(MlflowException, match="Not a proper models"): _parse_model_uri(uri)
def test_parse_models_uri_invalid_input(uri): with pytest.raises(MlflowException): _parse_model_uri(uri)