Fix title cutting off, move comment back

Author: Agent6-6-6

sectionproperties/post/post.py

@@ -7,7 +7,9 @@
 
 
 @contextlib.contextmanager
-def plotting_context(ax=None, pause=True, title="", filename="", render=True, axis_index=None, **kwargs):
+def plotting_context(
+    ax=None, pause=True, title="", filename="", render=True, axis_index=None, **kwargs
+):
     """Executes code required to set up a matplotlib figure.
 
     :param ax: Axes object on which to plot
@@ -47,7 +49,9 @@ def plotting_context(ax=None, pause=True, title="", filename="", render=True, ax
         except (AttributeError, TypeError):
             pass  # only 1 axis, not an array
         except IndexError as exc:
-            raise ValueError(f"axis_index={axis_index} is not compatible with arguments to subplots: {kwargs}") from exc
+            raise ValueError(
+                f"axis_index={axis_index} is not compatible with arguments to subplots: {kwargs}"
+            ) from exc
     else:
         fig = ax.get_figure()
         ax_supplied = True
@@ -56,12 +60,11 @@ def plotting_context(ax=None, pause=True, title="", filename="", render=True, ax
 
     yield fig, ax
 
-    # if no axes was supplied, finish the plot and return the figure and axes
+    ax.set_title(title)
     plt.tight_layout()
-
     ax.set_aspect("equal", anchor="C")
-    ax.set_title(title)
 
+    # if no axes was supplied, finish the plot and return the figure and axes
     if ax_supplied:
         # if an axis was supplied, don't continue with displaying or configuring the plot
         return