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)