matplotlib
diff --git a/‎lib/matplotlib/__init__.py‎
Lines changed: 1 addition & 0 deletions b/‎lib/matplotlib/__init__.py‎
Lines changed: 1 addition & 0 deletions
diff --git a/‎lib/matplotlib/streamplot.py‎
Lines changed: 2 additions & 0 deletions b/‎lib/matplotlib/streamplot.py‎
Lines changed: 2 additions & 0 deletions
diff --git a/‎lib/matplotlib/tests/baseline_images/test_streamplot/streamplot_colormap_test_image.pdf‎
256 KB b/‎lib/matplotlib/tests/baseline_images/test_streamplot/streamplot_colormap_test_image.pdf‎
256 KB
diff --git a/‎lib/matplotlib/tests/baseline_images/test_streamplot/streamplot_colormap_test_image.png‎
86.2 KB b/‎lib/matplotlib/tests/baseline_images/test_streamplot/streamplot_colormap_test_image.png‎
86.2 KB
@@ -1105,6 +1105,7 @@ def tk_window_focus():
     'matplotlib.tests.test_scale',
     'matplotlib.tests.test_simplification',
     'matplotlib.tests.test_spines',
+    'matplotlib.tests.test_streamplot',
     'matplotlib.tests.test_subplots',
     'matplotlib.tests.test_text',
     'matplotlib.tests.test_ticker',
 
@@ -85,6 +85,8 @@ def streamplot(axes, x, y, u, v, density=1, linewidth=None, color=None,
     if use_multicolor_lines:
         assert color.shape == grid.shape
         line_colors = []
+        if np.any(np.isnan(color)):
+            color = np.ma.array(color, mask=np.isnan(color))
     else:
         line_kw['color'] = color
         arrow_kw['color'] = color