axisartist cleanups.

Mostly just unwrapping long lists of arguments into a single line where
possible.

Author: anntzer

lib/mpl_toolkits/axisartist/axislines.py

@@ -153,12 +153,10 @@ def get_axislabel_pos_angle(self, axes):
     # TICK
 
     def get_tick_transform(self, axes):
-        return [axes.get_xaxis_transform(),
-                axes.get_yaxis_transform()][self.nth_coord]
+        return [axes.get_xaxis_transform(), axes.get_yaxis_transform()][self.nth_coord]
 
 
 class _FloatingAxisArtistHelperBase(_AxisArtistHelperBase):
-
     def __init__(self, nth_coord, value):
         self.nth_coord = nth_coord
         self._value = value
@@ -168,8 +166,7 @@ def get_nth_coord(self):
         return self.nth_coord
 
     def get_line(self, axes):
-        raise RuntimeError(
-            "get_line method should be defined by the derived class")
+        raise RuntimeError("get_line method should be defined by the derived class")
 
 
 class FixedAxisArtistHelperRectilinear(_FixedAxisArtistHelperBase):
@@ -187,10 +184,7 @@ def __init__(self, axes, loc, nth_coord=None):
 
     def get_tick_iterators(self, axes):
         """tick_loc, tick_angle, tick_label"""
-        if self._loc in ["bottom", "top"]:
-            angle_normal, angle_tangent = 90, 0
-        else:  # "left", "right"
-            angle_normal, angle_tangent = 0, 90
+        angle_normal, angle_tangent = {0: (90, 0), 1: (0, 90)}[self.nth_coord]
 
         major = self.axis.major
         major_locs = major.locator()
@@ -207,8 +201,7 @@ def _f(locs, labels):
                 c = self._to_xy(loc, const=self._pos)
                 # check if the tick point is inside axes
                 c2 = tick_to_axes.transform(c)
-                if mpl.transforms._interval_contains_close(
-                        (0, 1), c2[self.nth_coord]):
+                if mpl.transforms._interval_contains_close((0, 1), c2[self.nth_coord]):
                     yield c, angle_normal, angle_tangent, label
 
         return _f(major_locs, major_labels), _f(minor_locs, minor_labels)
@@ -245,20 +238,14 @@ def get_axislabel_pos_angle(self, axes):
         data_to_axes = axes.transData - axes.transAxes
         p = data_to_axes.transform([self._value, self._value])
         verts = self._to_xy(0.5, const=p[fixed_coord])
-        if 0 <= verts[fixed_coord] <= 1:
-            return verts, angle
-        else:
-            return None, None
+        return (verts, angle) if 0 <= verts[fixed_coord] <= 1 else (None, None)
 
     def get_tick_transform(self, axes):
         return axes.transData
 
     def get_tick_iterators(self, axes):
         """tick_loc, tick_angle, tick_label"""
-        if self.nth_coord == 0:
-            angle_normal, angle_tangent = 90, 0
-        else:
-            angle_normal, angle_tangent = 0, 90
+        angle_normal, angle_tangent = {0: (90, 0), 1: (0, 90)}[self.nth_coord]
 
         major = self.axis.major
         major_locs = major.locator()
@@ -326,27 +313,18 @@ def __init__(self, axes):
 
     @_api.delete_parameter(
         "3.9", "nth_coord", addendum="'nth_coord' is now inferred from 'loc'.")
-    def new_fixed_axis(self, loc,
-                       nth_coord=None,
-                       axis_direction=None,
-                       offset=None,
-                       axes=None,
-                       ):
+    def new_fixed_axis(
+            self, loc, nth_coord=None, axis_direction=None, offset=None, axes=None):
         if axes is None:
             _api.warn_external(
                 "'new_fixed_axis' explicitly requires the axes keyword.")
             axes = self.axes
         if axis_direction is None:
             axis_direction = loc
-        helper = FixedAxisArtistHelperRectilinear(axes, loc)
-        axisline = AxisArtist(axes, helper, offset=offset,
-                              axis_direction=axis_direction)
-        return axisline
+        return AxisArtist(axes, FixedAxisArtistHelperRectilinear(axes, loc),
+                          offset=offset, axis_direction=axis_direction)
 
-    def new_floating_axis(self, nth_coord, value,
-                          axis_direction="bottom",
-                          axes=None,
-                          ):
+    def new_floating_axis(self, nth_coord, value, axis_direction="bottom", axes=None):
         if axes is None:
             _api.warn_external(
                 "'new_floating_axis' explicitly requires the axes keyword.")
@@ -404,8 +382,7 @@ def __call__(self, *args, **kwargs):
 
     def __init__(self, *args, grid_helper=None, **kwargs):
         self._axisline_on = True
