File size: 672 Bytes
2f3c093 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 | import unittest
import agent_zero
class TestAgentZero(unittest.TestCase):
def setUp(self):
self.agent = agent_zero.AgentZero()
def test_initialize(self):
self.agent.initialize()
self.assertTrue(self.agent.is_initialized)
def test_get_status(self):
self.agent.initialize()
status = self.agent.get_status()
self.assertEqual(status, {"status": "initialized"})
def test_perform_action(self):
self.agent.initialize()
result = self.agent.perform_action("test_action")
self.assertEqual(result, "Action test_action performed successfully")
if __name__ == "__main__":
unittest.main()
|