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