Apply formatting changes

Author: guilyx

jupyddl/data_analyst.py

@@ -27,15 +27,17 @@ def __get_all_pddl_from_data(self):
         if "DISPLAY" in os.environ:
             for root, _, files in os.walk("pddl-examples/", topdown=False):
                 for name in files:
-                    #if ".gitkeep" in name:
+                    # if ".gitkeep" in name:
                     #    continue
                     tested_files.append(os.getcwd() + "/" + os.path.join(root, name))
                     if i % 2 != 0:
                         domains_problems.append((tested_files[i - 1], tested_files[i]))
                     i += 1
             return domains_problems
-        return [("pddl-examples/flip/problem.pddl", "pddl-examples/flip/domain.pddl"),
-                ("pddl-examples/dinner/problem.pddl", "pddl-examples/dinner/domain.pddl")]
+        return [
+            ("pddl-examples/flip/problem.pddl", "pddl-examples/flip/domain.pddl"),
+            ("pddl-examples/dinner/problem.pddl", "pddl-examples/dinner/domain.pddl"),
+        ]
 
     def __plot_data(self, times, total_nodes, plot_title):
         data = dict()

tests/test_astar.py

@@ -9,20 +9,26 @@
 
 
 def test_astar_init():
-    apla = AutomatedPlanner("pddl-examples/flip/domain.pddl", "pddl-examples/flip/problem.pddl")
+    apla = AutomatedPlanner(
+        "pddl-examples/flip/domain.pddl", "pddl-examples/flip/problem.pddl"
+    )
     astar = AStarBestFirstSearch(apla, apla.available_heuristics["goal_count"])
     assert astar.init.h_cost == apla.available_heuristics["goal_count"](
         apla.initial_state, apla
     )
 
 
 def test_astar_goal():
-    apla = AutomatedPlanner("pddl-examples/flip/domain.pddl", "pddl-examples/flip/problem.pddl")
+    apla = AutomatedPlanner(
+        "pddl-examples/flip/domain.pddl", "pddl-examples/flip/problem.pddl"
+    )
     path, _, _ = apla.astar_best_first_search()
     assert apla.available_heuristics["goal_count"](path[-1].state, apla) == 0
 
 
 def test_astar_path_length():
-    apla = AutomatedPlanner("pddl-examples/flip/domain.pddl", "pddl-examples/flip/problem.pddl")
+    apla = AutomatedPlanner(
+        "pddl-examples/flip/domain.pddl", "pddl-examples/flip/problem.pddl"
+    )
     path, _, _ = apla.astar_best_first_search()
     assert len(path) > 0

tests/test_automated_planner.py

@@ -8,29 +8,39 @@
 
 
 def test_parsing():
-    apla = AutomatedPlanner("pddl-examples/flip/domain.pddl", "pddl-examples/flip/problem.pddl")
+    apla = AutomatedPlanner(
+        "pddl-examples/flip/domain.pddl", "pddl-examples/flip/problem.pddl"
+    )
     assert str(apla.problem) != "" and str(apla.domain) != ""
 
 
 def test_available_actions():
-    apla = AutomatedPlanner("pddl-examples/flip/domain.pddl", "pddl-examples/flip/problem.pddl")
+    apla = AutomatedPlanner(
+        "pddl-examples/flip/domain.pddl", "pddl-examples/flip/problem.pddl"
+    )
     actions = apla.available_actions(apla.initial_state)
     assert len(actions) > 0
 
 
 def test_execute_action():
-    apla = AutomatedPlanner("pddl-examples/flip/domain.pddl", "pddl-examples/flip/problem.pddl")
+    apla = AutomatedPlanner(
+        "pddl-examples/flip/domain.pddl", "pddl-examples/flip/problem.pddl"
+    )
     actions = apla.available_actions(apla.initial_state)
     new_state = apla.transition(apla.initial_state, actions[0])
     assert str(new_state) != str(apla.initial_state)
 
 
 def test_state_has_term():
-    apla = AutomatedPlanner("pddl-examples/flip/domain.pddl", "pddl-examples/flip/problem.pddl")
+    apla = AutomatedPlanner(
+        "pddl-examples/flip/domain.pddl", "pddl-examples/flip/problem.pddl"
+    )
     is_goal = apla.state_has_term(apla.initial_state, apla.goals[0])
     assert not is_goal
 
 
 def test_state_assertion():
