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 the test of the config system

Browse Source
This commit is contained in:
Lincoln Stein 2023-07-11 17:37:02 -04:00 committed by Kent Keirsey
parent e0a7ec6e95
commit c3adb301a0
1 changed files with 14 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
tests/test_config.py
View File

@ -15,7 +15,7 @@ InvokeAI:
Features:
nsfw_checker: False
Memory/Performance:
max_loaded_models: 5
max_cache_size: 5
'''
)
@ -25,7 +25,7 @@ InvokeAI:
Features:
nsfw_checker: true
Memory/Performance:
max_loaded_models: 2
max_cache_size: 2
'''
)
@ -36,46 +36,46 @@ def test_use_init():
conf1 = InvokeAIAppConfig.get_config()
assert conf1
conf1.parse_args(conf=init1,argv=[])
assert conf1.max_loaded_models==5
assert conf1.max_cache_size==5
assert not conf1.nsfw_checker
conf2 = InvokeAIAppConfig.get_config()
assert conf2
conf2.parse_args(conf=init2,argv=[])
assert conf2.nsfw_checker
assert conf2.max_loaded_models==2
assert conf2.max_cache_size==2
assert not hasattr(conf2,'invalid_attribute')
def test_argv_override():
conf = InvokeAIAppConfig.get_config()
conf.parse_args(conf=init1,argv=['--nsfw_checker','--max_loaded=10'])
conf.parse_args(conf=init1,argv=['--nsfw_checker','--max_cache=10'])
assert conf.nsfw_checker
assert conf.max_loaded_models==10
assert conf.max_cache_size==10
assert conf.outdir==Path('outputs') # this is the default
def test_env_override():
# argv overrides
conf = InvokeAIAppConfig()
conf.parse_args(conf=init1,argv=['--max_loaded=10'])
conf.parse_args(conf=init1,argv=['--max_cache=10'])
assert conf.nsfw_checker==False
os.environ['INVOKEAI_nsfw_checker'] = 'True'
conf.parse_args(conf=init1,argv=['--max_loaded=10'])
conf.parse_args(conf=init1,argv=['--max_cache=10'])
assert conf.nsfw_checker==True
# environment variables should be case insensitive
os.environ['InvokeAI_Max_Loaded_Models'] = '15'
os.environ['InvokeAI_Max_Cache_Size'] = '15'
conf = InvokeAIAppConfig()
conf.parse_args(conf=init1,argv=[])
assert conf.max_loaded_models == 15
assert conf.max_cache_size == 15
conf = InvokeAIAppConfig()
conf.parse_args(conf=init1,argv=['--no-nsfw_checker','--max_loaded=10'])
conf.parse_args(conf=init1,argv=['--no-nsfw_checker','--max_cache=10'])
assert conf.nsfw_checker==False
assert conf.max_loaded_models==10
assert conf.max_cache_size==10
conf = InvokeAIAppConfig.get_config(max_loaded_models=20)
conf = InvokeAIAppConfig.get_config(max_cache_size=20)
conf.parse_args(conf=init1,argv=[])
assert conf.max_loaded_models==20
assert conf.max_cache_size==20
def test_type_coercion():
conf = InvokeAIAppConfig().get_config()