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: Mock simple_send_with_retries method in test setup

Browse source
This commit is contained in:
Paul Gauthier (aider) 2025-02-04 12:11:43 -08:00
parent 9553478384
commit af8f7e95b0
1 changed files with 3 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
tests/basic/test_history.py
View file

@ -7,6 +7,7 @@ from aider.models import Model
class TestChatSummary(TestCase): class TestChatSummary(TestCase):
def setUp(self): def setUp(self):
self.mock_model = mock.Mock(spec=Model) self.mock_model = mock.Mock(spec=Model)
del self.mock_model.simple_send_with_retries
self.mock_model.name = "gpt-3.5-turbo" self.mock_model.name = "gpt-3.5-turbo"
self.mock_model.token_count = lambda msg: len(msg["content"].split()) self.mock_model.token_count = lambda msg: len(msg["content"].split())
self.mock_model.info = {"max_input_tokens": 4096} self.mock_model.info = {"max_input_tokens": 4096}
@ -72,8 +73,10 @@ class TestChatSummary(TestCase):
@mock.patch("aider.models.Model.simple_send_with_retries") @mock.patch("aider.models.Model.simple_send_with_retries")
def test_fallback_to_second_model(self, mock_send): def test_fallback_to_second_model(self, mock_send):
mock_model1 = mock.Mock(spec=Model) mock_model1 = mock.Mock(spec=Model)
del mock_model1.simple_send_with_retries
mock_model1.name = "gpt-4" mock_model1.name = "gpt-4"
mock_model2 = mock.Mock(spec=Model) mock_model2 = mock.Mock(spec=Model)
del mock_model2.simple_send_with_retries
mock_model2.name = "gpt-3.5-turbo" mock_model2.name = "gpt-3.5-turbo"
chat_summary = ChatSummary([mock_model1, mock_model2], max_tokens=100) chat_summary = ChatSummary([mock_model1, mock_model2], max_tokens=100)