-    apla = AutomatedPlanner("pddl-examples/flip/domain.pddl", "pddl-examples/flip/problem.pddl")
+    apla = AutomatedPlanner(
+        "pddl-examples/flip/domain.pddl", "pddl-examples/flip/problem.pddl"
+    )
     assert not apla.satisfies(apla.problem.goal, apla.initial_state)

tests/test_data_analyst.py

@@ -11,27 +11,40 @@ def test_data_analyst_constructor():
     _ = DataAnalyst()
     assert True
 
+
 def test_data_analyst_plot_dfs_one_pddl():
     da = DataAnalyst()
-    da.plot_dfs(domain="pddl-examples/flip/domain.pddl", problem="pddl-examples/flip/problem.pddl")
+    da.plot_dfs(
+        domain="pddl-examples/flip/domain.pddl",
+        problem="pddl-examples/flip/problem.pddl",
+    )
     assert True
 
 
 def test_data_analyst_plot_bfs_one_pddl():
     da = DataAnalyst()
-    da.plot_bfs(domain="pddl-examples/flip/domain.pddl", problem="pddl-examples/flip/problem.pddl")
+    da.plot_bfs(
+        domain="pddl-examples/flip/domain.pddl",
+        problem="pddl-examples/flip/problem.pddl",
+    )
     assert True
 
 
 def test_data_analyst_plot_dijkstra_one_pddl():
     da = DataAnalyst()
-    da.plot_dijkstra(domain="pddl-examples/flip/domain.pddl", problem="pddl-examples/flip/problem.pddl")
+    da.plot_dijkstra(
+        domain="pddl-examples/flip/domain.pddl",
+        problem="pddl-examples/flip/problem.pddl",
+    )
     assert True
 
 
 def test_data_analyst_plot_astar_h_goal_count_one_pddl():
     da = DataAnalyst()
-    da.plot_astar_data(domain="pddl-examples/flip/domain.pddl", problem="pddl-examples/flip/problem.pddl")
+    da.plot_astar_data(
+        domain="pddl-examples/flip/domain.pddl",
+        problem="pddl-examples/flip/problem.pddl",
+    )
     assert True
 
 
@@ -98,22 +111,29 @@ def test_comparative_no_dfs():
 def test_comparative_one_pddl():
     da = DataAnalyst()
     da.comparative_data_plot(
-        dfs=False, bfs=False, domain="pddl-examples/flip/domain.pddl", problem="pddl-examples/flip/problem.pddl"
+        dfs=False,
+        bfs=False,
+        domain="pddl-examples/flip/domain.pddl",
+        problem="pddl-examples/flip/problem.pddl",
     )
     assert True
 
 
 def test_comparative_use_data_json():
     da = DataAnalyst()
     da.comparative_data_plot(
-        domain="pddl-examples/flip/domain.pddl", problem="pddl-examples/flip/problem.pddl", collect_new_data=False
+        domain="pddl-examples/flip/domain.pddl",
+        problem="pddl-examples/flip/problem.pddl",
+        collect_new_data=False,
     )
     assert True
 
 
 def test_comparative_zero_h():
     da = DataAnalyst()
     da.comparative_data_plot(
-        domain="pddl-examples/flip/domain.pddl", problem="pddl-examples/flip/problem.pddl", heuristic_key="zero"
+        domain="pddl-examples/flip/domain.pddl",
+        problem="pddl-examples/flip/problem.pddl",
+        heuristic_key="zero",
     )
     assert True

tests/test_heuristics.py

@@ -15,12 +15,16 @@
 
 
 def test_zero_heuristic():
-    apla = AutomatedPlanner("pddl-examples/flip/domain.pddl", "pddl-examples/flip/problem.pddl")
+    apla = AutomatedPlanner(
+        "pddl-examples/flip/domain.pddl", "pddl-examples/flip/problem.pddl"
+    )
     heuristic = hs.zero_heuristic(apla.initial_state, apla)
     assert heuristic == 0
 
 
 def test_goal_count_heuristic():
-    apla = AutomatedPlanner("pddl-examples/flip/domain.pddl", "pddl-examples/flip/problem.pddl")
+    apla = AutomatedPlanner(
+        "pddl-examples/flip/domain.pddl", "pddl-examples/flip/problem.pddl"
+    )
     heuristic = hs.goal_count_heuristic(apla.initial_state, apla)
     assert heuristic == 1