Added test for simpleProblemSolvingAgentProgram (#784)

nouman-10 · norvig · commit 3f888808bea2 · 2018-03-01T16:53:52.000-08:00
* Added test for simpleProblemSolvingAgent

* Some Style fixes

* Fixed update_state in test_search.py
diff --git a/tests/test_search.py b/tests/test_search.py
@@ -201,6 +201,50 @@ def GA_GraphColoringInts(edges, fitness):
     return genetic_algorithm(population, fitness)
 
 
+def test_simpleProblemSolvingAgent():
+    class vacuumAgent(SimpleProblemSolvingAgentProgram):
+        def update_state(self, state, percept):
+            return percept
+
+        def formulate_goal(self, state):
+            goal = [state7, state8]
+            return goal  
+
+        def formulate_problem(self, state, goal):
+            problem = state
+            return problem   
+    
+        def search(self, problem):
+            if problem == state1:
+                seq = ["Suck", "Right", "Suck"]
+            elif problem == state2:
+                seq = ["Suck", "Left", "Suck"]
+            elif problem == state3:
+                seq = ["Right", "Suck"]
+            elif problem == state4:
+                seq = ["Suck"]
+            elif problem == state5:
+                seq = ["Suck"]
+            elif problem == state6:
+                seq = ["Left", "Suck"]
+            return seq
+
+    state1 = [(0, 0), [(0, 0), "Dirty"], [(1, 0), ["Dirty"]]]
+    state2 = [(1, 0), [(0, 0), "Dirty"], [(1, 0), ["Dirty"]]]
+    state3 = [(0, 0), [(0, 0), "Clean"], [(1, 0), ["Dirty"]]]
+    state4 = [(1, 0), [(0, 0), "Clean"], [(1, 0), ["Dirty"]]]
+    state5 = [(0, 0), [(0, 0), "Dirty"], [(1, 0), ["Clean"]]]
+    state6 = [(1, 0), [(0, 0), "Dirty"], [(1, 0), ["Clean"]]]
+    state7 = [(0, 0), [(0, 0), "Clean"], [(1, 0), ["Clean"]]]
+    state8 = [(1, 0), [(0, 0), "Clean"], [(1, 0), ["Clean"]]]
+
+    a = vacuumAgent(state1)
+
+    assert a(state6) == "Left"
+    assert a(state1) == "Suck"
+    assert a(state3) == "Right"
+    
+
 
 # TODO: for .ipynb:
 """
