Ismael Sampaio
a7b292f437
- Initial agent development, starting by PDF fix: - Fixed test discovery on subdirectories test: - Started tests for PDF agent
21 lines
587 B
Python
21 lines
587 B
Python
from math import e
|
|
import unittest
|
|
|
|
from src.interfaces.agents.agent import Agent
|
|
from src.modules.pdf.pdfAgent import PDFAgent
|
|
|
|
class TestPDFAgent(unittest.TestCase):
|
|
def setUp(self):
|
|
self.agent: Agent
|
|
return super().setUp()
|
|
|
|
|
|
def createAgent(self):
|
|
self.agent = PDFAgent()
|
|
self.agent.createAgent()
|
|
self.assertIsNone(self.agent)
|
|
self.assertEqual(self.agent.name, "PDF Agent")
|
|
self.assertEqual(self.agent.description, "Agent to handle PDF operations")
|
|
|
|
self.assertIsInstance(self.agent.tools, list)
|