Formatting

Author: max-models

src/maxplotlib/backends/matplotlib/utils_old.py

@@ -31,7 +31,6 @@ def __init__(self, hex_color):
         self.pgf_col_str += self.rgb_dec_str[2] + "}%"
 
     def invert(self):
-
         inverted_color = Color(self.hx)
 
     def define_color_str(self, name):
@@ -261,7 +260,8 @@ def create_lineplot(self):
             self.ny_subplots,
             self.nx_subplots,
             figsize=self.set_size(
-                self.width, ratio=self.ratio  #
+                self.width,
+                ratio=self.ratio,  #
             ),  # sharex=True,#sharex='all', sharey='all',
             dpi=self.dpi,
             constrained_layout=False,
@@ -280,7 +280,6 @@ def setup_plotstyle(
         grid_alpha=0.0,
         grid_linestyle="dotted",
     ):
-
         if tex_fonts:
             plt.rcParams.update(self.tex_fonts)
 
@@ -405,7 +404,6 @@ def scale_axis(
         nticks=5,
         locs_labels=None,
     ):
-
         # https://matplotlib.org/3.1.1/api/_as_gen/matplotlib.pyplot.xticks.html
         # if subfigure_x == -1 and subfigure_y == -1 and nx_subplots > 1 and ny_subplots > 1:
         #    print('enter subfigure_x and subfigure_y!')
@@ -596,7 +594,6 @@ def savefig(
 
         if format_folder:
             for format in formats:
-
                 # print('self.directory',self.directory)
                 Path(self.directory + "/" + format).mkdir(parents=True, exist_ok=True)
 
@@ -764,7 +761,6 @@ def dump(
         self,
         filename="",
     ):
-
         if filename == "":
             filename = self.filename + "_dump.txt"
         Path(self.directory + "/dump").mkdir(parents=True, exist_ok=True)

src/maxplotlib/canvas/canvas.py

@@ -44,6 +44,7 @@ def layers(self):
         for (row, col), subplot in self.subplots.items():
             layers.extend(subplot.layers)
         return list(set(layers))
+
     def generate_new_rowcol(self, row, col):
         if row is None:
             for irow in range(self.nrows):
@@ -115,21 +116,32 @@ def add_subplot(self, **kwargs):
             self.subplots[label] = line_plot
         return line_plot
 
-    def savefig(self, filename, extension='pdf', backend="matplotlib", layers = None, layer_by_layer=False):
+    def savefig(
+        self,
+        filename,
+        extension="pdf",
+        backend="matplotlib",
+        layers=None,
+        layer_by_layer=False,
+    ):
         if backend == "matplotlib":
             if layer_by_layer:
                 layers = []
                 for layer in self.layers:
                     layers.append(layer)
-                    fig, axs = self.plot(show=False, backend="matplotlib", savefig=True, layers=layers)
+                    fig, axs = self.plot(
+                        show=False, backend="matplotlib", savefig=True, layers=layers
+                    )
                     fig.savefig(f"{filename}_{layers}.{extension}")
             else:
                 if layers is None:
                     layers = self.layers
                     full_filepath = f"{filename}.{extension}"
                 else:
                     full_filepath = f"{filename}_{layers}.{extension}"
-                fig, axs = self.plot(show=False, backend="matplotlib", savefig=True, layers=layers)
+                fig, axs = self.plot(
+                    show=False, backend="matplotlib", savefig=True, layers=layers
+                )
                 fig.savefig(full_filepath)
 
     # def add_line(self, label, x_data, y_data, **kwargs):

src/maxplotlib/subfigure/line_plot.py

@@ -39,7 +39,7 @@ def __init__(self, **kwargs):
     def add_caption(self, caption):
         self._caption = caption
 
-    def add_line(self, x_data, y_data, layer = 0, **kwargs):
+    def add_line(self, x_data, y_data, layer=0, **kwargs):
         """
         Add a line to the plot.
 
@@ -49,7 +49,12 @@ def add_line(self, x_data, y_data, layer = 0, **kwargs):
         y_data (list): Y-axis data.
         **kwargs: Additional keyword arguments for the plot (e.g., color, linestyle).
         """
-        ld = {"x": np.array(x_data), "y": np.array(y_data), "layer": layer, "kwargs": kwargs}
+        ld = {
+            "x": np.array(x_data),
+            "y": np.array(y_data),
+            "layer": layer,
+            "kwargs": kwargs,
+        }
         self.line_data.append(ld)
         if layer in self.layered_line_data:
             self.layered_line_data[layer].append(ld)
@@ -62,6 +67,7 @@ def layers(self):
         for layer_name, layer_lines in self.layered_line_data.items():
             layers.append(layer_name)
         return layers
+
     def plot_matplotlib(self, ax, layers=None):
         """
         Plot all lines on the provided axis.

src/maxplotlib/subfigure/subfigure.py

@@ -1,3 +1,3 @@
 class Subfigure:
     def __init__(self, **kwargs):
-        self.kwargs = kwargs
+        self.kwargs = kwargs

src/maxplotlib/tests/test_imports.py

@@ -4,7 +4,6 @@
 @pytest.mark.parametrize("x", [0])
 def import_modules(x):
     import matplotlib
-
     import maxplotlib