Fix test function naming and the actual test.

GNUmakefile

@@ -1,6 +1,6 @@
 #
 .DEFAULT:	all
-.PHONY:		all clean
+.PHONY:		all clean test
 
 #
 DIST?=	dist/
@@ -18,3 +18,6 @@ dist/ir-tfidf:: cmd/ir-tfidf/* internal/**
 
 clean::
 	rm -rf "${DIST}"
+
+test::
+	go test ./...

cmd/ir-tfidf/main.go

@@ -11,6 +11,11 @@ import (
 )
 
 func main() {
+	if len(os.Args) < 2 {
+		fmt.Printf("You need to specify a keyword to search.\n")
+		os.Exit(1)
+	}
+
 	articles := artifact.Read("data/articles.json")
 
 	tokens := make(map[string][]string)
@@ -49,7 +54,7 @@ func main() {
 		}
 
 		if score > 0 {
-			fmt.Printf("Article %v: %v\n", article.Id, score)
+			fmt.Printf("Article %v (https://blog.gslin.org/?p=%v): %v\n", article.Id, article.Id, score)
 		}
 	}
 }

internal/ngram/ngram_test.go

@@ -7,6 +7,21 @@ import (
 	"github.com/gslin/go-ir-playground/internal/ngram"
 )
 
+func TestBigram(t *testing.T) {
+	a := ngram.Bigram("test")
+	assert.Equal(t, len(a), 0)
+
+	a = ngram.Bigram("測試")
+	assert.Equal(t, len(a), 1)
+	assert.Equal(t, a[0], "測試")
+
+	a = ngram.Bigram("中文測試")
+	assert.Equal(t, len(a), 3)
+	assert.Equal(t, a[0], "中文")
+	assert.Equal(t, a[1], "文測")
+	assert.Equal(t, a[2], "測試")
+}
+
 func TestUnigram(t *testing.T) {
 	a := ngram.Unigram("test")
 	assert.Equal(t, len(a), 1)

internal/tokenizer/tokenizer_test.go

@@ -0,0 +1,14 @@
+package tokenizer_test
+
+import (
+	"testing"
+	"github.com/stretchr/testify/assert"
+
+	"github.com/gslin/go-ir-playground/internal/tokenizer"
+)
+
+func TestTokenize(t *testing.T) {
+	a := tokenizer.Tokenize("test")
+	assert.Equal(t, len(a), 1)
+	assert.Equal(t, a[0], "test")
+}