Merge pull request #170 from nschloe/text-test-coverage

Text test coverage

Author: nschloe

test/helpers.py

@@ -2,13 +2,14 @@
 #
 import matplotlib2tikz
 
+import imagehash
+import matplotlib
+import matplotlib.pyplot as plt
 import os
+from PIL import Image
 import shutil
-import tempfile
 import subprocess
-from PIL import Image
-import imagehash
-from matplotlib import pyplot as plt
+import tempfile
 
 
 def compute_phash(fig):
@@ -145,3 +146,16 @@ def assert_phash(fig, reference_phash):
     assert reference_phash == phash
 
     return
+
+
+def print_tree(obj, indent=''):
+    '''Recursively prints the tree structure of the matplotlib object.
+    '''
+    if isinstance(obj, matplotlib.text.Text):
+        print(indent, type(obj).__name__, '("%s")' % obj.get_text())
+    else:
+        print(indent, type(obj).__name__)
+
+    for child in obj.get_children():
+        print_tree(child, indent + '   ')
+    return

test/show

@@ -193,3 +193,10 @@ def plot():
 
 def test():
     helpers.assert_phash(plot(), '9d2327dc23cdd81a')
+
+
+if __name__ == '__main__':
+    plot()
+    plt.show()
+    # phash, _, _, _, _, _, _ = helpers.compute_phash(plot2())
+    # print(phash)

test/test_fancybox.py

@@ -1,34 +1,38 @@
 # -*- coding: utf-8 -*-
 #
-desc = 'Regular plot with overlay text'
-phash = '370da93449d3f64c'
+import helpers
 
+import matplotlib.pyplot as plt
+import numpy
 
-def plot():
-    from matplotlib import pyplot as pp
-    import numpy as np
 
-    fig = pp.figure()
+def plot():
+    fig = plt.figure()
 
-    xxx = np.linspace(0, 5)
+    xxx = numpy.linspace(0, 5)
     yyy = xxx**2
-    pp.text(1, 5, 'test1', size=50, rotation=30.,
+    plt.text(
+            1, 5, 'test1', size=50, rotation=30.,
             ha='center', va='bottom', color='r', style='italic',
             weight='light',
-            bbox=dict(boxstyle='round, pad=0.2',
-                      ec=(1., 0.5, 0.5),
-                      fc=(1., 0.8, 0.8),
-                      ls='dashdot'
-                      )
+            bbox=dict(
+                boxstyle='round, pad=0.2',
+                ec=(1., 0.5, 0.5),
+                fc=(1., 0.8, 0.8),
+                ls='dashdot'
+                )
             )
-    pp.text(3, 6, 'test2', size=50, rotation=-30.,
+    plt.text(
+            3, 6, 'test2', size=50, rotation=-30.,
             ha='center', va='center', color='b', weight='bold',
-            bbox=dict(boxstyle='square',
-                      ec=(1., 0.5, 0.5),
-                      fc=(1., 0.8, 0.8),
-                      )
+            bbox=dict(
+                boxstyle='square',
+                ec=(1., 0.5, 0.5),
+                fc=(1., 0.8, 0.8),
+                )
             )
-    pp.text(4, 8, 'test3', size=20, rotation=90.0,
+    plt.text(
+            4, 8, 'test3', size=20, rotation=90.0,
             ha='center', va='center', color='b', weight='demi',
             bbox=dict(
                 boxstyle='rarrow',
@@ -37,7 +41,8 @@ def plot():
                 fc=(1., 0.8, 0.8),
                 )
             )
-    pp.text(4, 16, 'test4', size=20, rotation=90.0,
+    plt.text(
+            4, 16, 'test4', size=20, rotation=90.0,
             ha='center', va='center', color='b', weight='heavy',
             bbox=dict(
                 boxstyle='larrow',
@@ -46,39 +51,54 @@ def plot():
                 fc=(1., 0.8, 0.8),
                 )
             )
-    pp.text(2, 18, 'test5', size=20,
+    plt.text(
+            2, 18, 'test5', size=20,
             ha='center', va='center', color='b',
             bbox=dict(
                 boxstyle='darrow',
                 ec=(1., 0.5, 0.5),
                 fc=(1., 0.8, 0.8),
                 )
             )
-    pp.text(1, 20, 'test6', size=20,
+    plt.text(
+            1, 20, 'test6', size=20,
             ha='center', va='center', color='b',
             bbox=dict(
                 boxstyle='circle',
                 ec=(1., 0.5, 0.5),
                 fc=(1., 0.8, 0.8),
                 )
             )
-    pp.text(3, 23, 'test7', size=20,
+    plt.text(
+            3, 23, 'test7', size=20,
             ha='center', va='center', color='b',
             bbox=dict(
                 boxstyle='roundtooth',
                 ec=(1., 0.5, 0.5),
                 fc=(1., 0.8, 0.8),
                 )
             )
-    pp.text(3, 20, 'test8', size=20,
+    plt.text(
+            3, 20, 'test8', size=20,
             ha='center', va='center', color='b',
             bbox=dict(
                 boxstyle='sawtooth',
                 ec=(1., 0.5, 0.5),
                 fc=(1., 0.8, 0.8),
                 )
             )
-    pp.plot(xxx, yyy, label='a graph')
-    pp.legend()
+    plt.plot(xxx, yyy, label='a graph')
+    plt.legend()
 
     return fig
+
+
+def test():
+    helpers.assert_phash(plot(), '370da93449d3f64c')
+
+
+if __name__ == '__main__':
+    plot()
+    plt.show()
+    # phash, _, _, _, _, _, _ = helpers.compute_phash(plot2())
+    # print(phash)

test/text_overlay.py renamed to test/test_text_overlay.py

@@ -20,3 +20,7 @@ def plot():
 
 def test():
     helpers.assert_phash(plot(), 'fd7e03fc03bc0381')
+
+
+if __name__ == '__main__':
+    helpers.print_tree(plot())