correction for Linthub (code style)

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()

matplotlib2tikz/save.py

@@ -30,7 +30,7 @@ def get_tikz_code(
         extra_tikzpicture_parameters=None,
         dpi=None,
         show_info=True
-        ):
+):
     '''Main function. Here, the recursion into the image starts and the
     contents are picked up. The actual file gets written in this routine.
 

test/test_image_plot.py

@@ -11,9 +11,11 @@
 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.
+# 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
+# test for monochrome picture
+def plot1():
 
     this_dir = os.path.dirname(os.path.realpath(__file__))
     lena = Image.open(os.path.join(this_dir, 'lena.png'))
@@ -29,7 +31,9 @@ def plot1(): #test for monochrome picture
     pp.colorbar()
     return fig
 
-def plot2(): #test for rgb picture
+
+# test for rgb picture
+def plot2():
 
     this_dir = os.path.dirname(os.path.realpath(__file__))
     lena = Image.open(os.path.join(this_dir, 'lena.png'))
@@ -44,14 +48,13 @@ def plot2(): #test for rgb picture
     pp.colorbar()
     return fig
 
+
 @pytest.mark.parametrize(
     'plot, phash', [
         (plot1, '455361ec211d72fb'),
         (plot2, '7558d3b30f634b06'),
     ]
     )
-
 def test(plot, phash):
     helpers.assert_phash(plot(), phash)
-    return
-
+    return