X-Git-Url: https://wannabe.guru.org/gitweb/?a=blobdiff_plain;f=collect%2Ftrie.py;h=70d57b1338ca8e19e4bde07e33e440a846bdc79e;hb=244e8476c95d14a480be7160042b2b27b693ca63;hp=3e4c9172fbbf3b01202f6c9ccc5b2d4ff607fcc1;hpb=fa4298fa508e00759565c246aef423ba28fedf31;p=python_utils.git diff --git a/collect/trie.py b/collect/trie.py index 3e4c917..70d57b1 100644 --- a/collect/trie.py +++ b/collect/trie.py @@ -11,6 +11,7 @@ class Trie(object): for examples. """ + def __init__(self): self.root = {} self.end = "~END~" @@ -241,7 +242,14 @@ class Trie(object): return None return [x for x in node if x != self.end] - def repr_fancy(self, padding: str, pointer: str, parent: str, node: Any, has_sibling: bool): + def repr_fancy( + self, + padding: str, + pointer: str, + parent: str, + node: Any, + has_sibling: bool, + ): if node is None: return if node is not self.root: @@ -328,4 +336,5 @@ class Trie(object): if __name__ == '__main__': import doctest + doctest.testmod()