From 8fde389664ca59773c38dc0ec1a434a98bc2428b Mon Sep 17 00:00:00 2001 From: Martin Bielik Date: Mon, 16 Dec 2024 23:46:17 +0100 Subject: command-type only roles --- tests/roles_test.py | 16 +++++++++++++--- 1 file changed, 13 insertions(+), 3 deletions(-) (limited to 'tests/roles_test.py') diff --git a/tests/roles_test.py b/tests/roles_test.py index 3230329..ac5525d 100644 --- a/tests/roles_test.py +++ b/tests/roles_test.py @@ -1,10 +1,20 @@ from roles import load_ai_role_names def test_role_completion(): - role_names = load_ai_role_names() - assert role_names == [ + role_names = load_ai_role_names('complete') + assert set(role_names) == { 'test-role-simple', 'test-role', 'deprecated-test-role-simple', 'deprecated-test-role', - ] + } + +def test_role_chat_only(): + role_names = load_ai_role_names('chat') + assert set(role_names) == { + 'test-role-simple', + 'test-role', + 'chat-only-role', + 'deprecated-test-role-simple', + 'deprecated-test-role', + } -- cgit v1.2.3