Testing 2d to 1d conversion function in

danuo · danuo · commit 369ba34281bd · 2020-09-16T15:40:08.000+02:00
colors.to_rgba()

+ Rebase with master
+ Change .reshape((-1)) to .reshape(-1)
diff --git a/lib/matplotlib/tests/test_colors.py b/lib/matplotlib/tests/test_colors.py
@@ -1211,3 +1211,10 @@ def test_colormap_bad_data_with_alpha():
     assert_array_equal(c[0, 0], (0, 0, 0, 0))
     c = cmap([[np.nan, 0.5], [0, 0]], alpha=np.full((2, 2), 0.5))
     assert_array_equal(c[0, 0], (0, 0, 0, 0))
+    
+    
+def test_2d_to_rgba():
+    color = np.array([0.1, 0.2, 0.3])
+    rgba_1d = mcolors.to_rgba(color.reshape(-1))
+    rgba_2d = mcolors.to_rgba(color.reshape((1, -1)))
+    assert rgba_1d == rgba_2d
