Spaces:
Sleeping
Sleeping
File size: 2,664 Bytes
503c07a 8b358c4 503c07a 8b358c4 |
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 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 |
'''Unittests for agent tools.'''
import unittest
from functions.tools import (
google_search,
wikipedia_search,
get_wikipedia_page
)
class TestGoogleSearch(unittest.TestCase):
'''Tests for the google search tool.'''
def setUp(self):
google_search_query = 'Python programming language'
self.search_results = google_search(google_search_query)
def test_result_type(self):
'''Search results should be a dictionary.'''
self.assertIsInstance(self.search_results, dict)
def test_result_length(self):
'''Search results should contain 5 items.'''
self.assertEqual(len(self.search_results), 5)
def test_result_content(self):
'''Each search result should contain three elements: title, link, and snippet.'''
for _, result in self.search_results.items():
self.assertIsInstance(result, dict)
self.assertIn('title', result)
self.assertIn('url', result)
self.assertIn('description', result)
self.assertIsInstance(result['title'], str)
self.assertIsInstance(result['url'], str)
self.assertIsInstance(result['description'], str)
class TestWikipediaSearch(unittest.TestCase):
'''Tests for the wikipedia search tool.'''
def setUp(self):
wikipedia_search_query = 'Python programming language'
self.search_results = wikipedia_search(wikipedia_search_query)
def test_result_type(self):
'''Search results should be a dictionary.'''
self.assertIsInstance(self.search_results, dict)
def test_result_length(self):
'''Search results should contain 5 items.'''
self.assertEqual(len(self.search_results), 5)
def test_result_content(self):
'''Each search result should contain three elements: title, link, and snippet.'''
for _, result in self.search_results.items():
self.assertIsInstance(result, dict)
self.assertIn('title', result)
self.assertIn('description', result)
self.assertIsInstance(result['title'], str)
self.assertIsInstance(result['description'], str)
class TestGetWikipediaPage(unittest.TestCase):
'''Tests for the get_wikipedia_page tool.'''
def setUp(self):
self.page_content = get_wikipedia_page('Mercedes Sosa')
def test_page_content_type(self):
'''Page content should be a string.'''
self.assertIsInstance(self.page_content, str)
def test_page_content_not_empty(self):
'''Page content should not be empty.'''
self.assertTrue(len(self.page_content) > 0)
|