refactor: Added type hints and docstrings to helper functions. Remove…

…d unused helper functions

tests/test_plots.py

@@ -76,18 +76,18 @@ def is_github_ci():
 show_plots = False
 
 
-# class TestPlots(unittest.TestCase):
-#
-#     def __init__(self, *args, **kwargs):
-#         super().__init__(*args, **kwargs)  # Call the parent constructor
-#         # Define the save directory
-#         self.artifacts = os.path.join(os.path.dirname(__file__), "artifacts")
-#         self.save_dir = os.path.join(self.artifacts, "plots")
-#         os.makedirs(self.save_dir, exist_ok=True)
-#
-#     def savefig(self, ax, name):
-#         ax.figure.savefig(os.path.join(self.save_dir, name))
-#
+class TestPlots(unittest.TestCase):
+
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)  # Call the parent constructor
+        # Define the save directory
+        self.artifacts = os.path.join(os.path.dirname(__file__), "artifacts")
+        self.save_dir = os.path.join(self.artifacts, "plots")
+        os.makedirs(self.save_dir, exist_ok=True)
+
+    def savefig(self, ax, name):
+        ax.figure.savefig(os.path.join(self.save_dir, name))
+
 #
 # class TestTimeSeriesPlots(TestPlots):
 #