Merge pull request #16433 from anntzer/demo_curvelinear_grid2

timhoffm · web-flow · commit 731e53db30e7 · 2020-02-11T22:58:37.000+01:00
Simplify demo_curvelinear_grid2.
diff --git a/examples/axisartist/demo_curvelinear_grid2.py b/examples/axisartist/demo_curvelinear_grid2.py
@@ -13,13 +13,12 @@
 import numpy as np
 import matplotlib.pyplot as plt
 
-from mpl_toolkits.axisartist.grid_helper_curvelinear import \
-    GridHelperCurveLinear
-from mpl_toolkits.axisartist.grid_finder import MaxNLocator
+from mpl_toolkits.axisartist.grid_helper_curvelinear import (
+    GridHelperCurveLinear)
+from mpl_toolkits.axisartist.grid_finder import (
+    ExtremeFinderSimple, MaxNLocator)
 from mpl_toolkits.axisartist.axislines import Subplot
 
-import mpl_toolkits.axisartist.angle_helper as angle_helper
-
 
 def curvelinear_test1(fig):
     """Grid for custom transform."""
@@ -34,25 +33,16 @@ def inv_tr(x, y):
         x, y = np.asarray(x), np.asarray(y)
         return sgn*x**2, y
 
-    extreme_finder = angle_helper.ExtremeFinderCycle(20, 20,
-                                                     lon_cycle=None,
-                                                     lat_cycle=None,
-                                                     # (0, np.inf),
-                                                     lon_minmax=None,
-                                                     lat_minmax=None,
-                                                     )
-
-    grid_helper = GridHelperCurveLinear((tr, inv_tr),
-                                        extreme_finder=extreme_finder,
-                                        # better tick density
-                                        grid_locator1=MaxNLocator(nbins=6),
-                                        grid_locator2=MaxNLocator(nbins=6))
+    grid_helper = GridHelperCurveLinear(
+        (tr, inv_tr),
+        extreme_finder=ExtremeFinderSimple(20, 20),
+        # better tick density
+        grid_locator1=MaxNLocator(nbins=6), grid_locator2=MaxNLocator(nbins=6))
 
     ax1 = Subplot(fig, 111, grid_helper=grid_helper)
     # ax1 will have a ticks and gridlines defined by the given
-    # transform (+ transData of the Axes). Note that the transform of
-    # the Axes itself (i.e., transData) is not affected by the given
-    # transform.
+    # transform (+ transData of the Axes). Note that the transform of the Axes
+    # itself (i.e., transData) is not affected by the given transform.
 
     fig.add_subplot(ax1)
 
