From 98835d1f6d7266cbcfad78f4898ebd190eb8a85f Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Tue, 3 Sep 2024 17:21:06 -0700 Subject: [PATCH] test: update sanity check tests to use tool_output instead of tool_warning --- tests/basic/test_models.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/basic/test_models.py b/tests/basic/test_models.py index af6e7081c..0cac3fddc 100644 --- a/tests/basic/test_models.py +++ b/tests/basic/test_models.py @@ -40,8 +40,8 @@ class TestModels(unittest.TestCase): sanity_check_model(mock_io, model) - mock_io.tool_warning.assert_called() - calls = mock_io.tool_warning.call_args_list + mock_io.tool_output.assert_called() + calls = mock_io.tool_output.call_args_list self.assertIn("- API_KEY1: ✓ Set", str(calls)) self.assertIn("- API_KEY2: ✓ Set", str(calls)) @@ -57,8 +57,8 @@ class TestModels(unittest.TestCase): sanity_check_model(mock_io, model) - mock_io.tool_warning.assert_called() - calls = mock_io.tool_warning.call_args_list + mock_io.tool_output.assert_called() + calls = mock_io.tool_output.call_args_list self.assertIn("- API_KEY1: ✗ Not set", str(calls)) self.assertIn("- API_KEY2: ✗ Not set", str(calls))