Make axisartist.grid_finder pseudo-tests real.

QuLogic · QuLogic · commit bd36aa32e027 · 2018-02-05T20:42:48.000-05:00
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]
