Foreign-Mirrors/Fooocus
1
0
Fork 0
mirror of https://github.com/lllyasviel/Fooocus.git synced 2026-08-16 13:13:16 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix: update unit tests

Browse Source
This commit is contained in:
Manuel Schmid
2024-05-20 17:29:11 +02:00
parent 0d1310d9e9
commit 8820f82b41
1 changed files with 28 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_utils.py
+27 -12
View File
@@ -8,12 +8,16 @@ class TestUtils(unittest.TestCase):
test_cases = [ test_cases = [
{ {
"input": ("some prompt, very cool, <lora:hey-lora:0.4>, cool <lora:you-lora:0.2>", [], 5), "input": ("some prompt, very cool, <lora:hey-lora:0.4>, cool <lora:you-lora:0.2>", [], 5),
"output": [("hey-lora.safetensors", 0.4), ("you-lora.safetensors", 0.2)], "output": (
[('hey-lora.safetensors', 0.4), ('you-lora.safetensors', 0.2)], 'some prompt, very cool, cool'),
}, },
# Test can not exceed limit # Test can not exceed limit
{ {
"input": ("some prompt, very cool, <lora:hey-lora:0.4>, cool <lora:you-lora:0.2>", [], 1), "input": ("some prompt, very cool, <lora:hey-lora:0.4>, cool <lora:you-lora:0.2>", [], 1),
"output": [("hey-lora.safetensors", 0.4)], "output": (
[('hey-lora.safetensors', 0.4)],
'some prompt, very cool, cool'
),
}, },
# test Loras from UI take precedence over prompt # test Loras from UI take precedence over prompt
{ {
@@ -22,22 +26,33 @@ class TestUtils(unittest.TestCase):
[("hey-lora.safetensors", 0.4)], [("hey-lora.safetensors", 0.4)],
5, 5,
), ),
"output": [ "output": (
("hey-lora.safetensors", 0.4), [
("l1.safetensors", 0.4), ('hey-lora.safetensors', 0.4),
("l2.safetensors", -0.2), ('l1.safetensors', 0.4),
("l3.safetensors", 0.3), ('l2.safetensors', -0.2),
("l4.safetensors", 0.5), ('l3.safetensors', 0.3),
('l4.safetensors', 0.5)
], ],
'some prompt, very cool'
)
}, },
# Test lora specification not separated by comma are ignored, only latest specified is used
{ {
"input": ("some prompt, very cool, <lora:hey-lora:0.4><lora:you-lora:0.2>", [], 3), "input": ("some prompt, very cool, <lora:hey-lora:0.4><lora:you-lora:0.2>", [], 3),
"output": [("you-lora.safetensors", 0.2)], "output": (
[
('hey-lora.safetensors', 0.4),
('you-lora.safetensors', 0.2)
],
'some prompt, very cool, <lora:you-lora:0.2><lora:hey-lora:0.4>'
),
}, },
{ {
"input": ("<lora:foo:1..2>, <lora:bar:.>, <lora:baz:+> and <lora:quux:>", [], 6), "input": ("<lora:foo:1..2>, <lora:bar:.>, <test:1.0>, <lora:baz:+> and <lora:quux:>", [], 6),
"output": [] "output": (
[],
'<lora:foo:1..2>, <lora:bar:.>, <test:1.0>, <lora:baz:+> and <lora:quux:>'
)
} }
] ]