added some test cases for legend

jdh2358 · jdh2358 · commit 56ef1a8423c5 · 2007-04-07T19:56:15.000Z
svn path=/trunk/matplotlib/; revision=3169
diff --git a/lib/matplotlib/axes.py b/lib/matplotlib/axes.py
@@ -2358,11 +2358,15 @@ def vlines(self, x, ymin, ymax, colors='k', linestyle='solid', **kwargs):
         determined by ymin and ymax
 
 
-        colors is a line collections color args, either a single color or a len(x) list of colors
+        colors is a line collections color args, either a single color
+        or a len(x) list of colors
 
         linestyle is one of solid|dashed|dashdot|dotted
 
         Returns the LineCollection that was added
+
+        kwargs are LineCollection properties:
+        %(LineCollection)s
         """
 
         if kwargs.get('fmt') is not None:
@@ -2393,7 +2397,7 @@ def vlines(self, x, ymin, ymax, colors='k', linestyle='solid', **kwargs):
         verts = [ ((thisx, thisymin), (thisx, thisymax)) for thisx, (thisymin, thisymax) in zip(x,Y)]
         coll = LineCollection(verts, colors=colors, linestyle=linestyle)
         self.add_collection(coll)
-
+        coll.update(kwargs)
 
         minx = nx.amin(x)
         maxx = nx.amax(x)
@@ -2404,7 +2408,7 @@ def vlines(self, x, ymin, ymax, colors='k', linestyle='solid', **kwargs):
         self.autoscale_view()
 
         return coll
-    vlines.__doc__ = dedent(vlines.__doc__)
+    vlines.__doc__ = dedent(vlines.__doc__) % artist.kwdocd
 
     #### Basic plotting
     def plot(self, *args, **kwargs):
diff --git a/lib/matplotlib/collections.py b/lib/matplotlib/collections.py
@@ -647,11 +647,20 @@ def get_transoffset(self):
         return self._transOffset
 
     def set_segments(self, segments):
+        """
+        set the line segments
+
+        ACCEPTS: a list of [(x1,y1), (x2, x2)...] segments
+        """
         if segments is None: return
         self._segments = [asarray(seg) for seg in segments]
         if self._uniform_offsets is not None:
             self._add_offsets()
 
+    def get_segments(self):
+        'get the line segments'
+        return self._segments
+
     set_verts = set_segments # for compatibility with PolyCollection
 
     def _add_offsets(self):
diff --git a/lib/matplotlib/legend.py b/lib/matplotlib/legend.py
@@ -328,7 +328,7 @@ def get_handles(ax):
 
         inv = ax.transAxes.inverse_xy_tup
         for handle in handles:
-
+            print 'autolegend', handle, type(handle)
             if isinstance(handle, Line2D):
 
                 xdata = handle.get_xdata(orig=False)
diff --git a/unit/legend_unit.py b/unit/legend_unit.py
@@ -0,0 +1,20 @@
+from pylab import figure, show, nx
+
+# scatter creates a RegPolyCollection
+fig = figure()
+ax = fig.add_subplot(111)
+N = 100
+x, y = 0.9*nx.mlab.rand(2,N)
+area = nx.pi*(10 * nx.mlab.rand(N))**2 # 0 to 10 point radiuses
+ax.scatter(x,y,s=area, marker='^', c='r', label='scatter')
+ax.legend()
+fig.savefig('legend_unit_polycoll')
+
+# vlines creates a LineCollection
+fig = figure()
+ax = fig.add_subplot(111)
+t = nx.arange(0.0, 1.0, 0.05)
+ax.vlines(t, [0], nx.sin(2*nx.pi*t), label='vlines')
+ax.legend()
+fig.savefig('legend_unit_linecoll')
+show()
