mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-20 12:24:59 +00:00
36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
import unittest
|
|
|
|
from aider.help import Help, install_help_extra
|
|
from aider.io import InputOutput
|
|
|
|
|
|
class TestHelp(unittest.TestCase):
|
|
def setUp(self):
|
|
io = InputOutput(yes=True)
|
|
install_help_extra(io)
|
|
|
|
def test_init(self):
|
|
help_inst = Help()
|
|
self.assertIsNotNone(help_inst.retriever)
|
|
|
|
def test_ask_without_mock(self):
|
|
help_instance = Help()
|
|
question = "What is aider?"
|
|
result = help_instance.ask(question)
|
|
|
|
self.assertIn(f"# Question: {question}", result)
|
|
self.assertIn("<doc", result)
|
|
self.assertIn("</doc>", result)
|
|
self.assertGreater(len(result), 100) # Ensure we got a substantial response
|
|
|
|
# Check for some expected content (adjust based on your actual help content)
|
|
self.assertIn("aider", result.lower())
|
|
self.assertIn("ai", result.lower())
|
|
self.assertIn("chat", result.lower())
|
|
|
|
# Assert that there are more than 5 <doc> entries
|
|
self.assertGreater(result.count("<doc"), 5)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|