func TestMarks(tt *testing.T) {
t := trie.NewTrie()
trie.AddWord(t, "hello")
trie.AddWord(t, "goodbye")
trie.AddWord(t, "goodday")
m := trie.DescendPath(t, "hello")
m.Mark(10)
if m.GetMark() != 10 {
tt.Fail()
}
}
func TestReverseLookup(tt *testing.T) {
t := trie.NewTrie()
trie.AddWord(t, "hello")
trie.AddWord(t, "goodbye")
trie.AddWord(t, "goodday")
m := trie.DescendPath(t, "hello")
rev := trie.ReverseLookup(t, m)
if rev != "hello" {
tt.Fail()
}
}
func TestStartsWordDescend(tt *testing.T) {
t := trie.NewTrie()
trie.AddWord(t, "hello")
trie.AddWord(t, "goodbye")
trie.AddWord(t, "goodday")
if !t.StartsWord('h' - 'a') {
tt.Fail()
}
if !t.StartsWord('g' - 'a') {
tt.Fail()
}
if t.StartsWord('x' - 'a') {
tt.Fail()
}
th := t.Descend('h' - 'a')
if th == nil {
tt.Fail()
}
if th.IsWord() {
tt.Fail()
}
to := th.Descend('e' - 'a').Descend('l' - 'a').Descend('l' - 'a').Descend('o' - 'a')
if to == nil {
tt.Fail()
}
if !to.IsWord() {
tt.Fail()
}
t_hell := trie.DescendPath(t, "hell")
if t_hell == nil {
tt.Fail()
}
if t_hell.IsWord() {
tt.Fail()
}
// this test fails, oddly, even though fmt.Println(t_hell.Descend('h' - 'a')) prints '<nil>'
if t_hell.Descend('h'-'a') != nil {
tt.Fail()
}
t_hello := t_hell.Descend('o' - 'a')
if !t_hello.IsWord() {
tt.Fail()
}
}