def test_get_depth_tuples(self): tree = SuffixTree.from_seq(['a', 'a', 'b']) expect(tree.items()).to( equal( set(( (('a', '$'), 2), (('a', 'a', '$'), 1), (('a', 'a', 'b', '$'), 1), (('a', 'b', '$'), 1), (('b', '$'), 1), ))))
def test_single_suffix_tree(self): tree = SuffixTree.from_seq(['a']) expect(tree['a'].count).to(equal(1)) expect(tree['b'].count).to(equal(0)) expect(tree['a']['$'].count).to(equal(1))
def test_elements_are_reversed(self): tree = SuffixTree.from_seq(['a', 'a', 'b']) expect(tree['a']['b'].count).to(equal(1)) expect(tree['b']['a'].count).to(equal(0))
def test_two_elements_in_tree(self): tree = SuffixTree.from_seq(['a', 'a']) expect(tree['a'].count).to(equal(2)) expect(tree['a']['a'].count).to(equal(1))