added unit test for job_shop_problem

aimacode · kaivalyar · Mar 26, 2017 · Mar 26, 2017 · Mar 26, 2017 · Mar 26, 2017
commit 27425fce83f1a5f628b081c471843e01dfb663f6
diff --git a/planning.py b/planning.py
@@ -681,7 +681,7 @@ def act(self, action):
         if list_action is None:
             raise Exception("Action '{}' not found".format(action.name))
         list_action.do_action(self.jobs, self.resources, self.kb, args)
-        print(self.resources)
+        # print(self.resources)
 
 
 def job_shop_problem():

diff --git a/tests/test_planning.py b/tests/test_planning.py
@@ -81,3 +81,20 @@ def test_graph_call():
     graph()
 
     assert levels_size == len(graph.levels) - 1
+
+
+def test_job_shop_problem():
+    p = job_shop_problem()
+    assert p.goal_test() is False
+
+    solution = [p.jobs[1][0],
+                p.jobs[0][0],
+                p.jobs[0][1],
+                p.jobs[0][2],
+                p.jobs[1][1],
+                p.jobs[1][2]]
+
+    for action in solution:
+        p.act(action)
+
+    assert p.goal_test()