-
Notifications
You must be signed in to change notification settings - Fork 0
/
levenstein_test.go
68 lines (56 loc) · 1.53 KB
/
levenstein_test.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
package levenstein
import "testing"
func BenchmarkLevensteinParallel(b *testing.B) {
left := "xxxx"
right := "xxxy"
b.RunParallel(func(pb *testing.PB) {
for pb.Next() {
Distance(left, right)
}
})
}
func BenchmarkLevensteinSerial(b *testing.B) {
left := "xxxx"
right := "xxxy"
for n := 0; n < b.N; n++ {
Distance(left, right)
}
}
func TestCalculations(t *testing.T) {
expectDistance := func(a string, b string, c int) {
d := Distance(a, b)
if d != c {
t.Errorf("a: `" + a + "`, b: `" + b + "` expected " + string(rune(c+48)) + " , got " + string(rune(d+48)))
}
}
// matches
expectDistance("", "", 0)
expectDistance("a", "a", 0)
expectDistance("abc", "abc", 0)
// empty on one side
expectDistance("a", "", 1)
expectDistance("", "a", 1)
expectDistance("abc", "", 3)
expectDistance("", "abc", 3)
// inserts only
expectDistance("a", "ab", 1)
expectDistance("b", "ba", 1)
expectDistance("ac", "abc", 1)
expectDistance("abcdefg", "xabxcdxxefxgx", 6)
// deletes only
expectDistance("ab", "a", 1)
expectDistance("ab", "b", 1)
expectDistance("abc", "ac", 1)
expectDistance("xabxcdxxefxgx", "abcdefg", 6)
// swaps only
expectDistance("a", "b", 1)
expectDistance("ab", "ac", 1)
expectDistance("ac", "bc", 1)
expectDistance("abc", "axc", 1)
expectDistance("xabxcdxxefxgx", "1ab2cd34ef5g6", 6)
// combination of insert + delete + swap
expectDistance("example", "samples", 3)
expectDistance("sturgeon", "urgently", 6)
expectDistance("levenshtein", "frankenstein", 6)
expectDistance("distance", "difference", 5)
}