Added set_params() method to AsinhLocator

rwpenney · rwpenney · commit 2002cb34d641 · 2021-10-21T16:57:45.000+01:00
diff --git a/lib/matplotlib/tests/test_ticker.py b/lib/matplotlib/tests/test_ticker.py
@@ -443,6 +443,14 @@ def test_set_params(self):
         assert sym.numticks == 8
 
 
+class TestAsinhLocator:
+    def test_set_params(self):
+        lctr = mticker.AsinhLocator(linear_width=5, numticks=17)
+        assert lctr.numticks == 17
+        lctr.set_params(numticks=23)
+        assert lctr.numticks == 23
+
+
 class TestScalarFormatter:
     offset_data = [
         (123, 189, 0),
diff --git a/lib/matplotlib/ticker.py b/lib/matplotlib/ticker.py
@@ -148,7 +148,7 @@
            'Locator', 'IndexLocator', 'FixedLocator', 'NullLocator',
            'LinearLocator', 'LogLocator', 'AutoLocator',
            'MultipleLocator', 'MaxNLocator', 'AutoMinorLocator',
-           'SymmetricalLogLocator', 'LogitLocator')
+           'SymmetricalLogLocator', 'AsinhLocator', 'LogitLocator')
 
 
 class _DummyAxis:
@@ -2606,6 +2606,11 @@ def __init__(self, linear_width, numticks=12):
         self.linear_width = linear_width
         self.numticks = numticks
 
+    def set_params(self, numticks=None):
+        """Set parameters within this locator."""
+        if numticks is not None:
+            self.numticks = numticks
+
     def __call__(self):
         dmin, dmax = self.axis.get_data_interval()
         return self.tick_values(dmin, dmax)
