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: default settings union evaluation

Browse Source
This commit is contained in:
psychedelicious 2024-03-25 16:35:17 +11:00 committed by Kent Keirsey
parent 2ec03ae95c
commit d63d26924b
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tests/app/services/model_records/test_model_records_sql.py
View File

@ -18,7 +18,9 @@ from invokeai.app.services.model_records import (
from invokeai.app.services.model_records.model_records_base import ModelRecordChanges
from invokeai.backend.model_manager.config import (
BaseModelType,
ControlAdapterDefaultSettings,
MainDiffusersConfig,
MainModelDefaultSettings,
ModelFormat,
ModelSourceType,
ModelType,
@ -288,3 +290,12 @@ def test_filter_2(store: ModelRecordServiceBase):
model_name="dup_name1",
)
assert len(matches) == 1
def test_model_record_changes():
# This test guards against some unexpected behaviours from pydantic's union evaluation. See #6035
changes = ModelRecordChanges.model_validate({"default_settings": {"preprocessor": "value"}})
assert isinstance(changes.default_settings, ControlAdapterDefaultSettings)
changes = ModelRecordChanges.model_validate({"default_settings": {"vae": "value"}})
assert isinstance(changes.default_settings, MainModelDefaultSettings)