You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
83 lines
3.1 KiB
83 lines
3.1 KiB
1 year ago
|
import unittest
|
||
|
|
from unittest.mock import patch
|
||
|
|
from Sswarms.models.revgptv1 import RevChatGPTModelv1
|
||
|
|
|
||
|
|
class TestRevChatGPT(unittest.TestCase):
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
self.access_token = "<your_access_token>"
|
||
|
|
self.model = RevChatGPTModelv1(access_token=self.access_token)
|
||
|
|
|
||
|
|
def test_run(self):
|
||
|
|
prompt = "What is the capital of France?"
|
||
|
|
response = self.model.run(prompt)
|
||
|
|
self.assertEqual(response, "The capital of France is Paris.")
|
||
|
|
|
||
|
|
def test_run_time(self):
|
||
|
|
prompt = "Generate a 300 word essay about technology."
|
||
|
|
response = self.model.run(prompt)
|
||
|
|
self.assertLess(self.model.end_time - self.model.start_time, 60)
|
||
|
|
|
||
|
|
def test_generate_summary(self):
|
||
|
|
text = "This is a sample text to summarize. It has multiple sentences and details. The summary should be concise."
|
||
|
|
summary = self.model.generate_summary(text)
|
||
|
|
self.assertLess(len(summary), len(text)/2)
|
||
|
|
|
||
|
|
def test_enable_plugin(self):
|
||
|
|
plugin_id = "some_plugin_id"
|
||
|
|
self.model.enable_plugin(plugin_id)
|
||
|
|
self.assertIn(plugin_id, self.model.config["plugin_ids"])
|
||
|
|
|
||
|
|
def test_list_plugins(self):
|
||
|
|
plugins = self.model.list_plugins()
|
||
|
|
self.assertGreater(len(plugins), 0)
|
||
|
|
self.assertIsInstance(plugins[0], dict)
|
||
|
|
self.assertIn("id", plugins[0])
|
||
|
|
self.assertIn("name", plugins[0])
|
||
|
|
|
||
|
|
def test_get_conversations(self):
|
||
|
|
conversations = self.model.chatbot.get_conversations()
|
||
|
|
self.assertIsInstance(conversations, list)
|
||
|
|
|
||
|
|
@patch("RevChatGPTModelv1.Chatbot.get_msg_history")
|
||
|
|
def test_get_msg_history(self, mock_get_msg_history):
|
||
|
|
conversation_id = "convo_id"
|
||
|
|
self.model.chatbot.get_msg_history(conversation_id)
|
||
|
|
mock_get_msg_history.assert_called_with(conversation_id)
|
||
|
|
|
||
|
|
@patch("RevChatGPTModelv1.Chatbot.share_conversation")
|
||
|
|
def test_share_conversation(self, mock_share_conversation):
|
||
|
|
self.model.chatbot.share_conversation()
|
||
|
|
mock_share_conversation.assert_called()
|
||
|
|
|
||
|
|
def test_gen_title(self):
|
||
|
|
convo_id = "123"
|
||
|
|
message_id = "456"
|
||
|
|
title = self.model.chatbot.gen_title(convo_id, message_id)
|
||
|
|
self.assertIsInstance(title, str)
|
||
|
|
|
||
|
|
def test_change_title(self):
|
||
|
|
convo_id = "123"
|
||
|
|
title = "New Title"
|
||
|
|
self.model.chatbot.change_title(convo_id, title)
|
||
|
|
self.assertEqual(self.model.chatbot.get_msg_history(convo_id)["title"], title)
|
||
|
|
|
||
|
|
def test_delete_conversation(self):
|
||
|
|
convo_id = "123"
|
||
|
|
self.model.chatbot.delete_conversation(convo_id)
|
||
|
|
with self.assertRaises(Exception):
|
||
|
|
self.model.chatbot.get_msg_history(convo_id)
|
||
|
|
|
||
|
|
def test_clear_conversations(self):
|
||
|
|
self.model.chatbot.clear_conversations()
|
||
|
|
conversations = self.model.chatbot.get_conversations()
|
||
|
|
self.assertEqual(len(conversations), 0)
|
||
|
|
|
||
|
|
def test_rollback_conversation(self):
|
||
|
|
original_convo_id = self.model.chatbot.conversation_id
|
||
|
|
self.model.chatbot.rollback_conversation(1)
|
||
|
|
self.assertNotEqual(original_convo_id, self.model.chatbot.conversation_id)
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
unittest.main()
|