| 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() |
|
|
