changed test_image_plot into two test one for rgb and another for monochrome pictures

Author: danielhkl

matplotlib2tikz/image.py

@@ -46,7 +46,7 @@ def draw_image(data, obj):
         assert len(dims) == 3 and dims[2] in [3, 4]
         # convert to PIL image
         image = PIL.Image.fromarray(img_array)
-        image.save(filename)
+        image.save(filename, origin=obj.origin)
 
     # write the corresponding information to the TikZ file
     extent = obj.get_extent()

test/test_image_plot.py

@@ -1,16 +1,19 @@
 # -*- coding: utf-8 -*-
 #
 import helpers
+from matplotlib import rcParams
+from matplotlib import pyplot as pp
+import pytest
+import os
 
+try:
+    from PIL import Image
+except ImportError:
+    raise RuntimeError('PIL must be installed to run this example')
 
-def plot():
-    from matplotlib import rcParams
-    from matplotlib import pyplot as pp
-    import os
-    try:
-        from PIL import Image
-    except ImportError:
-        raise RuntimeError('PIL must be installed to run this example')
+#the picture 'lena.png' with origin='lower' is flipped upside-down. So it has to be upside-down in the pdf-file as well.
+
+def plot1(): #test for monochrome picture
 
     this_dir = os.path.dirname(os.path.realpath(__file__))
     lena = Image.open(os.path.join(this_dir, 'lena.png'))
@@ -26,6 +29,29 @@ def plot():
     pp.colorbar()
     return fig
 
+def plot2(): #test for rgb picture
+
+    this_dir = os.path.dirname(os.path.realpath(__file__))
+    lena = Image.open(os.path.join(this_dir, 'lena.png'))
+    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, cmap='viridis', origin='lower')
+    # Set the current color map to HSV.
+    pp.hsv()
+    pp.colorbar()
+    return fig
+
+@pytest.mark.parametrize(
+    'plot, phash', [
+        (plot1, '455361ec211d72fb'),
+        (plot2, '7558d3b30f634b06'),
+    ]
+    )
+
+def test(plot, phash):
+    helpers.assert_phash(plot(), phash)
+    return
 
-def test():
-    helpers.assert_phash(plot(), '455361ec211d72fb')