mirror of
https://github.com/brycedrennan/imaginAIry
synced 2024-11-05 12:00:15 +00:00
25 lines
698 B
Python
25 lines
698 B
Python
from imaginairy import config
|
|
from imaginairy.model_manager import resolve_model_paths
|
|
|
|
|
|
def test_resolved_paths():
|
|
"""Test that the resolved model path is correct."""
|
|
(
|
|
model_metadata,
|
|
weights_path,
|
|
config_path,
|
|
control_weights_path,
|
|
) = resolve_model_paths()
|
|
assert model_metadata.short_name == config.DEFAULT_MODEL
|
|
assert model_metadata.config_path == config_path
|
|
default_config_path = config_path
|
|
|
|
(
|
|
model_metadata,
|
|
weights_path,
|
|
config_path,
|
|
control_weights_path,
|
|
) = resolve_model_paths(weights_path="foo.ckpt")
|
|
assert weights_path == "foo.ckpt"
|
|
assert config_path == default_config_path
|