diff --git a/tests/test_config.py b/tests/test_config.py index cea4991d12..d6c9d95d7e 100644 --- a/tests/test_config.py +++ b/tests/test_config.py @@ -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()