Add scatter color len mismatch ValueError test

bcongdon · bcongdon · commit e218bcfb3114 · 2016-11-08T15:14:27.000-06:00
diff --git a/lib/matplotlib/tests/test_axes.py b/lib/matplotlib/tests/test_axes.py
@@ -4800,15 +4800,10 @@ def test_fillbetween_cycle():
 
 
 @cleanup
-def test_log_margins():
-    plt.rcParams['axes.autolimit_mode'] = 'data'
+def test_color_length_mismatch():
+    N = 500
+    x, y = np.random.rand(N), np.random.rand(N)
+    colors = np.random.rand(N+1)
     fig, ax = plt.subplots()
-    margin = 0.05
-    ax.set_xmargin(margin)
-    ax.semilogx([1, 10], [1, 10])
-    xlim0, xlim1 = ax.get_xlim()
-    transform = ax.xaxis.get_transform()
-    xlim0t, xlim1t = transform.transform([xlim0, xlim1])
-    x0t, x1t = transform.transform([1, 10])
-    delta = (x1t - x0t) * margin
-    assert_allclose([xlim0t + delta, xlim1t - delta], [x0t, x1t])
+    with pytest.raises(ValueError):
+        ax.scatter(x, y, c=colors)
