remove six.moves.xrange from legend tests

phobson · phobson · commit 92ac2684dd2c · 2017-01-22T20:44:18.000-08:00
diff --git a/lib/matplotlib/tests/test_legend.py b/lib/matplotlib/tests/test_legend.py
@@ -1,7 +1,6 @@
 from __future__ import (absolute_import, division, print_function,
                         unicode_literals)
 
-from six.moves import xrange
 try:
     # mock in python 3.3+
     from unittest import mock
@@ -58,9 +57,9 @@ def test_various_labels():
     # tests all sorts of label types
     fig = plt.figure()
     ax = fig.add_subplot(121)
-    ax.plot(list(xrange(4)), 'o', label=1)
+    ax.plot(np.arage(4), 'o', label=1)
     ax.plot(np.linspace(4, 4.1), 'o', label='D\xe9velopp\xe9s')
-    ax.plot(list(xrange(4, 1, -1)), 'o', label='__nolegend__')
+    ax.plot(np.arage(4, 1, -1), 'o', label='__nolegend__')
     ax.legend(numpoints=1, loc=0)
 
 
@@ -122,9 +121,9 @@ def test_alpha_rcparam():
 def test_fancy():
     # using subplot triggers some offsetbox functionality untested elsewhere
     plt.subplot(121)
-    plt.scatter(list(xrange(10)), list(xrange(10, 0, -1)), label='XX\nXX')
+    plt.scatter(np.arange(10), np.arange(10, 0, -1), label='XX\nXX')
     plt.plot([5] * 10, 'o--', label='XX')
-    plt.errorbar(list(xrange(10)), list(xrange(10)), xerr=0.5,
+    plt.errorbar(np.arange(10), np.arange(10), xerr=0.5,
                  yerr=0.5, label='XX')
     plt.legend(loc="center left", bbox_to_anchor=[1.0, 0.5],
                ncol=2, shadow=True, title="My legend", numpoints=1)
@@ -144,14 +143,14 @@ def test_rc():
     # using subplot triggers some offsetbox functionality untested elsewhere
     plt.figure()
     ax = plt.subplot(121)
-    ax.scatter(list(xrange(10)), list(xrange(10, 0, -1)), label='three')
+    ax.scatter(np.arange(10), np.arange(10, 0, -1), label='three')
     ax.legend(loc="center left", bbox_to_anchor=[1.0, 0.5],
               title="My legend")
 
     mpl.rcParams['legend.scatterpoints'] = 1
     plt.figure()
     ax = plt.subplot(121)
-    ax.scatter(list(xrange(10)), list(xrange(10, 0, -1)), label='one')
+    ax.scatter(np.arange(10), np.arange(10, 0, -1), label='one')
     ax.legend(loc="center left", bbox_to_anchor=[1.0, 0.5],
               title="My legend")
 
