default image role tests

madox2 · Dec 22, 2024 · 380d5cd · 380d5cd
1 parent fbc2bfb
commit 380d5cd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/roles_test.py b/tests/roles_test.py
@@ -25,4 +25,4 @@ def test_role_chat_only():
 
 def test_explicit_image_roles():
     role_names = load_ai_role_names('image')
-    assert set(role_names) == { 'hd-image' }
+    assert set(role_names) == { 'hd-image', 'hd', 'natural' }