test fix

Author: nschloe

test/test_annotate.py

@@ -39,3 +39,8 @@ def plot():
 def test():
     phash = Phash(plot())
     assert phash.phash == 'ab8a79a1549654be', phash.get_details()
+
+
+if __name__ == '__main__':
+    plot()
+    plt.show()

test/test_basic_sin.py

@@ -2,12 +2,11 @@
 #
 from helpers import Phash
 
-from matplotlib import pyplot as plt
-from matplotlib import style
-import numpy as np
-
 
 def plot():
+    from matplotlib import pyplot as plt
+    import numpy as np
+
     fig = plt.figure()
     with plt.style.context(('ggplot')):
         t = np.arange(0.0, 2.0, 0.1)

test/test_image_plot.py

@@ -1,52 +1,52 @@
 # -*- 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')
 
 # the picture 'lena.png' with origin='lower' is flipped upside-down.
 # So it has to be upside-down in the pdf-file as well.
 
 
 # test for monochrome picture
 def plot1():
+    from matplotlib import rcParams
+    import matplotlib.pyplot as plt
+    from PIL import Image
+    import os
 
     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)
+    fig = plt.figure(figsize=figsize)
+    ax = plt.axes([0, 0, 1, 1], frameon=False)
     ax.set_axis_off()
-    pp.imshow(lena, cmap='viridis', origin='lower')
+    plt.imshow(lena, cmap='viridis', origin='lower')
     # Set the current color map to HSV.
-    pp.hsv()
-    pp.colorbar()
+    plt.hsv()
+    plt.colorbar()
     return fig
 
 
 # test for rgb picture
 def plot2():
+    from matplotlib import rcParams
+    import matplotlib.pyplot as plt
+    from PIL import Image
+    import os
 
     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)
+    fig = plt.figure(figsize=figsize)
+    ax = plt.axes([0, 0, 1, 1], frameon=False)
     ax.set_axis_off()
-    pp.imshow(lena, cmap='viridis', origin='lower')
+    plt.imshow(lena, cmap='viridis', origin='lower')
     # Set the current color map to HSV.
-    pp.hsv()
-    pp.colorbar()
+    plt.hsv()
+    plt.colorbar()
     return fig
 
 

test/test_scatter.py

@@ -5,8 +5,8 @@
 
 def plot():
     from matplotlib import pyplot as plt
-    from matplotlib import style
     import numpy as np
+
     fig = plt.figure()
     with plt.style.context(('fivethirtyeight')):
         np.random.seed(123)