Add some inspection methods

enaeseth · enaeseth · commit 58031284cd12 · 2011-11-13T15:39:54.000-08:00
diff --git a/fp_growth.py b/fp_growth.py
@@ -174,6 +174,17 @@ def collect_path(node):
 
         return (collect_path(node) for node in self.nodes(item))
 
+    def inspect(self):
+        print 'Tree:'
+        self.root.inspect(1)
+
+        print
+        print 'Routes:'
+        for item, nodes in self.items():
+            print '  %r' % item
+            for node in nodes:
+                print '    %r' % node
+
     def _removed(self, node):
         """Called when `node` is removed from the tree; performs cleanup."""
 
@@ -366,6 +377,10 @@ def children(self):
         """The nodes that are children of this node."""
         return tuple(self._children.itervalues())
 
+    def inspect(self, depth=0):
+        print ('  ' * depth) + repr(self)
+        for child in self.children:
+            child.inspect(depth + 1)
 
     def __repr__(self):
         if self.root:
