blob: c2425d5c4278f8fa19cebc101eaf27894f6dcff1 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
|
import generate
def test_registry_integrity():
assert generate.DEFAULT_MODEL in generate.MODELS
for key, repo in generate.MODELS.items():
assert isinstance(key, str) and key
assert isinstance(repo, str) and "/" in repo
def test_resolve_known_key():
assert generate.resolve_model("sdxl") == generate.MODELS["sdxl"]
assert generate.resolve_model("realvis") == generate.MODELS["realvis"]
def test_resolve_unknown_key_raises_listing_valid():
try:
generate.resolve_model("nope")
except SystemExit as e:
assert "sdxl" in str(e) and "realvis" in str(e) and "pony" in str(e)
else:
assert False, "expected SystemExit for unknown key"
if __name__ == "__main__":
test_registry_integrity()
test_resolve_known_key()
test_resolve_unknown_key_raises_listing_valid()
print("ok")
|