-        self._grid_helper = (grid_helper if grid_helper
-                             else GridHelperRectlinear(self))
+        self._grid_helper = grid_helper if grid_helper else GridHelperRectlinear(self)
         super().__init__(*args, **kwargs)
         self.toggle_axisline(True)
 

lib/mpl_toolkits/axisartist/floating_axes.py

@@ -158,11 +158,8 @@ def get_data_boundary(self, side):
                     bottom=(lat1, 1),
                     top=(lat2, 1))[side]
 
-    def new_fixed_axis(self, loc,
-                       nth_coord=None,
-                       axis_direction=None,
-                       offset=None,
-                       axes=None):
+    def new_fixed_axis(
+            self, loc, nth_coord=None, axis_direction=None, offset=None, axes=None):
         if axes is None:
             axes = self.axes
         if axis_direction is None:
@@ -179,23 +176,16 @@ def new_fixed_axis(self, loc,
 
     # new_floating_axis will inherit the grid_helper's extremes.
 
-    # def new_floating_axis(self, nth_coord,
-    #                       value,
-    #                       axes=None,
-    #                       axis_direction="bottom"
-    #                       ):
-
+    # def new_floating_axis(self, nth_coord, value, axes=None, axis_direction="bottom"):
     #     axis = super(GridHelperCurveLinear,
     #                  self).new_floating_axis(nth_coord,
     #                                          value, axes=axes,
     #                                          axis_direction=axis_direction)
-
     #     # set extreme values of the axis helper
     #     if nth_coord == 1:
     #         axis.get_helper().set_extremes(*self._extremes[:2])
     #     elif nth_coord == 0:
     #         axis.get_helper().set_extremes(*self._extremes[2:])
-
     #     return axis
 
     def _update_grid(self, x1, y1, x2, y2):
@@ -291,8 +281,6 @@ def adjust_axes_lim(self):
         self.set_ylim(bbox.ymin, bbox.ymax)
 
 
-floatingaxes_class_factory = cbook._make_class_factory(
-    FloatingAxesBase, "Floating{}")
-FloatingAxes = floatingaxes_class_factory(
-    host_axes_class_factory(axislines.Axes))
+floatingaxes_class_factory = cbook._make_class_factory(FloatingAxesBase, "Floating{}")
+FloatingAxes = floatingaxes_class_factory(host_axes_class_factory(axislines.Axes))
 FloatingSubplot = FloatingAxes

lib/mpl_toolkits/axisartist/grid_helper_curvelinear.py

@@ -272,11 +272,8 @@ def update_grid_finder(self, aux_trans=None, **kwargs):
         self._old_limits = None  # Force revalidation.
 
     @_api.make_keyword_only("3.9", "nth_coord")
-    def new_fixed_axis(self, loc,
-                       nth_coord=None,
-                       axis_direction=None,
-                       offset=None,
-                       axes=None):
+    def new_fixed_axis(
+            self, loc, nth_coord=None, axis_direction=None, offset=None, axes=None):
         if axes is None:
             axes = self.axes
         if axis_direction is None:
@@ -287,11 +284,7 @@ def new_fixed_axis(self, loc,
         # the floating_axig.GridHelperCurveLinear subclass?
         return axisline
 
-    def new_floating_axis(self, nth_coord,
-                          value,
-                          axes=None,
-                          axis_direction="bottom"
-                          ):
+    def new_floating_axis(self, nth_coord, value, axes=None, axis_direction="bottom"):
         if axes is None:
             axes = self.axes
         helper = FloatingAxisArtistHelper(
@@ -317,22 +310,15 @@ def get_gridlines(self, which="major", axis="both"):
         return grid_lines
 
     def get_tick_iterator(self, nth_coord, axis_side, minor=False):
-
-        # axisnr = dict(left=0, bottom=1, right=2, top=3)[axis_side]
         angle_tangent = dict(left=90, right=90, bottom=0, top=0)[axis_side]
-        # angle = [0, 90, 180, 270][axisnr]
         lon_or_lat = ["lon", "lat"][nth_coord]
         if not minor:  # major ticks
-            for (xy, a), l in zip(
+            for (xy, angle_normal), l in zip(
                     self._grid_info[lon_or_lat]["tick_locs"][axis_side],
                     self._grid_info[lon_or_lat]["tick_labels"][axis_side]):
-                angle_normal = a
                 yield xy, angle_normal, angle_tangent, l
         else:
-            for (xy, a), l in zip(
+            for (xy, angle_normal), l in zip(
                     self._grid_info[lon_or_lat]["tick_locs"][axis_side],
                     self._grid_info[lon_or_lat]["tick_labels"][axis_side]):
-                angle_normal = a
                 yield xy, angle_normal, angle_tangent, ""
-            # for xy, a, l in self._grid_info[lon_or_lat]["ticks"][axis_side]:
-            #     yield xy, a, ""