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

build(nodes): remove outdated metadata test

Browse Source 
I will add tests for the new service soon
This commit is contained in:
psychedelicious 2023-05-25 01:17:14 +10:00 committed by Kent Keirsey
parent ff6b345d45
commit 295b98a13c
1 changed files with 0 additions and 53 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

53
tests/nodes/test_png_metadata_service.py
View File

@ -1,53 +0,0 @@
import json
import os
from PIL import Image, PngImagePlugin
from invokeai.app.invocations.generate import TextToImageInvocation
from invokeai.app.services.metadata import PngMetadataService
valid_metadata = {
"session_id": "1",
"node": {
"id": "1",
"type": "txt2img",
"prompt": "dog",
"seed": 178785523,
"steps": 30,
"width": 512,
"height": 512,
"cfg_scale": 7.5,
"scheduler": "lms",
"model": "stable-diffusion-1.5",
},
}
metadata_service = PngMetadataService()
def test_can_load_and_parse_invokeai_metadata(tmp_path):
raw_metadata = {"session_id": "123", "node": {"id": "456", "type": "test_type"}}
temp_image = Image.new("RGB", (512, 512))
temp_image_path = os.path.join(tmp_path, "test.png")
pnginfo = PngImagePlugin.PngInfo()
pnginfo.add_text("invokeai", json.dumps(raw_metadata))
temp_image.save(temp_image_path, pnginfo=pnginfo)
image = Image.open(temp_image_path)
loaded_metadata = metadata_service.get_metadata(image)
assert loaded_metadata is not None
assert raw_metadata == loaded_metadata
def test_can_build_invokeai_metadata():
session_id = valid_metadata["session_id"]
node = TextToImageInvocation(**valid_metadata["node"])
metadata = metadata_service.build_metadata(session_id=session_id, node=node)
assert valid_metadata == metadata