add headless mode

Author: guilyx

jupyddl/data_analyst.py

@@ -6,16 +6,25 @@
 mpl.use("TkAgg")
 mpl.set_loglevel("WARNING")
 import matplotlib.pyplot as plt
-
 plt.style.use("ggplot")
 from .automated_planner import AutomatedPlanner
 from os import path
 import json
 
 
 class DataAnalyst:
-    def __init__(self):
+    def __init__(self, is_headless=False):
         logging.info("Instantiating data analyst...")
+        self.is_headless = is_headless
+        if is_headless:
+            mpl.set_loglevel("WARNING")
+            import matplotlib.pyplot as plt
+            plt.style.use("ggplot")
+        else:
+            mpl.use("TkAgg")
+            mpl.set_loglevel("WARNING")
+            import matplotlib.pyplot as plt
+            plt.style.use("ggplot")
 
     def __get_all_pddl_from_data(self):
         tested_files = []

tests/test_data_analyst.py

@@ -8,88 +8,88 @@
 
 
 def test_data_analyst_constructor():
-    _ = DataAnalyst()
+    _ = DataAnalyst(is_headless=True)
     assert True
 
 
 def test_data_analyst_plot_dfs():
-    da = DataAnalyst()
+    da = DataAnalyst(is_headless=True)
     da.plot_dfs()
     assert True
 
 
 def test_data_analyst_plot_bfs():
-    da = DataAnalyst()
+    da = DataAnalyst(is_headless=True)
     da.plot_bfs()
     assert True
 
 
 def test_data_analyst_plot_dijkstra():
-    da = DataAnalyst()
+    da = DataAnalyst(is_headless=True)
     da.plot_dijkstra()
     assert True
 
 
 def test_data_analyst_plot_astar_h_goal_count():
-    da = DataAnalyst()
+    da = DataAnalyst(is_headless=True)
     da.plot_astar_data()
     assert True
 
 
 def test_data_analyst_plot_astar_h_zero():
-    da = DataAnalyst()
+    da = DataAnalyst(is_headless=True)
     da.plot_astar_data(heuristic_key="zero")
     assert True
 
 
 def test_comparative_no_restrictions():
-    da = DataAnalyst()
+    da = DataAnalyst(is_headless=True)
     da.comparative_data_plot()
     assert True
 
 
 def test_comparative_no_astar():
-    da = DataAnalyst()
+    da = DataAnalyst(is_headless=True)
     da.comparative_data_plot(astar=False)
     assert True
 
 
 def test_comparative_no_bfs():
-    da = DataAnalyst()
+    da = DataAnalyst(is_headless=True)
     da.comparative_data_plot(bfs=False)
     assert True
 
 
 def test_comparative_no_dijkstra():
-    da = DataAnalyst()
+    da = DataAnalyst(is_headless=True)
     da.comparative_data_plot(dijkstra=False)
     assert True
 
 
 def test_comparative_no_dfs():
-    da = DataAnalyst()
+    da = DataAnalyst(is_headless=True)
     da.comparative_data_plot(dfs=False)
     assert True
 
 
 def test_comparative_one_pddl():
-    da = DataAnalyst()
+    da = DataAnalyst(is_headless=True)
     da.comparative_data_plot(
         dfs=False, bfs=False, domain="data/domain.pddl", problem="data/problem.pddl"
     )
     assert True
 
 
 def test_comparative_use_data_json():
-    da = DataAnalyst()
+    da = DataAnalyst(is_headless=True)
     da.comparative_data_plot(
         domain="data/domain.pddl", problem="data/problem.pddl", collect_new_data=False
     )
     assert True
 
 
 def test_comparative_zero_h():
-    da = DataAnalyst()
+    da = DataAnalyst(is_headless=True)
     da.comparative_data_plot(
         domain="data/domain.pddl", problem="data/problem.pddl", heuristic_key="zero"
     )