updated legend unit

jdh2358 · jdh2358 · commit 623f43e014a6 · 2007-04-07T20:16:55.000Z
svn path=/trunk/matplotlib/; revision=3172
diff --git a/unit/legend_unit.py b/unit/legend_unit.py
@@ -1,20 +1,27 @@
 from pylab import figure, show, nx
 
+Ntests = 3
+t = nx.arange(0.0, 1.0, 0.05)
+s = nx.sin(2*nx.pi*t)
+
 # scatter creates a RegPolyCollection
 fig = figure()
-ax = fig.add_subplot(111)
+ax = fig.add_subplot(Ntests, 1, 1)
 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 = fig.add_subplot(Ntests, 1, 2)
 ax.vlines(t, [0], nx.sin(2*nx.pi*t), label='vlines')
 ax.legend()
-fig.savefig('legend_unit_linecoll')
+
+# vlines creates a LineCollection
+ax = fig.add_subplot(Ntests, 1, 3)
+ax.plot(t, s, 'b-', lw=2, label='a line')
+ax.legend()
+
+fig.savefig('legend_unit')
 show()
