Merge pull request #16049 from anntzer/subplotspecrepr

jklymak · web-flow · commit 48487876505d · 2020-01-06T08:58:20.000-08:00
Add __repr__ to SubplotSpec.
diff --git a/lib/matplotlib/gridspec.py b/lib/matplotlib/gridspec.py
@@ -522,6 +522,11 @@ def __init__(self, gridspec, num1, num2=None):
         else:
             self._layoutbox = None
 
+    def __repr__(self):
+        return (f"{self.get_gridspec()}["
+                f"{self.rowspan.start}:{self.rowspan.stop}, "
+                f"{self.colspan.start}:{self.colspan.stop}]")
+
     @staticmethod
     def _from_subplot_args(figure, args):
         """
diff --git a/lib/matplotlib/tests/test_gridspec.py b/lib/matplotlib/tests/test_gridspec.py
@@ -24,3 +24,8 @@ def test_height_ratios():
     """
     with pytest.raises(ValueError):
         gridspec.GridSpec(1, 1, height_ratios=[2, 1, 3])
+
+
+def test_repr():
+    ss = gridspec.GridSpec(3, 3)[2, 1:3]
+    assert repr(ss) == "GridSpec(3, 3)[2:3, 1:3]"
