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')