Add test for axisline_style

oscargus · oscargus · commit 9140a3d8c131 · 2022-10-04T10:07:01.000+02:00
diff --git a/lib/mpl_toolkits/axisartist/axisline_style.py b/lib/mpl_toolkits/axisartist/axisline_style.py
@@ -54,10 +54,10 @@ def set_path(self, path):
         def draw(self, renderer):
             """
             Draw the axis line.
-             1) transform the path to the display coordinate.
-             2) extend the path to make a room for arrow
-             3) update the path of the FancyArrowPatch.
-             4) draw
+             1) Transform the path to the display coordinate.
+             2) Extend the path to make a room for arrow.
+             3) Update the path of the FancyArrowPatch.
+             4) Draw.
             """
             path_in_disp = self._line_transform.transform_path(self._line_path)
             mutation_size = self.get_mutation_scale()  # line_mutation_scale()
@@ -67,7 +67,7 @@ def draw(self, renderer):
             FancyArrowPatch.draw(self, renderer)
 
     class FilledArrow(SimpleArrow):
-        """The artist class that will be returned for SimpleArrow style."""
+        """The artist class that will be returned for FilledArrow style."""
         _ARROW_STYLE = "-|>"
 
 
diff --git a/lib/mpl_toolkits/tests/baseline_images/test_axisartist_axislines/axisline_style.png b/lib/mpl_toolkits/tests/baseline_images/test_axisartist_axislines/axisline_style.png
diff --git a/lib/mpl_toolkits/tests/test_axisartist_axislines.py b/lib/mpl_toolkits/tests/test_axisartist_axislines.py
@@ -3,7 +3,7 @@
 from matplotlib.testing.decorators import image_comparison
 from matplotlib.transforms import IdentityTransform
 
-from mpl_toolkits.axisartist.axislines import SubplotZero, Subplot
+from mpl_toolkits.axisartist.axislines import AxesZero, SubplotZero, Subplot
 from mpl_toolkits.axisartist import Axes, SubplotHost
 
 
@@ -90,3 +90,16 @@ def test_ParasiteAxesAuxTrans():
         ax1.set_ylim((0, 5))
 
     ax2.contour(xx, yy, data, colors='k')
+
+
+@image_comparison(['axisline_style.png'], remove_text=True, style='mpl20')
+def test_axisline_style():
+    fig = plt.figure(figsize=(2, 2))
+    ax = fig.add_subplot(axes_class=AxesZero)
+    ax.axis["xzero"].set_axisline_style("-|>,size=1.5")
+    ax.axis["xzero"].set_visible(True)
+    ax.axis["yzero"].set_axisline_style("->")
+    ax.axis["yzero"].set_visible(True)
+
+    for direction in ("left", "right", "bottom", "top"):
+        ax.axis[direction].set_visible(False)
