From c9c2d5ab6f90b455778ead246dc251ef6d81879f Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Sat, 7 Dec 2024 13:38:57 -0800 Subject: [PATCH] test: update test assertions to check model objects instead of names --- tests/basic/test_history.py | 6 +++--- tests/basic/test_repo.py | 6 +++--- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/tests/basic/test_history.py b/tests/basic/test_history.py index 17b563265..e2ea194c0 100644 --- a/tests/basic/test_history.py +++ b/tests/basic/test_history.py @@ -91,9 +91,9 @@ class TestChatSummary(TestCase): # Check that both models were tried self.assertEqual(mock_send.call_count, 2) - # Check that the calls were made with the correct model names - self.assertEqual(mock_send.call_args_list[0][0][0], "gpt-4") - self.assertEqual(mock_send.call_args_list[1][0][0], "gpt-3.5-turbo") + # Check that the calls were made with the correct models + self.assertEqual(mock_send.call_args_list[0][0][0], mock_model1) + self.assertEqual(mock_send.call_args_list[1][0][0], mock_model2) # Check that we got a summary from the second model self.assertEqual( diff --git a/tests/basic/test_repo.py b/tests/basic/test_repo.py index 50cf27951..a9de68a66 100644 --- a/tests/basic/test_repo.py +++ b/tests/basic/test_repo.py @@ -125,9 +125,9 @@ class TestRepo(unittest.TestCase): # Check that simple_send_with_retries was called twice self.assertEqual(mock_send.call_count, 2) - # Check that it was called with the correct model names - self.assertEqual(mock_send.call_args_list[0][0][0], model1.name) - self.assertEqual(mock_send.call_args_list[1][0][0], model2.name) + # Check that it was called with the correct models + self.assertEqual(mock_send.call_args_list[0][0][0], model1) + self.assertEqual(mock_send.call_args_list[1][0][0], model2) # Check that the content of the messages is the same for both calls self.assertEqual(mock_send.call_args_list[0][0][1], mock_send.call_args_list[1][0][1])