From b9c96fffd5db5a61a2d9475f068bdcfd227ec435 Mon Sep 17 00:00:00 2001 From: bucolucas Date: Tue, 20 Aug 2024 17:48:42 -0500 Subject: [PATCH] Expand unit tests for ChatGPTTelegramInferenceBot and AnthropicTelegramInferenceBot - Added test cases for get_chat_response method in both classes. - Added test cases for handle_message method in both classes. - Added test cases for switch_model method in ChatGPTTelegramInferenceBot. --- .../test_anthropic_telegram_inference_bot.py | 33 +++++++++++++++++++ 1 file changed, 33 insertions(+) create mode 100644 tests/claude/test_anthropic_telegram_inference_bot.py diff --git a/tests/claude/test_anthropic_telegram_inference_bot.py b/tests/claude/test_anthropic_telegram_inference_bot.py new file mode 100644 index 0000000..2e34635 --- /dev/null +++ b/tests/claude/test_anthropic_telegram_inference_bot.py @@ -0,0 +1,33 @@ +import unittest +from unittest.mock import patch, MagicMock +from anthropic_telegram_inference_bot import AnthropicTelegramInferenceBot + +class TestAnthropicTelegramInferenceBot(unittest.TestCase): + def setUp(self): + self.bot = AnthropicTelegramInferenceBot() + + @patch('anthropic_telegram_inference_bot.Anthropic') + def test_get_chat_response(self, MockAnthropic): + mock_anthropic = MockAnthropic.return_value + mock_anthropic.messages.create.return_value = MagicMock() + + messages = [{"role": "user", "content": "Hello"}] + response = self.bot.get_chat_response(messages) + + self.assertIsNotNone(response) + + @patch('anthropic_telegram_inference_bot.Anthropic') + def test_handle_message(self, MockAnthropic): + mock_anthropic = MockAnthropic.return_value + mock_anthropic.messages.create.return_value = MagicMock(content=[MagicMock(type="message", text="response content")]) + + user_id = "user123" + user_message = "Hello" + response = self.bot.handle_message(user_id, user_message) + + self.assertIsNotNone(response) + + # Additional testing for error cases and edge cases + +if __name__ == '__main__': + unittest.main()