Fix Test : Agent -> test_run

pyAmakCore/classes/agent.py

@@ -72,6 +72,12 @@ class Agent:
         """
         return self.__environment
 
+    def get_phase(self) -> Phase:
+        """
+        return current phase of agent
+        """
+        return self.__phase
+
     def add_neighbour(self, agent: 'Agent') -> None:
         """
         add agent into neighbours of current agent

pyAmakCore/tests/test_agent/test_run.py

@@ -6,6 +6,7 @@ from unittest import TestCase, main
 from pyAmakCore.classes.agent import Agent
 from pyAmakCore.classes.amas import Amas
 from pyAmakCore.classes.environment import Environment
+from pyAmakCore.enumeration.agent_phase import Phase
 from pyAmakCore.enumeration.executionPolicy import ExecutionPolicy
 
 
@@ -14,15 +15,38 @@ class TestAgentRun(TestCase):
     Test class for Agent run
     """
 
-    def test_run(self):
+    def test_run_one_phase(self):
+        """
+        test that run() work and return correct phase with 1 phase policy
+        """
         Agent.reset_agent()
         environment = Environment()
         amas = Amas(environment)
         amas.set_execution_policy(ExecutionPolicy.ONE_PHASE)
 
         agent = Agent(amas)
+        self.assertEqual(agent.get_phase(), Phase.INITIALIZING)
         agent.run()
-        self.assertEqual()
+        self.assertEqual(agent.get_phase(), Phase.DECISION_AND_ACTION_DONE)
+        agent.run()
+        self.assertEqual(agent.get_phase(), Phase.DECISION_AND_ACTION_DONE)
+
+    def test_run_two_phase(self):
+        """
+        test that run() work and return correct phase with 2 phase policy
+        """
+        Agent.reset_agent()
+        environment = Environment()
+        amas = Amas(environment)
+        amas.set_execution_policy(ExecutionPolicy.TWO_PHASES)
+
+        agent = Agent(amas)
+        self.assertEqual(agent.get_phase(), Phase.INITIALIZING)
+        agent.run()
+        self.assertEqual(agent.get_phase(), Phase.PERCEPTION_DONE)
+        agent.run()
+        self.assertEqual(agent.get_phase(), Phase.DECISION_AND_ACTION_DONE)
+
 
 if __name__ == '__main__':
     main()