mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
216 lines
6.4 KiB
Python
216 lines
6.4 KiB
Python
import os
|
|
from pathlib import Path
|
|
from typing import Any
|
|
|
|
import pytest
|
|
from omegaconf import OmegaConf
|
|
from pydantic import ValidationError
|
|
|
|
|
|
@pytest.fixture
|
|
def patch_rootdir(tmp_path: Path, monkeypatch: Any) -> None:
|
|
"""This may be overkill since the current tests don't need the root dir to exist"""
|
|
monkeypatch.setenv("INVOKEAI_ROOT", str(tmp_path))
|
|
|
|
|
|
init1 = OmegaConf.create(
|
|
"""
|
|
InvokeAI:
|
|
Features:
|
|
always_use_cpu: false
|
|
Memory/Performance:
|
|
max_cache_size: 5
|
|
tiled_decode: false
|
|
"""
|
|
)
|
|
|
|
init2 = OmegaConf.create(
|
|
"""
|
|
InvokeAI:
|
|
Features:
|
|
always_use_cpu: true
|
|
Memory/Performance:
|
|
max_cache_size: 2
|
|
tiled_decode: true
|
|
"""
|
|
)
|
|
|
|
init3 = OmegaConf.create(
|
|
"""
|
|
InvokeAI:
|
|
Generation:
|
|
sequential_guidance: true
|
|
attention_type: xformers
|
|
attention_slice_size: 7
|
|
forced_tiled_decode: True
|
|
Device:
|
|
device: cpu
|
|
Model Cache:
|
|
ram: 1.25
|
|
"""
|
|
)
|
|
|
|
|
|
def test_use_init(patch_rootdir):
|
|
# note that we explicitly set omegaconf dict and argv here
|
|
# so that the values aren't read from ~invokeai/invokeai.yaml and
|
|
# sys.argv respectively.
|
|
from invokeai.app.services.config import InvokeAIAppConfig
|
|
|
|
conf1 = InvokeAIAppConfig.get_config()
|
|
assert conf1
|
|
conf1.parse_args(conf=init1, argv=[])
|
|
assert not conf1.tiled_decode
|
|
assert conf1.max_cache_size == 5
|
|
assert not conf1.always_use_cpu
|
|
|
|
conf2 = InvokeAIAppConfig.get_config()
|
|
assert conf2
|
|
conf2.parse_args(conf=init2, argv=[])
|
|
assert conf2.tiled_decode
|
|
assert conf2.max_cache_size == 2
|
|
assert not hasattr(conf2, "invalid_attribute")
|
|
|
|
|
|
def test_legacy():
|
|
from invokeai.app.services.config import InvokeAIAppConfig
|
|
|
|
conf = InvokeAIAppConfig.get_config()
|
|
assert conf
|
|
conf.parse_args(conf=init3, argv=[])
|
|
assert conf.xformers_enabled
|
|
assert conf.device == "cpu"
|
|
assert conf.use_cpu
|
|
assert conf.ram == 1.25
|
|
assert conf.ram_cache_size == 1.25
|
|
|
|
|
|
def test_argv_override():
|
|
from invokeai.app.services.config import InvokeAIAppConfig
|
|
|
|
conf = InvokeAIAppConfig.get_config()
|
|
conf.parse_args(conf=init1, argv=["--always_use_cpu", "--max_cache=10"])
|
|
assert conf.always_use_cpu
|
|
assert conf.max_cache_size == 10
|
|
assert conf.outdir == Path("outputs") # this is the default
|
|
|
|
|
|
def test_env_override(patch_rootdir):
|
|
from invokeai.app.services.config import InvokeAIAppConfig
|
|
|
|
# argv overrides
|
|
conf = InvokeAIAppConfig()
|
|
conf.parse_args(conf=init1, argv=["--max_cache=10"])
|
|
assert conf.always_use_cpu is False
|
|
os.environ["INVOKEAI_always_use_cpu"] = "True"
|
|
conf.parse_args(conf=init1, argv=["--max_cache=10"])
|
|
assert conf.always_use_cpu is True
|
|
|
|
# environment variables should be case insensitive
|
|
os.environ["InvokeAI_Max_Cache_Size"] = "15"
|
|
conf = InvokeAIAppConfig()
|
|
conf.parse_args(conf=init1, argv=[])
|
|
assert conf.max_cache_size == 15
|
|
|
|
conf = InvokeAIAppConfig()
|
|
conf.parse_args(conf=init1, argv=["--no-always_use_cpu", "--max_cache=10"])
|
|
assert conf.always_use_cpu is False
|
|
assert conf.max_cache_size == 10
|
|
|
|
conf = InvokeAIAppConfig.get_config(max_cache_size=20)
|
|
conf.parse_args(conf=init1, argv=[])
|
|
assert conf.max_cache_size == 20
|
|
|
|
# make sure that prefix is respected
|
|
del os.environ["INVOKEAI_always_use_cpu"]
|
|
os.environ["always_use_cpu"] = "True"
|
|
conf.parse_args(conf=init1, argv=[])
|
|
assert conf.always_use_cpu is False
|
|
|
|
|
|
def test_root_resists_cwd(patch_rootdir):
|
|
from invokeai.app.services.config import InvokeAIAppConfig
|
|
|
|
previous = os.environ["INVOKEAI_ROOT"]
|
|
cwd = Path(os.getcwd()).resolve()
|
|
|
|
os.environ["INVOKEAI_ROOT"] = "."
|
|
conf = InvokeAIAppConfig.get_config()
|
|
conf.parse_args([])
|
|
assert conf.root_path == cwd
|
|
|
|
os.chdir("..")
|
|
assert conf.root_path == cwd
|
|
os.environ["INVOKEAI_ROOT"] = previous
|
|
os.chdir(cwd)
|
|
|
|
|
|
def test_type_coercion(patch_rootdir):
|
|
from invokeai.app.services.config import InvokeAIAppConfig
|
|
|
|
conf = InvokeAIAppConfig().get_config()
|
|
conf.parse_args(argv=["--root=/tmp/foobar"])
|
|
assert conf.root == Path("/tmp/foobar")
|
|
assert isinstance(conf.root, Path)
|
|
conf = InvokeAIAppConfig.get_config(root="/tmp/different")
|
|
conf.parse_args(argv=["--root=/tmp/foobar"])
|
|
assert conf.root == Path("/tmp/different")
|
|
assert isinstance(conf.root, Path)
|
|
|
|
|
|
@pytest.mark.xfail(
|
|
reason="""
|
|
This test fails when run as part of the full test suite.
|
|
|
|
This test needs to deny nodes from being included in the InvocationsUnion by providing
|
|
an app configuration as a test fixture. Pytest executes all test files before running
|
|
tests, so the app configuration is already initialized by the time this test runs, and
|
|
the InvocationUnion is already created and the denied nodes are not omitted from it.
|
|
|
|
This test passes when `test_config.py` is tested in isolation.
|
|
|
|
Perhaps a solution would be to call `InvokeAIAppConfig.get_config().parse_args()` in
|
|
other test files?
|
|
"""
|
|
)
|
|
def test_deny_nodes(patch_rootdir):
|
|
from invokeai.app.services.config import InvokeAIAppConfig
|
|
|
|
# Allow integer, string and float, but explicitly deny float
|
|
allow_deny_nodes_conf = OmegaConf.create(
|
|
"""
|
|
InvokeAI:
|
|
Nodes:
|
|
allow_nodes:
|
|
- integer
|
|
- string
|
|
- float
|
|
deny_nodes:
|
|
- float
|
|
"""
|
|
)
|
|
# must parse config before importing Graph, so its nodes union uses the config
|
|
conf = InvokeAIAppConfig().get_config()
|
|
conf.parse_args(conf=allow_deny_nodes_conf, argv=[])
|
|
from invokeai.app.services.shared.graph import Graph
|
|
|
|
# confirm graph validation fails when using denied node
|
|
Graph(nodes={"1": {"id": "1", "type": "integer"}})
|
|
Graph(nodes={"1": {"id": "1", "type": "string"}})
|
|
|
|
with pytest.raises(ValidationError):
|
|
Graph(nodes={"1": {"id": "1", "type": "float"}})
|
|
|
|
from invokeai.app.invocations.baseinvocation import BaseInvocation
|
|
|
|
# confirm invocations union will not have denied nodes
|
|
all_invocations = BaseInvocation.get_invocations()
|
|
|
|
has_integer = len([i for i in all_invocations if i.__fields__.get("type").default == "integer"]) == 1
|
|
has_string = len([i for i in all_invocations if i.__fields__.get("type").default == "string"]) == 1
|
|
has_float = len([i for i in all_invocations if i.__fields__.get("type").default == "float"]) == 1
|
|
|
|
assert has_integer
|
|
assert has_string
|
|
assert not has_float
|