From 9063ea91731d05221cfdd60e7d3b0f020d0fe286 Mon Sep 17 00:00:00 2001 From: psychedelicious <4822129+psychedelicious@users.noreply.github.com> Date: Fri, 15 Mar 2024 21:56:11 +1100 Subject: [PATCH] tests: comprehensive config migration tests Add testing for the settings that had to be explicitly migrated. --- tests/test_config.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/tests/test_config.py b/tests/test_config.py index 38a9413604..b74e26debe 100644 --- a/tests/test_config.py +++ b/tests/test_config.py @@ -38,6 +38,10 @@ InvokeAI: ignore_missing_core_models: false Paths: outdir: /some/outputs/dir + conf_path: /custom/models.yaml + Model Cache: + max_cache_size: 100 + max_vram_cache_size: 50 """ invalid_config = """ @@ -77,6 +81,9 @@ def test_migrate_v3_config_from_file(tmp_path: Path): assert config.outputs_dir == Path("/some/outputs/dir") assert config.host == "192.168.1.1" assert config.port == 8080 + assert config.ram == 100 + assert config.vram == 50 + assert config.legacy_models_yaml_path == Path("/custom/models.yaml") # This should be stripped out assert not hasattr(config, "esrgan")