lemmy_test.go

package main

import (
	"net/http"
	"strings"
	"testing"
)

var words = map[string]string{
	"et":        "et",
	"oratio":    "oratio",
	"conviciis": "convicium",
	"est":       "sum",
}

func TestLemmatizeWord(t *testing.T) {
	httpClient := &http.Client{}
	for base, lemmyd := range words {
		lemTest, _ := LemmatizeWord(httpClient, base)
		t.Logf("%s:%s:%s", base, lemTest, lemmyd)
		if lemTest != lemmyd {
			t.Error("Lemmatize Failed for word " + base)
		}
	}
}

func TestLemmatizeText(t *testing.T) {
	var in string
	var out []string
	mx := 10
	cache_size := 10000
	MAX_REQUESTS = &mx
	CACHE_SIZE = &cache_size
	for base, lemmyd := range words {
		in += base + " "
		out = append(out, lemmyd)
	}
	in = "//'\"`" + in
	t.Log(in)
	lr := NewLemmaReader(strings.NewReader(in))

	i := 0
	for w, done := lr.Read(); !done; w, done = lr.Read() {
		t.Logf("%s:%s", w, out[i])
		if w != out[i] {
			t.Errorf("Lemmatize Failed for word index %d", i)
		}
		i++
	}

	if i != len(out) {
		t.Error("Input and output counts do not match")
	}
}