1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
|
from roles import load_ai_role_names
def test_role_completion():
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',
# default roles
'right',
'below',
'tab',
}
def test_explicit_image_roles():
role_names = load_ai_role_names('image')
assert set(role_names) == { 'hd-image', 'hd', 'natural' }
|
