Added new-style formatter function.

tacaswell · tacaswell · commit 04c734cc9df2 · 2013-11-30T12:23:28.000-06:00
Added class `StrMethodFormatter` which is identical to `FormatStrFormatter`
only for new-style ('{}'.format) formatting methods.
diff --git a/doc/api/api_changes.rst b/doc/api/api_changes.rst
@@ -103,8 +103,9 @@ original location:
 * Added the rcParam `axes.fromatter.useoffset` to control the default value
   of `useOffset` in `ticker.ScalarFormatter`
 
-
-
+* Added `Formatter` sub-class `StrMethodFormatter` which
+  does the exact same thing as `FormatStrFormatter`, but for new-style
+  formatting strings.
 
 .. _changes_in_1_3:
 
diff --git a/doc/users/whats_new.rst b/doc/users/whats_new.rst
@@ -55,9 +55,9 @@ He also increased the performance for all of these functions and plot types.
 
 Support for detrending and windowing 2D arrays in mlab
 ``````````````````````````````````````````````````````
-Todd Jennings added support for 2D arrays in the 
+Todd Jennings added support for 2D arrays in the
 :func:`~matplotlib.mlab.detrend_mean`, :func:`~matplotlib.mlab.detrend_none`,
-and :func:`~matplotlib.mlab.detrend`, as well as adding 
+and :func:`~matplotlib.mlab.detrend`, as well as adding
 :func:`~matplotlib.mlab.apply_window` which support windowing 2D arrays.
 
 Support for strides in mlab
@@ -68,6 +68,11 @@ strides to create memory-efficient 2D arrays.  This includes
 array, and :func:`~matplotlib.mlab.stride_windows`, which uses a moving window
 to create a 2D array from a 1D array.
 
+Formatter for new-style formatting strings
+``````````````````````````````````````````
+Added `FormatStrFormatterNewStyle` which does the same job as
+`FormatStrFormatter`, but accepts new-style formatting strings
+instead of printf-style formatting strings
 
 Date handling
 -------------
diff --git a/lib/matplotlib/tests/test_ticker.py b/lib/matplotlib/tests/test_ticker.py
@@ -7,6 +7,7 @@
 from numpy.testing import assert_almost_equal
 import numpy as np
 import matplotlib
+from itertools import izip
 import matplotlib.ticker as mticker
 
 
@@ -49,13 +50,21 @@ def test_LogLocator():
     test_value = np.array([0.5, 1., 2., 4., 8., 16., 32., 64., 128., 256.])
     assert_almost_equal(loc.tick_values(1, 100), test_value)
 
-
 def test_use_offset():
     for use_offset in [True, False]:
         with matplotlib.rc_context({'axes.formatter.useoffset': use_offset}):
             tmp_form = mticker.ScalarFormatter()
             nose.tools.assert_equal(use_offset, tmp_form.get_useOffset())
 
+def test_formatstrformatter():
+    # test % style formatter
+    tmp_form = mticker.FormatStrFormatter('%05d')
+    nose.tools.assert_equal('00002', tmp_form(2))
+
+    # test str.format() style formatter
+    tmp_form = mticker.StrMethodFormatter('{:05d}')
+    nose.tools.assert_equal('00002', tmp_form(2))
+
 
 if __name__ == '__main__':
     import nose
diff --git a/lib/matplotlib/ticker.py b/lib/matplotlib/ticker.py
@@ -292,7 +292,7 @@ def __call__(self, x, pos=None):
 
 class FormatStrFormatter(Formatter):
     """
-    Use a format string to format the tick
+    Use an old-style ('%' operator) format string to format the tick
     """
     def __init__(self, fmt):
         self.fmt = fmt
@@ -302,6 +302,18 @@ def __call__(self, x, pos=None):
         return self.fmt % x
 
 
+class StrMethodFormatter(Formatter):
+    """
+    Use a new-style format string (as used by `str.format()`) to format the tick
+    """
+    def __init__(self, fmt):
+        self.fmt = fmt
+
+    def __call__(self, x, pos=None):
+        'Return the format for tick val *x* at position *pos*'
+        return self.fmt.format(x)
+
+
 class OldScalarFormatter(Formatter):
     """
     Tick location is a plain old number.
