fix test

Author: guilyx

tests/test_automated_planner.py

@@ -44,3 +44,41 @@ def test_state_assertion():
         "pddl-examples/dinner/domain.pddl", "pddl-examples/dinner/problem.pddl"
     )
     assert not apla.satisfies(apla.problem.goal, apla.initial_state)
+
+def test_bfs():
+    apla = AutomatedPlanner(
+        "pddl-examples/dinner/domain.pddl", "pddl-examples/dinner/problem.pddl"
+    )
+    path, _, _ = apla.breadth_first_search()
+    plan = apla.get_actions_from_path(path)
+    plan_state = apla.get_state_def_from_path(path)
+    assert plan and plan_state
+
+def test_dfs():
+    apla = AutomatedPlanner(
+        "pddl-examples/dinner/domain.pddl", "pddl-examples/dinner/problem.pddl"
+    )
+    path, _, _ = apla.depth_first_search()
+    plan = apla.get_actions_from_path(path)
+    plan_state = apla.get_state_def_from_path(path)
+    assert plan and plan_state
+
+def test_dij():
+    apla = AutomatedPlanner(
+        "pddl-examples/dinner/domain.pddl", "pddl-examples/dinner/problem.pddl"
+    )
+    path, _, _ = apla.dijktra_best_first_search()
+    plan = apla.get_actions_from_path(path)
+    plan_state = apla.get_state_def_from_path(path)
+    assert plan and plan_state
+
+
+def test_astar():
+    apla = AutomatedPlanner(
+        "pddl-examples/dinner/domain.pddl", "pddl-examples/dinner/problem.pddl"
+    )
+    path, _, _ = apla.astar_best_first_search()
+    plan = apla.get_actions_from_path(path)
+    plan_state = apla.get_state_def_from_path(path)
+    assert plan and plan_state
+

tests/test_basic_search.py

@@ -87,27 +87,4 @@ def test_search_astar_basic_no_path():
 
 def test_zero_heuristic():
     assert zero_heuristic() == 0
-
-
-def test_search_getter_costs():
-    apla = AutomatedPlanner(
-        "pddl-examples/tsp/domain.pddl", "pddl-examples/tsp/problem.pddl"
-    )
-    bfs = BreadthFirstSearch(apla)
-    path, _, _ = bfs.search()  # Path, computation time, opened nodes
-    plan = apla.get_actions_from_path(path)
-    state_plan = apla.get_state_def_from_path(path)
-
-    assert path and plan and state_plan
-
-
-def test_search_getter_no_costs():
-    apla = AutomatedPlanner(
-        "pddl-examples/cargo/domain.pddl", "pddl-examples/cargo/problem.pddl"
-    )
-    bfs = BreadthFirstSearch(apla)
-    path, _, _ = bfs.search()  # Path, computation time, opened nodes
-    plan = apla.get_actions_from_path(path)
-    state_plan = apla.get_state_def_from_path(path)
-
-    assert path and plan and state_plan
+