summaryrefslogtreecommitdiff
path: root/tests/roles_test.py
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--tests/roles_test.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/tests/roles_test.py b/tests/roles_test.py
index a0cc26a..ad51995 100644
--- 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' }