From 100cca5dbf54d38d4356dffc8adf369a747489d8 Mon Sep 17 00:00:00 2001 From: Paul Gauthier Date: Fri, 23 Aug 2024 15:17:15 -0700 Subject: [PATCH] feat: Add model dumps to test_repo.py --- tests/basic/test_repo.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tests/basic/test_repo.py b/tests/basic/test_repo.py index 919ae9560..21e36b91d 100644 --- a/tests/basic/test_repo.py +++ b/tests/basic/test_repo.py @@ -112,6 +112,8 @@ class TestRepo(unittest.TestCase): model1 = Model("gpt-3.5-turbo") model2 = Model("gpt-4") + dump(model1) + dump(model2) repo = GitRepo(InputOutput(), None, None, models=[model1, model2]) # Call the get_commit_message method with dummy diff and context @@ -123,6 +125,8 @@ class TestRepo(unittest.TestCase): # Check that simple_send_with_retries was called twice self.assertEqual(mock_send.call_count, 2) + dump(mock_send.call_args) + # Check that it was called with the correct model names mock_send.assert_any_call(model1.name, mock_send.call_args[0][1]) mock_send.assert_any_call(model2.name, mock_send.call_args[0][1])