New properties: FPTree#root gives access to the tree's root node, and FPNode#leaf is true iff the node is a leaf in its tree.

Eric Naeseth · Eric Naeseth · commit a6e13030da62 · 2009-03-10T03:24:15.000-05:00
diff --git a/fp_growth.py b/fp_growth.py
@@ -23,6 +23,11 @@ def __init__(self):
         # "neighbors" that will hit every node containing that item.
         self._routes = {}
         
+    @property
+    def root(self):
+        """The root node of the tree."""
+        return self._root
+        
     def add(self, transaction):
         """
         Adds a transaction to the tree.
@@ -133,6 +138,11 @@ def root(self):
         """True if this node is the root of a tree; false if otherwise."""
         return self._item is None and self._count is None
         
+    @property
+    def leaf(self):
+        """True if this node is a leaf in the tree; false if otherwise."""
+        return len(self._children) == 0
+        
     def parent():
         doc = "The node's parent."
         def fget(self):
