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 tests

Browse Source 
The tests were testing deprecated settings (not the settings themselves, just the class's functionality).
This commit is contained in:
psychedelicious 2024-03-08 12:37:37 +11:00 committed by Brandon
parent fb1ae55010
commit 0696094d95
1 changed files with 12 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
tests/test_config.py
View File

@ -18,9 +18,10 @@ init1 = OmegaConf.create(
InvokeAI: InvokeAI:
Features: Features:
always_use_cpu: false always_use_cpu: false
Memory/Performance: Model Cache:
max_cache_size: 5 convert_cache: 5
tiled_decode: false Generation:
force_tiled_decode: false
""" """
) )
@ -29,9 +30,10 @@ init2 = OmegaConf.create(
InvokeAI: InvokeAI:
Features: Features:
always_use_cpu: true always_use_cpu: true
Memory/Performance: Model Cache:
max_cache_size: 2 convert_cache: 2
tiled_decode: true Generation:
force_tiled_decode: true
""" """
) )
@ -60,15 +62,15 @@ def test_use_init(patch_rootdir):
conf1 = InvokeAIAppConfig.get_config() conf1 = InvokeAIAppConfig.get_config()
assert conf1 assert conf1
conf1.parse_args(conf=init1, argv=[]) conf1.parse_args(conf=init1, argv=[])
assert not conf1.tiled_decode assert not conf1.force_tiled_decode
assert conf1.max_cache_size == 5 assert conf1.convert_cache == 5
assert not conf1.always_use_cpu assert not conf1.always_use_cpu
conf2 = InvokeAIAppConfig.get_config() conf2 = InvokeAIAppConfig.get_config()
assert conf2 assert conf2
conf2.parse_args(conf=init2, argv=[]) conf2.parse_args(conf=init2, argv=[])
assert conf2.tiled_decode assert conf2.force_tiled_decode
assert conf2.max_cache_size == 2 assert conf2.convert_cache == 2
assert not hasattr(conf2, "invalid_attribute") assert not hasattr(conf2, "invalid_attribute")