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)