Tests have been added that check a figure created

danuo · danuo · commit 05f135944a44 · 2020-09-10T01:38:05.000+02:00
with 1d color arrays against one with 2d color arrays.
They should be identical if 2d arrays are read correctly.
diff --git a/lib/matplotlib/tests/test_axes.py b/lib/matplotlib/tests/test_axes.py
@@ -6513,3 +6513,22 @@ def test_multiplot_autoscale():
     ax2.axhspan(-5, 5)
     xlim = ax1.get_xlim()
     assert np.allclose(xlim, [0.5, 4.5])
+
+
+@check_figures_equal(extensions=["pdf"])
+def test_2dcolor_plot(fig_test, fig_ref):
+    color = np.array([0.1, 0.2, 0.3])
+    # plot with 1D-color:
+    axs = fig_test.subplots(5)
+    axs[0].plot([1, 2], [1, 2], c=color.reshape((-1)))
+    axs[1].scatter([1, 2], [1, 2], c=color.reshape((-1)))
+    axs[2].step([1, 2], [1, 2], c=color.reshape((-1)))
+    axs[3].hist(np.arange(10), color=color.reshape((-1)))
+    axs[4].bar(np.arange(10), np.arange(10), color=color.reshape((-1)))
+    # plot with 2D-color:
+    axs = fig_ref.subplots(5)
+    axs[0].plot([1, 2], [1, 2], c=color.reshape((1, -1)))
+    axs[1].scatter([1, 2], [1, 2], c=color.reshape((1, -1)))
+    axs[2].step([1, 2], [1, 2], c=color.reshape((1, -1)))
+    axs[3].hist(np.arange(10), color=color.reshape((1, -1)))
+    axs[4].bar(np.arange(10), np.arange(10), color=color.reshape((1, -1)))
