def test_select_url_custom_format():
key = "not_available"
urls = {}
def format(key):
return f"custom format for '{key}'"
expected = format(key)
with pytest.raises(RuntimeError) as exc_info:
enc.select_url(urls, key, format=format)
actual = exc_info.value.args[0]
assert actual.endswith(expected)
def state_dict_url(self, framework: str) -> str:
return enc.select_url(multi_layer_encoder_urls, framework)
def test_select_url_not_available():
key = "not_available"
urls = {}
with pytest.raises(RuntimeError):
enc.select_url(urls, key)
def test_select_url():
key = "available"
url = "https://pystiche.org"
assert enc.select_url({key: url}, key) == url
