diff options
Diffstat (limited to 'tests/config_test.py')
| -rw-r--r-- | tests/config_test.py | 127 |
1 files changed, 0 insertions, 127 deletions
diff --git a/tests/config_test.py b/tests/config_test.py deleted file mode 100644 index a13d89f..0000000 --- a/tests/config_test.py +++ /dev/null @@ -1,127 +0,0 @@ -import vim -from config import make_config_and_prompt, make_prompt - -default_config = { - "options": { - "model": "gpt-4o", - "endpoint_url": "https://api.openai.com/v1/chat/completions", - "max_tokens": "0", - "max_completion_tokens": "0", - "temperature": "1", - "request_timeout": "20", - "stream": "1", - "enable_auth": "1", - "token_file_path": "", - "selection_boundary": "", - "initial_prompt": "You are a general assistant.", - }, - "ui": { - "open_chat_command": "preset_below", - "scratch_buffer_keep_open": "0", - "populate_options": "0", - "code_syntax_enabled": "1", - "paste_mode": "1", - }, -} - -def test_default_config(): - actual_output = make_config_and_prompt({ - 'config_default': default_config, - 'config_extension': {}, - 'user_instruction': 'translate to Slovak', - 'user_selection': 'Hello world!', - 'command_type': 'chat', - }) - expected_output = { - 'config': default_config, - 'prompt': 'translate to Slovak:\nHello world!', - } - assert expected_output == actual_output - -def test_param_config(): - actual_config = make_config_and_prompt({ - 'config_default': default_config, - 'config_extension': { - 'options': { - 'max_tokens': '1000', - }, - }, - 'user_instruction': 'hello', - 'user_selection': '', - 'command_type': 'chat', - })['config'] - assert '1000' == actual_config['options']['max_tokens'] - assert 'gpt-4o' == actual_config['options']['model'] - -def test_role_config(): - config = make_config_and_prompt({ - 'config_default': default_config, - 'config_extension': {}, - 'user_instruction': '/test-role-simple user instruction', - 'user_selection': 'selected text', - 'command_type': 'chat', - }) - actual_config = config['config'] - actual_prompt = config['prompt'] - assert 'o1-preview' == actual_config['options']['model'] - assert 'simple role prompt:\nuser instruction:\nselected text' == actual_prompt - -def test_role_config_different_commands(): - config = make_config_and_prompt({ - 'config_default': default_config, - 'config_extension': {}, - 'user_instruction': '/test-role hello', - 'user_selection': '', - 'command_type': 'chat', - }) - actual_config = config['config'] - actual_prompt = config['prompt'] - assert 'model-common' == actual_config['options']['model'] - assert 'https://localhost/chat' == actual_config['options']['endpoint_url'] - assert '0' == actual_config['ui']['paste_mode'] - assert 'preset_tab' == actual_config['ui']['open_chat_command'] - assert 'hello' == actual_prompt - - config = make_config_and_prompt({ - 'config_default': default_config, - 'config_extension': {}, - 'user_instruction': '/test-role hello', - 'user_selection': '', - 'command_type': 'complete', - }) - actual_config = config['config'] - actual_prompt = config['prompt'] - assert 'model-common' == actual_config['options']['model'] - assert 'https://localhost/complete' == actual_config['options']['endpoint_url'] - assert '0' == actual_config['ui']['paste_mode'] - assert 'hello' == actual_prompt - -def test_multiple_role_configs(): - config = make_config_and_prompt({ - 'config_default': default_config, - 'config_extension': {}, - 'user_instruction': '/test-role /test-role-simple hello', - 'user_selection': '', - 'command_type': 'chat', - }) - actual_config = config['config'] - actual_prompt = config['prompt'] - assert 'o1-preview' == actual_config['options']['model'] - assert 'https://localhost/chat' == actual_config['options']['endpoint_url'] - assert 'simple role prompt:\nhello' == actual_prompt - -def test_user_prompt(): - assert 'fix grammar: helo word' == make_prompt( '', 'fix grammar: helo word', '', '') - assert 'fix grammar:\nhelo word' == make_prompt( '', 'fix grammar', 'helo word', '') - -def test_role_prompt(): - assert 'fix grammar:\nhelo word' == make_prompt( 'fix grammar', 'helo word', '', '') - assert 'fix grammar:\nhelo word' == make_prompt( 'fix grammar', '', 'helo word', '') - assert 'fix grammar:\nand spelling:\nhelo word' == make_prompt( 'fix grammar', 'and spelling', 'helo word', '') - -def test_selection_prompt(): - assert 'fix grammar:\nhelo word' == make_prompt( '', '', 'fix grammar:\nhelo word', '') - -def test_selection_boundary(): - assert 'fix grammar:\n###\nhelo word\n###' == make_prompt( '', 'fix grammar', 'helo word', '###') - assert 'fix grammar:\n###\nhelo word\n###' == make_prompt( 'fix grammar', '', 'helo word', '###') |