Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

tests: fix nodes tests

Browse Source 
they now require a valid output
This commit is contained in:
psychedelicious 2024-04-27 19:24:21 +10:00
parent 5a4dc9bbce
commit f526f0ae5f
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tests/test_node_graph.py
View File

@ -34,6 +34,7 @@ from .test_nodes import (
PolymorphicStringTestInvocation, PolymorphicStringTestInvocation,
PromptCollectionTestInvocation, PromptCollectionTestInvocation,
PromptTestInvocation, PromptTestInvocation,
PromptTestInvocationOutput,
TextToImageTestInvocation, TextToImageTestInvocation,
) )
@ -509,7 +510,7 @@ def test_invocation_decorator():
@invocation(invocation_type, title=title, tags=tags, category=category, version=version) @invocation(invocation_type, title=title, tags=tags, category=category, version=version)
class TestInvocation(BaseInvocation): class TestInvocation(BaseInvocation):
def invoke(self): def invoke(self) -> PromptTestInvocationOutput:
pass pass
schema = TestInvocation.model_json_schema() schema = TestInvocation.model_json_schema()
@ -527,7 +528,7 @@ def test_invocation_version_must_be_semver():
@invocation("test_invocation_version_valid", version=valid_version) @invocation("test_invocation_version_valid", version=valid_version)
class ValidVersionInvocation(BaseInvocation): class ValidVersionInvocation(BaseInvocation):
def invoke(self): def invoke(self) -> PromptTestInvocationOutput:
pass pass
with pytest.raises(InvalidVersionError): with pytest.raises(InvalidVersionError):