add __eq__ to matplotlib.gridspec.SubplotSpec

tacaswell · tacaswell · commit f328498e0693 · 2013-11-16T13:53:18.000-06:00
This is to address the issue discovered in http://stackoverflow.com/questions/20017314/how-are-existing-subplot-axes-retrieved-using-subplot2grid-gridspec-in-matplotli where multiple calls to gs = gridspec.GridSpec(2, 1) ax1 = fig.add_subplot(gs[0, 0]) ax2 = fig.add_subplot(gs[0, 0]) will return a new axes each time (unlike calls integers or tuples).
diff --git a/lib/matplotlib/gridspec.py b/lib/matplotlib/gridspec.py
@@ -457,3 +457,8 @@ def get_topmost_subplotspec(self):
             return gridspec.get_topmost_subplotspec()
         else:
             return self
+
+    def __eq__(self, other):
+        return all((self._gridspec == other._gridspec,
+                    self.num1 == other.num1,
+                    self.num2 == other.num2))
