From fbc2bfb445c71985e9fc399d3fac2def2fc6854e Mon Sep 17 00:00:00 2001
From: Martin Bielik <martin.bielik@instea.sk>
Date: Sun, 22 Dec 2024 14:55:15 +0100
Subject: added image generation

---
 tests/roles_test.py | 4 ++++
 1 file changed, 4 insertions(+)

(limited to 'tests/roles_test.py')

diff --git a/tests/roles_test.py b/tests/roles_test.py
index da9b9a4..a0cc26a 100644
--- a/tests/roles_test.py
+++ b/tests/roles_test.py
@@ -22,3 +22,7 @@ def test_role_chat_only():
         'below',
         'tab',
     }
+
+def test_explicit_image_roles():
+    role_names = load_ai_role_names('image')
+    assert set(role_names) == { 'hd-image' }
-- 
cgit v1.2.3