Spaces:
Sleeping
Sleeping
File size: 1,334 Bytes
da82b2b 0b497e7 da82b2b 0b497e7 da82b2b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 |
import unittest
from unittest.mock import Mock, patch
import requests
from frontend.app_ui import get_similar_prompts
class TestGetSimilarPrompts(unittest.TestCase):
@patch("frontend.app_ui.requests.post")
def test_get_similar_prompts_success(self, mock_post):
# Mock the response object to simulate a successful API call
mock_response = Mock()
mock_response.status_code = 200
mock_response.json.return_value = {"prompts": ["prompt1", "prompt2", "prompt3"]}
mock_post.return_value = mock_response
# Call the function with a sample query and number
result = get_similar_prompts("test query", 3)
# Assertions
self.assertIsInstance(result, dict)
self.assertEqual(result, {"prompts": ["prompt1", "prompt2", "prompt3"]})
@patch("frontend.app_ui.requests.post")
def test_get_similar_prompts_failure(self, mock_post):
# Mock the response object to simulate a failed API call
mock_post.side_effect = requests.RequestException("Mock request exception")
# Call the function with a sample query and number
result = get_similar_prompts("test query", 3)
# Assertions
self.assertIsInstance(result, dict)
self.assertEqual(len(result), 0)
if __name__ == "__main__":
unittest.main()
|