Add test for TableDrivenAgentProgram.

roberthoenig · roberthoenig · commit e1a4b2d76b17 · 2018-02-16T13:24:25.000+01:00
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]           |      | 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         |
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
+                   RandomVacuumAgent, TableDrivenAgentProgram
 
 
 random.seed("aima-python")
@@ -111,6 +111,20 @@ def test_compare_agents() :
     assert performance_ReflexVacummAgent <= performance_ModelBasedVacummAgent
 
 
+def test_TableDrivenAgentProgram():
+    table = {(('foo', 1),): 'action1',
+             (('foo', 2),): 'action2',
+             (('bar', 1),): 'action3',
+             (('bar', 2),): 'action1',
+             (('foo', 1), ('foo', 1),): 'action2',
+             (('foo', 1), ('foo', 2),): 'action3',
+             }
+    agent_program = TableDrivenAgentProgram(table)
+    assert agent_program(('foo', 1)) == 'action1'
+    assert agent_program(('foo', 2)) == 'action3'
+    assert agent_program(('invalid percept',)) == None
+
+
 def test_Agent():
     def constant_prog(percept):
         return percept
