From 55ad14e7905ad70eab1720ae7c4060f58065d9f7 Mon Sep 17 00:00:00 2001
From: Gea-Suan Lin <gslin@gslin.org>
Date: Fri, 16 Feb 2024 20:54:28 +0800
Subject: [PATCH] Add test cases for bigram.

---
 internal/ngram/ngram_test.go | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/internal/ngram/ngram_test.go b/internal/ngram/ngram_test.go
index a1953cd..d95b0b5 100644
--- a/internal/ngram/ngram_test.go
+++ b/internal/ngram/ngram_test.go
@@ -7,6 +7,16 @@ import (
 	"github.com/gslin/go-ir-playground/internal/ngram"
 )
 
+func testBigram(t *testing.T) {
+	a := ngram.Bigram("test")
+	assert.Equal(t, len(a), 1)
+	assert.Equal(t, a[0], "test")
+
+	a = ngram.Bigram("測試")
+	assert.Equal(t, len(a), 1)
+	assert.Equal(t, a[0], "測試")
+}
+
 func TestUnigram(t *testing.T) {
 	a := ngram.Unigram("test")
 	assert.Equal(t, len(a), 1)