You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
24 lines
679 B
24 lines
679 B
|
|
import unittest
|
|
from swarms.structs.agent import Agent
|
|
|
|
class TestBasicExample(unittest.TestCase):
|
|
def setUp(self):
|
|
self.agent = Agent(
|
|
agent_name="Test-Agent",
|
|
agent_description="A test agent",
|
|
system_prompt="You are a helpful assistant.",
|
|
model_name="gpt-4",
|
|
)
|
|
|
|
def test_agent_initialization(self):
|
|
self.assertEqual(self.agent.agent_name, "Test-Agent")
|
|
self.assertEqual(self.agent.agent_description, "A test agent")
|
|
|
|
def test_agent_run(self):
|
|
response = self.agent.run("What is 2+2?")
|
|
self.assertIsNotNone(response)
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|