Merge pull request #1988 from pelson/pickle_bars

mdboom · mdboom · commit 1f07cb8af690 · 2013-05-10T11:32:43.000-07:00
Added bar plot pickle support.
diff --git a/lib/matplotlib/container.py b/lib/matplotlib/container.py
@@ -32,6 +32,13 @@ def remove(self):
         if self._remove_method:
             self._remove_method(self)
 
+    def __getstate__(self):
+        d = self.__dict__.copy()
+        # remove the unpicklable remove method, this will get re-added on load
+        # (by the axes) if the artist lives on an axes.
+        d['_remove_method'] = None
+        return d
+
     def get_label(self):
         """
         Get the label used for this artist in the legend.
diff --git a/lib/matplotlib/tests/test_pickle.py b/lib/matplotlib/tests/test_pickle.py
@@ -114,6 +114,10 @@ def test_simple():
 #    ax = plt.subplot(121, projection='hammer')
 #    recursive_pickle(ax, 'figure')
 #    pickle.dump(ax, BytesIO(), pickle.HIGHEST_PROTOCOL)
+    
+    plt.figure()
+    plt.bar(left=range(10), height=range(10))
+    pickle.dump(plt.gca(), BytesIO(), pickle.HIGHEST_PROTOCOL)
 
     fig = plt.figure()
     ax = plt.axes()
