From 924eeb43de2f1b0a673fd3fb0a0b55968437d7d2 Mon Sep 17 00:00:00 2001 From: Paul Gauthier Date: Mon, 23 Sep 2024 11:34:49 -0700 Subject: [PATCH] fix: Update test assertions for API key checks --- 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 0cac3fddc..330fa79a5 100644 --- a/tests/basic/test_models.py +++ b/tests/basic/test_models.py @@ -42,8 +42,8 @@ class TestModels(unittest.TestCase): 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)) + self.assertIn("- API_KEY1: Set", str(calls)) + self.assertIn("- API_KEY2: Set", str(calls)) @patch("os.environ") def test_sanity_check_model_not_set(self, mock_environ): @@ -59,8 +59,8 @@ class TestModels(unittest.TestCase): 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)) + self.assertIn("- API_KEY1: Not set", str(calls)) + self.assertIn("- API_KEY2: Not set", str(calls)) if __name__ == "__main__":