def testCreateNode(self):
n = Node('leaf', token='dummy')
self.assertEqual(n.leafs, ['leaf'])
self.assertEqual(n.edges, {'token': 'dummy'})
def testCreateDictionary(self):
d = Dictionary([('key', 'The Term', 42, 21)], DictionaryTests.tokenizer)
n = Node(The=Node(Term=Node(([42, 21], 'key'))))
self.assertEqual(d.root, n)
def testSetLeaf(self):
n = Node((1, 'a'))
n.setLeaf('c', 3)
n.setLeaf('b', 2)
self.assertEqual(n.leafs, [(1, 'a'), (2, 'b'), (3, 'c')])
def testKey(self): n = Node((1, 'a')) self.assertEqual(n.key, 'a')
def testCreateOrGetNode(self):
n = Node(token='dummy')
self.assertEqual(n.createOrGet('token'), 'dummy')
