Refactor AxisArtistHelpers

Oscar Gustafsson · Oscar Gustafsson · commit 85e0f3ca9bcd · 2023-11-20T16:29:31.000+01:00
diff --git a/lib/mpl_toolkits/axisartist/axislines.py b/lib/mpl_toolkits/axisartist/axislines.py
@@ -84,9 +84,15 @@ def get_tick_iterators(self, axes):
             return iter_major, iter_minor
     """
 
+    def __init__(self, nth_coord):
+        self.nth_coord = nth_coord
+
     def update_lim(self, axes):
         pass
 
+    def get_nth_coord(self):
+        return self.nth_coord
+
     def _to_xy(self, values, const):
         """
         Create a (*values.shape, 2)-shape array representing (x, y) pairs.
@@ -115,17 +121,13 @@ class _FixedAxisArtistHelperBase(_AxisArtistHelperBase):
     @_api.delete_parameter("3.9", "nth_coord")
     def __init__(self, loc, nth_coord=None):
         """``nth_coord = 0``: x-axis; ``nth_coord = 1``: y-axis."""
-        self.nth_coord = _api.check_getitem(
-            {"bottom": 0, "top": 0, "left": 1, "right": 1}, loc=loc)
+        super().__init__(_api.check_getitem(
+            {"bottom": 0, "top": 0, "left": 1, "right": 1}, loc=loc))
         self._loc = loc
         self._pos = {"bottom": 0, "top": 1, "left": 0, "right": 1}[loc]
-        super().__init__()
         # axis line in transAxes
         self._path = Path(self._to_xy((0, 1), const=self._pos))
 
-    def get_nth_coord(self):
-        return self.nth_coord
-
     # LINE
 
     def get_line(self, axes):
@@ -158,12 +160,8 @@ def get_tick_transform(self, axes):
 
 class _FloatingAxisArtistHelperBase(_AxisArtistHelperBase):
     def __init__(self, nth_coord, value):
-        self.nth_coord = nth_coord
         self._value = value
-        super().__init__()
-
-    def get_nth_coord(self):
-        return self.nth_coord
+        super().__init__(nth_coord)
 
     def get_line(self, axes):
         raise RuntimeError("get_line method should be defined by the derived class")
