dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-25 14:55:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

test: update AutoCompleter test for command matching

Browse source
This commit is contained in:
Paul Gauthier 2024-08-28 16:03:40 -07:00 • committed by Paul Gauthier (aider)
parent d2bc24cb18
commit e11cd8ecbb
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
tests/basic/test_io.py
View file

@ -169,14 +169,14 @@ class TestInputOutput(unittest.TestCase):
commands = MagicMock() commands = MagicMock()
commands.get_commands.return_value = ["model", "chat", "help"] commands.get_commands.return_value = ["model", "chat", "help"]
commands.get_completions.return_value = ["gpt-3.5-turbo", "gpt-4"] commands.get_completions.return_value = ["gpt-3.5-turbo", "gpt-4"]
commands.matching_commands.return_value = (["model"], None, None) commands.matching_commands.return_value = (["/model", "/models"], None, None)
autocompleter = AutoCompleter(root, rel_fnames, addable_rel_fnames, commands, "utf-8") autocompleter = AutoCompleter(root, rel_fnames, addable_rel_fnames, commands, "utf-8")
# Test case for "/model gpt" # Test case for "/model gpt"
result = autocompleter.get_command_completions("/model gpt", ["/model", "gpt"]) result = autocompleter.get_command_completions("/model gpt", ["/model", "gpt"])
self.assertEqual(result, ["gpt-3.5-turbo", "gpt-4"]) self.assertEqual(result, ["gpt-3.5-turbo", "gpt-4"])
commands.get_completions.assert_called_once_with("model") commands.get_completions.assert_called_once_with("/model")
commands.matching_commands.assert_called_once_with("/model") commands.matching_commands.assert_called_once_with("/model")