Change return value of streamplot.

tonysyu · tonysyu · commit 9875323ed797 · 2012-09-02T09:16:37.000-04:00
- Return object that derives from `object` since deriving from `Container` was not beneficial.
- This return value is a stopgap; the final return value should allow users to set the colormap, alpha, etc. for both lines and arrows.
diff --git a/lib/matplotlib/streamplot.py b/lib/matplotlib/streamplot.py
@@ -8,7 +8,6 @@
 import matplotlib.colors as mcolors
 import matplotlib.collections as mcollections
 import matplotlib.patches as patches
-import matplotlib.container as container
 
 
 __all__ = ['streamplot']
@@ -50,7 +49,7 @@ def streamplot(axes, x, y, u, v, density=1, linewidth=None, color=None,
 
     Returns
     -------
-    *stream_container* : StreamplotContainer
+    *stream_container* : StreamplotSet
         Container object with attributes
             lines : `matplotlib.collections.LineCollection` of streamlines
             arrows : collection of `matplotlib.patches.FancyArrowPatch` objects
@@ -159,19 +158,15 @@ def streamplot(axes, x, y, u, v, density=1, linewidth=None, color=None,
     axes.autoscale_view(tight=True)
 
     ac = matplotlib.collections.PatchCollection(arrows)
-    stream_container = StreamplotContainer(lc, arrows=ac)
+    stream_container = StreamplotSet(lc, ac)
     return stream_container
 
 
-class StreamplotContainer(container.Container):
+class StreamplotSet(object):
 
-    def __new__(cls, *kl, **kwargs):
-        return tuple.__new__(cls)
-
-    def __init__(self, lines, arrows=None, **kwargs):
+    def __init__(self, lines, arrows, **kwargs):
         self.lines = lines
         self.arrows = arrows
-        container.Container.__init__(self, lines, **kwargs)
 
 
 # Coordinate definitions
