Make axisartist.grid_finder pseudo-tests real.

matplotlib · efiring · Feb 9, 2018 · Dec 1, 2016 · Dec 1, 2016 · Dec 1, 2016
commit 9488d1c82da00c04f6f4c5e2805175370c966fd7
diff --git a/lib/mpl_toolkits/axisartist/grid_finder.py b/lib/mpl_toolkits/axisartist/grid_finder.py
@@ -338,12 +338,3 @@ def __call__(self, direction, factor, values):
        r = [self._format_dict.get(k, v) for k, v in zip(values,
                                                         fallback_strings)]
        return r
-
-
-if __name__ == "__main__":
-    locator = MaxNLocator()
-    locs, nloc, factor = locator(0, 100)
-
-    fmt = FormatterPrettyPrint()
-
-    print(fmt("left", None, locs))
diff --git a/lib/mpl_toolkits/tests/test_axisartist_grid_finder.py b/lib/mpl_toolkits/tests/test_axisartist_grid_finder.py
@@ -0,0 +1,16 @@
+from __future__ import (absolute_import, division, print_function,
+                        unicode_literals)
+
+from mpl_toolkits.axisartist.grid_finder import (
+    FormatterPrettyPrint,
+    MaxNLocator)
+
+
+def test_pretty_print_format():
+    locator = MaxNLocator()
+    locs, nloc, factor = locator(0, 100)
+
+    fmt = FormatterPrettyPrint()
+
+    assert fmt("left", None, locs) == \
+        [r'$\mathdefault{%d}$' % (l, ) for l in locs]