test lookup functions, removed testing library

Author: antonbaumann

go.mod

@@ -1,5 +1,3 @@
 module github.com/antonbaumann/german-go-stemmer
 
 go 1.13
-
-require github.com/stretchr/testify v1.3.0

lookup/lookup.go internal/lookup/lookup.go

@@ -0,0 +1,79 @@
+package lookup
+
+import "testing"
+
+func TestIsVowelPositive(t *testing.T) {
+	vowels := "aeiouyAOV"
+	for _, v := range vowels {
+		if !IsVowel(uint8(v)) {
+			t.Errorf("`%c` should be a vowel", v)
+		}
+	}
+}
+
+func TestIsVowelNegative(t *testing.T) {
+	consonants := "bcdfghjklmnpqrstvwxz"
+	for _, v := range consonants {
+		if IsVowel(uint8(v)) {
+			t.Errorf("`%c` should not be a vowel", v)
+		}
+	}
+}
+
+func TestIsStopWordPositive(t *testing.T) {
+	stopWords := []string {
+		"damit", "dann", "der", "derselbe", "manche",
+	}
+	for _, word := range stopWords {
+		if !IsStopWord(word) {
+			t.Errorf("`%s` should be a stop word", word)
+		}
+	}
+}
+
+func TestIsStopWordNegative(t *testing.T) {
+	stopWords := []string {
+		"weithin", "wachsam", "umfassender",
+	}
+	for _, word := range stopWords {
+		if IsStopWord(word) {
+			t.Errorf("`%s` should not be a stop word", word)
+		}
+	}
+}
+
+func TestIsSEndingPositive(t *testing.T) {
+	sEndings := "bdfghklmnrt"
+	for _, v := range sEndings {
+		if !IsSEnding(uint8(v)) {
+			t.Errorf("`%c` should be a s-ending", v)
+		}
+	}
+}
+
+func TestIsSEndingNegative(t *testing.T) {
+	sEndings := "aceijopqsuvwxyz"
+	for _, v := range sEndings {
+		if IsSEnding(uint8(v)) {
+			t.Errorf("`%c` should not be a s-ending", v)
+		}
+	}
+}
+
+func TestIsStEndingPositive(t *testing.T) {
+	sEndings := "bdfghklmnt"
+	for _, v := range sEndings {
+		if !IsStEnding(uint8(v)) {
+			t.Errorf("`%c` should be a st-ending", v)
+		}
+	}
+}
+
+func TestIsStEndingNegative(t *testing.T) {
+	sEndings := "aceijopqrsuvwxyz"
+	for _, v := range sEndings {
+		if IsStEnding(uint8(v)) {
+			t.Errorf("`%c` should not be a s-ending", v)
+		}
+	}
+}

internal/lookup/lookup_test.go

@@ -1,7 +1,7 @@
 package stemmer
 
 import (
-	"github.com/antonbaumann/german-go-stemmer/lookup"
+	"github.com/antonbaumann/german-go-stemmer/internal/lookup"
 	"regexp"
 	"strings"
 )

stemmer.go

@@ -2,8 +2,8 @@ package stemmer
 
 import (
 	"bufio"
-	"github.com/stretchr/testify/assert"
 	"os"
+	"strings"
 	"testing"
 )
 
@@ -14,8 +14,12 @@ type testSet struct {
 
 func TestStemQuery(t *testing.T) {
 	query := "wie wird das wetter  \t morgen in \nmünchen"
+	expected := "wett morg munch"
 	result := Stem(query)
-	assert.Equal(t, "wett morg munch", result)
+
+	if !strings.EqualFold(expected, result) {
+		t.Errorf("test failed.\nexpected:\t %v\ngot:\t\t %v", expected, result)
+	}
 }
 
 func TestStemWords(t *testing.T) {
@@ -36,7 +40,11 @@ func TestStemWords(t *testing.T) {
 		t.Fail()
 	}
 
-	assert.Equal(t, expected, StemWords(words))
+	result := StemWords(words)
+
+	if !slicesEqual(expected, result) {
+		t.Errorf("test failed.\nexpected:\t %v\ngot:\t\t %v", expected, result)
+	}
 }
 
 func TestStemWord(t *testing.T) {
@@ -64,7 +72,9 @@ func TestStemWord(t *testing.T) {
 
 	for _, test := range tests {
 		result := stem(test.value)
-		assert.Equal(t, test.expected, result)
+		if !strings.EqualFold(test.expected, result) {
+			t.Errorf("test failed.\nexpected:\t %v\ngot:\t\t %v", test.expected, result)
+		}
 	}
 }
 
@@ -78,7 +88,9 @@ func TestStep1(t *testing.T) {
 	for _, test := range tests {
 		r1, _ := getRegions(test.value)
 		result := step1(test.value, r1)
-		assert.Equal(t, test.expected, result)
+		if !strings.EqualFold(test.expected, result) {
+			t.Errorf("test failed.\nexpected:\t %v\ngot:\t\t %v", test.expected, result)
+		}
 	}
 }
 
@@ -89,7 +101,9 @@ func TestStep2(t *testing.T) {
 	for _, test := range tests {
 		r1, _ := getRegions(test.value)
 		result := step2(test.value, r1)
-		assert.Equal(t, test.expected, result)
+		if !strings.EqualFold(test.expected, result) {
+			t.Errorf("test failed.\nexpected:\t %v\ngot:\t\t %v", test.expected, result)
+		}
 	}
 }
 
@@ -109,3 +123,15 @@ func readWordList(filePath string) ([]string, error) {
 
 	return words, nil
 }
+
+func slicesEqual(a, b []string) bool {
+	if len(a) != len(b) {
+		return false
+	}
+	for i, v := range a {
+		if !strings.EqualFold(v, b[i]) {
+			return false
+		}
+	}
+	return true
+}