added custom figure class hook

jdh2358 · jdh2358 · commit d074f646549e · 2006-06-21T13:26:57.000Z
svn path=/trunk/matplotlib/; revision=2503
diff --git a/examples/custom_figure_class.py b/examples/custom_figure_class.py
@@ -7,13 +7,13 @@
 class MyFigure(Figure):
     def __init__(self, *args, **kwargs):
         """
-        custom kwarg figtitle is passed on to 
+        custom kwarg figtitle is a figure title
         """
+        figtitle = kwargs.pop('figtitle', 'hi mom')
         Figure.__init__(self, *args, **kwargs)
-        figtitle = kwargs.get('figtitle', 'hi mom')
         self.text(0.5, 0.95, figtitle, ha='center')
 
-fig = figure(FigureClass=MyFigure)
+fig = figure(FigureClass=MyFigure, figtitle='my title')
 ax = fig.add_subplot(111)
 ax.plot([1,2,3])
 fig.savefig('test.png')
