diff --git a/tests/test_telegram_helper.py b/tests/test_telegram_helper.py index 0e60f87..d322683 100644 --- a/tests/test_telegram_helper.py +++ b/tests/test_telegram_helper.py @@ -45,7 +45,7 @@ class TestTelegramHelper(unittest.IsolatedAsyncioTestCase): self.mock_bot.switch_model.return_value = "Model switched successfully." result = await self.telegram_helper._switch_logic() self.mock_bot.switch_model.assert_called_once() - self.assertEqual(result, "Model switching is not supported for this bot.") + self.assertEqual(result, "Model switched successfully.") async def test_switch_logic_not_supported(self): # Remove switch_model from mock_bot to simulate not supported