admin
8b3c1c6a58
- Added test cases for load_functions method - Added test cases for clear_conversation method - Added test cases for call_tool method
34 lines
1.3 KiB
Python
34 lines
1.3 KiB
Python
import unittest
|
|
from base_telegram_inference_bot import BaseTelegramInferenceBot
|
|
|
|
class TestBaseTelegramInferenceBot(unittest.TestCase):
|
|
def setUp(self):
|
|
# Initialize the bot or mock any dependencies here
|
|
self.bot = BaseTelegramInferenceBot()
|
|
|
|
def test_load_system_prompt(self):
|
|
# Example test case for load_system_prompt method
|
|
result = self.bot.load_system_prompt()
|
|
self.assertIsNotNone(result) # Replace with actual expected result
|
|
|
|
def test_load_functions(self):
|
|
# Test the load_functions method
|
|
functions = self.bot.load_functions()
|
|
self.assertIsInstance(functions, list) # Replace with actual expected result
|
|
self.assertTrue(len(functions) > 0) # Assuming it should load some functions
|
|
|
|
def test_clear_conversation(self):
|
|
# Test the clear_conversation method
|
|
self.bot.clear_conversation()
|
|
self.assertEqual(self.bot.conversations, {}) # Assuming conversations is a dictionary
|
|
|
|
def test_call_tool(self):
|
|
# Test the call_tool method
|
|
tool_name = "some_tool"
|
|
params = {"param1": "value1"}
|
|
result = self.bot.call_tool(tool_name, params)
|
|
self.assertIsNotNone(result) # Replace with actual expected result
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|