def test_flatten_words(self):
node = Node()
node.add_word(1, 0)
node.add_word(2, 1)
node['a'].add_word(3, 0)
node['a'].add_word(4, 1)
node['b'].add_word(5, 2)
node['b']['c'].add_word(6, 0)
words = node.flatten_words()
expected = [(1, 0), (2, 1), (4, 1), (3, 0), (5, 2), (6, 0)]
for e in expected:
self.assertTrue(e in words)
def test_add_word(self):
node = Node()
node.add_word(1, 0)
node.add_word(None, 2)
node.add_word(3, None)
node.add_word(4, -1)
self.assertTrue((1, 0) in node.words)
self.assertFalse((None, 2) in node.words)
self.assertFalse((3, None) in node.words)
self.assertFalse((4, -1) in node.words)
def test_add_children(self):
node = Node()
node.children['a'] = 1
node.children['b'] = 2
self.assertEqual(node['a'], 1)
self.assertEqual(node['b'], 2)
