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