formatting

Author: max-models

src/maxplotlib/tests/test_canvas.py

@@ -79,8 +79,8 @@ def test_canvas_plot_tikzfigure_vertical_not_supported():
 
 def test_canvas_matplotlib_gridspec_kw_affects_row_spacing():
     """Test that hspace changes the vertical spacing between rows."""
-    import numpy as np
     import matplotlib.pyplot as plt
+    import numpy as np
 
     from maxplotlib import Canvas
 
@@ -123,8 +123,8 @@ def test_canvas_matplotlib_gridspec_kw_affects_row_spacing():
 
 def test_canvas_matplotlib_gridspec_kw_affects_2x2_line_spacing():
     """Test that wspace/hspace change spacing for 2×2 line subplot grids."""
-    import numpy as np
     import matplotlib.pyplot as plt
+    import numpy as np
 
     from maxplotlib import Canvas
 
@@ -184,8 +184,8 @@ def test_canvas_matplotlib_gridspec_kw_affects_2x2_line_spacing():
 
 def test_canvas_matplotlib_gridspec_kw_affects_2x2_imshow_spacing():
     """Test spacing control also works for 2×2 color (imshow) subplot grids."""
-    import numpy as np
     import matplotlib.pyplot as plt
+    import numpy as np
 
     from maxplotlib import Canvas
 

src/maxplotlib/tests/test_plot.py

@@ -20,8 +20,12 @@ def test_python_example_nxm_line_subplots_spacing_changes():
         for j, ax in enumerate(row):
             ax.plot(x, np.sin((i + 1) * (j + 1) * x))
     tight_fig, tight_m_axes = tight_canvas.plot(backend="matplotlib")
-    tight_hgap = tight_m_axes[0, 1].get_position().x0 - tight_m_axes[0, 0].get_position().x1
-    tight_vgap = tight_m_axes[0, 0].get_position().y0 - tight_m_axes[1, 0].get_position().y1
+    tight_hgap = (
+        tight_m_axes[0, 1].get_position().x0 - tight_m_axes[0, 0].get_position().x1
+    )
+    tight_vgap = (
+        tight_m_axes[0, 0].get_position().y0 - tight_m_axes[1, 0].get_position().y1
+    )
 
     loose_canvas, loose_axes = Canvas.subplots(
         nrows=2,
@@ -35,8 +39,12 @@ def test_python_example_nxm_line_subplots_spacing_changes():
         for j, ax in enumerate(row):
             ax.plot(x, np.sin((i + 1) * (j + 1) * x))
     loose_fig, loose_m_axes = loose_canvas.plot(backend="matplotlib")
-    loose_hgap = loose_m_axes[0, 1].get_position().x0 - loose_m_axes[0, 0].get_position().x1
-    loose_vgap = loose_m_axes[0, 0].get_position().y0 - loose_m_axes[1, 0].get_position().y1
+    loose_hgap = (
+        loose_m_axes[0, 1].get_position().x0 - loose_m_axes[0, 0].get_position().x1
+    )
+    loose_vgap = (
+        loose_m_axes[0, 0].get_position().y0 - loose_m_axes[1, 0].get_position().y1
+    )
 
     assert loose_hgap > tight_hgap
     assert loose_vgap > tight_vgap
@@ -62,8 +70,12 @@ def test_python_example_nxm_color_subplots_spacing_changes():
             ax.add_imshow(base + idx, cmap="viridis")
             idx += 1
     tight_fig, tight_m_axes = tight_canvas.plot(backend="matplotlib")
-    tight_hgap = tight_m_axes[0, 1].get_position().x0 - tight_m_axes[0, 0].get_position().x1
-    tight_vgap = tight_m_axes[0, 0].get_position().y0 - tight_m_axes[1, 0].get_position().y1
+    tight_hgap = (
+        tight_m_axes[0, 1].get_position().x0 - tight_m_axes[0, 0].get_position().x1
+    )
+    tight_vgap = (
+        tight_m_axes[0, 0].get_position().y0 - tight_m_axes[1, 0].get_position().y1
+    )
 
     loose_canvas, loose_axes = Canvas.subplots(
         nrows=2,
@@ -79,8 +91,12 @@ def test_python_example_nxm_color_subplots_spacing_changes():
             ax.add_imshow(base + idx, cmap="viridis")
             idx += 1
     loose_fig, loose_m_axes = loose_canvas.plot(backend="matplotlib")
-    loose_hgap = loose_m_axes[0, 1].get_position().x0 - loose_m_axes[0, 0].get_position().x1
-    loose_vgap = loose_m_axes[0, 0].get_position().y0 - loose_m_axes[1, 0].get_position().y1
+    loose_hgap = (
+        loose_m_axes[0, 1].get_position().x0 - loose_m_axes[0, 0].get_position().x1
+    )
+    loose_vgap = (
+        loose_m_axes[0, 0].get_position().y0 - loose_m_axes[1, 0].get_position().y1
+    )
 
     assert loose_hgap > tight_hgap
     assert loose_vgap > tight_vgap