ENH: Added __repr__ for Figure

richardjgowers · richardjgowers · commit d443db18af57 · 2017-10-17T12:03:07.000+01:00
diff --git a/lib/matplotlib/figure.py b/lib/matplotlib/figure.py
@@ -273,6 +273,13 @@ class Figure(Artist):
     def __str__(self):
         return "Figure(%gx%g)" % tuple(self.bbox.size)
 
+    def __repr__(self):
+        return "<{clsname} size {h:g}x{w:g} with {naxes} axes>".format(
+            clsname=self.__class__.__name__,
+            h=self.bbox.size[0], w=self.bbox.size[1],
+            naxes=len(self.axes),
+        )
+
     def __init__(self,
                  figsize=None,  # defaults to rc figure.figsize
                  dpi=None,  # defaults to rc figure.dpi
diff --git a/lib/matplotlib/tests/test_figure.py b/lib/matplotlib/tests/test_figure.py
@@ -326,3 +326,8 @@ def test_savefig():
     msg = "savefig() takes 2 positional arguments but 3 were given"
     with pytest.raises(TypeError, message=msg):
         fig.savefig("fname1.png", "fname2.png")
+
+
+def test_figure_repr():
+    fig = plt.figure(figsize=(10, 20), dpi=10)
+    assert repr(fig) == "<Figure size 100x200 with 0 axes>"
