move print_tree() to helpers

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/test_viridis.py

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