dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-21 04:44:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

test: add test case for get_command_completions with "/model gpt" input

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-08-28 16:00:45 -07:00
parent ac6c76d530
commit d02e96f773
1 changed files with 14 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

14
tests/basic/test_io.py
View file

@ -162,6 +162,20 @@ class TestInputOutput(unittest.TestCase):
mock_prompt.assert_called_once()
mock_prompt.reset_mock()
def test_get_command_completions(self):
root = ""
rel_fnames = []
addable_rel_fnames = []
commands = MagicMock()
commands.get_commands.return_value = ["model", "chat", "help"]
commands.get_completions.return_value = ["gpt-3.5-turbo", "gpt-4"]
autocompleter = AutoCompleter(root, rel_fnames, addable_rel_fnames, commands, "utf-8")
# Test case for "/model gpt"
result = autocompleter.get_command_completions("/model gpt", ["/model", "gpt"])
self.assertEqual(result, ["gpt-3.5-turbo", "gpt-4"])
commands.get_completions.assert_called_once_with("model")
if __name__ == "__main__":
unittest.main()