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

fix failing pytest for config module

Browse Source
This commit is contained in:
Lincoln Stein 2023-06-20 13:22:39 -04:00 committed by Kent Keirsey
parent d339c8627f
commit 257e972599
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tests/test_config.py
View File

@ -8,8 +8,6 @@ from pathlib import Path
os.environ['INVOKEAI_ROOT']='/tmp' os.environ['INVOKEAI_ROOT']='/tmp'
from invokeai.app.services.config import InvokeAIAppConfig from invokeai.app.services.config import InvokeAIAppConfig
from invokeai.app.invocations.generate import TextToImageInvocation
init1 = OmegaConf.create( init1 = OmegaConf.create(
''' '''
@ -37,13 +35,13 @@ def test_use_init():
# sys.argv respectively. # sys.argv respectively.
conf1 = InvokeAIAppConfig.get_config() conf1 = InvokeAIAppConfig.get_config()
assert conf1 assert conf1
conf1.parse_args(conf=init1) conf1.parse_args(conf=init1,argv=[])
assert conf1.max_loaded_models==5 assert conf1.max_loaded_models==5
assert not conf1.nsfw_checker assert not conf1.nsfw_checker
conf2 = InvokeAIAppConfig.get_config() conf2 = InvokeAIAppConfig.get_config()
assert conf2 assert conf2
conf2.parse_args(conf=init2) conf2.parse_args(conf=init2,argv=[])
assert conf2.nsfw_checker assert conf2.nsfw_checker
assert conf2.max_loaded_models==2 assert conf2.max_loaded_models==2
assert not hasattr(conf2,'invalid_attribute') assert not hasattr(conf2,'invalid_attribute')
@ -67,7 +65,7 @@ def test_env_override():
# environment variables should be case insensitive # environment variables should be case insensitive
os.environ['InvokeAI_Max_Loaded_Models'] = '15' os.environ['InvokeAI_Max_Loaded_Models'] = '15'
conf = InvokeAIAppConfig() conf = InvokeAIAppConfig()
conf.parse_args(conf=init1) conf.parse_args(conf=init1,argv=[])
assert conf.max_loaded_models == 15 assert conf.max_loaded_models == 15
conf = InvokeAIAppConfig() conf = InvokeAIAppConfig()