test_imgage_plot is now sensitive to the matplotlib.imshow origin parameter

Author: danielhkl

matplotlib2tikz/image.py

@@ -29,7 +29,7 @@ def draw_image(data, obj):
 
     if obj.origin == "lower":
         img_array = numpy.flipud(img_array)
-
+        
     dims = img_array.shape
     if len(dims) == 2:  # the values are given as one real number: look at cmap
         clims = obj.get_clim()
@@ -45,7 +45,6 @@ def draw_image(data, obj):
         # RGB (+alpha) information at each point
         assert len(dims) == 3 and dims[2] in [3, 4]
         # convert to PIL image
-
         image = PIL.Image.fromarray(img_array)
         image.save(filename)
 

test/test_image_plot.py

@@ -14,17 +14,18 @@ def plot():
 
     this_dir = os.path.dirname(os.path.realpath(__file__))
     lena = Image.open(os.path.join(this_dir, 'lena.png'))
+    lena = lena.convert('L')
     dpi = rcParams['figure.dpi']
     figsize = lena.size[0]/dpi, lena.size[1]/dpi
     fig = pp.figure(figsize=figsize)
     ax = pp.axes([0, 0, 1, 1], frameon=False)
     ax.set_axis_off()
-    pp.imshow(lena)
+    pp.imshow(lena, cmap='viridis', origin='lower')
     # Set the current color map to HSV.
     pp.hsv()
     pp.colorbar()
     return fig
 
 
 def test():
-    helpers.assert_phash(plot(), '75c3d34d1d0d4b91')
+    helpers.assert_phash(plot(), '455361ec211d72fb')