From 380d5cdd9538c2522dfc8d03a8a261760bb0439a Mon Sep 17 00:00:00 2001
From: Martin Bielik <martin.bielik@instea.sk>
Date: Sun, 22 Dec 2024 15:00:37 +0100
Subject: default image role tests

---
 tests/roles_test.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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' }
-- 
cgit v1.2.3