pngsuite test: plot images in grayscale if the input array is 2 dimensional

astraw · astraw · commit e2c86e479e72 · 2009-04-05T22:06:05.000Z
svn path=/branches/v0_98_5_maint/; revision=7027
diff --git a/examples/tests/pngsuite/pngsuite.py b/examples/tests/pngsuite/pngsuite.py
@@ -8,6 +8,7 @@
 """
 
 from matplotlib import pyplot as plt
+import matplotlib.cm as cm
 import glob
 
 files = glob.glob("basn*.png")
@@ -17,7 +18,11 @@
 
 for i, fname in enumerate(files):
     data = plt.imread(fname)
-    plt.imshow(data, extent=[i,i+1,0,1])
+    cmap = None # use default colormap
+    if data.ndim==2:
+        # keep grayscale images gray
+        cmap = cm.gray
+    plt.imshow(data, extent=[i,i+1,0,1], cmap=cmap)
 
 plt.gca().get_frame().set_facecolor("#ddffff")
 plt.gca().set_xlim(0, len(files))
