Rebased to latest master. Added set_params to new LogitLocator. Added testing functions to test_ticker.py

leeonadoh · leeonadoh · commit 27fbe7e8ef55 · 2015-03-03T14:29:35.000-05:00
diff --git a/lib/matplotlib/tests/test_ticker.py b/lib/matplotlib/tests/test_ticker.py
@@ -111,6 +111,15 @@ def test_MultipleLocator_set_params():
     mult.set_params(base=1.7)
     nose.tools.assert_equal(mult._base, 1.7)
 
+def test_LogitLocator_set_params():
+    """
+    Create logit locator with default minor=False, and change it to something
+    else. See if change was successful. Should not exception.
+    """
+    loc = mticker.LogitLocator()  # Defaults to false.
+    loc.set_params(minor=True)
+    nose.tools.assert_true(loc.minor)
+
 
 def test_FixedLocator_set_params():
     """
diff --git a/lib/matplotlib/ticker.py b/lib/matplotlib/ticker.py
@@ -1781,6 +1781,11 @@ def __init__(self, minor=False):
         """
         self.minor = minor
 
+    def set_params(self, minor=None):
+        """Set parameters within this locator."""
+        if minor is not None:
+            self.minor = minor
+
     def __call__(self):
         'Return the locations of the ticks'
         vmin, vmax = self.axis.get_view_interval()
