Merge pull request #11075 from anntzer/dropalpha-when-saving-diffimage

jklymak · web-flow · commit 7e073a018d85 · 2018-04-18T12:05:22.000-07:00
Drop alpha channel when saving comparison failure diff image.
diff --git a/lib/matplotlib/testing/compare.py b/lib/matplotlib/testing/compare.py
@@ -423,8 +423,9 @@ def compare_images(expected, actual, tol, in_decorator=False):
 
 
 def save_diff_image(expected, actual, output):
-    expectedImage = _png.read_png(expected)
-    actualImage = _png.read_png(actual)
+    # Drop alpha channels, similarly to compare_images.
+    expectedImage = _png.read_png(expected)[..., :3]
+    actualImage = _png.read_png(actual)[..., :3]
     actualImage, expectedImage = crop_to_same(
         actual, actualImage, expected, expectedImage)
     expectedImage = np.array(expectedImage).astype(float)
