diff options
| author | Martin Bielik <martin.bielik@instea.sk> | 2024-12-22 14:55:15 +0100 |
|---|---|---|
| committer | Martin Bielik <martin.bielik@instea.sk> | 2024-12-22 14:55:15 +0100 |
| commit | fbc2bfb445c71985e9fc399d3fac2def2fc6854e (patch) | |
| tree | e93db9bf177d8790c608e3871d9f55d259510855 /tests/roles_test.py | |
| parent | 4f1c95386437c8fa6e189c136681950a91fb25c9 (diff) | |
| download | vim-ai-fbc2bfb445c71985e9fc399d3fac2def2fc6854e.tar.gz | |
added image generation
Diffstat (limited to 'tests/roles_test.py')
| -rw-r--r-- | tests/roles_test.py | 4 |
1 files changed, 4 insertions, 0 deletions
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' } |