Fixed failing test on travis.

pelson · pelson · commit 65390b58318e · 2013-11-28T10:08:36.000Z
diff --git a/lib/matplotlib/tests/test_collections.py b/lib/matplotlib/tests/test_collections.py
@@ -416,6 +416,13 @@ def test_add_collection():
 
 @cleanup
 def test_quiver_limits():
+    ax = plt.axes()
+    x, y = np.arange(8), np.arange(10)
+    data = u = v = np.linspace(0, 10, 80).reshape(10, 8)
+    q = plt.quiver(x, y, u, v)
+    assert_equal(q.get_datalim(ax.transData).bounds, (0., 0., 7., 9.))
+
+    plt.figure()
     ax = plt.axes()
     x = np.linspace(-5, 10, 20)
     y = np.linspace(-2, 4, 10)
diff --git a/lib/matplotlib/tests/test_pickle.py b/lib/matplotlib/tests/test_pickle.py
@@ -158,7 +158,9 @@ def test_complete():
     plt.subplot(3, 3, 6)
     plt.streamplot(x, y, u, v)
 
-    plt.subplot(3, 3, 7)
+    ax = plt.subplot(3, 3, 7)
+    ax.set_xlim(0, 7)
+    ax.set_ylim(0, 9)
     plt.quiver(x, y, u, v)
 
     plt.subplot(3, 3, 8)
