🎨 Format Python code with psf/black

Author: ccaprani

src/ospgrid/grid.py

@@ -225,14 +225,14 @@ def __init__(self):
 
         """
         self.clear()
-        
+
         # Set color preferences
         ospv.fmt_undefo["color"] = "black"
         ospv.fmt_defo["color"] = "red"
         ospv.fmt_model["color"] = "black"
         ospv.fmt_secforce1["color"] = "red"
         ospv.fmt_secforce2["color"] = "red"
-        
+
         pass
 
     def clear(self):
@@ -793,8 +793,7 @@ def plot_grid(
         if figsize is None:
             figsize = self.FIGSIZE
 
-        ospv.plot_model(fig_wi_he=figsize,
-                        node_supports=False)
+        ospv.plot_model(fig_wi_he=figsize, node_supports=False)
         fig = plt.gcf()
         fig.set_figwidth(figsize[0])
         fig.set_figheight(figsize[1])
@@ -805,19 +804,20 @@ def plot_grid(
             plt.gca().set_axis_off()
 
         fig.tight_layout()
-        
+
     def _plot_model(self, ax):
         """
         Plots the background model for the section force diagrams.
         """
-        
-        ospv.plot_model(ax=ax,
-                        node_labels=False,
-                        element_labels=False,
-                        node_supports=False,
-                        #fmt_model={"color":"k"},
-                        local_axes=False)
 
+        ospv.plot_model(
+            ax=ax,
+            node_labels=False,
+            element_labels=False,
+            node_supports=False,
+            # fmt_model={"color":"k"},
+            local_axes=False,
+        )
 
     def plot_dsd(
         self,
@@ -857,25 +857,24 @@ def plot_dsd(
             grid_size = max(max(x) - min(x), max(y) - min(y))
 
             # in case of very small nodal values
-            max_disp = ospv.max_u_abs_from_beam_defo_interp_3d(max_disp,nep=21)
-            
+            max_disp = ospv.max_u_abs_from_beam_defo_interp_3d(max_disp, nep=21)
+
             # target about 1/4 the dimension of the grid
             sf = 0.25 * grid_size / max_disp
             # But round to some sensible values
             mag = 10 ** int(np.ceil(np.log10(sf)))
             scale_factor = round(10 * sf / mag) * mag / 10
 
-        
-
         fig = plt.figure()
         ax = fig.add_subplot(111, projection=Axes3D.name)
-        self._plot_model(ax)        
-        ospv.plot_defo(sfac=scale_factor,
-                       unDefoFlag=False,
-                       ax=ax,
-                       endDispFlag=False,
-                       node_supports=False
-                       )
+        self._plot_model(ax)
+        ospv.plot_defo(
+            sfac=scale_factor,
+            unDefoFlag=False,
+            ax=ax,
+            endDispFlag=False,
+            node_supports=False,
+        )
         fig = plt.gcf()
         fig.set_figwidth(figsize[0])
         fig.set_figheight(figsize[1])
@@ -918,11 +917,13 @@ def plot_bmd(
         if figsize is None:
             figsize = self.FIGSIZE
 
-        _,_,ax = ospv.section_force_diagram_3d("My", 
-                                      sfac=scale_factor, 
-                                      end_max_values=values,
-                                      node_supports=False,
-                                      alt_model_plot=2)
+        _, _, ax = ospv.section_force_diagram_3d(
+            "My",
+            sfac=scale_factor,
+            end_max_values=values,
+            node_supports=False,
+            alt_model_plot=2,
+        )
         self._plot_model(ax)
         fig = plt.gcf()
         fig.set_figwidth(figsize[0])
@@ -969,11 +970,13 @@ def plot_sfd(
         if figsize is None:
             figsize = self.FIGSIZE
 
-        _,_,ax = ospv.section_force_diagram_3d("Vz", 
-                                      sfac=-scale_factor, 
-                                      end_max_values=values,
-                                      node_supports=False,
-                                      alt_model_plot=2)
+        _, _, ax = ospv.section_force_diagram_3d(
+            "Vz",
+            sfac=-scale_factor,
+            end_max_values=values,
+            node_supports=False,
+            alt_model_plot=2,
+        )
         self._plot_model(ax)
         fig = plt.gcf()
         fig.set_figwidth(figsize[0])
@@ -1020,14 +1023,16 @@ def plot_tmd(
         if figsize is None:
             figsize = self.FIGSIZE
 
-        _,_,ax = ospv.section_force_diagram_3d("T", 
-                                      sfac=-scale_factor, 
-                                      dir_plt=2, 
-                                      end_max_values=values, 
-                                      node_supports=False,
-                                      #fmt_secforce1={"color":"r"},
-                                      #fmt_secforce2={"color":"r"},
-                                      alt_model_plot=2)
+        _, _, ax = ospv.section_force_diagram_3d(
+            "T",
+            sfac=-scale_factor,
+            dir_plt=2,
+            end_max_values=values,
+            node_supports=False,
+            # fmt_secforce1={"color":"r"},
+            # fmt_secforce2={"color":"r"},
+            alt_model_plot=2,
+        )
         self._plot_model(ax)
         fig = plt.gcf()
         fig.set_figwidth(figsize[0])

tests/test_basic.py

@@ -70,7 +70,6 @@ def getF():
 
 
 def do_ospg_analysis():
-
     params = grid_params()
     Lae, Lbe, Lce, Lde, EI, GJ, P, Mx, My = params