From 0ce8472562a78456d3dfd535db93c7f5193fde69 Mon Sep 17 00:00:00 2001
From: Lincoln Stein <lstein@gmail.com>
Date: Wed, 26 Jul 2023 07:29:33 -0400
Subject: [PATCH] adjust unit test to account for nsfw always being true now

---
 tests/test_config.py | 25 ++++++++++++++-----------
 1 file changed, 14 insertions(+), 11 deletions(-)

diff --git a/tests/test_config.py b/tests/test_config.py
index d6c9d95d7e..498a47748e 100644
--- a/tests/test_config.py
+++ b/tests/test_config.py
@@ -13,9 +13,10 @@ init1 = OmegaConf.create(
 '''
 InvokeAI:
   Features:
-    nsfw_checker: False
+    always_use_cpu: false
   Memory/Performance:
     max_cache_size: 5
+    tiled_decode: false
 '''
 )
 
@@ -23,9 +24,10 @@ init2 = OmegaConf.create(
 '''
 InvokeAI:
   Features:
-    nsfw_checker: true
+    always_use_cpu: true
   Memory/Performance:
     max_cache_size: 2
+    tiled_decode: true
 '''
 )
 
@@ -36,20 +38,21 @@ def test_use_init():
     conf1 = InvokeAIAppConfig.get_config()
     assert conf1
     conf1.parse_args(conf=init1,argv=[])
+    assert not conf1.tiled_decode
     assert conf1.max_cache_size==5
-    assert not conf1.nsfw_checker
+    assert not conf1.always_use_cpu
 
     conf2 = InvokeAIAppConfig.get_config()
     assert conf2
     conf2.parse_args(conf=init2,argv=[])
-    assert conf2.nsfw_checker
+    assert conf2.tiled_decode
     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_cache=10'])
-    assert conf.nsfw_checker
+    conf.parse_args(conf=init1,argv=['--always_use_cpu','--max_cache=10'])
+    assert conf.always_use_cpu
     assert conf.max_cache_size==10
     assert conf.outdir==Path('outputs')  # this is the default
     
@@ -57,10 +60,10 @@ def test_env_override():
     # argv overrides 
     conf = InvokeAIAppConfig()
     conf.parse_args(conf=init1,argv=['--max_cache=10'])
-    assert conf.nsfw_checker==False
-    os.environ['INVOKEAI_nsfw_checker'] = 'True'
+    assert conf.always_use_cpu==False
+    os.environ['INVOKEAI_always_use_cpu'] = 'True'
     conf.parse_args(conf=init1,argv=['--max_cache=10'])
-    assert conf.nsfw_checker==True
+    assert conf.always_use_cpu==True
 
     # environment variables should be case insensitive
     os.environ['InvokeAI_Max_Cache_Size'] = '15'
@@ -69,8 +72,8 @@ def test_env_override():
     assert conf.max_cache_size == 15
 
     conf = InvokeAIAppConfig()
-    conf.parse_args(conf=init1,argv=['--no-nsfw_checker','--max_cache=10'])
-    assert conf.nsfw_checker==False
+    conf.parse_args(conf=init1,argv=['--no-always_use_cpu','--max_cache=10'])
+    assert conf.always_use_cpu==False
     assert conf.max_cache_size==10
 
     conf = InvokeAIAppConfig.get_config(max_cache_size=20)