Add test for table_driven_agent_program and Random_agent_program (aimacode#770)

nouman-10 · norvig · commit c7fff61d1d2b · 2018-02-27T11:18:07.000-08:00
* Add test for table driven agent

* Some style fixes

* Added done to tabledrivenagent test in readme

* Added randomAgentProgram test to test_agents.py

* Added Import randomAgentProgram

* Style fixes

* Added the done tag tp tabledrivenagent test
diff --git a/README.md b/README.md
@@ -35,7 +35,7 @@ Here is a table of algorithms, the figure, name of the algorithm in the book and
 | 2.1    | Environment                       | `Environment`                 | [`agents.py`][agents]           | Done | Included |
 | 2.1    | Agent                             | `Agent`                       | [`agents.py`][agents]           | Done | Included |
 | 2.3    | Table-Driven-Vacuum-Agent         | `TableDrivenVacuumAgent`      | [`agents.py`][agents]           | Done | Included |
-| 2.7    | Table-Driven-Agent                | `TableDrivenAgent`            | [`agents.py`][agents]           |      | Included |
+| 2.7    | Table-Driven-Agent                | `TableDrivenAgent`            | [`agents.py`][agents]           | Done | Included |
 | 2.8    | Reflex-Vacuum-Agent               | `ReflexVacuumAgent`           | [`agents.py`][agents]           | Done | Included |
 | 2.10   | Simple-Reflex-Agent               | `SimpleReflexAgent`           | [`agents.py`][agents]           |      | Included |
 | 2.12   | Model-Based-Reflex-Agent          | `ReflexAgentWithState`        | [`agents.py`][agents]           |      | Included |
@@ -160,4 +160,4 @@ Many thanks for contributions over the years. I got bug reports, corrected code,
 [rl]:../master/rl.py
 [search]:../master/search.py
 [utils]:../master/utils.py
-[text]:../master/text.py
+[text]:../master/text.py
diff --git a/tests/test_agents.py b/tests/test_agents.py
@@ -2,7 +2,7 @@
 from agents import Direction
 from agents import Agent
 from agents import ReflexVacuumAgent, ModelBasedVacuumAgent, TrivialVacuumEnvironment, compare_agents,\
-                   RandomVacuumAgent, TableDrivenVacuumAgent
+                   RandomVacuumAgent, TableDrivenVacuumAgent, TableDrivenAgentProgram, RandomAgentProgram
 
 
 random.seed("aima-python")
@@ -54,6 +54,21 @@ def test_add():
     assert l1.direction == Direction.U
     assert l2.direction == Direction.D
 
+def test_RandomAgentProgram() :
+    #create a list of all the actions a vacuum cleaner can perform
+    list = ['Right', 'Left', 'Suck', 'NoOp']
+    # create a program and then an object of the RandomAgentProgram
+    program = RandomAgentProgram(list)
+    
+    agent = Agent(program)
+    # create an object of TrivialVacuumEnvironment
+    environment = TrivialVacuumEnvironment()
+    # add agent to the environment
+    environment.add_thing(agent)
+    # run the environment
+    environment.run()
+    # check final status of the environment
+    assert environment.status == {(1, 0): 'Clean' , (0, 0): 'Clean'}
 
 def test_RandomVacuumAgent() :
     # create an object of the RandomVacuumAgent
@@ -68,6 +83,34 @@ def test_RandomVacuumAgent() :
     assert environment.status == {(1,0):'Clean' , (0,0) : 'Clean'}
 
 
+def test_TableDrivenAgent() :
+    #create a table that would consist of all the possible states of the agent
+    loc_A, loc_B = (0, 0), (1, 0)
+    
+    table = {((loc_A, 'Clean'),): 'Right',
+             ((loc_A, 'Dirty'),): 'Suck',
+             ((loc_B, 'Clean'),): 'Left',
+             ((loc_B, 'Dirty'),): 'Suck',
+             ((loc_A, 'Dirty'), (loc_A, 'Clean')): 'Right',
+             ((loc_A, 'Clean'), (loc_B, 'Dirty')): 'Suck',
+             ((loc_B, 'Clean'), (loc_A, 'Dirty')): 'Suck',
+             ((loc_B, 'Dirty'), (loc_B, 'Clean')): 'Left',
+             ((loc_A, 'Dirty'), (loc_A, 'Clean'), (loc_B, 'Dirty')): 'Suck',
+             ((loc_B, 'Dirty'), (loc_B, 'Clean'), (loc_A, 'Dirty')): 'Suck'
+             }
+    # create an program and then an object of the TableDrivenAgent
+    program = TableDrivenAgentProgram(table)
+    agent = Agent(program)
+    # create an object of the TrivialVacuumEnvironment
+    environment = TrivialVacuumEnvironment()
+    # add agent to the environment
+    environment.add_thing(agent)
+    # run the environment
+    environment.run()
+    # check final status of the environment
+    assert environment.status == {(1, 0): 'Clean', (0, 0): 'Clean'}
+
+
 def test_ReflexVacuumAgent() :
     # create an object of the ReflexVacuumAgent
     agent = ReflexVacuumAgent()
